compatible_handler.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531
  1. package relay
  2. import (
  3. "bytes"
  4. "fmt"
  5. "io"
  6. "net/http"
  7. "strings"
  8. "time"
  9. "github.com/QuantumNous/new-api/common"
  10. "github.com/QuantumNous/new-api/constant"
  11. "github.com/QuantumNous/new-api/dto"
  12. "github.com/QuantumNous/new-api/logger"
  13. "github.com/QuantumNous/new-api/model"
  14. relaycommon "github.com/QuantumNous/new-api/relay/common"
  15. relayconstant "github.com/QuantumNous/new-api/relay/constant"
  16. "github.com/QuantumNous/new-api/relay/helper"
  17. "github.com/QuantumNous/new-api/service"
  18. "github.com/QuantumNous/new-api/setting/model_setting"
  19. "github.com/QuantumNous/new-api/setting/operation_setting"
  20. "github.com/QuantumNous/new-api/setting/ratio_setting"
  21. "github.com/QuantumNous/new-api/types"
  22. "github.com/shopspring/decimal"
  23. "github.com/gin-gonic/gin"
  24. )
  25. func TextHelper(c *gin.Context, info *relaycommon.RelayInfo) (newAPIError *types.NewAPIError) {
  26. info.InitChannelMeta(c)
  27. textReq, ok := info.Request.(*dto.GeneralOpenAIRequest)
  28. if !ok {
  29. return types.NewErrorWithStatusCode(fmt.Errorf("invalid request type, expected dto.GeneralOpenAIRequest, got %T", info.Request), types.ErrorCodeInvalidRequest, http.StatusBadRequest, types.ErrOptionWithSkipRetry())
  30. }
  31. request, err := common.DeepCopy(textReq)
  32. if err != nil {
  33. return types.NewError(fmt.Errorf("failed to copy request to GeneralOpenAIRequest: %w", err), types.ErrorCodeInvalidRequest, types.ErrOptionWithSkipRetry())
  34. }
  35. if request.WebSearchOptions != nil {
  36. c.Set("chat_completion_web_search_context_size", request.WebSearchOptions.SearchContextSize)
  37. }
  38. err = helper.ModelMappedHelper(c, info, request)
  39. if err != nil {
  40. return types.NewError(err, types.ErrorCodeChannelModelMappedError, types.ErrOptionWithSkipRetry())
  41. }
  42. includeUsage := true
  43. // 判断用户是否需要返回使用情况
  44. if request.StreamOptions != nil {
  45. includeUsage = request.StreamOptions.IncludeUsage
  46. }
  47. // 如果不支持StreamOptions,将StreamOptions设置为nil
  48. if !info.SupportStreamOptions || !request.Stream {
  49. request.StreamOptions = nil
  50. } else {
  51. // 如果支持StreamOptions,且请求中没有设置StreamOptions,根据配置文件设置StreamOptions
  52. if constant.ForceStreamOption {
  53. request.StreamOptions = &dto.StreamOptions{
  54. IncludeUsage: true,
  55. }
  56. }
  57. }
  58. info.ShouldIncludeUsage = includeUsage
  59. adaptor := GetAdaptor(info.ApiType)
  60. if adaptor == nil {
  61. return types.NewError(fmt.Errorf("invalid api type: %d", info.ApiType), types.ErrorCodeInvalidApiType, types.ErrOptionWithSkipRetry())
  62. }
  63. adaptor.Init(info)
  64. passThroughGlobal := model_setting.GetGlobalSettings().PassThroughRequestEnabled
  65. if info.RelayMode == relayconstant.RelayModeChatCompletions &&
  66. !passThroughGlobal &&
  67. !info.ChannelSetting.PassThroughBodyEnabled &&
  68. shouldChatCompletionsViaResponses(info) {
  69. applySystemPromptIfNeeded(c, info, request)
  70. usage, newApiErr := chatCompletionsViaResponses(c, info, adaptor, request)
  71. if newApiErr != nil {
  72. return newApiErr
  73. }
  74. var containAudioTokens = usage.CompletionTokenDetails.AudioTokens > 0 || usage.PromptTokensDetails.AudioTokens > 0
  75. var containsAudioRatios = ratio_setting.ContainsAudioRatio(info.OriginModelName) || ratio_setting.ContainsAudioCompletionRatio(info.OriginModelName)
  76. if containAudioTokens && containsAudioRatios {
  77. service.PostAudioConsumeQuota(c, info, usage, "")
  78. } else {
  79. postConsumeQuota(c, info, usage)
  80. }
  81. return nil
  82. }
  83. var requestBody io.Reader
  84. if passThroughGlobal || info.ChannelSetting.PassThroughBodyEnabled {
  85. body, err := common.GetRequestBody(c)
  86. if err != nil {
  87. return types.NewErrorWithStatusCode(err, types.ErrorCodeReadRequestBodyFailed, http.StatusBadRequest, types.ErrOptionWithSkipRetry())
  88. }
  89. if common.DebugEnabled {
  90. println("requestBody: ", string(body))
  91. }
  92. requestBody = bytes.NewBuffer(body)
  93. } else {
  94. convertedRequest, err := adaptor.ConvertOpenAIRequest(c, info, request)
  95. if err != nil {
  96. return types.NewError(err, types.ErrorCodeConvertRequestFailed, types.ErrOptionWithSkipRetry())
  97. }
  98. relaycommon.AppendRequestConversionFromRequest(info, convertedRequest)
  99. if info.ChannelSetting.SystemPrompt != "" {
  100. // 如果有系统提示,则将其添加到请求中
  101. request, ok := convertedRequest.(*dto.GeneralOpenAIRequest)
  102. if ok {
  103. containSystemPrompt := false
  104. for _, message := range request.Messages {
  105. if message.Role == request.GetSystemRoleName() {
  106. containSystemPrompt = true
  107. break
  108. }
  109. }
  110. if !containSystemPrompt {
  111. // 如果没有系统提示,则添加系统提示
  112. systemMessage := dto.Message{
  113. Role: request.GetSystemRoleName(),
  114. Content: info.ChannelSetting.SystemPrompt,
  115. }
  116. request.Messages = append([]dto.Message{systemMessage}, request.Messages...)
  117. } else if info.ChannelSetting.SystemPromptOverride {
  118. common.SetContextKey(c, constant.ContextKeySystemPromptOverride, true)
  119. // 如果有系统提示,且允许覆盖,则拼接到前面
  120. for i, message := range request.Messages {
  121. if message.Role == request.GetSystemRoleName() {
  122. if message.IsStringContent() {
  123. request.Messages[i].SetStringContent(info.ChannelSetting.SystemPrompt + "\n" + message.StringContent())
  124. } else {
  125. contents := message.ParseContent()
  126. contents = append([]dto.MediaContent{
  127. {
  128. Type: dto.ContentTypeText,
  129. Text: info.ChannelSetting.SystemPrompt,
  130. },
  131. }, contents...)
  132. request.Messages[i].Content = contents
  133. }
  134. break
  135. }
  136. }
  137. }
  138. }
  139. }
  140. jsonData, err := common.Marshal(convertedRequest)
  141. if err != nil {
  142. return types.NewError(err, types.ErrorCodeJsonMarshalFailed, types.ErrOptionWithSkipRetry())
  143. }
  144. // remove disabled fields for OpenAI API
  145. jsonData, err = relaycommon.RemoveDisabledFields(jsonData, info.ChannelOtherSettings)
  146. if err != nil {
  147. return types.NewError(err, types.ErrorCodeConvertRequestFailed, types.ErrOptionWithSkipRetry())
  148. }
  149. // apply param override
  150. if len(info.ParamOverride) > 0 {
  151. jsonData, err = relaycommon.ApplyParamOverride(jsonData, info.ParamOverride, relaycommon.BuildParamOverrideContext(info))
  152. if err != nil {
  153. return types.NewError(err, types.ErrorCodeChannelParamOverrideInvalid, types.ErrOptionWithSkipRetry())
  154. }
  155. }
  156. logger.LogDebug(c, fmt.Sprintf("text request body: %s", string(jsonData)))
  157. requestBody = bytes.NewBuffer(jsonData)
  158. }
  159. var httpResp *http.Response
  160. resp, err := adaptor.DoRequest(c, info, requestBody)
  161. if err != nil {
  162. return types.NewOpenAIError(err, types.ErrorCodeDoRequestFailed, http.StatusInternalServerError)
  163. }
  164. statusCodeMappingStr := c.GetString("status_code_mapping")
  165. if resp != nil {
  166. httpResp = resp.(*http.Response)
  167. info.IsStream = info.IsStream || strings.HasPrefix(httpResp.Header.Get("Content-Type"), "text/event-stream")
  168. if httpResp.StatusCode != http.StatusOK {
  169. newApiErr := service.RelayErrorHandler(c.Request.Context(), httpResp, false)
  170. // reset status code 重置状态码
  171. service.ResetStatusCode(newApiErr, statusCodeMappingStr)
  172. return newApiErr
  173. }
  174. }
  175. usage, newApiErr := adaptor.DoResponse(c, httpResp, info)
  176. if newApiErr != nil {
  177. // reset status code 重置状态码
  178. service.ResetStatusCode(newApiErr, statusCodeMappingStr)
  179. return newApiErr
  180. }
  181. var containAudioTokens = usage.(*dto.Usage).CompletionTokenDetails.AudioTokens > 0 || usage.(*dto.Usage).PromptTokensDetails.AudioTokens > 0
  182. var containsAudioRatios = ratio_setting.ContainsAudioRatio(info.OriginModelName) || ratio_setting.ContainsAudioCompletionRatio(info.OriginModelName)
  183. if containAudioTokens && containsAudioRatios {
  184. service.PostAudioConsumeQuota(c, info, usage.(*dto.Usage), "")
  185. } else {
  186. postConsumeQuota(c, info, usage.(*dto.Usage))
  187. }
  188. return nil
  189. }
  190. func shouldChatCompletionsViaResponses(info *relaycommon.RelayInfo) bool {
  191. if info == nil {
  192. return false
  193. }
  194. if info.RelayMode != relayconstant.RelayModeChatCompletions {
  195. return false
  196. }
  197. return service.ShouldChatCompletionsUseResponsesGlobal(info.ChannelId, info.OriginModelName)
  198. }
  199. func postConsumeQuota(ctx *gin.Context, relayInfo *relaycommon.RelayInfo, usage *dto.Usage, extraContent ...string) {
  200. if usage == nil {
  201. usage = &dto.Usage{
  202. PromptTokens: relayInfo.GetEstimatePromptTokens(),
  203. CompletionTokens: 0,
  204. TotalTokens: relayInfo.GetEstimatePromptTokens(),
  205. }
  206. extraContent = append(extraContent, "上游无计费信息")
  207. }
  208. adminRejectReason := common.GetContextKeyString(ctx, constant.ContextKeyAdminRejectReason)
  209. useTimeSeconds := time.Now().Unix() - relayInfo.StartTime.Unix()
  210. promptTokens := usage.PromptTokens
  211. cacheTokens := usage.PromptTokensDetails.CachedTokens
  212. imageTokens := usage.PromptTokensDetails.ImageTokens
  213. audioTokens := usage.PromptTokensDetails.AudioTokens
  214. completionTokens := usage.CompletionTokens
  215. cachedCreationTokens := usage.PromptTokensDetails.CachedCreationTokens
  216. modelName := relayInfo.OriginModelName
  217. tokenName := ctx.GetString("token_name")
  218. completionRatio := relayInfo.PriceData.CompletionRatio
  219. cacheRatio := relayInfo.PriceData.CacheRatio
  220. imageRatio := relayInfo.PriceData.ImageRatio
  221. modelRatio := relayInfo.PriceData.ModelRatio
  222. groupRatio := relayInfo.PriceData.GroupRatioInfo.GroupRatio
  223. modelPrice := relayInfo.PriceData.ModelPrice
  224. cachedCreationRatio := relayInfo.PriceData.CacheCreationRatio
  225. // Convert values to decimal for precise calculation
  226. dPromptTokens := decimal.NewFromInt(int64(promptTokens))
  227. dCacheTokens := decimal.NewFromInt(int64(cacheTokens))
  228. dImageTokens := decimal.NewFromInt(int64(imageTokens))
  229. dAudioTokens := decimal.NewFromInt(int64(audioTokens))
  230. dCompletionTokens := decimal.NewFromInt(int64(completionTokens))
  231. dCachedCreationTokens := decimal.NewFromInt(int64(cachedCreationTokens))
  232. dCompletionRatio := decimal.NewFromFloat(completionRatio)
  233. dCacheRatio := decimal.NewFromFloat(cacheRatio)
  234. dImageRatio := decimal.NewFromFloat(imageRatio)
  235. dModelRatio := decimal.NewFromFloat(modelRatio)
  236. dGroupRatio := decimal.NewFromFloat(groupRatio)
  237. dModelPrice := decimal.NewFromFloat(modelPrice)
  238. dCachedCreationRatio := decimal.NewFromFloat(cachedCreationRatio)
  239. dQuotaPerUnit := decimal.NewFromFloat(common.QuotaPerUnit)
  240. ratio := dModelRatio.Mul(dGroupRatio)
  241. // openai web search 工具计费
  242. var dWebSearchQuota decimal.Decimal
  243. var webSearchPrice float64
  244. // response api 格式工具计费
  245. if relayInfo.ResponsesUsageInfo != nil {
  246. if webSearchTool, exists := relayInfo.ResponsesUsageInfo.BuiltInTools[dto.BuildInToolWebSearchPreview]; exists && webSearchTool.CallCount > 0 {
  247. // 计算 web search 调用的配额 (配额 = 价格 * 调用次数 / 1000 * 分组倍率)
  248. webSearchPrice = operation_setting.GetWebSearchPricePerThousand(modelName, webSearchTool.SearchContextSize)
  249. dWebSearchQuota = decimal.NewFromFloat(webSearchPrice).
  250. Mul(decimal.NewFromInt(int64(webSearchTool.CallCount))).
  251. Div(decimal.NewFromInt(1000)).Mul(dGroupRatio).Mul(dQuotaPerUnit)
  252. extraContent = append(extraContent, fmt.Sprintf("Web Search 调用 %d 次,上下文大小 %s,调用花费 %s",
  253. webSearchTool.CallCount, webSearchTool.SearchContextSize, dWebSearchQuota.String()))
  254. }
  255. } else if strings.HasSuffix(modelName, "search-preview") {
  256. // search-preview 模型不支持 response api
  257. searchContextSize := ctx.GetString("chat_completion_web_search_context_size")
  258. if searchContextSize == "" {
  259. searchContextSize = "medium"
  260. }
  261. webSearchPrice = operation_setting.GetWebSearchPricePerThousand(modelName, searchContextSize)
  262. dWebSearchQuota = decimal.NewFromFloat(webSearchPrice).
  263. Div(decimal.NewFromInt(1000)).Mul(dGroupRatio).Mul(dQuotaPerUnit)
  264. extraContent = append(extraContent, fmt.Sprintf("Web Search 调用 1 次,上下文大小 %s,调用花费 %s",
  265. searchContextSize, dWebSearchQuota.String()))
  266. }
  267. // claude web search tool 计费
  268. var dClaudeWebSearchQuota decimal.Decimal
  269. var claudeWebSearchPrice float64
  270. claudeWebSearchCallCount := ctx.GetInt("claude_web_search_requests")
  271. if claudeWebSearchCallCount > 0 {
  272. claudeWebSearchPrice = operation_setting.GetClaudeWebSearchPricePerThousand()
  273. dClaudeWebSearchQuota = decimal.NewFromFloat(claudeWebSearchPrice).
  274. Div(decimal.NewFromInt(1000)).Mul(dGroupRatio).Mul(dQuotaPerUnit).Mul(decimal.NewFromInt(int64(claudeWebSearchCallCount)))
  275. extraContent = append(extraContent, fmt.Sprintf("Claude Web Search 调用 %d 次,调用花费 %s",
  276. claudeWebSearchCallCount, dClaudeWebSearchQuota.String()))
  277. }
  278. // file search tool 计费
  279. var dFileSearchQuota decimal.Decimal
  280. var fileSearchPrice float64
  281. if relayInfo.ResponsesUsageInfo != nil {
  282. if fileSearchTool, exists := relayInfo.ResponsesUsageInfo.BuiltInTools[dto.BuildInToolFileSearch]; exists && fileSearchTool.CallCount > 0 {
  283. fileSearchPrice = operation_setting.GetFileSearchPricePerThousand()
  284. dFileSearchQuota = decimal.NewFromFloat(fileSearchPrice).
  285. Mul(decimal.NewFromInt(int64(fileSearchTool.CallCount))).
  286. Div(decimal.NewFromInt(1000)).Mul(dGroupRatio).Mul(dQuotaPerUnit)
  287. extraContent = append(extraContent, fmt.Sprintf("File Search 调用 %d 次,调用花费 %s",
  288. fileSearchTool.CallCount, dFileSearchQuota.String()))
  289. }
  290. }
  291. var dImageGenerationCallQuota decimal.Decimal
  292. var imageGenerationCallPrice float64
  293. if ctx.GetBool("image_generation_call") {
  294. imageGenerationCallPrice = operation_setting.GetGPTImage1PriceOnceCall(ctx.GetString("image_generation_call_quality"), ctx.GetString("image_generation_call_size"))
  295. dImageGenerationCallQuota = decimal.NewFromFloat(imageGenerationCallPrice).Mul(dGroupRatio).Mul(dQuotaPerUnit)
  296. extraContent = append(extraContent, fmt.Sprintf("Image Generation Call 花费 %s", dImageGenerationCallQuota.String()))
  297. }
  298. var quotaCalculateDecimal decimal.Decimal
  299. var audioInputQuota decimal.Decimal
  300. var audioInputPrice float64
  301. isClaudeUsageSemantic := relayInfo.ChannelType == constant.ChannelTypeAnthropic
  302. if !relayInfo.PriceData.UsePrice {
  303. baseTokens := dPromptTokens
  304. // 减去 cached tokens
  305. // Anthropic API 的 input_tokens 已经不包含缓存 tokens,不需要减去
  306. // OpenAI/OpenRouter 等 API 的 prompt_tokens 包含缓存 tokens,需要减去
  307. var cachedTokensWithRatio decimal.Decimal
  308. if !dCacheTokens.IsZero() {
  309. if !isClaudeUsageSemantic {
  310. baseTokens = baseTokens.Sub(dCacheTokens)
  311. }
  312. cachedTokensWithRatio = dCacheTokens.Mul(dCacheRatio)
  313. }
  314. var dCachedCreationTokensWithRatio decimal.Decimal
  315. if !dCachedCreationTokens.IsZero() {
  316. if !isClaudeUsageSemantic {
  317. baseTokens = baseTokens.Sub(dCachedCreationTokens)
  318. }
  319. dCachedCreationTokensWithRatio = dCachedCreationTokens.Mul(dCachedCreationRatio)
  320. }
  321. // 减去 image tokens
  322. var imageTokensWithRatio decimal.Decimal
  323. if !dImageTokens.IsZero() {
  324. baseTokens = baseTokens.Sub(dImageTokens)
  325. imageTokensWithRatio = dImageTokens.Mul(dImageRatio)
  326. }
  327. // 减去 Gemini audio tokens
  328. if !dAudioTokens.IsZero() {
  329. audioInputPrice = operation_setting.GetGeminiInputAudioPricePerMillionTokens(modelName)
  330. if audioInputPrice > 0 {
  331. // 重新计算 base tokens
  332. baseTokens = baseTokens.Sub(dAudioTokens)
  333. audioInputQuota = decimal.NewFromFloat(audioInputPrice).Div(decimal.NewFromInt(1000000)).Mul(dAudioTokens).Mul(dGroupRatio).Mul(dQuotaPerUnit)
  334. extraContent = append(extraContent, fmt.Sprintf("Audio Input 花费 %s", audioInputQuota.String()))
  335. }
  336. }
  337. promptQuota := baseTokens.Add(cachedTokensWithRatio).
  338. Add(imageTokensWithRatio).
  339. Add(dCachedCreationTokensWithRatio)
  340. completionQuota := dCompletionTokens.Mul(dCompletionRatio)
  341. quotaCalculateDecimal = promptQuota.Add(completionQuota).Mul(ratio)
  342. if !ratio.IsZero() && quotaCalculateDecimal.LessThanOrEqual(decimal.Zero) {
  343. quotaCalculateDecimal = decimal.NewFromInt(1)
  344. }
  345. } else {
  346. quotaCalculateDecimal = dModelPrice.Mul(dQuotaPerUnit).Mul(dGroupRatio)
  347. }
  348. // 添加 responses tools call 调用的配额
  349. quotaCalculateDecimal = quotaCalculateDecimal.Add(dWebSearchQuota)
  350. quotaCalculateDecimal = quotaCalculateDecimal.Add(dFileSearchQuota)
  351. // 添加 audio input 独立计费
  352. quotaCalculateDecimal = quotaCalculateDecimal.Add(audioInputQuota)
  353. // 添加 image generation call 计费
  354. quotaCalculateDecimal = quotaCalculateDecimal.Add(dImageGenerationCallQuota)
  355. if len(relayInfo.PriceData.OtherRatios) > 0 {
  356. for key, otherRatio := range relayInfo.PriceData.OtherRatios {
  357. dOtherRatio := decimal.NewFromFloat(otherRatio)
  358. quotaCalculateDecimal = quotaCalculateDecimal.Mul(dOtherRatio)
  359. extraContent = append(extraContent, fmt.Sprintf("其他倍率 %s: %f", key, otherRatio))
  360. }
  361. }
  362. quota := int(quotaCalculateDecimal.Round(0).IntPart())
  363. totalTokens := promptTokens + completionTokens
  364. //var logContent string
  365. // record all the consume log even if quota is 0
  366. if totalTokens == 0 {
  367. // in this case, must be some error happened
  368. // we cannot just return, because we may have to return the pre-consumed quota
  369. quota = 0
  370. extraContent = append(extraContent, "上游没有返回计费信息,无法扣费(可能是上游超时)")
  371. logger.LogError(ctx, fmt.Sprintf("total tokens is 0, cannot consume quota, userId %d, channelId %d, "+
  372. "tokenId %d, model %s, pre-consumed quota %d", relayInfo.UserId, relayInfo.ChannelId, relayInfo.TokenId, modelName, relayInfo.FinalPreConsumedQuota))
  373. } else {
  374. if !ratio.IsZero() && quota == 0 {
  375. quota = 1
  376. }
  377. model.UpdateUserUsedQuotaAndRequestCount(relayInfo.UserId, quota)
  378. model.UpdateChannelUsedQuota(relayInfo.ChannelId, quota)
  379. }
  380. quotaDelta := quota - relayInfo.FinalPreConsumedQuota
  381. //logger.LogInfo(ctx, fmt.Sprintf("request quota delta: %s", logger.FormatQuota(quotaDelta)))
  382. if quotaDelta > 0 {
  383. logger.LogInfo(ctx, fmt.Sprintf("预扣费后补扣费:%s(实际消耗:%s,预扣费:%s)",
  384. logger.FormatQuota(quotaDelta),
  385. logger.FormatQuota(quota),
  386. logger.FormatQuota(relayInfo.FinalPreConsumedQuota),
  387. ))
  388. } else if quotaDelta < 0 {
  389. logger.LogInfo(ctx, fmt.Sprintf("预扣费后返还扣费:%s(实际消耗:%s,预扣费:%s)",
  390. logger.FormatQuota(-quotaDelta),
  391. logger.FormatQuota(quota),
  392. logger.FormatQuota(relayInfo.FinalPreConsumedQuota),
  393. ))
  394. }
  395. if quotaDelta != 0 {
  396. err := service.PostConsumeQuota(relayInfo, quotaDelta, relayInfo.FinalPreConsumedQuota, true)
  397. if err != nil {
  398. logger.LogError(ctx, "error consuming token remain quota: "+err.Error())
  399. }
  400. }
  401. logModel := modelName
  402. if strings.HasPrefix(logModel, "gpt-4-gizmo") {
  403. logModel = "gpt-4-gizmo-*"
  404. extraContent = append(extraContent, fmt.Sprintf("模型 %s", modelName))
  405. }
  406. if strings.HasPrefix(logModel, "gpt-4o-gizmo") {
  407. logModel = "gpt-4o-gizmo-*"
  408. extraContent = append(extraContent, fmt.Sprintf("模型 %s", modelName))
  409. }
  410. logContent := strings.Join(extraContent, ", ")
  411. other := service.GenerateTextOtherInfo(ctx, relayInfo, modelRatio, groupRatio, completionRatio, cacheTokens, cacheRatio, modelPrice, relayInfo.PriceData.GroupRatioInfo.GroupSpecialRatio)
  412. if adminRejectReason != "" {
  413. other["reject_reason"] = adminRejectReason
  414. }
  415. // For chat-based calls to the Claude model, tagging is required. Using Claude's rendering logs, the two approaches handle input rendering differently.
  416. if isClaudeUsageSemantic {
  417. other["claude"] = true
  418. other["usage_semantic"] = "anthropic"
  419. }
  420. if imageTokens != 0 {
  421. other["image"] = true
  422. other["image_ratio"] = imageRatio
  423. other["image_output"] = imageTokens
  424. }
  425. if cachedCreationTokens != 0 {
  426. other["cache_creation_tokens"] = cachedCreationTokens
  427. other["cache_creation_ratio"] = cachedCreationRatio
  428. }
  429. if !dWebSearchQuota.IsZero() {
  430. if relayInfo.ResponsesUsageInfo != nil {
  431. if webSearchTool, exists := relayInfo.ResponsesUsageInfo.BuiltInTools[dto.BuildInToolWebSearchPreview]; exists {
  432. other["web_search"] = true
  433. other["web_search_call_count"] = webSearchTool.CallCount
  434. other["web_search_price"] = webSearchPrice
  435. }
  436. } else if strings.HasSuffix(modelName, "search-preview") {
  437. other["web_search"] = true
  438. other["web_search_call_count"] = 1
  439. other["web_search_price"] = webSearchPrice
  440. }
  441. } else if !dClaudeWebSearchQuota.IsZero() {
  442. other["web_search"] = true
  443. other["web_search_call_count"] = claudeWebSearchCallCount
  444. other["web_search_price"] = claudeWebSearchPrice
  445. }
  446. if !dFileSearchQuota.IsZero() && relayInfo.ResponsesUsageInfo != nil {
  447. if fileSearchTool, exists := relayInfo.ResponsesUsageInfo.BuiltInTools[dto.BuildInToolFileSearch]; exists {
  448. other["file_search"] = true
  449. other["file_search_call_count"] = fileSearchTool.CallCount
  450. other["file_search_price"] = fileSearchPrice
  451. }
  452. }
  453. if !audioInputQuota.IsZero() {
  454. other["audio_input_seperate_price"] = true
  455. other["audio_input_token_count"] = audioTokens
  456. other["audio_input_price"] = audioInputPrice
  457. }
  458. if !dImageGenerationCallQuota.IsZero() {
  459. other["image_generation_call"] = true
  460. other["image_generation_call_price"] = imageGenerationCallPrice
  461. }
  462. model.RecordConsumeLog(ctx, relayInfo.UserId, model.RecordConsumeLogParams{
  463. ChannelId: relayInfo.ChannelId,
  464. PromptTokens: promptTokens,
  465. CompletionTokens: completionTokens,
  466. ModelName: logModel,
  467. TokenName: tokenName,
  468. Quota: quota,
  469. Content: logContent,
  470. TokenId: relayInfo.TokenId,
  471. UseTimeSeconds: int(useTimeSeconds),
  472. IsStream: relayInfo.IsStream,
  473. Group: relayInfo.UsingGroup,
  474. Other: other,
  475. })
  476. }