diff --git a/cognee/context_global_variables.py b/cognee/context_global_variables.py index 6a0f767ff..d4cedc187 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 311f89ad7..a2e053b3d 100644 --- a/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py +++ b/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py @@ -40,12 +40,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}" async with db_engine.get_async_session() as session: # Create dataset if it doesn't exist