@livekit/agents 0.1.0

package/dist/utils.d.ts

@@ -0,0 +1,4 @@
+import { AudioFrame } from '@livekit/rtc-node';
+export type AudioBuffer = AudioFrame[] | AudioFrame;
+export declare const mergeFrames: (buffer: AudioBuffer) => AudioFrame;
+//# sourceMappingURL=utils.d.ts.map

package/dist/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,MAAM,MAAM,WAAW,GAAG,UAAU,EAAE,GAAG,UAAU,CAAC;AAEpD,eAAO,MAAM,WAAW,WAAY,WAAW,KAAG,UA6BjD,CAAC"}

package/dist/utils.js

@@ -0,0 +1,29 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { AudioFrame } from '@livekit/rtc-node';
+export const mergeFrames = (buffer) => {
+    if (Array.isArray(buffer)) {
+        buffer = buffer;
+        if (buffer.length == 0) {
+            throw new TypeError('buffer is empty');
+        }
+        const sampleRate = buffer[0].sampleRate;
+        const channels = buffer[0].channels;
+        let samplesPerChannel = 0;
+        let data = new Uint16Array();
+        for (const frame of buffer) {
+            if (frame.sampleRate !== sampleRate) {
+                throw new TypeError('sample rate mismatch');
+            }
+            if (frame.channels !== channels) {
+                throw new TypeError('channel count mismatch');
+            }
+            data = new Uint16Array([...data, ...frame.data]);
+            samplesPerChannel += frame.samplesPerChannel;
+        }
+        return new AudioFrame(data, sampleRate, channels, samplesPerChannel);
+    }
+    return buffer;
+};
+//# sourceMappingURL=utils.js.map

package/dist/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../src/utils.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AACtC,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAI/C,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,MAAmB,EAAc,EAAE;IAC7D,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,MAAM,GAAG,MAAsB,CAAC;QAChC,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,SAAS,CAAC,iBAAiB,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QACxC,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;QACpC,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAC1B,IAAI,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC;QAE7B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,KAAK,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACpC,MAAM,IAAI,SAAS,CAAC,sBAAsB,CAAC,CAAC;YAC9C,CAAC;YAED,IAAI,KAAK,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAChC,MAAM,IAAI,SAAS,CAAC,wBAAwB,CAAC,CAAC;YAChD,CAAC;YAED,IAAI,GAAG,IAAI,WAAW,CAAC,CAAC,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;YACjD,iBAAiB,IAAI,KAAK,CAAC,iBAAiB,CAAC;QAC/C,CAAC;QAED,OAAO,IAAI,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,iBAAiB,CAAC,CAAC;IACvE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC,CAAC"}

package/dist/vad.d.ts

@@ -0,0 +1,28 @@
+import type { AudioFrame } from '@livekit/rtc-node';
+export declare enum VADEventType {
+    START_OF_SPEECH = 1,
+    SPEAKING = 2,
+    END_OF_SPEECH = 3
+}
+export interface VADEvent {
+    type: VADEventType;
+    samplesIndex: number;
+    duration: number;
+    speech: AudioFrame[];
+}
+export declare abstract class VAD {
+    abstract stream({ minSpeakingDuration, minSilenceDuration, paddingDuration, sampleRate, maxBufferedSpeech, }: {
+        minSpeakingDuration: number;
+        minSilenceDuration: number;
+        paddingDuration: number;
+        sampleRate: number;
+        maxBufferedSpeech: number;
+    }): VADStream;
+}
+export declare abstract class VADStream implements IterableIterator<VADEvent> {
+    abstract pushFrame(frame: AudioFrame): void;
+    abstract close(wait: boolean): Promise<void>;
+    abstract next(): IteratorResult<VADEvent>;
+    [Symbol.iterator](): VADStream;
+}
+//# sourceMappingURL=vad.d.ts.map

package/dist/vad.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vad.d.ts","sourceRoot":"","sources":["../src/vad.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEpD,oBAAY,YAAY;IACtB,eAAe,IAAI;IACnB,QAAQ,IAAI;IACZ,aAAa,IAAI;CAClB;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,YAAY,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,UAAU,EAAE,CAAC;CACtB;AAED,8BAAsB,GAAG;IACvB,QAAQ,CAAC,MAAM,CAAC,EACd,mBAAmB,EACnB,kBAAkB,EAClB,eAAe,EACf,UAAU,EACV,iBAAiB,GAClB,EAAE;QACD,mBAAmB,EAAE,MAAM,CAAC;QAC5B,kBAAkB,EAAE,MAAM,CAAC;QAC3B,eAAe,EAAE,MAAM,CAAC;QACxB,UAAU,EAAE,MAAM,CAAC;QACnB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,GAAG,SAAS;CACd;AAED,8BAAsB,SAAU,YAAW,gBAAgB,CAAC,QAAQ,CAAC;IACnE,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI;IAC3C,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAC5C,QAAQ,CAAC,IAAI,IAAI,cAAc,CAAC,QAAQ,CAAC;IACzC,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,SAAS;CAG/B"}

package/dist/vad.js

@@ -0,0 +1,14 @@
+export var VADEventType;
+(function (VADEventType) {
+    VADEventType[VADEventType["START_OF_SPEECH"] = 1] = "START_OF_SPEECH";
+    VADEventType[VADEventType["SPEAKING"] = 2] = "SPEAKING";
+    VADEventType[VADEventType["END_OF_SPEECH"] = 3] = "END_OF_SPEECH";
+})(VADEventType || (VADEventType = {}));
+export class VAD {
+}
+export class VADStream {
+    [Symbol.iterator]() {
+        return this;
+    }
+}
+//# sourceMappingURL=vad.js.map

package/dist/vad.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vad.js","sourceRoot":"","sources":["../src/vad.ts"],"names":[],"mappings":"AAKA,MAAM,CAAN,IAAY,YAIX;AAJD,WAAY,YAAY;IACtB,qEAAmB,CAAA;IACnB,uDAAY,CAAA;IACZ,iEAAiB,CAAA;AACnB,CAAC,EAJW,YAAY,KAAZ,YAAY,QAIvB;AASD,MAAM,OAAgB,GAAG;CAcxB;AAED,MAAM,OAAgB,SAAS;IAI7B,CAAC,MAAM,CAAC,QAAQ,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

package/dist/version.d.ts

@@ -0,0 +1,2 @@
+export declare const version = "0.1.0";
+//# sourceMappingURL=version.d.ts.map

package/dist/version.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["../src/version.ts"],"names":[],"mappings":"AAIA,eAAO,MAAM,OAAO,UAAU,CAAC"}

package/dist/version.js

@@ -0,0 +1,5 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+export const version = '0.1.0';
+//# sourceMappingURL=version.js.map

package/dist/version.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.js","sourceRoot":"","sources":["../src/version.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AAEtC,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC"}

package/dist/worker.d.ts

@@ -0,0 +1,84 @@
+/// <reference types="node" resolution-mode="require"/>
+import { type AvailabilityRequest, type Job, type JobAssignment, JobType } from '@livekit/protocol';
+import { EventEmitter } from 'events';
+import { WebSocket } from 'ws';
+import { HTTPServer } from './http_server.js';
+import { JobProcess } from './ipc/job_process.js';
+import { JobRequest } from './job_request.js';
+import type { AcceptData } from './job_request.js';
+declare class WorkerPermissions {
+    canPublish: boolean;
+    canSubscribe: boolean;
+    canPublishData: boolean;
+    canUpdateMetadata: boolean;
+    hidden: boolean;
+    constructor(canPublish?: boolean, canSubscribe?: boolean, canPublishData?: boolean, canUpdateMetadata?: boolean, hidden?: boolean);
+}
+export declare class WorkerOptions {
+    requestFunc: (arg: JobRequest) => Promise<void>;
+    cpuLoadFunc: () => number;
+    namespace: string;
+    permissions: WorkerPermissions;
+    workerType: JobType;
+    maxRetry: number;
+    wsURL: string;
+    apiKey?: string;
+    apiSecret?: string;
+    host: string;
+    port: number;
+    constructor({ requestFunc, cpuLoadFunc, namespace, permissions, workerType, maxRetry, wsURL, apiKey, apiSecret, host, port, }: {
+        requestFunc: (arg: JobRequest) => Promise<void>;
+        cpuLoadFunc?: () => number;
+        namespace?: string;
+        permissions?: WorkerPermissions;
+        workerType?: JobType;
+        maxRetry?: number;
+        wsURL?: string;
+        apiKey?: string;
+        apiSecret?: string;
+        host?: string;
+        port?: number;
+    });
+}
+declare class ActiveJob {
+    job: Job;
+    acceptData: AcceptData;
+    constructor(job: Job, acceptData: AcceptData);
+}
+type AssignmentPair = {
+    raw: string;
+    asgn: JobAssignment;
+};
+declare class PendingAssignment {
+    promise: Promise<AssignmentPair>;
+    resolve(arg: AssignmentPair): void;
+}
+export declare class Worker {
+    #private;
+    opts: WorkerOptions;
+    session: WebSocket | undefined;
+    closed: boolean;
+    httpServer: HTTPServer;
+    logger: import("pino").default.Logger<never>;
+    event: EventEmitter<[never]>;
+    pending: {
+        [id: string]: {
+            value: PendingAssignment;
+        };
+    };
+    processes: {
+        [id: string]: {
+            proc: JobProcess;
+            activeJob: ActiveJob;
+        };
+    };
+    constructor(opts: WorkerOptions);
+    get id(): string;
+    run(): Promise<void>;
+    startProcess(job: Job, acceptData: AcceptData, raw: string): void;
+    runWS(ws: WebSocket): void;
+    availability(msg: AvailabilityRequest): Promise<void>;
+    close(): Promise<void>;
+}
+export {};
+//# sourceMappingURL=worker.d.ts.map

