langchain versions

app.py

@@ -9,7 +9,6 @@ from typing import IO, Dict
 from io import BytesIO
 from langchain_core.messages import HumanMessage, SystemMessage
 from langgraph.graph import StateGraph
-from langgraph.prebuilt import ToolNode
 import base64
 from google.ai.generativelanguage_v1beta.types import Tool as GenAITool
 from google import genai
@@ -201,7 +200,7 @@ class BasicAgent:
 
         # Define nodes: these do the work
         self.builder.add_node("assistant", self.assistant)
-        self.builder.add_node("tools", ToolNode(self.tools))
+        self.builder.add_node("tools", self.tools_node)
 
         # Define edges: these determine how the control flow moves
         self.builder.add_edge("start", "assistant")
@@ -214,6 +213,14 @@ class BasicAgent:
     def assistant(self, state):
         return {"messages": [self.agent.invoke([self.sys_msg] + state["messages"])]}
     
+    def tools_node(self, state):
+        # Execute the tool and return the result
+        tool_name = state["messages"][-1].content
+        for tool in self.tools:
+            if tool.__name__ == tool_name:
+                return {"messages": [tool(*state["args"])]}
+        return {"messages": [f"Tool {tool_name} not found"]}
+    
     async def __call__(self, question: str, task_id: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = "This is a default answer."