@assistant-ui/core 0.1.9 → 0.1.11

package/src/tests/remote-thread-list-reactive-threadId.test.ts

@@ -0,0 +1,93 @@
+import { describe, it, expect, vi } from "vitest";
+
+/**
+ * Tests for the reactive threadId useEffect logic in useRemoteThreadListRuntime.
+ *
+ * The useEffect compares options.threadId against a prevThreadIdRef and calls
+ * switchToThread or switchToNewThread when it changes. We test this logic
+ * by simulating the ref + comparison pattern.
+ */
+
+// Mirrors the useEffect in useRemoteThreadListRuntimeImpl.
+// Keep in sync if that implementation changes.
+function simulateThreadIdEffect(
+  prevRef: { current: string | undefined },
+  threadId: string | undefined,
+  switchToThread: (id: string) => void,
+  switchToNewThread: () => void,
+) {
+  if (threadId === prevRef.current) return;
+  prevRef.current = threadId;
+  if (threadId) {
+    switchToThread(threadId);
+  } else {
+    switchToNewThread();
+  }
+}
+
+describe("threadId reactive effect", () => {
+  it("does nothing when threadId stays the same", () => {
+    const ref = { current: "thread-1" };
+    const switchToThread = vi.fn();
+    const switchToNewThread = vi.fn();
+
+    simulateThreadIdEffect(ref, "thread-1", switchToThread, switchToNewThread);
+
+    expect(switchToThread).not.toHaveBeenCalled();
+    expect(switchToNewThread).not.toHaveBeenCalled();
+  });
+
+  it("calls switchToThread when threadId changes", () => {
+    const ref = { current: "thread-1" };
+    const switchToThread = vi.fn();
+    const switchToNewThread = vi.fn();
+
+    simulateThreadIdEffect(ref, "thread-2", switchToThread, switchToNewThread);
+
+    expect(switchToThread).toHaveBeenCalledWith("thread-2");
+    expect(ref.current).toBe("thread-2");
+  });
+
+  it("calls switchToNewThread when threadId becomes undefined", () => {
+    const ref = { current: "thread-1" };
+    const switchToThread = vi.fn();
+    const switchToNewThread = vi.fn();
+
+    simulateThreadIdEffect(ref, undefined, switchToThread, switchToNewThread);
+
+    expect(switchToNewThread).toHaveBeenCalledOnce();
+    expect(ref.current).toBeUndefined();
+  });
+
+  it("skips switchToNewThread on first render when threadId is already undefined", () => {
+    const ref = { current: undefined as string | undefined };
+    const switchToThread = vi.fn();
+    const switchToNewThread = vi.fn();
+
+    // First render: undefined === undefined → skipped
+    simulateThreadIdEffect(ref, undefined, switchToThread, switchToNewThread);
+    expect(switchToNewThread).not.toHaveBeenCalled();
+  });
+
+  it("handles full navigation cycle", () => {
+    const ref = { current: undefined as string | undefined };
+    const switchToThread = vi.fn();
+    const switchToNewThread = vi.fn();
+
+    // Mount with thread-1
+    simulateThreadIdEffect(ref, "thread-1", switchToThread, switchToNewThread);
+    expect(switchToThread).toHaveBeenCalledWith("thread-1");
+
+    // Navigate to thread-2
+    simulateThreadIdEffect(ref, "thread-2", switchToThread, switchToNewThread);
+    expect(switchToThread).toHaveBeenCalledWith("thread-2");
+
+    // Navigate to new thread
+    simulateThreadIdEffect(ref, undefined, switchToThread, switchToNewThread);
+    expect(switchToNewThread).toHaveBeenCalledOnce();
+
+    // Navigate back to thread-1
+    simulateThreadIdEffect(ref, "thread-1", switchToThread, switchToNewThread);
+    expect(switchToThread).toHaveBeenCalledTimes(3);
+  });
+});

package/src/tests/thread-list-runtime-getLoadThreadsPromise.test.ts

@@ -0,0 +1,75 @@
+import { describe, it, expect } from "vitest";
+import { ThreadListRuntimeImpl } from "../runtime/api/thread-list-runtime";
+import type { ThreadListRuntimeCore } from "../runtime/interfaces/thread-list-runtime-core";
+import { EMPTY_THREAD_CORE } from "../runtimes/remote-thread-list/empty-thread-core";
+
+const MAIN_THREAD_ID = "main-thread";
+
+const createMockCore = (
+  loadPromise: Promise<void> = Promise.resolve(),
+): ThreadListRuntimeCore => ({
+  isLoading: false,
+  mainThreadId: MAIN_THREAD_ID,
+  newThreadId: undefined,
+  threadIds: [MAIN_THREAD_ID],
+  archivedThreadIds: [],
+  threadItems: {
+    [MAIN_THREAD_ID]: {
+      id: MAIN_THREAD_ID,
+      remoteId: MAIN_THREAD_ID,
+      externalId: undefined,
+      status: "regular",
+      title: undefined,
+    },
+  },
+  getMainThreadRuntimeCore: () => EMPTY_THREAD_CORE,
+  getThreadRuntimeCore: () => EMPTY_THREAD_CORE,
+  getItemById: (id) => {
+    if (id === MAIN_THREAD_ID) {
+      return {
+        id: MAIN_THREAD_ID,
+        remoteId: MAIN_THREAD_ID,
+        externalId: undefined,
+        status: "regular",
+        title: undefined,
+      };
+    }
+    return undefined;
+  },
+  switchToThread: () => Promise.resolve(),
+  switchToNewThread: () => Promise.resolve(),
+  getLoadThreadsPromise: () => loadPromise,
+  detach: () => Promise.resolve(),
+  rename: () => Promise.resolve(),
+  archive: () => Promise.resolve(),
+  unarchive: () => Promise.resolve(),
+  delete: () => Promise.resolve(),
+  initialize: () => Promise.resolve({ remoteId: "r", externalId: undefined }),
+  generateTitle: () => Promise.resolve(),
+  subscribe: () => () => {},
+});
+
+describe("ThreadListRuntime.getLoadThreadsPromise", () => {
+  it("returns a Promise that resolves", async () => {
+    const runtime = new ThreadListRuntimeImpl(createMockCore());
+    const result = runtime.getLoadThreadsPromise();
+    expect(result).toBeInstanceOf(Promise);
+    await expect(result).resolves.toBeUndefined();
+  });
+
+  it("delegates to the core implementation", async () => {
+    let resolved = false;
+    const loadPromise = new Promise<void>((resolve) => {
+      setTimeout(() => {
+        resolved = true;
+        resolve();
+      }, 10);
+    });
+
+    const runtime = new ThreadListRuntimeImpl(createMockCore(loadPromise));
+
+    expect(resolved).toBe(false);
+    await runtime.getLoadThreadsPromise();
+    expect(resolved).toBe(true);
+  });
+});

package/src/types/message.ts

@@ -40,6 +40,7 @@ export type FileMessagePart = {
   readonly filename?: string;
   readonly data: string;
   readonly mimeType: string;
+  readonly parentId?: string;
 };
 
 export type Unstable_AudioMessagePart = {

package/dist/react/model-context/makeInteractable.d.ts

@@ -1,10 +0,0 @@
-import { FC } from "react";
-import { type UseInteractableConfig } from "./useInteractable.js";
-export type InteractableConfig<TState> = UseInteractableConfig<TState> & {
-    name: string;
-};
-export type AssistantInteractable = FC & {
-    unstable_interactable: InteractableConfig<any>;
-};
-export declare const makeInteractable: <TState>(config: InteractableConfig<TState>) => AssistantInteractable;
-//# sourceMappingURL=makeInteractable.d.ts.map

package/dist/react/model-context/makeInteractable.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"makeInteractable.d.ts","sourceRoot":"","sources":["../../../src/react/model-context/makeInteractable.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC;AAC3B,OAAO,EAAmB,KAAK,qBAAqB,EAAE,6BAA0B;AAEhF,MAAM,MAAM,kBAAkB,CAAC,MAAM,IAAI,qBAAqB,CAAC,MAAM,CAAC,GAAG;IACvE,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,EAAE,GAAG;IACvC,qBAAqB,EAAE,kBAAkB,CAAC,GAAG,CAAC,CAAC;CAChD,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAI,MAAM,EACrC,QAAQ,kBAAkB,CAAC,MAAM,CAAC,0BAQnC,CAAC"}

package/dist/react/model-context/makeInteractable.js

@@ -1,10 +0,0 @@
-import { useInteractable } from "./useInteractable.js";
-export const makeInteractable = (config) => {
-    const Interactable = () => {
-        useInteractable(config.name, config);
-        return null;
-    };
-    Interactable.unstable_interactable = config;
-    return Interactable;
-};
-//# sourceMappingURL=makeInteractable.js.map

package/dist/react/model-context/makeInteractable.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"makeInteractable.js","sourceRoot":"","sources":["../../../src/react/model-context/makeInteractable.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAA8B,6BAA0B;AAUhF,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAC9B,MAAkC,EAClC,EAAE;IACF,MAAM,YAAY,GAA0B,GAAG,EAAE;QAC/C,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;IACF,YAAY,CAAC,qBAAqB,GAAG,MAAM,CAAC;IAC5C,OAAO,YAAY,CAAC;AACtB,CAAC,CAAC"}

package/dist/react/model-context/useInteractable.d.ts

@@ -1,16 +0,0 @@
-import type { InteractableStateSchema } from "../types/scopes/interactables.js";
-export type UseInteractableConfig<TState> = {
-    description: string;
-    stateSchema: InteractableStateSchema;
-    initialState: TState;
-    id?: string;
-    selected?: boolean;
-};
-type StateUpdater<TState> = TState | ((prev: TState) => TState);
-export type UseInteractableMetadata = {
-    id: string;
-    setSelected: (selected: boolean) => void;
-};
-export declare const useInteractable: <TState>(name: string, config: UseInteractableConfig<TState>) => [TState, (updater: StateUpdater<TState>) => void, UseInteractableMetadata];
-export {};
-//# sourceMappingURL=useInteractable.d.ts.map

package/dist/react/model-context/useInteractable.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"useInteractable.d.ts","sourceRoot":"","sources":["../../../src/react/model-context/useInteractable.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,uBAAuB,EAAE,yCAAsC;AAE7E,MAAM,MAAM,qBAAqB,CAAC,MAAM,IAAI;IAC1C,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,uBAAuB,CAAC;IACrC,YAAY,EAAE,MAAM,CAAC;IACrB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB,CAAC;AAEF,KAAK,YAAY,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,CAAC,CAAC;AAEhE,MAAM,MAAM,uBAAuB,GAAG;IACpC,EAAE,EAAE,MAAM,CAAC;IACX,WAAW,EAAE,CAAC,QAAQ,EAAE,OAAO,KAAK,IAAI,CAAC;CAC1C,CAAC;AAEF,eAAO,MAAM,eAAe,GAAI,MAAM,EACpC,MAAM,MAAM,EACZ,QAAQ,qBAAqB,CAAC,MAAM,CAAC,KACpC,CACD,MAAM,EACN,CAAC,OAAO,EAAE,YAAY,CAAC,MAAM,CAAC,KAAK,IAAI,EACvC,uBAAuB,CA+CxB,CAAC"}

package/dist/react/model-context/useInteractable.js

@@ -1,36 +0,0 @@
-import { useEffect, useCallback, useId, useRef } from "react";
-import { useAui, useAuiState } from "@assistant-ui/store";
-export const useInteractable = (name, config) => {
-    const aui = useAui();
-    const autoId = useId().replace(/[^a-zA-Z0-9]/g, "");
-    const id = config.id ?? autoId;
-    const stateSchemaRef = useRef(config.stateSchema);
-    stateSchemaRef.current = config.stateSchema;
-    const initialStateRef = useRef(config.initialState);
-    initialStateRef.current = config.initialState;
-    useEffect(() => {
-        return aui.interactables().register({
-            id,
-            name,
-            description: config.description,
-            stateSchema: stateSchemaRef.current,
-            initialState: initialStateRef.current,
-            selected: config.selected,
-        });
-    }, [aui, id, name, config.description, config.selected]);
-    const state = useAuiState((s) => s.interactables.definitions[id]?.state) ??
-        config.initialState;
-    const setState = useCallback((updater) => {
-        aui.interactables().setState(id, (prev) => {
-            if (typeof updater === "function") {
-                return updater(prev);
-            }
-            return updater;
-        });
-    }, [aui, id]);
-    const setSelected = useCallback((selected) => {
-        aui.interactables().setSelected(id, selected);
-    }, [aui, id]);
-    return [state, setState, { id, setSelected }];
-};
-//# sourceMappingURL=useInteractable.js.map

package/dist/react/model-context/useInteractable.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"useInteractable.js","sourceRoot":"","sources":["../../../src/react/model-context/useInteractable.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAkB1D,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,IAAY,EACZ,MAAqC,EAKrC,EAAE;IACF,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IAErB,MAAM,MAAM,GAAG,KAAK,EAAE,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;IACpD,MAAM,EAAE,GAAG,MAAM,CAAC,EAAE,IAAI,MAAM,CAAC;IAE/B,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAClD,cAAc,CAAC,OAAO,GAAG,MAAM,CAAC,WAAW,CAAC;IAC5C,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACpD,eAAe,CAAC,OAAO,GAAG,MAAM,CAAC,YAAY,CAAC;IAE9C,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,GAAG,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC;YAClC,EAAE;YACF,IAAI;YACJ,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,WAAW,EAAE,cAAc,CAAC,OAAO;YACnC,YAAY,EAAE,eAAe,CAAC,OAAO;YACrC,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEzD,MAAM,KAAK,GACR,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,KAAK,CAAY;QACtE,MAAM,CAAC,YAAY,CAAC;IAEtB,MAAM,QAAQ,GAAG,WAAW,CAC1B,CAAC,OAA6B,EAAE,EAAE;QAChC,GAAG,CAAC,aAAa,EAAE,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,EAAE;YACxC,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;gBAClC,OAAQ,OAAoC,CAAC,IAAc,CAAC,CAAC;YAC/D,CAAC;YACD,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,GAAG,EAAE,EAAE,CAAC,CACV,CAAC;IAEF,MAAM,WAAW,GAAG,WAAW,CAC7B,CAAC,QAAiB,EAAE,EAAE;QACpB,GAAG,CAAC,aAAa,EAAE,CAAC,WAAW,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;IAChD,CAAC,EACD,CAAC,GAAG,EAAE,EAAE,CAAC,CACV,CAAC;IAEF,OAAO,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC,CAAC;AAChD,CAAC,CAAC"}

package/src/react/model-context/makeInteractable.ts

@@ -1,21 +0,0 @@
-import { FC } from "react";
-import { useInteractable, type UseInteractableConfig } from "./useInteractable";
-
-export type InteractableConfig<TState> = UseInteractableConfig<TState> & {
-  name: string;
-};
-
-export type AssistantInteractable = FC & {
-  unstable_interactable: InteractableConfig<any>;
-};
-
-export const makeInteractable = <TState>(
-  config: InteractableConfig<TState>,
-) => {
-  const Interactable: AssistantInteractable = () => {
-    useInteractable(config.name, config);
-    return null;
-  };
-  Interactable.unstable_interactable = config;
-  return Interactable;
-};

package/src/react/model-context/useInteractable.ts

@@ -1,73 +0,0 @@
-import { useEffect, useCallback, useId, useRef } from "react";
-import { useAui, useAuiState } from "@assistant-ui/store";
-import type { InteractableStateSchema } from "../types/scopes/interactables";
-
-export type UseInteractableConfig<TState> = {
-  description: string;
-  stateSchema: InteractableStateSchema;
-  initialState: TState;
-  id?: string;
-  selected?: boolean;
-};
-
-type StateUpdater<TState> = TState | ((prev: TState) => TState);
-
-export type UseInteractableMetadata = {
-  id: string;
-  setSelected: (selected: boolean) => void;
-};
-
-export const useInteractable = <TState>(
-  name: string,
-  config: UseInteractableConfig<TState>,
-): [
-  TState,
-  (updater: StateUpdater<TState>) => void,
-  UseInteractableMetadata,
-] => {
-  const aui = useAui();
-
-  const autoId = useId().replace(/[^a-zA-Z0-9]/g, "");
-  const id = config.id ?? autoId;
-
-  const stateSchemaRef = useRef(config.stateSchema);
-  stateSchemaRef.current = config.stateSchema;
-  const initialStateRef = useRef(config.initialState);
-  initialStateRef.current = config.initialState;
-
-  useEffect(() => {
-    return aui.interactables().register({
-      id,
-      name,
-      description: config.description,
-      stateSchema: stateSchemaRef.current,
-      initialState: initialStateRef.current,
-      selected: config.selected,
-    });
-  }, [aui, id, name, config.description, config.selected]);
-
-  const state =
-    (useAuiState((s) => s.interactables.definitions[id]?.state) as TState) ??
-    config.initialState;
-
-  const setState = useCallback(
-    (updater: StateUpdater<TState>) => {
-      aui.interactables().setState(id, (prev) => {
-        if (typeof updater === "function") {
-          return (updater as (prev: TState) => TState)(prev as TState);
-        }
-        return updater;
-      });
-    },
-    [aui, id],
-  );
-
-  const setSelected = useCallback(
-    (selected: boolean) => {
-      aui.interactables().setSelected(id, selected);
-    },
-    [aui, id],
-  );
-
-  return [state, setState, { id, setSelected }];
-};