package/dist/worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../src/worker.ts"],"names":[],"mappings":";AAGA,OAAO,EACL,KAAK,mBAAmB,EACxB,KAAK,GAAG,EACR,KAAK,aAAa,EAClB,OAAO,EAIR,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAGtC,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAiB,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC7D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAkBnD,cAAM,iBAAiB;IACrB,UAAU,EAAE,OAAO,CAAC;IACpB,YAAY,EAAE,OAAO,CAAC;IACtB,cAAc,EAAE,OAAO,CAAC;IACxB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,MAAM,EAAE,OAAO,CAAC;gBAGd,UAAU,UAAO,EACjB,YAAY,UAAO,EACnB,cAAc,UAAO,EACrB,iBAAiB,UAAO,EACxB,MAAM,UAAQ;CAQjB;AAED,qBAAa,aAAa;IACxB,WAAW,EAAE,CAAC,GAAG,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAChD,WAAW,EAAE,MAAM,MAAM,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,iBAAiB,CAAC;IAC/B,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;gBAED,EACV,WAAW,EACX,WAAqB,EACrB,SAAqB,EACrB,WAAqC,EACrC,UAAiC,EACjC,QAAiC,EACjC,KAA6B,EAC7B,MAAkB,EAClB,SAAqB,EACrB,IAAkB,EAClB,IAAW,GACZ,EAAE;QACD,WAAW,EAAE,CAAC,GAAG,EAAE,UAAU,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAChD,WAAW,CAAC,EAAE,MAAM,MAAM,CAAC;QAC3B,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,iBAAiB,CAAC;QAChC,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;KACf;CAaF;AAED,cAAM,SAAS;IACb,GAAG,EAAE,GAAG,CAAC;IACT,UAAU,EAAE,UAAU,CAAC;gBAEX,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,UAAU;CAI7C;AAED,KAAK,cAAc,GAAG;IAGpB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,aAAa,CAAC;CACrB,CAAC;AAEF,cAAM,iBAAiB;IACrB,OAAO,0BAEJ;IACH,OAAO,CAAC,GAAG,EAAE,cAAc;CAG5B;AAED,qBAAa,MAAM;;IACjB,IAAI,EAAE,aAAa,CAAC;IAEpB,OAAO,EAAE,SAAS,GAAG,SAAS,CAAa;IAC3C,MAAM,UAAS;IACf,UAAU,EAAE,UAAU,CAAC;IACvB,MAAM,uCAA0B;IAChC,KAAK,wBAAsB;IAC3B,OAAO,EAAE;QAAE,CAAC,EAAE,EAAE,MAAM,GAAG;YAAE,KAAK,EAAE,iBAAiB,CAAA;SAAE,CAAA;KAAE,CAAM;IAC7D,SAAS,EAAE;QAAE,CAAC,EAAE,EAAE,MAAM,GAAG;YAAE,IAAI,EAAE,UAAU,CAAC;YAAC,SAAS,EAAE,SAAS,CAAA;SAAE,CAAA;KAAE,CAAM;gBAEjE,IAAI,EAAE,aAAa;IAS/B,IAAI,EAAE,IAAI,MAAM,CAEf;IAEK,GAAG;IAoDT,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM;IAc1D,KAAK,CAAC,EAAE,EAAE,SAAS;IA4Fb,YAAY,CAAC,GAAG,EAAE,mBAAmB;IAsDrC,KAAK;CAUZ"}

package/dist/worker.js

