From bacf733503dcce45640af8191d4fe9afc042dd1e Mon Sep 17 00:00:00 2001
From: Emir Karabeg <emirkarabeg@berkeley.edu>
Date: Tue, 20 May 2025 11:51:06 -0700
Subject: [PATCH] improvement(memory): consolidated expected return value

---
 apps/sim/blocks/blocks/memory.ts    | 1 -
 apps/sim/tools/memory/add_memory.ts | 6 +++---
 apps/sim/tools/memory/get_memory.ts | 6 +++---
 3 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/apps/sim/blocks/blocks/memory.ts b/apps/sim/blocks/blocks/memory.ts
index f8712537be..9bfa92d8d4 100644
--- a/apps/sim/blocks/blocks/memory.ts
+++ b/apps/sim/blocks/blocks/memory.ts
@@ -118,7 +118,6 @@ export const MemoryBlock: BlockConfig = {
   outputs: {
     response: {
       type: {
-        memory: 'any',
         memories: 'any',
         id: 'string',
       },
diff --git a/apps/sim/tools/memory/add_memory.ts b/apps/sim/tools/memory/add_memory.ts
index 7a85660d30..f6c04f8bda 100644
--- a/apps/sim/tools/memory/add_memory.ts
+++ b/apps/sim/tools/memory/add_memory.ts
@@ -142,7 +142,7 @@ export const memoryAddTool: ToolConfig<any, MemoryResponse> = {
       return {
         success: true,
         output: {
-          memory: data.data,
+          memories: data.data,
           message: isNewMemory ? 'Memory created successfully' : 'Memory appended successfully'
         },
       }
@@ -150,7 +150,7 @@ export const memoryAddTool: ToolConfig<any, MemoryResponse> = {
       return {
         success: false,
         output: {
-          memory: undefined,
+          memories: undefined,
           message: `Failed to add memory: ${error.message || 'Unknown error occurred'}`
         },
       }
@@ -161,7 +161,7 @@ export const memoryAddTool: ToolConfig<any, MemoryResponse> = {
     return {
       success: false,
       output: {
-        memory: undefined,
+        memories: undefined,
         message: `Memory operation failed: ${error.message || 'Unknown error occurred'}`
       },
       error: errorMessage
diff --git a/apps/sim/tools/memory/get_memory.ts b/apps/sim/tools/memory/get_memory.ts
index e97aad358d..9a3e18b651 100644
--- a/apps/sim/tools/memory/get_memory.ts
+++ b/apps/sim/tools/memory/get_memory.ts
@@ -56,7 +56,7 @@ export const memoryGetTool: ToolConfig<any, MemoryResponse> = {
       return {
         success: true,
         output: {
-          memory: data.data,
+          memories: data.data,
           message: 'Memory retrieved successfully'
         },
       }
@@ -64,7 +64,7 @@ export const memoryGetTool: ToolConfig<any, MemoryResponse> = {
       return {
         success: false,
         output: {
-          memory: undefined,
+          memories: undefined,
           message: `Failed to retrieve memory: ${error.message || 'Unknown error'}`
         },
         error: `Failed to retrieve memory: ${error.message || 'Unknown error'}`
@@ -76,7 +76,7 @@ export const memoryGetTool: ToolConfig<any, MemoryResponse> = {
     return {
       success: false,
       output: {
-        memory: undefined,
+        memories: undefined,
         message: errorMessage
       },
       error: errorMessage