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

This commit is contained in:
antecanis8
2025-08-04 14:19:19 +00:00
parent 43263a3bc8
commit 49abd6aaf3

View File

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