Update agent.py

agent.py

@@ -4,7 +4,7 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
-from tools import extract_text, describe_image, wiki_search, web_search, arxiv_search
+from tools import extract_text_from_image, read_excel, read_python, describe_image, wiki_search, web_search, arxiv_search
 from langchain_openai import ChatOpenAI
 from typing import TypedDict, Annotated, Optional
 
@@ -16,7 +16,7 @@ class AgentState(TypedDict):
 class BasicAgent():
     def __init__(self, chat, vision_llm):
         self.vision_llm = vision_llm
-        self.tools = [extract_text, describe_image, wiki_search, web_search, arxiv_search]
+        self.tools = [extract_text_from_image, read_excel, read_python, describe_image, wiki_search, web_search, arxiv_search]
         self.chat_with_tools = chat.bind_tools(self.tools)
         self._initialize_graph()
         print("BasicAgent initialized.")