257 lines
8.5 KiB
Python
257 lines
8.5 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
from collections.abc import Mapping
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
from urllib.parse import urlparse
|
|
|
|
|
|
class HostAgentConfigurationError(ValueError):
|
|
"""Raised when Host Agent process configuration is invalid."""
|
|
|
|
|
|
_LOOPBACK_BIND_HOSTS = frozenset({"127.0.0.1", "localhost", "::1"})
|
|
_AI_PLANNER_TRANSPORTS = frozenset({"direct", "cloud"})
|
|
_HOST_AGENT_MODES = frozenset({"cloud", "local"})
|
|
_REMOVED_RUNTIME_SUPERVISION_SETTINGS = (
|
|
"HOST_AGENT_RUNTIME_SUPERVISED",
|
|
"HOST_AGENT_RUNTIME_HOST",
|
|
"HOST_AGENT_RUNTIME_PORT",
|
|
)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class HostAgentConfig:
|
|
control_plane_url: str = ""
|
|
mode: str = "cloud"
|
|
host_id: str = ""
|
|
token: str = field(default="", repr=False)
|
|
identity_path: Path = Path("tasks/host_identity.json")
|
|
local_account_path: Path = Path("tasks/host_local_account.json")
|
|
enrollment_managed: bool = False
|
|
display_name: str | None = None
|
|
heartbeat_interval_seconds: float = 30.0
|
|
poll_timeout_seconds: float = 20.0
|
|
retry_backoff_seconds: float = 1.0
|
|
max_retry_backoff_seconds: float = 30.0
|
|
max_retry_attempts: int = 5
|
|
console_bind_host: str = "127.0.0.1"
|
|
console_port: int = 8765
|
|
console_allow_non_loopback: bool = False
|
|
console_session_ttl_seconds: float = 43200.0
|
|
console_history_limit: int = 200
|
|
ai_planner_transport: str = "cloud"
|
|
dependency_supervisor_enabled: bool = False
|
|
appium_supervised: bool = False
|
|
appium_host: str = "127.0.0.1"
|
|
appium_port: int = 4723
|
|
dependency_restart_max_attempts: int = 5
|
|
task_progress_db_path: Path = Path("host_agent_data/task_progress.sqlite3")
|
|
task_artifact_dir: Path = Path("host_agent_data/history")
|
|
task_retention_max_count: int = 50
|
|
task_retention_max_age_days: int = 7
|
|
skill_sync_interval_seconds: float = 300.0
|
|
|
|
|
|
def load_host_agent_config(
|
|
env: Mapping[str, str] | None = None,
|
|
) -> HostAgentConfig:
|
|
values = os.environ if env is None else env
|
|
_reject_removed_runtime_supervision_settings(values)
|
|
mode = values.get("HOST_AGENT_MODE", "cloud").strip().lower()
|
|
if mode not in _HOST_AGENT_MODES:
|
|
raise HostAgentConfigurationError("HOST_AGENT_MODE must be cloud or local")
|
|
control_plane_url = (
|
|
values.get(
|
|
"HOST_AGENT_CONTROL_PLANE_URL",
|
|
"https://amcp.home.jerryyan.top" if mode == "cloud" else "",
|
|
)
|
|
.strip()
|
|
.rstrip("/")
|
|
)
|
|
parsed_url = urlparse(control_plane_url)
|
|
if mode == "cloud" and (parsed_url.scheme not in {"http", "https"} or not parsed_url.netloc):
|
|
raise HostAgentConfigurationError(
|
|
"HOST_AGENT_CONTROL_PLANE_URL must be an HTTP(S) URL"
|
|
)
|
|
|
|
identity_path = Path(
|
|
values.get("HOST_AGENT_IDENTITY_PATH", "tasks/host_identity.json").strip()
|
|
)
|
|
local_account_path = Path(
|
|
values.get(
|
|
"HOST_AGENT_LOCAL_ACCOUNT_PATH", "tasks/host_local_account.json"
|
|
).strip()
|
|
)
|
|
|
|
config = HostAgentConfig(
|
|
control_plane_url=control_plane_url,
|
|
mode=mode,
|
|
identity_path=identity_path,
|
|
local_account_path=local_account_path,
|
|
enrollment_managed=mode == "cloud",
|
|
display_name=values.get("HOST_AGENT_DISPLAY_NAME") or None,
|
|
heartbeat_interval_seconds=_positive_float(
|
|
values,
|
|
"HOST_AGENT_HEARTBEAT_INTERVAL_SECONDS",
|
|
30.0,
|
|
),
|
|
poll_timeout_seconds=_positive_float(
|
|
values,
|
|
"HOST_AGENT_POLL_TIMEOUT_SECONDS",
|
|
20.0,
|
|
),
|
|
retry_backoff_seconds=_positive_float(
|
|
values,
|
|
"HOST_AGENT_RETRY_BACKOFF_SECONDS",
|
|
1.0,
|
|
),
|
|
max_retry_backoff_seconds=_positive_float(
|
|
values,
|
|
"HOST_AGENT_MAX_RETRY_BACKOFF_SECONDS",
|
|
30.0,
|
|
),
|
|
max_retry_attempts=_positive_int(
|
|
values,
|
|
"HOST_AGENT_MAX_RETRY_ATTEMPTS",
|
|
5,
|
|
),
|
|
console_bind_host=values.get(
|
|
"HOST_AGENT_CONSOLE_BIND_HOST", "127.0.0.1"
|
|
).strip(),
|
|
console_port=_positive_int(values, "HOST_AGENT_CONSOLE_PORT", 8765),
|
|
console_allow_non_loopback=_truthy(
|
|
values, "HOST_AGENT_CONSOLE_ALLOW_NON_LOOPBACK", False
|
|
),
|
|
console_session_ttl_seconds=_positive_float(
|
|
values,
|
|
"HOST_AGENT_CONSOLE_SESSION_TTL_SECONDS",
|
|
43200.0,
|
|
),
|
|
console_history_limit=_positive_int(
|
|
values,
|
|
"HOST_AGENT_CONSOLE_HISTORY_LIMIT",
|
|
200,
|
|
),
|
|
ai_planner_transport=("direct" if mode == "local" else _parse_ai_planner_transport(values.get("AI_PLANNER_TRANSPORT"))),
|
|
dependency_supervisor_enabled=_truthy(
|
|
values,
|
|
"HOST_AGENT_DEPENDENCY_SUPERVISOR_ENABLED",
|
|
mode == "local",
|
|
),
|
|
appium_supervised=_truthy(
|
|
values, "HOST_AGENT_APPIUM_SUPERVISED", mode == "local"
|
|
),
|
|
appium_host=values.get("HOST_AGENT_APPIUM_HOST", "127.0.0.1").strip(),
|
|
appium_port=_positive_int(values, "HOST_AGENT_APPIUM_PORT", 4723),
|
|
dependency_restart_max_attempts=_positive_int(
|
|
values, "HOST_AGENT_DEPENDENCY_RESTART_MAX_ATTEMPTS", 5
|
|
),
|
|
task_progress_db_path=Path(
|
|
values.get(
|
|
"HOST_AGENT_TASK_PROGRESS_DB_PATH",
|
|
"host_agent_data/task_progress.sqlite3",
|
|
).strip()
|
|
),
|
|
task_artifact_dir=Path(
|
|
values.get(
|
|
"HOST_AGENT_TASK_ARTIFACT_DIR", "host_agent_data/history"
|
|
).strip()
|
|
),
|
|
task_retention_max_count=_positive_int(
|
|
values, "HOST_AGENT_TASK_RETENTION_MAX_COUNT", 50
|
|
),
|
|
task_retention_max_age_days=_positive_int(
|
|
values, "HOST_AGENT_TASK_RETENTION_MAX_AGE_DAYS", 7
|
|
),
|
|
skill_sync_interval_seconds=_positive_float(
|
|
values, "HOST_AGENT_SKILL_SYNC_INTERVAL_SECONDS", 300.0
|
|
),
|
|
)
|
|
if config.max_retry_backoff_seconds < config.retry_backoff_seconds:
|
|
raise HostAgentConfigurationError(
|
|
"maximum retry backoff must not be less than initial backoff"
|
|
)
|
|
if (
|
|
config.console_bind_host not in _LOOPBACK_BIND_HOSTS
|
|
and not config.console_allow_non_loopback
|
|
):
|
|
raise HostAgentConfigurationError(
|
|
"HOST_AGENT_CONSOLE_BIND_HOST must be a loopback address "
|
|
f"({', '.join(sorted(_LOOPBACK_BIND_HOSTS))}) unless "
|
|
"HOST_AGENT_CONSOLE_ALLOW_NON_LOOPBACK is set"
|
|
)
|
|
return config
|
|
|
|
|
|
def _reject_removed_runtime_supervision_settings(values: Mapping[str, str]) -> None:
|
|
configured = [
|
|
setting
|
|
for setting in _REMOVED_RUNTIME_SUPERVISION_SETTINGS
|
|
if setting in values
|
|
]
|
|
if configured:
|
|
raise HostAgentConfigurationError(
|
|
f"{', '.join(configured)} has been removed with the standalone "
|
|
"Runtime service. Use the Host Agent console for task evidence "
|
|
"and HOST_AGENT_APPIUM_SUPERVISED for optional Appium supervision."
|
|
)
|
|
|
|
|
|
def _parse_ai_planner_transport(value: str | None) -> str:
|
|
if value is None:
|
|
return "cloud"
|
|
transport = value.strip().lower()
|
|
if transport not in _AI_PLANNER_TRANSPORTS:
|
|
raise HostAgentConfigurationError(
|
|
"AI_PLANNER_TRANSPORT must be one of "
|
|
f"{', '.join(sorted(_AI_PLANNER_TRANSPORTS))}"
|
|
)
|
|
return transport
|
|
|
|
|
|
def _positive_float(
|
|
values: Mapping[str, str],
|
|
name: str,
|
|
default: float,
|
|
) -> float:
|
|
raw_value = values.get(name)
|
|
if raw_value is None:
|
|
return default
|
|
try:
|
|
value = float(raw_value)
|
|
except ValueError as exc:
|
|
raise HostAgentConfigurationError(f"{name} must be a number") from exc
|
|
if value <= 0:
|
|
raise HostAgentConfigurationError(f"{name} must be greater than zero")
|
|
return value
|
|
|
|
|
|
def _positive_int(
|
|
values: Mapping[str, str],
|
|
name: str,
|
|
default: int,
|
|
) -> int:
|
|
raw_value = values.get(name)
|
|
if raw_value is None:
|
|
return default
|
|
try:
|
|
value = int(raw_value)
|
|
except ValueError as exc:
|
|
raise HostAgentConfigurationError(f"{name} must be an integer") from exc
|
|
if value <= 0:
|
|
raise HostAgentConfigurationError(f"{name} must be greater than zero")
|
|
return value
|
|
|
|
|
|
def _truthy(
|
|
values: Mapping[str, str],
|
|
name: str,
|
|
default: bool,
|
|
) -> bool:
|
|
raw_value = values.get(name)
|
|
if raw_value is None:
|
|
return default
|
|
return raw_value.strip().lower() in {"true", "1"}
|