From cf6ad5355962c32b3b719da3f69a5aaca554e80f Mon Sep 17 00:00:00 2001
From: Henry <hzj94@hotmail.com>
Date: Tue, 11 Jul 2023 12:52:32 +0100
Subject: [PATCH] update Zep memory node

---
 .../nodes/memory/ZepMemory/ZepMemory.ts       | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/packages/components/nodes/memory/ZepMemory/ZepMemory.ts b/packages/components/nodes/memory/ZepMemory/ZepMemory.ts
index 89d07b8fc..6e1a14bd7 100644
--- a/packages/components/nodes/memory/ZepMemory/ZepMemory.ts
+++ b/packages/components/nodes/memory/ZepMemory/ZepMemory.ts
@@ -44,6 +44,20 @@ class ZepMemory_Memory implements INode {
                 additionalParams: true,
                 optional: true
             },
+            {
+                label: 'API Key',
+                name: 'apiKey',
+                type: 'string',
+                additionalParams: true,
+                optional: true
+            },
+            {
+                label: 'Size',
+                name: 'k',
+                type: 'number',
+                default: '10',
+                description: 'Window of size k to surface the last k back-and-forths to use as memory.'
+            },
             {
                 label: 'Auto Summary Template',
                 name: 'autoSummaryTemplate',
@@ -98,6 +112,8 @@ class ZepMemory_Memory implements INode {
         const autoSummaryTemplate = nodeData.inputs?.autoSummaryTemplate as string
         const autoSummary = nodeData.inputs?.autoSummary as boolean
         const sessionId = nodeData.inputs?.sessionId as string
+        const apiKey = nodeData.inputs?.apiKey as string
+        const k = nodeData.inputs?.k as string
 
         const chatId = options?.chatId as string
 
@@ -110,6 +126,7 @@ class ZepMemory_Memory implements INode {
             memoryKey,
             inputKey
         }
+        if (apiKey) obj.apiKey = apiKey
 
         let zep = new ZepMemory(obj)
 
@@ -118,7 +135,7 @@ class ZepMemory_Memory implements INode {
         zep.loadMemoryVariables = async (values) => {
             let data = await tmpFunc.bind(zep, values)()
             if (autoSummary && zep.returnMessages && data[zep.memoryKey] && data[zep.memoryKey].length) {
-                const memory = await zep.zepClient.getMemory(zep.sessionId, 10)
+                const memory = await zep.zepClient.getMemory(zep.sessionId, parseInt(k, 10) ?? 10)
                 if (memory?.summary) {
                     let summary = autoSummaryTemplate.replace(/{summary}/g, memory.summary.content)
                     // eslint-disable-next-line no-console