Update app.py

app.py

@@ -155,27 +155,24 @@ def safe_calculator(expression: str) -> str:
 class HfAgentWrapper:
     def __init__(self):
         print("Initializing HfAgentWrapper...")
-        model_id_or_path = "bigcode/starcoderbase-1b"  # Using a model compatible with transformers 4.36.0
+        model_id_or_path = "bigcode/starcoderbase-1b" # Compatible model
         
         try:
-            print(f"Attempting to initialize HfAgent directly with model: {model_id_or_path}")
-            
-            hf_auth_token = os.getenv("HF_TOKEN")
-            if not hf_auth_token:
-                # Starcoderbase might be gated or require login for some operations/configs
-                print("WARNING: HF_AUTH_TOKEN secret not found. This might cause issues if the model requires full authentication.")
-                # For now, we'll let it proceed and see if HfAgent can fetch it without an explicit token here
-                # if it's publicly accessible enough. If not, this will error out.
+            print(f"Attempting to initialize HfAgent with model: {model_id_or_path}")
+
+            # Optional: Check if the secret is present for better logging.
+            # HfAgent will look for 'HF_TOKEN' on its own.
+            if not os.getenv("HF_TOKEN"):
+                print("CRITICAL WARNING: Space Secret 'HF_TOKEN' is not set. HfAgent will likely fail on gated models.")
+                # You can raise an error to stop if the token is essential
+                raise ValueError("HF_TOKEN secret is missing and is required for this model.")
             else:
-                print(f"HF_AUTH_TOKEN found (length: {len(hf_auth_token)}), will pass it to HfAgent.")
+                print("HF_TOKEN secret is present. HfAgent will use it automatically.")
 
-            # --- Ensure Tool names are correct ---
-            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.")
+            # --- CRITICAL: Ensure Tool names are correct ---
+            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.")
 
             time_tool_obj = Tool(
                 name=get_current_time_in_timezone.__name__,
@@ -195,17 +192,15 @@ class HfAgentWrapper:
             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]
-            # This print MUST show your actual tool names, not empty strings
+            # This print MUST show your actual tool names. If it shows empty strings, the agent will not work.
             print(f"Prepared Tool objects with names: {tool_names_for_log} and descriptions.") 
             # --- End Tool Name Check ---
 
             # Let HfAgent create its own pipeline internally.
-            # Pass model_id_or_path as the FIRST POSITIONAL argument.
-            # Pass hf_token as a keyword argument.
+            # We REMOVE the hf_token argument. HfAgent will find the HF_TOKEN environment variable itself.
             self.agent = HfAgent(
-                model_id_or_path,  # <<<--- First argument, NO 'llm_endpoint=' keyword
-                additional_tools=self.actual_tools_for_agent,
-                hf_token=hf_auth_token  # HfAgent uses this for its internal LLMProxy
+                model_id_or_path,  # Pass model ID string as the first positional argument
+                additional_tools=self.actual_tools_for_agent
             )
             print(f"HfAgent successfully instantiated for model {model_id_or_path}.")
 
@@ -239,7 +234,6 @@ class HfAgentWrapper:
 
         except Exception as e:
             print(f"CRITICAL ERROR: Failed to initialize HfAgent: {e}")
-            # import traceback # Already imported at the top of your file
             print("Full traceback of HfAgent initialization error:")
             traceback.print_exc() 
             raise RuntimeError(f"HfAgent initialization failed: {e}") from e
@@ -253,7 +247,6 @@ class HfAgentWrapper:
             return str(answer)
         except Exception as e:
             print(f"ERROR: HfAgent execution failed for question '{question[:50]}...': {e}")
-            # import traceback # Already imported at the top of your file
             print("Full traceback of HfAgent execution error:")
             traceback.print_exc()
             return f"Agent Error: Failed to process the question. Details: {e}"