Backport: fix: vertex google tool errors

.changeset/fuzzy-cherries-fail.md

@@ -0,0 +1,6 @@
+---
+'@ai-sdk/google-vertex': patch
+'@ai-sdk/google': patch
+---
+
+Fix adding google search along with url context in vertex ai

packages/google-vertex/src/google-vertex-embedding-model.test.ts

@@ -18,7 +18,7 @@ const dummyEmbeddings = [
 const testValues = ['test text one', 'test text two'];
 
 const DEFAULT_URL =
-  'https://us-central1-aiplatform.googleapis.com/v1/projects/test-project/locations/us-central1/publishers/google/models/textembedding-gecko@001:predict';
+  'https://us-central1-aiplatform.googleapis.com/v1beta1/projects/test-project/locations/us-central1/publishers/google/models/textembedding-gecko@001:predict';
 
 const CUSTOM_URL =
   'https://custom-endpoint.com/models/textembedding-gecko@001:predict';
@@ -43,7 +43,7 @@ describe('GoogleVertexEmbeddingModel', () => {
     project: 'test-project',
     headers: () => ({}),
     baseURL:
-      'https://us-central1-aiplatform.googleapis.com/v1/projects/test-project/locations/us-central1/publishers/google',
+      'https://us-central1-aiplatform.googleapis.com/v1beta1/projects/test-project/locations/us-central1/publishers/google',
   };
 
   const model = new GoogleVertexEmbeddingModel(mockModelId, mockConfig);

packages/google-vertex/src/google-vertex-provider.test.ts

@@ -45,7 +45,7 @@ describe('google-vertex-provider', () => {
       expect.objectContaining({
         provider: 'google.vertex.chat',
         baseURL:
-          'https://test-location-aiplatform.googleapis.com/v1/projects/test-project/locations/test-location/publishers/google',
+          'https://test-location-aiplatform.googleapis.com/v1beta1/projects/test-project/locations/test-location/publishers/google',
         headers: expect.any(Function),
         generateId: expect.any(Function),
       }),
@@ -73,7 +73,7 @@ describe('google-vertex-provider', () => {
         provider: 'google.vertex.embedding',
         headers: expect.any(Function),
         baseURL:
-          'https://test-location-aiplatform.googleapis.com/v1/projects/test-project/locations/test-location/publishers/google',
+          'https://test-location-aiplatform.googleapis.com/v1beta1/projects/test-project/locations/test-location/publishers/google',
       }),
     );
   });
@@ -154,7 +154,7 @@ describe('google-vertex-provider', () => {
       expect.objectContaining({
         provider: 'google.vertex.image',
         baseURL:
-          'https://test-location-aiplatform.googleapis.com/v1/projects/test-project/locations/test-location/publishers/google',
+          'https://test-location-aiplatform.googleapis.com/v1beta1/projects/test-project/locations/test-location/publishers/google',
         headers: expect.any(Function),
       }),
     );
@@ -172,7 +172,7 @@ describe('google-vertex-provider', () => {
       expect.objectContaining({
         provider: 'google.vertex.chat',
         baseURL:
-          'https://aiplatform.googleapis.com/v1/projects/test-project/locations/global/publishers/google',
+          'https://aiplatform.googleapis.com/v1beta1/projects/test-project/locations/global/publishers/google',
         headers: expect.any(Function),
         generateId: expect.any(Function),
       }),
@@ -192,7 +192,7 @@ describe('google-vertex-provider', () => {
         provider: 'google.vertex.embedding',
         headers: expect.any(Function),
         baseURL:
-          'https://aiplatform.googleapis.com/v1/projects/test-project/locations/global/publishers/google',
+          'https://aiplatform.googleapis.com/v1beta1/projects/test-project/locations/global/publishers/google',
       }),
     );
   });
@@ -209,7 +209,7 @@ describe('google-vertex-provider', () => {
       expect.objectContaining({
         provider: 'google.vertex.image',
         baseURL:
-          'https://aiplatform.googleapis.com/v1/projects/test-project/locations/global/publishers/google',
+          'https://aiplatform.googleapis.com/v1beta1/projects/test-project/locations/global/publishers/google',
         headers: expect.any(Function),
       }),
     );
@@ -239,7 +239,7 @@ describe('google-vertex-provider', () => {
       expect.objectContaining({
         provider: 'google.vertex.chat',
         baseURL:
-          'https://us-central1-aiplatform.googleapis.com/v1/projects/test-project/locations/us-central1/publishers/google',
+          'https://us-central1-aiplatform.googleapis.com/v1beta1/projects/test-project/locations/us-central1/publishers/google',
         headers: expect.any(Function),
         generateId: expect.any(Function),
       }),

packages/google-vertex/src/google-vertex-provider.ts

@@ -106,7 +106,7 @@ export function createVertex(
 
     return (
       withoutTrailingSlash(options.baseURL) ??
-      `https://${baseHost}/v1/projects/${project}/locations/${region}/publishers/google`
+      `https://${baseHost}/v1beta1/projects/${project}/locations/${region}/publishers/google`
     );
   };
 

packages/google/src/google-generative-ai-language-model.test.ts

@@ -1071,7 +1071,7 @@ describe('doGenerate', () => {
     });
 
     const requestBody = await server.calls[0].requestBodyJson;
-    expect(requestBody.tools).toEqual({ codeExecution: {} });
+    expect(requestBody.tools).toEqual([{ codeExecution: {} }]);
 
     expect(content).toEqual([
       {
@@ -1119,7 +1119,7 @@ describe('doGenerate', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { googleSearch: {} },
+        tools: [{ googleSearch: {} }],
       });
     });
 
@@ -1142,7 +1142,7 @@ describe('doGenerate', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { googleSearch: {} },
+        tools: [{ googleSearch: {} }],
       });
     });
 
@@ -1165,7 +1165,7 @@ describe('doGenerate', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { googleSearchRetrieval: {} },
+        tools: [{ googleSearchRetrieval: {} }],
       });
     });
 
