From ba1665aeeafc753de973ebf513e79f7ade682097 Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Sun, 26 May 2024 14:34:16 +0200 Subject: [PATCH] Updates to the configs --- cognee/infrastructure/databases/vector/embeddings/config.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/config.py b/cognee/infrastructure/databases/vector/embeddings/config.py index a6750c767..9837fdd03 100644 --- a/cognee/infrastructure/databases/vector/embeddings/config.py +++ b/cognee/infrastructure/databases/vector/embeddings/config.py @@ -7,9 +7,9 @@ from cognee.infrastructure.databases.vector.embeddings.DefaultEmbeddingEngine im class EmbeddingConfig(BaseSettings): openai_embedding_model: str = "text-embedding-3-large" openai_embedding_dimensions: int = 3072 - litellm_embedding_model: str = "text-embedding-3-large" - litellm_embedding_dimensions: int = 3072 - embedding_engine:object = DefaultEmbeddingEngine(embedding_model=openai_embedding_model, embedding_dimensions=openai_embedding_dimensions) + litellm_embedding_model: str = "BAAI/bge-large-en-v1.5" + litellm_embedding_dimensions: int = 1024 + embedding_engine:object = DefaultEmbeddingEngine(embedding_model=litellm_embedding_model, embedding_dimensions=litellm_embedding_dimensions) model_config = SettingsConfigDict(env_file = ".env", extra = "allow")