@subako-ai/assistant-ui 0.2.0 → 0.2.1

package/dist/convert.d.ts

@@ -1,5 +1,5 @@
 import type { ThreadMessageLike } from "@assistant-ui/react";
-import type { TranscriptMessage } from "@subako-ai/sdk";
+import { type LiveMessage, type TranscriptMessage } from "@subako-ai/sdk";
 /**
  * One message of the log as assistant-ui reads it: `text` becomes a text part,
  * `thinking` a reasoning part, and every `tool_call` the joined call, so its
@@ -11,4 +11,11 @@ import type { TranscriptMessage } from "@subako-ai/sdk";
  * handing back the same message until an event touches it.
  */
 export declare function convertMessage(message: TranscriptMessage): ThreadMessageLike;
+/**
+ * A message a model call is still producing, as assistant-ui reads it: an
+ * assistant message marked running, whose tool calls carry the argument text
+ * streamed so far. Its id is the model call's, so it keeps its place in the
+ * thread from one frame to the next until the logged message replaces it.
+ */
+export declare function convertLiveMessage(message: LiveMessage): ThreadMessageLike;
 //# sourceMappingURL=convert.d.ts.map

package/dist/convert.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"convert.d.ts","sourceRoot":"","sources":["../src/convert.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,KAAK,EAAoB,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAkD1E;;;;;;;;;GASG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,iBAAiB,GAAG,iBAAiB,CA0B5E"}
+{"version":3,"file":"convert.d.ts","sourceRoot":"","sources":["../src/convert.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAC7D,OAAO,EAGN,KAAK,WAAW,EAEhB,KAAK,iBAAiB,EACtB,MAAM,gBAAgB,CAAC;AAkDxB;;;;;;;;;GASG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,iBAAiB,GAAG,iBAAiB,CA0B5E;AAyCD;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,WAAW,GAAG,iBAAiB,CAO1E"}

package/dist/convert.js

@@ -1,3 +1,4 @@
+import { liveKey, } from "@subako-ai/sdk";
 /** The arguments arrive as JSON off the wire; assistant-ui asks for them typed as JSON. */
 const asArguments = (args) => args;
 /** A tool result is content blocks; assistant-ui renders one value, so the blocks are read as text. */
@@ -68,4 +69,56 @@ export function convertMessage(message) {
         ...(message.source === undefined ? {} : { metadata: { custom: { source: message.source } } }),
     };
 }
+/**
+ * The arguments a call streamed so far, once they parse; `undefined` while
+ * they are still partial. JSON that ends in `}` and parses is an object, as
+ * arguments are, and most frames leave the text mid-value, so nothing else is
+ * parsed at all.
+ */
+function partialArguments(text) {
+    if (!text.trimEnd().endsWith("}")) {
+        return undefined;
+    }
+    try {
+        return asArguments(JSON.parse(text));
+    }
+    catch {
+        return undefined;
+    }
+}
+function livePart(block) {
+    switch (block.type) {
+        case "text":
+            return { type: "text", text: block.text };
+        case "thinking":
+            return { type: "reasoning", text: block.thinking };
+        case "tool_call": {
+            const args = partialArguments(block.arguments);
+            return {
+                type: "tool-call",
+                toolCallId: block.id,
+                toolName: block.name,
+                argsText: block.arguments,
+                ...(args === undefined ? {} : { args }),
+            };
+        }
+        // COVERAGE(never): a live block has no other type, which `satisfies never` proves.
+        default:
+            return block;
+    }
+}
+/**
+ * A message a model call is still producing, as assistant-ui reads it: an
+ * assistant message marked running, whose tool calls carry the argument text
+ * streamed so far. Its id is the model call's, so it keeps its place in the
+ * thread from one frame to the next until the logged message replaces it.
+ */
+export function convertLiveMessage(message) {
+    return {
+        role: "assistant",
+        id: `live:${liveKey(message.target)}`,
+        content: message.content.map(livePart),
+        status: { type: "running" },
+    };
+}
 //# sourceMappingURL=convert.js.map

package/dist/convert.js.map

