Final_Assignment_Template

Sleeping

sqfoo commited on May 15, 2025

Commit

b0ded06

verified ·

1 Parent(s): e3e0d7c

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -101,7 +101,7 @@ class BasicAgent:
         self.model = ChatGoogleGenerativeAI(
             model="gemini-2.0-flash",
             temperature=0,
-            max_tokens=1024,
             timeout=None,
             max_retries=2,
             google_api_key="AIzaSyAxVUPaGJIgdxB46ZR0RWPKSjB9a63Z80o",

         self.model = ChatGoogleGenerativeAI(
             model="gemini-2.0-flash",
             temperature=0,
+            max_tokens=128,
             timeout=None,
             max_retries=2,
             google_api_key="AIzaSyAxVUPaGJIgdxB46ZR0RWPKSjB9a63Z80o",