@livekit/agents 0.4.5 → 0.5.0

package/dist/job.js

@@ -1,205 +1,202 @@
-import { ParticipantKind, RoomEvent, TrackKind } from '@livekit/rtc-node';
-import { log } from './log.js';
-/** Which tracks, if any, should the agent automatically subscribe to? */
-export var AutoSubscribe;
-(function (AutoSubscribe) {
-    AutoSubscribe[AutoSubscribe["SUBSCRIBE_ALL"] = 0] = "SUBSCRIBE_ALL";
-    AutoSubscribe[AutoSubscribe["SUBSCRIBE_NONE"] = 1] = "SUBSCRIBE_NONE";
-    AutoSubscribe[AutoSubscribe["VIDEO_ONLY"] = 2] = "VIDEO_ONLY";
-    AutoSubscribe[AutoSubscribe["AUDIO_ONLY"] = 3] = "AUDIO_ONLY";
-})(AutoSubscribe || (AutoSubscribe = {}));
-/** Attempted to add a function callback, but the function already exists. */
-export class FunctionExistsError extends Error {
-    constructor(msg) {
-        super(msg);
-        Object.setPrototypeOf(this, new.target.prototype);
-    }
+import { ParticipantKind, RoomEvent, TrackKind } from "@livekit/rtc-node";
+import { log } from "./log.js";
+var AutoSubscribe = /* @__PURE__ */ ((AutoSubscribe2) => {
+  AutoSubscribe2[AutoSubscribe2["SUBSCRIBE_ALL"] = 0] = "SUBSCRIBE_ALL";
+  AutoSubscribe2[AutoSubscribe2["SUBSCRIBE_NONE"] = 1] = "SUBSCRIBE_NONE";
+  AutoSubscribe2[AutoSubscribe2["VIDEO_ONLY"] = 2] = "VIDEO_ONLY";
+  AutoSubscribe2[AutoSubscribe2["AUDIO_ONLY"] = 3] = "AUDIO_ONLY";
+  return AutoSubscribe2;
+})(AutoSubscribe || {});
+class FunctionExistsError extends Error {
+  constructor(msg) {
+    super(msg);
+    Object.setPrototypeOf(this, new.target.prototype);
+  }
 }
-/** The job and environment context as seen by the agent, accessible by the entrypoint function. */
-export class JobContext {
-    #proc;
-    #info;
-    #room;
-    #onConnect;
-    #onShutdown;
-    /** @internal */
-    shutdownCallbacks = [];
-    #participantEntrypoints = [];
-    #participantTasks = {};
-    #logger;
-    constructor(proc, info, room, onConnect, onShutdown) {
-        this.#proc = proc;
-        this.#info = info;
-        this.#room = room;
-        this.#onConnect = onConnect;
-        this.#onShutdown = onShutdown;
-        this.onParticipantConnected = this.onParticipantConnected.bind(this);
-        this.#room.on(RoomEvent.ParticipantConnected, this.onParticipantConnected);
-        this.#logger = log().child({ info: this.#info });
-    }
-    get proc() {
-        return this.#proc;
-    }
-    get job() {
-        return this.#info.job;
-    }
-    /** @returns The room the agent was called into */
-    get room() {
-        return this.#room;
-    }
-    /** @returns The agent's participant if connected to the room, otherwise `undefined` */
-    get agent() {
-        return this.#room.localParticipant;
-    }
-    /** Adds a promise to be awaited when {@link JobContext.shutdown | shutdown} is called. */
-    addShutdownCallback(callback) {
-        this.shutdownCallbacks.push(callback);
-    }
-    async waitForParticipant(identity) {
-        if (!this.#room.isConnected) {
-            throw new Error('room is not connected');
+class JobContext {
+  #proc;
+  #info;
+  #room;
+  #onConnect;
+  #onShutdown;
+  /** @internal */
+  shutdownCallbacks = [];
+  #participantEntrypoints = [];
+  #participantTasks = {};
+  #logger;
+  constructor(proc, info, room, onConnect, onShutdown) {
+    this.#proc = proc;
+    this.#info = info;
+    this.#room = room;
+    this.#onConnect = onConnect;
+    this.#onShutdown = onShutdown;
+    this.onParticipantConnected = this.onParticipantConnected.bind(this);
+    this.#room.on(RoomEvent.ParticipantConnected, this.onParticipantConnected);
+    this.#logger = log().child({ info: this.#info });
+  }
+  get proc() {
+    return this.#proc;
+  }
+  get job() {
+    return this.#info.job;
+  }
+  /** @returns The room the agent was called into */
+  get room() {
+    return this.#room;
+  }
+  /** @returns The agent's participant if connected to the room, otherwise `undefined` */
+  get agent() {
+    return this.#room.localParticipant;
+  }
+  /** Adds a promise to be awaited when {@link JobContext.shutdown | shutdown} is called. */
+  addShutdownCallback(callback) {
+    this.shutdownCallbacks.push(callback);
+  }
+  async waitForParticipant(identity) {
+    if (!this.#room.isConnected) {
+      throw new Error("room is not connected");
+    }
+    for (const p of this.#room.remoteParticipants.values()) {
+      if ((!identity || p.identity === identity) && p.info.kind != ParticipantKind.AGENT) {
+        return p;
+      }
+    }
+    return new Promise((resolve, reject) => {
+      const onParticipantConnected = (participant) => {
+        if ((!identity || participant.identity === identity) && participant.info.kind != ParticipantKind.AGENT) {
+          clearHandlers();
+          resolve(participant);
         }
-        for (const p of this.#room.remoteParticipants.values()) {
-            if ((!identity || p.identity === identity) && p.info.kind != ParticipantKind.AGENT) {
-                return p;
-            }
-        }
-        return new Promise((resolve, reject) => {
-            const onParticipantConnected = (participant) => {
-                if ((!identity || participant.identity === identity) &&
-                    participant.info.kind != ParticipantKind.AGENT) {
-                    clearHandlers();
-                    resolve(participant);
-                }
-            };
-            const onDisconnected = () => {
-                clearHandlers();
-                reject(new Error('Room disconnected while waiting for participant'));
-            };
-            const clearHandlers = () => {
-                this.#room.off(RoomEvent.ParticipantConnected, onParticipantConnected);
-                this.#room.off(RoomEvent.Disconnected, onDisconnected);
-            };
-            this.#room.on(RoomEvent.ParticipantConnected, onParticipantConnected);
-            this.#room.on(RoomEvent.Disconnected, onDisconnected);
+      };
+      const onDisconnected = () => {
+        clearHandlers();
+        reject(new Error("Room disconnected while waiting for participant"));
+      };
+      const clearHandlers = () => {
+        this.#room.off(RoomEvent.ParticipantConnected, onParticipantConnected);
+        this.#room.off(RoomEvent.Disconnected, onDisconnected);
+      };
+      this.#room.on(RoomEvent.ParticipantConnected, onParticipantConnected);
+      this.#room.on(RoomEvent.Disconnected, onDisconnected);
+    });
+  }
+  /**
+   * Connects the agent to the room.
+   *
+   * @remarks
+   * It is recommended to run this command as early in the function as possible, as executing it
+   * later may cause noticeable delay between user and agent joins.
+   *
+   * @see {@link https://github.com/livekit/node-sdks/tree/main/packages/livekit-rtc#readme |
+   * @livekit/rtc-node} for more information about the parameters.
+   */
+  async connect(e2ee, autoSubscribe = 0 /* SUBSCRIBE_ALL */, rtcConfig) {
+    const opts = {
+      e2ee,
+      autoSubscribe: autoSubscribe == 0 /* SUBSCRIBE_ALL */,
+      rtcConfig,
+      dynacast: false
+    };
+    await this.#room.connect(this.#info.url, this.#info.token, opts);
+    this.#onConnect();
+    this.#room.remoteParticipants.forEach(this.onParticipantConnected);
+    if ([3 /* AUDIO_ONLY */, 2 /* VIDEO_ONLY */].includes(autoSubscribe)) {
+      this.#room.remoteParticipants.forEach((p) => {
+        p.trackPublications.forEach((pub) => {
+          if (autoSubscribe === 3 /* AUDIO_ONLY */ && pub.kind === TrackKind.KIND_AUDIO || autoSubscribe === 2 /* VIDEO_ONLY */ && pub.kind === TrackKind.KIND_VIDEO) {
+            pub.setSubscribed(true);
+          }
         });
-    }
-    /**
-     * Connects the agent to the room.
-     *
-     * @remarks
-     * It is recommended to run this command as early in the function as possible, as executing it
-     * later may cause noticeable delay between user and agent joins.
-     *
-     * @see {@link https://github.com/livekit/node-sdks/tree/main/packages/livekit-rtc#readme |
-     * @livekit/rtc-node} for more information about the parameters.
-     */
-    async connect(e2ee, autoSubscribe = AutoSubscribe.SUBSCRIBE_ALL, rtcConfig) {
-        const opts = {
-            e2ee,
-            autoSubscribe: autoSubscribe == AutoSubscribe.SUBSCRIBE_ALL,
-            rtcConfig,
-            dynacast: false,
-        };
-        await this.#room.connect(this.#info.url, this.#info.token, opts);
-        this.#onConnect();
-        this.#room.remoteParticipants.forEach(this.onParticipantConnected);
-        if ([AutoSubscribe.AUDIO_ONLY, AutoSubscribe.VIDEO_ONLY].includes(autoSubscribe)) {
-            this.#room.remoteParticipants.forEach((p) => {
-                p.trackPublications.forEach((pub) => {
-                    if ((autoSubscribe === AutoSubscribe.AUDIO_ONLY && pub.kind === TrackKind.KIND_AUDIO) ||
-                        (autoSubscribe === AutoSubscribe.VIDEO_ONLY && pub.kind === TrackKind.KIND_VIDEO)) {
-                        pub.setSubscribed(true);
-                    }
-                });
-            });
-        }
-    }
-    /**
-     * Gracefully shuts down the job, and runs all shutdown promises.
-     *
-     * @param reason - Optional reason for shutdown
-     */
-    shutdown(reason = '') {
-        this.#onShutdown(reason);
-    }
-    /** @internal */
-    onParticipantConnected(p) {
-        for (const callback of this.#participantEntrypoints) {
-            if (this.#participantTasks[p.identity]?.callback == callback) {
-                this.#logger.warn('a participant has joined before a prior prticipant task matching the same identity has finished:', p.identity);
-            }
-            const result = callback(this, p);
-            result.finally(() => delete this.#participantTasks[p.identity]);
-            this.#participantTasks[p.identity] = { callback, result };
-        }
-    }
-    /**
-     * Adds a promise to be awaited whenever a new participant joins the room.
-     *
-     * @throws {@link FunctionExistsError} if an entrypoint already exists
-     */
-    addParticipantEntrypoint(callback) {
-        if (this.#participantEntrypoints.includes(callback)) {
-            throw new FunctionExistsError('entrypoints cannot be added more than once');
-        }
-        this.#participantEntrypoints.push(callback);
-    }
+      });
+    }
+  }
+  /**
+   * Gracefully shuts down the job, and runs all shutdown promises.
+   *
+   * @param reason - Optional reason for shutdown
+   */
+  shutdown(reason = "") {
+    this.#onShutdown(reason);
+  }
+  /** @internal */
+  onParticipantConnected(p) {
+    var _a;
+    for (const callback of this.#participantEntrypoints) {
+      if (((_a = this.#participantTasks[p.identity]) == null ? void 0 : _a.callback) == callback) {
+        this.#logger.warn(
+          "a participant has joined before a prior prticipant task matching the same identity has finished:",
+          p.identity
+        );
+      }
+      const result = callback(this, p);
+      result.finally(() => delete this.#participantTasks[p.identity]);
+      this.#participantTasks[p.identity] = { callback, result };
+    }
+  }
+  /**
+   * Adds a promise to be awaited whenever a new participant joins the room.
+   *
+   * @throws {@link FunctionExistsError} if an entrypoint already exists
+   */
+  addParticipantEntrypoint(callback) {
+    if (this.#participantEntrypoints.includes(callback)) {
+      throw new FunctionExistsError("entrypoints cannot be added more than once");
+    }
+    this.#participantEntrypoints.push(callback);
+  }
 }
-export class JobProcess {
-    #pid = process.pid;
-    userData = {};
-    get pid() {
-        return this.#pid;
-    }
+class JobProcess {
+  #pid = process.pid;
+  userData = {};
+  get pid() {
+    return this.#pid;
+  }
 }
-/**
- * A request sent by the server to spawn a new agent job.
- *
- * @remarks
- * For most applications, this is best left to the default, which simply accepts the job and
- * handles the logic inside the entrypoint function. This class is useful for vetting which
- * requests should fill idle processes and which should be outright rejected.
- */
-export class JobRequest {
-    #job;
-    #onReject;
-    #onAccept;
-    /** @internal */
-    constructor(job, onReject, onAccept) {
-        this.#job = job;
-        this.#onReject = onReject;
-        this.#onAccept = onAccept;
-    }
-    /** @returns The ID of the job, set by the LiveKit server */
-    get id() {
-        return this.#job.id;
-    }
-    /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */
-    get job() {
-        return this.#job;
-    }
-    /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */
-    get room() {
-        return this.#job.room;
-    }
-    /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */
-    get publisher() {
-        return this.#job.participant;
-    }
-    /** @returns The agent's name, as set in {@link WorkerOptions} */
-    get agentName() {
-        return this.#job.agentName;
-    }
-    /** Rejects the job. */
-    async reject() {
-        await this.#onReject();
-    }
-    /** Accepts the job, launching it on an idle child process. */
-    async accept(name = '', identity = '', metadata = '') {
-        if (identity === '')
-            identity = 'agent-' + this.id;
-        this.#onAccept({ name, identity, metadata });
-    }
+class JobRequest {
+  #job;
+  #onReject;
+  #onAccept;
+  /** @internal */
+  constructor(job, onReject, onAccept) {
+    this.#job = job;
+    this.#onReject = onReject;
+    this.#onAccept = onAccept;
+  }
+  /** @returns The ID of the job, set by the LiveKit server */
+  get id() {
+    return this.#job.id;
+  }
+  /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */
+  get job() {
+    return this.#job;
+  }
+  /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */
+  get room() {
+    return this.#job.room;
+  }
+  /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */
+  get publisher() {
+    return this.#job.participant;
+  }
+  /** @returns The agent's name, as set in {@link WorkerOptions} */
+  get agentName() {
+    return this.#job.agentName;
+  }
+  /** Rejects the job. */
+  async reject() {
+    await this.#onReject();
+  }
+  /** Accepts the job, launching it on an idle child process. */
+  async accept(name = "", identity = "", metadata = "") {
+    if (identity === "") identity = "agent-" + this.id;
+    this.#onAccept({ name, identity, metadata });
+  }
 }
+export {
+  AutoSubscribe,
+  FunctionExistsError,
+  JobContext,
+  JobProcess,
+  JobRequest
+};
 //# sourceMappingURL=job.js.map

package/dist/job.js.map

@@ -1 +1 @@
-{"version":3,"file":"job.js","sourceRoot":"","sources":["../src/job.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAE1E,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAE/B,yEAAyE;AACzE,MAAM,CAAN,IAAY,aAKX;AALD,WAAY,aAAa;IACvB,mEAAa,CAAA;IACb,qEAAc,CAAA;IACd,6DAAU,CAAA;IACV,6DAAU,CAAA;AACZ,CAAC,EALW,aAAa,KAAb,aAAa,QAKxB;AAeD,6EAA6E;AAC7E,MAAM,OAAO,mBAAoB,SAAQ,KAAK;IAC5C,YAAY,GAAY;QACtB,KAAK,CAAC,GAAG,CAAC,CAAC;QACX,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACpD,CAAC;CACF;AAED,mGAAmG;AACnG,MAAM,OAAO,UAAU;IACrB,KAAK,CAAa;IAClB,KAAK,CAAiB;IACtB,KAAK,CAAO;IACZ,UAAU,CAAa;IACvB,WAAW,CAAsB;IACjC,gBAAgB;IAChB,iBAAiB,GAA4B,EAAE,CAAC;IAChD,uBAAuB,GAAiE,EAAE,CAAC;IAC3F,iBAAiB,GAKb,EAAE,CAAC;IACP,OAAO,CAAS;IAEhB,YACE,IAAgB,EAChB,IAAoB,EACpB,IAAU,EACV,SAAqB,EACrB,UAA+B;QAE/B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,oBAAoB,EAAE,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC3E,IAAI,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC;IACxB,CAAC;IAED,kDAAkD;IAClD,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,uFAAuF;IACvF,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,CAAC;IACrC,CAAC;IAED,0FAA0F;IAC1F,mBAAmB,CAAC,QAA6B;QAC/C,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,kBAAkB,CAAC,QAAiB;QACxC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,MAAM,EAAE,EAAE,CAAC;YACvD,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,QAAQ,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,IAAI,eAAe,CAAC,KAAK,EAAE,CAAC;gBACnF,OAAO,CAAC,CAAC;YACX,CAAC;QACH,CAAC;QAED,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,sBAAsB,GAAG,CAAC,WAA8B,EAAE,EAAE;gBAChE,IACE,CAAC,CAAC,QAAQ,IAAI,WAAW,CAAC,QAAQ,KAAK,QAAQ,CAAC;oBAChD,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,eAAe,CAAC,KAAK,EAC9C,CAAC;oBACD,aAAa,EAAE,CAAC;oBAChB,OAAO,CAAC,WAAW,CAAC,CAAC;gBACvB,CAAC;YACH,CAAC,CAAC;YACF,MAAM,cAAc,GAAG,GAAG,EAAE;gBAC1B,aAAa,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC,CAAC;YACvE,CAAC,CAAC;YAEF,MAAM,aAAa,GAAG,GAAG,EAAE;gBACzB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,oBAAoB,EAAE,sBAAsB,CAAC,CAAC;gBACvE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;YACzD,CAAC,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,oBAAoB,EAAE,sBAAsB,CAAC,CAAC;YACtE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;;;;;;;;OASG;IACH,KAAK,CAAC,OAAO,CACX,IAAkB,EAClB,gBAA+B,aAAa,CAAC,aAAa,EAC1D,SAA4B;QAE5B,MAAM,IAAI,GAAG;YACX,IAAI;YACJ,aAAa,EAAE,aAAa,IAAI,aAAa,CAAC,aAAa;YAC3D,SAAS;YACT,QAAQ,EAAE,KAAK;SAChB,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACjE,IAAI,CAAC,UAAU,EAAE,CAAC;QAElB,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAEnE,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,UAAU,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;YACjF,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBAC1C,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;oBAClC,IACE,CAAC,aAAa,KAAK,aAAa,CAAC,UAAU,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,UAAU,CAAC;wBACjF,CAAC,aAAa,KAAK,aAAa,CAAC,UAAU,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,CAAC,UAAU,CAAC,EACjF,CAAC;wBACD,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;oBAC1B,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,QAAQ,CAAC,MAAM,GAAG,EAAE;QAClB,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;IAED,gBAAgB;IAChB,sBAAsB,CAAC,CAAoB;QACzC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACpD,IAAI,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,QAAQ,IAAI,QAAQ,EAAE,CAAC;gBAC7D,IAAI,CAAC,OAAO,CAAC,IAAI,CACf,kGAAkG,EAClG,CAAC,CAAC,QAAQ,CACX,CAAC;YACJ,CAAC;YACD,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YACjC,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;YAChE,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QAC5D,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,wBAAwB,CAAC,QAAkE;QACzF,IAAI,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpD,MAAM,IAAI,mBAAmB,CAAC,4CAA4C,CAAC,CAAC;QAC9E,CAAC;QAED,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC;CACF;AAED,MAAM,OAAO,UAAU;IACrB,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC;IACnB,QAAQ,GAA8B,EAAE,CAAC;IAEzC,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;CACF;AAED;;;;;;;GAOG;AACH,MAAM,OAAO,UAAU;IACrB,IAAI,CAAY;IAChB,SAAS,CAAsB;IAC/B,SAAS,CAA8C;IAEvD,gBAAgB;IAChB,YACE,GAAc,EACd,QAA6B,EAC7B,QAAqD;QAErD,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC;QAChB,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;IAC5B,CAAC;IAED,4DAA4D;IAC5D,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IACtB,CAAC;IAED,uFAAuF;IACvF,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,uFAAuF;IACvF,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC;IACxB,CAAC;IAED,uFAAuF;IACvF,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC;IAC/B,CAAC;IAED,iEAAiE;IACjE,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC;IAC7B,CAAC;IAED,uBAAuB;IACvB,KAAK,CAAC,MAAM;QACV,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;IACzB,CAAC;IAED,8DAA8D;IAC9D,KAAK,CAAC,MAAM,CAAC,IAAI,GAAG,EAAE,EAAE,QAAQ,GAAG,EAAE,EAAE,QAAQ,GAAG,EAAE;QAClD,IAAI,QAAQ,KAAK,EAAE;YAAE,QAAQ,GAAG,QAAQ,GAAG,IAAI,CAAC,EAAE,CAAC;QAEnD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC/C,CAAC;CACF"}
+{"version":3,"sources":["../src/job.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport type * as proto from '@livekit/protocol';\nimport type {\n  E2EEOptions,\n  LocalParticipant,\n  RemoteParticipant,\n  Room,\n  RtcConfiguration,\n} from '@livekit/rtc-node';\nimport { ParticipantKind, RoomEvent, TrackKind } from '@livekit/rtc-node';\nimport type { Logger } from 'pino';\nimport { log } from './log.js';\n\n/** Which tracks, if any, should the agent automatically subscribe to? */\nexport enum AutoSubscribe {\n  SUBSCRIBE_ALL,\n  SUBSCRIBE_NONE,\n  VIDEO_ONLY,\n  AUDIO_ONLY,\n}\n\nexport type JobAcceptArguments = {\n  name: string;\n  identity: string;\n  metadata: string;\n};\n\nexport type RunningJobInfo = {\n  acceptArguments: JobAcceptArguments;\n  job: proto.Job;\n  url: string;\n  token: string;\n};\n\n/** Attempted to add a function callback, but the function already exists. */\nexport class FunctionExistsError extends Error {\n  constructor(msg?: string) {\n    super(msg);\n    Object.setPrototypeOf(this, new.target.prototype);\n  }\n}\n\n/** The job and environment context as seen by the agent, accessible by the entrypoint function. */\nexport class JobContext {\n  #proc: JobProcess;\n  #info: RunningJobInfo;\n  #room: Room;\n  #onConnect: () => void;\n  #onShutdown: (s: string) => void;\n  /** @internal */\n  shutdownCallbacks: (() => Promise<void>)[] = [];\n  #participantEntrypoints: ((job: JobContext, p: RemoteParticipant) => Promise<void>)[] = [];\n  #participantTasks: {\n    [id: string]: {\n      callback: (job: JobContext, p: RemoteParticipant) => Promise<void>;\n      result: Promise<void>;\n    };\n  } = {};\n  #logger: Logger;\n\n  constructor(\n    proc: JobProcess,\n    info: RunningJobInfo,\n    room: Room,\n    onConnect: () => void,\n    onShutdown: (s: string) => void,\n  ) {\n    this.#proc = proc;\n    this.#info = info;\n    this.#room = room;\n    this.#onConnect = onConnect;\n    this.#onShutdown = onShutdown;\n    this.onParticipantConnected = this.onParticipantConnected.bind(this);\n    this.#room.on(RoomEvent.ParticipantConnected, this.onParticipantConnected);\n    this.#logger = log().child({ info: this.#info });\n  }\n\n  get proc(): JobProcess {\n    return this.#proc;\n  }\n\n  get job(): proto.Job {\n    return this.#info.job;\n  }\n\n  /** @returns The room the agent was called into */\n  get room(): Room {\n    return this.#room;\n  }\n\n  /** @returns The agent's participant if connected to the room, otherwise `undefined` */\n  get agent(): LocalParticipant | undefined {\n    return this.#room.localParticipant;\n  }\n\n  /** Adds a promise to be awaited when {@link JobContext.shutdown | shutdown} is called. */\n  addShutdownCallback(callback: () => Promise<void>) {\n    this.shutdownCallbacks.push(callback);\n  }\n\n  async waitForParticipant(identity?: string): Promise<RemoteParticipant> {\n    if (!this.#room.isConnected) {\n      throw new Error('room is not connected');\n    }\n\n    for (const p of this.#room.remoteParticipants.values()) {\n      if ((!identity || p.identity === identity) && p.info.kind != ParticipantKind.AGENT) {\n        return p;\n      }\n    }\n\n    return new Promise((resolve, reject) => {\n      const onParticipantConnected = (participant: RemoteParticipant) => {\n        if (\n          (!identity || participant.identity === identity) &&\n          participant.info.kind != ParticipantKind.AGENT\n        ) {\n          clearHandlers();\n          resolve(participant);\n        }\n      };\n      const onDisconnected = () => {\n        clearHandlers();\n        reject(new Error('Room disconnected while waiting for participant'));\n      };\n\n      const clearHandlers = () => {\n        this.#room.off(RoomEvent.ParticipantConnected, onParticipantConnected);\n        this.#room.off(RoomEvent.Disconnected, onDisconnected);\n      };\n\n      this.#room.on(RoomEvent.ParticipantConnected, onParticipantConnected);\n      this.#room.on(RoomEvent.Disconnected, onDisconnected);\n    });\n  }\n\n  /**\n   * Connects the agent to the room.\n   *\n   * @remarks\n   * It is recommended to run this command as early in the function as possible, as executing it\n   * later may cause noticeable delay between user and agent joins.\n   *\n   * @see {@link https://github.com/livekit/node-sdks/tree/main/packages/livekit-rtc#readme |\n   * @livekit/rtc-node} for more information about the parameters.\n   */\n  async connect(\n    e2ee?: E2EEOptions,\n    autoSubscribe: AutoSubscribe = AutoSubscribe.SUBSCRIBE_ALL,\n    rtcConfig?: RtcConfiguration,\n  ) {\n    const opts = {\n      e2ee,\n      autoSubscribe: autoSubscribe == AutoSubscribe.SUBSCRIBE_ALL,\n      rtcConfig,\n      dynacast: false,\n    };\n\n    await this.#room.connect(this.#info.url, this.#info.token, opts);\n    this.#onConnect();\n\n    this.#room.remoteParticipants.forEach(this.onParticipantConnected);\n\n    if ([AutoSubscribe.AUDIO_ONLY, AutoSubscribe.VIDEO_ONLY].includes(autoSubscribe)) {\n      this.#room.remoteParticipants.forEach((p) => {\n        p.trackPublications.forEach((pub) => {\n          if (\n            (autoSubscribe === AutoSubscribe.AUDIO_ONLY && pub.kind === TrackKind.KIND_AUDIO) ||\n            (autoSubscribe === AutoSubscribe.VIDEO_ONLY && pub.kind === TrackKind.KIND_VIDEO)\n          ) {\n            pub.setSubscribed(true);\n          }\n        });\n      });\n    }\n  }\n\n  /**\n   * Gracefully shuts down the job, and runs all shutdown promises.\n   *\n   * @param reason - Optional reason for shutdown\n   */\n  shutdown(reason = '') {\n    this.#onShutdown(reason);\n  }\n\n  /** @internal */\n  onParticipantConnected(p: RemoteParticipant) {\n    for (const callback of this.#participantEntrypoints) {\n      if (this.#participantTasks[p.identity]?.callback == callback) {\n        this.#logger.warn(\n          'a participant has joined before a prior prticipant task matching the same identity has finished:',\n          p.identity,\n        );\n      }\n      const result = callback(this, p);\n      result.finally(() => delete this.#participantTasks[p.identity]);\n      this.#participantTasks[p.identity] = { callback, result };\n    }\n  }\n\n  /**\n   * Adds a promise to be awaited whenever a new participant joins the room.\n   *\n   * @throws {@link FunctionExistsError} if an entrypoint already exists\n   */\n  addParticipantEntrypoint(callback: (job: JobContext, p: RemoteParticipant) => Promise<void>) {\n    if (this.#participantEntrypoints.includes(callback)) {\n      throw new FunctionExistsError('entrypoints cannot be added more than once');\n    }\n\n    this.#participantEntrypoints.push(callback);\n  }\n}\n\nexport class JobProcess {\n  #pid = process.pid;\n  userData: { [id: string]: unknown } = {};\n\n  get pid(): number {\n    return this.#pid;\n  }\n}\n\n/**\n * A request sent by the server to spawn a new agent job.\n *\n * @remarks\n * For most applications, this is best left to the default, which simply accepts the job and\n * handles the logic inside the entrypoint function. This class is useful for vetting which\n * requests should fill idle processes and which should be outright rejected.\n */\nexport class JobRequest {\n  #job: proto.Job;\n  #onReject: () => Promise<void>;\n  #onAccept: (args: JobAcceptArguments) => Promise<void>;\n\n  /** @internal */\n  constructor(\n    job: proto.Job,\n    onReject: () => Promise<void>,\n    onAccept: (args: JobAcceptArguments) => Promise<void>,\n  ) {\n    this.#job = job;\n    this.#onReject = onReject;\n    this.#onAccept = onAccept;\n  }\n\n  /** @returns The ID of the job, set by the LiveKit server */\n  get id(): string {\n    return this.#job.id;\n  }\n\n  /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */\n  get job(): proto.Job {\n    return this.#job;\n  }\n\n  /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */\n  get room(): proto.Room | undefined {\n    return this.#job.room;\n  }\n\n  /** @see {@link https://www.npmjs.com/package/@livekit/protocol | @livekit/protocol} */\n  get publisher(): proto.ParticipantInfo | undefined {\n    return this.#job.participant;\n  }\n\n  /** @returns The agent's name, as set in {@link WorkerOptions} */\n  get agentName(): string {\n    return this.#job.agentName;\n  }\n\n  /** Rejects the job. */\n  async reject() {\n    await this.#onReject();\n  }\n\n  /** Accepts the job, launching it on an idle child process. */\n  async accept(name = '', identity = '', metadata = '') {\n    if (identity === '') identity = 'agent-' + this.id;\n\n    this.#onAccept({ name, identity, metadata });\n  }\n}\n"],"mappings":"AAWA,SAAS,iBAAiB,WAAW,iBAAiB;AAEtD,SAAS,WAAW;AAGb,IAAK,gBAAL,kBAAKA,mBAAL;AACL,EAAAA,8BAAA;AACA,EAAAA,8BAAA;AACA,EAAAA,8BAAA;AACA,EAAAA,8BAAA;AAJU,SAAAA;AAAA,GAAA;AAqBL,MAAM,4BAA4B,MAAM;AAAA,EAC7C,YAAY,KAAc;AACxB,UAAM,GAAG;AACT,WAAO,eAAe,MAAM,WAAW,SAAS;AAAA,EAClD;AACF;AAGO,MAAM,WAAW;AAAA,EACtB;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAEA,oBAA6C,CAAC;AAAA,EAC9C,0BAAwF,CAAC;AAAA,EACzF,oBAKI,CAAC;AAAA,EACL;AAAA,EAEA,YACE,MACA,MACA,MACA,WACA,YACA;AACA,SAAK,QAAQ;AACb,SAAK,QAAQ;AACb,SAAK,QAAQ;AACb,SAAK,aAAa;AAClB,SAAK,cAAc;AACnB,SAAK,yBAAyB,KAAK,uBAAuB,KAAK,IAAI;AACnE,SAAK,MAAM,GAAG,UAAU,sBAAsB,KAAK,sBAAsB;AACzE,SAAK,UAAU,IAAI,EAAE,MAAM,EAAE,MAAM,KAAK,MAAM,CAAC;AAAA,EACjD;AAAA,EAEA,IAAI,OAAmB;AACrB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,MAAiB;AACnB,WAAO,KAAK,MAAM;AAAA,EACpB;AAAA;AAAA,EAGA,IAAI,OAAa;AACf,WAAO,KAAK;AAAA,EACd;AAAA;AAAA,EAGA,IAAI,QAAsC;AACxC,WAAO,KAAK,MAAM;AAAA,EACpB;AAAA;AAAA,EAGA,oBAAoB,UAA+B;AACjD,SAAK,kBAAkB,KAAK,QAAQ;AAAA,EACtC;AAAA,EAEA,MAAM,mBAAmB,UAA+C;AACtE,QAAI,CAAC,KAAK,MAAM,aAAa;AAC3B,YAAM,IAAI,MAAM,uBAAuB;AAAA,IACzC;AAEA,eAAW,KAAK,KAAK,MAAM,mBAAmB,OAAO,GAAG;AACtD,WAAK,CAAC,YAAY,EAAE,aAAa,aAAa,EAAE,KAAK,QAAQ,gBAAgB,OAAO;AAClF,eAAO;AAAA,MACT;AAAA,IACF;AAEA,WAAO,IAAI,QAAQ,CAAC,SAAS,WAAW;AACtC,YAAM,yBAAyB,CAAC,gBAAmC;AACjE,aACG,CAAC,YAAY,YAAY,aAAa,aACvC,YAAY,KAAK,QAAQ,gBAAgB,OACzC;AACA,wBAAc;AACd,kBAAQ,WAAW;AAAA,QACrB;AAAA,MACF;AACA,YAAM,iBAAiB,MAAM;AAC3B,sBAAc;AACd,eAAO,IAAI,MAAM,iDAAiD,CAAC;AAAA,MACrE;AAEA,YAAM,gBAAgB,MAAM;AAC1B,aAAK,MAAM,IAAI,UAAU,sBAAsB,sBAAsB;AACrE,aAAK,MAAM,IAAI,UAAU,cAAc,cAAc;AAAA,MACvD;AAEA,WAAK,MAAM,GAAG,UAAU,sBAAsB,sBAAsB;AACpE,WAAK,MAAM,GAAG,UAAU,cAAc,cAAc;AAAA,IACtD,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,QACJ,MACA,gBAA+B,uBAC/B,WACA;AACA,UAAM,OAAO;AAAA,MACX;AAAA,MACA,eAAe,iBAAiB;AAAA,MAChC;AAAA,MACA,UAAU;AAAA,IACZ;AAEA,UAAM,KAAK,MAAM,QAAQ,KAAK,MAAM,KAAK,KAAK,MAAM,OAAO,IAAI;AAC/D,SAAK,WAAW;AAEhB,SAAK,MAAM,mBAAmB,QAAQ,KAAK,sBAAsB;AAEjE,QAAI,CAAC,oBAA0B,kBAAwB,EAAE,SAAS,aAAa,GAAG;AAChF,WAAK,MAAM,mBAAmB,QAAQ,CAAC,MAAM;AAC3C,UAAE,kBAAkB,QAAQ,CAAC,QAAQ;AACnC,cACG,kBAAkB,sBAA4B,IAAI,SAAS,UAAU,cACrE,kBAAkB,sBAA4B,IAAI,SAAS,UAAU,YACtE;AACA,gBAAI,cAAc,IAAI;AAAA,UACxB;AAAA,QACF,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,SAAS,SAAS,IAAI;AACpB,SAAK,YAAY,MAAM;AAAA,EACzB;AAAA;AAAA,EAGA,uBAAuB,GAAsB;AA7L/C;AA8LI,eAAW,YAAY,KAAK,yBAAyB;AACnD,YAAI,UAAK,kBAAkB,EAAE,QAAQ,MAAjC,mBAAoC,aAAY,UAAU;AAC5D,aAAK,QAAQ;AAAA,UACX;AAAA,UACA,EAAE;AAAA,QACJ;AAAA,MACF;AACA,YAAM,SAAS,SAAS,MAAM,CAAC;AAC/B,aAAO,QAAQ,MAAM,OAAO,KAAK,kBAAkB,EAAE,QAAQ,CAAC;AAC9D,WAAK,kBAAkB,EAAE,QAAQ,IAAI,EAAE,UAAU,OAAO;AAAA,IAC1D;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,yBAAyB,UAAoE;AAC3F,QAAI,KAAK,wBAAwB,SAAS,QAAQ,GAAG;AACnD,YAAM,IAAI,oBAAoB,4CAA4C;AAAA,IAC5E;AAEA,SAAK,wBAAwB,KAAK,QAAQ;AAAA,EAC5C;AACF;AAEO,MAAM,WAAW;AAAA,EACtB,OAAO,QAAQ;AAAA,EACf,WAAsC,CAAC;AAAA,EAEvC,IAAI,MAAc;AAChB,WAAO,KAAK;AAAA,EACd;AACF;AAUO,MAAM,WAAW;AAAA,EACtB;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAGA,YACE,KACA,UACA,UACA;AACA,SAAK,OAAO;AACZ,SAAK,YAAY;AACjB,SAAK,YAAY;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,KAAa;AACf,WAAO,KAAK,KAAK;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,MAAiB;AACnB,WAAO,KAAK;AAAA,EACd;AAAA;AAAA,EAGA,IAAI,OAA+B;AACjC,WAAO,KAAK,KAAK;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,YAA+C;AACjD,WAAO,KAAK,KAAK;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,YAAoB;AACtB,WAAO,KAAK,KAAK;AAAA,EACnB;AAAA;AAAA,EAGA,MAAM,SAAS;AACb,UAAM,KAAK,UAAU;AAAA,EACvB;AAAA;AAAA,EAGA,MAAM,OAAO,OAAO,IAAI,WAAW,IAAI,WAAW,IAAI;AACpD,QAAI,aAAa,GAAI,YAAW,WAAW,KAAK;AAEhD,SAAK,UAAU,EAAE,MAAM,UAAU,SAAS,CAAC;AAAA,EAC7C;AACF;","names":["AutoSubscribe"]}

package/dist/llm/chat_context.cjs

@@ -0,0 +1,131 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var chat_context_exports = {};
+__export(chat_context_exports, {
+  ChatContext: () => ChatContext,
+  ChatMessage: () => ChatMessage,
+  ChatRole: () => ChatRole
+});
+module.exports = __toCommonJS(chat_context_exports);
+var ChatRole = /* @__PURE__ */ ((ChatRole2) => {
+  ChatRole2[ChatRole2["SYSTEM"] = 0] = "SYSTEM";
+  ChatRole2[ChatRole2["USER"] = 1] = "USER";
+  ChatRole2[ChatRole2["ASSISTANT"] = 2] = "ASSISTANT";
+  ChatRole2[ChatRole2["TOOL"] = 3] = "TOOL";
+  return ChatRole2;
+})(ChatRole || {});
+const defaultCreateChatMessage = {
+  text: "",
+  images: [],
+  role: 0 /* SYSTEM */
+};
+class ChatMessage {
+  role;
+  id;
+  name;
+  content;
+  toolCalls;
+  toolCallId;
+  toolException;
+  /** @internal */
+  constructor({
+    role,
+    id,
+    name,
+    content,
+    toolCalls,
+    toolCallId,
+    toolException
+  }) {
+    this.role = role;
+    this.id = id;
+    this.name = name;
+    this.content = content;
+    this.toolCalls = toolCalls;
+    this.toolCallId = toolCallId;
+    this.toolException = toolException;
+  }
+  static createToolFromFunctionResult(func) {
+    if (!func.result && !func.error) {
+      throw new TypeError("CallableFunctionResult must include result or error");
+    }
+    return new ChatMessage({
+      role: 3 /* TOOL */,
+      name: func.name,
+      content: func.result || `Error: ${func.error}`,
+      toolCallId: func.toolCallId,
+      toolException: func.error
+    });
+  }
+  static createToolCalls(toolCalls, text = "") {
+    return new ChatMessage({
+      role: 2 /* ASSISTANT */,
+      toolCalls,
+      content: text
+    });
+  }
+  static create(options) {
+    const { text, images, role } = { ...defaultCreateChatMessage, ...options };
+    if (!images.length) {
+      return new ChatMessage({
+        role,
+        content: text
+      });
+    } else {
+      return new ChatMessage({
+        role,
+        content: [...text ? [text] : [], ...images]
+      });
+    }
+  }
+  /** Returns a structured clone of this message. */
+  copy() {
+    return new ChatMessage({
+      role: this.role,
+      id: this.id,
+      name: this.name,
+      content: this.content,
+      toolCalls: this.toolCalls,
+      toolCallId: this.toolCallId,
+      toolException: this.toolException
+    });
+  }
+}
+class ChatContext {
+  messages = [];
+  metadata = {};
+  append(msg) {
+    this.messages.push(ChatMessage.create(msg));
+    return this;
+  }
+  /** Returns a structured clone of this context. */
+  copy() {
+    const ctx = new ChatContext();
+    ctx.messages.push(...this.messages.map((msg) => msg.copy()));
+    ctx.metadata = structuredClone(this.metadata);
+    return ctx;
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ChatContext,
+  ChatMessage,
+  ChatRole
+});
+//# sourceMappingURL=chat_context.cjs.map

package/dist/llm/chat_context.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/llm/chat_context.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport type { AudioFrame, VideoFrame } from '@livekit/rtc-node';\nimport type { CallableFunctionResult, FunctionCallInfo } from './function_context.js';\n\nexport enum ChatRole {\n  SYSTEM,\n  USER,\n  ASSISTANT,\n  TOOL,\n}\n\nexport interface ChatImage {\n  image: string | VideoFrame;\n  inferenceWidth?: number;\n  inferenceHeight?: number;\n  /**\n   * @internal\n   * Used by LLM implementations to store a processed version of the image for later use.\n   */\n  cache: { [id: string | number | symbol]: any };\n}\n\nexport interface ChatAudio {\n  frame: AudioFrame | AudioFrame[];\n}\n\nexport type ChatContent = string | ChatImage | ChatAudio;\n\nconst defaultCreateChatMessage = {\n  text: '',\n  images: [],\n  role: ChatRole.SYSTEM,\n};\n\nexport class ChatMessage {\n  readonly role: ChatRole;\n  readonly id?: string;\n  readonly name?: string;\n  readonly content?: ChatContent | ChatContent[];\n  readonly toolCalls?: FunctionCallInfo[];\n  readonly toolCallId?: string;\n  readonly toolException?: Error;\n\n  /** @internal */\n  constructor({\n    role,\n    id,\n    name,\n    content,\n    toolCalls,\n    toolCallId,\n    toolException,\n  }: {\n    role: ChatRole;\n    id?: string;\n    name?: string;\n    content?: ChatContent | ChatContent[];\n    toolCalls?: FunctionCallInfo[];\n    toolCallId?: string;\n    toolException?: Error;\n  }) {\n    this.role = role;\n    this.id = id;\n    this.name = name;\n    this.content = content;\n    this.toolCalls = toolCalls;\n    this.toolCallId = toolCallId;\n    this.toolException = toolException;\n  }\n\n  static createToolFromFunctionResult(func: CallableFunctionResult): ChatMessage {\n    if (!func.result && !func.error) {\n      throw new TypeError('CallableFunctionResult must include result or error');\n    }\n\n    return new ChatMessage({\n      role: ChatRole.TOOL,\n      name: func.name,\n      content: func.result || `Error: ${func.error}`,\n      toolCallId: func.toolCallId,\n      toolException: func.error,\n    });\n  }\n\n  static createToolCalls(toolCalls: FunctionCallInfo[], text = '') {\n    return new ChatMessage({\n      role: ChatRole.ASSISTANT,\n      toolCalls,\n      content: text,\n    });\n  }\n\n  static create(\n    options: Partial<{\n      text?: string;\n      images: ChatImage[];\n      role: ChatRole;\n    }>,\n  ): ChatMessage {\n    const { text, images, role } = { ...defaultCreateChatMessage, ...options };\n\n    if (!images.length) {\n      return new ChatMessage({\n        role,\n        content: text,\n      });\n    } else {\n      return new ChatMessage({\n        role,\n        content: [...(text ? [text] : []), ...images],\n      });\n    }\n  }\n\n  /** Returns a structured clone of this message. */\n  copy(): ChatMessage {\n    return new ChatMessage({\n      role: this.role,\n      id: this.id,\n      name: this.name,\n      content: this.content,\n      toolCalls: this.toolCalls,\n      toolCallId: this.toolCallId,\n      toolException: this.toolException,\n    });\n  }\n}\n\nexport class ChatContext {\n  messages: ChatMessage[] = [];\n  metadata: { [id: string]: any } = {};\n\n  append(msg: { text?: string; images?: ChatImage[]; role: ChatRole }): ChatContext {\n    this.messages.push(ChatMessage.create(msg));\n    return this;\n  }\n\n  /** Returns a structured clone of this context. */\n  copy(): ChatContext {\n    const ctx = new ChatContext();\n    ctx.messages.push(...this.messages.map((msg) => msg.copy()));\n    ctx.metadata = structuredClone(this.metadata);\n    return ctx;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAMO,IAAK,WAAL,kBAAKA,cAAL;AACL,EAAAA,oBAAA;AACA,EAAAA,oBAAA;AACA,EAAAA,oBAAA;AACA,EAAAA,oBAAA;AAJU,SAAAA;AAAA,GAAA;AAwBZ,MAAM,2BAA2B;AAAA,EAC/B,MAAM;AAAA,EACN,QAAQ,CAAC;AAAA,EACT,MAAM;AACR;AAEO,MAAM,YAAY;AAAA,EACd;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA;AAAA,EAGT,YAAY;AAAA,IACV;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAQG;AACD,SAAK,OAAO;AACZ,SAAK,KAAK;AACV,SAAK,OAAO;AACZ,SAAK,UAAU;AACf,SAAK,YAAY;AACjB,SAAK,aAAa;AAClB,SAAK,gBAAgB;AAAA,EACvB;AAAA,EAEA,OAAO,6BAA6B,MAA2C;AAC7E,QAAI,CAAC,KAAK,UAAU,CAAC,KAAK,OAAO;AAC/B,YAAM,IAAI,UAAU,qDAAqD;AAAA,IAC3E;AAEA,WAAO,IAAI,YAAY;AAAA,MACrB,MAAM;AAAA,MACN,MAAM,KAAK;AAAA,MACX,SAAS,KAAK,UAAU,UAAU,KAAK,KAAK;AAAA,MAC5C,YAAY,KAAK;AAAA,MACjB,eAAe,KAAK;AAAA,IACtB,CAAC;AAAA,EACH;AAAA,EAEA,OAAO,gBAAgB,WAA+B,OAAO,IAAI;AAC/D,WAAO,IAAI,YAAY;AAAA,MACrB,MAAM;AAAA,MACN;AAAA,MACA,SAAS;AAAA,IACX,CAAC;AAAA,EACH;AAAA,EAEA,OAAO,OACL,SAKa;AACb,UAAM,EAAE,MAAM,QAAQ,KAAK,IAAI,EAAE,GAAG,0BAA0B,GAAG,QAAQ;AAEzE,QAAI,CAAC,OAAO,QAAQ;AAClB,aAAO,IAAI,YAAY;AAAA,QACrB;AAAA,QACA,SAAS;AAAA,MACX,CAAC;AAAA,IACH,OAAO;AACL,aAAO,IAAI,YAAY;AAAA,QACrB;AAAA,QACA,SAAS,CAAC,GAAI,OAAO,CAAC,IAAI,IAAI,CAAC,GAAI,GAAG,MAAM;AAAA,MAC9C,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA,EAGA,OAAoB;AAClB,WAAO,IAAI,YAAY;AAAA,MACrB,MAAM,KAAK;AAAA,MACX,IAAI,KAAK;AAAA,MACT,MAAM,KAAK;AAAA,MACX,SAAS,KAAK;AAAA,MACd,WAAW,KAAK;AAAA,MAChB,YAAY,KAAK;AAAA,MACjB,eAAe,KAAK;AAAA,IACtB,CAAC;AAAA,EACH;AACF;AAEO,MAAM,YAAY;AAAA,EACvB,WAA0B,CAAC;AAAA,EAC3B,WAAkC,CAAC;AAAA,EAEnC,OAAO,KAA2E;AAChF,SAAK,SAAS,KAAK,YAAY,OAAO,GAAG,CAAC;AAC1C,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,OAAoB;AAClB,UAAM,MAAM,IAAI,YAAY;AAC5B,QAAI,SAAS,KAAK,GAAG,KAAK,SAAS,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC,CAAC;AAC3D,QAAI,WAAW,gBAAgB,KAAK,QAAQ;AAC5C,WAAO;AAAA,EACT;AACF;","names":["ChatRole"]}