Update app.py

app.py

@@ -147,26 +147,22 @@ 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...")
-        model_repo = "google/gemma-1.1-7b-it"  # Consider Mixtral for more power if resources allow
-
+        model_repo = "google/gemma-1.1-7b-it"  # Or your preferred model
         print(f"Loading HfAgent with model: {model_repo}")
+
         try:
             # 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__: 
+            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__: 
+            if not web_search.__doc__:
                 raise ValueError("Tool 'web_search' is missing a docstring.")
-            if not safe_calculator.__doc__: 
+            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)
+            # Manually wrap your functions into Tool objects using the Tool constructor directly
             time_tool_obj = Tool(
                 name=get_current_time_in_timezone.__name__,
                 func=get_current_time_in_timezone,
@@ -183,28 +179,56 @@ class HfAgentWrapper:
                 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]
             
             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} and descriptions.")
 
+            # Initialize HfAgent with the list of Tool objects
             self.agent = HfAgent(model_repo, additional_tools=self.actual_tools_for_agent)
-            
-            print(f"HfAgent loaded successfully with model {model_repo}.")
-            # 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()]}")
-
+            print(f"HfAgent successfully instantiated with model {model_repo}.")
+
+            # --- New robust way to inspect the toolbox ---
+            print(f"Inspecting self.agent.toolbox...")
+            if hasattr(self.agent, 'toolbox'):
+                toolbox_attr = self.agent.toolbox
+                print(f"Type of self.agent.toolbox: {type(toolbox_attr)}")
+
+                if isinstance(toolbox_attr, dict):
+                    print(f"Toolbox is a dict. Keys: {list(toolbox_attr.keys())}")
+                    tools_info_from_dict = {}
+                    # Inspect a few items if it's a large dictionary
+                    for key, value in list(toolbox_attr.items())[:5]: 
+                        tool_name_in_dict = key # Default to key if value has no 'name'
+                        if hasattr(value, 'name'):
+                            tool_name_in_dict = value.name
+                        elif isinstance(value, dict) and 'name' in value:
+                            tool_name_in_dict = value['name']
+                        tools_info_from_dict[key] = {'name_attr': tool_name_in_dict, 'type': str(type(value))}
+                    print(f"Sample of toolbox dict content: {tools_info_from_dict}")
+                    # Compare with the tools we intended to pass
+                    print(f"Names of tools we passed to HfAgent: {[tool.name for tool in self.actual_tools_for_agent]}")
+
+                elif hasattr(toolbox_attr, 'tools') and callable(toolbox_attr.tools):
+                    try:
+                        tools_in_toolbox = toolbox_attr.tools()
+                        print(f"Toolbox has .tools() method. Tool names: {[tool.name for tool in tools_in_toolbox if hasattr(tool, 'name')]}")
+                    except Exception as e_toolbox_call:
+                        print(f"Error calling or processing self.agent.toolbox.tools(): {e_toolbox_call}")
+                else:
+                    print(f"Toolbox is of type {type(toolbox_attr)} but not a dict or recognized Toolbox object for this debug print.")
+            else:
+                print("self.agent does not have a 'toolbox' attribute (this would be unexpected).")
+            # --- End of new robust toolbox inspection ---
 
         except Exception as e:
             print(f"CRITICAL ERROR: Failed to initialize HfAgent: {e}")
-            import traceback
+            import traceback # Import for full traceback
             print("Full traceback of HfAgent initialization error:")
             traceback.print_exc() 
             raise RuntimeError(f"HfAgent initialization failed: {e}") from e
 
     def __call__(self, question: str) -> str:
-        # ... (your __call__ method remains the same) ...
         print(f"\n--- HfAgentWrapper received question (first 100 chars): {question[:100]}... ---")
         try:
             answer = self.agent.run(question)
@@ -212,6 +236,10 @@ class HfAgentWrapper:
             return str(answer)
         except Exception as e:
             print(f"ERROR: HfAgent execution failed for question '{question[:50]}...': {e}")
+            # It might be helpful to see the full traceback for agent execution errors too
+            import traceback
+            print("Full traceback of HfAgent execution error:")
+            traceback.print_exc()
             return f"Agent Error: Failed to process the question. Details: {e}"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):