beeai-framework 0.1.21 → 0.1.22

package/dist/middleware/trajectory.cjs

@@ -0,0 +1,322 @@
+'use strict';
+
+var context_cjs = require('../context.cjs');
+var emitter_cjs = require('../emitter/emitter.cjs');
+var errors_cjs = require('../errors.cjs');
+var logger_cjs = require('../logger/logger.cjs');
+var base_cjs = require('../agents/base.cjs');
+var remeda = require('remeda');
+var chat_cjs = require('../backend/chat.cjs');
+var base_cjs$1 = require('../tools/base.cjs');
+var requirement_cjs = require('../agents/requirement/requirements/requirement.cjs');
+var utils = require('sequelize/lib/utils');
+var serializer_cjs = require('../serializer/serializer.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+class GlobalTrajectoryMiddleware extends context_cjs.Middleware {
+  static {
+    __name(this, "GlobalTrajectoryMiddleware");
+  }
+  enabled;
+  included;
+  excluded;
+  cleanups = [];
+  target;
+  ctx = null;
+  pretty;
+  traceLevel = /* @__PURE__ */ new Map();
+  prefixByType;
+  matchNested;
+  emitterPriority;
+  formatter;
+  emitter;
+  constructor(options = {}) {
+    super();
+    this.enabled = options.enabled ?? true;
+    this.included = options.included ?? [];
+    this.excluded = options.excluded ?? [];
+    this.target = this.createTarget(options.target);
+    this.pretty = options.pretty ?? false;
+    this.matchNested = options.matchNested ?? true;
+    this.emitterPriority = options.emitterPriority ?? -1;
+    this.prefixByType = new Map([
+      [
+        base_cjs.BaseAgent,
+        "\u{1F916} "
+      ],
+      [
+        chat_cjs.ChatModel,
+        "\u{1F4AC} "
+      ],
+      [
+        base_cjs$1.Tool,
+        "\u{1F6E0}\uFE0F "
+      ],
+      [
+        requirement_cjs.Requirement,
+        "\u{1F50E} "
+      ],
+      ...options.prefixByType ? Array.from(options.prefixByType.entries()) : []
+    ]);
+    this.formatter = options.formatter ?? ((x) => `${x.prefix}${x.className}[${x.instanceName || x.className}][${x.eventName}]`);
+    this.emitter = emitter_cjs.Emitter.root.child({
+      namespace: [
+        "middleware",
+        "globalTrajectory"
+      ]
+    });
+  }
+  /**
+  * Bind the middleware to a run context
+  */
+  bind(ctx) {
+    while (this.cleanups.length > 0) {
+      this.cleanups.pop()();
+    }
+    this.traceLevel.clear();
+    this.traceLevel.set(ctx.runId, {
+      relative: 0,
+      absolute: 0
+    });
+    this.ctx = ctx;
+    this.bindEmitter(ctx.emitter);
+  }
+  createTarget(input) {
+    if (input === null || input === void 0) {
+      return (msg) => console.log(msg);
+    } else if (input instanceof logger_cjs.Logger) {
+      return (msg) => input.debug(msg);
+    } else {
+      return input;
+    }
+  }
+  bindEmitter(emitter) {
+    this.cleanups.push(emitter.match("*.*", (_, event) => this.logTraceId(event), {
+      matchNested: true
+    }));
+    const handleTopLevelEvent = /* @__PURE__ */ __name(async (data, meta) => {
+      if (!(meta.creator instanceof context_cjs.RunContext)) {
+        return;
+      }
+      if (!meta.trace) {
+        return;
+      }
+      this.logTraceId(meta);
+      if (!this.isAllowed(meta)) {
+        return;
+      }
+      if (!this.enabled) {
+        return;
+      }
+      const eventName = remeda.capitalize(meta.name);
+      await this[`onInternal${eventName}`].call(this, data, meta);
+    }, "handleTopLevelEvent");
+    this.cleanups.push(emitter.match((event) => [
+      "start",
+      "success",
+      "error",
+      "finish"
+    ].includes(event.name), handleTopLevelEvent, {
+      matchNested: false,
+      isBlocking: true,
+      priority: this.emitterPriority
+    }));
+    if (this.matchNested) {
+      const handleNestedEvent = /* @__PURE__ */ __name(async (data, meta) => {
+        if (!(meta.creator instanceof context_cjs.RunContext)) {
+          return;
+        }
+        if (meta.creator.emitter !== emitter) {
+          await handleTopLevelEvent(data, meta);
+          this.bindEmitter(meta.creator.emitter);
+        }
+      }, "handleNestedEvent");
+      this.cleanups.push(emitter.match((event) => event.name === "start", handleNestedEvent, {
+        matchNested: true,
+        isBlocking: true,
+        priority: this.emitterPriority
+      }));
+    }
+  }
+  logTraceId(meta) {
+    if (!meta.trace?.runId) {
+      return;
+    }
+    if (this.traceLevel.has(meta.trace.runId)) {
+      return;
+    }
+    if (meta.trace.parentRunId === meta.trace.runId) {
+      return;
+    }
+    if (meta.trace.parentRunId) {
+      const allowed = this.isAllowed(meta);
+      const parentTrace = this.traceLevel.get(meta.trace.parentRunId) ?? {
+        relative: 0,
+        absolute: 0
+      };
+      this.traceLevel.set(meta.trace.runId, {
+        relative: parentTrace.relative + (allowed ? 1 : 0),
+        absolute: parentTrace.absolute + 1
+      });
+    }
+  }
+  isAllowed(meta) {
+    let target = meta.creator;
+    if (target instanceof context_cjs.RunContext) {
+      target = target.instance;
+    }
+    for (const excluded of this.excluded) {
+      if (target instanceof excluded) {
+        return false;
+      }
+    }
+    if (this.included.length === 0) {
+      return true;
+    }
+    return this.included.some((included) => target instanceof included);
+  }
+  extractName(meta) {
+    let target = meta.creator;
+    if (target instanceof context_cjs.RunContext) {
+      target = target.instance;
+    }
+    const className = target.constructor.name;
+    let targetName = null;
+    if (target instanceof base_cjs.BaseAgent && target.meta?.name) {
+      targetName = target.meta.name;
+    } else if ("name" in target && typeof target.name === "string") {
+      targetName = target.name;
+    }
+    let prefix = "";
+    for (const [type, typePrefix] of this.prefixByType.entries()) {
+      if (target instanceof type) {
+        prefix = typePrefix;
+        break;
+      }
+    }
+    const input = {
+      prefix,
+      className,
+      instanceName: targetName,
+      eventName: meta.name
+    };
+    return this.formatter(input);
+  }
+  formatPrefix(meta) {
+    if (!meta.trace) {
+      return "";
+    }
+    const indent = this.getTraceLevel(meta, "self").relative;
+    const indentParent = this.getTraceLevel(meta, "parent").relative;
+    const indentDiff = indent - indentParent;
+    let prefix = "";
+    prefix += "  ".repeat(indentParent * 2);
+    if (meta.name !== "start" && indent) {
+      prefix += "<";
+    }
+    prefix += "--".repeat(indentDiff);
+    if (meta.name === "start" && prefix && indent) {
+      prefix += ">";
+    }
+    if (prefix) {
+      prefix = `${prefix} `;
+    }
+    const name = this.extractName(meta);
+    return `${prefix}${name}: `;
+  }
+  getTraceLevel(meta, type = "self") {
+    if (!meta.trace) {
+      return {
+        relative: 0,
+        absolute: 0
+      };
+    }
+    const runId = type === "parent" ? meta.trace.parentRunId || "" : meta.trace.runId;
+    return this.traceLevel.get(runId) ?? {
+      relative: 0,
+      absolute: 0
+    };
+  }
+  async formatPayload(value) {
+    if (utils.isPrimitive(value)) {
+      return String(value);
+    }
+    if (value instanceof errors_cjs.FrameworkError) {
+      return value.explain();
+    }
+    const serialized = await serializer_cjs.Serializer.serialize(value);
+    return JSON.stringify(await serializer_cjs.Serializer.deserialize(serialized, [], true), /* @__PURE__ */ (() => {
+      const excludedKeys = /* @__PURE__ */ new Set([
+        "emitter",
+        "cleanups",
+        "creator",
+        "listeners"
+      ]);
+      return (key, value2) => {
+        if (excludedKeys.has(key)) {
+          return void 0;
+        }
+        if (value2 && value2 instanceof base_cjs$1.Tool) {
+          return value2.name;
+        }
+        return value2;
+      };
+    })(), this.pretty ? 2 : void 0);
+  }
+  async onInternalStart(payload, meta) {
+    const prefix = this.formatPrefix(meta);
+    const message = `${prefix}${await this.formatPayload(payload.input)}`;
+    await this.emitter.emit("start", {
+      message,
+      level: this.getTraceLevel(meta),
+      origin: [
+        payload,
+        meta
+      ]
+    });
+    this.target(message);
+  }
+  async onInternalSuccess(payload, meta) {
+    const prefix = this.formatPrefix(meta);
+    const message = `${prefix}${await this.formatPayload(payload)}`;
+    await this.emitter.emit("success", {
+      message,
+      level: this.getTraceLevel(meta),
+      origin: [
+        payload,
+        meta
+      ]
+    });
+    this.target(message);
+  }
+  async onInternalError(payload, meta) {
+    const prefix = this.formatPrefix(meta);
+    const message = `${prefix}${await this.formatPayload(payload)}`;
+    await this.emitter.emit("error", {
+      message,
+      level: this.getTraceLevel(meta),
+      origin: [
+        payload,
+        meta
+      ]
+    });
+    this.target(message);
+  }
+  async onInternalFinish(payload, meta) {
+    const prefix = this.formatPrefix(meta);
+    const message = `${prefix}${await this.formatPayload(payload.error || payload.output)}`;
+    await this.emitter.emit("finish", {
+      message,
+      level: this.getTraceLevel(meta),
+      origin: [
+        payload,
+        meta
+      ]
+    });
+  }
+}
+
+exports.GlobalTrajectoryMiddleware = GlobalTrajectoryMiddleware;
+//# sourceMappingURL=trajectory.cjs.map
+//# sourceMappingURL=trajectory.cjs.map

