xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-05 16:53:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fixed bug for chat request's token reach the max token limit for OpenAI

Browse Source
This commit is contained in:
RockYang
2023-08-01 17:58:03 +08:00
parent 9eb8da2789
commit fdd659f393
3 changed files with 30 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/core/types/chat.go
View File

@@ -53,3 +53,10 @@ type ApiError struct {
const PromptMsg = "prompt" // prompt message
const ReplyMsg = "reply" // reply message
var ModelToTokens = map[string]int{
"gpt-3.5-turbo": 4096,
"gpt-3.5-turbo-16k": 16384,
"gpt-4": 8192,
"gpt-4-32k": 32768,
}