fix(stage-ui): user-sent message lost (#1215)

typed-sigterm · web-flow · commit 5b821c0ee03b · 2026-03-09T20:45:16.000+08:00
diff --git a/packages/stage-ui/src/stores/chat.ts b/packages/stage-ui/src/stores/chat.ts
@@ -170,7 +170,10 @@ export const useChatOrchestratorStore = defineStore('chat-orchestrator', () => {
       if (shouldAbort())
         return
 
-      const sessionMessagesForSend = chatSession.getSessionMessages(sessionId)
+      const sessionMessagesForSend = chatSession.sessionMessages[sessionId]
+      if (!sessionMessagesForSend) {
+        throw new Error('Session messages not found')
+      }
       sessionMessagesForSend.push({ role: 'user', content: finalContent, createdAt: sendingCreatedAt, id: nanoid() })
       chatSession.persistSessionMessages(sessionId)
 
diff --git a/packages/stage-ui/src/stores/chat/session-store.ts b/packages/stage-ui/src/stores/chat/session-store.ts
@@ -546,6 +546,8 @@ export const useChatSessionStore = defineStore('chat-session', () => {
     setSessionMessages,
     persistSessionMessages,
     getSessionMessages,
+    sessionMessages,
+    sessionMetas,
     getSessionGeneration,
     bumpSessionGeneration,
     getSessionGenerationValue,
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
