@deepnoodle/mobius 0.0.21 → 0.0.22

package/dist/worker.d.ts

@@ -1,190 +1,92 @@
 import { Client } from "./client.js";
 export type InstanceIDSource = "configured" | "cloud_run_revision_instance" | "hostname" | "fly_machine_id" | "railway_replica_id" | "render_instance_id" | "system_hostname" | "generated_uuid";
-/**
- * Resolve a per-process `worker_instance_id` from the runtime
- * environment. Order: explicit → Cloud Run K_REVISION + metadata →
- * HOSTNAME env → FLY_MACHINE_ID → RAILWAY_REPLICA_ID →
- * RENDER_INSTANCE_ID → `os.hostname()` suffixed with a per-boot
- * random tag (laptops and bare metal) → generated UUID.
- *
- * The system_hostname rung carries a random suffix because
- * `os.hostname()` identifies the host, not the process — two
- * processes started on the same machine would otherwise auto-detect
- * the same identifier and collide on the server's conflict detector.
- * Set {@link WorkerConfig.workerInstanceId} explicitly only when a
- * stable identity across restarts is desired (named singleton
- * workers).
- *
- * The returned source is informational only — workers log it once at
- * startup so operators can confirm the right platform was picked up.
- */
 export declare function resolveInstanceID(explicit: string | undefined): Promise<{
     id: string;
     source: InstanceIDSource;
 }>;
-/**
- * Logger interface used by Worker for status and error output. Compatible with
- * the global `console` object, so `console` is a valid logger. Pass `null` to
- * `WorkerConfig.logger` to silence all output.
- */
 export interface Logger {
     info(...args: unknown[]): void;
     warn(...args: unknown[]): void;
     error(...args: unknown[]): void;
 }
+export interface ModelCapability {
+    provider: string;
+    model: string;
+}
 export interface WorkerConfig {
-    /**
-     * Identifier for this worker process. Leave omitted for the
-     * common case: the SDK auto-detects an identifier that is unique
-     * per running process. Resolution prefers platform-native IDs
-     * that are unique-per-replica by design (Cloud Run revision
-     * instance, Kubernetes HOSTNAME, Fly machine, Railway replica,
-     * Render instance), then falls back to OS hostname plus a
-     * per-boot random suffix so two processes on the same host
-     * (back-to-back tests, parallel CI workers) cannot collide.
-     *
-     * Set this explicitly only when you want a stable identity
-     * across restarts of a named singleton worker. Two live
-     * processes using the same explicit value in the same project
-     * will collide and the second will fail with
-     * {@link WorkerInstanceConflictError}.
-     */
     workerInstanceId?: string;
-    /**
-     * Maximum number of jobs this worker holds in flight simultaneously.
-     * Defaults to 1; raise to claim several jobs from one worker process
-     * while still surfacing as a single row on the workers page.
-     */
     concurrency?: number;
-    /** Human-readable name reported to Mobius (e.g. "billing-worker"). */
     name?: string;
-    /** Version string reported to Mobius (e.g. "1.0.0"). */
     version?: string;
-    /**
-     * Queue names this worker subscribes to. Empty (or omitted) means claim
-     * jobs from any queue in the project. Runs default to the "default"
-     * queue.
-     */
     queues?: string[];
-    /**
-     * Optional filter of action names this worker will claim. When empty the
-     * worker will only claim jobs for actions it has registered.
-     */
     actions?: string[];
-    /** Long-poll window per claim request in seconds (0–30). Defaults to 20. */
-    pollWaitSeconds?: number;
-    /**
-     * Heartbeat interval in milliseconds. When unset, the SDK uses the interval
-     * advertised by the server in the claim response, falling back to 10_000.
-     */
+    models?: ModelCapability[];
+    reconnectDelayMs?: number;
     heartbeatIntervalMs?: number;
-    /**
-     * Logger for worker status and error output. Defaults to `console`. Pass
-     * `null` to silence all worker output, or supply a custom {@link Logger}
-     * (e.g. pino, winston) to route messages elsewhere.
-     */
     logger?: Logger | null;
-    /** Max buffered custom events per in-flight job before dropping oldest. */
-    eventQueueSize?: number;
-    /** Max custom events per HTTP batch. Defaults to 20. */
-    eventBatchSize?: number;
 }
-/**
- * ActionFn implements a single Mobius action. Receives the JSON-decoded
- * parameters delivered with the job claim and returns a JSON-serialisable
- * result.
- */
-export type ActionFn = (params: Record<string, unknown>, signal: AbortSignal, ctx: ActionContext) => Promise<unknown>;
 export interface ActionContext {
     jobId: string;
-    runId: string;
+    runId?: string;
+    sessionId?: string;
+    agentTurnId?: string;
+    toolCallId?: string;
     projectId?: string;
     workerInstanceId: string;
     attempt: number;
     queue?: string;
-    workflowName?: string;
-    stepName?: string;
+    stepId?: string;
     action?: string;
     emitEvent(type: string, payload: Record<string, unknown>): void;
 }
