add defensive check for from_converse response

Author: liustve

instrumentation/opentelemetry-instrumentation-botocore/src/opentelemetry/instrumentation/botocore/extensions/bedrock_utils.py

@@ -528,7 +528,8 @@ def __init__(
     def from_converse(
         cls, response: dict[str, Any], capture_content: bool
     ) -> _Choice:
-        orig_message = response["output"]["message"]
+        output = response.get("output", {})
+        orig_message = output.get("message", {})
         if role := orig_message.get("role"):
             message = {"role": role}
         else:

instrumentation/opentelemetry-instrumentation-botocore/tests/test_botocore_bedrock.py

@@ -26,6 +26,7 @@
 from botocore.response import StreamingBody
 
 from opentelemetry.instrumentation.botocore.extensions.bedrock_utils import (
+    ConverseStreamWrapper,
     InvokeModelWithResponseStreamWrapper,
 )
 from opentelemetry.semconv._incubating.attributes.error_attributes import (
@@ -3049,7 +3050,21 @@ def stream_error_callback(exc, ended):
         assert isinstance(tool_block["input"], dict)
     else:
         assert "input" not in tool_block
-
+def test_converse_stream_with_malformed_response():
+    """Test that converse stream handles malformed response missing output key."""
+    def stream_done_callback(response, ended):
+        pass
+    
+    wrapper = ConverseStreamWrapper(
+        stream=mock.MagicMock(),
+        stream_done_callback=stream_done_callback,
+        model_id="amazon.nova-micro-v1:0",
+    )
+    
+    malformed_response = {"stopReason": "end_turn"}
+    result = wrapper._complete_stream(malformed_response)
+    assert result is None
+    
 
 def amazon_nova_messages():
     return [