dtamind-components
Version:
DTAmindai Components
162 lines • 6.07 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const zep_1 = require("@langchain/community/memory/zep");
const utils_1 = require("../../../src/utils");
class ZepMemory_Memory {
constructor() {
this.label = 'Zep Memory - Open Source';
this.name = 'ZepMemory';
this.version = 2.0;
this.type = 'ZepMemory';
this.icon = 'zep.svg';
this.category = 'Memory';
this.description = 'Summarizes the conversation and stores the memory in zep server';
this.baseClasses = [this.type, ...(0, utils_1.getBaseClasses)(zep_1.ZepMemory)];
this.credential = {
label: 'Connect Credential',
name: 'credential',
type: 'credential',
optional: true,
description: 'Configure JWT authentication on your Zep instance (Optional)',
credentialNames: ['zepMemoryApi']
};
this.inputs = [
{
label: 'Base URL',
name: 'baseURL',
type: 'string',
default: 'http://127.0.0.1:8000'
},
{
label: 'Session Id',
name: 'sessionId',
type: 'string',
description: 'If not specified, a random id will be used. Learn <a target="_blank" href="https://docs.flowiseai.com/memory/long-term-memory#ui-and-embedded-chat">more</a>',
default: '',
additionalParams: true,
optional: true
},
{
label: 'Size',
name: 'k',
type: 'number',
default: '10',
description: 'Window of size k to surface the last k back-and-forth to use as memory.',
additionalParams: true
},
{
label: 'AI Prefix',
name: 'aiPrefix',
type: 'string',
default: 'ai',
additionalParams: true
},
{
label: 'Human Prefix',
name: 'humanPrefix',
type: 'string',
default: 'human',
additionalParams: true
},
{
label: 'Memory Key',
name: 'memoryKey',
type: 'string',
default: 'chat_history',
additionalParams: true
},
{
label: 'Input Key',
name: 'inputKey',
type: 'string',
default: 'input',
additionalParams: true
},
{
label: 'Output Key',
name: 'outputKey',
type: 'string',
default: 'text',
additionalParams: true
}
];
}
async init(nodeData, _, options) {
return await initializeZep(nodeData, options);
}
}
const initializeZep = async (nodeData, options) => {
const baseURL = nodeData.inputs?.baseURL;
const aiPrefix = nodeData.inputs?.aiPrefix;
const humanPrefix = nodeData.inputs?.humanPrefix;
const memoryKey = nodeData.inputs?.memoryKey;
const inputKey = nodeData.inputs?.inputKey;
const k = nodeData.inputs?.k;
const sessionId = nodeData.inputs?.sessionId;
const credentialData = await (0, utils_1.getCredentialData)(nodeData.credential ?? '', options);
const apiKey = (0, utils_1.getCredentialParam)('apiKey', credentialData, nodeData);
const orgId = options.orgId;
const obj = {
baseURL,
aiPrefix,
humanPrefix,
returnMessages: true,
memoryKey,
inputKey,
sessionId,
orgId,
k: k ? parseInt(k, 10) : undefined
};
if (apiKey)
obj.apiKey = apiKey;
return new ZepMemoryExtended(obj);
};
class ZepMemoryExtended extends zep_1.ZepMemory {
constructor(fields) {
super(fields);
this.orgId = '';
this.lastN = fields.k;
this.orgId = fields.orgId;
}
async loadMemoryVariables(values, overrideSessionId = '') {
if (overrideSessionId) {
this.sessionId = overrideSessionId;
}
return super.loadMemoryVariables({ ...values, lastN: this.lastN });
}
async saveContext(inputValues, outputValues, overrideSessionId = '') {
if (overrideSessionId) {
this.sessionId = overrideSessionId;
}
return super.saveContext(inputValues, outputValues);
}
async clear(overrideSessionId = '') {
if (overrideSessionId) {
this.sessionId = overrideSessionId;
}
return super.clear();
}
async getChatMessages(overrideSessionId = '', returnBaseMessages = false, prependMessages) {
const id = overrideSessionId ? overrideSessionId : this.sessionId;
const memoryVariables = await this.loadMemoryVariables({}, id);
const baseMessages = memoryVariables[this.memoryKey];
if (prependMessages?.length) {
baseMessages.unshift(...(await (0, utils_1.mapChatMessageToBaseMessage)(prependMessages, this.orgId)));
}
return returnBaseMessages ? baseMessages : (0, utils_1.convertBaseMessagetoIMessage)(baseMessages);
}
async addChatMessages(msgArray, overrideSessionId = '') {
const id = overrideSessionId ? overrideSessionId : this.sessionId;
const input = msgArray.find((msg) => msg.type === 'userMessage');
const output = msgArray.find((msg) => msg.type === 'apiMessage');
const inputValues = { [this.inputKey ?? 'input']: input?.text };
const outputValues = { output: output?.text };
await this.saveContext(inputValues, outputValues, id);
}
async clearChatMessages(overrideSessionId = '') {
const id = overrideSessionId ? overrideSessionId : this.sessionId;
await this.clear(id);
}
}
module.exports = { nodeClass: ZepMemory_Memory };
//# sourceMappingURL=ZepMemory.js.map