pi-shadow-mind 0.1.5

package/dist/runtime.js

@@ -0,0 +1,274 @@
+import { randomUUID } from "node:crypto";
+import { buildSessionContext, } from "@earendil-works/pi-coding-agent";
+import { getAgentDir } from "@earendil-works/pi-coding-agent";
+import { Text } from "@earendil-works/pi-tui";
+import { ConfigStore } from "./config.js";
+import { EntityStore } from "./entity-store.js";
+import { registerManagementTools } from "./management-tools.js";
+import { ShadowRegistry } from "./registry.js";
+import { ReportBatcher, formatReportBatch } from "./report-batcher.js";
+import { createRandom } from "./random.js";
+import { decideHeartbeat } from "./scheduler.js";
+import { ShadowRunner, resolveShadowTools } from "./shadow-runner.js";
+import { waitForSettled } from "./shutdown-drain.js";
+export class ShadowMindRuntime {
+    pi;
+    agentDir = getAgentDir();
+    configStore = new ConfigStore(this.agentDir);
+    registry = new ShadowRegistry(this.agentDir);
+    entityStore = new EntityStore(this.registry, this.configStore.configPath);
+    runner = new ShadowRunner();
+    active = new Map();
+    recentEvents = [];
+    batcher;
+    epoch = 0;
+    modelCalls = 0;
+    paused = false;
+    panelVisible = false;
+    latestContext;
+    diagnostics = [];
+    shadowCount = 0;
+    random = Math.random;
+    constructor(pi) {
+        this.pi = pi;
+        this.batcher = new ReportBatcher(this.configStore.current.resultBatchWindowMs, (reports) => this.deliverReports(reports));
+    }
+    register() {
+        registerManagementTools(this.pi, this.entityStore, () => this.configStore.current);
+        this.registerEvents();
+        this.registerUi();
+    }
+    registerEvents() {
+        this.pi.on("session_start", async (_event, ctx) => {
+            this.latestContext = ctx;
+            this.modelCalls = 0;
+            await this.configStore.initialize();
+            this.random = createRandom(this.configStore.current.randomSeed);
+            await this.registry.initialize();
+            await this.refresh(ctx);
+            this.record("session-config", { randomSeed: this.configStore.current.randomSeed ?? "random" });
+            this.updateStatus(ctx);
+        });
+        this.pi.on("input", (event, ctx) => {
+            this.latestContext = ctx;
+            if (event.source === "extension")
+                return;
+            this.epoch += 1;
+            this.abortAll("new-user-input");
+        });
+        this.pi.on("after_provider_response", (_event, ctx) => {
+            this.latestContext = ctx;
+            this.modelCalls += 1;
+        });
+        this.pi.on("turn_end", async (_event, ctx) => {
+            this.latestContext = ctx;
+            await this.onHeartbeat(ctx);
+        });
+        this.pi.on("session_shutdown", async (event, ctx) => {
+            this.latestContext = ctx;
+            if (event.reason === "quit" && (ctx.mode === "print" || ctx.mode === "json")) {
+                await this.drainHeadless(ctx);
+            }
+            this.epoch += 1;
+            this.abortAll("session-shutdown");
+            ctx.ui.setStatus("shadow-mind", undefined);
+            ctx.ui.setWidget("shadow-mind-panel", undefined);
+        });
+    }
+    registerUi() {
+        this.pi.registerCommand("shadow", {
+            description: "Show Shadow Mind status, or pause/resume it",
+            handler: async (args, ctx) => {
+                this.latestContext = ctx;
+                const command = args.trim().toLowerCase();
+                if (command === "pause") {
+                    this.paused = true;
+                    this.abortAll("paused");
+                    ctx.ui.notify("Shadow Mind paused", "info");
+                }
+                else if (command === "resume") {
+                    this.paused = false;
+                    ctx.ui.notify("Shadow Mind resumed", "info");
+                }
+                else if (command === "status") {
+                    await this.refresh(ctx);
+                    ctx.ui.notify(this.statusLines().join("\n"), this.diagnostics.length ? "warning" : "info");
+                }
+                else if (command === "hide") {
+                    this.panelVisible = false;
+                    ctx.ui.setWidget("shadow-mind-panel", undefined);
+                }
+                else {
+                    await this.refresh(ctx);
+                    this.panelVisible = !this.panelVisible;
+                }
+                this.updateStatus(ctx);
+            },
+        });
+        this.pi.registerMessageRenderer("shadow-report", (message, _options, theme) => {
+            const content = typeof message.content === "string" ? message.content : "Shadow report";
+            const prefix = theme.fg("accent", "🐙 shadow · ");
+            return new Text(`${prefix}${content}`, 0, 0);
+        });
+    }
+    async onHeartbeat(ctx) {
+        await this.refresh(ctx);
+        if (this.paused || !ctx.model) {
+            this.record("heartbeat-skipped", { reason: this.paused ? "paused" : "no-model", modelCalls: this.modelCalls });
+            return;
+        }
+        const snapshot = await this.registry.load();
+        const fullModelId = `${ctx.model.provider}/${ctx.model.id}`;
+        const decision = decideHeartbeat({
+            heartbeatProbability: this.configStore.current.heartbeatProbability,
+            availableSlots: Math.max(0, this.configStore.current.maxParallelShadows - this.active.size),
+            shadows: snapshot.shadows,
+            activeShadowIds: new Set([...this.active.values()].map(({ shadow }) => shadow.id)),
+            mainModelId: fullModelId,
+            random: this.random,
+        });
+        this.record("heartbeat", {
+            modelCalls: this.modelCalls,
+            roll: decision.heartbeatRoll,
+            candidates: decision.candidates,
+            activated: decision.activated.map(({ shadow, roll }) => ({ id: shadow.id, roll })),
+            ...(decision.modelFiltered.length ? { modelFiltered: decision.modelFiltered } : {}),
+            ...(decision.runningExcluded.length ? { runningExcluded: decision.runningExcluded } : {}),
+        });
+        if (!decision.activated.length)
+            return;
+        const context = buildSessionContext(ctx.sessionManager.getEntries(), ctx.sessionManager.getLeafId());
+        const availableTools = new Set(this.pi.getAllTools().map((tool) => tool.name));
+        for (const { shadow } of decision.activated) {
+            this.launchShadow(ctx, shadow, ctx.model, fullModelId, context, availableTools);
+        }
+    }
+    launchShadow(ctx, shadow, mainModel, fullModelId, context, availableTools) {
+        const runId = randomUUID();
+        const runEpoch = this.epoch;
+        const { tools, missing } = resolveShadowTools(shadow.tools, availableTools);
+        this.active.set(runId, { shadow, epoch: runEpoch });
+        this.record("run-start", {
+            runId,
+            shadowId: shadow.id,
+            model: shadow.runWithModel ?? this.configStore.current.defaultShadowModel ?? fullModelId,
+            ...(missing.length ? { missingTools: missing } : {}),
+        });
+        this.updateStatus(ctx);
+        void this.runner.run({
+            shadow: structuredClone(shadow),
+            config: structuredClone(this.configStore.current),
+            epoch: runEpoch,
+            runId,
+            cwd: ctx.cwd,
+            agentDir: this.agentDir,
+            mainSystemPrompt: ctx.getSystemPrompt(),
+            messages: context.messages,
+            mainModel,
+            tools,
+            resolveModel: (id) => resolveModel(ctx, id),
+            modelAuthOk: (model) => ctx.modelRegistry.hasConfiguredAuth(model) || ctx.modelRegistry.isUsingOAuth(model),
+            mainThinkingLevel: ctx.thinkingLevel,
+            onReport: (report) => this.acceptReport(report),
+        }).then((result) => this.handleRunEnd(runId, shadow, result));
+    }
+    handleRunEnd(runId, shadow, result) {
+        this.active.delete(runId);
+        this.record("run-end", { runId, shadowId: shadow.id, ...result });
+        if (this.latestContext)
+            this.updateStatus(this.latestContext);
+    }
+    acceptReport(report) {
+        if (report.epoch !== this.epoch)
+            return;
+        this.batcher.add(report);
+    }
+    async deliverReports(reports) {
+        const current = reports.filter((report) => report.epoch === this.epoch);
+        if (!current.length)
+            return;
+        const content = formatReportBatch(current);
+        this.record("report-delivered", { runIds: current.map((report) => report.runId), count: current.length });
+        const idle = this.latestContext?.isIdle() ?? true;
+        this.pi.sendMessage({
+            customType: "shadow-report",
+            content,
+            display: true,
+            details: { reports: current.map(({ shadowId, runId }) => ({ shadowId, runId })) },
+        }, { triggerTurn: true, deliverAs: idle ? "followUp" : "steer" });
+    }
+    async refresh(ctx) {
+        const config = await this.configStore.reload();
+        const registry = await this.registry.load();
+        this.batcher.setWindow(config.config.resultBatchWindowMs);
+        this.shadowCount = registry.shadows.length;
+        this.diagnostics = [
+            ...(config.error ? [`config: ${config.error}`] : []),
+            ...registry.diagnostics.map((item) => `${item.filePath}: ${item.message}`),
+        ];
+        this.updateStatus(ctx);
+    }
+    abortAll(reason) {
+        this.runner.abortAll();
+        this.batcher.clear();
+        this.record("runs-aborted", { reason, count: this.active.size });
+    }
+    async drainHeadless(ctx) {
+        if (this.active.size === 0 && !this.batcher.hasPending)
+            return;
+        const timeoutMs = this.configStore.current.headlessDrainTimeoutSeconds * 1000;
+        this.record("headless-drain-start", { timeoutMs, active: this.active.size });
+        const result = await waitForSettled({
+            timeoutMs,
+            isSettled: () => this.active.size === 0
+                && !this.batcher.hasPending
+                && ctx.isIdle()
+                && !ctx.hasPendingMessages(),
+        });
+        this.record(result.settled ? "headless-drain-complete" : "headless-drain-timeout", {
+            durationMs: result.durationMs,
+            active: this.active.size,
+        });
+        if (!result.settled)
+            this.abortAll("headless-drain-timeout");
+    }
+    record(kind, data) {
+        const event = { at: new Date().toISOString(), kind, epoch: this.epoch, data };
+        this.recentEvents.push(event);
+        if (this.recentEvents.length > 20)
+            this.recentEvents.shift();
+        this.pi.appendEntry("shadow-mind-event", event);
+    }
+    updateStatus(ctx) {
+        ctx.ui.setStatus("shadow-mind", `🐙 ${this.active.size}${this.paused ? " paused" : ""}${this.diagnostics.length || this.hasRecentRunErrors() ? " !" : ""}`);
+        if (this.panelVisible)
+            ctx.ui.setWidget("shadow-mind-panel", this.statusLines(), { placement: "aboveEditor" });
+    }
+    hasRecentRunErrors() {
+        return this.recentEvents.slice(-3).some((event) => event.kind === "run-end" && (event.data?.reason === "error" || event.data?.reason === "timeout"));
+    }
+    statusLines() {
+        const config = this.configStore.current;
+        return [
+            `🐙 Shadow Mind · ${this.paused ? "paused" : "active"} · running ${this.active.size}/${config.maxParallelShadows}`,
+            `heartbeat ${formatNumber(config.heartbeatProbability)} · batch ${config.resultBatchWindowMs}ms · timeout ${config.defaultShadowTimeoutSeconds}s · drain ${config.headlessDrainTimeoutSeconds}s · thinking ${config.defaultThinkingLevel}`,
+            `definitions: ${this.shadowCount} valid · ${this.diagnostics.length} invalid`,
+            ...this.recentEvents.slice(-5).map((event) => {
+                const failed = event.kind === "run-end" && (event.data?.reason === "error" || event.data?.reason === "timeout");
+                const detail = failed ? ` ${event.data?.error ?? event.data?.reason}` : "";
+                return `${event.at.slice(11, 19)} ${event.kind}${detail}`;
+            }),
+            "Commands: /shadow pause | resume | status | hide",
+        ];
+    }
+}
+function resolveModel(ctx, fullId) {
+    const separator = fullId.indexOf("/");
+    if (separator <= 0 || separator === fullId.length - 1)
+        return undefined;
+    return ctx.modelRegistry.find(fullId.slice(0, separator), fullId.slice(separator + 1));
+}
+function formatNumber(value) {
+    return Number(value.toFixed(3)).toString();
+}
+//# sourceMappingURL=runtime.js.map

