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, HuggingFaceInferenceAPI
+from langchain_huggingface import HuggingFaceEndpoint
 from tools import TOOLS  
 
 # --- Read local QA data for retriever ---
@@ -17,8 +17,7 @@ def build_graph():
     # Initialize Mistral model
     llm = HuggingFaceEndpoint(
         # repo_id="mistralai/Mistral-7B-Instruct-v0.3",
-        llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct"),
-        # task="conver"
+        task="conversation",
         huggingfacehub_api_token=os.environ["HF_TOKEN"]
     )