package/dist/middleware/trajectory.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/middleware/trajectory.ts"],"names":["GlobalTrajectoryMiddleware","Middleware","enabled","included","excluded","cleanups","target","ctx","pretty","traceLevel","Map","prefixByType","matchNested","emitterPriority","formatter","emitter","options","createTarget","BaseAgent","ChatModel","Tool","Requirement","Array","from","entries","x","prefix","className","instanceName","eventName","Emitter","root","child","namespace","bind","length","pop","clear","set","runId","relative","absolute","bindEmitter","input","undefined","msg","console","log","Logger","debug","push","match","_","event","logTraceId","handleTopLevelEvent","data","meta","creator","RunContext","trace","isAllowed","capitalize","name","call","includes","isBlocking","priority","handleNestedEvent","has","parentRunId","allowed","parentTrace","get","instance","some","extractName","targetName","type","typePrefix","formatPrefix","indent","getTraceLevel","indentParent","indentDiff","repeat","formatPayload","value","isPrimitive","String","FrameworkError","explain","serialized","Serializer","serialize","JSON","stringify","deserialize","excludedKeys","Set","key","onInternalStart","payload","message","emit","level","origin","onInternalSuccess","onInternalError","onInternalFinish","error","output"],"mappings":";;;;;;;;;;;;;;;;AA0FO,MAAMA,mCAAgEC,sBAAAA,CAAAA;EA1F7E;;;AA2FYC,EAAAA,OAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA;AACAC,EAAAA,QAAAA,GAA2B,EAAA;AAC3BC,EAAAA,MAAAA;EACAC,GAAAA,GAA4B,IAAA;AAC5BC,EAAAA,MAAAA;AACAC,EAAAA,UAAAA,uBAAiBC,GAAAA,EAAAA;AACjBC,EAAAA,YAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,eAAAA;AACAC,EAAAA,SAAAA;AACMC,EAAAA,OAAAA;EAEhB,WAAA,CAAYC,OAAAA,GAA6C,EAAC,EAAG;AAC3D,IAAA,KAAA,EAAK;AAEL,IAAA,IAAA,CAAKd,OAAAA,GAAUc,QAAQd,OAAAA,IAAW,IAAA;AAClC,IAAA,IAAA,CAAKC,QAAAA,GAAWa,OAAAA,CAAQb,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKC,QAAAA,GAAWY,OAAAA,CAAQZ,QAAAA,IAAY,EAAA;AACpC,IAAA,IAAA,CAAKE,MAAAA,GAAS,IAAA,CAAKW,YAAAA,CAAaD,OAAAA,CAAQV,MAAM,CAAA;AAC9C,IAAA,IAAA,CAAKE,MAAAA,GAASQ,QAAQR,MAAAA,IAAU,KAAA;AAChC,IAAA,IAAA,CAAKI,WAAAA,GAAcI,QAAQJ,WAAAA,IAAe,IAAA;AAC1C,IAAA,IAAA,CAAKC,eAAAA,GAAkBG,QAAQH,eAAAA,IAAmB,EAAA;AAGlD,IAAA,IAAA,CAAKF,YAAAA,GAAe,IAAID,GAAAA,CAAI;AAC1B,MAAA;AAACQ,QAAAA,kBAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,kBAAAA;AAAW,QAAA;;AACZ,MAAA;AAACC,QAAAA,eAAAA;AAAM,QAAA;;AACP,MAAA;AAACC,QAAAA,2BAAAA;AAAa,QAAA;;SACTL,OAAAA,CAAQL,YAAAA,GAAeW,MAAMC,IAAAA,CAAKP,OAAAA,CAAQL,aAAaa,OAAAA,EAAO,IAAM;AAC1E,KAAA,CAAA;AAED,IAAA,IAAA,CAAKV,YACHE,OAAAA,CAAQF,SAAAA,KACP,CAACW,CAAAA,KAAM,CAAA,EAAGA,EAAEC,MAAM,CAAA,EAAGD,CAAAA,CAAEE,SAAS,IAAIF,CAAAA,CAAEG,YAAAA,IAAgBH,EAAEE,SAAS,CAAA,EAAA,EAAKF,EAAEI,SAAS,CAAA,CAAA,CAAA,CAAA;AAEpF,IAAA,IAAA,CAAKd,OAAAA,GAAUe,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAA2C;MACrEC,SAAAA,EAAW;AAAC,QAAA,YAAA;AAAc,QAAA;;KAC5B,CAAA;AACF;;;;AAKAC,EAAAA,IAAAA,CAAK3B,GAAAA,EAA0B;AAE7B,IAAA,OAAO,IAAA,CAAKF,QAAAA,CAAS8B,MAAAA,GAAS,CAAA,EAAG;AAC/B,MAAA,IAAA,CAAK9B,QAAAA,CAAS+B,KAAG,EAAA;AACnB;AAEA,IAAA,IAAA,CAAK3B,WAAW4B,KAAAA,EAAK;AACrB,IAAA,IAAA,CAAK5B,UAAAA,CAAW6B,GAAAA,CAAI/B,GAAAA,CAAIgC,KAAAA,EAAO;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;KAAE,CAAA;AAC1D,IAAA,IAAA,CAAKlC,GAAAA,GAAMA,GAAAA;AAEX,IAAA,IAAA,CAAKmC,WAAAA,CAAYnC,IAAIQ,OAAO,CAAA;AAC9B;AAEQE,EAAAA,YAAAA,CAAa0B,KAAAA,EAAsC;AACzD,IAAA,IAAIA,KAAAA,KAAU,IAAA,IAAQA,KAAAA,KAAUC,MAAAA,EAAW;AAEzC,MAAA,OAAO,CAACC,GAAAA,KAAQC,OAAAA,CAAQC,GAAAA,CAAIF,GAAAA,CAAAA;AAC9B,KAAA,MAAA,IAAWF,iBAAiBK,iBAAAA,EAAQ;AAClC,MAAA,OAAO,CAACH,GAAAA,KAAQF,KAAAA,CAAMM,KAAAA,CAAMJ,GAAAA,CAAAA;KAC9B,MAAO;AACL,MAAA,OAAOF,KAAAA;AACT;AACF;AAEQD,EAAAA,WAAAA,CAAY3B,OAAAA,EAA6B;AAE/C,IAAA,IAAA,CAAKV,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CAAM,KAAA,EAAO,CAACC,CAAAA,EAAGC,KAAAA,KAAU,IAAA,CAAKC,UAAAA,CAAWD,KAAAA,CAAAA,EAAQ;MACzDzC,WAAAA,EAAa;AACf,KAAA,CAAA,CAAA;AAIF,IAAA,MAAM2C,mBAAAA,mBAAsB,MAAA,CAAA,OAAOC,IAAAA,EAAWC,IAAAA,KAAAA;AAC5C,MAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,sBAAAA,CAAAA,EAAa;AACzC,QAAA;AACF;AACA,MAAA,IAAI,CAACF,KAAKG,KAAAA,EAAO;AACf,QAAA;AACF;AAEA,MAAA,IAAA,CAAKN,WAAWG,IAAAA,CAAAA;AAChB,MAAA,IAAI,CAAC,IAAA,CAAKI,SAAAA,CAAUJ,IAAAA,CAAAA,EAAO;AACzB,QAAA;AACF;AACA,MAAA,IAAI,CAAC,KAAKvD,OAAAA,EAAS;AACjB,QAAA;AACF;AAEA,MAAA,MAAM2B,SAAAA,GAAYiC,iBAAAA,CAAWL,IAAAA,CAAKM,IAAI,CAAA;AACtC,MAAA,MAAM,IAAA,CAAK,aAAalC,SAAAA,CAAAA,CAAW,EAAWmC,IAAAA,CAAK,IAAA,EAAMR,MAAMC,IAAAA,CAAAA;KACjE,EAlB4B,qBAAA,CAAA;AAoB5B,IAAA,IAAA,CAAKpD,QAAAA,CAAS6C,IAAAA,CACZnC,OAAAA,CAAQoC,KAAAA,CACN,CAACE,KAAAA,KAAU;AAAC,MAAA,OAAA;AAAS,MAAA,SAAA;AAAW,MAAA,OAAA;AAAS,MAAA;MAAUY,QAAAA,CAASZ,KAAAA,CAAMU,IAAI,CAAA,EACtER,mBAAAA,EACA;MACE3C,WAAAA,EAAa,KAAA;MACbsD,UAAAA,EAAY,IAAA;AACZC,MAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,KAAA,CAAA,CAAA;AAKJ,IAAA,IAAI,KAAKD,WAAAA,EAAa;AACpB,MAAA,MAAMwD,iBAAAA,mBAAoB,MAAA,CAAA,OAAOZ,IAAAA,EAAWC,IAAAA,KAAAA;AAC1C,QAAA,IAAI,EAAEA,IAAAA,CAAKC,OAAAA,YAAmBC,sBAAAA,CAAAA,EAAa;AACzC,UAAA;AACF;AACA,QAAA,IAAIF,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAAA,KAAYA,OAAAA,EAAS;AACpC,UAAA,MAAMwC,mBAAAA,CAAoBC,MAAMC,IAAAA,CAAAA;AAChC,UAAA,IAAA,CAAKf,WAAAA,CAAYe,IAAAA,CAAKC,OAAAA,CAAQ3C,OAAO,CAAA;AACvC;OACF,EAR0B,mBAAA,CAAA;AAU1B,MAAA,IAAA,CAAKV,QAAAA,CAAS6C,KACZnC,OAAAA,CAAQoC,KAAAA,CAAM,CAACE,KAAAA,KAAUA,KAAAA,CAAMU,IAAAA,KAAS,OAAA,EAASK,iBAAAA,EAAmB;QAClExD,WAAAA,EAAa,IAAA;QACbsD,UAAAA,EAAY,IAAA;AACZC,QAAAA,QAAAA,EAAU,IAAA,CAAKtD;AACjB,OAAA,CAAA,CAAA;AAEJ;AACF;AAEQyC,EAAAA,UAAAA,CAAWG,IAAAA,EAAuB;AACxC,IAAA,IAAI,CAACA,IAAAA,CAAKG,KAAAA,EAAOrB,KAAAA,EAAO;AACtB,MAAA;AACF;AACA,IAAA,IAAI,KAAK9B,UAAAA,CAAW4D,GAAAA,CAAIZ,IAAAA,CAAKG,KAAAA,CAAMrB,KAAK,CAAA,EAAG;AACzC,MAAA;AACF;AACA,IAAA,IAAIkB,IAAAA,CAAKG,KAAAA,CAAMU,WAAAA,KAAgBb,IAAAA,CAAKG,MAAMrB,KAAAA,EAAO;AAC/C,MAAA;AACF;AAEA,IAAA,IAAIkB,IAAAA,CAAKG,MAAMU,WAAAA,EAAa;AAC1B,MAAA,MAAMC,OAAAA,GAAU,IAAA,CAAKV,SAAAA,CAAUJ,IAAAA,CAAAA;AAC/B,MAAA,MAAMe,cAAc,IAAA,CAAK/D,UAAAA,CAAWgE,IAAIhB,IAAAA,CAAKG,KAAAA,CAAMU,WAAW,CAAA,IAAK;QACjE9B,QAAAA,EAAU,CAAA;QACVC,QAAAA,EAAU;AACZ,OAAA;AACA,MAAA,IAAA,CAAKhC,UAAAA,CAAW6B,GAAAA,CAAImB,IAAAA,CAAKG,KAAAA,CAAMrB,KAAAA,EAAO;QACpCC,QAAAA,EAAUgC,WAAAA,CAAYhC,QAAAA,IAAY+B,OAAAA,GAAU,CAAA,GAAI,CAAA,CAAA;AAChD9B,QAAAA,QAAAA,EAAU+B,YAAY/B,QAAAA,GAAW;OACnC,CAAA;AACF;AACF;AAEQoB,EAAAA,SAAAA,CAAUJ,IAAAA,EAA0B;AAC1C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,sBAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,KAAA,MAAWtE,QAAAA,IAAY,KAAKA,QAAAA,EAAU;AACpC,MAAA,IAAIE,kBAAkBF,QAAAA,EAAU;AAC9B,QAAA,OAAO,KAAA;AACT;AACF;AAEA,IAAA,IAAI,IAAA,CAAKD,QAAAA,CAASgC,MAAAA,KAAW,CAAA,EAAG;AAC9B,MAAA,OAAO,IAAA;AACT;AAEA,IAAA,OAAO,KAAKhC,QAAAA,CAASwE,IAAAA,CAAK,CAACxE,QAAAA,KAAaG,kBAAkBH,QAAAA,CAAAA;AAC5D;AAEQyE,EAAAA,WAAAA,CAAYnB,IAAAA,EAAyB;AAC3C,IAAA,IAAInD,SAAcmD,IAAAA,CAAKC,OAAAA;AACvB,IAAA,IAAIpD,kBAAkBqD,sBAAAA,EAAY;AAChCrD,MAAAA,MAAAA,GAASA,MAAAA,CAAOoE,QAAAA;AAClB;AAEA,IAAA,MAAM/C,SAAAA,GAAYrB,OAAO,WAAA,CAAYyD,IAAAA;AACrC,IAAA,IAAIc,UAAAA,GAA4B,IAAA;AAEhC,IAAA,IAAIvE,MAAAA,YAAkBY,kBAAAA,IAAaZ,MAAAA,CAAOmD,IAAAA,EAAMM,IAAAA,EAAM;AACpDc,MAAAA,UAAAA,GAAavE,OAAOmD,IAAAA,CAAKM,IAAAA;AAC3B,KAAA,MAAA,IAAW,MAAA,IAAUzD,MAAAA,IAAU,OAAOA,MAAAA,CAAOyD,SAAS,QAAA,EAAU;AAC9Dc,MAAAA,UAAAA,GAAavE,MAAAA,CAAOyD,IAAAA;AACtB;AAEA,IAAA,IAAIrC,MAAAA,GAAS,EAAA;AACb,IAAA,KAAA,MAAW,CAACoD,IAAAA,EAAMC,UAAAA,KAAe,IAAA,CAAKpE,YAAAA,CAAaa,SAAO,EAAI;AAC5D,MAAA,IAAIlB,kBAAkBwE,IAAAA,EAAM;AAC1BpD,QAAAA,MAAAA,GAASqD,UAAAA;AACT,QAAA;AACF;AACF;AAEA,IAAA,MAAMpC,KAAAA,GAAkD;AACtDjB,MAAAA,MAAAA;AACAC,MAAAA,SAAAA;MACAC,YAAAA,EAAciD,UAAAA;AACdhD,MAAAA,SAAAA,EAAW4B,IAAAA,CAAKM;AAClB,KAAA;AAEA,IAAA,OAAO,IAAA,CAAKjD,UAAU6B,KAAAA,CAAAA;AACxB;AAEQqC,EAAAA,YAAAA,CAAavB,IAAAA,EAAyB;AAC5C,IAAA,IAAI,CAACA,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO,EAAA;AACT;AAEA,IAAA,MAAMqB,MAAAA,GAAS,IAAA,CAAKC,aAAAA,CAAczB,IAAAA,EAAM,MAAA,CAAA,CAAQjB,QAAAA;AAChD,IAAA,MAAM2C,YAAAA,GAAe,IAAA,CAAKD,aAAAA,CAAczB,IAAAA,EAAM,QAAA,CAAA,CAAUjB,QAAAA;AACxD,IAAA,MAAM4C,aAAaH,MAAAA,GAASE,YAAAA;AAE5B,IAAA,IAAIzD,MAAAA,GAAS,EAAA;AACbA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,MAAAA,CAAOF,YAAAA,GAAe,CAAA,CAAA;AAErC,IAAA,IAAI1B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWkB,MAAAA,EAAQ;AACnCvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEAA,IAAAA,MAAAA,IAAU,IAAA,CAAK2D,OAAOD,UAAAA,CAAAA;AAEtB,IAAA,IAAI3B,IAAAA,CAAKM,IAAAA,KAAS,OAAA,IAAWrC,MAAAA,IAAUuD,MAAAA,EAAQ;AAC7CvD,MAAAA,MAAAA,IAAU,GAAA;AACZ;AAEA,IAAA,IAAIA,MAAAA,EAAQ;AACVA,MAAAA,MAAAA,GAAS,GAAGA,MAAAA,CAAAA,CAAAA,CAAAA;AACd;AAEA,IAAA,MAAMqC,IAAAA,GAAO,IAAA,CAAKa,WAAAA,CAAYnB,IAAAA,CAAAA;AAC9B,IAAA,OAAO,CAAA,EAAG/B,MAAAA,CAAAA,EAASqC,IAAAA,CAAAA,EAAAA,CAAAA;AACrB;EAEQmB,aAAAA,CAAczB,IAAAA,EAAiBqB,OAA0B,MAAA,EAAoB;AACnF,IAAA,IAAI,CAACrB,KAAKG,KAAAA,EAAO;AACf,MAAA,OAAO;QAAEpB,QAAAA,EAAU,CAAA;QAAGC,QAAAA,EAAU;AAAE,OAAA;AACpC;AAEA,IAAA,MAAMF,KAAAA,GAAQuC,SAAS,QAAA,GAAWrB,IAAAA,CAAKG,MAAMU,WAAAA,IAAe,EAAA,GAAKb,KAAKG,KAAAA,CAAMrB,KAAAA;AAC5E,IAAA,OAAO,IAAA,CAAK9B,UAAAA,CAAWgE,GAAAA,CAAIlC,KAAAA,CAAAA,IAAU;MAAEC,QAAAA,EAAU,CAAA;MAAGC,QAAAA,EAAU;AAAE,KAAA;AAClE;AAEA,EAAA,MAAc6C,cAAcC,KAAAA,EAA6B;AACvD,IAAA,IAAIC,iBAAAA,CAAYD,KAAAA,CAAAA,EAAQ;AACtB,MAAA,OAAOE,OAAOF,KAAAA,CAAAA;AAChB;AAEA,IAAA,IAAIA,iBAAiBG,yBAAAA,EAAgB;AACnC,MAAA,OAAOH,MAAMI,OAAAA,EAAO;AACtB;AAEA,IAAA,MAAMC,UAAAA,GAAa,MAAMC,yBAAAA,CAAWC,SAAAA,CAAUP,KAAAA,CAAAA;AAC9C,IAAA,OAAOQ,IAAAA,CAAKC,SAAAA,CACV,MAAMH,yBAAAA,CAAWI,WAAAA,CAAYL,YAAY,EAAA,EAAI,IAAA,CAAA,kBAC5C,CAAA,MAAA;AACC,MAAA,MAAMM,YAAAA,uBAAmBC,GAAAA,CAAI;AAAC,QAAA,SAAA;AAAW,QAAA,UAAA;AAAY,QAAA,SAAA;AAAW,QAAA;AAAY,OAAA,CAAA;AAC5E,MAAA,OAAO,CAACC,KAAKb,MAAAA,KAAAA;AACX,QAAA,IAAIW,YAAAA,CAAa7B,GAAAA,CAAI+B,GAAAA,CAAAA,EAAM;AACzB,UAAA,OAAOxD,MAAAA;AACT;AACA,QAAA,IAAI2C,MAAAA,IAASA,kBAAiBnE,eAAAA,EAAM;AAClC,UAAA,OAAOmE,MAAAA,CAAMxB,IAAAA;AACf;AACA,QAAA,OAAOwB,MAAAA;AACT,OAAA;AACF,KAAA,GAAA,EACA,IAAA,CAAK/E,MAAAA,GAAS,CAAA,GAAIoC,MAAAA,CAAAA;AAEtB;EAEA,MAAcyD,eAAAA,CACZC,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQ3D,KAAK,CAAA,CAAA,CAAA;AAElE,IAAA,MAAM,IAAA,CAAK5B,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcI,iBAAAA,CACZL,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AAEjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,SAAA,EAAW;AACjCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcK,eAAAA,CACZN,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,GAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAAA,CAAAA,CAAAA;AAErD,IAAA,MAAM,IAAA,CAAKvF,OAAAA,CAAQyF,IAAAA,CAAK,OAAA,EAAS;AAC/BD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AAEA,IAAA,IAAA,CAAKnD,OAAOiG,OAAAA,CAAAA;AACd;EAEA,MAAcM,gBAAAA,CACZP,SACA7C,IAAAA,EACe;AACf,IAAA,MAAM/B,MAAAA,GAAS,IAAA,CAAKsD,YAAAA,CAAavB,IAAAA,CAAAA;AACjC,IAAA,MAAM8C,OAAAA,GAAU,CAAA,EAAG7E,MAAAA,CAAAA,EAAS,MAAM,IAAA,CAAK4D,aAAAA,CAAcgB,OAAAA,CAAQQ,KAAAA,IAASR,OAAAA,CAAQS,MAAM,CAAA,CAAA,CAAA;AACpF,IAAA,MAAM,IAAA,CAAKhG,OAAAA,CAAQyF,IAAAA,CAAK,QAAA,EAAU;AAChCD,MAAAA,OAAAA;MACAE,KAAAA,EAAO,IAAA,CAAKvB,cAAczB,IAAAA,CAAAA;MAC1BiD,MAAAA,EAAQ;AAACJ,QAAAA,OAAAA;AAAS7C,QAAAA;;KACpB,CAAA;AACF;AACF","file":"trajectory.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { Middleware, RunContext, RunContextCallbacks, RunInstance } from \"@/context.js\";\nimport { Callback, Emitter, EventMeta } from \"@/emitter/emitter.js\";\nimport { FrameworkError } from \"@/errors.js\";\nimport { Logger } from \"@/logger/logger.js\";\nimport { BaseAgent } from \"@/agents/base.js\";\nimport type { AnyConstructable } from \"@/internals/types.js\";\nimport { capitalize } from \"remeda\";\nimport { ChatModel } from \"@/backend/chat.js\";\nimport { Tool } from \"@/tools/base.js\";\nimport { Requirement } from \"@/agents/requirement/requirements/requirement.js\";\nimport { isPrimitive } from \"sequelize/lib/utils\";\nimport type { InferCallbackValue } from \"@/emitter/types.js\";\nimport { Serializer } from \"@/serializer/serializer.js\";\n\n/**\n * Information about how deep the given entity is in the execution tree.\n */\nexport interface TraceLevel {\n  /** Relative depth to the included (observed) elements */\n  relative: number;\n  /** Absolute depth from the root */\n  absolute: number;\n}\n\n/**\n * Input for custom formatter function\n */\nexport interface GlobalTrajectoryMiddlewareFormatterInput {\n  prefix: string;\n  className: string;\n  eventName: string;\n  instanceName: string | null;\n}\n\nexport interface GlobalTrajectoryMiddlewareCallbacks {\n  start: Callback<{\n    message: string;\n    level: TraceLevel;\n    origin: [InferCallbackValue<RunContextCallbacks[\"start\"]>, EventMeta];\n  }>;\n  success: Callback<{\n    message: string;\n    level: TraceLevel;\n    origin: [InferCallbackValue<RunContextCallbacks[\"success\"]>, EventMeta];\n  }>;\n  error: Callback<{\n    message: string;\n    level: TraceLevel;\n    origin: [InferCallbackValue<RunContextCallbacks[\"error\"]>, EventMeta];\n  }>;\n  finish: Callback<{\n    message: string;\n    level: TraceLevel;\n    origin: [InferCallbackValue<RunContextCallbacks[\"finish\"]>, EventMeta];\n  }>;\n}\n\ntype OutputTargetFn = (message: string) => void;\ntype OutputTarget = Logger | OutputTargetFn;\n\nexport interface GlobalTrajectoryMiddlewareOptions {\n  /** Specify output target: 'console', Logger instance, custom function, or null to disable */\n  target?: OutputTarget;\n  /** List of classes to include in the trajectory */\n  included?: AnyConstructable[];\n  /** List of classes to exclude from the trajectory */\n  excluded?: AnyConstructable[];\n  /** Use pretty formatting for the trajectory */\n  pretty?: boolean;\n  /** Customize how instances of individual classes should be printed */\n  prefixByType?: Map<AnyConstructable, string>;\n  /** Enable/Disable the logging */\n  enabled?: boolean;\n  /** Whether to observe trajectories of nested run contexts */\n  matchNested?: boolean;\n  /** Defines a priority for registered events */\n  emitterPriority?: number;\n  /** Custom formatter function */\n  formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n}\n\n/**\n * Middleware for capturing and logging execution flow of agents, tools, and models.\n * Provides hierarchical visualization with indentation to show the call stack.\n */\nexport class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {\n  protected enabled: boolean;\n  protected included: AnyConstructable[];\n  protected excluded: AnyConstructable[];\n  protected cleanups: (() => void)[] = [];\n  protected target: (message: string) => void;\n  protected ctx: RunContext<T> | null = null;\n  protected pretty: boolean;\n  protected traceLevel = new Map<string, TraceLevel>();\n  protected prefixByType: Map<any, string>;\n  protected matchNested: boolean;\n  protected emitterPriority: number;\n  protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;\n  public readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;\n\n  constructor(options: GlobalTrajectoryMiddlewareOptions = {}) {\n    super();\n\n    this.enabled = options.enabled ?? true;\n    this.included = options.included ?? [];\n    this.excluded = options.excluded ?? [];\n    this.target = this.createTarget(options.target);\n    this.pretty = options.pretty ?? false;\n    this.matchNested = options.matchNested ?? true;\n    this.emitterPriority = options.emitterPriority ?? -1; // run later\n\n    // Default prefixes\n    this.prefixByType = new Map([\n      [BaseAgent, \"🤖 \"],\n      [ChatModel, \"💬 \"],\n      [Tool, \"🛠️ \"],\n      [Requirement, \"🔎 \"],\n      ...((options.prefixByType ? Array.from(options.prefixByType.entries()) : []) as any),\n    ]);\n\n    this.formatter =\n      options.formatter ??\n      ((x) => `${x.prefix}${x.className}[${x.instanceName || x.className}][${x.eventName}]`);\n\n    this.emitter = Emitter.root.child<GlobalTrajectoryMiddlewareCallbacks>({\n      namespace: [\"middleware\", \"globalTrajectory\"],\n    });\n  }\n\n  /**\n   * Bind the middleware to a run context\n   */\n  bind(ctx: RunContext<T>): void {\n    // Cleanup previous bindings\n    while (this.cleanups.length > 0) {\n      this.cleanups.pop()!();\n    }\n\n    this.traceLevel.clear();\n    this.traceLevel.set(ctx.runId, { relative: 0, absolute: 0 });\n    this.ctx = ctx;\n\n    this.bindEmitter(ctx.emitter);\n  }\n\n  private createTarget(input?: OutputTarget): OutputTargetFn {\n    if (input === null || input === undefined) {\n      // eslint-disable-next-line no-console\n      return (msg) => console.log(msg);\n    } else if (input instanceof Logger) {\n      return (msg) => input.debug(msg);\n    } else {\n      return input;\n    }\n  }\n\n  private bindEmitter(emitter: Emitter<any>): void {\n    // Track all events for trace ID logging\n    this.cleanups.push(\n      emitter.match(\"*.*\", (_, event) => this.logTraceId(event), {\n        matchNested: true,\n      }),\n    );\n\n    // Handle top-level events\n    const handleTopLevelEvent = async (data: any, meta: EventMeta) => {\n      if (!(meta.creator instanceof RunContext)) {\n        return;\n      }\n      if (!meta.trace) {\n        return;\n      }\n\n      this.logTraceId(meta);\n      if (!this.isAllowed(meta)) {\n        return;\n      }\n      if (!this.enabled) {\n        return;\n      }\n\n      const eventName = capitalize(meta.name) as \"Start\" | \"Success\" | \"Error\" | \"Finish\";\n      await this[`onInternal${eventName}` as const].call(this, data, meta);\n    };\n\n    this.cleanups.push(\n      emitter.match(\n        (event) => [\"start\", \"success\", \"error\", \"finish\"].includes(event.name),\n        handleTopLevelEvent,\n        {\n          matchNested: false,\n          isBlocking: true,\n          priority: this.emitterPriority,\n        },\n      ),\n    );\n\n    // Handle nested events if enabled\n    if (this.matchNested) {\n      const handleNestedEvent = async (data: any, meta: EventMeta) => {\n        if (!(meta.creator instanceof RunContext)) {\n          return;\n        }\n        if (meta.creator.emitter !== emitter) {\n          await handleTopLevelEvent(data, meta);\n          this.bindEmitter(meta.creator.emitter);\n        }\n      };\n\n      this.cleanups.push(\n        emitter.match((event) => event.name === \"start\", handleNestedEvent, {\n          matchNested: true,\n          isBlocking: true,\n          priority: this.emitterPriority,\n        }),\n      );\n    }\n  }\n\n  private logTraceId(meta: EventMeta): void {\n    if (!meta.trace?.runId) {\n      return;\n    }\n    if (this.traceLevel.has(meta.trace.runId)) {\n      return;\n    }\n    if (meta.trace.parentRunId === meta.trace.runId) {\n      return;\n    }\n\n    if (meta.trace.parentRunId) {\n      const allowed = this.isAllowed(meta);\n      const parentTrace = this.traceLevel.get(meta.trace.parentRunId) ?? {\n        relative: 0,\n        absolute: 0,\n      };\n      this.traceLevel.set(meta.trace.runId, {\n        relative: parentTrace.relative + (allowed ? 1 : 0),\n        absolute: parentTrace.absolute + 1,\n      });\n    }\n  }\n\n  private isAllowed(meta: EventMeta): boolean {\n    let target: any = meta.creator;\n    if (target instanceof RunContext) {\n      target = target.instance;\n    }\n\n    for (const excluded of this.excluded) {\n      if (target instanceof excluded) {\n        return false;\n      }\n    }\n\n    if (this.included.length === 0) {\n      return true;\n    }\n\n    return this.included.some((included) => target instanceof included);\n  }\n\n  private extractName(meta: EventMeta): string {\n    let target: any = meta.creator;\n    if (target instanceof RunContext) {\n      target = target.instance;\n    }\n\n    const className = target.constructor.name;\n    let targetName: string | null = null;\n\n    if (target instanceof BaseAgent && target.meta?.name) {\n      targetName = target.meta.name;\n    } else if (\"name\" in target && typeof target.name === \"string\") {\n      targetName = target.name;\n    }\n\n    let prefix = \"\";\n    for (const [type, typePrefix] of this.prefixByType.entries()) {\n      if (target instanceof type) {\n        prefix = typePrefix;\n        break;\n      }\n    }\n\n    const input: GlobalTrajectoryMiddlewareFormatterInput = {\n      prefix,\n      className,\n      instanceName: targetName,\n      eventName: meta.name,\n    };\n\n    return this.formatter(input);\n  }\n\n  private formatPrefix(meta: EventMeta): string {\n    if (!meta.trace) {\n      return \"\";\n    }\n\n    const indent = this.getTraceLevel(meta, \"self\").relative;\n    const indentParent = this.getTraceLevel(meta, \"parent\").relative;\n    const indentDiff = indent - indentParent;\n\n    let prefix = \"\";\n    prefix += \"  \".repeat(indentParent * 2);\n\n    if (meta.name !== \"start\" && indent) {\n      prefix += \"<\";\n    }\n\n    prefix += \"--\".repeat(indentDiff);\n\n    if (meta.name === \"start\" && prefix && indent) {\n      prefix += \">\";\n    }\n\n    if (prefix) {\n      prefix = `${prefix} `;\n    }\n\n    const name = this.extractName(meta);\n    return `${prefix}${name}: `;\n  }\n\n  private getTraceLevel(meta: EventMeta, type: \"self\" | \"parent\" = \"self\"): TraceLevel {\n    if (!meta.trace) {\n      return { relative: 0, absolute: 0 };\n    }\n\n    const runId = type === \"parent\" ? meta.trace.parentRunId || \"\" : meta.trace.runId;\n    return this.traceLevel.get(runId) ?? { relative: 0, absolute: 0 };\n  }\n\n  private async formatPayload(value: any): Promise<string> {\n    if (isPrimitive(value)) {\n      return String(value);\n    }\n\n    if (value instanceof FrameworkError) {\n      return value.explain();\n    }\n\n    const serialized = await Serializer.serialize(value);\n    return JSON.stringify(\n      await Serializer.deserialize(serialized, [], true),\n      (() => {\n        const excludedKeys = new Set([\"emitter\", \"cleanups\", \"creator\", \"listeners\"]);\n        return (key, value) => {\n          if (excludedKeys.has(key)) {\n            return undefined;\n          }\n          if (value && value instanceof Tool) {\n            return value.name;\n          }\n          return value;\n        };\n      })(),\n      this.pretty ? 2 : undefined,\n    );\n  }\n\n  private async onInternalStart(\n    payload: InferCallbackValue<RunContextCallbacks[\"start\"]>,\n    meta: EventMeta,\n  ): Promise<void> {\n    const prefix = this.formatPrefix(meta);\n    const message = `${prefix}${await this.formatPayload(payload.input)}`;\n\n    await this.emitter.emit(\"start\", {\n      message,\n      level: this.getTraceLevel(meta),\n      origin: [payload, meta],\n    });\n\n    this.target(message);\n  }\n\n  private async onInternalSuccess(\n    payload: InferCallbackValue<RunContextCallbacks[\"success\"]>,\n    meta: EventMeta,\n  ): Promise<void> {\n    const prefix = this.formatPrefix(meta);\n    // TODO: change to payload.output once available\n    const message = `${prefix}${await this.formatPayload(payload)}`;\n\n    await this.emitter.emit(\"success\", {\n      message,\n      level: this.getTraceLevel(meta),\n      origin: [payload, meta],\n    });\n\n    this.target(message);\n  }\n\n  private async onInternalError(\n    payload: InferCallbackValue<RunContextCallbacks[\"error\"]>,\n    meta: EventMeta,\n  ): Promise<void> {\n    const prefix = this.formatPrefix(meta);\n    const message = `${prefix}${await this.formatPayload(payload)}`;\n\n    await this.emitter.emit(\"error\", {\n      message,\n      level: this.getTraceLevel(meta),\n      origin: [payload, meta],\n    });\n\n    this.target(message);\n  }\n\n  private async onInternalFinish(\n    payload: InferCallbackValue<RunContextCallbacks[\"finish\"]>,\n    meta: EventMeta,\n  ): Promise<void> {\n    const prefix = this.formatPrefix(meta);\n    const message = `${prefix}${await this.formatPayload(payload.error || payload.output)}`;\n    await this.emitter.emit(\"finish\", {\n      message,\n      level: this.getTraceLevel(meta),\n      origin: [payload, meta],\n    });\n  }\n}\n"]}

