# Copyright (c) Microsoft. All rights reserved.
# type: ignore
# Because the Bedrock client does not have typing, we are ignoring type issues in this module.
from __future__ import annotations

import asyncio
import copy
import json
import logging
import sys
from collections import deque
from collections.abc import AsyncIterable, Awaitable, Mapping, MutableMapping, Sequence
from typing import Any, ClassVar, Generic, Literal, TypedDict
from uuid import uuid4

from agent_framework import (
    BaseChatClient,
    ChatAndFunctionMiddlewareTypes,
    ChatMiddlewareLayer,
    ChatOptions,
    ChatResponse,
    ChatResponseUpdate,
    Content,
    FinishReasonLiteral,
    FunctionInvocationConfiguration,
    FunctionInvocationLayer,
    FunctionTool,
    Message,
    ResponseStream,
    UsageDetails,
    validate_tool_mode,
)
from agent_framework._settings import SecretString, load_settings
from agent_framework._telemetry import get_user_agent
from agent_framework.exceptions import ChatClientInvalidResponseException
from agent_framework.observability import ChatTelemetryLayer
from boto3.session import Session as Boto3Session
from botocore.client import BaseClient
from botocore.config import Config as BotoConfig
from botocore.exceptions import ClientError
from pydantic import BaseModel

if sys.version_info >= (3, 13):
    from typing import TypeVar  # type: ignore # pragma: no cover
else:
    from typing_extensions import TypeVar  # type: ignore # pragma: no cover
if sys.version_info >= (3, 12):
    from typing import override  # type: ignore # pragma: no cover
else:
    from typing_extensions import override  # type: ignore # pragma: no cover
if sys.version_info >= (3, 11):
    from typing import TypedDict  # type: ignore # pragma: no cover
else:
    from typing_extensions import TypedDict  # type: ignore # pragma: no cover

logger = logging.getLogger("agent_framework.bedrock")


__all__ = [
    "BedrockChatClient",
    "BedrockChatOptions",
    "BedrockGuardrailConfig",
    "BedrockSettings",
]

ResponseModelT = TypeVar("ResponseModelT", bound=BaseModel | None, default=None)


# region Bedrock Chat Options TypedDict


DEFAULT_REGION = "us-east-1"
DEFAULT_MAX_TOKENS = 1024


class BedrockGuardrailConfig(TypedDict, total=False):
    """Amazon Bedrock Guardrails configuration.

    See: https://docs.aws.amazon.com/bedrock/latest/userguide/guardrails.html
    """

    guardrailIdentifier: str
    """The identifier of the guardrail to apply."""

    guardrailVersion: str
    """The version of the guardrail to use."""

    trace: Literal["enabled", "disabled"]
    """Whether to include guardrail trace information in the response."""

    streamProcessingMode: Literal["sync", "async"]
    """How to process guardrails during streaming (sync blocks, async does not)."""