-/**
- * Worker claims jobs from Mobius and dispatches each to the corresponding
- * registered action function. A *job* is a single action invocation on
- * behalf of a workflow run; the backend owns the workflow engine.
- *
- * With `concurrency > 1` the worker holds up to N jobs in flight
- * simultaneously, all reporting the same `worker_instance_id` and
- * session token; the admin UI shows one row with a saturation bar
- * rather than N independent rows.
- */
+export type ActionFn = (params: Record<string, unknown>, signal: AbortSignal, ctx: ActionContext) => Promise<unknown>;
+export interface GenerationJob {
+    jobId: string;
+    runId?: string;
+    sessionId?: string;
+    agentTurnId?: string;
+    toolCallId?: string;
+    provider?: string;
+    model?: string;
+    spec: Record<string, unknown>;
+}
+export type GenerationEmitter = (delta: Record<string, unknown>) => void;
+export type GenerationFn = (ctx: ActionContext, job: GenerationJob, emit: GenerationEmitter) => Promise<Record<string, unknown>>;
 export declare class Worker {
     private readonly client;
     private readonly config;
     private readonly logger;
     private readonly actions;
-    private readonly sessionToken;
-    private instanceIDResolved;
-    private abortController;
-    private authRevoked;
-    constructor(client: Client, config: WorkerConfig, actions?: Map<string, ActionFn>);
-    /** Register an action function under the given name. */
+    private readonly generators;
+    private sessionToken;
+    private stopping;
+    private readonly stopController;
+    constructor(client: Client, config?: WorkerConfig, actions?: Map<string, ActionFn>);
     register(name: string, fn: ActionFn): this;
-    /**
-     * Start the claim loop. Returns a promise that resolves when the worker is
-     * stopped via {@link stop} or the given signal is aborted. Throws
-     * {@link AuthRevokedError} when the credential is revoked mid-flight
-     * and {@link WorkerInstanceConflictError} when another live process
-     * has already registered this worker_instance_id.
-     *
-     * Cancellation has two distinct shapes:
-     *
-     * - {@link stop} aborts the claim loop only — in-flight jobs continue
-     *   to run to natural completion and the returned promise resolves
-     *   once they all settle. This is graceful drain.
-     * - The caller's `signal`, when aborted, propagates into both the
-     *   claim loop and every running action. This is the emergency-stop
-     *   path (e.g. SIGTERM with a hard deadline).
-     */
-    run(signal?: AbortSignal): Promise<void>;
-    /** Gracefully stop the worker after in-flight jobs complete. */
+    registerGenerator(provider: string, model: string, fn: GenerationFn): this;
     stop(): void;
+    run(signal?: AbortSignal): Promise<void>;
+    private runSocket;
+    private registerFrame;
+    private claim;
+    private modelCapabilities;
     private executeJob;
-    private heartbeatInterval;
-    private failJob;
+    private generator;
+    private actionContext;
+    private report;
 }
 export interface WorkerPoolConfig extends Omit<WorkerConfig, "workerInstanceId"> {
-    /** Number of worker instances to run. Defaults to 1. */
     count?: number;
-    /**
-     * Prefix used to derive child instance IDs as `<prefix>-<index>`.
-     * When omitted, the SDK generates a per-boot prefix; child workers
-     * each get their own session token, so a pool of N produces N rows
-     * on the workers page.
-     */
     workerInstanceIdPrefix?: string;
 }
-/**
- * Runs multiple worker instances in one process. Most callers do not
- * need a pool — to run several jobs from one process, set
- * {@link WorkerConfig.concurrency} on a single {@link Worker} and the
- * admin UI will show one row with a saturation bar. Reach for a pool
- * only when each child should surface as its own row on the workers
- * page (independent draining, in-flight isolation).
- */
 export declare class WorkerPool {
     private readonly client;
     private readonly config;
     private readonly actions;
-    private abortController;
+    private readonly generators;
     constructor(client: Client, config: WorkerPoolConfig);
-    /** Register an action function under the given name for every pool worker. */
     register(name: string, fn: ActionFn): this;
-    /**
-     * Start all workers in the pool. Rejects with AuthRevokedError if any child
-     * worker sees credential revocation.
-     */
+    registerGenerator(provider: string, model: string, fn: GenerationFn): this;
     run(signal?: AbortSignal): Promise<void>;
-    /** Gracefully stop every worker in the pool after in-flight jobs complete. */
-    stop(): void;
 }
 //# sourceMappingURL=worker.d.ts.map

