Test_Magus

Running

SergeyO7 commited on 14 days ago

Commit

6a0cd84

verified ·

1 Parent(s): 705dc8a

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -61,7 +61,7 @@ def ImageAnalysisTool(question: str) -> str:
         return f"Error fetching image: {e}"
     model = LiteLLMModel(
-        model_id="gemini/gemini-2.0-flash",
         api_key=os.environ.get("GEMINI_KEY"),
         max_tokens=8192
     )

         return f"Error fetching image: {e}"
     model = LiteLLMModel(
+        model_id="gemini/gemini-2.5-pro",
         api_key=os.environ.get("GEMINI_KEY"),
         max_tokens=8192
     )