# agents/registry.py
from pathlib import Path

PROVIDERS = {
    "gemini": {
        "name": "Google Gemini",
        "type": "gemini",
        "key": "tu GOOGLE_API_KEY aquí o desde os.getenv",
        "models": ["gemini-1.5-flash", "gemini-1.5-pro"]
    },
    "openrouter": {
        "name": "OpenRouter",
        "type": "openai_compat",
        "key": "tu OPENROUTER_API_KEY",
        "base_url": "https://openrouter.ai/api/v1",
        "headers": {
            "HTTP-Referer": "https://huggingface.co/spaces/vfven/mission-control-ui",
            "X-Title": "Mission Control AI"
        },
        "models": [
            "google/gemma-3-27b-it:free",
            "google/gemma-3-12b-it:free",
            "meta-llama/llama-3.3-70b-instruct:free",
            "mistralai/mistral-small-3.1-24b-instruct:free"
        ]
    },
    "groq": {
        "name": "Groq",
        "type": "openai_compat",
        "key": "tu GROQ_API_KEY",
        "base_url": "https://api.groq.com/openai/v1",
        "headers": {},
        "models": ["llama-3.1-70b-versatile", "gemma2-27b-it"]
    }
}

# Diccionario global de agentes (se carga una vez)
AGENT_REGISTRY = {}

def register_default_agents():
    global AGENT_REGISTRY
    AGENT_REGISTRY = {
        "manager": {
            "key": "manager",
            "name": "Manager",
            "provider": "gemini",
            "role": "... (tu prompt completo del manager aquí) ...",
            "models": ["gemini-1.5-flash"]
        },
        "backend_dev": {
            "key": "backend_dev",
            "name": "Backend",
            "provider": "openrouter",
            "role": "... (tu prompt completo) ...",
            "models": PROVIDERS["openrouter"]["models"]
        },
        # ... el resto de agentes ...
    }

# Llamar al inicio
register_default_agents()