package/dist/runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../src/runtime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACL,mBAAmB,GAGpB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,IAAI,EAAE,MAAM,wBAAwB,CAAC;AAE9C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAwB,MAAM,oBAAoB,CAAC;AAC5F,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAGrD,MAAM,OAAO,iBAAiB;IAkBC;IAjBZ,QAAQ,GAAG,WAAW,EAAE,CAAC;IACzB,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7C,QAAQ,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7C,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAC1E,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IAC5B,MAAM,GAAG,IAAI,GAAG,EAAuD,CAAC;IACxE,YAAY,GAAmB,EAAE,CAAC;IAClC,OAAO,CAAgB;IAChC,KAAK,GAAG,CAAC,CAAC;IACV,UAAU,GAAG,CAAC,CAAC;IACf,MAAM,GAAG,KAAK,CAAC;IACf,YAAY,GAAG,KAAK,CAAC;IACrB,aAAa,CAAoB;IACjC,WAAW,GAAa,EAAE,CAAC;IAC3B,WAAW,GAAG,CAAC,CAAC;IAChB,MAAM,GAAiB,IAAI,CAAC,MAAM,CAAC;IAE3C,YAA6B,EAAgB;QAAhB,OAAE,GAAF,EAAE,CAAc;QAC3C,IAAI,CAAC,OAAO,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,mBAAmB,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC;IAC5H,CAAC;IAED,QAAQ;QACN,uBAAuB,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACnF,IAAI,CAAC,cAAc,EAAE,CAAC;QACtB,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;YAChD,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;YACzB,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;YACpB,MAAM,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;YACpC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAChE,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACxB,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,UAAU,IAAI,QAAQ,EAAE,CAAC,CAAC;YAC/F,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;YACjC,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;YACzB,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW;gBAAE,OAAO;YACzC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,yBAAyB,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;YACpD,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;YACzB,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;YAC3C,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;YACzB,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YAClD,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;YACzB,IAAI,KAAK,CAAC,MAAM,KAAK,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,OAAO,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;gBAC7E,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;YAChC,CAAC;YACD,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;YAClC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;YAC3C,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE;YAChC,WAAW,EAAE,6CAA6C;YAC1D,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;gBAC3B,IAAI,CAAC,aAAa,GAAG,GAAG,CAAC;gBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;gBAC1C,IAAI,OAAO,KAAK,OAAO,EAAE,CAAC;oBACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;oBACnB,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;oBACxB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,oBAAoB,EAAE,MAAM,CAAC,CAAC;gBAC9C,CAAC;qBAAM,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAChC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;oBACpB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;gBAC/C,CAAC;qBAAM,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBAChC,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;oBACxB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;gBAC7F,CAAC;qBAAM,IAAI,OAAO,KAAK,MAAM,EAAE,CAAC;oBAC9B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;oBAC1B,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;gBACnD,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;oBACxB,IAAI,CAAC,YAAY,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC;gBACzC,CAAC;gBACD,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACzB,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,uBAAuB,CAAC,eAAe,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE;YAC5E,MAAM,OAAO,GAAG,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;YACxF,MAAM,MAAM,GAAG,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;YAClD,OAAO,IAAI,IAAI,CAAC,GAAG,MAAM,GAAG,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,GAAqB;QAC7C,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;YAC/G,OAAO;QACT,CAAC;QACD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC5C,MAAM,WAAW,GAAG,GAAG,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;QAC5D,MAAM,QAAQ,GAAG,eAAe,CAAC;YAC/B,oBAAoB,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,oBAAoB;YACnE,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YAC3F,OAAO,EAAE,QAAQ,CAAC,OAAO;YACzB,eAAe,EAAE,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAClF,WAAW,EAAE,WAAW;YACxB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,IAAI,EAAE,QAAQ,CAAC,aAAa;YAC5B,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,SAAS,EAAE,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;YAClF,GAAG,CAAC,QAAQ,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnF,GAAG,CAAC,QAAQ,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,QAAQ,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC1F,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM;YAAE,OAAO;QAEvC,MAAM,OAAO,GAAG,mBAAmB,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC,CAAC;QACrG,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC/E,KAAK,MAAM,EAAE,MAAM,EAAE,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC5C,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;QAClF,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,GAAqB,EAAE,MAAwB,EAAE,SAAqB,EAAE,WAAmB,EAAE,OAA+C,EAAE,cAA2B;QAC5L,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC;QAC5B,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAC5E,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE;YACvB,KAAK;YACL,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,KAAK,EAAE,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,kBAAkB,IAAI,WAAW;YACxF,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACrD,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QACvB,KAAK,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;YACnB,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC;YAC/B,MAAM,EAAE,eAAe,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;YACjD,KAAK,EAAE,QAAQ;YACf,KAAK;YACL,GAAG,EAAE,GAAG,CAAC,GAAG;YACZ,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,gBAAgB,EAAE,GAAG,CAAC,eAAe,EAAE;YACvC,QAAQ,EAAE,OAAO,CAAC,QAAgD;YAClE,SAAS;YACT,KAAK;YACL,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,YAAY,CAAC,GAAG,EAAE,EAAE,CAAC;YAC3C,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,CAAC,aAAa,CAAC,iBAAiB,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,aAAa,CAAC,YAAY,CAAC,KAAK,CAAC;YAC3G,iBAAiB,EAAE,GAAG,CAAC,aAAa;YACpC,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC;SAChD,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAChE,CAAC;IAEO,YAAY,CAAC,KAAa,EAAE,MAAwB,EAAE,MAAuB;QACnF,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAClE,IAAI,IAAI,CAAC,aAAa;YAAE,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAChE,CAAC;IAEO,YAAY,CAAC,MAAoB;QACvC,IAAI,MAAM,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK;YAAE,OAAO;QACxC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,OAAuB;QAClD,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC;QACxE,IAAI,CAAC,OAAO,CAAC,MAAM;YAAE,OAAO;QAC5B,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1G,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,EAAE,MAAM,EAAE,IAAI,IAAI,CAAC;QAClD,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC;YAClB,UAAU,EAAE,eAAe;YAC3B,OAAO;YACP,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE;SAClF,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;IACpE,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,GAAqB;QACzC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QAC1D,IAAI,CAAC,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC;QAC3C,IAAI,CAAC,WAAW,GAAG;YACjB,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACpD,GAAG,QAAQ,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC;SAC3E,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAEO,QAAQ,CAAC,MAAc;QAC7B,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACvB,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,GAAqB;QAC/C,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU;YAAE,OAAO;QAC/D,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,2BAA2B,GAAG,IAAI,CAAC;QAC9E,IAAI,CAAC,MAAM,CAAC,sBAAsB,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7E,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;YAClC,SAAS;YACT,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC;mBAClC,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU;mBACxB,GAAG,CAAC,MAAM,EAAE;mBACZ,CAAC,GAAG,CAAC,kBAAkB,EAAE;SAC/B,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,wBAAwB,EAAE;YACjF,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;SACzB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;IAC/D,CAAC;IAEO,MAAM,CAAC,IAAY,EAAE,IAA8B;QACzD,MAAM,KAAK,GAAiB,EAAE,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC;QAC5F,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,EAAE;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC7D,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;IAClD,CAAC;IAEO,YAAY,CAAC,GAAqB;QACxC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5J,IAAI,IAAI,CAAC,YAAY;YAAE,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,mBAAmB,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CAAC;IACjH,CAAC;IAEO,kBAAkB;QACxB,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC;IACvJ,CAAC;IAEO,WAAW;QACjB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC;QACxC,OAAO;YACL,oBAAoB,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,cAAc,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,kBAAkB,EAAE;YAClH,aAAa,YAAY,CAAC,MAAM,CAAC,oBAAoB,CAAC,YAAY,MAAM,CAAC,mBAAmB,gBAAgB,MAAM,CAAC,2BAA2B,aAAa,MAAM,CAAC,2BAA2B,gBAAgB,MAAM,CAAC,oBAAoB,EAAE;YAC1O,gBAAgB,IAAI,CAAC,WAAW,YAAY,IAAI,CAAC,WAAW,CAAC,MAAM,UAAU;YAC7E,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC;gBAChH,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3E,OAAO,GAAG,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,IAAI,GAAG,MAAM,EAAE,CAAC;YAC5D,CAAC,CAAC;YACF,kDAAkD;SACnD,CAAC;IACJ,CAAC;CACF;AAED,SAAS,YAAY,CAAC,GAAqB,EAAE,MAAc;IACzD,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,IAAI,SAAS,IAAI,CAAC,IAAI,SAAS,KAAK,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IACxE,OAAO,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC;AACzF,CAAC;AAED,SAAS,YAAY,CAAC,KAAa;IACjC,OAAO,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;AAC7C,CAAC"}

