fix-tests

Signed-off-by: yxia216 <[email protected]>

Author: hustxiayang

internal/translator/openai_gcpvertexai.go

@@ -361,11 +361,6 @@ func (o *openAIToGCPVertexAITranslatorV1ChatCompletion) openAIMessageToGeminiMes
 	if err != nil {
 		return nil, fmt.Errorf("error converting tool choice: %w", err)
 	}
-	if openAIReq.EnterpriseWebSearch {
-		tools = append(tools, genai.Tool{
-			EnterpriseWebSearch: &genai.EnterpriseWebSearch{},
-		})
-	}
 
 	// Convert generation config.
 	generationConfig, responseMode, err := openAIReqToGeminiGenerationConfig(openAIReq, requestModel)
@@ -411,6 +406,12 @@ func (o *openAIToGCPVertexAITranslatorV1ChatCompletion) applyVendorSpecificField
 	if gcpVendorFields.SafetySettings != nil {
 		gcr.SafetySettings = gcpVendorFields.SafetySettings
 	}
+
+	if gcpVendorFields.EnterpriseWebSearch {
+		gcr.Tools = append(gcr.Tools, genai.Tool{
+			EnterpriseWebSearch: &genai.EnterpriseWebSearch{},
+		})
+	}
 }
 
 func (o *openAIToGCPVertexAITranslatorV1ChatCompletion) geminiResponseToOpenAIMessage(gcr genai.GenerateContentResponse, responseModel string) (*openai.ChatCompletionResponse, error) {

internal/translator/openai_gcpvertexai_test.go

@@ -307,13 +307,13 @@ func TestOpenAIToGCPVertexAITranslatorV1ChatCompletion_RequestBody(t *testing.T)
     ],
     "tools": [
         {
-			"enterpriseWebSearch": {},
+            "enterpriseWebSearch": {}
         }
     ],
     "generation_config": {
         "maxOutputTokens": 1024,
         "temperature": 0.7
-    },
+    }
 }`)
 
 	tests := []struct {
@@ -707,7 +707,7 @@ func TestOpenAIToGCPVertexAITranslatorV1ChatCompletion_RequestBody(t *testing.T)
 					{
 						Header: &corev3.HeaderValue{
 							Key:      "Content-Length",
-							RawValue: []byte("395"),
+							RawValue: []byte("190"),
 						},
 					},
 				},