From 3a7ba5725cacc63cb45d1aa8399f38f2374443c6 Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Wed, 18 Jun 2025 00:37:22 +0800
Subject: [PATCH] fix(relay): ensure consistent setting of web search context
 size in TextHelper function

---
 relay/relay-text.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/relay/relay-text.go b/relay/relay-text.go
index c94e0f50..24fb8155 100644
--- a/relay/relay-text.go
+++ b/relay/relay-text.go
@@ -90,15 +90,16 @@ func TextHelper(c *gin.Context) (openaiErr *dto.OpenAIErrorWithStatusCode) {
 
 	// get & validate textRequest 获取并验证文本请求
 	textRequest, err := getAndValidateTextRequest(c, relayInfo)
-	if textRequest.WebSearchOptions != nil {
-		c.Set("chat_completion_web_search_context_size", textRequest.WebSearchOptions.SearchContextSize)
-	}
 
 	if err != nil {
 		common.LogError(c, fmt.Sprintf("getAndValidateTextRequest failed: %s", err.Error()))
 		return service.OpenAIErrorWrapperLocal(err, "invalid_text_request", http.StatusBadRequest)
 	}
 
+	if textRequest.WebSearchOptions != nil {
+		c.Set("chat_completion_web_search_context_size", textRequest.WebSearchOptions.SearchContextSize)
+	}
+
 	if setting.ShouldCheckPromptSensitive() {
 		words, err := checkRequestSensitive(textRequest, relayInfo)
 		if err != nil {