Update agent.py

agent.py

@@ -14,12 +14,10 @@ class AgentState(TypedDict):
 
 
 class BasicAgent():
-    def __init__(self, llm):
-        chat = ChatHuggingFace(llm=llm, verbose=True)
-        
-        vision_llm = ChatOpenAI(model="gpt-4o")
+    def __init__(self, chat, vision_llm):
+        self.vision_llm = vision_llm
         self.tools = [extract_text, describe_image, wiki_search, web_search, arxiv_search]
-        self.chat_with_tools = chat.bind_tools(self.tools) # need to change
+        self.chat_with_tools = chat.bind_tools(self.tools)
         self._initialize_graph()
         print("BasicAgent initialized.")