From 48825d0d84c9cb346bca66ffd428fba1c1678501 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Tue, 17 Dec 2024 14:22:51 +0100 Subject: [PATCH] chore: Resolve typo in getting documents code Resolve typo in code chore COG-912 --- .../users/permissions/methods/get_document_ids_for_user.py | 6 +++--- cognee/tests/test_pgvector.py | 1 - 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/cognee/modules/users/permissions/methods/get_document_ids_for_user.py b/cognee/modules/users/permissions/methods/get_document_ids_for_user.py index d726e9002..d439fb4f5 100644 --- a/cognee/modules/users/permissions/methods/get_document_ids_for_user.py +++ b/cognee/modules/users/permissions/methods/get_document_ids_for_user.py @@ -21,7 +21,7 @@ async def get_document_ids_for_user(user_id: UUID, datasets: list[str] = None) - )).all() if datasets: - documnets_ids_in_dataset = set() + documents_ids_in_dataset = set() # If datasets are specified filter out documents that aren't part of the specified datasets for dataset in datasets: # Find dataset id for dataset element @@ -45,6 +45,6 @@ async def get_document_ids_for_user(user_id: UUID, datasets: list[str] = None) - # If document is related to dataset added it to return value if data_id: - documnets_ids_in_dataset.add(document_id) - return list(documnets_ids_in_dataset) + documents_ids_in_dataset.add(document_id) + return list(documents_ids_in_dataset) return document_ids diff --git a/cognee/tests/test_pgvector.py b/cognee/tests/test_pgvector.py index 3aaed016a..9554a3f9d 100644 --- a/cognee/tests/test_pgvector.py +++ b/cognee/tests/test_pgvector.py @@ -74,7 +74,6 @@ async def main(): assert len(document_ids) == 1, f"Number of expected documents doesn't match {len(document_ids)} != 1" # Test getting of documents for search when no dataset is provided - from cognee.modules.users.permissions.methods import get_document_ids_for_user user = await get_default_user() document_ids = await get_document_ids_for_user(user.id) assert len(document_ids) == 2, f"Number of expected documents doesn't match {len(document_ids)} != 2"