class BedrockChatOptions(ChatOptions[ResponseModelT], Generic[ResponseModelT], total=False):
    """Amazon Bedrock Converse API-specific chat options dict.

    Extends base ChatOptions with Bedrock-specific parameters.
    Bedrock uses a unified Converse API that works across multiple
    foundation models (Claude, Titan, Llama, etc.).

    See: https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_Converse.html

    Keys:
        # Inherited from ChatOptions (mapped to Bedrock):
        model: The Bedrock model identifier,
            translates to ``modelId`` in Bedrock API.
        temperature: Sampling temperature,
            translates to ``inferenceConfig.temperature``.
        top_p: Nucleus sampling parameter,
            translates to ``inferenceConfig.topP``.
        max_tokens: Maximum number of tokens to generate,
            translates to ``inferenceConfig.maxTokens``.
        stop: Stop sequences,
            translates to ``inferenceConfig.stopSequences``.
        tools: List of tools available to the model,
            translates to ``toolConfig.tools``.
        tool_choice: How the model should use tools,
            translates to ``toolConfig.toolChoice``.
        response_format: Structured output format. Accepts a Pydantic BaseModel
            subclass or an OpenAI-style dict schema
            (``{"json_schema": {"name": ..., "schema": ...}}``).
            When provided, the Converse API request includes
            ``outputConfig.textFormat`` with the schema serialized as a JSON
            string. ``ChatResponse.value`` will be populated with the parsed
            model instance. Only supported on models that support
            ``outputConfig.textFormat``. Unsupported models raise a ValueError.

        # Options not supported in Bedrock Converse API:
        seed: Not supported.
        frequency_penalty: Not supported.
        presence_penalty: Not supported.
        allow_multiple_tool_calls: Not supported (models handle parallel calls automatically).
        user: Not supported.
        store: Not supported.
        logit_bias: Not supported.
        metadata: Not supported (use additional_properties for additionalModelRequestFields).

        # Bedrock-specific options:
        guardrailConfig: Guardrails configuration for content filtering.
        performanceConfig: Performance optimization settings.
        requestMetadata: Key-value metadata for the request.
        promptVariables: Variables for prompt management (if using managed prompts).
    """

    # Bedrock-specific options
    guardrailConfig: BedrockGuardrailConfig
    """Guardrails configuration for content filtering and safety."""

    performanceConfig: dict[str, Any]
    """Performance optimization settings (e.g., latency optimization).
    See: https://docs.aws.amazon.com/bedrock/latest/userguide/inference-performance.html"""

    requestMetadata: dict[str, str]
    """Key-value metadata for the request (max 2048 characters total)."""

    promptVariables: dict[str, dict[str, str]]
    """Variables for prompt management when using managed prompts."""

    # ChatOptions fields not supported in Bedrock
    seed: None  # type: ignore[misc]
    """Not supported in Bedrock Converse API."""

    frequency_penalty: None  # type: ignore[misc]
    """Not supported in Bedrock Converse API."""

    presence_penalty: None  # type: ignore[misc]
    """Not supported in Bedrock Converse API."""

    allow_multiple_tool_calls: None  # type: ignore[misc]
    """Not supported. Bedrock models handle parallel tool calls automatically."""

    user: None  # type: ignore[misc]
    """Not supported in Bedrock Converse API."""

    store: None  # type: ignore[misc]
    """Not supported in Bedrock Converse API."""

    logit_bias: None  # type: ignore[misc]
    """Not supported in Bedrock Converse API."""


BEDROCK_OPTION_TRANSLATIONS: dict[str, str] = {
    "model": "modelId",
    "max_tokens": "maxTokens",
    "top_p": "topP",
    "stop": "stopSequences",
}
"""Maps ChatOptions keys to Bedrock Converse API parameter names."""

BedrockChatOptionsT = TypeVar("BedrockChatOptionsT", bound=TypedDict, default="BedrockChatOptions", covariant=True)  # type: ignore[valid-type]


# endregion


ROLE_MAP: dict[str, str] = {
    "user": "user",
    "assistant": "assistant",
    "system": "user",
    "tool": "user",
}

FINISH_REASON_MAP: dict[str, FinishReasonLiteral] = {
    "end_turn": "stop",
    "stop_sequence": "stop",
    "max_tokens": "length",
    "length": "length",
    "content_filtered": "content_filter",
    "tool_use": "tool_calls",
}


class BedrockSettings(TypedDict, total=False):
    """Bedrock configuration settings pulled from environment variables or .env files."""

    region: str | None
    chat_model: str | None
    access_key: SecretString | None
    secret_key: SecretString | None
    session_token: SecretString | None