package/dist/middleware/trajectory.d.cts

@@ -0,0 +1,115 @@
+import { RunContextCallbacks, RunInstance, Middleware, RunContext } from '../context.cjs';
+import { C as Callback, I as InferCallbackValue, a as EventMeta, E as Emitter } from '../emitter-CGKd_-kK.cjs';
+import { Logger } from '../logger/logger.cjs';
+import { AnyConstructable } from '../internals/types.cjs';
+import '../internals/serializable.cjs';
+import '../internals/helpers/guards.cjs';
+import '../internals/helpers/promise.cjs';
+import '../errors.cjs';
+import 'pino';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+/**
+ * Information about how deep the given entity is in the execution tree.
+ */
+interface TraceLevel {
+    /** Relative depth to the included (observed) elements */
+    relative: number;
+    /** Absolute depth from the root */
+    absolute: number;
+}
+/**
+ * Input for custom formatter function
+ */
+interface GlobalTrajectoryMiddlewareFormatterInput {
+    prefix: string;
+    className: string;
+    eventName: string;
+    instanceName: string | null;
+}
+interface GlobalTrajectoryMiddlewareCallbacks {
+    start: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["start"]>, EventMeta];
+    }>;
+    success: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["success"]>, EventMeta];
+    }>;
+    error: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["error"]>, EventMeta];
+    }>;
+    finish: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["finish"]>, EventMeta];
+    }>;
+}
+type OutputTargetFn = (message: string) => void;
+type OutputTarget = Logger | OutputTargetFn;
+interface GlobalTrajectoryMiddlewareOptions {
+    /** Specify output target: 'console', Logger instance, custom function, or null to disable */
+    target?: OutputTarget;
+    /** List of classes to include in the trajectory */
+    included?: AnyConstructable[];
+    /** List of classes to exclude from the trajectory */
+    excluded?: AnyConstructable[];
+    /** Use pretty formatting for the trajectory */
+    pretty?: boolean;
+    /** Customize how instances of individual classes should be printed */
+    prefixByType?: Map<AnyConstructable, string>;
+    /** Enable/Disable the logging */
+    enabled?: boolean;
+    /** Whether to observe trajectories of nested run contexts */
+    matchNested?: boolean;
+    /** Defines a priority for registered events */
+    emitterPriority?: number;
+    /** Custom formatter function */
+    formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;
+}
+/**
+ * Middleware for capturing and logging execution flow of agents, tools, and models.
+ * Provides hierarchical visualization with indentation to show the call stack.
+ */
+declare class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {
+    protected enabled: boolean;
+    protected included: AnyConstructable[];
+    protected excluded: AnyConstructable[];
+    protected cleanups: (() => void)[];
+    protected target: (message: string) => void;
+    protected ctx: RunContext<T> | null;
+    protected pretty: boolean;
+    protected traceLevel: Map<string, TraceLevel>;
+    protected prefixByType: Map<any, string>;
+    protected matchNested: boolean;
+    protected emitterPriority: number;
+    protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;
+    readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;
+    constructor(options?: GlobalTrajectoryMiddlewareOptions);
+    /**
+     * Bind the middleware to a run context
+     */
+    bind(ctx: RunContext<T>): void;
+    private createTarget;
+    private bindEmitter;
+    private logTraceId;
+    private isAllowed;
+    private extractName;
+    private formatPrefix;
+    private getTraceLevel;
+    private formatPayload;
+    private onInternalStart;
+    private onInternalSuccess;
+    private onInternalError;
+    private onInternalFinish;
+}
+
+export { GlobalTrajectoryMiddleware, type GlobalTrajectoryMiddlewareCallbacks, type GlobalTrajectoryMiddlewareFormatterInput, type GlobalTrajectoryMiddlewareOptions, type TraceLevel };

