lint fixes

This commit is contained in:
vinodkiran 2023-12-22 12:13:52 +05:30
parent f2f8ed6a9c
commit be79adb07c
2 changed files with 3 additions and 4 deletions

View File

@ -39,7 +39,7 @@ class LLMFilterCompressionRetriever_Retrievers implements INode {
name: 'model',
type: 'BaseLanguageModel',
optional: true
},
}
]
}

View File

@ -1,4 +1,4 @@
import { INodeData } from '../../src'
import { INodeData } from "../../src";
export const resolveVectorStoreOrRetriever = (nodeData: INodeData, vectorStore: any) => {
const output = nodeData.outputs?.output as string
@ -12,7 +12,7 @@ export const resolveVectorStoreOrRetriever = (nodeData: INodeData, vectorStore:
const lambda = nodeData.inputs?.lambda as string
const f = fetchK ? parseInt(fetchK) : 20
const l = lambda ? parseFloat(lambda) : 0.5
const retriever = vectorStore.asRetriever({
return vectorStore.asRetriever({
searchType: 'mmr',
k: k,
searchKwargs: {
@ -20,7 +20,6 @@ export const resolveVectorStoreOrRetriever = (nodeData: INodeData, vectorStore:
lambda: l
}
})
return retriever
} else {
// "searchType" is "similarity"
return vectorStore.asRetriever(k)