gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Updates to the configs

Browse source
This commit is contained in:
Vasilije 2024-05-26 14:34:16 +02:00
parent 499670402f
commit ba1665aeea
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/infrastructure/databases/vector/embeddings/config.py
View file

@ -7,9 +7,9 @@ from cognee.infrastructure.databases.vector.embeddings.DefaultEmbeddingEngine im
class EmbeddingConfig(BaseSettings): class EmbeddingConfig(BaseSettings):
openai_embedding_model: str = "text-embedding-3-large" openai_embedding_model: str = "text-embedding-3-large"
openai_embedding_dimensions: int = 3072 openai_embedding_dimensions: int = 3072
litellm_embedding_model: str = "text-embedding-3-large" litellm_embedding_model: str = "BAAI/bge-large-en-v1.5"
litellm_embedding_dimensions: int = 3072 litellm_embedding_dimensions: int = 1024
embedding_engine:object = DefaultEmbeddingEngine(embedding_model=openai_embedding_model, embedding_dimensions=openai_embedding_dimensions) embedding_engine:object = DefaultEmbeddingEngine(embedding_model=litellm_embedding_model, embedding_dimensions=litellm_embedding_dimensions)
model_config = SettingsConfigDict(env_file = ".env", extra = "allow") model_config = SettingsConfigDict(env_file = ".env", extra = "allow")