feat: Support newer Gemini models (#188)

2025-05-06 10:04:30 +02:00
parent 408ee1ba7d
commit 7d88283f8a
1 changed files with 1 additions and 1 deletions
--- a/mm_agents/agent.py
+++ b/mm_agents/agent.py
@@ -837,7 +837,7 @@ class PromptAgent:
            )
            return response.text

-        elif self.model == "gemini-1.5-pro-latest":
+        elif self.model.startswith("gemini"):
            messages = payload["messages"]
            max_tokens = payload["max_tokens"]
            top_p = payload["top_p"]