Update app.py

app.py

@@ -1,17 +1,26 @@
-from smolagents import CodeAgent, HfApiModel, load_tool, tool
+#!/usr/bin/env python
+# coding=utf-8
+
+import copy
 import datetime
 import requests
 import pytz
 import yaml
 
-# Import our custom tools from the tools folder
+# smolagents imports:
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
+from smolagents.prompts import DEFAULT_PROMPT_TEMPLATES
+
+# Tools from your local folder
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
 
+# Gradio UI from your local file
 from Gradio_UI import GradioUI
 
-# Example Tool (non-functioning): provided purely as a template.
+###############################################################
+# Example custom tool: a non-functional template
 @tool
 def my_custom_tool(arg1: str, arg2: int) -> str:
     """Example Tool: A non-functional tool provided as a template.
@@ -22,7 +31,8 @@ def my_custom_tool(arg1: str, arg2: int) -> str:
     """
     return "What magic will you build ?"
 
-# Working Tool: Fetches the current local time in a specified timezone.
+###############################################################
+# A working tool: get the current time in a given timezone
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Working Tool: Fetches the current local time in a specified timezone.
@@ -37,37 +47,56 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Instantiate the FinalAnswerTool (required for returning final answers)
+###############################################################
+# Instantiate final_answer (required for returning final answers)
 final_answer = FinalAnswerTool()
 
-# Instantiate working local tools:
-# Web Search Tool: Uses DuckDuckGo to perform a web search.
-search_tool = DuckDuckGoSearchTool()   # Defined in tools/web_search.py
+# Other local tools: web search & visit page
+search_tool = DuckDuckGoSearchTool()
+visit_tool = VisitWebpageTool()
 
-# Webpage Visit Tool: Visits a URL and converts its content to Markdown.
-visit_tool = VisitWebpageTool()         # Defined in tools/visit_webpage.py
-
-# Define the model configuration using HfApiModel
+###############################################################
+# Define model configuration using HfApiModel
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen3-32B',  # Adjust if this model is overloaded.
+    model_id='Qwen/Qwen3-32B',  # Adjust if overloaded or unavailable
     custom_role_conversions=None,
 )
 
-# Load prompt templates from prompts.yaml
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
+###############################################################
+# 1) Read your custom prompts.yaml
+try:
+    with open("prompts.yaml", 'r') as stream:
+        custom_prompts = yaml.safe_load(stream)
+except Exception as e:
+    print("Error loading prompts.yaml:", e)
+    custom_prompts = {}
+
+# 2) Merge with the built-in defaults so no keys or subkeys are missing
+prompt_templates = copy.deepcopy(DEFAULT_PROMPT_TEMPLATES)
+
+def deep_merge(base_dict, override_dict):
+    """Recursively merge override_dict into base_dict."""
+    for k, v in override_dict.items():
+        if isinstance(v, dict) and isinstance(base_dict.get(k), dict):
+            deep_merge(base_dict[k], v)
+        else:
+            base_dict[k] = v
+
+# Perform the merge
+deep_merge(prompt_templates, custom_prompts)
 
-# Create the CodeAgent, including our working tools and the example tool.
+###############################################################
+# Create the CodeAgent with the merged prompt templates
 agent = CodeAgent(
     model=model,
     tools=[
-        final_answer,               # Must remain included.
-        search_tool,                # Enables web search.
-        visit_tool,                 # Enables webpage visits.
-        get_current_time_in_timezone,  # Working tool for time queries.
-        my_custom_tool,             # Example (non-functioning) tool.
+        final_answer,
+        search_tool,
+        visit_tool,
+        get_current_time_in_timezone,
+        my_custom_tool,  # Example tool
     ],
     max_steps=6,
     verbosity_level=1,
@@ -75,8 +104,10 @@ agent = CodeAgent(
     planning_interval=None,
     name=None,
     description=None,
+    # The merged dictionary—ensures all required keys & structure are present
     prompt_templates=prompt_templates
 )
 
-# Launch the Gradio UI for interactive use
-GradioUI(agent).launch()
+###############################################################
+# Launch Gradio UI
+GradioUI(agent).launch()