fix: PR comment changes

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,
     }

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,