config.py

from __future__ import annotations

import json
import os
import sys
from pathlib import Path
from typing import Any, Dict, List, Optional

from pydantic import BaseModel, Field


class MCPServerConfig(BaseModel):
    id: str
    transport: str = Field(default="stdio", description="stdio (demo)")
    command: List[str]
    env: Dict[str, str] = Field(default_factory=dict)
    cwd: Optional[str] = None


def _repo_root() -> Path:
    # core/mcp/config.py -> core/mcp -> core -> repo_root
    return Path(__file__).resolve().parents[2]


def default_mcp_servers() -> List[MCPServerConfig]:
    root = _repo_root()
    return [
        MCPServerConfig(
            id="acmetech",
            transport="stdio",
            command=[sys.executable, "-m", "mcp_servers.acmetech_server"],
            env={
                "PYTHONUNBUFFERED": "1",
                "MCP_SERVER_ID": "acmetech",
            },
            cwd=str(root),
        ),
        MCPServerConfig(
            id="finetune",
            transport="stdio",
            command=[sys.executable, "-m", "mcp_servers.finetune_server"],
            env={
                "PYTHONUNBUFFERED": "1",
                "MCP_SERVER_ID": "finetune",
                "FINETUNE_DB_PATH": str(root / "data" / "finetune_jobs.sqlite"),
            },
            cwd=str(root),
        ),
    ]


def load_mcp_servers() -> List[MCPServerConfig]:
    """Load MCP servers from env MCP_SERVERS (JSON array) or fall back to default demo server."""
    raw = os.getenv("MCP_SERVERS", "").strip()
    if not raw:
        return default_mcp_servers()
    try:
        data = json.loads(raw)
    except Exception as e:
        raise ValueError(f"Invalid MCP_SERVERS JSON: {e}") from e
    if not isinstance(data, list):
        raise ValueError("MCP_SERVERS must be a JSON array")
    out: List[MCPServerConfig] = []
    for item in data:
        if not isinstance(item, dict):
            raise ValueError("MCP_SERVERS array items must be objects")
        out.append(MCPServerConfig.model_validate(item))
    return out