feat: Enhance usage tracking across response handlers

2026-02-03 00:25:23 +07:00
parent c1e88a9d01
commit a33924b336
8 changed files with 299 additions and 66 deletions
--- a/open-sse/translator/response/kiro-to-openai.js
+++ b/open-sse/translator/response/kiro-to-openai.js
@@ -17,9 +17,6 @@ export function convertKiroToOpenAI(chunk, state) {
  if (chunk.object === "chat.completion.chunk" && chunk.choices) {
    return chunk;
  }
-
-
-  console.log("chunk", chunk);
  
  // Handle string chunk (raw SSE data)
  let data = chunk;
@@ -161,6 +158,11 @@ export function convertKiroToOpenAI(chunk, state) {
      }]
    };

+    // Include usage in final chunk if available
+    if (state.usage && typeof state.usage === "object") {
+      openaiChunk.usage = state.usage;
+    }
+
    return openaiChunk;
  }