From 229dd3a12388dab3caa79e5f84d87fba929f87af Mon Sep 17 00:00:00 2001
From: CaIon <i@caion.me>
Date: Sat, 9 Aug 2025 17:51:49 +0800
Subject: [PATCH] feat: update relay-text to conditionally include usage based
 on StreamOptions #696

---
 relay/relay-text.go | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/relay/relay-text.go b/relay/relay-text.go
index 1e014615..50d574f3 100644
--- a/relay/relay-text.go
+++ b/relay/relay-text.go
@@ -140,10 +140,10 @@ func TextHelper(c *gin.Context) (newAPIError *types.NewAPIError) {
 			returnPreConsumedQuota(c, relayInfo, userQuota, preConsumedQuota)
 		}
 	}()
-	includeUsage := false
+	includeUsage := true
 	// 判断用户是否需要返回使用情况
-	if textRequest.StreamOptions != nil && textRequest.StreamOptions.IncludeUsage {
-		includeUsage = true
+	if textRequest.StreamOptions != nil {
+		includeUsage = textRequest.StreamOptions.IncludeUsage
 	}
 
 	// 如果不支持StreamOptions，将StreamOptions设置为nil
@@ -158,9 +158,7 @@ func TextHelper(c *gin.Context) (newAPIError *types.NewAPIError) {
 		}
 	}
 
-	if includeUsage {
-		relayInfo.ShouldIncludeUsage = true
-	}
+	relayInfo.ShouldIncludeUsage = includeUsage
 
 	adaptor := GetAdaptor(relayInfo.ApiType)
 	if adaptor == nil {