farcasclaudiu/Flowise
1
0
Fork 0
mirror of https://github.com/farcasclaudiu/Flowise.git synced 2026-06-22 11:01:22 +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 responseSynthesizerObj = nodeData.inputs?.responseSynthesizer
let queryEngine = new RetrieverQueryEngine(vectorStoreRetriever)
if (responseSynthesizerObj) {
if (responseSynthesizerObj.type === 'TreeSummarize') {
const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new TreeSummarize(vectorStoreRetriever.serviceContext, responseSynthesizerObj.textQAPromptTemplate),
serviceContext: vectorStoreRetriever.serviceContext
})
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} else if (responseSynthesizerObj.type === 'CompactAndRefine') {
const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new CompactAndRefine(
@@ -84,7 +86,7 @@ class QueryEngine_LlamaIndex implements INode {
),
serviceContext: vectorStoreRetriever.serviceContext
})
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} else if (responseSynthesizerObj.type === 'Refine') {
const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new Refine(
@@ -94,18 +96,16 @@ class QueryEngine_LlamaIndex implements INode {
),
serviceContext: vectorStoreRetriever.serviceContext
})
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
} else if (responseSynthesizerObj.type === 'SimpleResponseBuilder') {
const responseSynthesizer = new ResponseSynthesizer({
responseBuilder: new SimpleResponseBuilder(vectorStoreRetriever.serviceContext),
serviceContext: vectorStoreRetriever.serviceContext
})
return new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
queryEngine = new RetrieverQueryEngine(vectorStoreRetriever, responseSynthesizer)
}
}
const queryEngine = new RetrieverQueryEngine(vectorStoreRetriever)
let text = ''
let sourceDocuments: ICommonObject[] = []
let sourceNodes: BaseNode<Metadata>[] = []