Update agent.py

agent.py

@@ -20,7 +20,7 @@ class BasicAgent():
         
         search_tool = DuckDuckGoSearchRun()
         vision_llm = ChatOpenAI(model="gpt-4o")
-        self.tools = [describe_image, search_tool]
+        self.tools = [extract_text, describe_image, search_tool]
         self.chat_with_tools = chat.bind_tools(self.tools)
         self._initialize_graph()
         print("BasicAgent initialized.")
@@ -57,7 +57,7 @@ class BasicAgent():
         You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
         YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
         """)
-
+        print("Calling assistant...", state["messages"])
         return {
             "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],            
         }