From d646a922eefdcce76a1f40ce79624c9d3d1285a0 Mon Sep 17 00:00:00 2001
From: CaIon <i@caion.me>
Date: Fri, 15 Aug 2025 15:55:01 +0800
Subject: [PATCH] refactor: set prompt tokens when not provided in usage

---
 relay/channel/openai/relay_responses.go | 4 ++++
 1 file changed, 4 insertions(+)

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
+			}
 		}
 	}