diff --git a/cognee/context_global_variables.py b/cognee/context_global_variables.py index c2e9e82a9..62e06fc64 100644 --- a/cognee/context_global_variables.py +++ b/cognee/context_global_variables.py @@ -16,8 +16,8 @@ vector_db_config = ContextVar("vector_db_config", default=None) graph_db_config = ContextVar("graph_db_config", default=None) session_user = ContextVar("session_user", default=None) -VECTOR_DBS_WITH_MULTI_USER_SUPPORT = ["lancedb"] -GRAPH_DBS_WITH_MULTI_USER_SUPPORT = ["kuzu"] +VECTOR_DBS_WITH_MULTI_USER_SUPPORT = ["lancedb", "falkor"] +GRAPH_DBS_WITH_MULTI_USER_SUPPORT = ["kuzu", "falkor"] async def set_session_user_context_variable(user): 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 0df3502ba..3684bb100 100644 --- a/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py +++ b/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py @@ -43,12 +43,12 @@ async def get_or_create_dataset_database( if graph_config.graph_database_provider == "kuzu": graph_db_name = f"{dataset_id}.pkl" else: - graph_db_name = dataset_id + graph_db_name = f"{dataset_id}" if vector_config.vector_db_provider == "lancedb": vector_db_name = f"{dataset_id}.lance.db" else: - vector_db_name = dataset_id + vector_db_name = f"{dataset_id}" base_config = get_base_config() databases_directory_path = os.path.join(