make mypy happy

Author: alexander-alderman-webb

sentry_sdk/integrations/boto3/_services/bedrock.py

@@ -23,7 +23,7 @@
 
 
 if TYPE_CHECKING:
-    from typing import Any, Iterator, Optional, Union
+    from typing import Any, Iterator, Mapping, Optional, Union
 
     from mypy_boto3_bedrock_runtime.type_defs import (
         ConverseResponseTypeDef,
@@ -171,7 +171,7 @@ def _transform_system_instructions(
 
 
 def _transform_messages(
-    input_messages: "Union[list[dict[str, Any]], tuple[dict[str, Any], ...]]",
+    input_messages: "Union[list[Mapping[str, Any]], tuple[Mapping[str, Any], ...]]",
 ) -> "list[InputMessage]":
     """
     Unlike other providers, AWS declares that input and output messages use the same message type.
@@ -512,6 +512,10 @@ def on_success_response(
 
             set_on_span(SPANDATA.GEN_AI_USAGE_INPUT_TOKENS, input_tokens)
 
+            set_on_span(
+                SPANDATA.GEN_AI_RESPONSE_FINISH_REASONS, [response["stopReason"]]
+            )
+
             client = sentry_sdk.get_client()
             if "message" in response["output"] and (
                 (