@assistant-ui/core 0.2.23 → 0.3.0

package/src/react/primitives/threadList/ThreadListItems.tsx

@@ -78,7 +78,7 @@ const ThreadListPrimitiveItemsInner: FC<{
         >
           <RenderChildrenWithAccessor
             getItemState={(aui) =>
-              aui.threads().item({ index, archived }).getState()
+              aui.threads.item({ index, archived }).getState()
             }
           >
             {(getItem) =>

package/src/react/providers/AttachmentByIndexProvider.tsx

@@ -10,7 +10,7 @@ export const MessageAttachmentByIndexProvider: FC<
     attachment: Derived({
       source: "message",
       query: { type: "index", index },
-      get: (aui) => aui.message().attachment({ index }),
+      get: (aui) => aui.message.attachment({ index }),
     }),
   });
 
@@ -26,7 +26,7 @@ export const ComposerAttachmentByIndexProvider: FC<
     attachment: Derived({
       source: "composer",
       query: { type: "index", index },
-      get: (aui) => aui.composer().attachment({ index }),
+      get: (aui) => aui.composer.attachment({ index }),
     }),
   });
 

package/src/react/providers/ChainOfThoughtByIndicesProvider.tsx

@@ -25,7 +25,7 @@ export const ChainOfThoughtByIndicesProvider: FC<
             `ChainOfThought part index ${index} is out of bounds (0..${parts.length - 1})`,
           );
         }
-        return parentAui.message().part({ index: startIndex + index });
+        return parentAui.message.part({ index: startIndex + index });
       },
     }),
   });

package/src/react/providers/ChainOfThoughtPartByIndexProvider.tsx

@@ -1,33 +1,16 @@
-import { useMemo, type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
-import type { PartMethods } from "../../store/scopes/part";
 
 export const ChainOfThoughtPartByIndexProvider: FC<
   PropsWithChildren<{
     index: number;
   }>
 > = ({ index, children }) => {
-  const lastPartRef = useMemo(
-    () => ({ index, current: null as PartMethods | null }),
-    [index],
-  );
   const aui = useAui({
     part: Derived({
       source: "chainOfThought",
       query: { type: "index", index },
-      get: (aui) => {
-        const chainOfThought = aui.chainOfThought();
-        if (
-          index >= chainOfThought.getState().parts.length &&
-          lastPartRef.current
-        ) {
-          return lastPartRef.current;
-        }
-
-        const part = chainOfThought.part({ index });
-        lastPartRef.current = part;
-        return part;
-      },
+      get: (aui) => aui.chainOfThought.part({ index }),
     }),
   });
 

package/src/react/providers/MessageByIdProvider.tsx

@@ -10,12 +10,12 @@ export const MessageByIdProvider: FC<
     message: Derived({
       source: "thread",
       query: { type: "id", id },
-      get: (aui) => aui.thread().message({ id }),
+      get: (aui) => aui.thread.message({ id }),
     }),
     composer: Derived({
       source: "message",
       query: {},
-      get: (aui) => aui.thread().message({ id }).composer(),
+      get: (aui) => aui.thread.message({ id }).composer(),
     }),
   });
 

package/src/react/providers/MessageByIndexProvider.tsx

@@ -10,12 +10,12 @@ export const MessageByIndexProvider: FC<
     message: Derived({
       source: "thread",
       query: { type: "index", index },
-      get: (aui) => aui.thread().message({ index }),
+      get: (aui) => aui.thread.message({ index }),
     }),
     composer: Derived({
       source: "message",
       query: {},
-      get: (aui) => aui.thread().message({ index }).composer(),
+      get: (aui) => aui.thread.message({ index }).composer(),
     }),
   });
 

package/src/react/providers/PartByIndexProvider.tsx

@@ -1,30 +1,16 @@
-import { useMemo, type FC, type PropsWithChildren } from "react";
+import type { FC, PropsWithChildren } from "react";
 import { useAui, AuiProvider, Derived } from "@assistant-ui/store";
