Update agent.py

agent.py

@@ -91,13 +91,13 @@ class BasicAgent:
                 There are few tools provided: web_search, visit_webpage, read_file and image_caption. 
                 Here are few examples demonstrating how to call and use the tools.
         """
-        tools = [web_search, visit_webpage, read_file, image_caption]
-        prompt = ChatPromptTemplate.from_messages([
+        self.tools = [web_search, visit_webpage, read_file, image_caption]
+        self.prompt = ChatPromptTemplate.from_messages([
             ("system", self.sys_prompt),
-            ("human", "Question: {input}")
+            ("human", "{input}")
         ])
-        self.agent = create_tool_calling_agent(self.model, tools, prompt)
-        self.agent_exe = AgentExecutor(agent=self.agent, tools=tools, verbose=True)
+        self.agent = create_tool_calling_agent(self.model, self.tools, self.prompt)
+        self.agent_exe = AgentExecutor(agent=self.agent, tools=self.tools, verbose=True)
         print("BasicAgent initialized.")
     
     def __call__(self, question: str) -> str: