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

fix: update comments for o1 model handling

Browse Source
This commit is contained in:
Laisky.Cai 2025-01-05 02:42:37 +00:00
parent d70d5656ab
commit 73a928f970
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
relay/adaptor/openai/adaptor.go
View File

@ -85,8 +85,8 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
request.StreamOptions.IncludeUsage = true request.StreamOptions.IncludeUsage = true
} }
// o1 do not support system prompt and max_tokens // o1/o1-mini/o1-preview do not support system prompt and max_tokens
if strings.HasPrefix(request.Model, "o1-mini") || strings.HasPrefix(request.Model, "o1-preview") { if strings.HasPrefix(request.Model, "o1") {
request.MaxTokens = 0 request.MaxTokens = 0
request.Messages = func(raw []model.Message) (filtered []model.Message) { request.Messages = func(raw []model.Message) (filtered []model.Message) {
for i := range raw { for i := range raw {

2
relay/billing/ratio/model.go
View File

@ -467,7 +467,7 @@ func GetCompletionRatio(name string, channelType int) float64 {
} }
return 2 return 2
} }
// including o1, o1-preview, o1-mini // including o1/o1-preview/o1-mini
if strings.HasPrefix(name, "o1") { if strings.HasPrefix(name, "o1") {
return 4 return 4
} }