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

update query engine

Browse Source
This commit is contained in:
Henry
2024-01-26 00:22:24 +00:00
parent 1fb3e25f53
commit 3d670fec81
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/components/nodes/engine/QueryEngine/QueryEngine.ts
+6 -6
View File
@@ -68,13 +68,15 @@ class QueryEngine_LlamaIndex implements INode {
const vectorStoreRetriever = nodeData.inputs?.vectorStoreRetriever const vectorStoreRetriever = nodeData.inputs?.vectorStoreRetriever
const responseSynthesizerObj = nodeData.inputs?.responseSynthesizer const responseSynthesizerObj = nodeData.inputs?.responseSynthesizer
let queryEngine = new RetrieverQueryEngine(vectorStoreRetriever)
if (responseSynthesizerObj) { if (responseSynthesizerObj) {
if (responseSynthesizerObj.type === 'TreeSummarize') { if (responseSynthesizerObj.type === 'TreeSummarize') {
const responseSynthesizer = new ResponseSynthesizer({ const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new TreeSummarize(vectorStoreRetriever.serviceContext, responseSynthesizerObj.textQAPromptTemplate), responseBuilder: new TreeSummarize(vectorStoreRetriever.serviceContext, responseSynthesizerObj.textQAPromptTemplate),
serviceContext: vectorStoreRetriever.serviceContext serviceContext: vectorStoreRetriever.serviceContext
}) })
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer) queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} else if (responseSynthesizerObj.type === 'CompactAndRefine') { } else if (responseSynthesizerObj.type === 'CompactAndRefine') {
const responseSynthesizer = new ResponseSynthesizer({ const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new CompactAndRefine( responseBuilder: new CompactAndRefine(
@@ -84,7 +86,7 @@ class QueryEngine_LlamaIndex implements INode {
), ),
serviceContext: vectorStoreRetriever.serviceContext serviceContext: vectorStoreRetriever.serviceContext
}) })
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer) queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} else if (responseSynthesizerObj.type === 'Refine') { } else if (responseSynthesizerObj.type === 'Refine') {
const responseSynthesizer = new ResponseSynthesizer({ const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new Refine( responseBuilder: new Refine(
@@ -94,18 +96,16 @@ class QueryEngine_LlamaIndex implements INode {
), ),
serviceContext: vectorStoreRetriever.serviceContext serviceContext: vectorStoreRetriever.serviceContext
}) })
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer) queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} else if (responseSynthesizerObj.type === 'SimpleResponseBuilder') { } else if (responseSynthesizerObj.type === 'SimpleResponseBuilder') {
const responseSynthesizer = new ResponseSynthesizer({ const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new SimpleResponseBuilder(vectorStoreRetriever.serviceContext), responseBuilder: new SimpleResponseBuilder(vectorStoreRetriever.serviceContext),
serviceContext: vectorStoreRetriever.serviceContext serviceContext: vectorStoreRetriever.serviceContext
}) })
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer) queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} }
} }
const queryEngine = new RetrieverQueryEngine(vectorStoreRetriever)
let text = '' let text = ''
let sourceDocuments: ICommonObject[] = [] let sourceDocuments: ICommonObject[] = []
let sourceNodes: BaseNode<Metadata>[] = [] let sourceNodes: BaseNode<Metadata>[] = []