fix(js/plugins/googleai): work around googleai forced tool calling in…

…finite loop
firebase · Dec 13, 2024 · 863549a · 863549a
1 parent 7893efd
commit 863549a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/js/plugins/googleai/src/gemini.ts b/js/plugins/googleai/src/gemini.ts
@@ -666,7 +666,12 @@ export function defineGoogleAIModel(
       }
 
       let toolConfig: ToolConfig | undefined;
-      if (requestConfig.functionCallingConfig) {
+      if (
+        requestConfig.functionCallingConfig &&
+        // This is a workround for issue: https://github.com/firebase/genkit/issues/1520
+        // TODO: remove this when the issue is resolved upstream in the Gemini API
+        !messages[messages.length - 1].content.find((c) => c.toolResponse)
+      ) {
         toolConfig = {
           functionCallingConfig: {
             allowedFunctionNames: