@assistant-ui/react 0.10.50 → 0.11.0

package/dist/types/EventTypes.d.ts

@@ -0,0 +1,49 @@
+export type EventSource<T extends keyof AssistantEvents = keyof AssistantEvents> = T extends `${infer Source}.${string}` ? Source : never;
+type ScopeConfig = {
+    composer: "thread" | "message";
+    thread: never;
+    "thread-list-item": never;
+};
+export type SourceByScope<TScope extends AssistantEventScope<keyof AssistantEvents>> = (TScope extends "*" ? EventSource : never) | (TScope extends keyof ScopeConfig ? TScope : never) | {
+    [K in keyof ScopeConfig]: TScope extends ScopeConfig[K] ? K : never;
+}[keyof ScopeConfig];
+export type AssistantEventScope<TEvent extends keyof AssistantEvents> = "*" | EventSource<TEvent> | ScopeConfig[EventSource<TEvent>];
+export type AssistantEventSelector<TEvent extends keyof AssistantEvents> = TEvent | {
+    scope: AssistantEventScope<TEvent>;
+    event: TEvent;
+};
+export type AssistantEvents = {
+    "thread.run-start": {
+        threadId: string;
+    };
+    "thread.run-end": {
+        threadId: string;
+    };
+    "thread.initialize": {
+        threadId: string;
+    };
+    "thread.model-context-update": {
+        threadId: string;
+    };
+    "composer.send": {
+        threadId: string;
+        messageId?: string;
+    };
+    "composer.attachment-add": {
+        threadId: string;
+        messageId?: string;
+    };
+    "thread-list-item.switched-to": {
+        threadId: string;
+    };
+    "thread-list-item.switched-away": {
+        threadId: string;
+    };
+};
+export declare const normalizeEventSelector: <TEvent extends keyof AssistantEvents>(selector: AssistantEventSelector<TEvent>) => {
+    scope: AssistantEventScope<TEvent>;
+    event: TEvent;
+};
+export declare const checkEventScope: <TEvent extends keyof AssistantEvents, TExpectedScope extends AssistantEventScope<keyof AssistantEvents>>(expectedScope: TExpectedScope, scope: AssistantEventScope<TEvent>, _event: TEvent) => _event is Extract<TEvent, `${SourceByScope<TExpectedScope>}.${string}`>;
+export {};
+//# sourceMappingURL=EventTypes.d.ts.map

package/dist/types/EventTypes.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"EventTypes.d.ts","sourceRoot":"","sources":["../../src/types/EventTypes.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,WAAW,CACrB,CAAC,SAAS,MAAM,eAAe,GAAG,MAAM,eAAe,IACrD,CAAC,SAAS,GAAG,MAAM,MAAM,IAAI,MAAM,EAAE,GAAG,MAAM,GAAG,KAAK,CAAC;AAE3D,KAAK,WAAW,GAAG;IACjB,QAAQ,EAAE,QAAQ,GAAG,SAAS,CAAC;IAC/B,MAAM,EAAE,KAAK,CAAC;IACd,kBAAkB,EAAE,KAAK,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,aAAa,CACvB,MAAM,SAAS,mBAAmB,CAAC,MAAM,eAAe,CAAC,IAEvD,CAAC,MAAM,SAAS,GAAG,GAAG,WAAW,GAAG,KAAK,CAAC,GAC1C,CAAC,MAAM,SAAS,MAAM,WAAW,GAAG,MAAM,GAAG,KAAK,CAAC,GACnD;KACG,CAAC,IAAI,MAAM,WAAW,GAAG,MAAM,SAAS,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK;CACpE,CAAC,MAAM,WAAW,CAAC,CAAC;AAEzB,MAAM,MAAM,mBAAmB,CAAC,MAAM,SAAS,MAAM,eAAe,IAChE,GAAG,GACH,WAAW,CAAC,MAAM,CAAC,GACnB,WAAW,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;AAErC,MAAM,MAAM,sBAAsB,CAAC,MAAM,SAAS,MAAM,eAAe,IACnE,MAAM,GACN;IACE,KAAK,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC;IACnC,KAAK,EAAE,MAAM,CAAC;CACf,CAAC;AAEN,MAAM,MAAM,eAAe,GAAG;IAE5B,kBAAkB,EAAE;QAClB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,gBAAgB,EAAE;QAChB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,mBAAmB,EAAE;QACnB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,6BAA6B,EAAE;QAC7B,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IAGF,eAAe,EAAE;QACf,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,yBAAyB,EAAE;QACzB,QAAQ,EAAE,MAAM,CAAC;QACjB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,CAAC;IAGF,8BAA8B,EAAE;QAC9B,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,gCAAgC,EAAE;QAChC,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;CACH,CAAC;AAEF,eAAO,MAAM,sBAAsB,GAAI,MAAM,SAAS,MAAM,eAAe,EACzE,UAAU,sBAAsB,CAAC,MAAM,CAAC;;;CAczC,CAAC;AAEF,eAAO,MAAM,eAAe,GAC1B,MAAM,SAAS,MAAM,eAAe,EACpC,cAAc,SAAS,mBAAmB,CAAC,MAAM,eAAe,CAAC,EAEjE,eAAe,cAAc,EAC7B,OAAO,mBAAmB,CAAC,MAAM,CAAC,EAClC,QAAQ,MAAM,KACb,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,GAAG,aAAa,CAAC,cAAc,CAAC,IAAI,MAAM,EAAE,CAExE,CAAC"}

package/dist/types/EventTypes.js

@@ -0,0 +1,22 @@
+// src/types/EventTypes.ts
+var normalizeEventSelector = (selector) => {
+  if (typeof selector === "string") {
+    const source = selector.split(".")[0];
+    return {
+      scope: source,
+      event: selector
+    };
+  }
+  return {
+    scope: selector.scope,
+    event: selector.event
+  };
+};
+var checkEventScope = (expectedScope, scope, _event) => {
+  return scope === expectedScope;
+};
+export {
+  checkEventScope,
+  normalizeEventSelector
+};
+//# sourceMappingURL=EventTypes.js.map

package/dist/types/EventTypes.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/types/EventTypes.ts"],"sourcesContent":["export type EventSource<\n  T extends keyof AssistantEvents = keyof AssistantEvents,\n> = T extends `${infer Source}.${string}` ? Source : never;\n\ntype ScopeConfig = {\n  composer: \"thread\" | \"message\";\n  thread: never;\n  \"thread-list-item\": never;\n};\n\nexport type SourceByScope<\n  TScope extends AssistantEventScope<keyof AssistantEvents>,\n> =\n  | (TScope extends \"*\" ? EventSource : never)\n  | (TScope extends keyof ScopeConfig ? TScope : never)\n  | {\n      [K in keyof ScopeConfig]: TScope extends ScopeConfig[K] ? K : never;\n    }[keyof ScopeConfig];\n\nexport type AssistantEventScope<TEvent extends keyof AssistantEvents> =\n  | \"*\"\n  | EventSource<TEvent>\n  | ScopeConfig[EventSource<TEvent>];\n\nexport type AssistantEventSelector<TEvent extends keyof AssistantEvents> =\n  | TEvent\n  | {\n      scope: AssistantEventScope<TEvent>;\n      event: TEvent;\n    };\n\nexport type AssistantEvents = {\n  // Thread events (from ThreadRuntimeEventType)\n  \"thread.run-start\": {\n    threadId: string;\n  };\n  \"thread.run-end\": {\n    threadId: string;\n  };\n  \"thread.initialize\": {\n    threadId: string;\n  };\n  \"thread.model-context-update\": {\n    threadId: string;\n  };\n\n  // Composer events (from ComposerRuntimeEventType)\n  \"composer.send\": {\n    threadId: string;\n    messageId?: string;\n  };\n  \"composer.attachment-add\": {\n    threadId: string;\n    messageId?: string;\n  };\n\n  // Thread list item events (from ThreadListItemEventType)\n  \"thread-list-item.switched-to\": {\n    threadId: string;\n  };\n  \"thread-list-item.switched-away\": {\n    threadId: string;\n  };\n};\n\nexport const normalizeEventSelector = <TEvent extends keyof AssistantEvents>(\n  selector: AssistantEventSelector<TEvent>,\n) => {\n  if (typeof selector === \"string\") {\n    const source = selector.split(\".\")[0] as AssistantEventScope<TEvent>;\n    return {\n      scope: source,\n      event: selector,\n    };\n  }\n\n  return {\n    scope: selector.scope,\n    event: selector.event,\n  };\n};\n\nexport const checkEventScope = <\n  TEvent extends keyof AssistantEvents,\n  TExpectedScope extends AssistantEventScope<keyof AssistantEvents>,\n>(\n  expectedScope: TExpectedScope,\n  scope: AssistantEventScope<TEvent>,\n  _event: TEvent,\n): _event is Extract<TEvent, `${SourceByScope<TExpectedScope>}.${string}`> => {\n  return scope === expectedScope;\n};\n"],"mappings":";AAiEO,IAAM,yBAAyB,CACpC,aACG;AACH,MAAI,OAAO,aAAa,UAAU;AAChC,UAAM,SAAS,SAAS,MAAM,GAAG,EAAE,CAAC;AACpC,WAAO;AAAA,MACL,OAAO;AAAA,MACP,OAAO;AAAA,IACT;AAAA,EACF;AAEA,SAAO;AAAA,IACL,OAAO,SAAS;AAAA,IAChB,OAAO,SAAS;AAAA,EAClB;AACF;AAEO,IAAM,kBAAkB,CAI7B,eACA,OACA,WAC4E;AAC5E,SAAO,UAAU;AACnB;","names":[]}

