debug

app.py

@@ -13,7 +13,7 @@ import base64
 from google.ai.generativelanguage_v1beta.types import Tool as GenAITool
 import google.generativeai as genai
 import operator
-from langgraph.prebuilt import ToolExecutor
+from langgraph.prebuilt import ToolNode
 from langchain_core.tools import tool
 from utilities import get_file
 import time
@@ -165,7 +165,7 @@ class BasicAgent:
             get_file, analyse_excel, add_numbers, transcribe_audio,
             python_code, open_image, open_youtube_video, google_search
         ]
-        self.tool_executor = ToolExecutor(self.tools)
+        self.tool_node = ToolNode(self.tools)
         
         # System message
         self.sys_msg = SystemMessage('''You are a general AI assistant. I will ask you a question. Follow these steps:
@@ -194,7 +194,8 @@ class BasicAgent:
         - Never make up information - if you can't find it, say so''')
 
         # Create the graph
-        self.workflow = StateGraph(AgentState, config={"recursion_limit": 50})
+        self.workflow = StateGraph(AgentState)
+        self.workflow.set_config({"recursion_limit": 50})
         
         # Add nodes
         self.workflow.add_node("agent", self.call_model)
@@ -321,7 +322,7 @@ class BasicAgent:
                             logger.info(f"Executing tool: {tool_name}")
                             logger.info(f"Tool arguments: {json.dumps(tool_args, indent=2)}")
                             
-                            result = self.tool_executor.invoke(tool_name, tool_args)
+                            result = self.tool_node.invoke({"name": tool_name, "args": tool_args})
                             logger.info(f"Tool result: {result}")
                             
                             messages.append(AIMessage(content=f"Tool result: {result}"))