adding whoami to system prompt

agent/context_manager/manager.py

@@ -8,6 +8,7 @@ from pathlib import Path
 from typing import Any
 
 import yaml
+from huggingface_hub import HfApi
 from jinja2 import Template
 from litellm import Message, acompletion
 
@@ -24,7 +25,8 @@ class ContextManager:
         prompt_file_suffix: str = "system_prompt_v2.yaml",
     ):
         self.system_prompt = self._load_system_prompt(
-            tool_specs or [], prompt_file_suffix="system_prompt_v2.yaml"
+            tool_specs or [],
+            prompt_file_suffix="system_prompt_v2.yaml",
         )
         self.max_context = max_context
         self.compact_size = int(max_context * compact_size)
@@ -58,6 +60,7 @@ class ContextManager:
             current_date=current_date,
             current_time=current_time,
             current_timezone=current_timezone,
+            hf_user_info=HfApi().whoami().get("name"),
         )
 
     def add_message(self, message: Message, token_count: int = None) -> None:

agent/core/tools.py

@@ -35,8 +35,6 @@ from agent.tools.github_read_file import (
     GITHUB_READ_FILE_TOOL_SPEC,
     github_read_file_handler,
 )
-from agent.tools.jobs_tool import HF_JOBS_TOOL_SPEC, hf_jobs_handler
-from agent.tools.plan_tool import PLAN_TOOL_SPEC, plan_tool_handler
 from agent.tools.hf_repo_files_tool import (
     HF_REPO_FILES_TOOL_SPEC,
     hf_repo_files_handler,
@@ -45,6 +43,8 @@ from agent.tools.hf_repo_git_tool import (
     HF_REPO_GIT_TOOL_SPEC,
     hf_repo_git_handler,
 )
+from agent.tools.jobs_tool import HF_JOBS_TOOL_SPEC, hf_jobs_handler
+from agent.tools.plan_tool import PLAN_TOOL_SPEC, plan_tool_handler
 
 # NOTE: Private HF repo tool disabled - replaced by hf_repo_files and hf_repo_git
 # from agent.tools.private_hf_repo_tools import (
@@ -57,7 +57,7 @@ warnings.filterwarnings(
     "ignore", category=DeprecationWarning, module="aiohttp.connector"
 )
 
-NOT_ALLOWED_TOOL_NAMES = ["hf_jobs", "hf_doc_search", "hf_doc_fetch"]
+NOT_ALLOWED_TOOL_NAMES = ["hf_jobs", "hf_doc_search", "hf_doc_fetch", "hf_whoami"]
 
 
 def convert_mcp_content_to_string(content: list) -> str:
@@ -301,8 +301,6 @@ def create_builtin_tools() -> list[ToolSpec]:
             parameters=HF_REPO_GIT_TOOL_SPEC["parameters"],
             handler=hf_repo_git_handler,
         ),
-
-
         # NOTE: Github search code tool disabled - a bit buggy
         # ToolSpec(
         #     name=GITHUB_SEARCH_CODE_TOOL_SPEC["name"],

agent/prompts/system_prompt_v2.yaml

@@ -2,6 +2,7 @@ system_prompt: |
   You are Hugging Face Agent, a skilled AI assistant for machine learning engineering with deep expertise in the Hugging Face ecosystem. You help users accomplish ML tasks (training, fine-tuning, data processing, inference, evaluation) by interacting with Hugging Face services via {{ num_tools }} specialized tools.
 
   _Current Time: **{{ current_date }} {{ current_time }} ({{ current_timezone }})**_
+  {% if hf_user_info %}_AUTHENTICATED ON HF AS: **{{ hf_user_info }}**_{% endif %}
 
   # Core Mission & Behavior
 
@@ -330,11 +331,6 @@ system_prompt: |
   - Check model size, architecture, requirements
   - Verify dataset columns, splits, size
 
-  **hf_whoami:**
-  - Check authentication status
-  - Verify token has correct permissions
-  - Use before operations requiring write access
-
   ## Execution & Storage Tools
 
   **hf_jobs:**
@@ -456,8 +452,6 @@ system_prompt: |
       hub_model_id="username/model-name",  # ← Must be set
       # ...
   )
-
-  # Verify token: hf_whoami()
   ```
 
   ### Dataset Format Mismatch