"""LLM Provider 注册表"""
from app.providers.base import BaseLLMProvider
from app.providers.openai_provider import OpenAIProvider
from app.providers.qwen_provider import QwenProvider

PROVIDERS = {
    "openai": OpenAIProvider,
    "qwen": QwenProvider,
}


def get_provider(
    provider_type: str,
    api_key: str,
    model: str,
    base_url: str | None = None,
) -> BaseLLMProvider:
    """根据类型获取 Provider 实例"""
    if provider_type not in PROVIDERS:
        raise ValueError(f"Unknown provider: {provider_type}")

    provider_cls = PROVIDERS[provider_type]
    return provider_cls(api_key=api_key, model=model, base_url=base_url)