Update agent.py

agent.py

@@ -12,12 +12,12 @@ qa_dict = {
     for _, row in qa_pairs.iterrows()
 }
 
-def build_graph():
-    """Construct a LangGraph agent with retriever and fallback LLM."""
+from langchain_huggingface import HuggingFaceEndpoint
 
+def build_graph():
     llm = HuggingFaceEndpoint(
-        repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-        task="chat-completion",  
+        repo_id="Qwen/Qwen2.5-32B-Instruct",
+        task="text-generation",
         huggingfacehub_api_token=os.environ["HF_TOKEN"]
     )
 
@@ -30,9 +30,9 @@ def build_graph():
         return {"messages": state["messages"]}
 
     def assistant_node(state: MessagesState):
-        # <--- Use invoke with list of messages to get correct "chat" format
-        result = llm.invoke(state["messages"])
-        return {"messages": [result]}
+        query = state["messages"][-1].content.strip()
+        response = llm.invoke(query)
+        return {"messages": [AIMessage(content=response)]}
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever_node)
@@ -44,6 +44,7 @@ def build_graph():
 
     return builder.compile()
 
+
 # --- Agent class wrapper for app.py ---
 class BasicAgent:
     def __init__(self):