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

chore: ruff formatting

Browse source
This commit is contained in:
Andrej Milicevic 2025-10-28 18:01:32 +01:00
parent bbcd8baf3a
commit 9c9395851c
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/context_global_variables.py
View file

@ -60,7 +60,7 @@ async def set_database_global_context_variables(dataset: Union[str, UUID], user_
"vector_db_provider": dataset_database.vector_database_provider, "vector_db_provider": dataset_database.vector_database_provider,
"vector_db_url": dataset_database.vector_database_url, "vector_db_url": dataset_database.vector_database_url,
# TODO: Maybe add key to dataset_database, and put it here?? # TODO: Maybe add key to dataset_database, and put it here??
"vector_db_key": "" "vector_db_key": "",
} }
# vector_config = { # vector_config = {

1
cognee/infrastructure/databases/utils/get_or_create_dataset_database.py
View file

@ -23,6 +23,7 @@ default_graph_db_key = None
vector_dbs_with_multi_user_support = ["lancedb", "falkor"] vector_dbs_with_multi_user_support = ["lancedb", "falkor"]
graph_dbs_with_multi_user_support = ["kuzu", "falkor"] graph_dbs_with_multi_user_support = ["kuzu", "falkor"]
async def get_or_create_dataset_database( async def get_or_create_dataset_database(
dataset: Union[str, UUID], dataset: Union[str, UUID],
user: User, user: User,