@assistant-ui/react-ag-ui 0.1.3 → 0.1.5

package/dist/useAgUiRuntime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useAgUiRuntime.d.ts","sourceRoot":"","sources":["../src/useAgUiRuntime.ts"],"names":[],"mappings":"AAuBA,OAAO,KAAK,EACV,gBAAgB,EAIjB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EACV,aAAa,EACb,eAAe,EACf,qBAAqB,EACtB,2BAAwB;AASzB,MAAM,MAAM,oBAAoB,GAAG,gBAAgB,GAAG;IACpD;;;OAGG;IACH,6BAA6B,EAAE,MAAM,SAAS,aAAa,EAAE,CAAC;IAC9D;;;;OAIG;IACH,iCAAiC,EAAE,CACjC,SAAS,EAAE,SAAS,eAAe,EAAE,KAClC,OAAO,CAAC,IAAI,CAAC,CAAC;CACpB,CAAC;AAEF,wBAAgB,cAAc,CAC5B,OAAO,EAAE,qBAAqB,GAC7B,oBAAoB,CAwWtB"}
+{"version":3,"file":"useAgUiRuntime.d.ts","sourceRoot":"","sources":["../src/useAgUiRuntime.ts"],"names":[],"mappings":"AAuBA,OAAO,KAAK,EACV,gBAAgB,EAIjB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EACV,aAAa,EACb,eAAe,EACf,qBAAqB,EACtB,2BAAwB;AASzB,MAAM,MAAM,oBAAoB,GAAG,gBAAgB,GAAG;IACpD;;;OAGG;IACH,6BAA6B,EAAE,MAAM,SAAS,aAAa,EAAE,CAAC;IAC9D;;;;OAIG;IACH,iCAAiC,EAAE,CACjC,SAAS,EAAE,SAAS,eAAe,EAAE,KAClC,OAAO,CAAC,IAAI,CAAC,CAAC;CACpB,CAAC;AAEF,wBAAgB,cAAc,CAC5B,OAAO,EAAE,qBAAqB,GAC7B,oBAAoB,CAyWtB"}

package/dist/useAgUiRuntime.js

@@ -199,6 +199,7 @@ function useAgUiRuntime(options) {
 				queueController?.clear();
 				return core.reload(parentId, config);
 			},
