fix-test

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

Author: hustxiayang

internal/translator/openai_gcpvertexai.go

@@ -361,7 +361,7 @@ func (o *openAIToGCPVertexAITranslatorV1ChatCompletion) openAIMessageToGeminiMes
 	if err != nil {
 		return nil, fmt.Errorf("error converting tool choice: %w", err)
 	}
-	if openAIReq.GCPVertexAIVendorFields.EnterpriseWebSearch {
+	if openAIReq.EnterpriseWebSearch {
 		tools = append(tools, genai.Tool{
 			EnterpriseWebSearch: &genai.EnterpriseWebSearch{},
 		})

internal/translator/openai_gcpvertexai_test.go

@@ -294,6 +294,28 @@ func TestOpenAIToGCPVertexAITranslatorV1ChatCompletion_RequestBody(t *testing.T)
   }
 }`)
 
+	wantBdyWithEnterpriseWebSearch := []byte(`{
+    "contents": [
+        {
+            "parts": [
+                {
+                    "text": "Test with web grounding for enterprise"
+                }
+            ],
+            "role": "user"
+        }
+    ],
+    "tools": [
+        {
+			"enterpriseWebSearch": {},
+        }
+    ],
+    "generation_config": {
+        "maxOutputTokens": 1024,
+        "temperature": 0.7
+    },
+}`)
+
 	tests := []struct {
 		name              string
 		modelNameOverride internalapi.ModelNameOverride
@@ -654,6 +676,48 @@ func TestOpenAIToGCPVertexAITranslatorV1ChatCompletion_RequestBody(t *testing.T)
 			},
 			wantBody: wantBdyWithGuidedRegex,
 		},
+		{
+			name: "Request with gcp web grounding for enterprise",
+			input: openai.ChatCompletionRequest{
+				Model:       "gemini-1.5-pro",
+				Temperature: ptr.To(0.7),
+				MaxTokens:   ptr.To(int64(1024)),
+				Messages: []openai.ChatCompletionMessageParamUnion{
+					{
+						OfUser: &openai.ChatCompletionUserMessageParam{
+							Role:    openai.ChatMessageRoleUser,
+							Content: openai.StringOrUserRoleContentUnion{Value: "Test with web grounding for enterprise"},
+						},
+					},
+				},
+				GCPVertexAIVendorFields: &openai.GCPVertexAIVendorFields{
+					EnterpriseWebSearch: true,
+				},
+			},
+			onRetry:   false,
+			wantError: false,
+			wantHeaderMut: &extprocv3.HeaderMutation{
+				SetHeaders: []*corev3.HeaderValueOption{
+					{
+						Header: &corev3.HeaderValue{
+							Key:      ":path",
+							RawValue: []byte("publishers/google/models/gemini-1.5-pro:generateContent"),
+						},
+					},
+					{
+						Header: &corev3.HeaderValue{
+							Key:      "Content-Length",
+							RawValue: []byte("395"),
+						},
+					},
+				},
+			},
+			wantBody: &extprocv3.BodyMutation{
+				Mutation: &extprocv3.BodyMutation_Body{
+					Body: wantBdyWithEnterpriseWebSearch,
+				},
+			},
+		},
 	}
 
 	for _, tc := range tests {