Update agent.py

agent.py

@@ -91,10 +91,12 @@ def build_graph():
 
     # First create the HuggingFaceEndpoint
     llm_endpoint = HuggingFaceEndpoint(
-        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+        repo_id="gemini/gemini-2.0-flash",
+        #"Qwen/Qwen2.5-Coder-32B-Instruct",
+        #huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
+        api_key=os.getenv("GEMINI_API_KEY"),
         temperature=0.3,
-        max_new_tokens=2048,
+        max_new_tokens=512,
         timeout=60,
     )