@@ -1 +1 @@
-{"version":3,"file":"convert.js","sourceRoot":"","sources":["../src/convert.ts"],"names":[],"mappings":"AAYA,2FAA2F;AAC3F,MAAM,WAAW,GAAG,CAAC,IAA6B,EAAqC,EAAE,CACxF,IAAyC,CAAC;AAE3C,uGAAuG;AACvG,MAAM,UAAU,GAAG,CAAC,OAAoC,EAAU,EAAE,CACnE,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAE3G;;;GAGG;AACH,SAAS,UAAU,CAAC,IAAoB;IACvC,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,KAAK,SAAS;YACb,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QACxB,KAAK,SAAS;YACb,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QACxC,KAAK,QAAQ;YACZ,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACzC,2EAA2E;QAC3E,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACb,OAAO,SAAS,CAAC;IACnB,CAAC;AACF,CAAC;AAED,SAAS,YAAY,CAAC,IAAoB;IACzC,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAClC,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,UAAU,EAAE,IAAI,CAAC,EAAE;QACnB,QAAQ,EAAE,IAAI,CAAC,IAAI;QACnB,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC;QACjC,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QAChH,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC;KAC/C,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,cAAc,CAAC,OAA0B;IACxD,8FAA8F;IAC9F,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IAChF,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC7C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,MAAM;gBACV,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtD,MAAM;YACP,KAAK,UAAU;gBACd,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC/D,MAAM;YACP,KAAK,WAAW;gBACf,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC1E,MAAM;YACP;gBACC,MAAM;QACR,CAAC;IACF,CAAC;IACD,OAAO;QACN,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;QACvB,OAAO;QACP,sFAAsF;QACtF,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;KAC7F,CAAC;AACH,CAAC"}
+{"version":3,"file":"convert.js","sourceRoot":"","sources":["../src/convert.ts"],"names":[],"mappings":"AACA,OAAO,EAIN,OAAO,GAEP,MAAM,gBAAgB,CAAC;AAWxB,2FAA2F;AAC3F,MAAM,WAAW,GAAG,CAAC,IAA6B,EAAqC,EAAE,CACxF,IAAyC,CAAC;AAE3C,uGAAuG;AACvG,MAAM,UAAU,GAAG,CAAC,OAAoC,EAAU,EAAE,CACnE,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAE3G;;;GAGG;AACH,SAAS,UAAU,CAAC,IAAoB;IACvC,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,KAAK,SAAS;YACb,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC;QACxB,KAAK,SAAS;YACb,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QACxC,KAAK,QAAQ;YACZ,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACzC,2EAA2E;QAC3E,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACb,OAAO,SAAS,CAAC;IACnB,CAAC;AACF,CAAC;AAED,SAAS,YAAY,CAAC,IAAoB;IACzC,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;IAClC,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,UAAU,EAAE,IAAI,CAAC,EAAE;QACnB,QAAQ,EAAE,IAAI,CAAC,IAAI;QACnB,IAAI,EAAE,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC;QACjC,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QAChH,GAAG,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC;KAC/C,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,MAAM,UAAU,cAAc,CAAC,OAA0B;IACxD,8FAA8F;IAC9F,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;IAChF,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC7C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACpB,KAAK,MAAM;gBACV,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBACtD,MAAM;YACP,KAAK,UAAU;gBACd,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC/D,MAAM;YACP,KAAK,WAAW;gBACf,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC1E,MAAM;YACP;gBACC,MAAM;QACR,CAAC;IACF,CAAC;IACD,OAAO;QACN,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,EAAE,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC;QACvB,OAAO;QACP,sFAAsF;QACtF,GAAG,CAAC,OAAO,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC;KAC7F,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,SAAS,gBAAgB,CAAC,IAAY;IACrC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;QACnC,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,IAAI,CAAC;QACJ,OAAO,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAA4B,CAAC,CAAC;IACjE,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,SAAS,CAAC;IAClB,CAAC;AACF,CAAC;AAED,SAAS,QAAQ,CAAC,KAAgB;IACjC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;QACpB,KAAK,MAAM;YACV,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;QAC3C,KAAK,UAAU;YACd,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;QACpD,KAAK,WAAW,EAAE,CAAC;YAClB,MAAM,IAAI,GAAG,gBAAgB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC/C,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,KAAK,CAAC,EAAE;gBACpB,QAAQ,EAAE,KAAK,CAAC,IAAI;gBACpB,QAAQ,EAAE,KAAK,CAAC,SAAS;gBACzB,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC;aACvC,CAAC;QACH,CAAC;QACD,mFAAmF;QACnF;YACC,OAAO,KAAqB,CAAC;IAC/B,CAAC;AACF,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAAoB;IACtD,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,EAAE,EAAE,QAAQ,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;QACrC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;QACtC,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;KAC3B,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -1,4 +1,4 @@
 export { SubakoChat, type SubakoChatProps, SubakoToolApproval, type SubakoToolApprovalProps } from "./chat.js";
-export { convertMessage } from "./convert.js";
+export { convertLiveMessage, convertMessage } from "./convert.js";
 export { type SubakoSession, useSubakoRuntime } from "./runtime.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,eAAe,EAAE,kBAAkB,EAAE,KAAK,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAC/G,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,KAAK,aAAa,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,KAAK,eAAe,EAAE,kBAAkB,EAAE,KAAK,uBAAuB,EAAE,MAAM,WAAW,CAAC;AAC/G,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAClE,OAAO,EAAE,KAAK,aAAa,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC"}

