From ac6c3ef9deeef9ef8069ba1aecdce8791987bd1e Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Tue, 11 Nov 2025 15:07:59 +0100 Subject: [PATCH] fix: fix names, add falkor to constants --- cognee/context_global_variables.py | 4 ++-- .../databases/utils/get_or_create_dataset_database.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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