Update app.py

app.py

@@ -19,20 +19,64 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+def reflect_and_summarize(thought: str) -> str:
+    """Condenses reasoning or intermediate thoughts into a concise summary.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        thought: Intermediate reasoning or notes from the agent
+    """
+    summary = thought.strip()
+    if len(summary) > 300:
+        summary = summary[:300] + "..."
+    return f"Key insight: {summary}"
+
+@tool
+def fetch_json(url: str) -> str:
+    """Fetches JSON data from a public API endpoint.
+    Args:
+        url: A publicly accessible API endpoint returning JSON
     """
     try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
+        response = requests.get(url, timeout=5)
+        response.raise_for_status()
+        return response.text[:2000]  # prevent token explosion
     except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+        return f"Failed to fetch data: {str(e)}"
+
 
+@tool
+def list_prompt_templates() -> str:
+    """Lists available prompt templates loaded from prompts.yaml."""
+    try:
+        return yaml.dump(prompt_templates, sort_keys=False)
+    except Exception as e:
+        return f"Error reading prompts: {str(e)}"
+        
+@tool
+def get_current_time(
+    timezone: str = "UTC",
+    fmt: str = "%Y-%m-%d %H:%M:%S"
+) -> str:
+    """Returns current time in a specified timezone.
+    Args:
+        timezone: Timezone name (default: UTC)
+        fmt: Datetime format string
+    """
+    try:
+        tz = pytz.timezone(timezone)
+        return datetime.datetime.now(tz).strftime(fmt)
+    except Exception as e:
+        return f"Invalid timezone or format: {e}"
+@tool
+def validate_final_answer(answer: str) -> str:
+    """Checks final answer for clarity and completeness."""
+    issues = []
+    if len(answer) < 20:
+        issues.append("Answer too short")
+    if "I don't know" in answer:
+        issues.append("Uncertain response detected")
+    if not issues:
+        return "Answer looks good."
+    return "Potential issues: " + ", ".join(issues)
 
 final_answer = FinalAnswerTool()
 
@@ -55,7 +99,13 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools = [
+    reflect_and_summarize,
+    fetch_json,
+    get_current_time,
+    list_prompt_templates,
+    validate_final_answer,
+    final_answer ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,