Extract response output and stream content into helper functions

Author: SrdjanLL

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

@@ -500,7 +500,9 @@ def _converse_on_success(
                 )
 
         # In case of an early stream closure, the result may not contain outputs
-        if "output" in result and "message" in result["output"]:
+        if self._stream_has_output(result) and self._stream_has_content(
+            result
+        ):
             event_logger = instrumentor_context.event_logger
             choice = _Choice.from_converse(result, capture_content)
             # this path is used by streaming apis, in that case we are already out of the span
@@ -786,7 +788,9 @@ def _handle_amazon_nova_response(
             )
 
         # In case of an early stream closure, the result may not contain outputs
-        if "output" in response_body and "message" in response_body["output"]:
+        if self._stream_has_output(response_body) and self._stream_has_content(
+            response_body
+        ):
             event_logger = instrumentor_context.event_logger
             choice = _Choice.from_converse(response_body, capture_content)
             event_logger.emit(choice.to_choice_event())
@@ -1010,3 +1014,9 @@ def on_error(
                 duration,
                 attributes=metrics_attributes,
             )
+
+    def _stream_has_output(self, response_body: dict[str, Any]):
+        return "output" in response_body
+
+    def _stream_has_content(self, response_body: dict[str, Any]):
+        return "message" in response_body["output"]