Bugfix/Save chunk's metadata (#2366)

save metadata chunk
This commit is contained in:
Henry Heng 2024-05-08 17:24:03 +01:00 committed by GitHub
parent 26e7a1ac35
commit a9f9c8874c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 3 deletions

View File

@ -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) {

View File

@ -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)