@assistant-ui/react 0.15.14 → 0.15.16

package/dist/mcp-apps/bridge.js.map

@@ -1 +1 @@
-{"version":3,"file":"bridge.js","names":["SandboxHostFrame","MCP_APP_PROTOCOL_VERSION","McpAppBridgeHandlers","McpAppDisplayMode","McpAppHostContext","McpAppHostInfo","McpAppJsonRpcMessage","McpAppJsonRpcNotification","McpAppJsonRpcRequest","McpAppJsonRpcResponse","isRecord","VALID_DISPLAY_MODES","const","McpAppBridgeFrame","CreateMcpAppBridgeOptions","frame","handlers","hostInfo","hostContext","McpAppBridge","onMessage","event","MessageEvent","dispose","notifyToolInput","input","notifyToolResult","result","notifyHostContextChanged","DEFAULT_HOST_INFO","name","version","METHOD_ALIASES","Record","normalizeMethod","method","JSONRPC_ERROR","parseError","invalidRequest","methodNotFound","invalidParams","internalError","isJsonRpcMessage","value","v","jsonrpc","isRequest","msg","isNotification","createMcpAppBridge","opts","post","sendMessage","respond","id","payload","error","code","message","data","res","errorResponse","undefined","handleRequest","req","params","requestedProtocolVersion","protocolVersion","host","capabilities","tools","callTool","resources","readResource","listResources","ui","openLink","requestDisplayMode","updateModelContext","hostCapabilities","openLinks","serverTools","serverResources","text","onLog","logging","callParams","arguments","allowedTools","includes","callArgs","Array","isArray","readParams","uri","linkParams","url","linkProtocol","URL","protocol","modeParams","mode","err","Error","String","onError","handleNotification","note","onInitialized","p","width","height","onSizeChange","onRequestTeardown","content","type","ctx"],"sources":["../../src/mcp-apps/bridge.ts"],"sourcesContent":["import type { SandboxHostFrame } from \"../sandbox-host/SandboxHost\";\nimport {\n  MCP_APP_PROTOCOL_VERSION,\n  type McpAppBridgeHandlers,\n  type McpAppDisplayMode,\n  type McpAppHostContext,\n  type McpAppHostInfo,\n  type McpAppJsonRpcMessage,\n  type McpAppJsonRpcNotification,\n  type McpAppJsonRpcRequest,\n  type McpAppJsonRpcResponse,\n} from \"./types\";\nimport { isRecord } from \"@assistant-ui/core/internal\";\n\nconst VALID_DISPLAY_MODES = [\n  \"inline\",\n  \"fullscreen\",\n  \"pip\",\n] as const satisfies readonly McpAppDisplayMode[];\n\nexport type McpAppBridgeFrame = SandboxHostFrame;\n\nexport type CreateMcpAppBridgeOptions = {\n  frame: McpAppBridgeFrame;\n  handlers?: McpAppBridgeHandlers | undefined;\n  hostInfo?: McpAppHostInfo | undefined;\n  hostContext?: McpAppHostContext | undefined;\n};\n\nexport type McpAppBridge = {\n  onMessage: (event: MessageEvent) => void;\n  dispose: () => void;\n  notifyToolInput: (input: unknown) => void;\n  notifyToolResult: (result: unknown) => void;\n  notifyHostContextChanged: (hostContext: McpAppHostContext) => void;\n};\n\nconst DEFAULT_HOST_INFO: McpAppHostInfo = {\n  name: \"assistant-ui\",\n  version: \"0.1\",\n};\n\n// Accept both the legacy method names and the MCP-UI 2026-01-26 names that\n// `ui/*` capable widgets (e.g. xmcp's host-bridge) emit. Normalize on input\n// so downstream switch statements only need to know the legacy names.\nconst METHOD_ALIASES: Record<string, string> = {\n  \"ui/notifications/initialized\": \"notifications/initialized\",\n  \"ui/notifications/size-changed\": \"notifications/size_changed\",\n  \"ui/request-display-mode\": \"requestDisplayMode\",\n  \"ui/open-link\": \"openLink\",\n  \"ui/update-model-context\": \"updateModelContext\",\n  \"ui/message\": \"sendMessage\",\n  \"notifications/message\": \"notifications/log\",\n};\n\nconst normalizeMethod = (method: string): string =>\n  METHOD_ALIASES[method] ?? method;\n\nconst JSONRPC_ERROR = {\n  parseError: -32700,\n  invalidRequest: -32600,\n  methodNotFound: -32601,\n  invalidParams: -32602,\n  internalError: -32603,\n} as const;\n\nfunction isJsonRpcMessage(value: unknown): value is McpAppJsonRpcMessage {\n  if (!value || typeof value !== \"object\") return false;\n  const v = value as Record<string, unknown>;\n  return v.jsonrpc === \"2.0\" && typeof v.method === \"string\";\n}\n\nfunction isRequest(msg: McpAppJsonRpcMessage): msg is McpAppJsonRpcRequest {\n  return \"id\" in msg;\n}\n\nfunction isNotification(\n  msg: McpAppJsonRpcMessage,\n): msg is McpAppJsonRpcNotification {\n  return !(\"id\" in msg);\n}\n\nexport function createMcpAppBridge(\n  opts: CreateMcpAppBridgeOptions,\n): McpAppBridge {\n  const {\n    frame,\n    handlers = {},\n    hostInfo = DEFAULT_HOST_INFO,\n    hostContext = {},\n  } = opts;\n\n  const post = (msg: McpAppJsonRpcMessage) => {\n    frame.sendMessage(msg);\n  };\n\n  const respond = (\n    id: McpAppJsonRpcRequest[\"id\"],\n    payload:\n      | { result: unknown }\n      | { error: { code: number; message: string; data?: unknown } },\n  ) => {\n    const res: McpAppJsonRpcResponse = {\n      jsonrpc: \"2.0\",\n      id,\n      ...payload,\n    };\n    post(res);\n  };\n\n  const errorResponse = (\n    id: McpAppJsonRpcRequest[\"id\"],\n    code: number,\n    message: string,\n    data?: unknown,\n  ) => {\n    respond(id, {\n      error: {\n        code,\n        message,\n        ...(data !== undefined ? { data } : {}),\n      },\n    });\n  };\n\n  const handleRequest = async (req: McpAppJsonRpcRequest) => {\n    try {\n      const params = req.params;\n\n      switch (normalizeMethod(req.method)) {\n        case \"ui/initialize\": {\n          const requestedProtocolVersion =\n            isRecord(params) && typeof params.protocolVersion === \"string\"\n              ? params.protocolVersion\n              : MCP_APP_PROTOCOL_VERSION;\n          respond(req.id, {\n            result: {\n              protocolVersion: requestedProtocolVersion,\n              host: hostInfo,\n              hostInfo,\n              hostContext,\n              capabilities: {\n                tools: handlers.callTool ? {} : undefined,\n                resources:\n                  handlers.readResource || handlers.listResources\n                    ? {}\n                    : undefined,\n                ui: {\n                  sendMessage: !!handlers.sendMessage,\n                  openLink: !!handlers.openLink,\n                  requestDisplayMode: !!handlers.requestDisplayMode,\n                  updateModelContext: !!handlers.updateModelContext,\n                },\n              },\n              hostCapabilities: {\n                ...(handlers.openLink ? { openLinks: {} } : {}),\n                ...(handlers.callTool ? { serverTools: {} } : {}),\n                ...(handlers.readResource || handlers.listResources\n                  ? { serverResources: {} }\n                  : {}),\n                ...(handlers.updateModelContext\n                  ? { updateModelContext: { text: {} } }\n                  : {}),\n                ...(handlers.sendMessage ? { message: { text: {} } } : {}),\n                ...(handlers.onLog ? { logging: {} } : {}),\n              },\n            },\n          });\n          return;\n        }\n\n        case \"tools/call\": {\n          if (!handlers.callTool) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"tools/call is not supported by this host\",\n            );\n            return;\n          }\n          const callParams = (params ?? {}) as {\n            name?: unknown;\n            arguments?: unknown;\n          };\n          if (typeof callParams.name !== \"string\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"tools/call requires a string 'name'\",\n            );\n            return;\n          }\n          if (\n            handlers.allowedTools &&\n            !handlers.allowedTools.includes(callParams.name)\n          ) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              `tool '${callParams.name}' is not allowed for this app`,\n            );\n            return;\n          }\n          let callArgs: Record<string, unknown> | undefined;\n          if (callParams.arguments !== undefined) {\n            if (\n              callParams.arguments === null ||\n              typeof callParams.arguments !== \"object\" ||\n              Array.isArray(callParams.arguments)\n            ) {\n              errorResponse(\n                req.id,\n                JSONRPC_ERROR.invalidParams,\n                \"tools/call 'arguments' must be an object\",\n              );\n              return;\n            }\n            callArgs = callParams.arguments as Record<string, unknown>;\n          }\n          const result = await handlers.callTool({\n            name: callParams.name,\n            ...(callArgs !== undefined ? { arguments: callArgs } : {}),\n          });\n          respond(req.id, { result });\n          return;\n        }\n\n        case \"resources/read\": {\n          if (!handlers.readResource) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"resources/read is not supported by this host\",\n            );\n            return;\n          }\n          const readParams = (params ?? {}) as { uri?: unknown };\n          if (typeof readParams.uri !== \"string\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"resources/read requires a string 'uri'\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: await handlers.readResource({ uri: readParams.uri }),\n          });\n          return;\n        }\n\n        case \"resources/list\": {\n          if (!handlers.listResources) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"resources/list is not supported by this host\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: (await handlers.listResources(params)) ?? null,\n          });\n          return;\n        }\n\n        case \"openLink\": {\n          if (!handlers.openLink) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"openLink is not supported by this host\",\n            );\n            return;\n          }\n          const linkParams = (params ?? {}) as { url?: unknown };\n          if (typeof linkParams.url !== \"string\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"openLink requires a string 'url'\",\n            );\n            return;\n          }\n          let linkProtocol: string;\n          try {\n            linkProtocol = new URL(linkParams.url).protocol;\n          } catch {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"openLink requires a valid URL\",\n            );\n            return;\n          }\n          if (linkProtocol !== \"https:\" && linkProtocol !== \"http:\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"openLink only accepts http(s) URLs\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: await handlers.openLink({ url: linkParams.url }),\n          });\n          return;\n        }\n\n        case \"sendMessage\": {\n          if (!handlers.sendMessage) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"sendMessage is not supported by this host\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: (await handlers.sendMessage(params)) ?? null,\n          });\n          return;\n        }\n\n        case \"updateModelContext\": {\n          if (!handlers.updateModelContext) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"updateModelContext is not supported by this host\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: (await handlers.updateModelContext(params)) ?? null,\n          });\n          return;\n        }\n\n        case \"requestDisplayMode\": {\n          if (!handlers.requestDisplayMode) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"requestDisplayMode is not supported by this host\",\n            );\n            return;\n          }\n          const modeParams = (params ?? {}) as { mode?: unknown };\n          if (\n            typeof modeParams.mode !== \"string\" ||\n            !VALID_DISPLAY_MODES.includes(modeParams.mode as McpAppDisplayMode)\n          ) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"requestDisplayMode requires a valid 'mode'\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: await handlers.requestDisplayMode({\n              mode: modeParams.mode as McpAppDisplayMode,\n            }),\n          });\n          return;\n        }\n\n        default: {\n          errorResponse(\n            req.id,\n            JSONRPC_ERROR.methodNotFound,\n            `Unknown method: ${req.method}`,\n          );\n        }\n      }\n    } catch (err) {\n      const error = err instanceof Error ? err : new Error(String(err));\n      handlers.onError?.(error);\n      errorResponse(req.id, JSONRPC_ERROR.internalError, error.message);\n    }\n  };\n\n  const handleNotification = (note: McpAppJsonRpcNotification) => {\n    switch (normalizeMethod(note.method)) {\n      case \"notifications/initialized\": {\n        handlers.onInitialized?.();\n        return;\n      }\n      case \"notifications/size_changed\": {\n        const p = (note.params ?? {}) as { width?: number; height?: number };\n        handlers.onSizeChange?.({\n          ...(typeof p.width === \"number\" ? { width: p.width } : {}),\n          ...(typeof p.height === \"number\" ? { height: p.height } : {}),\n        });\n        return;\n      }\n      case \"notifications/log\": {\n        handlers.onLog?.(note.params);\n        return;\n      }\n      case \"notifications/request_teardown\": {\n        handlers.onRequestTeardown?.(note.params);\n        return;\n      }\n      case \"notifications/error\": {\n        const p = (note.params ?? {}) as { message?: string };\n        handlers.onError?.(\n          new Error(typeof p.message === \"string\" ? p.message : \"Widget error\"),\n        );\n        return;\n      }\n      default:\n        return;\n    }\n  };\n\n  // The host applies the cross-origin guard before delegating; this only\n  // validates the JSON-RPC envelope.\n  const onMessage = (event: MessageEvent) => {\n    if (!isJsonRpcMessage(event.data)) return;\n\n    const msg = event.data;\n    if (isRequest(msg)) {\n      void handleRequest(msg);\n    } else if (isNotification(msg)) {\n      handleNotification(msg);\n    }\n  };\n\n  return {\n    onMessage,\n    dispose: () => {},\n    notifyToolInput: (input: unknown) => {\n      post({\n        jsonrpc: \"2.0\",\n        method: \"notifications/tools/call/input\",\n        params: { input },\n      });\n      post({\n        jsonrpc: \"2.0\",\n        method: \"ui/notifications/tool-input\",\n        params: isRecord(input) ? { arguments: input } : {},\n      });\n    },\n    notifyToolResult: (result: unknown) => {\n      post({\n        jsonrpc: \"2.0\",\n        method: \"notifications/tools/call/result\",\n        params: { result },\n      });\n      post({\n        jsonrpc: \"2.0\",\n        method: \"ui/notifications/tool-result\",\n        params: isRecord(result)\n          ? result\n          : { content: [{ type: \"text\", text: String(result) }] },\n      });\n    },\n    notifyHostContextChanged: (ctx: McpAppHostContext) => {\n      post({\n        jsonrpc: \"2.0\",\n        method: \"notifications/host_context/changed\",\n        params: ctx,\n      });\n      post({\n        jsonrpc: \"2.0\",\n        method: \"ui/notifications/host-context-changed\",\n        params: ctx,\n      });\n    },\n  };\n}\n"],"mappings":";;;AAcA,MAAMW,sBAAsB;CAC1B;CACA;CACA;AAAK;AAoBP,MAAMkB,oBAAoC;CACxCC,MAAM;CACNC,SAAS;AACX;AAKA,MAAMC,iBAAyC;CAC7C,gCAAgC;CAChC,iCAAiC;CACjC,2BAA2B;CAC3B,gBAAgB;CAChB,2BAA2B;CAC3B,cAAc;CACd,yBAAyB;AAC3B;AAEA,MAAME,mBAAmBC,WACvBH,eAAeG,WAAWA;AAE5B,MAAMC,gBAAgB;CACpBC,YAAY;CACZC,gBAAgB;CAChBC,gBAAgB;CAChBC,eAAe;CACfC,eAAe;AACjB;AAEA,SAASC,iBAAiBC,OAA+C;CACvE,IAAI,CAACA,SAAS,OAAOA,UAAU,UAAU,OAAO;CAChD,MAAMC,IAAID;CACV,OAAOC,EAAEC,YAAY,SAAS,OAAOD,EAAET,WAAW;AACpD;AAEA,SAASW,UAAUC,KAAwD;CACzE,OAAO,QAAQA;AACjB;AAEA,SAASC,eACPD,KACkC;CAClC,OAAO,EAAE,QAAQA;AACnB;AAEA,SAAgBE,mBACdC,MACc;CACd,MAAM,EACJnC,OACAC,WAAW,CAAC,GACZC,WAAWY,mBACXX,cAAc,CAAC,MACbgC;CAEJ,MAAMC,QAAQJ,QAA8B;EAC1ChC,MAAMqC,YAAYL,GAAG;CACvB;CAEA,MAAMM,WACJC,IACAC,YAGG;EACH,MAAMK,MAA6B;GACjCf,SAAS;GACTS;GACA,GAAGC;EACL;EACAJ,KAAKS,GAAG;CACV;CAEA,MAAMC,iBACJP,IACAG,MACAC,SACAC,SACG;EACHN,QAAQC,IAAI,EACVE,OAAO;GACLC;GACAC;GACA,GAAIC,SAASG,KAAAA,IAAY,EAAEH,KAAK,IAAI,CAAC;EACvC,EACF,CAAC;CACH;CAEA,MAAMI,gBAAgB,OAAOC,QAA8B;EACzD,IAAI;GACF,MAAMC,SAASD,IAAIC;GAEnB,QAAQ/B,gBAAgB8B,IAAI7B,MAAM,GAAlC;IACE,KAAK,iBAAiB;KACpB,MAAM+B,2BACJxD,SAASuD,MAAM,KAAK,OAAOA,OAAOE,oBAAoB,WAClDF,OAAOE,kBAAAA;KAEbd,QAAQW,IAAIV,IAAI,EACd3B,QAAQ;MACNwC,iBAAiBD;MACjBE,MAAMnD;MACNA;MACAC;MACAmD,cAAc;OACZC,OAAOtD,SAASuD,WAAW,CAAC,IAAIT,KAAAA;OAChCU,WACExD,SAASyD,gBAAgBzD,SAAS0D,gBAC9B,CAAC,IACDZ,KAAAA;OACNa,IAAI;QACFvB,aAAa,CAAC,CAACpC,SAASoC;QACxBwB,UAAU,CAAC,CAAC5D,SAAS4D;QACrBC,oBAAoB,CAAC,CAAC7D,SAAS6D;QAC/BC,oBAAoB,CAAC,CAAC9D,SAAS8D;OACjC;MACF;MACAC,kBAAkB;OAChB,GAAI/D,SAAS4D,WAAW,EAAEI,WAAW,CAAC,EAAE,IAAI,CAAC;OAC7C,GAAIhE,SAASuD,WAAW,EAAEU,aAAa,CAAC,EAAE,IAAI,CAAC;OAC/C,GAAIjE,SAASyD,gBAAgBzD,SAAS0D,gBAClC,EAAEQ,iBAAiB,CAAC,EAAE,IACtB,CAAC;OACL,GAAIlE,SAAS8D,qBACT,EAAEA,oBAAoB,EAAEK,MAAM,CAAC,EAAE,EAAE,IACnC,CAAC;OACL,GAAInE,SAASoC,cAAc,EAAEM,SAAS,EAAEyB,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC;OACxD,GAAInE,SAASoE,QAAQ,EAAEC,SAAS,CAAC,EAAE,IAAI,CAAC;MAC1C;KACF,EACF,CAAC;KACD;IACF;IAEA,KAAK,cAAc;KACjB,IAAI,CAACrE,SAASuD,UAAU;MACtBV,cACEG,IAAIV,IACJlB,cAAcG,gBACd,0CACF;MACA;KACF;KACA,MAAM+C,aAAcrB,UAAU,CAAC;KAI/B,IAAI,OAAOqB,WAAWxD,SAAS,UAAU;MACvC+B,cACEG,IAAIV,IACJlB,cAAcI,eACd,qCACF;MACA;KACF;KACA,IACExB,SAASwE,gBACT,CAACxE,SAASwE,aAAaC,SAASH,WAAWxD,IAAI,GAC/C;MACA+B,cACEG,IAAIV,IACJlB,cAAcI,eACd,SAAS8C,WAAWxD,KAAI,8BAC1B;MACA;KACF;KACA,IAAI4D;KACJ,IAAIJ,WAAWC,cAAczB,KAAAA,GAAW;MACtC,IACEwB,WAAWC,cAAc,QACzB,OAAOD,WAAWC,cAAc,YAChCI,MAAMC,QAAQN,WAAWC,SAAS,GAClC;OACA1B,cACEG,IAAIV,IACJlB,cAAcI,eACd,0CACF;OACA;MACF;MACAkD,WAAWJ,WAAWC;KACxB;KACA,MAAM5D,SAAS,MAAMX,SAASuD,SAAS;MACrCzC,MAAMwD,WAAWxD;MACjB,GAAI4D,aAAa5B,KAAAA,IAAY,EAAEyB,WAAWG,SAAS,IAAI,CAAC;KAC1D,CAAC;KACDrC,QAAQW,IAAIV,IAAI,EAAE3B,OAAO,CAAC;KAC1B;IACF;IAEA,KAAK,kBAAkB;KACrB,IAAI,CAACX,SAASyD,cAAc;MAC1BZ,cACEG,IAAIV,IACJlB,cAAcG,gBACd,8CACF;MACA;KACF;KACA,MAAMsD,aAAc5B,UAAU,CAAC;KAC/B,IAAI,OAAO4B,WAAWC,QAAQ,UAAU;MACtCjC,cACEG,IAAIV,IACJlB,cAAcI,eACd,wCACF;MACA;KACF;KACAa,QAAQW,IAAIV,IAAI,EACd3B,QAAQ,MAAMX,SAASyD,aAAa,EAAEqB,KAAKD,WAAWC,IAAI,CAAC,EAC7D,CAAC;KACD;IACF;IAEA,KAAK;KACH,IAAI,CAAC9E,SAAS0D,eAAe;MAC3Bb,cACEG,IAAIV,IACJlB,cAAcG,gBACd,8CACF;MACA;KACF;KACAc,QAAQW,IAAIV,IAAI,EACd3B,QAAS,MAAMX,SAAS0D,cAAcT,MAAM,KAAM,KACpD,CAAC;KACD;IAGF,KAAK,YAAY;KACf,IAAI,CAACjD,SAAS4D,UAAU;MACtBf,cACEG,IAAIV,IACJlB,cAAcG,gBACd,wCACF;MACA;KACF;KACA,MAAMwD,aAAc9B,UAAU,CAAC;KAC/B,IAAI,OAAO8B,WAAWC,QAAQ,UAAU;MACtCnC,cACEG,IAAIV,IACJlB,cAAcI,eACd,kCACF;MACA;KACF;KACA,IAAIyD;KACJ,IAAI;MACFA,eAAe,IAAIC,IAAIH,WAAWC,GAAG,CAAC,CAACG;KACzC,QAAQ;MACNtC,cACEG,IAAIV,IACJlB,cAAcI,eACd,+BACF;MACA;KACF;KACA,IAAIyD,iBAAiB,YAAYA,iBAAiB,SAAS;MACzDpC,cACEG,IAAIV,IACJlB,cAAcI,eACd,oCACF;MACA;KACF;KACAa,QAAQW,IAAIV,IAAI,EACd3B,QAAQ,MAAMX,SAAS4D,SAAS,EAAEoB,KAAKD,WAAWC,IAAI,CAAC,EACzD,CAAC;KACD;IACF;IAEA,KAAK;KACH,IAAI,CAAChF,SAASoC,aAAa;MACzBS,cACEG,IAAIV,IACJlB,cAAcG,gBACd,2CACF;MACA;KACF;KACAc,QAAQW,IAAIV,IAAI,EACd3B,QAAS,MAAMX,SAASoC,YAAYa,MAAM,KAAM,KAClD,CAAC;KACD;IAGF,KAAK;KACH,IAAI,CAACjD,SAAS8D,oBAAoB;MAChCjB,cACEG,IAAIV,IACJlB,cAAcG,gBACd,kDACF;MACA;KACF;KACAc,QAAQW,IAAIV,IAAI,EACd3B,QAAS,MAAMX,SAAS8D,mBAAmBb,MAAM,KAAM,KACzD,CAAC;KACD;IAGF,KAAK,sBAAsB;KACzB,IAAI,CAACjD,SAAS6D,oBAAoB;MAChChB,cACEG,IAAIV,IACJlB,cAAcG,gBACd,kDACF;MACA;KACF;KACA,MAAM6D,aAAcnC,UAAU,CAAC;KAC/B,IACE,OAAOmC,WAAWC,SAAS,YAC3B,CAAC1F,oBAAoB8E,SAASW,WAAWC,IAAyB,GAClE;MACAxC,cACEG,IAAIV,IACJlB,cAAcI,eACd,4CACF;MACA;KACF;KACAa,QAAQW,IAAIV,IAAI,EACd3B,QAAQ,MAAMX,SAAS6D,mBAAmB,EACxCwB,MAAMD,WAAWC,KACnB,CAAC,EACH,CAAC;KACD;IACF;IAEA,SACExC,cACEG,IAAIV,IACJlB,cAAcG,gBACd,mBAAmByB,IAAI7B,QACzB;GAEJ;EACF,SAASmE,KAAK;GACZ,MAAM9C,QAAQ8C,eAAeC,QAAQD,MAAM,IAAIC,MAAMC,OAAOF,GAAG,CAAC;GAChEtF,SAASyF,UAAUjD,KAAK;GACxBK,cAAcG,IAAIV,IAAIlB,cAAcK,eAAee,MAAME,OAAO;EAClE;CACF;CAEA,MAAMgD,sBAAsBC,SAAoC;EAC9D,QAAQzE,gBAAgByE,KAAKxE,MAAM,GAAnC;GACE,KAAK;IACHnB,SAAS4F,gBAAgB;IACzB;GAEF,KAAK,8BAA8B;IACjC,MAAMC,IAAKF,KAAK1C,UAAU,CAAC;IAC3BjD,SAASgG,eAAe;KACtB,GAAI,OAAOH,EAAEC,UAAU,WAAW,EAAEA,OAAOD,EAAEC,MAAM,IAAI,CAAC;KACxD,GAAI,OAAOD,EAAEE,WAAW,WAAW,EAAEA,QAAQF,EAAEE,OAAO,IAAI,CAAC;IAC7D,CAAC;IACD;GACF;GACA,KAAK;IACH/F,SAASoE,QAAQuB,KAAK1C,MAAM;IAC5B;GAEF,KAAK;IACHjD,SAASiG,oBAAoBN,KAAK1C,MAAM;IACxC;GAEF,KAAK,uBAAuB;IAC1B,MAAM4C,IAAKF,KAAK1C,UAAU,CAAC;IAC3BjD,SAASyF,UACP,IAAIF,MAAM,OAAOM,EAAEnD,YAAY,WAAWmD,EAAEnD,UAAU,cAAc,CACtE;IACA;GACF;GACA,SACE;EACJ;CACF;CAIA,MAAMtC,aAAaC,UAAwB;EACzC,IAAI,CAACqB,iBAAiBrB,MAAMsC,IAAI,GAAG;EAEnC,MAAMZ,MAAM1B,MAAMsC;EAClB,IAAIb,UAAUC,GAAG,GACf,cAAmBA,GAAG;OACjB,IAAIC,eAAeD,GAAG,GAC3B2D,mBAAmB3D,GAAG;CAE1B;CAEA,OAAO;EACL3B;EACAG,eAAe,CAAC;EAChBC,kBAAkBC,UAAmB;GACnC0B,KAAK;IACHN,SAAS;IACTV,QAAQ;IACR8B,QAAQ,EAAExC,MAAM;GAClB,CAAC;GACD0B,KAAK;IACHN,SAAS;IACTV,QAAQ;IACR8B,QAAQvD,SAASe,KAAK,IAAI,EAAE8D,WAAW9D,MAAM,IAAI,CAAC;GACpD,CAAC;EACH;EACAC,mBAAmBC,WAAoB;GACrCwB,KAAK;IACHN,SAAS;IACTV,QAAQ;IACR8B,QAAQ,EAAEtC,OAAO;GACnB,CAAC;GACDwB,KAAK;IACHN,SAAS;IACTV,QAAQ;IACR8B,QAAQvD,SAASiB,MAAM,IACnBA,SACA,EAAEuF,SAAS,CAAC;KAAEC,MAAM;KAAQhC,MAAMqB,OAAO7E,MAAM;IAAE,CAAC,EAAE;GAC1D,CAAC;EACH;EACAC,2BAA2BwF,QAA2B;GACpDjE,KAAK;IACHN,SAAS;IACTV,QAAQ;IACR8B,QAAQmD;GACV,CAAC;GACDjE,KAAK;IACHN,SAAS;IACTV,QAAQ;IACR8B,QAAQmD;GACV,CAAC;EACH;CACF;AACF"}
+{"version":3,"file":"bridge.js","names":["SandboxHostFrame","invokeCallbackSafely","MCP_APP_PROTOCOL_VERSION","McpAppBridgeHandlers","McpAppDisplayMode","McpAppHostContext","McpAppHostInfo","McpAppJsonRpcMessage","McpAppJsonRpcNotification","McpAppJsonRpcRequest","McpAppJsonRpcResponse","isRecord","VALID_DISPLAY_MODES","const","McpAppBridgeFrame","CreateMcpAppBridgeOptions","frame","handlers","hostInfo","hostContext","McpAppBridge","onMessage","event","MessageEvent","dispose","notifyToolInput","input","notifyToolResult","result","notifyHostContextChanged","DEFAULT_HOST_INFO","name","version","METHOD_ALIASES","Record","normalizeMethod","method","JSONRPC_ERROR","parseError","invalidRequest","methodNotFound","invalidParams","internalError","isJsonRpcMessage","value","v","jsonrpc","isRequest","msg","isNotification","createMcpAppBridge","opts","post","sendMessage","respond","id","payload","error","code","message","data","res","errorResponse","undefined","reportError","Error","onError","handleRequest","req","params","requestedProtocolVersion","protocolVersion","host","capabilities","tools","callTool","resources","readResource","listResources","ui","openLink","requestDisplayMode","updateModelContext","hostCapabilities","openLinks","serverTools","serverResources","text","onLog","logging","callParams","arguments","allowedTools","includes","callArgs","Array","isArray","readParams","uri","linkParams","url","linkProtocol","URL","protocol","modeParams","mode","err","String","handleNotification","note","onInitialized","p","width","height","onSizeChange","onRequestTeardown","content","type","ctx"],"sources":["../../src/mcp-apps/bridge.ts"],"sourcesContent":["import type { SandboxHostFrame } from \"../sandbox-host/SandboxHost\";\nimport { invokeCallbackSafely } from \"../utils/invokeCallbackSafely\";\nimport {\n  MCP_APP_PROTOCOL_VERSION,\n  type McpAppBridgeHandlers,\n  type McpAppDisplayMode,\n  type McpAppHostContext,\n  type McpAppHostInfo,\n  type McpAppJsonRpcMessage,\n  type McpAppJsonRpcNotification,\n  type McpAppJsonRpcRequest,\n  type McpAppJsonRpcResponse,\n} from \"./types\";\nimport { isRecord } from \"@assistant-ui/core/internal\";\n\nconst VALID_DISPLAY_MODES = [\n  \"inline\",\n  \"fullscreen\",\n  \"pip\",\n] as const satisfies readonly McpAppDisplayMode[];\n\nexport type McpAppBridgeFrame = SandboxHostFrame;\n\nexport type CreateMcpAppBridgeOptions = {\n  frame: McpAppBridgeFrame;\n  handlers?: McpAppBridgeHandlers | undefined;\n  hostInfo?: McpAppHostInfo | undefined;\n  hostContext?: McpAppHostContext | undefined;\n};\n\nexport type McpAppBridge = {\n  onMessage: (event: MessageEvent) => void;\n  dispose: () => void;\n  notifyToolInput: (input: unknown) => void;\n  notifyToolResult: (result: unknown) => void;\n  notifyHostContextChanged: (hostContext: McpAppHostContext) => void;\n};\n\nconst DEFAULT_HOST_INFO: McpAppHostInfo = {\n  name: \"assistant-ui\",\n  version: \"0.1\",\n};\n\n// Accept both the legacy method names and the MCP-UI 2026-01-26 names that\n// `ui/*` capable widgets (e.g. xmcp's host-bridge) emit. Normalize on input\n// so downstream switch statements only need to know the legacy names.\nconst METHOD_ALIASES: Record<string, string> = {\n  \"ui/notifications/initialized\": \"notifications/initialized\",\n  \"ui/notifications/size-changed\": \"notifications/size_changed\",\n  \"ui/request-display-mode\": \"requestDisplayMode\",\n  \"ui/open-link\": \"openLink\",\n  \"ui/update-model-context\": \"updateModelContext\",\n  \"ui/message\": \"sendMessage\",\n  \"notifications/message\": \"notifications/log\",\n};\n\nconst normalizeMethod = (method: string): string =>\n  METHOD_ALIASES[method] ?? method;\n\nconst JSONRPC_ERROR = {\n  parseError: -32700,\n  invalidRequest: -32600,\n  methodNotFound: -32601,\n  invalidParams: -32602,\n  internalError: -32603,\n} as const;\n\nfunction isJsonRpcMessage(value: unknown): value is McpAppJsonRpcMessage {\n  if (!value || typeof value !== \"object\") return false;\n  const v = value as Record<string, unknown>;\n  return v.jsonrpc === \"2.0\" && typeof v.method === \"string\";\n}\n\nfunction isRequest(msg: McpAppJsonRpcMessage): msg is McpAppJsonRpcRequest {\n  return \"id\" in msg;\n}\n\nfunction isNotification(\n  msg: McpAppJsonRpcMessage,\n): msg is McpAppJsonRpcNotification {\n  return !(\"id\" in msg);\n}\n\nexport function createMcpAppBridge(\n  opts: CreateMcpAppBridgeOptions,\n): McpAppBridge {\n  const {\n    frame,\n    handlers = {},\n    hostInfo = DEFAULT_HOST_INFO,\n    hostContext = {},\n  } = opts;\n\n  const post = (msg: McpAppJsonRpcMessage) => {\n    frame.sendMessage(msg);\n  };\n\n  const respond = (\n    id: McpAppJsonRpcRequest[\"id\"],\n    payload:\n      | { result: unknown }\n      | { error: { code: number; message: string; data?: unknown } },\n  ) => {\n    const res: McpAppJsonRpcResponse = {\n      jsonrpc: \"2.0\",\n      id,\n      ...payload,\n    };\n    post(res);\n  };\n\n  const errorResponse = (\n    id: McpAppJsonRpcRequest[\"id\"],\n    code: number,\n    message: string,\n    data?: unknown,\n  ) => {\n    respond(id, {\n      error: {\n        code,\n        message,\n        ...(data !== undefined ? { data } : {}),\n      },\n    });\n  };\n\n  const reportError = (error: Error) => {\n    invokeCallbackSafely(() => handlers.onError?.(error), \"MCP App onError\");\n  };\n\n  const handleRequest = async (req: McpAppJsonRpcRequest) => {\n    try {\n      const params = req.params;\n\n      switch (normalizeMethod(req.method)) {\n        case \"ui/initialize\": {\n          const requestedProtocolVersion =\n            isRecord(params) && typeof params.protocolVersion === \"string\"\n              ? params.protocolVersion\n              : MCP_APP_PROTOCOL_VERSION;\n          respond(req.id, {\n            result: {\n              protocolVersion: requestedProtocolVersion,\n              host: hostInfo,\n              hostInfo,\n              hostContext,\n              capabilities: {\n                tools: handlers.callTool ? {} : undefined,\n                resources:\n                  handlers.readResource || handlers.listResources\n                    ? {}\n                    : undefined,\n                ui: {\n                  sendMessage: !!handlers.sendMessage,\n                  openLink: !!handlers.openLink,\n                  requestDisplayMode: !!handlers.requestDisplayMode,\n                  updateModelContext: !!handlers.updateModelContext,\n                },\n              },\n              hostCapabilities: {\n                ...(handlers.openLink ? { openLinks: {} } : {}),\n                ...(handlers.callTool ? { serverTools: {} } : {}),\n                ...(handlers.readResource || handlers.listResources\n                  ? { serverResources: {} }\n                  : {}),\n                ...(handlers.updateModelContext\n                  ? { updateModelContext: { text: {} } }\n                  : {}),\n                ...(handlers.sendMessage ? { message: { text: {} } } : {}),\n                ...(handlers.onLog ? { logging: {} } : {}),\n              },\n            },\n          });\n          return;\n        }\n\n        case \"tools/call\": {\n          if (!handlers.callTool) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"tools/call is not supported by this host\",\n            );\n            return;\n          }\n          const callParams = (params ?? {}) as {\n            name?: unknown;\n            arguments?: unknown;\n          };\n          if (typeof callParams.name !== \"string\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"tools/call requires a string 'name'\",\n            );\n            return;\n          }\n          if (\n            handlers.allowedTools &&\n            !handlers.allowedTools.includes(callParams.name)\n          ) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              `tool '${callParams.name}' is not allowed for this app`,\n            );\n            return;\n          }\n          let callArgs: Record<string, unknown> | undefined;\n          if (callParams.arguments !== undefined) {\n            if (\n              callParams.arguments === null ||\n              typeof callParams.arguments !== \"object\" ||\n              Array.isArray(callParams.arguments)\n            ) {\n              errorResponse(\n                req.id,\n                JSONRPC_ERROR.invalidParams,\n                \"tools/call 'arguments' must be an object\",\n              );\n              return;\n            }\n            callArgs = callParams.arguments as Record<string, unknown>;\n          }\n          const result = await handlers.callTool({\n            name: callParams.name,\n            ...(callArgs !== undefined ? { arguments: callArgs } : {}),\n          });\n          respond(req.id, { result });\n          return;\n        }\n\n        case \"resources/read\": {\n          if (!handlers.readResource) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"resources/read is not supported by this host\",\n            );\n            return;\n          }\n          const readParams = (params ?? {}) as { uri?: unknown };\n          if (typeof readParams.uri !== \"string\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"resources/read requires a string 'uri'\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: await handlers.readResource({ uri: readParams.uri }),\n          });\n          return;\n        }\n\n        case \"resources/list\": {\n          if (!handlers.listResources) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"resources/list is not supported by this host\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: (await handlers.listResources(params)) ?? null,\n          });\n          return;\n        }\n\n        case \"openLink\": {\n          if (!handlers.openLink) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"openLink is not supported by this host\",\n            );\n            return;\n          }\n          const linkParams = (params ?? {}) as { url?: unknown };\n          if (typeof linkParams.url !== \"string\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"openLink requires a string 'url'\",\n            );\n            return;\n          }\n          let linkProtocol: string;\n          try {\n            linkProtocol = new URL(linkParams.url).protocol;\n          } catch {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"openLink requires a valid URL\",\n            );\n            return;\n          }\n          if (linkProtocol !== \"https:\" && linkProtocol !== \"http:\") {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"openLink only accepts http(s) URLs\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: await handlers.openLink({ url: linkParams.url }),\n          });\n          return;\n        }\n\n        case \"sendMessage\": {\n          if (!handlers.sendMessage) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"sendMessage is not supported by this host\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: (await handlers.sendMessage(params)) ?? null,\n          });\n          return;\n        }\n\n        case \"updateModelContext\": {\n          if (!handlers.updateModelContext) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"updateModelContext is not supported by this host\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: (await handlers.updateModelContext(params)) ?? null,\n          });\n          return;\n        }\n\n        case \"requestDisplayMode\": {\n          if (!handlers.requestDisplayMode) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.methodNotFound,\n              \"requestDisplayMode is not supported by this host\",\n            );\n            return;\n          }\n          const modeParams = (params ?? {}) as { mode?: unknown };\n          if (\n            typeof modeParams.mode !== \"string\" ||\n            !VALID_DISPLAY_MODES.includes(modeParams.mode as McpAppDisplayMode)\n          ) {\n            errorResponse(\n              req.id,\n              JSONRPC_ERROR.invalidParams,\n              \"requestDisplayMode requires a valid 'mode'\",\n            );\n            return;\n          }\n          respond(req.id, {\n            result: await handlers.requestDisplayMode({\n              mode: modeParams.mode as McpAppDisplayMode,\n            }),\n          });\n          return;\n        }\n\n        default: {\n          errorResponse(\n            req.id,\n            JSONRPC_ERROR.methodNotFound,\n            `Unknown method: ${req.method}`,\n          );\n        }\n      }\n    } catch (err) {\n      const error = err instanceof Error ? err : new Error(String(err));\n      reportError(error);\n      errorResponse(req.id, JSONRPC_ERROR.internalError, error.message);\n    }\n  };\n\n  const handleNotification = (note: McpAppJsonRpcNotification) => {\n    try {\n      switch (normalizeMethod(note.method)) {\n        case \"notifications/initialized\": {\n          handlers.onInitialized?.();\n          return;\n        }\n        case \"notifications/size_changed\": {\n          const p = (note.params ?? {}) as { width?: number; height?: number };\n          handlers.onSizeChange?.({\n            ...(typeof p.width === \"number\" ? { width: p.width } : {}),\n            ...(typeof p.height === \"number\" ? { height: p.height } : {}),\n          });\n          return;\n        }\n        case \"notifications/log\": {\n          handlers.onLog?.(note.params);\n          return;\n        }\n        case \"notifications/request_teardown\": {\n          handlers.onRequestTeardown?.(note.params);\n          return;\n        }\n        case \"notifications/error\": {\n          const p = (note.params ?? {}) as { message?: string };\n          reportError(\n            new Error(\n              typeof p.message === \"string\" ? p.message : \"Widget error\",\n            ),\n          );\n          return;\n        }\n        default:\n          return;\n      }\n    } catch (err) {\n      const error = err instanceof Error ? err : new Error(String(err));\n      reportError(error);\n    }\n  };\n\n  // The host applies the cross-origin guard before delegating; this only\n  // validates the JSON-RPC envelope.\n  const onMessage = (event: MessageEvent) => {\n    if (!isJsonRpcMessage(event.data)) return;\n\n    const msg = event.data;\n    if (isRequest(msg)) {\n      void handleRequest(msg);\n    } else if (isNotification(msg)) {\n      handleNotification(msg);\n    }\n  };\n\n  return {\n    onMessage,\n    dispose: () => {},\n    notifyToolInput: (input: unknown) => {\n      post({\n        jsonrpc: \"2.0\",\n        method: \"notifications/tools/call/input\",\n        params: { input },\n      });\n      post({\n        jsonrpc: \"2.0\",\n        method: \"ui/notifications/tool-input\",\n        params: isRecord(input) ? { arguments: input } : {},\n      });\n    },\n    notifyToolResult: (result: unknown) => {\n      post({\n        jsonrpc: \"2.0\",\n        method: \"notifications/tools/call/result\",\n        params: { result },\n      });\n      post({\n        jsonrpc: \"2.0\",\n        method: \"ui/notifications/tool-result\",\n        params: isRecord(result)\n          ? result\n          : { content: [{ type: \"text\", text: String(result) }] },\n      });\n    },\n    notifyHostContextChanged: (ctx: McpAppHostContext) => {\n      post({\n        jsonrpc: \"2.0\",\n        method: \"notifications/host_context/changed\",\n        params: ctx,\n      });\n      post({\n        jsonrpc: \"2.0\",\n        method: \"ui/notifications/host-context-changed\",\n        params: ctx,\n      });\n    },\n  };\n}\n"],"mappings":";;;;AAeA,MAAMY,sBAAsB;CAC1B;CACA;CACA;AAAK;AAoBP,MAAMkB,oBAAoC;CACxCC,MAAM;CACNC,SAAS;AACX;AAKA,MAAMC,iBAAyC;CAC7C,gCAAgC;CAChC,iCAAiC;CACjC,2BAA2B;CAC3B,gBAAgB;CAChB,2BAA2B;CAC3B,cAAc;CACd,yBAAyB;AAC3B;AAEA,MAAME,mBAAmBC,WACvBH,eAAeG,WAAWA;AAE5B,MAAMC,gBAAgB;CACpBC,YAAY;CACZC,gBAAgB;CAChBC,gBAAgB;CAChBC,eAAe;CACfC,eAAe;AACjB;AAEA,SAASC,iBAAiBC,OAA+C;CACvE,IAAI,CAACA,SAAS,OAAOA,UAAU,UAAU,OAAO;CAChD,MAAMC,IAAID;CACV,OAAOC,EAAEC,YAAY,SAAS,OAAOD,EAAET,WAAW;AACpD;AAEA,SAASW,UAAUC,KAAwD;CACzE,OAAO,QAAQA;AACjB;AAEA,SAASC,eACPD,KACkC;CAClC,OAAO,EAAE,QAAQA;AACnB;AAEA,SAAgBE,mBACdC,MACc;CACd,MAAM,EACJnC,OACAC,WAAW,CAAC,GACZC,WAAWY,mBACXX,cAAc,CAAC,MACbgC;CAEJ,MAAMC,QAAQJ,QAA8B;EAC1ChC,MAAMqC,YAAYL,GAAG;CACvB;CAEA,MAAMM,WACJC,IACAC,YAGG;EACH,MAAMK,MAA6B;GACjCf,SAAS;GACTS;GACA,GAAGC;EACL;EACAJ,KAAKS,GAAG;CACV;CAEA,MAAMC,iBACJP,IACAG,MACAC,SACAC,SACG;EACHN,QAAQC,IAAI,EACVE,OAAO;GACLC;GACAC;GACA,GAAIC,SAASG,KAAAA,IAAY,EAAEH,KAAK,IAAI,CAAC;EACvC,EACF,CAAC;CACH;CAEA,MAAMI,eAAeP,UAAiB;EACpCxD,2BAA2BgB,SAASiD,UAAUT,KAAK,GAAG,iBAAiB;CACzE;CAEA,MAAMU,gBAAgB,OAAOC,QAA8B;EACzD,IAAI;GACF,MAAMC,SAASD,IAAIC;GAEnB,QAAQlC,gBAAgBiC,IAAIhC,MAAM,GAAlC;IACE,KAAK,iBAAiB;KACpB,MAAMkC,2BACJ3D,SAAS0D,MAAM,KAAK,OAAOA,OAAOE,oBAAoB,WAClDF,OAAOE,kBAAAA;KAEbjB,QAAQc,IAAIb,IAAI,EACd3B,QAAQ;MACN2C,iBAAiBD;MACjBE,MAAMtD;MACNA;MACAC;MACAsD,cAAc;OACZC,OAAOzD,SAAS0D,WAAW,CAAC,IAAIZ,KAAAA;OAChCa,WACE3D,SAAS4D,gBAAgB5D,SAAS6D,gBAC9B,CAAC,IACDf,KAAAA;OACNgB,IAAI;QACF1B,aAAa,CAAC,CAACpC,SAASoC;QACxB2B,UAAU,CAAC,CAAC/D,SAAS+D;QACrBC,oBAAoB,CAAC,CAAChE,SAASgE;QAC/BC,oBAAoB,CAAC,CAACjE,SAASiE;OACjC;MACF;MACAC,kBAAkB;OAChB,GAAIlE,SAAS+D,WAAW,EAAEI,WAAW,CAAC,EAAE,IAAI,CAAC;OAC7C,GAAInE,SAAS0D,WAAW,EAAEU,aAAa,CAAC,EAAE,IAAI,CAAC;OAC/C,GAAIpE,SAAS4D,gBAAgB5D,SAAS6D,gBAClC,EAAEQ,iBAAiB,CAAC,EAAE,IACtB,CAAC;OACL,GAAIrE,SAASiE,qBACT,EAAEA,oBAAoB,EAAEK,MAAM,CAAC,EAAE,EAAE,IACnC,CAAC;OACL,GAAItE,SAASoC,cAAc,EAAEM,SAAS,EAAE4B,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC;OACxD,GAAItE,SAASuE,QAAQ,EAAEC,SAAS,CAAC,EAAE,IAAI,CAAC;MAC1C;KACF,EACF,CAAC;KACD;IACF;IAEA,KAAK,cAAc;KACjB,IAAI,CAACxE,SAAS0D,UAAU;MACtBb,cACEM,IAAIb,IACJlB,cAAcG,gBACd,0CACF;MACA;KACF;KACA,MAAMkD,aAAcrB,UAAU,CAAC;KAI/B,IAAI,OAAOqB,WAAW3D,SAAS,UAAU;MACvC+B,cACEM,IAAIb,IACJlB,cAAcI,eACd,qCACF;MACA;KACF;KACA,IACExB,SAAS2E,gBACT,CAAC3E,SAAS2E,aAAaC,SAASH,WAAW3D,IAAI,GAC/C;MACA+B,cACEM,IAAIb,IACJlB,cAAcI,eACd,SAASiD,WAAW3D,KAAI,8BAC1B;MACA;KACF;KACA,IAAI+D;KACJ,IAAIJ,WAAWC,cAAc5B,KAAAA,GAAW;MACtC,IACE2B,WAAWC,cAAc,QACzB,OAAOD,WAAWC,cAAc,YAChCI,MAAMC,QAAQN,WAAWC,SAAS,GAClC;OACA7B,cACEM,IAAIb,IACJlB,cAAcI,eACd,0CACF;OACA;MACF;MACAqD,WAAWJ,WAAWC;KACxB;KACA,MAAM/D,SAAS,MAAMX,SAAS0D,SAAS;MACrC5C,MAAM2D,WAAW3D;MACjB,GAAI+D,aAAa/B,KAAAA,IAAY,EAAE4B,WAAWG,SAAS,IAAI,CAAC;KAC1D,CAAC;KACDxC,QAAQc,IAAIb,IAAI,EAAE3B,OAAO,CAAC;KAC1B;IACF;IAEA,KAAK,kBAAkB;KACrB,IAAI,CAACX,SAAS4D,cAAc;MAC1Bf,cACEM,IAAIb,IACJlB,cAAcG,gBACd,8CACF;MACA;KACF;KACA,MAAMyD,aAAc5B,UAAU,CAAC;KAC/B,IAAI,OAAO4B,WAAWC,QAAQ,UAAU;MACtCpC,cACEM,IAAIb,IACJlB,cAAcI,eACd,wCACF;MACA;KACF;KACAa,QAAQc,IAAIb,IAAI,EACd3B,QAAQ,MAAMX,SAAS4D,aAAa,EAAEqB,KAAKD,WAAWC,IAAI,CAAC,EAC7D,CAAC;KACD;IACF;IAEA,KAAK;KACH,IAAI,CAACjF,SAAS6D,eAAe;MAC3BhB,cACEM,IAAIb,IACJlB,cAAcG,gBACd,8CACF;MACA;KACF;KACAc,QAAQc,IAAIb,IAAI,EACd3B,QAAS,MAAMX,SAAS6D,cAAcT,MAAM,KAAM,KACpD,CAAC;KACD;IAGF,KAAK,YAAY;KACf,IAAI,CAACpD,SAAS+D,UAAU;MACtBlB,cACEM,IAAIb,IACJlB,cAAcG,gBACd,wCACF;MACA;KACF;KACA,MAAM2D,aAAc9B,UAAU,CAAC;KAC/B,IAAI,OAAO8B,WAAWC,QAAQ,UAAU;MACtCtC,cACEM,IAAIb,IACJlB,cAAcI,eACd,kCACF;MACA;KACF;KACA,IAAI4D;KACJ,IAAI;MACFA,eAAe,IAAIC,IAAIH,WAAWC,GAAG,CAAC,CAACG;KACzC,QAAQ;MACNzC,cACEM,IAAIb,IACJlB,cAAcI,eACd,+BACF;MACA;KACF;KACA,IAAI4D,iBAAiB,YAAYA,iBAAiB,SAAS;MACzDvC,cACEM,IAAIb,IACJlB,cAAcI,eACd,oCACF;MACA;KACF;KACAa,QAAQc,IAAIb,IAAI,EACd3B,QAAQ,MAAMX,SAAS+D,SAAS,EAAEoB,KAAKD,WAAWC,IAAI,CAAC,EACzD,CAAC;KACD;IACF;IAEA,KAAK;KACH,IAAI,CAACnF,SAASoC,aAAa;MACzBS,cACEM,IAAIb,IACJlB,cAAcG,gBACd,2CACF;MACA;KACF;KACAc,QAAQc,IAAIb,IAAI,EACd3B,QAAS,MAAMX,SAASoC,YAAYgB,MAAM,KAAM,KAClD,CAAC;KACD;IAGF,KAAK;KACH,IAAI,CAACpD,SAASiE,oBAAoB;MAChCpB,cACEM,IAAIb,IACJlB,cAAcG,gBACd,kDACF;MACA;KACF;KACAc,QAAQc,IAAIb,IAAI,EACd3B,QAAS,MAAMX,SAASiE,mBAAmBb,MAAM,KAAM,KACzD,CAAC;KACD;IAGF,KAAK,sBAAsB;KACzB,IAAI,CAACpD,SAASgE,oBAAoB;MAChCnB,cACEM,IAAIb,IACJlB,cAAcG,gBACd,kDACF;MACA;KACF;KACA,MAAMgE,aAAcnC,UAAU,CAAC;KAC/B,IACE,OAAOmC,WAAWC,SAAS,YAC3B,CAAC7F,oBAAoBiF,SAASW,WAAWC,IAAyB,GAClE;MACA3C,cACEM,IAAIb,IACJlB,cAAcI,eACd,4CACF;MACA;KACF;KACAa,QAAQc,IAAIb,IAAI,EACd3B,QAAQ,MAAMX,SAASgE,mBAAmB,EACxCwB,MAAMD,WAAWC,KACnB,CAAC,EACH,CAAC;KACD;IACF;IAEA,SACE3C,cACEM,IAAIb,IACJlB,cAAcG,gBACd,mBAAmB4B,IAAIhC,QACzB;GAEJ;EACF,SAASsE,KAAK;GACZ,MAAMjD,QAAQiD,eAAezC,QAAQyC,MAAM,IAAIzC,MAAM0C,OAAOD,GAAG,CAAC;GAChE1C,YAAYP,KAAK;GACjBK,cAAcM,IAAIb,IAAIlB,cAAcK,eAAee,MAAME,OAAO;EAClE;CACF;CAEA,MAAMiD,sBAAsBC,SAAoC;EAC9D,IAAI;GACF,QAAQ1E,gBAAgB0E,KAAKzE,MAAM,GAAnC;IACE,KAAK;KACHnB,SAAS6F,gBAAgB;KACzB;IAEF,KAAK,8BAA8B;KACjC,MAAMC,IAAKF,KAAKxC,UAAU,CAAC;KAC3BpD,SAASiG,eAAe;MACtB,GAAI,OAAOH,EAAEC,UAAU,WAAW,EAAEA,OAAOD,EAAEC,MAAM,IAAI,CAAC;MACxD,GAAI,OAAOD,EAAEE,WAAW,WAAW,EAAEA,QAAQF,EAAEE,OAAO,IAAI,CAAC;KAC7D,CAAC;KACD;IACF;IACA,KAAK;KACHhG,SAASuE,QAAQqB,KAAKxC,MAAM;KAC5B;IAEF,KAAK;KACHpD,SAASkG,oBAAoBN,KAAKxC,MAAM;KACxC;IAEF,KAAK,uBAAuB;KAC1B,MAAM0C,IAAKF,KAAKxC,UAAU,CAAC;KAC3BL,YACE,IAAIC,MACF,OAAO8C,EAAEpD,YAAY,WAAWoD,EAAEpD,UAAU,cAC9C,CACF;KACA;IACF;IACA,SACE;GACJ;EACF,SAAS+C,KAAK;GACZ,MAAMjD,QAAQiD,eAAezC,QAAQyC,MAAM,IAAIzC,MAAM0C,OAAOD,GAAG,CAAC;GAChE1C,YAAYP,KAAK;EACnB;CACF;CAIA,MAAMpC,aAAaC,UAAwB;EACzC,IAAI,CAACqB,iBAAiBrB,MAAMsC,IAAI,GAAG;EAEnC,MAAMZ,MAAM1B,MAAMsC;EAClB,IAAIb,UAAUC,GAAG,GACf,cAAmBA,GAAG;OACjB,IAAIC,eAAeD,GAAG,GAC3B4D,mBAAmB5D,GAAG;CAE1B;CAEA,OAAO;EACL3B;EACAG,eAAe,CAAC;EAChBC,kBAAkBC,UAAmB;GACnC0B,KAAK;IACHN,SAAS;IACTV,QAAQ;IACRiC,QAAQ,EAAE3C,MAAM;GAClB,CAAC;GACD0B,KAAK;IACHN,SAAS;IACTV,QAAQ;IACRiC,QAAQ1D,SAASe,KAAK,IAAI,EAAEiE,WAAWjE,MAAM,IAAI,CAAC;GACpD,CAAC;EACH;EACAC,mBAAmBC,WAAoB;GACrCwB,KAAK;IACHN,SAAS;IACTV,QAAQ;IACRiC,QAAQ,EAAEzC,OAAO;GACnB,CAAC;GACDwB,KAAK;IACHN,SAAS;IACTV,QAAQ;IACRiC,QAAQ1D,SAASiB,MAAM,IACnBA,SACA,EAAEwF,SAAS,CAAC;KAAEC,MAAM;KAAQ9B,MAAMoB,OAAO/E,MAAM;IAAE,CAAC,EAAE;GAC1D,CAAC;EACH;EACAC,2BAA2ByF,QAA2B;GACpDlE,KAAK;IACHN,SAAS;IACTV,QAAQ;IACRiC,QAAQiD;GACV,CAAC;GACDlE,KAAK;IACHN,SAAS;IACTV,QAAQ;IACRiC,QAAQiD;GACV,CAAC;EACH;CACF;AACF"}

package/dist/model-context/makeAssistantVisible.js

@@ -1,6 +1,6 @@
 "use client";
-import { useAui } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui } from "@assistant-ui/store";
 import { tool } from "@assistant-ui/core";
 import { createContext, forwardRef, useContext, useEffect, useId, useRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";

package/dist/primitives/actionBar/ActionBarCopy.js

@@ -1,8 +1,8 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useActionBarCopy } from "@assistant-ui/core/react";
+import { useAuiState } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { composeEventHandlers } from "@radix-ui/primitive";

package/dist/primitives/actionBar/ActionBarExportMarkdown.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarExportMarkdown.d.ts","names":[],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"mappings":";;cAQM,+BAA8B,UAAA;EAIlC;EACA,aAAa,2BAA2B;aACpC;kBAgCW;OACH,UAAU;OACV,QAAQ,yBAAyB;;cAGlC,kDAAgC,0BAAA,KAAA,qBAAA,gBAAA,qCAAA,qBAAA;;;;;EAvChC;EACA,aAAE,2BAA2B;oBAsDxC,cAAA"}
+{"version":3,"file":"ActionBarExportMarkdown.d.ts","names":[],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"mappings":";;cAQM,+BAA8B,UAAA;EAIlC;EACA,aAAa,2BAA2B;aACpC;kBAgCW;OACH,UAAU;OACV,QAAQ,yBAAyB;;cAGlC,kDAAgC,0BAAA,KAAA,qBAAA,gBAAA,qCAAA,qBAAA;;;;;EAvChC;EACA,aAAE,2BAA2B;oBAwDxC,cAAA"}

