luulam/9router
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(open-sse): extract buildUsage helper, dedup token-details (B2)

Browse Source 
Add helpers/usageHelper.js for conditional prompt/completion token details.
Apply to gemini/codex/claude response translators; keep each provider's token
math intact. No behavior change; golden + gate: no regression.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
decolua
2026-06-13 17:18:15 +07:00
parent 34ea763d80
commit 39278e9613
5 changed files with 27 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
open-sse/translator/response/claude-to-openai.js
View File

@@ -1,6 +1,7 @@
import { register } from "../index.js";
import { FORMATS } from "../formats.js";
import { buildChunk } from "../helpers/chunkBuilder.js";
import { buildUsage } from "../helpers/usageHelper.js";
// Create OpenAI chunk helper
function createChunk(state, delta, finishReason = null) {
@@ -127,19 +128,13 @@ export function claudeToOpenAIResponse(chunk, state) {
const finalChunk = createChunk(state, {}, state.finishReason);
if (state.usage) {
finalChunk.usage = {
prompt_tokens: state.usage.prompt_tokens,
completion_tokens: state.usage.completion_tokens,
total_tokens: state.usage.total_tokens
};
const cacheRead = state.usage.cache_read_input_tokens;
const cacheCreate = state.usage.cache_creation_input_tokens;
if (cacheRead > 0 || cacheCreate > 0) {
finalChunk.usage.prompt_tokens_details = {};
if (cacheRead > 0) finalChunk.usage.prompt_tokens_details.cached_tokens = cacheRead;
if (cacheCreate > 0) finalChunk.usage.prompt_tokens_details.cache_creation_tokens = cacheCreate;
}
finalChunk.usage = buildUsage({
promptTokens: state.usage.prompt_tokens,
completionTokens: state.usage.completion_tokens,
totalTokens: state.usage.total_tokens,
cachedTokens: state.usage.cache_read_input_tokens || 0,
cacheCreationTokens: state.usage.cache_creation_input_tokens || 0
});
}
results.push(finalChunk);