GithubProxy/coai
1
0
Fork 0
mirror of https://github.com/coaidev/coai.git synced 2025-05-20 05:20:15 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix import cycle

Browse Source
This commit is contained in:
Zhang Minghan 2023-10-10 06:58:27 +08:00
parent fff9fd8b06
commit 5e9426b03e
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
utils/tokenizer.go
View File

@ -1,7 +1,6 @@
package utils
import (
"chat/adapter/zhipuai"
"chat/globals"
"fmt"
"github.com/pkoukk/tiktoken-go"
@ -115,9 +114,9 @@ func CountInputToken(model string, v []globals.Message) float32 {
return 0
case globals.Claude2100k:
return float32(CountTokenPrice(v, model)) / 1000 * 0.008
case zhipuai.ChatGLMPro:
case globals.ZhiPuChatGLMPro:
return float32(CountTokenPrice(v, model)) / 1000 * 0.1
case zhipuai.ChatGLMStd:
case globals.ZhiPuChatGLMStd:
return float32(CountTokenPrice(v, model)) / 1000 * 0.05
default:
return 0
@ -140,9 +139,9 @@ func CountOutputToken(model string, t int) float32 {
return 0
case globals.Claude2100k:
return float32(t*GetWeightByModel(model)) / 1000 * 0.008
case zhipuai.ChatGLMPro:
case globals.ZhiPuChatGLMPro:
return float32(t*GetWeightByModel(model)) / 1000 * 0.1
case zhipuai.ChatGLMStd:
case globals.ZhiPuChatGLMStd:
return float32(t*GetWeightByModel(model)) / 1000 * 0.05
default:
return 0