gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Add TODOs to handle refactoring

Browse source
This commit is contained in:
Igor Ilic 2025-05-20 13:54:02 +02:00
parent 472143df03
commit 1361203ead
3 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cognee/api/v1/cognify/routers/get_cognify_router.py
View file

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

1
cognee/infrastructure/databases/vector/get_vector_engine.py
View file

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

1
cognee/tasks/documents/check_permissions_on_dataset.py
View file

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