fix: update PromptTokens assignment logic in relay_responses

b0f86bd8 · CaIon · 0e34de8f · b0f86bd8
Commit b0f86bd8 authored Aug 26, 2025 by CaIon
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

relay/channel/openai/relay_responses.go
+1 -1

No files found.
--- a/relay/channel/openai/relay_responses.go
+++ b/relay/channel/openai/relay_responses.go
@@ -109,7 +109,7 @@ func OaiResponsesStreamHandler(c *gin.Context, info *relaycommon.RelayInfo, resp
 	}

 	if usage.PromptTokens == 0 && usage.CompletionTokens != 0 {
-		usage.PromptTokens = usage.CompletionTokens
+		usage.PromptTokens = info.PromptTokens
 	} else {
 		usage.TotalTokens = usage.PromptTokens + usage.CompletionTokens
 	}