Merge pull request #2 from amirbilu/usage-to-stream

Usage to stream

Author: amirbilu

.changeset/brown-meals-decide.md

@@ -0,0 +1,5 @@
+---
+'token.js': minor
+---
+
+Support usage metrics (e.g., totalTokens) in the Bedrock provider

src/handlers/bedrock.ts

@@ -549,7 +549,15 @@ async function* createCompletionResponseStreaming(
         created,
         model,
         object: 'chat.completion.chunk',
+        ...(stream.metadata?.usage && {
+          usage: {
+            prompt_tokens: stream.metadata.usage.inputTokens,
+            completion_tokens: stream.metadata.usage.outputTokens,
+            total_tokens: stream.metadata.usage.totalTokens,
+          },
+        }),
       }
+
       yield convertedResponse
     }
   }