diff --git a/src/services/code-index/embedders/gemini.ts b/src/services/code-index/embedders/gemini.ts index e0f6f1fcfd..6f032863e4 100644 --- a/src/services/code-index/embedders/gemini.ts +++ b/src/services/code-index/embedders/gemini.ts @@ -11,7 +11,7 @@ import { TelemetryService } from "@roo-code/telemetry" * * Supported models: * - text-embedding-004 (dimension: 768) - * - gemini-embedding-001 (dimension: 768) + * - gemini-embedding-001 (dimension: 3072) */ export class GeminiEmbedder implements IEmbedder { private readonly openAICompatibleEmbedder: OpenAICompatibleEmbedder diff --git a/src/shared/embeddingModels.ts b/src/shared/embeddingModels.ts index 5cb064174b..a3cd61e659 100644 --- a/src/shared/embeddingModels.ts +++ b/src/shared/embeddingModels.ts @@ -48,7 +48,7 @@ export const EMBEDDING_MODEL_PROFILES: EmbeddingModelProfiles = { }, gemini: { "text-embedding-004": { dimension: 768 }, - "gemini-embedding-001": { dimension: 768, scoreThreshold: 0.4 }, + "gemini-embedding-001": { dimension: 3072, scoreThreshold: 0.4 }, }, mistral: { "codestral-embed-2505": { dimension: 1536, scoreThreshold: 0.4 },