Update agent.py

agent.py

@@ -2,7 +2,7 @@ import os
 import pandas as pd
 from langchain_core.messages import HumanMessage, AIMessage
 from langgraph.graph import StateGraph, MessagesState
-from langchain_huggingface import HuggingFaceEndpoint
+from langchain_huggingface import HuggingFaceEndpoint, HuggingFaceInferenceAPI
 from tools import TOOLS  
 
 # --- Read local QA data for retriever ---
@@ -16,7 +16,9 @@ qa_dict = {
 def build_graph():
     # Initialize Mistral model
     llm = HuggingFaceEndpoint(
-        repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+        # repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+        llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct"),
+        # task="conver"
         huggingfacehub_api_token=os.environ["HF_TOKEN"]
     )