Add usage on streamed chat response (#398)

openai-php · May 27, 2024 · 769be96 · 769be96
1 parent 7ec1207
commit 769be96
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/Responses/Chat/CreateStreamedResponse.php b/src/Responses/Chat/CreateStreamedResponse.php
@@ -29,6 +29,7 @@ private function __construct(
         public readonly int $created,
         public readonly string $model,
         public readonly array $choices,
+        public readonly ?CreateResponseUsage $usage,
     ) {
     }
 
@@ -49,6 +50,7 @@ public static function from(array $attributes): self
             $attributes['created'],
             $attributes['model'],
             $choices,
+            isset($attributes['usage']) ? CreateResponseUsage::from($attributes['usage']) : null,
         );
     }
 
@@ -66,6 +68,7 @@ public function toArray(): array
                 static fn (CreateStreamedResponseChoice $result): array => $result->toArray(),
                 $this->choices,
             ),
+            'usage' => $this->usage?->toArray() ?? null,
         ];
     }
 }