@assistant-ui/core 0.3.8 → 0.3.10

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/core",
-  "version": "0.3.8",
+  "version": "0.3.10",
   "description": "Framework-agnostic core runtime for assistant-ui",
   "keywords": [
     "assistant",
@@ -59,8 +59,8 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-stream": "^0.3.35",
-    "nanoid": "^6.0.0"
+    "assistant-stream": "^0.3.36",
+    "nanoid": "^6.0.1"
   },
   "optionalDevDependencies": {
     "@assistant-ui/x-generative-compiler": ">=0.0.3"
@@ -89,17 +89,17 @@
   },
   "devDependencies": {
     "@testing-library/react": "^16.3.2",
-    "@types/react": "^19.2.17",
-    "jsdom": "^29.1.1",
+    "@types/react": "^19.2.18",
+    "jsdom": "^30.0.1",
     "react": "^19.2.8",
     "react-dom": "^19.2.8",
     "vitest": "^4.1.10",
     "zustand": "^5.0.14",
-    "@assistant-ui/store": "0.3.7",
-    "@assistant-ui/tap": "0.9.10",
-    "@assistant-ui/vite": "0.0.11",
-    "@assistant-ui/x-buildutils": "0.0.21",
-    "assistant-cloud": "0.1.38"
+    "@assistant-ui/store": "0.3.8",
+    "@assistant-ui/tap": "0.9.11",
+    "@assistant-ui/vite": "0.0.12",
+    "@assistant-ui/x-buildutils": "0.0.22",
+    "assistant-cloud": "0.1.39"
   },
   "publishConfig": {
     "access": "public",

package/src/runtimes/external-store/external-store-thread-runtime-core.ts

@@ -466,7 +466,11 @@ export class ExternalStoreThreadRuntimeCore
   }
 
   public async append(message: AppendMessage): Promise<void> {
-    const isEdit = message.parentId !== (this.messages.at(-1)?.id ?? null);
+    // sourceId marks an edit send; the parent may coincide with the head
+    // after a resync (e.g. cancelRun dropped the edited message).
+    const isEdit =
+      message.sourceId != null ||
+      message.parentId !== (this.messages.at(-1)?.id ?? null);
 
     // Buffering does not start a run, so the tool-abort below must wait until
     // the queue flushes. By then the prior run (and its tools) has settled.

package/src/runtimes/local/local-thread-runtime-core.test.ts

@@ -1186,12 +1186,10 @@ describe("LocalThreadRuntimeCore runs", () => {
   };
 
   it("appends the user message and the adapter result", async () => {
-    const run = vi.fn(
-      async (): Promise<ChatModelRunResult> => ({
-        content: [{ type: "text", text: "Hello!" }],
-        status: { type: "complete", reason: "stop" },
-      }),
-    );
+    const run = vi.fn(async (): Promise<ChatModelRunResult> => ({
+      content: [{ type: "text", text: "Hello!" }],
+      status: { type: "complete", reason: "stop" },
+    }));
     const thread = createPlainThread({ run });
 
     await thread.append(userMessage("Hi"));
@@ -1247,23 +1245,21 @@ describe("LocalThreadRuntimeCore runs", () => {
   });
 
   it("does not run again after a tool result once maxSteps is reached", async () => {
-    const run = vi.fn(
-      async (): Promise<ChatModelRunResult> => ({
-        content: [
-          {
-            type: "tool-call",
-            toolCallId: "tc1",
-            toolName: "myTool",
-            args: {},
-            argsText: "{}",
-          },
-        ],
-        status: { type: "requires-action", reason: "tool-calls" },
-        metadata: {
-          steps: [{ usage: { promptTokens: 10, completionTokens: 5 } }],
+    const run = vi.fn(async (): Promise<ChatModelRunResult> => ({
+      content: [
+        {
+          type: "tool-call",
+          toolCallId: "tc1",
+          toolName: "myTool",
+          args: {},
+          argsText: "{}",
         },
-      }),
-    );
+      ],
+      status: { type: "requires-action", reason: "tool-calls" },
+      metadata: {
+        steps: [{ usage: { promptTokens: 10, completionTokens: 5 } }],
+      },
+    }));
     const thread = createPlainThread({ run }, { maxSteps: 1 });
 
     await thread.append(userMessage("Tool call"));

package/src/tests/external-store-thread-runtime-core.test.ts

@@ -883,6 +883,42 @@ describe("ExternalStoreThreadRuntimeCore - message queue", () => {
     expect(queue.steer).toHaveBeenCalledTimes(1);
   });
 
+  it("routes an edit send to onEdit even when anchored at the head", async () => {
+    const queue = makeQueue();
+    const onNew = vi.fn();
+    const onEdit = vi.fn();
+    const runtime = new ExternalStoreThreadRuntimeCore(
+      mockContextProvider,
+      makeStore({ queue, onNew, onEdit }),
+    );
+
+    await runtime.append(appendMessage({ sourceId: "u1" }));
+
+    expect(onEdit).toHaveBeenCalledTimes(1);
+    expect(onEdit.mock.calls[0]![0]).toMatchObject({
+      sourceId: "u1",
+      parentId: null,
+    });
+    expect(onNew).not.toHaveBeenCalled();
+    expect(queue.enqueue).not.toHaveBeenCalled();
+    expect(queue.steer).not.toHaveBeenCalled();
+  });
+
+  it("throws the edit capability error for an edit send instead of queueing it", async () => {
+    const queue = makeQueue();
+    const onNew = vi.fn();
+    const runtime = new ExternalStoreThreadRuntimeCore(
+      mockContextProvider,
+      makeStore({ queue, onNew }),
+    );
+
+    await expect(
+      runtime.append(appendMessage({ sourceId: "u1" })),
+    ).rejects.toThrow("Runtime does not support editing messages.");
+    expect(onNew).not.toHaveBeenCalled();
+    expect(queue.enqueue).not.toHaveBeenCalled();
+  });
+
   it("defaults a mid-run send to steer and an idle send to enqueue", async () => {
     const queue = makeQueue();
     const running = new ExternalStoreThreadRuntimeCore(