package/dist/primitives/actionBar/ActionBarExportMarkdown.js

@@ -1,7 +1,7 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { useAui, useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui, useAuiState } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { composeEventHandlers } from "@radix-ui/primitive";
@@ -80,7 +80,7 @@ const ActionBarPrimitiveExportMarkdown = forwardRef((t0, forwardedRef) => {
 	let t3;
 	if ($[9] !== callback) {
 		t3 = () => {
-			callback?.();
+			callback?.().catch(_temp3);
 		};
 		$[9] = callback;
 		$[10] = t3;
@@ -116,6 +116,9 @@ function _temp(c) {
 function _temp2(s) {
 	return (s.message.role !== "assistant" || s.message.status?.type !== "running") && s.message.parts.some(_temp);
 }
+function _temp3(error) {
+	console.error("[assistant-ui] markdown export failed:", error);
+}
 //#endregion
 export { ActionBarPrimitiveExportMarkdown };
 

package/dist/primitives/actionBar/ActionBarExportMarkdown.js.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarExportMarkdown.js","names":["c","_c","forwardRef","useCallback","ActionButtonProps","composeEventHandlers","Primitive","useAuiState","useAui","useActionBarExportMarkdown","t0","$","t1","undefined","filename","onExport","aui","hasExportableContent","_temp2","t2","message","content","getCopyText","blob","Blob","type","url","URL","createObjectURL","a","document","createElement","href","download","Date","now","click","setTimeout","revokeObjectURL","callback","ActionBarPrimitiveExportMarkdown","Element","HTMLButtonElement","Props","forwardedRef","disabled","onClick","props","t3","t4","t5","displayName","_temp","text","length","s","role","status","parts","some"],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef, useCallback } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useActionBarExportMarkdown = ({\n  filename,\n  onExport,\n}: {\n  filename?: string | undefined;\n  onExport?: ((content: string) => void | Promise<void>) | undefined;\n} = {}) => {\n  const aui = useAui();\n  const hasExportableContent = useAuiState((s) => {\n    return (\n      (s.message.role !== \"assistant\" ||\n        s.message.status?.type !== \"running\") &&\n      s.message.parts.some((c) => c.type === \"text\" && c.text.length > 0)\n    );\n  });\n\n  const callback = useCallback(async () => {\n    const content = aui.message.getCopyText();\n    if (!content) return;\n\n    if (onExport) {\n      await onExport(content);\n      return;\n    }\n\n    const blob = new Blob([content], { type: \"text/markdown\" });\n    const url = URL.createObjectURL(blob);\n    const a = document.createElement(\"a\");\n    a.href = url;\n    a.download = filename ?? `message-${Date.now()}.md`;\n    a.click();\n    setTimeout(() => URL.revokeObjectURL(url), 40_000);\n  }, [aui, filename, onExport]);\n\n  if (!hasExportableContent) return null;\n  return callback;\n};\n\nexport namespace ActionBarPrimitiveExportMarkdown {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarExportMarkdown>;\n}\n\nexport const ActionBarPrimitiveExportMarkdown = forwardRef<\n  ActionBarPrimitiveExportMarkdown.Element,\n  ActionBarPrimitiveExportMarkdown.Props\n>(({ filename, onExport, onClick, disabled, ...props }, forwardedRef) => {\n  const callback = useActionBarExportMarkdown({ filename, onExport });\n  return (\n    <Primitive.button\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={disabled || !callback}\n      onClick={composeEventHandlers(onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveExportMarkdown.displayName =\n  \"ActionBarPrimitive.ExportMarkdown\";\n"],"mappings":";;;;;;;;AAQA,MAAMS,8BAA6BC,OAAA;CAAA,MAAAC,IAAAV,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAD,EAAA,OAAAD,IAAA;EAACE,KAAAF,OAAAG,KAAAA,IAAA,CAM/B,IAN+BH;EAM9BC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAN8B,MAAA,EAAAG,UAAAC,aAAAH;CAOlC,MAAAI,MAAYR,OAAO;CACnB,MAAAS,uBAA6BV,YAAYW,MAMxC;CAAE,IAAAC;CAAA,IAAAR,EAAA,OAAAK,IAAAI,WAAAT,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EAE0BI,KAAA,YAAA;GAC3B,MAAAE,UAAgBL,IAAGI,QAAQE,YAAa;GACxC,IAAI,CAACD,SAAO;GAEZ,IAAIN,UAAQ;IACV,MAAMA,SAASM,OAAO;IAAC;GAAA;GAIzB,MAAAE,OAAa,IAAIC,KAAK,CAACH,OAAO,GAAG,EAAAI,MAAQ,gBAAgB,CAAC;GAC1D,MAAAC,MAAYC,IAAGC,gBAAiBL,IAAI;GACpC,MAAAM,IAAUC,SAAQC,cAAe,GAAG;GACpCF,EAACG,OAAQN;GACTG,EAACI,WAAYnB,YAAA,WAAuBoB,KAAIC,IAAK,EAAC;GAC9CN,EAACO,MAAO;GACRC,iBAAiBV,IAAGW,gBAAiBZ,GAAG,GAAG,GAAM;EAAC;EACnDf,EAAA,KAAAK,IAAAI;EAAAT,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAhBD,MAAA4B,WAAiBpB;CAkBjB,IAAI,CAACF,sBAAoB,OAAS;CAAK,OAChCsB;AAAQ;AAQjB,MAAaC,mCAAmCtC,YAG9CQ,IAAAkC,iBAAA;CAAA,MAAAjC,IAAAV,EAAA,EAAA;CAAA,IAAA4C;CAAA,IAAA/B;CAAA,IAAAgC;CAAA,IAAA/B;CAAA,IAAAgC;CAAA,IAAApC,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAI,UAAAC,UAAA+B,SAAAD,aAAAE,SAAArC;EAAmDC,EAAA,KAAAD;EAAAC,EAAA,KAAAkC;EAAAlC,EAAA,KAAAG;EAAAH,EAAA,KAAAmC;EAAAnC,EAAA,KAAAI;EAAAJ,EAAA,KAAAoC;CAAA,OAAA;EAAAF,WAAAlC,EAAA;EAAAG,WAAAH,EAAA;EAAAmC,UAAAnC,EAAA;EAAAI,WAAAJ,EAAA;EAAAoC,QAAApC,EAAA;CAAA;CAAA,IAAAC;CAAA,IAAAD,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EACRH,KAAA;GAAAE;GAAAC;EAAqB;EAACJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAAlE,MAAA4B,WAAiB9B,2BAA2BG,EAAsB;CAMpD,MAAAO,KAAA0B,YAAA,CAAaN;CAAQ,IAAAS;CAAA,IAAArC,EAAA,OAAA4B,UAAA;EACQS,WAAA;GACrCT,WAAW;EAAC;EACb5B,EAAA,KAAA4B;EAAA5B,EAAA,MAAAqC;CAAA,OAAAA,KAAArC,EAAA;CAAA,IAAAsC;CAAA,IAAAtC,EAAA,QAAAmC,WAAAnC,EAAA,QAAAqC,IAAA;EAFQC,KAAA5C,qBAAqByC,SAASE,EAEtC;EAACrC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAqC;EAAArC,EAAA,MAAAsC;CAAA,OAAAA,KAAAtC,EAAA;CAAA,IAAAuC;CAAA,IAAAvC,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAoC,SAAApC,EAAA,QAAAQ,MAAAR,EAAA,QAAAsC,IAAA;EAPJC,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GAAQ,GACTH;GACCH,KAAAA;GACK,UAAAzB;GACD,SAAA8B;EAEP,CAAA;EACFtC,EAAA,MAAAiC;EAAAjC,EAAA,MAAAoC;EAAApC,EAAA,MAAAQ;EAAAR,EAAA,MAAAsC;EAAAtC,EAAA,MAAAuC;CAAA,OAAAA,KAAAvC,EAAA;CAAA,OARFuC;AAQE,CAEL;AAEDV,iCAAiCW,cAC/B;AA9DiC,SAAAC,MAAApD,GAAA;CAAA,OAYDA,EAACyB,SAAU,UAAUzB,EAACqD,KAAKC,SAAU;AAAC;AAZrC,SAAApC,OAAAqC,GAAA;CAAA,QAU5BA,EAACnC,QAAQoC,SAAU,eAClBD,EAACnC,QAAQqC,QAAahC,SAAK,cAC7B8B,EAACnC,QAAQsC,MAAMC,KAAMP,KAA6C;AAAC"}
+{"version":3,"file":"ActionBarExportMarkdown.js","names":["c","_c","forwardRef","useCallback","ActionButtonProps","composeEventHandlers","Primitive","useAuiState","useAui","useActionBarExportMarkdown","t0","$","t1","undefined","filename","onExport","aui","hasExportableContent","_temp2","t2","message","content","getCopyText","blob","Blob","type","url","URL","createObjectURL","a","document","createElement","href","download","Date","now","click","setTimeout","revokeObjectURL","callback","ActionBarPrimitiveExportMarkdown","Element","HTMLButtonElement","Props","forwardedRef","disabled","onClick","props","t3","catch","_temp3","t4","t5","displayName","_temp","text","length","s","role","status","parts","some","error","console"],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef, useCallback } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useActionBarExportMarkdown = ({\n  filename,\n  onExport,\n}: {\n  filename?: string | undefined;\n  onExport?: ((content: string) => void | Promise<void>) | undefined;\n} = {}) => {\n  const aui = useAui();\n  const hasExportableContent = useAuiState((s) => {\n    return (\n      (s.message.role !== \"assistant\" ||\n        s.message.status?.type !== \"running\") &&\n      s.message.parts.some((c) => c.type === \"text\" && c.text.length > 0)\n    );\n  });\n\n  const callback = useCallback(async () => {\n    const content = aui.message.getCopyText();\n    if (!content) return;\n\n    if (onExport) {\n      await onExport(content);\n      return;\n    }\n\n    const blob = new Blob([content], { type: \"text/markdown\" });\n    const url = URL.createObjectURL(blob);\n    const a = document.createElement(\"a\");\n    a.href = url;\n    a.download = filename ?? `message-${Date.now()}.md`;\n    a.click();\n    setTimeout(() => URL.revokeObjectURL(url), 40_000);\n  }, [aui, filename, onExport]);\n\n  if (!hasExportableContent) return null;\n  return callback;\n};\n\nexport namespace ActionBarPrimitiveExportMarkdown {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarExportMarkdown>;\n}\n\nexport const ActionBarPrimitiveExportMarkdown = forwardRef<\n  ActionBarPrimitiveExportMarkdown.Element,\n  ActionBarPrimitiveExportMarkdown.Props\n>(({ filename, onExport, onClick, disabled, ...props }, forwardedRef) => {\n  const callback = useActionBarExportMarkdown({ filename, onExport });\n  return (\n    <Primitive.button\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={disabled || !callback}\n      onClick={composeEventHandlers(onClick, () => {\n        void callback?.().catch((error: unknown) => {\n          console.error(\"[assistant-ui] markdown export failed:\", error);\n        });\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveExportMarkdown.displayName =\n  \"ActionBarPrimitive.ExportMarkdown\";\n"],"mappings":";;;;;;;;AAQA,MAAMS,8BAA6BC,OAAA;CAAA,MAAAC,IAAAV,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAD,EAAA,OAAAD,IAAA;EAACE,KAAAF,OAAAG,KAAAA,IAAA,CAM/B,IAN+BH;EAM9BC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAN8B,MAAA,EAAAG,UAAAC,aAAAH;CAOlC,MAAAI,MAAYR,OAAO;CACnB,MAAAS,uBAA6BV,YAAYW,MAMxC;CAAE,IAAAC;CAAA,IAAAR,EAAA,OAAAK,IAAAI,WAAAT,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EAE0BI,KAAA,YAAA;GAC3B,MAAAE,UAAgBL,IAAGI,QAAQE,YAAa;GACxC,IAAI,CAACD,SAAO;GAEZ,IAAIN,UAAQ;IACV,MAAMA,SAASM,OAAO;IAAC;GAAA;GAIzB,MAAAE,OAAa,IAAIC,KAAK,CAACH,OAAO,GAAG,EAAAI,MAAQ,gBAAgB,CAAC;GAC1D,MAAAC,MAAYC,IAAGC,gBAAiBL,IAAI;GACpC,MAAAM,IAAUC,SAAQC,cAAe,GAAG;GACpCF,EAACG,OAAQN;GACTG,EAACI,WAAYnB,YAAA,WAAuBoB,KAAIC,IAAK,EAAC;GAC9CN,EAACO,MAAO;GACRC,iBAAiBV,IAAGW,gBAAiBZ,GAAG,GAAG,GAAM;EAAC;EACnDf,EAAA,KAAAK,IAAAI;EAAAT,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAhBD,MAAA4B,WAAiBpB;CAkBjB,IAAI,CAACF,sBAAoB,OAAS;CAAK,OAChCsB;AAAQ;AAQjB,MAAaC,mCAAmCtC,YAG9CQ,IAAAkC,iBAAA;CAAA,MAAAjC,IAAAV,EAAA,EAAA;CAAA,IAAA4C;CAAA,IAAA/B;CAAA,IAAAgC;CAAA,IAAA/B;CAAA,IAAAgC;CAAA,IAAApC,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAI,UAAAC,UAAA+B,SAAAD,aAAAE,SAAArC;EAAmDC,EAAA,KAAAD;EAAAC,EAAA,KAAAkC;EAAAlC,EAAA,KAAAG;EAAAH,EAAA,KAAAmC;EAAAnC,EAAA,KAAAI;EAAAJ,EAAA,KAAAoC;CAAA,OAAA;EAAAF,WAAAlC,EAAA;EAAAG,WAAAH,EAAA;EAAAmC,UAAAnC,EAAA;EAAAI,WAAAJ,EAAA;EAAAoC,QAAApC,EAAA;CAAA;CAAA,IAAAC;CAAA,IAAAD,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EACRH,KAAA;GAAAE;GAAAC;EAAqB;EAACJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAAlE,MAAA4B,WAAiB9B,2BAA2BG,EAAsB;CAMpD,MAAAO,KAAA0B,YAAA,CAAaN;CAAQ,IAAAS;CAAA,IAAArC,EAAA,OAAA4B,UAAA;EACQS,WAAA;GAChCT,WAAiB,CAAC,CAAAU,MAACC,MAEvB;EAAC;EACHvC,EAAA,KAAA4B;EAAA5B,EAAA,MAAAqC;CAAA,OAAAA,KAAArC,EAAA;CAAA,IAAAwC;CAAA,IAAAxC,EAAA,QAAAmC,WAAAnC,EAAA,QAAAqC,IAAA;EAJQG,KAAA9C,qBAAqByC,SAASE,EAItC;EAACrC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAqC;EAAArC,EAAA,MAAAwC;CAAA,OAAAA,KAAAxC,EAAA;CAAA,IAAAyC;CAAA,IAAAzC,EAAA,QAAAiC,gBAAAjC,EAAA,QAAAoC,SAAApC,EAAA,QAAAQ,MAAAR,EAAA,QAAAwC,IAAA;EATJC,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GAAQ,GACTL;GACCH,KAAAA;GACK,UAAAzB;GACD,SAAAgC;EAIP,CAAA;EACFxC,EAAA,MAAAiC;EAAAjC,EAAA,MAAAoC;EAAApC,EAAA,MAAAQ;EAAAR,EAAA,MAAAwC;EAAAxC,EAAA,MAAAyC;CAAA,OAAAA,KAAAzC,EAAA;CAAA,OAVFyC;AAUE,CAEL;AAEDZ,iCAAiCa,cAC/B;AAhEiC,SAAAC,MAAAtD,GAAA;CAAA,OAYDA,EAACyB,SAAU,UAAUzB,EAACuD,KAAKC,SAAU;AAAC;AAZrC,SAAAtC,OAAAuC,GAAA;CAAA,QAU5BA,EAACrC,QAAQsC,SAAU,eAClBD,EAACrC,QAAQuC,QAAalC,SAAK,cAC7BgC,EAACrC,QAAQwC,MAAMC,KAAMP,KAA6C;AAAC;AAkCvE,SAAAJ,OAAAY,OAAA;CAUQC,QAAOD,MAAO,0CAA0CA,KAAK;AAAC"}

package/dist/primitives/actionBar/ActionBarFeedbackNegative.js

@@ -1,8 +1,8 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useActionBarFeedbackNegative } from "@assistant-ui/core/react";
+import { useAuiState } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { composeEventHandlers } from "@radix-ui/primitive";

package/dist/primitives/actionBar/ActionBarFeedbackPositive.js

@@ -1,8 +1,8 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useActionBarFeedbackPositive } from "@assistant-ui/core/react";
+import { useAuiState } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { composeEventHandlers } from "@radix-ui/primitive";

package/dist/primitives/actionBar/ActionBarStopSpeaking.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarStopSpeaking.d.ts","names":[],"sources":["../../../src/primitives/actionBar/ActionBarStopSpeaking.tsx"],"mappings":";;cASM;kBAMW;OACH,UAAU;OACV,QAAQ,yBAAyB;;cAGlC,gDAA8B,0BAAA,KAAA,qBAAA,gBAAA,qCAAA,qBAAA;;;;oBAyBzC,cAAA,6CAAA,cAAA"}
+{"version":3,"file":"ActionBarStopSpeaking.d.ts","names":[],"sources":["../../../src/primitives/actionBar/ActionBarStopSpeaking.tsx"],"mappings":";;cAQM;kBAMW;OACH,UAAU;OACV,QAAQ,yBAAyB;;cAGlC,gDAA8B,0BAAA,KAAA,qBAAA,gBAAA,qCAAA,qBAAA;;;;oBAiBzC,cAAA,6CAAA,cAAA"}

package/dist/primitives/actionBar/ActionBarStopSpeaking.js

@@ -5,7 +5,6 @@ import { useActionBarStopSpeaking } from "@assistant-ui/core/react";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { composeEventHandlers } from "@radix-ui/primitive";
-import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
 //#region src/primitives/actionBar/ActionBarStopSpeaking.tsx
 const useActionBarStopSpeaking$1 = () => {
 	const { disabled, stopSpeaking } = useActionBarStopSpeaking();
@@ -13,52 +12,40 @@ const useActionBarStopSpeaking$1 = () => {
 	return stopSpeaking;
 };
 const ActionBarPrimitiveStopSpeaking = forwardRef((props, ref) => {
-	const $ = c(12);
+	const $ = c(10);
 	const callback = useActionBarStopSpeaking$1();
-	let t0;
+	const t0 = !callback;
+	let t1;
 	if ($[0] !== callback) {
-		t0 = (e) => {
-			if (callback) {
-				e.preventDefault();
-				callback();
-			}
+		t1 = () => {
+			callback?.();
 		};
 		$[0] = callback;
-		$[1] = t0;
-	} else t0 = $[1];
-	useEscapeKeydown(t0);
-	const t1 = !callback;
+		$[1] = t1;
+	} else t1 = $[1];
 	let t2;
-	if ($[2] !== callback) {
-		t2 = () => {
-			callback?.();
-		};
-		$[2] = callback;
-		$[3] = t2;
-	} else t2 = $[3];
+	if ($[2] !== props.onClick || $[3] !== t1) {
+		t2 = composeEventHandlers(props.onClick, t1);
+		$[2] = props.onClick;
+		$[3] = t1;
+		$[4] = t2;
+	} else t2 = $[4];
 	let t3;
-	if ($[4] !== props.onClick || $[5] !== t2) {
-		t3 = composeEventHandlers(props.onClick, t2);
-		$[4] = props.onClick;
-		$[5] = t2;
-		$[6] = t3;
-	} else t3 = $[6];
-	let t4;
-	if ($[7] !== props || $[8] !== ref || $[9] !== t1 || $[10] !== t3) {
-		t4 = /* @__PURE__ */ jsx(Primitive.button, {
+	if ($[5] !== props || $[6] !== ref || $[7] !== t0 || $[8] !== t2) {
+		t3 = /* @__PURE__ */ jsx(Primitive.button, {
 			type: "button",
-			disabled: t1,
+			disabled: t0,
 			...props,
 			ref,
-			onClick: t3
+			onClick: t2
 		});
-		$[7] = props;
-		$[8] = ref;
-		$[9] = t1;
-		$[10] = t3;
-		$[11] = t4;
-	} else t4 = $[11];
-	return t4;
+		$[5] = props;
+		$[6] = ref;
+		$[7] = t0;
+		$[8] = t2;
+		$[9] = t3;
+	} else t3 = $[9];
+	return t3;
 });
 ActionBarPrimitiveStopSpeaking.displayName = "ActionBarPrimitive.StopSpeaking";
 //#endregion

package/dist/primitives/actionBar/ActionBarStopSpeaking.js.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarStopSpeaking.js","names":["c","_c","forwardRef","ActionButtonProps","useEscapeKeydown","Primitive","composeEventHandlers","useActionBarStopSpeaking","useActionBarStopSpeakingBehavior","disabled","stopSpeaking","ActionBarPrimitiveStopSpeaking","Element","HTMLButtonElement","Props","props","ref","$","callback","t0","e","preventDefault","t1","t2","t3","onClick","t4","displayName"],"sources":["../../../src/primitives/actionBar/ActionBarStopSpeaking.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { useEscapeKeydown } from \"@radix-ui/react-use-escape-keydown\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { useActionBarStopSpeaking as useActionBarStopSpeakingBehavior } from \"@assistant-ui/core/react\";\n\nconst useActionBarStopSpeaking = () => {\n  const { disabled, stopSpeaking } = useActionBarStopSpeakingBehavior();\n  if (disabled) return null;\n  return stopSpeaking;\n};\n\nexport namespace ActionBarPrimitiveStopSpeaking {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarStopSpeaking>;\n}\n\nexport const ActionBarPrimitiveStopSpeaking = forwardRef<\n  ActionBarPrimitiveStopSpeaking.Element,\n  ActionBarPrimitiveStopSpeaking.Props\n>((props, ref) => {\n  const callback = useActionBarStopSpeaking();\n\n  // TODO this stops working if the user is not hovering over an older message\n  useEscapeKeydown((e) => {\n    if (callback) {\n      e.preventDefault();\n      callback();\n    }\n  });\n\n  return (\n    <Primitive.button\n      type=\"button\"\n      disabled={!callback}\n      {...props}\n      ref={ref}\n      onClick={composeEventHandlers(props.onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveStopSpeaking.displayName = \"ActionBarPrimitive.StopSpeaking\";\n"],"mappings":";;;;;;;;;AASA,MAAMO,mCAA2B;CAC/B,MAAA,EAAAE,UAAAC,iBAAmCF,yBAAiC;CACpE,IAAIC,UAAQ,OAAS;CAAK,OACnBC;AAAY;AAQrB,MAAaC,iCAAiCT,YAG5Ca,OAAAC,QAAA;CAAA,MAAAC,IAAAhB,EAAA,EAAA;CACA,MAAAiB,WAAiBX,2BAAyB;CAAE,IAAAY;CAAA,IAAAF,EAAA,OAAAC,UAAA;EAG3BC,MAAAC,MAAA;GACf,IAAIF,UAAQ;IACVE,EAACC,eAAgB;IACjBH,SAAS;GAAC;EACX;EACFD,EAAA,KAAAC;EAAAD,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CALDb,iBAAiBe,EAKhB;CAKa,MAAAG,KAAA,CAACJ;CAAQ,IAAAK;CAAA,IAAAN,EAAA,OAAAC,UAAA;EAG0BK,WAAA;GAC3CL,WAAW;EAAC;EACbD,EAAA,KAAAC;EAAAD,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,IAAAO;CAAA,IAAAP,EAAA,OAAAF,MAAAU,WAAAR,EAAA,OAAAM,IAAA;EAFQC,KAAAlB,qBAAqBS,MAAKU,SAAUF,EAE5C;EAACN,EAAA,KAAAF,MAAAU;EAAAR,EAAA,KAAAM;EAAAN,EAAA,KAAAO;CAAA,OAAAA,KAAAP,EAAA;CAAA,IAAAS;CAAA,IAAAT,EAAA,OAAAF,SAAAE,EAAA,OAAAD,OAAAC,EAAA,OAAAK,MAAAL,EAAA,QAAAO,IAAA;EAPJE,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GACK,UAAAJ;GAAS,GACfP;GACCC;GACI,SAAAQ;EAEP,CAAA;EACFP,EAAA,KAAAF;EAAAE,EAAA,KAAAD;EAAAC,EAAA,KAAAK;EAAAL,EAAA,MAAAO;EAAAP,EAAA,MAAAS;CAAA,OAAAA,KAAAT,EAAA;CAAA,OARFS;AAQE,CAEL;AAEDf,+BAA+BgB,cAAc"}
+{"version":3,"file":"ActionBarStopSpeaking.js","names":["c","_c","forwardRef","ActionButtonProps","Primitive","composeEventHandlers","useActionBarStopSpeaking","useActionBarStopSpeakingBehavior","disabled","stopSpeaking","ActionBarPrimitiveStopSpeaking","Element","HTMLButtonElement","Props","props","ref","$","callback","t0","t1","t2","onClick","t3","displayName"],"sources":["../../../src/primitives/actionBar/ActionBarStopSpeaking.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { useActionBarStopSpeaking as useActionBarStopSpeakingBehavior } from \"@assistant-ui/core/react\";\n\nconst useActionBarStopSpeaking = () => {\n  const { disabled, stopSpeaking } = useActionBarStopSpeakingBehavior();\n  if (disabled) return null;\n  return stopSpeaking;\n};\n\nexport namespace ActionBarPrimitiveStopSpeaking {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarStopSpeaking>;\n}\n\nexport const ActionBarPrimitiveStopSpeaking = forwardRef<\n  ActionBarPrimitiveStopSpeaking.Element,\n  ActionBarPrimitiveStopSpeaking.Props\n>((props, ref) => {\n  const callback = useActionBarStopSpeaking();\n\n  return (\n    <Primitive.button\n      type=\"button\"\n      disabled={!callback}\n      {...props}\n      ref={ref}\n      onClick={composeEventHandlers(props.onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveStopSpeaking.displayName = \"ActionBarPrimitive.StopSpeaking\";\n"],"mappings":";;;;;;;;AAQA,MAAMM,mCAA2B;CAC/B,MAAA,EAAAE,UAAAC,iBAAmCF,yBAAiC;CACpE,IAAIC,UAAQ,OAAS;CAAK,OACnBC;AAAY;AAQrB,MAAaC,iCAAiCR,YAG5CY,OAAAC,QAAA;CAAA,MAAAC,IAAAf,EAAA,EAAA;CACA,MAAAgB,WAAiBX,2BAAyB;CAK5B,MAAAY,KAAA,CAACD;CAAQ,IAAAE;CAAA,IAAAH,EAAA,OAAAC,UAAA;EAG0BE,WAAA;GAC3CF,WAAW;EAAC;EACbD,EAAA,KAAAC;EAAAD,EAAA,KAAAG;CAAA,OAAAA,KAAAH,EAAA;CAAA,IAAAI;CAAA,IAAAJ,EAAA,OAAAF,MAAAO,WAAAL,EAAA,OAAAG,IAAA;EAFQC,KAAAf,qBAAqBS,MAAKO,SAAUF,EAE5C;EAACH,EAAA,KAAAF,MAAAO;EAAAL,EAAA,KAAAG;EAAAH,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAM;CAAA,IAAAN,EAAA,OAAAF,SAAAE,EAAA,OAAAD,OAAAC,EAAA,OAAAE,MAAAF,EAAA,OAAAI,IAAA;EAPJE,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GACK,UAAAJ;GAAS,GACfJ;GACCC;GACI,SAAAK;EAEP,CAAA;EACFJ,EAAA,KAAAF;EAAAE,EAAA,KAAAD;EAAAC,EAAA,KAAAE;EAAAF,EAAA,KAAAI;EAAAJ,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OARFM;AAQE,CAEL;AAEDZ,+BAA+Ba,cAAc"}

package/dist/primitives/actionBar/useActionBarFloatStatus.js

@@ -1,6 +1,6 @@
 "use client";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAuiState } from "@assistant-ui/store";
 //#region src/primitives/actionBar/useActionBarFloatStatus.ts
 const HideAndFloatStatus = {
 	Hidden: "hidden",

package/dist/primitives/assistantModal/AssistantModalRoot.js

@@ -1,7 +1,7 @@
 "use client";
 import { usePopoverScope } from "./scope.js";
-import { useAui } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui } from "@assistant-ui/store";
 import { useEffect, useState } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { Popover } from "radix-ui";

package/dist/primitives/attachment/AttachmentName.js

@@ -1,6 +1,6 @@
 "use client";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAuiState } from "@assistant-ui/store";
 import { Fragment, jsx } from "react/jsx-runtime";
 //#region src/primitives/attachment/AttachmentName.tsx
 const AttachmentPrimitiveName = () => {

package/dist/primitives/attachment/AttachmentRemove.js

@@ -1,7 +1,7 @@
 "use client";
 import { createActionButton } from "../../utils/createActionButton.js";
-import { useAui } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui } from "@assistant-ui/store";
 import "@assistant-ui/tap/react-shim";
 //#region src/primitives/attachment/AttachmentRemove.ts
 const useAttachmentRemove = () => {

package/dist/primitives/attachment/AttachmentThumb.js

@@ -1,7 +1,7 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAuiState } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 //#region src/primitives/attachment/AttachmentThumb.tsx

package/dist/primitives/branchPicker/BranchPickerCount.js

@@ -1,6 +1,6 @@
 "use client";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAuiState } from "@assistant-ui/store";
 import { Fragment, jsx } from "react/jsx-runtime";
 //#region src/primitives/branchPicker/BranchPickerCount.tsx
 const useBranchPickerCount = () => {

package/dist/primitives/branchPicker/BranchPickerNumber.js

@@ -1,6 +1,6 @@
 "use client";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAuiState } from "@assistant-ui/store";
 import { Fragment, jsx } from "react/jsx-runtime";
 //#region src/primitives/branchPicker/BranchPickerNumber.tsx
 const useBranchPickerNumber = () => {

package/dist/primitives/branchPicker/BranchPickerRoot.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BranchPickerRoot.d.ts","names":[],"sources":["../../../src/primitives/branchPicker/BranchPickerRoot.tsx"],"mappings":";;;kBAUiB;OACH,UAAU,oBAAoB,UAAU;OACxC,QAAQ,gCAAgC,UAAU;;;;;;IAM5D;;;;;;;;;;;;;;;;;;;;cAqBS,2CAAyB,0BAAA,KAAA,qBAAA,gBAAA,kCAAA,eAAA;;;;;;;;;;EArBX;oBA8BzB,cAAA"}
+{"version":3,"file":"BranchPickerRoot.d.ts","names":[],"sources":["../../../src/primitives/branchPicker/BranchPickerRoot.tsx"],"mappings":";;;kBAUiB;OACH,UAAU,oBAAoB,UAAU;OACxC,QAAQ,gCAAgC,UAAU;;;;;;IAM5D;;;;;;;;;;;;;;;;;;;;cAqBS,2CAAyB,0BAAA,KAAA,qBAAA,gBAAA,kCAAA,eAAA;;;;;;;;;;EArBX;oBAgCzB,cAAA"}

package/dist/primitives/branchPicker/BranchPickerRoot.js

@@ -1,7 +1,7 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { MessagePrimitiveIf } from "../message/MessageIf.js";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { AuiIf } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 //#region src/primitives/branchPicker/BranchPickerRoot.tsx
@@ -23,7 +23,7 @@ import { jsx } from "react/jsx-runtime";
 * ```
 */
 const BranchPickerPrimitiveRoot = forwardRef((t0, ref) => {
-	const $ = c(9);
+	const $ = c(11);
 	let hideWhenSingleBranch;
 	let rest;
 	if ($[0] !== t0) {
@@ -35,27 +35,32 @@ const BranchPickerPrimitiveRoot = forwardRef((t0, ref) => {
 		hideWhenSingleBranch = $[1];
 		rest = $[2];
 	}
-	const t1 = hideWhenSingleBranch ? true : void 0;
+	let t1;
+	if ($[3] !== hideWhenSingleBranch) {
+		t1 = (s) => !hideWhenSingleBranch || s.message.branchCount >= 2;
+		$[3] = hideWhenSingleBranch;
+		$[4] = t1;
+	} else t1 = $[4];
 	let t2;
-	if ($[3] !== ref || $[4] !== rest) {
+	if ($[5] !== ref || $[6] !== rest) {
 		t2 = /* @__PURE__ */ jsx(Primitive.div, {
 			...rest,
 			ref
 		});
-		$[3] = ref;
-		$[4] = rest;
-		$[5] = t2;
-	} else t2 = $[5];
+		$[5] = ref;
+		$[6] = rest;
+		$[7] = t2;
+	} else t2 = $[7];
 	let t3;
-	if ($[6] !== t1 || $[7] !== t2) {
-		t3 = /* @__PURE__ */ jsx(MessagePrimitiveIf, {
-			hasBranches: t1,
+	if ($[8] !== t1 || $[9] !== t2) {
+		t3 = /* @__PURE__ */ jsx(AuiIf, {
+			condition: t1,
 			children: t2
 		});
-		$[6] = t1;
-		$[7] = t2;
-		$[8] = t3;
-	} else t3 = $[8];
+		$[8] = t1;
+		$[9] = t2;
+		$[10] = t3;
+	} else t3 = $[10];
 	return t3;
 });
 BranchPickerPrimitiveRoot.displayName = "BranchPickerPrimitive.Root";

package/dist/primitives/branchPicker/BranchPickerRoot.js.map

@@ -1 +1 @@
-{"version":3,"file":"BranchPickerRoot.js","names":["c","_c","Primitive","ComponentRef","forwardRef","ComponentPropsWithoutRef","MessagePrimitiveIf","If","BranchPickerPrimitiveRoot","Element","div","Props","hideWhenSingleBranch","t0","ref","$","rest","t1","undefined","t2","t3","displayName"],"sources":["../../../src/primitives/branchPicker/BranchPickerRoot.tsx"],"sourcesContent":["\"use client\";\n\nimport { Primitive } from \"../../utils/Primitive\";\nimport {\n  type ComponentRef,\n  forwardRef,\n  type ComponentPropsWithoutRef,\n} from \"react\";\nimport { MessagePrimitiveIf as If } from \"../message/MessageIf\";\n\nexport namespace BranchPickerPrimitiveRoot {\n  export type Element = ComponentRef<typeof Primitive.div>;\n  export type Props = ComponentPropsWithoutRef<typeof Primitive.div> & {\n    /**\n     * Whether to hide the branch picker when there's only one branch available.\n     * When true, the component will only render when multiple branches exist.\n     * @default false\n     */\n    hideWhenSingleBranch?: boolean | undefined;\n  };\n}\n\n/**\n * The root container for branch picker components.\n *\n * This component provides a container for branch navigation controls,\n * with optional conditional rendering based on the number of available branches.\n * It integrates with the message branching system to allow users to navigate\n * between different response variations.\n *\n * @example\n * ```tsx\n * <BranchPickerPrimitive.Root hideWhenSingleBranch={true}>\n *   <BranchPickerPrimitive.Previous />\n *   <BranchPickerPrimitive.Count />\n *   <BranchPickerPrimitive.Next />\n * </BranchPickerPrimitive.Root>\n * ```\n */\nexport const BranchPickerPrimitiveRoot = forwardRef<\n  BranchPickerPrimitiveRoot.Element,\n  BranchPickerPrimitiveRoot.Props\n>(({ hideWhenSingleBranch, ...rest }, ref) => {\n  return (\n    <If hasBranches={hideWhenSingleBranch ? true : undefined}>\n      <Primitive.div {...rest} ref={ref} />\n    </If>\n  );\n});\n\nBranchPickerPrimitiveRoot.displayName = \"BranchPickerPrimitive.Root\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAuCA,MAAaQ,4BAA4BJ,YAGvCS,IAAAC,QAAA;CAAA,MAAAC,IAAAd,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAI;CAAA,IAAAD,EAAA,OAAAF,IAAA;EAAC,CAAA,CAAAD,yBAAAI,QAAAH;EAAiCE,EAAA,KAAAF;EAAAE,EAAA,KAAAH;EAAAG,EAAA,KAAAC;CAAA,OAAA;EAAAJ,uBAAAG,EAAA;EAAAC,OAAAD,EAAA;CAAA;CAEf,MAAAE,KAAAL,uBAAA,OAAAM,KAAAA;CAAuC,IAAAC;CAAA,IAAAJ,EAAA,OAAAD,OAAAC,EAAA,OAAAC,MAAA;EACtDG,KAAA,oBAAA,UAAA,KAAA;GAAA,GAAmBH;GAAWF;EAAG,CAAA;EAAIC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;EAAAD,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAA,IAAAK;CAAA,IAAAL,EAAA,OAAAE,MAAAF,EAAA,OAAAI,IAAA;EADvCC,KAAA,oBAAC,oBAAD;GAAiB,aAAAH;GACfE,UAAAA;EADC,CAAA;EAEEJ,EAAA,KAAAE;EAAAF,EAAA,KAAAI;EAAAJ,EAAA,KAAAK;CAAA,OAAAA,KAAAL,EAAA;CAAA,OAFLK;AAEK,CAER;AAEDZ,0BAA0Ba,cAAc"}
+{"version":3,"file":"BranchPickerRoot.js","names":["c","_c","Primitive","ComponentRef","forwardRef","ComponentPropsWithoutRef","AuiIf","BranchPickerPrimitiveRoot","Element","div","Props","hideWhenSingleBranch","t0","ref","$","rest","t1","s","message","branchCount","t2","t3","displayName"],"sources":["../../../src/primitives/branchPicker/BranchPickerRoot.tsx"],"sourcesContent":["\"use client\";\n\nimport { Primitive } from \"../../utils/Primitive\";\nimport {\n  type ComponentRef,\n  forwardRef,\n  type ComponentPropsWithoutRef,\n} from \"react\";\nimport { AuiIf } from \"@assistant-ui/store\";\n\nexport namespace BranchPickerPrimitiveRoot {\n  export type Element = ComponentRef<typeof Primitive.div>;\n  export type Props = ComponentPropsWithoutRef<typeof Primitive.div> & {\n    /**\n     * Whether to hide the branch picker when there's only one branch available.\n     * When true, the component will only render when multiple branches exist.\n     * @default false\n     */\n    hideWhenSingleBranch?: boolean | undefined;\n  };\n}\n\n/**\n * The root container for branch picker components.\n *\n * This component provides a container for branch navigation controls,\n * with optional conditional rendering based on the number of available branches.\n * It integrates with the message branching system to allow users to navigate\n * between different response variations.\n *\n * @example\n * ```tsx\n * <BranchPickerPrimitive.Root hideWhenSingleBranch={true}>\n *   <BranchPickerPrimitive.Previous />\n *   <BranchPickerPrimitive.Count />\n *   <BranchPickerPrimitive.Next />\n * </BranchPickerPrimitive.Root>\n * ```\n */\nexport const BranchPickerPrimitiveRoot = forwardRef<\n  BranchPickerPrimitiveRoot.Element,\n  BranchPickerPrimitiveRoot.Props\n>(({ hideWhenSingleBranch, ...rest }, ref) => {\n  return (\n    <AuiIf\n      condition={(s) => !hideWhenSingleBranch || s.message.branchCount >= 2}\n    >\n      <Primitive.div {...rest} ref={ref} />\n    </AuiIf>\n  );\n});\n\nBranchPickerPrimitiveRoot.displayName = \"BranchPickerPrimitive.Root\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAuCA,MAAaO,4BAA4BH,YAGvCQ,IAAAC,QAAA;CAAA,MAAAC,IAAAb,EAAA,EAAA;CAAA,IAAAU;CAAA,IAAAI;CAAA,IAAAD,EAAA,OAAAF,IAAA;EAAC,CAAA,CAAAD,yBAAAI,QAAAH;EAAiCE,EAAA,KAAAF;EAAAE,EAAA,KAAAH;EAAAG,EAAA,KAAAC;CAAA,OAAA;EAAAJ,uBAAAG,EAAA;EAAAC,OAAAD,EAAA;CAAA;CAAA,IAAAE;CAAA,IAAAF,EAAA,OAAAH,sBAAA;EAGnBK,MAAAC,MAAO,CAACN,wBAAwBM,EAACC,QAAQC,eAAgB;EAACL,EAAA,KAAAH;EAAAG,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,IAAAM;CAAA,IAAAN,EAAA,OAAAD,OAAAC,EAAA,OAAAC,MAAA;EAErEK,KAAA,oBAAA,UAAA,KAAA;GAAA,GAAmBL;GAAWF;EAAG,CAAA;EAAIC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;EAAAD,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,IAAAO;CAAA,IAAAP,EAAA,OAAAE,MAAAF,EAAA,OAAAM,IAAA;EAHvCC,KAAA,oBAAC,OAAD;GACa,WAAAL;GAEXI,UAAAA;EAHI,CAAA;EAIEN,EAAA,KAAAE;EAAAF,EAAA,KAAAM;EAAAN,EAAA,MAAAO;CAAA,OAAAA,KAAAP,EAAA;CAAA,OAJRO;AAIQ,CAEX;AAEDd,0BAA0Be,cAAc"}

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js

@@ -1,7 +1,7 @@
 "use client";
 import { createActionButton } from "../../utils/createActionButton.js";
-import { useAui, useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui, useAuiState } from "@assistant-ui/store";
 import "@assistant-ui/tap/react-shim";
 //#region src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.ts
 const useChainOfThoughtAccordionTrigger = () => {

package/dist/primitives/composer/ComposerAddAttachment.js

@@ -1,8 +1,8 @@
 "use client";
 import { createActionButton } from "../../utils/createActionButton.js";
-import { useAui } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
 import { useComposerAddAttachment } from "@assistant-ui/core/react";
+import { useAui } from "@assistant-ui/store";
 import "@assistant-ui/tap/react-shim";
 //#region src/primitives/composer/ComposerAddAttachment.ts
 const useComposerAddAttachment$1 = (t0) => {

package/dist/primitives/composer/ComposerAttachmentDropzone.js

@@ -1,6 +1,6 @@
 "use client";
-import { useAui } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAui } from "@assistant-ui/store";
 import { cloneElement, forwardRef, isValidElement, useState } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 import { composeEventHandlers } from "@radix-ui/primitive";

package/dist/primitives/composer/ComposerDictationTranscript.d.ts

@@ -13,11 +13,11 @@ declare namespace ComposerPrimitiveDictationTranscript {
  *
  * @example
  * ```tsx
- * <ComposerPrimitive.If dictation>
+ * <AuiIf condition={(s) => s.composer.dictation != null}>
  *   <div className="dictation-preview">
  *     <ComposerPrimitive.DictationTranscript />
  *   </div>
- * </ComposerPrimitive.If>
+ * </AuiIf>
  * ```
  */
 declare const ComposerPrimitiveDictationTranscript: import("react").ForwardRefExoticComponent<Omit<Omit<import("react").ClassAttributes<HTMLSpanElement> & import("react").HTMLAttributes<HTMLSpanElement> & {

package/dist/primitives/composer/ComposerDictationTranscript.js

@@ -1,7 +1,7 @@
 "use client";
 import { Primitive } from "../../utils/Primitive.js";
-import { useAuiState } from "@assistant-ui/store";
 import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAuiState } from "@assistant-ui/store";
 import { forwardRef } from "@assistant-ui/tap/react-shim";
 import { jsx } from "react/jsx-runtime";
 //#region src/primitives/composer/ComposerDictationTranscript.tsx
@@ -13,11 +13,11 @@ import { jsx } from "react/jsx-runtime";
 *
 * @example
 * ```tsx
-* <ComposerPrimitive.If dictation>
+* <AuiIf condition={(s) => s.composer.dictation != null}>
 *   <div className="dictation-preview">
 *     <ComposerPrimitive.DictationTranscript />
 *   </div>
-* </ComposerPrimitive.If>
+* </AuiIf>
 * ```
 */
 const ComposerPrimitiveDictationTranscript = forwardRef((t0, forwardRef) => {

package/dist/primitives/composer/ComposerDictationTranscript.js.map

@@ -1 +1 @@
-{"version":3,"file":"ComposerDictationTranscript.js","names":["c","_c","Primitive","ComponentRef","forwardRef","ComponentPropsWithoutRef","useAuiState","ComposerPrimitiveDictationTranscript","Element","span","Props","t0","$","children","props","transcript","_temp","t1","t2","displayName","s","composer","dictation"],"sources":["../../../src/primitives/composer/ComposerDictationTranscript.tsx"],"sourcesContent":["\"use client\";\n\nimport { Primitive } from \"../../utils/Primitive\";\nimport {\n  type ComponentRef,\n  forwardRef,\n  type ComponentPropsWithoutRef,\n} from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport namespace ComposerPrimitiveDictationTranscript {\n  export type Element = ComponentRef<typeof Primitive.span>;\n  export type Props = ComponentPropsWithoutRef<typeof Primitive.span>;\n}\n\n/**\n * Renders the current interim (partial) transcript while dictation is active.\n *\n * This component displays real-time feedback of what the user is saying before\n * the transcription is finalized and committed to the composer input.\n *\n * @example\n * ```tsx\n * <ComposerPrimitive.If dictation>\n *   <div className=\"dictation-preview\">\n *     <ComposerPrimitive.DictationTranscript />\n *   </div>\n * </ComposerPrimitive.If>\n * ```\n */\nexport const ComposerPrimitiveDictationTranscript = forwardRef<\n  ComposerPrimitiveDictationTranscript.Element,\n  ComposerPrimitiveDictationTranscript.Props\n>(({ children, ...props }, forwardRef) => {\n  const transcript = useAuiState((s) => s.composer.dictation?.transcript);\n\n  if (!transcript) return null;\n\n  return (\n    <Primitive.span {...props} ref={forwardRef}>\n      {children ?? transcript}\n    </Primitive.span>\n  );\n});\n\nComposerPrimitiveDictationTranscript.displayName =\n  \"ComposerPrimitive.DictationTranscript\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AA8BA,MAAaO,uCAAuCH,YAGlDO,IAAAP,eAAA;CAAA,MAAAQ,IAAAX,EAAA,CAAA;CAAA,IAAAY;CAAA,IAAAC;CAAA,IAAAF,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAE,aAAAC,SAAAH;EAAsBC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;EAAAD,EAAA,KAAAE;CAAA,OAAA;EAAAD,WAAAD,EAAA;EAAAE,QAAAF,EAAA;CAAA;CACvB,MAAAG,aAAmBT,YAAYU,KAAuC;CAEtE,IAAI,CAACD,YAAU,OAAS;CAInB,MAAAE,KAAAJ,YAAAE;CAAsB,IAAAG;CAAA,IAAAN,EAAA,OAAAR,cAAAQ,EAAA,OAAAE,SAAAF,EAAA,OAAAK,IAAA;EADzBC,KAAA,oBAAA,UAAA,MAAA;GAAA,GAAoBJ;GAAYV,KAAAA;GAC7Ba,UAAAA;EACH,CAAA;EAAiBL,EAAA,KAAAR;EAAAQ,EAAA,KAAAE;EAAAF,EAAA,KAAAK;EAAAL,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAFjBM;AAEiB,CAEpB;AAEDX,qCAAqCY,cACnC;AAbA,SAAAH,MAAAI,GAAA;CAAA,OACsCA,EAACC,SAASC,WAAsBP;AAAA"}
+{"version":3,"file":"ComposerDictationTranscript.js","names":["c","_c","Primitive","ComponentRef","forwardRef","ComponentPropsWithoutRef","useAuiState","ComposerPrimitiveDictationTranscript","Element","span","Props","t0","$","children","props","transcript","_temp","t1","t2","displayName","s","composer","dictation"],"sources":["../../../src/primitives/composer/ComposerDictationTranscript.tsx"],"sourcesContent":["\"use client\";\n\nimport { Primitive } from \"../../utils/Primitive\";\nimport {\n  type ComponentRef,\n  forwardRef,\n  type ComponentPropsWithoutRef,\n} from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\n\nexport namespace ComposerPrimitiveDictationTranscript {\n  export type Element = ComponentRef<typeof Primitive.span>;\n  export type Props = ComponentPropsWithoutRef<typeof Primitive.span>;\n}\n\n/**\n * Renders the current interim (partial) transcript while dictation is active.\n *\n * This component displays real-time feedback of what the user is saying before\n * the transcription is finalized and committed to the composer input.\n *\n * @example\n * ```tsx\n * <AuiIf condition={(s) => s.composer.dictation != null}>\n *   <div className=\"dictation-preview\">\n *     <ComposerPrimitive.DictationTranscript />\n *   </div>\n * </AuiIf>\n * ```\n */\nexport const ComposerPrimitiveDictationTranscript = forwardRef<\n  ComposerPrimitiveDictationTranscript.Element,\n  ComposerPrimitiveDictationTranscript.Props\n>(({ children, ...props }, forwardRef) => {\n  const transcript = useAuiState((s) => s.composer.dictation?.transcript);\n\n  if (!transcript) return null;\n\n  return (\n    <Primitive.span {...props} ref={forwardRef}>\n      {children ?? transcript}\n    </Primitive.span>\n  );\n});\n\nComposerPrimitiveDictationTranscript.displayName =\n  \"ComposerPrimitive.DictationTranscript\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AA8BA,MAAaO,uCAAuCH,YAGlDO,IAAAP,eAAA;CAAA,MAAAQ,IAAAX,EAAA,CAAA;CAAA,IAAAY;CAAA,IAAAC;CAAA,IAAAF,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAE,aAAAC,SAAAH;EAAsBC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;EAAAD,EAAA,KAAAE;CAAA,OAAA;EAAAD,WAAAD,EAAA;EAAAE,QAAAF,EAAA;CAAA;CACvB,MAAAG,aAAmBT,YAAYU,KAAuC;CAEtE,IAAI,CAACD,YAAU,OAAS;CAInB,MAAAE,KAAAJ,YAAAE;CAAsB,IAAAG;CAAA,IAAAN,EAAA,OAAAR,cAAAQ,EAAA,OAAAE,SAAAF,EAAA,OAAAK,IAAA;EADzBC,KAAA,oBAAA,UAAA,MAAA;GAAA,GAAoBJ;GAAYV,KAAAA;GAC7Ba,UAAAA;EACH,CAAA;EAAiBL,EAAA,KAAAR;EAAAQ,EAAA,KAAAE;EAAAF,EAAA,KAAAK;EAAAL,EAAA,KAAAM;CAAA,OAAAA,KAAAN,EAAA;CAAA,OAFjBM;AAEiB,CAEpB;AAEDX,qCAAqCY,cACnC;AAbA,SAAAH,MAAAI,GAAA;CAAA,OACsCA,EAACC,SAASC,WAAsBP;AAAA"}

package/dist/primitives/composer/ComposerInput.js

@@ -9,9 +9,9 @@ import { cloneElement, forwardRef, isValidElement, useCallback, useEffect, useRe
 import { jsx } from "react/jsx-runtime";
 import { useComposedRefs } from "@radix-ui/react-compose-refs";
 import { composeEventHandlers } from "@radix-ui/primitive";
-import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
 import { Slot } from "radix-ui";
 import TextareaAutosize from "react-textarea-autosize";
+import { useEscapeKeydown } from "@radix-ui/react-use-escape-keydown";
 import { flushTapSync } from "@assistant-ui/tap";
 //#region src/primitives/composer/ComposerInput.tsx
 const TOUCH_PRIMARY_QUERY = "(pointer: coarse) and (not (any-pointer: fine))";
@@ -142,7 +142,7 @@ const ComposerPrimitiveInput = forwardRef(({ autoFocus = false, asChild, render,
 	]);
 	useEffect(() => {
 		if (aui.composer.getState().type !== "thread" || !unstable_focusOnThreadSwitched) return void 0;
-		return aui.on("threadListItem.switchedTo", focus);
+		return aui.on("threads.selectionChanged", focus);
 	}, [
 		unstable_focusOnThreadSwitched,
 		focus,