Refactor: Move prompt templates to separate Python file and update app.py

app.py

@@ -7,6 +7,31 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 import re
 import os
+import gradio as gr
+from smolagents import Agent, Tool
+from smolagents.tools import (
+    SearchTool,
+    CalculatorTool,
+    WikipediaTool,
+    FileSystemTool,
+    CodeInterpreterTool,
+    WebScrapingTool,
+    WeatherTool,
+    NewsTool,
+    ImageGenerationTool,
+    TextToSpeechTool,
+    TranslationTool,
+    EmailTool,
+    CalendarTool,
+    MapTool,
+    MusicTool,
+    VideoTool,
+    DatabaseTool,
+    APITool,
+    BrowserTool,
+    TerminalTool
+)
+from prompt_templates import PROMPT_TEMPLATES
 
 from Gradio_UI import GradioUI
 
@@ -68,129 +93,93 @@ model = OpenAIModel(
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Define prompt templates directly as a Python dictionary
-prompt_templates = {
-    "system_prompt": """You are an expert assistant who can solve any task using code blobs. You will be given a task to solve as best you can.
-To do so, you have been given access to a list of tools: these tools are basically Python functions which you can call with code.
-To solve the task, you must plan forward to proceed in a series of steps, in a cycle of 'Thought:', 'Code:', and 'Observation:' sequences.
-At each step, in the 'Thought:' sequence, you should first explain your reasoning towards solving the task and the tools that you want to use.
-Then in the 'Code:' sequence, you should write the code in simple Python. The code sequence must end with '<end_code>' sequence.
-During each intermediate step, you can use 'print()' to save whatever important information you will then need.
-These print outputs will then appear in the 'Observation:' field, which will be available as input for the next step.
-
-IMPORTANT RULES:
-1. You MUST use the final_answer tool to submit your answer. Never return a default answer.
-2. If you're unsure about an answer, use the search tool to find information.
-3. Always try to find the answer rather than returning a default response.
-4. If you can't find a definitive answer, provide your best reasoning based on available information.
-5. Never return "This is a default answer" - always provide your best attempt at answering the question.
-6. For each question, you must:
-   a. Analyze what information is needed
-   b. Use appropriate tools to gather that information
-   c. Process the information to form an answer
-   d. Use final_answer to submit your response""",
-
-    "final_answer": """You have reached the final step of your task. Use the final_answer tool to submit your answer.
-The answer should be clear, concise, and directly address the original question.
-Format: final_answer(your_answer)""",
-
-    "planning": {
-        "initial_facts": """Below I will present you a task.
-You will now build a comprehensive preparatory survey of which facts we have at our disposal and which ones we still need.
-To do so, you will have to read the task and identify things that must be discovered in order to successfully complete it.
-Don't make any assumptions. For each item, provide a thorough reasoning. Here is how you will structure this survey:
-
----
-### 1. Facts given in the task
-List here the specific facts given in the task that could help you (there might be nothing here).
-
-### 2. Facts to look up
-List here any facts that we may need to look up.
-Also list where to find each of these, for instance a website, a file... - maybe the task contains some sources that you should re-use here.
-
-### 3. Facts to derive
-List here anything that we want to derive from the above by logical reasoning, for instance computation or simulation.""",
-
-        "initial_plan": """You are a world expert at making efficient plans to solve any task using a set of carefully crafted tools.
-Now for the given task, develop a step-by-step high-level plan taking into account the above inputs and list of facts.
-This plan should involve individual tasks based on the available tools, that if executed correctly will yield the correct answer.
-Do not skip steps, do not add any superfluous steps. Only write the high-level plan, DO NOT DETAIL INDIVIDUAL TOOL CALLS.
-After writing the final step of the plan, write the '\n<end_plan>' tag and stop there.""",
-
-        "update_facts_pre_messages": """You are a world expert at gathering known and unknown facts based on a conversation.
-Below you will find a task, and a history of attempts made to solve the task. You will have to produce a list of these:
-### 1. Facts given in the task
-### 2. Facts that we have learned
-### 3. Facts still to look up
-### 4. Facts still to derive
-Find the task and history below:""",
-
-        "update_facts_post_messages": """Earlier we've built a list of facts.
-But since in your previous steps you may have learned useful new facts or invalidated some false ones.
-Please update your list of facts based on the previous history, and provide these headings:
-### 1. Facts given in the task
-### 2. Facts that we have learned
-### 3. Facts still to look up
-### 4. Facts still to derive
-
-Now write your new list of facts below.""",
-
-        "update_plan_pre_messages": """You are a world expert at making efficient plans to solve any task using a set of carefully crafted tools.
-You have been given a task:
-```
-{{task}}
-```
-
-Find below the record of what has been tried so far to solve it. Then you will be asked to make an updated plan to solve the task.
-If the previous tries so far have met some success, you can make an updated plan based on these actions.
-If you are stalled, you can make a completely new plan starting from scratch.""",
-
-        "update_plan_post_messages": """You're still working towards solving this task:
-```
-{{task}}
-```
-
-Now for the given task, develop a step-by-step high-level plan taking into account the above inputs and list of facts.
-This plan should involve individual tasks based on the available tools, that if executed correctly will yield the correct answer.
-Beware that you have {remaining_steps} steps remaining.
-Do not skip steps, do not add any superfluous steps. Only write the high-level plan, DO NOT DETAIL INDIVIDUAL TOOL CALLS.
-After writing the final step of the plan, write the '\n<end_plan>' tag and stop there.
-
-Now write your new plan below."""
-    },
-
-    "managed_agent": {
-        "task": """You're a helpful agent named '{{name}}'.
-You have been submitted this task by your manager.
----
-Task:
-{{task}}
----
-You're helping your manager solve a wider task: so make sure to not provide a one-line answer, but give as much information as possible to give them a clear understanding of the answer.
-Your final_answer WILL HAVE to contain these parts:
-### 1. Task outcome (short version):
-### 2. Task outcome (extremely detailed version):
-### 3. Additional context (if relevant):
-
-Put all these in your final_answer tool, everything that you do not pass as an argument to final_answer will be lost.
-And even if your task resolution is not successful, please return as much context as possible, so that your manager can act upon this feedback.""",
-
-        "report": """Here is the final answer from your managed agent '{{name}}':
-{{final_answer}}"""
-    }
-}
+# Initialize tools
+tools = [
+    SearchTool(),
+    CalculatorTool(),
+    WikipediaTool(),
+    FileSystemTool(),
+    CodeInterpreterTool(),
+    WebScrapingTool(),
+    WeatherTool(),
+    NewsTool(),
+    ImageGenerationTool(),
+    TextToSpeechTool(),
+    TranslationTool(),
+    EmailTool(),
+    CalendarTool(),
+    MapTool(),
+    MusicTool(),
+    VideoTool(),
+    DatabaseTool(),
+    APITool(),
+    BrowserTool(),
+    TerminalTool()
+]
+
+# Initialize agents
+managed_agent = Agent(
+    tools=tools,
+    prompt_template=PROMPT_TEMPLATES["managed_agent"]
+)
+
+tool_agent = Agent(
+    tools=tools,
+    prompt_template=PROMPT_TEMPLATES["tool_agent"]
+)
+
+tool_agent_with_memory = Agent(
+    tools=tools,
+    prompt_template=PROMPT_TEMPLATES["tool_agent_with_memory"],
+    memory=True
+)
+
+tool_agent_with_memory_and_planning = Agent(
+    tools=tools,
+    prompt_template=PROMPT_TEMPLATES["tool_agent_with_memory_and_planning"],
+    memory=True,
+    planning=True
+)
+
+# Create Gradio interface
+def chat(message, history, agent_type):
+    # Select agent based on type
+    if agent_type == "Managed Agent":
+        agent = managed_agent
+    elif agent_type == "Tool Agent":
+        agent = tool_agent
+    elif agent_type == "Tool Agent with Memory":
+        agent = tool_agent_with_memory
+    else:  # Tool Agent with Memory and Planning
+        agent = tool_agent_with_memory_and_planning
     
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer, DuckDuckGoSearchTool(), calculate_min_price, extract_price_from_snippet, get_current_time_in_timezone],
-    max_steps=15,  # Increased max steps for more complex reasoning
-    verbosity_level=2,  # Increased verbosity for better debugging
-    grammar=None,
-    planning_interval=1,  # Added planning interval to ensure proper planning
-    name="Question Answering Agent",
-    description="An agent specialized in answering various types of questions using available tools. The agent must use the final_answer tool to submit its answer.",
-    prompt_templates=prompt_templates
+    # Get response from agent
+    response = agent.chat(message)
+    return response
+
+# Create the interface
+iface = gr.Interface(
+    fn=chat,
+    inputs=[
+        gr.Textbox(label="Message", placeholder="Type your message here..."),
+        gr.State([]),  # For chat history
+        gr.Radio(
+            choices=[
+                "Managed Agent",
+                "Tool Agent",
+                "Tool Agent with Memory",
+                "Tool Agent with Memory and Planning"
+            ],
+            label="Agent Type",
+            value="Managed Agent"
+        )
+    ],
+    outputs=gr.Textbox(label="Response"),
+    title="AI Assistant",
+    description="Choose an agent type and start chatting!",
+    theme=gr.themes.Soft()
 )
 
