diff options
| -rw-r--r-- | packages/openai-stream/src/provider.test.ts | 1 | ||||
| -rw-r--r-- | packages/openai-stream/src/stream.ts | 1 |
2 files changed, 2 insertions, 0 deletions
diff --git a/packages/openai-stream/src/provider.test.ts b/packages/openai-stream/src/provider.test.ts index 7b2d938..434c405 100644 --- a/packages/openai-stream/src/provider.test.ts +++ b/packages/openai-stream/src/provider.test.ts @@ -145,6 +145,7 @@ describe("transformBody", () => { model: "test-model", messages: [{ role: "user", content: "Hello" }], stream: true, + stream_options: { include_usage: true }, temperature: 0.5, max_tokens: 42, }); diff --git a/packages/openai-stream/src/stream.ts b/packages/openai-stream/src/stream.ts index 2916432..e801cc5 100644 --- a/packages/openai-stream/src/stream.ts +++ b/packages/openai-stream/src/stream.ts @@ -69,6 +69,7 @@ export async function* streamChat( model: opts?.model ?? config.model, messages: finalMessages, stream: true, + stream_options: { include_usage: true }, }; if (openaiTools.length > 0) { |