From c0edd6918fd34d35a6486263c1907459aac366ca Mon Sep 17 00:00:00 2001
From: Luca Stauble <luca.stauble@gmail.com>
Date: Thu, 19 Sep 2024 14:57:30 +0200
Subject: [PATCH] fix: update UI when remove tags from search results

---
 src/store/zustand/search/store.ts | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/src/store/zustand/search/store.ts b/src/store/zustand/search/store.ts
index 6304e9bfb..8aaa5fef4 100644
--- a/src/store/zustand/search/store.ts
+++ b/src/store/zustand/search/store.ts
@@ -133,10 +133,10 @@ export function updateConversationsOnly(conversations: Array<NormalizedConversat
 	useMessageStore.setState(
 		produce(({ populatedItems }: PopulatedItemsSliceState) => {
 			conversations.forEach((conversation) => {
-				populatedItems.conversations[conversation.id] = merge(
-					populatedItems.conversations[conversation.id],
-					conversation
-				);
+				populatedItems.conversations[conversation.id] = {
+					...merge(populatedItems.conversations[conversation.id], conversation),
+					tags: conversation.tags
+				};
 			});
 		})
 	);
@@ -170,7 +170,10 @@ export function updateMessagesOnly(messages: Array<IncompleteMessage>): void {
 	useMessageStore.setState(
 		produce(({ populatedItems }: PopulatedItemsSliceState) => {
 			messages.forEach((message) => {
-				populatedItems.messages[message.id] = merge(populatedItems.messages[message.id], message);
+				populatedItems.messages[message.id] = {
+					...merge(populatedItems.messages[message.id], message),
+					tags: message.tags
+				};
 			});
 		})
 	);