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 10:27:53 +02:00
parent 6c739f3357
commit dad26b3904
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -2,17 +2,19 @@ import os
from functools import lru_cache from functools import lru_cache
from pydantic_settings import BaseSettings, SettingsConfigDict 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.relational.config import get_relationaldb_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 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()
relational_config = get_relationaldb_config()
class VectorConfig(BaseSettings): class VectorConfig(BaseSettings):
vector_db_url: str = "" vector_db_url: str = ""
vector_db_key: str = "" vector_db_key: str = ""
vector_db_path: str = os.path.join(base_config.database_directory_path + "cognee.lancedb") vector_db_path: str = os.path.join(relational_config.database_directory_path + "cognee.lancedb")
vector_engine: object = LanceDBAdapter( vector_engine: object = LanceDBAdapter(
url = vector_db_path, url = vector_db_path,
api_key = None, api_key = None,