xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-09-27 05:36:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

强制使用0613模型

Browse Source
This commit is contained in:
quzard 2023-06-17 10:03:26 +08:00
parent 9d85e8f79d
commit 735fb0c7bd
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
controller/relay.go
View File

@ -182,6 +182,12 @@ func relayHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
} else if channelType == common.ChannelTypePaLM { } else if channelType == common.ChannelTypePaLM {
err := relayPaLM(textRequest, c) err := relayPaLM(textRequest, c)
return err return err
} else {
// 强制使用0613模型
textRequest.Model = strings.TrimSuffix(textRequest.Model, "-0301")
textRequest.Model = strings.TrimSuffix(textRequest.Model, "-0314")
textRequest.Model = strings.TrimSuffix(textRequest.Model, "-0613")
textRequest.Model = textRequest.Model + "-0613"
} }
var promptTokens int var promptTokens int
switch relayMode { switch relayMode {