fix(sagemaker): Improve _handle_call to safely parse JSON, CSV, and byte inputs (#2963)

Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
Co-authored-by: Nir Gazit <[email protected]>

Author: 3 people

packages/opentelemetry-instrumentation-sagemaker/opentelemetry/instrumentation/sagemaker/__init__.py

@@ -145,16 +145,31 @@ def stream_done(response_body):
     response["Body"] = StreamingWrapper(response["Body"], stream_done)
 
 
+def _try_parse_json(value):
+    """Try to decode JSON if it's a string or bytes; fallback to raw value."""
+    try:
+        if isinstance(value, bytes):
+            value = value.decode("utf-8").strip()
+        if isinstance(value, str):
+            return json.loads(value)
+        return value
+    except (json.JSONDecodeError, UnicodeDecodeError):
+        return str(value)
+
+
 @dont_throw
 def _handle_call(span, kwargs, response):
     response["Body"] = ReusableStreamingBody(
         response["Body"]._raw_stream, response["Body"]._content_length
     )
-    request_body = json.loads(kwargs.get("Body"))
-    response_body = json.loads(response.get("Body").read())
 
-    endpoint_name = kwargs.get("EndpointName")
+    raw_request = kwargs.get("Body")
+    raw_response = response["Body"].read()
 
+    request_body = _try_parse_json(raw_request)
+    response_body = _try_parse_json(raw_response)
+
+    endpoint_name = kwargs.get("EndpointName")
     _set_span_attribute(span, SpanAttributes.LLM_REQUEST_MODEL, endpoint_name)
     _set_span_attribute(
         span, SpanAttributes.TRACELOOP_ENTITY_INPUT, json.dumps(request_body)