Skip to content

bedrock_adapter

AWS Bedrock provider adapter implementation.

This module defines BedrockAdapter, a ProviderAdapter implementation for AWS Bedrock chat models. It validates that incoming model configuration objects are instances of BedrockModelConfig and prepares AWS authentication environment variables before chat client creation.

BedrockAdapter

Bases: ProviderAdapter

Provider adapter for AWS Bedrock chat models.

This adapter validates Bedrock-specific model configuration objects and sets AWS credential-related environment variables before constructing a chat client.

Attributes:

Name Type Description
provider_name

Name of the provider handled by this adapter.
chat_client

Chat client class used to create Bedrock chat clients.

Methods:

Name Description
validate_model_config

Validate that model_config is a BedrockModelConfig instance.
_set_env_if_allowed

Set an environment variable if permitted by the current configuration.
pre_create

Apply environment-based credential configuration before client creation.
Source code in src/mada/core/chat_clients/bedrock_adapter.py 
 22
 23
 24
 25
 26
 27
 28
 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
121
122
123
124
125
126
127
128
129
130
class BedrockAdapter(ProviderAdapter):
    """
    Provider adapter for AWS Bedrock chat models.

    This adapter validates Bedrock-specific model configuration objects and sets
    AWS credential-related environment variables before constructing a chat
    client.

    Attributes:
        provider_name:
            Name of the provider handled by this adapter.
        chat_client:
            Chat client class used to create Bedrock chat clients.

    Methods:
        validate_model_config:
            Validate that `model_config` is a `BedrockModelConfig` instance.
        _set_env_if_allowed:
            Set an environment variable if permitted by the current
            configuration.
        pre_create:
            Apply environment-based credential configuration before client
            creation.
    """

    provider_name = "aws-bedrock"
    chat_client = BedrockChatClient

    def validate_model_config(self, model_config: BaseModelConfig) -> None:
        """
        Validate that `model_config` is compatible with the Bedrock adapter.

        Args:
            model_config:
                Model configuration to validate.

        Raises:
            TypeError:
                Raised if `model_config` is not a `BedrockModelConfig`
                instance.
        """
        if not isinstance(model_config, BedrockModelConfig):
            raise TypeError(
                f"{self.provider_name} adapter requires BedrockModelConfig, "
                f"got {type(model_config).__name__}"
            )

    def _set_env_if_allowed(
        self, name: str, value: str, override: bool = False
    ) -> None:
        """
        Set an environment variable if a value is provided and overwriting is
        allowed.

        Args:
            name:
                Name of the environment variable to set.
            value:
                Value to assign to `name`.
            override:
                Whether an existing conflicting environment variable may be
                overwritten.

        Raises:
            RuntimeError:
                Raised if `name` is already set to a different value and
                `override` is `False`.
        """
        if not value:
            return

        existing = os.environ.get(name)
        if existing and existing != value and not override:
            raise RuntimeError(
                f"Refusing to overwrite existing environment variable '{name}'. "
                f"Set override_env_credentials=True to allow this."
            )

        os.environ[name] = value

    def pre_create(self, model_config: BedrockModelConfig) -> None:
        """
        Prepare AWS credential environment variables before client creation.

        Args:
            model_config:
                Bedrock model configuration containing credential values and
                override settings.

        Raises:
            RuntimeError:
                Raised if an existing environment variable would be overwritten
                without permission.
        """
        override = getattr(model_config, "override_env_credentials", False)

        self._set_env_if_allowed(
            "AWS_BEARER_TOKEN_BEDROCK", model_config.bearer_token, override
        )
        self._set_env_if_allowed("AWS_PROFILE", model_config.aws_profile, override)
        self._set_env_if_allowed(
            "AWS_ACCESS_KEY_ID", model_config.aws_access_key_id, override
        )
        self._set_env_if_allowed(
            "AWS_SECRET_ACCESS_KEY", model_config.aws_secret_access_key, override
        )
        self._set_env_if_allowed(
            "AWS_SESSION_TOKEN", model_config.aws_session_token, override
        )

pre_create(model_config)

Prepare AWS credential environment variables before client creation.

Parameters:

Name Type Description Default
model_config BedrockModelConfig

Bedrock model configuration containing credential values and override settings.

 required

Raises:

Type Description
RuntimeError

Raised if an existing environment variable would be overwritten without permission.
Source code in src/mada/core/chat_clients/bedrock_adapter.py 
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
def pre_create(self, model_config: BedrockModelConfig) -> None:
    """
    Prepare AWS credential environment variables before client creation.

    Args:
        model_config:
            Bedrock model configuration containing credential values and
            override settings.

    Raises:
        RuntimeError:
            Raised if an existing environment variable would be overwritten
            without permission.
    """
    override = getattr(model_config, "override_env_credentials", False)

    self._set_env_if_allowed(
        "AWS_BEARER_TOKEN_BEDROCK", model_config.bearer_token, override
    )
    self._set_env_if_allowed("AWS_PROFILE", model_config.aws_profile, override)
    self._set_env_if_allowed(
        "AWS_ACCESS_KEY_ID", model_config.aws_access_key_id, override
    )
    self._set_env_if_allowed(
        "AWS_SECRET_ACCESS_KEY", model_config.aws_secret_access_key, override
    )
    self._set_env_if_allowed(
        "AWS_SESSION_TOKEN", model_config.aws_session_token, override
    )

validate_model_config(model_config)

Validate that model_config is compatible with the Bedrock adapter.

Parameters:

Name Type Description Default
model_config BaseModelConfig

Model configuration to validate.

 required

Raises:

Type Description
TypeError

Raised if model_config is not a BedrockModelConfig instance.
Source code in src/mada/core/chat_clients/bedrock_adapter.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
def validate_model_config(self, model_config: BaseModelConfig) -> None:
    """
    Validate that `model_config` is compatible with the Bedrock adapter.

    Args:
        model_config:
            Model configuration to validate.

    Raises:
        TypeError:
            Raised if `model_config` is not a `BedrockModelConfig`
            instance.
    """
    if not isinstance(model_config, BedrockModelConfig):
        raise TypeError(
            f"{self.provider_name} adapter requires BedrockModelConfig, "
            f"got {type(model_config).__name__}"
        )