@assemblyline-agents/runtime 0.1.0

package/dist/hook-dispatch.js

@@ -0,0 +1,37 @@
+export function createHookDispatcher(options) {
+    let loaded;
+    const load = async () => {
+        const hooks = [];
+        for (const hook of options.hooks) {
+            try {
+                const module = await options.loadModule(hook.source);
+                const events = module.default?.events;
+                if (!events || typeof events !== "object") {
+                    options.logger.warn("hook.invalid", { hook: hook.name, detail: "default export must declare an events map" });
+                    continue;
+                }
+                hooks.push({ name: hook.name, events });
+            }
+            catch (error) {
+                options.logger.warn("hook.load_failed", { hook: hook.name, error: String(error) });
+            }
+        }
+        return hooks;
+    };
+    return async (runId, event) => {
+        loaded ??= load();
+        const hooks = await loaded;
+        for (const hook of hooks) {
+            const handlers = [hook.events[event.type], hook.events["*"]].filter((handler) => typeof handler === "function");
+            for (const handler of handlers) {
+                try {
+                    await handler({ type: event.type, timestamp: event.timestamp, data: event.data, ...(event.sequence !== undefined ? { sequence: event.sequence } : {}) }, { runId, agent: { name: options.agentName } });
+                }
+                catch (error) {
+                    options.logger.warn("hook.failed", { hook: hook.name, eventType: event.type, error: String(error) });
+                }
+            }
+        }
+    };
+}
+//# sourceMappingURL=hook-dispatch.js.map

package/dist/hook-dispatch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hook-dispatch.js","sourceRoot":"","sources":["../src/hook-dispatch.ts"],"names":[],"mappings":"AA0BA,MAAM,UAAU,oBAAoB,CAAC,OAA8B;IACjE,IAAI,MAAyC,CAAC;IAE9C,MAAM,IAAI,GAAG,KAAK,IAA2B,EAAE;QAC7C,MAAM,KAAK,GAAiB,EAAE,CAAC;QAC/B,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAA+B,IAAI,CAAC,MAAM,CAAC,CAAC;gBACnF,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC;gBACtC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;oBAC1C,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,2CAA2C,EAAE,CAAC,CAAC;oBAC9G,SAAS;gBACX,CAAC;gBACD,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACrF,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;IAEF,OAAO,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,KAAK,IAAI,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC;QAC3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CACjE,CAAC,OAAO,EAA+B,EAAE,CAAC,OAAO,OAAO,KAAK,UAAU,CACxE,CAAC;YACF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,IAAI,CAAC;oBACH,MAAM,OAAO,CACX,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EACzI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,CAC9C,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACvG,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}

package/dist/http.d.ts

@@ -0,0 +1,26 @@
+import type { JsonObject } from "@assemblyline-agents/core";
+export interface HttpRequestPolicy {
+    timeoutMs?: number;
+    maxAttempts?: number;
+    minDelayMs?: number;
+    maxDelayMs?: number;
+    maxResponseBytes?: number;
+    retryOn?: (response: Response) => boolean;
+    /** Override the retry delay for a response-based retry (e.g. Telegram's parameters.retry_after in the body). */
+    retryDelayMsFor?: (response: Response, attempt: number) => number | undefined;
+}
+export interface HttpJsonResult {
+    response: Response;
+    status: number;
+    ok: boolean;
+    json: JsonObject;
+    text: string;
+}
+export declare class HttpResponseTooLargeError extends Error {
+    constructor(limitBytes: number);
+}
+export declare function retryAfterMs(response: Response, fallbackMs: number, capMs: number): number;
+export declare function fetchWithPolicy(fetchImpl: typeof fetch, url: string, init?: RequestInit, policy?: HttpRequestPolicy): Promise<Response>;
+export declare function readBodyCapped(response: Response, maxBytes?: number): Promise<string>;
+export declare function fetchJson(fetchImpl: typeof fetch, url: string, init?: RequestInit, policy?: HttpRequestPolicy): Promise<HttpJsonResult>;
+//# sourceMappingURL=http.d.ts.map

package/dist/http.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../src/http.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAQ5D,MAAM,WAAW,iBAAiB;IAChC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,KAAK,OAAO,CAAC;IAC1C,gHAAgH;IAChH,eAAe,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,KAAK,MAAM,GAAG,SAAS,CAAC;CAC/E;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,QAAQ,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,EAAE,EAAE,OAAO,CAAC;IACZ,IAAI,EAAE,UAAU,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,qBAAa,yBAA0B,SAAQ,KAAK;gBACtC,UAAU,EAAE,MAAM;CAI/B;AAED,wBAAgB,YAAY,CAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,CAS1F;AAED,wBAAsB,eAAe,CACnC,SAAS,EAAE,OAAO,KAAK,EACvB,GAAG,EAAE,MAAM,EACX,IAAI,GAAE,WAAgB,EACtB,MAAM,GAAE,iBAAsB,GAC7B,OAAO,CAAC,QAAQ,CAAC,CAqCnB;AAED,wBAAsB,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,GAAE,MAAmC,GAAG,OAAO,CAAC,MAAM,CAAC,CAmBvH;AAED,wBAAsB,SAAS,CAC7B,SAAS,EAAE,OAAO,KAAK,EACvB,GAAG,EAAE,MAAM,EACX,IAAI,GAAE,WAAgB,EACtB,MAAM,GAAE,iBAAsB,GAC7B,OAAO,CAAC,cAAc,CAAC,CAazB"}

package/dist/http.js

