openai - catch usage payload

pgrayy · pgrayy · commit a76b2331301c · 2025-05-21T14:10:30.000Z
diff --git a/src/strands/models/openai.py b/src/strands/models/openai.py
@@ -113,6 +113,6 @@ def stream(self, request: dict[str, Any]) -> Iterable[dict[str, Any]]:
 
         yield {"chunk_type": "message_stop", "data": choice.finish_reason}
 
-        event = next(response)
-        if hasattr(event, "usage"):
-            yield {"chunk_type": "metadata", "data": event.usage}
+        for event in response:
+            if hasattr(event, "usage"):
+                yield {"chunk_type": "metadata", "data": event.usage}
