diff --git a/agent/main.py b/agent/main.py index a7262707..2d7b27bf 100644 --- a/agent/main.py +++ b/agent/main.py @@ -9,6 +9,7 @@ import argparse import asyncio import json +import logging import os import signal import sys @@ -56,6 +57,8 @@ CLI_CONFIG_PATH = Path(__file__).parent.parent / "configs" / "cli_agent_config.json" +logger = logging.getLogger(__name__) + def _is_scheduled_hf_job_tool(tool_info: dict[str, Any]) -> bool: if tool_info.get("tool") != "hf_jobs": @@ -97,6 +100,7 @@ def _get_hf_user(token: str | None) -> str | None: return HfApi(token=token).whoami().get("name") except Exception: + logger.exception("HF whoami failed while resolving username for token") return None