-# Configure Gradio UI with sharing enabled
-GradioUI(agent).launch(share=True)
+# Launch the app
+if __name__ == "__main__":
+    iface.launch()

prompt_templates.py

@@ -0,0 +1,127 @@
+PROMPT_TEMPLATES = {
+    "managed_agent": """You are a helpful AI assistant that helps users with their tasks. You have access to the following tools:
+
+{tools}
+
+To use a tool, please use the following format:
+
+```
+Thought: I need to use a tool to help the user
+Action: the_tool_name
+Action Input: {{"input": "the input to the tool"}}
+```
+
+The user will then respond with the output of the tool. After using a tool, you should analyze the output and either:
+1. Use another tool if needed
+2. Provide a final answer to the user
+
+If you don't need to use any tools, you can respond directly to the user.
+
+Remember to:
+- Always think about what tool would be most helpful
+- Format your responses clearly
+- Be helpful and friendly
+- If you're not sure what to do, ask the user for clarification
+
+Human: {input}
+
+Assistant: Let me help you with that. First, I'll analyze what tools might be useful for your request.
+
+{agent_scratchpad}""",
+    
+    "tool_agent": """You are a helpful AI assistant that helps users with their tasks. You have access to the following tools:
+
+{tools}
+
+To use a tool, please use the following format:
+
+```
+Thought: I need to use a tool to help the user
+Action: the_tool_name
+Action Input: {{"input": "the input to the tool"}}
+```
+
+The user will then respond with the output of the tool. After using a tool, you should analyze the output and either:
+1. Use another tool if needed
+2. Provide a final answer to the user
+
+If you don't need to use any tools, you can respond directly to the user.
+
+Remember to:
+- Always think about what tool would be most helpful
+- Format your responses clearly
+- Be helpful and friendly
+- If you're not sure what to do, ask the user for clarification
+
+Human: {input}
+
+Assistant: Let me help you with that. First, I'll analyze what tools might be useful for your request.
+
+{agent_scratchpad}""",
+    
+    "tool_agent_with_memory": """You are a helpful AI assistant that helps users with their tasks. You have access to the following tools:
+
+{tools}
+
+To use a tool, please use the following format:
+
+```
+Thought: I need to use a tool to help the user
+Action: the_tool_name
+Action Input: {{"input": "the input to the tool"}}
+```
+
+The user will then respond with the output of the tool. After using a tool, you should analyze the output and either:
+1. Use another tool if needed
+2. Provide a final answer to the user
+
+If you don't need to use any tools, you can respond directly to the user.
+
+Remember to:
+- Always think about what tool would be most helpful
+- Format your responses clearly
+- Be helpful and friendly
+- If you're not sure what to do, ask the user for clarification
+
+Previous conversation:
+{chat_history}
+
+Human: {input}
+
+Assistant: Let me help you with that. First, I'll analyze what tools might be useful for your request.
+
+{agent_scratchpad}""",
+    
+    "tool_agent_with_memory_and_planning": """You are a helpful AI assistant that helps users with their tasks. You have access to the following tools:
+
+{tools}
+
+To use a tool, please use the following format:
+
+```
+Thought: I need to use a tool to help the user
+Action: the_tool_name
+Action Input: {{"input": "the input to the tool"}}
+```
+
+The user will then respond with the output of the tool. After using a tool, you should analyze the output and either:
+1. Use another tool if needed
+2. Provide a final answer to the user
+
+If you don't need to use any tools, you can respond directly to the user.
+
+Remember to:
+- Always think about what tool would be most helpful
+- Format your responses clearly
+- Be helpful and friendly
+- If you're not sure what to do, ask the user for clarification
+
+Previous conversation:
+{chat_history}
+
+Human: {input}
+
+Assistant: Let me help you with that. First, I'll analyze what tools might be useful for your request.
+
+{agent_scratchpad}"""
+}