fix docstring for time fn

app.py

@@ -4,21 +4,34 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+
 from Gradio_UI import GradioUI
 
+# Below is an example of a tool that does nothing. Amaze us with your creativity!
+@tool
+def my_custom_tool(arg1: str, arg2: int) -> str:
+    """
+    A tool that does nothing yet.
 
+    Args:
+        arg1 (str): The first argument.
+        arg2 (int): The second argument.
+    
+    Returns:
+        str: A placeholder message.
+    """
+    return "What magic will you build ?"
 
-# Tool to get current local time in a specified timezone
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """
     A tool that fetches the current local time in a specified timezone.
-    
+
     Args:
         timezone (str): A valid timezone string (e.g., 'America/New_York').
-    
+
     Returns:
-        str: A message with the current local time or an error message if the timezone is invalid.
+        str: A message with the current local time in the given timezone or an error message if the timezone is invalid.
     """
     try:
         tz = pytz.timezone(timezone)
@@ -27,21 +40,20 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Calculator tool that evaluates mathematical expressions
 @tool
 def calculator_tool(expression: str) -> str:
     """
     A calculator tool that evaluates a mathematical expression.
-    
+
     Args:
         expression (str): A string representing a mathematical expression.
                           Supported operators: +, -, *, /, **, and %.
-    
+
     Returns:
         str: The result of the calculation as a string, or an error message if evaluation fails.
     """
     try:
-        # Evaluate the expression with a restricted environment for safety.
+        # Evaluate the expression in a restricted environment for safety.
         result = eval(expression, {"__builtins__": {}}, {})
         if isinstance(result, float) and result.is_integer():
             result = int(result)
@@ -49,28 +61,24 @@ def calculator_tool(expression: str) -> str:
     except Exception as e:
         return f"Error evaluating expression: {e}"
 
-# Instantiate the final answer tool
 final_answer = FinalAnswerTool()
 
-# Create an instance of the language model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # It is possible that this model may be overloaded.
     custom_role_conversions=None,
 )
 
-# Import tool from Hub (e.g., for image generation)
+# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load prompt templates from YAML file
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
-# Create the CodeAgent with the desired tools. Notice that we add our calculator_tool
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, calculator_tool],  # You can add more tools here
+    tools=[final_answer, calculator_tool],  # Added our calculator tool along with final_answer.
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -80,5 +88,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch the Gradio UI for interacting with the agent
 GradioUI(agent).launch()