Update agent.py

agent.py

@@ -7,7 +7,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_groq import ChatGroq
 from langchain_core.tools import tool
 from langchain_core.messages import HumanMessage, SystemMessage
-from langchain_core.prompts import ChatPromptTemplate
+from langchain_core.prompts import ChatPromptTemplate, PromptTemplate
 from langgraph.prebuilt import ToolNode
 from langgraph.prebuilt import tools_condition
 
@@ -237,6 +237,20 @@ def divide(a: float, b: float) -> float:
 # - subtract: Subtract A by B with passing A as the first argument
 # - divide: Divide A by B with passing A as the first argument
 
+# You have access to the following tools:
+# - serper_websearch: web search the content of the query by passing the query as input with Serper Search Engine
+# - duckduck_websearch: web search the content of the query by passing the query as input with DuckDuckGo Search Engine
+# - visit_webpage: visit the given webpage url by passing the url as input
+# - wiki_search: wiki search the content of the query by passing the query as input if the question asks for wiki search it
+# - text_splitter: split text into chunks
+# - youtube_transcript: fetch the transcript of the Youtube video by passing the video url as input if the question asks for watching a Youtube video
+# - read_file: read the content of the attached file by passing the TASK-ID as input
+# - excel_read: read the content of the attached excel file by passing the TASK-ID as input
+# - csv_read: read the content of the attached csv file by passing the TASK-ID as input
+# - mp3_listen: listen to the content of the attached mp3 file by passing the TASK-ID as input
+# - image_caption: understand the visual content of the attached image by passing the TASK-ID as input
+# - run_python: run the python code
+
 # ("human", f"Question: {question}\nReport to validate: {final_answer}")
 class BasicAgent:
     def __init__(self):
@@ -267,31 +281,27 @@ 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:
-                - serper_websearch: web search the content of the query by passing the query as input with Serper Search Engine
-                - duckduck_websearch: web search the content of the query by passing the query as input with DuckDuckGo Search Engine
-                - visit_webpage: visit the given webpage url by passing the url as input
-                - wiki_search: wiki search the content of the query by passing the query as input if the question asks for wiki search it
-                - text_splitter: split text into chunks
-                - youtube_transcript: fetch the transcript of the Youtube video by passing the video url as input if the question asks for watching a Youtube video
-                - read_file: read the content of the attached file by passing the TASK-ID as input
-                - excel_read: read the content of the attached excel file by passing the TASK-ID as input
-                - csv_read: read the content of the attached csv file by passing the TASK-ID as input
-                - mp3_listen: listen to the content of the attached mp3 file by passing the TASK-ID as input
-                - image_caption: understand the visual content of the attached image by passing the TASK-ID as input
-                - run_python: run the python code
-
                 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.
+
+                Follow this format in your response:
+                THOUGHT: [Describe your reasoning here]
+                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]
         """
         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)
         # self.sys_msg = SystemMessage(content=self.sys_prompt)
         
-        self.prompt = ChatPromptTemplate.from_messages([
-            ("system", self.sys_prompt),
-            ("human", "{input}")
-        ])
+        # self.prompt = ChatPromptTemplate.from_messages([
+        #     ("system", self.sys_prompt),
+        #     ("human", "{input}")
+        # ])
+        self.prompt = PromptTemplate(
+            input_variables=["input", "tool_names"],
+            template=self.sys_prompt
+        )
         # self.agent = initialize_agent(
         #     tools=self.tools,
         #     llm=self.model,