@@ -0,0 +1,114 @@
+const DEFAULT_TIMEOUT_MS = 15_000;
+const DEFAULT_MAX_ATTEMPTS = 3;
+const DEFAULT_MIN_DELAY_MS = 250;
+const DEFAULT_MAX_DELAY_MS = 30_000;
+const DEFAULT_MAX_RESPONSE_BYTES = 512 * 1024;
+export class HttpResponseTooLargeError extends Error {
+    constructor(limitBytes) {
+        super(`HTTP response exceeded the ${limitBytes}-byte limit.`);
+        this.name = "HttpResponseTooLargeError";
+    }
+}
+export function retryAfterMs(response, fallbackMs, capMs) {
+    const header = response.headers.get("retry-after");
+    if (header) {
+        const seconds = Number(header);
+        if (Number.isFinite(seconds) && seconds >= 0)
+            return Math.min(seconds * 1000, capMs);
+        const dateMs = Date.parse(header);
+        if (Number.isFinite(dateMs))
+            return Math.min(Math.max(dateMs - Date.now(), 0), capMs);
+    }
+    return Math.min(fallbackMs, capMs);
+}
+export async function fetchWithPolicy(fetchImpl, url, init = {}, policy = {}) {
+    const timeoutMs = policy.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+    const maxAttempts = Math.max(1, policy.maxAttempts ?? DEFAULT_MAX_ATTEMPTS);
+    const minDelayMs = policy.minDelayMs ?? DEFAULT_MIN_DELAY_MS;
+    const maxDelayMs = policy.maxDelayMs ?? DEFAULT_MAX_DELAY_MS;
+    const shouldRetry = policy.retryOn ?? ((response) => response.status === 429 || response.status >= 500);
+    let lastError;
+    for (let attempt = 1; attempt <= maxAttempts; attempt += 1) {
+        const controller = new AbortController();
+        const timer = setTimeout(() => controller.abort(new Error(`HTTP request timed out after ${timeoutMs}ms.`)), timeoutMs);
+        const upstreamAbort = init.signal;
+        const onUpstreamAbort = () => controller.abort(upstreamAbort?.reason);
+        if (upstreamAbort) {
+            if (upstreamAbort.aborted)
+                controller.abort(upstreamAbort.reason);
+            else
+                upstreamAbort.addEventListener("abort", onUpstreamAbort, { once: true });
+        }
+        try {
+            const response = await fetchImpl(url, { ...init, signal: controller.signal });
+            if (!shouldRetry(response) || attempt >= maxAttempts)
+                return response;
+            const overrideDelay = policy.retryDelayMsFor?.(response, attempt);
+            const delayMs = overrideDelay !== undefined
+                ? Math.min(overrideDelay, maxDelayMs)
+                : retryAfterMs(response, backoffDelayMs(attempt, minDelayMs, maxDelayMs), maxDelayMs);
+            await response.body?.cancel().catch(() => undefined);
+            await sleep(delayMs);
+        }
+        catch (error) {
+            if (upstreamAbort?.aborted)
+                throw error;
+            lastError = error;
+            if (attempt >= maxAttempts)
+                throw error;
+            await sleep(backoffDelayMs(attempt, minDelayMs, maxDelayMs));
+        }
+        finally {
+            clearTimeout(timer);
+            if (upstreamAbort)
+                upstreamAbort.removeEventListener("abort", onUpstreamAbort);
+        }
+    }
+    throw lastError instanceof Error ? lastError : new Error("HTTP request failed.");
+}
+export async function readBodyCapped(response, maxBytes = DEFAULT_MAX_RESPONSE_BYTES) {
+    const body = response.body;
+    if (!body)
+        return "";
+    const reader = body.getReader();
+    const chunks = [];
+    let total = 0;
+    for (;;) {
+        const { done, value } = await reader.read();
+        if (done)
+            break;
+        if (value) {
+            total += value.byteLength;
+            if (total > maxBytes) {
+                await reader.cancel().catch(() => undefined);
+                throw new HttpResponseTooLargeError(maxBytes);
+            }
+            chunks.push(value);
+        }
+    }
+    return Buffer.concat(chunks).toString("utf8");
+}
+export async function fetchJson(fetchImpl, url, init = {}, policy = {}) {
+    const response = await fetchWithPolicy(fetchImpl, url, init, policy);
+    const text = await readBodyCapped(response, policy.maxResponseBytes ?? DEFAULT_MAX_RESPONSE_BYTES);
+    let json = {};
+    if (text.length > 0) {
+        try {
+            const parsed = JSON.parse(text);
+            if (parsed && typeof parsed === "object" && !Array.isArray(parsed))
+                json = parsed;
+        }
+        catch {
+            // Non-JSON bodies are surfaced through `text`; callers decide how strict to be.
+        }
+    }
+    return { response, status: response.status, ok: response.ok, json, text };
+}
+function backoffDelayMs(attempt, minDelayMs, maxDelayMs) {
+    const exponential = Math.min(minDelayMs * 2 ** (attempt - 1), maxDelayMs);
+    return Math.floor(exponential / 2 + Math.random() * (exponential / 2));
+}
+function sleep(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+//# sourceMappingURL=http.js.map

package/dist/http.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../src/http.ts"],"names":[],"mappings":"AAEA,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAC/B,MAAM,oBAAoB,GAAG,GAAG,CAAC;AACjC,MAAM,oBAAoB,GAAG,MAAM,CAAC;AACpC,MAAM,0BAA0B,GAAG,GAAG,GAAG,IAAI,CAAC;AAqB9C,MAAM,OAAO,yBAA0B,SAAQ,KAAK;IAClD,YAAY,UAAkB;QAC5B,KAAK,CAAC,8BAA8B,UAAU,cAAc,CAAC,CAAC;QAC9D,IAAI,CAAC,IAAI,GAAG,2BAA2B,CAAC;IAC1C,CAAC;CACF;AAED,MAAM,UAAU,YAAY,CAAC,QAAkB,EAAE,UAAkB,EAAE,KAAa;IAChF,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACnD,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,GAAG,IAAI,EAAE,KAAK,CAAC,CAAC;QACrF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAClC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IACxF,CAAC;IACD,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,SAAuB,EACvB,GAAW,EACX,OAAoB,EAAE,EACtB,SAA4B,EAAE;IAE9B,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,kBAAkB,CAAC;IACzD,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,IAAI,oBAAoB,CAAC,CAAC;IAC5E,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,oBAAoB,CAAC;IAC7D,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,oBAAoB,CAAC;IAC7D,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,QAAkB,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC,CAAC;IAElH,IAAI,SAAkB,CAAC;IACvB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,IAAI,CAAC,EAAE,CAAC;QAC3D,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,gCAAgC,SAAS,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QACvH,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC;QAClC,MAAM,eAAe,GAAG,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QACtE,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,aAAa,CAAC,OAAO;gBAAE,UAAU,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;;gBAC7D,aAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,eAAe,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;YAC9E,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,OAAO,IAAI,WAAW;gBAAE,OAAO,QAAQ,CAAC;YACtE,MAAM,aAAa,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAClE,MAAM,OAAO,GAAG,aAAa,KAAK,SAAS;gBACzC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,UAAU,CAAC;gBACrC,CAAC,CAAC,YAAY,CAAC,QAAQ,EAAE,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,CAAC,EAAE,UAAU,CAAC,CAAC;YACxF,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACrD,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,aAAa,EAAE,OAAO;gBAAE,MAAM,KAAK,CAAC;YACxC,SAAS,GAAG,KAAK,CAAC;YAClB,IAAI,OAAO,IAAI,WAAW;gBAAE,MAAM,KAAK,CAAC;YACxC,MAAM,KAAK,CAAC,cAAc,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC;QAC/D,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,aAAa;gBAAE,aAAa,CAAC,mBAAmB,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IACD,MAAM,SAAS,YAAY,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;AACnF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,QAAkB,EAAE,WAAmB,0BAA0B;IACpG,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IAC3B,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,CAAC;IACrB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,SAAS,CAAC;QACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,IAAI;YAAE,MAAM;QAChB,IAAI,KAAK,EAAE,CAAC;YACV,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC;YAC1B,IAAI,KAAK,GAAG,QAAQ,EAAE,CAAC;gBACrB,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAC7C,MAAM,IAAI,yBAAyB,CAAC,QAAQ,CAAC,CAAC;YAChD,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,SAAS,CAC7B,SAAuB,EACvB,GAAW,EACX,OAAoB,EAAE,EACtB,SAA4B,EAAE;IAE9B,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;IACrE,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,QAAQ,EAAE,MAAM,CAAC,gBAAgB,IAAI,0BAA0B,CAAC,CAAC;IACnG,IAAI,IAAI,GAAe,EAAE,CAAC;IAC1B,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAY,CAAC;YAC3C,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;gBAAE,IAAI,GAAG,MAAoB,CAAC;QAClG,CAAC;QAAC,MAAM,CAAC;YACP,gFAAgF;QAClF,CAAC;IACH,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC,MAAM,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;AAC5E,CAAC;AAED,SAAS,cAAc,CAAC,OAAe,EAAE,UAAkB,EAAE,UAAkB;IAC7E,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,GAAG,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;IAC1E,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}

package/dist/in-memory-conversation-turn-store.d.ts

@@ -0,0 +1,29 @@
+import type { ConversationTurnRecord, ConversationTurnStatus, ConversationTurnStore } from "./state.js";
+export declare class InMemoryConversationTurnStore implements ConversationTurnStore {
+    private readonly persist;
+    turns: Map<string, ConversationTurnRecord>;
+    constructor(persist?: () => Promise<void>);
+    enqueueConversationTurn(input: Parameters<ConversationTurnStore["enqueueConversationTurn"]>[0]): Promise<{
+        turn: ConversationTurnRecord;
+        inserted: boolean;
+    }>;
+    leaseNextConversationTurn(options: Parameters<ConversationTurnStore["leaseNextConversationTurn"]>[0]): Promise<ConversationTurnRecord | undefined>;
+    attachConversationTurnRun(id: string, leaseToken: string, runId: string): Promise<ConversationTurnRecord>;
+    settleConversationTurn(id: string, leaseToken: string, status: "blocked" | "completed" | "failed" | "cancelled", options?: {
+        runId?: string;
+        error?: string;
+    }): Promise<ConversationTurnRecord>;
+    settleConversationTurnByRunId(runId: string, status: "blocked" | "completed" | "failed" | "cancelled", options?: {
+        error?: string;
+    }): Promise<ConversationTurnRecord | undefined>;
+    retryConversationTurn(id: string, leaseToken: string, error: string): Promise<ConversationTurnRecord>;
+    recoverExpiredConversationTurns(now?: Date): Promise<ConversationTurnRecord[]>;
+    listConversationTurns(options?: {
+        agentScope?: string;
+        conversationId?: string;
+        status?: ConversationTurnStatus;
+    }): Promise<ConversationTurnRecord[]>;
+    private requireLease;
+    private settle;
+}
+//# sourceMappingURL=in-memory-conversation-turn-store.d.ts.map

package/dist/in-memory-conversation-turn-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory-conversation-turn-store.d.ts","sourceRoot":"","sources":["../src/in-memory-conversation-turn-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,sBAAsB,EACtB,sBAAsB,EACtB,qBAAqB,EACtB,MAAM,YAAY,CAAC;AAapB,qBAAa,6BAA8B,YAAW,qBAAqB;IAG7D,OAAO,CAAC,QAAQ,CAAC,OAAO;IAFpC,KAAK,sCAA6C;gBAErB,OAAO,GAAE,MAAM,OAAO,CAAC,IAAI,CAAa;IAE/D,uBAAuB,CAAC,KAAK,EAAE,UAAU,CAAC,qBAAqB,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC;;;;IA6B9F,yBAAyB,CAC7B,OAAO,EAAE,UAAU,CAAC,qBAAqB,CAAC,2BAA2B,CAAC,CAAC,CAAC,CAAC,CAAC,GACzE,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAiCxC,yBAAyB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAQzG,sBAAsB,CAC1B,EAAE,EAAE,MAAM,EACV,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,EACxD,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAO,GAC/C,OAAO,CAAC,sBAAsB,CAAC;IAK5B,6BAA6B,CACjC,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,EACxD,OAAO,GAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAO,GAC/B,OAAO,CAAC,sBAAsB,GAAG,SAAS,CAAC;IAQxC,qBAAqB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,CAAC;IAoBrG,+BAA+B,CAAC,GAAG,OAAa,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAyBpF,qBAAqB,CAAC,OAAO,GAAE;QACnC,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,MAAM,CAAC,EAAE,sBAAsB,CAAC;KAC5B,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAS1C,OAAO,CAAC,YAAY;YASN,MAAM;CAkBrB"}

package/dist/in-memory-conversation-turn-store.js

