Final_Assignment_Agent_vtony

Sleeping

vtony commited on Jun 30

Commit

fd0075c

verified ·

1 Parent(s): 1352b8d

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -113,9 +113,9 @@ tools = [
 # --- Graph Builder ---
 def build_graph():
-    # Initialize model (Gemini 2.0 Flash)
     llm = ChatGoogleGenerativeAI(
-        model="gemini-2.0-flash-exp",
         temperature=0.3,
         google_api_key=google_api_key,
         max_retries=3

 # --- Graph Builder ---
 def build_graph():
+    # Initialize model (Gemini 2.5 Flash)
     llm = ChatGoogleGenerativeAI(
+        model="gemini-2.5-flash",
         temperature=0.3,
         google_api_key=google_api_key,
         max_retries=3