updated custom prompt_templates

agent.py

@@ -1,5 +1,9 @@
 from smolagents import (
     ToolCallingAgent,
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    FinalAnswerTool,
+    VisitWebpageTool,
     PromptTemplates,
     PlanningPromptTemplate,
     ManagedAgentPromptTemplate,
@@ -19,22 +23,25 @@ class BaseAgent:
             model=openai_model,
             tools=[
                 addition_tool,
+                # DuckDuckGoSearchTool(),
+                # VisitWebpageTool(),
+                # FinalAnswerTool(),
             ],
             verbosity_level=2,
             prompt_templates=PromptTemplates(
                 system_prompt="You are a general AI assistant. 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 comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. 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 be put in the list is a number or a string.",
-                # planning=PlanningPromptTemplate(
-                #     initial_plan="",
-                #     update_plan_pre_messages="",
-                #     update_plan_post_messages="",
-                # ),
-                # managed_agent=ManagedAgentPromptTemplate(
-                #     task="",
-                #     report="",
-                # ),
-                # final_answer=FinalAnswerPromptTemplate(
-                #     pre_messages="", post_messages=""
-                # ),
+                planning=PlanningPromptTemplate(
+                    initial_plan="",
+                    update_plan_pre_messages="",
+                    update_plan_post_messages="",
+                ),
+                managed_agent=ManagedAgentPromptTemplate(
+                    task="",
+                    report="",
+                ),
+                final_answer=FinalAnswerPromptTemplate(
+                    pre_messages="", post_messages=""
+                ),
             ),
         )