Skip to content

chat_client_factory

Factory utilities for creating chat clients from provider-specific adapters.

This module exposes MADAChatClientFactory, which registers built-in provider adapters and creates agent_framework.BaseChatClient instances from a BaseModelConfig.

MADAChatClientFactory

Factory for registering provider adapters and creating chat clients.

The factory maintains a mapping of provider names to ProviderAdapter instances. Each adapter is responsible for validating configuration and constructing the provider-specific client.

Attributes:

Name Type Description
_adapters Dict[str, ProviderAdapter]

Mapping of provider names to registered provider adapters.

Methods:

Name Description
register_provider_adapter

Register a provider adapter with the factory.
get_adapter

Return the adapter registered for a provider.
create

Create a chat client from a model configuration.
Source code in src/mada/core/chat_clients/chat_client_factory.py 
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
class MADAChatClientFactory:
    """
    Factory for registering provider adapters and creating chat clients.

    The factory maintains a mapping of provider names to `ProviderAdapter`
    instances. Each adapter is responsible for validating configuration and
    constructing the provider-specific client.

    Attributes:
        _adapters:
            Mapping of provider names to registered provider adapters.

    Methods:
        register_provider_adapter:
            Register a provider adapter with the factory.
        get_adapter:
            Return the adapter registered for a provider.
        create:
            Create a chat client from a model configuration.
    """

    def __init__(self):
        """
        Initialize the factory and register built-in provider adapters.
        """
        self._adapters: Dict[str, ProviderAdapter] = {}
        self._register_builtins()

    def _register_builtins(self) -> None:
        """
        Register the default provider adapters supported by the application.
        """
        self.register_provider_adapter(OpenAIAdapter())
        self.register_provider_adapter(LivAIAdapter())
        self.register_provider_adapter(BedrockAdapter())

    def register_provider_adapter(self, adapter: ProviderAdapter) -> None:
        """
        Register a provider adapter with the factory.

        Args:
            adapter:
                Provider adapter to register, keyed by its `provider_name`
                attribute.
        """
        self._adapters[adapter.provider_name] = adapter
        LOG.debug("Registered provider adapter '%s'", adapter.provider_name)

    def get_adapter(self, provider: str) -> Optional[ProviderAdapter]:
        """
        Return the registered adapter for a provider, if available.

        Args:
            provider:
                Provider name.

        Returns:
            The matching provider adapter, or `None` if the provider is not
            registered.
        """
        return self._adapters.get(provider)

    def create(self, model_config: BaseModelConfig) -> BaseChatClient:
        """
        Create a chat client for the provider and model in `model_config`.

        Args:
            model_config:
                Model configuration containing the target provider and model
                details.

        Returns:
            A provider-specific chat client instance.

        Raises:
            ValueError:
                If no adapter is registered for the requested provider, or if
                client creation fails.
        """
        adapter = self._adapters.get(model_config.provider)
        if adapter is None:
            raise ValueError(
                f"No provider adapter registered for provider '{model_config.provider}'"
            )

        try:
            return adapter.create_client(model_config)
        except Exception as e:
            raise ValueError(
                f"Failed to create chat client for provider='{model_config.provider}', "
                f"model='{model_config.model}': {e}"
            ) from e

__init__()

Initialize the factory and register built-in provider adapters.

Source code in src/mada/core/chat_clients/chat_client_factory.py 
50
51
52
53
54
55
def __init__(self):
    """
    Initialize the factory and register built-in provider adapters.
    """
    self._adapters: Dict[str, ProviderAdapter] = {}
    self._register_builtins()

create(model_config)

Create a chat client for the provider and model in model_config.

Parameters:

Name Type Description Default
model_config BaseModelConfig

Model configuration containing the target provider and model details.

 required

Returns:

Type Description
BaseChatClient

A provider-specific chat client instance.

Raises:

Type Description
ValueError

If no adapter is registered for the requested provider, or if client creation fails.
Source code in src/mada/core/chat_clients/chat_client_factory.py 
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
def create(self, model_config: BaseModelConfig) -> BaseChatClient:
    """
    Create a chat client for the provider and model in `model_config`.

    Args:
        model_config:
            Model configuration containing the target provider and model
            details.

    Returns:
        A provider-specific chat client instance.

    Raises:
        ValueError:
            If no adapter is registered for the requested provider, or if
            client creation fails.
    """
    adapter = self._adapters.get(model_config.provider)
    if adapter is None:
        raise ValueError(
            f"No provider adapter registered for provider '{model_config.provider}'"
        )

    try:
        return adapter.create_client(model_config)
    except Exception as e:
        raise ValueError(
            f"Failed to create chat client for provider='{model_config.provider}', "
            f"model='{model_config.model}': {e}"
        ) from e

get_adapter(provider)

Return the registered adapter for a provider, if available.

Parameters:

Name Type Description Default
provider str

Provider name.

 required

Returns:

Type Description
Optional[ProviderAdapter]

The matching provider adapter, or None if the provider is not
Optional[ProviderAdapter]

registered.
Source code in src/mada/core/chat_clients/chat_client_factory.py 
77
78
79
80
81
82
83
84
85
86
87
88
89
def get_adapter(self, provider: str) -> Optional[ProviderAdapter]:
    """
    Return the registered adapter for a provider, if available.

    Args:
        provider:
            Provider name.

    Returns:
        The matching provider adapter, or `None` if the provider is not
        registered.
    """
    return self._adapters.get(provider)

register_provider_adapter(adapter)

Register a provider adapter with the factory.

Parameters:

Name Type Description Default
adapter ProviderAdapter

Provider adapter to register, keyed by its provider_name attribute.

 required
Source code in src/mada/core/chat_clients/chat_client_factory.py 
65
66
67
68
69
70
71
72
73
74
75
def register_provider_adapter(self, adapter: ProviderAdapter) -> None:
    """
    Register a provider adapter with the factory.

    Args:
        adapter:
            Provider adapter to register, keyed by its `provider_name`
            attribute.
    """
    self._adapters[adapter.provider_name] = adapter
    LOG.debug("Registered provider adapter '%s'", adapter.provider_name)