Test_Magus

Running

SergeyO7 commited on 8 days ago

Commit

bc906fa

verified ·

1 Parent(s): 6f3253b

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -9,7 +9,7 @@ class MagAgent:
         """Initialize the MagAgent with search tools."""
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
-            model_id="gemini/gemini-2.0-flash",
             api_key= os.environ.get("GEMINI_KEY"),
             max_tokens=8192
         )

         """Initialize the MagAgent with search tools."""
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
+            model_id="gemini/gemini-2.0-flash-lite",
             api_key= os.environ.get("GEMINI_KEY"),
             max_tokens=8192
         )