@assistant-ui/core 0.1.5 → 0.1.7

package/src/runtimes/readonly/ReadonlyThreadRuntimeCore.ts

@@ -0,0 +1,223 @@
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { ThreadMessage } from "../../types/message";
+import type { ThreadRuntimeCore } from "../../runtime/interfaces/thread-runtime-core";
+import { BaseSubscribable } from "../../subscribable/subscribable";
+
+const READONLY_THREAD_ERROR = new Error(
+  "This is a readonly thread. You cannot perform mutations on readonly threads.",
+);
+
+export class ReadonlyThreadRuntimeCore
+  extends BaseSubscribable
+  implements ThreadRuntimeCore
+{
+  private _messages: readonly ThreadMessage[] = [];
+
+  get messages() {
+    return this._messages;
+  }
+
+  setMessages(messages: readonly ThreadMessage[]) {
+    if (this._messages === messages) return;
+    this._messages = messages;
+    this._notifySubscribers();
+  }
+
+  getMessageById(messageId: string) {
+    const idx = this._messages.findIndex((m) => m.id === messageId);
+    if (idx === -1) return undefined;
+    return {
+      parentId: this._messages[idx - 1]?.id ?? null,
+      message: this._messages[idx]!,
+      index: idx,
+    };
+  }
+
+  getBranches(messageId: string) {
+    // no branching in readonly threads
+    const idx = this._messages.findIndex((m) => m.id === messageId);
+    if (idx === -1) return [];
+    return [messageId];
+  }
+
+  switchToBranch(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  append(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  startRun(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  resumeRun(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  cancelRun(): void {
+    // noop - nothing to cancel
+  }
+
+  addToolResult(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  resumeToolCall(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  speak(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  stopSpeaking(): void {
+    // noop
+  }
+
+  submitFeedback(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  getModelContext() {
+    return {};
+  }
+
+  exportExternalState() {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  importExternalState(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  unstable_loadExternalState(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  composer = {
+    attachments: [] as never[],
+    attachmentAccept: "*",
+
+    async addAttachment() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    async removeAttachment() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    isEditing: false as const,
+    canCancel: false,
+    isEmpty: true,
+    text: "",
+
+    setText() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    role: "user" as const,
+
+    setRole() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    runConfig: {},
+
+    setRunConfig() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    async reset() {
+      // noop
+    },
+
+    async clearAttachments() {
+      // noop
+    },
+
+    send() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    cancel() {
+      // noop
+    },
+
+    dictation: undefined,
+
+    startDictation() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    stopDictation() {
+      // noop
+    },
+
+    quote: undefined,
+
+    setQuote() {
+      throw READONLY_THREAD_ERROR;
+    },
+
+    subscribe() {
+      return () => {};
+    },
+
+    unstable_on() {
+      return () => {};
+    },
+  };
+
+  getEditComposer() {
+    return undefined;
+  }
+
+  beginEdit(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  speech = undefined;
+
+  capabilities = {
+    switchToBranch: false,
+    switchBranchDuringRun: false,
+    edit: false,
+    reload: false,
+    cancel: false,
+    unstable_copy: false,
+    speech: false,
+    dictation: false,
+    attachments: false,
+    feedback: false,
+  } as const;
+
+  isDisabled = false;
+  isLoading = false;
+
+  state = null;
+  suggestions = [] as never[];
+  extras = undefined;
+
+  import(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  export() {
+    return {
+      messages: this._messages.map((message, idx) => ({
+        message,
+        parentId: this._messages[idx - 1]?.id ?? null,
+      })),
+    };
+  }
+
+  reset(): void {
+    throw READONLY_THREAD_ERROR;
+  }
+
+  unstable_on(): Unsubscribe {
+    return () => {};
+  }
+}

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

@@ -1,4 +1,4 @@
-import type { ThreadMessage } from "../../types";
+import type { ThreadMessage } from "../../types/message";
 import type { AssistantRuntime } from "../../runtime/api/assistant-runtime";
 import { AssistantStream } from "assistant-stream";
 

package/src/store/clients/chain-of-thought-client.ts

@@ -4,7 +4,7 @@ import type {
   ChainOfThoughtState,
   ChainOfThoughtPart,
 } from "../scopes/chain-of-thought";
-import type { MessagePartStatus } from "../../types";
+import type { MessagePartStatus } from "../../types/message";
 import type { PartMethods } from "../scopes/part";
 
 const COMPLETE_STATUS: MessagePartStatus = Object.freeze({

package/src/store/clients/model-context-client.ts

@@ -1,7 +1,7 @@
 import { resource, tapMemo, tapState } from "@assistant-ui/tap";
 import type { ClientOutput } from "@assistant-ui/store";
-import { CompositeContextProvider } from "../../utils";
-import type { ModelContextState } from "../scopes";
+import { CompositeContextProvider } from "../../utils/composite-context-provider";
+import type { ModelContextState } from "../scopes/model-context";
 
 const version = 1;
 

package/src/store/clients/no-op-composer-client.ts

@@ -1,6 +1,6 @@
 import { resource, tapMemo } from "@assistant-ui/tap";
 import type { ClientOutput } from "@assistant-ui/store";
-import type { ComposerState } from "../scopes";
+import type { ComposerState } from "../scopes/composer";
 
 export const NoOpComposerClient = resource(
   ({ type }: { type: "edit" | "thread" }): ClientOutput<"composer"> => {

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

@@ -1,6 +1,6 @@
 import { resource, tapEffect, tapResource } from "@assistant-ui/tap";
 import type { AssistantRuntime } from "../../runtime/api/assistant-runtime";
-import { ThreadListClient } from "../runtime-clients";
+import { ThreadListClient } from "../runtime-clients/thread-list-runtime-client";
 import {
   tapAssistantClientRef,
   Derived,

package/src/store/clients/suggestions.ts

@@ -1,7 +1,8 @@
 import { resource, tapState, withKey } from "@assistant-ui/tap";
 import type { ClientOutput } from "@assistant-ui/store";
 import { tapClientLookup } from "@assistant-ui/store";
-import type { SuggestionsState, SuggestionState } from "../scopes";
+import type { SuggestionsState } from "../scopes/suggestions";
+import type { SuggestionState } from "../scopes/suggestion";
 
 export type SuggestionConfig =
   | string

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

@@ -1,9 +1,9 @@
 import type {
   ThreadAssistantMessagePart,
   ThreadUserMessagePart,
-  Attachment,
   ThreadMessage,
-} from "../../types";
+} from "../../types/message";
+import type { Attachment } from "../../types/attachment";
 import {
   resource,
   tapMemo,
@@ -13,7 +13,8 @@ import {
 } from "@assistant-ui/tap";
 import type { ClientOutput } from "@assistant-ui/store";
 import { tapClientLookup } from "@assistant-ui/store";
-import type { MessageState, PartState } from "../scopes";
+import type { MessageState } from "../scopes/message";
+import type { PartState } from "../scopes/part";
 import { NoOpComposerClient } from "./no-op-composer-client";
 
 const ThreadMessagePartClient = resource(

package/src/store/index.ts

@@ -1,7 +1,81 @@
 /// <reference path="./scope-registration.ts" />
 
 // scopes
-export * from "./scopes";
+export type {
+  ThreadsState,
+  ThreadsMethods,
+  ThreadsClientSchema,
+} from "./scopes/threads";
+export type {
+  ThreadListItemState,
+  ThreadListItemMethods,
+  ThreadListItemMeta,
+  ThreadListItemEvents,
+  ThreadListItemClientSchema,
+} from "./scopes/thread-list-item";
+export type {
+  ThreadState,
+  ThreadMethods,
+  ThreadMeta,
+  ThreadEvents,
+  ThreadClientSchema,
+} from "./scopes/thread";
+export type {
+  MessageState,
+  MessageMethods,
+  MessageMeta,
+  MessageClientSchema,
+} from "./scopes/message";
+export type {
+  PartState,
+  PartMethods,
+  PartMeta,
+  PartClientSchema,
+} from "./scopes/part";
+export type {
+  ComposerState,
+  ComposerMethods,
+  ComposerMeta,
+  ComposerEvents,
+  ComposerClientSchema,
+} from "./scopes/composer";
+export type {
+  AttachmentState,
+  AttachmentMethods,
+  AttachmentMeta,
+  AttachmentClientSchema,
+} from "./scopes/attachment";
+export type {
+  SuggestionsState,
+  SuggestionsMethods,
+  SuggestionsClientSchema,
+  Suggestion,
+} from "./scopes/suggestions";
+export type {
+  SuggestionState,
+  SuggestionMethods,
+  SuggestionMeta,
+  SuggestionClientSchema,
+} from "./scopes/suggestion";
+export type {
+  ModelContextState,
+  ModelContextMethods,
+  ModelContextClientSchema,
+} from "./scopes/model-context";
+export type {
+  ChainOfThoughtState,
+  ChainOfThoughtMethods,
+  ChainOfThoughtMeta,
+  ChainOfThoughtClientSchema,
+  ChainOfThoughtPart,
+} from "./scopes/chain-of-thought";
 
 // clients
-export * from "./clients";
+export { NoOpComposerClient } from "./clients/no-op-composer-client";
+export { Suggestions, type SuggestionConfig } from "./clients/suggestions";
+export { ChainOfThoughtClient } from "./clients/chain-of-thought-client";
+export {
+  ThreadMessageClient,
+  type ThreadMessageClientProps,
+} from "./clients/thread-message-client";
+export { ModelContext } from "./clients/model-context-client";

package/src/store/internal.ts

@@ -1,4 +1,10 @@
-export * from "./runtime-clients";
+export { AttachmentRuntimeClient } from "./runtime-clients/attachment-runtime-client";
+export { MessagePartClient } from "./runtime-clients/message-part-runtime-client";
+export { ComposerClient } from "./runtime-clients/composer-runtime-client";
+export { MessageClient } from "./runtime-clients/message-runtime-client";
+export { ThreadClient } from "./runtime-clients/thread-runtime-client";
+export { ThreadListItemClient } from "./runtime-clients/thread-list-item-runtime-client";
+export { ThreadListClient } from "./runtime-clients/thread-list-runtime-client";
 export {
   RuntimeAdapterResource,
   baseRuntimeAdapterTransformScopes,

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

@@ -1,6 +1,6 @@
 import { resource } from "@assistant-ui/tap";
 import { type ClientOutput } from "@assistant-ui/store";
-import { AttachmentRuntime } from "../../runtime";
+import { AttachmentRuntime } from "../../runtime/api/attachment-runtime";
 import { tapSubscribable } from "./tap-subscribable";
 
 export const AttachmentRuntimeClient = resource(

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

@@ -1,4 +1,4 @@
-import type { Unsubscribe } from "../../types";
+import type { Unsubscribe } from "../../types/unsubscribe";
 import {
   resource,
   tapMemo,
@@ -15,9 +15,9 @@ import {
 import {
   ComposerRuntime,
   EditComposerRuntime,
-  ComposerRuntimeEventType,
-} from "../../runtime";
-import { ComposerState } from "../scopes";
+} from "../../runtime/api/composer-runtime";
+import { ComposerRuntimeEventType } from "../../runtime/interfaces/composer-runtime-core";
+import { ComposerState } from "../scopes/composer";
 import { AttachmentRuntimeClient } from "./attachment-runtime-client";
 import { tapSubscribable } from "./tap-subscribable";
 

package/src/store/runtime-clients/message-part-runtime-client.ts

@@ -1,6 +1,6 @@
 import { resource } from "@assistant-ui/tap";
 import { type ClientOutput } from "@assistant-ui/store";
-import { MessagePartRuntime } from "../../runtime";
+import { MessagePartRuntime } from "../../runtime/api/message-part-runtime";
 import { tapSubscribable } from "./tap-subscribable";
 
 export const MessagePartClient = resource(

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

@@ -11,11 +11,11 @@ import {
   tapClientLookup,
   tapClientResource,
 } from "@assistant-ui/store";
-import { MessageRuntime } from "../../runtime";
+import { MessageRuntime } from "../../runtime/api/message-runtime";
 import { tapSubscribable } from "./tap-subscribable";
 import { ComposerClient } from "./composer-runtime-client";
 import { MessagePartClient } from "./message-part-runtime-client";
-import { MessageState } from "../scopes";
+import { MessageState } from "../scopes/message";
 import { AttachmentRuntimeClient } from "./attachment-runtime-client";
 
 const MessageAttachmentClientByIndex = resource(

package/src/store/runtime-clients/tap-subscribable.ts

@@ -1,5 +1,5 @@
 import { tapState, tapEffect } from "@assistant-ui/tap";
-import type { SubscribableWithState } from "../../subscribable";
+import type { SubscribableWithState } from "../../subscribable/subscribable";
 
 export const tapSubscribable = <T>(
   subscribable: Omit<SubscribableWithState<T, any>, "path">,

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

@@ -1,7 +1,10 @@
-import type { Unsubscribe } from "../../types";
+import type { Unsubscribe } from "../../types/unsubscribe";
 import { resource, tapEffect } from "@assistant-ui/tap";
 import { type ClientOutput, tapAssistantEmit } from "@assistant-ui/store";
-import { ThreadListItemEventType, ThreadListItemRuntime } from "../../runtime";
+import {
+  ThreadListItemEventType,
+  ThreadListItemRuntime,
+} from "../../runtime/api/thread-list-item-runtime";
 import { tapSubscribable } from "./tap-subscribable";
 
 export const ThreadListItemClient = resource(

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

@@ -4,11 +4,12 @@ import {
   tapClientLookup,
   tapClientResource,
 } from "@assistant-ui/store";
-import { ThreadListRuntime, AssistantRuntime } from "../../runtime";
+import { ThreadListRuntime } from "../../runtime/api/thread-list-runtime";
+import { AssistantRuntime } from "../../runtime/api/assistant-runtime";
 import { tapSubscribable } from "./tap-subscribable";
 import { ThreadListItemClient } from "./thread-list-item-runtime-client";
 import { ThreadClient } from "./thread-runtime-client";
-import { ThreadsState } from "../scopes";
+import { ThreadsState } from "../scopes/threads";
 
 const ThreadListItemClientById = resource(
   ({ runtime, id }: { runtime: ThreadListRuntime; id: string }) => {

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

@@ -1,5 +1,6 @@
-import type { Unsubscribe } from "../../types";
-import { ThreadRuntimeEventType, ThreadRuntime } from "../../runtime";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import { ThreadRuntimeEventType } from "../../runtime/interfaces/thread-runtime-core";
+import { ThreadRuntime } from "../../runtime/api/thread-runtime";
 import {
   resource,
   tapResource,
@@ -17,7 +18,7 @@ import {
 import { ComposerClient } from "./composer-runtime-client";
 import { MessageClient } from "./message-runtime-client";
 import { tapSubscribable } from "./tap-subscribable";
-import { ThreadState } from "../scopes";
+import { ThreadState } from "../scopes/thread";
 
 const MessageClientById = resource(
   ({
@@ -115,7 +116,8 @@ export const ThreadClient = resource(
       composer: () => composer.methods,
       append: runtime.append,
       startRun: runtime.startRun,
-      unstable_resumeRun: runtime.unstable_resumeRun,
+      resumeRun: runtime.resumeRun,
+      unstable_resumeRun: runtime.resumeRun,
       cancelRun: runtime.cancelRun,
       getModelContext: runtime.getModelContext,
       export: runtime.export,

package/src/store/scopes/attachment.ts

@@ -1,5 +1,5 @@
-import type { Attachment } from "../../types";
-import type { AttachmentRuntime } from "../../runtime";
+import type { Attachment } from "../../types/attachment";
+import type { AttachmentRuntime } from "../../runtime/api/attachment-runtime";
 
 export type AttachmentState = Attachment;
 

package/src/store/scopes/chain-of-thought.ts

@@ -1,4 +1,5 @@
-import type { MessagePartStatus, ToolCallMessagePartStatus } from "../../types";
+import type { ToolCallMessagePartStatus } from "../../types/message";
+import type { MessagePartStatus } from "../../types/message";
 import type { PartMethods, PartState } from "./part";
 
 export type ChainOfThoughtPart = Extract<

package/src/store/scopes/composer.ts

@@ -1,11 +1,9 @@
-import type {
-  Attachment,
-  CreateAttachment,
-  MessageRole,
-  RunConfig,
-  QuoteInfo,
-} from "../../types";
-import type { ComposerRuntime, DictationState } from "../../runtime";
+import type { Attachment, CreateAttachment } from "../../types/attachment";
+import type { MessageRole } from "../../types/message";
+import type { QuoteInfo } from "../../types/quote";
+import type { RunConfig } from "../../types/message";
+import type { ComposerRuntime } from "../../runtime/api/composer-runtime";
+import type { DictationState } from "../../runtime/interfaces/composer-runtime-core";
 import type { AttachmentMethods } from "./attachment";
 
 export type ComposerState = {

package/src/store/scopes/message.ts

@@ -1,9 +1,10 @@
-import type { ThreadMessage, RunConfig } from "../../types";
+import type { ThreadMessage } from "../../types/message";
+import type { RunConfig } from "../../types/message";
 import type {
   SpeechState,
   SubmittedFeedback,
-  MessageRuntime,
-} from "../../runtime";
+} 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";
 import type { AttachmentMethods } from "./attachment";

package/src/store/scopes/model-context.ts

@@ -1,5 +1,5 @@
-import type { Unsubscribe } from "../../types";
-import type { ModelContextProvider } from "../../model-context";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { ModelContextProvider } from "../../model-context/types";
 
 export type ModelContextState = Record<string, never>;
 

package/src/store/scopes/part.ts

@@ -2,10 +2,10 @@ import type { ToolResponse } from "assistant-stream";
 import type {
   ThreadUserMessagePart,
   ThreadAssistantMessagePart,
-  MessagePartStatus,
   ToolCallMessagePartStatus,
-} from "../../types";
-import type { MessagePartRuntime } from "../../runtime";
+} from "../../types/message";
+import type { MessagePartStatus } from "../../types/message";
+import type { MessagePartRuntime } from "../../runtime/api/message-part-runtime";
 
 export type PartState = (ThreadUserMessagePart | ThreadAssistantMessagePart) & {
   readonly status: MessagePartStatus | ToolCallMessagePartStatus;

package/src/store/scopes/thread-list-item.ts

@@ -1,7 +1,5 @@
-import type {
-  ThreadListItemRuntime,
-  ThreadListItemStatus,
-} from "../../runtime";
+import type { ThreadListItemRuntime } from "../../runtime/api/thread-list-item-runtime";
+import type { ThreadListItemStatus } from "../../runtime/interfaces/thread-list-runtime-core";
 
 export type ThreadListItemState = {
   readonly id: string;

package/src/store/scopes/thread.ts

@@ -3,14 +3,16 @@ import type {
   RuntimeCapabilities,
   SpeechState,
   ThreadSuggestion,
-  ExportedMessageRepository,
-  ThreadMessageLike,
+} from "../../runtime/interfaces/thread-runtime-core";
+import type { ExportedMessageRepository } from "../../runtime/utils/message-repository";
+import type { ThreadMessageLike } from "../../runtime/utils/thread-message-like";
+import type {
   CreateAppendMessage,
   CreateStartRunConfig,
   CreateResumeRunConfig,
   ThreadRuntime,
-} from "../../runtime";
-import type { ModelContext } from "../../model-context";
+} from "../../runtime/api/thread-runtime";
+import type { ModelContext } from "../../model-context/types";
 import type { MessageMethods, MessageState } from "./message";
 import type { ComposerMethods, ComposerState } from "./composer";
 
@@ -92,6 +94,11 @@ export type ThreadMethods = {
    * Resume a run with the given configuration.
    * @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;

package/src/store/scopes/threads.ts

@@ -1,4 +1,4 @@
-import type { AssistantRuntime } from "../../runtime";
+import type { AssistantRuntime } from "../../runtime/api/assistant-runtime";
 import type {
   ThreadListItemMethods,
   ThreadListItemState,

package/src/subscribable/subscribable.ts

@@ -1,4 +1,4 @@
-import type { Unsubscribe } from "../types";
+import type { Unsubscribe } from "../types/unsubscribe";
 
 // =============================================================================
 // Sentinel

package/src/tests/MessageRepository.test.ts

@@ -3,7 +3,8 @@ import {
   MessageRepository,
   ExportedMessageRepository,
 } from "../runtime/utils/message-repository";
-import type { ThreadMessage, TextMessagePart } from "../types";
+import type { ThreadMessage } from "../types/message";
+import type { TextMessagePart } from "../types/message";
 import type { ThreadMessageLike } from "../runtime/utils/thread-message-like";
 
 // Mock generateId and generateOptimisticId to make tests deterministic