Update agent.py

agent.py

@@ -18,9 +18,11 @@ class AgentState(TypedDict):
 
 
 class BasicAgent():
-    def __init__(self, chat, vision_llm):
-        extract_text_from_image = tool(ExtractTextFromImage(vision_llm).__call__)
-        describe_image = tool(DescribeImage(vision_llm).__call__)
+    def __init__(self, chat):
+        self.vision_llm = ChatOpenAI(model="gpt-4o")
+        
+        extract_text_from_image = tool(ExtractTextFromImage(self.vision_llm).__call__)
+        describe_image = tool(DescribeImage(self.vision_llm).__call__)
 
         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)