From a144869ddfc4df16f3c95760a7af9a93c2659c52 Mon Sep 17 00:00:00 2001
From: Gustavo Cid <gustavo@openlayer.com>
Date: Thu, 17 Jul 2025 09:57:08 -0300
Subject: [PATCH] fix(bedrock): accept size positional argument

---
 src/openlayer/lib/integrations/bedrock_tracer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/openlayer/lib/integrations/bedrock_tracer.py b/src/openlayer/lib/integrations/bedrock_tracer.py
index 255b0e31..e1dca78c 100644
--- a/src/openlayer/lib/integrations/bedrock_tracer.py
+++ b/src/openlayer/lib/integrations/bedrock_tracer.py
@@ -156,8 +156,11 @@ def handle_non_streaming_invoke(
         )
 
     # Reset response body for return (since we read it)
+    response_bytes = json.dumps(response_data).encode("utf-8")
     response["body"] = type(
-        "MockBody", (), {"read": lambda: json.dumps(response_data).encode("utf-8")}
+        "MockBody",
+        (),
+        {"read": lambda size=-1: response_bytes[:size] if size > 0 else response_bytes},
     )()
     return response