@@ -0,0 +1,296 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { JobType, ParticipantPermission, ServerMessage, WorkerMessage, } from '@livekit/protocol';
+import { EventEmitter } from 'events';
+import { AccessToken } from 'livekit-server-sdk';
+import os from 'os';
+import { WebSocket } from 'ws';
+import { HTTPServer } from './http_server.js';
+import { JobProcess } from './ipc/job_process.js';
+import { JobRequest } from './job_request.js';
+import { log } from './log.js';
+import { version } from './version.js';
+const MAX_RECONNECT_ATTEMPTS = 10;
+const ASSIGNMENT_TIMEOUT = 15 * 1000;
+const LOAD_INTERVAL = 5 * 1000;
+const cpuLoad = () => (os
+    .cpus()
+    .reduce((acc, x) => acc + x.times.idle / Object.values(x.times).reduce((acc, x) => acc + x, 0), 0) /
+    os.cpus().length) *
+    100;
+class WorkerPermissions {
+    canPublish;
+    canSubscribe;
+    canPublishData;
+    canUpdateMetadata;
+    hidden;
+    constructor(canPublish = true, canSubscribe = true, canPublishData = true, canUpdateMetadata = true, hidden = false) {
+        this.canPublish = canPublish;
+        this.canSubscribe = canSubscribe;
+        this.canPublishData = canPublishData;
+        this.canUpdateMetadata = canUpdateMetadata;
+        this.hidden = hidden;
+    }
+}
+export class WorkerOptions {
+    requestFunc;
+    cpuLoadFunc;
+    namespace;
+    permissions;
+    workerType;
+    maxRetry;
+    wsURL;
+    apiKey;
+    apiSecret;
+    host;
+    port;
+    constructor({ requestFunc, cpuLoadFunc = cpuLoad, namespace = 'default', permissions = new WorkerPermissions(), workerType = JobType.JT_PUBLISHER, maxRetry = MAX_RECONNECT_ATTEMPTS, wsURL = 'ws://localhost:7880', apiKey = undefined, apiSecret = undefined, host = 'localhost', port = 8081, }) {
+        this.requestFunc = requestFunc;
+        this.cpuLoadFunc = cpuLoadFunc;
+        this.namespace = namespace;
+        this.permissions = permissions;
+        this.workerType = workerType;
+        this.maxRetry = maxRetry;
+        this.wsURL = wsURL;
+        this.apiKey = apiKey;
+        this.apiSecret = apiSecret;
+        this.host = host;
+        this.port = port;
+    }
+}
+class ActiveJob {
+    job;
+    acceptData;
+    constructor(job, acceptData) {
+        this.job = job;
+        this.acceptData = acceptData;
+    }
+}
+class PendingAssignment {
+    promise = new Promise((resolve) => {
+        this.resolve = resolve; // oh, JavaScript.
+    });
+    resolve(arg) {
+        arg;
+    }
+}
+export class Worker {
+    opts;
+    #id = 'unregistered';
+    session = undefined;
+    closed = false;
+    httpServer;
+    logger = log.child({ version });
+    event = new EventEmitter();
+    pending = {};
+    processes = {};
+    constructor(opts) {
+        opts.wsURL = opts.wsURL || process.env.LIVEKIT_URL || '';
+        opts.apiKey = opts.apiKey || process.env.LIVEKIT_API_KEY || '';
+        opts.apiSecret = opts.apiSecret || process.env.LIVEKIT_API_SECRET || '';
+        this.opts = opts;
+        this.httpServer = new HTTPServer(opts.host, opts.port);
+    }
+    get id() {
+        return this.#id;
+    }
+    async run() {
+        this.logger.info('starting worker');
+        if (this.opts.wsURL === '')
+            throw new Error('--url is required, or set LIVEKIT_URL env var');
+        if (this.opts.apiKey === '')
+            throw new Error('--api-key is required, or set LIVEKIT_API_KEY env var');
+        if (this.opts.apiSecret === '')
+            throw new Error('--api-secret is required, or set LIVEKIT_API_SECRET env var');
+        const workerWS = async () => {
+            let retries = 0;
+            while (!this.closed) {
+                const token = new AccessToken(this.opts.apiKey, this.opts.apiSecret);
+                token.addGrant({ agent: true });
+                const jwt = await token.toJwt();
+                const url = new URL(this.opts.wsURL);
+                url.protocol = url.protocol.replace('http', 'ws');
+                this.session = new WebSocket(url + 'agent', {
+                    headers: { authorization: 'Bearer ' + jwt },
+                });
+                try {
+                    await new Promise((resolve, reject) => {
+                        this.session.on('open', resolve);
+                        this.session.on('error', (error) => reject(error));
+                        this.session.on('close', (code) => reject(`WebSocket returned ${code}`));
+                    });
+                    this.runWS(this.session);
+                    return;
+                }
+                catch (e) {
+                    if (this.closed)
+                        return;
+                    if (retries >= this.opts.maxRetry) {
+                        throw new Error(`failed to connect to LiveKit server after ${retries} attempts: ${e}`);
+                    }
+                    retries++;
+                    const delay = Math.min(retries * 2, 10);
+                    this.logger.warn(`failed to connect to LiveKit server, retrying in ${delay} seconds: ${e} (${retries}/${this.opts.maxRetry})`);
+                    await new Promise((resolve) => setTimeout(resolve, delay * 1000));
+                }
+            }
+        };
+        await Promise.all([workerWS(), this.httpServer.run()]);
+    }
+    startProcess(job, acceptData, raw) {
+        const proc = new JobProcess(job, acceptData, raw, this.opts.wsURL);
+        this.processes[job.id] = { proc, activeJob: new ActiveJob(job, acceptData) };
+        proc
+            .run()
+            .catch((e) => {
+            proc.logger.error(`error running job process ${proc.job.id}: ${e}`);
+        })
+            .finally(() => {
+            proc.clear();
+            delete this.processes[job.id];
+        });
+    }
+    runWS(ws) {
+        let closingWS = false;
+        const send = (msg) => {
+            if (closingWS) {
+                this.event.off('worker_msg', send);
+                return;
+            }
+            ws.send(msg.toBinary());
+        };
+        this.event.on('worker_msg', send);
+        ws.addEventListener('close', () => {
+            closingWS = true;
+            this.logger.error('worker connection closed unexpectedly');
+            this.close();
+        });
+        ws.addEventListener('message', (event) => {
+            if (event.type !== 'message') {
+                this.logger.warn('unexpected message type: ' + event.type);
+                return;
+            }
+            const msg = new ServerMessage();
+            msg.fromBinary(event.data);
+            switch (msg.message.case) {
+                case 'register': {
+                    this.#id = msg.message.value.workerId;
+                    log
+                        .child({ id: this.id, server_info: msg.message.value.serverInfo })
+                        .info('registered worker');
+                    break;
+                }
+                case 'availability': {
+                    this.availability(msg.message.value);
+                    break;
+                }
+                case 'assignment': {
+                    const job = msg.message.value.job;
+                    if (job.id in this.pending) {
+                        const task = this.pending[job.id];
+                        delete this.pending[job.id];
+                        task.value.resolve({
+                            asgn: msg.message.value,
+                            raw: msg.toJsonString(),
+                        });
+                    }
+                    else {
+                        log.child({ job }).warn('received assignment for unknown job ' + job.id);
+                    }
+                    break;
+                }
+            }
+        });
+        this.event.emit('worker_msg', new WorkerMessage({
+            message: {
+                case: 'register',
+                value: {
+                    type: this.opts.workerType,
+                    namespace: this.opts.namespace,
+                    allowedPermissions: new ParticipantPermission({
+                        canPublish: this.opts.permissions.canPublish,
+                        canSubscribe: this.opts.permissions.canSubscribe,
+                        canPublishData: this.opts.permissions.canPublishData,
+                        hidden: this.opts.permissions.hidden,
+                        agent: true,
+                    }),
+                    version,
+                },
+            },
+        }));
+        const loadMonitor = setInterval(() => {
+            if (closingWS)
+                clearInterval(loadMonitor);
+            this.event.emit('worker_msg', new WorkerMessage({
+                message: {
+                    case: 'updateWorker',
+                    value: {
+                        load: cpuLoad(),
+                    },
+                },
+            }));
+        }, LOAD_INTERVAL);
+    }
+    async availability(msg) {
+        const tx = new EventEmitter();
+        const req = new JobRequest(msg.job, tx);
+        tx.on('recv', async (av) => {
+            const msg = new WorkerMessage({
+                message: {
+                    case: 'availability',
+                    value: {
+                        available: av.avail,
+                        jobId: req.id,
+                        participantIdentity: av.data?.identity,
+                        participantName: av.data?.name,
+                        participantMetadata: av.data?.metadata,
+                    },
+                },
+            });
+            this.pending[req.id] = { value: new PendingAssignment() };
+            this.event.emit('worker_msg', msg);
+            if (!av.avail)
+                return;
+            const timer = setTimeout(() => {
+                log.child({ req }).warn(`assignment for job ${req.id} timed out`);
+                return;
+            }, ASSIGNMENT_TIMEOUT);
+            this.pending[req.id].value.promise.then(({ asgn, raw }) => {
+                clearTimeout(timer);
+                this.startProcess(asgn.job, av.data, raw);
+            });
+        });
+        try {
+            this.opts.requestFunc(req);
+        }
+        catch (e) {
+            log.child({ req }).error(`user request handler for job ${req.id} failed`);
+        }
+        finally {
+            if (!req.answered) {
+                log.child({ req }).error(`no answer for job ${req.id}, automatically rejecting the job`);
+                this.event.emit('worker_msg', new WorkerMessage({
+                    message: {
+                        case: 'availability',
+                        value: {
+                            available: false,
+                        },
+                    },
+                }));
+            }
+        }
+    }
+    async close() {
+        if (this.closed)
+            return;
+        this.closed = true;
+        this.logger.debug('shutting down worker');
+        await this.httpServer.close();
+        for await (const value of Object.values(this.processes)) {
+            await value.proc.close();
+        }
+        this.session?.close();
+    }
+}
+//# sourceMappingURL=worker.js.map

