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

sxjeru · web-flow · commit cde0d1a6c13b · 2024-12-12T22:39:01.000+08:00
* Update index.ts

* Update index.ts
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);
