diff --git a/packages/components/nodes/memory/ZepMemory/ZepMemory.ts b/packages/components/nodes/memory/ZepMemory/ZepMemory.ts
index d511e7b22..2715706cb 100644
--- a/packages/components/nodes/memory/ZepMemory/ZepMemory.ts
+++ b/packages/components/nodes/memory/ZepMemory/ZepMemory.ts
@@ -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
diff --git a/packages/components/nodes/memory/ZepMemoryCloud/ZepMemoryCloud.ts b/packages/components/nodes/memory/ZepMemoryCloud/ZepMemoryCloud.ts
index 336fbc4d0..f5926d873 100644
--- a/packages/components/nodes/memory/ZepMemoryCloud/ZepMemoryCloud.ts
+++ b/packages/components/nodes/memory/ZepMemoryCloud/ZepMemoryCloud.ts
@@ -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)
diff --git a/packages/components/nodes/vectorstores/Zep/Zep.ts b/packages/components/nodes/vectorstores/Zep/Zep.ts
index c5a8f0d44..7bd7a08c8 100644
--- a/packages/components/nodes/vectorstores/Zep/Zep.ts
+++ b/packages/components/nodes/vectorstores/Zep/Zep.ts
@@ -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'
diff --git a/packages/components/nodes/vectorstores/Zep/Zep_Existing.ts b/packages/components/nodes/vectorstores/Zep/Zep_Existing.ts
index 75c2bd3fd..a8c57236a 100644
--- a/packages/components/nodes/vectorstores/Zep/Zep_Existing.ts
+++ b/packages/components/nodes/vectorstores/Zep/Zep_Existing.ts
@@ -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'
diff --git a/packages/components/nodes/vectorstores/Zep/Zep_Upsert.ts b/packages/components/nodes/vectorstores/Zep/Zep_Upsert.ts
index f05fc207b..83dcf710f 100644
--- a/packages/components/nodes/vectorstores/Zep/Zep_Upsert.ts
+++ b/packages/components/nodes/vectorstores/Zep/Zep_Upsert.ts
@@ -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'
diff --git a/packages/components/nodes/vectorstores/ZepCloud/ZepCloud.ts b/packages/components/nodes/vectorstores/ZepCloud/ZepCloud.ts
index 10847bb8f..74f143ed3 100644
--- a/packages/components/nodes/vectorstores/ZepCloud/ZepCloud.ts
+++ b/packages/components/nodes/vectorstores/ZepCloud/ZepCloud.ts
@@ -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'