package/dist/middleware/trajectory.d.ts

@@ -0,0 +1,115 @@
+import { RunContextCallbacks, RunInstance, Middleware, RunContext } from '../context.js';
+import { C as Callback, I as InferCallbackValue, a as EventMeta, E as Emitter } from '../emitter-CN7I4uSV.js';
+import { Logger } from '../logger/logger.js';
+import { AnyConstructable } from '../internals/types.js';
+import '../internals/serializable.js';
+import '../internals/helpers/guards.js';
+import '../internals/helpers/promise.js';
+import '../errors.js';
+import 'pino';
+
+/**
+ * Copyright 2025 © BeeAI a Series of LF Projects, LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+/**
+ * Information about how deep the given entity is in the execution tree.
+ */
+interface TraceLevel {
+    /** Relative depth to the included (observed) elements */
+    relative: number;
+    /** Absolute depth from the root */
+    absolute: number;
+}
+/**
+ * Input for custom formatter function
+ */
+interface GlobalTrajectoryMiddlewareFormatterInput {
+    prefix: string;
+    className: string;
+    eventName: string;
+    instanceName: string | null;
+}
+interface GlobalTrajectoryMiddlewareCallbacks {
+    start: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["start"]>, EventMeta];
+    }>;
+    success: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["success"]>, EventMeta];
+    }>;
+    error: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["error"]>, EventMeta];
+    }>;
+    finish: Callback<{
+        message: string;
+        level: TraceLevel;
+        origin: [InferCallbackValue<RunContextCallbacks["finish"]>, EventMeta];
+    }>;
+}
+type OutputTargetFn = (message: string) => void;
+type OutputTarget = Logger | OutputTargetFn;
+interface GlobalTrajectoryMiddlewareOptions {
+    /** Specify output target: 'console', Logger instance, custom function, or null to disable */
+    target?: OutputTarget;
+    /** List of classes to include in the trajectory */
+    included?: AnyConstructable[];
+    /** List of classes to exclude from the trajectory */
+    excluded?: AnyConstructable[];
+    /** Use pretty formatting for the trajectory */
+    pretty?: boolean;
+    /** Customize how instances of individual classes should be printed */
+    prefixByType?: Map<AnyConstructable, string>;
+    /** Enable/Disable the logging */
+    enabled?: boolean;
+    /** Whether to observe trajectories of nested run contexts */
+    matchNested?: boolean;
+    /** Defines a priority for registered events */
+    emitterPriority?: number;
+    /** Custom formatter function */
+    formatter?: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;
+}
+/**
+ * Middleware for capturing and logging execution flow of agents, tools, and models.
+ * Provides hierarchical visualization with indentation to show the call stack.
+ */
+declare class GlobalTrajectoryMiddleware<T extends RunInstance = any> extends Middleware<T> {
+    protected enabled: boolean;
+    protected included: AnyConstructable[];
+    protected excluded: AnyConstructable[];
+    protected cleanups: (() => void)[];
+    protected target: (message: string) => void;
+    protected ctx: RunContext<T> | null;
+    protected pretty: boolean;
+    protected traceLevel: Map<string, TraceLevel>;
+    protected prefixByType: Map<any, string>;
+    protected matchNested: boolean;
+    protected emitterPriority: number;
+    protected formatter: (input: GlobalTrajectoryMiddlewareFormatterInput) => string;
+    readonly emitter: Emitter<GlobalTrajectoryMiddlewareCallbacks>;
+    constructor(options?: GlobalTrajectoryMiddlewareOptions);
+    /**
+     * Bind the middleware to a run context
+     */
+    bind(ctx: RunContext<T>): void;
+    private createTarget;
+    private bindEmitter;
+    private logTraceId;
+    private isAllowed;
+    private extractName;
+    private formatPrefix;
+    private getTraceLevel;
+    private formatPayload;
+    private onInternalStart;
+    private onInternalSuccess;
+    private onInternalError;
+    private onInternalFinish;
+}
+
+export { GlobalTrajectoryMiddleware, type GlobalTrajectoryMiddlewareCallbacks, type GlobalTrajectoryMiddlewareFormatterInput, type GlobalTrajectoryMiddlewareOptions, type TraceLevel };