fix: switch embedding model to gemini-embedding-001

text-embedding-004 and text-multilingual-embedding-002 are not available
through this API key. gemini-embedding-001 (768-dim, multilingual) is.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Vadym Samoilenko 2026-04-29 16:02:12 +01:00
parent be0bffe459
commit bfb3a18d65

View file

@ -18,7 +18,7 @@ from ..core.logging import get_logger
logger = get_logger(__name__)
_EMBED_MODEL = "text-embedding-004"
_EMBED_MODEL = "gemini-embedding-001"
_BATCH_SIZE = 100
_MAX_RETRIES = 3
_INITIAL_BACKOFF = 2.0