@assistant-ui/core 0.3.14 → 0.3.16

package/dist/adapters/voice.js

@@ -66,6 +66,13 @@ function createVoiceSession(options, setup) {
 			if (disconnected) return;
 			disconnected = true;
 			detachAbortHandler();
+			if (currentStatus.type !== "ended") {
+				currentStatus = {
+					type: "ended",
+					reason: "cancelled"
+				};
+				notifyEventListeners(statusCbs, currentStatus, "Voice session");
+			}
 			try {
 				controls?.disconnect();
 			} finally {
@@ -100,6 +107,10 @@ function createVoiceSession(options, setup) {
 	if (abortSignal) {
 		abortHandler = () => session.disconnect();
 		abortSignal.addEventListener("abort", abortHandler, { once: true });
+		if (abortSignal.aborted) {
+			session.disconnect();
+			return session;
+		}
 	}
 	const doSetup = async () => {
 		try {

package/dist/adapters/voice.js.map

@@ -1 +1 @@
-{"version":3,"file":"voice.js","names":["Unsubscribe","notifyEventListeners","RealtimeVoiceAdapter","Status","type","reason","error","Mode","TranscriptItem","role","text","isFinal","Session","status","isMuted","disconnect","mute","unmute","onStatusChange","callback","onTranscript","transcript","onModeChange","mode","onVolumeChange","volume","connect","options","abortSignal","AbortSignal","VoiceSessionControls","VoiceSessionHelpers","setStatus","end","emitTranscript","item","emitMode","emitVolume","isDisposed","createVoiceSession","setup","helpers","Promise","statusCbs","Set","s","transcriptCbs","t","modeCbs","m","volumeCbs","v","currentStatus","disposed","disconnected","controls","abortHandler","detachAbortHandler","removeEventListener","undefined","cleanup","clear","session","cb","add","delete","addEventListener","once","doSetup"],"sources":["../../src/adapters/voice.ts"],"sourcesContent":["import type { Unsubscribe } from \"../types/unsubscribe\";\nimport { notifyEventListeners } from \"../utils/notify-event-listeners\";\n\nexport namespace RealtimeVoiceAdapter {\n  export type Status =\n    | {\n        type: \"starting\" | \"running\";\n      }\n    | {\n        type: \"ended\";\n        reason: \"finished\" | \"cancelled\" | \"error\";\n        error?: unknown;\n      };\n\n  export type Mode = \"listening\" | \"speaking\";\n\n  export type TranscriptItem = {\n    role: \"user\" | \"assistant\";\n    text: string;\n    isFinal?: boolean;\n  };\n\n  export type Session = {\n    status: Status;\n    isMuted: boolean;\n\n    disconnect: () => void;\n    mute: () => void;\n    unmute: () => void;\n\n    onStatusChange: (callback: (status: Status) => void) => Unsubscribe;\n    onTranscript: (\n      callback: (transcript: TranscriptItem) => void,\n    ) => Unsubscribe;\n    onModeChange: (callback: (mode: Mode) => void) => Unsubscribe;\n    onVolumeChange: (callback: (volume: number) => void) => Unsubscribe;\n  };\n}\n\nexport type RealtimeVoiceAdapter = {\n  connect: (options: {\n    abortSignal?: AbortSignal;\n  }) => RealtimeVoiceAdapter.Session;\n};\n\nexport type VoiceSessionControls = {\n  disconnect: () => void;\n  mute: () => void;\n  unmute: () => void;\n};\n\nexport type VoiceSessionHelpers = {\n  setStatus: (status: RealtimeVoiceAdapter.Status) => void;\n  end: (reason: \"finished\" | \"cancelled\" | \"error\", error?: unknown) => void;\n  emitTranscript: (item: RealtimeVoiceAdapter.TranscriptItem) => void;\n  emitMode: (mode: RealtimeVoiceAdapter.Mode) => void;\n  emitVolume: (volume: number) => void;\n  isDisposed: () => boolean;\n};\n\nexport function createVoiceSession(\n  options: { abortSignal?: AbortSignal },\n  setup: (helpers: VoiceSessionHelpers) => Promise<VoiceSessionControls>,\n): RealtimeVoiceAdapter.Session {\n  const statusCbs = new Set<(s: RealtimeVoiceAdapter.Status) => void>();\n  const transcriptCbs = new Set<\n    (t: RealtimeVoiceAdapter.TranscriptItem) => void\n  >();\n  const modeCbs = new Set<(m: RealtimeVoiceAdapter.Mode) => void>();\n  const volumeCbs = new Set<(v: number) => void>();\n\n  let currentStatus: RealtimeVoiceAdapter.Status = { type: \"starting\" };\n  let isMuted = false;\n  let disposed = false;\n  let disconnected = false;\n  let controls: VoiceSessionControls | null = null;\n  const abortSignal = options.abortSignal;\n  let abortHandler: (() => void) | undefined;\n\n  const detachAbortHandler = () => {\n    if (abortHandler) {\n      abortSignal?.removeEventListener(\"abort\", abortHandler);\n      abortHandler = undefined;\n    }\n  };\n\n  const cleanup = () => {\n    disposed = true;\n    statusCbs.clear();\n    transcriptCbs.clear();\n    modeCbs.clear();\n    volumeCbs.clear();\n  };\n\n  const helpers: VoiceSessionHelpers = {\n    setStatus: (status) => {\n      if (disposed) return;\n      currentStatus = status;\n      notifyEventListeners(statusCbs, status, \"Voice session\");\n    },\n    end: (reason, error?) => {\n      if (disposed) return;\n      currentStatus = { type: \"ended\", reason, error };\n      notifyEventListeners(statusCbs, currentStatus, \"Voice session\");\n      cleanup();\n    },\n    emitTranscript: (item) => {\n      if (disposed) return;\n      notifyEventListeners(transcriptCbs, item, \"Voice session\");\n    },\n    emitMode: (mode) => {\n      if (disposed) return;\n      notifyEventListeners(modeCbs, mode, \"Voice session\");\n    },\n    emitVolume: (volume) => {\n      if (disposed) return;\n      notifyEventListeners(volumeCbs, volume, \"Voice session\");\n    },\n    isDisposed: () => disposed,\n  };\n\n  const session: RealtimeVoiceAdapter.Session = {\n    get status() {\n      return currentStatus;\n    },\n    get isMuted() {\n      return isMuted;\n    },\n    disconnect: () => {\n      if (disconnected) return;\n      disconnected = true;\n      detachAbortHandler();\n      try {\n        controls?.disconnect();\n      } finally {\n        cleanup();\n      }\n    },\n    mute: () => {\n      controls?.mute();\n      isMuted = true;\n    },\n    unmute: () => {\n      controls?.unmute();\n      isMuted = false;\n    },\n    onStatusChange: (cb) => {\n      statusCbs.add(cb);\n      return () => statusCbs.delete(cb);\n    },\n    onTranscript: (cb) => {\n      transcriptCbs.add(cb);\n      return () => transcriptCbs.delete(cb);\n    },\n    onModeChange: (cb) => {\n      modeCbs.add(cb);\n      return () => modeCbs.delete(cb);\n    },\n    onVolumeChange: (cb) => {\n      volumeCbs.add(cb);\n      return () => volumeCbs.delete(cb);\n    },\n  };\n\n  if (abortSignal) {\n    abortHandler = () => session.disconnect();\n    abortSignal.addEventListener(\"abort\", abortHandler, { once: true });\n  }\n\n  const doSetup = async () => {\n    try {\n      if (disposed) return;\n      controls = await setup(helpers);\n      if (disposed) {\n        controls.disconnect();\n      } else if (isMuted) {\n        controls.mute();\n      }\n    } catch (error) {\n      helpers.end(\"error\", error);\n    }\n  };\n\n  doSetup();\n  return session;\n}\n"],"mappings":";;AA4DA,SAAgBuC,mBACdZ,SACAa,OAC8B;CAC9B,MAAMG,4BAAY,IAAIC,IAA8C;CACpE,MAAME,gCAAgB,IAAIF,IAExB;CACF,MAAMI,0BAAU,IAAIJ,IAA4C;CAChE,MAAMM,4BAAY,IAAIN,IAAyB;CAE/C,IAAIQ,gBAA6C,EAAEhD,MAAM,WAAW;CACpE,IAAIU,UAAU;CACd,IAAIuC,WAAW;CACf,IAAIC,eAAe;CACnB,IAAIC,WAAwC;CAC5C,MAAM3B,cAAcD,QAAQC;CAC5B,IAAI4B;CAEJ,MAAMC,2BAA2B;EAC/B,IAAID,cAAc;GAChB5B,aAAa8B,oBAAoB,SAASF,YAAY;GACtDA,eAAeG,KAAAA;EACjB;CACF;CAEA,MAAMC,gBAAgB;EACpBP,WAAW;EACXV,UAAUkB,MAAM;EAChBf,cAAce,MAAM;EACpBb,QAAQa,MAAM;EACdX,UAAUW,MAAM;CAClB;CAEA,MAAMpB,UAA+B;EACnCT,YAAYnB,WAAW;GACrB,IAAIwC,UAAU;GACdD,gBAAgBvC;GAChBZ,qBAAqB0C,WAAW9B,QAAQ,eAAe;EACzD;EACAoB,MAAM5B,QAAQC,UAAW;GACvB,IAAI+C,UAAU;GACdD,gBAAgB;IAAEhD,MAAM;IAASC;IAAQC;GAAM;GAC/CL,qBAAqB0C,WAAWS,eAAe,eAAe;GAC9DQ,QAAQ;EACV;EACA1B,iBAAiBC,SAAS;GACxB,IAAIkB,UAAU;GACdpD,qBAAqB6C,eAAeX,MAAM,eAAe;EAC3D;EACAC,WAAWb,SAAS;GAClB,IAAI8B,UAAU;GACdpD,qBAAqB+C,SAASzB,MAAM,eAAe;EACrD;EACAc,aAAaZ,WAAW;GACtB,IAAI4B,UAAU;GACdpD,qBAAqBiD,WAAWzB,QAAQ,eAAe;EACzD;EACAa,kBAAkBe;CACpB;CAEA,MAAMS,UAAwC;EAC5C,IAAIjD,SAAS;GACX,OAAOuC;EACT;EACA,IAAItC,UAAU;GACZ,OAAOA;EACT;EACAC,kBAAkB;GAChB,IAAIuC,cAAc;GAClBA,eAAe;GACfG,mBAAmB;GACnB,IAAI;IACFF,UAAUxC,WAAW;GACvB,UAAU;IACR6C,QAAQ;GACV;EACF;EACA5C,YAAY;GACVuC,UAAUvC,KAAK;GACfF,UAAU;EACZ;EACAG,cAAc;GACZsC,UAAUtC,OAAO;GACjBH,UAAU;EACZ;EACAI,iBAAiB6C,OAAO;GACtBpB,UAAUqB,IAAID,EAAE;GAChB,aAAapB,UAAUsB,OAAOF,EAAE;EAClC;EACA3C,eAAe2C,OAAO;GACpBjB,cAAckB,IAAID,EAAE;GACpB,aAAajB,cAAcmB,OAAOF,EAAE;EACtC;EACAzC,eAAeyC,OAAO;GACpBf,QAAQgB,IAAID,EAAE;GACd,aAAaf,QAAQiB,OAAOF,EAAE;EAChC;EACAvC,iBAAiBuC,OAAO;GACtBb,UAAUc,IAAID,EAAE;GAChB,aAAab,UAAUe,OAAOF,EAAE;EAClC;CACF;CAEA,IAAInC,aAAa;EACf4B,qBAAqBM,QAAQ/C,WAAW;EACxCa,YAAYsC,iBAAiB,SAASV,cAAc,EAAEW,MAAM,KAAK,CAAC;CACpE;CAEA,MAAMC,UAAU,YAAY;EAC1B,IAAI;GACF,IAAIf,UAAU;GACdE,WAAW,MAAMf,MAAMC,OAAO;GAC9B,IAAIY,UACFE,SAASxC,WAAW;QACf,IAAID,SACTyC,SAASvC,KAAK;EAElB,SAASV,OAAO;GACdmC,QAAQR,IAAI,SAAS3B,KAAK;EAC5B;CACF;CAEA8D,QAAQ;CACR,OAAON;AACT"}
+{"version":3,"file":"voice.js","names":["Unsubscribe","notifyEventListeners","RealtimeVoiceAdapter","Status","type","reason","error","Mode","TranscriptItem","role","text","isFinal","Session","status","isMuted","disconnect","mute","unmute","onStatusChange","callback","onTranscript","transcript","onModeChange","mode","onVolumeChange","volume","connect","options","abortSignal","AbortSignal","VoiceSessionControls","VoiceSessionHelpers","setStatus","end","emitTranscript","item","emitMode","emitVolume","isDisposed","createVoiceSession","setup","helpers","Promise","statusCbs","Set","s","transcriptCbs","t","modeCbs","m","volumeCbs","v","currentStatus","disposed","disconnected","controls","abortHandler","detachAbortHandler","removeEventListener","undefined","cleanup","clear","session","cb","add","delete","addEventListener","once","aborted","doSetup"],"sources":["../../src/adapters/voice.ts"],"sourcesContent":["import type { Unsubscribe } from \"../types/unsubscribe\";\nimport { notifyEventListeners } from \"../utils/notify-event-listeners\";\n\nexport namespace RealtimeVoiceAdapter {\n  export type Status =\n    | {\n        type: \"starting\" | \"running\";\n      }\n    | {\n        type: \"ended\";\n        reason: \"finished\" | \"cancelled\" | \"error\";\n        error?: unknown;\n      };\n\n  export type Mode = \"listening\" | \"speaking\";\n\n  export type TranscriptItem = {\n    role: \"user\" | \"assistant\";\n    text: string;\n    isFinal?: boolean;\n  };\n\n  export type Session = {\n    status: Status;\n    isMuted: boolean;\n\n    disconnect: () => void;\n    mute: () => void;\n    unmute: () => void;\n\n    onStatusChange: (callback: (status: Status) => void) => Unsubscribe;\n    onTranscript: (\n      callback: (transcript: TranscriptItem) => void,\n    ) => Unsubscribe;\n    onModeChange: (callback: (mode: Mode) => void) => Unsubscribe;\n    onVolumeChange: (callback: (volume: number) => void) => Unsubscribe;\n  };\n}\n\nexport type RealtimeVoiceAdapter = {\n  connect: (options: {\n    abortSignal?: AbortSignal;\n  }) => RealtimeVoiceAdapter.Session;\n};\n\nexport type VoiceSessionControls = {\n  disconnect: () => void;\n  mute: () => void;\n  unmute: () => void;\n};\n\nexport type VoiceSessionHelpers = {\n  setStatus: (status: RealtimeVoiceAdapter.Status) => void;\n  end: (reason: \"finished\" | \"cancelled\" | \"error\", error?: unknown) => void;\n  emitTranscript: (item: RealtimeVoiceAdapter.TranscriptItem) => void;\n  emitMode: (mode: RealtimeVoiceAdapter.Mode) => void;\n  emitVolume: (volume: number) => void;\n  isDisposed: () => boolean;\n};\n\nexport function createVoiceSession(\n  options: { abortSignal?: AbortSignal },\n  setup: (helpers: VoiceSessionHelpers) => Promise<VoiceSessionControls>,\n): RealtimeVoiceAdapter.Session {\n  const statusCbs = new Set<(s: RealtimeVoiceAdapter.Status) => void>();\n  const transcriptCbs = new Set<\n    (t: RealtimeVoiceAdapter.TranscriptItem) => void\n  >();\n  const modeCbs = new Set<(m: RealtimeVoiceAdapter.Mode) => void>();\n  const volumeCbs = new Set<(v: number) => void>();\n\n  let currentStatus: RealtimeVoiceAdapter.Status = { type: \"starting\" };\n  let isMuted = false;\n  let disposed = false;\n  let disconnected = false;\n  let controls: VoiceSessionControls | null = null;\n  const abortSignal = options.abortSignal;\n  let abortHandler: (() => void) | undefined;\n\n  const detachAbortHandler = () => {\n    if (abortHandler) {\n      abortSignal?.removeEventListener(\"abort\", abortHandler);\n      abortHandler = undefined;\n    }\n  };\n\n  const cleanup = () => {\n    disposed = true;\n    statusCbs.clear();\n    transcriptCbs.clear();\n    modeCbs.clear();\n    volumeCbs.clear();\n  };\n\n  const helpers: VoiceSessionHelpers = {\n    setStatus: (status) => {\n      if (disposed) return;\n      currentStatus = status;\n      notifyEventListeners(statusCbs, status, \"Voice session\");\n    },\n    end: (reason, error?) => {\n      if (disposed) return;\n      currentStatus = { type: \"ended\", reason, error };\n      notifyEventListeners(statusCbs, currentStatus, \"Voice session\");\n      cleanup();\n    },\n    emitTranscript: (item) => {\n      if (disposed) return;\n      notifyEventListeners(transcriptCbs, item, \"Voice session\");\n    },\n    emitMode: (mode) => {\n      if (disposed) return;\n      notifyEventListeners(modeCbs, mode, \"Voice session\");\n    },\n    emitVolume: (volume) => {\n      if (disposed) return;\n      notifyEventListeners(volumeCbs, volume, \"Voice session\");\n    },\n    isDisposed: () => disposed,\n  };\n\n  const session: RealtimeVoiceAdapter.Session = {\n    get status() {\n      return currentStatus;\n    },\n    get isMuted() {\n      return isMuted;\n    },\n    disconnect: () => {\n      if (disconnected) return;\n      disconnected = true;\n      detachAbortHandler();\n      if (currentStatus.type !== \"ended\") {\n        currentStatus = { type: \"ended\", reason: \"cancelled\" };\n        notifyEventListeners(statusCbs, currentStatus, \"Voice session\");\n      }\n      try {\n        controls?.disconnect();\n      } finally {\n        cleanup();\n      }\n    },\n    mute: () => {\n      controls?.mute();\n      isMuted = true;\n    },\n    unmute: () => {\n      controls?.unmute();\n      isMuted = false;\n    },\n    onStatusChange: (cb) => {\n      statusCbs.add(cb);\n      return () => statusCbs.delete(cb);\n    },\n    onTranscript: (cb) => {\n      transcriptCbs.add(cb);\n      return () => transcriptCbs.delete(cb);\n    },\n    onModeChange: (cb) => {\n      modeCbs.add(cb);\n      return () => modeCbs.delete(cb);\n    },\n    onVolumeChange: (cb) => {\n      volumeCbs.add(cb);\n      return () => volumeCbs.delete(cb);\n    },\n  };\n\n  if (abortSignal) {\n    abortHandler = () => session.disconnect();\n    abortSignal.addEventListener(\"abort\", abortHandler, { once: true });\n    if (abortSignal.aborted) {\n      session.disconnect();\n      return session;\n    }\n  }\n\n  const doSetup = async () => {\n    try {\n      if (disposed) return;\n      controls = await setup(helpers);\n      if (disposed) {\n        controls.disconnect();\n      } else if (isMuted) {\n        controls.mute();\n      }\n    } catch (error) {\n      helpers.end(\"error\", error);\n    }\n  };\n\n  doSetup();\n  return session;\n}\n"],"mappings":";;AA4DA,SAAgBuC,mBACdZ,SACAa,OAC8B;CAC9B,MAAMG,4BAAY,IAAIC,IAA8C;CACpE,MAAME,gCAAgB,IAAIF,IAExB;CACF,MAAMI,0BAAU,IAAIJ,IAA4C;CAChE,MAAMM,4BAAY,IAAIN,IAAyB;CAE/C,IAAIQ,gBAA6C,EAAEhD,MAAM,WAAW;CACpE,IAAIU,UAAU;CACd,IAAIuC,WAAW;CACf,IAAIC,eAAe;CACnB,IAAIC,WAAwC;CAC5C,MAAM3B,cAAcD,QAAQC;CAC5B,IAAI4B;CAEJ,MAAMC,2BAA2B;EAC/B,IAAID,cAAc;GAChB5B,aAAa8B,oBAAoB,SAASF,YAAY;GACtDA,eAAeG,KAAAA;EACjB;CACF;CAEA,MAAMC,gBAAgB;EACpBP,WAAW;EACXV,UAAUkB,MAAM;EAChBf,cAAce,MAAM;EACpBb,QAAQa,MAAM;EACdX,UAAUW,MAAM;CAClB;CAEA,MAAMpB,UAA+B;EACnCT,YAAYnB,WAAW;GACrB,IAAIwC,UAAU;GACdD,gBAAgBvC;GAChBZ,qBAAqB0C,WAAW9B,QAAQ,eAAe;EACzD;EACAoB,MAAM5B,QAAQC,UAAW;GACvB,IAAI+C,UAAU;GACdD,gBAAgB;IAAEhD,MAAM;IAASC;IAAQC;GAAM;GAC/CL,qBAAqB0C,WAAWS,eAAe,eAAe;GAC9DQ,QAAQ;EACV;EACA1B,iBAAiBC,SAAS;GACxB,IAAIkB,UAAU;GACdpD,qBAAqB6C,eAAeX,MAAM,eAAe;EAC3D;EACAC,WAAWb,SAAS;GAClB,IAAI8B,UAAU;GACdpD,qBAAqB+C,SAASzB,MAAM,eAAe;EACrD;EACAc,aAAaZ,WAAW;GACtB,IAAI4B,UAAU;GACdpD,qBAAqBiD,WAAWzB,QAAQ,eAAe;EACzD;EACAa,kBAAkBe;CACpB;CAEA,MAAMS,UAAwC;EAC5C,IAAIjD,SAAS;GACX,OAAOuC;EACT;EACA,IAAItC,UAAU;GACZ,OAAOA;EACT;EACAC,kBAAkB;GAChB,IAAIuC,cAAc;GAClBA,eAAe;GACfG,mBAAmB;GACnB,IAAIL,cAAchD,SAAS,SAAS;IAClCgD,gBAAgB;KAAEhD,MAAM;KAASC,QAAQ;IAAY;IACrDJ,qBAAqB0C,WAAWS,eAAe,eAAe;GAChE;GACA,IAAI;IACFG,UAAUxC,WAAW;GACvB,UAAU;IACR6C,QAAQ;GACV;EACF;EACA5C,YAAY;GACVuC,UAAUvC,KAAK;GACfF,UAAU;EACZ;EACAG,cAAc;GACZsC,UAAUtC,OAAO;GACjBH,UAAU;EACZ;EACAI,iBAAiB6C,OAAO;GACtBpB,UAAUqB,IAAID,EAAE;GAChB,aAAapB,UAAUsB,OAAOF,EAAE;EAClC;EACA3C,eAAe2C,OAAO;GACpBjB,cAAckB,IAAID,EAAE;GACpB,aAAajB,cAAcmB,OAAOF,EAAE;EACtC;EACAzC,eAAeyC,OAAO;GACpBf,QAAQgB,IAAID,EAAE;GACd,aAAaf,QAAQiB,OAAOF,EAAE;EAChC;EACAvC,iBAAiBuC,OAAO;GACtBb,UAAUc,IAAID,EAAE;GAChB,aAAab,UAAUe,OAAOF,EAAE;EAClC;CACF;CAEA,IAAInC,aAAa;EACf4B,qBAAqBM,QAAQ/C,WAAW;EACxCa,YAAYsC,iBAAiB,SAASV,cAAc,EAAEW,MAAM,KAAK,CAAC;EAClE,IAAIvC,YAAYwC,SAAS;GACvBN,QAAQ/C,WAAW;GACnB,OAAO+C;EACT;CACF;CAEA,MAAMO,UAAU,YAAY;EAC1B,IAAI;GACF,IAAIhB,UAAU;GACdE,WAAW,MAAMf,MAAMC,OAAO;GAC9B,IAAIY,UACFE,SAASxC,WAAW;QACf,IAAID,SACTyC,SAASvC,KAAK;EAElB,SAASV,OAAO;GACdmC,QAAQR,IAAI,SAAS3B,KAAK;EAC5B;CACF;CAEA+D,QAAQ;CACR,OAAOP;AACT"}

package/dist/index.js

@@ -4,10 +4,10 @@ import { mergeModelContexts } from "./model-context/types.js";
 import { tool } from "./model-context/tool.js";
 import { unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "./model-context/interactable-composer-metadata.js";
 import { ModelContextRegistry } from "./model-context/registry.js";
+import { generateId } from "./utils/id.js";
 import { FRAME_MESSAGE_CHANNEL } from "./model-context/frame/types.js";
 import { AssistantFrameHost } from "./model-context/frame/host.js";
 import { AssistantFrameProvider } from "./model-context/frame/provider.js";
-import { generateId } from "./utils/id.js";
 import { CompositeAttachmentAdapter, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter } from "./adapters/attachment.js";
 import { WebSpeechDictationAdapter, WebSpeechSynthesisAdapter } from "./adapters/speech.js";
 import { createVoiceSession } from "./adapters/voice.js";

package/dist/internal.d.ts

@@ -8,13 +8,18 @@ import { BaseSubject, BaseSubscribable, EventSubscribable, EventSubscriptionSubj
 import { generateErrorMessageId, generateId, isErrorMessageId } from "./utils/id.js";
 import { getThreadMessageText } from "./utils/text.js";
 import { toMessagePartStatus } from "./utils/normalizePartStatus.js";
+import { notifyEventListeners } from "./utils/notify-event-listeners.js";
 import { resolveToolApprovalResponse } from "./runtime/utils/resolveToolApprovalResponse.js";
 import { CompositeContextProvider } from "./utils/composite-context-provider.js";
 import { stableStringifyToolArgs, trackToolArgsKeyOrder } from "./utils/json/stable-stringify-tool-args.js";
 import { isJSONValue, isRecord } from "./utils/json/is-json.js";
-import { dataUrlMediaType, httpUrlPattern, isParsableUrl, parseDataUrl } from "./utils/data-url.js";
+import { FilePartSource, dataUrlMediaType, httpUrlPattern, isParsableUrl, parseDataUrl, resolveFilePartSource } from "./utils/data-url.js";
+import { invokeUserCallback } from "./utils/invoke-user-callback.js";
 import { detectImageMediaType } from "./utils/image-media-type.js";
 import { resolveFileMediaType, resolveImageMediaType, toMediaWireUrl } from "./utils/wire-media.js";
+import { createToolCallCancellationStub, scanPendingToolCalls } from "./runtime/utils/pending-tool-calls.js";
+import { AbortableThreadLoadPurpose, createAbortableThreadLoad } from "./runtime/utils/abortable-thread-load.js";
+import { createCloudThreadListAdapterCreateFallback } from "./react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.js";
 import { ComposerRuntimeCoreBinding, EditComposerRuntimeCoreBinding, MessageStateBinding, ThreadComposerRuntimeCoreBinding } from "./runtime/api/bindings.js";
 import { BaseAssistantRuntimeCore } from "./runtime/base/base-assistant-runtime-core.js";
 import { BaseComposerRuntimeCore } from "./runtime/base/base-composer-runtime-core.js";
@@ -29,6 +34,7 @@ import { MessagePartRuntimeImpl } from "./runtime/api/message-part-runtime.js";
 import { MessageRuntimeImpl } from "./runtime/api/message-runtime.js";
 import { symbolInnerMessage } from "./runtime/utils/external-store-message.js";
 import { getAutoStatus, isAutoStatus } from "./runtime/utils/auto-status.js";
+import { MessageRepositorySession, MessageRepositorySessionOptions, createMessageRepositorySession } from "./runtime/utils/message-repository-session.js";
 import "./runtime/internal.js";
 import { LocalRuntimeOptionsBase } from "./runtimes/local/local-runtime-options.js";
 import { LocalThreadRuntimeCore } from "./runtimes/local/local-thread-runtime-core.js";
@@ -47,4 +53,4 @@ import { RemoteThreadData, RemoteThreadState, THREAD_MAPPING_ID, createThreadMap
 import "./runtimes/internal.js";
 import { ThreadListItemRuntimeBinding, ThreadRuntimeCoreBinding, ThreadRuntimeImpl, getThreadState } from "./runtime/api/thread-runtime.js";
 import { AssistantRuntimeImpl } from "./runtime/api/assistant-runtime.js";
-export { AssistantRuntimeImpl, AttachmentRuntimeImpl, BaseAssistantRuntimeCore, BaseComposerRuntimeCore, BaseSubject, BaseSubscribable, BaseThreadRuntimeCore, type ComposerRuntimeCoreBinding, ComposerRuntimeImpl, CompositeContextProvider, type ConverterCallback, DefaultEditComposerRuntimeCore, DefaultThreadComposerRuntimeCore, EMPTY_THREAD_CORE, EditComposerAttachmentRuntimeImpl, type EditComposerRuntimeCoreBinding, EditComposerRuntimeImpl, type EventSubscribable, EventSubscriptionSubject, ExportedMessageRepository, type ExportedMessageRepositoryItem, ExternalStoreRuntimeCore, type ExternalStoreThreadFactory, ExternalStoreThreadListRuntimeCore, ExternalStoreThreadRuntimeCore, LazyMemoizeSubject, LocalRuntimeCore, type LocalRuntimeOptionsBase, type LocalThreadFactory, LocalThreadListRuntimeCore, LocalThreadRuntimeCore, MessageAttachmentRuntimeImpl, MessagePartRuntimeImpl, MessageRepository, MessageRuntimeImpl, type MessageStateBinding, type NestedSubscribable, NestedSubscriptionSubject, OptimisticState, ReadonlyThreadRuntimeCore, type RemoteThreadData, type RemoteThreadInitializeResponse, type RemoteThreadListOptions, type RemoteThreadState, SKIP_UPDATE, type SKIP_UPDATE as SKIP_UPDATE_TYPE, ShallowMemoizeSubject, type Subscribable, type SubscribableWithState, type THREAD_MAPPING_ID, ThreadComposerAttachmentRuntimeImpl, type ThreadComposerRuntimeCoreBinding, ThreadComposerRuntimeImpl, type ThreadListItemRuntimeBinding, ThreadListItemRuntimeImpl, type ThreadListItemStateBinding, type ThreadListRuntimeCoreBinding, ThreadListRuntimeImpl, ThreadMessageConverter, type ThreadRuntimeCoreBinding, ThreadRuntimeImpl, ToolInvocationTracker, consumeSuggestionResult, createThreadMappingId, dataUrlMediaType, detectImageMediaType, fileMatchesAccept, fromThreadMessageLike, generateErrorMessageId, generateId, getAutoStatus, getFileDataURL, getThreadData, getThreadMessageText, getThreadState, hasUpcomingMessage, httpUrlPattern, isAutoStatus, isCreateAttachment, isErrorMessageId, isJSONValue, isParsableUrl, isRecord, parseDataUrl, resolveFileMediaType, resolveImageMediaType, resolveToolApprovalResponse, shouldContinue, stableStringifyToolArgs, symbolInnerMessage, toMediaWireUrl, toMessagePartStatus, trackToolArgsKeyOrder, updateStatusReducer };
+export { type AbortableThreadLoadPurpose, AssistantRuntimeImpl, AttachmentRuntimeImpl, BaseAssistantRuntimeCore, BaseComposerRuntimeCore, BaseSubject, BaseSubscribable, BaseThreadRuntimeCore, type ComposerRuntimeCoreBinding, ComposerRuntimeImpl, CompositeContextProvider, type ConverterCallback, DefaultEditComposerRuntimeCore, DefaultThreadComposerRuntimeCore, EMPTY_THREAD_CORE, EditComposerAttachmentRuntimeImpl, type EditComposerRuntimeCoreBinding, EditComposerRuntimeImpl, type EventSubscribable, EventSubscriptionSubject, ExportedMessageRepository, type ExportedMessageRepositoryItem, ExternalStoreRuntimeCore, type ExternalStoreThreadFactory, ExternalStoreThreadListRuntimeCore, ExternalStoreThreadRuntimeCore, type FilePartSource, LazyMemoizeSubject, LocalRuntimeCore, type LocalRuntimeOptionsBase, type LocalThreadFactory, LocalThreadListRuntimeCore, LocalThreadRuntimeCore, MessageAttachmentRuntimeImpl, MessagePartRuntimeImpl, MessageRepository, type MessageRepositorySession, type MessageRepositorySessionOptions, MessageRuntimeImpl, type MessageStateBinding, type NestedSubscribable, NestedSubscriptionSubject, OptimisticState, ReadonlyThreadRuntimeCore, type RemoteThreadData, type RemoteThreadInitializeResponse, type RemoteThreadListOptions, type RemoteThreadState, SKIP_UPDATE, type SKIP_UPDATE as SKIP_UPDATE_TYPE, ShallowMemoizeSubject, type Subscribable, type SubscribableWithState, type THREAD_MAPPING_ID, ThreadComposerAttachmentRuntimeImpl, type ThreadComposerRuntimeCoreBinding, ThreadComposerRuntimeImpl, type ThreadListItemRuntimeBinding, ThreadListItemRuntimeImpl, type ThreadListItemStateBinding, type ThreadListRuntimeCoreBinding, ThreadListRuntimeImpl, ThreadMessageConverter, type ThreadRuntimeCoreBinding, ThreadRuntimeImpl, ToolInvocationTracker, consumeSuggestionResult, createAbortableThreadLoad, createCloudThreadListAdapterCreateFallback, createMessageRepositorySession, createThreadMappingId, createToolCallCancellationStub, dataUrlMediaType, detectImageMediaType, fileMatchesAccept, fromThreadMessageLike, generateErrorMessageId, generateId, getAutoStatus, getFileDataURL, getThreadData, getThreadMessageText, getThreadState, hasUpcomingMessage, httpUrlPattern, invokeUserCallback, isAutoStatus, isCreateAttachment, isErrorMessageId, isJSONValue, isParsableUrl, isRecord, notifyEventListeners, parseDataUrl, resolveFileMediaType, resolveFilePartSource, resolveImageMediaType, resolveToolApprovalResponse, scanPendingToolCalls, shouldContinue, stableStringifyToolArgs, symbolInnerMessage, toMediaWireUrl, toMessagePartStatus, trackToolArgsKeyOrder, updateStatusReducer };

package/dist/internal.js

@@ -1,21 +1,26 @@
 import { isJSONValue, isRecord } from "./utils/json/is-json.js";
+import { notifyEventListeners } from "./utils/notify-event-listeners.js";
 import { BaseSubject, BaseSubscribable, EventSubscriptionSubject, LazyMemoizeSubject, NestedSubscriptionSubject, SKIP_UPDATE, ShallowMemoizeSubject } from "./subscribable/subscribable.js";
+import { CompositeContextProvider } from "./utils/composite-context-provider.js";
 import { generateErrorMessageId, generateId, isErrorMessageId } from "./utils/id.js";
 import { fileMatchesAccept, getFileDataURL } from "./adapters/attachment.js";
 import { getThreadMessageText } from "./utils/text.js";
 import { consumeSuggestionResult } from "./adapters/suggestion.js";
-import { dataUrlMediaType, httpUrlPattern, isParsableUrl, parseDataUrl } from "./utils/data-url.js";
+import { dataUrlMediaType, httpUrlPattern, isParsableUrl, parseDataUrl, resolveFilePartSource } from "./utils/data-url.js";
 import { fromThreadMessageLike } from "./runtime/utils/thread-message-like.js";
 import { symbolInnerMessage } from "./runtime/utils/external-store-message.js";
 import { getAutoStatus, isAutoStatus } from "./runtime/utils/auto-status.js";
 import { ExportedMessageRepository, MessageRepository } from "./runtime/utils/message-repository.js";
 import { toMessagePartStatus } from "./utils/normalizePartStatus.js";
 import { resolveToolApprovalResponse } from "./runtime/utils/resolveToolApprovalResponse.js";
-import { CompositeContextProvider } from "./utils/composite-context-provider.js";
 import { isCreateAttachment } from "./types/attachment.js";
 import { stableStringifyToolArgs, trackToolArgsKeyOrder } from "./utils/json/stable-stringify-tool-args.js";
+import { invokeUserCallback } from "./utils/invoke-user-callback.js";
 import { detectImageMediaType } from "./utils/image-media-type.js";
 import { resolveFileMediaType, resolveImageMediaType, toMediaWireUrl } from "./utils/wire-media.js";
+import { createToolCallCancellationStub, scanPendingToolCalls } from "./runtime/utils/pending-tool-calls.js";
+import { createAbortableThreadLoad } from "./runtime/utils/abortable-thread-load.js";
+import { createCloudThreadListAdapterCreateFallback } from "./react/runtimes/cloud/createCloudThreadListAdapterCreateFallback.js";
 import { BaseAssistantRuntimeCore } from "./runtime/base/base-assistant-runtime-core.js";
 import { AttachmentRuntimeImpl, EditComposerAttachmentRuntimeImpl, MessageAttachmentRuntimeImpl, ThreadComposerAttachmentRuntimeImpl } from "./runtime/api/attachment-runtime.js";
 import { ComposerRuntimeImpl, EditComposerRuntimeImpl, ThreadComposerRuntimeImpl } from "./runtime/api/composer-runtime.js";
@@ -29,6 +34,7 @@ import { BaseThreadRuntimeCore } from "./runtime/base/base-thread-runtime-core.j
 import { ThreadListItemRuntimeImpl } from "./runtime/api/thread-list-item-runtime.js";
 import { ThreadListRuntimeImpl } from "./runtime/api/thread-list-runtime.js";
 import { AssistantRuntimeImpl } from "./runtime/api/assistant-runtime.js";
+import { createMessageRepositorySession } from "./runtime/utils/message-repository-session.js";
 import { shouldContinue } from "./runtimes/local/should-continue.js";
 import { LocalThreadRuntimeCore } from "./runtimes/local/local-thread-runtime-core.js";
 import { LocalThreadListRuntimeCore } from "./runtimes/local/local-thread-list-runtime-core.js";
@@ -42,4 +48,4 @@ import { ReadonlyThreadRuntimeCore } from "./runtimes/readonly/ReadonlyThreadRun
 import { OptimisticState } from "./runtimes/remote-thread-list/optimistic-state.js";
 import { EMPTY_THREAD_CORE } from "./runtimes/remote-thread-list/empty-thread-core.js";
 import { createThreadMappingId, getThreadData, updateStatusReducer } from "./runtimes/remote-thread-list/remote-thread-state.js";
-export { AssistantRuntimeImpl, AttachmentRuntimeImpl, BaseAssistantRuntimeCore, BaseComposerRuntimeCore, BaseSubject, BaseSubscribable, BaseThreadRuntimeCore, ComposerRuntimeImpl, CompositeContextProvider, DefaultEditComposerRuntimeCore, DefaultThreadComposerRuntimeCore, EMPTY_THREAD_CORE, EditComposerAttachmentRuntimeImpl, EditComposerRuntimeImpl, EventSubscriptionSubject, ExportedMessageRepository, ExternalStoreRuntimeCore, ExternalStoreThreadListRuntimeCore, ExternalStoreThreadRuntimeCore, LazyMemoizeSubject, LocalRuntimeCore, LocalThreadListRuntimeCore, LocalThreadRuntimeCore, MessageAttachmentRuntimeImpl, MessagePartRuntimeImpl, MessageRepository, MessageRuntimeImpl, NestedSubscriptionSubject, OptimisticState, ReadonlyThreadRuntimeCore, SKIP_UPDATE, ShallowMemoizeSubject, ThreadComposerAttachmentRuntimeImpl, ThreadComposerRuntimeImpl, ThreadListItemRuntimeImpl, ThreadListRuntimeImpl, ThreadMessageConverter, ThreadRuntimeImpl, ToolInvocationTracker, consumeSuggestionResult, createThreadMappingId, dataUrlMediaType, detectImageMediaType, fileMatchesAccept, fromThreadMessageLike, generateErrorMessageId, generateId, getAutoStatus, getFileDataURL, getThreadData, getThreadMessageText, getThreadState, hasUpcomingMessage, httpUrlPattern, isAutoStatus, isCreateAttachment, isErrorMessageId, isJSONValue, isParsableUrl, isRecord, parseDataUrl, resolveFileMediaType, resolveImageMediaType, resolveToolApprovalResponse, shouldContinue, stableStringifyToolArgs, symbolInnerMessage, toMediaWireUrl, toMessagePartStatus, trackToolArgsKeyOrder, updateStatusReducer };
+export { AssistantRuntimeImpl, AttachmentRuntimeImpl, BaseAssistantRuntimeCore, BaseComposerRuntimeCore, BaseSubject, BaseSubscribable, BaseThreadRuntimeCore, ComposerRuntimeImpl, CompositeContextProvider, DefaultEditComposerRuntimeCore, DefaultThreadComposerRuntimeCore, EMPTY_THREAD_CORE, EditComposerAttachmentRuntimeImpl, EditComposerRuntimeImpl, EventSubscriptionSubject, ExportedMessageRepository, ExternalStoreRuntimeCore, ExternalStoreThreadListRuntimeCore, ExternalStoreThreadRuntimeCore, LazyMemoizeSubject, LocalRuntimeCore, LocalThreadListRuntimeCore, LocalThreadRuntimeCore, MessageAttachmentRuntimeImpl, MessagePartRuntimeImpl, MessageRepository, MessageRuntimeImpl, NestedSubscriptionSubject, OptimisticState, ReadonlyThreadRuntimeCore, SKIP_UPDATE, ShallowMemoizeSubject, ThreadComposerAttachmentRuntimeImpl, ThreadComposerRuntimeImpl, ThreadListItemRuntimeImpl, ThreadListRuntimeImpl, ThreadMessageConverter, ThreadRuntimeImpl, ToolInvocationTracker, consumeSuggestionResult, createAbortableThreadLoad, createCloudThreadListAdapterCreateFallback, createMessageRepositorySession, createThreadMappingId, createToolCallCancellationStub, dataUrlMediaType, detectImageMediaType, fileMatchesAccept, fromThreadMessageLike, generateErrorMessageId, generateId, getAutoStatus, getFileDataURL, getThreadData, getThreadMessageText, getThreadState, hasUpcomingMessage, httpUrlPattern, invokeUserCallback, isAutoStatus, isCreateAttachment, isErrorMessageId, isJSONValue, isParsableUrl, isRecord, notifyEventListeners, parseDataUrl, resolveFileMediaType, resolveFilePartSource, resolveImageMediaType, resolveToolApprovalResponse, scanPendingToolCalls, shouldContinue, stableStringifyToolArgs, symbolInnerMessage, toMediaWireUrl, toMessagePartStatus, trackToolArgsKeyOrder, updateStatusReducer };

package/dist/model-context/frame/host.d.ts

@@ -5,7 +5,6 @@ declare class AssistantFrameHost implements ModelContextProvider {
   private _context;
   private _subscribers;
   private _pendingRequests;
-  private _requestCounter;
   private _iframeWindow;
   private _targetOrigin;
   private _disposed;
@@ -14,6 +13,7 @@ declare class AssistantFrameHost implements ModelContextProvider {
   private updateContext;
   private callTool;
   private sendRequest;
+  private cancelToolCall;
   private requestContext;
   private notifySubscribers;
   getModelContext(): ModelContext;

package/dist/model-context/frame/host.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"host.d.ts","names":[],"sources":["../../../src/model-context/frame/host.ts"],"mappings":";;;cAkDa,8BAA8B;UACjC;UACA;UACA;UAOA;UACA;UACA;UACA;EAEI,YAAA,cAAc,QAAQ;UAU1B;UA6BA;UAoBA;UAkBA;UAuDA;UAYA;EAIR,mBAAmB;EAInB,UAAU,uBAAuB;EAKjC"}
+{"version":3,"file":"host.d.ts","names":[],"sources":["../../../src/model-context/frame/host.ts"],"mappings":";;;cAqDa,8BAA8B;UACjC;UACA;UACA;UAOA;UACA;UACA;EAGN,YAAA,cAAc,QACd;UAWM;UA6BA;UAoBA;UAkBA;UAyDA;UAUA;UAYA;EAIR,mBAAmB;EAInB,UAAU,uBAAuB;EAKjC"}

package/dist/model-context/frame/host.js

@@ -1,6 +1,8 @@
 import { notifySubscribers } from "../../subscribable/subscribable.js";
+import { generateId } from "../../utils/id.js";
 import { FRAME_MESSAGE_CHANNEL } from "./types.js";
 //#region src/model-context/frame/host.ts
+const getDefaultTargetOrigin = () => window.location.origin;
 /**
 * Deserializes tools from JSON Schema format back to Tool objects
 */
@@ -27,11 +29,10 @@ var AssistantFrameHost = class {
 	_context = {};
 	_subscribers = /* @__PURE__ */ new Set();
 	_pendingRequests = /* @__PURE__ */ new Map();
-	_requestCounter = 0;
 	_iframeWindow;
 	_targetOrigin;
 	_disposed = false;
-	constructor(iframeWindow, targetOrigin = "*") {
+	constructor(iframeWindow, targetOrigin = getDefaultTargetOrigin()) {
 		this._iframeWindow = iframeWindow;
 		this._targetOrigin = targetOrigin;
 		this.handleMessage = this.handleMessage.bind(this);
@@ -50,7 +51,7 @@ var AssistantFrameHost = class {
 			case "tool-result": {
 				const pending = this._pendingRequests.get(message.id);
 				if (pending) {
-					if (message.error) pending.reject(new Error(message.error));
+					if (typeof message.error === "string") pending.reject(new Error(message.error));
 					else pending.resolve(message.result);
 					this._pendingRequests.delete(message.id);
 				}
@@ -72,7 +73,7 @@ var AssistantFrameHost = class {
 	callTool(toolName, args, abortSignal) {
 		return this.sendRequest({
 			type: "tool-call",
-			id: `tool-${this._requestCounter++}`,
+			id: `tool-${generateId()}`,
 			toolName,
 			args
 		}, 3e4, `Tool call "${toolName}" timed out`, abortSignal);
@@ -86,6 +87,7 @@ var AssistantFrameHost = class {
 				if (!abortSignal) return;
 				const pending = this._pendingRequests.get(message.id);
 				if (pending) {
+					this.cancelToolCall(message.id);
 					pending.reject(getAbortReason(abortSignal));
 					this._pendingRequests.delete(message.id);
 				}
@@ -107,6 +109,7 @@ var AssistantFrameHost = class {
 			timeoutId = setTimeout(() => {
 				const pending = this._pendingRequests.get(message.id);
 				if (pending) {
+					this.cancelToolCall(message.id);
 					pending.reject(new Error(timeoutMessage));
 					this._pendingRequests.delete(message.id);
 				}
@@ -118,6 +121,15 @@ var AssistantFrameHost = class {
 			}, this._targetOrigin);
 		});
 	}
+	cancelToolCall(id) {
+		this._iframeWindow.postMessage({
+			channel: FRAME_MESSAGE_CHANNEL,
+			message: {
+				type: "tool-cancel",
+				id
+			}
+		}, this._targetOrigin);
+	}
 	requestContext() {
 		this._iframeWindow.postMessage({
 			channel: FRAME_MESSAGE_CHANNEL,
@@ -139,7 +151,10 @@ var AssistantFrameHost = class {
 		window.removeEventListener("message", this.handleMessage);
 		this._subscribers.clear();
 		const error = /* @__PURE__ */ new Error("AssistantFrameHost has been disposed");
-		for (const pending of this._pendingRequests.values()) pending.reject(error);
+		for (const [id, pending] of this._pendingRequests) {
+			this.cancelToolCall(id);
+			pending.reject(error);
+		}
 		this._pendingRequests.clear();
 	}
 };

package/dist/model-context/frame/host.js.map

@@ -1 +1 @@
-{"version":3,"file":"host.js","names":["ModelContextProvider","ModelContext","Unsubscribe","Tool","notifySubscribers","notifyStateSubscribers","FrameMessage","FRAME_MESSAGE_CHANNEL","SerializedModelContext","SerializedTool","deserializeTool","serializedTool","parameters","description","disabled","undefined","type","deserializeModelContext","serialized","system","tools","Object","fromEntries","entries","map","name","tool","getAbortReason","signal","AbortSignal","reason","error","Error","AssistantFrameHost","_context","_subscribers","Set","_pendingRequests","Map","resolve","value","reject","_requestCounter","_iframeWindow","Window","_targetOrigin","_disposed","constructor","iframeWindow","targetOrigin","handleMessage","bind","window","addEventListener","requestContext","event","MessageEvent","origin","source","data","channel","message","updateContext","context","pending","get","id","result","delete","serializedContext","execute","args","abortSignal","callTool","toolName","Promise","sendRequest","T","timeout","timeoutMessage","aborted","timeoutId","ReturnType","setTimeout","onAbort","cleanup","clearTimeout","removeEventListener","set","once","postMessage","getModelContext","subscribe","callback","add","dispose","clear","values"],"sources":["../../../src/model-context/frame/host.ts"],"sourcesContent":["import type { ModelContextProvider, ModelContext } from \"../types\";\nimport type { Unsubscribe } from \"../../types/unsubscribe\";\nimport type { Tool } from \"assistant-stream\";\nimport { notifySubscribers as notifyStateSubscribers } from \"../../subscribable/subscribable\";\nimport {\n  type FrameMessage,\n  FRAME_MESSAGE_CHANNEL,\n  type SerializedModelContext,\n  type SerializedTool,\n} from \"./types\";\n\n/**\n * Deserializes tools from JSON Schema format back to Tool objects\n */\nconst deserializeTool = (serializedTool: SerializedTool): Tool<any, any> =>\n  ({\n    parameters: serializedTool.parameters,\n    ...(serializedTool.description && {\n      description: serializedTool.description,\n    }),\n    ...(serializedTool.disabled !== undefined && {\n      disabled: serializedTool.disabled,\n    }),\n    ...(serializedTool.type && { type: serializedTool.type }),\n  }) as Tool<any, any>;\n\n/**\n * Deserializes a ModelContext from transmission format\n */\nconst deserializeModelContext = (\n  serialized: SerializedModelContext,\n): ModelContext => ({\n  ...(serialized.system !== undefined && { system: serialized.system }),\n  ...(serialized.tools && {\n    tools: Object.fromEntries(\n      Object.entries(serialized.tools).map(([name, tool]) => [\n        name,\n        deserializeTool(tool),\n      ]),\n    ),\n  }),\n});\n\nconst getAbortReason = (signal: AbortSignal): unknown => {\n  if (signal.reason !== undefined) return signal.reason;\n  const error = new Error(\"Tool call was aborted\");\n  error.name = \"AbortError\";\n  return error;\n};\n\nexport class AssistantFrameHost implements ModelContextProvider {\n  private _context: ModelContext = {};\n  private _subscribers = new Set<() => void>();\n  private _pendingRequests = new Map<\n    string,\n    {\n      resolve: (value: any) => void;\n      reject: (error: any) => void;\n    }\n  >();\n  private _requestCounter = 0;\n  private _iframeWindow: Window;\n  private _targetOrigin: string;\n  private _disposed = false;\n\n  constructor(iframeWindow: Window, targetOrigin: string = \"*\") {\n    this._iframeWindow = iframeWindow;\n    this._targetOrigin = targetOrigin;\n\n    this.handleMessage = this.handleMessage.bind(this);\n    window.addEventListener(\"message\", this.handleMessage);\n\n    this.requestContext();\n  }\n\n  private handleMessage(event: MessageEvent) {\n    if (this._targetOrigin !== \"*\" && event.origin !== this._targetOrigin)\n      return;\n    if (event.source !== this._iframeWindow) return;\n    if (event.data?.channel !== FRAME_MESSAGE_CHANNEL) return;\n\n    const message = event.data.message as FrameMessage;\n\n    switch (message.type) {\n      case \"model-context-update\": {\n        this.updateContext(message.context);\n        break;\n      }\n\n      case \"tool-result\": {\n        const pending = this._pendingRequests.get(message.id);\n        if (pending) {\n          if (message.error) {\n            pending.reject(new Error(message.error));\n          } else {\n            pending.resolve(message.result);\n          }\n          this._pendingRequests.delete(message.id);\n        }\n        break;\n      }\n    }\n  }\n\n  private updateContext(serializedContext: SerializedModelContext) {\n    const context = deserializeModelContext(serializedContext);\n    this._context = {\n      ...context,\n      tools:\n        context.tools &&\n        Object.fromEntries(\n          Object.entries(context.tools).map(([name, tool]) => [\n            name,\n            {\n              ...tool,\n              execute: (args: any, context: { abortSignal: AbortSignal }) =>\n                this.callTool(name, args, context.abortSignal),\n            } as Tool<any, any>,\n          ]),\n        ),\n    };\n    this.notifySubscribers();\n  }\n\n  private callTool(\n    toolName: string,\n    args: any,\n    abortSignal: AbortSignal,\n  ): Promise<any> {\n    return this.sendRequest(\n      {\n        type: \"tool-call\",\n        id: `tool-${this._requestCounter++}`,\n        toolName,\n        args,\n      },\n      30000,\n      `Tool call \"${toolName}\" timed out`,\n      abortSignal,\n    );\n  }\n\n  private sendRequest<T extends FrameMessage & { id: string }>(\n    message: T,\n    timeout = 30000,\n    timeoutMessage = \"Request timed out\",\n    abortSignal?: AbortSignal,\n  ): Promise<any> {\n    if (this._disposed) {\n      return Promise.reject(new Error(\"AssistantFrameHost has been disposed\"));\n    }\n    if (abortSignal?.aborted) {\n      return Promise.reject(getAbortReason(abortSignal));\n    }\n\n    return new Promise((resolve, reject) => {\n      let timeoutId: ReturnType<typeof setTimeout> | undefined;\n      const onAbort = () => {\n        if (!abortSignal) return;\n        const pending = this._pendingRequests.get(message.id);\n        if (pending) {\n          pending.reject(getAbortReason(abortSignal));\n          this._pendingRequests.delete(message.id);\n        }\n      };\n      const cleanup = () => {\n        if (timeoutId !== undefined) clearTimeout(timeoutId);\n        abortSignal?.removeEventListener(\"abort\", onAbort);\n      };\n\n      this._pendingRequests.set(message.id, {\n        resolve: (value: any) => {\n          cleanup();\n          resolve(value);\n        },\n        reject: (error: any) => {\n          cleanup();\n          reject(error);\n        },\n      });\n\n      timeoutId = setTimeout(() => {\n        const pending = this._pendingRequests.get(message.id);\n        if (pending) {\n          pending.reject(new Error(timeoutMessage));\n          this._pendingRequests.delete(message.id);\n        }\n      }, timeout);\n      abortSignal?.addEventListener(\"abort\", onAbort, { once: true });\n\n      this._iframeWindow.postMessage(\n        { channel: FRAME_MESSAGE_CHANNEL, message },\n        this._targetOrigin,\n      );\n    });\n  }\n\n  private requestContext() {\n    this._iframeWindow.postMessage(\n      {\n        channel: FRAME_MESSAGE_CHANNEL,\n        message: {\n          type: \"model-context-request\",\n        } as FrameMessage,\n      },\n      this._targetOrigin,\n    );\n  }\n\n  private notifySubscribers() {\n    notifyStateSubscribers(this._subscribers);\n  }\n\n  getModelContext(): ModelContext {\n    return this._context;\n  }\n\n  subscribe(callback: () => void): Unsubscribe {\n    this._subscribers.add(callback);\n    return () => this._subscribers.delete(callback);\n  }\n\n  dispose() {\n    this._disposed = true;\n    window.removeEventListener(\"message\", this.handleMessage);\n    this._subscribers.clear();\n    const error = new Error(\"AssistantFrameHost has been disposed\");\n    for (const pending of this._pendingRequests.values()) {\n      pending.reject(error);\n    }\n    this._pendingRequests.clear();\n  }\n}\n"],"mappings":";;;;;;AAcA,MAAMU,mBAAmBC,oBACtB;CACCC,YAAYD,eAAeC;CAC3B,GAAID,eAAeE,eAAe,EAChCA,aAAaF,eAAeE,YAC9B;CACA,GAAIF,eAAeG,aAAaC,KAAAA,KAAa,EAC3CD,UAAUH,eAAeG,SAC3B;CACA,GAAIH,eAAeK,QAAQ,EAAEA,MAAML,eAAeK,KAAK;AACzD;;;;AAKF,MAAMC,2BACJC,gBACkB;CAClB,GAAIA,WAAWC,WAAWJ,KAAAA,KAAa,EAAEI,QAAQD,WAAWC,OAAO;CACnE,GAAID,WAAWE,SAAS,EACtBA,OAAOC,OAAOC,YACZD,OAAOE,QAAQL,WAAWE,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMC,UAAU,CACrDD,MACAf,gBAAgBgB,IAAI,CAAC,CACtB,CACH,EACF;AACF;AAEA,MAAMC,kBAAkBC,WAAiC;CACvD,IAAIA,OAAOE,WAAWf,KAAAA,GAAW,OAAOa,OAAOE;CAC/C,MAAMC,wBAAQ,IAAIC,MAAM,uBAAuB;CAC/CD,MAAMN,OAAO;CACb,OAAOM;AACT;AAEA,IAAaE,qBAAb,MAAgE;CAC9D,WAAiC,CAAC;CAClC,+BAAuB,IAAIG,IAAgB;CAC3C,mCAA2B,IAAIE,IAM7B;CACF,kBAA0B;CAC1B;CACA;CACA,YAAoB;CAEpBS,YAAYC,cAAsBC,eAAuB,KAAK;EAC5D,KAAKN,gBAAgBK;EACrB,KAAKH,gBAAgBI;EAErB,KAAKC,gBAAgB,KAAKA,cAAcC,KAAK,IAAI;EACjDC,OAAOC,iBAAiB,WAAW,KAAKH,aAAa;EAErD,KAAKI,eAAe;CACtB;CAEA,cAAsBC,OAAqB;EACzC,IAAI,KAAKV,kBAAkB,OAAOU,MAAME,WAAW,KAAKZ,eACtD;EACF,IAAIU,MAAMG,WAAW,KAAKf,eAAe;EACzC,IAAIY,MAAMI,MAAMC,YAAAA,sBAAmC;EAEnD,MAAMC,UAAUN,MAAMI,KAAKE;EAE3B,QAAQA,QAAQ7C,MAAhB;GACE,KAAK;IACH,KAAK8C,cAAcD,QAAQE,OAAO;IAClC;GAGF,KAAK,eAAe;IAClB,MAAMC,UAAU,KAAK3B,iBAAiB4B,IAAIJ,QAAQK,EAAE;IACpD,IAAIF,SAAS;KACX,IAAIH,QAAQ9B,OACViC,QAAQvB,OAAO,IAAIT,MAAM6B,QAAQ9B,KAAK,CAAC;UAEvCiC,QAAQzB,QAAQsB,QAAQM,MAAM;KAEhC,KAAK9B,iBAAiB+B,OAAOP,QAAQK,EAAE;IACzC;IACA;GACF;EACF;CACF;CAEA,cAAsBG,mBAA2C;EAC/D,MAAMN,UAAU9C,wBAAwBoD,iBAAiB;EACzD,KAAKnC,WAAW;GACd,GAAG6B;GACH3C,OACE2C,QAAQ3C,SACRC,OAAOC,YACLD,OAAOE,QAAQwC,QAAQ3C,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMC,UAAU,CAClDD,MACA;IACE,GAAGC;IACH4C,UAAUC,MAAWR,YACnB,KAAKU,SAAShD,MAAM8C,MAAMR,QAAQS,WAAW;GACjD,CAAmB,CACpB,CACH;EACJ;EACA,KAAKpE,kBAAkB;CACzB;CAEA,SACEsE,UACAH,MACAC,aACc;EACd,OAAO,KAAKI,YACV;GACE5D,MAAM;GACNkD,IAAI,QAAQ,KAAKxB;GACjBgC;GACAH;EACF,GACA,KACA,cAAcG,SAAQ,cACtBF,WACF;CACF;CAEA,YACEX,SACAiB,UAAU,KACVC,iBAAiB,qBACjBP,aACc;EACd,IAAI,KAAK1B,WACP,OAAO6B,QAAQlC,uBAAO,IAAIT,MAAM,sCAAsC,CAAC;EAEzE,IAAIwC,aAAaQ,SACf,OAAOL,QAAQlC,OAAOd,eAAe6C,WAAW,CAAC;EAGnD,OAAO,IAAIG,SAASpC,SAASE,WAAW;GACtC,IAAIwC;GACJ,MAAMG,gBAAgB;IACpB,IAAI,CAACZ,aAAa;IAClB,MAAMR,UAAU,KAAK3B,iBAAiB4B,IAAIJ,QAAQK,EAAE;IACpD,IAAIF,SAAS;KACXA,QAAQvB,OAAOd,eAAe6C,WAAW,CAAC;KAC1C,KAAKnC,iBAAiB+B,OAAOP,QAAQK,EAAE;IACzC;GACF;GACA,MAAMmB,gBAAgB;IACpB,IAAIJ,cAAclE,KAAAA,GAAWuE,aAAaL,SAAS;IACnDT,aAAae,oBAAoB,SAASH,OAAO;GACnD;GAEA,KAAK/C,iBAAiBmD,IAAI3B,QAAQK,IAAI;IACpC3B,UAAUC,UAAe;KACvB6C,QAAQ;KACR9C,QAAQC,KAAK;IACf;IACAC,SAASV,UAAe;KACtBsD,QAAQ;KACR5C,OAAOV,KAAK;IACd;GACF,CAAC;GAEDkD,YAAYE,iBAAiB;IAC3B,MAAMnB,UAAU,KAAK3B,iBAAiB4B,IAAIJ,QAAQK,EAAE;IACpD,IAAIF,SAAS;KACXA,QAAQvB,OAAO,IAAIT,MAAM+C,cAAc,CAAC;KACxC,KAAK1C,iBAAiB+B,OAAOP,QAAQK,EAAE;IACzC;GACF,GAAGY,OAAO;GACVN,aAAanB,iBAAiB,SAAS+B,SAAS,EAAEK,MAAM,KAAK,CAAC;GAE9D,KAAK9C,cAAc+C,YACjB;IAAE9B,SAASrD;IAAuBsD;GAAQ,GAC1C,KAAKhB,aACP;EACF,CAAC;CACH;CAEA,iBAAyB;EACvB,KAAKF,cAAc+C,YACjB;GACE9B,SAASrD;GACTsD,SAAS,EACP7C,MAAM,wBACR;EACF,GACA,KAAK6B,aACP;CACF;CAEA,oBAA4B;EAC1BxC,kBAAuB,KAAK8B,YAAY;CAC1C;CAEAwD,kBAAgC;EAC9B,OAAO,KAAKzD;CACd;CAEA0D,UAAUC,UAAmC;EAC3C,KAAK1D,aAAa2D,IAAID,QAAQ;EAC9B,aAAa,KAAK1D,aAAaiC,OAAOyB,QAAQ;CAChD;CAEAE,UAAU;EACR,KAAKjD,YAAY;EACjBM,OAAOmC,oBAAoB,WAAW,KAAKrC,aAAa;EACxD,KAAKf,aAAa6D,MAAM;EACxB,MAAMjE,wBAAQ,IAAIC,MAAM,sCAAsC;EAC9D,KAAK,MAAMgC,WAAW,KAAK3B,iBAAiB4D,OAAO,GACjDjC,QAAQvB,OAAOV,KAAK;EAEtB,KAAKM,iBAAiB2D,MAAM;CAC9B;AACF"}
+{"version":3,"file":"host.js","names":["ModelContextProvider","ModelContext","Unsubscribe","Tool","notifySubscribers","notifyStateSubscribers","generateId","FrameMessage","FRAME_MESSAGE_CHANNEL","SerializedModelContext","SerializedTool","getDefaultTargetOrigin","window","location","origin","deserializeTool","serializedTool","parameters","description","disabled","undefined","type","deserializeModelContext","serialized","system","tools","Object","fromEntries","entries","map","name","tool","getAbortReason","signal","AbortSignal","reason","error","Error","AssistantFrameHost","_context","_subscribers","Set","_pendingRequests","Map","resolve","value","reject","_iframeWindow","Window","_targetOrigin","_disposed","constructor","iframeWindow","targetOrigin","handleMessage","bind","addEventListener","requestContext","event","MessageEvent","source","data","channel","message","updateContext","context","pending","get","id","result","delete","serializedContext","execute","args","abortSignal","callTool","toolName","Promise","sendRequest","T","timeout","timeoutMessage","aborted","timeoutId","ReturnType","setTimeout","onAbort","cancelToolCall","cleanup","clearTimeout","removeEventListener","set","once","postMessage","getModelContext","subscribe","callback","add","dispose","clear"],"sources":["../../../src/model-context/frame/host.ts"],"sourcesContent":["import type { ModelContextProvider, ModelContext } from \"../types\";\nimport type { Unsubscribe } from \"../../types/unsubscribe\";\nimport type { Tool } from \"assistant-stream\";\nimport { notifySubscribers as notifyStateSubscribers } from \"../../subscribable/subscribable\";\nimport { generateId } from \"../../utils/id\";\nimport {\n  type FrameMessage,\n  FRAME_MESSAGE_CHANNEL,\n  type SerializedModelContext,\n  type SerializedTool,\n} from \"./types\";\n\nconst getDefaultTargetOrigin = () => window.location.origin;\n\n/**\n * Deserializes tools from JSON Schema format back to Tool objects\n */\nconst deserializeTool = (serializedTool: SerializedTool): Tool<any, any> =>\n  ({\n    parameters: serializedTool.parameters,\n    ...(serializedTool.description && {\n      description: serializedTool.description,\n    }),\n    ...(serializedTool.disabled !== undefined && {\n      disabled: serializedTool.disabled,\n    }),\n    ...(serializedTool.type && { type: serializedTool.type }),\n  }) as Tool<any, any>;\n\n/**\n * Deserializes a ModelContext from transmission format\n */\nconst deserializeModelContext = (\n  serialized: SerializedModelContext,\n): ModelContext => ({\n  ...(serialized.system !== undefined && { system: serialized.system }),\n  ...(serialized.tools && {\n    tools: Object.fromEntries(\n      Object.entries(serialized.tools).map(([name, tool]) => [\n        name,\n        deserializeTool(tool),\n      ]),\n    ),\n  }),\n});\n\nconst getAbortReason = (signal: AbortSignal): unknown => {\n  if (signal.reason !== undefined) return signal.reason;\n  const error = new Error(\"Tool call was aborted\");\n  error.name = \"AbortError\";\n  return error;\n};\n\nexport class AssistantFrameHost implements ModelContextProvider {\n  private _context: ModelContext = {};\n  private _subscribers = new Set<() => void>();\n  private _pendingRequests = new Map<\n    string,\n    {\n      resolve: (value: any) => void;\n      reject: (error: any) => void;\n    }\n  >();\n  private _iframeWindow: Window;\n  private _targetOrigin: string;\n  private _disposed = false;\n\n  constructor(\n    iframeWindow: Window,\n    targetOrigin: string = getDefaultTargetOrigin(),\n  ) {\n    this._iframeWindow = iframeWindow;\n    this._targetOrigin = targetOrigin;\n\n    this.handleMessage = this.handleMessage.bind(this);\n    window.addEventListener(\"message\", this.handleMessage);\n\n    this.requestContext();\n  }\n\n  private handleMessage(event: MessageEvent) {\n    if (this._targetOrigin !== \"*\" && event.origin !== this._targetOrigin)\n      return;\n    if (event.source !== this._iframeWindow) return;\n    if (event.data?.channel !== FRAME_MESSAGE_CHANNEL) return;\n\n    const message = event.data.message as FrameMessage;\n\n    switch (message.type) {\n      case \"model-context-update\": {\n        this.updateContext(message.context);\n        break;\n      }\n\n      case \"tool-result\": {\n        const pending = this._pendingRequests.get(message.id);\n        if (pending) {\n          if (typeof message.error === \"string\") {\n            pending.reject(new Error(message.error));\n          } else {\n            pending.resolve(message.result);\n          }\n          this._pendingRequests.delete(message.id);\n        }\n        break;\n      }\n    }\n  }\n\n  private updateContext(serializedContext: SerializedModelContext) {\n    const context = deserializeModelContext(serializedContext);\n    this._context = {\n      ...context,\n      tools:\n        context.tools &&\n        Object.fromEntries(\n          Object.entries(context.tools).map(([name, tool]) => [\n            name,\n            {\n              ...tool,\n              execute: (args: any, context: { abortSignal: AbortSignal }) =>\n                this.callTool(name, args, context.abortSignal),\n            } as Tool<any, any>,\n          ]),\n        ),\n    };\n    this.notifySubscribers();\n  }\n\n  private callTool(\n    toolName: string,\n    args: any,\n    abortSignal: AbortSignal,\n  ): Promise<any> {\n    return this.sendRequest(\n      {\n        type: \"tool-call\",\n        id: `tool-${generateId()}`,\n        toolName,\n        args,\n      },\n      30000,\n      `Tool call \"${toolName}\" timed out`,\n      abortSignal,\n    );\n  }\n\n  private sendRequest<T extends FrameMessage & { id: string }>(\n    message: T,\n    timeout = 30000,\n    timeoutMessage = \"Request timed out\",\n    abortSignal?: AbortSignal,\n  ): Promise<any> {\n    if (this._disposed) {\n      return Promise.reject(new Error(\"AssistantFrameHost has been disposed\"));\n    }\n    if (abortSignal?.aborted) {\n      return Promise.reject(getAbortReason(abortSignal));\n    }\n\n    return new Promise((resolve, reject) => {\n      let timeoutId: ReturnType<typeof setTimeout> | undefined;\n      const onAbort = () => {\n        if (!abortSignal) return;\n        const pending = this._pendingRequests.get(message.id);\n        if (pending) {\n          this.cancelToolCall(message.id);\n          pending.reject(getAbortReason(abortSignal));\n          this._pendingRequests.delete(message.id);\n        }\n      };\n      const cleanup = () => {\n        if (timeoutId !== undefined) clearTimeout(timeoutId);\n        abortSignal?.removeEventListener(\"abort\", onAbort);\n      };\n\n      this._pendingRequests.set(message.id, {\n        resolve: (value: any) => {\n          cleanup();\n          resolve(value);\n        },\n        reject: (error: any) => {\n          cleanup();\n          reject(error);\n        },\n      });\n\n      timeoutId = setTimeout(() => {\n        const pending = this._pendingRequests.get(message.id);\n        if (pending) {\n          this.cancelToolCall(message.id);\n          pending.reject(new Error(timeoutMessage));\n          this._pendingRequests.delete(message.id);\n        }\n      }, timeout);\n      abortSignal?.addEventListener(\"abort\", onAbort, { once: true });\n\n      this._iframeWindow.postMessage(\n        { channel: FRAME_MESSAGE_CHANNEL, message },\n        this._targetOrigin,\n      );\n    });\n  }\n\n  private cancelToolCall(id: string) {\n    this._iframeWindow.postMessage(\n      {\n        channel: FRAME_MESSAGE_CHANNEL,\n        message: { type: \"tool-cancel\", id } satisfies FrameMessage,\n      },\n      this._targetOrigin,\n    );\n  }\n\n  private requestContext() {\n    this._iframeWindow.postMessage(\n      {\n        channel: FRAME_MESSAGE_CHANNEL,\n        message: {\n          type: \"model-context-request\",\n        } as FrameMessage,\n      },\n      this._targetOrigin,\n    );\n  }\n\n  private notifySubscribers() {\n    notifyStateSubscribers(this._subscribers);\n  }\n\n  getModelContext(): ModelContext {\n    return this._context;\n  }\n\n  subscribe(callback: () => void): Unsubscribe {\n    this._subscribers.add(callback);\n    return () => this._subscribers.delete(callback);\n  }\n\n  dispose() {\n    this._disposed = true;\n    window.removeEventListener(\"message\", this.handleMessage);\n    this._subscribers.clear();\n    const error = new Error(\"AssistantFrameHost has been disposed\");\n    for (const [id, pending] of this._pendingRequests) {\n      this.cancelToolCall(id);\n      pending.reject(error);\n    }\n    this._pendingRequests.clear();\n  }\n}\n"],"mappings":";;;;AAYA,MAAMW,+BAA+BC,OAAOC,SAASC;;;;AAKrD,MAAMC,mBAAmBC,oBACtB;CACCC,YAAYD,eAAeC;CAC3B,GAAID,eAAeE,eAAe,EAChCA,aAAaF,eAAeE,YAC9B;CACA,GAAIF,eAAeG,aAAaC,KAAAA,KAAa,EAC3CD,UAAUH,eAAeG,SAC3B;CACA,GAAIH,eAAeK,QAAQ,EAAEA,MAAML,eAAeK,KAAK;AACzD;;;;AAKF,MAAMC,2BACJC,gBACkB;CAClB,GAAIA,WAAWC,WAAWJ,KAAAA,KAAa,EAAEI,QAAQD,WAAWC,OAAO;CACnE,GAAID,WAAWE,SAAS,EACtBA,OAAOC,OAAOC,YACZD,OAAOE,QAAQL,WAAWE,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMC,UAAU,CACrDD,MACAf,gBAAgBgB,IAAI,CAAC,CACtB,CACH,EACF;AACF;AAEA,MAAMC,kBAAkBC,WAAiC;CACvD,IAAIA,OAAOE,WAAWf,KAAAA,GAAW,OAAOa,OAAOE;CAC/C,MAAMC,wBAAQ,IAAIC,MAAM,uBAAuB;CAC/CD,MAAMN,OAAO;CACb,OAAOM;AACT;AAEA,IAAaE,qBAAb,MAAgE;CAC9D,WAAiC,CAAC;CAClC,+BAAuB,IAAIG,IAAgB;CAC3C,mCAA2B,IAAIE,IAM7B;CACF;CACA;CACA,YAAoB;CAEpBQ,YACEC,cACAC,eAAuB1C,uBAAuB,GAC9C;EACA,KAAKoC,gBAAgBK;EACrB,KAAKH,gBAAgBI;EAErB,KAAKC,gBAAgB,KAAKA,cAAcC,KAAK,IAAI;EACjD3C,OAAO4C,iBAAiB,WAAW,KAAKF,aAAa;EAErD,KAAKG,eAAe;CACtB;CAEA,cAAsBC,OAAqB;EACzC,IAAI,KAAKT,kBAAkB,OAAOS,MAAM5C,WAAW,KAAKmC,eACtD;EACF,IAAIS,MAAME,WAAW,KAAKb,eAAe;EACzC,IAAIW,MAAMG,MAAMC,YAAAA,sBAAmC;EAEnD,MAAMC,UAAUL,MAAMG,KAAKE;EAE3B,QAAQA,QAAQ1C,MAAhB;GACE,KAAK;IACH,KAAK2C,cAAcD,QAAQE,OAAO;IAClC;GAGF,KAAK,eAAe;IAClB,MAAMC,UAAU,KAAKxB,iBAAiByB,IAAIJ,QAAQK,EAAE;IACpD,IAAIF,SAAS;KACX,IAAI,OAAOH,QAAQ3B,UAAU,UAC3B8B,QAAQpB,OAAO,IAAIT,MAAM0B,QAAQ3B,KAAK,CAAC;UAEvC8B,QAAQtB,QAAQmB,QAAQM,MAAM;KAEhC,KAAK3B,iBAAiB4B,OAAOP,QAAQK,EAAE;IACzC;IACA;GACF;EACF;CACF;CAEA,cAAsBG,mBAA2C;EAC/D,MAAMN,UAAU3C,wBAAwBiD,iBAAiB;EACzD,KAAKhC,WAAW;GACd,GAAG0B;GACHxC,OACEwC,QAAQxC,SACRC,OAAOC,YACLD,OAAOE,QAAQqC,QAAQxC,KAAK,CAAC,CAACI,KAAK,CAACC,MAAMC,UAAU,CAClDD,MACA;IACE,GAAGC;IACHyC,UAAUC,MAAWR,YACnB,KAAKU,SAAS7C,MAAM2C,MAAMR,QAAQS,WAAW;GACjD,CAAmB,CACpB,CACH;EACJ;EACA,KAAKtE,kBAAkB;CACzB;CAEA,SACEwE,UACAH,MACAC,aACc;EACd,OAAO,KAAKI,YACV;GACEzD,MAAM;GACN+C,IAAI,QAAQ9D,WAAW;GACvBsE;GACAH;EACF,GACA,KACA,cAAcG,SAAQ,cACtBF,WACF;CACF;CAEA,YACEX,SACAiB,UAAU,KACVC,iBAAiB,qBACjBP,aACc;EACd,IAAI,KAAKxB,WACP,OAAO2B,QAAQ/B,uBAAO,IAAIT,MAAM,sCAAsC,CAAC;EAEzE,IAAIqC,aAAaQ,SACf,OAAOL,QAAQ/B,OAAOd,eAAe0C,WAAW,CAAC;EAGnD,OAAO,IAAIG,SAASjC,SAASE,WAAW;GACtC,IAAIqC;GACJ,MAAMG,gBAAgB;IACpB,IAAI,CAACZ,aAAa;IAClB,MAAMR,UAAU,KAAKxB,iBAAiByB,IAAIJ,QAAQK,EAAE;IACpD,IAAIF,SAAS;KACX,KAAKqB,eAAexB,QAAQK,EAAE;KAC9BF,QAAQpB,OAAOd,eAAe0C,WAAW,CAAC;KAC1C,KAAKhC,iBAAiB4B,OAAOP,QAAQK,EAAE;IACzC;GACF;GACA,MAAMoB,gBAAgB;IACpB,IAAIL,cAAc/D,KAAAA,GAAWqE,aAAaN,SAAS;IACnDT,aAAagB,oBAAoB,SAASJ,OAAO;GACnD;GAEA,KAAK5C,iBAAiBiD,IAAI5B,QAAQK,IAAI;IACpCxB,UAAUC,UAAe;KACvB2C,QAAQ;KACR5C,QAAQC,KAAK;IACf;IACAC,SAASV,UAAe;KACtBoD,QAAQ;KACR1C,OAAOV,KAAK;IACd;GACF,CAAC;GAED+C,YAAYE,iBAAiB;IAC3B,MAAMnB,UAAU,KAAKxB,iBAAiByB,IAAIJ,QAAQK,EAAE;IACpD,IAAIF,SAAS;KACX,KAAKqB,eAAexB,QAAQK,EAAE;KAC9BF,QAAQpB,OAAO,IAAIT,MAAM4C,cAAc,CAAC;KACxC,KAAKvC,iBAAiB4B,OAAOP,QAAQK,EAAE;IACzC;GACF,GAAGY,OAAO;GACVN,aAAalB,iBAAiB,SAAS8B,SAAS,EAAEM,MAAM,KAAK,CAAC;GAE9D,KAAK7C,cAAc8C,YACjB;IAAE/B,SAAStD;IAAuBuD;GAAQ,GAC1C,KAAKd,aACP;EACF,CAAC;CACH;CAEA,eAAuBmB,IAAY;EACjC,KAAKrB,cAAc8C,YACjB;GACE/B,SAAStD;GACTuD,SAAS;IAAE1C,MAAM;IAAe+C;GAAG;EACrC,GACA,KAAKnB,aACP;CACF;CAEA,iBAAyB;EACvB,KAAKF,cAAc8C,YACjB;GACE/B,SAAStD;GACTuD,SAAS,EACP1C,MAAM,wBACR;EACF,GACA,KAAK4B,aACP;CACF;CAEA,oBAA4B;EAC1B5C,kBAAuB,KAAKmC,YAAY;CAC1C;CAEAsD,kBAAgC;EAC9B,OAAO,KAAKvD;CACd;CAEAwD,UAAUC,UAAmC;EAC3C,KAAKxD,aAAayD,IAAID,QAAQ;EAC9B,aAAa,KAAKxD,aAAa8B,OAAO0B,QAAQ;CAChD;CAEAE,UAAU;EACR,KAAKhD,YAAY;EACjBtC,OAAO8E,oBAAoB,WAAW,KAAKpC,aAAa;EACxD,KAAKd,aAAa2D,MAAM;EACxB,MAAM/D,wBAAQ,IAAIC,MAAM,sCAAsC;EAC9D,KAAK,MAAM,CAAC+B,IAAIF,YAAY,KAAKxB,kBAAkB;GACjD,KAAK6C,eAAenB,EAAE;GACtBF,QAAQpB,OAAOV,KAAK;EACtB;EACA,KAAKM,iBAAiByD,MAAM;CAC9B;AACF"}

package/dist/model-context/frame/provider.d.ts

@@ -5,14 +5,20 @@ declare class AssistantFrameProvider {
   private static _instance;
   private _providers;
   private _providerUnsubscribes;
+  private _activeToolCalls;
   private _targetOrigin;
+  private _strictRegistrations;
+  private _wildcardRegistrations;
   private constructor();
   private static getInstance;
+  private reconcileTargetOrigin;
   private handleMessage;
   private handleToolCall;
+  private cancelToolCall;
   private sendMessage;
   private getModelContext;
   private broadcastUpdate;
+  private removeProvider;
   static addModelContextProvider(provider: ModelContextProvider, targetOrigin?: string): Unsubscribe;
   static dispose(): void;
 }

package/dist/model-context/frame/provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"provider.d.ts","names":[],"sources":["../../../src/model-context/frame/provider.ts"],"mappings":";;;cA+Ba;iBACI;UAEP;UACA;UAIA;UAED;iBAQQ;UASP;UAsBM;UAoCN;UAOA;UAkBA;SAcD,wBACL,UAAU,sBACV,wBACC;SAmBI"}
+{"version":3,"file":"provider.d.ts","names":[],"sources":["../../../src/model-context/frame/provider.ts"],"mappings":";;;cAiCa;iBACI;UAEP;UACA;UACA;UAIA;UACA;UACA;UAED;iBAQQ;UAWP;UAsBA;UA0BM;UA2CN;UAOA;UAOA;UAkBA;UAcA;SAsBD,wBACL,UAAU,sBACV,wBACC;SA2DI"}

package/dist/model-context/frame/provider.js

@@ -11,12 +11,16 @@ const serializeModelContext = (context) => ({
 	...context.system !== void 0 && { system: context.system },
 	...context.tools && { tools: Object.fromEntries(Object.entries(context.tools).map(([name, tool]) => [name, serializeTool(tool)])) }
 });
+const getDefaultTargetOrigin = () => window.location.origin;
 var AssistantFrameProvider = class AssistantFrameProvider {
 	static _instance = null;
-	_providers = /* @__PURE__ */ new Set();
+	_providers = /* @__PURE__ */ new Map();
 	_providerUnsubscribes = /* @__PURE__ */ new Map();
+	_activeToolCalls = /* @__PURE__ */ new Map();
 	_targetOrigin;
-	constructor(targetOrigin = "*") {
+	_strictRegistrations = 0;
+	_wildcardRegistrations = 0;
+	constructor(targetOrigin = getDefaultTargetOrigin()) {
 		this._targetOrigin = targetOrigin;
 		this.handleMessage = this.handleMessage.bind(this);
 		window.addEventListener("message", this.handleMessage);
@@ -24,8 +28,22 @@ var AssistantFrameProvider = class AssistantFrameProvider {
 	}
 	static getInstance(targetOrigin) {
 		if (!AssistantFrameProvider._instance) AssistantFrameProvider._instance = new AssistantFrameProvider(targetOrigin);
+		else AssistantFrameProvider._instance.reconcileTargetOrigin(targetOrigin);
 		return AssistantFrameProvider._instance;
 	}
+	reconcileTargetOrigin(targetOrigin = getDefaultTargetOrigin()) {
+		if (targetOrigin === this._targetOrigin) return;
+		if (this._providers.size === 0) {
+			this._targetOrigin = targetOrigin;
+			return;
+		}
+		if (targetOrigin === "*") return;
+		if (this._targetOrigin === "*") {
+			this._targetOrigin = targetOrigin;
+			return;
+		}
+		throw new Error(`AssistantFrameProvider cannot register conflicting target origins: "${this._targetOrigin}" and "${targetOrigin}"`);
+	}
 	handleMessage(event) {
 		if (this._targetOrigin !== "*" && event.origin !== this._targetOrigin) return;
 		if (event.source !== window.parent) return;
@@ -38,18 +56,28 @@ var AssistantFrameProvider = class AssistantFrameProvider {
 					context: serializeModelContext(this.getModelContext())
 				});
 				break;
-			case "tool-call": this.handleToolCall(message, event);
+			case "tool-call":
+				this.handleToolCall(message, event);
+				break;
+			case "tool-cancel": this.cancelToolCall(message.id);
 		}
 	}
 	async handleToolCall(message, event) {
 		const tool = this.getModelContext().tools?.[message.toolName];
+		const abortController = new AbortController();
+		this._activeToolCalls.get(message.id)?.abortController.abort();
+		const activeCall = {
+			abortController,
+			event
+		};
+		this._activeToolCalls.set(message.id, activeCall);
 		let result;
 		let error;
 		if (!tool) error = `Tool "${message.toolName}" not found`;
 		else try {
 			result = tool.execute ? await tool.execute(message.args, {
 				toolCallId: message.id,
-				abortSignal: new AbortController().signal,
+				abortSignal: abortController.signal,
 				human: async () => {
 					throw new Error("Tool human input is not supported in frame context");
 				}
@@ -57,12 +85,20 @@ var AssistantFrameProvider = class AssistantFrameProvider {
 		} catch (e) {
 			error = e instanceof Error ? e.message : String(e);
 		}
+		if (this._activeToolCalls.get(message.id) !== activeCall) return;
+		this._activeToolCalls.delete(message.id);
 		this.sendMessage(event, {
 			type: "tool-result",
 			id: message.id,
-			...error ? { error } : { result }
+			...error !== void 0 ? { error } : { result }
 		});
 	}
+	cancelToolCall(id) {
+		const activeCall = this._activeToolCalls.get(id);
+		if (!activeCall) return;
+		this._activeToolCalls.delete(id);
+		activeCall.abortController.abort();
+	}
 	sendMessage(event, message) {
 		event.source?.postMessage({
 			channel: FRAME_MESSAGE_CHANNEL,
@@ -70,7 +106,7 @@ var AssistantFrameProvider = class AssistantFrameProvider {
 		}, { targetOrigin: event.origin });
 	}
 	getModelContext() {
-		return Array.from(this._providers).map((p) => p.getModelContext()).reduce((merged, context) => ({
+		return Array.from(new Set(this._providers.values())).map((p) => p.getModelContext()).reduce((merged, context) => ({
 			system: context.system ? merged.system ? `${merged.system}\n\n${context.system}` : context.system : merged.system,
 			tools: {
 				...merged.tools || {},
@@ -90,27 +126,101 @@ var AssistantFrameProvider = class AssistantFrameProvider {
 			}, this._targetOrigin);
 		}
 	}
+	removeProvider(id, origin) {
+		this._providers.delete(id);
+		const unsubscribe = this._providerUnsubscribes.get(id);
+		this._providerUnsubscribes.delete(id);
+		if (origin === "*") {
+			this._wildcardRegistrations -= 1;
+			if (this._wildcardRegistrations === 0 && this._strictRegistrations === 0) this._targetOrigin = getDefaultTargetOrigin();
+		} else {
+			this._strictRegistrations -= 1;
+			if (this._strictRegistrations === 0) this._targetOrigin = this._wildcardRegistrations > 0 ? "*" : getDefaultTargetOrigin();
+		}
+		return unsubscribe;
+	}
 	static addModelContextProvider(provider, targetOrigin) {
-		const instance = AssistantFrameProvider.getInstance(targetOrigin);
-		instance._providers.add(provider);
-		const unsubscribe = provider.subscribe?.(() => instance.broadcastUpdate());
-		if (unsubscribe) instance._providerUnsubscribes.set(provider, unsubscribe);
-		instance.broadcastUpdate();
-		return () => {
-			instance._providers.delete(provider);
-			instance._providerUnsubscribes.get(provider)?.();
-			instance._providerUnsubscribes.delete(provider);
+		const origin = targetOrigin ?? getDefaultTargetOrigin();
+		const instance = AssistantFrameProvider.getInstance(origin);
+		const id = Symbol();
+		instance._providers.set(id, provider);
+		if (origin === "*") instance._wildcardRegistrations += 1;
+		else instance._strictRegistrations += 1;
+		try {
+			const unsubscribe = provider.subscribe?.(() => instance.broadcastUpdate());
+			if (unsubscribe) instance._providerUnsubscribes.set(id, unsubscribe);
 			instance.broadcastUpdate();
+		} catch (error) {
+			const unsubscribe = instance.removeProvider(id, origin);
+			try {
+				unsubscribe?.();
+			} catch (unsubscribeError) {
+				console.error(unsubscribeError);
+			}
+			try {
+				instance.broadcastUpdate();
+			} catch (broadcastError) {
+				console.error(broadcastError);
+			}
+			throw error;
+		}
+		let released = false;
+		return () => {
+			if (released) return;
+			released = true;
+			const unsubscribe = instance.removeProvider(id, origin);
+			let unsubscribeFailed = false;
+			let unsubscribeError;
+			try {
+				unsubscribe?.();
+			} catch (error) {
+				unsubscribeFailed = true;
+				unsubscribeError = error;
+			}
+			try {
+				instance.broadcastUpdate();
+			} catch (error) {
+				if (!unsubscribeFailed) throw error;
+				console.error(error);
+			}
+			if (unsubscribeFailed) throw unsubscribeError;
 		};
 	}
 	static dispose() {
 		if (AssistantFrameProvider._instance) {
 			const instance = AssistantFrameProvider._instance;
 			window.removeEventListener("message", instance.handleMessage);
-			instance._providerUnsubscribes.forEach((unsubscribe) => unsubscribe?.());
+			let cleanupFailed = false;
+			let cleanupError;
+			const runCleanup = (cleanup) => {
+				try {
+					cleanup();
+				} catch (error) {
+					if (cleanupFailed) console.error(error);
+					else {
+						cleanupFailed = true;
+						cleanupError = error;
+					}
+				}
+			};
+			instance._providerUnsubscribes.forEach((unsubscribe) => {
+				if (unsubscribe) runCleanup(unsubscribe);
+			});
 			instance._providerUnsubscribes.clear();
 			instance._providers.clear();
+			instance._activeToolCalls.forEach(({ abortController, event }, id) => {
+				runCleanup(() => {
+					abortController.abort();
+					instance.sendMessage(event, {
+						type: "tool-result",
+						id,
+						error: "AssistantFrameProvider has been disposed"
+					});
+				});
+			});
+			instance._activeToolCalls.clear();
 			AssistantFrameProvider._instance = null;
+			if (cleanupFailed) throw cleanupError;
 		}
 	}
 };