src/llm/__init__.py

from .base import LLMClient
from .mock_client import MockLLMClient
from .litellm_client import LiteLLMClient

def get_llm_client(provider: str, model_name: str, **kwargs) -> LLMClient:
    """
    Factory function to get the appropriate LLMClient.
    All providers (except mock) are now routed through LiteLLMClient.
    """
    if provider.lower() == "mock":
        return MockLLMClient(**kwargs)
    
    return LiteLLMClient(model_name=model_name, provider=provider, **kwargs)