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

🐛 fix(JSONDocumentBuilder.py): handle both single document and list of documents when building JSON document

Browse source 
✨ feat(JSONDocumentBuilder.py): add support for building JSON document from a list of documents
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-22 14:59:00 -03:00
commit 5fa8281e94
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/components/utilities/JSONDocumentBuilder.py
View file

@ -31,4 +31,12 @@ class JSONDocumentBuilder(CustomComponent):
key: str, key: str,
document: Document, document: Document,
) -> Document: ) -> Document:
return Document(page_content={key: document.page_content}) documents = None
if isinstance(document, list):
documents = [
Document(page_content={key: doc.page_content}) for doc in document
]
else:
documents = Document(page_content={key: document.page_content})
self.repr_value = documents
return documents