from __future__ import annotations

import importlib
import sys
import types
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

pkg = types.ModuleType("support_ops_env")
pkg.__path__ = [str(ROOT)]
sys.modules.setdefault("support_ops_env", pkg)

server_pkg = types.ModuleType("support_ops_env.server")
server_pkg.__path__ = [str(ROOT / "server")]
sys.modules.setdefault("support_ops_env.server", server_pkg)

MODULE_ALIASES = {
    "support_ops_env.models": "models",
    "support_ops_env.tasks": "tasks",
    "support_ops_env.graders": "graders",
    "support_ops_env.client": "client",
    "support_ops_env.inference": "inference",
    "support_ops_env.train": "train",
    "support_ops_env.train_action_json": "train_action_json",
    "support_ops_env.train_obs_format": "train_obs_format",
    "support_ops_env.train_reward_signal": "train_reward_signal",
    "support_ops_env.train_rollout_episode": "train_rollout_episode",
    "support_ops_env.train_system_prompt": "train_system_prompt",
    "support_ops_env.train_tool_bonuses": "train_tool_bonuses",
    "support_ops_env.train_vllm_compat": "train_vllm_compat",
    "support_ops_env.train_reward_chart": "train_reward_chart",
    "support_ops_env.server.app": "server.app",
    "support_ops_env.server.driftshield_environment": "server.driftshield_environment",
    "support_ops_env.server.support_ops_tools": "server.support_ops_tools",
    "support_ops_env.server.support_ops_environment": "server.driftshield_environment",
}

for alias, target in MODULE_ALIASES.items():
    sys.modules.setdefault(alias, importlib.import_module(target))