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', name: 'model',
type: 'BaseLanguageModel', type: 'BaseLanguageModel',
optional: true optional: true
}, }
] ]
} }

View File

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