This commit is contained in:
Ofer Mendelevitch
2023-12-31 21:53:48 -08:00
parent dc3e4fd059
commit b44a0f1d39
2 changed files with 6 additions and 12 deletions
@@ -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