package/dist/types/index.d.ts

@@ -5,4 +5,5 @@ export type { EmptyMessagePartComponent, EmptyMessagePartProps, TextMessagePartC
 export type { EmptyMessagePartComponent as EmptyContentPartComponent, EmptyMessagePartProps as EmptyContentPartProps, TextMessagePartComponent as TextContentPartComponent, TextMessagePartProps as TextContentPartProps, ReasoningMessagePartComponent as ReasoningContentPartComponent, ReasoningMessagePartProps as ReasoningContentPartProps, SourceMessagePartComponent as SourceContentPartComponent, SourceMessagePartProps as SourceContentPartProps, ImageMessagePartComponent as ImageContentPartComponent, ImageMessagePartProps as ImageContentPartProps, FileMessagePartComponent as FileContentPartComponent, FileMessagePartProps as FileContentPartProps, Unstable_AudioMessagePartComponent as AudioContentPartComponent, Unstable_AudioMessagePartProps as AudioContentPartProps, ToolCallMessagePartComponent as ToolCallContentPartComponent, ToolCallMessagePartProps as ToolCallContentPartProps, } from "./MessagePartComponentTypes";
 export type { ThreadListItemStatus } from "../api/ThreadListItemRuntime";
 export type { Unsubscribe } from "./Unsubscribe";
+export type { AssistantEventScope, AssistantEventSelector, AssistantEvents, } from "./EventTypes";
 //# sourceMappingURL=index.d.ts.map

package/dist/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,UAAU,EACV,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,mBAAmB,CAAC;AAE3B,YAAY,EACV,aAAa,EACb,eAAe,EACf,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,EAChB,eAAe,EACf,yBAAyB,EACzB,mBAAmB,EACnB,aAAa,EACb,iBAAiB,EACjB,yBAAyB,EAGzB,qBAAqB,EACrB,0BAA0B,EAC1B,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,EACjB,aAAa,GACd,MAAM,kBAAkB,CAAC;AAG1B,YAAY,EACV,eAAe,IAAI,eAAe,EAClC,oBAAoB,IAAI,oBAAoB,EAC5C,iBAAiB,IAAI,iBAAiB,EACtC,gBAAgB,IAAI,gBAAgB,EACpC,eAAe,IAAI,eAAe,EAClC,yBAAyB,IAAI,gBAAgB,EAC7C,mBAAmB,IAAI,mBAAmB,EAC1C,iBAAiB,IAAI,iBAAiB,EACtC,yBAAyB,IAAI,yBAAyB,EAGtD,qBAAqB,IAAI,qBAAqB,EAC9C,0BAA0B,IAAI,0BAA0B,EACxD,mBAAmB,IAAI,uBAAuB,EAC9C,sBAAsB,IAAI,sBAAsB,EAChD,iBAAiB,IAAI,iBAAiB,EACtC,aAAa,IAAI,aAAa,GAC/B,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EACV,yBAAyB,EACzB,qBAAqB,EACrB,wBAAwB,EACxB,oBAAoB,EACpB,6BAA6B,EAC7B,yBAAyB,EACzB,0BAA0B,EAC1B,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,wBAAwB,EACxB,oBAAoB,EACpB,kCAAkC,EAClC,8BAA8B,EAC9B,4BAA4B,EAC5B,wBAAwB,GACzB,MAAM,6BAA6B,CAAC;AAGrC,YAAY,EACV,yBAAyB,IAAI,yBAAyB,EACtD,qBAAqB,IAAI,qBAAqB,EAC9C,wBAAwB,IAAI,wBAAwB,EACpD,oBAAoB,IAAI,oBAAoB,EAC5C,6BAA6B,IAAI,6BAA6B,EAC9D,yBAAyB,IAAI,yBAAyB,EACtD,0BAA0B,IAAI,0BAA0B,EACxD,sBAAsB,IAAI,sBAAsB,EAChD,yBAAyB,IAAI,yBAAyB,EACtD,qBAAqB,IAAI,qBAAqB,EAC9C,wBAAwB,IAAI,wBAAwB,EACpD,oBAAoB,IAAI,oBAAoB,EAC5C,kCAAkC,IAAI,yBAAyB,EAC/D,8BAA8B,IAAI,qBAAqB,EACvD,4BAA4B,IAAI,4BAA4B,EAC5D,wBAAwB,IAAI,wBAAwB,GACrD,MAAM,6BAA6B,CAAC;AAGrC,YAAY,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AAEzE,YAAY,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,UAAU,EACV,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,mBAAmB,CAAC;AAE3B,YAAY,EACV,aAAa,EACb,eAAe,EACf,oBAAoB,EACpB,iBAAiB,EACjB,gBAAgB,EAChB,eAAe,EACf,yBAAyB,EACzB,mBAAmB,EACnB,aAAa,EACb,iBAAiB,EACjB,yBAAyB,EAGzB,qBAAqB,EACrB,0BAA0B,EAC1B,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,EACjB,aAAa,GACd,MAAM,kBAAkB,CAAC;AAG1B,YAAY,EACV,eAAe,IAAI,eAAe,EAClC,oBAAoB,IAAI,oBAAoB,EAC5C,iBAAiB,IAAI,iBAAiB,EACtC,gBAAgB,IAAI,gBAAgB,EACpC,eAAe,IAAI,eAAe,EAClC,yBAAyB,IAAI,gBAAgB,EAC7C,mBAAmB,IAAI,mBAAmB,EAC1C,iBAAiB,IAAI,iBAAiB,EACtC,yBAAyB,IAAI,yBAAyB,EAGtD,qBAAqB,IAAI,qBAAqB,EAC9C,0BAA0B,IAAI,0BAA0B,EACxD,mBAAmB,IAAI,uBAAuB,EAC9C,sBAAsB,IAAI,sBAAsB,EAChD,iBAAiB,IAAI,iBAAiB,EACtC,aAAa,IAAI,aAAa,GAC/B,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EACV,yBAAyB,EACzB,qBAAqB,EACrB,wBAAwB,EACxB,oBAAoB,EACpB,6BAA6B,EAC7B,yBAAyB,EACzB,0BAA0B,EAC1B,sBAAsB,EACtB,yBAAyB,EACzB,qBAAqB,EACrB,wBAAwB,EACxB,oBAAoB,EACpB,kCAAkC,EAClC,8BAA8B,EAC9B,4BAA4B,EAC5B,wBAAwB,GACzB,MAAM,6BAA6B,CAAC;AAGrC,YAAY,EACV,yBAAyB,IAAI,yBAAyB,EACtD,qBAAqB,IAAI,qBAAqB,EAC9C,wBAAwB,IAAI,wBAAwB,EACpD,oBAAoB,IAAI,oBAAoB,EAC5C,6BAA6B,IAAI,6BAA6B,EAC9D,yBAAyB,IAAI,yBAAyB,EACtD,0BAA0B,IAAI,0BAA0B,EACxD,sBAAsB,IAAI,sBAAsB,EAChD,yBAAyB,IAAI,yBAAyB,EACtD,qBAAqB,IAAI,qBAAqB,EAC9C,wBAAwB,IAAI,wBAAwB,EACpD,oBAAoB,IAAI,oBAAoB,EAC5C,kCAAkC,IAAI,yBAAyB,EAC/D,8BAA8B,IAAI,qBAAqB,EACvD,4BAA4B,IAAI,4BAA4B,EAC5D,wBAAwB,IAAI,wBAAwB,GACrD,MAAM,6BAA6B,CAAC;AAGrC,YAAY,EAAE,oBAAoB,EAAE,MAAM,8BAA8B,CAAC;AAEzE,YAAY,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAEjD,YAAY,EACV,mBAAmB,EACnB,sBAAsB,EACtB,eAAe,GAChB,MAAM,cAAc,CAAC"}

package/dist/utils/smooth/SmoothContext.js

@@ -8,7 +8,7 @@ import {
   useState
 } from "react";
 import { create } from "zustand";
-import { useMessagePartRuntime } from "../../context/react/MessagePartContext.js";
+import { useAssistantApi } from "../../context/index.js";
 import { createContextStoreHook } from "../../context/react/utils/createContextStoreHook.js";
 import { jsx } from "react/jsx-runtime";
 var SmoothContext = createContext(null);
@@ -18,9 +18,9 @@ var makeSmoothContext = (initialState) => {
 };
 var SmoothContextProvider = ({ children }) => {
   const outer = useSmoothContext({ optional: true });
-  const MessagePartRuntime = useMessagePartRuntime();
+  const api = useAssistantApi();
   const [context] = useState(
-    () => makeSmoothContext(MessagePartRuntime.getState().status)
+    () => makeSmoothContext(api.part().getState().status)
   );
   if (outer) return children;
   return /* @__PURE__ */ jsx(SmoothContext.Provider, { value: context, children });

package/dist/utils/smooth/SmoothContext.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/utils/smooth/SmoothContext.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  ComponentType,\n  createContext,\n  FC,\n  forwardRef,\n  PropsWithChildren,\n  useContext,\n  useState,\n} from \"react\";\nimport { ReadonlyStore } from \"../../context/ReadonlyStore\";\nimport { create, UseBoundStore } from \"zustand\";\nimport {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../types/AssistantTypes\";\nimport { useMessagePartRuntime } from \"../../context/react/MessagePartContext\";\nimport { createContextStoreHook } from \"../../context/react/utils/createContextStoreHook\";\n\ntype SmoothContextValue = {\n  useSmoothStatus: UseBoundStore<\n    ReadonlyStore<MessagePartStatus | ToolCallMessagePartStatus>\n  >;\n};\n\nconst SmoothContext = createContext<SmoothContextValue | null>(null);\n\nconst makeSmoothContext = (\n  initialState: MessagePartStatus | ToolCallMessagePartStatus,\n) => {\n  const useSmoothStatus = create(() => initialState);\n  return { useSmoothStatus };\n};\n\nexport const SmoothContextProvider: FC<PropsWithChildren> = ({ children }) => {\n  const outer = useSmoothContext({ optional: true });\n  const MessagePartRuntime = useMessagePartRuntime();\n\n  const [context] = useState(() =>\n    makeSmoothContext(MessagePartRuntime.getState().status),\n  );\n\n  // do not wrap if there is an outer SmoothContextProvider\n  if (outer) return children;\n\n  return (\n    <SmoothContext.Provider value={context}>{children}</SmoothContext.Provider>\n  );\n};\n\nexport const withSmoothContextProvider = <C extends ComponentType<any>>(\n  Component: C,\n): C => {\n  const Wrapped = forwardRef((props, ref) => {\n    return (\n      <SmoothContextProvider>\n        <Component {...(props as any)} ref={ref} />\n      </SmoothContextProvider>\n    );\n  });\n  Wrapped.displayName = Component.displayName;\n  return Wrapped as any;\n};\n\nfunction useSmoothContext(options?: {\n  optional?: false | undefined;\n}): SmoothContextValue;\nfunction useSmoothContext(options?: {\n  optional?: boolean | undefined;\n}): SmoothContextValue | null;\nfunction useSmoothContext(options?: { optional?: boolean | undefined }) {\n  const context = useContext(SmoothContext);\n  if (!options?.optional && !context)\n    throw new Error(\n      \"This component must be used within a SmoothContextProvider.\",\n    );\n  return context;\n}\n\nexport const { useSmoothStatus, useSmoothStatusStore } = createContextStoreHook(\n  useSmoothContext,\n  \"useSmoothStatus\",\n);\n"],"mappings":";;;AAEA;AAAA,EAEE;AAAA,EAEA;AAAA,EAEA;AAAA,EACA;AAAA,OACK;AAEP,SAAS,cAA6B;AAKtC,SAAS,6BAA6B;AACtC,SAAS,8BAA8B;AA6BnC;AArBJ,IAAM,gBAAgB,cAAyC,IAAI;AAEnE,IAAM,oBAAoB,CACxB,iBACG;AACH,QAAMA,mBAAkB,OAAO,MAAM,YAAY;AACjD,SAAO,EAAE,iBAAAA,iBAAgB;AAC3B;AAEO,IAAM,wBAA+C,CAAC,EAAE,SAAS,MAAM;AAC5E,QAAM,QAAQ,iBAAiB,EAAE,UAAU,KAAK,CAAC;AACjD,QAAM,qBAAqB,sBAAsB;AAEjD,QAAM,CAAC,OAAO,IAAI;AAAA,IAAS,MACzB,kBAAkB,mBAAmB,SAAS,EAAE,MAAM;AAAA,EACxD;AAGA,MAAI,MAAO,QAAO;AAElB,SACE,oBAAC,cAAc,UAAd,EAAuB,OAAO,SAAU,UAAS;AAEtD;AAEO,IAAM,4BAA4B,CACvC,cACM;AACN,QAAM,UAAU,WAAW,CAAC,OAAO,QAAQ;AACzC,WACE,oBAAC,yBACC,8BAAC,aAAW,GAAI,OAAe,KAAU,GAC3C;AAAA,EAEJ,CAAC;AACD,UAAQ,cAAc,UAAU;AAChC,SAAO;AACT;AAQA,SAAS,iBAAiB,SAA8C;AACtE,QAAM,UAAU,WAAW,aAAa;AACxC,MAAI,CAAC,SAAS,YAAY,CAAC;AACzB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;AAEO,IAAM,EAAE,iBAAiB,qBAAqB,IAAI;AAAA,EACvD;AAAA,EACA;AACF;","names":["useSmoothStatus"]}
+{"version":3,"sources":["../../../src/utils/smooth/SmoothContext.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  ComponentType,\n  createContext,\n  FC,\n  forwardRef,\n  PropsWithChildren,\n  useContext,\n  useState,\n} from \"react\";\nimport { ReadonlyStore } from \"../../context/ReadonlyStore\";\nimport { create, UseBoundStore } from \"zustand\";\nimport {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../types/AssistantTypes\";\nimport { useAssistantApi } from \"../../context\";\nimport { createContextStoreHook } from \"../../context/react/utils/createContextStoreHook\";\n\ntype SmoothContextValue = {\n  useSmoothStatus: UseBoundStore<\n    ReadonlyStore<MessagePartStatus | ToolCallMessagePartStatus>\n  >;\n};\n\nconst SmoothContext = createContext<SmoothContextValue | null>(null);\n\nconst makeSmoothContext = (\n  initialState: MessagePartStatus | ToolCallMessagePartStatus,\n) => {\n  const useSmoothStatus = create(() => initialState);\n  return { useSmoothStatus };\n};\n\nexport const SmoothContextProvider: FC<PropsWithChildren> = ({ children }) => {\n  const outer = useSmoothContext({ optional: true });\n  const api = useAssistantApi();\n\n  const [context] = useState(() =>\n    makeSmoothContext(api.part().getState().status),\n  );\n\n  // do not wrap if there is an outer SmoothContextProvider\n  if (outer) return children;\n\n  return (\n    <SmoothContext.Provider value={context}>{children}</SmoothContext.Provider>\n  );\n};\n\nexport const withSmoothContextProvider = <C extends ComponentType<any>>(\n  Component: C,\n): C => {\n  const Wrapped = forwardRef((props, ref) => {\n    return (\n      <SmoothContextProvider>\n        <Component {...(props as any)} ref={ref} />\n      </SmoothContextProvider>\n    );\n  });\n  Wrapped.displayName = Component.displayName;\n  return Wrapped as any;\n};\n\nfunction useSmoothContext(options?: {\n  optional?: false | undefined;\n}): SmoothContextValue;\nfunction useSmoothContext(options?: {\n  optional?: boolean | undefined;\n}): SmoothContextValue | null;\nfunction useSmoothContext(options?: { optional?: boolean | undefined }) {\n  const context = useContext(SmoothContext);\n  if (!options?.optional && !context)\n    throw new Error(\n      \"This component must be used within a SmoothContextProvider.\",\n    );\n  return context;\n}\n\nexport const { useSmoothStatus, useSmoothStatusStore } = createContextStoreHook(\n  useSmoothContext,\n  \"useSmoothStatus\",\n);\n"],"mappings":";;;AAEA;AAAA,EAEE;AAAA,EAEA;AAAA,EAEA;AAAA,EACA;AAAA,OACK;AAEP,SAAS,cAA6B;AAKtC,SAAS,uBAAuB;AAChC,SAAS,8BAA8B;AA6BnC;AArBJ,IAAM,gBAAgB,cAAyC,IAAI;AAEnE,IAAM,oBAAoB,CACxB,iBACG;AACH,QAAMA,mBAAkB,OAAO,MAAM,YAAY;AACjD,SAAO,EAAE,iBAAAA,iBAAgB;AAC3B;AAEO,IAAM,wBAA+C,CAAC,EAAE,SAAS,MAAM;AAC5E,QAAM,QAAQ,iBAAiB,EAAE,UAAU,KAAK,CAAC;AACjD,QAAM,MAAM,gBAAgB;AAE5B,QAAM,CAAC,OAAO,IAAI;AAAA,IAAS,MACzB,kBAAkB,IAAI,KAAK,EAAE,SAAS,EAAE,MAAM;AAAA,EAChD;AAGA,MAAI,MAAO,QAAO;AAElB,SACE,oBAAC,cAAc,UAAd,EAAuB,OAAO,SAAU,UAAS;AAEtD;AAEO,IAAM,4BAA4B,CACvC,cACM;AACN,QAAM,UAAU,WAAW,CAAC,OAAO,QAAQ;AACzC,WACE,oBAAC,yBACC,8BAAC,aAAW,GAAI,OAAe,KAAU,GAC3C;AAAA,EAEJ,CAAC;AACD,UAAQ,cAAc,UAAU;AAChC,SAAO;AACT;AAQA,SAAS,iBAAiB,SAA8C;AACtE,QAAM,UAAU,WAAW,aAAa;AACxC,MAAI,CAAC,SAAS,YAAY,CAAC;AACzB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AACF,SAAO;AACT;AAEO,IAAM,EAAE,iBAAiB,qBAAqB,IAAI;AAAA,EACvD;AAAA,EACA;AACF;","names":["useSmoothStatus"]}

package/dist/utils/smooth/useSmooth.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"useSmooth.d.ts","sourceRoot":"","sources":["../../../src/utils/smooth/useSmooth.tsx"],"names":[],"mappings":"AAIA,OAAO,EAEL,oBAAoB,EACpB,eAAe,EAChB,MAAM,4BAA4B,CAAC;AAIpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AA4DhE,eAAO,MAAM,SAAS,GACpB,OAAO,gBAAgB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CAAC,EAClE,SAAQ,OAAe,KACtB,gBAAgB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CA2E5D,CAAC"}
+{"version":3,"file":"useSmooth.d.ts","sourceRoot":"","sources":["../../../src/utils/smooth/useSmooth.tsx"],"names":[],"mappings":"AAIA,OAAO,EAEL,oBAAoB,EACpB,eAAe,EAChB,MAAM,4BAA4B,CAAC;AAIpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,8BAA8B,CAAC;AA4DhE,eAAO,MAAM,SAAS,GACpB,OAAO,gBAAgB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CAAC,EAClE,SAAQ,OAAe,KACtB,gBAAgB,GAAG,CAAC,eAAe,GAAG,oBAAoB,CAwE5D,CAAC"}

package/dist/utils/smooth/useSmooth.js

@@ -2,7 +2,7 @@
 
 // src/utils/smooth/useSmooth.tsx
 import { useEffect, useMemo, useRef, useState } from "react";
-import { useMessage } from "../../context/index.js";
+import { useAssistantState } from "../../context/index.js";
 import { useCallbackRef } from "@radix-ui/react-use-callback-ref";
 import { useSmoothStatusStore } from "./SmoothContext.js";
 import { writableStore } from "../../context/ReadonlyStore.js";
@@ -55,10 +55,7 @@ var SMOOTH_STATUS = Object.freeze({
 });
 var useSmooth = (state, smooth = false) => {
   const { text } = state;
-  const id = useMessage({
-    optional: true,
-    selector: (m) => m.id
-  });
+  const id = useAssistantState(({ message }) => message.id);
   const idRef = useRef(id);
   const [displayedText, setDisplayedText] = useState(text);
   const smoothStatusStore = useSmoothStatusStore({ optional: true });

package/dist/utils/smooth/useSmooth.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/utils/smooth/useSmooth.tsx"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport { useMessage } from \"../../context\";\nimport {\n  MessagePartStatus,\n  ReasoningMessagePart,\n  TextMessagePart,\n} from \"../../types/AssistantTypes\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { useSmoothStatusStore } from \"./SmoothContext\";\nimport { writableStore } from \"../../context/ReadonlyStore\";\nimport { MessagePartState } from \"../../api/MessagePartRuntime\";\n\nclass TextStreamAnimator {\n  private animationFrameId: number | null = null;\n  private lastUpdateTime: number = Date.now();\n\n  public targetText: string = \"\";\n\n  constructor(\n    public currentText: string,\n    private setText: (newText: string) => void,\n  ) {}\n\n  start() {\n    if (this.animationFrameId !== null) return;\n    this.lastUpdateTime = Date.now();\n    this.animate();\n  }\n\n  stop() {\n    if (this.animationFrameId !== null) {\n      cancelAnimationFrame(this.animationFrameId);\n      this.animationFrameId = null;\n    }\n  }\n\n  private animate = () => {\n    const currentTime = Date.now();\n    const deltaTime = currentTime - this.lastUpdateTime;\n    let timeToConsume = deltaTime;\n\n    const remainingChars = this.targetText.length - this.currentText.length;\n    const baseTimePerChar = Math.min(5, 250 / remainingChars);\n\n    let charsToAdd = 0;\n    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {\n      charsToAdd++;\n      timeToConsume -= baseTimePerChar;\n    }\n\n    if (charsToAdd !== remainingChars) {\n      this.animationFrameId = requestAnimationFrame(this.animate);\n    } else {\n      this.animationFrameId = null;\n    }\n    if (charsToAdd === 0) return;\n\n    this.currentText = this.targetText.slice(\n      0,\n      this.currentText.length + charsToAdd,\n    );\n    this.lastUpdateTime = currentTime - timeToConsume;\n    this.setText(this.currentText);\n  };\n}\n\nconst SMOOTH_STATUS: MessagePartStatus = Object.freeze({\n  type: \"running\",\n});\n\nexport const useSmooth = (\n  state: MessagePartState & (TextMessagePart | ReasoningMessagePart),\n  smooth: boolean = false,\n): MessagePartState & (TextMessagePart | ReasoningMessagePart) => {\n  const { text } = state;\n  const id = useMessage({\n    optional: true,\n    selector: (m: { id: string }) => m.id,\n  });\n\n  const idRef = useRef(id);\n  const [displayedText, setDisplayedText] = useState(text);\n\n  const smoothStatusStore = useSmoothStatusStore({ optional: true });\n  const setText = useCallbackRef((text: string) => {\n    setDisplayedText(text);\n    if (smoothStatusStore) {\n      const target =\n        displayedText !== text || state.status.type === \"running\"\n          ? SMOOTH_STATUS\n          : state.status;\n      writableStore(smoothStatusStore).setState(target, true);\n    }\n  });\n\n  // TODO this is hacky\n  useEffect(() => {\n    if (smoothStatusStore) {\n      const target =\n        displayedText !== text || state.status.type === \"running\"\n          ? SMOOTH_STATUS\n          : state.status;\n      writableStore(smoothStatusStore).setState(target, true);\n    }\n  }, [smoothStatusStore, text, displayedText, state.status]);\n\n  const [animatorRef] = useState<TextStreamAnimator>(\n    new TextStreamAnimator(text, setText),\n  );\n\n  useEffect(() => {\n    if (!smooth) {\n      animatorRef.stop();\n      return;\n    }\n\n    if (idRef.current !== id || !text.startsWith(animatorRef.targetText)) {\n      idRef.current = id;\n      setText(text);\n\n      animatorRef.currentText = text;\n      animatorRef.targetText = text;\n      animatorRef.stop();\n\n      return;\n    }\n\n    animatorRef.targetText = text;\n    animatorRef.start();\n  }, [setText, animatorRef, id, smooth, text]);\n\n  useEffect(() => {\n    return () => {\n      animatorRef.stop();\n    };\n  }, [animatorRef]);\n\n  return useMemo(\n    () =>\n      smooth\n        ? {\n            type: \"text\",\n            text: displayedText,\n            status: text === displayedText ? state.status : SMOOTH_STATUS,\n          }\n        : state,\n    [smooth, displayedText, state, text],\n  );\n};\n"],"mappings":";;;AAEA,SAAS,WAAW,SAAS,QAAQ,gBAAgB;AACrD,SAAS,kBAAkB;AAM3B,SAAS,sBAAsB;AAC/B,SAAS,4BAA4B;AACrC,SAAS,qBAAqB;AAG9B,IAAM,qBAAN,MAAyB;AAAA,EAMvB,YACS,aACC,SACR;AAFO;AACC;AAAA,EACP;AAAA,EARK,mBAAkC;AAAA,EAClC,iBAAyB,KAAK,IAAI;AAAA,EAEnC,aAAqB;AAAA,EAO5B,QAAQ;AACN,QAAI,KAAK,qBAAqB,KAAM;AACpC,SAAK,iBAAiB,KAAK,IAAI;AAC/B,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,OAAO;AACL,QAAI,KAAK,qBAAqB,MAAM;AAClC,2BAAqB,KAAK,gBAAgB;AAC1C,WAAK,mBAAmB;AAAA,IAC1B;AAAA,EACF;AAAA,EAEQ,UAAU,MAAM;AACtB,UAAM,cAAc,KAAK,IAAI;AAC7B,UAAM,YAAY,cAAc,KAAK;AACrC,QAAI,gBAAgB;AAEpB,UAAM,iBAAiB,KAAK,WAAW,SAAS,KAAK,YAAY;AACjE,UAAM,kBAAkB,KAAK,IAAI,GAAG,MAAM,cAAc;AAExD,QAAI,aAAa;AACjB,WAAO,iBAAiB,mBAAmB,aAAa,gBAAgB;AACtE;AACA,uBAAiB;AAAA,IACnB;AAEA,QAAI,eAAe,gBAAgB;AACjC,WAAK,mBAAmB,sBAAsB,KAAK,OAAO;AAAA,IAC5D,OAAO;AACL,WAAK,mBAAmB;AAAA,IAC1B;AACA,QAAI,eAAe,EAAG;AAEtB,SAAK,cAAc,KAAK,WAAW;AAAA,MACjC;AAAA,MACA,KAAK,YAAY,SAAS;AAAA,IAC5B;AACA,SAAK,iBAAiB,cAAc;AACpC,SAAK,QAAQ,KAAK,WAAW;AAAA,EAC/B;AACF;AAEA,IAAM,gBAAmC,OAAO,OAAO;AAAA,EACrD,MAAM;AACR,CAAC;AAEM,IAAM,YAAY,CACvB,OACA,SAAkB,UAC8C;AAChE,QAAM,EAAE,KAAK,IAAI;AACjB,QAAM,KAAK,WAAW;AAAA,IACpB,UAAU;AAAA,IACV,UAAU,CAAC,MAAsB,EAAE;AAAA,EACrC,CAAC;AAED,QAAM,QAAQ,OAAO,EAAE;AACvB,QAAM,CAAC,eAAe,gBAAgB,IAAI,SAAS,IAAI;AAEvD,QAAM,oBAAoB,qBAAqB,EAAE,UAAU,KAAK,CAAC;AACjE,QAAM,UAAU,eAAe,CAACA,UAAiB;AAC/C,qBAAiBA,KAAI;AACrB,QAAI,mBAAmB;AACrB,YAAM,SACJ,kBAAkBA,SAAQ,MAAM,OAAO,SAAS,YAC5C,gBACA,MAAM;AACZ,oBAAc,iBAAiB,EAAE,SAAS,QAAQ,IAAI;AAAA,IACxD;AAAA,EACF,CAAC;AAGD,YAAU,MAAM;AACd,QAAI,mBAAmB;AACrB,YAAM,SACJ,kBAAkB,QAAQ,MAAM,OAAO,SAAS,YAC5C,gBACA,MAAM;AACZ,oBAAc,iBAAiB,EAAE,SAAS,QAAQ,IAAI;AAAA,IACxD;AAAA,EACF,GAAG,CAAC,mBAAmB,MAAM,eAAe,MAAM,MAAM,CAAC;AAEzD,QAAM,CAAC,WAAW,IAAI;AAAA,IACpB,IAAI,mBAAmB,MAAM,OAAO;AAAA,EACtC;AAEA,YAAU,MAAM;AACd,QAAI,CAAC,QAAQ;AACX,kBAAY,KAAK;AACjB;AAAA,IACF;AAEA,QAAI,MAAM,YAAY,MAAM,CAAC,KAAK,WAAW,YAAY,UAAU,GAAG;AACpE,YAAM,UAAU;AAChB,cAAQ,IAAI;AAEZ,kBAAY,cAAc;AAC1B,kBAAY,aAAa;AACzB,kBAAY,KAAK;AAEjB;AAAA,IACF;AAEA,gBAAY,aAAa;AACzB,gBAAY,MAAM;AAAA,EACpB,GAAG,CAAC,SAAS,aAAa,IAAI,QAAQ,IAAI,CAAC;AAE3C,YAAU,MAAM;AACd,WAAO,MAAM;AACX,kBAAY,KAAK;AAAA,IACnB;AAAA,EACF,GAAG,CAAC,WAAW,CAAC;AAEhB,SAAO;AAAA,IACL,MACE,SACI;AAAA,MACE,MAAM;AAAA,MACN,MAAM;AAAA,MACN,QAAQ,SAAS,gBAAgB,MAAM,SAAS;AAAA,IAClD,IACA;AAAA,IACN,CAAC,QAAQ,eAAe,OAAO,IAAI;AAAA,EACrC;AACF;","names":["text"]}
+{"version":3,"sources":["../../../src/utils/smooth/useSmooth.tsx"],"sourcesContent":["\"use client\";\n\nimport { useEffect, useMemo, useRef, useState } from \"react\";\nimport { useAssistantState } from \"../../context\";\nimport {\n  MessagePartStatus,\n  ReasoningMessagePart,\n  TextMessagePart,\n} from \"../../types/AssistantTypes\";\nimport { useCallbackRef } from \"@radix-ui/react-use-callback-ref\";\nimport { useSmoothStatusStore } from \"./SmoothContext\";\nimport { writableStore } from \"../../context/ReadonlyStore\";\nimport { MessagePartState } from \"../../api/MessagePartRuntime\";\n\nclass TextStreamAnimator {\n  private animationFrameId: number | null = null;\n  private lastUpdateTime: number = Date.now();\n\n  public targetText: string = \"\";\n\n  constructor(\n    public currentText: string,\n    private setText: (newText: string) => void,\n  ) {}\n\n  start() {\n    if (this.animationFrameId !== null) return;\n    this.lastUpdateTime = Date.now();\n    this.animate();\n  }\n\n  stop() {\n    if (this.animationFrameId !== null) {\n      cancelAnimationFrame(this.animationFrameId);\n      this.animationFrameId = null;\n    }\n  }\n\n  private animate = () => {\n    const currentTime = Date.now();\n    const deltaTime = currentTime - this.lastUpdateTime;\n    let timeToConsume = deltaTime;\n\n    const remainingChars = this.targetText.length - this.currentText.length;\n    const baseTimePerChar = Math.min(5, 250 / remainingChars);\n\n    let charsToAdd = 0;\n    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {\n      charsToAdd++;\n      timeToConsume -= baseTimePerChar;\n    }\n\n    if (charsToAdd !== remainingChars) {\n      this.animationFrameId = requestAnimationFrame(this.animate);\n    } else {\n      this.animationFrameId = null;\n    }\n    if (charsToAdd === 0) return;\n\n    this.currentText = this.targetText.slice(\n      0,\n      this.currentText.length + charsToAdd,\n    );\n    this.lastUpdateTime = currentTime - timeToConsume;\n    this.setText(this.currentText);\n  };\n}\n\nconst SMOOTH_STATUS: MessagePartStatus = Object.freeze({\n  type: \"running\",\n});\n\nexport const useSmooth = (\n  state: MessagePartState & (TextMessagePart | ReasoningMessagePart),\n  smooth: boolean = false,\n): MessagePartState & (TextMessagePart | ReasoningMessagePart) => {\n  const { text } = state;\n  const id = useAssistantState(({ message }) => message.id);\n\n  const idRef = useRef(id);\n  const [displayedText, setDisplayedText] = useState(text);\n\n  const smoothStatusStore = useSmoothStatusStore({ optional: true });\n  const setText = useCallbackRef((text: string) => {\n    setDisplayedText(text);\n    if (smoothStatusStore) {\n      const target =\n        displayedText !== text || state.status.type === \"running\"\n          ? SMOOTH_STATUS\n          : state.status;\n      writableStore(smoothStatusStore).setState(target, true);\n    }\n  });\n\n  // TODO this is hacky\n  useEffect(() => {\n    if (smoothStatusStore) {\n      const target =\n        displayedText !== text || state.status.type === \"running\"\n          ? SMOOTH_STATUS\n          : state.status;\n      writableStore(smoothStatusStore).setState(target, true);\n    }\n  }, [smoothStatusStore, text, displayedText, state.status]);\n\n  const [animatorRef] = useState<TextStreamAnimator>(\n    new TextStreamAnimator(text, setText),\n  );\n\n  useEffect(() => {\n    if (!smooth) {\n      animatorRef.stop();\n      return;\n    }\n\n    if (idRef.current !== id || !text.startsWith(animatorRef.targetText)) {\n      idRef.current = id;\n      setText(text);\n\n      animatorRef.currentText = text;\n      animatorRef.targetText = text;\n      animatorRef.stop();\n\n      return;\n    }\n\n    animatorRef.targetText = text;\n    animatorRef.start();\n  }, [setText, animatorRef, id, smooth, text]);\n\n  useEffect(() => {\n    return () => {\n      animatorRef.stop();\n    };\n  }, [animatorRef]);\n\n  return useMemo(\n    () =>\n      smooth\n        ? {\n            type: \"text\",\n            text: displayedText,\n            status: text === displayedText ? state.status : SMOOTH_STATUS,\n          }\n        : state,\n    [smooth, displayedText, state, text],\n  );\n};\n"],"mappings":";;;AAEA,SAAS,WAAW,SAAS,QAAQ,gBAAgB;AACrD,SAAS,yBAAyB;AAMlC,SAAS,sBAAsB;AAC/B,SAAS,4BAA4B;AACrC,SAAS,qBAAqB;AAG9B,IAAM,qBAAN,MAAyB;AAAA,EAMvB,YACS,aACC,SACR;AAFO;AACC;AAAA,EACP;AAAA,EARK,mBAAkC;AAAA,EAClC,iBAAyB,KAAK,IAAI;AAAA,EAEnC,aAAqB;AAAA,EAO5B,QAAQ;AACN,QAAI,KAAK,qBAAqB,KAAM;AACpC,SAAK,iBAAiB,KAAK,IAAI;AAC/B,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,OAAO;AACL,QAAI,KAAK,qBAAqB,MAAM;AAClC,2BAAqB,KAAK,gBAAgB;AAC1C,WAAK,mBAAmB;AAAA,IAC1B;AAAA,EACF;AAAA,EAEQ,UAAU,MAAM;AACtB,UAAM,cAAc,KAAK,IAAI;AAC7B,UAAM,YAAY,cAAc,KAAK;AACrC,QAAI,gBAAgB;AAEpB,UAAM,iBAAiB,KAAK,WAAW,SAAS,KAAK,YAAY;AACjE,UAAM,kBAAkB,KAAK,IAAI,GAAG,MAAM,cAAc;AAExD,QAAI,aAAa;AACjB,WAAO,iBAAiB,mBAAmB,aAAa,gBAAgB;AACtE;AACA,uBAAiB;AAAA,IACnB;AAEA,QAAI,eAAe,gBAAgB;AACjC,WAAK,mBAAmB,sBAAsB,KAAK,OAAO;AAAA,IAC5D,OAAO;AACL,WAAK,mBAAmB;AAAA,IAC1B;AACA,QAAI,eAAe,EAAG;AAEtB,SAAK,cAAc,KAAK,WAAW;AAAA,MACjC;AAAA,MACA,KAAK,YAAY,SAAS;AAAA,IAC5B;AACA,SAAK,iBAAiB,cAAc;AACpC,SAAK,QAAQ,KAAK,WAAW;AAAA,EAC/B;AACF;AAEA,IAAM,gBAAmC,OAAO,OAAO;AAAA,EACrD,MAAM;AACR,CAAC;AAEM,IAAM,YAAY,CACvB,OACA,SAAkB,UAC8C;AAChE,QAAM,EAAE,KAAK,IAAI;AACjB,QAAM,KAAK,kBAAkB,CAAC,EAAE,QAAQ,MAAM,QAAQ,EAAE;AAExD,QAAM,QAAQ,OAAO,EAAE;AACvB,QAAM,CAAC,eAAe,gBAAgB,IAAI,SAAS,IAAI;AAEvD,QAAM,oBAAoB,qBAAqB,EAAE,UAAU,KAAK,CAAC;AACjE,QAAM,UAAU,eAAe,CAACA,UAAiB;AAC/C,qBAAiBA,KAAI;AACrB,QAAI,mBAAmB;AACrB,YAAM,SACJ,kBAAkBA,SAAQ,MAAM,OAAO,SAAS,YAC5C,gBACA,MAAM;AACZ,oBAAc,iBAAiB,EAAE,SAAS,QAAQ,IAAI;AAAA,IACxD;AAAA,EACF,CAAC;AAGD,YAAU,MAAM;AACd,QAAI,mBAAmB;AACrB,YAAM,SACJ,kBAAkB,QAAQ,MAAM,OAAO,SAAS,YAC5C,gBACA,MAAM;AACZ,oBAAc,iBAAiB,EAAE,SAAS,QAAQ,IAAI;AAAA,IACxD;AAAA,EACF,GAAG,CAAC,mBAAmB,MAAM,eAAe,MAAM,MAAM,CAAC;AAEzD,QAAM,CAAC,WAAW,IAAI;AAAA,IACpB,IAAI,mBAAmB,MAAM,OAAO;AAAA,EACtC;AAEA,YAAU,MAAM;AACd,QAAI,CAAC,QAAQ;AACX,kBAAY,KAAK;AACjB;AAAA,IACF;AAEA,QAAI,MAAM,YAAY,MAAM,CAAC,KAAK,WAAW,YAAY,UAAU,GAAG;AACpE,YAAM,UAAU;AAChB,cAAQ,IAAI;AAEZ,kBAAY,cAAc;AAC1B,kBAAY,aAAa;AACzB,kBAAY,KAAK;AAEjB;AAAA,IACF;AAEA,gBAAY,aAAa;AACzB,gBAAY,MAAM;AAAA,EACpB,GAAG,CAAC,SAAS,aAAa,IAAI,QAAQ,IAAI,CAAC;AAE3C,YAAU,MAAM;AACd,WAAO,MAAM;AACX,kBAAY,KAAK;AAAA,IACnB;AAAA,EACF,GAAG,CAAC,WAAW,CAAC;AAEhB,SAAO;AAAA,IACL,MACE,SACI;AAAA,MACE,MAAM;AAAA,MACN,MAAM;AAAA,MACN,QAAQ,SAAS,gBAAgB,MAAM,SAAS;AAAA,IAClD,IACA;AAAA,IACN,CAAC,QAAQ,eAAe,OAAO,IAAI;AAAA,EACrC;AACF;","names":["text"]}