@@ -0,0 +1,166 @@
+import { randomUUID } from "node:crypto";
+import { noPersist, timestamp } from "./state-store-helpers.js";
+const ACTIVE_STATUSES = new Set(["running", "blocked"]);
+function clone(record) {
+    return structuredClone(record);
+}
+function retryDelayMs(attemptCount) {
+    return Math.min(60_000, 1_000 * (2 ** Math.max(0, attemptCount - 1)));
+}
+export class InMemoryConversationTurnStore {
+    persist;
+    turns = new Map();
+    constructor(persist = noPersist) {
+        this.persist = persist;
+    }
+    async enqueueConversationTurn(input) {
+        const existing = [...this.turns.values()].find((turn) => turn.agentScope === input.agentScope &&
+            turn.idempotencyScope === input.idempotencyScope &&
+            turn.idempotencyKey === input.idempotencyKey);
+        if (existing)
+            return { turn: clone(existing), inserted: false };
+        const now = timestamp();
+        const turn = {
+            id: randomUUID(),
+            agentScope: input.agentScope,
+            conversationId: input.conversationId,
+            channel: input.channel,
+            eventId: input.eventId,
+            payload: structuredClone(input.payload),
+            idempotencyScope: input.idempotencyScope,
+            idempotencyKey: input.idempotencyKey,
+            status: "pending",
+            attemptCount: 0,
+            maxAttempts: Math.max(1, Math.floor(input.maxAttempts ?? 5)),
+            nextAttemptAt: now,
+            createdAt: now,
+            updatedAt: now
+        };
+        this.turns.set(turn.id, turn);
+        await this.persist();
+        return { turn: clone(turn), inserted: true };
+    }
+    async leaseNextConversationTurn(options) {
+        const now = options.now ?? new Date();
+        const nowIso = now.toISOString();
+        const activeConversations = new Set([...this.turns.values()]
+            .filter((turn) => turn.agentScope === options.agentScope && ACTIVE_STATUSES.has(turn.status))
+            .map((turn) => turn.conversationId));
+        const candidate = [...this.turns.values()]
+            .filter((turn) => turn.agentScope === options.agentScope &&
+            turn.status === "pending" &&
+            turn.nextAttemptAt <= nowIso &&
+            !activeConversations.has(turn.conversationId))
+            // Map iteration and modern Array#sort are stable, preserving insertion
+            // order when multiple events arrive in the same clock millisecond.
+            .sort((a, b) => a.createdAt.localeCompare(b.createdAt))[0];
+        if (!candidate)
+            return undefined;
+        const leaseToken = randomUUID();
+        const updated = {
+            ...candidate,
+            status: "running",
+            attemptCount: candidate.attemptCount + 1,
+            leaseToken,
+            leaseExpiresAt: new Date(now.getTime() + Math.max(1, options.leaseForMs)).toISOString(),
+            updatedAt: nowIso
+        };
+        this.turns.set(updated.id, updated);
+        await this.persist();
+        return clone(updated);
+    }
+    async attachConversationTurnRun(id, leaseToken, runId) {
+        const current = this.requireLease(id, leaseToken);
+        const updated = { ...current, runId, updatedAt: timestamp() };
+        this.turns.set(id, updated);
+        await this.persist();
+        return clone(updated);
+    }
+    async settleConversationTurn(id, leaseToken, status, options = {}) {
+        const current = this.requireLease(id, leaseToken);
+        return this.settle(current, status, options);
+    }
+    async settleConversationTurnByRunId(runId, status, options = {}) {
+        const current = [...this.turns.values()].find((turn) => turn.runId === runId && ACTIVE_STATUSES.has(turn.status));
+        if (!current)
+            return undefined;
+        return this.settle(current, status, options);
+    }
+    async retryConversationTurn(id, leaseToken, error) {
+        const current = this.requireLease(id, leaseToken);
+        const now = new Date();
+        const exhausted = current.attemptCount >= current.maxAttempts;
+        const updated = {
+            ...current,
+            status: exhausted ? "failed" : "pending",
+            nextAttemptAt: exhausted
+                ? now.toISOString()
+                : new Date(now.getTime() + retryDelayMs(current.attemptCount)).toISOString(),
+            lastError: error,
+            updatedAt: now.toISOString()
+        };
+        delete updated.leaseToken;
+        delete updated.leaseExpiresAt;
+        this.turns.set(id, updated);
+        await this.persist();
+        return clone(updated);
+    }
+    async recoverExpiredConversationTurns(now = new Date()) {
+        const recovered = [];
+        for (const current of this.turns.values()) {
+            if (current.status !== "running" ||
+                !current.leaseExpiresAt ||
+                current.leaseExpiresAt > now.toISOString() ||
+                current.runId)
+                continue;
+            const updated = {
+                ...current,
+                status: current.attemptCount >= current.maxAttempts ? "failed" : "pending",
+                nextAttemptAt: now.toISOString(),
+                updatedAt: now.toISOString(),
+                lastError: current.lastError ?? "Conversation turn lease expired before a run was attached."
+            };
+            delete updated.leaseToken;
+            delete updated.leaseExpiresAt;
+            this.turns.set(updated.id, updated);
+            recovered.push(clone(updated));
+        }
+        if (recovered.length > 0)
+            await this.persist();
+        return recovered;
+    }
+    async listConversationTurns(options = {}) {
+        return [...this.turns.values()]
+            .filter((turn) => !options.agentScope || turn.agentScope === options.agentScope)
+            .filter((turn) => !options.conversationId || turn.conversationId === options.conversationId)
+            .filter((turn) => !options.status || turn.status === options.status)
+            .sort((a, b) => a.createdAt.localeCompare(b.createdAt))
+            .map(clone);
+    }
+    requireLease(id, leaseToken) {
+        const current = this.turns.get(id);
+        if (!current)
+            throw new Error(`Conversation turn not found: ${id}`);
+        if (current.status !== "running" || current.leaseToken !== leaseToken) {
+            throw new Error(`Conversation turn lease mismatch: ${id}`);
+        }
+        return current;
+    }
+    async settle(current, status, options) {
+        const updated = {
+            ...current,
+            status,
+            updatedAt: timestamp()
+        };
+        if (options.runId)
+            updated.runId = options.runId;
+        if (options.error)
+            updated.lastError = options.error;
+        delete updated.leaseToken;
+        delete updated.leaseExpiresAt;
+        this.turns.set(updated.id, updated);
+        await this.persist();
+        return clone(updated);
+    }
+}
+//# sourceMappingURL=in-memory-conversation-turn-store.js.map

