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,