Update agent.py

agent.py

@@ -281,6 +281,9 @@ class BasicAgent:
                 If you are asked for a string, don't use articles, neither abbreviations (eg. 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 put in the list is a number or a string.
 
+                You have access to the following tools:
+                {tools}
+                Here are the tools you can use: {tool_names}
                 If Task ID is included in the question, remember to call the relevant read tools [ie. read_file, excel_read, csv_read, mp3_listen, image_caption]
                 Note: python_tool is called when the question mentions the term "Python" or any math calculation.
 
@@ -289,6 +292,9 @@ class BasicAgent:
                 ACTION: [Specify the action/tool to use and any relevant input]
                 OBSERVATIOn: [Result of the action/tool, provided by the system]
                 FINAL ANSWER: [Provide your final response to the user]
+
+                User Input: {input}
+                {agent_scratchpad}
         """
         self.tools = [duckduck_websearch, serper_websearch, visit_webpage, wiki_search, text_splitter, youtube_transcript, read_file, excel_read, csv_read, mp3_listen, image_caption, run_python]
         # self.model_with_tools = self.model.bind_tools(self.tools)
@@ -299,7 +305,7 @@ class BasicAgent:
         #     ("human", "{input}")
         # ])
         self.prompt = PromptTemplate(
-            input_variables=["input", "tool_names"],
+            input_variables=["input", "tools", "tool_names", "agent_scratchpad"],
             template=self.sys_prompt
         )
         # self.agent = initialize_agent(