From 4f5771230e6177db1f87b448abdc851067317608 Mon Sep 17 00:00:00 2001 From: Andrej Milicevic Date: Tue, 11 Nov 2025 14:22:42 +0100 Subject: [PATCH] fix: PR comment changes --- cognee/context_global_variables.py | 9 ++++++++- .../databases/utils/get_or_create_dataset_database.py | 1 + 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/cognee/context_global_variables.py b/cognee/context_global_variables.py index 2d711a8b2..6a0f767ff 100644 --- a/cognee/context_global_variables.py +++ b/cognee/context_global_variables.py @@ -86,10 +86,17 @@ async def set_database_global_context_variables(dataset: Union[str, UUID], user_ base_config.system_root_directory, "databases", str(user.id) ) + if dataset_database.vector_database_provider == "lancedb": + vector_db_url = os.path.join( + databases_directory_path, dataset_database.vector_database_name + ) + else: + vector_db_url = dataset_database.vector_database_url + # Set vector and graph database configuration based on dataset database information vector_config = { "vector_db_provider": dataset_database.vector_database_provider, - "vector_db_url": dataset_database.vector_database_url, + "vector_db_url": vector_db_url, "vector_db_key": dataset_database.vector_database_key, "vector_db_name": dataset_database.vector_database_name, } 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 61d7840c0..b00616671 100644 --- a/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py +++ b/cognee/infrastructure/databases/utils/get_or_create_dataset_database.py @@ -12,6 +12,7 @@ from cognee.modules.data.methods import get_unique_dataset_id from cognee.modules.users.models import DatasetDatabase from cognee.modules.users.models import User + async def get_or_create_dataset_database( dataset: Union[str, UUID], user: User,