class BedrockChatClient(
    FunctionInvocationLayer[BedrockChatOptionsT],
    ChatMiddlewareLayer[BedrockChatOptionsT],
    ChatTelemetryLayer[BedrockChatOptionsT],
    BaseChatClient[BedrockChatOptionsT],
    Generic[BedrockChatOptionsT],
):
    """Async chat client for Amazon Bedrock's Converse API with middleware, telemetry, and function invocation."""

    OTEL_PROVIDER_NAME: ClassVar[str] = "aws.bedrock"  # type: ignore[reportIncompatibleVariableOverride, misc]

    def __init__(
        self,
        *,
        region: str | None = None,
        model: str | None = None,
        access_key: str | None = None,
        secret_key: str | None = None,
        session_token: str | None = None,
        client: BaseClient | None = None,
        boto3_session: Boto3Session | None = None,
        additional_properties: dict[str, Any] | None = None,
        middleware: Sequence[ChatAndFunctionMiddlewareTypes] | None = None,
        function_invocation_configuration: FunctionInvocationConfiguration | None = None,
        env_file_path: str | None = None,
        env_file_encoding: str | None = None,
    ) -> None:
        """Create a Bedrock chat client and load AWS credentials.

        Args:
            region: Region to send Bedrock requests to; falls back to BEDROCK_REGION.
            model: Default model identifier; falls back to BEDROCK_CHAT_MODEL.
            access_key: Optional AWS access key for manual credential injection.
            secret_key: Optional AWS secret key paired with ``access_key``.
            session_token: Optional AWS session token for temporary credentials.
            client: Preconfigured Bedrock runtime client; when omitted a boto3 session is created.
            boto3_session: Custom boto3 session used to build the runtime client if provided.
            additional_properties: Additional properties stored on the client instance.
            middleware: Optional sequence of middlewares to include.
            function_invocation_configuration: Optional function invocation configuration
            env_file_path: Optional .env file path used by ``BedrockSettings`` to load defaults.
            env_file_encoding: Encoding for the optional .env file.

        Examples:
            .. code-block:: python

                from agent_framework.amazon import BedrockChatClient

                # Basic usage with default credentials
                client = BedrockChatClient(model="<model name>")

                # Using custom ChatOptions with type safety:
                from typing import TypedDict
                from agent_framework_bedrock import BedrockChatOptions


                class MyOptions(BedrockChatOptions, total=False):
                    my_custom_option: str


                client = BedrockChatClient[MyOptions](model="<model name>")
                response = await client.get_response("Hello", options={"my_custom_option": "value"})
        """
        settings = load_settings(
            BedrockSettings,
            env_prefix="BEDROCK_",
            region=region,
            chat_model=model,
            access_key=access_key,
            secret_key=secret_key,
            session_token=session_token,
            env_file_path=env_file_path,
            env_file_encoding=env_file_encoding,
        )
        region = settings.get("region") or DEFAULT_REGION
        chat_model = settings.get("chat_model")

        if client:
            self._bedrock_client = client
        else:
            session = boto3_session or self._create_session(settings)
            self._bedrock_client = session.client(
                "bedrock-runtime",
                region_name=region,
                config=BotoConfig(user_agent_extra=get_user_agent()),
            )

        super().__init__(
            additional_properties=additional_properties,
            middleware=middleware,
            function_invocation_configuration=function_invocation_configuration,
        )
        self.model = chat_model
        self.region = region

    @staticmethod
    def _create_session(settings: BedrockSettings) -> Boto3Session:
        session_kwargs: dict[str, Any] = {"region_name": settings.get("region") or DEFAULT_REGION}
        access_key = settings.get("access_key")
        secret_key = settings.get("secret_key")
        session_token = settings.get("session_token")
        if access_key is not None and secret_key is not None:
            session_kwargs["aws_access_key_id"] = access_key.get_secret_value()
            session_kwargs["aws_secret_access_key"] = secret_key.get_secret_value()
        if session_token is not None:
            session_kwargs["aws_session_token"] = session_token.get_secret_value()
        return Boto3Session(**session_kwargs)

    def _invoke_converse(self, request: Mapping[str, Any]) -> dict[str, Any]:
        try:
            response = self._bedrock_client.converse(**request)
            if not isinstance(response, Mapping):
                raise ChatClientInvalidResponseException("Bedrock converse response must be a mapping.")
            return response
        except ClientError as e:
            error_details = e.response.get("Error", {})
            error_code = error_details.get("Code", "")
            error_message = error_details.get("Message", "")
            # "outputConfig" in error_message catches cases where Bedrock explicitly
            # rejects the outputConfig field (unsupported model). Other ValidationExceptions
            # (e.g. malformed schema shape, invalid property values) will not mention
            # "outputConfig" and will bubble up as raw ClientError without being misdiagnosed.
            if error_code == "ValidationException" and (
                "outputconfig" in error_message.lower() or "outputconfig" in str(e).lower()
            ):
                raise ValueError(
                    f"Model '{self.model}' does not support structured output via outputConfig.textFormat. "
                    "Check the model's Bedrock Converse outputConfig/textFormat support. "
                    f"AWS error Code: {error_code}. AWS error Message: {error_message}"
                ) from e
            raise

    @override
    def _inner_get_response(
        self,
        *,
        messages: Sequence[Message],
        options: Mapping[str, Any],
        stream: bool = False,
        **kwargs: Any,
    ) -> Awaitable[ChatResponse] | ResponseStream[ChatResponseUpdate, ChatResponse]:
        request = self._prepare_options(messages, options, **kwargs)

        if stream:
            # Streaming mode - simulate streaming by yielding a single update
            async def _stream() -> AsyncIterable[ChatResponseUpdate]:
                response = await asyncio.to_thread(self._invoke_converse, request)
                parsed_response = self._process_converse_response(response, options)
                contents = list(parsed_response.messages[0].contents if parsed_response.messages else [])
                if parsed_response.usage_details:
                    contents.append(Content.from_usage(usage_details=parsed_response.usage_details))  # type: ignore[arg-type]
                raw_finish_reason = (
                    parsed_response.finish_reason if isinstance(parsed_response.finish_reason, str) else None
                )
                finish_reason = self._map_finish_reason(raw_finish_reason)
                yield ChatResponseUpdate(
                    response_id=parsed_response.response_id,
                    contents=contents,
                    model=parsed_response.model,
                    finish_reason=finish_reason,
                    raw_representation=parsed_response.raw_representation,
                )

            return self._build_response_stream(_stream(), response_format=options.get("response_format"))

        # Non-streaming mode
        async def _get_response() -> ChatResponse:
            raw_response = await asyncio.to_thread(self._invoke_converse, request)
            return self._process_converse_response(raw_response, options)

        return _get_response()

    def _prepare_options(
        self,
        messages: Sequence[Message],
        options: Mapping[str, Any],
        **kwargs: Any,
    ) -> dict[str, Any]:
        model = options.get("model") or self.model
        if not model:
            raise ValueError(
                "Bedrock model is required. Set via chat options or BEDROCK_CHAT_MODEL environment variable."
            )

        system_prompts, conversation = self._prepare_bedrock_messages(messages)
        if not conversation:
            raise ValueError("At least one non-system message is required for Bedrock requests.")
        # Prepend instructions from options if they exist
        if instructions := options.get("instructions"):
            system_prompts = [{"text": instructions}, *system_prompts]

        run_options: dict[str, Any] = {
            "modelId": model,
            "messages": conversation,
            "inferenceConfig": {"maxTokens": options.get("max_tokens", DEFAULT_MAX_TOKENS)},
        }
        if system_prompts:
            run_options["system"] = system_prompts

        if (temperature := options.get("temperature")) is not None:
            run_options["inferenceConfig"]["temperature"] = temperature
        if (top_p := options.get("top_p")) is not None:
            run_options["inferenceConfig"]["topP"] = top_p
        if (stop := options.get("stop")) is not None:
            run_options["inferenceConfig"]["stopSequences"] = stop

        tool_config = self._prepare_tools(options.get("tools"))
        if tool_mode := validate_tool_mode(options.get("tool_choice")):
            if "allowed_tools" in tool_mode:
                logger.warning("allowed_tools is not supported by Bedrock; the setting will be ignored")
            match tool_mode.get("mode"):
                case "none":
                    # Bedrock doesn't support toolChoice "none".
                    # Omit toolConfig entirely so the model won't attempt tool calls.
                    tool_config = None
                case "auto":
                    if tool_config and "tools" in tool_config:
                        tool_config["toolChoice"] = {"auto": {}}
                case "required":
                    if not (tool_config and "tools" in tool_config):
                        raise ValueError(
                            "tool_choice='required' requires at least one tool to be configured, "
                            "but no tools were provided."
                        )
                    if required_name := tool_mode.get("required_function_name"):
                        tool_config["toolChoice"] = {"tool": {"name": required_name}}
                    else:
                        tool_config["toolChoice"] = {"any": {}}
                case _:
                    raise ValueError(f"Unsupported tool mode for Bedrock: {tool_mode.get('mode')}")
        if tool_config:
            run_options["toolConfig"] = tool_config

        if output_config := self._prepare_output_config(options.get("response_format")):
            run_options["outputConfig"] = output_config

        return run_options

    def _prepare_bedrock_messages(
        self, messages: Sequence[Message]
    ) -> tuple[list[dict[str, str]], list[dict[str, Any]]]:
        prompts: list[dict[str, str]] = []
        conversation: list[dict[str, Any]] = []
        pending_tool_use_ids: deque[str] = deque()
        for message in messages:
            if message.role == "system":
                text_value = message.text
                if text_value:
                    prompts.append({"text": text_value})
                continue

            content_blocks = self._convert_message_to_content_blocks(message)
            if not content_blocks:
                continue

            role = ROLE_MAP.get(message.role, "user")
            if role == "assistant":
                pending_tool_use_ids = deque(
                    block["toolUse"]["toolUseId"]
                    for block in content_blocks
                    if isinstance(block, MutableMapping) and "toolUse" in block
                )
            elif message.role == "tool":
                content_blocks = self._align_tool_results_with_pending(content_blocks, pending_tool_use_ids)
                pending_tool_use_ids.clear()
                if not content_blocks:
                    continue
            else:
                pending_tool_use_ids.clear()

            conversation.append({"role": role, "content": content_blocks})

        return prompts, conversation

    def _align_tool_results_with_pending(
        self, content_blocks: list[dict[str, Any]], pending_tool_use_ids: deque[str]
    ) -> list[dict[str, Any]]:
        if not content_blocks:
            return content_blocks
        if not pending_tool_use_ids:
            # No pending tool calls; drop toolResult blocks to avoid Bedrock validation errors
            return [
                block for block in content_blocks if not (isinstance(block, MutableMapping) and "toolResult" in block)
            ]

        aligned_blocks: list[dict[str, Any]] = []
        pending = deque(pending_tool_use_ids)
        for block in content_blocks:
            if not isinstance(block, MutableMapping):
                aligned_blocks.append(block)
                continue
            tool_result = block.get("toolResult")
            if not tool_result:
                aligned_blocks.append(block)
                continue
            if not pending:
                logger.debug("Dropping extra tool result block due to missing pending tool uses: %s", block)
                continue
            tool_use_id = tool_result.get("toolUseId")
            if tool_use_id:
                try:
                    pending.remove(tool_use_id)
                except ValueError:
                    logger.debug("Tool result references unknown toolUseId '%s'. Dropping block.", tool_use_id)
                    continue
            else:
                tool_result["toolUseId"] = pending.popleft()
            aligned_blocks.append(block)

        return aligned_blocks

    def _convert_message_to_content_blocks(self, message: Message) -> list[dict[str, Any]]:
        blocks: list[dict[str, Any]] = []
        for content in message.contents:
            block = self._convert_content_to_bedrock_block(content)
            if block is None:
                logger.debug("Skipping unsupported content type for Bedrock: %s", type(content))
                continue
            blocks.append(block)
        return blocks

    def _convert_content_to_bedrock_block(self, content: Content) -> dict[str, Any] | None:
        match content.type:
            case "text":
                return {"text": content.text}
            case "function_call":
                arguments = content.parse_arguments() or {}
                return {
                    "toolUse": {
                        "toolUseId": content.call_id or self._generate_tool_call_id(),
                        "name": content.name,
                        "input": arguments,
                    }
                }
            case "function_result":
                if content.items:
                    text_parts = [item.text or "" for item in content.items if item.type == "text"]
                    rich_items = [item for item in content.items if item.type in ("data", "uri")]
                    if rich_items:
                        logger.warning(
                            "Bedrock does not support rich content (images, audio) in tool results. "
                            "Rich content items will be omitted."
                        )
                    tool_result_text = "\n".join(text_parts) if text_parts else ""
                    tool_result_blocks = self._convert_tool_result_to_blocks(tool_result_text)
                else:
                    tool_result_blocks = self._convert_tool_result_to_blocks(content.result)
                tool_result_block = {
                    "toolResult": {
                        "toolUseId": content.call_id,
                        "content": tool_result_blocks,
                        "status": "error" if content.exception else "success",
                    }
                }
                if content.exception:
                    tool_result = tool_result_block["toolResult"]
                    existing_content = tool_result.get("content")
                    content_list: list[dict[str, Any]]
                    if isinstance(existing_content, list):
                        content_list = existing_content
                    else:
                        content_list = []
                        tool_result["content"] = content_list
                    content_list.append({"text": str(content.exception)})
                return tool_result_block
            case _:
                # Bedrock does not support other content types at this time
                pass
        return None

    def _convert_tool_result_to_blocks(self, result: Any) -> list[dict[str, Any]]:
        if isinstance(result, str):
            prepared_result = result
        else:
            parsed = FunctionTool.parse_result(result)
            text_parts = [c.text or "" for c in parsed if c.type == "text"]
            prepared_result = "\n".join(text_parts) if text_parts else str(result)
        try:
            parsed_result: object = json.loads(prepared_result)
        except json.JSONDecodeError:
            return [{"text": prepared_result}]

        return self._convert_prepared_tool_result_to_blocks(parsed_result)

    def _convert_prepared_tool_result_to_blocks(self, value: object) -> list[dict[str, Any]]:
        if isinstance(value, Sequence) and not isinstance(value, (str, bytes, bytearray)):
            blocks: list[dict[str, Any]] = []
            for item in value:
                blocks.extend(self._convert_prepared_tool_result_to_blocks(item))
            return blocks or [{"text": ""}]
        return [self._normalize_tool_result_value(value)]

    def _normalize_tool_result_value(self, value: object) -> dict[str, Any]:
        if isinstance(value, dict):
            return {"json": value}
        if isinstance(value, Sequence) and not isinstance(value, (str, bytes, bytearray)):
            return {"json": [item for item in value]}
        if isinstance(value, str):
            return {"text": value}
        if isinstance(value, (int, float, bool)) or value is None:
            return {"json": value}
        if isinstance(value, Content) and value.type == "text":
            return {"text": value.text}
        if hasattr(value, "to_dict"):
            try:
                return {"json": value.to_dict()}  # type: ignore[call-arg]
            except Exception:  # pragma: no cover - defensive
                return {"text": str(value)}
        return {"text": str(value)}

    def _prepare_tools(self, tools: list[FunctionTool | MutableMapping[str, Any]] | None) -> dict[str, Any] | None:
        converted: list[dict[str, Any]] = []
        if not tools:
            return None
        for tool in tools:
            if isinstance(tool, MutableMapping):
                converted.append(dict(tool))
                continue
            if isinstance(tool, FunctionTool):
                converted.append({
                    "toolSpec": {
                        "name": tool.name,
                        "description": tool.description or "",
                        "inputSchema": {"json": tool.parameters()},
                    }
                })
                continue
            logger.debug("Ignoring unsupported tool type for Bedrock: %s", type(tool))
        return {"tools": converted} if converted else None

    @staticmethod
    def _generate_tool_call_id() -> str:
        return f"tool-call-{uuid4().hex}"

    def _process_converse_response(
        self, response: dict[str, Any], options: Mapping[str, Any] | None = None
    ) -> ChatResponse:
        """Convert Bedrock Converse API response to ChatResponse."""
        output = response.get("output") or {}
        message = output.get("message") or {}
        content_blocks = message.get("content") or []
        contents = self._parse_message_contents(content_blocks)
        chat_message = Message(role="assistant", contents=contents, raw_representation=message)
        usage_source = response.get("usage") or output.get("usage")
        usage_details = self._parse_usage(usage_source)
        finish_reason = self._map_finish_reason(output.get("completionReason") or response.get("stopReason"))
        response_id = response.get("responseId") or message.get("id")
        model = response.get("modelId") or output.get("modelId") or self.model
        return ChatResponse(
            response_id=response_id,
            messages=[chat_message],
            usage_details=usage_details,
            model=model,
            finish_reason=finish_reason,
            response_format=options.get("response_format") if options else None,
            raw_representation=response,
        )

    def _parse_usage(self, usage: dict[str, Any] | None) -> UsageDetails | None:
        if not usage:
            return None
        details: UsageDetails = {}
        if (input_tokens := usage.get("inputTokens")) is not None:
            details["input_token_count"] = input_tokens
        if (output_tokens := usage.get("outputTokens")) is not None:
            details["output_token_count"] = output_tokens
        if (total_tokens := usage.get("totalTokens")) is not None:
            details["total_token_count"] = total_tokens
        return details

    def _parse_message_contents(self, content_blocks: Sequence[dict[str, Any]]) -> list[Any]:
        contents: list[Any] = []
        for block in content_blocks:
            if text_value := block.get("text"):
                contents.append(Content.from_text(text=text_value, raw_representation=block))
                continue
            if (json_value := block.get("json")) is not None:
                contents.append(Content.from_text(text=json.dumps(json_value), raw_representation=block))
                continue
            tool_use_value = block.get("toolUse")
            tool_use = (
                tool_use_value
                if isinstance(tool_use_value, dict)
                else dict(tool_use_value)
                if isinstance(tool_use_value, Mapping)
                else None
            )
            if tool_use is not None:
                tool_name_value = tool_use.get("name")
                tool_name = tool_name_value if isinstance(tool_name_value, str) else None
                if not tool_name:
                    raise ChatClientInvalidResponseException(
                        "Bedrock response missing required tool name in toolUse block."
                    )
                tool_use_id = tool_use.get("toolUseId")
                contents.append(
                    Content.from_function_call(
                        call_id=tool_use_id if isinstance(tool_use_id, str) else self._generate_tool_call_id(),
                        name=tool_name,
                        arguments=tool_use.get("input"),
                        raw_representation=block,
                    )
                )
                continue
            tool_result_value = block.get("toolResult")
            tool_result = (
                tool_result_value
                if isinstance(tool_result_value, dict)
                else dict(tool_result_value)
                if isinstance(tool_result_value, Mapping)
                else None
            )
            if tool_result is not None:
                status_value = tool_result.get("status")
                status = (status_value if isinstance(status_value, str) else "success").lower()
                exception = None
                if status not in {"success", "ok"}:
                    exception = RuntimeError(f"Bedrock tool result status: {status}")
                result_value = self._convert_bedrock_tool_result_to_value(tool_result.get("content"))
                tool_use_id = tool_result.get("toolUseId")
                contents.append(
                    Content.from_function_result(
                        call_id=tool_use_id if isinstance(tool_use_id, str) else self._generate_tool_call_id(),
                        result=result_value,
                        exception=str(exception) if exception else None,  # type: ignore[arg-type]
                        raw_representation=block,
                    )
                )
                continue
            logger.debug("Ignoring unsupported Bedrock content block: %s", block)
        return contents

    def _map_finish_reason(self, reason: str | None) -> FinishReasonLiteral | None:
        if not reason:
            return None
        return FINISH_REASON_MAP.get(reason.lower())

    def _prepare_output_config(self, response_format: Any | None) -> dict[str, Any] | None:
        """Convert response_format into the AWS Bedrock outputConfig wire format.

        Args:
            response_format: A Pydantic model class or a dict schema, or None.

        Returns:
            A dict for the Converse API ``outputConfig`` parameter, or None if
            response_format is not set.
        """
        if response_format is None:
            return None

        if isinstance(response_format, Mapping):
            if "json_schema" in response_format:
                # Shape A — OpenAI-style wrapper
                json_schema_config = response_format["json_schema"]
                schema_src = json_schema_config.get("schema", {})
                name = json_schema_config.get("name", "output_schema")
            elif "schema" in response_format:
                # Shape B — inner shape directly {"name": ..., "schema": ...}
                schema_src = response_format["schema"]
                name = response_format.get("name", "output_schema")
            else:
                # Shape C — assume entire dict is the raw schema
                logger.warning(
                    "response_format dict has no 'json_schema' or 'schema' key; "
                    "treating entire dict as raw JSON schema."
                )
                schema_src = dict(response_format)
                name = "output_schema"

            if isinstance(schema_src, str):
                schema_src = json.loads(schema_src)
            schema = copy.deepcopy(schema_src)
        else:
            if not isinstance(response_format, type) or not issubclass(response_format, BaseModel):
                raise TypeError(
                    "response_format must be None, a dict JSON schema, "
                    "or a Pydantic BaseModel subclass."
                )
            # response_format is a Pydantic model class
            schema = response_format.model_json_schema()
            name = response_format.__name__

        self._set_additional_properties_false(schema)

        json_schema: dict[str, Any] = {
            "name": name,
            "schema": json.dumps(schema),
        }

        description = getattr(response_format, "__doc__", None) if not isinstance(response_format, Mapping) else None
        if description and isinstance(description, str) and description.strip():
            json_schema["description"] = description.strip()

        return {
            "textFormat": {
                "type": "json_schema",
                "structure": {
                    "jsonSchema": json_schema
                },
            }
        }

    def _set_additional_properties_false(self, schema: dict[str, Any]) -> None:
        """Recursively set additionalProperties: false on all object types in a JSON schema.

        AWS requires strict schema enforcement. This mirrors the approach used by
        AnthropicChatClient._prepare_response_format().

        Args:
            schema: The JSON schema dict to modify in-place.
        """
        visited: set[int] = set()

        def walk(node: Any) -> None:
            if isinstance(node, dict):
                node_id = id(node)
                if node_id in visited:
                    return
                visited.add(node_id)
                if node.get("type") == "object" or (
                    "properties" in node and "type" not in node
                ):
                    existing = node.get("additionalProperties")
                    if existing is None or existing is True:
                        node["additionalProperties"] = False
                for value in node.values():
                    if isinstance(value, (dict, list)):
                        walk(value)
            elif isinstance(node, list):
                node_id = id(node)
                if node_id in visited:
                    return
                visited.add(node_id)
                for item in node:
                    if isinstance(item, (dict, list)):
                        walk(item)

        walk(schema)

    def service_url(self) -> str:
        """Returns the service URL for the Bedrock runtime in the configured AWS region.

        Returns:
            str: The Bedrock runtime service URL.
        """
        return f"https://bedrock-runtime.{self.region}.amazonaws.com"

    def _convert_bedrock_tool_result_to_value(self, content: object) -> object:
        if not content:
            return None
        if isinstance(content, Sequence) and not isinstance(content, (str, bytes, bytearray)):
            values: list[object] = []
            for item in content:
                item_dict = item if isinstance(item, dict) else dict(item) if isinstance(item, Mapping) else None
                if item_dict is not None:
                    text_value = item_dict.get("text")
                    if isinstance(text_value, str):
                        values.append(text_value)
                        continue
                    if "json" in item_dict:
                        values.append(item_dict["json"])
                        continue
                values.append(item)
            return values[0] if len(values) == 1 else values
        content_dict = content if isinstance(content, dict) else dict(content) if isinstance(content, Mapping) else None
        if content_dict is not None:
            text_value = content_dict.get("text")
            if isinstance(text_value, str):
                return text_value
            if "json" in content_dict:
                return content_dict["json"]
        return content
