@assistant-ui/core 0.1.3 → 0.1.7

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

@@ -1,10 +1,12 @@
-import type { Unsubscribe } from "../../types";
+import type { Unsubscribe } from "../../types/unsubscribe";
 import {
   LazyMemoizeSubject,
+  NestedSubscriptionSubject,
+} from "../../subscribable/subscribable";
+import {
   SKIP_UPDATE,
   ShallowMemoizeSubject,
-  NestedSubscriptionSubject,
-} from "../../subscribable";
+} from "../../subscribable/subscribable";
 import type { ThreadListRuntimeCore } from "../interfaces/thread-list-runtime-core";
 import {
   ThreadListItemRuntime,

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

@@ -14,11 +14,11 @@ import {
   MessageRuntimeImpl,
   MessageState,
 } from "./message-runtime";
+import { NestedSubscriptionSubject } from "../../subscribable/subscribable";
 import {
-  NestedSubscriptionSubject,
   ShallowMemoizeSubject,
   SKIP_UPDATE,
-} from "../../subscribable";
+} from "../../subscribable/subscribable";
 import type { SubscribableWithState } from "../../subscribable/subscribable";
 import {
   ThreadComposerRuntime,
@@ -30,15 +30,12 @@ import {
   ThreadRuntimePath,
 } from "./paths";
 import type { ThreadListItemState } from "./bindings";
-import type {
-  RunConfig,
-  AppendMessage,
-  ThreadMessage,
-  Unsubscribe,
-} from "../../types";
-import { EventSubscriptionSubject } from "../../subscribable";
+import type { AppendMessage, ThreadMessage } from "../../types/message";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { RunConfig } from "../../types/message";
+import { EventSubscriptionSubject } from "../../subscribable/subscribable";
 import { symbolInnerMessage } from "../utils/external-store-message";
-import { ModelContext } from "../../model-context";
+import { ModelContext } from "../../model-context/types";
 import {
   ChatModelRunOptions,
   ChatModelRunResult,
@@ -263,6 +260,11 @@ export type ThreadRuntime = {
    * 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;
 
   /**
@@ -377,6 +379,7 @@ 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);
@@ -431,10 +434,15 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     return this._threadBinding.getState().startRun(toStartRunConfig(config));
   }
 
-  public unstable_resumeRun(config: CreateResumeRunConfig) {
+  public resumeRun(config: CreateResumeRunConfig) {
     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();
   }

package/src/runtime/base/base-assistant-runtime-core.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";
 import { CompositeContextProvider } from "../../utils/composite-context-provider";
 import type { AssistantRuntimeCore } from "../interfaces/assistant-runtime-core";
 import type { ThreadListRuntimeCore } from "../interfaces/thread-list-runtime-core";

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

@@ -3,13 +3,12 @@ import type {
   CompleteAttachment,
   CreateAttachment,
   PendingAttachment,
-  MessageRole,
-  RunConfig,
-  QuoteInfo,
-  AppendMessage,
-  Unsubscribe,
-} from "../../types";
-import { BaseSubscribable } from "../../subscribable";
+} from "../../types/attachment";
+import type { MessageRole, AppendMessage } from "../../types/message";
+import type { QuoteInfo } from "../../types/quote";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { RunConfig } from "../../types/message";
+import { BaseSubscribable } from "../../subscribable/subscribable";
 import type { AttachmentAdapter } from "../../adapters/attachment";
 import type {
   ComposerRuntimeCore,

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

@@ -1,5 +1,6 @@
-import type { AppendMessage, ThreadMessage, Unsubscribe } from "../../types";
-import type { ModelContextProvider } from "../../model-context";
+import type { AppendMessage, ThreadMessage } from "../../types/message";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { ModelContextProvider } from "../../model-context/types";
 import { getThreadMessageText } from "../../utils/text";
 import {
   ExportedMessageRepository,

package/src/runtime/base/default-edit-composer-runtime-core.ts

@@ -1,4 +1,4 @@
-import type { AppendMessage, ThreadMessage } from "../../types";
+import type { AppendMessage, ThreadMessage } from "../../types/message";
 import { getThreadMessageText } from "../../utils/text";
 import type { AttachmentAdapter } from "../../adapters/attachment";
 import type { DictationAdapter } from "../../adapters/speech";

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

@@ -1,4 +1,4 @@
-import type { AppendMessage } from "../../types";
+import type { AppendMessage } from "../../types/message";
 import type { AttachmentAdapter } from "../../adapters/attachment";
 import type { DictationAdapter } from "../../adapters/speech";
 import type { ThreadComposerRuntimeCore } from "../interfaces/composer-runtime-core";

package/src/runtime/interfaces/assistant-runtime-core.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";
 import type { ThreadListRuntimeCore } from "./thread-list-runtime-core";
 
 export type AssistantRuntimeCore = {

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

@@ -1,11 +1,8 @@
-import type {
-  MessageRole,
-  RunConfig,
-  QuoteInfo,
-  Attachment,
-  CreateAttachment,
-  Unsubscribe,
-} from "../../types";
+import type { MessageRole } from "../../types/message";
+import type { QuoteInfo } from "../../types/quote";
+import type { Attachment, CreateAttachment } from "../../types/attachment";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { RunConfig } from "../../types/message";
 import type { DictationAdapter } from "../../adapters/speech";
 
 export type ComposerRuntimeEventType = "send" | "attachmentAdd";

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

@@ -1,4 +1,4 @@
-import type { Unsubscribe } from "../../types";
+import type { Unsubscribe } from "../../types/unsubscribe";
 import type { ThreadRuntimeCore } from "./thread-runtime-core";
 
 export type ThreadListItemStatus = "archived" | "regular" | "new" | "deleted";

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

@@ -1,11 +1,8 @@
 import type { ReadonlyJSONValue } from "assistant-stream/utils";
-import type { ModelContext } from "../../model-context";
-import type {
-  RunConfig,
-  Unsubscribe,
-  AppendMessage,
-  ThreadMessage,
-} from "../../types";
+import type { ModelContext } from "../../model-context/types";
+import type { Unsubscribe } from "../../types/unsubscribe";
+import type { AppendMessage, ThreadMessage } from "../../types/message";
+import type { RunConfig } from "../../types/message";
 import type { SpeechSynthesisAdapter } from "../../adapters/speech";
 import type {
   ChatModelRunOptions,

package/src/runtime/utils/auto-status.ts

@@ -1,4 +1,4 @@
-import type { MessageStatus } from "../../types";
+import type { MessageStatus } from "../../types/message";
 import { ReadonlyJSONValue } from "assistant-stream/utils";
 
 const symbolAutoStatus = Symbol("autoStatus");

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

@@ -1,17 +1,19 @@
 import type {
   FileMessagePart,
   MessageStatus,
-  MessageTiming,
   ReasoningMessagePart,
-  RunConfig,
   SourceMessagePart,
-  TextMessagePart,
   ThreadAssistantMessagePart,
   ThreadMessage,
   ThreadStep,
   ToolCallMessagePart,
-} from "../../types";
-import type { ModelContext } from "../../model-context";
+} from "../../types/message";
+import type {
+  MessageTiming,
+  RunConfig,
+  TextMessagePart,
+} from "../../types/message";
+import type { ModelContext } from "../../model-context/types";
 import { ReadonlyJSONValue } from "assistant-stream/utils";
 
 export type ChatModelRunUpdate = {

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

@@ -1,4 +1,4 @@
-import type { ThreadMessage } from "../../types";
+import type { ThreadMessage } from "../../types/message";
 
 export const symbolInnerMessage = Symbol("innerMessage");
 const symbolInnerMessages = Symbol("innerMessages");

package/src/runtime/utils/message-repository.ts

@@ -1,4 +1,5 @@
-import type { RunConfig, ThreadMessage } from "../../types";
+import type { ThreadMessage } from "../../types/message";
+import type { RunConfig } from "../../types/message";
 import { generateId, generateOptimisticId } from "../../utils/id";
 import type { ThreadMessageLike } from "./thread-message-like";
 import { getAutoStatus } from "./auto-status";

package/src/runtime/utils/thread-message-like.ts

@@ -1,12 +1,10 @@
 import { parsePartialJsonObject } from "assistant-stream/utils";
 import { generateId } from "../../utils/id";
 import type {
-  MessageTiming,
   ReasoningMessagePart,
   SourceMessagePart,
   ThreadStep,
   MessageStatus,
-  TextMessagePart,
   ImageMessagePart,
   ThreadMessage,
   ThreadAssistantMessagePart,
@@ -14,11 +12,12 @@ import type {
   ThreadUserMessagePart,
   ThreadUserMessage,
   ThreadSystemMessage,
-  CompleteAttachment,
   FileMessagePart,
   DataMessagePart,
   Unstable_AudioMessagePart,
-} from "../../types";
+} from "../../types/message";
+import type { CompleteAttachment } from "../../types/attachment";
+import type { MessageTiming, TextMessagePart } from "../../types/message";
 import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
 
 type DataPrefixedPart = {

package/src/runtimes/external-store/external-store-adapter.ts

@@ -1,4 +1,4 @@
-import type { AppendMessage, ThreadMessage } from "../../types";
+import type { AppendMessage, ThreadMessage } from "../../types/message";
 import type { ThreadMessageLike } from "../../runtime/utils/thread-message-like";
 import type { AttachmentAdapter } from "../../adapters/attachment";
 import type {

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

@@ -1,4 +1,4 @@
-import type { Unsubscribe } from "../../types";
+import type { Unsubscribe } from "../../types/unsubscribe";
 import type { ExternalStoreThreadRuntimeCore } from "./external-store-thread-runtime-core";
 import type {
   ThreadListItemCoreState,

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

@@ -1,4 +1,4 @@
-import type { AppendMessage, ThreadMessage } from "../../types";
+import type { AppendMessage, ThreadMessage } from "../../types/message";
 import type {
   AddToolResultOptions,
   ResumeRunConfig,
@@ -24,7 +24,7 @@ import type {
   ThreadRuntimeCore,
 } from "../../runtime/interfaces/thread-runtime-core";
 import { BaseThreadRuntimeCore } from "../../runtime/base/base-thread-runtime-core";
-import type { ModelContextProvider } from "../../model-context";
+import type { ModelContextProvider } from "../../model-context/types";
 import {
   ExportedMessageRepository,
   MessageRepository,

package/src/runtimes/external-store/thread-message-converter.ts

@@ -1,4 +1,4 @@
-import type { ThreadMessage } from "../../types";
+import type { ThreadMessage } from "../../types/message";
 
 export type ConverterCallback<TIn> = (
   cache: ThreadMessage | undefined,

package/src/runtimes/internal.ts

@@ -17,6 +17,9 @@ export {
 export { ThreadMessageConverter } from "./external-store/thread-message-converter";
 export type { ConverterCallback } from "./external-store/thread-message-converter";
 
+// Readonly Runtime
+export { ReadonlyThreadRuntimeCore } from "./readonly/ReadonlyThreadRuntimeCore";
+
 // Remote Thread List
 export { OptimisticState } from "./remote-thread-list/optimistic-state";
 export { EMPTY_THREAD_CORE } from "./remote-thread-list/empty-thread-core";

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

@@ -1,5 +1,5 @@
 import type { ThreadListRuntimeCore } from "../../runtime/interfaces/thread-list-runtime-core";
-import { BaseSubscribable } from "../../subscribable";
+import { BaseSubscribable } from "../../subscribable/subscribable";
 import type { LocalThreadRuntimeCore } from "./local-thread-runtime-core";
 
 export type LocalThreadFactory = () => LocalThreadRuntimeCore;

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

@@ -16,11 +16,11 @@ import type {
 } from "../../runtime/interfaces/thread-runtime-core";
 import { BaseThreadRuntimeCore } from "../../runtime/base/base-thread-runtime-core";
 import type {
-  RunConfig,
   AppendMessage,
   ThreadAssistantMessage,
-} from "../../types";
-import type { ModelContextProvider } from "../../model-context";
+} from "../../types/message";
+import type { RunConfig } from "../../types/message";
+import type { ModelContextProvider } from "../../model-context/types";
 
 class AbortError extends Error {
   override name = "AbortError";

package/src/runtimes/local/should-continue.ts

@@ -1,4 +1,4 @@
-import type { ThreadAssistantMessage } from "../../types";
+import type { ThreadAssistantMessage } from "../../types/message";
 
 export const shouldContinue = (
   result: ThreadAssistantMessage,

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(