Update agent.py

agent.py

@@ -21,7 +21,7 @@ class BasicAgent():
         search_tool = DuckDuckGoSearchRun()
         vision_llm = ChatOpenAI(model="gpt-4o")
         self.tools = [extract_text, describe_image, search_tool]
-        self.chat_with_tools = vision_llm.bind_tools(self.tools)
+            self.chat_with_tools = vision_llm.bind_tools(self.tools) # need to change
         self._initialize_graph()
         print("BasicAgent initialized.")
 
@@ -42,9 +42,12 @@ class BasicAgent():
         self.agent = builder.compile()
 
     
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question: {question}.")
-        messages=[HumanMessage(content=question)]
+    def __call__(self, question: str, file_name : str) -> str:
+        print(f"Agent received question: {question}. /nProvided file: {file_name}.")
+        if file_name is not None:
+            messages=[HumanMessage(content=f"{question}. The filename you have access to is {file_name}.")]
+        else:
+            messages=[HumanMessage(content=question)]
         response = self.agent.invoke({"messages":messages})
         answer = response['messages'][-1].content
         print(f"Agent returning answer: {answer}")