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

feat: support gemini-2.0-flash

Browse Source
This commit is contained in:
Laisky.Cai
2025-02-06 02:34:41 +00:00
parent aa30c37e3c
commit 3e4708b77e
6 changed files with 30 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4,6 +4,7 @@ import (
"fmt"
"io"
"net/http"
"strings"
"github.com/gin-gonic/gin"
"github.com/pkg/errors"
@@ -24,13 +25,9 @@ func (a *Adaptor) Init(meta *meta.Meta) {
}
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
var defaultVersion string
switch meta.ActualModelName {
case "gemini-2.0-flash-exp",
"gemini-2.0-flash-thinking-exp-01-21":
defaultVersion := config.GeminiVersion
if strings.Contains(meta.ActualModelName, "-2.0") {
defaultVersion = "v1beta"
default:
defaultVersion = config.GeminiVersion
}
version := helper.AssignOrDefault(meta.Config.APIVersion, defaultVersion)