Refactor SlackDirectoryLoaderComponent to update document metadata

This commit is contained in:
anovazzi1 2024-01-15 22:12:36 -03:00
commit 3d4a4c1fb5

View file

@ -21,4 +21,11 @@ class SlackDirectoryLoaderComponent(CustomComponent):
metadata: Optional[Dict] = None,
workspace_url: Optional[str] = None,
) -> List[Document]:
return SlackDirectoryLoader(zip_path=zip_path, metadata=metadata, workspace_url=workspace_url).load()
documents = SlackDirectoryLoader(zip_path=zip_path,workspace_url=workspace_url).load()
if(metadata):
for document in documents:
if not document.metadata:
document.metadata = metadata
else:
document.metadata.update(metadata)
return documents