package/dist/in-memory-conversation-turn-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory-conversation-turn-store.js","sourceRoot":"","sources":["../src/in-memory-conversation-turn-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMzC,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAC;AAEhE,MAAM,eAAe,GAAG,IAAI,GAAG,CAAyB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;AAEhF,SAAS,KAAK,CAAC,MAA8B;IAC3C,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,YAAY,CAAC,YAAoB;IACxC,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;AACxE,CAAC;AAED,MAAM,OAAO,6BAA6B;IAGX;IAF7B,KAAK,GAAG,IAAI,GAAG,EAAkC,CAAC;IAElD,YAA6B,UAA+B,SAAS;QAAxC,YAAO,GAAP,OAAO,CAAiC;IAAG,CAAC;IAEzE,KAAK,CAAC,uBAAuB,CAAC,KAAsE;QAClG,MAAM,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CACtD,IAAI,CAAC,UAAU,KAAK,KAAK,CAAC,UAAU;YACpC,IAAI,CAAC,gBAAgB,KAAK,KAAK,CAAC,gBAAgB;YAChD,IAAI,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc,CAC7C,CAAC;QACF,IAAI,QAAQ;YAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QAChE,MAAM,GAAG,GAAG,SAAS,EAAE,CAAC;QACxB,MAAM,IAAI,GAA2B;YACnC,EAAE,EAAE,UAAU,EAAE;YAChB,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,OAAO,EAAE,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC;YACvC,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,MAAM,EAAE,SAAS;YACjB,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;YAC5D,aAAa,EAAE,GAAG;YAClB,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;SACf,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC9B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,yBAAyB,CAC7B,OAA0E;QAE1E,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC;QACtC,MAAM,MAAM,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC;QACjC,MAAM,mBAAmB,GAAG,IAAI,GAAG,CACjC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;aACrB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,KAAK,OAAO,CAAC,UAAU,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;aAC5F,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CACtC,CAAC;QACF,MAAM,SAAS,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;aACvC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CACf,IAAI,CAAC,UAAU,KAAK,OAAO,CAAC,UAAU;YACtC,IAAI,CAAC,MAAM,KAAK,SAAS;YACzB,IAAI,CAAC,aAAa,IAAI,MAAM;YAC5B,CAAC,mBAAmB,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAC9C;YACD,uEAAuE;YACvE,mEAAmE;aAClE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,IAAI,CAAC,SAAS;YAAE,OAAO,SAAS,CAAC;QACjC,MAAM,UAAU,GAAG,UAAU,EAAE,CAAC;QAChC,MAAM,OAAO,GAA2B;YACtC,GAAG,SAAS;YACZ,MAAM,EAAE,SAAS;YACjB,YAAY,EAAE,SAAS,CAAC,YAAY,GAAG,CAAC;YACxC,UAAU;YACV,cAAc,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,EAAE;YACvF,SAAS,EAAE,MAAM;SAClB,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,EAAU,EAAE,UAAkB,EAAE,KAAa;QAC3E,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,CAAC;QAC9D,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC5B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,sBAAsB,CAC1B,EAAU,EACV,UAAkB,EAClB,MAAwD,EACxD,UAA8C,EAAE;QAEhD,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAClD,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,6BAA6B,CACjC,KAAa,EACb,MAAwD,EACxD,UAA8B,EAAE;QAEhC,MAAM,OAAO,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CACrD,IAAI,CAAC,KAAK,KAAK,KAAK,IAAI,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CACzD,CAAC;QACF,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAC/B,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,EAAU,EAAE,UAAkB,EAAE,KAAa;QACvE,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,WAAW,CAAC;QAC9D,MAAM,OAAO,GAA2B;YACtC,GAAG,OAAO;YACV,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;YACxC,aAAa,EAAE,SAAS;gBACtB,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE;gBACnB,CAAC,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,YAAY,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW,EAAE;YAC9E,SAAS,EAAE,KAAK;YAChB,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;SAC7B,CAAC;QACF,OAAO,OAAO,CAAC,UAAU,CAAC;QAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC5B,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,+BAA+B,CAAC,GAAG,GAAG,IAAI,IAAI,EAAE;QACpD,MAAM,SAAS,GAA6B,EAAE,CAAC;QAC/C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;YAC1C,IACE,OAAO,CAAC,MAAM,KAAK,SAAS;gBAC5B,CAAC,OAAO,CAAC,cAAc;gBACvB,OAAO,CAAC,cAAc,GAAG,GAAG,CAAC,WAAW,EAAE;gBAC1C,OAAO,CAAC,KAAK;gBACb,SAAS;YACX,MAAM,OAAO,GAA2B;gBACtC,GAAG,OAAO;gBACV,MAAM,EAAE,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;gBAC1E,aAAa,EAAE,GAAG,CAAC,WAAW,EAAE;gBAChC,SAAS,EAAE,GAAG,CAAC,WAAW,EAAE;gBAC5B,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,4DAA4D;aAC7F,CAAC;YACF,OAAO,OAAO,CAAC,UAAU,CAAC;YAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;YAC9B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACpC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACjC,CAAC;QACD,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC;YAAE,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QAC/C,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,UAIxB,EAAE;QACJ,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;aAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,CAAC,UAAU,CAAC;aAC/E,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,IAAI,IAAI,CAAC,cAAc,KAAK,OAAO,CAAC,cAAc,CAAC;aAC3F,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC;aACnE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;aACtD,GAAG,CAAC,KAAK,CAAC,CAAC;IAChB,CAAC;IAEO,YAAY,CAAC,EAAU,EAAE,UAAkB;QACjD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACnC,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,EAAE,EAAE,CAAC,CAAC;QACpE,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,IAAI,OAAO,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;YACtE,MAAM,IAAI,KAAK,CAAC,qCAAqC,EAAE,EAAE,CAAC,CAAC;QAC7D,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,MAAM,CAClB,OAA+B,EAC/B,MAAwD,EACxD,OAA2C;QAE3C,MAAM,OAAO,GAA2B;YACtC,GAAG,OAAO;YACV,MAAM;YACN,SAAS,EAAE,SAAS,EAAE;SACvB,CAAC;QACF,IAAI,OAAO,CAAC,KAAK;YAAE,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QACjD,IAAI,OAAO,CAAC,KAAK;YAAE,OAAO,CAAC,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC;QACrD,OAAO,OAAO,CAAC,UAAU,CAAC;QAC1B,OAAO,OAAO,CAAC,cAAc,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;QACrB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC;CACF"}

