fix: gemini tool optional

ikechan8370 · Nov 18, 2024 · 30f9c82 · 30f9c82
1 parent 7c2961c
commit 30f9c82
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/client/CustomGoogleGeminiClient.js b/client/CustomGoogleGeminiClient.js
@@ -163,13 +163,20 @@ export class CustomGoogleGeminiClient extends GoogleGeminiClient {
         temperature: opt.temperature || 0.9,
         topP: opt.topP || 0.95,
         topK: opt.tokK || 16
-      },
-      tools: [
+      }
+    }
+    if (this.tools?.length > 0) {
+      body.tools = [
         {
-          functionDeclarations: this.tools.map(tool => tool.function())
+          function_declarations: this.tools.map(tool => tool.function())
           // codeExecution: {}
         }
       ]
+      body.tool_config = {
+        function_calling_config: {
+          mode: 'ANY'
+        }
+      }
     }
     if (opt.image) {
       delete body.tools