From 20b81ddb80c63ef08f9555f56d7c2caf75932184 Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Tue, 4 Mar 2025 14:47:16 +0100
Subject: [PATCH] Add optimistic values for editing

---
 src/components/thread/messages/human.tsx | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/src/components/thread/messages/human.tsx b/src/components/thread/messages/human.tsx
index aeef2dc..340a634 100644
--- a/src/components/thread/messages/human.tsx
+++ b/src/components/thread/messages/human.tsx
@@ -49,9 +49,23 @@ export function HumanMessage({
 
   const handleSubmitEdit = () => {
     setIsEditing(false);
+
+    const newMessage: Message = { type: "human", content: value };
     thread.submit(
-      { messages: [{ type: "human", content: value }] },
-      { checkpoint: parentCheckpoint },
+      { messages: [newMessage] },
+      {
+        checkpoint: parentCheckpoint,
+        streamMode: ["values"],
+        optimisticValues: (prev) => {
+          const values = meta?.firstSeenState?.values;
+          if (!values) return prev;
+
+          return {
+            ...values,
+            messages: [...(values.messages ?? []), newMessage],
+          };
+        },
+      },
     );
   };