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

fix: Remove data handling based on type in resolving directory function

Browse source 
No need to handle different data types in resolving directories, focus on just handling case when it's a directory

Fix COG-656
This commit is contained in:
Igor Ilic 2024-12-13 09:55:47 +01:00
parent 7100a4994a
commit 92d0122b46
1 changed files with 2 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cognee/tasks/ingestion/resolve_data_directories.py
View file

@ -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