package/dist/scheduler.d.ts

@@ -0,0 +1,10 @@
+import type { HeartbeatDecision, ShadowDefinition } from "./types.js";
+export declare function decideHeartbeat(options: {
+    heartbeatProbability: number;
+    availableSlots: number;
+    shadows: readonly ShadowDefinition[];
+    activeShadowIds: ReadonlySet<string>;
+    mainModelId: string;
+    random?: () => number;
+}): HeartbeatDecision;
+export declare function matchesModel(shadow: ShadowDefinition, fullModelId: string): boolean;

package/dist/scheduler.js

@@ -0,0 +1,46 @@
+export function decideHeartbeat(options) {
+    const random = options.random ?? Math.random;
+    const heartbeatRoll = random();
+    if (heartbeatRoll >= options.heartbeatProbability || options.availableSlots <= 0) {
+        return { heartbeatRoll, activated: [], candidates: [], modelFiltered: [], runningExcluded: [] };
+    }
+    const modelFiltered = [];
+    const runningExcluded = [];
+    const rolls = options.shadows
+        .filter((shadow) => {
+        if (!shadow.enabled)
+            return false;
+        if (options.activeShadowIds.has(shadow.id)) {
+            runningExcluded.push(shadow.id);
+            return false;
+        }
+        if (!matchesModel(shadow, options.mainModelId)) {
+            modelFiltered.push(shadow.id);
+            return false;
+        }
+        return true;
+    })
+        .map((shadow) => ({ shadow, roll: random() }));
+    const hits = rolls.filter(({ shadow, roll }) => roll < shadow.activationProbability);
+    const selected = sample(hits, Math.min(options.availableSlots, hits.length), random);
+    const selectedIds = new Set(selected.map(({ shadow }) => shadow.id));
+    return {
+        heartbeatRoll,
+        activated: selected,
+        candidates: rolls.map(({ shadow, roll }) => ({ shadowId: shadow.id, roll, selected: selectedIds.has(shadow.id) })),
+        modelFiltered,
+        runningExcluded,
+    };
+}
+export function matchesModel(shadow, fullModelId) {
+    return shadow.activeForModels.includes("*") || shadow.activeForModels.includes(fullModelId);
+}
+function sample(values, count, random) {
+    const copy = [...values];
+    for (let index = copy.length - 1; index > 0; index -= 1) {
+        const swap = Math.floor(random() * (index + 1));
+        [copy[index], copy[swap]] = [copy[swap], copy[index]];
+    }
+    return copy.slice(0, count);
+}
+//# sourceMappingURL=scheduler.js.map

