refactor: Add TODOs to handle refactoring

cognee/api/v1/cognify/routers/get_cognify_router.py

@@ -29,6 +29,7 @@ def get_cognify_router() -> APIRouter:
         try:
             # TODO: Make async gather per dataset
             for dataset in payload.datasets:
+                # TODO: Move context setting handler outside of APIs
                 dataset_database = await get_or_create_dataset_database(dataset, user)
                 vector_config = {
                     "vector_db_url": dataset_database.vector_database_name,

cognee/infrastructure/databases/vector/get_vector_engine.py

@@ -4,6 +4,7 @@ from cognee.context_global_variables import vector_db_config
 
 
 def get_vector_engine():
+    # TODO: Create new get_vector_db_context_config function to handle context variables
     if vector_db_config.get():
         return create_vector_engine(**vector_db_config.get())
     return create_vector_engine(**get_vectordb_config().to_dict())

cognee/tasks/documents/check_permissions_on_dataset.py

@@ -1,7 +1,6 @@
 from cognee.modules.data.processing.document_types import Document
 from cognee.modules.users.permissions.methods import check_permission_on_dataset
 from typing import List
-from uuid import UUID
 
 
 async def check_permissions_on_dataset(