beeai-framework 0.1.21 → 0.1.23

package/dist/backend/embedding.d.ts

@@ -1,8 +1,8 @@
 import { Run, GetRunContext } from '../context.js';
 import { Serializable } from '../internals/serializable.js';
-import { C as Callback, E as Emitter } from '../emitter-CZFbzlUi.js';
+import { C as Callback, E as Emitter } from '../emitter-36-9MnvA.js';
 import { FrameworkError } from '../errors.js';
-import { F as FullModelName } from '../chat-C0s-o6ll.js';
+import { F as FullModelName } from '../chat-DrqwUgD8.js';
 import { ProviderName } from './constants.js';
 import '../internals/helpers/promise.js';
 import '../internals/types.js';

package/dist/backend/message.cjs

@@ -36,9 +36,11 @@ class Message extends serializable_cjs.Serializable {
     __name(this, "Message");
   }
   meta;
+  id;
   content;
-  constructor(content, meta = {}) {
+  constructor(content, meta = {}, id = void 0) {
     super(), this.meta = meta;
+    this.id = id;
     if (!meta?.createdAt) {
       meta.createdAt = /* @__PURE__ */ new Date();
     }
@@ -60,17 +62,17 @@ class Message extends serializable_cjs.Serializable {
   }
   assertContent(content) {
   }
-  static of({ role, text, meta }) {
+  static of({ role, text, meta, id }) {
     if (role === "user") {
-      return new UserMessage(text, meta);
+      return new UserMessage(text, meta, id);
     } else if (role === "assistant") {
-      return new AssistantMessage(text, meta);
+      return new AssistantMessage(text, meta, id);
     } else if (role === "system") {
-      return new SystemMessage(text, meta);
+      return new SystemMessage(text, meta, id);
     } else if (role === "tool") {
-      return new ToolMessage(text, meta);
+      return new ToolMessage(text, meta, id);
     } else {
-      return new CustomMessage(role, text, meta);
+      return new CustomMessage(role, text, meta, id);
     }
   }
   get text() {
@@ -81,6 +83,7 @@ class Message extends serializable_cjs.Serializable {
   }
   createSnapshot() {
     return {
+      id: this.id,
       content: utils_cjs.shallowCopy(this.content),
       meta: utils_cjs.shallowCopy(this.meta),
       role: this.role
@@ -91,11 +94,13 @@ class Message extends serializable_cjs.Serializable {
   }
   toPlain() {
     return {
+      id: this.id,
       role: this.role,
       content: utils_cjs.shallowCopy(this.content)
     };
   }
   merge(other) {
+    this.id = this.id || other.id;
     Object.assign(this, other.meta);
     this.content.push(...other.content);
   }
@@ -128,10 +133,12 @@ class AssistantMessage extends Message {
   assertContent(content) {
     if (content.type === "tool-call") {
       const key = "args";
-      const args = object_cjs.popProp(content, key);
-      if (args !== null && !object_cjs.hasProp(content, "input")) {
-        logger_cjs.Logger.root.warn(`The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`);
-        content.input = args;
+      if (!Object.getOwnPropertyDescriptor(content, key)) {
+        const args = object_cjs.popProp(content, key);
+        if (args !== null && !object_cjs.hasProp(content, "input")) {
+          logger_cjs.Logger.root.warn(`The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`);
+          content.input = args;
+        }
       }
       object_cjs.safeDefineProperty(content, key, () => {
         logger_cjs.Logger.root.warn(`The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`);
@@ -240,8 +247,8 @@ class CustomMessage extends Message {
     __name(this, "CustomMessage");
   }
   role;
-  constructor(role, content, meta = {}) {
-    super(content, meta);
+  constructor(role, content, meta = {}, id) {
+    super(content, meta, id);
     if (!role) {
       throw new errors_cjs.ValueError(`Role "${role}" must be specified!`);
     }

package/dist/backend/message.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/backend/message.ts"],"names":["isText","content","type","isImage","isFile","isToolCall","isToolResult","Message","Serializable","meta","createdAt","Date","fromString","Array","isArray","chunk","assertContent","watchArray","onAdd","of","role","text","UserMessage","AssistantMessage","SystemMessage","ToolMessage","CustomMessage","getTexts","map","c","join","filter","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign","toPlain","merge","other","push","fromChunks","chunks","instance","forEach","Symbol","iterator","register","getToolCalls","key","args","popProp","hasProp","Logger","root","warn","input","safeDefineProperty","getToolResults","schema","z","object","literal","toolName","string","toolCallId","and","output","any","or","result","success","data","safeParse","ValueError","isError","isString","value","includes","getImages","getFiles","Role","ASSISTANT","SYSTEM","USER"],"mappings":";;;;;;;;;;;;;AA6BA,SAASA,OAAOC,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSF,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASG,QAAQF,OAAAA,EAA2B;AAC1C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,OAAA;AAC1B;AAFSC,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAGT,SAASC,OAAOH,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSE,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASC,WAAWJ,OAAAA,EAA2B;AAC7C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,WAAA;AAC1B;AAFSG,MAAAA,CAAAA,UAAAA,EAAAA,YAAAA,CAAAA;AAGT,SAASC,aAAaL,OAAAA,EAA2B;AAC/C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,aAAA;AAC1B;AAFSI,MAAAA,CAAAA,YAAAA,EAAAA,cAAAA,CAAAA;AAIF,MAAeC,gBAGZC,6BAAAA,CAAAA;EAhDV;;;;AAkDkBP,EAAAA,OAAAA;EAEhB,WAAA,CACEA,OAAAA,EACgBQ,IAAAA,GAAoB,EAAC,EACrC;AACA,IAAA,KAAA,EAAK,EAAA,KAFWA,IAAAA,GAAAA,IAAAA;AAGhB,IAAA,IAAI,CAACA,MAAMC,SAAAA,EAAW;AACpBD,MAAAA,IAAAA,CAAKC,SAAAA,uBAAgBC,IAAAA,EAAAA;AACvB;AACA,IAAA,IAAI,OAAOV,YAAY,QAAA,EAAU;AAC/B,MAAA,IAAA,CAAKA,OAAAA,GAAU;AAAC,QAAA,IAAA,CAAKW,WAAWX,OAAAA;;KAClC,MAAO;AACL,MAAA,IAAA,CAAKA,OAAAA,GAAUY,KAAAA,CAAMC,OAAAA,CAAQb,OAAAA,IAAWA,OAAAA,GAAU;AAACA,QAAAA;;AACrD;AACA,IAAA,KAAA,MAAWc,KAAAA,IAAS,KAAKd,OAAAA,EAAS;AAChC,MAAA,IAAA,CAAKe,cAAcD,KAAAA,CAAAA;AACrB;AACA,IAAA,IAAA,CAAKd,OAAAA,GAAUgB,oBAAAA,CAAW,IAAA,CAAKhB,OAAAA,EAAS;AACtCiB,MAAAA,KAAAA,EAAO,IAAA,CAAKF;KACd,CAAA;AACF;AAEUA,EAAAA,aAAAA,CAAcf,OAAAA,EAAkC;AAAC;AAI3D,EAAA,OAAOkB,EAAAA,CAAG,EAAEC,IAAAA,EAAMC,IAAAA,EAAMZ,MAAI,EAA2B;AACrD,IAAA,IAAIW,SAAS,MAAA,EAAQ;AACnB,MAAA,OAAO,IAAIE,WAAAA,CAAYD,IAAAA,EAAMZ,IAAAA,CAAAA;AAC/B,KAAA,MAAA,IAAWW,SAAS,WAAA,EAAa;AAC/B,MAAA,OAAO,IAAIG,gBAAAA,CAAiBF,IAAAA,EAAMZ,IAAAA,CAAAA;AACpC,KAAA,MAAA,IAAWW,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,IAAII,aAAAA,CAAcH,IAAAA,EAAMZ,IAAAA,CAAAA;AACjC,KAAA,MAAA,IAAWW,SAAS,MAAA,EAAQ;AAC1B,MAAA,OAAO,IAAIK,WAAAA,CAAYJ,IAAAA,EAAMZ,IAAAA,CAAAA;KAC/B,MAAO;AACL,MAAA,OAAO,IAAIiB,aAAAA,CAAcN,IAAAA,EAAMC,IAAAA,EAAMZ,IAAAA,CAAAA;AACvC;AACF;AAEA,EAAA,IAAIY,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKM,QAAAA,EAAQ,CACjBC,GAAAA,CAAI,CAACC,MAAMA,CAAAA,CAAER,IAAI,CAAA,CACjBS,IAAAA,CAAK,EAAA,CAAA;AACV;EAEAH,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK1B,OAAAA,CAAQ8B,MAAAA,CAAO/B,MAAAA,CAAAA;AAC7B;EAEAgC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAE/B,OAAAA,EAASgC,qBAAAA,CAAY,KAAKhC,OAAO,CAAA;MAAGQ,IAAAA,EAAMwB,qBAAAA,CAAY,KAAKxB,IAAI,CAAA;AAAGW,MAAAA,IAAAA,EAAM,IAAA,CAAKA;AAAK,KAAA;AAC7F;AAEAc,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;EAEAG,OAAAA,GAAU;AACR,IAAA,OAAO;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMnB,OAAAA,EAASgC,qBAAAA,CAAY,KAAKhC,OAAO;AAAE,KAAA;AAC/D;AAEAsC,EAAAA,KAAAA,CAAMC,KAAAA,EAAsB;AAC1BJ,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,EAAMG,KAAAA,CAAM/B,IAAI,CAAA;AAC9B,IAAA,IAAA,CAAKR,OAAAA,CAAQwC,IAAAA,CAAI,GAAID,KAAAA,CAAMvC,OAAO,CAAA;AACpC;AAEA,EAAA,OAAOyC,WAAiEC,MAAAA,EAAkB;AACxF,IAAA,MAAMC,QAAAA,GAAW,IAAI,IAAA,CAAK,EAAE,CAAA;AAC5BD,IAAAA,MAAAA,CAAOE,QAAQ,CAAC9B,KAAAA,KAAU6B,QAAAA,CAASL,KAAAA,CAAMxB,KAAAA,CAAAA,CAAAA;AACzC,IAAA,OAAO6B,QAAAA;AACT;EAEA,CAACE,MAAAA,CAAOC,QAAQ,CAAA,GAAiB;AAC/B,IAAA,OAAO,IAAA,CAAK9C,OAAAA,CAAQ6C,MAAAA,CAAOC,QAAQ,CAAA,EAAC;AACtC;AACF;AAEO,MAAMxB,yBAAyBhB,OAAAA,CAAAA;EAjItC;;;EAkIkBa,IAAAA,GAAO,WAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAC,YAAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAKhD,OAAAA,CAAQ8B,MAAAA,CAAO1B,UAAAA,CAAAA;AAC7B;AAEUO,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AAEUL,EAAAA,aAAAA,CACRf,OAAAA,EACuD;AACvD,IAAA,IAAIA,OAAAA,CAAQC,SAAS,WAAA,EAAa;AAChC,MAAA,MAAMgD,GAAAA,GAAM,MAAA;AACZ,MAAA,MAAMC,IAAAA,GAAOC,kBAAAA,CAAQnD,OAAAA,EAAgBiD,GAAAA,CAAAA;AAErC,MAAA,IAAIC,SAAS,IAAA,IAAQ,CAACE,kBAAAA,CAAQpD,OAAAA,EAAS,OAAA,CAAA,EAAU;AAC/CqD,QAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQN,GAAAA,CAAAA,0IAAAA,CAA+I,CAAA;AAEzJjD,QAAAA,OAAAA,CAAQwD,KAAAA,GAAQN,IAAAA;AAClB;AAEAO,MAAAA,6BAAAA,CAAmBzD,OAAAA,EAASiD,KAA6B,MAAA;AACvDI,QAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQN,GAAAA,CAAAA,gGAAAA,CAAqG,CAAA;AAE/G,QAAA,OAAOjD,OAAAA,CAAQwD,KAAAA;OACjB,CAAA;AACF;AACF;AACF;AAEO,MAAMhC,oBAAoBlB,OAAAA,CAAAA;EAxKjC;;;EAyKkBa,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO,IAAA,CAAK1D,OAAAA,CAAQ8B,MAAAA,CAAOzB,YAAAA,CAAAA;AAC7B;AAEUM,EAAAA,UAAAA,CAAWS,IAAAA,EAA8B;AACjD,IAAA,MAAMuC,MAAAA,GAASC,MACZC,MAAAA,CAAO;MACN5D,IAAAA,EAAM2D,KAAAA,CAAEE,QAAQ,aAAA,CAAA;AAChBC,MAAAA,QAAAA,EAAUH,MAAEI,MAAAA,EAAM;AAClBC,MAAAA,UAAAA,EAAYL,MAAEI,MAAAA;KAChB,CAAA,CACCE,GAAAA,CAAIN,KAAAA,CAAEC,MAAAA,CAAO;AAAEM,MAAAA,MAAAA,EAAQP,MAAEQ,GAAAA;KAAM,CAAA,CAAGC,EAAAA,CAAGT,KAAAA,CAAEC,MAAAA,CAAO;AAAES,MAAAA,MAAAA,EAAQV,MAAEQ,GAAAA;AAAM,KAAA,CAAA,CAAA,CAAA;AAEnE,IAAA,MAAM,EAAEG,OAAAA,EAASC,IAAAA,EAAI,GAAKb,MAAAA,CAAOc,UAAUrD,IAAAA,CAAAA;AAC3C,IAAA,IAAI,CAACmD,OAAAA,EAAS;AACZ,MAAA,MAAM,IAAIG,qBAAAA,CAAW,CAAA,oCAAA,EAAuCtD,IAAAA,CAAAA,EAAAA,CAAQ,CAAA;AACtE;AAEA,IAAA,OAAOoD,IAAAA;AACT;AAEUzD,EAAAA,aAAAA,CAAcf,OAAAA,EAA4D;AAClF,IAAA,MAAMsE,MAAAA,GAASnB,kBAAAA,CAAQnD,OAAAA,EAAgB,QAAA,EAAU,IAAA,CAAA;AACjD,IAAA,MAAM2E,OAAAA,GAAUxB,kBAAAA,CAAQnD,OAAAA,EAAgB,SAAA,EAAW,KAAA,CAAA;AAEnD,IAAA,IAAIsE,WAAW,IAAA,IAAQ,CAAClB,kBAAAA,CAAQpD,OAAAA,EAAS,QAAA,CAAA,EAAW;AAClDqD,MAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,KACV,yIAAA,CAAA;AAEF,MAAA,IAAIqB,eAAAA,CAASN,MAAAA,CAAAA,EAAS;AACpBtE,QAAAA,OAAAA,CAAQmE,MAAAA,GAAS;AAAElE,UAAAA,IAAAA,EAAM0E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;OAC1E,MAAO;AACLtE,QAAAA,OAAAA,CAAQmE,MAAAA,GAAS;AAAElE,UAAAA,IAAAA,EAAM0E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;AAC1E;AACF;AAEAb,IAAAA,6BAAAA,CAAmBzD,OAAAA,EAAS,UAAiB,MAAA;AAC3CqD,MAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAOvD,QAAQmE,MAAAA,CAAOU,KAAAA;KACxB,CAAA;AACApB,IAAAA,6BAAAA,CAAmBzD,OAAAA,EAAS,WAAkB,MAAA;AAC5CqD,MAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAOvD,OAAAA,CAAQmE,MAAAA,CAAOlE,IAAAA,CAAK6E,QAAAA,CAAS,OAAA,CAAA;KACtC,CAAA;AACF;AACF;AAEO,MAAMvD,sBAAsBjB,OAAAA,CAAAA;EAlOnC;;;EAmOkBa,IAAAA,GAAoB,QAAA;EAEpC;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;AAEUpC,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMC,oBAAoBf,OAAAA,CAAAA;EA9OjC;;;EA+OkBa,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAgC,SAAAA,GAAY;AACV,IAAA,OAAO,IAAA,CAAK/E,OAAAA,CAAQ8B,MAAAA,CAAO5B,OAAAA,CAAAA;AAC7B;EAEA8E,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAKhF,OAAAA,CAAQ8B,MAAAA,CAAO3B,MAAAA,CAAAA;AAC7B;AAEUQ,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAM6D,IAAAA,GAAO;EAClBC,SAAAA,EAAW,WAAA;EACXC,MAAAA,EAAQ,QAAA;EACRC,IAAAA,EAAM;AACR;AAEO,MAAM3D,sBAAsBnB,OAAAA,CAAAA;EAxQnC;;;AAyQSa,EAAAA,IAAAA;AAEP,EAAA,WAAA,CAAYA,IAAAA,EAAcnB,OAAAA,EAAsCQ,IAAAA,GAAoB,EAAC,EAAG;AACtF,IAAA,KAAA,CAAMR,SAASQ,IAAAA,CAAAA;AACf,IAAA,IAAI,CAACW,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIuD,qBAAAA,CAAW,CAAA,MAAA,EAASvD,IAAAA,CAAAA,oBAAAA,CAA0B,CAAA;AAC1D;AACA,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA;AACd;AAEUR,EAAAA,UAAAA,CAAW6C,KAAAA,EAAmC;AACtD,IAAA,OAAO;MAAEvD,IAAAA,EAAM,MAAA;MAAQmB,IAAAA,EAAMoC;AAAM,KAAA;AACrC;AACF","file":"message.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { FilePart, ImagePart, TextPart, ToolCallPart, ToolResultPart } from \"ai\";\nimport { z } from \"zod\";\nimport { ValueError } from \"@/errors.js\";\nimport { hasProp, popProp, safeDefineProperty } from \"@/internals/helpers/object.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { isString } from \"remeda\";\nimport { watchArray } from \"@/internals/helpers/array.js\";\n\nexport type MessageRole = \"user\" | \"system\" | \"tool\" | \"assistant\";\nexport type MessageContentPart = TextPart | ToolCallPart | ImagePart | FilePart | ToolResultPart;\n\nexport interface MessageMeta {\n  [key: string]: any;\n  createdAt?: Date;\n}\n\nexport interface MessageInput {\n  role: MessageRole;\n  text: string; // TODO\n  meta?: MessageMeta;\n}\n\nfunction isText(content: MessageContentPart): content is TextPart {\n  return content.type === \"text\";\n}\nfunction isImage(content: MessageContentPart): content is ImagePart {\n  return content.type === \"image\";\n}\nfunction isFile(content: MessageContentPart): content is FilePart {\n  return content.type === \"file\";\n}\nfunction isToolCall(content: MessageContentPart): content is ToolCallPart {\n  return content.type === \"tool-call\";\n}\nfunction isToolResult(content: MessageContentPart): content is ToolResultPart {\n  return content.type === \"tool-result\";\n}\n\nexport abstract class Message<\n  T extends MessageContentPart = MessageContentPart,\n  R extends string = MessageRole | string,\n> extends Serializable {\n  public abstract readonly role: R;\n  public readonly content: T[];\n\n  constructor(\n    content: T | T[] | string,\n    public readonly meta: MessageMeta = {},\n  ) {\n    super();\n    if (!meta?.createdAt) {\n      meta.createdAt = new Date();\n    }\n    if (typeof content === \"string\") {\n      this.content = [this.fromString(content)];\n    } else {\n      this.content = Array.isArray(content) ? content : [content];\n    }\n    for (const chunk of this.content) {\n      this.assertContent(chunk);\n    }\n    this.content = watchArray(this.content, {\n      onAdd: this.assertContent,\n    });\n  }\n\n  protected assertContent(content: T): asserts content is T {}\n\n  protected abstract fromString(input: string): T;\n\n  static of({ role, text, meta }: MessageInput): Message {\n    if (role === \"user\") {\n      return new UserMessage(text, meta);\n    } else if (role === \"assistant\") {\n      return new AssistantMessage(text, meta);\n    } else if (role === \"system\") {\n      return new SystemMessage(text, meta);\n    } else if (role === \"tool\") {\n      return new ToolMessage(text, meta);\n    } else {\n      return new CustomMessage(role, text, meta);\n    }\n  }\n\n  get text() {\n    return this.getTexts()\n      .map((c) => c.text)\n      .join(\"\");\n  }\n\n  getTexts() {\n    return this.content.filter(isText) as TextPart[];\n  }\n\n  createSnapshot() {\n    return { content: shallowCopy(this.content), meta: shallowCopy(this.meta), role: this.role };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n\n  toPlain() {\n    return { role: this.role, content: shallowCopy(this.content) } as const;\n  }\n\n  merge(other: Message<T, R>) {\n    Object.assign(this, other.meta);\n    this.content.push(...other.content);\n  }\n\n  static fromChunks<M2 extends Message>(this: new (...args: any[]) => M2, chunks: M2[]): M2 {\n    const instance = new this([]);\n    chunks.forEach((chunk) => instance.merge(chunk));\n    return instance;\n  }\n\n  [Symbol.iterator](): Iterator<T> {\n    return this.content[Symbol.iterator]();\n  }\n}\n\nexport class AssistantMessage extends Message<TextPart | ToolCallPart | FilePart> {\n  public readonly role = \"assistant\";\n\n  static {\n    this.register();\n  }\n\n  getToolCalls() {\n    return this.content.filter(isToolCall);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n\n  protected assertContent(\n    content: TextPart | ToolCallPart | FilePart,\n  ): asserts content is TextPart | ToolCallPart | FilePart {\n    if (content.type === \"tool-call\") {\n      const key = \"args\";\n      const args = popProp(content as any, key);\n\n      if (args !== null && !hasProp(content, \"input\")) {\n        Logger.root.warn(\n          `The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`,\n        );\n        content.input = args;\n      }\n\n      safeDefineProperty(content, key as keyof typeof content, () => {\n        Logger.root.warn(\n          `The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`,\n        );\n        return content.input;\n      });\n    }\n  }\n}\n\nexport class ToolMessage extends Message<ToolResultPart> {\n  public readonly role = \"tool\";\n\n  static {\n    this.register();\n  }\n\n  getToolResults() {\n    return this.content.filter(isToolResult);\n  }\n\n  protected fromString(text: string): ToolResultPart {\n    const schema = z\n      .object({\n        type: z.literal(\"tool-result\"),\n        toolName: z.string(),\n        toolCallId: z.string(),\n      })\n      .and(z.object({ output: z.any() }).or(z.object({ result: z.any() })));\n\n    const { success, data } = schema.safeParse(text);\n    if (!success) {\n      throw new ValueError(`ToolMessage cannot be created from '${text}'!`);\n    }\n\n    return data as ToolResultPart;\n  }\n\n  protected assertContent(content: ToolResultPart): asserts content is ToolResultPart {\n    const result = popProp(content as any, \"result\", null);\n    const isError = popProp(content as any, \"isError\", false);\n\n    if (result !== null && !hasProp(content, \"output\")) {\n      Logger.root.warn(\n        \"The 'result' property in the ToolMessage class is deprecated and will be removed. Use the 'output' property for content chunks instead.\",\n      );\n      if (isString(result)) {\n        content.output = { type: isError ? \"error-text\" : \"text\", value: result };\n      } else {\n        content.output = { type: isError ? \"error-json\" : \"json\", value: result };\n      }\n    }\n\n    safeDefineProperty(content, \"result\" as any, () => {\n      Logger.root.warn(\n        \"The 'result' property of Tool Message is deprecated and will be removed. Use the 'output.value' property instead.\",\n      );\n      return content.output.value as any;\n    });\n    safeDefineProperty(content, \"isError\" as any, () => {\n      Logger.root.warn(\n        \"The 'isError' property of Tool Message is deprecated and will be removed. Use the 'output.type' property instead.\",\n      );\n      return content.output.type.includes(\"error\");\n    });\n  }\n}\n\nexport class SystemMessage extends Message<TextPart> {\n  public readonly role: MessageRole = \"system\";\n\n  static {\n    this.register();\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class UserMessage extends Message<TextPart | ImagePart | FilePart> {\n  public readonly role = \"user\";\n\n  static {\n    this.register();\n  }\n\n  getImages() {\n    return this.content.filter(isImage);\n  }\n\n  getFiles() {\n    return this.content.filter(isFile);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport const Role = {\n  ASSISTANT: \"assistant\",\n  SYSTEM: \"system\",\n  USER: \"user\",\n} as const;\n\nexport class CustomMessage extends Message<MessageContentPart, string> {\n  public role: string;\n\n  constructor(role: string, content: MessageContentPart | string, meta: MessageMeta = {}) {\n    super(content, meta);\n    if (!role) {\n      throw new ValueError(`Role \"${role}\" must be specified!`);\n    }\n    this.role = role;\n  }\n\n  protected fromString(input: string): MessageContentPart {\n    return { type: \"text\", text: input };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/backend/message.ts"],"names":["isText","content","type","isImage","isFile","isToolCall","isToolResult","Message","Serializable","id","meta","undefined","createdAt","Date","fromString","Array","isArray","chunk","assertContent","watchArray","onAdd","of","role","text","UserMessage","AssistantMessage","SystemMessage","ToolMessage","CustomMessage","getTexts","map","c","join","filter","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign","toPlain","merge","other","push","fromChunks","chunks","instance","forEach","Symbol","iterator","register","getToolCalls","key","getOwnPropertyDescriptor","args","popProp","hasProp","Logger","root","warn","input","safeDefineProperty","getToolResults","schema","z","object","literal","toolName","string","toolCallId","and","output","any","or","result","success","data","safeParse","ValueError","isError","isString","value","includes","getImages","getFiles","Role","ASSISTANT","SYSTEM","USER"],"mappings":";;;;;;;;;;;;;AA8BA,SAASA,OAAOC,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSF,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASG,QAAQF,OAAAA,EAA2B;AAC1C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,OAAA;AAC1B;AAFSC,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAGT,SAASC,OAAOH,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSE,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASC,WAAWJ,OAAAA,EAA2B;AAC7C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,WAAA;AAC1B;AAFSG,MAAAA,CAAAA,UAAAA,EAAAA,YAAAA,CAAAA;AAGT,SAASC,aAAaL,OAAAA,EAA2B;AAC/C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,aAAA;AAC1B;AAFSI,MAAAA,CAAAA,YAAAA,EAAAA,cAAAA,CAAAA;AAIF,MAAeC,gBAGZC,6BAAAA,CAAAA;EAjDV;;;;AAkDSC,EAAAA,EAAAA;AAESR,EAAAA,OAAAA;AAEhB,EAAA,WAAA,CACEA,OAAAA,EACgBS,IAAAA,GAAoB,EAAC,EACrCD,KAAyBE,MAAAA,EACzB;AACA,IAAA,KAAA,EAAK,EAAA,KAHWD,IAAAA,GAAAA,IAAAA;AAIhB,IAAA,IAAA,CAAKD,EAAAA,GAAKA,EAAAA;AACV,IAAA,IAAI,CAACC,MAAME,SAAAA,EAAW;AACpBF,MAAAA,IAAAA,CAAKE,SAAAA,uBAAgBC,IAAAA,EAAAA;AACvB;AACA,IAAA,IAAI,OAAOZ,YAAY,QAAA,EAAU;AAC/B,MAAA,IAAA,CAAKA,OAAAA,GAAU;AAAC,QAAA,IAAA,CAAKa,WAAWb,OAAAA;;KAClC,MAAO;AACL,MAAA,IAAA,CAAKA,OAAAA,GAAUc,KAAAA,CAAMC,OAAAA,CAAQf,OAAAA,IAAWA,OAAAA,GAAU;AAACA,QAAAA;;AACrD;AACA,IAAA,KAAA,MAAWgB,KAAAA,IAAS,KAAKhB,OAAAA,EAAS;AAChC,MAAA,IAAA,CAAKiB,cAAcD,KAAAA,CAAAA;AACrB;AACA,IAAA,IAAA,CAAKhB,OAAAA,GAAUkB,oBAAAA,CAAW,IAAA,CAAKlB,OAAAA,EAAS;AACtCmB,MAAAA,KAAAA,EAAO,IAAA,CAAKF;KACd,CAAA;AACF;AAEUA,EAAAA,aAAAA,CAAcjB,OAAAA,EAAkC;AAAC;AAI3D,EAAA,OAAOoB,GAAG,EAAEC,IAAAA,EAAMC,IAAAA,EAAMb,IAAAA,EAAMD,IAAE,EAA2B;AACzD,IAAA,IAAIa,SAAS,MAAA,EAAQ;AACnB,MAAA,OAAO,IAAIE,WAAAA,CAAYD,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;AACrC,KAAA,MAAA,IAAWa,SAAS,WAAA,EAAa;AAC/B,MAAA,OAAO,IAAIG,gBAAAA,CAAiBF,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;AAC1C,KAAA,MAAA,IAAWa,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,IAAII,aAAAA,CAAcH,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;AACvC,KAAA,MAAA,IAAWa,SAAS,MAAA,EAAQ;AAC1B,MAAA,OAAO,IAAIK,WAAAA,CAAYJ,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;KACrC,MAAO;AACL,MAAA,OAAO,IAAImB,aAAAA,CAAcN,IAAAA,EAAMC,IAAAA,EAAMb,MAAMD,EAAAA,CAAAA;AAC7C;AACF;AAEA,EAAA,IAAIc,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKM,QAAAA,EAAQ,CACjBC,GAAAA,CAAI,CAACC,MAAMA,CAAAA,CAAER,IAAI,CAAA,CACjBS,IAAAA,CAAK,EAAA,CAAA;AACV;EAEAH,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK5B,OAAAA,CAAQgC,MAAAA,CAAOjC,MAAAA,CAAAA;AAC7B;EAEAkC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLzB,MAAAA,EAAAA,EAAI,IAAA,CAAKA,EAAAA;MACTR,OAAAA,EAASkC,qBAAAA,CAAY,KAAKlC,OAAO,CAAA;MACjCS,IAAAA,EAAMyB,qBAAAA,CAAY,KAAKzB,IAAI,CAAA;AAC3BY,MAAAA,IAAAA,EAAM,IAAA,CAAKA;AACb,KAAA;AACF;AAEAc,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;EAEAG,OAAAA,GAAU;AACR,IAAA,OAAO;AAAE/B,MAAAA,EAAAA,EAAI,IAAA,CAAKA,EAAAA;AAAIa,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMrB,OAAAA,EAASkC,qBAAAA,CAAY,KAAKlC,OAAO;AAAE,KAAA;AAC5E;AAEAwC,EAAAA,KAAAA,CAAMC,KAAAA,EAAsB;AAC1B,IAAA,IAAA,CAAKjC,EAAAA,GAAK,IAAA,CAAKA,EAAAA,IAAMiC,KAAAA,CAAMjC,EAAAA;AAC3B6B,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,EAAMG,KAAAA,CAAMhC,IAAI,CAAA;AAC9B,IAAA,IAAA,CAAKT,OAAAA,CAAQ0C,IAAAA,CAAI,GAAID,KAAAA,CAAMzC,OAAO,CAAA;AACpC;AAEA,EAAA,OAAO2C,WAAiEC,MAAAA,EAAkB;AACxF,IAAA,MAAMC,QAAAA,GAAW,IAAI,IAAA,CAAK,EAAE,CAAA;AAC5BD,IAAAA,MAAAA,CAAOE,QAAQ,CAAC9B,KAAAA,KAAU6B,QAAAA,CAASL,KAAAA,CAAMxB,KAAAA,CAAAA,CAAAA;AACzC,IAAA,OAAO6B,QAAAA;AACT;EAEA,CAACE,MAAAA,CAAOC,QAAQ,CAAA,GAAiB;AAC/B,IAAA,OAAO,IAAA,CAAKhD,OAAAA,CAAQ+C,MAAAA,CAAOC,QAAQ,CAAA,EAAC;AACtC;AACF;AAEO,MAAMxB,yBAAyBlB,OAAAA,CAAAA;EA3ItC;;;EA4IkBe,IAAAA,GAAO,WAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAC,YAAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAKlD,OAAAA,CAAQgC,MAAAA,CAAO5B,UAAAA,CAAAA;AAC7B;AAEUS,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAErB,IAAAA,EAAM,MAAA;AAAQqB,MAAAA;AAAK,KAAA;AAC9B;AAEUL,EAAAA,aAAAA,CACRjB,OAAAA,EACuD;AACvD,IAAA,IAAIA,OAAAA,CAAQC,SAAS,WAAA,EAAa;AAChC,MAAA,MAAMkD,GAAAA,GAAM,MAAA;AACZ,MAAA,IAAI,CAACd,MAAAA,CAAOe,wBAAAA,CAAyBpD,OAAAA,EAASmD,GAAAA,CAAAA,EAAM;AAClD,QAAA,MAAME,IAAAA,GAAOC,kBAAAA,CAAQtD,OAAAA,EAAgBmD,GAAAA,CAAAA;AAErC,QAAA,IAAIE,SAAS,IAAA,IAAQ,CAACE,kBAAAA,CAAQvD,OAAAA,EAAS,OAAA,CAAA,EAAU;AAC/CwD,UAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQP,GAAAA,CAAAA,0IAAAA,CAA+I,CAAA;AAEzJnD,UAAAA,OAAAA,CAAQ2D,KAAAA,GAAQN,IAAAA;AAClB;AACF;AAEAO,MAAAA,6BAAAA,CAAmB5D,OAAAA,EAASmD,KAA6B,MAAA;AACvDK,QAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQP,GAAAA,CAAAA,gGAAAA,CAAqG,CAAA;AAE/G,QAAA,OAAOnD,OAAAA,CAAQ2D,KAAAA;OACjB,CAAA;AACF;AACF;AACF;AAEO,MAAMjC,oBAAoBpB,OAAAA,CAAAA;EApLjC;;;EAqLkBe,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAY,cAAAA,GAAiB;AACf,IAAA,OAAO,IAAA,CAAK7D,OAAAA,CAAQgC,MAAAA,CAAO3B,YAAAA,CAAAA;AAC7B;AAEUQ,EAAAA,UAAAA,CAAWS,IAAAA,EAA8B;AACjD,IAAA,MAAMwC,MAAAA,GAASC,MACZC,MAAAA,CAAO;MACN/D,IAAAA,EAAM8D,KAAAA,CAAEE,QAAQ,aAAA,CAAA;AAChBC,MAAAA,QAAAA,EAAUH,MAAEI,MAAAA,EAAM;AAClBC,MAAAA,UAAAA,EAAYL,MAAEI,MAAAA;KAChB,CAAA,CACCE,GAAAA,CAAIN,KAAAA,CAAEC,MAAAA,CAAO;AAAEM,MAAAA,MAAAA,EAAQP,MAAEQ,GAAAA;KAAM,CAAA,CAAGC,EAAAA,CAAGT,KAAAA,CAAEC,MAAAA,CAAO;AAAES,MAAAA,MAAAA,EAAQV,MAAEQ,GAAAA;AAAM,KAAA,CAAA,CAAA,CAAA;AAEnE,IAAA,MAAM,EAAEG,OAAAA,EAASC,IAAAA,EAAI,GAAKb,MAAAA,CAAOc,UAAUtD,IAAAA,CAAAA;AAC3C,IAAA,IAAI,CAACoD,OAAAA,EAAS;AACZ,MAAA,MAAM,IAAIG,qBAAAA,CAAW,CAAA,oCAAA,EAAuCvD,IAAAA,CAAAA,EAAAA,CAAQ,CAAA;AACtE;AAEA,IAAA,OAAOqD,IAAAA;AACT;AAEU1D,EAAAA,aAAAA,CAAcjB,OAAAA,EAA4D;AAClF,IAAA,MAAMyE,MAAAA,GAASnB,kBAAAA,CAAQtD,OAAAA,EAAgB,QAAA,EAAU,IAAA,CAAA;AACjD,IAAA,MAAM8E,OAAAA,GAAUxB,kBAAAA,CAAQtD,OAAAA,EAAgB,SAAA,EAAW,KAAA,CAAA;AAEnD,IAAA,IAAIyE,WAAW,IAAA,IAAQ,CAAClB,kBAAAA,CAAQvD,OAAAA,EAAS,QAAA,CAAA,EAAW;AAClDwD,MAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,KACV,yIAAA,CAAA;AAEF,MAAA,IAAIqB,eAAAA,CAASN,MAAAA,CAAAA,EAAS;AACpBzE,QAAAA,OAAAA,CAAQsE,MAAAA,GAAS;AAAErE,UAAAA,IAAAA,EAAM6E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;OAC1E,MAAO;AACLzE,QAAAA,OAAAA,CAAQsE,MAAAA,GAAS;AAAErE,UAAAA,IAAAA,EAAM6E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;AAC1E;AACF;AAEAb,IAAAA,6BAAAA,CAAmB5D,OAAAA,EAAS,UAAiB,MAAA;AAC3CwD,MAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAO1D,QAAQsE,MAAAA,CAAOU,KAAAA;KACxB,CAAA;AACApB,IAAAA,6BAAAA,CAAmB5D,OAAAA,EAAS,WAAkB,MAAA;AAC5CwD,MAAAA,iBAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAO1D,OAAAA,CAAQsE,MAAAA,CAAOrE,IAAAA,CAAKgF,QAAAA,CAAS,OAAA,CAAA;KACtC,CAAA;AACF;AACF;AAEO,MAAMxD,sBAAsBnB,OAAAA,CAAAA;EA9OnC;;;EA+OkBe,IAAAA,GAAoB,QAAA;EAEpC;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;AAEUpC,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAErB,IAAAA,EAAM,MAAA;AAAQqB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMC,oBAAoBjB,OAAAA,CAAAA;EA1PjC;;;EA2PkBe,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAiC,SAAAA,GAAY;AACV,IAAA,OAAO,IAAA,CAAKlF,OAAAA,CAAQgC,MAAAA,CAAO9B,OAAAA,CAAAA;AAC7B;EAEAiF,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAKnF,OAAAA,CAAQgC,MAAAA,CAAO7B,MAAAA,CAAAA;AAC7B;AAEUU,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAErB,IAAAA,EAAM,MAAA;AAAQqB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAM8D,IAAAA,GAAO;EAClBC,SAAAA,EAAW,WAAA;EACXC,MAAAA,EAAQ,QAAA;EACRC,IAAAA,EAAM;AACR;AAEO,MAAM5D,sBAAsBrB,OAAAA,CAAAA;EApRnC;;;AAqRSe,EAAAA,IAAAA;AAEP,EAAA,WAAA,CACEA,IAAAA,EACArB,OAAAA,EACAS,IAAAA,GAAoB,IACpBD,EAAAA,EACA;AACA,IAAA,KAAA,CAAMR,OAAAA,EAASS,MAAMD,EAAAA,CAAAA;AACrB,IAAA,IAAI,CAACa,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIwD,qBAAAA,CAAW,CAAA,MAAA,EAASxD,IAAAA,CAAAA,oBAAAA,CAA0B,CAAA;AAC1D;AACA,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA;AACd;AAEUR,EAAAA,UAAAA,CAAW8C,KAAAA,EAAmC;AACtD,IAAA,OAAO;MAAE1D,IAAAA,EAAM,MAAA;MAAQqB,IAAAA,EAAMqC;AAAM,KAAA;AACrC;AACF","file":"message.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { FilePart, ImagePart, TextPart, ToolCallPart, ToolResultPart } from \"ai\";\nimport { z } from \"zod\";\nimport { ValueError } from \"@/errors.js\";\nimport { hasProp, popProp, safeDefineProperty } from \"@/internals/helpers/object.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { isString } from \"remeda\";\nimport { watchArray } from \"@/internals/helpers/array.js\";\n\nexport type MessageRole = \"user\" | \"system\" | \"tool\" | \"assistant\";\nexport type MessageContentPart = TextPart | ToolCallPart | ImagePart | FilePart | ToolResultPart;\n\nexport interface MessageMeta {\n  [key: string]: any;\n  createdAt?: Date;\n}\n\nexport interface MessageInput {\n  id?: string;\n  role: MessageRole;\n  text: string; // TODO\n  meta?: MessageMeta;\n}\n\nfunction isText(content: MessageContentPart): content is TextPart {\n  return content.type === \"text\";\n}\nfunction isImage(content: MessageContentPart): content is ImagePart {\n  return content.type === \"image\";\n}\nfunction isFile(content: MessageContentPart): content is FilePart {\n  return content.type === \"file\";\n}\nfunction isToolCall(content: MessageContentPart): content is ToolCallPart {\n  return content.type === \"tool-call\";\n}\nfunction isToolResult(content: MessageContentPart): content is ToolResultPart {\n  return content.type === \"tool-result\";\n}\n\nexport abstract class Message<\n  T extends MessageContentPart = MessageContentPart,\n  R extends string = MessageRole | string,\n> extends Serializable {\n  public id: string | undefined;\n  public abstract readonly role: R;\n  public readonly content: T[];\n\n  constructor(\n    content: T | T[] | string,\n    public readonly meta: MessageMeta = {},\n    id: string | undefined = undefined,\n  ) {\n    super();\n    this.id = id;\n    if (!meta?.createdAt) {\n      meta.createdAt = new Date();\n    }\n    if (typeof content === \"string\") {\n      this.content = [this.fromString(content)];\n    } else {\n      this.content = Array.isArray(content) ? content : [content];\n    }\n    for (const chunk of this.content) {\n      this.assertContent(chunk);\n    }\n    this.content = watchArray(this.content, {\n      onAdd: this.assertContent,\n    });\n  }\n\n  protected assertContent(content: T): asserts content is T {}\n\n  protected abstract fromString(input: string): T;\n\n  static of({ role, text, meta, id }: MessageInput): Message {\n    if (role === \"user\") {\n      return new UserMessage(text, meta, id);\n    } else if (role === \"assistant\") {\n      return new AssistantMessage(text, meta, id);\n    } else if (role === \"system\") {\n      return new SystemMessage(text, meta, id);\n    } else if (role === \"tool\") {\n      return new ToolMessage(text, meta, id);\n    } else {\n      return new CustomMessage(role, text, meta, id);\n    }\n  }\n\n  get text() {\n    return this.getTexts()\n      .map((c) => c.text)\n      .join(\"\");\n  }\n\n  getTexts() {\n    return this.content.filter(isText) as TextPart[];\n  }\n\n  createSnapshot() {\n    return {\n      id: this.id,\n      content: shallowCopy(this.content),\n      meta: shallowCopy(this.meta),\n      role: this.role,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n\n  toPlain() {\n    return { id: this.id, role: this.role, content: shallowCopy(this.content) } as const;\n  }\n\n  merge(other: Message<T, R>) {\n    this.id = this.id || other.id;\n    Object.assign(this, other.meta);\n    this.content.push(...other.content);\n  }\n\n  static fromChunks<M2 extends Message>(this: new (...args: any[]) => M2, chunks: M2[]): M2 {\n    const instance = new this([]);\n    chunks.forEach((chunk) => instance.merge(chunk));\n    return instance;\n  }\n\n  [Symbol.iterator](): Iterator<T> {\n    return this.content[Symbol.iterator]();\n  }\n}\n\nexport class AssistantMessage extends Message<TextPart | ToolCallPart | FilePart> {\n  public readonly role = \"assistant\";\n\n  static {\n    this.register();\n  }\n\n  getToolCalls() {\n    return this.content.filter(isToolCall);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n\n  protected assertContent(\n    content: TextPart | ToolCallPart | FilePart,\n  ): asserts content is TextPart | ToolCallPart | FilePart {\n    if (content.type === \"tool-call\") {\n      const key = \"args\";\n      if (!Object.getOwnPropertyDescriptor(content, key)) {\n        const args = popProp(content as any, key);\n\n        if (args !== null && !hasProp(content, \"input\")) {\n          Logger.root.warn(\n            `The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`,\n          );\n          content.input = args;\n        }\n      }\n\n      safeDefineProperty(content, key as keyof typeof content, () => {\n        Logger.root.warn(\n          `The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`,\n        );\n        return content.input;\n      });\n    }\n  }\n}\n\nexport class ToolMessage extends Message<ToolResultPart> {\n  public readonly role = \"tool\";\n\n  static {\n    this.register();\n  }\n\n  getToolResults() {\n    return this.content.filter(isToolResult);\n  }\n\n  protected fromString(text: string): ToolResultPart {\n    const schema = z\n      .object({\n        type: z.literal(\"tool-result\"),\n        toolName: z.string(),\n        toolCallId: z.string(),\n      })\n      .and(z.object({ output: z.any() }).or(z.object({ result: z.any() })));\n\n    const { success, data } = schema.safeParse(text);\n    if (!success) {\n      throw new ValueError(`ToolMessage cannot be created from '${text}'!`);\n    }\n\n    return data as ToolResultPart;\n  }\n\n  protected assertContent(content: ToolResultPart): asserts content is ToolResultPart {\n    const result = popProp(content as any, \"result\", null);\n    const isError = popProp(content as any, \"isError\", false);\n\n    if (result !== null && !hasProp(content, \"output\")) {\n      Logger.root.warn(\n        \"The 'result' property in the ToolMessage class is deprecated and will be removed. Use the 'output' property for content chunks instead.\",\n      );\n      if (isString(result)) {\n        content.output = { type: isError ? \"error-text\" : \"text\", value: result };\n      } else {\n        content.output = { type: isError ? \"error-json\" : \"json\", value: result };\n      }\n    }\n\n    safeDefineProperty(content, \"result\" as any, () => {\n      Logger.root.warn(\n        \"The 'result' property of Tool Message is deprecated and will be removed. Use the 'output.value' property instead.\",\n      );\n      return content.output.value as any;\n    });\n    safeDefineProperty(content, \"isError\" as any, () => {\n      Logger.root.warn(\n        \"The 'isError' property of Tool Message is deprecated and will be removed. Use the 'output.type' property instead.\",\n      );\n      return content.output.type.includes(\"error\");\n    });\n  }\n}\n\nexport class SystemMessage extends Message<TextPart> {\n  public readonly role: MessageRole = \"system\";\n\n  static {\n    this.register();\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class UserMessage extends Message<TextPart | ImagePart | FilePart> {\n  public readonly role = \"user\";\n\n  static {\n    this.register();\n  }\n\n  getImages() {\n    return this.content.filter(isImage);\n  }\n\n  getFiles() {\n    return this.content.filter(isFile);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport const Role = {\n  ASSISTANT: \"assistant\",\n  SYSTEM: \"system\",\n  USER: \"user\",\n} as const;\n\nexport class CustomMessage extends Message<MessageContentPart, string> {\n  public role: string;\n\n  constructor(\n    role: string,\n    content: MessageContentPart | string,\n    meta: MessageMeta = {},\n    id?: string,\n  ) {\n    super(content, meta, id);\n    if (!role) {\n      throw new ValueError(`Role \"${role}\" must be specified!`);\n    }\n    this.role = role;\n  }\n\n  protected fromString(input: string): MessageContentPart {\n    return { type: \"text\", text: input };\n  }\n}\n"]}

package/dist/backend/message.d.cts

@@ -15,27 +15,31 @@ interface MessageMeta {
     createdAt?: Date;
 }
 interface MessageInput {
+    id?: string;
     role: MessageRole;
     text: string;
     meta?: MessageMeta;
 }
 declare abstract class Message<T extends MessageContentPart = MessageContentPart, R extends string = MessageRole | string> extends Serializable {
     readonly meta: MessageMeta;
+    id: string | undefined;
     abstract readonly role: R;
     readonly content: T[];
-    constructor(content: T | T[] | string, meta?: MessageMeta);
+    constructor(content: T | T[] | string, meta?: MessageMeta, id?: string | undefined);
     protected assertContent(content: T): asserts content is T;
     protected abstract fromString(input: string): T;
-    static of({ role, text, meta }: MessageInput): Message;
+    static of({ role, text, meta, id }: MessageInput): Message;
     get text(): string;
     getTexts(): TextPart[];
     createSnapshot(): {
+        id: string | undefined;
         content: T[];
         meta: MessageMeta;
         role: R;
     };
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
     toPlain(): {
+        readonly id: string | undefined;
         readonly role: R;
         readonly content: T[];
     };
@@ -72,7 +76,7 @@ declare const Role: {
 };
 declare class CustomMessage extends Message<MessageContentPart, string> {
     role: string;
-    constructor(role: string, content: MessageContentPart | string, meta?: MessageMeta);
+    constructor(role: string, content: MessageContentPart | string, meta?: MessageMeta, id?: string);
     protected fromString(input: string): MessageContentPart;
 }
 

package/dist/backend/message.d.ts

@@ -15,27 +15,31 @@ interface MessageMeta {
     createdAt?: Date;
 }
 interface MessageInput {
+    id?: string;
     role: MessageRole;
     text: string;
     meta?: MessageMeta;
 }
 declare abstract class Message<T extends MessageContentPart = MessageContentPart, R extends string = MessageRole | string> extends Serializable {
     readonly meta: MessageMeta;
+    id: string | undefined;
     abstract readonly role: R;
     readonly content: T[];
-    constructor(content: T | T[] | string, meta?: MessageMeta);
+    constructor(content: T | T[] | string, meta?: MessageMeta, id?: string | undefined);
     protected assertContent(content: T): asserts content is T;
     protected abstract fromString(input: string): T;
-    static of({ role, text, meta }: MessageInput): Message;
+    static of({ role, text, meta, id }: MessageInput): Message;
     get text(): string;
     getTexts(): TextPart[];
     createSnapshot(): {
+        id: string | undefined;
         content: T[];
         meta: MessageMeta;
         role: R;
     };
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
     toPlain(): {
+        readonly id: string | undefined;
         readonly role: R;
         readonly content: T[];
     };
@@ -72,7 +76,7 @@ declare const Role: {
 };
 declare class CustomMessage extends Message<MessageContentPart, string> {
     role: string;
-    constructor(role: string, content: MessageContentPart | string, meta?: MessageMeta);
+    constructor(role: string, content: MessageContentPart | string, meta?: MessageMeta, id?: string);
     protected fromString(input: string): MessageContentPart;
 }
 

package/dist/backend/message.js

@@ -34,9 +34,11 @@ class Message extends Serializable {
     __name(this, "Message");
   }
   meta;
+  id;
   content;
-  constructor(content, meta = {}) {
+  constructor(content, meta = {}, id = void 0) {
     super(), this.meta = meta;
+    this.id = id;
     if (!meta?.createdAt) {
       meta.createdAt = /* @__PURE__ */ new Date();
     }
@@ -58,17 +60,17 @@ class Message extends Serializable {
   }
   assertContent(content) {
   }
-  static of({ role, text, meta }) {
+  static of({ role, text, meta, id }) {
     if (role === "user") {
-      return new UserMessage(text, meta);
+      return new UserMessage(text, meta, id);
     } else if (role === "assistant") {
-      return new AssistantMessage(text, meta);
+      return new AssistantMessage(text, meta, id);
     } else if (role === "system") {
-      return new SystemMessage(text, meta);
+      return new SystemMessage(text, meta, id);
     } else if (role === "tool") {
-      return new ToolMessage(text, meta);
+      return new ToolMessage(text, meta, id);
     } else {
-      return new CustomMessage(role, text, meta);
+      return new CustomMessage(role, text, meta, id);
     }
   }
   get text() {
@@ -79,6 +81,7 @@ class Message extends Serializable {
   }
   createSnapshot() {
     return {
+      id: this.id,
       content: shallowCopy(this.content),
       meta: shallowCopy(this.meta),
       role: this.role
@@ -89,11 +92,13 @@ class Message extends Serializable {
   }
   toPlain() {
     return {
+      id: this.id,
       role: this.role,
       content: shallowCopy(this.content)
     };
   }
   merge(other) {
+    this.id = this.id || other.id;
     Object.assign(this, other.meta);
     this.content.push(...other.content);
   }
@@ -126,10 +131,12 @@ class AssistantMessage extends Message {
   assertContent(content) {
     if (content.type === "tool-call") {
       const key = "args";
-      const args = popProp(content, key);
-      if (args !== null && !hasProp(content, "input")) {
-        Logger.root.warn(`The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`);
-        content.input = args;
+      if (!Object.getOwnPropertyDescriptor(content, key)) {
+        const args = popProp(content, key);
+        if (args !== null && !hasProp(content, "input")) {
+          Logger.root.warn(`The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`);
+          content.input = args;
+        }
       }
       safeDefineProperty(content, key, () => {
         Logger.root.warn(`The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`);
@@ -238,8 +245,8 @@ class CustomMessage extends Message {
     __name(this, "CustomMessage");
   }
   role;
-  constructor(role, content, meta = {}) {
-    super(content, meta);
+  constructor(role, content, meta = {}, id) {
+    super(content, meta, id);
     if (!role) {
       throw new ValueError(`Role "${role}" must be specified!`);
     }

package/dist/backend/message.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/backend/message.ts"],"names":["isText","content","type","isImage","isFile","isToolCall","isToolResult","Message","Serializable","meta","createdAt","Date","fromString","Array","isArray","chunk","assertContent","watchArray","onAdd","of","role","text","UserMessage","AssistantMessage","SystemMessage","ToolMessage","CustomMessage","getTexts","map","c","join","filter","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign","toPlain","merge","other","push","fromChunks","chunks","instance","forEach","Symbol","iterator","register","getToolCalls","key","args","popProp","hasProp","Logger","root","warn","input","safeDefineProperty","getToolResults","schema","z","object","literal","toolName","string","toolCallId","and","output","any","or","result","success","data","safeParse","ValueError","isError","isString","value","includes","getImages","getFiles","Role","ASSISTANT","SYSTEM","USER"],"mappings":";;;;;;;;;;;AA6BA,SAASA,OAAOC,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSF,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASG,QAAQF,OAAAA,EAA2B;AAC1C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,OAAA;AAC1B;AAFSC,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAGT,SAASC,OAAOH,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSE,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASC,WAAWJ,OAAAA,EAA2B;AAC7C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,WAAA;AAC1B;AAFSG,MAAAA,CAAAA,UAAAA,EAAAA,YAAAA,CAAAA;AAGT,SAASC,aAAaL,OAAAA,EAA2B;AAC/C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,aAAA;AAC1B;AAFSI,MAAAA,CAAAA,YAAAA,EAAAA,cAAAA,CAAAA;AAIF,MAAeC,gBAGZC,YAAAA,CAAAA;EAhDV;;;;AAkDkBP,EAAAA,OAAAA;EAEhB,WAAA,CACEA,OAAAA,EACgBQ,IAAAA,GAAoB,EAAC,EACrC;AACA,IAAA,KAAA,EAAK,EAAA,KAFWA,IAAAA,GAAAA,IAAAA;AAGhB,IAAA,IAAI,CAACA,MAAMC,SAAAA,EAAW;AACpBD,MAAAA,IAAAA,CAAKC,SAAAA,uBAAgBC,IAAAA,EAAAA;AACvB;AACA,IAAA,IAAI,OAAOV,YAAY,QAAA,EAAU;AAC/B,MAAA,IAAA,CAAKA,OAAAA,GAAU;AAAC,QAAA,IAAA,CAAKW,WAAWX,OAAAA;;KAClC,MAAO;AACL,MAAA,IAAA,CAAKA,OAAAA,GAAUY,KAAAA,CAAMC,OAAAA,CAAQb,OAAAA,IAAWA,OAAAA,GAAU;AAACA,QAAAA;;AACrD;AACA,IAAA,KAAA,MAAWc,KAAAA,IAAS,KAAKd,OAAAA,EAAS;AAChC,MAAA,IAAA,CAAKe,cAAcD,KAAAA,CAAAA;AACrB;AACA,IAAA,IAAA,CAAKd,OAAAA,GAAUgB,UAAAA,CAAW,IAAA,CAAKhB,OAAAA,EAAS;AACtCiB,MAAAA,KAAAA,EAAO,IAAA,CAAKF;KACd,CAAA;AACF;AAEUA,EAAAA,aAAAA,CAAcf,OAAAA,EAAkC;AAAC;AAI3D,EAAA,OAAOkB,EAAAA,CAAG,EAAEC,IAAAA,EAAMC,IAAAA,EAAMZ,MAAI,EAA2B;AACrD,IAAA,IAAIW,SAAS,MAAA,EAAQ;AACnB,MAAA,OAAO,IAAIE,WAAAA,CAAYD,IAAAA,EAAMZ,IAAAA,CAAAA;AAC/B,KAAA,MAAA,IAAWW,SAAS,WAAA,EAAa;AAC/B,MAAA,OAAO,IAAIG,gBAAAA,CAAiBF,IAAAA,EAAMZ,IAAAA,CAAAA;AACpC,KAAA,MAAA,IAAWW,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,IAAII,aAAAA,CAAcH,IAAAA,EAAMZ,IAAAA,CAAAA;AACjC,KAAA,MAAA,IAAWW,SAAS,MAAA,EAAQ;AAC1B,MAAA,OAAO,IAAIK,WAAAA,CAAYJ,IAAAA,EAAMZ,IAAAA,CAAAA;KAC/B,MAAO;AACL,MAAA,OAAO,IAAIiB,aAAAA,CAAcN,IAAAA,EAAMC,IAAAA,EAAMZ,IAAAA,CAAAA;AACvC;AACF;AAEA,EAAA,IAAIY,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKM,QAAAA,EAAQ,CACjBC,GAAAA,CAAI,CAACC,MAAMA,CAAAA,CAAER,IAAI,CAAA,CACjBS,IAAAA,CAAK,EAAA,CAAA;AACV;EAEAH,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK1B,OAAAA,CAAQ8B,MAAAA,CAAO/B,MAAAA,CAAAA;AAC7B;EAEAgC,cAAAA,GAAiB;AACf,IAAA,OAAO;MAAE/B,OAAAA,EAASgC,WAAAA,CAAY,KAAKhC,OAAO,CAAA;MAAGQ,IAAAA,EAAMwB,WAAAA,CAAY,KAAKxB,IAAI,CAAA;AAAGW,MAAAA,IAAAA,EAAM,IAAA,CAAKA;AAAK,KAAA;AAC7F;AAEAc,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;EAEAG,OAAAA,GAAU;AACR,IAAA,OAAO;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMnB,OAAAA,EAASgC,WAAAA,CAAY,KAAKhC,OAAO;AAAE,KAAA;AAC/D;AAEAsC,EAAAA,KAAAA,CAAMC,KAAAA,EAAsB;AAC1BJ,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,EAAMG,KAAAA,CAAM/B,IAAI,CAAA;AAC9B,IAAA,IAAA,CAAKR,OAAAA,CAAQwC,IAAAA,CAAI,GAAID,KAAAA,CAAMvC,OAAO,CAAA;AACpC;AAEA,EAAA,OAAOyC,WAAiEC,MAAAA,EAAkB;AACxF,IAAA,MAAMC,QAAAA,GAAW,IAAI,IAAA,CAAK,EAAE,CAAA;AAC5BD,IAAAA,MAAAA,CAAOE,QAAQ,CAAC9B,KAAAA,KAAU6B,QAAAA,CAASL,KAAAA,CAAMxB,KAAAA,CAAAA,CAAAA;AACzC,IAAA,OAAO6B,QAAAA;AACT;EAEA,CAACE,MAAAA,CAAOC,QAAQ,CAAA,GAAiB;AAC/B,IAAA,OAAO,IAAA,CAAK9C,OAAAA,CAAQ6C,MAAAA,CAAOC,QAAQ,CAAA,EAAC;AACtC;AACF;AAEO,MAAMxB,yBAAyBhB,OAAAA,CAAAA;EAjItC;;;EAkIkBa,IAAAA,GAAO,WAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAC,YAAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAKhD,OAAAA,CAAQ8B,MAAAA,CAAO1B,UAAAA,CAAAA;AAC7B;AAEUO,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AAEUL,EAAAA,aAAAA,CACRf,OAAAA,EACuD;AACvD,IAAA,IAAIA,OAAAA,CAAQC,SAAS,WAAA,EAAa;AAChC,MAAA,MAAMgD,GAAAA,GAAM,MAAA;AACZ,MAAA,MAAMC,IAAAA,GAAOC,OAAAA,CAAQnD,OAAAA,EAAgBiD,GAAAA,CAAAA;AAErC,MAAA,IAAIC,SAAS,IAAA,IAAQ,CAACE,OAAAA,CAAQpD,OAAAA,EAAS,OAAA,CAAA,EAAU;AAC/CqD,QAAAA,MAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQN,GAAAA,CAAAA,0IAAAA,CAA+I,CAAA;AAEzJjD,QAAAA,OAAAA,CAAQwD,KAAAA,GAAQN,IAAAA;AAClB;AAEAO,MAAAA,kBAAAA,CAAmBzD,OAAAA,EAASiD,KAA6B,MAAA;AACvDI,QAAAA,MAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQN,GAAAA,CAAAA,gGAAAA,CAAqG,CAAA;AAE/G,QAAA,OAAOjD,OAAAA,CAAQwD,KAAAA;OACjB,CAAA;AACF;AACF;AACF;AAEO,MAAMhC,oBAAoBlB,OAAAA,CAAAA;EAxKjC;;;EAyKkBa,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAW,cAAAA,GAAiB;AACf,IAAA,OAAO,IAAA,CAAK1D,OAAAA,CAAQ8B,MAAAA,CAAOzB,YAAAA,CAAAA;AAC7B;AAEUM,EAAAA,UAAAA,CAAWS,IAAAA,EAA8B;AACjD,IAAA,MAAMuC,MAAAA,GAASC,EACZC,MAAAA,CAAO;MACN5D,IAAAA,EAAM2D,CAAAA,CAAEE,QAAQ,aAAA,CAAA;AAChBC,MAAAA,QAAAA,EAAUH,EAAEI,MAAAA,EAAM;AAClBC,MAAAA,UAAAA,EAAYL,EAAEI,MAAAA;KAChB,CAAA,CACCE,GAAAA,CAAIN,CAAAA,CAAEC,MAAAA,CAAO;AAAEM,MAAAA,MAAAA,EAAQP,EAAEQ,GAAAA;KAAM,CAAA,CAAGC,EAAAA,CAAGT,CAAAA,CAAEC,MAAAA,CAAO;AAAES,MAAAA,MAAAA,EAAQV,EAAEQ,GAAAA;AAAM,KAAA,CAAA,CAAA,CAAA;AAEnE,IAAA,MAAM,EAAEG,OAAAA,EAASC,IAAAA,EAAI,GAAKb,MAAAA,CAAOc,UAAUrD,IAAAA,CAAAA;AAC3C,IAAA,IAAI,CAACmD,OAAAA,EAAS;AACZ,MAAA,MAAM,IAAIG,UAAAA,CAAW,CAAA,oCAAA,EAAuCtD,IAAAA,CAAAA,EAAAA,CAAQ,CAAA;AACtE;AAEA,IAAA,OAAOoD,IAAAA;AACT;AAEUzD,EAAAA,aAAAA,CAAcf,OAAAA,EAA4D;AAClF,IAAA,MAAMsE,MAAAA,GAASnB,OAAAA,CAAQnD,OAAAA,EAAgB,QAAA,EAAU,IAAA,CAAA;AACjD,IAAA,MAAM2E,OAAAA,GAAUxB,OAAAA,CAAQnD,OAAAA,EAAgB,SAAA,EAAW,KAAA,CAAA;AAEnD,IAAA,IAAIsE,WAAW,IAAA,IAAQ,CAAClB,OAAAA,CAAQpD,OAAAA,EAAS,QAAA,CAAA,EAAW;AAClDqD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,yIAAA,CAAA;AAEF,MAAA,IAAIqB,QAAAA,CAASN,MAAAA,CAAAA,EAAS;AACpBtE,QAAAA,OAAAA,CAAQmE,MAAAA,GAAS;AAAElE,UAAAA,IAAAA,EAAM0E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;OAC1E,MAAO;AACLtE,QAAAA,OAAAA,CAAQmE,MAAAA,GAAS;AAAElE,UAAAA,IAAAA,EAAM0E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;AAC1E;AACF;AAEAb,IAAAA,kBAAAA,CAAmBzD,OAAAA,EAAS,UAAiB,MAAA;AAC3CqD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAOvD,QAAQmE,MAAAA,CAAOU,KAAAA;KACxB,CAAA;AACApB,IAAAA,kBAAAA,CAAmBzD,OAAAA,EAAS,WAAkB,MAAA;AAC5CqD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAOvD,OAAAA,CAAQmE,MAAAA,CAAOlE,IAAAA,CAAK6E,QAAAA,CAAS,OAAA,CAAA;KACtC,CAAA;AACF;AACF;AAEO,MAAMvD,sBAAsBjB,OAAAA,CAAAA;EAlOnC;;;EAmOkBa,IAAAA,GAAoB,QAAA;EAEpC;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;AAEUpC,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMC,oBAAoBf,OAAAA,CAAAA;EA9OjC;;;EA+OkBa,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAgC,SAAAA,GAAY;AACV,IAAA,OAAO,IAAA,CAAK/E,OAAAA,CAAQ8B,MAAAA,CAAO5B,OAAAA,CAAAA;AAC7B;EAEA8E,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAKhF,OAAAA,CAAQ8B,MAAAA,CAAO3B,MAAAA,CAAAA;AAC7B;AAEUQ,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAEnB,IAAAA,EAAM,MAAA;AAAQmB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAM6D,IAAAA,GAAO;EAClBC,SAAAA,EAAW,WAAA;EACXC,MAAAA,EAAQ,QAAA;EACRC,IAAAA,EAAM;AACR;AAEO,MAAM3D,sBAAsBnB,OAAAA,CAAAA;EAxQnC;;;AAyQSa,EAAAA,IAAAA;AAEP,EAAA,WAAA,CAAYA,IAAAA,EAAcnB,OAAAA,EAAsCQ,IAAAA,GAAoB,EAAC,EAAG;AACtF,IAAA,KAAA,CAAMR,SAASQ,IAAAA,CAAAA;AACf,IAAA,IAAI,CAACW,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIuD,UAAAA,CAAW,CAAA,MAAA,EAASvD,IAAAA,CAAAA,oBAAAA,CAA0B,CAAA;AAC1D;AACA,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA;AACd;AAEUR,EAAAA,UAAAA,CAAW6C,KAAAA,EAAmC;AACtD,IAAA,OAAO;MAAEvD,IAAAA,EAAM,MAAA;MAAQmB,IAAAA,EAAMoC;AAAM,KAAA;AACrC;AACF","file":"message.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { FilePart, ImagePart, TextPart, ToolCallPart, ToolResultPart } from \"ai\";\nimport { z } from \"zod\";\nimport { ValueError } from \"@/errors.js\";\nimport { hasProp, popProp, safeDefineProperty } from \"@/internals/helpers/object.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { isString } from \"remeda\";\nimport { watchArray } from \"@/internals/helpers/array.js\";\n\nexport type MessageRole = \"user\" | \"system\" | \"tool\" | \"assistant\";\nexport type MessageContentPart = TextPart | ToolCallPart | ImagePart | FilePart | ToolResultPart;\n\nexport interface MessageMeta {\n  [key: string]: any;\n  createdAt?: Date;\n}\n\nexport interface MessageInput {\n  role: MessageRole;\n  text: string; // TODO\n  meta?: MessageMeta;\n}\n\nfunction isText(content: MessageContentPart): content is TextPart {\n  return content.type === \"text\";\n}\nfunction isImage(content: MessageContentPart): content is ImagePart {\n  return content.type === \"image\";\n}\nfunction isFile(content: MessageContentPart): content is FilePart {\n  return content.type === \"file\";\n}\nfunction isToolCall(content: MessageContentPart): content is ToolCallPart {\n  return content.type === \"tool-call\";\n}\nfunction isToolResult(content: MessageContentPart): content is ToolResultPart {\n  return content.type === \"tool-result\";\n}\n\nexport abstract class Message<\n  T extends MessageContentPart = MessageContentPart,\n  R extends string = MessageRole | string,\n> extends Serializable {\n  public abstract readonly role: R;\n  public readonly content: T[];\n\n  constructor(\n    content: T | T[] | string,\n    public readonly meta: MessageMeta = {},\n  ) {\n    super();\n    if (!meta?.createdAt) {\n      meta.createdAt = new Date();\n    }\n    if (typeof content === \"string\") {\n      this.content = [this.fromString(content)];\n    } else {\n      this.content = Array.isArray(content) ? content : [content];\n    }\n    for (const chunk of this.content) {\n      this.assertContent(chunk);\n    }\n    this.content = watchArray(this.content, {\n      onAdd: this.assertContent,\n    });\n  }\n\n  protected assertContent(content: T): asserts content is T {}\n\n  protected abstract fromString(input: string): T;\n\n  static of({ role, text, meta }: MessageInput): Message {\n    if (role === \"user\") {\n      return new UserMessage(text, meta);\n    } else if (role === \"assistant\") {\n      return new AssistantMessage(text, meta);\n    } else if (role === \"system\") {\n      return new SystemMessage(text, meta);\n    } else if (role === \"tool\") {\n      return new ToolMessage(text, meta);\n    } else {\n      return new CustomMessage(role, text, meta);\n    }\n  }\n\n  get text() {\n    return this.getTexts()\n      .map((c) => c.text)\n      .join(\"\");\n  }\n\n  getTexts() {\n    return this.content.filter(isText) as TextPart[];\n  }\n\n  createSnapshot() {\n    return { content: shallowCopy(this.content), meta: shallowCopy(this.meta), role: this.role };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n\n  toPlain() {\n    return { role: this.role, content: shallowCopy(this.content) } as const;\n  }\n\n  merge(other: Message<T, R>) {\n    Object.assign(this, other.meta);\n    this.content.push(...other.content);\n  }\n\n  static fromChunks<M2 extends Message>(this: new (...args: any[]) => M2, chunks: M2[]): M2 {\n    const instance = new this([]);\n    chunks.forEach((chunk) => instance.merge(chunk));\n    return instance;\n  }\n\n  [Symbol.iterator](): Iterator<T> {\n    return this.content[Symbol.iterator]();\n  }\n}\n\nexport class AssistantMessage extends Message<TextPart | ToolCallPart | FilePart> {\n  public readonly role = \"assistant\";\n\n  static {\n    this.register();\n  }\n\n  getToolCalls() {\n    return this.content.filter(isToolCall);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n\n  protected assertContent(\n    content: TextPart | ToolCallPart | FilePart,\n  ): asserts content is TextPart | ToolCallPart | FilePart {\n    if (content.type === \"tool-call\") {\n      const key = \"args\";\n      const args = popProp(content as any, key);\n\n      if (args !== null && !hasProp(content, \"input\")) {\n        Logger.root.warn(\n          `The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`,\n        );\n        content.input = args;\n      }\n\n      safeDefineProperty(content, key as keyof typeof content, () => {\n        Logger.root.warn(\n          `The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`,\n        );\n        return content.input;\n      });\n    }\n  }\n}\n\nexport class ToolMessage extends Message<ToolResultPart> {\n  public readonly role = \"tool\";\n\n  static {\n    this.register();\n  }\n\n  getToolResults() {\n    return this.content.filter(isToolResult);\n  }\n\n  protected fromString(text: string): ToolResultPart {\n    const schema = z\n      .object({\n        type: z.literal(\"tool-result\"),\n        toolName: z.string(),\n        toolCallId: z.string(),\n      })\n      .and(z.object({ output: z.any() }).or(z.object({ result: z.any() })));\n\n    const { success, data } = schema.safeParse(text);\n    if (!success) {\n      throw new ValueError(`ToolMessage cannot be created from '${text}'!`);\n    }\n\n    return data as ToolResultPart;\n  }\n\n  protected assertContent(content: ToolResultPart): asserts content is ToolResultPart {\n    const result = popProp(content as any, \"result\", null);\n    const isError = popProp(content as any, \"isError\", false);\n\n    if (result !== null && !hasProp(content, \"output\")) {\n      Logger.root.warn(\n        \"The 'result' property in the ToolMessage class is deprecated and will be removed. Use the 'output' property for content chunks instead.\",\n      );\n      if (isString(result)) {\n        content.output = { type: isError ? \"error-text\" : \"text\", value: result };\n      } else {\n        content.output = { type: isError ? \"error-json\" : \"json\", value: result };\n      }\n    }\n\n    safeDefineProperty(content, \"result\" as any, () => {\n      Logger.root.warn(\n        \"The 'result' property of Tool Message is deprecated and will be removed. Use the 'output.value' property instead.\",\n      );\n      return content.output.value as any;\n    });\n    safeDefineProperty(content, \"isError\" as any, () => {\n      Logger.root.warn(\n        \"The 'isError' property of Tool Message is deprecated and will be removed. Use the 'output.type' property instead.\",\n      );\n      return content.output.type.includes(\"error\");\n    });\n  }\n}\n\nexport class SystemMessage extends Message<TextPart> {\n  public readonly role: MessageRole = \"system\";\n\n  static {\n    this.register();\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class UserMessage extends Message<TextPart | ImagePart | FilePart> {\n  public readonly role = \"user\";\n\n  static {\n    this.register();\n  }\n\n  getImages() {\n    return this.content.filter(isImage);\n  }\n\n  getFiles() {\n    return this.content.filter(isFile);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport const Role = {\n  ASSISTANT: \"assistant\",\n  SYSTEM: \"system\",\n  USER: \"user\",\n} as const;\n\nexport class CustomMessage extends Message<MessageContentPart, string> {\n  public role: string;\n\n  constructor(role: string, content: MessageContentPart | string, meta: MessageMeta = {}) {\n    super(content, meta);\n    if (!role) {\n      throw new ValueError(`Role \"${role}\" must be specified!`);\n    }\n    this.role = role;\n  }\n\n  protected fromString(input: string): MessageContentPart {\n    return { type: \"text\", text: input };\n  }\n}\n"]}
+{"version":3,"sources":["../../src/backend/message.ts"],"names":["isText","content","type","isImage","isFile","isToolCall","isToolResult","Message","Serializable","id","meta","undefined","createdAt","Date","fromString","Array","isArray","chunk","assertContent","watchArray","onAdd","of","role","text","UserMessage","AssistantMessage","SystemMessage","ToolMessage","CustomMessage","getTexts","map","c","join","filter","createSnapshot","shallowCopy","loadSnapshot","snapshot","Object","assign","toPlain","merge","other","push","fromChunks","chunks","instance","forEach","Symbol","iterator","register","getToolCalls","key","getOwnPropertyDescriptor","args","popProp","hasProp","Logger","root","warn","input","safeDefineProperty","getToolResults","schema","z","object","literal","toolName","string","toolCallId","and","output","any","or","result","success","data","safeParse","ValueError","isError","isString","value","includes","getImages","getFiles","Role","ASSISTANT","SYSTEM","USER"],"mappings":";;;;;;;;;;;AA8BA,SAASA,OAAOC,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSF,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASG,QAAQF,OAAAA,EAA2B;AAC1C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,OAAA;AAC1B;AAFSC,MAAAA,CAAAA,OAAAA,EAAAA,SAAAA,CAAAA;AAGT,SAASC,OAAOH,OAAAA,EAA2B;AACzC,EAAA,OAAOA,QAAQC,IAAAA,KAAS,MAAA;AAC1B;AAFSE,MAAAA,CAAAA,MAAAA,EAAAA,QAAAA,CAAAA;AAGT,SAASC,WAAWJ,OAAAA,EAA2B;AAC7C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,WAAA;AAC1B;AAFSG,MAAAA,CAAAA,UAAAA,EAAAA,YAAAA,CAAAA;AAGT,SAASC,aAAaL,OAAAA,EAA2B;AAC/C,EAAA,OAAOA,QAAQC,IAAAA,KAAS,aAAA;AAC1B;AAFSI,MAAAA,CAAAA,YAAAA,EAAAA,cAAAA,CAAAA;AAIF,MAAeC,gBAGZC,YAAAA,CAAAA;EAjDV;;;;AAkDSC,EAAAA,EAAAA;AAESR,EAAAA,OAAAA;AAEhB,EAAA,WAAA,CACEA,OAAAA,EACgBS,IAAAA,GAAoB,EAAC,EACrCD,KAAyBE,MAAAA,EACzB;AACA,IAAA,KAAA,EAAK,EAAA,KAHWD,IAAAA,GAAAA,IAAAA;AAIhB,IAAA,IAAA,CAAKD,EAAAA,GAAKA,EAAAA;AACV,IAAA,IAAI,CAACC,MAAME,SAAAA,EAAW;AACpBF,MAAAA,IAAAA,CAAKE,SAAAA,uBAAgBC,IAAAA,EAAAA;AACvB;AACA,IAAA,IAAI,OAAOZ,YAAY,QAAA,EAAU;AAC/B,MAAA,IAAA,CAAKA,OAAAA,GAAU;AAAC,QAAA,IAAA,CAAKa,WAAWb,OAAAA;;KAClC,MAAO;AACL,MAAA,IAAA,CAAKA,OAAAA,GAAUc,KAAAA,CAAMC,OAAAA,CAAQf,OAAAA,IAAWA,OAAAA,GAAU;AAACA,QAAAA;;AACrD;AACA,IAAA,KAAA,MAAWgB,KAAAA,IAAS,KAAKhB,OAAAA,EAAS;AAChC,MAAA,IAAA,CAAKiB,cAAcD,KAAAA,CAAAA;AACrB;AACA,IAAA,IAAA,CAAKhB,OAAAA,GAAUkB,UAAAA,CAAW,IAAA,CAAKlB,OAAAA,EAAS;AACtCmB,MAAAA,KAAAA,EAAO,IAAA,CAAKF;KACd,CAAA;AACF;AAEUA,EAAAA,aAAAA,CAAcjB,OAAAA,EAAkC;AAAC;AAI3D,EAAA,OAAOoB,GAAG,EAAEC,IAAAA,EAAMC,IAAAA,EAAMb,IAAAA,EAAMD,IAAE,EAA2B;AACzD,IAAA,IAAIa,SAAS,MAAA,EAAQ;AACnB,MAAA,OAAO,IAAIE,WAAAA,CAAYD,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;AACrC,KAAA,MAAA,IAAWa,SAAS,WAAA,EAAa;AAC/B,MAAA,OAAO,IAAIG,gBAAAA,CAAiBF,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;AAC1C,KAAA,MAAA,IAAWa,SAAS,QAAA,EAAU;AAC5B,MAAA,OAAO,IAAII,aAAAA,CAAcH,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;AACvC,KAAA,MAAA,IAAWa,SAAS,MAAA,EAAQ;AAC1B,MAAA,OAAO,IAAIK,WAAAA,CAAYJ,IAAAA,EAAMb,IAAAA,EAAMD,EAAAA,CAAAA;KACrC,MAAO;AACL,MAAA,OAAO,IAAImB,aAAAA,CAAcN,IAAAA,EAAMC,IAAAA,EAAMb,MAAMD,EAAAA,CAAAA;AAC7C;AACF;AAEA,EAAA,IAAIc,IAAAA,GAAO;AACT,IAAA,OAAO,IAAA,CAAKM,QAAAA,EAAQ,CACjBC,GAAAA,CAAI,CAACC,MAAMA,CAAAA,CAAER,IAAI,CAAA,CACjBS,IAAAA,CAAK,EAAA,CAAA;AACV;EAEAH,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAK5B,OAAAA,CAAQgC,MAAAA,CAAOjC,MAAAA,CAAAA;AAC7B;EAEAkC,cAAAA,GAAiB;AACf,IAAA,OAAO;AACLzB,MAAAA,EAAAA,EAAI,IAAA,CAAKA,EAAAA;MACTR,OAAAA,EAASkC,WAAAA,CAAY,KAAKlC,OAAO,CAAA;MACjCS,IAAAA,EAAMyB,WAAAA,CAAY,KAAKzB,IAAI,CAAA;AAC3BY,MAAAA,IAAAA,EAAM,IAAA,CAAKA;AACb,KAAA;AACF;AAEAc,EAAAA,YAAAA,CAAaC,QAAAA,EAAkD;AAC7DC,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,MAAMF,QAAAA,CAAAA;AACtB;EAEAG,OAAAA,GAAU;AACR,IAAA,OAAO;AAAE/B,MAAAA,EAAAA,EAAI,IAAA,CAAKA,EAAAA;AAAIa,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMrB,OAAAA,EAASkC,WAAAA,CAAY,KAAKlC,OAAO;AAAE,KAAA;AAC5E;AAEAwC,EAAAA,KAAAA,CAAMC,KAAAA,EAAsB;AAC1B,IAAA,IAAA,CAAKjC,EAAAA,GAAK,IAAA,CAAKA,EAAAA,IAAMiC,KAAAA,CAAMjC,EAAAA;AAC3B6B,IAAAA,MAAAA,CAAOC,MAAAA,CAAO,IAAA,EAAMG,KAAAA,CAAMhC,IAAI,CAAA;AAC9B,IAAA,IAAA,CAAKT,OAAAA,CAAQ0C,IAAAA,CAAI,GAAID,KAAAA,CAAMzC,OAAO,CAAA;AACpC;AAEA,EAAA,OAAO2C,WAAiEC,MAAAA,EAAkB;AACxF,IAAA,MAAMC,QAAAA,GAAW,IAAI,IAAA,CAAK,EAAE,CAAA;AAC5BD,IAAAA,MAAAA,CAAOE,QAAQ,CAAC9B,KAAAA,KAAU6B,QAAAA,CAASL,KAAAA,CAAMxB,KAAAA,CAAAA,CAAAA;AACzC,IAAA,OAAO6B,QAAAA;AACT;EAEA,CAACE,MAAAA,CAAOC,QAAQ,CAAA,GAAiB;AAC/B,IAAA,OAAO,IAAA,CAAKhD,OAAAA,CAAQ+C,MAAAA,CAAOC,QAAQ,CAAA,EAAC;AACtC;AACF;AAEO,MAAMxB,yBAAyBlB,OAAAA,CAAAA;EA3ItC;;;EA4IkBe,IAAAA,GAAO,WAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAC,YAAAA,GAAe;AACb,IAAA,OAAO,IAAA,CAAKlD,OAAAA,CAAQgC,MAAAA,CAAO5B,UAAAA,CAAAA;AAC7B;AAEUS,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAErB,IAAAA,EAAM,MAAA;AAAQqB,MAAAA;AAAK,KAAA;AAC9B;AAEUL,EAAAA,aAAAA,CACRjB,OAAAA,EACuD;AACvD,IAAA,IAAIA,OAAAA,CAAQC,SAAS,WAAA,EAAa;AAChC,MAAA,MAAMkD,GAAAA,GAAM,MAAA;AACZ,MAAA,IAAI,CAACd,MAAAA,CAAOe,wBAAAA,CAAyBpD,OAAAA,EAASmD,GAAAA,CAAAA,EAAM;AAClD,QAAA,MAAME,IAAAA,GAAOC,OAAAA,CAAQtD,OAAAA,EAAgBmD,GAAAA,CAAAA;AAErC,QAAA,IAAIE,SAAS,IAAA,IAAQ,CAACE,OAAAA,CAAQvD,OAAAA,EAAS,OAAA,CAAA,EAAU;AAC/CwD,UAAAA,MAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQP,GAAAA,CAAAA,0IAAAA,CAA+I,CAAA;AAEzJnD,UAAAA,OAAAA,CAAQ2D,KAAAA,GAAQN,IAAAA;AAClB;AACF;AAEAO,MAAAA,kBAAAA,CAAmB5D,OAAAA,EAASmD,KAA6B,MAAA;AACvDK,QAAAA,MAAAA,CAAOC,IAAAA,CAAKC,IAAAA,CACV,CAAA,KAAA,EAAQP,GAAAA,CAAAA,gGAAAA,CAAqG,CAAA;AAE/G,QAAA,OAAOnD,OAAAA,CAAQ2D,KAAAA;OACjB,CAAA;AACF;AACF;AACF;AAEO,MAAMjC,oBAAoBpB,OAAAA,CAAAA;EApLjC;;;EAqLkBe,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAY,cAAAA,GAAiB;AACf,IAAA,OAAO,IAAA,CAAK7D,OAAAA,CAAQgC,MAAAA,CAAO3B,YAAAA,CAAAA;AAC7B;AAEUQ,EAAAA,UAAAA,CAAWS,IAAAA,EAA8B;AACjD,IAAA,MAAMwC,MAAAA,GAASC,EACZC,MAAAA,CAAO;MACN/D,IAAAA,EAAM8D,CAAAA,CAAEE,QAAQ,aAAA,CAAA;AAChBC,MAAAA,QAAAA,EAAUH,EAAEI,MAAAA,EAAM;AAClBC,MAAAA,UAAAA,EAAYL,EAAEI,MAAAA;KAChB,CAAA,CACCE,GAAAA,CAAIN,CAAAA,CAAEC,MAAAA,CAAO;AAAEM,MAAAA,MAAAA,EAAQP,EAAEQ,GAAAA;KAAM,CAAA,CAAGC,EAAAA,CAAGT,CAAAA,CAAEC,MAAAA,CAAO;AAAES,MAAAA,MAAAA,EAAQV,EAAEQ,GAAAA;AAAM,KAAA,CAAA,CAAA,CAAA;AAEnE,IAAA,MAAM,EAAEG,OAAAA,EAASC,IAAAA,EAAI,GAAKb,MAAAA,CAAOc,UAAUtD,IAAAA,CAAAA;AAC3C,IAAA,IAAI,CAACoD,OAAAA,EAAS;AACZ,MAAA,MAAM,IAAIG,UAAAA,CAAW,CAAA,oCAAA,EAAuCvD,IAAAA,CAAAA,EAAAA,CAAQ,CAAA;AACtE;AAEA,IAAA,OAAOqD,IAAAA;AACT;AAEU1D,EAAAA,aAAAA,CAAcjB,OAAAA,EAA4D;AAClF,IAAA,MAAMyE,MAAAA,GAASnB,OAAAA,CAAQtD,OAAAA,EAAgB,QAAA,EAAU,IAAA,CAAA;AACjD,IAAA,MAAM8E,OAAAA,GAAUxB,OAAAA,CAAQtD,OAAAA,EAAgB,SAAA,EAAW,KAAA,CAAA;AAEnD,IAAA,IAAIyE,WAAW,IAAA,IAAQ,CAAClB,OAAAA,CAAQvD,OAAAA,EAAS,QAAA,CAAA,EAAW;AAClDwD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,yIAAA,CAAA;AAEF,MAAA,IAAIqB,QAAAA,CAASN,MAAAA,CAAAA,EAAS;AACpBzE,QAAAA,OAAAA,CAAQsE,MAAAA,GAAS;AAAErE,UAAAA,IAAAA,EAAM6E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;OAC1E,MAAO;AACLzE,QAAAA,OAAAA,CAAQsE,MAAAA,GAAS;AAAErE,UAAAA,IAAAA,EAAM6E,UAAU,YAAA,GAAe,MAAA;UAAQE,KAAAA,EAAOP;AAAO,SAAA;AAC1E;AACF;AAEAb,IAAAA,kBAAAA,CAAmB5D,OAAAA,EAAS,UAAiB,MAAA;AAC3CwD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAO1D,QAAQsE,MAAAA,CAAOU,KAAAA;KACxB,CAAA;AACApB,IAAAA,kBAAAA,CAAmB5D,OAAAA,EAAS,WAAkB,MAAA;AAC5CwD,MAAAA,MAAAA,CAAOC,IAAAA,CAAKC,KACV,mHAAA,CAAA;AAEF,MAAA,OAAO1D,OAAAA,CAAQsE,MAAAA,CAAOrE,IAAAA,CAAKgF,QAAAA,CAAS,OAAA,CAAA;KACtC,CAAA;AACF;AACF;AAEO,MAAMxD,sBAAsBnB,OAAAA,CAAAA;EA9OnC;;;EA+OkBe,IAAAA,GAAoB,QAAA;EAEpC;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;AAEUpC,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAErB,IAAAA,EAAM,MAAA;AAAQqB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAMC,oBAAoBjB,OAAAA,CAAAA;EA1PjC;;;EA2PkBe,IAAAA,GAAO,MAAA;EAEvB;AACE,IAAA,IAAA,CAAK4B,QAAAA,EAAQ;AACf;EAEAiC,SAAAA,GAAY;AACV,IAAA,OAAO,IAAA,CAAKlF,OAAAA,CAAQgC,MAAAA,CAAO9B,OAAAA,CAAAA;AAC7B;EAEAiF,QAAAA,GAAW;AACT,IAAA,OAAO,IAAA,CAAKnF,OAAAA,CAAQgC,MAAAA,CAAO7B,MAAAA,CAAAA;AAC7B;AAEUU,EAAAA,UAAAA,CAAWS,IAAAA,EAAwB;AAC3C,IAAA,OAAO;MAAErB,IAAAA,EAAM,MAAA;AAAQqB,MAAAA;AAAK,KAAA;AAC9B;AACF;AAEO,MAAM8D,IAAAA,GAAO;EAClBC,SAAAA,EAAW,WAAA;EACXC,MAAAA,EAAQ,QAAA;EACRC,IAAAA,EAAM;AACR;AAEO,MAAM5D,sBAAsBrB,OAAAA,CAAAA;EApRnC;;;AAqRSe,EAAAA,IAAAA;AAEP,EAAA,WAAA,CACEA,IAAAA,EACArB,OAAAA,EACAS,IAAAA,GAAoB,IACpBD,EAAAA,EACA;AACA,IAAA,KAAA,CAAMR,OAAAA,EAASS,MAAMD,EAAAA,CAAAA;AACrB,IAAA,IAAI,CAACa,IAAAA,EAAM;AACT,MAAA,MAAM,IAAIwD,UAAAA,CAAW,CAAA,MAAA,EAASxD,IAAAA,CAAAA,oBAAAA,CAA0B,CAAA;AAC1D;AACA,IAAA,IAAA,CAAKA,IAAAA,GAAOA,IAAAA;AACd;AAEUR,EAAAA,UAAAA,CAAW8C,KAAAA,EAAmC;AACtD,IAAA,OAAO;MAAE1D,IAAAA,EAAM,MAAA;MAAQqB,IAAAA,EAAMqC;AAAM,KAAA;AACrC;AACF","file":"message.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Serializable } from \"@/internals/serializable.js\";\nimport { shallowCopy } from \"@/serializer/utils.js\";\nimport { FilePart, ImagePart, TextPart, ToolCallPart, ToolResultPart } from \"ai\";\nimport { z } from \"zod\";\nimport { ValueError } from \"@/errors.js\";\nimport { hasProp, popProp, safeDefineProperty } from \"@/internals/helpers/object.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { isString } from \"remeda\";\nimport { watchArray } from \"@/internals/helpers/array.js\";\n\nexport type MessageRole = \"user\" | \"system\" | \"tool\" | \"assistant\";\nexport type MessageContentPart = TextPart | ToolCallPart | ImagePart | FilePart | ToolResultPart;\n\nexport interface MessageMeta {\n  [key: string]: any;\n  createdAt?: Date;\n}\n\nexport interface MessageInput {\n  id?: string;\n  role: MessageRole;\n  text: string; // TODO\n  meta?: MessageMeta;\n}\n\nfunction isText(content: MessageContentPart): content is TextPart {\n  return content.type === \"text\";\n}\nfunction isImage(content: MessageContentPart): content is ImagePart {\n  return content.type === \"image\";\n}\nfunction isFile(content: MessageContentPart): content is FilePart {\n  return content.type === \"file\";\n}\nfunction isToolCall(content: MessageContentPart): content is ToolCallPart {\n  return content.type === \"tool-call\";\n}\nfunction isToolResult(content: MessageContentPart): content is ToolResultPart {\n  return content.type === \"tool-result\";\n}\n\nexport abstract class Message<\n  T extends MessageContentPart = MessageContentPart,\n  R extends string = MessageRole | string,\n> extends Serializable {\n  public id: string | undefined;\n  public abstract readonly role: R;\n  public readonly content: T[];\n\n  constructor(\n    content: T | T[] | string,\n    public readonly meta: MessageMeta = {},\n    id: string | undefined = undefined,\n  ) {\n    super();\n    this.id = id;\n    if (!meta?.createdAt) {\n      meta.createdAt = new Date();\n    }\n    if (typeof content === \"string\") {\n      this.content = [this.fromString(content)];\n    } else {\n      this.content = Array.isArray(content) ? content : [content];\n    }\n    for (const chunk of this.content) {\n      this.assertContent(chunk);\n    }\n    this.content = watchArray(this.content, {\n      onAdd: this.assertContent,\n    });\n  }\n\n  protected assertContent(content: T): asserts content is T {}\n\n  protected abstract fromString(input: string): T;\n\n  static of({ role, text, meta, id }: MessageInput): Message {\n    if (role === \"user\") {\n      return new UserMessage(text, meta, id);\n    } else if (role === \"assistant\") {\n      return new AssistantMessage(text, meta, id);\n    } else if (role === \"system\") {\n      return new SystemMessage(text, meta, id);\n    } else if (role === \"tool\") {\n      return new ToolMessage(text, meta, id);\n    } else {\n      return new CustomMessage(role, text, meta, id);\n    }\n  }\n\n  get text() {\n    return this.getTexts()\n      .map((c) => c.text)\n      .join(\"\");\n  }\n\n  getTexts() {\n    return this.content.filter(isText) as TextPart[];\n  }\n\n  createSnapshot() {\n    return {\n      id: this.id,\n      content: shallowCopy(this.content),\n      meta: shallowCopy(this.meta),\n      role: this.role,\n    };\n  }\n\n  loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>) {\n    Object.assign(this, snapshot);\n  }\n\n  toPlain() {\n    return { id: this.id, role: this.role, content: shallowCopy(this.content) } as const;\n  }\n\n  merge(other: Message<T, R>) {\n    this.id = this.id || other.id;\n    Object.assign(this, other.meta);\n    this.content.push(...other.content);\n  }\n\n  static fromChunks<M2 extends Message>(this: new (...args: any[]) => M2, chunks: M2[]): M2 {\n    const instance = new this([]);\n    chunks.forEach((chunk) => instance.merge(chunk));\n    return instance;\n  }\n\n  [Symbol.iterator](): Iterator<T> {\n    return this.content[Symbol.iterator]();\n  }\n}\n\nexport class AssistantMessage extends Message<TextPart | ToolCallPart | FilePart> {\n  public readonly role = \"assistant\";\n\n  static {\n    this.register();\n  }\n\n  getToolCalls() {\n    return this.content.filter(isToolCall);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n\n  protected assertContent(\n    content: TextPart | ToolCallPart | FilePart,\n  ): asserts content is TextPart | ToolCallPart | FilePart {\n    if (content.type === \"tool-call\") {\n      const key = \"args\";\n      if (!Object.getOwnPropertyDescriptor(content, key)) {\n        const args = popProp(content as any, key);\n\n        if (args !== null && !hasProp(content, \"input\")) {\n          Logger.root.warn(\n            `The '${key}' property in the AssistantMessage class is deprecated and will be removed. Use the 'input' property for Tool Call content chunks instead.`,\n          );\n          content.input = args;\n        }\n      }\n\n      safeDefineProperty(content, key as keyof typeof content, () => {\n        Logger.root.warn(\n          `The '${key}' property in the Tool Call is deprecated and will be removed. Use the 'input' property instead.`,\n        );\n        return content.input;\n      });\n    }\n  }\n}\n\nexport class ToolMessage extends Message<ToolResultPart> {\n  public readonly role = \"tool\";\n\n  static {\n    this.register();\n  }\n\n  getToolResults() {\n    return this.content.filter(isToolResult);\n  }\n\n  protected fromString(text: string): ToolResultPart {\n    const schema = z\n      .object({\n        type: z.literal(\"tool-result\"),\n        toolName: z.string(),\n        toolCallId: z.string(),\n      })\n      .and(z.object({ output: z.any() }).or(z.object({ result: z.any() })));\n\n    const { success, data } = schema.safeParse(text);\n    if (!success) {\n      throw new ValueError(`ToolMessage cannot be created from '${text}'!`);\n    }\n\n    return data as ToolResultPart;\n  }\n\n  protected assertContent(content: ToolResultPart): asserts content is ToolResultPart {\n    const result = popProp(content as any, \"result\", null);\n    const isError = popProp(content as any, \"isError\", false);\n\n    if (result !== null && !hasProp(content, \"output\")) {\n      Logger.root.warn(\n        \"The 'result' property in the ToolMessage class is deprecated and will be removed. Use the 'output' property for content chunks instead.\",\n      );\n      if (isString(result)) {\n        content.output = { type: isError ? \"error-text\" : \"text\", value: result };\n      } else {\n        content.output = { type: isError ? \"error-json\" : \"json\", value: result };\n      }\n    }\n\n    safeDefineProperty(content, \"result\" as any, () => {\n      Logger.root.warn(\n        \"The 'result' property of Tool Message is deprecated and will be removed. Use the 'output.value' property instead.\",\n      );\n      return content.output.value as any;\n    });\n    safeDefineProperty(content, \"isError\" as any, () => {\n      Logger.root.warn(\n        \"The 'isError' property of Tool Message is deprecated and will be removed. Use the 'output.type' property instead.\",\n      );\n      return content.output.type.includes(\"error\");\n    });\n  }\n}\n\nexport class SystemMessage extends Message<TextPart> {\n  public readonly role: MessageRole = \"system\";\n\n  static {\n    this.register();\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport class UserMessage extends Message<TextPart | ImagePart | FilePart> {\n  public readonly role = \"user\";\n\n  static {\n    this.register();\n  }\n\n  getImages() {\n    return this.content.filter(isImage);\n  }\n\n  getFiles() {\n    return this.content.filter(isFile);\n  }\n\n  protected fromString(text: string): TextPart {\n    return { type: \"text\", text };\n  }\n}\n\nexport const Role = {\n  ASSISTANT: \"assistant\",\n  SYSTEM: \"system\",\n  USER: \"user\",\n} as const;\n\nexport class CustomMessage extends Message<MessageContentPart, string> {\n  public role: string;\n\n  constructor(\n    role: string,\n    content: MessageContentPart | string,\n    meta: MessageMeta = {},\n    id?: string,\n  ) {\n    super(content, meta, id);\n    if (!role) {\n      throw new ValueError(`Role \"${role}\" must be specified!`);\n    }\n    this.role = role;\n  }\n\n  protected fromString(input: string): MessageContentPart {\n    return { type: \"text\", text: input };\n  }\n}\n"]}

package/dist/backend/utils.d.cts

@@ -1,12 +1,12 @@
 import '../internals/types.cjs';
 import './constants.cjs';
 import '../tools/base.cjs';
-export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-CRb3vUVg.cjs';
+export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-BOb9n7AJ.cjs';
 import 'zod';
 import '../internals/helpers/guards.cjs';
 import 'ajv';
 import '../context.cjs';
-import '../emitter-DpqUYjXH.cjs';
+import '../emitter-jN55XZZq.cjs';
 import '../internals/serializable.cjs';
 import '../internals/helpers/promise.cjs';
 import '../errors.cjs';

package/dist/backend/utils.d.ts

@@ -1,12 +1,12 @@
 import '../internals/types.js';
 import './constants.js';
 import '../tools/base.js';
-export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-C0s-o6ll.js';
+export { F as FullModelName, r as filterToolsByToolChoice, q as generateToolUnionSchema, o as loadModel, p as parseModel } from '../chat-DrqwUgD8.js';
 import 'zod';
 import '../internals/helpers/guards.js';
 import 'ajv';
 import '../context.js';
-import '../emitter-CZFbzlUi.js';
+import '../emitter-36-9MnvA.js';
 import '../internals/serializable.js';
 import '../internals/helpers/promise.js';
 import '../errors.js';

package/dist/{base-BQgDlcGa.d.ts → base-CyFx7BRL.d.ts}

@@ -1,7 +1,7 @@
-import { Run, GetRunContext } from './context.js';
+import { MiddlewareType, Run, GetRunContext } from './context.js';
 import { FrameworkError } from './errors.js';
 import { Serializable } from './internals/serializable.js';
-import { E as Emitter } from './emitter-CZFbzlUi.js';
+import { E as Emitter } from './emitter-36-9MnvA.js';
 import { BaseMemory } from './memory/base.js';
 import { OmitEmpty } from './internals/types.js';
 import { AnyTool } from './tools/base.js';
@@ -39,6 +39,7 @@ interface BaseAgentRunOptions {
 declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunOptions = BaseAgentRunOptions> extends Serializable {
     protected isRunning: boolean;
     abstract readonly emitter: Emitter<unknown>;
+    readonly middlewares: MiddlewareType<typeof this>[];
     run(...[input, options]: Partial<TOptions> extends TOptions ? [input: TInput, options?: TOptions] : [input: TInput, options: TOptions]): Run<TOutput, this, readonly [TInput, TOptions | undefined]>;
     protected abstract _run(input: TInput, options: TOptions, run: GetRunContext<typeof this>): Promise<TOutput>;
     destroy(): void;
@@ -48,8 +49,9 @@ declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunO
     createSnapshot(): {
         isRunning: boolean;
         emitter: Emitter<unknown>;
+        middlewares: MiddlewareType<any>[];
     };
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
 }
 
-export { AgentError as A, type BaseAgentRunOptions as B, type GetAgentInput as G, type InternalAgentCallbackValue as I, type PublicAgentCallbackValue as P, BaseAgent as a, type AgentMeta as b, type AnyAgent as c, type AgentCallbackValue as d, type AgentCallback as e, type GetAgentOutput as f };
+export { AgentError as A, type BaseAgentRunOptions as B, type GetAgentInput as G, type InternalAgentCallbackValue as I, type PublicAgentCallbackValue as P, BaseAgent as a, type AnyAgent as b, type AgentMeta as c, type AgentCallbackValue as d, type AgentCallback as e, type GetAgentOutput as f };

package/dist/{base-CwcxwybB.d.cts → base-XfWeXQ-S.d.cts}

@@ -1,7 +1,7 @@
-import { Run, GetRunContext } from './context.cjs';
+import { MiddlewareType, Run, GetRunContext } from './context.cjs';
 import { FrameworkError } from './errors.cjs';
 import { Serializable } from './internals/serializable.cjs';
-import { E as Emitter } from './emitter-DpqUYjXH.cjs';
+import { E as Emitter } from './emitter-jN55XZZq.cjs';
 import { BaseMemory } from './memory/base.cjs';
 import { OmitEmpty } from './internals/types.cjs';
 import { AnyTool } from './tools/base.cjs';
@@ -39,6 +39,7 @@ interface BaseAgentRunOptions {
 declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunOptions = BaseAgentRunOptions> extends Serializable {
     protected isRunning: boolean;
     abstract readonly emitter: Emitter<unknown>;
+    readonly middlewares: MiddlewareType<typeof this>[];
     run(...[input, options]: Partial<TOptions> extends TOptions ? [input: TInput, options?: TOptions] : [input: TInput, options: TOptions]): Run<TOutput, this, readonly [TInput, TOptions | undefined]>;
     protected abstract _run(input: TInput, options: TOptions, run: GetRunContext<typeof this>): Promise<TOutput>;
     destroy(): void;
@@ -48,8 +49,9 @@ declare abstract class BaseAgent<TInput, TOutput, TOptions extends BaseAgentRunO
     createSnapshot(): {
         isRunning: boolean;
         emitter: Emitter<unknown>;
+        middlewares: MiddlewareType<any>[];
     };
     loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void;
 }
 
-export { AgentError as A, type BaseAgentRunOptions as B, type GetAgentInput as G, type InternalAgentCallbackValue as I, type PublicAgentCallbackValue as P, BaseAgent as a, type AgentMeta as b, type AnyAgent as c, type AgentCallbackValue as d, type AgentCallback as e, type GetAgentOutput as f };
+export { AgentError as A, type BaseAgentRunOptions as B, type GetAgentInput as G, type InternalAgentCallbackValue as I, type PublicAgentCallbackValue as P, BaseAgent as a, type AnyAgent as b, type AgentMeta as c, type AgentCallbackValue as d, type AgentCallback as e, type GetAgentOutput as f };

package/dist/{chat-CRb3vUVg.d.cts → chat-BOb9n7AJ.d.cts}

@@ -1,8 +1,7 @@
 import { Message, AssistantMessage, MessageContentPart } from './backend/message.cjs';
-import * as ai from 'ai';
 import { Run, GetRunContext } from './context.cjs';
 import { Serializable } from './internals/serializable.cjs';
-import { C as Callback, E as Emitter } from './emitter-DpqUYjXH.cjs';
+import { C as Callback, E as Emitter } from './emitter-jN55XZZq.cjs';
 import { FrameworkError } from './errors.cjs';
 import { Task } from 'promise-based-task';
 import { BaseCache } from './cache/base.cjs';
@@ -13,6 +12,7 @@ import { ZodSchema, z, ZodType } from 'zod';
 import { SchemaObject } from 'ajv';
 import { PromptTemplate } from './template.cjs';
 import { Logger } from './logger/logger.cjs';
+import { ToolCallPart } from 'ai';
 
 /**
  * Copyright 2025 © BeeAI a Series of LF Projects, LLC
@@ -39,6 +39,7 @@ interface ChatModelParameters {
     presencePenalty?: number;
     seed?: number;
     stopSequences?: string[];
+    stream?: boolean;
 }
 interface ResponseObjectJson {
     type: "object-json";
@@ -67,6 +68,7 @@ interface ChatModelInput extends ChatModelParameters {
     responseFormat?: ZodSchema | ResponseObjectJson;
     toolChoice?: ChatModelToolChoice;
     messages: Message[];
+    streamPartialToolCalls?: boolean;
 }
 type ChatModelFinishReason = "stop" | "length" | "content-filter" | "tool-calls" | "error" | "other" | "unknown";
 interface ChatModelUsage {
@@ -113,11 +115,7 @@ declare abstract class ChatModel extends Serializable {
     readonly modelSupportsToolCalling: boolean;
     abstract get modelId(): string;
     abstract get providerId(): string;
-    create(input: ChatModelInput & {
-        stream?: boolean;
-    }): Run<ChatModelOutput, this, readonly [ChatModelInput & {
-        stream?: boolean;
-    }]>;
+    create(input: ChatModelInput): Run<ChatModelOutput, this, readonly [ChatModelInput]>;
     createStructure<T>(input: ChatModelObjectInput<T>): Run<ChatModelObjectOutput<T>, this, readonly [ChatModelObjectInput<T>]>;
     config({ cache, parameters }: ChatConfig): void;
     static fromName(name: FullModelName | ProviderName, options?: ChatModelParameters): Promise<ChatModel>;
@@ -150,7 +148,8 @@ declare class ChatModelOutput extends Serializable {
     constructor(messages: Message[], usage?: ChatModelUsage | undefined, finishReason?: ChatModelFinishReason | undefined);
     static fromChunks(chunks: ChatModelOutput[]): ChatModelOutput;
     merge(other: ChatModelOutput): void;
-    getToolCalls(): ai.ToolCallPart[];
+    dedupe(): void;
+    getToolCalls(): ToolCallPart[];
     getTextMessages(): AssistantMessage[];
     getTextContent(): string;
     toString(): string;