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

bugfix

This commit is contained in:
Ofer Mendelevitch 2023-12-31 21:53:48 -08:00 committed by Ilango
parent 356137b88b
commit 0c2252c642
2 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
packages/components/nodes/chains/VectaraChain/VectaraChain.ts
View File

@ -249,14 +249,14 @@ class VectaraChain_Chains implements INode {
}))
const mmrRerankerId = 272725718 // Vectara reranker ID for MMR
const mmrEnabled = vectaraFilter?.mmrConfig?.mmrDiversityBias > 0
const mmrEnabled = vectaraFilter?.mmrConfig?.enabled
const data = {
query: [
{
query: input,
start: 0,
numResults: mmrEnabled ? vectaraFilter?.mmrK : topK,
numResults: mmrEnabled ? vectaraFilter?.mmrTopK : topK,
corpusKey: corpusKeys,
contextConfig: {
sentencesAfter: vectaraFilter?.contextConfig?.sentencesAfter ?? 2,

14
packages/components/nodes/vectorstores/Vectara/Vectara.ts
View File

@ -1,12 +1,5 @@
import { flatten } from 'lodash'
import {
VectaraStore,
VectaraLibArgs,
VectaraFilter,
VectaraContextConfig,
VectaraFile,
VectaraMMRConfig
} from 'langchain/vectorstores/vectara'
import { VectaraStore, VectaraLibArgs, VectaraFilter, VectaraContextConfig, VectaraFile, MMRConfig } from 'langchain/vectorstores/vectara'
import { Document } from 'langchain/document'
import { Embeddings } from 'langchain/embeddings/base'
import { ICommonObject, INode, INodeData, INodeOutputsValue, INodeParams } from '../../../src/Interface'
@ -237,8 +230,9 @@ class Vectara_VectorStores implements INode {
if (sentencesBefore) vectaraContextConfig.sentencesBefore = sentencesBefore
if (sentencesAfter) vectaraContextConfig.sentencesAfter = sentencesAfter
vectaraFilter.contextConfig = vectaraContextConfig
const mmrConfig: VectaraMMRConfig = {}
mmrConfig.mmrK = mmrK
const mmrConfig: MMRConfig = {}
mmrConfig.enabled = mmrDiversityBias > 0
mmrConfig.mmrTopK = mmrK
mmrConfig.diversityBias = mmrDiversityBias
vectaraFilter.mmrConfig = mmrConfig