xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-04 15:53:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: extend model check in GetModelModalities function to include additional models

Browse Source
This commit is contained in:
Laisky.Cai
2025-03-26 03:54:07 +00:00
parent 5f6b515bb3
commit e6d70c8bf7
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
relay/adaptor/geminiv2/constants.go
View File

@@ -31,7 +31,10 @@ func GetModelModalities(model string) []string {
}
// Until 2025-03-26, the following models do not accept the responseModalities field
if model == "gemini-2.5-pro-exp-03-25" {
if model == "gemini-2.5-pro-exp-03-25" ||
model == "aqa" ||
strings.HasPrefix(model, "gemma") ||
strings.HasPrefix(model, "text-embed") {
return nil
}