Skip to content

Commit a48b2b0

Browse files
committed
fix: update dimension for google/gemini-embedding-001 model
1 parent 1c1438d commit a48b2b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/shared/embeddingModels.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ export const EMBEDDING_MODEL_PROFILES: EmbeddingModelProfiles = {
6161
// Cohere models
6262
"cohere/embed-v4.0": { dimension: 1024, scoreThreshold: 0.4 },
6363
// Google models
64-
"google/gemini-embedding-001": { dimension: 768, scoreThreshold: 0.4 },
64+
"google/gemini-embedding-001": { dimension: 3072, scoreThreshold: 0.4 },
6565
"google/text-embedding-005": { dimension: 768, scoreThreshold: 0.4 },
6666
"google/text-multilingual-embedding-002": { dimension: 768, scoreThreshold: 0.4 },
6767
// Amazon models

0 commit comments

Comments
 (0)