-import type { PartMethods } from "../../store/scopes/part";
 
 export const PartByIndexProvider: FC<
   PropsWithChildren<{
     index: number;
   }>
 > = ({ index, children }) => {
-  const lastPartRef = useMemo(
-    () => ({ index, current: null as PartMethods | null }),
-    [index],
-  );
   const aui = useAui({
     part: Derived({
       source: "message",
       query: { type: "index", index },
-      get: (aui) => {
-        const message = aui.message();
-        if (index >= message.getState().parts.length && lastPartRef.current) {
-          return lastPartRef.current;
-        }
-
-        const part = message.part({ index });
-        lastPartRef.current = part;
-        return part;
-      },
+      get: (aui) => aui.message.part({ index }),
     }),
   });
 

package/src/react/providers/QueueItemByIndexProvider.tsx

@@ -13,7 +13,7 @@ export const QueueItemByIndexProvider: FC<QueueItemByIndexProviderProps> = ({
     queueItem: Derived({
       source: "composer",
       query: { index },
-      get: (aui) => aui.composer().queueItem({ index }),
+      get: (aui) => aui.composer.queueItem({ index }),
     }),
   });
 

package/src/react/providers/ReadonlyThreadProvider.tsx

@@ -76,7 +76,7 @@ export const ReadonlyThreadProvider: FC<ReadonlyThreadProvider.Props> = ({
     composer: Derived({
       source: "thread",
       query: {},
-      get: (aui) => aui.thread().composer(),
+      get: (aui) => aui.thread.composer(),
     }),
   });
 

package/src/react/providers/SuggestionByIndexProvider.tsx

@@ -13,7 +13,7 @@ export const SuggestionByIndexProvider: FC<SuggestionByIndexProviderProps> = ({
     suggestion: Derived({
       source: "suggestions",
       query: { index },
-      get: (aui) => aui.suggestions().suggestion({ index }),
+      get: (aui) => aui.suggestions.suggestion({ index }),
     }),
   });
 

package/src/react/providers/ThreadListItemByIndexProvider.tsx

@@ -11,7 +11,7 @@ export const ThreadListItemByIndexProvider: FC<
     threadListItem: Derived({
       source: "threads",
       query: { type: "index", index, archived },
-      get: (aui) => aui.threads().item({ index, archived }),
+      get: (aui) => aui.threads.item({ index, archived }),
     }),
   });
 

package/src/react/runtimes/RemoteThreadListHookInstanceManager.tsx

@@ -147,15 +147,15 @@ export class RemoteThreadListHookInstanceManager extends BaseSubscribable {
       return runtime.threads.main.unstable_on("initialize", () => {
         if (hasInitializedRef.current) return;
 
-        const state = aui.threadListItem().getState();
+        const state = aui.threadListItem.getState();
         if (state.status !== "new") return;
         hasInitializedRef.current = true;
 
-        initPromiseRef.current = aui.threadListItem().initialize();
+        initPromiseRef.current = aui.threadListItem.initialize();
 
         const dispose = runtime.thread.unstable_on("runEnd", () => {
           dispose();
-          aui.threadListItem().generateTitle();
+          aui.threadListItem.generateTitle();
         });
       });
     }, [runtime, aui]);

package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.test.tsx

@@ -6,20 +6,22 @@ import { describe, expect, it, vi } from "vitest";
 import { useAssistantCloudThreadHistoryAdapter } from "./AssistantCloudThreadHistoryAdapter";
 
 const mocks = vi.hoisted(() => {
-  const threadListItem = {
-    getState: () => ({ remoteId: "thread-1" }),
-  };
+  const makeClient = (remoteId: string) =>
+    ({
+      threadListItem: {
+        getState: () => ({ remoteId }),
+      },
+    }) as unknown as import("@assistant-ui/store").AssistantClient;
 
   return {
-    assistantClient: {
-      threadListItem: () => threadListItem,
-    },
+    makeClient,
+    aui: makeClient("thread-1"),
   };
 });
 
 vi.mock("@assistant-ui/store", async (importOriginal) => ({
   ...(await importOriginal<typeof import("@assistant-ui/store")>()),
-  useAui: () => mocks.assistantClient,
+  useAui: () => mocks.aui,
 }));
 
 const makeCloud = () =>
@@ -33,6 +35,7 @@ const makeCloud = () =>
 
 describe("useAssistantCloudThreadHistoryAdapter", () => {
   it("refreshes formatted persistence when the Cloud client changes", async () => {
+    mocks.aui = mocks.makeClient("thread-1");
     const firstCloud = makeCloud();
     const secondCloud = makeCloud();
     const cloudRef = { current: firstCloud };
@@ -66,4 +69,26 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
       format: "test",
     });
   });
+
+  it("resolves the aui client at call time instead of capturing it", async () => {
+    mocks.aui = mocks.makeClient("thread-1");
+    const cloud = makeCloud();
+    const cloudRef = { current: cloud };
+    const { result, rerender } = renderHook(() =>
+      useAssistantCloudThreadHistoryAdapter(cloudRef),
+    );
+
+    await result.current.load();
+    expect(cloud.threads.messages.list).toHaveBeenCalledWith("thread-1", {
+      format: "aui/v0",
+    });
+
+    mocks.aui = mocks.makeClient("thread-2");
+    rerender();
+
+    await result.current.load();
+    expect(cloud.threads.messages.list).toHaveBeenCalledWith("thread-2", {
+      format: "aui/v0",
+    });
+  });
 });

package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.ts

@@ -1,4 +1,4 @@
-import { type RefObject, useState } from "react";
+import { type RefObject, useEffect, useRef, useState } from "react";
 import type {
   GenericThreadHistoryAdapter,
   ThreadHistoryAdapter,
@@ -13,25 +13,31 @@ import {
   createFormattedPersistence,
 } from "assistant-cloud";
 import { auiV0Decode, auiV0Encode } from "./auiV0";
-import { type AssistantClient, useAui } from "@assistant-ui/store";
-import type { ThreadListItemMethods } from "../../../store/scopes/thread-list-item";
+import { type AssistantClient, getClientId, useAui } from "@assistant-ui/store";
 
 const globalPersistence = new WeakMap<
-  ThreadListItemMethods,
+  getClientId.ClientId,
   CloudMessagePersistence
 >();
 
 class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
   private cloudRef: RefObject<AssistantCloud>;
-  private aui: AssistantClient;
+  private auiRef: RefObject<AssistantClient>;
 
-  constructor(cloudRef: RefObject<AssistantCloud>, aui: AssistantClient) {
+  constructor(
+    cloudRef: RefObject<AssistantCloud>,
+    auiRef: RefObject<AssistantClient>,
+  ) {
     this.cloudRef = cloudRef;
-    this.aui = aui;
+    this.auiRef = auiRef;
+  }
+
+  private get aui(): AssistantClient {
+    return this.auiRef.current;
   }
 
   private get _persistence(): CloudMessagePersistence {
-    const key = this.aui.threadListItem();
+    const key = getClientId(this.aui.threadListItem);
     if (!globalPersistence.has(key)) {
       globalPersistence.set(
         key,
@@ -52,11 +58,11 @@ class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
     return {
       // Note: callers must also call reportTelemetry() for run tracking
       async append(item: MessageFormatItem<TMessage>) {
-        const { remoteId } = await adapter.aui.threadListItem().initialize();
+        const { remoteId } = await adapter.aui.threadListItem.initialize();
         await formatted.append(remoteId, item);
       },
       async update(item: MessageFormatItem<TMessage>, localMessageId: string) {
-        const remoteId = adapter.aui.threadListItem().getState().remoteId;
+        const remoteId = adapter.aui.threadListItem.getState().remoteId;
         if (!remoteId) return;
         await formatted.update?.(remoteId, item, localMessageId);
       },
@@ -82,7 +88,7 @@ class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
         );
       },
       async load(): Promise<MessageFormatRepository<TMessage>> {
-        const remoteId = adapter.aui.threadListItem().getState().remoteId;
+        const remoteId = adapter.aui.threadListItem.getState().remoteId;
         if (!remoteId) return { messages: [] };
         return formatted.load(remoteId);
       },
@@ -90,7 +96,7 @@ class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
   }
 
   async append({ parentId, message }: ExportedMessageRepositoryItem) {
-    const { remoteId } = await this.aui.threadListItem().initialize();
+    const { remoteId } = await this.aui.threadListItem.initialize();
     const encoded = auiV0Encode(message);
     await this._persistence.append(
       remoteId,
@@ -110,7 +116,7 @@ class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
       return this.append(item);
     }
     const { message } = item;
-    const remoteId = this.aui.threadListItem().getState().remoteId;
+    const remoteId = this.aui.threadListItem.getState().remoteId;
     if (!remoteId) return;
     const encoded = auiV0Encode(message);
     await this._persistence.update(remoteId, message.id, "aui/v0", encoded);
@@ -127,7 +133,7 @@ class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
   }
 
   async load() {
-    const remoteId = this.aui.threadListItem().getState().remoteId;
+    const remoteId = this.aui.threadListItem.getState().remoteId;
     if (!remoteId) return { messages: [] };
     const messages = await this._persistence.load(remoteId, "aui/v0");
     return {
@@ -150,7 +156,7 @@ class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
   ) {
     if (!this.cloudRef.current.telemetry.enabled) return;
 
-    const remoteId = this.aui.threadListItem().getState().remoteId;
+    const remoteId = this.aui.threadListItem.getState().remoteId;
     if (!remoteId) return;
 
     const extracted = extractRunTelemetry(format, runMessages);
@@ -810,8 +816,12 @@ export function useAssistantCloudThreadHistoryAdapter(
   cloudRef: RefObject<AssistantCloud>,
 ): ThreadHistoryAdapter {
   const aui = useAui();
+  const auiRef = useRef(aui);
+  useEffect(() => {
+    auiRef.current = aui;
+  });
   const [adapter] = useState(
-    () => new AssistantCloudThreadHistoryAdapter(cloudRef, aui),
+    () => new AssistantCloudThreadHistoryAdapter(cloudRef, auiRef),
   );
   return adapter;
 }

package/src/react/runtimes/createRuntimeExtras.test.ts

@@ -6,7 +6,7 @@ type Extras = { value: number; greet: () => string };
 
 const clientWith = (extras: unknown) =>
   ({
-    thread: () => ({ getState: () => ({ extras }) }),
+    thread: { getState: () => ({ extras }) },
   }) as unknown as AssistantClient;
 
 describe("createRuntimeExtras", () => {

package/src/react/runtimes/createRuntimeExtras.ts

@@ -46,7 +46,7 @@ export const createRuntimeExtras = <T extends object>(
     is(extras) ? extras : undefined;
 
   const get = (client: AssistantClient): T =>
-    assert(client.thread().getState().extras);
+    assert(client.thread.getState().extras);
 
   function use<S>(
     select?: (extras: T) => S,

package/src/react/types/scopes/tools.ts

@@ -20,12 +20,6 @@ export type ToolsState = {
   /** Registered tool UIs (renderer + presentation options) keyed by tool name. */
   toolUIs: Record<string, readonly ToolRegistration[]>;
   mcpApp?: McpAppResourceOutput | undefined;
-  /**
-   * @deprecated Use {@link toolUIs} instead, whose entries carry the renderer
-   * alongside its presentation options. This component-only map is kept for
-   * back-compat and will be removed in v0.15.
-   */
-  tools: Record<string, ToolCallMessagePartComponent[]>;
 };
 
 export type ToolsMethods = {

package/src/react/utils/groupParts.ts

@@ -47,10 +47,8 @@ export const GROUPBY_MEMO_KEY: unique symbol = Symbol.for(
  *   `display: "standalone"`). Resolving the registry-driven cases reads the
  *   {@link GroupByContext} passed to the `groupBy` function. Takes precedence
  *   over the `"tool-call"` entry.
- * - `"mcp-app"` — **deprecated**, kept for back-compat. Matches only MCP-app
- *   tool calls. Prefer `"standalone-tool-call"`, which is a superset.
  */
-type GroupPartType = PartState["type"] | "standalone-tool-call" | "mcp-app";
+type GroupPartType = PartState["type"] | "standalone-tool-call";
 
 /**
  * Build a `groupBy` from a `part.type → group-key path` lookup.
@@ -93,10 +91,6 @@ export const groupPartByType = <TKey extends `group-${string}`>(
       if (isStandalone && lookup["standalone-tool-call"] !== undefined) {
         return lookup["standalone-tool-call"]!;
       }
-      // TODO(v0.15): drop the deprecated "mcp-app" key (superseded by "standalone-tool-call").
-      if (isMcpApp && lookup["mcp-app"] !== undefined) {
-        return lookup["mcp-app"]!;
-      }
     }
     return lookup[part.type] ?? [];
   }) as ((part: PartState, context?: GroupByContext) => readonly TKey[]) & {

package/src/store/clients/runtime-adapter.ts

@@ -13,17 +13,17 @@ export const baseRuntimeAdapterTransformScopes = (
   scopes.thread ??= Derived({
     source: "threads",
     query: { type: "main" },
-    get: (aui) => aui.threads().thread("main"),
+    get: (aui) => aui.threads.thread("main"),
   });
   scopes.threadListItem ??= Derived({
     source: "threads",
     query: { type: "main" },
-    get: (aui) => aui.threads().item("main"),
+    get: (aui) => aui.threads.item("main"),
   });
   scopes.composer ??= Derived({
     source: "thread",
     query: {},
-    get: (aui) => aui.threads().thread("main").composer(),
+    get: (aui) => aui.threads.thread("main").composer(),
   });
 
   if (!scopes.modelContext && parent.modelContext.source === null) {

package/src/tests/groupParts.test.ts

@@ -119,25 +119,7 @@ describe("groupPartByType", () => {
     expect(fn(part({ type: "text" }))).toEqual([]);
   });
 
-  it("routes MCP-app tool calls through the 'mcp-app' entry when present", () => {
-    const fn = groupPartByType({
-      "tool-call": ["group-tool"],
-      "mcp-app": [],
-    });
-    const mcpApp = part({
-      type: "tool-call",
-      toolName: "render",
-      mcp: { app: { resourceUri: "ui://my-app" } },
-    } as Partial<PartState>);
-    const regular = part({
-      type: "tool-call",
-      toolName: "search",
-    } as Partial<PartState>);
-    expect(fn(mcpApp)).toEqual([]);
-    expect(fn(regular)).toEqual(["group-tool"]);
-  });
-
-  it("falls back to 'tool-call' for MCP-app parts when 'mcp-app' is absent", () => {
+  it("falls back to 'tool-call' for MCP-app parts when 'standalone-tool-call' is absent", () => {
     const fn = groupPartByType({ "tool-call": ["group-tool"] });
     const mcpApp = part({
       type: "tool-call",
@@ -147,10 +129,10 @@ describe("groupPartByType", () => {
     expect(fn(mcpApp)).toEqual(["group-tool"]);
   });
 
-  it("does not route non-`ui://` tool calls through 'mcp-app'", () => {
+  it("does not treat non-`ui://` tool calls as standalone", () => {
     const fn = groupPartByType({
       "tool-call": ["group-tool"],
-      "mcp-app": ["group-mcp"],
+      "standalone-tool-call": ["group-standalone"],
     });
     const notMcp = part({
       type: "tool-call",
@@ -235,33 +217,6 @@ describe("groupPartByType", () => {
     expect(fn(mcpApp)).toEqual([]);
   });
 
-  it("routes MCP-app parts through the deprecated 'mcp-app' entry", () => {
-    const fn = groupPartByType({
-      "tool-call": ["group-tool"],
-      "mcp-app": ["group-mcp"],
-    });
-    const mcpApp = part({
-      type: "tool-call",
-      toolName: "render",
-      mcp: { app: { resourceUri: "ui://my-app" } },
-    } as Partial<PartState>);
-    expect(fn(mcpApp)).toEqual(["group-mcp"]);
-  });
-
-  it("prefers 'standalone-tool-call' over the deprecated 'mcp-app' entry", () => {
-    const fn = groupPartByType({
-      "tool-call": ["group-tool"],
-      "standalone-tool-call": ["group-standalone"],
-      "mcp-app": ["group-mcp"],
-    });
-    const mcpApp = part({
-      type: "tool-call",
-      toolName: "render",
-      mcp: { app: { resourceUri: "ui://x" } },
-    } as Partial<PartState>);
-    expect(fn(mcpApp)).toEqual(["group-standalone"]);
-  });
-
   it("tags the function with a GROUPBY_MEMO_KEY fingerprint", () => {
     const fn = groupPartByType({ reasoning: ["group-r"] });
     const memoKey = (fn as unknown as { [GROUPBY_MEMO_KEY]: string })[