ocmm 0.5.3 → 0.6.0

package/dist/runtime-fallback/event-handler-test-fixtures.js

@@ -0,0 +1,150 @@
+import assert from "node:assert/strict";
+import { OcmmConfigSchema } from "../config/schema.js";
+export function makeMockClient() {
+    const calls = [];
+    const messagesResp = {
+        messages: [
+            { role: "user", parts: [{ type: "text", text: "hello" }] },
+            { role: "assistant", parts: [{ type: "text", text: "hi" }] },
+        ],
+    };
+    const client = {
+        session: {
+            async abort() {
+                return undefined;
+            },
+            async messages() {
+                return messagesResp;
+            },
+            async prompt(args) {
+                calls.push({
+                    sessionID: args.path.id,
+                    body: args.body,
+                    ...(args.query?.directory !== undefined ? { directory: args.query.directory } : {}),
+                });
+                return undefined;
+            },
+        },
+    };
+    return { client, calls, messagesResp };
+}
+export function makeConfig(overrides = {}) {
+    return OcmmConfigSchema.parse({
+        agents: {
+            orchestrator: {
+                model: "hoo/primary-model",
+                fallbackModels: ["hoo/fallback-a", "hoo/fallback-b"],
+            },
+        },
+        runtimeFallback: overrides,
+    });
+}
+export function makeErrorEvent(sessionID, error, extras = {}) {
+    return {
+        event: {
+            type: "session.error",
+            properties: { sessionID, error, ...extras },
+        },
+    };
+}
+export function makeIdleEvent(sessionID) {
+    return { event: { type: "session.idle", properties: { sessionID } } };
+}
+export class FakeHandlerScheduler {
+    tasks = [];
+    schedule(delayMs, run) {
+        const task = { delayMs, run, cancelled: false };
+        this.tasks.push(task);
+        return () => { task.cancelled = true; };
+    }
+    async run(index) {
+        const task = this.tasks[index];
+        assert.ok(task, `missing scheduled task ${index}`);
+        if (!task.cancelled)
+            await task.run();
+    }
+}
+export function deferred() {
+    let resolve;
+    let reject;
+    const promise = new Promise((resolvePromise, rejectPromise) => {
+        resolve = resolvePromise;
+        reject = rejectPromise;
+    });
+    return { promise, resolve, reject };
+}
+export async function flushHandler() {
+    await new Promise((resolve) => setImmediate(resolve));
+    await Promise.resolve();
+}
+export function makeHandlerConfig(chain, runtimeFallback = {}, requirementOptions = {}) {
+    return OcmmConfigSchema.parse({
+        agents: { worker: { requirement: { fallbackChain: chain, ...requirementOptions } } },
+        runtimeFallback,
+    });
+}
+export function standardHandlerChain() {
+    return [
+        { providers: ["provider-a"], model: "primary", variant: "high" },
+        { providers: ["provider-b"], model: "fallback-a", variant: "medium" },
+        { providers: ["provider-c"], model: "fallback-b", variant: "low" },
+    ];
+}
+export function publishWorkerRouteSnapshot(registry, chain = standardHandlerChain()) {
+    const generation = registry.beginBuild();
+    const routes = new Map([
+        ["worker", {
+                model: "provider-a/primary",
+                requirement: { fallbackChain: chain },
+                requirementSource: "user-config",
+                primarySource: "user-requirement",
+            }],
+    ]);
+    assert.equal(registry.publish(generation, routes), true);
+    return registry.snapshot().snapshotId;
+}
+export function makeControlledClient(promptResults = [], phases = {}) {
+    const calls = [];
+    let aborts = 0;
+    let messages = 0;
+    let messagesResp = {
+        messages: [{ role: "user", parts: [{ type: "text", text: "retry this" }] }],
+    };
+    const client = {
+        session: {
+            async abort() {
+                aborts++;
+                return phases.abortResults?.shift() ?? undefined;
+            },
+            async messages() {
+                messages++;
+                const response = await (phases.messagesResults?.shift() ?? Promise.resolve(messagesResp));
+                phases.onMessagesResolved?.();
+                return response;
+            },
+            async prompt(args) {
+                calls.push({ sessionID: args.path.id, body: args.body });
+                return promptResults.shift() ?? Promise.resolve();
+            },
+        },
+    };
+    return {
+        client,
+        calls,
+        get aborts() { return aborts; },
+        get messages() { return messages; },
+        setMessages(response) { messagesResp = response; },
+    };
+}
+export function makeCreatedEvent(sessionID, parent = {}) {
+    return { event: { type: "session.created", properties: { sessionID, ...parent } } };
+}
+export function modelFor(modelID, providerID = "provider-a") {
+    return { providerID, modelID };
+}
+export function dispatchedModels(calls) {
+    return calls
+        .filter((call) => typeof call.body.modelID === "string")
+        .map((call) => `${call.body.providerID}/${call.body.modelID}`);
+}
+//# sourceMappingURL=event-handler-test-fixtures.js.map

