Skip to content

Commit cc97f80

Browse files
authored
Merge pull request #973 from Portkey-AI/fix/openrouter-streaming-tokens
chore: map usage object in openrouter stream chunk
2 parents d23db6b + dc4271c commit cc97f80

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/providers/openrouter/chatComplete.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,11 @@ interface OpenrouterStreamChunk {
7979
object: string;
8080
created: number;
8181
model: string;
82+
usage?: {
83+
prompt_tokens: number;
84+
completion_tokens: number;
85+
total_tokens: number;
86+
};
8287
choices: {
8388
delta: {
8489
role?: string | null;
@@ -170,6 +175,7 @@ export const OpenrouterChatCompleteStreamChunkTransform: (
170175
finish_reason: parsedChunk.choices[0].finish_reason,
171176
},
172177
],
178+
...(parsedChunk.usage && { usage: parsedChunk.usage }),
173179
})}` + '\n\n'
174180
);
175181
};

0 commit comments

Comments
 (0)