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

* Update index.ts * Update index.ts
2026-06-15 12:10:16 +00:00 · 2024-12-12 22:39:01 +08:00
parent a7eb8008ea
commit cde0d1a6c1
1 changed files with 6 additions and 1 deletions
@@ -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);