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

feat: add support for gemini-2.0-flash-thinking-exp model

Browse Source
This commit is contained in:
Jinjun Liu
2024-12-21 19:46:35 -06:00
parent 42dfcd6167
commit c4afcc8337
5 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
relay/adaptor/gemini/adaptor.go
View File

@@ -25,7 +25,8 @@ func (a *Adaptor) Init(meta *meta.Meta) {
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
defaultVersion := config.GeminiVersion
if meta.ActualModelName == "gemini-2.0-flash-exp" {
// gemini-2.0-flash-exp and gemini-2.0-flash-thinking-exp use v1beta
if meta.ActualModelName == "gemini-2.0-flash-exp" || meta.ActualModelName == "gemini-2.0-flash-thinking-exp" {
defaultVersion = "v1beta"
}