Update app.py

app.py

@@ -14,52 +14,8 @@ model = OpenAIServerModel(
     api_base="https://generativelanguage.googleapis.com/v1beta/openai/"
 )
 
-# prompt = PromptTemplates(
-#     system_prompt="""
-# You are a helpful assistant tasked with answering questions using a set of tools.
-# {{tool_descriptions}}
-
-# Now, 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 commas or units like $ or %, unless explicitly required.
-# - If you are asked for a string, do not include articles or abbreviations, and write digits in full unless otherwise specified.
-# - If asked for a comma-separated list, apply the above rules depending on whether the list elements are numbers or strings.
-
-# Your answer should **only** start with `FINAL ANSWER: ` followed by the answer.
-# """,
-
-#     planning="""
-# You have received the following task:
-# {{task_description}}
-
-# Think step-by-step about what tools or steps are needed to solve it. Plan clearly before proceeding.
-# """,
-
-#     managed_agent="""
-# You are executing the following step:
-# {{step_description}}
-
-# Use available tools wisely. Think before acting, and ensure your actions follow the original instruction format.
-
-# Remember to ultimately return the answer in the format:
-# FINAL ANSWER: [YOUR FINAL ANSWER]
-# """,
-
-#     final_answer="""
-# You have completed all steps.
-
-# Now, return your final response using the following format:
-# FINAL ANSWER: [YOUR FINAL ANSWER]
-
-# YOUR FINAL ANSWER should be a number, a few words, or a comma-separated list, following the formatting rules provided earlier.
-# """
-# )
-
-prompt = {
-    'system_prompt':"""
+prompt = PromptTemplates(
+    system_prompt="""
 You are a helpful assistant tasked with answering questions using a set of tools.
 {{tool_descriptions}}
 
@@ -75,14 +31,14 @@ YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma sepa
 Your answer should **only** start with `FINAL ANSWER: ` followed by the answer.
 """,
 
-    'planning':"""
+    planning="""
 You have received the following task:
 {{task_description}}
 
 Think step-by-step about what tools or steps are needed to solve it. Plan clearly before proceeding.
 """,
 
-    'managed_agent':"""
+    managed_agent="""
 You are executing the following step:
 {{step_description}}
 
@@ -92,7 +48,7 @@ Remember to ultimately return the answer in the format:
 FINAL ANSWER: [YOUR FINAL ANSWER]
 """,
 
-    'final_answer':"""
+    final_answer="""
 You have completed all steps.
 
 Now, return your final response using the following format:
@@ -100,7 +56,51 @@ FINAL ANSWER: [YOUR FINAL ANSWER]
 
 YOUR FINAL ANSWER should be a number, a few words, or a comma-separated list, following the formatting rules provided earlier.
 """
-}
+)
+
+# prompt = {
+#     'system_prompt':"""
+# You are a helpful assistant tasked with answering questions using a set of tools.
+# {{tool_descriptions}}
+
+# Now, 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 commas or units like $ or %, unless explicitly required.
+# - If you are asked for a string, do not include articles or abbreviations, and write digits in full unless otherwise specified.
+# - If asked for a comma-separated list, apply the above rules depending on whether the list elements are numbers or strings.
+
+# Your answer should **only** start with `FINAL ANSWER: ` followed by the answer.
+# """,
+
+#     'planning':"""
+# You have received the following task:
+# {{task_description}}
+
+# Think step-by-step about what tools or steps are needed to solve it. Plan clearly before proceeding.
+# """,
+
+#     'managed_agent':"""
+# You are executing the following step:
+# {{step_description}}
+
+# Use available tools wisely. Think before acting, and ensure your actions follow the original instruction format.
+
+# Remember to ultimately return the answer in the format:
+# FINAL ANSWER: [YOUR FINAL ANSWER]
+# """,
+
+#     'final_answer':"""
+# You have completed all steps.
+
+# Now, return your final response using the following format:
+# FINAL ANSWER: [YOUR FINAL ANSWER]
+
+# YOUR FINAL ANSWER should be a number, a few words, or a comma-separated list, following the formatting rules provided earlier.
+# """
+# }
 
 
 # (Keep Constants as is)
@@ -115,8 +115,9 @@ class BasicAgent:
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             model = model,
             add_base_tools=True,
-            system_prompt = prompt
         )
+
+        self.agent.prompt_templates["system_prompt"] = prompt
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")