diff --git a/relay/channel/openai/relay_responses.go b/relay/channel/openai/relay_responses.go index 45534252..c5ff6d24 100644 --- a/relay/channel/openai/relay_responses.go +++ b/relay/channel/openai/relay_responses.go @@ -103,6 +103,10 @@ func OaiResponsesStreamHandler(c *gin.Context, info *relaycommon.RelayInfo, resp // 非正常结束,使用输出文本的 token 数量 completionTokens := service.CountTextToken(tempStr, info.UpstreamModelName) usage.CompletionTokens = completionTokens + + if usage.PromptTokens == 0 { + usage.PromptTokens = info.PromptTokens + } } }