bugfix
This commit is contained in:
parent
dc3e4fd059
commit
b44a0f1d39
|
|
@ -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,
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue