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,
  ];
  }
 }