Merge pull request Portkey-AI#923 from narengogi/vertex/google-search

add support for google search tool
Create-Inc · Feb 8, 2025 · 79e2dcb · 79e2dcb
2 parents 1d31738 + 73acd67
commit 79e2dcb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/providers/google-vertex-ai/chatComplete.ts b/src/providers/google-vertex-ai/chatComplete.ts
@@ -287,7 +287,13 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
           recursivelyDeleteUnsupportedParameters(tool.function?.parameters);
           delete tool.function?.strict;
 
-          if (tool.function.name === 'googleSearchRetrieval') {
+          if (['googleSearch', 'google_search'].includes(tool.function.name)) {
+            tools.push({ googleSearch: {} });
+          } else if (
+            ['googleSearchRetrieval', 'google_search_retrieval'].includes(
+              tool.function.name
+            )
+          ) {
             tools.push(buildGoogleSearchRetrievalTool(tool));
           } else {
             functionDeclarations.push(tool.function);

diff --git a/src/providers/google/chatComplete.ts b/src/providers/google/chatComplete.ts
@@ -359,7 +359,13 @@ export const GoogleChatCompleteConfig: ProviderConfig = {
           recursivelyDeleteUnsupportedParameters(tool.function?.parameters);
           delete tool.function?.strict;
 
-          if (tool.function.name === 'googleSearchRetrieval') {
+          if (['googleSearch', 'google_search'].includes(tool.function.name)) {
+            tools.push({ googleSearch: {} });
+          } else if (
+            ['googleSearchRetrieval', 'google_search_retrieval'].includes(
+              tool.function.name
+            )
+          ) {
             tools.push(buildGoogleSearchRetrievalTool(tool));
           } else {
             functionDeclarations.push(tool.function);