Update agent.py

agent.py

@@ -12,7 +12,6 @@ from typing import TypedDict, Annotated, Optional
 
 class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
-    input_file: Optional[str]  # Contains file path (PDF/PNG)
 
 
 class BasicAgent():
@@ -63,6 +62,7 @@ class BasicAgent():
         
             Returns:
                 A single string containing the concatenated text extracted from each image.
+        
         search_tool(query: str) -> str:
             Search the web using the DuckDuckGoSearchRun to perform a search query and return a summarized textual result.
 
@@ -75,11 +75,10 @@ class BasicAgent():
         sys_msg = SystemMessage(content=f"""
         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.
-        You have access to the following tools:\n{textual_description_of_tool} \n You may have access to some optional images. Currently the loaded image is: {image}.""")
+        You have access to the following tools:\n{textual_description_of_tool}\n""")
 
         return {
             "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],            
-            "input_file": image
         }