@assistant-ui/react 0.5.91 → 0.5.93

package/dist/runtimes/local/LocalRuntimeOptions.mjs

@@ -0,0 +1,16 @@
+const splitLocalRuntimeOptions = (options) => {
+  const { initialMessages, maxToolRoundtrips, maxSteps, adapters, ...rest } = options;
+  return {
+    localRuntimeOptions: {
+      initialMessages,
+      maxToolRoundtrips,
+      maxSteps,
+      adapters
+    },
+    otherOptions: rest
+  };
+};
+export {
+  splitLocalRuntimeOptions
+};
+//# sourceMappingURL=LocalRuntimeOptions.mjs.map

package/dist/runtimes/local/LocalRuntimeOptions.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/runtimes/local/LocalRuntimeOptions.tsx"],"sourcesContent":["import type { CoreMessage } from \"../../types\";\nimport { AttachmentAdapter } from \"../attachment/AttachmentAdapter\";\nimport { FeedbackAdapter } from \"../feedback/FeedbackAdapter\";\nimport { SpeechSynthesisAdapter } from \"../speech/SpeechAdapterTypes\";\nimport { ChatModelAdapter } from \"./ChatModelAdapter\";\n\nexport type LocalRuntimeOptionsBase = {\n  maxSteps?: number | undefined;\n  /**\n   * @deprecated Use `maxSteps` (which is `maxToolRoundtrips` + 1; if you set `maxToolRoundtrips` to 2, set `maxSteps` to 3) instead. This field will be removed in v0.6.\n   */\n  maxToolRoundtrips?: number | undefined;\n  adapters: {\n    chatModel: ChatModelAdapter;\n    attachments?: AttachmentAdapter | undefined;\n    speech?: SpeechSynthesisAdapter | undefined;\n    feedback?: FeedbackAdapter | undefined;\n  };\n};\n\n// TODO align LocalRuntimeOptions with LocalRuntimeOptionsBase\nexport type LocalRuntimeOptions = Omit<LocalRuntimeOptionsBase, \"adapters\"> & {\n  initialMessages?: readonly CoreMessage[] | undefined;\n  adapters?: Omit<LocalRuntimeOptionsBase[\"adapters\"], \"chatModel\"> | undefined;\n};\n\nexport const splitLocalRuntimeOptions = <T extends LocalRuntimeOptions>(\n  options: T,\n) => {\n  const { initialMessages, maxToolRoundtrips, maxSteps, adapters, ...rest } =\n    options;\n\n  return {\n    localRuntimeOptions: {\n      initialMessages,\n      maxToolRoundtrips,\n      maxSteps,\n      adapters,\n    },\n    otherOptions: rest,\n  };\n};\n"],"mappings":"AA0BO,MAAM,2BAA2B,CACtC,YACG;AACH,QAAM,EAAE,iBAAiB,mBAAmB,UAAU,UAAU,GAAG,KAAK,IACtE;AAEF,SAAO;AAAA,IACL,qBAAqB;AAAA,MACnB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,IACA,cAAc;AAAA,EAChB;AACF;","names":[]}

package/dist/runtimes/local/LocalThreadManagerRuntimeCore.d.ts

@@ -0,0 +1,33 @@
+import type { Unsubscribe } from "../../types";
+import { ThreadManagerMetadata, ThreadManagerRuntimeCore } from "../core/ThreadManagerRuntimeCore";
+import { ExportedMessageRepository } from "../utils/MessageRepository";
+import { LocalThreadRuntimeCore } from "./LocalThreadRuntimeCore";
+export type LocalThreadData = {
+    data: ExportedMessageRepository;
+    metadata: ThreadManagerMetadata;
+    isArchived: boolean;
+};
+export type LocalThreadFactory = (threadId: string, data: ExportedMessageRepository) => LocalThreadRuntimeCore;
+export declare class LocalThreadManagerRuntimeCore implements ThreadManagerRuntimeCore {
+    private _threadFactory;
+    private _threadData;
+    private _threads;
+    private _archivedThreads;
+    get threads(): readonly ThreadManagerMetadata[];
+    get archivedThreads(): readonly ThreadManagerMetadata[];
+    private _mainThread;
+    get mainThread(): LocalThreadRuntimeCore;
+    constructor(_threadFactory: LocalThreadFactory);
+    switchToThread(threadId: string): void;
+    switchToNewThread(): void;
+    private _performThreadSwitch;
+    private _performMoveOp;
+    rename(threadId: string, newTitle: string): Promise<void>;
+    archive(threadId: string): Promise<void>;
+    unarchive(threadId: string): Promise<void>;
+    delete(threadId: string): Promise<void>;
+    private _subscriptions;
+    subscribe(callback: () => void): Unsubscribe;
+    private _notifySubscribers;
+}
+//# sourceMappingURL=LocalThreadManagerRuntimeCore.d.ts.map

package/dist/runtimes/local/LocalThreadManagerRuntimeCore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LocalThreadManagerRuntimeCore.d.ts","sourceRoot":"","sources":["../../../src/runtimes/local/LocalThreadManagerRuntimeCore.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/C,OAAO,EACL,qBAAqB,EACrB,wBAAwB,EACzB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,EAAE,yBAAyB,EAAE,MAAM,4BAA4B,CAAC;AAEvE,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAElE,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,EAAE,yBAAyB,CAAC;IAChC,QAAQ,EAAE,qBAAqB,CAAC;IAChC,UAAU,EAAE,OAAO,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG,CAC/B,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,yBAAyB,KAC5B,sBAAsB,CAAC;AAE5B,qBAAa,6BAA8B,YAAW,wBAAwB;IAoBhE,OAAO,CAAC,cAAc;IAnBlC,OAAO,CAAC,WAAW,CAAsC;IAEzD,OAAO,CAAC,QAAQ,CAAwC;IACxD,OAAO,CAAC,gBAAgB,CAAwC;IAEhE,IAAW,OAAO,qCAEjB;IAED,IAAW,eAAe,qCAEzB;IAED,OAAO,CAAC,WAAW,CAAyB;IAE5C,IAAW,UAAU,IAAI,sBAAsB,CAE9C;gBAEmB,cAAc,EAAE,kBAAkB;IAW/C,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAUtC,iBAAiB,IAAI,IAAI;IAYhC,OAAO,CAAC,oBAAoB;IAc5B,OAAO,CAAC,cAAc;IAyCT,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBzD,OAAO,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxC,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI1C,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIpD,OAAO,CAAC,cAAc,CAAyB;IAExC,SAAS,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,WAAW;IAKnD,OAAO,CAAC,kBAAkB;CAG3B"}

package/dist/runtimes/local/LocalThreadManagerRuntimeCore.js

@@ -0,0 +1,143 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var LocalThreadManagerRuntimeCore_exports = {};
+__export(LocalThreadManagerRuntimeCore_exports, {
+  LocalThreadManagerRuntimeCore: () => LocalThreadManagerRuntimeCore
+});
+module.exports = __toCommonJS(LocalThreadManagerRuntimeCore_exports);
+var import_idUtils = require("../../utils/idUtils");
+class LocalThreadManagerRuntimeCore {
+  constructor(_threadFactory) {
+    this._threadFactory = _threadFactory;
+    const threadId = (0, import_idUtils.generateId)();
+    this._threadData.set(threadId, {
+      data: { messages: [] },
+      metadata: { threadId },
+      isArchived: false
+    });
+    this._threads = [{ threadId }];
+    this._mainThread = this._threadFactory(threadId, { messages: [] });
+  }
+  _threadData = /* @__PURE__ */ new Map();
+  _threads = [];
+  _archivedThreads = [];
+  get threads() {
+    return this._threads;
+  }
+  get archivedThreads() {
+    return this._archivedThreads;
+  }
+  _mainThread;
+  get mainThread() {
+    return this._mainThread;
+  }
+  switchToThread(threadId) {
+    if (this._mainThread.threadId === threadId) return;
+    const data = this._threadData.get(threadId);
+    if (!data) throw new Error("Thread not found");
+    const thread = this._threadFactory(threadId, data.data);
+    this._performThreadSwitch(thread);
+  }
+  switchToNewThread() {
+    const threadId = (0, import_idUtils.generateId)();
+    this._threadData.set(threadId, {
+      data: { messages: [] },
+      metadata: { threadId },
+      isArchived: false
+    });
+    this._threads = [{ threadId }];
+    const thread = this._threadFactory(threadId, { messages: [] });
+    this._performThreadSwitch(thread);
+  }
+  _performThreadSwitch(newThreadCore) {
+    const data = this._threadData.get(this._mainThread.threadId);
+    if (!data) throw new Error("Thread not found");
+    const exprt = this._mainThread.export();
+    data.data = exprt;
+    this._mainThread._notifyEventSubscribers("switched-away");
+    this._mainThread = newThreadCore;
+    newThreadCore._notifyEventSubscribers("switched-to");
+    this._notifySubscribers();
+  }
+  _performMoveOp(threadId, operation) {
+    const data = this._threadData.get(threadId);
+    if (!data) throw new Error("Thread not found");
+    if (operation === "archive" && data.isArchived) return;
+    if (operation === "unarchive" && !data.isArchived) return;
+    if (operation === "archive") {
+      data.isArchived = true;
+      this._archivedThreads = [...this._archivedThreads, data.metadata];
+    }
+    if (operation === "unarchive") {
+      data.isArchived = false;
+      this._threads = [...this._threads, data.metadata];
+    }
+    if (operation === "delete") {
+      this._threadData.delete(threadId);
+    }
+    if (operation === "archive" || operation === "delete" && data.isArchived) {
+      this._archivedThreads = this._archivedThreads.filter(
+        (t) => t.threadId !== threadId
+      );
+    }
+    if (operation === "unarchive" || operation === "delete" && !data.isArchived) {
+      this._threads = this._threads.filter((t) => t.threadId !== threadId);
+    }
+    this._notifySubscribers();
+  }
+  async rename(threadId, newTitle) {
+    const data = this._threadData.get(threadId);
+    if (!data) throw new Error("Thread not found");
+    data.metadata = {
+      ...data.metadata,
+      title: newTitle
+    };
+    const threadList = data.isArchived ? this.archivedThreads : this.threads;
+    const idx = threadList.findIndex((t) => t.threadId === threadId);
+    const updatedThreadList = threadList.toSpliced(idx, 1, data.metadata);
+    if (data.isArchived) {
+      this._archivedThreads = updatedThreadList;
+    } else {
+      this._threads = updatedThreadList;
+    }
+    this._notifySubscribers();
+  }
+  async archive(threadId) {
+    this._performMoveOp(threadId, "archive");
+  }
+  async unarchive(threadId) {
+    this._performMoveOp(threadId, "unarchive");
+  }
+  async delete(threadId) {
+    this._performMoveOp(threadId, "delete");
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+  _notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  LocalThreadManagerRuntimeCore
+});
+//# sourceMappingURL=LocalThreadManagerRuntimeCore.js.map

package/dist/runtimes/local/LocalThreadManagerRuntimeCore.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/runtimes/local/LocalThreadManagerRuntimeCore.tsx"],"sourcesContent":["import type { Unsubscribe } from \"../../types\";\nimport {\n  ThreadManagerMetadata,\n  ThreadManagerRuntimeCore,\n} from \"../core/ThreadManagerRuntimeCore\";\nimport { ExportedMessageRepository } from \"../utils/MessageRepository\";\nimport { generateId } from \"../../utils/idUtils\";\nimport { LocalThreadRuntimeCore } from \"./LocalThreadRuntimeCore\";\n\nexport type LocalThreadData = {\n  data: ExportedMessageRepository;\n  metadata: ThreadManagerMetadata;\n  isArchived: boolean;\n};\n\nexport type LocalThreadFactory = (\n  threadId: string,\n  data: ExportedMessageRepository,\n) => LocalThreadRuntimeCore;\n\nexport class LocalThreadManagerRuntimeCore implements ThreadManagerRuntimeCore {\n  private _threadData = new Map<string, LocalThreadData>();\n\n  private _threads: readonly ThreadManagerMetadata[] = [];\n  private _archivedThreads: readonly ThreadManagerMetadata[] = [];\n\n  public get threads() {\n    return this._threads;\n  }\n\n  public get archivedThreads() {\n    return this._archivedThreads;\n  }\n\n  private _mainThread: LocalThreadRuntimeCore;\n\n  public get mainThread(): LocalThreadRuntimeCore {\n    return this._mainThread;\n  }\n\n  constructor(private _threadFactory: LocalThreadFactory) {\n    const threadId = generateId();\n    this._threadData.set(threadId, {\n      data: { messages: [] },\n      metadata: { threadId },\n      isArchived: false,\n    });\n    this._threads = [{ threadId }];\n    this._mainThread = this._threadFactory(threadId, { messages: [] });\n  }\n\n  public switchToThread(threadId: string): void {\n    if (this._mainThread.threadId === threadId) return;\n\n    const data = this._threadData.get(threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    const thread = this._threadFactory(threadId, data.data);\n    this._performThreadSwitch(thread);\n  }\n\n  public switchToNewThread(): void {\n    const threadId = generateId();\n    this._threadData.set(threadId, {\n      data: { messages: [] },\n      metadata: { threadId },\n      isArchived: false,\n    });\n    this._threads = [{ threadId }];\n    const thread = this._threadFactory(threadId, { messages: [] });\n    this._performThreadSwitch(thread);\n  }\n\n  private _performThreadSwitch(newThreadCore: LocalThreadRuntimeCore) {\n    const data = this._threadData.get(this._mainThread.threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    const exprt = this._mainThread.export();\n    data.data = exprt;\n\n    this._mainThread._notifyEventSubscribers(\"switched-away\");\n    this._mainThread = newThreadCore;\n    newThreadCore._notifyEventSubscribers(\"switched-to\");\n\n    this._notifySubscribers();\n  }\n\n  private _performMoveOp(\n    threadId: string,\n    operation: \"archive\" | \"unarchive\" | \"delete\",\n  ) {\n    const data = this._threadData.get(threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    if (operation === \"archive\" && data.isArchived) return;\n    if (operation === \"unarchive\" && !data.isArchived) return;\n\n    if (operation === \"archive\") {\n      data.isArchived = true;\n      this._archivedThreads = [...this._archivedThreads, data.metadata];\n    }\n    if (operation === \"unarchive\") {\n      data.isArchived = false;\n      this._threads = [...this._threads, data.metadata];\n    }\n    if (operation === \"delete\") {\n      this._threadData.delete(threadId);\n    }\n\n    if (\n      operation === \"archive\" ||\n      (operation === \"delete\" && data.isArchived)\n    ) {\n      this._archivedThreads = this._archivedThreads.filter(\n        (t) => t.threadId !== threadId,\n      );\n    }\n\n    if (\n      operation === \"unarchive\" ||\n      (operation === \"delete\" && !data.isArchived)\n    ) {\n      this._threads = this._threads.filter((t) => t.threadId !== threadId);\n    }\n\n    this._notifySubscribers();\n  }\n\n  public async rename(threadId: string, newTitle: string): Promise<void> {\n    const data = this._threadData.get(threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    data.metadata = {\n      ...data.metadata,\n      title: newTitle,\n    };\n\n    const threadList = data.isArchived ? this.archivedThreads : this.threads;\n    const idx = threadList.findIndex((t) => t.threadId === threadId);\n    const updatedThreadList = threadList.toSpliced(idx, 1, data.metadata);\n    if (data.isArchived) {\n      this._archivedThreads = updatedThreadList;\n    } else {\n      this._threads = updatedThreadList;\n    }\n    this._notifySubscribers();\n  }\n\n  public async archive(threadId: string): Promise<void> {\n    this._performMoveOp(threadId, \"archive\");\n  }\n\n  public async unarchive(threadId: string): Promise<void> {\n    this._performMoveOp(threadId, \"unarchive\");\n  }\n\n  public async delete(threadId: string): Promise<void> {\n    this._performMoveOp(threadId, \"delete\");\n  }\n\n  private _subscriptions = new Set<() => void>();\n\n  public subscribe(callback: () => void): Unsubscribe {\n    this._subscriptions.add(callback);\n    return () => this._subscriptions.delete(callback);\n  }\n\n  private _notifySubscribers() {\n    for (const callback of this._subscriptions) callback();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAMA,qBAA2B;AAcpB,MAAM,8BAAkE;AAAA,EAoB7E,YAAoB,gBAAoC;AAApC;AAClB,UAAM,eAAW,2BAAW;AAC5B,SAAK,YAAY,IAAI,UAAU;AAAA,MAC7B,MAAM,EAAE,UAAU,CAAC,EAAE;AAAA,MACrB,UAAU,EAAE,SAAS;AAAA,MACrB,YAAY;AAAA,IACd,CAAC;AACD,SAAK,WAAW,CAAC,EAAE,SAAS,CAAC;AAC7B,SAAK,cAAc,KAAK,eAAe,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;AAAA,EACnE;AAAA,EA5BQ,cAAc,oBAAI,IAA6B;AAAA,EAE/C,WAA6C,CAAC;AAAA,EAC9C,mBAAqD,CAAC;AAAA,EAE9D,IAAW,UAAU;AACnB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAW,kBAAkB;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ;AAAA,EAER,IAAW,aAAqC;AAC9C,WAAO,KAAK;AAAA,EACd;AAAA,EAaO,eAAe,UAAwB;AAC5C,QAAI,KAAK,YAAY,aAAa,SAAU;AAE5C,UAAM,OAAO,KAAK,YAAY,IAAI,QAAQ;AAC1C,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,UAAM,SAAS,KAAK,eAAe,UAAU,KAAK,IAAI;AACtD,SAAK,qBAAqB,MAAM;AAAA,EAClC;AAAA,EAEO,oBAA0B;AAC/B,UAAM,eAAW,2BAAW;AAC5B,SAAK,YAAY,IAAI,UAAU;AAAA,MAC7B,MAAM,EAAE,UAAU,CAAC,EAAE;AAAA,MACrB,UAAU,EAAE,SAAS;AAAA,MACrB,YAAY;AAAA,IACd,CAAC;AACD,SAAK,WAAW,CAAC,EAAE,SAAS,CAAC;AAC7B,UAAM,SAAS,KAAK,eAAe,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;AAC7D,SAAK,qBAAqB,MAAM;AAAA,EAClC;AAAA,EAEQ,qBAAqB,eAAuC;AAClE,UAAM,OAAO,KAAK,YAAY,IAAI,KAAK,YAAY,QAAQ;AAC3D,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,UAAM,QAAQ,KAAK,YAAY,OAAO;AACtC,SAAK,OAAO;AAEZ,SAAK,YAAY,wBAAwB,eAAe;AACxD,SAAK,cAAc;AACnB,kBAAc,wBAAwB,aAAa;AAEnD,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEQ,eACN,UACA,WACA;AACA,UAAM,OAAO,KAAK,YAAY,IAAI,QAAQ;AAC1C,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,QAAI,cAAc,aAAa,KAAK,WAAY;AAChD,QAAI,cAAc,eAAe,CAAC,KAAK,WAAY;AAEnD,QAAI,cAAc,WAAW;AAC3B,WAAK,aAAa;AAClB,WAAK,mBAAmB,CAAC,GAAG,KAAK,kBAAkB,KAAK,QAAQ;AAAA,IAClE;AACA,QAAI,cAAc,aAAa;AAC7B,WAAK,aAAa;AAClB,WAAK,WAAW,CAAC,GAAG,KAAK,UAAU,KAAK,QAAQ;AAAA,IAClD;AACA,QAAI,cAAc,UAAU;AAC1B,WAAK,YAAY,OAAO,QAAQ;AAAA,IAClC;AAEA,QACE,cAAc,aACb,cAAc,YAAY,KAAK,YAChC;AACA,WAAK,mBAAmB,KAAK,iBAAiB;AAAA,QAC5C,CAAC,MAAM,EAAE,aAAa;AAAA,MACxB;AAAA,IACF;AAEA,QACE,cAAc,eACb,cAAc,YAAY,CAAC,KAAK,YACjC;AACA,WAAK,WAAW,KAAK,SAAS,OAAO,CAAC,MAAM,EAAE,aAAa,QAAQ;AAAA,IACrE;AAEA,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAa,OAAO,UAAkB,UAAiC;AACrE,UAAM,OAAO,KAAK,YAAY,IAAI,QAAQ;AAC1C,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,SAAK,WAAW;AAAA,MACd,GAAG,KAAK;AAAA,MACR,OAAO;AAAA,IACT;AAEA,UAAM,aAAa,KAAK,aAAa,KAAK,kBAAkB,KAAK;AACjE,UAAM,MAAM,WAAW,UAAU,CAAC,MAAM,EAAE,aAAa,QAAQ;AAC/D,UAAM,oBAAoB,WAAW,UAAU,KAAK,GAAG,KAAK,QAAQ;AACpE,QAAI,KAAK,YAAY;AACnB,WAAK,mBAAmB;AAAA,IAC1B,OAAO;AACL,WAAK,WAAW;AAAA,IAClB;AACA,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAa,QAAQ,UAAiC;AACpD,SAAK,eAAe,UAAU,SAAS;AAAA,EACzC;AAAA,EAEA,MAAa,UAAU,UAAiC;AACtD,SAAK,eAAe,UAAU,WAAW;AAAA,EAC3C;AAAA,EAEA,MAAa,OAAO,UAAiC;AACnD,SAAK,eAAe,UAAU,QAAQ;AAAA,EACxC;AAAA,EAEQ,iBAAiB,oBAAI,IAAgB;AAAA,EAEtC,UAAU,UAAmC;AAClD,SAAK,eAAe,IAAI,QAAQ;AAChC,WAAO,MAAM,KAAK,eAAe,OAAO,QAAQ;AAAA,EAClD;AAAA,EAEQ,qBAAqB;AAC3B,eAAW,YAAY,KAAK,eAAgB,UAAS;AAAA,EACvD;AACF;","names":[]}

package/dist/runtimes/local/LocalThreadManagerRuntimeCore.mjs

@@ -0,0 +1,119 @@
+import { generateId } from "../../utils/idUtils";
+class LocalThreadManagerRuntimeCore {
+  constructor(_threadFactory) {
+    this._threadFactory = _threadFactory;
+    const threadId = generateId();
+    this._threadData.set(threadId, {
+      data: { messages: [] },
+      metadata: { threadId },
+      isArchived: false
+    });
+    this._threads = [{ threadId }];
+    this._mainThread = this._threadFactory(threadId, { messages: [] });
+  }
+  _threadData = /* @__PURE__ */ new Map();
+  _threads = [];
+  _archivedThreads = [];
+  get threads() {
+    return this._threads;
+  }
+  get archivedThreads() {
+    return this._archivedThreads;
+  }
+  _mainThread;
+  get mainThread() {
+    return this._mainThread;
+  }
+  switchToThread(threadId) {
+    if (this._mainThread.threadId === threadId) return;
+    const data = this._threadData.get(threadId);
+    if (!data) throw new Error("Thread not found");
+    const thread = this._threadFactory(threadId, data.data);
+    this._performThreadSwitch(thread);
+  }
+  switchToNewThread() {
+    const threadId = generateId();
+    this._threadData.set(threadId, {
+      data: { messages: [] },
+      metadata: { threadId },
+      isArchived: false
+    });
+    this._threads = [{ threadId }];
+    const thread = this._threadFactory(threadId, { messages: [] });
+    this._performThreadSwitch(thread);
+  }
+  _performThreadSwitch(newThreadCore) {
+    const data = this._threadData.get(this._mainThread.threadId);
+    if (!data) throw new Error("Thread not found");
+    const exprt = this._mainThread.export();
+    data.data = exprt;
+    this._mainThread._notifyEventSubscribers("switched-away");
+    this._mainThread = newThreadCore;
+    newThreadCore._notifyEventSubscribers("switched-to");
+    this._notifySubscribers();
+  }
+  _performMoveOp(threadId, operation) {
+    const data = this._threadData.get(threadId);
+    if (!data) throw new Error("Thread not found");
+    if (operation === "archive" && data.isArchived) return;
+    if (operation === "unarchive" && !data.isArchived) return;
+    if (operation === "archive") {
+      data.isArchived = true;
+      this._archivedThreads = [...this._archivedThreads, data.metadata];
+    }
+    if (operation === "unarchive") {
+      data.isArchived = false;
+      this._threads = [...this._threads, data.metadata];
+    }
+    if (operation === "delete") {
+      this._threadData.delete(threadId);
+    }
+    if (operation === "archive" || operation === "delete" && data.isArchived) {
+      this._archivedThreads = this._archivedThreads.filter(
+        (t) => t.threadId !== threadId
+      );
+    }
+    if (operation === "unarchive" || operation === "delete" && !data.isArchived) {
+      this._threads = this._threads.filter((t) => t.threadId !== threadId);
+    }
+    this._notifySubscribers();
+  }
+  async rename(threadId, newTitle) {
+    const data = this._threadData.get(threadId);
+    if (!data) throw new Error("Thread not found");
+    data.metadata = {
+      ...data.metadata,
+      title: newTitle
+    };
+    const threadList = data.isArchived ? this.archivedThreads : this.threads;
+    const idx = threadList.findIndex((t) => t.threadId === threadId);
+    const updatedThreadList = threadList.toSpliced(idx, 1, data.metadata);
+    if (data.isArchived) {
+      this._archivedThreads = updatedThreadList;
+    } else {
+      this._threads = updatedThreadList;
+    }
+    this._notifySubscribers();
+  }
+  async archive(threadId) {
+    this._performMoveOp(threadId, "archive");
+  }
+  async unarchive(threadId) {
+    this._performMoveOp(threadId, "unarchive");
+  }
+  async delete(threadId) {
+    this._performMoveOp(threadId, "delete");
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+  _notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
+  }
+}
+export {
+  LocalThreadManagerRuntimeCore
+};
+//# sourceMappingURL=LocalThreadManagerRuntimeCore.mjs.map

package/dist/runtimes/local/LocalThreadManagerRuntimeCore.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/runtimes/local/LocalThreadManagerRuntimeCore.tsx"],"sourcesContent":["import type { Unsubscribe } from \"../../types\";\nimport {\n  ThreadManagerMetadata,\n  ThreadManagerRuntimeCore,\n} from \"../core/ThreadManagerRuntimeCore\";\nimport { ExportedMessageRepository } from \"../utils/MessageRepository\";\nimport { generateId } from \"../../utils/idUtils\";\nimport { LocalThreadRuntimeCore } from \"./LocalThreadRuntimeCore\";\n\nexport type LocalThreadData = {\n  data: ExportedMessageRepository;\n  metadata: ThreadManagerMetadata;\n  isArchived: boolean;\n};\n\nexport type LocalThreadFactory = (\n  threadId: string,\n  data: ExportedMessageRepository,\n) => LocalThreadRuntimeCore;\n\nexport class LocalThreadManagerRuntimeCore implements ThreadManagerRuntimeCore {\n  private _threadData = new Map<string, LocalThreadData>();\n\n  private _threads: readonly ThreadManagerMetadata[] = [];\n  private _archivedThreads: readonly ThreadManagerMetadata[] = [];\n\n  public get threads() {\n    return this._threads;\n  }\n\n  public get archivedThreads() {\n    return this._archivedThreads;\n  }\n\n  private _mainThread: LocalThreadRuntimeCore;\n\n  public get mainThread(): LocalThreadRuntimeCore {\n    return this._mainThread;\n  }\n\n  constructor(private _threadFactory: LocalThreadFactory) {\n    const threadId = generateId();\n    this._threadData.set(threadId, {\n      data: { messages: [] },\n      metadata: { threadId },\n      isArchived: false,\n    });\n    this._threads = [{ threadId }];\n    this._mainThread = this._threadFactory(threadId, { messages: [] });\n  }\n\n  public switchToThread(threadId: string): void {\n    if (this._mainThread.threadId === threadId) return;\n\n    const data = this._threadData.get(threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    const thread = this._threadFactory(threadId, data.data);\n    this._performThreadSwitch(thread);\n  }\n\n  public switchToNewThread(): void {\n    const threadId = generateId();\n    this._threadData.set(threadId, {\n      data: { messages: [] },\n      metadata: { threadId },\n      isArchived: false,\n    });\n    this._threads = [{ threadId }];\n    const thread = this._threadFactory(threadId, { messages: [] });\n    this._performThreadSwitch(thread);\n  }\n\n  private _performThreadSwitch(newThreadCore: LocalThreadRuntimeCore) {\n    const data = this._threadData.get(this._mainThread.threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    const exprt = this._mainThread.export();\n    data.data = exprt;\n\n    this._mainThread._notifyEventSubscribers(\"switched-away\");\n    this._mainThread = newThreadCore;\n    newThreadCore._notifyEventSubscribers(\"switched-to\");\n\n    this._notifySubscribers();\n  }\n\n  private _performMoveOp(\n    threadId: string,\n    operation: \"archive\" | \"unarchive\" | \"delete\",\n  ) {\n    const data = this._threadData.get(threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    if (operation === \"archive\" && data.isArchived) return;\n    if (operation === \"unarchive\" && !data.isArchived) return;\n\n    if (operation === \"archive\") {\n      data.isArchived = true;\n      this._archivedThreads = [...this._archivedThreads, data.metadata];\n    }\n    if (operation === \"unarchive\") {\n      data.isArchived = false;\n      this._threads = [...this._threads, data.metadata];\n    }\n    if (operation === \"delete\") {\n      this._threadData.delete(threadId);\n    }\n\n    if (\n      operation === \"archive\" ||\n      (operation === \"delete\" && data.isArchived)\n    ) {\n      this._archivedThreads = this._archivedThreads.filter(\n        (t) => t.threadId !== threadId,\n      );\n    }\n\n    if (\n      operation === \"unarchive\" ||\n      (operation === \"delete\" && !data.isArchived)\n    ) {\n      this._threads = this._threads.filter((t) => t.threadId !== threadId);\n    }\n\n    this._notifySubscribers();\n  }\n\n  public async rename(threadId: string, newTitle: string): Promise<void> {\n    const data = this._threadData.get(threadId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    data.metadata = {\n      ...data.metadata,\n      title: newTitle,\n    };\n\n    const threadList = data.isArchived ? this.archivedThreads : this.threads;\n    const idx = threadList.findIndex((t) => t.threadId === threadId);\n    const updatedThreadList = threadList.toSpliced(idx, 1, data.metadata);\n    if (data.isArchived) {\n      this._archivedThreads = updatedThreadList;\n    } else {\n      this._threads = updatedThreadList;\n    }\n    this._notifySubscribers();\n  }\n\n  public async archive(threadId: string): Promise<void> {\n    this._performMoveOp(threadId, \"archive\");\n  }\n\n  public async unarchive(threadId: string): Promise<void> {\n    this._performMoveOp(threadId, \"unarchive\");\n  }\n\n  public async delete(threadId: string): Promise<void> {\n    this._performMoveOp(threadId, \"delete\");\n  }\n\n  private _subscriptions = new Set<() => void>();\n\n  public subscribe(callback: () => void): Unsubscribe {\n    this._subscriptions.add(callback);\n    return () => this._subscriptions.delete(callback);\n  }\n\n  private _notifySubscribers() {\n    for (const callback of this._subscriptions) callback();\n  }\n}\n"],"mappings":"AAMA,SAAS,kBAAkB;AAcpB,MAAM,8BAAkE;AAAA,EAoB7E,YAAoB,gBAAoC;AAApC;AAClB,UAAM,WAAW,WAAW;AAC5B,SAAK,YAAY,IAAI,UAAU;AAAA,MAC7B,MAAM,EAAE,UAAU,CAAC,EAAE;AAAA,MACrB,UAAU,EAAE,SAAS;AAAA,MACrB,YAAY;AAAA,IACd,CAAC;AACD,SAAK,WAAW,CAAC,EAAE,SAAS,CAAC;AAC7B,SAAK,cAAc,KAAK,eAAe,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;AAAA,EACnE;AAAA,EA5BQ,cAAc,oBAAI,IAA6B;AAAA,EAE/C,WAA6C,CAAC;AAAA,EAC9C,mBAAqD,CAAC;AAAA,EAE9D,IAAW,UAAU;AACnB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAW,kBAAkB;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA,EAEQ;AAAA,EAER,IAAW,aAAqC;AAC9C,WAAO,KAAK;AAAA,EACd;AAAA,EAaO,eAAe,UAAwB;AAC5C,QAAI,KAAK,YAAY,aAAa,SAAU;AAE5C,UAAM,OAAO,KAAK,YAAY,IAAI,QAAQ;AAC1C,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,UAAM,SAAS,KAAK,eAAe,UAAU,KAAK,IAAI;AACtD,SAAK,qBAAqB,MAAM;AAAA,EAClC;AAAA,EAEO,oBAA0B;AAC/B,UAAM,WAAW,WAAW;AAC5B,SAAK,YAAY,IAAI,UAAU;AAAA,MAC7B,MAAM,EAAE,UAAU,CAAC,EAAE;AAAA,MACrB,UAAU,EAAE,SAAS;AAAA,MACrB,YAAY;AAAA,IACd,CAAC;AACD,SAAK,WAAW,CAAC,EAAE,SAAS,CAAC;AAC7B,UAAM,SAAS,KAAK,eAAe,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;AAC7D,SAAK,qBAAqB,MAAM;AAAA,EAClC;AAAA,EAEQ,qBAAqB,eAAuC;AAClE,UAAM,OAAO,KAAK,YAAY,IAAI,KAAK,YAAY,QAAQ;AAC3D,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,UAAM,QAAQ,KAAK,YAAY,OAAO;AACtC,SAAK,OAAO;AAEZ,SAAK,YAAY,wBAAwB,eAAe;AACxD,SAAK,cAAc;AACnB,kBAAc,wBAAwB,aAAa;AAEnD,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEQ,eACN,UACA,WACA;AACA,UAAM,OAAO,KAAK,YAAY,IAAI,QAAQ;AAC1C,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,QAAI,cAAc,aAAa,KAAK,WAAY;AAChD,QAAI,cAAc,eAAe,CAAC,KAAK,WAAY;AAEnD,QAAI,cAAc,WAAW;AAC3B,WAAK,aAAa;AAClB,WAAK,mBAAmB,CAAC,GAAG,KAAK,kBAAkB,KAAK,QAAQ;AAAA,IAClE;AACA,QAAI,cAAc,aAAa;AAC7B,WAAK,aAAa;AAClB,WAAK,WAAW,CAAC,GAAG,KAAK,UAAU,KAAK,QAAQ;AAAA,IAClD;AACA,QAAI,cAAc,UAAU;AAC1B,WAAK,YAAY,OAAO,QAAQ;AAAA,IAClC;AAEA,QACE,cAAc,aACb,cAAc,YAAY,KAAK,YAChC;AACA,WAAK,mBAAmB,KAAK,iBAAiB;AAAA,QAC5C,CAAC,MAAM,EAAE,aAAa;AAAA,MACxB;AAAA,IACF;AAEA,QACE,cAAc,eACb,cAAc,YAAY,CAAC,KAAK,YACjC;AACA,WAAK,WAAW,KAAK,SAAS,OAAO,CAAC,MAAM,EAAE,aAAa,QAAQ;AAAA,IACrE;AAEA,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAa,OAAO,UAAkB,UAAiC;AACrE,UAAM,OAAO,KAAK,YAAY,IAAI,QAAQ;AAC1C,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,SAAK,WAAW;AAAA,MACd,GAAG,KAAK;AAAA,MACR,OAAO;AAAA,IACT;AAEA,UAAM,aAAa,KAAK,aAAa,KAAK,kBAAkB,KAAK;AACjE,UAAM,MAAM,WAAW,UAAU,CAAC,MAAM,EAAE,aAAa,QAAQ;AAC/D,UAAM,oBAAoB,WAAW,UAAU,KAAK,GAAG,KAAK,QAAQ;AACpE,QAAI,KAAK,YAAY;AACnB,WAAK,mBAAmB;AAAA,IAC1B,OAAO;AACL,WAAK,WAAW;AAAA,IAClB;AACA,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAa,QAAQ,UAAiC;AACpD,SAAK,eAAe,UAAU,SAAS;AAAA,EACzC;AAAA,EAEA,MAAa,UAAU,UAAiC;AACtD,SAAK,eAAe,UAAU,WAAW;AAAA,EAC3C;AAAA,EAEA,MAAa,OAAO,UAAiC;AACnD,SAAK,eAAe,UAAU,QAAQ;AAAA,EACxC;AAAA,EAEQ,iBAAiB,oBAAI,IAAgB;AAAA,EAEtC,UAAU,UAAmC;AAClD,SAAK,eAAe,IAAI,QAAQ;AAChC,WAAO,MAAM,KAAK,eAAe,OAAO,QAAQ;AAAA,EAClD;AAAA,EAEQ,qBAAqB;AAC3B,eAAW,YAAY,KAAK,eAAgB,UAAS;AAAA,EACvD;AACF;","names":[]}

package/dist/runtimes/local/LocalThreadRuntimeCore.d.ts

@@ -0,0 +1,36 @@
+import type { ModelConfigProvider, AppendMessage } from "../../types";
+import { LocalRuntimeOptionsBase } from "./LocalRuntimeOptions";
+import { AddToolResultOptions, ThreadSuggestion, ThreadRuntimeCore } from "../core/ThreadRuntimeCore";
+import { BaseThreadRuntimeCore } from "../core/BaseThreadRuntimeCore";
+export declare class LocalThreadRuntimeCore extends BaseThreadRuntimeCore implements ThreadRuntimeCore {
+    readonly threadId: string;
+    readonly capabilities: {
+        switchToBranch: boolean;
+        edit: boolean;
+        reload: boolean;
+        cancel: boolean;
+        unstable_copy: boolean;
+        speech: boolean;
+        attachments: boolean;
+        feedback: boolean;
+    };
+    private abortController;
+    readonly isDisabled = false;
+    readonly suggestions: readonly ThreadSuggestion[];
+    get adapters(): {
+        chatModel: import("./ChatModelAdapter").ChatModelAdapter;
+        attachments?: import("..").AttachmentAdapter | undefined;
+        speech?: import("..").SpeechSynthesisAdapter | undefined;
+        feedback?: import("..").FeedbackAdapter | undefined;
+    };
+    constructor(configProvider: ModelConfigProvider, threadId: string, options: LocalRuntimeOptionsBase);
+    private _options;
+    get extras(): undefined;
+    setOptions(options: LocalRuntimeOptionsBase): void;
+    append(message: AppendMessage): Promise<void>;
+    startRun(parentId: string | null): Promise<void>;
+    private performRoundtrip;
+    cancelRun(): void;
+    addToolResult({ messageId, toolCallId, result, }: AddToolResultOptions): void;
+}
+//# sourceMappingURL=LocalThreadRuntimeCore.d.ts.map

package/dist/runtimes/local/LocalThreadRuntimeCore.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"LocalThreadRuntimeCore.d.ts","sourceRoot":"","sources":["../../../src/runtimes/local/LocalThreadRuntimeCore.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,mBAAmB,EACnB,aAAa,EAEd,MAAM,aAAa,CAAC;AAIrB,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EACL,oBAAoB,EACpB,gBAAgB,EAChB,iBAAiB,EAClB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAEtE,qBAAa,sBACX,SAAQ,qBACR,YAAW,iBAAiB;aAwBV,QAAQ,EAAE,MAAM;IAtBlC,SAAgB,YAAY;;;;;;;;;MAS1B;IAEF,OAAO,CAAC,eAAe,CAAgC;IAEvD,SAAgB,UAAU,SAAS;IACnC,SAAgB,WAAW,EAAE,SAAS,gBAAgB,EAAE,CAAM;IAE9D,IAAW,QAAQ;;;;;MAElB;gBAGC,cAAc,EAAE,mBAAmB,EACnB,QAAQ,EAAE,MAAM,EAChC,OAAO,EAAE,uBAAuB;IAOlC,OAAO,CAAC,QAAQ,CAA2B;IAE3C,IAAW,MAAM,cAEhB;IAEM,UAAU,CAAC,OAAO,EAAE,uBAAuB;IA4BrC,MAAM,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAc7C,QAAQ,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;YAoB/C,gBAAgB;IA+GvB,SAAS;IAKT,aAAa,CAAC,EACnB,SAAS,EACT,UAAU,EACV,MAAM,GACP,EAAE,oBAAoB;CAkCxB"}

package/dist/runtimes/local/LocalThreadRuntimeCore.js

@@ -0,0 +1,226 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var LocalThreadRuntimeCore_exports = {};
+__export(LocalThreadRuntimeCore_exports, {
+  LocalThreadRuntimeCore: () => LocalThreadRuntimeCore
+});
+module.exports = __toCommonJS(LocalThreadRuntimeCore_exports);
+var import_internal = require("../../internal");
+var import_edge = require("../edge");
+var import_shouldContinue = require("./shouldContinue");
+var import_BaseThreadRuntimeCore = require("../core/BaseThreadRuntimeCore");
+class LocalThreadRuntimeCore extends import_BaseThreadRuntimeCore.BaseThreadRuntimeCore {
+  constructor(configProvider, threadId, options) {
+    super(configProvider);
+    this.threadId = threadId;
+    this.setOptions(options);
+  }
+  capabilities = {
+    switchToBranch: true,
+    edit: true,
+    reload: true,
+    cancel: true,
+    unstable_copy: true,
+    speech: false,
+    attachments: false,
+    feedback: false
+  };
+  abortController = null;
+  isDisabled = false;
+  suggestions = [];
+  get adapters() {
+    return this._options.adapters;
+  }
+  _options;
+  get extras() {
+    return void 0;
+  }
+  setOptions(options) {
+    if (this._options === options) return;
+    this._options = options;
+    let hasUpdates = false;
+    const canSpeak = options.adapters?.speech !== void 0;
+    if (this.capabilities.speech !== canSpeak) {
+      this.capabilities.speech = canSpeak;
+      hasUpdates = true;
+    }
+    const canAttach = options.adapters?.attachments !== void 0;
+    if (this.capabilities.attachments !== canAttach) {
+      this.capabilities.attachments = canAttach;
+      hasUpdates = true;
+    }
+    const canFeedback = options.adapters?.feedback !== void 0;
+    if (this.capabilities.feedback !== canFeedback) {
+      this.capabilities.feedback = canFeedback;
+      hasUpdates = true;
+    }
+    if (hasUpdates) this._notifySubscribers();
+  }
+  async append(message) {
+    const newMessage = (0, import_edge.fromCoreMessage)(message, {
+      attachments: message.attachments
+    });
+    this.repository.addOrUpdateMessage(message.parentId, newMessage);
+    if (message.role === "user") {
+      await this.startRun(newMessage.id);
+    } else {
+      this.repository.resetHead(newMessage.id);
+      this._notifySubscribers();
+    }
+  }
+  async startRun(parentId) {
+    this.repository.resetHead(parentId);
+    const id = (0, import_internal.generateId)();
+    let message = {
+      id,
+      role: "assistant",
+      status: { type: "running" },
+      content: [],
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this._notifyEventSubscribers("run-start");
+    do {
+      message = await this.performRoundtrip(parentId, message);
+    } while ((0, import_shouldContinue.shouldContinue)(message));
+  }
+  async performRoundtrip(parentId, message) {
+    const messages = this.repository.getMessages();
+    this.abortController?.abort();
+    this.abortController = new AbortController();
+    const initialContent = message.content;
+    const initialSteps = message.metadata?.steps;
+    const initalCustom = message.metadata?.custom;
+    const updateMessage = (m) => {
+      const newSteps = m.metadata?.steps || m.metadata?.roundtrips;
+      const steps2 = newSteps ? [...initialSteps ?? [], ...newSteps] : void 0;
+      message = {
+        ...message,
+        ...m.content ? { content: [...initialContent, ...m.content ?? []] } : void 0,
+        status: m.status ?? message.status,
+        // TODO deprecated, remove in v0.6
+        ...steps2 ? { roundtrips: steps2 } : void 0,
+        ...m.metadata ? {
+          metadata: {
+            ...message.metadata,
+            ...steps2 ? { roundtrips: steps2, steps: steps2 } : void 0,
+            ...m.metadata?.custom ? {
+              custom: { ...initalCustom ?? {}, ...m.metadata.custom }
+            } : void 0
+          }
+        } : void 0
+      };
+      this.repository.addOrUpdateMessage(parentId, message);
+      this._notifySubscribers();
+    };
+    const maxSteps = this._options.maxSteps ? this._options.maxSteps : (this._options.maxToolRoundtrips ?? 1) + 1;
+    const steps = message.metadata?.steps?.length ?? 0;
+    if (steps >= maxSteps) {
+      updateMessage({
+        status: {
+          type: "incomplete",
+          reason: "tool-calls"
+        }
+      });
+      return message;
+    } else {
+      updateMessage({
+        status: {
+          type: "running"
+        }
+      });
+    }
+    try {
+      const promiseOrGenerator = this.adapters.chatModel.run({
+        messages,
+        abortSignal: this.abortController.signal,
+        config: this.getModelConfig(),
+        onUpdate: updateMessage,
+        unstable_assistantMessageId: message.id
+      });
+      if (Symbol.asyncIterator in promiseOrGenerator) {
+        for await (const r of promiseOrGenerator) {
+          updateMessage(r);
+        }
+      } else {
+        updateMessage(await promiseOrGenerator);
+      }
+      this.abortController = null;
+      if (message.status.type === "running") {
+        updateMessage({
+          status: { type: "complete", reason: "unknown" }
+        });
+      }
+    } catch (e) {
+      this.abortController = null;
+      if (e instanceof Error && e.name === "AbortError") {
+        updateMessage({
+          status: { type: "incomplete", reason: "cancelled" }
+        });
+      } else {
+        updateMessage({
+          status: { type: "incomplete", reason: "error", error: e }
+        });
+        throw e;
+      }
+    }
+    return message;
+  }
+  cancelRun() {
+    this.abortController?.abort();
+    this.abortController = null;
+  }
+  addToolResult({
+    messageId,
+    toolCallId,
+    result
+  }) {
+    const messageData = this.repository.getMessage(messageId);
+    const { parentId } = messageData;
+    let { message } = messageData;
+    if (message.role !== "assistant")
+      throw new Error("Tried to add tool result to non-assistant message");
+    let added = false;
+    let found = false;
+    const newContent = message.content.map((c) => {
+      if (c.type !== "tool-call") return c;
+      if (c.toolCallId !== toolCallId) return c;
+      found = true;
+      if (!c.result) added = true;
+      return {
+        ...c,
+        result
+      };
+    });
+    if (!found)
+      throw new Error("Tried to add tool result to non-existing tool call");
+    message = {
+      ...message,
+      content: newContent
+    };
+    this.repository.addOrUpdateMessage(parentId, message);
+    if (added && (0, import_shouldContinue.shouldContinue)(message)) {
+      this.performRoundtrip(parentId, message);
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  LocalThreadRuntimeCore
+});
+//# sourceMappingURL=LocalThreadRuntimeCore.js.map

package/dist/runtimes/local/LocalThreadRuntimeCore.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/runtimes/local/LocalThreadRuntimeCore.tsx"],"sourcesContent":["import { generateId } from \"../../internal\";\nimport type {\n  ModelConfigProvider,\n  AppendMessage,\n  ThreadAssistantMessage,\n} from \"../../types\";\nimport { fromCoreMessage } from \"../edge\";\nimport type { ChatModelRunResult } from \"./ChatModelAdapter\";\nimport { shouldContinue } from \"./shouldContinue\";\nimport { LocalRuntimeOptionsBase } from \"./LocalRuntimeOptions\";\nimport {\n  AddToolResultOptions,\n  ThreadSuggestion,\n  ThreadRuntimeCore,\n} from \"../core/ThreadRuntimeCore\";\nimport { BaseThreadRuntimeCore } from \"../core/BaseThreadRuntimeCore\";\n\nexport class LocalThreadRuntimeCore\n  extends BaseThreadRuntimeCore\n  implements ThreadRuntimeCore\n{\n  public readonly capabilities = {\n    switchToBranch: true,\n    edit: true,\n    reload: true,\n    cancel: true,\n    unstable_copy: true,\n    speech: false,\n    attachments: false,\n    feedback: false,\n  };\n\n  private abortController: AbortController | null = null;\n\n  public readonly isDisabled = false;\n  public readonly suggestions: readonly ThreadSuggestion[] = [];\n\n  public get adapters() {\n    return this._options.adapters;\n  }\n\n  constructor(\n    configProvider: ModelConfigProvider,\n    public readonly threadId: string,\n    options: LocalRuntimeOptionsBase,\n  ) {\n    super(configProvider);\n\n    this.setOptions(options);\n  }\n\n  private _options!: LocalRuntimeOptionsBase;\n\n  public get extras() {\n    return undefined;\n  }\n\n  public setOptions(options: LocalRuntimeOptionsBase) {\n    if (this._options === options) return;\n\n    this._options = options;\n\n    let hasUpdates = false;\n\n    const canSpeak = options.adapters?.speech !== undefined;\n    if (this.capabilities.speech !== canSpeak) {\n      this.capabilities.speech = canSpeak;\n      hasUpdates = true;\n    }\n\n    const canAttach = options.adapters?.attachments !== undefined;\n    if (this.capabilities.attachments !== canAttach) {\n      this.capabilities.attachments = canAttach;\n      hasUpdates = true;\n    }\n\n    const canFeedback = options.adapters?.feedback !== undefined;\n    if (this.capabilities.feedback !== canFeedback) {\n      this.capabilities.feedback = canFeedback;\n      hasUpdates = true;\n    }\n\n    if (hasUpdates) this._notifySubscribers();\n  }\n\n  public async append(message: AppendMessage): Promise<void> {\n    const newMessage = fromCoreMessage(message, {\n      attachments: message.attachments,\n    });\n    this.repository.addOrUpdateMessage(message.parentId, newMessage);\n\n    if (message.role === \"user\") {\n      await this.startRun(newMessage.id);\n    } else {\n      this.repository.resetHead(newMessage.id);\n      this._notifySubscribers();\n    }\n  }\n\n  public async startRun(parentId: string | null): Promise<void> {\n    this.repository.resetHead(parentId);\n\n    // add assistant message\n    const id = generateId();\n    let message: ThreadAssistantMessage = {\n      id,\n      role: \"assistant\",\n      status: { type: \"running\" },\n      content: [],\n      createdAt: new Date(),\n    };\n\n    this._notifyEventSubscribers(\"run-start\");\n\n    do {\n      message = await this.performRoundtrip(parentId, message);\n    } while (shouldContinue(message));\n  }\n\n  private async performRoundtrip(\n    parentId: string | null,\n    message: ThreadAssistantMessage,\n  ) {\n    const messages = this.repository.getMessages();\n\n    // abort existing run\n    this.abortController?.abort();\n    this.abortController = new AbortController();\n\n    const initialContent = message.content;\n    const initialSteps = message.metadata?.steps;\n    const initalCustom = message.metadata?.custom;\n    const updateMessage = (m: Partial<ChatModelRunResult>) => {\n      const newSteps = m.metadata?.steps || m.metadata?.roundtrips;\n      const steps = newSteps\n        ? [...(initialSteps ?? []), ...newSteps]\n        : undefined;\n\n      message = {\n        ...message,\n        ...(m.content\n          ? { content: [...initialContent, ...(m.content ?? [])] }\n          : undefined),\n        status: m.status ?? message.status,\n        // TODO deprecated, remove in v0.6\n        ...(steps ? { roundtrips: steps } : undefined),\n        ...(m.metadata\n          ? {\n              metadata: {\n                ...message.metadata,\n                ...(steps ? { roundtrips: steps, steps } : undefined),\n                ...(m.metadata?.custom\n                  ? {\n                      custom: { ...(initalCustom ?? {}), ...m.metadata.custom },\n                    }\n                  : undefined),\n              },\n            }\n          : undefined),\n      };\n      this.repository.addOrUpdateMessage(parentId, message);\n      this._notifySubscribers();\n    };\n\n    const maxSteps = this._options.maxSteps\n      ? this._options.maxSteps\n      : (this._options.maxToolRoundtrips ?? 1) + 1;\n\n    const steps = message.metadata?.steps?.length ?? 0;\n    if (steps >= maxSteps) {\n      // reached max tool steps\n      updateMessage({\n        status: {\n          type: \"incomplete\",\n          reason: \"tool-calls\",\n        },\n      });\n      return message;\n    } else {\n      updateMessage({\n        status: {\n          type: \"running\",\n        },\n      });\n    }\n\n    try {\n      const promiseOrGenerator = this.adapters.chatModel.run({\n        messages,\n        abortSignal: this.abortController.signal,\n        config: this.getModelConfig(),\n        onUpdate: updateMessage,\n        unstable_assistantMessageId: message.id,\n      });\n\n      // handle async iterator for streaming results\n      if (Symbol.asyncIterator in promiseOrGenerator) {\n        for await (const r of promiseOrGenerator) {\n          updateMessage(r);\n        }\n      } else {\n        updateMessage(await promiseOrGenerator);\n      }\n\n      this.abortController = null;\n\n      if (message.status.type === \"running\") {\n        updateMessage({\n          status: { type: \"complete\", reason: \"unknown\" },\n        });\n      }\n    } catch (e) {\n      this.abortController = null;\n\n      // TODO this should be handled by the run result stream\n      if (e instanceof Error && e.name === \"AbortError\") {\n        updateMessage({\n          status: { type: \"incomplete\", reason: \"cancelled\" },\n        });\n      } else {\n        updateMessage({\n          status: { type: \"incomplete\", reason: \"error\", error: e },\n        });\n\n        throw e;\n      }\n    }\n    return message;\n  }\n\n  public cancelRun() {\n    this.abortController?.abort();\n    this.abortController = null;\n  }\n\n  public addToolResult({\n    messageId,\n    toolCallId,\n    result,\n  }: AddToolResultOptions) {\n    const messageData = this.repository.getMessage(messageId);\n    const { parentId } = messageData;\n    let { message } = messageData;\n\n    if (message.role !== \"assistant\")\n      throw new Error(\"Tried to add tool result to non-assistant message\");\n\n    let added = false;\n    let found = false;\n    const newContent = message.content.map((c) => {\n      if (c.type !== \"tool-call\") return c;\n      if (c.toolCallId !== toolCallId) return c;\n      found = true;\n      if (!c.result) added = true;\n      return {\n        ...c,\n        result,\n      };\n    });\n\n    if (!found)\n      throw new Error(\"Tried to add tool result to non-existing tool call\");\n\n    message = {\n      ...message,\n      content: newContent,\n    };\n    this.repository.addOrUpdateMessage(parentId, message);\n\n    if (added && shouldContinue(message)) {\n      this.performRoundtrip(parentId, message);\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,sBAA2B;AAM3B,kBAAgC;AAEhC,4BAA+B;AAO/B,mCAAsC;AAE/B,MAAM,+BACH,mDAEV;AAAA,EAqBE,YACE,gBACgB,UAChB,SACA;AACA,UAAM,cAAc;AAHJ;AAKhB,SAAK,WAAW,OAAO;AAAA,EACzB;AAAA,EA5BgB,eAAe;AAAA,IAC7B,gBAAgB;AAAA,IAChB,MAAM;AAAA,IACN,QAAQ;AAAA,IACR,QAAQ;AAAA,IACR,eAAe;AAAA,IACf,QAAQ;AAAA,IACR,aAAa;AAAA,IACb,UAAU;AAAA,EACZ;AAAA,EAEQ,kBAA0C;AAAA,EAElC,aAAa;AAAA,EACb,cAA2C,CAAC;AAAA,EAE5D,IAAW,WAAW;AACpB,WAAO,KAAK,SAAS;AAAA,EACvB;AAAA,EAYQ;AAAA,EAER,IAAW,SAAS;AAClB,WAAO;AAAA,EACT;AAAA,EAEO,WAAW,SAAkC;AAClD,QAAI,KAAK,aAAa,QAAS;AAE/B,SAAK,WAAW;AAEhB,QAAI,aAAa;AAEjB,UAAM,WAAW,QAAQ,UAAU,WAAW;AAC9C,QAAI,KAAK,aAAa,WAAW,UAAU;AACzC,WAAK,aAAa,SAAS;AAC3B,mBAAa;AAAA,IACf;AAEA,UAAM,YAAY,QAAQ,UAAU,gBAAgB;AACpD,QAAI,KAAK,aAAa,gBAAgB,WAAW;AAC/C,WAAK,aAAa,cAAc;AAChC,mBAAa;AAAA,IACf;AAEA,UAAM,cAAc,QAAQ,UAAU,aAAa;AACnD,QAAI,KAAK,aAAa,aAAa,aAAa;AAC9C,WAAK,aAAa,WAAW;AAC7B,mBAAa;AAAA,IACf;AAEA,QAAI,WAAY,MAAK,mBAAmB;AAAA,EAC1C;AAAA,EAEA,MAAa,OAAO,SAAuC;AACzD,UAAM,iBAAa,6BAAgB,SAAS;AAAA,MAC1C,aAAa,QAAQ;AAAA,IACvB,CAAC;AACD,SAAK,WAAW,mBAAmB,QAAQ,UAAU,UAAU;AAE/D,QAAI,QAAQ,SAAS,QAAQ;AAC3B,YAAM,KAAK,SAAS,WAAW,EAAE;AAAA,IACnC,OAAO;AACL,WAAK,WAAW,UAAU,WAAW,EAAE;AACvC,WAAK,mBAAmB;AAAA,IAC1B;AAAA,EACF;AAAA,EAEA,MAAa,SAAS,UAAwC;AAC5D,SAAK,WAAW,UAAU,QAAQ;AAGlC,UAAM,SAAK,4BAAW;AACtB,QAAI,UAAkC;AAAA,MACpC;AAAA,MACA,MAAM;AAAA,MACN,QAAQ,EAAE,MAAM,UAAU;AAAA,MAC1B,SAAS,CAAC;AAAA,MACV,WAAW,oBAAI,KAAK;AAAA,IACtB;AAEA,SAAK,wBAAwB,WAAW;AAExC,OAAG;AACD,gBAAU,MAAM,KAAK,iBAAiB,UAAU,OAAO;AAAA,IACzD,aAAS,sCAAe,OAAO;AAAA,EACjC;AAAA,EAEA,MAAc,iBACZ,UACA,SACA;AACA,UAAM,WAAW,KAAK,WAAW,YAAY;AAG7C,SAAK,iBAAiB,MAAM;AAC5B,SAAK,kBAAkB,IAAI,gBAAgB;AAE3C,UAAM,iBAAiB,QAAQ;AAC/B,UAAM,eAAe,QAAQ,UAAU;AACvC,UAAM,eAAe,QAAQ,UAAU;AACvC,UAAM,gBAAgB,CAAC,MAAmC;AACxD,YAAM,WAAW,EAAE,UAAU,SAAS,EAAE,UAAU;AAClD,YAAMA,SAAQ,WACV,CAAC,GAAI,gBAAgB,CAAC,GAAI,GAAG,QAAQ,IACrC;AAEJ,gBAAU;AAAA,QACR,GAAG;AAAA,QACH,GAAI,EAAE,UACF,EAAE,SAAS,CAAC,GAAG,gBAAgB,GAAI,EAAE,WAAW,CAAC,CAAE,EAAE,IACrD;AAAA,QACJ,QAAQ,EAAE,UAAU,QAAQ;AAAA;AAAA,QAE5B,GAAIA,SAAQ,EAAE,YAAYA,OAAM,IAAI;AAAA,QACpC,GAAI,EAAE,WACF;AAAA,UACE,UAAU;AAAA,YACR,GAAG,QAAQ;AAAA,YACX,GAAIA,SAAQ,EAAE,YAAYA,QAAO,OAAAA,OAAM,IAAI;AAAA,YAC3C,GAAI,EAAE,UAAU,SACZ;AAAA,cACE,QAAQ,EAAE,GAAI,gBAAgB,CAAC,GAAI,GAAG,EAAE,SAAS,OAAO;AAAA,YAC1D,IACA;AAAA,UACN;AAAA,QACF,IACA;AAAA,MACN;AACA,WAAK,WAAW,mBAAmB,UAAU,OAAO;AACpD,WAAK,mBAAmB;AAAA,IAC1B;AAEA,UAAM,WAAW,KAAK,SAAS,WAC3B,KAAK,SAAS,YACb,KAAK,SAAS,qBAAqB,KAAK;AAE7C,UAAM,QAAQ,QAAQ,UAAU,OAAO,UAAU;AACjD,QAAI,SAAS,UAAU;AAErB,oBAAc;AAAA,QACZ,QAAQ;AAAA,UACN,MAAM;AAAA,UACN,QAAQ;AAAA,QACV;AAAA,MACF,CAAC;AACD,aAAO;AAAA,IACT,OAAO;AACL,oBAAc;AAAA,QACZ,QAAQ;AAAA,UACN,MAAM;AAAA,QACR;AAAA,MACF,CAAC;AAAA,IACH;AAEA,QAAI;AACF,YAAM,qBAAqB,KAAK,SAAS,UAAU,IAAI;AAAA,QACrD;AAAA,QACA,aAAa,KAAK,gBAAgB;AAAA,QAClC,QAAQ,KAAK,eAAe;AAAA,QAC5B,UAAU;AAAA,QACV,6BAA6B,QAAQ;AAAA,MACvC,CAAC;AAGD,UAAI,OAAO,iBAAiB,oBAAoB;AAC9C,yBAAiB,KAAK,oBAAoB;AACxC,wBAAc,CAAC;AAAA,QACjB;AAAA,MACF,OAAO;AACL,sBAAc,MAAM,kBAAkB;AAAA,MACxC;AAEA,WAAK,kBAAkB;AAEvB,UAAI,QAAQ,OAAO,SAAS,WAAW;AACrC,sBAAc;AAAA,UACZ,QAAQ,EAAE,MAAM,YAAY,QAAQ,UAAU;AAAA,QAChD,CAAC;AAAA,MACH;AAAA,IACF,SAAS,GAAG;AACV,WAAK,kBAAkB;AAGvB,UAAI,aAAa,SAAS,EAAE,SAAS,cAAc;AACjD,sBAAc;AAAA,UACZ,QAAQ,EAAE,MAAM,cAAc,QAAQ,YAAY;AAAA,QACpD,CAAC;AAAA,MACH,OAAO;AACL,sBAAc;AAAA,UACZ,QAAQ,EAAE,MAAM,cAAc,QAAQ,SAAS,OAAO,EAAE;AAAA,QAC1D,CAAC;AAED,cAAM;AAAA,MACR;AAAA,IACF;AACA,WAAO;AAAA,EACT;AAAA,EAEO,YAAY;AACjB,SAAK,iBAAiB,MAAM;AAC5B,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEO,cAAc;AAAA,IACnB;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAAyB;AACvB,UAAM,cAAc,KAAK,WAAW,WAAW,SAAS;AACxD,UAAM,EAAE,SAAS,IAAI;AACrB,QAAI,EAAE,QAAQ,IAAI;AAElB,QAAI,QAAQ,SAAS;AACnB,YAAM,IAAI,MAAM,mDAAmD;AAErE,QAAI,QAAQ;AACZ,QAAI,QAAQ;AACZ,UAAM,aAAa,QAAQ,QAAQ,IAAI,CAAC,MAAM;AAC5C,UAAI,EAAE,SAAS,YAAa,QAAO;AACnC,UAAI,EAAE,eAAe,WAAY,QAAO;AACxC,cAAQ;AACR,UAAI,CAAC,EAAE,OAAQ,SAAQ;AACvB,aAAO;AAAA,QACL,GAAG;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAED,QAAI,CAAC;AACH,YAAM,IAAI,MAAM,oDAAoD;AAEtE,cAAU;AAAA,MACR,GAAG;AAAA,MACH,SAAS;AAAA,IACX;AACA,SAAK,WAAW,mBAAmB,UAAU,OAAO;AAEpD,QAAI,aAAS,sCAAe,OAAO,GAAG;AACpC,WAAK,iBAAiB,UAAU,OAAO;AAAA,IACzC;AAAA,EACF;AACF;","names":["steps"]}