farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-28 19:00:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Bugfix/Missing Filter for VectorStore to Document (#2285)

Browse Source 
add filter for vector store to document
This commit is contained in:
Henry Heng
2024-04-29 22:25:40 +01:00
committed by GitHub
parent c4eb75ddde
commit 2b1273ca31
17 changed files with 67 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/documentloaders/VectorStoreToDocument/VectorStoreToDocument.ts
+9 -2
View File
@@ -70,12 +70,19 @@ class VectorStoreToDocument_DocumentLoaders implements INode {
const output = nodeData.outputs?.output as string
const topK = (vectorStore as any)?.k ?? 4
const _filter = (vectorStore as any)?.filter
const docs = await vectorStore.similaritySearchWithScore(query ?? input, topK)
// If it is already pre-defined in lc_kwargs, then don't pass it again
const filter = vectorStore.lc_kwargs.filter ? undefined : _filter
if (vectorStore.lc_kwargs.filter) {
;(vectorStore as any).filter = vectorStore.lc_kwargs.filter
}
const docs = await vectorStore.similaritySearchWithScore(query ?? input, topK, filter)
// eslint-disable-next-line no-console
console.log('\x1b[94m\x1b[1m\n*****VectorStore Documents*****\n\x1b[0m\x1b[0m')
// eslint-disable-next-line no-console
console.log(docs)
console.log(JSON.stringify(docs, null, 2))
if (output === 'document') {
let finaldocs = []