Update agent.py

agent.py

@@ -23,10 +23,16 @@ def build_graph():
     # llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
     # llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     # llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
+    # llm = HuggingFaceEndpoint(
+    #     url="https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3",
+    #     huggingfacehub_api_token=os.environ["HF_TOKEN"]
+    # )
     llm = HuggingFaceEndpoint(
-        url="https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3",
-        huggingfacehub_api_token=os.environ["HF_TOKEN"]
+        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        huggingfacehub_api_token=HF_TOKEN,
     )
+
+    
     llm_with_tools = llm.bind_tools(TOOLS)
 
     # Step 1: Retriever node