package/dist/runtime-fallback/event-handler-test-fixtures.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"event-handler-test-fixtures.js","sourceRoot":"","sources":["../../src/runtime-fallback/event-handler-test-fixtures.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAA;AAIvC,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAA;AAUtD,MAAM,UAAU,cAAc;IAC5B,MAAM,KAAK,GAAiB,EAAE,CAAA;IAC9B,MAAM,YAAY,GAAY;QAC5B,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;YAC1D,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE;SAC7D;KACF,CAAA;IACD,MAAM,MAAM,GAAe;QACzB,OAAO,EAAE;YACP,KAAK,CAAC,KAAK;gBACT,OAAO,SAAS,CAAA;YAClB,CAAC;YACD,KAAK,CAAC,QAAQ;gBACZ,OAAO,YAAY,CAAA;YACrB,CAAC;YACD,KAAK,CAAC,MAAM,CAAC,IAA6F;gBACxG,KAAK,CAAC,IAAI,CAAC;oBACT,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE;oBACvB,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACpF,CAAC,CAAA;gBACF,OAAO,SAAS,CAAA;YAClB,CAAC;SACF;KACF,CAAA;IACD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,CAAA;AACxC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,YAAqC,EAAE;IAChE,OAAO,gBAAgB,CAAC,KAAK,CAAC;QAC5B,MAAM,EAAE;YACN,YAAY,EAAE;gBACZ,KAAK,EAAE,mBAAmB;gBAC1B,cAAc,EAAE,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;aACrD;SACF;QACD,eAAe,EAAE,SAAS;KAC3B,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAC5B,SAAiB,EACjB,KAAc,EACd,SAAkC,EAAE;IAEpC,OAAO;QACL,KAAK,EAAE;YACL,IAAI,EAAE,eAAe;YACrB,UAAU,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,MAAM,EAAE;SAC5C;KACF,CAAA;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,SAAiB;IAC7C,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAA;AACvE,CAAC;AAQD,MAAM,OAAO,oBAAoB;IACtB,KAAK,GAA2B,EAAE,CAAA;IAE3C,QAAQ,CAAC,OAAe,EAAE,GAAwB;QAChD,MAAM,IAAI,GAAyB,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,CAAA;QACrE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACrB,OAAO,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAA,CAAC,CAAC,CAAA;IACxC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAa;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QAC9B,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,0BAA0B,KAAK,EAAE,CAAC,CAAA;QAClD,IAAI,CAAC,IAAI,CAAC,SAAS;YAAE,MAAM,IAAI,CAAC,GAAG,EAAE,CAAA;IACvC,CAAC;CACF;AAED,MAAM,UAAU,QAAQ;IAKtB,IAAI,OAA4B,CAAA;IAChC,IAAI,MAAmC,CAAA;IACvC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAI,CAAC,cAAc,EAAE,aAAa,EAAE,EAAE;QAC/D,OAAO,GAAG,cAAc,CAAA;QACxB,MAAM,GAAG,aAAa,CAAA;IACxB,CAAC,CAAC,CAAA;IACF,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAA;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY;IAChC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAA;IAC3D,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;AACzB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,KAAyF,EACzF,kBAA2C,EAAE,EAC7C,qBAA8C,EAAE;IAEhD,OAAO,gBAAgB,CAAC,KAAK,CAAC;QAC5B,MAAM,EAAE,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,GAAG,kBAAkB,EAAE,EAAE,EAAE;QACpF,eAAe;KAChB,CAAC,CAAA;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB;IAClC,OAAO;QACL,EAAE,SAAS,EAAE,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,MAAe,EAAE;QACzE,EAAE,SAAS,EAAE,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,QAAiB,EAAE;QAC9E,EAAE,SAAS,EAAE,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,KAAc,EAAE;KAC5E,CAAA;AACH,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,QAAgC,EAChC,QAA6D,oBAAoB,EAAE;IAEnF,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,EAAE,CAAA;IACxC,MAAM,MAAM,GAAG,IAAI,GAAG,CAA8B;QAClD,CAAC,QAAQ,EAAE;gBACT,KAAK,EAAE,oBAAoB;gBAC3B,WAAW,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE;gBACrC,iBAAiB,EAAE,aAAa;gBAChC,aAAa,EAAE,kBAAkB;aAClC,CAAC;KACH,CAAC,CAAA;IACF,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAA;IACxD,OAAO,QAAQ,CAAC,QAAQ,EAAE,CAAC,UAAU,CAAA;AACvC,CAAC;AAQD,MAAM,UAAU,oBAAoB,CAClC,gBAAyC,EAAE,EAC3C,SAAiC,EAAE;IAEnC,MAAM,KAAK,GAAiB,EAAE,CAAA;IAC9B,IAAI,MAAM,GAAG,CAAC,CAAA;IACd,IAAI,QAAQ,GAAG,CAAC,CAAA;IAChB,IAAI,YAAY,GAAY;QAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;KAC5E,CAAA;IACD,MAAM,MAAM,GAAe;QACzB,OAAO,EAAE;YACP,KAAK,CAAC,KAAK;gBACT,MAAM,EAAE,CAAA;gBACR,OAAO,MAAM,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,SAAS,CAAA;YAClD,CAAC;YACD,KAAK,CAAC,QAAQ;gBACZ,QAAQ,EAAE,CAAA;gBACV,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,eAAe,EAAE,KAAK,EAAE,IAAI,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAA;gBACzF,MAAM,CAAC,kBAAkB,EAAE,EAAE,CAAA;gBAC7B,OAAO,QAAQ,CAAA;YACjB,CAAC;YACD,KAAK,CAAC,MAAM,CAAC,IAA6F;gBACxG,KAAK,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA;gBACxD,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAA;YACnD,CAAC;SACF;KACF,CAAA;IACD,OAAO;QACL,MAAM;QACN,KAAK;QACL,IAAI,MAAM,KAAK,OAAO,MAAM,CAAA,CAAC,CAAC;QAC9B,IAAI,QAAQ,KAAK,OAAO,QAAQ,CAAA,CAAC,CAAC;QAClC,WAAW,CAAC,QAAiB,IAAI,YAAY,GAAG,QAAQ,CAAA,CAAC,CAAC;KAC3D,CAAA;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAiB,EAAE,SAAiC,EAAE;IACrF,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,UAAU,EAAE,EAAE,SAAS,EAAE,GAAG,MAAM,EAAE,EAAE,EAAE,CAAA;AACrF,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,OAAe,EAAE,UAAU,GAAG,YAAY;IACjE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAA;AAChC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,KAAmB;IAClD,OAAO,KAAK;SACT,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC;SACvD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,UAAoB,IAAI,IAAI,CAAC,IAAI,CAAC,OAAiB,EAAE,CAAC,CAAA;AACtF,CAAC"}

