@assistant-ui/core 0.1.17 → 0.2.0

package/src/runtime/api/message-runtime.ts

@@ -1,7 +1,4 @@
-import type {
-  SpeechState,
-  SubmittedFeedback,
-} from "../interfaces/thread-runtime-core";
+import type { SpeechState } from "../interfaces/thread-runtime-core";
 import { symbolInnerMessage } from "../utils/external-store-message";
 import type {
   ToolCallMessagePartStatus,
@@ -90,10 +87,6 @@ export type MessageState = ThreadMessage & {
    * @deprecated This API is still under active development and might change without notice.
    */
   readonly speech: SpeechState | undefined;
-  /**
-   * @deprecated Use `message.metadata.submittedFeedback` instead. This will be removed in 0.12.0.
-   */
-  readonly submittedFeedback: SubmittedFeedback | undefined;
 };
 
 export type { MessageStateBinding } from "./bindings";

package/src/runtime/api/thread-list-runtime.ts

@@ -28,6 +28,8 @@ export type ThreadListState = {
   readonly threadIds: readonly string[];
   readonly archivedThreadIds: readonly string[];
   readonly isLoading: boolean;
+  readonly isLoadingMore: boolean;
+  readonly hasMore: boolean;
   readonly threadItems: Readonly<
     Record<string, Omit<ThreadListItemState, "isMain" | "threadId">>
   >;
@@ -51,6 +53,7 @@ export type ThreadListRuntime = {
 
   getLoadThreadsPromise(): Promise<void>;
   reload(): Promise<void>;
+  loadMore(): Promise<void>;
 };
 
 const getThreadListState = (
@@ -62,6 +65,8 @@ const getThreadListState = (
     threadIds: threadList.threadIds,
     archivedThreadIds: threadList.archivedThreadIds,
     isLoading: threadList.isLoading,
+    isLoadingMore: threadList.isLoadingMore ?? false,
+    hasMore: threadList.hasMore ?? false,
     threadItems: threadList.threadItems,
   };
 };
@@ -138,6 +143,7 @@ export class ThreadListRuntimeImpl implements ThreadListRuntime {
     this.switchToNewThread = this.switchToNewThread.bind(this);
     this.getLoadThreadsPromise = this.getLoadThreadsPromise.bind(this);
     this.reload = this.reload.bind(this);
+    this.loadMore = this.loadMore.bind(this);
     this.getState = this.getState.bind(this);
     this.subscribe = this.subscribe.bind(this);
     this.getById = this.getById.bind(this);
@@ -162,6 +168,10 @@ export class ThreadListRuntimeImpl implements ThreadListRuntime {
     return this._core.reload?.() ?? RESOLVED_PROMISE;
   }
 
+  public loadMore(): Promise<void> {
+    return this._core.loadMore?.() ?? RESOLVED_PROMISE;
+  }
+
   public getState(): ThreadListState {
     return this._getState();
   }

package/src/runtime/api/thread-runtime.ts

@@ -252,10 +252,6 @@ export type ThreadRuntime = {
    */
   append(message: CreateAppendMessage): void;
 
-  /**
-   * @deprecated pass an object with `parentId` instead. This will be removed in 0.12.0.
-   */
-  startRun(parentId: string | null): void;
   /**
    * Start a new run with the given configuration.
    * @param config The configuration for starting the run
@@ -268,11 +264,6 @@ export type ThreadRuntime = {
    **/
   resumeRun(config: CreateResumeRunConfig): void;
 
-  /**
-   * @deprecated Use `resumeRun` instead.
-   */
-  unstable_resumeRun(config: CreateResumeRunConfig): void;
-
   /**
    * Export the thread state in the external store format.
    * For AI SDK runtimes, this returns the AI SDK message format.
@@ -289,22 +280,10 @@ export type ThreadRuntime = {
    */
   importExternalState(state: any): void;
 
-  /**
-   * Load external state into the thread.
-   * @deprecated Use importExternalState instead. This method will be removed in 0.12.0.
-   * @param state The state to load into the thread
-   */
-  unstable_loadExternalState(state: any): void;
-
   subscribe(callback: () => void): Unsubscribe;
   cancelRun(): void;
   getModelContext(): ModelContext;
 
-  /**
-   * @deprecated This method was renamed to `getModelContext`.
-   */
-  getModelConfig(): ModelContext;
-
   export(): ExportedMessageRepository;
   import(repository: ExportedMessageRepository): void;
 
@@ -396,9 +375,6 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
   protected __internal_bindMethods() {
     this.append = this.append.bind(this);
     this.resumeRun = this.resumeRun.bind(this);
-    this.unstable_resumeRun = this.unstable_resumeRun.bind(this);
-    this.unstable_loadExternalState =
-      this.unstable_loadExternalState.bind(this);
     this.importExternalState = this.importExternalState.bind(this);
     this.exportExternalState = this.exportExternalState.bind(this);
     this.startRun = this.startRun.bind(this);
@@ -418,7 +394,6 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     this.subscribe = this.subscribe.bind(this);
     this.unstable_on = this.unstable_on.bind(this);
     this.getModelContext = this.getModelContext.bind(this);
-    this.getModelConfig = this.getModelConfig.bind(this);
     this.getState = this.getState.bind(this);
   }
 
@@ -444,15 +419,7 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     return this._threadBinding.getState().getModelContext();
   }
 
-  public getModelConfig() {
-    return this.getModelContext();
-  }
-
-  public startRun(configOrParentId: string | null | CreateStartRunConfig) {
-    const config =
-      configOrParentId === null || typeof configOrParentId === "string"
-        ? { parentId: configOrParentId }
-        : configOrParentId;
+  public startRun(config: CreateStartRunConfig) {
     return this._threadBinding.getState().startRun(toStartRunConfig(config));
   }
 
@@ -460,11 +427,6 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     return this._threadBinding.getState().resumeRun(toResumeRunConfig(config));
   }
 
-  /** @deprecated Use `resumeRun` instead. */
-  public unstable_resumeRun(config: CreateResumeRunConfig) {
-    return this.resumeRun(config);
-  }
-
   public exportExternalState() {
     return this._threadBinding.getState().exportExternalState();
   }
@@ -473,10 +435,6 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     this._threadBinding.getState().importExternalState(state);
   }
 
-  public unstable_loadExternalState(state: any) {
-    this._threadBinding.getState().unstable_loadExternalState(state);
-  }
-
   public cancelRun() {
     this._threadBinding.getState().cancelRun();
   }
@@ -575,7 +533,6 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
           const thread = this._threadBinding.getState();
 
           const branches = thread.getBranches(message.id);
-          const submittedFeedback = message.metadata.submittedFeedback;
 
           return {
             ...message,
@@ -590,8 +547,6 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
 
             speech:
               speechState?.messageId === message.id ? speechState : undefined,
-
-            submittedFeedback,
           } satisfies MessageState;
         },
         subscribe: (callback) => this._threadBinding.subscribe(callback),

package/src/runtime/base/base-thread-runtime-core.ts

@@ -61,7 +61,6 @@ export abstract class BaseThreadRuntimeCore implements ThreadRuntimeCore {
   public abstract cancelRun(): void;
   public abstract exportExternalState(): any;
   public abstract importExternalState(state: any): void;
-  public abstract unstable_loadExternalState(state: any): void;
 
   protected _voiceMessages: ThreadMessage[] = [];
   protected _voiceGeneration = 0;

package/src/runtime/interfaces/thread-list-runtime-core.ts

@@ -17,6 +17,8 @@ export type ThreadListItemCoreState = {
 
 export type ThreadListRuntimeCore = {
   readonly isLoading: boolean;
+  readonly isLoadingMore?: boolean;
+  readonly hasMore?: boolean;
   mainThreadId: string;
   newThreadId: string | undefined;
 
@@ -35,6 +37,7 @@ export type ThreadListRuntimeCore = {
 
   getLoadThreadsPromise(): Promise<void>;
   reload?(): Promise<void>;
+  loadMore?(): Promise<void>;
 
   detach(threadId: string): Promise<void>;
   rename(threadId: string, newTitle: string): Promise<void>;

package/src/runtime/interfaces/thread-runtime-core.ts

@@ -194,9 +194,4 @@ export type ThreadRuntimeCore = Readonly<{
     event: E,
     callback: ThreadRuntimeEventCallback<E>,
   ): Unsubscribe;
-
-  /**
-   * @deprecated Use importExternalState instead. This method will be removed in 0.12.0.
-   */
-  unstable_loadExternalState: (state: any) => void;
 }>;

package/src/runtime/utils/chat-model-adapter.ts

@@ -50,11 +50,6 @@ export type ChatModelRunOptions = {
   readonly abortSignal: AbortSignal;
   readonly context: ModelContext;
 
-  /**
-   * @deprecated This field was renamed to `context`.
-   */
-  readonly config: ModelContext;
-
   readonly unstable_assistantMessageId?: string | undefined;
   readonly unstable_threadId?: string | undefined;
   readonly unstable_parentId?: string | null | undefined;

package/src/runtime/utils/external-store-message.ts

@@ -8,14 +8,6 @@ type WithInnerMessages<T> = {
   [symbolInnerMessages]?: T[];
 };
 
-/**
- * @deprecated Use `getExternalStoreMessages` (plural) instead. This function will be removed in 0.12.0.
- */
-export const getExternalStoreMessage = <T>(input: ThreadMessage) => {
-  const withInnerMessages = input as WithInnerMessages<T>;
-  return withInnerMessages[symbolInnerMessage];
-};
-
 const EMPTY_ARRAY: never[] = [];
 
 /**

package/src/runtimes/assistant-transport/utils.ts

@@ -1,31 +1,3 @@
-import {
-  type Tool,
-  toToolsJSONSchema,
-  type ToolJSONSchema,
-} from "assistant-stream";
-
-/**
- * @deprecated Use `toToolsJSONSchema` from `assistant-stream` instead.
- */
-export function toAISDKTools(
-  tools: Record<string, Tool>,
-): Record<string, ToolJSONSchema> {
-  return toToolsJSONSchema(tools, { filter: () => true });
-}
-
-/**
- * @deprecated Use `toToolsJSONSchema` from `assistant-stream` instead, which includes filtering by default.
- */
-export function getEnabledTools(
-  tools: Record<string, Tool>,
-): Record<string, Tool> {
-  return Object.fromEntries(
-    Object.entries(tools).filter(
-      ([, tool]) => !tool.disabled && tool.type !== "backend",
-    ),
-  );
-}
-
 export async function createRequestHeaders(
   headersValue:
     | Record<string, string>

package/src/runtimes/external-store/external-store-thread-runtime-core.ts

@@ -316,10 +316,6 @@ export class ExternalStoreThreadRuntimeCore
     this._store.onLoadExternalState(state);
   }
 
-  public unstable_loadExternalState(state: any): void {
-    this.importExternalState(state);
-  }
-
   public cancelRun(): void {
     if (!this._store.onCancel)
       throw new Error("Runtime does not support cancelling runs.");

package/src/runtimes/local/local-thread-runtime-core.ts

@@ -226,10 +226,6 @@ export class LocalThreadRuntimeCore
     throw new Error("Runtime does not support importing external states.");
   }
 
-  public unstable_loadExternalState(): void {
-    throw new Error("Runtime does not support importing external states.");
-  }
-
   public async startRun(
     { parentId, runConfig }: StartRunConfig,
     runCallback?: ChatModelAdapter["run"],
@@ -405,7 +401,6 @@ export class LocalThreadRuntimeCore
         runConfig: this._lastRunConfig,
         abortSignal,
         context,
-        config: context,
         unstable_assistantMessageId: message.id,
         unstable_threadId: threadId,
         unstable_parentId: parentId,

package/src/runtimes/readonly/ReadonlyThreadRuntimeCore.ts

@@ -104,10 +104,6 @@ export class ReadonlyThreadRuntimeCore
     throw READONLY_THREAD_ERROR;
   }
 
-  unstable_loadExternalState(): void {
-    throw READONLY_THREAD_ERROR;
-  }
-
   composer = {
     attachments: [] as never[],
     attachmentAccept: "*",

package/src/runtimes/remote-thread-list/empty-thread-core.ts

@@ -83,10 +83,6 @@ export const EMPTY_THREAD_CORE: ThreadRuntimeCore = {
     throw EMPTY_THREAD_ERROR;
   },
 
-  unstable_loadExternalState() {
-    throw EMPTY_THREAD_ERROR;
-  },
-
   composer: {
     attachments: [],
     attachmentAccept: "*",

package/src/runtimes/remote-thread-list/remote-thread-state.ts

@@ -1,4 +1,7 @@
-import type { RemoteThreadInitializeResponse } from "./types";
+import type {
+  RemoteThreadInitializeResponse,
+  RemoteThreadMetadata,
+} from "./types";
 
 export type RemoteThreadData =
   | {
@@ -34,8 +37,58 @@ export function createThreadMappingId(id: string): THREAD_MAPPING_ID {
   return id as THREAD_MAPPING_ID;
 }
 
+export const normalizeCursor = (c: string | undefined): string | undefined =>
+  c || undefined;
+
+type ClassifyAccumulator = {
+  threadIds: string[];
+  archivedThreadIds: string[];
+  threadIdMap: Record<string, THREAD_MAPPING_ID>;
+  threadData: Record<THREAD_MAPPING_ID, RemoteThreadData>;
+};
+
+export const classifyThreads = (
+  threads: readonly RemoteThreadMetadata[],
+  acc: ClassifyAccumulator,
+): ClassifyAccumulator => {
+  for (const thread of threads) {
+    if (acc.threadIdMap[thread.remoteId] !== undefined) continue;
+
+    switch (thread.status) {
+      case "regular":
+        acc.threadIds.push(thread.remoteId);
+        break;
+      case "archived":
+        acc.archivedThreadIds.push(thread.remoteId);
+        break;
+      default: {
+        const _exhaustiveCheck: never = thread.status;
+        throw new Error(`Unsupported state: ${_exhaustiveCheck}`);
+      }
+    }
+
+    const mappingId = createThreadMappingId(thread.remoteId);
+    acc.threadIdMap[thread.remoteId] = mappingId;
+    acc.threadData[mappingId] = {
+      id: thread.remoteId,
+      remoteId: thread.remoteId,
+      externalId: thread.externalId,
+      status: thread.status,
+      title: thread.title,
+      custom: thread.custom,
+      initializeTask: Promise.resolve({
+        remoteId: thread.remoteId,
+        externalId: thread.externalId,
+      }),
+    };
+  }
+  return acc;
+};
+
 export type RemoteThreadState = {
   readonly isLoading: boolean;
+  readonly isLoadingMore: boolean;
+  readonly cursor: string | undefined;
   readonly newThreadId: string | undefined;
   readonly threadIds: readonly string[];
   readonly archivedThreadIds: readonly string[];

package/src/runtimes/remote-thread-list/types.ts

@@ -18,10 +18,15 @@ export type RemoteThreadMetadata = {
 
 export type RemoteThreadListResponse = {
   threads: RemoteThreadMetadata[];
+  nextCursor?: string | undefined;
+};
+
+export type RemoteThreadListPageOptions = {
+  after?: string | undefined;
 };
 
 export type RemoteThreadListAdapter = {
-  list(): Promise<RemoteThreadListResponse>;
+  list(params?: RemoteThreadListPageOptions): Promise<RemoteThreadListResponse>;
 
   rename(remoteId: string, newTitle: string): Promise<void>;
   archive(remoteId: string): Promise<void>;

package/src/store/clients/thread-message-client.ts

@@ -106,7 +106,6 @@ export const ThreadMessageClient = resource(
         branchNumber,
         branchCount,
         speech: undefined,
-        submittedFeedback: message.metadata.submittedFeedback,
         isCopied: isCopiedState,
         isHovering: isHoveringState,
       };

package/src/store/runtime-clients/thread-list-runtime-client.ts

@@ -53,6 +53,8 @@ export const ThreadListClient = resource(
         mainThreadId: runtimeState.mainThreadId,
         newThreadId: runtimeState.newThreadId ?? null,
         isLoading: runtimeState.isLoading,
+        isLoadingMore: runtimeState.isLoadingMore,
+        hasMore: runtimeState.hasMore,
         threadIds: runtimeState.threadIds,
         archivedThreadIds: runtimeState.archivedThreadIds,
         threadItems: threadItems.state,
@@ -87,6 +89,7 @@ export const ThreadListClient = resource(
       },
       getLoadThreadsPromise: () => runtime.getLoadThreadsPromise(),
       reload: () => runtime.reload(),
+      loadMore: () => runtime.loadMore(),
       __internal_getAssistantRuntime: () => __internal_assistantRuntime,
     };
   },

package/src/store/runtime-clients/thread-runtime-client.ts

@@ -116,7 +116,6 @@ export const ThreadClient = resource(
       append: runtime.append,
       startRun: runtime.startRun,
       resumeRun: runtime.resumeRun,
-      unstable_resumeRun: runtime.resumeRun,
       cancelRun: runtime.cancelRun,
       getModelContext: runtime.getModelContext,
       export: runtime.export,

package/src/store/scopes/message.ts

@@ -1,9 +1,6 @@
 import type { ThreadMessage } from "../../types/message";
 import type { RunConfig } from "../../types/message";
-import type {
-  SpeechState,
-  SubmittedFeedback,
-} from "../../runtime/interfaces/thread-runtime-core";
+import type { SpeechState } from "../../runtime/interfaces/thread-runtime-core";
 import type { MessageRuntime } from "../../runtime/api/message-runtime";
 import type { ComposerMethods, ComposerState } from "./composer";
 import type { PartMethods, PartState } from "./part";
@@ -31,8 +28,6 @@ export type MessageState = ThreadMessage & {
    * });
    */
   readonly speech: SpeechState | undefined;
-  /** @deprecated Use `message.metadata.submittedFeedback` instead. This will be removed in 0.12.0. */
-  readonly submittedFeedback: SubmittedFeedback | undefined;
   readonly composer: ComposerState;
   readonly parts: readonly PartState[];
   readonly isCopied: boolean;

package/src/store/scopes/thread.ts

@@ -98,11 +98,6 @@ export type ThreadMethods = {
    * @param config The configuration for resuming the run
    */
   resumeRun(config: CreateResumeRunConfig): void;
-
-  /**
-   * @deprecated Use `resumeRun` instead.
-   */
-  unstable_resumeRun(config: CreateResumeRunConfig): void;
   cancelRun(): void;
   getModelContext(): ModelContext;
   export(): ExportedMessageRepository;

package/src/store/scopes/threads.ts

@@ -9,6 +9,8 @@ export type ThreadsState = {
   readonly mainThreadId: string;
   readonly newThreadId: string | null;
   readonly isLoading: boolean;
+  readonly isLoadingMore: boolean;
+  readonly hasMore: boolean;
   readonly threadIds: readonly string[];
   readonly archivedThreadIds: readonly string[];
   readonly threadItems: readonly ThreadListItemState[];
@@ -28,6 +30,7 @@ export type ThreadsMethods = {
   thread(selector: "main"): ThreadMethods;
   getLoadThreadsPromise(): Promise<void>;
   reload(): Promise<void>;
+  loadMore(): Promise<void>;
   __internal_getAssistantRuntime?(): AssistantRuntime;
 };