try

sxjeru · Dec 12, 2024 · a43bbd4 · a43bbd4
1 parent b1af766
commit a43bbd4
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/libs/agent-runtime/google/index.ts b/src/libs/agent-runtime/google/index.ts
@@ -91,7 +91,13 @@ export class LobeGoogleAI implements LobeRuntimeAI {
         .generateContentStream({
           contents,
           systemInstruction: payload.system as string,
-          tools: this.buildGoogleTools(payload.tools),
+          tools: (() => {
+            const processedTools = this.buildGoogleTools(payload.tools);
+            if (model.startsWith('gemini-2.0')) {
+              return [...(processedTools || []), { googleSearch: {} }];
+            }
+            return processedTools;
+          })(),
         });
 
       const googleStream = convertIterableToStream(geminiStreamResult.stream);