requirements update for llm compat

app.py

@@ -5,7 +5,7 @@ import inspect
 import pandas as pd
 import asyncio
 from langchain_google_genai import ChatGoogleGenerativeAI
-from typing import IO, Dict, TypedDict, Annotated, Sequence
+from typing import IO, Dict, TypedDict, Annotated, Sequence, Any, Callable
 from io import BytesIO
 from langchain_core.messages import HumanMessage, SystemMessage, BaseMessage, AIMessage
 from langgraph.graph import StateGraph, END
@@ -13,7 +13,6 @@ import base64
 from google.ai.generativelanguage_v1beta.types import Tool as GenAITool
 import google.generativeai as genai
 import operator
-from langgraph.prebuilt.tool_node import ToolNode
 from langchain_core.tools import tool
 from utilities import get_file
 import time
@@ -36,6 +35,13 @@ class AgentState(TypedDict):
     messages: Annotated[Sequence[BaseMessage], operator.add]
     next: str
 
+# Tool execution function
+def execute_tool(tool_name: str, tool_args: Dict[str, Any], tools: Dict[str, Callable]) -> Any:
+    """Execute a tool with the given arguments."""
+    if tool_name not in tools:
+        raise ValueError(f"Tool {tool_name} not found")
+    return tools[tool_name](**tool_args)
+
 # Convert existing functions to tools
 @tool
 def analyse_excel(task_id: str) -> Dict[str, float]:
@@ -221,11 +227,16 @@ class BasicAgent:
             )
         
         # Create tool executor
-        self.tools = [
-            get_file, analyse_excel, add_numbers, transcribe_audio,
-            python_code, open_image, open_youtube_video, google_search
-        ]
-        self.tool_node = ToolNode(self.tools)
+        self.tools = {
+            "get_file": get_file,
+            "analyse_excel": analyse_excel,
+            "add_numbers": add_numbers,
+            "transcribe_audio": transcribe_audio,
+            "python_code": python_code,
+            "open_image": open_image,
+            "open_youtube_video": open_youtube_video,
+            "google_search": google_search
+        }
         
         # System message
         self.sys_msg = SystemMessage('''You are a general AI assistant. I will ask you a question. Follow these steps:
@@ -435,7 +446,7 @@ class BasicAgent:
                             logger.info(f"Executing tool: {tool_name}")
                             logger.info(f"Tool arguments: {json.dumps(tool_args, indent=2)}")
                             
-                            result = self.tool_node.invoke({"name": tool_name, "args": tool_args})
+                            result = execute_tool(tool_name, tool_args, self.tools)
                             logger.info(f"Tool result: {result}")
                             
                             messages.append(AIMessage(content=f"Tool result: {result}"))