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 09:52:23 +02:00
parent 41ef5cf194
commit bbf55f9547
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/infrastructure/databases/vector/config.py
View file

@ -4,6 +4,8 @@ from pydantic_settings import BaseSettings, SettingsConfigDict
from cognee.base_config import get_base_config from cognee.base_config import get_base_config
from cognee.infrastructure.databases.vector.lancedb.LanceDBAdapter import LanceDBAdapter from cognee.infrastructure.databases.vector.lancedb.LanceDBAdapter import LanceDBAdapter
from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config
from cognee.infrastructure.files.storage import LocalStorage
embeddings_config = get_embedding_config() embeddings_config = get_embedding_config()
base_config = get_base_config() base_config = get_base_config()
@ -18,6 +20,8 @@ class VectorConfig(BaseSettings):
) )
vector_engine_choice:str = "lancedb" vector_engine_choice:str = "lancedb"
LocalStorage.ensure_directory_exists(vector_db_path)
model_config = SettingsConfigDict(env_file = ".env", extra = "allow") model_config = SettingsConfigDict(env_file = ".env", extra = "allow")
def to_dict(self) -> dict: def to_dict(self) -> dict: