@assistant-ui/react 0.14.29 → 0.15.1

package/src/tests/external-thread-parity.test.tsx

@@ -66,7 +66,7 @@ describe("ExternalThread part status", () => {
       isRunning: true,
     });
     const part = (toolCallId: string) =>
-      aui().thread().message({ id: "a1" }).part({ toolCallId }).getState();
+      aui().thread.message({ id: "a1" }).part({ toolCallId }).getState();
     expect(part("tc1").status).toEqual({ type: "running" });
   });
 
@@ -75,7 +75,7 @@ describe("ExternalThread part status", () => {
       messages: [assistantMessage({ type: "running" }, "ok")],
       isRunning: true,
     });
-    const state = aui().thread().message({ id: "a1" }).getState();
+    const state = aui().thread.message({ id: "a1" }).getState();
     expect(state.parts[0]!.status).toEqual({ type: "complete" });
     expect(state.parts[1]!.status).toEqual({ type: "complete" });
   });
@@ -96,7 +96,7 @@ describe("ExternalThread part status", () => {
       isRunning: false,
     });
     expect(
-      aui().thread().message({ id: "u1" }).part({ index: 0 }).getState().status,
+      aui().thread.message({ id: "u1" }).part({ index: 0 }).getState().status,
     ).toEqual({ type: "complete" });
     expect(
       aui()
@@ -106,7 +106,7 @@ describe("ExternalThread part status", () => {
         .getState().status,
     ).toEqual({ type: "requires-action", reason: "tool-calls" });
     expect(
-      aui().thread().message({ id: "a1" }).part({ index: 0 }).getState().status,
+      aui().thread.message({ id: "a1" }).part({ index: 0 }).getState().status,
     ).toEqual({ type: "complete" });
   });
 });
@@ -120,7 +120,7 @@ describe("ExternalThread unset optional callbacks", () => {
       isRunning: false,
     });
     const part = () =>
-      aui().thread().message({ id: "a1" }).part({ toolCallId: "tc1" });
+      aui().thread.message({ id: "a1" }).part({ toolCallId: "tc1" });
 
     expect(() => part().addToolResult("ok")).toThrow(
       "Runtime does not support tool results (onAddToolResult is not set).",
@@ -128,10 +128,10 @@ describe("ExternalThread unset optional callbacks", () => {
     expect(() => part().resumeToolCall(undefined)).toThrow(
       "Runtime does not support resuming tool calls (onResumeToolCall is not set).",
     );
-    expect(() => aui().thread().resumeRun()).toThrow(
+    expect(() => aui().thread.resumeRun()).toThrow(
       "Runtime does not support resuming runs (onResume is not set).",
     );
-    expect(() => aui().thread().importExternalState({})).toThrow(
+    expect(() => aui().thread.importExternalState({})).toThrow(
       "Runtime does not support importing external states (onLoadExternalState is not set).",
     );
   });
@@ -142,16 +142,14 @@ describe("ExternalThread composer", () => {
     const onNew = vi.fn();
     const { aui } = renderThread({ messages: [], isRunning: false, onNew });
 
-    aui().thread().composer().setText("hello");
-    aui().thread().composer().send();
+    aui().thread.composer().setText("hello");
+    aui().thread.composer().send();
 
     await waitFor(() => expect(onNew).toHaveBeenCalledTimes(1));
     expect(onNew.mock.calls[0]![0].content).toEqual([
       { type: "text", text: "hello" },
     ]);
-    await waitFor(() =>
-      expect(aui().thread().getState().composer.text).toBe(""),
-    );
+    await waitFor(() => expect(aui().thread.getState().composer.text).toBe(""));
   });
 
   it("stamps the thread head as parentId on queue-adapter sends", async () => {
@@ -177,8 +175,8 @@ describe("ExternalThread composer", () => {
       },
     });
 
-    aui().thread().composer().setText("queued");
-    aui().thread().composer().send();
+    aui().thread.composer().setText("queued");
+    aui().thread.composer().send();
 
     await waitFor(() => expect(enqueue).toHaveBeenCalledTimes(1));
     expect(enqueue.mock.calls[0]![0].parentId).toBe("u1");
@@ -188,10 +186,10 @@ describe("ExternalThread composer", () => {
     const onNew = vi.fn();
     const { aui } = renderThread({ messages: [], isRunning: false, onNew });
 
-    aui().thread().composer().send();
-    aui().thread().composer().setText("first");
-    aui().thread().composer().send();
-    aui().thread().composer().send();
+    aui().thread.composer().send();
+    aui().thread.composer().setText("first");
+    aui().thread.composer().send();
+    aui().thread.composer().send();
 
     await new Promise((resolve) => setTimeout(resolve, 0));
     expect(onNew).toHaveBeenCalledTimes(1);
@@ -219,10 +217,10 @@ describe("ExternalThread duplicate message ids", () => {
         ],
       });
 
-      const state = aui().thread().getState();
+      const state = aui().thread.getState();
       expect(state.messages.map((m) => m.id)).toEqual(["u1", "dup"]);
 
-      const dup = aui().thread().message({ id: "dup" }).getState();
+      const dup = aui().thread.message({ id: "dup" }).getState();
       expect(dup.parts[0]).toMatchObject({ type: "text", text: "fresh" });
       expect(warn).toHaveBeenCalledWith(expect.stringContaining('"dup"'));
     } finally {

package/src/tests/local-runtime-queue.test.tsx

@@ -27,8 +27,7 @@ const createCountingAdapter = () => {
 };
 
 const userTexts = (aui: ReturnType<typeof useAui>) =>
-  aui
-    .thread()
+  aui.thread
     .getState()
     .messages.filter((m) => m.role === "user")
     .map((m) =>
@@ -57,8 +56,8 @@ const renderWithRuntime = (adapter: ChatModelAdapter, enableQueue: boolean) => {
 
 const send = async (aui: ReturnType<typeof useAui>, text: string) => {
   await act(async () => {
-    aui.thread().composer().setText(text);
-    aui.thread().composer().send();
+    aui.thread.composer().setText(text);
+    aui.thread.composer().send();
     await flush();
   });
 };
@@ -69,13 +68,12 @@ describe("local runtime message queue", () => {
     const aui = renderWithRuntime(adapter, true);
 
     await send(aui, "first");
-    expect(aui.thread().getState().isRunning).toBe(true);
-    expect(aui.thread().getState().capabilities.queue).toBe(true);
+    expect(aui.thread.getState().isRunning).toBe(true);
+    expect(aui.thread.getState().capabilities.queue).toBe(true);
 
     await send(aui, "second");
     expect(
-      aui
-        .thread()
+      aui.thread
         .composer()
         .getState()
         .queue.map((q) => q.prompt),
@@ -87,7 +85,7 @@ describe("local runtime message queue", () => {
       await flush();
       await flush();
     });
-    expect(aui.thread().composer().getState().queue).toEqual([]);
+    expect(aui.thread.composer().getState().queue).toEqual([]);
     expect(userTexts(aui)).toContain("second");
   });
 
@@ -100,7 +98,7 @@ describe("local runtime message queue", () => {
 
     await send(aui, "a");
     await send(aui, "b");
-    expect(aui.thread().composer().getState().queue).toHaveLength(2);
+    expect(aui.thread.composer().getState().queue).toHaveLength(2);
 
     await act(async () => {
       releases[0]!();
@@ -109,8 +107,7 @@ describe("local runtime message queue", () => {
     });
     expect(getRunCount()).toBe(2);
     expect(
-      aui
-        .thread()
+      aui.thread
         .composer()
         .getState()
         .queue.map((q) => q.prompt),
@@ -124,13 +121,13 @@ describe("local runtime message queue", () => {
     await send(aui, "first");
     await send(aui, "a");
     await send(aui, "b");
-    expect(aui.thread().composer().getState().queue).toHaveLength(2);
+    expect(aui.thread.composer().getState().queue).toHaveLength(2);
 
     await act(async () => {
-      aui.thread().composer().queueItem({ index: 0 }).remove();
+      aui.thread.composer().queueItem({ index: 0 }).remove();
       await flush();
     });
-    const queue = aui.thread().composer().getState().queue;
+    const queue = aui.thread.composer().getState().queue;
     expect(queue).toHaveLength(1);
     expect(queue[0]!.prompt).toBe("b");
   });
@@ -142,15 +139,15 @@ describe("local runtime message queue", () => {
     await send(aui, "first");
     await send(aui, "a");
     await send(aui, "b");
-    expect(aui.thread().composer().getState().queue).toHaveLength(2);
+    expect(aui.thread.composer().getState().queue).toHaveLength(2);
 
     await act(async () => {
-      aui.thread().cancelRun();
+      aui.thread.cancelRun();
       await flush();
       await flush();
     });
 
-    expect(aui.thread().composer().getState().queue).toEqual([]);
+    expect(aui.thread.composer().getState().queue).toEqual([]);
     // cancelling must not start the next queued message
     expect(getRunCount()).toBe(1);
   });
@@ -168,11 +165,11 @@ describe("local runtime message queue", () => {
     // start a second run and queue a message behind it
     await send(aui, "second");
     await send(aui, "queued");
-    expect(aui.thread().composer().getState().queue).toHaveLength(1);
+    expect(aui.thread.composer().getState().queue).toHaveLength(1);
 
     // edit the first message while the run is in progress
     await act(async () => {
-      const message = aui.thread().message({ index: 0 });
+      const message = aui.thread.message({ index: 0 });
       message.composer().beginEdit();
       message.composer().setText("edited");
       message.composer().send();
@@ -180,7 +177,7 @@ describe("local runtime message queue", () => {
     });
 
     // the edit is applied (branches the thread) and the stale queue is cleared
-    expect(aui.thread().composer().getState().queue).toEqual([]);
+    expect(aui.thread.composer().getState().queue).toEqual([]);
   });
 
   it("buffers a send during a regenerate instead of interrupting it", async () => {
@@ -196,22 +193,21 @@ describe("local runtime message queue", () => {
 
     // regenerate the assistant message: a run started outside the queue
     await act(async () => {
-      aui.thread().message({ index: 1 }).reload();
+      aui.thread.message({ index: 1 }).reload();
       await flush();
     });
     expect(getRunCount()).toBe(2);
-    expect(aui.thread().getState().isRunning).toBe(true);
+    expect(aui.thread.getState().isRunning).toBe(true);
 
     // sending now must buffer, not interrupt the regenerate
     await act(async () => {
-      aui.thread().composer().setText("Y");
-      aui.thread().composer().send();
+      aui.thread.composer().setText("Y");
+      aui.thread.composer().send();
       await flush();
     });
     expect(getRunCount()).toBe(2);
     expect(
-      aui
-        .thread()
+      aui.thread
         .composer()
         .getState()
         .queue.map((q) => q.prompt),
@@ -239,7 +235,7 @@ describe("local runtime message queue", () => {
     await send(aui, "first");
     await send(aui, "a");
     await send(aui, "b");
-    expect(aui.thread().composer().getState().queue).toHaveLength(2);
+    expect(aui.thread.composer().getState().queue).toHaveLength(2);
 
     // run 1 settles -> "a" drains (run 2 throws) -> "b" drains (run 3)
     await act(async () => {
@@ -248,18 +244,17 @@ describe("local runtime message queue", () => {
       await flush();
     });
     expect(runCount).toBe(3);
-    expect(aui.thread().composer().getState().queue).toEqual([]);
+    expect(aui.thread.composer().getState().queue).toEqual([]);
 
     // "b" is running; a new send must buffer behind it, not interrupt it
     await act(async () => {
-      aui.thread().composer().setText("c");
-      aui.thread().composer().send();
+      aui.thread.composer().setText("c");
+      aui.thread.composer().send();
       await flush();
     });
     expect(runCount).toBe(3);
     expect(
-      aui
-        .thread()
+      aui.thread
         .composer()
         .getState()
         .queue.map((q) => q.prompt),
@@ -269,7 +264,7 @@ describe("local runtime message queue", () => {
   it("does not expose the queue capability when the flag is off", async () => {
     const { adapter } = createCountingAdapter();
     const aui = renderWithRuntime(adapter, false);
-    expect(aui.thread().getState().capabilities.queue).toBe(false);
+    expect(aui.thread.getState().capabilities.queue).toBe(false);
   });
 
   it("tears the queue down when the flag is toggled off at runtime", async () => {

package/src/tests/messagePartSwitchRace.test.tsx

@@ -2,7 +2,7 @@
 
 import { act, cleanup, render, waitFor } from "@testing-library/react";
 import { flushSync } from "react-dom";
-import { useEffect, useState, type FC } from "react";
+import { Component, useEffect, useState, type FC, type ReactNode } from "react";
 import { afterEach, describe, expect, it, vi } from "vitest";
 import { useAui, useAuiState } from "@assistant-ui/store";
 import { AssistantRuntimeProvider, PartByIndexProvider } from "../context";
@@ -85,16 +85,33 @@ const ChildrenMessage: FC = () => (
 const InvalidPartReader: FC = () => {
   const part = useAuiState((s) => s.part);
   return (
-    <p data-testid="clamped-part">
+    <p data-testid="invalid-part">
       {part.type === "text" ? part.text : part.type}
     </p>
   );
 };
 
+class PartErrorBoundary extends Component<
+  { children: ReactNode },
+  { error: Error | null }
+> {
+  override state: { error: Error | null } = { error: null };
+  static getDerivedStateFromError(error: Error) {
+    return { error };
+  }
+  override render() {
+    if (this.state.error)
+      return <p data-testid="part-error">{this.state.error.message}</p>;
+    return this.props.children;
+  }
+}
+
 const InvalidPartMessage: FC = () => (
-  <PartByIndexProvider index={2}>
-    <InvalidPartReader />
-  </PartByIndexProvider>
+  <PartErrorBoundary>
+    <PartByIndexProvider index={2}>
+      <InvalidPartReader />
+    </PartByIndexProvider>
+  </PartErrorBoundary>
 );
 
 const App: FC<{ MessageComponent?: FC }> = ({ MessageComponent = Message }) => {
@@ -152,21 +169,21 @@ describe("part hooks under a thread-switch race", () => {
     },
   );
 
-  it("warns and clamps an index that was never valid", async () => {
-    const warn = vi.spyOn(console, "warn").mockImplementation(() => {});
+  it("throws to the error boundary for an index that was never valid", async () => {
+    const consoleError = vi
+      .spyOn(console, "error")
+      .mockImplementation(() => {});
     try {
       let view!: ReturnType<typeof render>;
       await act(async () => {
         view = render(<App MessageComponent={InvalidPartMessage} />);
       });
-      expect(warn).toHaveBeenCalledWith(
-        expect.stringContaining(
-          "useClientLookup: Clamped stale index 2 to 1 (length: 2)",
-        ),
+      expect(view.queryByTestId("invalid-part")).toBeNull();
+      expect(view.getByTestId("part-error").textContent).toContain(
+        "useClientLookup: index 2 out of bounds (length: 2)",
       );
-      expect(view.getByTestId("clamped-part").textContent).toBe("world");
     } finally {
-      warn.mockRestore();
+      consoleError.mockRestore();
     }
   });
 });

package/src/unstable/useComposerInput.test.tsx

@@ -18,11 +18,11 @@ const flushTapSync = vi.fn((fn: () => void) => fn());
 
 vi.mock("@assistant-ui/store", () => ({
   useAui: () => ({
-    composer: () => ({
+    composer: {
       getState: () => ({ isEditing: fixture.composer.isEditing }),
       setText: composerSetText,
       send: composerSend,
-    }),
+    },
   }),
   useAuiState: (selector: (s: unknown) => unknown) =>
     selector({

package/src/unstable/useComposerInput.ts

@@ -83,9 +83,9 @@ export function unstable_useComposerInput(
 
   const setText = useCallback(
     (text: string) => {
-      if (!aui.composer().getState().isEditing) return;
+      if (!aui.composer.getState().isEditing) return;
       flushTapSync(() => {
-        aui.composer().setText(text);
+        aui.composer.setText(text);
       });
     },
     [aui],

package/src/unstable/useComposerInputHistory.test.tsx

@@ -12,11 +12,11 @@ const setText = vi.fn();
 
 vi.mock("@assistant-ui/store", () => ({
   useAui: () => ({
-    composer: () => ({
+    composer: {
       getState: () => ({ type: fixture.composerType }),
       setText,
-    }),
-    thread: () => ({ getState: () => ({ messages: fixture.messages }) }),
+    },
+    thread: { getState: () => ({ messages: fixture.messages }) },
     on: (_event: string, cb: () => void) => {
       fixture.switchedToHandlers.push(cb);
       return () => {};

package/src/unstable/useComposerInputHistory.ts

@@ -70,7 +70,7 @@ export function unstable_useComposerInputHistory(): Unstable_ComposerInputHistor
   const browseRef = useRef<BrowseState | null>(null);
 
   useEffect(() => {
-    if (aui.composer().getState().type !== "thread") return undefined;
+    if (aui.composer.getState().type !== "thread") return undefined;
 
     return aui.on("threadListItem.switchedTo", () => {
       browseRef.current = null;
@@ -84,7 +84,7 @@ export function unstable_useComposerInputHistory(): Unstable_ComposerInputHistor
       if (e.nativeEvent.isComposing) return;
       if (e.shiftKey || e.ctrlKey || e.metaKey || e.altKey) return;
       if (popoverCtx && popoverCtx.getActiveAria() !== null) return;
-      if (aui.composer().getState().type !== "thread") return;
+      if (aui.composer.getState().type !== "thread") return;
 
       const textarea = e.currentTarget;
       const { selectionStart, selectionEnd, value } = textarea;
@@ -96,7 +96,7 @@ export function unstable_useComposerInputHistory(): Unstable_ComposerInputHistor
       const browse = browseRef.current;
 
       const commitText = (text: string): void => {
-        flushTapSync(() => aui.composer().setText(text));
+        flushTapSync(() => aui.composer.setText(text));
         // React's controlled-value commit restores the pre-recall caret;
         // reposition after the commit, before paint.
         requestAnimationFrame(() => {
@@ -124,13 +124,13 @@ export function unstable_useComposerInputHistory(): Unstable_ComposerInputHistor
 
         if (!browse) {
           if (value.trim() !== "") return;
-          const history = deriveHistory(aui.thread().getState().messages);
+          const history = deriveHistory(aui.thread.getState().messages);
           if (history.length === 0) return;
           recall(history, 0, value);
           return;
         }
 
-        const history = deriveHistory(aui.thread().getState().messages);
+        const history = deriveHistory(aui.thread.getState().messages);
         const next = browse.cursor + 1;
         if (next >= history.length) {
           e.preventDefault();
@@ -150,7 +150,7 @@ export function unstable_useComposerInputHistory(): Unstable_ComposerInputHistor
         return;
       }
 
-      const history = deriveHistory(aui.thread().getState().messages);
+      const history = deriveHistory(aui.thread.getState().messages);
       recall(history, next, browse.draftSnapshot);
     },
     [aui, popoverCtx],

package/src/unstable/useMentionAdapter.ts

@@ -107,7 +107,7 @@ export function unstable_useMentionAdapter(
   const adapter = useMemo<Unstable_TriggerAdapter>(() => {
     const getModelContextTools = (): Unstable_TriggerItem[] => {
       if (!wantsTools) return [];
-      const ctx = aui.thread().getModelContext();
+      const ctx = aui.thread.getModelContext();
       const tools = ctx.tools;
       if (!tools) return [];
       const formatLabel = toolsConfig?.formatLabel;

package/src/utils/smooth/SmoothContext.tsx

@@ -38,7 +38,7 @@ export const SmoothContextProvider: FC<PropsWithChildren> = ({ children }) => {
   const aui = useAui();
 
   const [context] = useState(() =>
-    makeSmoothContext(aui.part().getState().status),
+    makeSmoothContext(aui.part.getState().status),
   );
 
   // do not wrap if there is an outer SmoothContextProvider

package/src/utils/smooth/useSmooth.test.tsx

@@ -9,7 +9,7 @@ import type {
 
 const part = {};
 vi.mock("@assistant-ui/store", () => ({
-  useAui: () => ({ part: () => part }),
+  useAui: () => ({ part }),
   useAuiState: (selector: () => unknown) => selector(),
 }));
 vi.mock("./SmoothContext", () => ({

package/src/utils/smooth/useSmooth.ts

@@ -181,7 +181,7 @@ export const useSmooth = (
   // of relying on a render-time proxy reference that may be stable
   // across thread swaps.
   const aui = useAui();
-  const part = useAuiState(() => aui.part());
+  const part = useAuiState(() => aui.part);
   const [prevPart, setPrevPart] = useState(part);
   if (part !== prevPart || !text.startsWith(displayedText)) {
     setPrevPart(part);

package/dist/context/react/utils/createStateHookForRuntime.d.ts

@@ -1,34 +0,0 @@
-import { SubscribableRuntime } from "./useRuntimeState.js";
-//#region src/context/react/utils/createStateHookForRuntime.d.ts
-declare function createStateHookForRuntime<TState>(useRuntime: (options: {
-  optional: boolean | undefined;
-}) => SubscribableRuntime<TState> | null): {
-  (): TState;
-  <TSelected>(selector: (state: TState) => TSelected): TSelected;
-  <TSelected>(selector: ((state: TState) => TSelected) | undefined): TSelected | TState;
-  (options: {
-    optional?: false | undefined;
-  }): TState;
-  (options: {
-    optional?: boolean | undefined;
-  }): TState | null;
-  <TSelected>(options: {
-    optional?: false | undefined;
-    selector: (state: TState) => TSelected;
-  }): TSelected;
-  <TSelected>(options: {
-    optional?: false | undefined;
-    selector: ((state: TState) => TSelected) | undefined;
-  }): TSelected | TState;
-  <TSelected>(options: {
-    optional?: boolean | undefined;
-    selector: (state: TState) => TSelected;
-  }): TSelected | null;
-  <TSelected>(options: {
-    optional?: boolean | undefined;
-    selector: ((state: TState) => TSelected) | undefined;
-  }): TSelected | TState | null;
-};
-//#endregion
-export { createStateHookForRuntime };
-//# sourceMappingURL=createStateHookForRuntime.d.ts.map

package/dist/context/react/utils/createStateHookForRuntime.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"createStateHookForRuntime.d.ts","names":[],"sources":["../../../../src/context/react/utils/createStateHookForRuntime.ts"],"mappings":";;iBAKgB,0BAA0B,QACxC,aAAa;EACX;MACI,oBAAoB;MAGD;GAGH,WAAS,WAClB,OAAO,WAAW,YAC5B;GAGmB,WAAS,YACjB,OAAO,WAAW,yBAC7B,YAAY;GAGgB;IAAE;MAAiC;GAGnC;IAC7B;MACE;GAGkB,WAAS;IAC7B;IACA,WAAW,OAAO,WAAW;MAC3B;GAGkB,WAAS;IAC7B;IACA,YAAY,OAAO,WAAW;MAC5B,YAAY;GAGM,WAAS;IAC7B;IACA,WAAW,OAAO,WAAW;MAC3B;GAGkB,WAAS;IAC7B;IACA,YAAY,OAAO,WAAW;MAC5B,YAAY"}

package/dist/context/react/utils/createStateHookForRuntime.js

@@ -1,21 +0,0 @@
-import { useRuntimeStateInternal } from "./useRuntimeState.js";
-//#region src/context/react/utils/createStateHookForRuntime.ts
-function createStateHookForRuntime(useRuntime) {
-	function useStoreHook(param) {
-		let optional = false;
-		let selector;
-		if (typeof param === "function") selector = param;
-		else if (param) {
-			optional = !!param.optional;
-			selector = param.selector;
-		}
-		const store = useRuntime({ optional });
-		if (!store) return null;
-		return useRuntimeStateInternal(store, selector);
-	}
-	return useStoreHook;
-}
-//#endregion
-export { createStateHookForRuntime };
-
-//# sourceMappingURL=createStateHookForRuntime.js.map

package/dist/context/react/utils/createStateHookForRuntime.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"createStateHookForRuntime.js","names":["SubscribableRuntime","useRuntimeStateInternal","createStateHookForRuntime","useRuntime","options","optional","TState","useStoreHook","selector","state","TSelected","param","store"],"sources":["../../../../src/context/react/utils/createStateHookForRuntime.ts"],"sourcesContent":["import {\n  type SubscribableRuntime,\n  useRuntimeStateInternal,\n} from \"./useRuntimeState\";\n\nexport function createStateHookForRuntime<TState>(\n  useRuntime: (options: {\n    optional: boolean | undefined;\n  }) => SubscribableRuntime<TState> | null,\n) {\n  // empty\n  function useStoreHook(): TState;\n\n  // selector\n  function useStoreHook<TSelected>(\n    selector: (state: TState) => TSelected,\n  ): TSelected;\n\n  // selector?\n  function useStoreHook<TSelected>(\n    selector: ((state: TState) => TSelected) | undefined,\n  ): TSelected | TState;\n\n  // optional=false\n  function useStoreHook(options: { optional?: false | undefined }): TState;\n\n  // optional?\n  function useStoreHook(options: {\n    optional?: boolean | undefined;\n  }): TState | null;\n\n  // optional=false, selector\n  function useStoreHook<TSelected>(options: {\n    optional?: false | undefined;\n    selector: (state: TState) => TSelected;\n  }): TSelected;\n\n  // optional=false, selector?\n  function useStoreHook<TSelected>(options: {\n    optional?: false | undefined;\n    selector: ((state: TState) => TSelected) | undefined;\n  }): TSelected | TState;\n\n  // optional?, selector\n  function useStoreHook<TSelected>(options: {\n    optional?: boolean | undefined;\n    selector: (state: TState) => TSelected;\n  }): TSelected | null;\n\n  // optional?, selector?\n  function useStoreHook<TSelected>(options: {\n    optional?: boolean | undefined;\n    selector: ((state: TState) => TSelected) | undefined;\n  }): TSelected | TState | null;\n\n  function useStoreHook<TSelected>(\n    param?:\n      | ((state: TState) => TSelected)\n      | {\n          optional?: boolean | undefined;\n          selector?: ((state: TState) => TSelected) | undefined;\n        },\n  ): TSelected | TState | null {\n    let optional = false;\n    let selector: ((state: TState) => TSelected) | undefined;\n\n    if (typeof param === \"function\") {\n      selector = param;\n    } else if (param) {\n      optional = !!param.optional;\n      selector = param.selector;\n    }\n\n    const store = useRuntime({ optional });\n    if (!store) return null;\n\n    // it is ok to call useRuntimeStateInternal conditionally because it will never become null if its available\n    return useRuntimeStateInternal(store, selector);\n  }\n\n  return useStoreHook;\n}\n"],"mappings":";;AAKA,SAAgBE,0BACdC,YAGA;CA8CA,SAASI,aACPI,OAM2B;EAC3B,IAAIN,WAAW;EACf,IAAIG;EAEJ,IAAI,OAAOG,UAAU,YACnBH,WAAWG;OACN,IAAIA,OAAO;GAChBN,WAAW,CAAC,CAACM,MAAMN;GACnBG,WAAWG,MAAMH;EACnB;EAEA,MAAMI,QAAQT,WAAW,EAAEE,SAAS,CAAC;EACrC,IAAI,CAACO,OAAO,OAAO;EAGnB,OAAOX,wBAAwBW,OAAOJ,QAAQ;CAChD;CAEA,OAAOD;AACT"}

package/dist/legacy-runtime/hooks/AssistantContext.d.ts

@@ -1,74 +0,0 @@
-import { AssistantRuntime } from "../runtime/AssistantRuntime.js";
-//#region src/legacy-runtime/hooks/AssistantContext.d.ts
-/**
- * @deprecated Use {@link useAui} instead. See the {@link https://assistant-ui.com/docs/migrations/v0-12 migration guide}.
- *
- * Hook to access the AssistantRuntime from the current context.
- *
- * The AssistantRuntime provides access to the top-level assistant state and actions,
- * including thread management, tool registration, and configuration.
- *
- * @param options Configuration options
- * @param options.optional Whether the hook should return null if no context is found
- * @returns The AssistantRuntime instance, or null if optional is true and no context exists
- *
- * @example
- * ```tsx
- * // Before:
- * function MyComponent() {
- *   const runtime = useAssistantRuntime();
- *   const handleNewThread = () => {
- *     runtime.switchToNewThread();
- *   };
- *   return <button onClick={handleNewThread}>New Thread</button>;
- * }
- *
- * // After:
- * function MyComponent() {
- *   const aui = useAui();
- *   const handleNewThread = () => {
- *     aui.threads().switchToNewThread();
- *   };
- *   return <button onClick={handleNewThread}>New Thread</button>;
- * }
- * ```
- */
-declare function useAssistantRuntime(options?: {
-  optional?: false | undefined;
-}): AssistantRuntime;
-declare function useAssistantRuntime(options?: {
-  optional?: boolean | undefined;
-}): AssistantRuntime | null;
-/**
- * @deprecated Use {@link useAuiState}: `useAuiState((s) => s.threads)`. See the {@link https://assistant-ui.com/docs/migrations/v0-12 migration guide}.
- */
-declare const useThreadList: {
-  (): import("@assistant-ui/core").ThreadListState;
-  <TSelected>(selector: (state: import("@assistant-ui/core").ThreadListState) => TSelected): TSelected;
-  <TSelected>(selector: ((state: import("@assistant-ui/core").ThreadListState) => TSelected) | undefined): import("@assistant-ui/core").ThreadListState | TSelected;
-  (options: {
-    optional?: false | undefined;
-  }): import("@assistant-ui/core").ThreadListState;
-  (options: {
-    optional?: boolean | undefined;
-  }): import("@assistant-ui/core").ThreadListState | null;
-  <TSelected>(options: {
-    optional?: false | undefined;
-    selector: (state: import("@assistant-ui/core").ThreadListState) => TSelected;
-  }): TSelected;
-  <TSelected>(options: {
-    optional?: false | undefined;
-    selector: ((state: import("@assistant-ui/core").ThreadListState) => TSelected) | undefined;
-  }): import("@assistant-ui/core").ThreadListState | TSelected;
-  <TSelected>(options: {
-    optional?: boolean | undefined;
-    selector: (state: import("@assistant-ui/core").ThreadListState) => TSelected;
-  }): TSelected | null;
-  <TSelected>(options: {
-    optional?: boolean | undefined;
-    selector: ((state: import("@assistant-ui/core").ThreadListState) => TSelected) | undefined;
-  }): import("@assistant-ui/core").ThreadListState | TSelected | null;
-};
-//#endregion
-export { useAssistantRuntime, useThreadList };
-//# sourceMappingURL=AssistantContext.d.ts.map

package/dist/legacy-runtime/hooks/AssistantContext.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"AssistantContext.d.ts","names":[],"sources":["../../../src/legacy-runtime/hooks/AssistantContext.ts"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAwCgB,oBAAoB;EAClC;IACE;iBACY,oBAAoB;EAClC;IACE;;;;cAqBS"}