package/dist/scheduler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scheduler.js","sourceRoot":"","sources":["../src/scheduler.ts"],"names":[],"mappings":"AAEA,MAAM,UAAU,eAAe,CAAC,OAO/B;IACC,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC;IAC7C,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC;IAC/B,IAAI,aAAa,IAAI,OAAO,CAAC,oBAAoB,IAAI,OAAO,CAAC,cAAc,IAAI,CAAC,EAAE,CAAC;QACjF,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;IAClG,CAAC;IAED,MAAM,aAAa,GAAa,EAAE,CAAC;IACnC,MAAM,eAAe,GAAa,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO;SAC1B,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACjB,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,KAAK,CAAC;QAClC,IAAI,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;YAC3C,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAChC,OAAO,KAAK,CAAC;QACf,CAAC;QACD,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;YAC/C,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAC9B,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC;IACjD,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,GAAG,MAAM,CAAC,qBAAqB,CAAC,CAAC;IAErF,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,CAAC;IACrF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IACrE,OAAO;QACL,aAAa;QACb,SAAS,EAAE,QAAQ;QACnB,UAAU,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QAClH,aAAa;QACb,eAAe;KAChB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,MAAwB,EAAE,WAAmB;IACxE,OAAO,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AAC9F,CAAC;AAED,SAAS,MAAM,CAAI,MAAoB,EAAE,KAAa,EAAE,MAAoB;IAC1E,MAAM,IAAI,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;IACzB,KAAK,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;IACxD,CAAC;IACD,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;AAC9B,CAAC"}

