Просмотр исходного кода

fix: 修复claude渠道流模式计费可能异常

CaIon 2 лет назад
Родитель
Сommit
7ab6c6c303
2 измененных файлов с 7 добавлено и 3 удалено
  1. 4 0
      relay/channel/claude/relay-claude.go
  2. 3 3
      web/src/constants/channel.constants.js

+ 4 - 0
relay/channel/claude/relay-claude.go

@@ -278,6 +278,10 @@ func claudeStreamHandler(requestMode int, modelName string, promptTokens int, c
 	}
 	if requestMode == RequestModeCompletion {
 		usage = *service.ResponseText2Usage(responseText, modelName, promptTokens)
+	} else {
+		if usage.CompletionTokens == 0 {
+			usage = *service.ResponseText2Usage(responseText, modelName, usage.PromptTokens)
+		}
 	}
 	return nil, &usage
 }

+ 3 - 3
web/src/constants/channel.constants.js

@@ -1,15 +1,15 @@
 export const CHANNEL_OPTIONS = [
     {key: 1, text: 'OpenAI', value: 1, color: 'green', label: 'OpenAI'},
     {key: 2, text: 'Midjourney Proxy', value: 2, color: 'light-blue', label: 'Midjourney Proxy'},
-    {key: 14, text: 'Anthropic Claude', value: 14, color: 'black', label: 'Anthropic Claude'},
-    {key: 3, text: 'Azure OpenAI', value: 3, color: 'olive', label: 'Azure OpenAI'},
+    {key: 14, text: 'Anthropic Claude', value: 14, color: 'indigo', label: 'Anthropic Claude'},
+    {key: 3, text: 'Azure OpenAI', value: 3, color: 'teal', label: 'Azure OpenAI'},
     {key: 11, text: 'Google PaLM2', value: 11, color: 'orange', label: 'Google PaLM2'},
     {key: 24, text: 'Google Gemini', value: 24, color: 'orange', label: 'Google Gemini'},
     {key: 15, text: '百度文心千帆', value: 15, color: 'blue', label: '百度文心千帆'},
     {key: 17, text: '阿里通义千问', value: 17, color: 'orange', label: '阿里通义千问'},
     {key: 18, text: '讯飞星火认知', value: 18, color: 'blue', label: '讯飞星火认知'},
     {key: 16, text: '智谱 ChatGLM', value: 16, color: 'violet', label: '智谱 ChatGLM'},
-    {key: 16, text: '智谱 GLM-4V', value: 26, color: 'green', label: '智谱 GLM-4V'},
+    {key: 16, text: '智谱 GLM-4V', value: 26, color: 'purple', label: '智谱 GLM-4V'},
     {key: 16, text: 'Moonshot', value: 25, color: 'green', label: 'Moonshot'},
     {key: 19, text: '360 智脑', value: 19, color: 'blue', label: '360 智脑'},
     {key: 23, text: '腾讯混元', value: 23, color: 'teal', label: '腾讯混元'},