@@ -1192,14 +1192,16 @@ describe('doGenerate', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: {
-          googleSearchRetrieval: {
-            dynamicRetrievalConfig: {
-              mode: 'MODE_DYNAMIC',
-              dynamicThreshold: 1,
+        tools: [
+          {
+            googleSearchRetrieval: {
+              dynamicRetrievalConfig: {
+                mode: 'MODE_DYNAMIC',
+                dynamicThreshold: 1,
+              },
             },
           },
-        },
+        ],
       });
     });
     it('should use urlContextTool for gemini-2.0-pro', async () => {
@@ -1221,7 +1223,7 @@ describe('doGenerate', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { urlContext: {} },
+        tools: [{ urlContext: {} }],
       });
     });
   });
@@ -2170,7 +2172,7 @@ describe('doStream', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { googleSearch: {} },
+        tools: [{ googleSearch: {} }],
       });
     });
 
@@ -2196,7 +2198,7 @@ describe('doStream', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { googleSearch: {} },
+        tools: [{ googleSearch: {} }],
       });
     });
 
@@ -2221,7 +2223,7 @@ describe('doStream', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: { googleSearchRetrieval: {} },
+        tools: [{ googleSearchRetrieval: {} }],
       });
     });
 
@@ -2250,14 +2252,16 @@ describe('doStream', () => {
       });
 
       expect(await server.calls[0].requestBodyJson).toMatchObject({
-        tools: {
-          googleSearchRetrieval: {
-            dynamicRetrievalConfig: {
-              mode: 'MODE_DYNAMIC',
-              dynamicThreshold: 1,
+        tools: [
+          {
+            googleSearchRetrieval: {
+              dynamicRetrievalConfig: {
+                mode: 'MODE_DYNAMIC',
+                dynamicThreshold: 1,
+              },
             },
           },
-        },
+        ],
       });
     });
   });

