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

feat: Finalize zep-cloud integration, update naming

This commit is contained in:
paulpaliychuk 2024-02-20 00:31:35 -05:00
parent 29c86f77bf
commit 9da7a1b4de
6 changed files with 27 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
packages/components/nodes/memory/ZepMemory/ZepMemory.ts
View File

@ -17,8 +17,8 @@ class ZepMemory_Memory implements INode {
inputs: INodeParams[]
constructor() {
this.label = 'Zep Memory'
this.name = 'ZepMemory'
this.label = 'Zep Memory - Open Source'
this.name = 'Zep Memory (Open Source)'
this.version = 2.0
this.type = 'ZepMemory'
this.icon = 'zep.svg'
@ -97,11 +97,11 @@ class ZepMemory_Memory implements INode {
}
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
return await initalizeZep(nodeData, options)
return await initializeZep(nodeData, options)
}
}
const initalizeZep = async (nodeData: INodeData, options: ICommonObject): Promise<ZepMemory> => {
const initializeZep = async (nodeData: INodeData, options: ICommonObject): Promise<ZepMemory> => {
const baseURL = nodeData.inputs?.baseURL as string
const aiPrefix = nodeData.inputs?.aiPrefix as string
const humanPrefix = nodeData.inputs?.humanPrefix as string

22
packages/components/nodes/memory/ZepMemoryCloud/ZepMemoryCloud.ts
View File

@ -20,7 +20,7 @@ class ZepMemoryCloud_Memory implements INode {
constructor() {
this.label = 'Zep Memory - Cloud'
this.name = 'ZepMemory (Cloud)'
this.name = 'Zep Memory (Cloud)'
this.version = 2.0
this.type = 'ZepMemory'
this.icon = 'zep.svg'
@ -75,6 +75,13 @@ class ZepMemoryCloud_Memory implements INode {
default: 'chat_history',
additionalParams: true
},
{
label: 'Input Key',
name: 'inputKey',
type: 'string',
default: 'input',
additionalParams: true
},
{
label: 'Output Key',
name: 'outputKey',
@ -86,20 +93,21 @@ class ZepMemoryCloud_Memory implements INode {
}
async init(nodeData: INodeData, _: string, options: ICommonObject): Promise<any> {
return await initalizeZep(nodeData, options)
return await initializeZep(nodeData, options)
}
}
const initalizeZep = async (nodeData: INodeData, options: ICommonObject): Promise<ZepMemory> => {
const initializeZep = async (nodeData: INodeData, options: ICommonObject): Promise<ZepMemory> => {
const aiPrefix = nodeData.inputs?.aiPrefix as string
const humanPrefix = nodeData.inputs?.humanPrefix as string
const memoryKey = nodeData.inputs?.memoryKey as string
const inputKey = nodeData.inputs?.inputKey as string
const memoryType = nodeData.inputs?.memoryType as 'perpetual' | 'message_window'
const sessionId = nodeData.inputs?.sessionId as string
const credentialData = await getCredentialData(nodeData.credential ?? '', options)
const apiKey = getCredentialParam('apiKey', credentialData, nodeData)
const obj: ZepMemoryInput & ZepMemoryExtendedInput = {
apiKey,
baseURL: 'https://api.development.getzep.com',
@ -107,7 +115,9 @@ const initalizeZep = async (nodeData: INodeData, options: ICommonObject): Promis
humanPrefix,
memoryKey,
sessionId,
memoryType: memoryType
inputKey,
memoryType: memoryType,
returnMessages: true
}
return new ZepMemoryExtended(obj)
@ -157,7 +167,7 @@ class ZepMemoryExtended extends ZepMemory implements MemoryMethods {
const id = overrideSessionId ? overrideSessionId : this.sessionId
const input = msgArray.find((msg) => msg.type === 'userMessage')
const output = msgArray.find((msg) => msg.type === 'apiMessage')
const inputValues = { ['input']: input?.text }
const inputValues = { [this.inputKey ?? 'input']: input?.text }
const outputValues = { output: output?.text }
await this.saveContext(inputValues, outputValues, id)

4
packages/components/nodes/vectorstores/Zep/Zep.ts
View File

@ -22,8 +22,8 @@ class Zep_VectorStores implements INode {
outputs: INodeOutputsValue[]
constructor() {
this.label = 'Zep'
this.name = 'zep'
this.label = 'Zep Collection - Open Source'
this.name = 'Zep Collection (Open Source)'
this.version = 2.0
this.type = 'Zep'
this.icon = 'zep.svg'

4
packages/components/nodes/vectorstores/Zep/Zep_Existing.ts
View File

@ -20,8 +20,8 @@ class Zep_Existing_VectorStores implements INode {
outputs: INodeOutputsValue[]
constructor() {
this.label = 'Zep Load Existing Index'
this.name = 'zepExistingIndex'
this.label = 'Zep Load Existing Index (Open Source)'
this.name = 'Zep Existing Index (Open Source)'
this.version = 1.0
this.type = 'Zep'
this.icon = 'zep.svg'

4
packages/components/nodes/vectorstores/Zep/Zep_Upsert.ts
View File

@ -20,8 +20,8 @@ class Zep_Upsert_VectorStores implements INode {
outputs: INodeOutputsValue[]
constructor() {
this.label = 'Zep Upsert Document'
this.name = 'zepUpsert'
this.label = 'Zep Upsert Document (Open Source)'
this.name = 'Zep Upsert (Open Source)'
this.version = 1.0
this.type = 'Zep'
this.icon = 'zep.svg'

2
packages/components/nodes/vectorstores/ZepCloud/ZepCloud.ts
View File

@ -23,7 +23,7 @@ class Zep_CloudVectorStores implements INode {
constructor() {
this.label = 'Zep Collection - Cloud'
this.name = 'zepCloud'
this.name = 'Zep Collection Cloud'
this.version = 2.0
this.type = 'Zep'
this.icon = 'zep.svg'