package/dist/worker.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../src/worker.ts"],"names":[],"mappings":"AAWA,OAAO,EAEL,MAAM,EAMP,MAAM,aAAa,CAAC;AAErB,MAAM,MAAM,gBAAgB,GACxB,YAAY,GACZ,6BAA6B,GAC7B,UAAU,GACV,gBAAgB,GAChB,oBAAoB,GACpB,oBAAoB,GACpB,iBAAiB,GACjB,gBAAgB,CAAC;AAWrB;;;;;;;;;;;;;;;;;GAiBG;AACH,wBAAsB,iBAAiB,CACrC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAC3B,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,gBAAgB,CAAA;CAAE,CAAC,CAyBnD;AA6BD;;;;GAIG;AACH,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAC/B,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;CACjC;AAcD,MAAM,WAAW,YAAY;IAC3B;;;;;;;;;;;;;;;OAeG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;;OAIG;IACH,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,sEAAsE;IACtE,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,wDAAwD;IACxD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,4EAA4E;IAC5E,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB;;;OAGG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B;;;;OAIG;IACH,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,2EAA2E;IAC3E,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,wDAAwD;IACxD,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;;;GAIG;AACH,MAAM,MAAM,QAAQ,GAAG,CACrB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,MAAM,EAAE,WAAW,EACnB,GAAG,EAAE,aAAa,KACf,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CACjE;AAED;;;;;;;;;GASG;AACH,qBAAa,MAAM;IAcf,OAAO,CAAC,QAAQ,CAAC,MAAM;IAbzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAIrB;IACF,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAwB;IAChD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,kBAAkB,CAAS;IACnC,OAAO,CAAC,eAAe,CAAyB;IAChD,OAAO,CAAC,WAAW,CAAS;gBAGT,MAAM,EAAE,MAAM,EAC/B,MAAM,EAAE,YAAY,EACpB,OAAO,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC;IAqBjC,wDAAwD;IACxD,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,GAAG,IAAI;IAK1C;;;;;;;;;;;;;;;OAeG;IACG,GAAG,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAwF9C,gEAAgE;IAChE,IAAI,IAAI,IAAI;YAME,UAAU;IA4IxB,OAAO,CAAC,iBAAiB;YASX,OAAO;CAyBtB;AAED,MAAM,WAAW,gBAAiB,SAAQ,IAAI,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAC9E,wDAAwD;IACxD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;;OAKG;IACH,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED;;;;;;;GAOG;AACH,qBAAa,UAAU;IAenB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAdzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CASrB;IACF,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA+B;IACvD,OAAO,CAAC,eAAe,CAAyB;gBAG7B,MAAM,EAAE,MAAM,EAC/B,MAAM,EAAE,gBAAgB;IAsB1B,8EAA8E;IAC9E,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,GAAG,IAAI;IAK1C;;;OAGG;IACG,GAAG,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;IAoD9C,8EAA8E;IAC9E,IAAI,IAAI,IAAI;CAGb"}
+{"version":3,"file":"worker.d.ts","sourceRoot":"","sources":["../src/worker.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,MAAM,gBAAgB,GACxB,YAAY,GACZ,6BAA6B,GAC7B,UAAU,GACV,gBAAgB,GAChB,oBAAoB,GACpB,oBAAoB,GACpB,iBAAiB,GACjB,gBAAgB,CAAC;AAKrB,wBAAsB,iBAAiB,CACrC,QAAQ,EAAE,MAAM,GAAG,SAAS,GAC3B,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,gBAAgB,CAAA;CAAE,CAAC,CAwBnD;AAwBD,MAAM,WAAW,MAAM;IACrB,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAC/B,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;IAC/B,KAAK,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;CACjC;AASD,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,YAAY;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,MAAM,CAAC,EAAE,eAAe,EAAE,CAAC;IAC3B,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACxB;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CACjE;AAED,MAAM,MAAM,QAAQ,GAAG,CACrB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,MAAM,EAAE,WAAW,EACnB,GAAG,EAAE,aAAa,KACf,OAAO,CAAC,OAAO,CAAC,CAAC;AAEtB,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,MAAM,MAAM,iBAAiB,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;AACzE,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,aAAa,EAClB,GAAG,EAAE,aAAa,EAClB,IAAI,EAAE,iBAAiB,KACpB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;AAEtC,qBAAa,MAAM;IASf,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,MAAM;IATzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA+B;IACvD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmC;IAC9D,OAAO,CAAC,YAAY,CAAM;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAyB;gBAGrC,MAAM,EAAE,MAAM,EACd,MAAM,GAAE,YAAiB,EAC1C,OAAO,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC;IASjC,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,GAAG,IAAI;IAK1C,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,YAAY,GAAG,IAAI;IAK1E,IAAI,IAAI,IAAI;IAKN,GAAG,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;YAgBhC,SAAS;IA+DvB,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,KAAK;IAuBb,OAAO,CAAC,iBAAiB;YAOX,UAAU;IA2DxB,OAAO,CAAC,SAAS;IAKjB,OAAO,CAAC,aAAa;IAmBrB,OAAO,CAAC,MAAM;CAoBf;AAED,MAAM,WAAW,gBAAiB,SAAQ,IAAI,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAC9E,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,qBAAa,UAAU;IAInB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAJzB,OAAO,CAAC,QAAQ,CAAC,OAAO,CAA+B;IACvD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmC;gBAE3C,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,gBAAgB;IAG3C,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,QAAQ,GAAG,IAAI;IAK1C,iBAAiB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,EAAE,YAAY,GAAG,IAAI;IAKpE,GAAG,CAAC,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;CAiB/C"}