Update agent.py

agent.py

@@ -135,6 +135,7 @@ tools = [
     web_search_tool,
     arvix_search_tool,
 ]
+chat_with_tools = chat.bind_tools(tools)
 
 # Build graph function
 def build_graph(provider: str = "huggingface"):
@@ -167,7 +168,7 @@ def build_graph(provider: str = "huggingface"):
     # Node
     def assistant(state: MessagesState):
         """Assistant node"""
-        return {"messages": [llm_with_tools.invoke(state["messages"])]}
+        return {"messages": [chat_with_tools.invoke(state["messages"])]}
     
     # def retriever(state: MessagesState):
       #  """Retriever node"""