oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add support for configuring output dimensionality for multiple Gemini new models

Browse Source
This commit is contained in:
antecanis8
2025-08-04 14:19:19 +00:00
parent 43263a3bc8
commit 49abd6aaf3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
relay/channel/gemini/adaptor.go
View File

@@ -173,8 +173,8 @@ func (a *Adaptor) ConvertEmbeddingRequest(c *gin.Context, info *relaycommon.Rela
// set specific parameters for different models // set specific parameters for different models
// https://ai.google.dev/api/embeddings?hl=zh-cn#method:-models.embedcontent // https://ai.google.dev/api/embeddings?hl=zh-cn#method:-models.embedcontent
switch info.UpstreamModelName { switch info.UpstreamModelName {
case "text-embedding-004": case "text-embedding-004","gemini-embedding-exp-03-07","gemini-embedding-001":
// except embedding-001 supports setting `OutputDimensionality` // Only newer models introduced after 2024 support OutputDimensionality
if request.Dimensions > 0 { if request.Dimensions > 0 {
geminiRequest["outputDimensionality"] = request.Dimensions geminiRequest["outputDimensionality"] = request.Dimensions
} }