package/dist/worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.js","sourceRoot":"","sources":["../src/worker.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAC7C,EAAE;AACF,sCAAsC;AACtC,OAAO,EAIL,OAAO,EACP,qBAAqB,EACrB,aAAa,EACb,aAAa,GACd,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAiB,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE7D,OAAO,EAAE,GAAG,EAAE,MAAM,UAAU,CAAC;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,MAAM,sBAAsB,GAAG,EAAE,CAAC;AAClC,MAAM,kBAAkB,GAAG,EAAE,GAAG,IAAI,CAAC;AACrC,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,CAAC;AAE/B,MAAM,OAAO,GAAG,GAAW,EAAE,CAC3B,CAAC,EAAE;KACA,IAAI,EAAE;KACN,MAAM,CACL,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC,EACtF,CAAC,CACF;IACD,EAAE,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC;IACnB,GAAG,CAAC;AAEN,MAAM,iBAAiB;IACrB,UAAU,CAAU;IACpB,YAAY,CAAU;IACtB,cAAc,CAAU;IACxB,iBAAiB,CAAU;IAC3B,MAAM,CAAU;IAEhB,YACE,UAAU,GAAG,IAAI,EACjB,YAAY,GAAG,IAAI,EACnB,cAAc,GAAG,IAAI,EACrB,iBAAiB,GAAG,IAAI,EACxB,MAAM,GAAG,KAAK;QAEd,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,aAAa;IACxB,WAAW,CAAqC;IAChD,WAAW,CAAe;IAC1B,SAAS,CAAS;IAClB,WAAW,CAAoB;IAC/B,UAAU,CAAU;IACpB,QAAQ,CAAS;IACjB,KAAK,CAAS;IACd,MAAM,CAAU;IAChB,SAAS,CAAU;IACnB,IAAI,CAAS;IACb,IAAI,CAAS;IAEb,YAAY,EACV,WAAW,EACX,WAAW,GAAG,OAAO,EACrB,SAAS,GAAG,SAAS,EACrB,WAAW,GAAG,IAAI,iBAAiB,EAAE,EACrC,UAAU,GAAG,OAAO,CAAC,YAAY,EACjC,QAAQ,GAAG,sBAAsB,EACjC,KAAK,GAAG,qBAAqB,EAC7B,MAAM,GAAG,SAAS,EAClB,SAAS,GAAG,SAAS,EACrB,IAAI,GAAG,WAAW,EAClB,IAAI,GAAG,IAAI,GAaZ;QACC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;IACnB,CAAC;CACF;AAED,MAAM,SAAS;IACb,GAAG,CAAM;IACT,UAAU,CAAa;IAEvB,YAAY,GAAQ,EAAE,UAAsB;QAC1C,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;CACF;AASD,MAAM,iBAAiB;IACrB,OAAO,GAAG,IAAI,OAAO,CAAiB,CAAC,OAAO,EAAE,EAAE;QAChD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,CAAC,kBAAkB;IAC5C,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,GAAmB;QACzB,GAAG,CAAC;IACN,CAAC;CACF;AAED,MAAM,OAAO,MAAM;IACjB,IAAI,CAAgB;IACpB,GAAG,GAAG,cAAc,CAAC;IACrB,OAAO,GAA0B,SAAS,CAAC;IAC3C,MAAM,GAAG,KAAK,CAAC;IACf,UAAU,CAAa;IACvB,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;IAChC,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC;IAC3B,OAAO,GAAmD,EAAE,CAAC;IAC7D,SAAS,GAAiE,EAAE,CAAC;IAE7E,YAAY,IAAmB;QAC7B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,EAAE,CAAC;QACzD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,EAAE,CAAC;QAC/D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,EAAE,CAAC;QAExE,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,UAAU,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACzD,CAAC;IAED,IAAI,EAAE;QACJ,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,GAAG;QACP,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAEpC,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,KAAK,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QAC7F,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,EAAE;YACzB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC3E,IAAI,IAAI,CAAC,IAAI,CAAC,SAAS,KAAK,EAAE;YAC5B,MAAM,IAAI,KAAK,CAAC,6DAA6D,CAAC,CAAC;QAEjF,MAAM,QAAQ,GAAG,KAAK,IAAI,EAAE;YAC1B,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACrE,KAAK,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAChC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;gBAEhC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACrC,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBAClD,IAAI,CAAC,OAAO,GAAG,IAAI,SAAS,CAAC,GAAG,GAAG,OAAO,EAAE;oBAC1C,OAAO,EAAE,EAAE,aAAa,EAAE,SAAS,GAAG,GAAG,EAAE;iBAC5C,CAAC,CAAC;gBAEH,IAAI,CAAC;oBACH,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;wBACpC,IAAI,CAAC,OAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;wBAClC,IAAI,CAAC,OAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;wBACpD,IAAI,CAAC,OAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,sBAAsB,IAAI,EAAE,CAAC,CAAC,CAAC;oBAC5E,CAAC,CAAC,CAAC;oBAEH,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAQ,CAAC,CAAC;oBAC1B,OAAO;gBACT,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,IAAI,IAAI,CAAC,MAAM;wBAAE,OAAO;oBACxB,IAAI,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;wBAClC,MAAM,IAAI,KAAK,CAAC,6CAA6C,OAAO,cAAc,CAAC,EAAE,CAAC,CAAC;oBACzF,CAAC;oBAED,OAAO,EAAE,CAAC;oBACV,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;oBAExC,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oDAAoD,KAAK,aAAa,CAAC,KAAK,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,CAC7G,CAAC;oBAEF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC;gBACpE,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;IAED,YAAY,CAAC,GAAQ,EAAE,UAAsB,EAAE,GAAW;QACxD,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,SAAS,CAAC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;QAC7E,IAAI;aACD,GAAG,EAAE;aACL,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACX,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,EAAa;QACjB,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,MAAM,IAAI,GAAG,CAAC,GAAkB,EAAE,EAAE;YAClC,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;gBACnC,OAAO;YACT,CAAC;YACD,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC1B,CAAC,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;QAElC,EAAE,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;YAChC,SAAS,GAAG,IAAI,CAAC;YACjB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC3D,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;YACvC,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC3D,OAAO;YACT,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,IAAkB,CAAC,CAAC;YACzC,QAAQ,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;gBACzB,KAAK,UAAU,CAAC,CAAC,CAAC;oBAChB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;oBACtC,GAAG;yBACA,KAAK,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,WAAW,EAAE,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;yBACjE,IAAI,CAAC,mBAAmB,CAAC,CAAC;oBAC7B,MAAM;gBACR,CAAC;gBACD,KAAK,cAAc,CAAC,CAAC,CAAC;oBACpB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBACrC,MAAM;gBACR,CAAC;gBACD,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAI,CAAC;oBACnC,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;wBAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBAClC,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBAC5B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;4BACjB,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,KAAK;4BACvB,GAAG,EAAE,GAAG,CAAC,YAAY,EAAE;yBACxB,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,sCAAsC,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;oBAC3E,CAAC;oBACD,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,IAAI,CACb,YAAY,EACZ,IAAI,aAAa,CAAC;YAChB,OAAO,EAAE;gBACP,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE;oBACL,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;oBAC1B,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;oBAC9B,kBAAkB,EAAE,IAAI,qBAAqB,CAAC;wBAC5C,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,UAAU;wBAC5C,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY;wBAChD,cAAc,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,cAAc;wBACpD,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM;wBACpC,KAAK,EAAE,IAAI;qBACZ,CAAC;oBACF,OAAO;iBACR;aACF;SACF,CAAC,CACH,CAAC;QAEF,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE;YACnC,IAAI,SAAS;gBAAE,aAAa,CAAC,WAAW,CAAC,CAAC;YAC1C,IAAI,CAAC,KAAK,CAAC,IAAI,CACb,YAAY,EACZ,IAAI,aAAa,CAAC;gBAChB,OAAO,EAAE;oBACP,IAAI,EAAE,cAAc;oBACpB,KAAK,EAAE;wBACL,IAAI,EAAE,OAAO,EAAE;qBAChB;iBACF;aACF,CAAC,CACH,CAAC;QACJ,CAAC,EAAE,aAAa,CAAC,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,GAAwB;QACzC,MAAM,EAAE,GAAG,IAAI,YAAY,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,GAAI,EAAE,EAAE,CAAC,CAAC;QAEzC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,EAAY,EAAE,EAAE;YACnC,MAAM,GAAG,GAAG,IAAI,aAAa,CAAC;gBAC5B,OAAO,EAAE;oBACP,IAAI,EAAE,cAAc;oBACpB,KAAK,EAAE;wBACL,SAAS,EAAE,EAAE,CAAC,KAAK;wBACnB,KAAK,EAAE,GAAG,CAAC,EAAE;wBACb,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,QAAQ;wBACtC,eAAe,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI;wBAC9B,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,QAAQ;qBACvC;iBACF;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,iBAAiB,EAAE,EAAE,CAAC;YAC1D,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;YACnC,IAAI,CAAC,EAAE,CAAC,KAAK;gBAAE,OAAO;YAEtB,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,sBAAsB,GAAG,CAAC,EAAE,YAAY,CAAC,CAAC;gBAClE,OAAO;YACT,CAAC,EAAE,kBAAkB,CAAC,CAAC;YACvB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE;gBACxD,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,IAAI,CAAC,YAAY,CAAC,IAAK,CAAC,GAAI,EAAE,EAAE,CAAC,IAAK,EAAE,GAAG,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,gCAAgC,GAAG,CAAC,EAAE,SAAS,CAAC,CAAC;QAC5E,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;gBAClB,GAAG,CAAC,KAAK,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,qBAAqB,GAAG,CAAC,EAAE,mCAAmC,CAAC,CAAC;gBACzF,IAAI,CAAC,KAAK,CAAC,IAAI,CACb,YAAY,EACZ,IAAI,aAAa,CAAC;oBAChB,OAAO,EAAE;wBACP,IAAI,EAAE,cAAc;wBACpB,KAAK,EAAE;4BACL,SAAS,EAAE,KAAK;yBACjB;qBACF;iBACF,CAAC,CACH,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAC9B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YACxD,MAAM,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;IACxB,CAAC;CACF"}

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "@livekit/agents",
+  "version": "0.1.0",
+  "description": "LiveKit Node Agents",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "author": "aoife cassidy <aoife@cassidy.sh>",
+  "type": "module",
+  "devDependencies": {
+    "@microsoft/api-extractor": "^7.35.0",
+    "@types/ws": "^8.5.10"
+  },
+  "peerDependencies": {
+    "typescript": "^5.0.0"
+  },
+  "dependencies": {
+    "@livekit/rtc-node": "^0.1.0",
+    "@livekit/protocol": "^1.13.0",
+    "commander": "^12.0.0",
+    "livekit-server-sdk": "^2.1.2",
+    "pino": "^8.19.0",
+    "pino-pretty": "^11.0.0",
+    "ws": "^8.16.0"
+  },
+  "scripts": {
+    "build": "tsc",
+    "lint": "eslint -f unix \"src/**/*.ts\"",
+    "api:check": "api-extractor run --typescript-compiler-folder ../node_modules/typescript",
+    "api:update": "api-extractor run --local --typescript-compiler-folder ../node_modules/typescript --verbose"
+  }
+}

