config.py

"""
Centralised, cached access to runtime configuration.
"""

from functools import lru_cache
from importlib import resources
import json
import os
from pathlib import Path
import yaml
from dotenv import load_dotenv


@lru_cache(maxsize=None)
def _load_prompt_template() -> str:
    with open("prompts/router_prompt.yml", "r") as fh:
        return yaml.safe_load(fh)["template"]


@lru_cache(maxsize=None)
def _load_mcp_configs() -> dict:
    cfg_path = Path(
        os.getenv(
            "MCP_CONFIG_PATH",
            Path(__file__).resolve().parent / "settings" / "mcp_config.json",
        )
    )
    with cfg_path.open() as fh:
        return json.load(fh)["mcpServers"]


load_dotenv()

PROMPT_TEMPLATE: str = _load_prompt_template()
MCP_CONFIGS: dict = _load_mcp_configs()
MODEL_API_KEY: str = os.getenv("MODEL_API_KEY")
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
SMITHERY_API_KEY = os.getenv("SMITHERY_API_KEY")
MCP_CONFIGS["healthcare-mcp-public"]["args"].append(SMITHERY_API_KEY)
MCP_CONFIGS["healthcare-mcp-public"]["url"] = MCP_CONFIGS["healthcare-mcp-public"]["url"].format(api_key=SMITHERY_API_KEY)