@assistant-ui/react-ag-ui 0.0.2

package/dist/runtime/adapter/run-aggregator.js

@@ -0,0 +1,302 @@
+"use client";
+
+// src/runtime/adapter/run-aggregator.ts
+var RunAggregator = class {
+  emitUpdate;
+  showThinking;
+  logger;
+  status;
+  textParts = /* @__PURE__ */ new Map();
+  activeTextMessageId;
+  reasoningBuffer = "";
+  reasoningActive = false;
+  toolCalls = /* @__PURE__ */ new Map();
+  partOrder = [];
+  hasReasoningPart = false;
+  textPartCounter = 0;
+  constructor(options) {
+    this.emitUpdate = options.emit;
+    this.showThinking = options.showThinking;
+    this.logger = options.logger;
+  }
+  handle(event) {
+    switch (event.type) {
+      case "RUN_STARTED": {
+        this.clearTextParts();
+        this.reasoningBuffer = "";
+        this.reasoningActive = false;
+        this.toolCalls.clear();
+        this.partOrder.length = 0;
+        this.hasReasoningPart = false;
+        this.textPartCounter = 0;
+        this.activeTextMessageId = void 0;
+        this.status = { type: "running" };
+        this.emit();
+        break;
+      }
+      case "RUN_FINISHED": {
+        this.status = { type: "complete", reason: "unknown" };
+        this.emit();
+        break;
+      }
+      case "RUN_ERROR": {
+        this.status = {
+          type: "incomplete",
+          reason: "error",
+          ...event.message !== void 0 ? { error: event.message } : {}
+        };
+        this.emit();
+        break;
+      }
+      case "RUN_CANCELLED": {
+        this.status = { type: "incomplete", reason: "cancelled" };
+        this.emit();
+        break;
+      }
+      case "TEXT_MESSAGE_START": {
+        const id = this.startTextMessage(event.messageId);
+        if (id) {
+          this.markTextPartTouched(id);
+        }
+        this.emit();
+        break;
+      }
+      case "TEXT_MESSAGE_CONTENT":
+      case "TEXT_MESSAGE_CHUNK": {
+        if (!event.delta) break;
+        const id = this.resolveTextMessageId(
+          "messageId" in event ? event.messageId : void 0
+        );
+        this.appendText(id, event.delta);
+        this.emit();
+        break;
+      }
+      case "TEXT_MESSAGE_END": {
+        if (event.messageId && this.activeTextMessageId === event.messageId) {
+          this.activeTextMessageId = void 0;
+        }
+        this.emit();
+        break;
+      }
+      case "THINKING_START":
+      case "THINKING_TEXT_MESSAGE_START": {
+        if (!this.showThinking) break;
+        this.reasoningActive = true;
+        if (!this.reasoningBuffer) this.reasoningBuffer = "";
+        this.ensureReasoningPart();
+        this.emit();
+        break;
+      }
+      case "THINKING_TEXT_MESSAGE_CONTENT": {
+        if (!this.showThinking || !event.delta) break;
+        this.reasoningBuffer += event.delta;
+        this.ensureReasoningPart();
+        this.emit();
+        break;
+      }
+      case "THINKING_TEXT_MESSAGE_END":
+      case "THINKING_END": {
+        if (!this.showThinking) break;
+        this.emit();
+        break;
+      }
+      case "TOOL_CALL_START": {
+        this.startToolCall(
+          event.toolCallId,
+          event.toolCallName,
+          event.parentMessageId
+        );
+        this.emit();
+        break;
+      }
+      case "TOOL_CALL_ARGS":
+      case "TOOL_CALL_CHUNK": {
+        if (!event.delta) break;
+        this.appendToolArgs(event.toolCallId, event.delta);
+        this.emit();
+        break;
+      }
+      case "TOOL_CALL_END": {
+        this.emit();
+        break;
+      }
+      case "TOOL_CALL_RESULT": {
+        this.finishToolCall(
+          event.toolCallId,
+          event.content ?? "",
+          event.role === "tool" ? false : void 0
+        );
+        this.emit();
+        break;
+      }
+      default: {
+        this.logger.debug?.("[agui] aggregator ignored event", event);
+      }
+    }
+  }
+  clearTextParts() {
+    this.textParts.clear();
+  }
+  generateTextKey() {
+    this.textPartCounter += 1;
+    return `text-${this.textPartCounter}`;
+  }
+  startTextMessage(messageId) {
+    const id = messageId ?? this.generateTextKey();
+    this.ensureTextPart(id);
+    this.activeTextMessageId = id;
+    return id;
+  }
+  resolveTextMessageId(messageId) {
+    if (messageId) {
+      this.ensureTextPart(messageId);
+      this.activeTextMessageId = messageId;
+      return messageId;
+    }
+    if (this.activeTextMessageId) {
+      return this.activeTextMessageId;
+    }
+    const generated = this.generateTextKey();
+    this.ensureTextPart(generated);
+    this.activeTextMessageId = generated;
+    return generated;
+  }
+  ensureTextPart(id) {
+    if (!this.textParts.has(id)) {
+      this.textParts.set(id, { buffer: "", touched: false });
+      if (!this.partOrder.some((part) => part.kind === "text" && part.key === id)) {
+        this.partOrder.push({ kind: "text", key: id });
+      }
+    }
+  }
+  markTextPartTouched(id) {
+    const entry = this.textParts.get(id);
+    if (!entry) return;
+    entry.touched = true;
+  }
+  appendText(id, delta) {
+    this.ensureTextPart(id);
+    const entry = this.textParts.get(id);
+    if (!entry) return;
+    entry.buffer += delta;
+    entry.touched = true;
+  }
+  startToolCall(id, name, parentMessageId) {
+    if (!id) return;
+    if (!this.partOrder.some(
+      (part) => part.kind === "tool-call" && part.toolCallId === id
+    )) {
+      this.partOrder.push({ kind: "tool-call", toolCallId: id });
+    }
+    const state = {
+      toolCallId: id,
+      toolCallName: name ?? "tool",
+      argsText: "",
+      parsedArgs: void 0,
+      result: void 0,
+      isError: void 0
+    };
+    if (parentMessageId) {
+      state.parentMessageId = parentMessageId;
+    }
+    this.toolCalls.set(id, state);
+  }
+  appendToolArgs(id, delta) {
+    const entry = id ? this.toolCalls.get(id) : void 0;
+    if (!entry) return;
+    entry.argsText += delta;
+    try {
+      const parsed = JSON.parse(entry.argsText);
+      if (parsed && typeof parsed === "object") {
+        entry.parsedArgs = parsed;
+      } else {
+        entry.parsedArgs = void 0;
+      }
+    } catch {
+      entry.parsedArgs = void 0;
+    }
+  }
+  finishToolCall(id, content, isError) {
+    if (!id) return;
+    let entry = this.toolCalls.get(id);
+    if (!entry) {
+      entry = {
+        toolCallId: id,
+        toolCallName: "tool",
+        argsText: "",
+        parsedArgs: void 0,
+        result: void 0,
+        isError: void 0
+      };
+      this.toolCalls.set(id, entry);
+    }
+    if (!this.partOrder.some(
+      (part) => part.kind === "tool-call" && part.toolCallId === id
+    )) {
+      this.partOrder.push({ kind: "tool-call", toolCallId: id });
+    }
+    entry.result = this.tryParseJSON(content);
+    entry.isError = isError;
+  }
+  tryParseJSON(value) {
+    if (!value) return value;
+    try {
+      return JSON.parse(value);
+    } catch {
+      return value;
+    }
+  }
+  emit() {
+    const snapshot = [];
+    for (const part of this.partOrder) {
+      if (part.kind === "reasoning") {
+        if (this.showThinking && (this.reasoningActive || this.reasoningBuffer.length > 0)) {
+          snapshot.push({
+            type: "reasoning",
+            text: this.reasoningBuffer
+          });
+        }
+        continue;
+      }
+      if (part.kind === "text") {
+        const entry2 = this.textParts.get(part.key);
+        if (entry2?.touched) {
+          snapshot.push({ type: "text", text: entry2.buffer });
+        }
+        continue;
+      }
+      const entry = this.toolCalls.get(part.toolCallId);
+      if (!entry) continue;
+      const toolPart = {
+        type: "tool-call",
+        toolCallId: entry.toolCallId,
+        toolName: entry.toolCallName,
+        args: entry.parsedArgs ?? {},
+        argsText: entry.argsText,
+        ...entry.result !== void 0 ? { result: entry.result } : {},
+        ...entry.isError !== void 0 ? { isError: entry.isError } : {},
+        ...entry.parentMessageId ? { parentId: entry.parentMessageId } : {}
+      };
+      snapshot.push(toolPart);
+    }
+    const result = {
+      content: snapshot,
+      ...this.status ? { status: this.status } : void 0
+    };
+    this.emitUpdate(result);
+  }
+  ensureReasoningPart() {
+    if (this.hasReasoningPart) return;
+    const textIndex = this.partOrder.findIndex((part) => part.kind === "text");
+    if (textIndex === -1) {
+      this.partOrder.push({ kind: "reasoning" });
+    } else {
+      this.partOrder.splice(textIndex, 0, { kind: "reasoning" });
+    }
+    this.hasReasoningPart = true;
+  }
+};
+export {
+  RunAggregator
+};
+//# sourceMappingURL=run-aggregator.js.map

package/dist/runtime/adapter/run-aggregator.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/runtime/adapter/run-aggregator.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  ChatModelRunResult,\n  ThreadAssistantMessagePart,\n  ToolCallMessagePart,\n} from \"@assistant-ui/react\";\nimport type { AgUiEvent } from \"../types\";\nimport type { Logger } from \"../logger\";\n\ntype Emit = (update: ChatModelRunResult) => void;\n\ntype ToolCallState = {\n  toolCallId: string;\n  toolCallName: string;\n  argsText: string;\n  parsedArgs: Record<string, unknown> | undefined;\n  result: unknown;\n  isError: boolean | undefined;\n  parentMessageId?: string;\n};\n\nexport type RunAggregatorOptions = {\n  showThinking: boolean;\n  logger: Logger;\n  emit: Emit;\n};\n\n/**\n * Collects AG-UI events into assistant-ui run snapshots that can be yielded from a ChatModelAdapter.\n *\n * The aggregator keeps a single assistant message worth of parts. Each incoming event updates the parts and\n * emits a fresh snapshot through the provided `emit` callback.\n */\nexport class RunAggregator {\n  private readonly emitUpdate: Emit;\n  private readonly showThinking: boolean;\n  private readonly logger: Logger;\n\n  private status: ChatModelRunResult[\"status\"] | undefined;\n  private readonly textParts = new Map<\n    string,\n    { buffer: string; touched: boolean }\n  >();\n  private activeTextMessageId: string | undefined;\n  private reasoningBuffer = \"\";\n  private reasoningActive = false;\n  private readonly toolCalls = new Map<string, ToolCallState>();\n  private readonly partOrder: (\n    | { kind: \"text\"; key: string }\n    | { kind: \"reasoning\" }\n    | { kind: \"tool-call\"; toolCallId: string }\n  )[] = [];\n  private hasReasoningPart = false;\n  private textPartCounter = 0;\n\n  constructor(options: RunAggregatorOptions) {\n    this.emitUpdate = options.emit;\n    this.showThinking = options.showThinking;\n    this.logger = options.logger;\n  }\n\n  handle(event: AgUiEvent): void {\n    switch (event.type) {\n      case \"RUN_STARTED\": {\n        this.clearTextParts();\n        this.reasoningBuffer = \"\";\n        this.reasoningActive = false;\n        this.toolCalls.clear();\n        this.partOrder.length = 0;\n        this.hasReasoningPart = false;\n        this.textPartCounter = 0;\n        this.activeTextMessageId = undefined;\n        this.status = { type: \"running\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_FINISHED\": {\n        this.status = { type: \"complete\", reason: \"unknown\" };\n        this.emit();\n        break;\n      }\n      case \"RUN_ERROR\": {\n        this.status = {\n          type: \"incomplete\",\n          reason: \"error\",\n          ...(event.message !== undefined ? { error: event.message } : {}),\n        };\n        this.emit();\n        break;\n      }\n      case \"RUN_CANCELLED\": {\n        this.status = { type: \"incomplete\", reason: \"cancelled\" };\n        this.emit();\n        break;\n      }\n\n      case \"TEXT_MESSAGE_START\": {\n        const id = this.startTextMessage(event.messageId);\n        if (id) {\n          this.markTextPartTouched(id);\n        }\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_CONTENT\":\n      case \"TEXT_MESSAGE_CHUNK\": {\n        if (!event.delta) break;\n        const id = this.resolveTextMessageId(\n          \"messageId\" in event ? event.messageId : undefined,\n        );\n        this.appendText(id, event.delta);\n        this.emit();\n        break;\n      }\n      case \"TEXT_MESSAGE_END\": {\n        if (event.messageId && this.activeTextMessageId === event.messageId) {\n          this.activeTextMessageId = undefined;\n        }\n        this.emit();\n        break;\n      }\n\n      case \"THINKING_START\":\n      case \"THINKING_TEXT_MESSAGE_START\": {\n        if (!this.showThinking) break;\n        this.reasoningActive = true;\n        if (!this.reasoningBuffer) this.reasoningBuffer = \"\";\n        this.ensureReasoningPart();\n        this.emit();\n        break;\n      }\n      case \"THINKING_TEXT_MESSAGE_CONTENT\": {\n        if (!this.showThinking || !event.delta) break;\n        this.reasoningBuffer += event.delta;\n        this.ensureReasoningPart();\n        this.emit();\n        break;\n      }\n      case \"THINKING_TEXT_MESSAGE_END\":\n      case \"THINKING_END\": {\n        if (!this.showThinking) break;\n        this.emit();\n        break;\n      }\n\n      case \"TOOL_CALL_START\": {\n        this.startToolCall(\n          event.toolCallId,\n          event.toolCallName,\n          event.parentMessageId,\n        );\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_ARGS\":\n      case \"TOOL_CALL_CHUNK\": {\n        if (!event.delta) break;\n        this.appendToolArgs(event.toolCallId, event.delta);\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_END\": {\n        this.emit();\n        break;\n      }\n      case \"TOOL_CALL_RESULT\": {\n        this.finishToolCall(\n          event.toolCallId,\n          event.content ?? \"\",\n          event.role === \"tool\" ? false : undefined,\n        );\n        this.emit();\n        break;\n      }\n\n      default: {\n        this.logger.debug?.(\"[agui] aggregator ignored event\", event);\n      }\n    }\n  }\n\n  private clearTextParts(): void {\n    this.textParts.clear();\n  }\n\n  private generateTextKey(): string {\n    this.textPartCounter += 1;\n    return `text-${this.textPartCounter}`;\n  }\n\n  private startTextMessage(messageId?: string): string {\n    const id = messageId ?? this.generateTextKey();\n    this.ensureTextPart(id);\n    this.activeTextMessageId = id;\n    return id;\n  }\n\n  private resolveTextMessageId(messageId?: string): string {\n    if (messageId) {\n      this.ensureTextPart(messageId);\n      this.activeTextMessageId = messageId;\n      return messageId;\n    }\n\n    if (this.activeTextMessageId) {\n      return this.activeTextMessageId;\n    }\n\n    const generated = this.generateTextKey();\n    this.ensureTextPart(generated);\n    this.activeTextMessageId = generated;\n    return generated;\n  }\n\n  private ensureTextPart(id: string): void {\n    if (!this.textParts.has(id)) {\n      this.textParts.set(id, { buffer: \"\", touched: false });\n      if (\n        !this.partOrder.some((part) => part.kind === \"text\" && part.key === id)\n      ) {\n        this.partOrder.push({ kind: \"text\", key: id });\n      }\n    }\n  }\n\n  private markTextPartTouched(id: string): void {\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.touched = true;\n  }\n\n  private appendText(id: string, delta: string): void {\n    this.ensureTextPart(id);\n    const entry = this.textParts.get(id);\n    if (!entry) return;\n    entry.buffer += delta;\n    entry.touched = true;\n  }\n\n  private startToolCall(\n    id: string | undefined,\n    name?: string,\n    parentMessageId?: string,\n  ) {\n    if (!id) return;\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    const state: ToolCallState = {\n      toolCallId: id,\n      toolCallName: name ?? \"tool\",\n      argsText: \"\",\n      parsedArgs: undefined,\n      result: undefined,\n      isError: undefined,\n    };\n    if (parentMessageId) {\n      state.parentMessageId = parentMessageId;\n    }\n    this.toolCalls.set(id, state);\n  }\n\n  private appendToolArgs(id: string | undefined, delta: string) {\n    const entry = id ? this.toolCalls.get(id) : undefined;\n    if (!entry) return;\n    entry.argsText += delta;\n    try {\n      const parsed = JSON.parse(entry.argsText);\n      if (parsed && typeof parsed === \"object\") {\n        entry.parsedArgs = parsed as Record<string, unknown>;\n      } else {\n        entry.parsedArgs = undefined;\n      }\n    } catch {\n      entry.parsedArgs = undefined;\n    }\n  }\n\n  private finishToolCall(id: string, content: string, isError?: boolean) {\n    if (!id) return;\n    let entry = this.toolCalls.get(id);\n    if (!entry) {\n      entry = {\n        toolCallId: id,\n        toolCallName: \"tool\",\n        argsText: \"\",\n        parsedArgs: undefined,\n        result: undefined,\n        isError: undefined,\n      };\n      this.toolCalls.set(id, entry);\n    }\n    if (\n      !this.partOrder.some(\n        (part) => part.kind === \"tool-call\" && part.toolCallId === id,\n      )\n    ) {\n      this.partOrder.push({ kind: \"tool-call\", toolCallId: id });\n    }\n    entry.result = this.tryParseJSON(content);\n    entry.isError = isError;\n  }\n\n  private tryParseJSON(value: string): unknown {\n    if (!value) return value;\n    try {\n      return JSON.parse(value);\n    } catch {\n      return value;\n    }\n  }\n\n  private emit(): void {\n    const snapshot: ThreadAssistantMessagePart[] = [];\n\n    for (const part of this.partOrder) {\n      if (part.kind === \"reasoning\") {\n        if (\n          this.showThinking &&\n          (this.reasoningActive || this.reasoningBuffer.length > 0)\n        ) {\n          snapshot.push({\n            type: \"reasoning\",\n            text: this.reasoningBuffer,\n          } as const);\n        }\n        continue;\n      }\n\n      if (part.kind === \"text\") {\n        const entry = this.textParts.get(part.key);\n        if (entry?.touched) {\n          snapshot.push({ type: \"text\", text: entry.buffer } as const);\n        }\n        continue;\n      }\n\n      const entry = this.toolCalls.get(part.toolCallId);\n      if (!entry) continue;\n      const toolPart: ToolCallMessagePart = {\n        type: \"tool-call\",\n        toolCallId: entry.toolCallId,\n        toolName: entry.toolCallName,\n        args: (entry.parsedArgs ?? {}) as any,\n        argsText: entry.argsText,\n        ...(entry.result !== undefined ? { result: entry.result } : {}),\n        ...(entry.isError !== undefined ? { isError: entry.isError } : {}),\n        ...(entry.parentMessageId ? { parentId: entry.parentMessageId } : {}),\n      } as ToolCallMessagePart;\n      snapshot.push(toolPart);\n    }\n\n    const result: ChatModelRunResult = {\n      content: snapshot,\n      ...(this.status ? { status: this.status } : undefined),\n    };\n    this.emitUpdate(result);\n  }\n\n  private ensureReasoningPart(): void {\n    if (this.hasReasoningPart) return;\n    // ensure reasoning appears before the first text segment if possible\n    const textIndex = this.partOrder.findIndex((part) => part.kind === \"text\");\n    if (textIndex === -1) {\n      this.partOrder.push({ kind: \"reasoning\" });\n    } else {\n      this.partOrder.splice(textIndex, 0, { kind: \"reasoning\" });\n    }\n    this.hasReasoningPart = true;\n  }\n}\n"],"mappings":";;;AAkCO,IAAM,gBAAN,MAAoB;AAAA,EACR;AAAA,EACA;AAAA,EACA;AAAA,EAET;AAAA,EACS,YAAY,oBAAI,IAG/B;AAAA,EACM;AAAA,EACA,kBAAkB;AAAA,EAClB,kBAAkB;AAAA,EACT,YAAY,oBAAI,IAA2B;AAAA,EAC3C,YAIX,CAAC;AAAA,EACC,mBAAmB;AAAA,EACnB,kBAAkB;AAAA,EAE1B,YAAY,SAA+B;AACzC,SAAK,aAAa,QAAQ;AAC1B,SAAK,eAAe,QAAQ;AAC5B,SAAK,SAAS,QAAQ;AAAA,EACxB;AAAA,EAEA,OAAO,OAAwB;AAC7B,YAAQ,MAAM,MAAM;AAAA,MAClB,KAAK,eAAe;AAClB,aAAK,eAAe;AACpB,aAAK,kBAAkB;AACvB,aAAK,kBAAkB;AACvB,aAAK,UAAU,MAAM;AACrB,aAAK,UAAU,SAAS;AACxB,aAAK,mBAAmB;AACxB,aAAK,kBAAkB;AACvB,aAAK,sBAAsB;AAC3B,aAAK,SAAS,EAAE,MAAM,UAAU;AAChC,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,gBAAgB;AACnB,aAAK,SAAS,EAAE,MAAM,YAAY,QAAQ,UAAU;AACpD,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,aAAa;AAChB,aAAK,SAAS;AAAA,UACZ,MAAM;AAAA,UACN,QAAQ;AAAA,UACR,GAAI,MAAM,YAAY,SAAY,EAAE,OAAO,MAAM,QAAQ,IAAI,CAAC;AAAA,QAChE;AACA,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,iBAAiB;AACpB,aAAK,SAAS,EAAE,MAAM,cAAc,QAAQ,YAAY;AACxD,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MAEA,KAAK,sBAAsB;AACzB,cAAM,KAAK,KAAK,iBAAiB,MAAM,SAAS;AAChD,YAAI,IAAI;AACN,eAAK,oBAAoB,EAAE;AAAA,QAC7B;AACA,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL,KAAK,sBAAsB;AACzB,YAAI,CAAC,MAAM,MAAO;AAClB,cAAM,KAAK,KAAK;AAAA,UACd,eAAe,QAAQ,MAAM,YAAY;AAAA,QAC3C;AACA,aAAK,WAAW,IAAI,MAAM,KAAK;AAC/B,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,oBAAoB;AACvB,YAAI,MAAM,aAAa,KAAK,wBAAwB,MAAM,WAAW;AACnE,eAAK,sBAAsB;AAAA,QAC7B;AACA,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MAEA,KAAK;AAAA,MACL,KAAK,+BAA+B;AAClC,YAAI,CAAC,KAAK,aAAc;AACxB,aAAK,kBAAkB;AACvB,YAAI,CAAC,KAAK,gBAAiB,MAAK,kBAAkB;AAClD,aAAK,oBAAoB;AACzB,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,iCAAiC;AACpC,YAAI,CAAC,KAAK,gBAAgB,CAAC,MAAM,MAAO;AACxC,aAAK,mBAAmB,MAAM;AAC9B,aAAK,oBAAoB;AACzB,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL,KAAK,gBAAgB;AACnB,YAAI,CAAC,KAAK,aAAc;AACxB,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MAEA,KAAK,mBAAmB;AACtB,aAAK;AAAA,UACH,MAAM;AAAA,UACN,MAAM;AAAA,UACN,MAAM;AAAA,QACR;AACA,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK;AAAA,MACL,KAAK,mBAAmB;AACtB,YAAI,CAAC,MAAM,MAAO;AAClB,aAAK,eAAe,MAAM,YAAY,MAAM,KAAK;AACjD,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,iBAAiB;AACpB,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MACA,KAAK,oBAAoB;AACvB,aAAK;AAAA,UACH,MAAM;AAAA,UACN,MAAM,WAAW;AAAA,UACjB,MAAM,SAAS,SAAS,QAAQ;AAAA,QAClC;AACA,aAAK,KAAK;AACV;AAAA,MACF;AAAA,MAEA,SAAS;AACP,aAAK,OAAO,QAAQ,mCAAmC,KAAK;AAAA,MAC9D;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,iBAAuB;AAC7B,SAAK,UAAU,MAAM;AAAA,EACvB;AAAA,EAEQ,kBAA0B;AAChC,SAAK,mBAAmB;AACxB,WAAO,QAAQ,KAAK,eAAe;AAAA,EACrC;AAAA,EAEQ,iBAAiB,WAA4B;AACnD,UAAM,KAAK,aAAa,KAAK,gBAAgB;AAC7C,SAAK,eAAe,EAAE;AACtB,SAAK,sBAAsB;AAC3B,WAAO;AAAA,EACT;AAAA,EAEQ,qBAAqB,WAA4B;AACvD,QAAI,WAAW;AACb,WAAK,eAAe,SAAS;AAC7B,WAAK,sBAAsB;AAC3B,aAAO;AAAA,IACT;AAEA,QAAI,KAAK,qBAAqB;AAC5B,aAAO,KAAK;AAAA,IACd;AAEA,UAAM,YAAY,KAAK,gBAAgB;AACvC,SAAK,eAAe,SAAS;AAC7B,SAAK,sBAAsB;AAC3B,WAAO;AAAA,EACT;AAAA,EAEQ,eAAe,IAAkB;AACvC,QAAI,CAAC,KAAK,UAAU,IAAI,EAAE,GAAG;AAC3B,WAAK,UAAU,IAAI,IAAI,EAAE,QAAQ,IAAI,SAAS,MAAM,CAAC;AACrD,UACE,CAAC,KAAK,UAAU,KAAK,CAAC,SAAS,KAAK,SAAS,UAAU,KAAK,QAAQ,EAAE,GACtE;AACA,aAAK,UAAU,KAAK,EAAE,MAAM,QAAQ,KAAK,GAAG,CAAC;AAAA,MAC/C;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,oBAAoB,IAAkB;AAC5C,UAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;AACnC,QAAI,CAAC,MAAO;AACZ,UAAM,UAAU;AAAA,EAClB;AAAA,EAEQ,WAAW,IAAY,OAAqB;AAClD,SAAK,eAAe,EAAE;AACtB,UAAM,QAAQ,KAAK,UAAU,IAAI,EAAE;AACnC,QAAI,CAAC,MAAO;AACZ,UAAM,UAAU;AAChB,UAAM,UAAU;AAAA,EAClB;AAAA,EAEQ,cACN,IACA,MACA,iBACA;AACA,QAAI,CAAC,GAAI;AACT,QACE,CAAC,KAAK,UAAU;AAAA,MACd,CAAC,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe;AAAA,IAC7D,GACA;AACA,WAAK,UAAU,KAAK,EAAE,MAAM,aAAa,YAAY,GAAG,CAAC;AAAA,IAC3D;AACA,UAAM,QAAuB;AAAA,MAC3B,YAAY;AAAA,MACZ,cAAc,QAAQ;AAAA,MACtB,UAAU;AAAA,MACV,YAAY;AAAA,MACZ,QAAQ;AAAA,MACR,SAAS;AAAA,IACX;AACA,QAAI,iBAAiB;AACnB,YAAM,kBAAkB;AAAA,IAC1B;AACA,SAAK,UAAU,IAAI,IAAI,KAAK;AAAA,EAC9B;AAAA,EAEQ,eAAe,IAAwB,OAAe;AAC5D,UAAM,QAAQ,KAAK,KAAK,UAAU,IAAI,EAAE,IAAI;AAC5C,QAAI,CAAC,MAAO;AACZ,UAAM,YAAY;AAClB,QAAI;AACF,YAAM,SAAS,KAAK,MAAM,MAAM,QAAQ;AACxC,UAAI,UAAU,OAAO,WAAW,UAAU;AACxC,cAAM,aAAa;AAAA,MACrB,OAAO;AACL,cAAM,aAAa;AAAA,MACrB;AAAA,IACF,QAAQ;AACN,YAAM,aAAa;AAAA,IACrB;AAAA,EACF;AAAA,EAEQ,eAAe,IAAY,SAAiB,SAAmB;AACrE,QAAI,CAAC,GAAI;AACT,QAAI,QAAQ,KAAK,UAAU,IAAI,EAAE;AACjC,QAAI,CAAC,OAAO;AACV,cAAQ;AAAA,QACN,YAAY;AAAA,QACZ,cAAc;AAAA,QACd,UAAU;AAAA,QACV,YAAY;AAAA,QACZ,QAAQ;AAAA,QACR,SAAS;AAAA,MACX;AACA,WAAK,UAAU,IAAI,IAAI,KAAK;AAAA,IAC9B;AACA,QACE,CAAC,KAAK,UAAU;AAAA,MACd,CAAC,SAAS,KAAK,SAAS,eAAe,KAAK,eAAe;AAAA,IAC7D,GACA;AACA,WAAK,UAAU,KAAK,EAAE,MAAM,aAAa,YAAY,GAAG,CAAC;AAAA,IAC3D;AACA,UAAM,SAAS,KAAK,aAAa,OAAO;AACxC,UAAM,UAAU;AAAA,EAClB;AAAA,EAEQ,aAAa,OAAwB;AAC3C,QAAI,CAAC,MAAO,QAAO;AACnB,QAAI;AACF,aAAO,KAAK,MAAM,KAAK;AAAA,IACzB,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEQ,OAAa;AACnB,UAAM,WAAyC,CAAC;AAEhD,eAAW,QAAQ,KAAK,WAAW;AACjC,UAAI,KAAK,SAAS,aAAa;AAC7B,YACE,KAAK,iBACJ,KAAK,mBAAmB,KAAK,gBAAgB,SAAS,IACvD;AACA,mBAAS,KAAK;AAAA,YACZ,MAAM;AAAA,YACN,MAAM,KAAK;AAAA,UACb,CAAU;AAAA,QACZ;AACA;AAAA,MACF;AAEA,UAAI,KAAK,SAAS,QAAQ;AACxB,cAAMA,SAAQ,KAAK,UAAU,IAAI,KAAK,GAAG;AACzC,YAAIA,QAAO,SAAS;AAClB,mBAAS,KAAK,EAAE,MAAM,QAAQ,MAAMA,OAAM,OAAO,CAAU;AAAA,QAC7D;AACA;AAAA,MACF;AAEA,YAAM,QAAQ,KAAK,UAAU,IAAI,KAAK,UAAU;AAChD,UAAI,CAAC,MAAO;AACZ,YAAM,WAAgC;AAAA,QACpC,MAAM;AAAA,QACN,YAAY,MAAM;AAAA,QAClB,UAAU,MAAM;AAAA,QAChB,MAAO,MAAM,cAAc,CAAC;AAAA,QAC5B,UAAU,MAAM;AAAA,QAChB,GAAI,MAAM,WAAW,SAAY,EAAE,QAAQ,MAAM,OAAO,IAAI,CAAC;AAAA,QAC7D,GAAI,MAAM,YAAY,SAAY,EAAE,SAAS,MAAM,QAAQ,IAAI,CAAC;AAAA,QAChE,GAAI,MAAM,kBAAkB,EAAE,UAAU,MAAM,gBAAgB,IAAI,CAAC;AAAA,MACrE;AACA,eAAS,KAAK,QAAQ;AAAA,IACxB;AAEA,UAAM,SAA6B;AAAA,MACjC,SAAS;AAAA,MACT,GAAI,KAAK,SAAS,EAAE,QAAQ,KAAK,OAAO,IAAI;AAAA,IAC9C;AACA,SAAK,WAAW,MAAM;AAAA,EACxB;AAAA,EAEQ,sBAA4B;AAClC,QAAI,KAAK,iBAAkB;AAE3B,UAAM,YAAY,KAAK,UAAU,UAAU,CAAC,SAAS,KAAK,SAAS,MAAM;AACzE,QAAI,cAAc,IAAI;AACpB,WAAK,UAAU,KAAK,EAAE,MAAM,YAAY,CAAC;AAAA,IAC3C,OAAO;AACL,WAAK,UAAU,OAAO,WAAW,GAAG,EAAE,MAAM,YAAY,CAAC;AAAA,IAC3D;AACA,SAAK,mBAAmB;AAAA,EAC1B;AACF;","names":["entry"]}

package/dist/runtime/adapter/subscriber.js

@@ -0,0 +1,66 @@
+"use client";
+
+// src/runtime/adapter/subscriber.ts
+import { parseAgUiEvent } from "../event-parser.js";
+var ensureEvent = (raw, type) => {
+  if (raw && typeof raw === "object") {
+    const payload = raw;
+    if (typeof payload["type"] === "string") {
+      return parseAgUiEvent(payload);
+    }
+    return parseAgUiEvent({ type, ...payload });
+  }
+  return parseAgUiEvent({ type });
+};
+var dispatchIfValid = (dispatch, raw, type) => {
+  const event = ensureEvent(raw, type);
+  if (!event) return;
+  dispatch(event);
+};
+var createAgUiSubscriber = (options) => {
+  const { dispatch, runId, onRunFailed } = options;
+  return {
+    onEvent: ({ event }) => {
+      const typeCandidate = event && typeof event === "object" ? event["type"] : void 0;
+      if (typeof typeCandidate === "string") {
+        return;
+      }
+      const parsed = parseAgUiEvent(event);
+      if (parsed) dispatch(parsed);
+    },
+    onTextMessageStartEvent: ({ event }) => dispatchIfValid(dispatch, event, "TEXT_MESSAGE_START"),
+    onTextMessageContentEvent: ({ event }) => dispatchIfValid(dispatch, event, "TEXT_MESSAGE_CONTENT"),
+    onTextMessageEndEvent: ({ event }) => dispatchIfValid(dispatch, event, "TEXT_MESSAGE_END"),
+    onTextMessageChunkEvent: ({ event }) => dispatchIfValid(dispatch, event, "TEXT_MESSAGE_CHUNK"),
+    onThinkingStartEvent: ({ event }) => dispatchIfValid(dispatch, event, "THINKING_START"),
+    onThinkingEndEvent: ({ event }) => dispatchIfValid(dispatch, event, "THINKING_END"),
+    onThinkingTextMessageStartEvent: ({ event }) => dispatchIfValid(dispatch, event, "THINKING_TEXT_MESSAGE_START"),
+    onThinkingTextMessageContentEvent: ({ event }) => dispatchIfValid(dispatch, event, "THINKING_TEXT_MESSAGE_CONTENT"),
+    onThinkingTextMessageEndEvent: ({ event }) => dispatchIfValid(dispatch, event, "THINKING_TEXT_MESSAGE_END"),
+    onToolCallStartEvent: ({ event }) => dispatchIfValid(dispatch, event, "TOOL_CALL_START"),
+    onToolCallArgsEvent: ({ event }) => dispatchIfValid(dispatch, event, "TOOL_CALL_ARGS"),
+    onToolCallEndEvent: ({ event }) => dispatchIfValid(dispatch, event, "TOOL_CALL_END"),
+    onToolCallChunkEvent: ({ event }) => dispatchIfValid(dispatch, event, "TOOL_CALL_CHUNK"),
+    onToolCallResultEvent: ({ event }) => dispatchIfValid(dispatch, event, "TOOL_CALL_RESULT"),
+    onStateSnapshotEvent: ({ event }) => dispatchIfValid(dispatch, event, "STATE_SNAPSHOT"),
+    onStateDeltaEvent: ({ event }) => dispatchIfValid(dispatch, event, "STATE_DELTA"),
+    onMessagesSnapshotEvent: ({ event }) => dispatchIfValid(dispatch, event, "MESSAGES_SNAPSHOT"),
+    onCustomEvent: ({ event }) => dispatchIfValid(dispatch, event, "CUSTOM"),
+    onRawEvent: ({ event }) => dispatchIfValid(dispatch, event, "RAW"),
+    onRunFinalized: () => dispatch({ type: "RUN_FINISHED", runId }),
+    onRunFailed: ({ error }) => {
+      onRunFailed?.(error);
+      const message = typeof error.message === "string" ? error.message : "Run failed";
+      const code = typeof error?.code === "string" ? error.code : void 0;
+      dispatch({
+        type: "RUN_ERROR",
+        ...message !== void 0 ? { message } : {},
+        ...code !== void 0 ? { code } : {}
+      });
+    }
+  };
+};
+export {
+  createAgUiSubscriber
+};
+//# sourceMappingURL=subscriber.js.map

package/dist/runtime/adapter/subscriber.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/runtime/adapter/subscriber.ts"],"sourcesContent":["\"use client\";\n\nimport type { AgUiEvent } from \"../types\";\nimport { parseAgUiEvent } from \"../event-parser\";\n\ntype Dispatch = (event: AgUiEvent) => void;\n\ntype Subscriber = {\n  onEvent?: (payload: { event: unknown }) => void;\n  onTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onTextMessageChunkEvent?: (payload: { event: unknown }) => void;\n  onThinkingStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingEndEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageStartEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageContentEvent?: (payload: { event: unknown }) => void;\n  onThinkingTextMessageEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallStartEvent?: (payload: { event: unknown }) => void;\n  onToolCallArgsEvent?: (payload: { event: unknown }) => void;\n  onToolCallEndEvent?: (payload: { event: unknown }) => void;\n  onToolCallChunkEvent?: (payload: { event: unknown }) => void;\n  onToolCallResultEvent?: (payload: { event: unknown }) => void;\n  onStateSnapshotEvent?: (payload: { event: unknown }) => void;\n  onStateDeltaEvent?: (payload: { event: unknown }) => void;\n  onMessagesSnapshotEvent?: (payload: { event: unknown }) => void;\n  onCustomEvent?: (payload: { event: unknown }) => void;\n  onRawEvent?: (payload: { event: unknown }) => void;\n  onRunFinalized?: () => void;\n  onRunFailed?: (payload: { error: Error }) => void;\n};\n\nconst ensureEvent = (\n  raw: unknown,\n  type: AgUiEvent[\"type\"],\n): AgUiEvent | null => {\n  if (raw && typeof raw === \"object\") {\n    const payload = raw as Record<string, unknown>;\n    if (typeof payload[\"type\"] === \"string\") {\n      return parseAgUiEvent(payload);\n    }\n    return parseAgUiEvent({ type, ...payload });\n  }\n  return parseAgUiEvent({ type });\n};\n\nconst dispatchIfValid = (\n  dispatch: Dispatch,\n  raw: unknown,\n  type: AgUiEvent[\"type\"],\n) => {\n  const event = ensureEvent(raw, type);\n  if (!event) return;\n  dispatch(event);\n};\n\ntype SubscriberOptions = {\n  dispatch: Dispatch;\n  runId: string;\n  onRunFailed?: (error: Error) => void;\n};\n\nexport const createAgUiSubscriber = (\n  options: SubscriberOptions,\n): Subscriber => {\n  const { dispatch, runId, onRunFailed } = options;\n  return {\n    onEvent: ({ event }) => {\n      const typeCandidate =\n        event && typeof event === \"object\"\n          ? (event as Record<string, unknown>)[\"type\"]\n          : undefined;\n      if (typeof typeCandidate === \"string\") {\n        // Typed handlers will receive this via the discriminated callbacks; avoid duplicates.\n        return;\n      }\n      const parsed = parseAgUiEvent(event);\n      if (parsed) dispatch(parsed);\n    },\n    onTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TEXT_MESSAGE_START\"),\n    onTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TEXT_MESSAGE_CONTENT\"),\n    onTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TEXT_MESSAGE_END\"),\n    onTextMessageChunkEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TEXT_MESSAGE_CHUNK\"),\n    onThinkingStartEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"THINKING_START\"),\n    onThinkingEndEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"THINKING_END\"),\n    onThinkingTextMessageStartEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"THINKING_TEXT_MESSAGE_START\"),\n    onThinkingTextMessageContentEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"THINKING_TEXT_MESSAGE_CONTENT\"),\n    onThinkingTextMessageEndEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"THINKING_TEXT_MESSAGE_END\"),\n    onToolCallStartEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TOOL_CALL_START\"),\n    onToolCallArgsEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TOOL_CALL_ARGS\"),\n    onToolCallEndEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TOOL_CALL_END\"),\n    onToolCallChunkEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TOOL_CALL_CHUNK\"),\n    onToolCallResultEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"TOOL_CALL_RESULT\"),\n    onStateSnapshotEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"STATE_SNAPSHOT\"),\n    onStateDeltaEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"STATE_DELTA\"),\n    onMessagesSnapshotEvent: ({ event }) =>\n      dispatchIfValid(dispatch, event, \"MESSAGES_SNAPSHOT\"),\n    onCustomEvent: ({ event }) => dispatchIfValid(dispatch, event, \"CUSTOM\"),\n    onRawEvent: ({ event }) => dispatchIfValid(dispatch, event, \"RAW\"),\n    onRunFinalized: () => dispatch({ type: \"RUN_FINISHED\", runId }),\n    onRunFailed: ({ error }) => {\n      onRunFailed?.(error);\n      const message =\n        typeof error.message === \"string\" ? error.message : \"Run failed\";\n      const code =\n        typeof (error as any)?.code === \"string\"\n          ? (error as any).code\n          : undefined;\n      dispatch({\n        type: \"RUN_ERROR\" as const,\n        ...(message !== undefined ? { message } : {}),\n        ...(code !== undefined ? { code } : {}),\n      } satisfies AgUiEvent);\n    },\n  };\n};\n\nexport type AgUiSubscriber = ReturnType<typeof createAgUiSubscriber>;\n"],"mappings":";;;AAGA,SAAS,sBAAsB;AA6B/B,IAAM,cAAc,CAClB,KACA,SACqB;AACrB,MAAI,OAAO,OAAO,QAAQ,UAAU;AAClC,UAAM,UAAU;AAChB,QAAI,OAAO,QAAQ,MAAM,MAAM,UAAU;AACvC,aAAO,eAAe,OAAO;AAAA,IAC/B;AACA,WAAO,eAAe,EAAE,MAAM,GAAG,QAAQ,CAAC;AAAA,EAC5C;AACA,SAAO,eAAe,EAAE,KAAK,CAAC;AAChC;AAEA,IAAM,kBAAkB,CACtB,UACA,KACA,SACG;AACH,QAAM,QAAQ,YAAY,KAAK,IAAI;AACnC,MAAI,CAAC,MAAO;AACZ,WAAS,KAAK;AAChB;AAQO,IAAM,uBAAuB,CAClC,YACe;AACf,QAAM,EAAE,UAAU,OAAO,YAAY,IAAI;AACzC,SAAO;AAAA,IACL,SAAS,CAAC,EAAE,MAAM,MAAM;AACtB,YAAM,gBACJ,SAAS,OAAO,UAAU,WACrB,MAAkC,MAAM,IACzC;AACN,UAAI,OAAO,kBAAkB,UAAU;AAErC;AAAA,MACF;AACA,YAAM,SAAS,eAAe,KAAK;AACnC,UAAI,OAAQ,UAAS,MAAM;AAAA,IAC7B;AAAA,IACA,yBAAyB,CAAC,EAAE,MAAM,MAChC,gBAAgB,UAAU,OAAO,oBAAoB;AAAA,IACvD,2BAA2B,CAAC,EAAE,MAAM,MAClC,gBAAgB,UAAU,OAAO,sBAAsB;AAAA,IACzD,uBAAuB,CAAC,EAAE,MAAM,MAC9B,gBAAgB,UAAU,OAAO,kBAAkB;AAAA,IACrD,yBAAyB,CAAC,EAAE,MAAM,MAChC,gBAAgB,UAAU,OAAO,oBAAoB;AAAA,IACvD,sBAAsB,CAAC,EAAE,MAAM,MAC7B,gBAAgB,UAAU,OAAO,gBAAgB;AAAA,IACnD,oBAAoB,CAAC,EAAE,MAAM,MAC3B,gBAAgB,UAAU,OAAO,cAAc;AAAA,IACjD,iCAAiC,CAAC,EAAE,MAAM,MACxC,gBAAgB,UAAU,OAAO,6BAA6B;AAAA,IAChE,mCAAmC,CAAC,EAAE,MAAM,MAC1C,gBAAgB,UAAU,OAAO,+BAA+B;AAAA,IAClE,+BAA+B,CAAC,EAAE,MAAM,MACtC,gBAAgB,UAAU,OAAO,2BAA2B;AAAA,IAC9D,sBAAsB,CAAC,EAAE,MAAM,MAC7B,gBAAgB,UAAU,OAAO,iBAAiB;AAAA,IACpD,qBAAqB,CAAC,EAAE,MAAM,MAC5B,gBAAgB,UAAU,OAAO,gBAAgB;AAAA,IACnD,oBAAoB,CAAC,EAAE,MAAM,MAC3B,gBAAgB,UAAU,OAAO,eAAe;AAAA,IAClD,sBAAsB,CAAC,EAAE,MAAM,MAC7B,gBAAgB,UAAU,OAAO,iBAAiB;AAAA,IACpD,uBAAuB,CAAC,EAAE,MAAM,MAC9B,gBAAgB,UAAU,OAAO,kBAAkB;AAAA,IACrD,sBAAsB,CAAC,EAAE,MAAM,MAC7B,gBAAgB,UAAU,OAAO,gBAAgB;AAAA,IACnD,mBAAmB,CAAC,EAAE,MAAM,MAC1B,gBAAgB,UAAU,OAAO,aAAa;AAAA,IAChD,yBAAyB,CAAC,EAAE,MAAM,MAChC,gBAAgB,UAAU,OAAO,mBAAmB;AAAA,IACtD,eAAe,CAAC,EAAE,MAAM,MAAM,gBAAgB,UAAU,OAAO,QAAQ;AAAA,IACvE,YAAY,CAAC,EAAE,MAAM,MAAM,gBAAgB,UAAU,OAAO,KAAK;AAAA,IACjE,gBAAgB,MAAM,SAAS,EAAE,MAAM,gBAAgB,MAAM,CAAC;AAAA,IAC9D,aAAa,CAAC,EAAE,MAAM,MAAM;AAC1B,oBAAc,KAAK;AACnB,YAAM,UACJ,OAAO,MAAM,YAAY,WAAW,MAAM,UAAU;AACtD,YAAM,OACJ,OAAQ,OAAe,SAAS,WAC3B,MAAc,OACf;AACN,eAAS;AAAA,QACP,MAAM;AAAA,QACN,GAAI,YAAY,SAAY,EAAE,QAAQ,IAAI,CAAC;AAAA,QAC3C,GAAI,SAAS,SAAY,EAAE,KAAK,IAAI,CAAC;AAAA,MACvC,CAAqB;AAAA,IACvB;AAAA,EACF;AACF;","names":[]}

package/dist/runtime/event-parser.js

@@ -0,0 +1,158 @@
+// src/runtime/event-parser.ts
+var isString = (value) => typeof value === "string";
+var isNonEmptyString = (value) => isString(value) && value.length > 0;
+var withOptional = (base, optionals) => {
+  const definedEntries = Object.entries(optionals).filter(
+    ([, value]) => value !== void 0
+  );
+  return definedEntries.length === 0 ? base : { ...base, ...Object.fromEntries(definedEntries) };
+};
+var parseAgUiEvent = (event) => {
+  if (!event || typeof event !== "object") return null;
+  const payload = event;
+  const typeValue = payload["type"];
+  if (!isString(typeValue)) return null;
+  const getString = (key) => isString(payload[key]) ? payload[key] : void 0;
+  switch (typeValue) {
+    case "RUN_STARTED": {
+      const runId = getString("runId");
+      return runId ? { type: "RUN_STARTED", runId } : null;
+    }
+    case "RUN_FINISHED": {
+      const runId = getString("runId");
+      return runId ? { type: "RUN_FINISHED", runId } : null;
+    }
+    case "RUN_CANCELLED": {
+      const runId = getString("runId");
+      return withOptional({ type: "RUN_CANCELLED" }, { runId });
+    }
+    case "RUN_ERROR": {
+      return withOptional(
+        { type: "RUN_ERROR" },
+        {
+          message: getString("message"),
+          code: getString("code")
+        }
+      );
+    }
+    case "TEXT_MESSAGE_START":
+      return withOptional(
+        { type: "TEXT_MESSAGE_START" },
+        { messageId: getString("messageId") }
+      );
+    case "TEXT_MESSAGE_CONTENT": {
+      const delta = getString("delta");
+      if (!isNonEmptyString(delta)) return null;
+      return withOptional(
+        { type: "TEXT_MESSAGE_CONTENT", delta },
+        { messageId: getString("messageId") }
+      );
+    }
+    case "TEXT_MESSAGE_END":
+      return withOptional(
+        { type: "TEXT_MESSAGE_END" },
+        { messageId: getString("messageId") }
+      );
+    case "TEXT_MESSAGE_CHUNK": {
+      const delta = getString("delta") ?? "";
+      return withOptional(
+        { type: "TEXT_MESSAGE_CHUNK", delta },
+        { messageId: getString("messageId") }
+      );
+    }
+    case "THINKING_START":
+      return withOptional(
+        { type: "THINKING_START" },
+        { title: getString("title") }
+      );
+    case "THINKING_TEXT_MESSAGE_START":
+      return { type: "THINKING_TEXT_MESSAGE_START" };
+    case "THINKING_TEXT_MESSAGE_CONTENT": {
+      const delta = getString("delta") ?? "";
+      return { type: "THINKING_TEXT_MESSAGE_CONTENT", delta };
+    }
+    case "THINKING_TEXT_MESSAGE_END":
+      return { type: "THINKING_TEXT_MESSAGE_END" };
+    case "THINKING_END":
+      return { type: "THINKING_END" };
+    case "TOOL_CALL_START": {
+      const toolCallId = getString("toolCallId");
+      if (!toolCallId) return null;
+      return withOptional(
+        { type: "TOOL_CALL_START", toolCallId },
+        {
+          toolCallName: getString("toolCallName"),
+          parentMessageId: getString("parentMessageId")
+        }
+      );
+    }
+    case "TOOL_CALL_ARGS": {
+      const toolCallId = getString("toolCallId");
+      if (!toolCallId) return null;
+      const delta = getString("delta") ?? "";
+      return { type: "TOOL_CALL_ARGS", toolCallId, delta };
+    }
+    case "TOOL_CALL_END": {
+      const toolCallId = getString("toolCallId");
+      return toolCallId ? { type: "TOOL_CALL_END", toolCallId } : null;
+    }
+    case "TOOL_CALL_CHUNK":
+      return withOptional(
+        { type: "TOOL_CALL_CHUNK" },
+        {
+          toolCallId: getString("toolCallId"),
+          toolCallName: getString("toolCallName"),
+          parentMessageId: getString("parentMessageId"),
+          delta: getString("delta")
+        }
+      );
+    case "TOOL_CALL_RESULT": {
+      const toolCallId = getString("toolCallId");
+      if (!toolCallId) return null;
+      return withOptional(
+        {
+          type: "TOOL_CALL_RESULT",
+          toolCallId,
+          content: getString("content") ?? ""
+        },
+        {
+          messageId: getString("messageId"),
+          role: payload["role"] === "tool" ? "tool" : void 0
+        }
+      );
+    }
+    case "STATE_SNAPSHOT":
+      return { type: "STATE_SNAPSHOT", snapshot: payload["snapshot"] };
+    case "STATE_DELTA":
+      return {
+        type: "STATE_DELTA",
+        delta: Array.isArray(payload["delta"]) ? payload["delta"] : []
+      };
+    case "MESSAGES_SNAPSHOT":
+      return {
+        type: "MESSAGES_SNAPSHOT",
+        messages: Array.isArray(payload["messages"]) ? payload["messages"] : []
+      };
+    case "RAW":
+      return withOptional(
+        { type: "RAW", event: payload["event"] },
+        { source: getString("source") }
+      );
+    case "CUSTOM": {
+      const name = getString("name");
+      if (!name) return null;
+      return { type: "CUSTOM", name, value: payload["value"] };
+    }
+    default:
+      return withOptional(
+        { type: "RAW", event: payload },
+        {
+          source: isString(payload["type"]) ? payload["type"] : void 0
+        }
+      );
+  }
+};
+export {
+  parseAgUiEvent
+};
+//# sourceMappingURL=event-parser.js.map

package/dist/runtime/event-parser.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/runtime/event-parser.ts"],"sourcesContent":["import type { AgUiEvent } from \"./types\";\n\nconst isString = (value: unknown): value is string => typeof value === \"string\";\nconst isNonEmptyString = (value: unknown): value is string =>\n  isString(value) && value.length > 0;\n\nconst withOptional = <T extends object>(\n  base: T,\n  optionals: Record<string, unknown>,\n) => {\n  const definedEntries = Object.entries(optionals).filter(\n    ([, value]) => value !== undefined,\n  );\n  return definedEntries.length === 0\n    ? base\n    : ({ ...base, ...Object.fromEntries(definedEntries) } as T);\n};\n\nexport const parseAgUiEvent = (event: unknown): AgUiEvent | null => {\n  if (!event || typeof event !== \"object\") return null;\n  const payload = event as Record<string, unknown>;\n  const typeValue = payload[\"type\"];\n  if (!isString(typeValue)) return null;\n\n  const getString = (key: string) =>\n    isString(payload[key]) ? (payload[key] as string) : undefined;\n\n  switch (typeValue) {\n    case \"RUN_STARTED\": {\n      const runId = getString(\"runId\");\n      return runId ? { type: \"RUN_STARTED\", runId } : null;\n    }\n    case \"RUN_FINISHED\": {\n      const runId = getString(\"runId\");\n      return runId ? { type: \"RUN_FINISHED\", runId } : null;\n    }\n    case \"RUN_CANCELLED\": {\n      const runId = getString(\"runId\");\n      return withOptional({ type: \"RUN_CANCELLED\" as const }, { runId });\n    }\n    case \"RUN_ERROR\": {\n      return withOptional(\n        { type: \"RUN_ERROR\" as const },\n        {\n          message: getString(\"message\"),\n          code: getString(\"code\"),\n        },\n      );\n    }\n    case \"TEXT_MESSAGE_START\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_START\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\");\n      if (!isNonEmptyString(delta)) return null;\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CONTENT\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"TEXT_MESSAGE_END\":\n      return withOptional(\n        { type: \"TEXT_MESSAGE_END\" as const },\n        { messageId: getString(\"messageId\") },\n      );\n    case \"TEXT_MESSAGE_CHUNK\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return withOptional(\n        { type: \"TEXT_MESSAGE_CHUNK\" as const, delta },\n        { messageId: getString(\"messageId\") },\n      );\n    }\n    case \"THINKING_START\":\n      return withOptional(\n        { type: \"THINKING_START\" as const },\n        { title: getString(\"title\") },\n      );\n    case \"THINKING_TEXT_MESSAGE_START\":\n      return { type: \"THINKING_TEXT_MESSAGE_START\" };\n    case \"THINKING_TEXT_MESSAGE_CONTENT\": {\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"THINKING_TEXT_MESSAGE_CONTENT\", delta };\n    }\n    case \"THINKING_TEXT_MESSAGE_END\":\n      return { type: \"THINKING_TEXT_MESSAGE_END\" };\n    case \"THINKING_END\":\n      return { type: \"THINKING_END\" };\n    case \"TOOL_CALL_START\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        { type: \"TOOL_CALL_START\" as const, toolCallId },\n        {\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n        },\n      );\n    }\n    case \"TOOL_CALL_ARGS\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      const delta = getString(\"delta\") ?? \"\";\n      return { type: \"TOOL_CALL_ARGS\", toolCallId, delta };\n    }\n    case \"TOOL_CALL_END\": {\n      const toolCallId = getString(\"toolCallId\");\n      return toolCallId ? { type: \"TOOL_CALL_END\", toolCallId } : null;\n    }\n    case \"TOOL_CALL_CHUNK\":\n      return withOptional(\n        { type: \"TOOL_CALL_CHUNK\" as const },\n        {\n          toolCallId: getString(\"toolCallId\"),\n          toolCallName: getString(\"toolCallName\"),\n          parentMessageId: getString(\"parentMessageId\"),\n          delta: getString(\"delta\"),\n        },\n      );\n    case \"TOOL_CALL_RESULT\": {\n      const toolCallId = getString(\"toolCallId\");\n      if (!toolCallId) return null;\n      return withOptional(\n        {\n          type: \"TOOL_CALL_RESULT\" as const,\n          toolCallId,\n          content: getString(\"content\") ?? \"\",\n        },\n        {\n          messageId: getString(\"messageId\"),\n          role: payload[\"role\"] === \"tool\" ? \"tool\" : undefined,\n        },\n      );\n    }\n    case \"STATE_SNAPSHOT\":\n      return { type: \"STATE_SNAPSHOT\", snapshot: payload[\"snapshot\"] };\n    case \"STATE_DELTA\":\n      return {\n        type: \"STATE_DELTA\",\n        delta: Array.isArray(payload[\"delta\"])\n          ? (payload[\"delta\"] as any[])\n          : [],\n      };\n    case \"MESSAGES_SNAPSHOT\":\n      return {\n        type: \"MESSAGES_SNAPSHOT\",\n        messages: Array.isArray(payload[\"messages\"])\n          ? (payload[\"messages\"] as any[])\n          : [],\n      };\n    case \"RAW\":\n      return withOptional(\n        { type: \"RAW\" as const, event: payload[\"event\"] },\n        { source: getString(\"source\") },\n      );\n    case \"CUSTOM\": {\n      const name = getString(\"name\");\n      if (!name) return null;\n      return { type: \"CUSTOM\", name, value: payload[\"value\"] };\n    }\n    default:\n      return withOptional(\n        { type: \"RAW\" as const, event: payload },\n        {\n          source: isString(payload[\"type\"])\n            ? (payload[\"type\"] as string)\n            : undefined,\n        },\n      );\n  }\n};\n"],"mappings":";AAEA,IAAM,WAAW,CAAC,UAAoC,OAAO,UAAU;AACvE,IAAM,mBAAmB,CAAC,UACxB,SAAS,KAAK,KAAK,MAAM,SAAS;AAEpC,IAAM,eAAe,CACnB,MACA,cACG;AACH,QAAM,iBAAiB,OAAO,QAAQ,SAAS,EAAE;AAAA,IAC/C,CAAC,CAAC,EAAE,KAAK,MAAM,UAAU;AAAA,EAC3B;AACA,SAAO,eAAe,WAAW,IAC7B,OACC,EAAE,GAAG,MAAM,GAAG,OAAO,YAAY,cAAc,EAAE;AACxD;AAEO,IAAM,iBAAiB,CAAC,UAAqC;AAClE,MAAI,CAAC,SAAS,OAAO,UAAU,SAAU,QAAO;AAChD,QAAM,UAAU;AAChB,QAAM,YAAY,QAAQ,MAAM;AAChC,MAAI,CAAC,SAAS,SAAS,EAAG,QAAO;AAEjC,QAAM,YAAY,CAAC,QACjB,SAAS,QAAQ,GAAG,CAAC,IAAK,QAAQ,GAAG,IAAe;AAEtD,UAAQ,WAAW;AAAA,IACjB,KAAK,eAAe;AAClB,YAAM,QAAQ,UAAU,OAAO;AAC/B,aAAO,QAAQ,EAAE,MAAM,eAAe,MAAM,IAAI;AAAA,IAClD;AAAA,IACA,KAAK,gBAAgB;AACnB,YAAM,QAAQ,UAAU,OAAO;AAC/B,aAAO,QAAQ,EAAE,MAAM,gBAAgB,MAAM,IAAI;AAAA,IACnD;AAAA,IACA,KAAK,iBAAiB;AACpB,YAAM,QAAQ,UAAU,OAAO;AAC/B,aAAO,aAAa,EAAE,MAAM,gBAAyB,GAAG,EAAE,MAAM,CAAC;AAAA,IACnE;AAAA,IACA,KAAK,aAAa;AAChB,aAAO;AAAA,QACL,EAAE,MAAM,YAAqB;AAAA,QAC7B;AAAA,UACE,SAAS,UAAU,SAAS;AAAA,UAC5B,MAAM,UAAU,MAAM;AAAA,QACxB;AAAA,MACF;AAAA,IACF;AAAA,IACA,KAAK;AACH,aAAO;AAAA,QACL,EAAE,MAAM,qBAA8B;AAAA,QACtC,EAAE,WAAW,UAAU,WAAW,EAAE;AAAA,MACtC;AAAA,IACF,KAAK,wBAAwB;AAC3B,YAAM,QAAQ,UAAU,OAAO;AAC/B,UAAI,CAAC,iBAAiB,KAAK,EAAG,QAAO;AACrC,aAAO;AAAA,QACL,EAAE,MAAM,wBAAiC,MAAM;AAAA,QAC/C,EAAE,WAAW,UAAU,WAAW,EAAE;AAAA,MACtC;AAAA,IACF;AAAA,IACA,KAAK;AACH,aAAO;AAAA,QACL,EAAE,MAAM,mBAA4B;AAAA,QACpC,EAAE,WAAW,UAAU,WAAW,EAAE;AAAA,MACtC;AAAA,IACF,KAAK,sBAAsB;AACzB,YAAM,QAAQ,UAAU,OAAO,KAAK;AACpC,aAAO;AAAA,QACL,EAAE,MAAM,sBAA+B,MAAM;AAAA,QAC7C,EAAE,WAAW,UAAU,WAAW,EAAE;AAAA,MACtC;AAAA,IACF;AAAA,IACA,KAAK;AACH,aAAO;AAAA,QACL,EAAE,MAAM,iBAA0B;AAAA,QAClC,EAAE,OAAO,UAAU,OAAO,EAAE;AAAA,MAC9B;AAAA,IACF,KAAK;AACH,aAAO,EAAE,MAAM,8BAA8B;AAAA,IAC/C,KAAK,iCAAiC;AACpC,YAAM,QAAQ,UAAU,OAAO,KAAK;AACpC,aAAO,EAAE,MAAM,iCAAiC,MAAM;AAAA,IACxD;AAAA,IACA,KAAK;AACH,aAAO,EAAE,MAAM,4BAA4B;AAAA,IAC7C,KAAK;AACH,aAAO,EAAE,MAAM,eAAe;AAAA,IAChC,KAAK,mBAAmB;AACtB,YAAM,aAAa,UAAU,YAAY;AACzC,UAAI,CAAC,WAAY,QAAO;AACxB,aAAO;AAAA,QACL,EAAE,MAAM,mBAA4B,WAAW;AAAA,QAC/C;AAAA,UACE,cAAc,UAAU,cAAc;AAAA,UACtC,iBAAiB,UAAU,iBAAiB;AAAA,QAC9C;AAAA,MACF;AAAA,IACF;AAAA,IACA,KAAK,kBAAkB;AACrB,YAAM,aAAa,UAAU,YAAY;AACzC,UAAI,CAAC,WAAY,QAAO;AACxB,YAAM,QAAQ,UAAU,OAAO,KAAK;AACpC,aAAO,EAAE,MAAM,kBAAkB,YAAY,MAAM;AAAA,IACrD;AAAA,IACA,KAAK,iBAAiB;AACpB,YAAM,aAAa,UAAU,YAAY;AACzC,aAAO,aAAa,EAAE,MAAM,iBAAiB,WAAW,IAAI;AAAA,IAC9D;AAAA,IACA,KAAK;AACH,aAAO;AAAA,QACL,EAAE,MAAM,kBAA2B;AAAA,QACnC;AAAA,UACE,YAAY,UAAU,YAAY;AAAA,UAClC,cAAc,UAAU,cAAc;AAAA,UACtC,iBAAiB,UAAU,iBAAiB;AAAA,UAC5C,OAAO,UAAU,OAAO;AAAA,QAC1B;AAAA,MACF;AAAA,IACF,KAAK,oBAAoB;AACvB,YAAM,aAAa,UAAU,YAAY;AACzC,UAAI,CAAC,WAAY,QAAO;AACxB,aAAO;AAAA,QACL;AAAA,UACE,MAAM;AAAA,UACN;AAAA,UACA,SAAS,UAAU,SAAS,KAAK;AAAA,QACnC;AAAA,QACA;AAAA,UACE,WAAW,UAAU,WAAW;AAAA,UAChC,MAAM,QAAQ,MAAM,MAAM,SAAS,SAAS;AAAA,QAC9C;AAAA,MACF;AAAA,IACF;AAAA,IACA,KAAK;AACH,aAAO,EAAE,MAAM,kBAAkB,UAAU,QAAQ,UAAU,EAAE;AAAA,IACjE,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,OAAO,MAAM,QAAQ,QAAQ,OAAO,CAAC,IAChC,QAAQ,OAAO,IAChB,CAAC;AAAA,MACP;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,MAAM;AAAA,QACN,UAAU,MAAM,QAAQ,QAAQ,UAAU,CAAC,IACtC,QAAQ,UAAU,IACnB,CAAC;AAAA,MACP;AAAA,IACF,KAAK;AACH,aAAO;AAAA,QACL,EAAE,MAAM,OAAgB,OAAO,QAAQ,OAAO,EAAE;AAAA,QAChD,EAAE,QAAQ,UAAU,QAAQ,EAAE;AAAA,MAChC;AAAA,IACF,KAAK,UAAU;AACb,YAAM,OAAO,UAAU,MAAM;AAC7B,UAAI,CAAC,KAAM,QAAO;AAClB,aAAO,EAAE,MAAM,UAAU,MAAM,OAAO,QAAQ,OAAO,EAAE;AAAA,IACzD;AAAA,IACA;AACE,aAAO;AAAA,QACL,EAAE,MAAM,OAAgB,OAAO,QAAQ;AAAA,QACvC;AAAA,UACE,QAAQ,SAAS,QAAQ,MAAM,CAAC,IAC3B,QAAQ,MAAM,IACf;AAAA,QACN;AAAA,MACF;AAAA,EACJ;AACF;","names":[]}

package/dist/runtime/logger.js

@@ -0,0 +1,9 @@
+// src/runtime/logger.ts
+var makeLogger = (l) => ({
+  debug: (...a) => l?.debug?.(...a),
+  error: (...a) => l?.error?.(...a)
+});
+export {
+  makeLogger
+};
+//# sourceMappingURL=logger.js.map

package/dist/runtime/logger.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/runtime/logger.ts"],"sourcesContent":["export type Logger = {\n  debug: (...a: any[]) => void;\n  error: (...a: any[]) => void;\n};\n\nexport const makeLogger = (l?: Partial<Logger>): Logger => ({\n  debug: (...a) => l?.debug?.(...a),\n  error: (...a) => l?.error?.(...a),\n});\n"],"mappings":";AAKO,IAAM,aAAa,CAAC,OAAiC;AAAA,EAC1D,OAAO,IAAI,MAAM,GAAG,QAAQ,GAAG,CAAC;AAAA,EAChC,OAAO,IAAI,MAAM,GAAG,QAAQ,GAAG,CAAC;AAClC;","names":[]}

package/dist/runtime/types.js

@@ -0,0 +1 @@
+//# sourceMappingURL=types.js.map

package/dist/runtime/types.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}