add top K to vector stores

2026-06-28 13:00:56 +03:00 · 2023-06-10 17:21:27 +01:00
parent 617b3bea96
commit d11cb5f4b4
36 changed files with 925 additions and 238 deletions
@@ -40,7 +40,10 @@ class VectorDBQAChain_Chains implements INode {
        const model = nodeData.inputs?.model as BaseLanguageModel
        const vectorStore = nodeData.inputs?.vectorStore as VectorStore

-        const chain = VectorDBQAChain.fromLLM(model, vectorStore, { verbose: process.env.DEBUG === 'true' ? true : false })
+        const chain = VectorDBQAChain.fromLLM(model, vectorStore, {
+            k: (vectorStore as any)?.k ?? 4,
+            verbose: process.env.DEBUG === 'true' ? true : false
+        })
        return chain
    }