package/dist/utils/tap-store/index.d.ts

@@ -0,0 +1,3 @@
+export { type Store, asStore } from "./store";
+export { tapApi, type ActionsObject } from "./tap-store-api";
+//# sourceMappingURL=index.d.ts.map

package/dist/utils/tap-store/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/utils/tap-store/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,KAAK,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAE,MAAM,EAAE,KAAK,aAAa,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/utils/tap-store/index.js

@@ -0,0 +1,8 @@
+// src/utils/tap-store/index.ts
+import { asStore } from "./store.js";
+import { tapApi } from "./tap-store-api.js";
+export {
+  asStore,
+  tapApi
+};
+//# sourceMappingURL=index.js.map

package/dist/utils/tap-store/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/utils/tap-store/index.ts"],"sourcesContent":["export { type Store, asStore } from \"./store\";\nexport { tapApi, type ActionsObject } from \"./tap-store-api\";\n"],"mappings":";AAAA,SAAqB,eAAe;AACpC,SAAS,cAAkC;","names":[]}

package/dist/utils/tap-store/store.d.ts

@@ -0,0 +1,21 @@
+import { ResourceElement, Unsubscribe } from "@assistant-ui/tap";
+import { StoreApi } from "./tap-store-api";
+export interface Store<TState, TActions> {
+    getApi(): StoreApi<TState, TActions>;
+    /**
+     * Subscribe to the store.
+     */
+    subscribe(listener: () => void): Unsubscribe;
+    /**
+     * Synchronously flush all the updates to the store.
+     */
+    flushSync(): void;
+}
+export declare const asStore: <TState, TActions, TProps>(props: ResourceElement<{
+    api: StoreApi<TState, TActions>;
+}, TProps>, options?: {
+    key?: string | number;
+} | undefined) => ResourceElement<Store<TState, TActions>, ResourceElement<{
+    api: StoreApi<TState, TActions>;
+}, TProps>>;
+//# sourceMappingURL=store.d.ts.map

