diff --git a/cognee/context_global_variables.py b/cognee/context_global_variables.py index 9a4f49763..ee2e37030 100644 --- a/cognee/context_global_variables.py +++ b/cognee/context_global_variables.py @@ -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_url": dataset_database.vector_database_url, # TODO: Maybe add key to dataset_database, and put it here?? - "vector_db_key": "" + "vector_db_key": "", } # vector_config = { diff --git a/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py b/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py index 0af94fd3a..1552a7bbc 100644 --- a/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py +++ b/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py @@ -23,6 +23,7 @@ default_graph_db_key = None vector_dbs_with_multi_user_support = ["lancedb", "falkor"] graph_dbs_with_multi_user_support = ["kuzu", "falkor"] + async def get_or_create_dataset_database( dataset: Union[str, UUID], user: User,