fixed spaces memories appearance-reappearance bug

2 files changed, 7 insertions, 4 deletions

diff --git a/apps/web/src/components/Sidebar/MemoriesBar.tsx b/apps/web/src/components/Sidebar/MemoriesBar.tsx
index 1c061451..f474262a 100644
--- a/apps/web/src/components/Sidebar/MemoriesBar.tsx
+++ b/apps/web/src/components/Sidebar/MemoriesBar.tsx
@@ -411,7 +411,6 @@ export function SpaceMoreButton({
           </DropdownMenuItem>
 					<DialogTrigger asChild>
 						<DropdownMenuItem
-							onClick={onDelete}
 							className="focus:bg-red-100 focus:text-red-400 dark:focus:bg-red-100/10"
 						>
 							<Trash2 className="mr-2 h-4 w-4" strokeWidth={1.5} />
diff --git a/apps/web/src/contexts/MemoryContext.tsx b/apps/web/src/contexts/MemoryContext.tsx
index 08ffbcf5..67e6250e 100644
--- a/apps/web/src/contexts/MemoryContext.tsx
+++ b/apps/web/src/contexts/MemoryContext.tsx
@@ -1,7 +1,7 @@
 "use client";
 import React, { useCallback } from "react";
 import { ChachedSpaceContent, StoredContent, storedContent, StoredSpace } from "@/server/db/schema";
-import { addMemory, searchMemoriesAndSpaces, addSpace, fetchContentForSpace, deleteSpace, deleteMemory } from "@/actions/db";
+import { addMemory, searchMemoriesAndSpaces, addSpace, fetchContentForSpace, deleteSpace, deleteMemory, fetchFreeMemories } from "@/actions/db";
 import { User } from "next-auth";
 
 export type SearchResult = {
@@ -54,6 +54,8 @@ export const MemoryProvider: React.FC<
 		setSpaces(prev => prev.filter(i => i.id !== deleted.id))
 		setCachedMemories(prev => prev.filter(i => i.space !== deleted.id))
 
+		setFreeMemories(await fetchFreeMemories())
+
 		return deleted
 	}
 
@@ -61,7 +63,7 @@ export const MemoryProvider: React.FC<
 		const deleted = (await deleteMemory(...params))!
 
 		setCachedMemories(prev => prev.filter(i => i.id !== deleted.id))
-		setFreeMemories(prev => prev.filter(i => i.id !== deleted.id))
+		setFreeMemories(await fetchFreeMemories())
 
 		return deleted
 	}
@@ -79,9 +81,11 @@ export const MemoryProvider: React.FC<
 			offset: 0,
 			limit: 3
 		})).map(m => ({ ...m, space: addedSpace.id }))
-
+		
 		setCachedMemories(prev => [...prev, ...cachedMemories])
 
+		setFreeMemories(await fetchFreeMemories())
+		
 		return {
 			space: addedSpace, addedMemories
 		}