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.
    """
    p = provider.lower()
    if p == "mock":
        return MockLLMClient(**kwargs)

    return LiteLLMClient(provider=provider, model_name=model_name, **kwargs)