import os from pathlib import Path DIR_ROOT = Path(__file__).resolve().parent VAL_DEFAULTS = { "HF_HOME": "/workspace/hf_cache", "PROFILER_AGENT_PORT": "8011", "FORENSIC_AGENT_PORT": "8002", } KEYS_DEF = ("HF_HOME", "HF_HUB_OFFLINE", "PROFILER_AGENT_PORT", "FORENSIC_AGENT_PORT") def apply() -> dict: for key in (*VAL_DEFAULTS, "HF_HUB_OFFLINE"): if os.environ.get(key) == "": os.environ.pop(key) try: from dotenv import load_dotenv load_dotenv(DIR_ROOT / ".env") except Exception: pass env = os.environ for key, default in VAL_DEFAULTS.items(): if not env.get(key): env[key] = default if not env.get("HF_HUB_OFFLINE"): hub = Path(env["HF_HOME"]) / "hub" if hub.is_dir() and any(hub.glob("models--*")): env["HF_HUB_OFFLINE"] = "1" else: env.pop("HF_HUB_OFFLINE", None) return {k: env.get(k) for k in KEYS_DEF} SETTINGS = apply() if __name__ == "__main__": for k in KEYS_DEF: print(f" {k}={SETTINGS[k] if SETTINGS[k] is not None else ''}")