Gen-AI Python Implementation

Author: Jeel Mehta

aws-opentelemetry-distro/src/amazon/opentelemetry/distro/patches/_bedrock_patches.py

@@ -363,12 +363,12 @@ def on_success(self, span: Span, result: Dict[str, Any]):
     def _handle_amazon_titan_response(self, span: Span, response_body: Dict[str, Any]):
         if "inputTextTokenCount" in response_body:
             span.set_attribute(GEN_AI_USAGE_INPUT_TOKENS, response_body["inputTextTokenCount"])
-            if "results" in response_body:
+            if "results" in response_body and response_body["results"]:
                 result = response_body["results"][0]
-            if "tokenCount" in result:
-                span.set_attribute(GEN_AI_USAGE_OUTPUT_TOKENS, result["tokenCount"])
-            if "completionReason" in result:
-                span.set_attribute(GEN_AI_RESPONSE_FINISH_REASONS, [result["completionReason"]])
+                if "tokenCount" in result:
+                    span.set_attribute(GEN_AI_USAGE_OUTPUT_TOKENS, result["tokenCount"])
+                if "completionReason" in result:
+                    span.set_attribute(GEN_AI_RESPONSE_FINISH_REASONS, [result["completionReason"]])
 
     # pylint: disable=no-self-use
     def _handle_anthropic_claude_response(self, span: Span, response_body: Dict[str, Any]):