package/src/cli.ts

@@ -0,0 +1,79 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import { Command, Option } from 'commander';
+import type { EventEmitter } from 'events';
+import { log } from './log.js';
+import { version } from './version.js';
+import { Worker, type WorkerOptions } from './worker.js';
+
+type CliArgs = {
+  opts: WorkerOptions;
+  logLevel: string;
+  production: boolean;
+  watch: boolean;
+  event?: EventEmitter;
+};
+
+const runWorker = async (args: CliArgs) => {
+  log.level = args.logLevel;
+  const worker = new Worker(args.opts);
+
+  process.on('SIGINT', async () => {
+    await worker.close();
+    log.info('worker closed');
+    process.exit(130); // SIGINT exit code
+  });
+
+  try {
+    await worker.run();
+  } catch {
+    log.fatal('worker failed');
+    process.exit(1);
+  }
+};
+
+export const runApp = (opts: WorkerOptions) => {
+  const program = new Command()
+    .name('agents')
+    .description('LiveKit Agents CLI')
+    .version(version)
+    .addOption(
+      new Option('--log-level <level>', 'Set the logging level')
+        .choices(['trace', 'debug', 'info', 'warn', 'error', 'fatal'])
+        .default('trace'),
+    )
+    .addOption(
+      new Option('--url <string>', 'LiveKit server or Cloud project websocket URL')
+        .makeOptionMandatory(true)
+        .env('LIVEKIT_URL'),
+    )
+    .addOption(
+      new Option('--api-key <string>', "LiveKit server or Cloud project's API key")
+        .makeOptionMandatory(true)
+        .env('LIVEKIT_API_KEY'),
+    )
+    .addOption(
+      new Option('--api-secret <string>', "LiveKit server or Cloud project's API secret")
+        .makeOptionMandatory(true)
+        .env('LIVEKIT_API_SECRET'),
+    );
+
+  program
+    .command('start')
+    .description('Start the worker in production mode')
+    .action(() => {
+      const options = program.optsWithGlobals();
+      opts.wsURL = options.url || opts.wsURL;
+      opts.apiKey = options.apiKey || opts.apiKey;
+      opts.apiSecret = options.apiSecret || opts.apiSecret;
+      runWorker({
+        opts,
+        production: true,
+        watch: false,
+        logLevel: options.logLevel,
+      });
+    });
+
+  program.parse();
+};

package/src/generator.ts

@@ -0,0 +1,18 @@
+// SPDX-FileCopyrightText: 2024 LiveKit, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+import type { JobContext } from './job_context.js';
+
+type entryFunction = (job: JobContext) => Promise<void>;
+
+export interface Agent {
+  entry: entryFunction;
+}
+
+/**
+ * Helper to define an agent according to the required interface.
+ * @example `export default defineAgent(myAgent);`
+ */
+export function defineAgent(entry: entryFunction): Agent {
+  return { entry };
+}