fix: stabilize streamed answer completion

2026-07-19 15:26:55 +08:00
parent c22dea9979
commit 695a969c57
4 changed files with 141 additions and 37 deletions
@@ -0,0 +1,33 @@
+export type ChatMessage = {
+  readonly role: "user" | "assistant";
+  readonly text: string;
+  readonly suggestions?: readonly string[];
+};
+
+export type ChatMessageView = ChatMessage & {
+  readonly renderKey: string;
+  readonly state: "settled" | "streaming" | "thinking";
+};
+
+export function chatMessageViews(
+  messages: readonly ChatMessage[],
+  loading: boolean,
+  streamingText: string,
+): readonly ChatMessageView[] {
+  const settled = messages.map((message, index) => ({
+    ...message,
+    renderKey: `message-${index}`,
+    state: "settled" as const,
+  }));
+  if (!loading || messages.at(-1)?.role === "assistant") return settled;
+
+  return [
+    ...settled,
+    {
+      role: "assistant",
+      text: streamingText,
+      renderKey: `message-${messages.length}`,
+      state: streamingText ? "streaming" : "thinking",
+    },
+  ];
+}