packages/google/src/google-prepare-tools.test.ts

@@ -47,7 +47,7 @@ it('should correctly prepare function tools', () => {
   expect(result.toolWarnings).toEqual([]);
 });
 
-it('should correctly prepare provider-defined tools', () => {
+it('should correctly prepare provider-defined tools as array', () => {
   const result = prepareTools({
     tools: [
       {
@@ -65,10 +65,24 @@ it('should correctly prepare provider-defined tools', () => {
     ],
     modelId: 'gemini-2.5-flash',
   });
-  expect(result.tools).toEqual({
-    googleSearch: {},
-    urlContext: {},
+  expect(result.tools).toEqual([{ googleSearch: {} }, { urlContext: {} }]);
+  expect(result.toolConfig).toBeUndefined();
+  expect(result.toolWarnings).toEqual([]);
+});
+
+it('should correctly prepare single provider-defined tool', () => {
+  const result = prepareTools({
+    tools: [
+      {
+        type: 'provider-defined',
+        id: 'google.google_search',
+        name: 'google_search',
+        args: {},
+      },
+    ],
+    modelId: 'gemini-2.5-flash',
   });
+  expect(result.tools).toEqual([{ googleSearch: {} }]);
   expect(result.toolConfig).toBeUndefined();
   expect(result.toolWarnings).toEqual([]);
 });
@@ -205,10 +219,8 @@ it('should warn when mixing function and provider-defined tools', () => {
     modelId: 'gemini-2.5-flash',
   });
 
-  // Should only include provider-defined tools
-  expect(result.tools).toEqual({
-    googleSearch: {},
-  });
+  // Should only include provider-defined tools as array
+  expect(result.tools).toEqual([{ googleSearch: {} }]);
 
   // Should have warning about mixed tool types
   expect(result.toolWarnings).toEqual([
@@ -248,10 +260,8 @@ it('should handle tool choice with mixed tools (provider-defined tools only)', (
     modelId: 'gemini-2.5-flash',
   });
 
-  // Should only include provider-defined tools
-  expect(result.tools).toEqual({
-    googleSearch: {},
-  });
+  // Should only include provider-defined tools as array
+  expect(result.tools).toEqual([{ googleSearch: {} }]);
 
   // Should apply tool choice to provider-defined tools
   expect(result.toolConfig).toEqual(undefined);
@@ -271,3 +281,54 @@ it('should handle tool choice with mixed tools (provider-defined tools only)', (
     },
   ]);
 });
+
+it('should handle latest modelId for provider-defined tools correctly', () => {
+  const result = prepareTools({
+    tools: [
+      {
+        type: 'provider-defined',
+        id: 'google.google_search',
+        name: 'google_search',
+        args: {},
+      },
+    ],
+    modelId: 'gemini-flash-latest',
+  });
+  expect(result.tools).toEqual([{ googleSearchRetrieval: {} }]);
+  expect(result.toolConfig).toBeUndefined();
+  expect(result.toolWarnings).toEqual([]);
+});
+
+it('should handle code execution tool', () => {
+  const result = prepareTools({
+    tools: [
+      {
+        type: 'provider-defined',
+        id: 'google.code_execution',
+        name: 'code_execution',
+        args: {},
+      },
+    ],
+    modelId: 'gemini-2.5-flash',
+  });
+  expect(result.tools).toEqual([{ codeExecution: {} }]);
+  expect(result.toolConfig).toBeUndefined();
+  expect(result.toolWarnings).toEqual([]);
+});
+
+it('should handle url context tool alone', () => {
+  const result = prepareTools({
+    tools: [
+      {
+        type: 'provider-defined',
+        id: 'google.url_context',
+        name: 'url_context',
+        args: {},
+      },
+    ],
+    modelId: 'gemini-2.5-flash',
+  });
+  expect(result.tools).toEqual([{ urlContext: {} }]);
+  expect(result.toolConfig).toBeUndefined();
+  expect(result.toolWarnings).toEqual([]);
+});