fixed spaces memories appearance-reappearance bug

1 files changed, 7 insertions, 3 deletions

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
 		}