lint fixes
This commit is contained in:
parent
f2f8ed6a9c
commit
be79adb07c
|
|
@ -39,7 +39,7 @@ class LLMFilterCompressionRetriever_Retrievers implements INode {
|
||||||
name: 'model',
|
name: 'model',
|
||||||
type: 'BaseLanguageModel',
|
type: 'BaseLanguageModel',
|
||||||
optional: true
|
optional: true
|
||||||
},
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue