@assistant-ui/react-ag-ui 0.0.6 → 0.0.7

package/dist/runtime/AgUiThreadRuntimeCore.js

@@ -1,376 +1,376 @@
 "use client";
-
-// src/runtime/AgUiThreadRuntimeCore.ts
 import { INTERNAL } from "@assistant-ui/react";
 import { RunAggregator } from "./adapter/run-aggregator.js";
 import { toAgUiMessages, toAgUiTools } from "./adapter/conversions.js";
 import { createAgUiSubscriber } from "./adapter/subscriber.js";
-var FALLBACK_USER_STATUS = { type: "complete", reason: "unknown" };
-var AgUiThreadRuntimeCore = class {
-  agent;
-  logger;
-  showThinking;
-  onError;
-  onCancel;
-  notifyUpdate;
-  runtime;
-  messages = [];
-  isRunningFlag = false;
-  abortController = null;
-  stateSnapshot;
-  pendingError = null;
-  history;
-  lastRunConfig;
-  assistantHistoryParents = /* @__PURE__ */ new Map();
-  recordedHistoryIds = /* @__PURE__ */ new Set();
-  constructor(options) {
-    this.agent = options.agent;
-    this.logger = options.logger;
-    this.showThinking = options.showThinking;
-    this.onError = options.onError;
-    this.onCancel = options.onCancel;
-    this.history = options.history;
-    this.notifyUpdate = options.notifyUpdate;
-  }
-  updateOptions(options) {
-    this.agent = options.agent;
-    this.logger = options.logger;
-    this.showThinking = options.showThinking;
-    this.onError = options.onError;
-    this.onCancel = options.onCancel;
-    this.history = options.history;
-  }
-  attachRuntime(runtime) {
-    this.runtime = runtime;
-  }
-  detachRuntime() {
-    this.runtime = void 0;
-  }
-  getMessages() {
-    return this.messages;
-  }
-  getState() {
-    return this.stateSnapshot;
-  }
-  isRunning() {
-    return this.isRunningFlag;
-  }
-  async append(message) {
-    const startRun = message.startRun ?? message.role === "user";
-    if (message.sourceId) {
-      this.messages = this.messages.filter(
-        (entry) => entry.id !== message.sourceId
-      );
+const FALLBACK_USER_STATUS = { type: "complete", reason: "unknown" };
+export class AgUiThreadRuntimeCore {
+    agent;
+    logger;
+    showThinking;
+    onError;
+    onCancel;
+    notifyUpdate;
+    runtime;
+    messages = [];
+    isRunningFlag = false;
+    abortController = null;
+    stateSnapshot;
+    pendingError = null;
+    history;
+    lastRunConfig;
+    assistantHistoryParents = new Map();
+    recordedHistoryIds = new Set();
+    constructor(options) {
+        this.agent = options.agent;
+        this.logger = options.logger;
+        this.showThinking = options.showThinking;
+        this.onError = options.onError;
+        this.onCancel = options.onCancel;
+        this.history = options.history;
+        this.notifyUpdate = options.notifyUpdate;
     }
-    this.resetHead(message.parentId);
-    const threadMessage = this.toThreadMessage(message);
-    this.messages = [...this.messages, threadMessage];
-    this.notifyUpdate();
-    this.recordHistoryEntry(message.parentId ?? null, threadMessage);
-    if (!startRun) return;
-    await this.startRun(threadMessage.id, message.runConfig);
-  }
-  async edit(message) {
-    await this.append(message);
-  }
-  async reload(parentId, config = {}) {
-    this.resetHead(parentId);
-    this.notifyUpdate();
-    await this.startRun(parentId, config.runConfig);
-  }
-  async cancel() {
-    if (!this.abortController) return;
-    this.abortController.abort();
-  }
-  async resume(config) {
-    if (config.stream) {
-      this.logger.debug?.(
-        "[agui] resume stream is not supported, falling back to regular run"
-      );
+    updateOptions(options) {
+        this.agent = options.agent;
+        this.logger = options.logger;
+        this.showThinking = options.showThinking;
+        this.onError = options.onError;
+        this.onCancel = options.onCancel;
+        this.history = options.history;
     }
-    await this.startRun(
-      config.parentId,
-      config.runConfig ?? this.lastRunConfig
-    );
-  }
-  addToolResult(options) {
-    let updated = false;
-    this.messages = this.messages.map((message) => {
-      if (message.id !== options.messageId || message.role !== "assistant")
-        return message;
-      updated = true;
-      const assistant = message;
-      const content = assistant.content.map((part) => {
-        if (part.type !== "tool-call" || part.toolCallId !== options.toolCallId)
-          return part;
+    attachRuntime(runtime) {
+        this.runtime = runtime;
+    }
+    detachRuntime() {
+        this.runtime = undefined;
+    }
+    getMessages() {
+        return this.messages;
+    }
+    getState() {
+        return this.stateSnapshot;
+    }
+    isRunning() {
+        return this.isRunningFlag;
+    }
+    async append(message) {
+        const startRun = message.startRun ?? message.role === "user";
+        if (message.sourceId) {
+            this.messages = this.messages.filter((entry) => entry.id !== message.sourceId);
+        }
+        this.resetHead(message.parentId);
+        const threadMessage = this.toThreadMessage(message);
+        this.messages = [...this.messages, threadMessage];
+        this.notifyUpdate();
+        this.recordHistoryEntry(message.parentId ?? null, threadMessage);
+        if (!startRun)
+            return;
+        await this.startRun(threadMessage.id, message.runConfig);
+    }
+    async edit(message) {
+        await this.append(message);
+    }
+    async reload(parentId, config = {}) {
+        this.resetHead(parentId);
+        this.notifyUpdate();
+        await this.startRun(parentId, config.runConfig);
+    }
+    async cancel() {
+        if (!this.abortController)
+            return;
+        this.abortController.abort();
+    }
+    async resume(config) {
+        if (config.stream) {
+            this.logger.debug?.("[agui] resume stream is not supported, falling back to regular run");
+        }
+        await this.startRun(config.parentId, config.runConfig ?? this.lastRunConfig);
+    }
+    addToolResult(options) {
+        let updated = false;
+        this.messages = this.messages.map((message) => {
+            if (message.id !== options.messageId || message.role !== "assistant")
+                return message;
+            updated = true;
+            const assistant = message;
+            const content = assistant.content.map((part) => {
+                if (part.type !== "tool-call" || part.toolCallId !== options.toolCallId)
+                    return part;
+                return {
+                    ...part,
+                    result: options.result,
+                    artifact: options.artifact,
+                    isError: options.isError,
+                };
+            });
+            return {
+                ...assistant,
+                content,
+            };
+        });
+        if (updated) {
+            this.notifyUpdate();
+        }
+    }
+    applyExternalMessages(messages) {
+        this.assistantHistoryParents.clear();
+        this.messages = [...messages];
+        this.recordedHistoryIds.clear();
+        for (const message of this.messages) {
+            this.recordedHistoryIds.add(message.id);
+        }
+        this.notifyUpdate();
+    }
+    loadExternalState(state) {
+        this.stateSnapshot = state;
+        this.notifyUpdate();
+    }
+    async startRun(parentId, runConfig) {
+        const normalizedRunConfig = runConfig ?? {};
+        this.lastRunConfig = normalizedRunConfig;
+        this.resetHead(parentId);
+        const historicalMessages = [...this.messages];
+        const runId = INTERNAL.generateId();
+        this.pendingError = null;
+        const input = this.buildRunInput(runId, normalizedRunConfig, historicalMessages);
+        const assistantParentId = parentId ?? this.messages.at(-1)?.id ?? null;
+        let assistantMessageId;
+        const ensureAssistant = () => {
+            if (assistantMessageId)
+                return assistantMessageId;
+            const created = this.insertAssistantPlaceholder();
+            assistantMessageId = created;
+            this.markPendingAssistantHistory(created, assistantParentId ?? null);
+            return created;
+        };
+        const aggregator = new RunAggregator({
+            showThinking: this.showThinking,
+            logger: this.logger,
+            emit: (update) => this.updateAssistantMessage(ensureAssistant(), update),
+        });
+        const dispatch = (event) => this.handleEvent(aggregator, event);
+        const abortController = new AbortController();
+        const abortSignal = abortController.signal;
+        this.abortController = abortController;
+        abortSignal.addEventListener("abort", () => {
+            dispatch({ type: "RUN_CANCELLED" });
+            this.finishRun(abortController);
+            this.onCancel?.();
+        }, { once: true });
+        const subscriber = createAgUiSubscriber({
+            dispatch,
+            runId,
+            onRunFailed: (error) => {
+                this.pendingError = error;
+                this.onError?.(error);
+            },
+        });
+        aggregator.handle({ type: "RUN_STARTED", runId });
+        this.setRunning(true);
+        try {
+            try {
+                this.agent.messages = input.messages;
+                this.agent.threadId = input.threadId;
+            }
+            catch {
+                // ignore
+            }
+            await this.agent.runAgent(input, subscriber, {
+                signal: abortSignal,
+            });
+        }
+        catch (error) {
+            if (!abortSignal.aborted) {
+                const err = error instanceof Error ? error : new Error(String(error));
+                dispatch({ type: "RUN_ERROR", message: err.message });
+                this.onError?.(err);
+                this.pendingError = this.pendingError ?? err;
+            }
+        }
+        finally {
+            this.finishRun(abortController);
+        }
+        if (this.pendingError) {
+            const err = this.pendingError;
+            this.pendingError = null;
+            throw err;
+        }
+    }
+    buildRunInput(runId, runConfig, historyMessages) {
+        const threadId = "main";
+        const messages = toAgUiMessages(historyMessages ?? this.messages);
+        const context = this.runtime?.thread.getModelContext();
         return {
-          ...part,
-          result: options.result,
-          artifact: options.artifact,
-          isError: options.isError
+            threadId,
+            runId,
+            state: this.stateSnapshot ?? null,
+            messages,
+            tools: toAgUiTools(context?.tools),
+            context: context?.system
+                ? [{ description: "system", value: context.system }]
+                : [],
+            forwardedProps: {
+                ...(context?.callSettings ?? {}),
+                ...(context?.config ?? {}),
+                ...(runConfig?.custom ? { runConfig: runConfig.custom } : {}),
+            },
         };
-      });
-      return {
-        ...assistant,
-        content
-      };
-    });
-    if (updated) {
-      this.notifyUpdate();
     }
-  }
-  applyExternalMessages(messages) {
-    this.assistantHistoryParents.clear();
-    this.messages = [...messages];
-    this.recordedHistoryIds.clear();
-    for (const message of this.messages) {
-      this.recordedHistoryIds.add(message.id);
+    setRunning(running) {
+        this.isRunningFlag = running;
+        this.notifyUpdate();
     }
-    this.notifyUpdate();
-  }
-  loadExternalState(state) {
-    this.stateSnapshot = state;
-    this.notifyUpdate();
-  }
-  async startRun(parentId, runConfig) {
-    const normalizedRunConfig = runConfig ?? {};
-    this.lastRunConfig = normalizedRunConfig;
-    this.resetHead(parentId);
-    const historicalMessages = [...this.messages];
-    const runId = INTERNAL.generateId();
-    this.pendingError = null;
-    const input = this.buildRunInput(
-      runId,
-      normalizedRunConfig,
-      historicalMessages
-    );
-    const assistantParentId = parentId ?? this.messages.at(-1)?.id ?? null;
-    let assistantMessageId;
-    const ensureAssistant = () => {
-      if (assistantMessageId) return assistantMessageId;
-      const created = this.insertAssistantPlaceholder();
-      assistantMessageId = created;
-      this.markPendingAssistantHistory(created, assistantParentId ?? null);
-      return created;
-    };
-    const aggregator = new RunAggregator({
-      showThinking: this.showThinking,
-      logger: this.logger,
-      emit: (update) => this.updateAssistantMessage(ensureAssistant(), update)
-    });
-    const dispatch = (event) => this.handleEvent(aggregator, event);
-    const abortController = new AbortController();
-    const abortSignal = abortController.signal;
-    this.abortController = abortController;
-    abortSignal.addEventListener(
-      "abort",
-      () => {
-        dispatch({ type: "RUN_CANCELLED" });
-        this.finishRun(abortController);
-        this.onCancel?.();
-      },
-      { once: true }
-    );
-    const subscriber = createAgUiSubscriber({
-      dispatch,
-      runId,
-      onRunFailed: (error) => {
-        this.pendingError = error;
-        this.onError?.(error);
-      }
-    });
-    aggregator.handle({ type: "RUN_STARTED", runId });
-    this.setRunning(true);
-    try {
-      try {
-        this.agent.messages = input.messages;
-        this.agent.threadId = input.threadId;
-      } catch {
-      }
-      await this.agent.runAgent(input, subscriber, {
-        signal: abortSignal
-      });
-    } catch (error) {
-      if (!abortSignal.aborted) {
-        const err = error instanceof Error ? error : new Error(String(error));
-        dispatch({ type: "RUN_ERROR", message: err.message });
-        this.onError?.(err);
-        this.pendingError = this.pendingError ?? err;
-      }
-    } finally {
-      this.finishRun(abortController);
+    finishRun(controller) {
+        if (this.abortController === controller) {
+            this.abortController = null;
+        }
+        this.setRunning(false);
     }
-    if (this.pendingError) {
-      const err = this.pendingError;
-      this.pendingError = null;
-      throw err;
+    insertAssistantPlaceholder() {
+        const id = INTERNAL.generateId();
+        const assistant = {
+            id,
+            role: "assistant",
+            createdAt: new Date(),
+            status: { type: "running" },
+            content: [],
+            metadata: {
+                unstable_state: this.stateSnapshot ?? null,
+                unstable_annotations: [],
+                unstable_data: [],
+                steps: [],
+                custom: {},
+            },
+        };
+        this.messages = [...this.messages, assistant];
+        this.notifyUpdate();
+        return id;
     }
-  }
-  buildRunInput(runId, runConfig, historyMessages) {
-    const threadId = "main";
-    const messages = toAgUiMessages(historyMessages ?? this.messages);
-    const context = this.runtime?.thread.getModelContext();
-    return {
-      threadId,
-      runId,
-      state: this.stateSnapshot ?? null,
-      messages,
-      tools: toAgUiTools(context?.tools),
-      context: context?.system ? [{ description: "system", value: context.system }] : [],
-      forwardedProps: {
-        ...context?.callSettings ?? {},
-        ...context?.config ?? {},
-        ...runConfig?.custom ? { runConfig: runConfig.custom } : {}
-      }
-    };
-  }
-  setRunning(running) {
-    this.isRunningFlag = running;
-    this.notifyUpdate();
-  }
-  finishRun(controller) {
-    if (this.abortController === controller) {
-      this.abortController = null;
+    updateAssistantMessage(messageId, update) {
+        let touched = false;
+        let latestStatus;
+        this.messages = this.messages.map((message) => {
+            if (message.id !== messageId || message.role !== "assistant")
+                return message;
+            touched = true;
+            const assistant = message;
+            const metadata = update.metadata
+                ? this.mergeAssistantMetadata(assistant.metadata, update.metadata)
+                : assistant.metadata;
+            latestStatus = update.status ?? assistant.status;
+            return {
+                ...assistant,
+                content: (update.content ??
+                    assistant.content),
+                status: latestStatus,
+                metadata,
+            };
+        });
+        if (touched) {
+            this.notifyUpdate();
+            if (this.isTerminalStatus(latestStatus)) {
+                this.persistAssistantHistory(messageId);
+            }
+        }
     }
-    this.setRunning(false);
-  }
-  insertAssistantPlaceholder() {
-    const id = INTERNAL.generateId();
-    const assistant = {
-      id,
-      role: "assistant",
-      createdAt: /* @__PURE__ */ new Date(),
-      status: { type: "running" },
-      content: [],
-      metadata: {
-        unstable_state: this.stateSnapshot ?? null,
-        unstable_annotations: [],
-        unstable_data: [],
-        steps: [],
-        custom: {}
-      }
-    };
-    this.messages = [...this.messages, assistant];
-    this.notifyUpdate();
-    return id;
-  }
-  updateAssistantMessage(messageId, update) {
-    let touched = false;
-    let latestStatus;
-    this.messages = this.messages.map((message) => {
-      if (message.id !== messageId || message.role !== "assistant")
-        return message;
-      touched = true;
-      const assistant = message;
-      const metadata = update.metadata ? this.mergeAssistantMetadata(assistant.metadata, update.metadata) : assistant.metadata;
-      latestStatus = update.status ?? assistant.status;
-      return {
-        ...assistant,
-        content: update.content ?? assistant.content,
-        status: latestStatus,
-        metadata
-      };
-    });
-    if (touched) {
-      this.notifyUpdate();
-      if (this.isTerminalStatus(latestStatus)) {
-        this.persistAssistantHistory(messageId);
-      }
+    mergeAssistantMetadata(current, incoming) {
+        const annotations = incoming.unstable_annotations
+            ? [...current.unstable_annotations, ...incoming.unstable_annotations]
+            : current.unstable_annotations;
+        const data = incoming.unstable_data
+            ? [...current.unstable_data, ...incoming.unstable_data]
+            : current.unstable_data;
+        const steps = incoming.steps
+            ? [...current.steps, ...incoming.steps]
+            : current.steps;
+        return {
+            unstable_state: incoming.unstable_state !== undefined
+                ? incoming.unstable_state
+                : current.unstable_state,
+            unstable_annotations: annotations,
+            unstable_data: data,
+            steps,
+            custom: incoming.custom
+                ? { ...current.custom, ...incoming.custom }
+                : current.custom,
+        };
     }
-  }
-  mergeAssistantMetadata(current, incoming) {
-    const annotations = incoming.unstable_annotations ? [...current.unstable_annotations, ...incoming.unstable_annotations] : current.unstable_annotations;
-    const data = incoming.unstable_data ? [...current.unstable_data, ...incoming.unstable_data] : current.unstable_data;
-    const steps = incoming.steps ? [...current.steps, ...incoming.steps] : current.steps;
-    return {
-      unstable_state: incoming.unstable_state !== void 0 ? incoming.unstable_state : current.unstable_state,
-      unstable_annotations: annotations,
-      unstable_data: data,
-      steps,
-      custom: incoming.custom ? { ...current.custom, ...incoming.custom } : current.custom
-    };
-  }
-  handleEvent(aggregator, event) {
-    switch (event.type) {
-      case "STATE_SNAPSHOT": {
-        this.stateSnapshot = event.snapshot;
-        this.notifyUpdate();
-        return;
-      }
-      case "STATE_DELTA": {
-        this.logger.debug?.("[agui] state delta event ignored", event.delta);
-        return;
-      }
-      case "MESSAGES_SNAPSHOT": {
-        this.importMessagesSnapshot(event.messages);
-        return;
-      }
-      default:
-        aggregator.handle(event);
+    handleEvent(aggregator, event) {
+        switch (event.type) {
+            case "STATE_SNAPSHOT": {
+                this.stateSnapshot = event.snapshot;
+                this.notifyUpdate();
+                return;
+            }
+            case "STATE_DELTA": {
+                this.logger.debug?.("[agui] state delta event ignored", event.delta);
+                return;
+            }
+            case "MESSAGES_SNAPSHOT": {
+                this.importMessagesSnapshot(event.messages);
+                return;
+            }
+            default:
+                aggregator.handle(event);
+        }
+    }
+    importMessagesSnapshot(rawMessages) {
+        try {
+            const converted = rawMessages.map((message) => INTERNAL.fromThreadMessageLike(message, INTERNAL.generateId(), FALLBACK_USER_STATUS));
+            this.applyExternalMessages(converted);
+        }
+        catch (error) {
+            this.logger.error?.("[agui] failed to import messages snapshot", error);
+        }
+    }
+    toThreadMessage(message) {
+        return INTERNAL.fromThreadMessageLike(message, INTERNAL.generateId(), FALLBACK_USER_STATUS);
+    }
+    resetHead(parentId) {
+        if (!parentId) {
+            if (this.messages.length) {
+                this.messages = [];
+            }
+            return;
+        }
+        const idx = this.messages.findIndex((message) => message.id === parentId);
+        if (idx === -1)
+            return;
+        this.messages = this.messages.slice(0, idx + 1);
+    }
+    isTerminalStatus(status) {
+        return status?.type === "complete" || status?.type === "incomplete";
+    }
+    recordHistoryEntry(parentId, message) {
+        this.appendHistoryItem(parentId, message);
+    }
+    markPendingAssistantHistory(messageId, parentId) {
+        if (!this.history)
+            return;
+        this.assistantHistoryParents.set(messageId, parentId);
     }
-  }
-  importMessagesSnapshot(rawMessages) {
-    try {
-      const converted = rawMessages.map(
-        (message) => INTERNAL.fromThreadMessageLike(
-          message,
-          INTERNAL.generateId(),
-          FALLBACK_USER_STATUS
-        )
-      );
-      this.applyExternalMessages(converted);
-    } catch (error) {
-      this.logger.error?.("[agui] failed to import messages snapshot", error);
+    persistAssistantHistory(messageId) {
+        if (!this.history)
+            return;
+        const parentId = this.assistantHistoryParents.get(messageId);
+        if (parentId === undefined)
+            return;
+        const message = this.messages.find((m) => m.id === messageId);
+        if (!message || message.role !== "assistant")
+            return;
+        if (!this.isTerminalStatus(message.status))
+            return;
+        this.assistantHistoryParents.delete(messageId);
+        this.appendHistoryItem(parentId, message);
     }
-  }
-  toThreadMessage(message) {
-    return INTERNAL.fromThreadMessageLike(
-      message,
-      INTERNAL.generateId(),
-      FALLBACK_USER_STATUS
-    );
-  }
-  resetHead(parentId) {
-    if (!parentId) {
-      if (this.messages.length) {
-        this.messages = [];
-      }
-      return;
+    appendHistoryItem(parentId, message) {
+        if (!this.history || this.recordedHistoryIds.has(message.id))
+            return;
+        this.recordedHistoryIds.add(message.id);
+        void this.history.append({ parentId, message }).catch((error) => {
+            this.recordedHistoryIds.delete(message.id);
+            this.logger.error?.("[agui] failed to append history entry", error);
+        });
     }
-    const idx = this.messages.findIndex((message) => message.id === parentId);
-    if (idx === -1) return;
-    this.messages = this.messages.slice(0, idx + 1);
-  }
-  isTerminalStatus(status) {
-    return status?.type === "complete" || status?.type === "incomplete";
-  }
-  recordHistoryEntry(parentId, message) {
-    this.appendHistoryItem(parentId, message);
-  }
-  markPendingAssistantHistory(messageId, parentId) {
-    if (!this.history) return;
-    this.assistantHistoryParents.set(messageId, parentId);
-  }
-  persistAssistantHistory(messageId) {
-    if (!this.history) return;
-    const parentId = this.assistantHistoryParents.get(messageId);
-    if (parentId === void 0) return;
-    const message = this.messages.find((m) => m.id === messageId);
-    if (!message || message.role !== "assistant") return;
-    if (!this.isTerminalStatus(message.status)) return;
-    this.assistantHistoryParents.delete(messageId);
-    this.appendHistoryItem(parentId, message);
-  }
-  appendHistoryItem(parentId, message) {
-    if (!this.history || this.recordedHistoryIds.has(message.id)) return;
-    this.recordedHistoryIds.add(message.id);
-    void this.history.append({ parentId, message }).catch((error) => {
-      this.recordedHistoryIds.delete(message.id);
-      this.logger.error?.("[agui] failed to append history entry", error);
-    });
-  }
-};
-export {
-  AgUiThreadRuntimeCore
-};
+}
 //# sourceMappingURL=AgUiThreadRuntimeCore.js.map