+			onDelete: (messageId) => core.deleteMessage(messageId),
 			onCancel: async () => {
 				queueController?.notifyCancelled();
 				core.cancel();

package/dist/useAgUiRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAgUiRuntime.js","names":[],"sources":["../src/useAgUiRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport {\n  useIsomorphicLayoutEffect,\n  useReplaySafeEffect,\n} from \"@assistant-ui/store/internal\";\nimport { createMessageQueue } from \"@assistant-ui/core\";\nimport type {\n  MessageQueueController,\n  ToolExecutionStatus,\n} from \"@assistant-ui/core\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { makeLogger } from \"./runtime/logger\";\nimport type {\n  AgUiInterrupt,\n  AgUiResumeEntry,\n  UseAgUiRuntimeOptions,\n} from \"./runtime/types\";\nimport { AgUiThreadController } from \"./runtime/AgUiThreadController\";\nimport { agUiExtras } from \"./agUiExtras\";\nimport type { QueueItemState } from \"@assistant-ui/core/store\";\n\nconst EMPTY_INTERRUPTS: readonly AgUiInterrupt[] = [];\nconst EMPTY_QUEUE_ITEMS: readonly QueueItemState[] = Object.freeze([]);\nconst subscribeNoop = () => () => {};\n\nexport type AgUiAssistantRuntime = AssistantRuntime & {\n  /**\n   * @deprecated Use the `useAgUiInterrupts()` hook instead. This method is kept\n   * for backward compatibility and will be removed in a future major release.\n   */\n  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];\n  /**\n   * @deprecated Use the `useAgUiSubmitInterruptResponses()` hook instead. This\n   * method is kept for backward compatibility and will be removed in a future\n   * major release.\n   */\n  unstable_submitInterruptResponses: (\n    responses: readonly AgUiResumeEntry[],\n  ) => Promise<void>;\n};\n\nexport function useAgUiRuntime(\n  options: UseAgUiRuntimeOptions,\n): AgUiAssistantRuntime {\n  const logger = useMemo(() => makeLogger(options.logger), [options.logger]);\n  const coreRef = useRef<AgUiThreadController | null>(null);\n  const threadSwitchGenerationRef = useRef(0);\n  const switchingGenerationRef = useRef<number | null>(null);\n  const runtimeAdapters = useRuntimeAdapters();\n\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  if (!coreRef.current) {\n    coreRef.current = new AgUiThreadController({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      joinStrategy: options.joinStrategy,\n      getRunContext: options.getRunContext,\n      resumeTranscript: options.resumeTranscript,\n      autoCancelPendingToolCalls: options.autoCancelPendingToolCalls,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n      isThreadSwitching: () => switchingGenerationRef.current !== null,\n    });\n  }\n\n  const core = coreRef.current;\n  const snapshot = useSyncExternalStore(\n    core.subscribe,\n    core.getSnapshot,\n    core.getSnapshot,\n  );\n  useEffect(() => {\n    core.updateOptions({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      joinStrategy: options.joinStrategy,\n      getRunContext: options.getRunContext,\n      resumeTranscript: options.resumeTranscript,\n      autoCancelPendingToolCalls: options.autoCancelPendingToolCalls,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n    });\n  });\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const isRunning = snapshot.isRunning || hasExecutingTools;\n\n  // The driver sends through the agent core rather than the runtime, because\n  // the runtime's append routes every tail append back into this queue.\n  const queueRef = useRef<MessageQueueController | null>(null);\n  // Counts rendered busy edges, whether a run start or an interrupt arriving.\n  // A dispatch whose count never moves was never observed as busy, so no\n  // falling edge is coming to release the queue.\n  const busyEdgesRef = useRef(0);\n  const lastQueueEdgeRef = useRef<{\n    controller: MessageQueueController;\n    busy: boolean;\n  } | null>(null);\n  if (options.unstable_enableMessageQueue && !queueRef.current) {\n    queueRef.current = createMessageQueue({\n      run: (message) => {\n        const controller = queueRef.current;\n        // A send dispatched mid-switch is dropped, and the queue released, or\n        // it would wait for an idle edge that never comes.\n        if (switchingGenerationRef.current !== null) {\n          controller?.notifyIdle();\n          return;\n        }\n        const edgesAtDispatch = busyEdgesRef.current;\n        // The queue drops the item before dispatching and stays busy until an\n        // idle edge. An append observed as busy is released by that falling\n        // edge, and releasing again here would advance the queue twice; one\n        // that never runs (a pre-run rejection, or startRun false) has to be\n        // released here or every later send buffers forever. An interrupt is\n        // rechecked because a run that resolves without yielding to React\n        // reaches here before the effect below has seen either edge.\n        const releaseIfNoRun = () => {\n          if (\n            queueRef.current !== controller ||\n            busyEdgesRef.current !== edgesAtDispatch ||\n            core.getPendingInterrupts() !== null\n          )\n            return;\n          controller?.notifyIdle();\n        };\n        void core.append(message).then(releaseIfNoRun, (e: unknown) => {\n          releaseIfNoRun();\n          logger.error?.(\"[agui] queued message failed to send\", e);\n        });\n      },\n    });\n  }\n  const queueController = options.unstable_enableMessageQueue\n    ? queueRef.current\n    : null;\n  useIsomorphicLayoutEffect(() => {\n    if (options.unstable_enableMessageQueue || !queueRef.current) return;\n    const controller = queueRef.current;\n    queueRef.current = null;\n    controller.clear();\n  }, [options.unstable_enableMessageQueue]);\n\n  // Feeds the store memo below: the runtime core skips an adapter whose\n  // identity is unchanged, so queue items have to move the store reference or\n  // subscribers never see composer.queue change.\n  const queueItems = useSyncExternalStore(\n    queueController?.subscribe ?? subscribeNoop,\n    () => queueController?.adapter.items ?? EMPTY_QUEUE_ITEMS,\n    () => EMPTY_QUEUE_ITEMS,\n  );\n  const steerQueueItems = useSyncExternalStore(\n    queueController?.subscribe ?? subscribeNoop,\n    () => queueController?.adapter.steerItems ?? EMPTY_QUEUE_ITEMS,\n    () => EMPTY_QUEUE_ITEMS,\n  );\n\n  // Holds the queue while client-side tools are still executing, and while an\n  // interrupt is unanswered: `append` refuses a new run until it is resolved,\n  // and the queue drops an item before dispatching it, so draining into that\n  // refusal would lose the message instead of keeping it visible.\n  const queueBusy = isRunning || core.getPendingInterrupts() !== null;\n  useEffect(() => {\n    if (options.isSendDisabled === true) {\n      queueController?.hold();\n    } else {\n      queueController?.release();\n    }\n  }, [options.isSendDisabled, queueController]);\n\n  useEffect(() => {\n    if (!queueController) return;\n    if (\n      lastQueueEdgeRef.current?.controller === queueController &&\n      lastQueueEdgeRef.current.busy === queueBusy\n    )\n      return;\n    lastQueueEdgeRef.current = { controller: queueController, busy: queueBusy };\n    if (queueBusy) {\n      busyEdgesRef.current++;\n      queueController.notifyBusy();\n    } else {\n      queueController.notifyIdle();\n    }\n  }, [queueBusy, queueController]);\n\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread, ...rest } =\n      threadListAdapter;\n\n    const abandonPreviousThread = () => {\n      queueRef.current?.notifyCancelled();\n      if (core.isRunning()) {\n        void core.cancel().catch((error: unknown) => {\n          logger.error?.(\n            \"[agui] cancelling the run on thread switch failed\",\n            error,\n          );\n        });\n      }\n      queueRef.current?.clear();\n    };\n\n    const releaseSwitch = (generation: number) => {\n      if (switchingGenerationRef.current === generation) {\n        switchingGenerationRef.current = null;\n      }\n    };\n\n    return {\n      ...rest,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n              await onSwitchToNewThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n            } finally {\n              releaseSwitch(generation);\n            }\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n              const result = await onSwitchToThread(threadId);\n              if (generation !== threadSwitchGenerationRef.current) return;\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n              core.applyExternalMessages(result.messages);\n              if (result.state !== undefined) {\n                core.loadExternalState(result.state);\n              }\n              if (generation !== threadSwitchGenerationRef.current) return;\n              releaseSwitch(generation);\n              if (result.unstable_resume) {\n                void core.resumeInFlightRun(result.messages);\n              }\n            } finally {\n              releaseSwitch(generation);\n            }\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core, logger]);\n\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(\n    () => {\n      return {\n        ...shared,\n        isLoading: snapshot.isLoading,\n        messageRepository: snapshot.messageRepository,\n        state: snapshot.state,\n        isRunning: snapshot.isRunning,\n        extras: agUiExtras.provide({\n          interrupts:\n            core.getPendingInterrupts()?.interrupts ?? EMPTY_INTERRUPTS,\n          sendA2uiAction: (action) => core.sendA2uiAction(action),\n          submitInterruptResponses: (responses) =>\n            core.submitInterruptResponses(responses),\n          steerAway: (message, responses) => core.steerAway(message, responses),\n          state: snapshot.state,\n          setState: (next) => core.setState(next),\n        }),\n        unstable_enableToolInvocations: true,\n        unstable_persistsHistory: true,\n        setToolStatuses,\n        onNew: (message: AppendMessage) =>\n          switchingGenerationRef.current !== null\n            ? Promise.resolve()\n            : core.append(message),\n        onVoiceTranscript: (message) => core.appendVoiceTranscript(message),\n        onEdit: (message: AppendMessage) => {\n          queueController?.clear();\n          return core.edit(message);\n        },\n        onReload: (parentId: string | null, config: { runConfig?: any }) => {\n          queueController?.clear();\n          return core.reload(parentId, config);\n        },\n        onCancel: async () => {\n          queueController?.notifyCancelled();\n          core.cancel();\n        },\n        onAddToolResult: (options) => core.addToolResult(options),\n        unstable_onRecordToolInteraction: (options) =>\n          core.recordToolInteraction(options),\n        onRespondToToolApproval: (options) =>\n          core.respondToToolApproval(options).catch((error: unknown) => {\n            core.reportError(error);\n            throw error;\n          }),\n        onResume: (config) => core.resume(config),\n        setMessages: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onImport: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onLoadExternalState: (state: ReadonlyJSONValue) =>\n          core.loadExternalState(state),\n        adapters: adapterAdapters,\n        ...(queueController && { queue: queueController.adapter }),\n      } satisfies ExternalStoreAdapter<ThreadMessage>;\n    },\n    // toolInvocations intentionally excluded: abort/resume use refs internally and work with stale captures\n    [\n      adapterAdapters,\n      core,\n      isRunning,\n      queueController,\n      queueItems,\n      steerQueueItems,\n      shared,\n      snapshot,\n    ],\n  );\n\n  const baseRuntime = useExternalStoreRuntime(store);\n\n  const createRuntime = (): AgUiAssistantRuntime => {\n    return {\n      threads: baseRuntime.threads,\n      get thread() {\n        return baseRuntime.thread;\n      },\n      registerModelContextProvider: (provider) =>\n        baseRuntime.registerModelContextProvider(provider),\n      unstable_getPendingInterrupts: () =>\n        core.getPendingInterrupts()?.interrupts ?? [],\n      unstable_submitInterruptResponses: (responses) =>\n        core.submitInterruptResponses(responses),\n    };\n  };\n  const [pinnedRuntime, setPinnedRuntime] = useState(() => ({\n    baseRuntime,\n    runtime: createRuntime(),\n  }));\n  let currentRuntime = pinnedRuntime;\n  if (pinnedRuntime.baseRuntime !== baseRuntime) {\n    currentRuntime = { baseRuntime, runtime: createRuntime() };\n    setPinnedRuntime(currentRuntime);\n  }\n  const runtime = currentRuntime.runtime;\n\n  useReplaySafeEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;;;;AAwCA,MAAM,mBAA6C,CAAC;AACpD,MAAM,oBAA+C,OAAO,OAAO,CAAC,CAAC;AACrE,MAAM,4BAA4B,CAAC;AAkBnC,SAAgB,eACd,SACsB;CACtB,MAAM,SAAS,cAAc,WAAW,QAAQ,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC;CACzE,MAAM,UAAU,OAAoC,IAAI;CACxD,MAAM,4BAA4B,OAAO,CAAC;CAC1C,MAAM,yBAAyB,OAAsB,IAAI;CACzD,MAAM,kBAAkB,mBAAmB;CAE3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,qBAAqB;EACzC,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,cAAc,QAAQ;EACtB,eAAe,QAAQ;EACvB,kBAAkB,QAAQ;EAC1B,4BAA4B,QAAQ;EACpC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAChD,yBAAyB,uBAAuB,YAAY;CAC9D,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,MAAM,WAAW,qBACf,KAAK,WACL,KAAK,aACL,KAAK,WACP;CACA,gBAAgB;EACd,KAAK,cAAc;GACjB,OAAO,QAAQ;GACf;GACA,cAAc,QAAQ,gBAAgB;GACtC,cAAc,QAAQ;GACtB,eAAe,QAAQ;GACvB,kBAAkB,QAAQ;GAC1B,4BAA4B,QAAQ;GACpC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;GAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;GACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAClD,CAAC;CACH,CAAC;CAED,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CAEJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YAAY,SAAS,aAAa;CAIxC,MAAM,WAAW,OAAsC,IAAI;CAI3D,MAAM,eAAe,OAAO,CAAC;CAC7B,MAAM,mBAAmB,OAGf,IAAI;CACd,IAAI,QAAQ,+BAA+B,CAAC,SAAS,SACnD,SAAS,UAAU,mBAAmB,EACpC,MAAM,YAAY;EAChB,MAAM,aAAa,SAAS;EAG5B,IAAI,uBAAuB,YAAY,MAAM;GAC3C,YAAY,WAAW;GACvB;EACF;EACA,MAAM,kBAAkB,aAAa;EAQrC,MAAM,uBAAuB;GAC3B,IACE,SAAS,YAAY,cACrB,aAAa,YAAY,mBACzB,KAAK,qBAAqB,MAAM,MAEhC;GACF,YAAY,WAAW;EACzB;EACA,KAAU,OAAO,OAAO,CAAC,CAAC,KAAK,iBAAiB,MAAe;GAC7D,eAAe;GACf,OAAO,QAAQ,wCAAwC,CAAC;EAC1D,CAAC;CACH,EACF,CAAC;CAEH,MAAM,kBAAkB,QAAQ,8BAC5B,SAAS,UACT;CACJ,gCAAgC;EAC9B,IAAI,QAAQ,+BAA+B,CAAC,SAAS,SAAS;EAC9D,MAAM,aAAa,SAAS;EAC5B,SAAS,UAAU;EACnB,WAAW,MAAM;CACnB,GAAG,CAAC,QAAQ,2BAA2B,CAAC;CAKxC,MAAM,aAAa,qBACjB,iBAAiB,aAAa,qBACxB,iBAAiB,QAAQ,SAAS,yBAClC,iBACR;CACA,MAAM,kBAAkB,qBACtB,iBAAiB,aAAa,qBACxB,iBAAiB,QAAQ,cAAc,yBACvC,iBACR;CAMA,MAAM,YAAY,aAAa,KAAK,qBAAqB,MAAM;CAC/D,gBAAgB;EACd,IAAI,QAAQ,mBAAmB,MAC7B,iBAAiB,KAAK;OAEtB,iBAAiB,QAAQ;CAE7B,GAAG,CAAC,QAAQ,gBAAgB,eAAe,CAAC;CAE5C,gBAAgB;EACd,IAAI,CAAC,iBAAiB;EACtB,IACE,iBAAiB,SAAS,eAAe,mBACzC,iBAAiB,QAAQ,SAAS,WAElC;EACF,iBAAiB,UAAU;GAAE,YAAY;GAAiB,MAAM;EAAU;EAC1E,IAAI,WAAW;GACb,aAAa;GACb,gBAAgB,WAAW;EAC7B,OACE,gBAAgB,WAAW;CAE/B,GAAG,CAAC,WAAW,eAAe,CAAC;CAE/B,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,kBAAkB,GAAG,SAChD;EAEF,MAAM,8BAA8B;GAClC,SAAS,SAAS,gBAAgB;GAClC,IAAI,KAAK,UAAU,GACjB,KAAU,OAAO,CAAC,CAAC,OAAO,UAAmB;IAC3C,OAAO,QACL,qDACA,KACF;GACF,CAAC;GAEH,SAAS,SAAS,MAAM;EAC1B;EAEA,MAAM,iBAAiB,eAAuB;GAC5C,IAAI,uBAAuB,YAAY,YACrC,uBAAuB,UAAU;EAErC;EAEA,OAAO;GACL,GAAG;GACH,qBAAqB,sBACjB,YAAY;IACV,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KACF,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KAGtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;KACtB,MAAM,oBAAoB;KAC1B,IAAI,eAAe,0BAA0B,SAAS;KACtD,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KACtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;IACxB,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KACF,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KAGtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;KACtB,MAAM,SAAS,MAAM,iBAAiB,QAAQ;KAC9C,IAAI,eAAe,0BAA0B,SAAS;KACtD,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KACtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;KACtB,KAAK,sBAAsB,OAAO,QAAQ;KAC1C,IAAI,OAAO,UAAU,KAAA,GACnB,KAAK,kBAAkB,OAAO,KAAK;KAErC,IAAI,eAAe,0BAA0B,SAAS;KACtD,cAAc,UAAU;KACxB,IAAI,OAAO,iBACT,KAAU,kBAAkB,OAAO,QAAQ;IAE/C,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,IACA,KAAA;EACN;CACF,GAAG;EAAC;EAAmB;EAAM;CAAM,CAAC;CAEpC,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAEA,MAAM,SAAS,8BAA8B,OAAO;CACpD,MAAM,QAAQ,cACN;EACJ,OAAO;GACL,GAAG;GACH,WAAW,SAAS;GACpB,mBAAmB,SAAS;GAC5B,OAAO,SAAS;GAChB,WAAW,SAAS;GACpB,QAAQ,WAAW,QAAQ;IACzB,YACE,KAAK,qBAAqB,CAAC,EAAE,cAAc;IAC7C,iBAAiB,WAAW,KAAK,eAAe,MAAM;IACtD,2BAA2B,cACzB,KAAK,yBAAyB,SAAS;IACzC,YAAY,SAAS,cAAc,KAAK,UAAU,SAAS,SAAS;IACpE,OAAO,SAAS;IAChB,WAAW,SAAS,KAAK,SAAS,IAAI;GACxC,CAAC;GACD,gCAAgC;GAChC,0BAA0B;GAC1B;GACA,QAAQ,YACN,uBAAuB,YAAY,OAC/B,QAAQ,QAAQ,IAChB,KAAK,OAAO,OAAO;GACzB,oBAAoB,YAAY,KAAK,sBAAsB,OAAO;GAClE,SAAS,YAA2B;IAClC,iBAAiB,MAAM;IACvB,OAAO,KAAK,KAAK,OAAO;GAC1B;GACA,WAAW,UAAyB,WAAgC;IAClE,iBAAiB,MAAM;IACvB,OAAO,KAAK,OAAO,UAAU,MAAM;GACrC;GACA,UAAU,YAAY;IACpB,iBAAiB,gBAAgB;IACjC,KAAK,OAAO;GACd;GACA,kBAAkB,YAAY,KAAK,cAAc,OAAO;GACxD,mCAAmC,YACjC,KAAK,sBAAsB,OAAO;GACpC,0BAA0B,YACxB,KAAK,sBAAsB,OAAO,CAAC,CAAC,OAAO,UAAmB;IAC5D,KAAK,YAAY,KAAK;IACtB,MAAM;GACR,CAAC;GACH,WAAW,WAAW,KAAK,OAAO,MAAM;GACxC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,sBAAsB,UACpB,KAAK,kBAAkB,KAAK;GAC9B,UAAU;GACV,GAAI,mBAAmB,EAAE,OAAO,gBAAgB,QAAQ;EAC1D;CACF,GAEA;EACE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;CACF,CACF;CAEA,MAAM,cAAc,wBAAwB,KAAK;CAEjD,MAAM,sBAA4C;EAChD,OAAO;GACL,SAAS,YAAY;GACrB,IAAI,SAAS;IACX,OAAO,YAAY;GACrB;GACA,+BAA+B,aAC7B,YAAY,6BAA6B,QAAQ;GACnD,qCACE,KAAK,qBAAqB,CAAC,EAAE,cAAc,CAAC;GAC9C,oCAAoC,cAClC,KAAK,yBAAyB,SAAS;EAC3C;CACF;CACA,MAAM,CAAC,eAAe,oBAAoB,gBAAgB;EACxD;EACA,SAAS,cAAc;CACzB,EAAE;CACF,IAAI,iBAAiB;CACrB,IAAI,cAAc,gBAAgB,aAAa;EAC7C,iBAAiB;GAAE;GAAa,SAAS,cAAc;EAAE;EACzD,iBAAiB,cAAc;CACjC;CACA,MAAM,UAAU,eAAe;CAE/B,0BAA0B;EACxB,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}
+{"version":3,"file":"useAgUiRuntime.js","names":[],"sources":["../src/useAgUiRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n  useSyncExternalStore,\n} from \"react\";\nimport {\n  useExternalStoreRuntime,\n  useExternalStoreSharedOptions,\n  useRuntimeAdapters,\n} from \"@assistant-ui/core/react\";\nimport {\n  useIsomorphicLayoutEffect,\n  useReplaySafeEffect,\n} from \"@assistant-ui/store/internal\";\nimport { createMessageQueue } from \"@assistant-ui/core\";\nimport type {\n  MessageQueueController,\n  ToolExecutionStatus,\n} from \"@assistant-ui/core\";\nimport type {\n  AssistantRuntime,\n  AppendMessage,\n  ExternalStoreAdapter,\n  ThreadMessage,\n} from \"@assistant-ui/core\";\nimport type { ReadonlyJSONValue } from \"assistant-stream/utils\";\nimport { makeLogger } from \"./runtime/logger\";\nimport type {\n  AgUiInterrupt,\n  AgUiResumeEntry,\n  UseAgUiRuntimeOptions,\n} from \"./runtime/types\";\nimport { AgUiThreadController } from \"./runtime/AgUiThreadController\";\nimport { agUiExtras } from \"./agUiExtras\";\nimport type { QueueItemState } from \"@assistant-ui/core/store\";\n\nconst EMPTY_INTERRUPTS: readonly AgUiInterrupt[] = [];\nconst EMPTY_QUEUE_ITEMS: readonly QueueItemState[] = Object.freeze([]);\nconst subscribeNoop = () => () => {};\n\nexport type AgUiAssistantRuntime = AssistantRuntime & {\n  /**\n   * @deprecated Use the `useAgUiInterrupts()` hook instead. This method is kept\n   * for backward compatibility and will be removed in a future major release.\n   */\n  unstable_getPendingInterrupts: () => readonly AgUiInterrupt[];\n  /**\n   * @deprecated Use the `useAgUiSubmitInterruptResponses()` hook instead. This\n   * method is kept for backward compatibility and will be removed in a future\n   * major release.\n   */\n  unstable_submitInterruptResponses: (\n    responses: readonly AgUiResumeEntry[],\n  ) => Promise<void>;\n};\n\nexport function useAgUiRuntime(\n  options: UseAgUiRuntimeOptions,\n): AgUiAssistantRuntime {\n  const logger = useMemo(() => makeLogger(options.logger), [options.logger]);\n  const coreRef = useRef<AgUiThreadController | null>(null);\n  const threadSwitchGenerationRef = useRef(0);\n  const switchingGenerationRef = useRef<number | null>(null);\n  const runtimeAdapters = useRuntimeAdapters();\n\n  const historyAdapter = options.adapters?.history ?? runtimeAdapters?.history;\n  const threadListAdapter = options.adapters?.threadList;\n\n  if (!coreRef.current) {\n    coreRef.current = new AgUiThreadController({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      joinStrategy: options.joinStrategy,\n      getRunContext: options.getRunContext,\n      resumeTranscript: options.resumeTranscript,\n      autoCancelPendingToolCalls: options.autoCancelPendingToolCalls,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n      isThreadSwitching: () => switchingGenerationRef.current !== null,\n    });\n  }\n\n  const core = coreRef.current;\n  const snapshot = useSyncExternalStore(\n    core.subscribe,\n    core.getSnapshot,\n    core.getSnapshot,\n  );\n  useEffect(() => {\n    core.updateOptions({\n      agent: options.agent,\n      logger,\n      showThinking: options.showThinking ?? true,\n      joinStrategy: options.joinStrategy,\n      getRunContext: options.getRunContext,\n      resumeTranscript: options.resumeTranscript,\n      autoCancelPendingToolCalls: options.autoCancelPendingToolCalls,\n      ...(options.onError && { onError: options.onError }),\n      ...(options.onCancel && { onCancel: options.onCancel }),\n      ...(historyAdapter && { history: historyAdapter }),\n    });\n  });\n\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  const hasExecutingTools = Object.values(toolStatuses).some(\n    (s) => s?.type === \"executing\",\n  );\n  const isRunning = snapshot.isRunning || hasExecutingTools;\n\n  // The driver sends through the agent core rather than the runtime, because\n  // the runtime's append routes every tail append back into this queue.\n  const queueRef = useRef<MessageQueueController | null>(null);\n  // Counts rendered busy edges, whether a run start or an interrupt arriving.\n  // A dispatch whose count never moves was never observed as busy, so no\n  // falling edge is coming to release the queue.\n  const busyEdgesRef = useRef(0);\n  const lastQueueEdgeRef = useRef<{\n    controller: MessageQueueController;\n    busy: boolean;\n  } | null>(null);\n  if (options.unstable_enableMessageQueue && !queueRef.current) {\n    queueRef.current = createMessageQueue({\n      run: (message) => {\n        const controller = queueRef.current;\n        // A send dispatched mid-switch is dropped, and the queue released, or\n        // it would wait for an idle edge that never comes.\n        if (switchingGenerationRef.current !== null) {\n          controller?.notifyIdle();\n          return;\n        }\n        const edgesAtDispatch = busyEdgesRef.current;\n        // The queue drops the item before dispatching and stays busy until an\n        // idle edge. An append observed as busy is released by that falling\n        // edge, and releasing again here would advance the queue twice; one\n        // that never runs (a pre-run rejection, or startRun false) has to be\n        // released here or every later send buffers forever. An interrupt is\n        // rechecked because a run that resolves without yielding to React\n        // reaches here before the effect below has seen either edge.\n        const releaseIfNoRun = () => {\n          if (\n            queueRef.current !== controller ||\n            busyEdgesRef.current !== edgesAtDispatch ||\n            core.getPendingInterrupts() !== null\n          )\n            return;\n          controller?.notifyIdle();\n        };\n        void core.append(message).then(releaseIfNoRun, (e: unknown) => {\n          releaseIfNoRun();\n          logger.error?.(\"[agui] queued message failed to send\", e);\n        });\n      },\n    });\n  }\n  const queueController = options.unstable_enableMessageQueue\n    ? queueRef.current\n    : null;\n  useIsomorphicLayoutEffect(() => {\n    if (options.unstable_enableMessageQueue || !queueRef.current) return;\n    const controller = queueRef.current;\n    queueRef.current = null;\n    controller.clear();\n  }, [options.unstable_enableMessageQueue]);\n\n  // Feeds the store memo below: the runtime core skips an adapter whose\n  // identity is unchanged, so queue items have to move the store reference or\n  // subscribers never see composer.queue change.\n  const queueItems = useSyncExternalStore(\n    queueController?.subscribe ?? subscribeNoop,\n    () => queueController?.adapter.items ?? EMPTY_QUEUE_ITEMS,\n    () => EMPTY_QUEUE_ITEMS,\n  );\n  const steerQueueItems = useSyncExternalStore(\n    queueController?.subscribe ?? subscribeNoop,\n    () => queueController?.adapter.steerItems ?? EMPTY_QUEUE_ITEMS,\n    () => EMPTY_QUEUE_ITEMS,\n  );\n\n  // Holds the queue while client-side tools are still executing, and while an\n  // interrupt is unanswered: `append` refuses a new run until it is resolved,\n  // and the queue drops an item before dispatching it, so draining into that\n  // refusal would lose the message instead of keeping it visible.\n  const queueBusy = isRunning || core.getPendingInterrupts() !== null;\n  useEffect(() => {\n    if (options.isSendDisabled === true) {\n      queueController?.hold();\n    } else {\n      queueController?.release();\n    }\n  }, [options.isSendDisabled, queueController]);\n\n  useEffect(() => {\n    if (!queueController) return;\n    if (\n      lastQueueEdgeRef.current?.controller === queueController &&\n      lastQueueEdgeRef.current.busy === queueBusy\n    )\n      return;\n    lastQueueEdgeRef.current = { controller: queueController, busy: queueBusy };\n    if (queueBusy) {\n      busyEdgesRef.current++;\n      queueController.notifyBusy();\n    } else {\n      queueController.notifyIdle();\n    }\n  }, [queueBusy, queueController]);\n\n  const threadList = useMemo(() => {\n    if (!threadListAdapter) return undefined;\n\n    const { onSwitchToNewThread, onSwitchToThread, ...rest } =\n      threadListAdapter;\n\n    const abandonPreviousThread = () => {\n      queueRef.current?.notifyCancelled();\n      if (core.isRunning()) {\n        void core.cancel().catch((error: unknown) => {\n          logger.error?.(\n            \"[agui] cancelling the run on thread switch failed\",\n            error,\n          );\n        });\n      }\n      queueRef.current?.clear();\n    };\n\n    const releaseSwitch = (generation: number) => {\n      if (switchingGenerationRef.current === generation) {\n        switchingGenerationRef.current = null;\n      }\n    };\n\n    return {\n      ...rest,\n      onSwitchToNewThread: onSwitchToNewThread\n        ? async () => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n              await onSwitchToNewThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n            } finally {\n              releaseSwitch(generation);\n            }\n          }\n        : undefined,\n      onSwitchToThread: onSwitchToThread\n        ? async (threadId: string) => {\n            const generation = ++threadSwitchGenerationRef.current;\n            switchingGenerationRef.current = generation;\n            try {\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              // Clear before the thread id flips, or the old messages leak\n              // into the new thread as a sibling branch.\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n              const result = await onSwitchToThread(threadId);\n              if (generation !== threadSwitchGenerationRef.current) return;\n              abandonPreviousThread();\n              if (generation !== threadSwitchGenerationRef.current) return;\n              core.applyExternalMessages([]);\n              core.resetThreadState();\n              core.applyExternalMessages(result.messages);\n              if (result.state !== undefined) {\n                core.loadExternalState(result.state);\n              }\n              if (generation !== threadSwitchGenerationRef.current) return;\n              releaseSwitch(generation);\n              if (result.unstable_resume) {\n                void core.resumeInFlightRun(result.messages);\n              }\n            } finally {\n              releaseSwitch(generation);\n            }\n          }\n        : undefined,\n    };\n  }, [threadListAdapter, core, logger]);\n\n  const adapters = options.adapters;\n  const adapterAdapters = useMemo(\n    () => ({\n      attachments: adapters?.attachments ?? runtimeAdapters?.attachments,\n      speech: adapters?.speech,\n      dictation: adapters?.dictation,\n      voice: adapters?.voice,\n      feedback: adapters?.feedback,\n      threadList,\n    }),\n    [adapters, runtimeAdapters, threadList],\n  );\n\n  const shared = useExternalStoreSharedOptions(options);\n  const store = useMemo(\n    () => {\n      return {\n        ...shared,\n        isLoading: snapshot.isLoading,\n        messageRepository: snapshot.messageRepository,\n        state: snapshot.state,\n        isRunning: snapshot.isRunning,\n        extras: agUiExtras.provide({\n          interrupts:\n            core.getPendingInterrupts()?.interrupts ?? EMPTY_INTERRUPTS,\n          sendA2uiAction: (action) => core.sendA2uiAction(action),\n          submitInterruptResponses: (responses) =>\n            core.submitInterruptResponses(responses),\n          steerAway: (message, responses) => core.steerAway(message, responses),\n          state: snapshot.state,\n          setState: (next) => core.setState(next),\n        }),\n        unstable_enableToolInvocations: true,\n        unstable_persistsHistory: true,\n        setToolStatuses,\n        onNew: (message: AppendMessage) =>\n          switchingGenerationRef.current !== null\n            ? Promise.resolve()\n            : core.append(message),\n        onVoiceTranscript: (message) => core.appendVoiceTranscript(message),\n        onEdit: (message: AppendMessage) => {\n          queueController?.clear();\n          return core.edit(message);\n        },\n        onReload: (parentId: string | null, config: { runConfig?: any }) => {\n          queueController?.clear();\n          return core.reload(parentId, config);\n        },\n        onDelete: (messageId: string) => core.deleteMessage(messageId),\n        onCancel: async () => {\n          queueController?.notifyCancelled();\n          core.cancel();\n        },\n        onAddToolResult: (options) => core.addToolResult(options),\n        unstable_onRecordToolInteraction: (options) =>\n          core.recordToolInteraction(options),\n        onRespondToToolApproval: (options) =>\n          core.respondToToolApproval(options).catch((error: unknown) => {\n            core.reportError(error);\n            throw error;\n          }),\n        onResume: (config) => core.resume(config),\n        setMessages: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onImport: (messages: readonly ThreadMessage[]) =>\n          core.applyExternalMessages(messages),\n        onLoadExternalState: (state: ReadonlyJSONValue) =>\n          core.loadExternalState(state),\n        adapters: adapterAdapters,\n        ...(queueController && { queue: queueController.adapter }),\n      } satisfies ExternalStoreAdapter<ThreadMessage>;\n    },\n    // toolInvocations intentionally excluded: abort/resume use refs internally and work with stale captures\n    [\n      adapterAdapters,\n      core,\n      isRunning,\n      queueController,\n      queueItems,\n      steerQueueItems,\n      shared,\n      snapshot,\n    ],\n  );\n\n  const baseRuntime = useExternalStoreRuntime(store);\n\n  const createRuntime = (): AgUiAssistantRuntime => {\n    return {\n      threads: baseRuntime.threads,\n      get thread() {\n        return baseRuntime.thread;\n      },\n      registerModelContextProvider: (provider) =>\n        baseRuntime.registerModelContextProvider(provider),\n      unstable_getPendingInterrupts: () =>\n        core.getPendingInterrupts()?.interrupts ?? [],\n      unstable_submitInterruptResponses: (responses) =>\n        core.submitInterruptResponses(responses),\n    };\n  };\n  const [pinnedRuntime, setPinnedRuntime] = useState(() => ({\n    baseRuntime,\n    runtime: createRuntime(),\n  }));\n  let currentRuntime = pinnedRuntime;\n  if (pinnedRuntime.baseRuntime !== baseRuntime) {\n    currentRuntime = { baseRuntime, runtime: createRuntime() };\n    setPinnedRuntime(currentRuntime);\n  }\n  const runtime = currentRuntime.runtime;\n\n  useReplaySafeEffect(() => {\n    core.attachRuntime(runtime);\n    return () => {\n      core.detachRuntime();\n    };\n  }, [core, runtime]);\n\n  useEffect(() => {\n    core.__internal_load();\n  }, [core]);\n\n  return runtime;\n}\n"],"mappings":";;;;;;;;;AAwCA,MAAM,mBAA6C,CAAC;AACpD,MAAM,oBAA+C,OAAO,OAAO,CAAC,CAAC;AACrE,MAAM,4BAA4B,CAAC;AAkBnC,SAAgB,eACd,SACsB;CACtB,MAAM,SAAS,cAAc,WAAW,QAAQ,MAAM,GAAG,CAAC,QAAQ,MAAM,CAAC;CACzE,MAAM,UAAU,OAAoC,IAAI;CACxD,MAAM,4BAA4B,OAAO,CAAC;CAC1C,MAAM,yBAAyB,OAAsB,IAAI;CACzD,MAAM,kBAAkB,mBAAmB;CAE3C,MAAM,iBAAiB,QAAQ,UAAU,WAAW,iBAAiB;CACrE,MAAM,oBAAoB,QAAQ,UAAU;CAE5C,IAAI,CAAC,QAAQ,SACX,QAAQ,UAAU,IAAI,qBAAqB;EACzC,OAAO,QAAQ;EACf;EACA,cAAc,QAAQ,gBAAgB;EACtC,cAAc,QAAQ;EACtB,eAAe,QAAQ;EACvB,kBAAkB,QAAQ;EAC1B,4BAA4B,QAAQ;EACpC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;EAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;EACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAChD,yBAAyB,uBAAuB,YAAY;CAC9D,CAAC;CAGH,MAAM,OAAO,QAAQ;CACrB,MAAM,WAAW,qBACf,KAAK,WACL,KAAK,aACL,KAAK,WACP;CACA,gBAAgB;EACd,KAAK,cAAc;GACjB,OAAO,QAAQ;GACf;GACA,cAAc,QAAQ,gBAAgB;GACtC,cAAc,QAAQ;GACtB,eAAe,QAAQ;GACvB,kBAAkB,QAAQ;GAC1B,4BAA4B,QAAQ;GACpC,GAAI,QAAQ,WAAW,EAAE,SAAS,QAAQ,QAAQ;GAClD,GAAI,QAAQ,YAAY,EAAE,UAAU,QAAQ,SAAS;GACrD,GAAI,kBAAkB,EAAE,SAAS,eAAe;EAClD,CAAC;CACH,CAAC;CAED,MAAM,CAAC,cAAc,mBAAmB,SAEtC,CAAC,CAAC;CAEJ,MAAM,oBAAoB,OAAO,OAAO,YAAY,CAAC,CAAC,MACnD,MAAM,GAAG,SAAS,WACrB;CACA,MAAM,YAAY,SAAS,aAAa;CAIxC,MAAM,WAAW,OAAsC,IAAI;CAI3D,MAAM,eAAe,OAAO,CAAC;CAC7B,MAAM,mBAAmB,OAGf,IAAI;CACd,IAAI,QAAQ,+BAA+B,CAAC,SAAS,SACnD,SAAS,UAAU,mBAAmB,EACpC,MAAM,YAAY;EAChB,MAAM,aAAa,SAAS;EAG5B,IAAI,uBAAuB,YAAY,MAAM;GAC3C,YAAY,WAAW;GACvB;EACF;EACA,MAAM,kBAAkB,aAAa;EAQrC,MAAM,uBAAuB;GAC3B,IACE,SAAS,YAAY,cACrB,aAAa,YAAY,mBACzB,KAAK,qBAAqB,MAAM,MAEhC;GACF,YAAY,WAAW;EACzB;EACA,KAAU,OAAO,OAAO,CAAC,CAAC,KAAK,iBAAiB,MAAe;GAC7D,eAAe;GACf,OAAO,QAAQ,wCAAwC,CAAC;EAC1D,CAAC;CACH,EACF,CAAC;CAEH,MAAM,kBAAkB,QAAQ,8BAC5B,SAAS,UACT;CACJ,gCAAgC;EAC9B,IAAI,QAAQ,+BAA+B,CAAC,SAAS,SAAS;EAC9D,MAAM,aAAa,SAAS;EAC5B,SAAS,UAAU;EACnB,WAAW,MAAM;CACnB,GAAG,CAAC,QAAQ,2BAA2B,CAAC;CAKxC,MAAM,aAAa,qBACjB,iBAAiB,aAAa,qBACxB,iBAAiB,QAAQ,SAAS,yBAClC,iBACR;CACA,MAAM,kBAAkB,qBACtB,iBAAiB,aAAa,qBACxB,iBAAiB,QAAQ,cAAc,yBACvC,iBACR;CAMA,MAAM,YAAY,aAAa,KAAK,qBAAqB,MAAM;CAC/D,gBAAgB;EACd,IAAI,QAAQ,mBAAmB,MAC7B,iBAAiB,KAAK;OAEtB,iBAAiB,QAAQ;CAE7B,GAAG,CAAC,QAAQ,gBAAgB,eAAe,CAAC;CAE5C,gBAAgB;EACd,IAAI,CAAC,iBAAiB;EACtB,IACE,iBAAiB,SAAS,eAAe,mBACzC,iBAAiB,QAAQ,SAAS,WAElC;EACF,iBAAiB,UAAU;GAAE,YAAY;GAAiB,MAAM;EAAU;EAC1E,IAAI,WAAW;GACb,aAAa;GACb,gBAAgB,WAAW;EAC7B,OACE,gBAAgB,WAAW;CAE/B,GAAG,CAAC,WAAW,eAAe,CAAC;CAE/B,MAAM,aAAa,cAAc;EAC/B,IAAI,CAAC,mBAAmB,OAAO,KAAA;EAE/B,MAAM,EAAE,qBAAqB,kBAAkB,GAAG,SAChD;EAEF,MAAM,8BAA8B;GAClC,SAAS,SAAS,gBAAgB;GAClC,IAAI,KAAK,UAAU,GACjB,KAAU,OAAO,CAAC,CAAC,OAAO,UAAmB;IAC3C,OAAO,QACL,qDACA,KACF;GACF,CAAC;GAEH,SAAS,SAAS,MAAM;EAC1B;EAEA,MAAM,iBAAiB,eAAuB;GAC5C,IAAI,uBAAuB,YAAY,YACrC,uBAAuB,UAAU;EAErC;EAEA,OAAO;GACL,GAAG;GACH,qBAAqB,sBACjB,YAAY;IACV,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KACF,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KAGtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;KACtB,MAAM,oBAAoB;KAC1B,IAAI,eAAe,0BAA0B,SAAS;KACtD,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KACtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;IACxB,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,IACA,KAAA;GACJ,kBAAkB,mBACd,OAAO,aAAqB;IAC1B,MAAM,aAAa,EAAE,0BAA0B;IAC/C,uBAAuB,UAAU;IACjC,IAAI;KACF,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KAGtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;KACtB,MAAM,SAAS,MAAM,iBAAiB,QAAQ;KAC9C,IAAI,eAAe,0BAA0B,SAAS;KACtD,sBAAsB;KACtB,IAAI,eAAe,0BAA0B,SAAS;KACtD,KAAK,sBAAsB,CAAC,CAAC;KAC7B,KAAK,iBAAiB;KACtB,KAAK,sBAAsB,OAAO,QAAQ;KAC1C,IAAI,OAAO,UAAU,KAAA,GACnB,KAAK,kBAAkB,OAAO,KAAK;KAErC,IAAI,eAAe,0BAA0B,SAAS;KACtD,cAAc,UAAU;KACxB,IAAI,OAAO,iBACT,KAAU,kBAAkB,OAAO,QAAQ;IAE/C,UAAU;KACR,cAAc,UAAU;IAC1B;GACF,IACA,KAAA;EACN;CACF,GAAG;EAAC;EAAmB;EAAM;CAAM,CAAC;CAEpC,MAAM,WAAW,QAAQ;CACzB,MAAM,kBAAkB,eACf;EACL,aAAa,UAAU,eAAe,iBAAiB;EACvD,QAAQ,UAAU;EAClB,WAAW,UAAU;EACrB,OAAO,UAAU;EACjB,UAAU,UAAU;EACpB;CACF,IACA;EAAC;EAAU;EAAiB;CAAU,CACxC;CAEA,MAAM,SAAS,8BAA8B,OAAO;CACpD,MAAM,QAAQ,cACN;EACJ,OAAO;GACL,GAAG;GACH,WAAW,SAAS;GACpB,mBAAmB,SAAS;GAC5B,OAAO,SAAS;GAChB,WAAW,SAAS;GACpB,QAAQ,WAAW,QAAQ;IACzB,YACE,KAAK,qBAAqB,CAAC,EAAE,cAAc;IAC7C,iBAAiB,WAAW,KAAK,eAAe,MAAM;IACtD,2BAA2B,cACzB,KAAK,yBAAyB,SAAS;IACzC,YAAY,SAAS,cAAc,KAAK,UAAU,SAAS,SAAS;IACpE,OAAO,SAAS;IAChB,WAAW,SAAS,KAAK,SAAS,IAAI;GACxC,CAAC;GACD,gCAAgC;GAChC,0BAA0B;GAC1B;GACA,QAAQ,YACN,uBAAuB,YAAY,OAC/B,QAAQ,QAAQ,IAChB,KAAK,OAAO,OAAO;GACzB,oBAAoB,YAAY,KAAK,sBAAsB,OAAO;GAClE,SAAS,YAA2B;IAClC,iBAAiB,MAAM;IACvB,OAAO,KAAK,KAAK,OAAO;GAC1B;GACA,WAAW,UAAyB,WAAgC;IAClE,iBAAiB,MAAM;IACvB,OAAO,KAAK,OAAO,UAAU,MAAM;GACrC;GACA,WAAW,cAAsB,KAAK,cAAc,SAAS;GAC7D,UAAU,YAAY;IACpB,iBAAiB,gBAAgB;IACjC,KAAK,OAAO;GACd;GACA,kBAAkB,YAAY,KAAK,cAAc,OAAO;GACxD,mCAAmC,YACjC,KAAK,sBAAsB,OAAO;GACpC,0BAA0B,YACxB,KAAK,sBAAsB,OAAO,CAAC,CAAC,OAAO,UAAmB;IAC5D,KAAK,YAAY,KAAK;IACtB,MAAM;GACR,CAAC;GACH,WAAW,WAAW,KAAK,OAAO,MAAM;GACxC,cAAc,aACZ,KAAK,sBAAsB,QAAQ;GACrC,WAAW,aACT,KAAK,sBAAsB,QAAQ;GACrC,sBAAsB,UACpB,KAAK,kBAAkB,KAAK;GAC9B,UAAU;GACV,GAAI,mBAAmB,EAAE,OAAO,gBAAgB,QAAQ;EAC1D;CACF,GAEA;EACE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;CACF,CACF;CAEA,MAAM,cAAc,wBAAwB,KAAK;CAEjD,MAAM,sBAA4C;EAChD,OAAO;GACL,SAAS,YAAY;GACrB,IAAI,SAAS;IACX,OAAO,YAAY;GACrB;GACA,+BAA+B,aAC7B,YAAY,6BAA6B,QAAQ;GACnD,qCACE,KAAK,qBAAqB,CAAC,EAAE,cAAc,CAAC;GAC9C,oCAAoC,cAClC,KAAK,yBAAyB,SAAS;EAC3C;CACF;CACA,MAAM,CAAC,eAAe,oBAAoB,gBAAgB;EACxD;EACA,SAAS,cAAc;CACzB,EAAE;CACF,IAAI,iBAAiB;CACrB,IAAI,cAAc,gBAAgB,aAAa;EAC7C,iBAAiB;GAAE;GAAa,SAAS,cAAc;EAAE;EACzD,iBAAiB,cAAc;CACjC;CACA,MAAM,UAAU,eAAe;CAE/B,0BAA0B;EACxB,KAAK,cAAc,OAAO;EAC1B,aAAa;GACX,KAAK,cAAc;EACrB;CACF,GAAG,CAAC,MAAM,OAAO,CAAC;CAElB,gBAAgB;EACd,KAAK,gBAAgB;CACvB,GAAG,CAAC,IAAI,CAAC;CAET,OAAO;AACT"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@assistant-ui/react-ag-ui",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "AG-UI protocol adapter for assistant-ui",
   "keywords": [
     "ag-ui",
@@ -30,7 +30,7 @@
   "sideEffects": false,
   "dependencies": {
     "@ag-ui/client": "^1.0.3",
-    "@assistant-ui/core": "^0.3.34",
+    "@assistant-ui/core": "^0.3.35",
     "@assistant-ui/generative-ui": "^0.0.10",
     "@assistant-ui/store": "^0.3.23",
     "assistant-stream": "^0.3.57",

package/src/runtime/AgUiThreadController.snapshot-head-history.test.ts

@@ -0,0 +1,226 @@
+import { describe, expect, it } from "vitest";
+import type { AppendMessage } from "@assistant-ui/core";
+import { HttpAgent } from "@ag-ui/client";
+import { AgUiThreadController } from "./AgUiThreadController";
+import { makeLogger } from "./logger";
+
+const tick = () => new Promise((r) => setTimeout(r, 0));
+const settle = async () => {
+  for (let i = 0; i < 8; i++) await tick();
+};
+const userMessage = {
+  role: "user",
+  content: [{ type: "text", text: "hello" }],
+  attachments: [],
+  metadata: { custom: {} },
+  createdAt: new Date(0),
+  parentId: null,
+  sourceId: null,
+  runConfig: {},
+  startRun: true,
+} as unknown as AppendMessage;
+
+type Row = { id: string; status: string | undefined };
+
+// Follows the ThreadHistoryAdapter contract: both writes upsert by message
+// id, and only append moves the stored head.
+function createHistory(writes: string[]) {
+  const items = new Map<string, any>();
+  let headId: string | null = null;
+  const store = (item: any, moveHead: boolean) => {
+    if (moveHead) headId = item.message.id;
+    items.set(item.message.id, structuredClone(item));
+  };
+  return {
+    rows: () =>
+      new Map<string, Row>(
+        [...items.values()].map((item) => [
+          item.message.id,
+          { id: item.message.id, status: item.message.status?.type },
+        ]),
+      ),
+    load: async () => ({ headId, messages: [...items.values()] }),
+    append: async (item: any) => {
+      writes.push(`append ${item.message.id}`);
+      store(item, true);
+    },
+    update: async (item: any) => {
+      writes.push(`update ${item.message.id}`);
+      store(item, false);
+    },
+  };
+}
+
+async function reload(history: ReturnType<typeof createHistory>) {
+  const core = new AgUiThreadController({
+    agent: new HttpAgent({ url: "https://agent.test" }),
+    logger: makeLogger(),
+    showThinking: false,
+    history: history as any,
+  });
+  await core.__internal_load();
+  return core.getMessages().map((m: any) => `${m.id}:${m.status?.type ?? "-"}`);
+}
+
+async function run(events: (requestMessages: any[]) => object[][]) {
+  const writes: string[] = [];
+  const history = createHistory(writes);
+  let write!: (e: object) => void;
+  let close!: () => void;
+  let requestMessages: any[] = [];
+  const agent = new HttpAgent({
+    url: "https://agent.test",
+    fetch: async (_url: unknown, init: RequestInit) => {
+      const encoder = new TextEncoder();
+      let controller!: ReadableStreamDefaultController<Uint8Array>;
+      const body = new ReadableStream<Uint8Array>({
+        start: (c) => void (controller = c),
+      });
+      requestMessages = JSON.parse(init.body as string).messages ?? [];
+      write = (e) =>
+        controller.enqueue(encoder.encode(`data: ${JSON.stringify(e)}\n\n`));
+      close = () => controller.close();
+      return new Response(body, {
+        headers: { "Content-Type": "text/event-stream" },
+      });
+    },
+  });
+  const core = new AgUiThreadController({
+    agent,
+    logger: makeLogger(),
+    showThinking: false,
+    history: history as any,
+  });
+  const sending = core.append(userMessage).catch(() => {});
+  for (let i = 0; i < 50 && !write; i++) await tick();
+  for (const batch of events(requestMessages)) {
+    for (const e of batch) write(e);
+    await settle();
+  }
+  close();
+  await sending;
+  await settle();
+  const shown = core
+    .getMessages()
+    .map((m: any) => `${m.id}:${m.status?.type ?? "-"}`);
+  return {
+    rows: history.rows(),
+    writes,
+    shown,
+    reloaded: await reload(history),
+  };
+}
+
+const start = [
+  { type: "RUN_STARTED", threadId: "t", runId: "r" },
+  { type: "TEXT_MESSAGE_START", messageId: "answer", role: "assistant" },
+  { type: "TEXT_MESSAGE_CONTENT", messageId: "answer", delta: "a1" },
+];
+const end = { type: "TEXT_MESSAGE_END", messageId: "answer" };
+const moveSnapshot = (req: any[]) => ({
+  type: "MESSAGES_SNAPSHOT",
+  messages: [
+    ...req,
+    { id: "answer", role: "assistant", content: "a1" },
+    { id: "other", role: "assistant", content: "o" },
+  ],
+});
+
+describe("ag-ui: a snapshot's new head and the history adapter", () => {
+  it("stores the new head after one head-move snapshot and RUN_FINISHED", async () => {
+    const { rows, writes, shown, reloaded } = await run((req) => [
+      start,
+      [moveSnapshot(req)],
+      [end, { type: "RUN_FINISHED", threadId: "t", runId: "r" }],
+    ]);
+    expect(shown).toContain("other:complete");
+    expect(rows.get("other")?.status, writes.join(", ")).toBe("complete");
+    expect(reloaded.slice(1), writes.join(", ")).toEqual(shown.slice(1));
+  });
+
+  it("stores the new head after one head-move snapshot and RUN_ERROR", async () => {
+    const { rows, writes, shown, reloaded } = await run((req) => [
+      start,
+      [moveSnapshot(req)],
+      [{ type: "RUN_ERROR", message: "boom" }],
+    ]);
+    // Which message takes the error depends on whether the run moved onto
+    // "other"; either way history holds what the thread shows.
+    const other = shown.find((entry) => entry.startsWith("other:"));
+    expect(["other:incomplete", "other:complete"]).toContain(other);
+    expect(`other:${rows.get("other")?.status}`, writes.join(", ")).toBe(other);
+    expect(reloaded.slice(1), writes.join(", ")).toEqual(shown.slice(1));
+  });
+
+  it("stores a snapshot-only answer that a second snapshot repeats", async () => {
+    const snapshot = (req: any[]) => ({
+      type: "MESSAGES_SNAPSHOT",
+      messages: [...req, { id: "only", role: "assistant", content: "o" }],
+    });
+    const { rows, writes, shown, reloaded } = await run((req) => [
+      [{ type: "RUN_STARTED", threadId: "t", runId: "r" }, snapshot(req)],
+      [snapshot(req)],
+      [{ type: "RUN_FINISHED", threadId: "t", runId: "r" }],
+    ]);
+    expect(rows.get("only")?.status, writes.join(", ")).toBe("complete");
+    expect(reloaded.slice(1), writes.join(", ")).toEqual(shown.slice(1));
+  });
+
+  it("stores an answer that arrives only in a snapshot", async () => {
+    const { rows, writes, shown, reloaded } = await run((req) => [
+      [
+        { type: "RUN_STARTED", threadId: "t", runId: "r" },
+        {
+          type: "MESSAGES_SNAPSHOT",
+          messages: [...req, { id: "only", role: "assistant", content: "o" }],
+        },
+      ],
+      [{ type: "RUN_FINISHED", threadId: "t", runId: "r" }],
+    ]);
+    expect(shown).toContain("only:complete");
+    expect(rows.get("only")?.status, writes.join(", ")).toBe("complete");
+    expect(writes.filter((w) => w.endsWith(" only"))).toHaveLength(1);
+    expect(reloaded.slice(1), writes.join(", ")).toEqual(shown.slice(1));
+  });
+
+  it("keeps the streamed answer as the head after a snapshot adds an earlier step", async () => {
+    const { writes, shown, reloaded } = await run((req) => [
+      start,
+      [
+        {
+          type: "MESSAGES_SNAPSHOT",
+          messages: [
+            ...req,
+            { id: "step", role: "assistant", content: "s" },
+            { id: "answer", role: "assistant", content: "a1" },
+          ],
+        },
+      ],
+      [end, { type: "RUN_FINISHED", threadId: "t", runId: "r" }],
+    ]);
+    expect(shown.slice(1)).toEqual(["step:complete", "answer:complete"]);
+    expect(reloaded.slice(1), writes.join(", ")).toEqual(shown.slice(1));
+  });
+
+  it("control: a second snapshot after the head move stores the new head", async () => {
+    const { rows, writes, shown, reloaded } = await run((req) => [
+      start,
+      [moveSnapshot(req)],
+      [moveSnapshot(req)],
+      [end, { type: "RUN_FINISHED", threadId: "t", runId: "r" }],
+    ]);
+    expect(rows.get("other")?.status, JSON.stringify({ writes, shown })).toBe(
+      "complete",
+    );
+    expect(reloaded.slice(1), writes.join(", ")).toEqual(shown.slice(1));
+  });
+
+  it("control: the streamed answer without a snapshot is stored", async () => {
+    const { rows, shown, reloaded } = await run(() => [
+      start,
+      [end, { type: "RUN_FINISHED", threadId: "t", runId: "r" }],
+    ]);
+    expect(rows.get("answer")?.status).toBe("complete");
+    expect(reloaded).toEqual(shown);
+  });
+});

package/src/runtime/AgUiThreadController.snapshot-nested-settle.test.ts

@@ -0,0 +1,210 @@
+import { describe, expect, it } from "vitest";
+import type { AppendMessage } from "@assistant-ui/core";
+import { HttpAgent } from "@ag-ui/client";
+import { AgUiThreadController } from "./AgUiThreadController";
+import { makeLogger } from "./logger";
+
+const tick = () => new Promise((r) => setTimeout(r, 0));
+const settle = async () => {
+  for (let i = 0; i < 8; i++) await tick();
+};
+const userMessage = {
+  role: "user",
+  content: [{ type: "text", text: "hello" }],
+  attachments: [],
+  metadata: { custom: {} },
+  createdAt: new Date(0),
+  parentId: null,
+  sourceId: null,
+  runConfig: {},
+  startRun: true,
+} as unknown as AppendMessage;
+
+type End = "close" | "error" | "stop" | "finish";
+
+async function run(end: End, withSnapshot = true, history?: unknown) {
+  let write!: (e: object) => void;
+  let close!: () => void;
+  let requestMessages: any[] = [];
+  const agent = new HttpAgent({
+    url: "https://agent.test",
+    fetch: async (_url: unknown, init: RequestInit) => {
+      const encoder = new TextEncoder();
+      let controller!: ReadableStreamDefaultController<Uint8Array>;
+      let closed = false;
+      const body = new ReadableStream<Uint8Array>({
+        start: (c) => void (controller = c),
+      });
+      requestMessages = JSON.parse(init.body as string).messages ?? [];
+      write = (e) => {
+        if (!closed)
+          controller.enqueue(encoder.encode(`data: ${JSON.stringify(e)}\n\n`));
+      };
+      close = () => {
+        if (closed) return;
+        closed = true;
+        controller.close();
+      };
+      init.signal?.addEventListener("abort", () => {
+        if (closed) return;
+        closed = true;
+        try {
+          controller.error(new DOMException("aborted", "AbortError"));
+        } catch {}
+      });
+      return new Response(body, {
+        headers: { "Content-Type": "text/event-stream" },
+      });
+    },
+  });
+  const core = new AgUiThreadController({
+    agent,
+    logger: makeLogger(),
+    showThinking: false,
+    ...(history ? { history: history as any } : {}),
+  });
+  const sending = core.append(userMessage).catch(() => {});
+  for (let i = 0; i < 50 && !write; i++) await tick();
+  const events: object[] = [
+    { type: "RUN_STARTED", threadId: "t", runId: "r" },
+    { type: "TEXT_MESSAGE_START", messageId: "answer", role: "assistant" },
+    { type: "TEXT_MESSAGE_CONTENT", messageId: "answer", delta: "a1" },
+    { type: "TEXT_MESSAGE_END", messageId: "answer" },
+    { type: "TOOL_CALL_START", toolCallId: "task", toolCallName: "task" },
+    { type: "TOOL_CALL_END", toolCallId: "task" },
+    {
+      type: "SUBAGENT_STARTED",
+      subagentRunId: "sub",
+      name: "worker",
+      parentToolCallId: "task",
+    },
+    {
+      type: "TOOL_CALL_START",
+      toolCallId: "ncall",
+      toolCallName: "search",
+      subagentRunId: "sub",
+    },
+    { type: "TOOL_CALL_END", toolCallId: "ncall", subagentRunId: "sub" },
+    { type: "SUBAGENT_FINISHED", subagentRunId: "sub" },
+  ];
+  for (const e of events) write(e);
+  await settle();
+  if (withSnapshot) {
+    write({
+      type: "MESSAGES_SNAPSHOT",
+      messages: [
+        ...requestMessages,
+        {
+          id: "answer",
+          role: "assistant",
+          content: "a1",
+          toolCalls: [
+            {
+              id: "task",
+              type: "function",
+              function: { name: "task", arguments: "{}" },
+            },
+          ],
+        },
+        {
+          id: "s-call",
+          role: "assistant",
+          content: "",
+          subagentRunId: "sub",
+          toolCalls: [
+            {
+              id: "ncall",
+              type: "function",
+              function: { name: "search", arguments: "{}" },
+            },
+          ],
+        },
+      ],
+    });
+    await settle();
+  }
+  const before = nested(core);
+  if (end === "close") close();
+  else if (end === "error")
+    write({ type: "RUN_ERROR", message: "server failed" });
+  else if (end === "finish")
+    write({ type: "RUN_FINISHED", threadId: "t", runId: "r" });
+  else await core.cancel();
+  await settle();
+  close();
+  await sending;
+  await settle();
+  return { before, after: nested(core) };
+}
+
+function nested(core: AgUiThreadController) {
+  const answer = core.getMessages().find((m: any) => m.id === "answer") as any;
+  const call = answer?.content.find(
+    (p: any) => p.type === "tool-call" && p.toolCallId === "task",
+  );
+  const sub = call?.messages?.find((m: any) => m.id === "sub");
+  return {
+    answer: answer?.status?.type,
+    sub: sub?.status?.type,
+    subReason: sub?.status?.reason,
+    ncallOpen: sub?.content.some(
+      (p: any) => p.toolCallId === "ncall" && p.result === undefined,
+    ),
+  };
+}
+
+describe("ag-ui: a nested message waiting on its own call after the run ends", () => {
+  for (const end of ["close", "error", "stop"] as const) {
+    it(`settles the nested message when the run ends by ${end}`, async () => {
+      const { before, after } = await run(end);
+      // the snapshot imported the sub-agent's message waiting on ncall
+      expect(before.sub, JSON.stringify(before)).toBe("requires-action");
+      expect(after.answer, JSON.stringify(after)).toBe("incomplete");
+      expect(after.sub, JSON.stringify(after)).not.toBe("requires-action");
+    });
+  }
+
+  it("control: after RUN_FINISHED the nested call stays open and is not settled as an error", async () => {
+    const { after } = await run("finish");
+    expect(after.answer, JSON.stringify(after)).toBe("requires-action");
+    expect(after.ncallOpen, JSON.stringify(after)).toBe(true);
+    expect(after.subReason, JSON.stringify(after)).not.toBe("error");
+  });
+
+  it("control: without the snapshot the nested message is the stream's (complete) and settles", async () => {
+    const { before, after } = await run("close", false);
+    expect(before.sub, JSON.stringify(before)).toBe("complete");
+    expect(after.sub, JSON.stringify(after)).not.toBe("requires-action");
+  });
+
+  it("appends the answer again when its first append fails under a queued update", async () => {
+    const writes: string[] = [];
+    let headId: string | null = null;
+    const rows = new Set<string>();
+    let failed = false;
+    const history = {
+      load: async () => ({ headId: null, messages: [] }),
+      append: async (item: any) => {
+        const id = item.message.id;
+        writes.push(`append ${id}`);
+        if (id === "answer" && !failed) {
+          failed = true;
+          await tick();
+          throw new Error("write failed");
+        }
+        rows.add(id);
+        headId = id;
+      },
+      // Follows the ThreadHistoryAdapter contract: update does not move the head.
+      update: async (item: any) => {
+        writes.push(`update ${item.message.id}`);
+        rows.add(item.message.id);
+      },
+    };
+    const { after } = await run("error", true, history);
+    expect(after.sub, JSON.stringify(after)).not.toBe("requires-action");
+    expect(writes, writes.join(", ")).not.toContain("update answer");
+    expect(rows.has("answer"), writes.join(", ")).toBe(true);
+    expect(headId, writes.join(", ")).toBe("answer");
+  });
+});