jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor IMSDbLoaderComponent to update document metadata

Browse source
This commit is contained in:
anovazzi1 2024-01-15 19:49:04 -03:00
commit 310886b0ba
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/components/documentloaders/IMSDbLoader.py
View file

@ -20,4 +20,11 @@ class IMSDbLoaderComponent(CustomComponent):
metadata: Optional[Dict] = None, metadata: Optional[Dict] = None,
web_path: str = "", web_path: str = "",
) -> Document: ) -> Document:
return IMSDbLoader(metadata=metadata, web_path=web_path) documents = IMSDbLoader(web_path=web_path).load()
if(metadata):
for document in documents:
if not document.metadata:
document.metadata = metadata
else:
document.metadata.update(metadata)
return documents