From 5fa8281e94e9984e379b3ae1d270351c705f309c Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 22 Aug 2023 14:59:00 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(JSONDocumentBuilder.py):=20h?=
 =?UTF-8?q?andle=20both=20single=20document=20and=20list=20of=20documents?=
 =?UTF-8?q?=20when=20building=20JSON=20document=20=E2=9C=A8=20feat(JSONDoc?=
 =?UTF-8?q?umentBuilder.py):=20add=20support=20for=20building=20JSON=20doc?=
 =?UTF-8?q?ument=20from=20a=20list=20of=20documents?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../components/utilities/JSONDocumentBuilder.py        | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/utilities/JSONDocumentBuilder.py b/src/backend/langflow/components/utilities/JSONDocumentBuilder.py
index 0336dcc1b..4880200da 100644
--- a/src/backend/langflow/components/utilities/JSONDocumentBuilder.py
+++ b/src/backend/langflow/components/utilities/JSONDocumentBuilder.py
@@ -31,4 +31,12 @@ class JSONDocumentBuilder(CustomComponent):
         key: str,
         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