Update agent.py

agent.py

@@ -221,8 +221,8 @@ def build_graph(provider: str = "huggingface"):
 
     elif provider == "google":
         # Google Gemini
-        #llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
-        llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
+        llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+        #llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
         
     else:
         raise ValueError("Invalid provider. Choose 'huggingface'.")