package/dist/utils/tap-store/store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"store.d.ts","sourceRoot":"","sources":["../../../src/utils/tap-store/store.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,eAAe,EAGf,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAE3C,MAAM,WAAW,KAAK,CAAC,MAAM,EAAE,QAAQ;IACrC,MAAM,IAAI,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAErC;;OAEG;IACH,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,WAAW,CAAC;IAE7C;;OAEG;IACH,SAAS,IAAI,IAAI,CAAC;CACnB;AAED,eAAO,MAAM,OAAO,GACjB,MAAM,EAAE,QAAQ,EAAE,MAAM;SAGd,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC;;;;SAA1B,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC;WAsBtC,CAAC"}

package/dist/utils/tap-store/store.js

@@ -0,0 +1,29 @@
+// src/utils/tap-store/store.ts
+import {
+  tapMemo,
+  tapEffect,
+  resource,
+  createResource
+} from "@assistant-ui/tap";
+var asStore = resource(
+  (element) => {
+    const resource2 = tapMemo(
+      () => createResource(element, true),
+      [element.type]
+    );
+    tapEffect(() => {
+      resource2.updateInput(element.props);
+    });
+    return tapMemo(() => {
+      return {
+        getApi: () => resource2.getState().api,
+        subscribe: resource2.subscribe,
+        flushSync: resource2.flushSync
+      };
+    }, [resource2]);
+  }
+);
+export {
+  asStore
+};
+//# sourceMappingURL=store.js.map

package/dist/utils/tap-store/store.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/utils/tap-store/store.ts"],"sourcesContent":["import {\n  tapMemo,\n  tapEffect,\n  ResourceElement,\n  resource,\n  createResource,\n  Unsubscribe,\n} from \"@assistant-ui/tap\";\nimport { StoreApi } from \"./tap-store-api\";\n\nexport interface Store<TState, TActions> {\n  getApi(): StoreApi<TState, TActions>;\n\n  /**\n   * Subscribe to the store.\n   */\n  subscribe(listener: () => void): Unsubscribe;\n\n  /**\n   * Synchronously flush all the updates to the store.\n   */\n  flushSync(): void;\n}\n\nexport const asStore = resource(\n  <TState, TActions, TProps>(\n    element: ResourceElement<\n      {\n        api: StoreApi<TState, TActions>;\n      },\n      TProps\n    >,\n  ): Store<TState, TActions> => {\n    const resource = tapMemo(\n      () => createResource(element, true),\n      [element.type],\n    );\n\n    tapEffect(() => {\n      resource.updateInput(element.props);\n    });\n\n    return tapMemo<Store<TState, TActions>>(() => {\n      return {\n        getApi: () => resource.getState().api,\n        subscribe: resource.subscribe,\n        flushSync: resource.flushSync,\n      };\n    }, [resource]);\n  },\n);\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,EACA;AAAA,OAEK;AAiBA,IAAM,UAAU;AAAA,EACrB,CACE,YAM4B;AAC5B,UAAMA,YAAW;AAAA,MACf,MAAM,eAAe,SAAS,IAAI;AAAA,MAClC,CAAC,QAAQ,IAAI;AAAA,IACf;AAEA,cAAU,MAAM;AACd,MAAAA,UAAS,YAAY,QAAQ,KAAK;AAAA,IACpC,CAAC;AAED,WAAO,QAAiC,MAAM;AAC5C,aAAO;AAAA,QACL,QAAQ,MAAMA,UAAS,SAAS,EAAE;AAAA,QAClC,WAAWA,UAAS;AAAA,QACpB,WAAWA,UAAS;AAAA,MACtB;AAAA,IACF,GAAG,CAACA,SAAQ,CAAC;AAAA,EACf;AACF;","names":["resource"]}

