From 92d0122b467d3d11538246ed6d8652bd238f4ae8 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 13 Dec 2024 09:55:47 +0100 Subject: [PATCH] fix: Remove data handling based on type in resolving directory function No need to handle different data types in resolving directories, focus on just handling case when it's a directory Fix COG-656 --- cognee/tasks/ingestion/resolve_data_directories.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/cognee/tasks/ingestion/resolve_data_directories.py b/cognee/tasks/ingestion/resolve_data_directories.py index 599a9b342..980756805 100644 --- a/cognee/tasks/ingestion/resolve_data_directories.py +++ b/cognee/tasks/ingestion/resolve_data_directories.py @@ -30,13 +30,8 @@ async def resolve_data_directories(data: Union[BinaryIO, List[BinaryIO], str, Li resolved_data.extend( [os.path.join(item, f) for f in os.listdir(item) if os.path.isfile(os.path.join(item, f))] ) - elif os.path.isfile(item): # If it's a file, add it to the resolved_data list + else: # If it's a file or text add it directly resolved_data.append(item) - else: # If it's just text add it directly - resolved_data.append(item) - elif isinstance(item, BinaryIO): # If it's a binary stream, add it directly + else: # If it's not a string add it directly resolved_data.append(item) - else: - raise TypeError(f"Unsupported type: {type(item)}. Expected str or BinaryIO.") - return resolved_data