package/dist/runtime-fallback/event-handler.d.ts

@@ -1,12 +1,24 @@
 import { type OcmmClient } from "./dispatcher.ts";
+import { type EffectiveRouteRegistry } from "../routing/route-registry.ts";
 import type { OcmmConfig } from "../config/schema.ts";
 import { type IdleContinuationState } from "./idle-state.ts";
+import { type Subagent429Scheduler } from "./subagent-429-controller.ts";
 export type RuntimeFallbackDeps = {
     getConfig: () => OcmmConfig;
     client?: OcmmClient;
     directory?: string;
     idleState?: IdleContinuationState;
     clearSessionIntent?: (sessionID: string) => void;
-    registeredAgentModels?: ReadonlyMap<string, string>;
+    routeRegistry?: EffectiveRouteRegistry;
+    scheduler?: Subagent429Scheduler;
+    clock?: () => number;
+    random?: () => number;
 };
+export type RuntimeFallbackRuntime = {
+    event: (input: unknown) => Promise<void>;
+    afterTask: (input: unknown, output: unknown) => Promise<void>;
+};
+export declare const SUPPRESSION_TOMBSTONE_GRACE_MS: number;
+export declare const MAX_SUPPRESSION_TOMBSTONES = 256;
+export declare function createRuntimeFallbackRuntime(deps: RuntimeFallbackDeps): RuntimeFallbackRuntime;
 export declare function createRuntimeFallbackEventHandler(deps: RuntimeFallbackDeps): (input: unknown) => Promise<void>;