package/dist/utils/tap-store/tap-store-api.d.ts

@@ -0,0 +1,8 @@
+export interface ActionsObject {
+    [key: string]: ((...args: any[]) => any) | ActionsObject;
+}
+export type StoreApi<TState, TActions> = TActions & {
+    getState(): TState;
+};
+export declare const tapApi: <TState, TActions extends ActionsObject>(state: TState, actions: TActions) => StoreApi<TState, TActions>;
+//# sourceMappingURL=tap-store-api.d.ts.map

package/dist/utils/tap-store/tap-store-api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tap-store-api.d.ts","sourceRoot":"","sources":["../../../src/utils/tap-store/tap-store-api.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,aAAa;IAC5B,CAAC,GAAG,EAAE,MAAM,GAAG,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,CAAC,GAAG,aAAa,CAAC;CAC1D;AAED,MAAM,MAAM,QAAQ,CAAC,MAAM,EAAE,QAAQ,IAAI,QAAQ,GAAG;IAClD,QAAQ,IAAI,MAAM,CAAC;CACpB,CAAC;AA8CF,eAAO,MAAM,MAAM,GAAI,MAAM,EAAE,QAAQ,SAAS,aAAa,EAC3D,OAAO,MAAM,EACb,SAAS,QAAQ,+BAkBlB,CAAC"}

package/dist/utils/tap-store/tap-store-api.js

@@ -0,0 +1,57 @@
+// src/utils/tap-store/tap-store-api.ts
+import { tapEffect, tapMemo, tapRef } from "@assistant-ui/tap";
+var ReadonlyStoreApiHandler = class {
+  constructor(getState, getActions) {
+    this.getState = getState;
+    this.getActions = getActions;
+  }
+  get(_, prop) {
+    if (prop === "getState") return this.getState;
+    return this.getActions()[prop];
+  }
+  ownKeys() {
+    return ["getState", ...Object.keys(this.getActions())];
+  }
+  has(_, prop) {
+    if (prop === "getState") return true;
+    return prop in this.getActions();
+  }
+  getOwnPropertyDescriptor(_, prop) {
+    if (prop === "getState")
+      return {
+        enumerable: true,
+        configurable: false,
+        get: this.getState
+      };
+    return Object.getOwnPropertyDescriptor(this.getActions(), prop);
+  }
+  set() {
+    return false;
+  }
+  defineProperty() {
+    return false;
+  }
+  deleteProperty() {
+    return false;
+  }
+};
+var tapApi = (state, actions) => {
+  const ref = tapRef(() => ({ state, actions }));
+  tapEffect(() => {
+    ref.current = { state, actions };
+  });
+  return tapMemo(
+    () => new Proxy(
+      {},
+      new ReadonlyStoreApiHandler(
+        () => ref.current.state,
+        () => ref.current.actions
+      )
+    ),
+    []
+  );
+};
+export {
+  tapApi
+};
+//# sourceMappingURL=tap-store-api.js.map

package/dist/utils/tap-store/tap-store-api.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/utils/tap-store/tap-store-api.ts"],"sourcesContent":["import { tapEffect, tapMemo, tapRef } from \"@assistant-ui/tap\";\n\nexport interface ActionsObject {\n  [key: string]: ((...args: any[]) => any) | ActionsObject;\n}\n\nexport type StoreApi<TState, TActions> = TActions & {\n  getState(): TState;\n};\n\nclass ReadonlyStoreApiHandler<TState, TActions>\n  implements ProxyHandler<StoreApi<TState, TActions>>\n{\n  constructor(\n    private readonly getState: () => TState,\n    private readonly getActions: () => TActions,\n  ) {}\n\n  get(_: unknown, prop: string | symbol) {\n    if (prop === \"getState\") return this.getState;\n    return this.getActions()[prop as keyof TActions];\n  }\n\n  ownKeys(): ArrayLike<string | symbol> {\n    return [\"getState\", ...Object.keys(this.getActions() as object)];\n  }\n\n  has(_: unknown, prop: string | symbol) {\n    if (prop === \"getState\") return true;\n    return prop in (this.getActions() as object);\n  }\n\n  getOwnPropertyDescriptor(_: unknown, prop: string | symbol) {\n    if (prop === \"getState\")\n      return {\n        enumerable: true,\n        configurable: false,\n        get: this.getState,\n      };\n\n    return Object.getOwnPropertyDescriptor(this.getActions(), prop);\n  }\n\n  set() {\n    return false;\n  }\n  defineProperty() {\n    return false;\n  }\n  deleteProperty() {\n    return false;\n  }\n}\n\nexport const tapApi = <TState, TActions extends ActionsObject>(\n  state: TState,\n  actions: TActions,\n) => {\n  const ref = tapRef(() => ({ state, actions }));\n  tapEffect(() => {\n    ref.current = { state, actions };\n  });\n\n  return tapMemo(\n    () =>\n      new Proxy<StoreApi<TState, TActions>>(\n        {} as StoreApi<TState, TActions>,\n        new ReadonlyStoreApiHandler(\n          () => ref.current.state,\n          () => ref.current.actions,\n        ),\n      ),\n    [],\n  );\n};\n"],"mappings":";AAAA,SAAS,WAAW,SAAS,cAAc;AAU3C,IAAM,0BAAN,MAEA;AAAA,EACE,YACmB,UACA,YACjB;AAFiB;AACA;AAAA,EAChB;AAAA,EAEH,IAAI,GAAY,MAAuB;AACrC,QAAI,SAAS,WAAY,QAAO,KAAK;AACrC,WAAO,KAAK,WAAW,EAAE,IAAsB;AAAA,EACjD;AAAA,EAEA,UAAsC;AACpC,WAAO,CAAC,YAAY,GAAG,OAAO,KAAK,KAAK,WAAW,CAAW,CAAC;AAAA,EACjE;AAAA,EAEA,IAAI,GAAY,MAAuB;AACrC,QAAI,SAAS,WAAY,QAAO;AAChC,WAAO,QAAS,KAAK,WAAW;AAAA,EAClC;AAAA,EAEA,yBAAyB,GAAY,MAAuB;AAC1D,QAAI,SAAS;AACX,aAAO;AAAA,QACL,YAAY;AAAA,QACZ,cAAc;AAAA,QACd,KAAK,KAAK;AAAA,MACZ;AAEF,WAAO,OAAO,yBAAyB,KAAK,WAAW,GAAG,IAAI;AAAA,EAChE;AAAA,EAEA,MAAM;AACJ,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AAAA,EACA,iBAAiB;AACf,WAAO;AAAA,EACT;AACF;AAEO,IAAM,SAAS,CACpB,OACA,YACG;AACH,QAAM,MAAM,OAAO,OAAO,EAAE,OAAO,QAAQ,EAAE;AAC7C,YAAU,MAAM;AACd,QAAI,UAAU,EAAE,OAAO,QAAQ;AAAA,EACjC,CAAC;AAED,SAAO;AAAA,IACL,MACE,IAAI;AAAA,MACF,CAAC;AAAA,MACD,IAAI;AAAA,QACF,MAAM,IAAI,QAAQ;AAAA,QAClB,MAAM,IAAI,QAAQ;AAAA,MACpB;AAAA,IACF;AAAA,IACF,CAAC;AAAA,EACH;AACF;","names":[]}

package/dist/utils/useToolArgsFieldStatus.js

@@ -1,17 +1,17 @@
 // src/utils/useToolArgsFieldStatus.tsx
 import { getPartialJsonObjectFieldState } from "assistant-stream/utils";
