From a9f9c8874c41e50db233a8734808411f8603a316 Mon Sep 17 00:00:00 2001
From: Henry Heng <henryheng@flowiseai.com>
Date: Wed, 8 May 2024 17:24:03 +0100
Subject: [PATCH] Bugfix/Save chunk's metadata (#2366)

save metadata chunk
---
 packages/server/src/controllers/documentstore/index.ts | 5 +++--
 packages/server/src/services/documentstore/index.ts    | 3 ++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/packages/server/src/controllers/documentstore/index.ts b/packages/server/src/controllers/documentstore/index.ts
index 6797eb12..073af3af 100644
--- a/packages/server/src/controllers/documentstore/index.ts
+++ b/packages/server/src/controllers/documentstore/index.ts
@@ -141,7 +141,7 @@ const editDocumentStoreFileChunk = async (req: Request, res: Response, next: Nex
             )
         }
         const body = req.body
-        if (typeof body === 'undefined' || body.pageContent === 'undefined' || body.pageContent === '') {
+        if (typeof body === 'undefined') {
             throw new InternalFlowiseError(
                 StatusCodes.PRECONDITION_FAILED,
                 `Error: documentStoreController.editDocumentStoreFileChunk - body not provided!`
@@ -151,7 +151,8 @@ const editDocumentStoreFileChunk = async (req: Request, res: Response, next: Nex
             req.params.storeId,
             req.params.loaderId,
             req.params.chunkId,
-            body.pageContent
+            body.pageContent,
+            body.metadata
         )
         return res.json(apiResponse)
     } catch (error) {
diff --git a/packages/server/src/services/documentstore/index.ts b/packages/server/src/services/documentstore/index.ts
index 3347e38c..d19860d4 100644
--- a/packages/server/src/services/documentstore/index.ts
+++ b/packages/server/src/services/documentstore/index.ts
@@ -282,7 +282,7 @@ const deleteDocumentStoreFileChunk = async (storeId: string, docId: string, chun
     }
 }
 
-const editDocumentStoreFileChunk = async (storeId: string, docId: string, chunkId: string, content: string) => {
+const editDocumentStoreFileChunk = async (storeId: string, docId: string, chunkId: string, content: string, metadata: ICommonObject) => {
     try {
         const appServer = getRunningExpressApp()
         const entity = await appServer.AppDataSource.getRepository(DocumentStore).findOneBy({
@@ -305,6 +305,7 @@ const editDocumentStoreFileChunk = async (storeId: string, docId: string, chunkI
         }
         found.totalChars -= editChunk.pageContent.length
         editChunk.pageContent = content
+        editChunk.metadata = JSON.stringify(metadata)
         found.totalChars += content.length
         await appServer.AppDataSource.getRepository(DocumentStoreFileChunk).save(editChunk)
         entity.loaders = JSON.stringify(loaders)