Update app.py

app.py

@@ -147,6 +147,7 @@ def safe_calculator(expression: str) -> str:
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# --- Agent Definition using HfAgent ---
 class HfAgentWrapper:
     def __init__(self):
         print("Initializing HfAgentWrapper...")
@@ -154,31 +155,49 @@ class HfAgentWrapper:
 
         print(f"Loading HfAgent with model: {model_repo}")
         try:
-            # Your original tool functions
-            # get_current_time_in_timezone, web_search, safe_calculator
-
-            # Manually wrap your functions into Tool objects
-            # Tool.from_function(func) uses func.__name__ for the tool name
-            # and func.__doc__ for the tool description.
-            time_tool_obj = Tool.from_function(get_current_time_in_timezone)
-            search_tool_obj = Tool.from_function(web_search)
-            calculator_tool_obj = Tool.from_function(safe_calculator)
+            # Ensure your tool functions have docstrings, as they are used for the description.
+            # (Your current tool functions look like they do, which is good)
+            if not get_current_time_in_timezone.__doc__: 
+                raise ValueError("Tool 'get_current_time_in_timezone' is missing a docstring.")
+            if not web_search.__doc__: 
+                raise ValueError("Tool 'web_search' is missing a docstring.")
+            if not safe_calculator.__doc__: 
+                raise ValueError("Tool 'safe_calculator' is missing a docstring.")
+
+            # Manually wrap your functions into Tool objects using the Tool constructor
+            # The constructor for Tool in this version is typically:
+            # Tool(name: str, func: Callable, description: str)
+            time_tool_obj = Tool(
+                name=get_current_time_in_timezone.__name__,
+                func=get_current_time_in_timezone,
+                description=get_current_time_in_timezone.__doc__
+            )
+            search_tool_obj = Tool(
+                name=web_search.__name__,
+                func=web_search,
+                description=web_search.__doc__
+            )
+            calculator_tool_obj = Tool(
+                name=safe_calculator.__name__,
+                func=safe_calculator,
+                description=safe_calculator.__doc__
+            )
 
             # This is the list of Tool objects that HfAgent will receive
             self.actual_tools_for_agent = [time_tool_obj, search_tool_obj, calculator_tool_obj]
-
-            # For printing/logging purposes, let's see their names
+            
             tool_names_for_log = [tool.name for tool in self.actual_tools_for_agent]
-            print(f"Prepared Tool objects with names: {tool_names_for_log}")
+            print(f"Prepared Tool objects with names: {tool_names_for_log} and descriptions.")
 
             self.agent = HfAgent(model_repo, additional_tools=self.actual_tools_for_agent)
-
+            
             print(f"HfAgent loaded successfully with model {model_repo}.")
-            print(f"HfAgent initialized with tools: {tool_names_for_log}") # Use the names from the Tool objects
+            # HfAgent's toolbox will also have its own representation of these tools
+            print(f"HfAgent internal toolbox tool names: {[tool.name for tool in self.agent.toolbox.tools()]}")
+
 
         except Exception as e:
             print(f"CRITICAL ERROR: Failed to initialize HfAgent: {e}")
-            # Ensure the full traceback of the original error 'e' is visible in logs
             import traceback
             print("Full traceback of HfAgent initialization error:")
             traceback.print_exc()