Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update query engine

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
packages/components/nodes/engine/QueryEngine/QueryEngine.ts
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>[] = []