Sultan
/
Flowise
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update LLMInMemoryCache.ts 

code cleanup
This commit is contained in:
Henry Heng 2023-10-06 23:34:07 +01:00 committed by GitHub
parent 91581250a7
commit 0b0cf357c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
packages/components/nodes/llmcache/LLMInMemoryCache/LLMInMemoryCache.ts
View File

@ -1,4 +1,4 @@
import { INode, INodeData, INodeOutputsValue, INodeParams } from '../../../src/Interface' import { INode, INodeParams } from '../../../src/Interface'
import { InMemoryCache } from 'langchain/cache' import { InMemoryCache } from 'langchain/cache'
import { getBaseClasses } from '../../../src' import { getBaseClasses } from '../../../src'
@ -12,7 +12,6 @@ class LLMInMemoryCache implements INode {
category: string category: string
baseClasses: string[] baseClasses: string[]
inputs: INodeParams[] inputs: INodeParams[]
outputs: INodeOutputsValue[]
inMemoryCache: any inMemoryCache: any
constructor() { constructor() {
@ -22,18 +21,11 @@ class LLMInMemoryCache implements INode {
this.type = 'LLMCache' this.type = 'LLMCache'
this.icon = 'memorycache.png' this.icon = 'memorycache.png'
this.category = 'LLM Cache' this.category = 'LLM Cache'
this.baseClasses = [this.type, 'LLMCacheBase'] this.baseClasses = [this.type, ...getBaseClasses(InMemoryCache)]
this.inputs = [] this.inputs = []
this.outputs = [
{
label: 'LLM Cache',
name: 'cache',
baseClasses: [this.type, ...getBaseClasses(InMemoryCache)]
}
]
} }
async init(nodeData: INodeData): Promise<any> { async init(): Promise<any> {
if (!this.inMemoryCache) { if (!this.inMemoryCache) {
this.inMemoryCache = InMemoryCache.global() this.inMemoryCache = InMemoryCache.global()
} }