Update app.py

app.py

@@ -14,6 +14,7 @@ print(f"--- Python version: {sys.version} ---")
 # print(f"--- Python sys.path (module search paths): {sys.path} ---") # Optional now
 
 import transformers
+from transformers.tools import Tool
 print(f"--- Expected Transformers Version: 4.36.0 ---")
 print(f"--- Actual Transformers Version: {transformers.__version__} ---")
 # print(f"--- Transformers module loaded from: {transformers.__file__} ---") # Optional now
@@ -146,7 +147,6 @@ 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,15 +154,38 @@ class HfAgentWrapper:
 
         print(f"Loading HfAgent with model: {model_repo}")
         try:
-            self.tools_list = [get_current_time_in_timezone, web_search, safe_calculator]
-            self.agent = HfAgent(model_repo, additional_tools=self.tools_list)
+            # 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)
+
+            # 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}")
+
+            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.__name__ for tool in self.tools_list]}")
+            print(f"HfAgent initialized with tools: {tool_names_for_log}") # Use the names from the Tool objects
+
         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() 
             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)