diff --git a/packages/ui/src/stores/message-v2/instance-store.ts b/packages/ui/src/stores/message-v2/instance-store.ts index df10c648..459e02ec 100644 --- a/packages/ui/src/stores/message-v2/instance-store.ts +++ b/packages/ui/src/stores/message-v2/instance-store.ts @@ -301,10 +301,10 @@ export function createInstanceMessageStore(instanceId: string): InstanceMessageS } } - setState("messages", reconcile(nextMessages)) - setState("messageInfoVersion", reconcile(nextMessageInfoVersion)) - setState("pendingParts", reconcile(nextPendingParts)) - setState("permissions", "byMessage", reconcile(nextPermissionsByMessage)) + setState("messages", (prev) => ({ ...prev, ...nextMessages })) + setState("messageInfoVersion", (prev) => ({ ...prev, ...nextMessageInfoVersion })) + setState("pendingParts", (prev) => ({ ...prev, ...nextPendingParts })) + setState("permissions", "byMessage", (prev) => ({ ...prev, ...nextPermissionsByMessage })) if (usageState) { setState("usage", sessionId, usageState)