Update agent.py

agent.py

@@ -916,15 +916,17 @@ def should_call_tool(state):
     if isinstance(last_msg, AIMessage) and last_msg.tool_calls:
         return "call_tool"
     return "end"
+    
 
 from typing import TypedDict, List, Optional, Union
 from langchain.schema import BaseMessage
 
 class AgentState(TypedDict):
-    messages: List[BaseMessage]   # chat history
-    input: str                    # original input
-    intent: str                   # derived or predicted intent
-    result: Optional[str]         # tool output, if any
+    messages: List[BaseMessage]    # Chat history
+    input: str                     # Original input
+    intent: str                    # Derived or predicted intent
+    result: Optional[str]          # Final or intermediate result
+
 
 
 
@@ -1133,11 +1135,6 @@ def build_graph() -> StateGraph:
     call_llm = llm.bind_tools(tools)
 
     # --- Build the graph ---
-from langgraph.graph import StateGraph
-
-def build_graph():
-    builder = StateGraph(AgentState)  # Keep your typed state graph
-
     builder.add_node("retriever", retriever)
     builder.add_node("assistant", assistant)
     builder.add_node("call_llm", call_llm)