💄 style: Enable googleSearch Tool for gemini-2.0-flash-exp (#4997)

* Update index.ts * Update index.ts
lobehub · Dec 12, 2024 · cde0d1a · cde0d1a
1 parent a7eb800
commit cde0d1a
Showing 1 changed file with 6 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,12 @@ export class LobeGoogleAI implements LobeRuntimeAI {
         .generateContentStream({
           contents,
           systemInstruction: payload.system as string,
-          tools: this.buildGoogleTools(payload.tools),
+          tools: (() => {
+            if (!payload.tools && model.startsWith('gemini-2.0')) {
+              return [{ googleSearch: {} } as GoogleFunctionCallTool];
+            }
+            return this.buildGoogleTools(payload.tools);
+          })(),
         });
 
       const googleStream = convertIterableToStream(geminiStreamResult.stream);