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_from_image, read_excel, read_python, describe_image, wiki_search, web_search, arxiv_search
+from tools import make_text_from_image_tool, make_describe_image_tool, read_excel, read_python, wiki_search, web_search, arxiv_search
 from langchain_openai import ChatOpenAI
 from typing import TypedDict, Annotated, Optional
 
@@ -15,8 +15,7 @@ class AgentState(TypedDict):
 
 class BasicAgent():
     def __init__(self, chat, vision_llm):
-        self.vision_llm = vision_llm
-        self.tools = [extract_text_from_image, read_excel, read_python, describe_image, wiki_search, web_search, arxiv_search]
+        self.tools = [make_text_from_image_tool(vision_llm), make_describe_image_tool(vision_llm), read_excel, read_python, wiki_search, web_search, arxiv_search]
         self.chat_with_tools = chat.bind_tools(self.tools)
         self._initialize_graph()
         print("BasicAgent initialized.")