Merge pull request #1500 from antecanis8/gemini_batchembedcontents

fix: Gemini embedding model only embeds the first text in a batch
This commit is contained in:
Calcium-Ion
2025-08-09 11:42:08 +08:00
committed by GitHub
3 changed files with 43 additions and 31 deletions

View File

@@ -216,10 +216,14 @@ type GeminiEmbeddingRequest struct {
OutputDimensionality int `json:"outputDimensionality,omitempty"`
}
type GeminiEmbeddingResponse struct {
Embedding ContentEmbedding `json:"embedding"`
type GeminiBatchEmbeddingRequest struct {
Requests []*GeminiEmbeddingRequest `json:"requests"`
}
type ContentEmbedding struct {
type GeminiEmbedding struct {
Values []float64 `json:"values"`
}
type GeminiBatchEmbeddingResponse struct {
Embeddings []*GeminiEmbedding `json:"embeddings"`
}