Update aws_client.py

cohere-ai · Aug 15, 2024 · 2a52d98 · 2a52d98
1 parent 5402078
commit 2a52d98
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/cohere/aws_client.py b/src/cohere/aws_client.py
@@ -161,7 +161,8 @@ def _hook(
         response.stream = Streamer(output)
 
         # reset response object to allow for re-reading
-        del response._content
+        if hasattr(response, "_content"):
+            del response._content
         response.is_stream_consumed = False
         response.is_closed = False