package/dist/in-memory-run-store.d.ts

@@ -0,0 +1,39 @@
+import type { JsonObject } from "@assemblyline-agents/core";
+import { type DeliveryFailureInput, type DeliveryLeaseOptions, type DeliveryObligationInput, type DeliveryObligationPatch } from "./delivery-queue.js";
+import type { CheckpointRecord, CheckpointRetentionOptions, CheckpointRetentionResult, DeliveryObligation, RunEvent, RunEventType, RunRecord, RunControlIntent, RunStore, ToolCallRecord } from "./state.js";
+export declare class InMemoryRunStore implements RunStore {
+    private readonly persist;
+    runs: Map<string, RunRecord>;
+    events: RunEvent[];
+    toolCalls: Map<string, ToolCallRecord>;
+    checkpoints: CheckpointRecord[];
+    deliveries: Map<string, DeliveryObligation>;
+    idempotency: Set<string>;
+    constructor(persist?: () => Promise<void>);
+    createRun(input: Omit<RunRecord, "id" | "createdAt" | "updatedAt">): Promise<RunRecord>;
+    updateRun(id: string, patch: Partial<Omit<RunRecord, "id" | "createdAt">>): Promise<RunRecord>;
+    touchRun(id: string): Promise<void>;
+    requestRunControl(id: string, intent: RunControlIntent): Promise<RunRecord>;
+    settleRunControl(id: string): Promise<RunRecord>;
+    getRun(id: string): Promise<RunRecord | undefined>;
+    listRuns(): Promise<RunRecord[]>;
+    appendEvent(runId: string, type: RunEventType, data?: JsonObject): Promise<RunEvent>;
+    listEvents(runId: string): Promise<RunEvent[]>;
+    createToolCall(input: Omit<ToolCallRecord, "id" | "createdAt" | "updatedAt">): Promise<ToolCallRecord>;
+    updateToolCall(id: string, patch: Partial<Omit<ToolCallRecord, "id" | "createdAt">>): Promise<ToolCallRecord>;
+    listToolCalls(runId: string): Promise<ToolCallRecord[]>;
+    createCheckpoint(input: Omit<CheckpointRecord, "id" | "createdAt">): Promise<CheckpointRecord>;
+    listCheckpoints(runId: string): Promise<CheckpointRecord[]>;
+    compactCheckpoints(options?: CheckpointRetentionOptions): Promise<CheckpointRetentionResult>;
+    createDelivery(input: DeliveryObligationInput): Promise<DeliveryObligation>;
+    updateDelivery(id: string, patch: DeliveryObligationPatch): Promise<DeliveryObligation>;
+    listDeliveries(runId: string): Promise<DeliveryObligation[]>;
+    leaseDueDeliveries(options?: DeliveryLeaseOptions): Promise<DeliveryObligation[]>;
+    completeDeliverySend(id: string, leaseToken: string, patch?: {
+        payload?: JsonObject;
+    }): Promise<DeliveryObligation>;
+    failDeliverySend(id: string, leaseToken: string, failure: DeliveryFailureInput): Promise<DeliveryObligation>;
+    recoverExpiredDeliveries(now?: Date): Promise<DeliveryObligation[]>;
+    reserveIdempotencyKey(scope: string, key: string, _ownerId?: string): Promise<boolean>;
+}
+//# sourceMappingURL=in-memory-run-store.d.ts.map