package/dist/shadow-runner.d.ts

@@ -0,0 +1,93 @@
+import { createAgentSession } from "@earendil-works/pi-coding-agent";
+import type { Model } from "@earendil-works/pi-ai";
+import type { ThinkingLevel } from "@earendil-works/pi-agent-core";
+import type { ShadowConfig, ShadowDefinition, ShadowReport } from "./types.js";
+/**
+ * Resolve the final tool allowlist for a Shadow run against the main session's
+ * tool registry, per the design contract:
+ *
+ * - the final set is `readOnlyTools defaults + whitelist + report_to_main`;
+ * - names that do not exist in the main session are dropped (ignored) and
+ *   returned as `missing`, while the rest are provided normally;
+ * - the built-in `report_to_main` is always kept.
+ */
+export declare function resolveShadowTools(whitelist: readonly string[], available: ReadonlySet<string>, defaults?: readonly string[]): {
+    tools: string[];
+    missing: string[];
+};
+export interface ShadowRunRequest {
+    shadow: ShadowDefinition;
+    config: ShadowConfig;
+    epoch: number;
+    runId: string;
+    cwd: string;
+    agentDir: string;
+    mainSystemPrompt: string;
+    messages: readonly Record<string, unknown>[];
+    mainModel: Model<any>;
+    /** Final tool allowlist, already resolved against the main session registry. */
+    tools: string[];
+    resolveModel(fullModelId: string): Model<any> | undefined;
+    /** Optional auth check for explicitly configured shadow models (runtime supplies it). */
+    modelAuthOk?(model: Model<any>): boolean;
+    /** The main session's effective thinking level at activation, used as the final fallback. */
+    mainThinkingLevel?: ThinkingLevel;
+    onReport(report: ShadowReport): void;
+}
+export interface ToolStat {
+    tool: string;
+    calls: number;
+    failures: number;
+}
+export interface ShadowRunResult {
+    reason: "report" | "silent" | "timeout" | "aborted" | "error";
+    error?: string;
+    durationMs: number;
+    toolNames: string[];
+    /** Requested tools that did not materialize in the shadow session. */
+    missingTools: string[];
+    toolCalls: number;
+    toolFailures: number;
+    /** Per-tool usage summary (called tools only). */
+    toolStats: ToolStat[];
+    /** The thinking level actually used for this run (after fallback resolution). */
+    thinkingLevel?: string;
+    sessionFile?: string;
+}
+type ShadowSession = Awaited<ReturnType<typeof createAgentSession>>["session"];
+/** Unify the three run() return paths so result shape changes touch one place. */
+export declare function buildRunResult(options: {
+    reason: ShadowRunResult["reason"];
+    error?: string;
+    durationMs: number;
+    session?: ShadowSession;
+    /** Message count before the shadow's own run started; only later tool results are counted. */
+    baseMessageCount: number;
+    missingTools: string[];
+    thinkingLevel?: string;
+}): ShadowRunResult;
+export declare class ShadowRunner {
+    private readonly controllers;
+    run(request: ShadowRunRequest): Promise<ShadowRunResult>;
+    /** Build the shadow AgentSession (loader, report tool, allowlist) and validate it. */
+    private bootstrapSession;
+    abort(runId: string): void;
+    abortAll(): void;
+}
+export declare function toolMetrics(messages: readonly {
+    role?: string;
+    toolName?: string;
+    isError?: boolean;
+}[]): {
+    toolCalls: number;
+    toolFailures: number;
+    toolStats: ToolStat[];
+};
+/**
+ * Resolve the thinking level for this run: shadow config → plugin default →
+ * the main session's effective level. The first candidate the model supports
+ * (per its thinkingLevelMap; null marks unsupported, a missing key or map means
+ * provider default, i.e. supported) wins. Fails only when none are supported.
+ */
+export declare function resolveRunThinkingLevel(model: Model<any>, request: ShadowRunRequest): ThinkingLevel;
+export {};