-import { useMessagePart } from "../context/index.js";
+import { useAssistantState } from "../context/index.js";
 var COMPLETE_STATUS = { type: "complete" };
 var useToolArgsFieldStatus = (fieldPath) => {
-  return useMessagePart((t) => {
-    if (t.type !== "tool-call")
+  return useAssistantState(({ part }) => {
+    if (part.type !== "tool-call")
       throw new Error(
         "useToolArgsFieldStatus can only be used inside tool-call message parts"
       );
-    const state = getPartialJsonObjectFieldState(t.args, fieldPath);
-    if (state === "complete" || t.status.type === "requires-action")
+    const state = getPartialJsonObjectFieldState(part.args, fieldPath);
+    if (state === "complete" || part.status?.type === "requires-action")
       return COMPLETE_STATUS;
-    return t.status;
+    return part.status;
   });
 };
 export {

package/dist/utils/useToolArgsFieldStatus.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/utils/useToolArgsFieldStatus.tsx"],"sourcesContent":["import { getPartialJsonObjectFieldState } from \"assistant-stream/utils\";\nimport { useMessagePart } from \"../context\";\n\nconst COMPLETE_STATUS = { type: \"complete\" };\n\nexport const useToolArgsFieldStatus = (fieldPath: (string | number)[]) => {\n  return useMessagePart((t) => {\n    if (t.type !== \"tool-call\")\n      throw new Error(\n        \"useToolArgsFieldStatus can only be used inside tool-call message parts\",\n      );\n\n    const state = getPartialJsonObjectFieldState(t.args, fieldPath);\n    if (state === \"complete\" || t.status.type === \"requires-action\")\n      return COMPLETE_STATUS;\n    return t.status;\n  });\n};\n"],"mappings":";AAAA,SAAS,sCAAsC;AAC/C,SAAS,sBAAsB;AAE/B,IAAM,kBAAkB,EAAE,MAAM,WAAW;AAEpC,IAAM,yBAAyB,CAAC,cAAmC;AACxE,SAAO,eAAe,CAAC,MAAM;AAC3B,QAAI,EAAE,SAAS;AACb,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,UAAM,QAAQ,+BAA+B,EAAE,MAAM,SAAS;AAC9D,QAAI,UAAU,cAAc,EAAE,OAAO,SAAS;AAC5C,aAAO;AACT,WAAO,EAAE;AAAA,EACX,CAAC;AACH;","names":[]}
+{"version":3,"sources":["../../src/utils/useToolArgsFieldStatus.tsx"],"sourcesContent":["import { getPartialJsonObjectFieldState } from \"assistant-stream/utils\";\nimport { useAssistantState } from \"../context\";\n\nconst COMPLETE_STATUS = { type: \"complete\" };\n\nexport const useToolArgsFieldStatus = (fieldPath: (string | number)[]) => {\n  return useAssistantState(({ part }) => {\n    if (part.type !== \"tool-call\")\n      throw new Error(\n        \"useToolArgsFieldStatus can only be used inside tool-call message parts\",\n      );\n\n    const state = getPartialJsonObjectFieldState(part.args, fieldPath);\n    if (state === \"complete\" || part.status?.type === \"requires-action\")\n      return COMPLETE_STATUS;\n    return part.status;\n  });\n};\n"],"mappings":";AAAA,SAAS,sCAAsC;AAC/C,SAAS,yBAAyB;AAElC,IAAM,kBAAkB,EAAE,MAAM,WAAW;AAEpC,IAAM,yBAAyB,CAAC,cAAmC;AACxE,SAAO,kBAAkB,CAAC,EAAE,KAAK,MAAM;AACrC,QAAI,KAAK,SAAS;AAChB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAEF,UAAM,QAAQ,+BAA+B,KAAK,MAAM,SAAS;AACjE,QAAI,UAAU,cAAc,KAAK,QAAQ,SAAS;AAChD,aAAO;AACT,WAAO,KAAK;AAAA,EACd,CAAC;AACH;","names":[]}

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.10.50",
+  "version": "0.11.0",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -49,6 +49,7 @@
   "sideEffects": false,
   "dependencies": {
     "assistant-cloud": "^0.1.1",
+    "@assistant-ui/tap": "^0.1.0",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.2",

package/src/api/AttachmentRuntime.ts

@@ -54,7 +54,9 @@ export abstract class AttachmentRuntimeImpl<
 
   public abstract get source(): Source;
 
-  constructor(private _core: AttachmentSnapshotBinding<Source>) {}
+  constructor(private _core: AttachmentSnapshotBinding<Source>) {
+    this.__internal_bindMethods();
+  }
 
   protected __internal_bindMethods() {
     this.getState = this.getState.bind(this);

package/src/api/ComposerRuntime.ts

@@ -39,6 +39,8 @@ type BaseComposerState = {
   readonly role: MessageRole;
   readonly attachments: readonly Attachment[];
   readonly runConfig: RunConfig;
+
+  readonly attachmentAccept: string;
 };
 
 export type ThreadComposerState = BaseComposerState & {
@@ -69,6 +71,7 @@ const getThreadComposerState = (
     text: runtime?.text ?? "",
     role: runtime?.role ?? "user",
     runConfig: runtime?.runConfig ?? EMPTY_OBJECT,
+    attachmentAccept: runtime?.attachmentAccept ?? "",
 
     value: runtime?.text ?? "",
   });
@@ -88,6 +91,7 @@ const getEditComposerState = (
     role: runtime?.role ?? "user",
     attachments: runtime?.attachments ?? EMPTY_ARRAY,
     runConfig: runtime?.runConfig ?? EMPTY_OBJECT,
+    attachmentAccept: runtime?.attachmentAccept ?? "",
 
     value: runtime?.text ?? "",
   });
@@ -102,8 +106,6 @@ export type ComposerRuntime = {
    */
   getState(): ComposerState;
 
-  getAttachmentAccept(): string;
-
   /**
    * Given a standard js File object, add it to the composer. A composer can have multiple attachments.
    * @param file The file to add to the composer.
@@ -166,6 +168,14 @@ export type ComposerRuntime = {
    * @param idx The index of the attachment to get.
    */
   getAttachmentByIndex(idx: number): AttachmentRuntime;
+
+  /**
+   * @deprecated This API is still under active development and might change without notice.
+   */
+  unstable_on(
+    event: ComposerRuntimeEventType,
+    callback: () => void,
+  ): Unsubscribe;
 };
 
 export abstract class ComposerRuntimeImpl implements ComposerRuntime {
@@ -188,7 +198,6 @@ export abstract class ComposerRuntimeImpl implements ComposerRuntime {
     this.send = this.send.bind(this);
     this.cancel = this.cancel.bind(this);
     this.setRole = this.setRole.bind(this);
-    this.getAttachmentAccept = this.getAttachmentAccept.bind(this);
     this.getAttachmentByIndex = this.getAttachmentByIndex.bind(this);
     this.unstable_on = this.unstable_on.bind(this);
   }
@@ -267,12 +276,6 @@ export abstract class ComposerRuntimeImpl implements ComposerRuntime {
     return subject.subscribe(callback);
   }
 
-  public getAttachmentAccept(): string {
-    const core = this._core.getState();
-    if (!core) throw new Error("Composer is not available");
-    return core.getAttachmentAccept();
-  }
-
   public abstract getAttachmentByIndex(idx: number): AttachmentRuntime;
 }
 
@@ -317,6 +320,8 @@ export class ThreadComposerRuntimeImpl
       subscribe: (callback) => stateBinding.subscribe(callback),
     });
     this._getState = stateBinding.getState.bind(stateBinding);
+
+    this.__internal_bindMethods();
   }
 
   public override getState(): ThreadComposerState {
@@ -394,6 +399,8 @@ export class EditComposerRuntimeImpl
     });
 
     this._getState = stateBinding.getState.bind(stateBinding);
+
+    this.__internal_bindMethods();
   }
 
   public override __internal_bindMethods() {

package/src/api/MessagePartRuntime.ts

@@ -44,7 +44,9 @@ export class MessagePartRuntimeImpl implements MessagePartRuntime {
     private contentBinding: MessagePartSnapshotBinding,
     private messageApi?: MessageStateBinding,
     private threadApi?: ThreadRuntimeCoreBinding,
-  ) {}
+  ) {
+    this.__internal_bindMethods();
+  }
 
   protected __internal_bindMethods() {
     this.addToolResult = this.addToolResult.bind(this);

package/src/api/MessageRuntime.ts

@@ -152,6 +152,8 @@ export class MessageRuntimeImpl implements MessageRuntime {
       }),
       () => this._threadBinding.getState().beginEdit(this._core.getState().id),
     );
+
+    this.__internal_bindMethods();
   }
 
   protected __internal_bindMethods() {

package/src/api/ThreadListItemRuntime.ts

@@ -36,6 +36,9 @@ export type ThreadListItemRuntime = {
     event: ThreadListItemEventType,
     callback: () => void,
   ): Unsubscribe;
+
+  /** @internal */
+  __internal_getRuntime(): ThreadListItemRuntime;
 };
 
 export type ThreadListItemStateBinding = SubscribableWithState<
@@ -51,7 +54,9 @@ export class ThreadListItemRuntimeImpl implements ThreadListItemRuntime {
   constructor(
     private _core: ThreadListItemStateBinding,
     private _threadListBinding: ThreadListRuntimeCoreBinding,
-  ) {}
+  ) {
+    this.__internal_bindMethods();
+  }
 
   protected __internal_bindMethods() {
     this.switchTo = this.switchTo.bind(this);
@@ -142,4 +147,9 @@ export class ThreadListItemRuntimeImpl implements ThreadListItemRuntime {
 
     this._threadListBinding.detach(state.id);
   }
+
+  /** @internal */
+  public __internal_getRuntime(): ThreadListItemRuntime {
+    return this;
+  }
 }