Robot MCP Server

"""
Factory for creating LLM providers based on model names.
"""

import os
from typing import Optional
from .base_provider import LLMProvider


def create_llm_provider(model_name: str, api_key: str = None) -> LLMProvider:
    """
    Create an LLM provider based on the model name.
    
    Args:
        model_name: Model name (e.g., "claude-3-7-sonnet-latest", "gemini-2.5-flash", "gpt", "o-series")
        api_key: Optional API key override
        
    Returns:
        LLMProvider instance
        
    Raises:
        ValueError: If model is not supported or API key is missing
    """
    model_lower = model_name.lower()
    
    if "claude" in model_lower:
        from .claude_provider import ClaudeProvider
        return ClaudeProvider(api_key=api_key, model=model_name)
    
    elif "gemini" in model_lower:
        from .gemini_provider import GeminiProvider
        return GeminiProvider(api_key=api_key, model=model_name)
    
    elif "gpt" in model_lower or model_lower.startswith('o1') or model_lower.startswith('o3') or model_lower.startswith('o4'):
        from .openai_provider import OpenAIProvider
        return OpenAIProvider(api_key=api_key, model=model_name)
    
    else:
        raise ValueError(f"Unsupported model provider for: {model_name}")