package/dist/index.js

@@ -1,4 +1,4 @@
 export { SubakoChat, SubakoToolApproval } from "./chat.js";
-export { convertMessage } from "./convert.js";
+export { convertLiveMessage, convertMessage } from "./convert.js";
 export { useSubakoRuntime } from "./runtime.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAwB,kBAAkB,EAAgC,MAAM,WAAW,CAAC;AAC/G,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAsB,gBAAgB,EAAE,MAAM,cAAc,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAwB,kBAAkB,EAAgC,MAAM,WAAW,CAAC;AAC/G,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAClE,OAAO,EAAsB,gBAAgB,EAAE,MAAM,cAAc,CAAC"}

package/dist/runtime.d.ts

@@ -22,7 +22,7 @@ export interface SubakoSession {
  *
  * | assistant-ui              | Subako                                        |
  * | ------------------------- | --------------------------------------------- |
- * | `messages`                | `state.transcript`, through `convertMessage`  |
+ * | `messages`                | `state.transcript`, then `state.live`         |
  * | `isRunning`               | `state.isRunning`                             |
  * | `onNew`                   | `send(text)`                                  |
  * | `onCancel`                | `cancel()`                                    |
@@ -36,6 +36,10 @@ export interface SubakoSession {
  * The session is a value, not something React re-renders for, so the runtime
  * subscribes to it here: a chat handed the session by a parent that does not
  * re-render still follows the log.
+ *
+ * A session opened with `deltas` also shows what a model is writing while it
+ * writes it: each of `state.live` follows the log as a running assistant
+ * message, which the logged message replaces once it lands.
  */
 export declare function useSubakoRuntime(session: SubakoSession | null): AssistantRuntime;
 //# sourceMappingURL=runtime.d.ts.map

package/dist/runtime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAiB,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAG3E,OAAO,KAAK,EAAE,YAAY,EAAqB,MAAM,gBAAgB,CAAC;AAGtE;;;;GAIG;AACH,MAAM,WAAW,aAAa;IAC7B,sGAAsG;IACtG,QAAQ,CAAC,KAAK,EAAE,YAAY,CAAC;IAC7B,kFAAkF;IAClF,QAAQ,CAAC,SAAS,EAAE,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,KAAK,MAAM,IAAI,CAAC;IAC5E,sEAAsE;IACtE,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClC,8BAA8B;IAC9B,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACxB,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7F;AAgBD;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,GAAG,gBAAgB,CAmBhF"}
+{"version":3,"file":"runtime.d.ts","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAiB,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAG3E,OAAO,KAAK,EAAe,YAAY,EAAqB,MAAM,gBAAgB,CAAC;AAInF;;;;GAIG;AACH,MAAM,WAAW,aAAa;IAC7B,sGAAsG;IACtG,QAAQ,CAAC,KAAK,EAAE,YAAY,CAAC;IAC7B,kFAAkF;IAClF,QAAQ,CAAC,SAAS,EAAE,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,YAAY,KAAK,IAAI,KAAK,MAAM,IAAI,CAAC;IAC5E,sEAAsE;IACtE,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAClC,8BAA8B;IAC9B,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IACxB,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,GAAG,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC7F;AAqBD;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI,GAAG,gBAAgB,CA+BhF"}

package/dist/runtime.js

@@ -1,9 +1,11 @@
 "use client";
 import { useExternalStoreRuntime } from "@assistant-ui/react";
 import { useSessionState } from "@subako-ai/react";
-import { convertMessage } from "./convert.js";
+import { useMemo } from "react";
+import { convertLiveMessage, convertMessage } from "./convert.js";
 /** The thread before the connection is open: no messages, and the same array on every render. */
 const NOTHING = [];
+const convertEntry = (entry) => ("target" in entry ? convertLiveMessage(entry) : convertMessage(entry));
 /** assistant-ui hands the composer's parts; a session takes one string. */
 function appendedText(message) {
     let text = "";
@@ -20,7 +22,7 @@ function appendedText(message) {
  *
  * | assistant-ui              | Subako                                        |
  * | ------------------------- | --------------------------------------------- |
- * | `messages`                | `state.transcript`, through `convertMessage`  |
+ * | `messages`                | `state.transcript`, then `state.live`         |
  * | `isRunning`               | `state.isRunning`                             |
  * | `onNew`                   | `send(text)`                                  |
  * | `onCancel`                | `cancel()`                                    |
@@ -34,15 +36,27 @@ function appendedText(message) {
  * The session is a value, not something React re-renders for, so the runtime
  * subscribes to it here: a chat handed the session by a parent that does not
  * re-render still follows the log.
+ *
+ * A session opened with `deltas` also shows what a model is writing while it
+ * writes it: each of `state.live` follows the log as a running assistant
+ * message, which the logged message replaces once it lands.
  */
 export function useSubakoRuntime(session) {
     // The snapshot this render was woken for, which is what the thread is built from.
     const state = useSessionState(session);
+    const transcript = state?.transcript;
+    const live = state?.live;
+    // Rebuilt only when either part changes, so assistant-ui's per-message cache stays warm.
+    const messages = useMemo(() => transcript === undefined
+        ? NOTHING
+        : live === undefined || live.length === 0
+            ? transcript
+            : [...transcript, ...live], [transcript, live]);
     return useExternalStoreRuntime({
-        messages: state?.transcript ?? NOTHING,
+        messages,
         isRunning: state?.isRunning ?? false,
         isDisabled: session === null,
-        convertMessage,
+        convertMessage: convertEntry,
         onNew: async (message) => {
             await session?.send(appendedText(message));
         },

package/dist/runtime.js.map

@@ -1 +1 @@
-{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAGb,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAmB9C,iGAAiG;AACjG,MAAM,OAAO,GAAiC,EAAE,CAAC;AAEjD,2EAA2E;AAC3E,SAAS,YAAY,CAAC,OAAsB;IAC3C,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;IACF,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,UAAU,gBAAgB,CAAC,OAA6B;IAC7D,kFAAkF;IAClF,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IACvC,OAAO,uBAAuB,CAAoB;QACjD,QAAQ,EAAE,KAAK,EAAE,UAAU,IAAI,OAAO;QACtC,SAAS,EAAE,KAAK,EAAE,SAAS,IAAI,KAAK;QACpC,UAAU,EAAE,OAAO,KAAK,IAAI;QAC5B,cAAc;QACd,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,MAAM,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;QAC5C,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACpB,MAAM,OAAO,EAAE,MAAM,EAAE,CAAC;QACzB,CAAC;QACD,gFAAgF;QAChF,uBAAuB,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE;YACnE,MAAM,OAAO,EAAE,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjF,CAAC;KACD,CAAC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAGb,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAEnD,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAsBlE,iGAAiG;AACjG,MAAM,OAAO,GAA2B,EAAE,CAAC;AAE3C,MAAM,YAAY,GAAG,CAAC,KAAkB,EAAE,EAAE,CAAC,CAAC,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,CAAC;AAErH,2EAA2E;AAC3E,SAAS,YAAY,CAAC,OAAsB;IAC3C,IAAI,IAAI,GAAG,EAAE,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC;QACnB,CAAC;IACF,CAAC;IACD,OAAO,IAAI,CAAC;AACb,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,UAAU,gBAAgB,CAAC,OAA6B;IAC7D,kFAAkF;IAClF,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IACvC,MAAM,UAAU,GAAG,KAAK,EAAE,UAAU,CAAC;IACrC,MAAM,IAAI,GAAG,KAAK,EAAE,IAAI,CAAC;IACzB,yFAAyF;IACzF,MAAM,QAAQ,GAAG,OAAO,CACvB,GAA2B,EAAE,CAC5B,UAAU,KAAK,SAAS;QACvB,CAAC,CAAC,OAAO;QACT,CAAC,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YACxC,CAAC,CAAC,UAAU;YACZ,CAAC,CAAC,CAAC,GAAG,UAAU,EAAE,GAAG,IAAI,CAAC,EAC7B,CAAC,UAAU,EAAE,IAAI,CAAC,CAClB,CAAC;IACF,OAAO,uBAAuB,CAAc;QAC3C,QAAQ;QACR,SAAS,EAAE,KAAK,EAAE,SAAS,IAAI,KAAK;QACpC,UAAU,EAAE,OAAO,KAAK,IAAI;QAC5B,cAAc,EAAE,YAAY;QAC5B,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,MAAM,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;QAC5C,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACpB,MAAM,OAAO,EAAE,MAAM,EAAE,CAAC;QACzB,CAAC;QACD,gFAAgF;QAChF,uBAAuB,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE;YACnE,MAAM,OAAO,EAAE,eAAe,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjF,CAAC;KACD,CAAC,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@subako-ai/assistant-ui",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "assistant-ui runtime and chat for Subako sessions",
   "homepage": "https://subako.ai",
   "license": "Apache-2.0",
@@ -39,15 +39,15 @@
     "react": "^19.2.0",
     "react-dom": "^19.2.0",
     "typescript": "^7",
-    "@subako-ai/sdk": "0.2.0",
-    "@subako-ai/react": "0.2.0"
+    "@subako-ai/react": "0.2.1",
+    "@subako-ai/sdk": "0.2.1"
   },
   "peerDependencies": {
     "@assistant-ui/react": "^0.15.0",
     "@assistant-ui/react-markdown": "^0.14.15",
     "react": "^19.0.0",
-    "@subako-ai/react": "^0.2.0",
-    "@subako-ai/sdk": "^0.2.0"
+    "@subako-ai/react": "^0.2.1",
+    "@subako-ai/sdk": "^0.2.1"
   },
   "engines": {
     "node": ">=22"

package/src/convert.ts

@@ -1,5 +1,11 @@
 import type { ThreadMessageLike } from "@assistant-ui/react";
-import type { ContentBlockBody, TranscriptMessage } from "@subako-ai/sdk";
+import {
+	type ContentBlockBody,
+	type LiveBlock,
+	type LiveMessage,
+	liveKey,
+	type TranscriptMessage,
+} from "@subako-ai/sdk";
 
 /** One part of an assistant-ui message; its content is either a string or these. */
 type ContentPart = Extract<ThreadMessageLike["content"], readonly unknown[]>[number];
@@ -86,3 +92,57 @@ export function convertMessage(message: TranscriptMessage): ThreadMessageLike {
 		...(message.source === undefined ? {} : { metadata: { custom: { source: message.source } } }),
 	};
 }
+
+/**
+ * The arguments a call streamed so far, once they parse; `undefined` while
+ * they are still partial. JSON that ends in `}` and parses is an object, as
+ * arguments are, and most frames leave the text mid-value, so nothing else is
+ * parsed at all.
+ */
+function partialArguments(text: string): NonNullable<ToolCallPart["args"]> | undefined {
+	if (!text.trimEnd().endsWith("}")) {
+		return undefined;
+	}
+	try {
+		return asArguments(JSON.parse(text) as Record<string, unknown>);
+	} catch {
+		return undefined;
+	}
+}
+
+function livePart(block: LiveBlock): ContentPart {
+	switch (block.type) {
+		case "text":
+			return { type: "text", text: block.text };
+		case "thinking":
+			return { type: "reasoning", text: block.thinking };
+		case "tool_call": {
+			const args = partialArguments(block.arguments);
+			return {
+				type: "tool-call",
+				toolCallId: block.id,
+				toolName: block.name,
+				argsText: block.arguments,
+				...(args === undefined ? {} : { args }),
+			};
+		}
+		// COVERAGE(never): a live block has no other type, which `satisfies never` proves.
+		default:
+			return block satisfies never;
+	}
+}
+
+/**
+ * A message a model call is still producing, as assistant-ui reads it: an
+ * assistant message marked running, whose tool calls carry the argument text
+ * streamed so far. Its id is the model call's, so it keeps its place in the
+ * thread from one frame to the next until the logged message replaces it.
+ */
+export function convertLiveMessage(message: LiveMessage): ThreadMessageLike {
+	return {
+		role: "assistant",
+		id: `live:${liveKey(message.target)}`,
+		content: message.content.map(livePart),
+		status: { type: "running" },
+	};
+}

package/src/index.ts

@@ -1,3 +1,3 @@
 export { SubakoChat, type SubakoChatProps, SubakoToolApproval, type SubakoToolApprovalProps } from "./chat.js";
-export { convertMessage } from "./convert.js";
+export { convertLiveMessage, convertMessage } from "./convert.js";
 export { type SubakoSession, useSubakoRuntime } from "./runtime.js";

package/src/runtime.ts

@@ -3,8 +3,9 @@
 import type { AppendMessage, AssistantRuntime } from "@assistant-ui/react";
 import { useExternalStoreRuntime } from "@assistant-ui/react";
 import { useSessionState } from "@subako-ai/react";
-import type { SessionState, TranscriptMessage } from "@subako-ai/sdk";
-import { convertMessage } from "./convert.js";
+import type { LiveMessage, SessionState, TranscriptMessage } from "@subako-ai/sdk";
+import { useMemo } from "react";
+import { convertLiveMessage, convertMessage } from "./convert.js";
 
 /**
  * What the runtime reads off a session: the SDK's `SessionConnection`, or
@@ -23,8 +24,13 @@ export interface SubakoSession {
 	resolveApproval(callId: string, decision: "allow" | "deny", message?: string): Promise<void>;
 }
 
+/** One message of the thread: one the log holds, or one a model call is still producing. */
+type ThreadEntry = TranscriptMessage | LiveMessage;
+
 /** The thread before the connection is open: no messages, and the same array on every render. */
-const NOTHING: readonly TranscriptMessage[] = [];
+const NOTHING: readonly ThreadEntry[] = [];
+
+const convertEntry = (entry: ThreadEntry) => ("target" in entry ? convertLiveMessage(entry) : convertMessage(entry));
 
 /** assistant-ui hands the composer's parts; a session takes one string. */
 function appendedText(message: AppendMessage): string {
@@ -43,7 +49,7 @@ function appendedText(message: AppendMessage): string {
  *
  * | assistant-ui              | Subako                                        |
  * | ------------------------- | --------------------------------------------- |
- * | `messages`                | `state.transcript`, through `convertMessage`  |
+ * | `messages`                | `state.transcript`, then `state.live`         |
  * | `isRunning`               | `state.isRunning`                             |
  * | `onNew`                   | `send(text)`                                  |
  * | `onCancel`                | `cancel()`                                    |
@@ -57,15 +63,31 @@ function appendedText(message: AppendMessage): string {
  * The session is a value, not something React re-renders for, so the runtime
  * subscribes to it here: a chat handed the session by a parent that does not
  * re-render still follows the log.
+ *
+ * A session opened with `deltas` also shows what a model is writing while it
+ * writes it: each of `state.live` follows the log as a running assistant
+ * message, which the logged message replaces once it lands.
  */
 export function useSubakoRuntime(session: SubakoSession | null): AssistantRuntime {
 	// The snapshot this render was woken for, which is what the thread is built from.
 	const state = useSessionState(session);
-	return useExternalStoreRuntime<TranscriptMessage>({
-		messages: state?.transcript ?? NOTHING,
+	const transcript = state?.transcript;
+	const live = state?.live;
+	// Rebuilt only when either part changes, so assistant-ui's per-message cache stays warm.
+	const messages = useMemo(
+		(): readonly ThreadEntry[] =>
+			transcript === undefined
+				? NOTHING
+				: live === undefined || live.length === 0
+					? transcript
+					: [...transcript, ...live],
+		[transcript, live],
+	);
+	return useExternalStoreRuntime<ThreadEntry>({
+		messages,
 		isRunning: state?.isRunning ?? false,
 		isDisabled: session === null,
-		convertMessage,
+		convertMessage: convertEntry,
 		onNew: async (message) => {
 			await session?.send(appendedText(message));
 		},