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