Skip to content

Commit 94366df

Browse files
gustavocidornelaswhoseoyster
authored andcommitted
fix(bedrock): accept size positional argument
1 parent 37c886b commit 94366df

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/openlayer/lib/integrations/bedrock_tracer.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,11 @@ def handle_non_streaming_invoke(
156156
)
157157

158158
# Reset response body for return (since we read it)
159+
response_bytes = json.dumps(response_data).encode("utf-8")
159160
response["body"] = type(
160-
"MockBody", (), {"read": lambda: json.dumps(response_data).encode("utf-8")}
161+
"MockBody",
162+
(),
163+
{"read": lambda size=-1: response_bytes[:size] if size > 0 else response_bytes},
161164
)()
162165
return response
163166

0 commit comments

Comments
 (0)