package/dist/in-memory-run-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"in-memory-run-store.d.ts","sourceRoot":"","sources":["../src/in-memory-run-store.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EAKL,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EACzB,KAAK,uBAAuB,EAC5B,KAAK,uBAAuB,EAC7B,MAAM,qBAAqB,CAAC;AAY7B,OAAO,KAAK,EACV,gBAAgB,EAChB,0BAA0B,EAE1B,yBAAyB,EACzB,kBAAkB,EAClB,QAAQ,EACR,YAAY,EACZ,SAAS,EACT,gBAAgB,EAChB,QAAQ,EACR,cAAc,EACf,MAAM,YAAY,CAAC;AAEpB,qBAAa,gBAAiB,YAAW,QAAQ;IAQnC,OAAO,CAAC,QAAQ,CAAC,OAAO;IAPpC,IAAI,yBAAgC;IACpC,MAAM,EAAE,QAAQ,EAAE,CAAM;IACxB,SAAS,8BAAqC;IAC9C,WAAW,EAAE,gBAAgB,EAAE,CAAM;IACrC,UAAU,kCAAyC;IACnD,WAAW,cAAqB;gBAEH,OAAO,GAAE,MAAM,OAAO,CAAC,IAAI,CAAa;IAE/D,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IAavF,SAAS,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC;IAS9F,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAOnC,iBAAiB,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC;IAsB3E,gBAAgB,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC;IAYhD,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,SAAS,CAAC;IAIlD,QAAQ,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAIhC,WAAW,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,GAAE,UAAe,GAAG,OAAO,CAAC,QAAQ,CAAC;IAcxF,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;IAI9C,cAAc,CAAC,KAAK,EAAE,IAAI,CAAC,cAAc,EAAE,IAAI,GAAG,WAAW,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAatG,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,GAAG,WAAW,CAAC,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAS7G,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAIvD,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,GAAG,WAAW,CAAC,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAW9F,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,EAAE,CAAC;IAI3D,kBAAkB,CAAC,OAAO,GAAE,0BAA+B,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAiEhG,cAAc,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAO3E,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IASvF,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAI5D,kBAAkB,CAAC,OAAO,GAAE,oBAAyB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IA2BrF,oBAAoB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,GAAE;QAAE,OAAO,CAAC,EAAE,UAAU,CAAA;KAAO,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAqBvH,gBAAgB,CAAC,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,kBAAkB,CAAC;IAuB5G,wBAAwB,CAAC,GAAG,GAAE,IAAiB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAmB/E,qBAAqB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAO7F"}