versun/claude-code
1
0
Fork 0
mirror of https://github.com/claude-code-best/claude-code.git synced 2026-06-22 00:05:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复 openai 的 cost 计算问题

Browse Source
This commit is contained in:
claude-code-best
2026-04-04 16:37:50 +08:00
parent ab7556e355
commit 462fe69d80
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/services/api/openai/index.ts
View File

@@ -11,6 +11,8 @@ import { normalizeMessagesForAPI } from '../../../utils/messages.js'
import { toolToAPISchema } from '../../../utils/api.js' import { toolToAPISchema } from '../../../utils/api.js'
import { getEmptyToolPermissionContext } from '../../../Tool.js' import { getEmptyToolPermissionContext } from '../../../Tool.js'
import { logForDebugging } from '../../../utils/debug.js' import { logForDebugging } from '../../../utils/debug.js'
import { addToTotalSessionCost } from '../../../cost-tracker.js'
import { calculateUSDCost } from '../../../utils/modelCost.js'
import type { Options } from '../claude.js' import type { Options } from '../claude.js'
import { randomUUID } from 'crypto' import { randomUUID } from 'crypto'
import { import {
@@ -189,6 +191,12 @@ export async function* queryModelOpenAI(
break break
} }
// Track cost and token usage (matching the Anthropic path in claude.ts)
if (event.type === 'message_stop' && usage.input_tokens + usage.output_tokens > 0) {
const costUSD = calculateUSDCost(openaiModel, usage as any)
addToTotalSessionCost(costUSD, usage as any, options.model)
}
// Also yield as StreamEvent for real-time display (matching Anthropic path) // Also yield as StreamEvent for real-time display (matching Anthropic path)
yield { yield {
type: 'stream_event', type: 'stream_event',