config.py

from __future__ import annotations

import os
import uuid
from typing import Any

from pydantic import BaseModel, Field, model_validator
from typing_extensions import Self

from ..environment import HaiAgentsEnvironment

API_KEY_ENV_VAR = "HAI_API_KEY"
BASE_URL_ENV_VAR = "HAI_API_BASE_URL"
AUTO_SIDECAR_ENV_VAR = "HAI_AUTO_SIDECAR"

DEFAULT_BASE_URL = HaiAgentsEnvironment.EU.value
KIND_TO_CAPABILITY = {"web": "browser", "desktop": "desktop"}
CAPABILITIES = frozenset(KIND_TO_CAPABILITY.values())


def auto_sidecars_enabled() -> bool:
    return os.getenv(AUTO_SIDECAR_ENV_VAR, "1").strip().lower() not in {"0", "false", "no"}


def session_id_from_environment_id(environment_id: str, api_key: str, capability: str) -> str:
    return str(uuid.uuid5(uuid.NAMESPACE_DNS, f"{api_key}.{environment_id}.{capability}"))


class SidecarConfig(BaseModel):
    capability: str
    environment_id: str
    api_key: str = Field(default_factory=lambda: os.getenv(API_KEY_ENV_VAR, ""))
    base_url: str = Field(default_factory=lambda: os.getenv(BASE_URL_ENV_VAR) or DEFAULT_BASE_URL)
    session_id: str = ""
    driver_options: dict[str, Any] = Field(default_factory=dict)

    @model_validator(mode="after")
    def _resolve_defaults(self) -> Self:
        if self.capability not in CAPABILITIES:
            raise ValueError(f"unknown capability {self.capability!r}; expected one of {sorted(CAPABILITIES)}")
        if not self.api_key:
            raise ValueError(f"api_key is required (or set {API_KEY_ENV_VAR})")
        if not self.session_id:
            self.session_id = session_id_from_environment_id(self.environment_id, self.api_key, self.capability)
        return self