Update agent.py

agent.py

@@ -144,11 +144,11 @@ def build_graph():
     # First create the HuggingFaceEndpoint
     llm_endpoint = HuggingFaceEndpoint(
         # repo_id="Qwen/Qwen2.5-14B-Instruct",
-        # repo_id="HuggingFaceH4/mistral-7b-anthropic",
-        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        repo_id="HuggingFaceH4/mistral-7b-anthropic",
+        # repo_id="Qwen/Qwen2.5-Coder-32B-Instruct", # this one is poor
         huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
         temperature=0.1,         # Maximum determinism
-        max_new_tokens=128,      # Even more restrictive
+        max_new_tokens=512,      # Even more restrictive with 128
         timeout=90,              # Moderate timeout
         do_sample=False,         # Completely deterministic
     )