@assemblyline-agents/runtime 5.0.10 → 6.0.1

package/dist/usage-reconciliation-support.js

@@ -0,0 +1,95 @@
+import { UsageAccountingError, sha256Json } from "./usage-records.js";
+const FETCH_TIMEOUT_MS = 10_000;
+export async function fetchUsageJson(url, key, fetchImpl) {
+    let lastError;
+    for (const delayMs of [0, 150, 750]) {
+        if (delayMs)
+            await new Promise((resolve) => setTimeout(resolve, delayMs));
+        const controller = new AbortController();
+        const timer = setTimeout(() => controller.abort(), FETCH_TIMEOUT_MS);
+        timer.unref?.();
+        try {
+            const response = await fetchImpl(url, {
+                headers: { authorization: `Bearer ${key}`, accept: "application/json" },
+                signal: controller.signal
+            });
+            if (response.ok)
+                return response.json();
+            const retryable = response.status === 429 || response.status >= 500;
+            const detail = (await response.text()).slice(0, 500);
+            lastError = new UsageAccountingError(`Provider reconciliation failed with HTTP ${response.status}${detail ? `: ${detail}` : "."}`, "usage_provider_reconciliation_failed");
+            if (!retryable)
+                throw lastError;
+        }
+        catch (error) {
+            lastError = error;
+        }
+        finally {
+            clearTimeout(timer);
+        }
+    }
+    throw lastError instanceof Error
+        ? lastError
+        : new UsageAccountingError("Provider reconciliation failed after retries.");
+}
+export function settledUsageRange(from, to, now, lagHours) {
+    if (!Number.isFinite(lagHours) || lagHours < 0)
+        throw new UsageAccountingError("Provider settlement lag must be non-negative.");
+    return normalizedUsageRange(from, to, utcDay(new Date(now.getTime() - lagHours * 3_600_000)));
+}
+export function completedUsageRange(from, to, now) {
+    return normalizedUsageRange(from, to, utcDay(now));
+}
+export function usageDates(from, to, maximum) {
+    const dates = [];
+    for (let cursor = new Date(from); cursor < new Date(to); cursor = new Date(cursor.getTime() + 86_400_000)) {
+        dates.push(cursor.toISOString().slice(0, 10));
+    }
+    if (dates.length > maximum)
+        throw new UsageAccountingError(`Provider activity reconciliation supports at most ${maximum} completed UTC days.`);
+    return dates;
+}
+export function usageControlKey(provider, source, bucket, dimensions) {
+    return `control:${provider}:${source}:${bucket}:${sha256Json(dimensions)}`;
+}
+export function bodyString(body, key) {
+    return stringValue(body[key]);
+}
+export function stringValue(value) {
+    return typeof value === "string" && value.trim() ? value.trim() : undefined;
+}
+export function numberValue(value) {
+    const parsed = typeof value === "number" ? value : typeof value === "string" ? Number(value) : NaN;
+    return Number.isFinite(parsed) && parsed >= 0 ? parsed : undefined;
+}
+export function integerValue(value) {
+    const parsed = numberValue(value);
+    return parsed !== undefined && Number.isSafeInteger(parsed) ? parsed : undefined;
+}
+export function objectValue(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value) ? value : undefined;
+}
+export function objectList(value) {
+    return Array.isArray(value) ? value.filter((entry) => Boolean(objectValue(entry))) : [];
+}
+export function compactJson(value) {
+    return Object.fromEntries(Object.entries(value).filter(([, entry]) => typeof entry === "string" || typeof entry === "number" || typeof entry === "boolean"));
+}
+function normalizedUsageRange(from, to, maximumTo) {
+    const normalizedFrom = iso(from, "from");
+    const requestedTo = to ? iso(to, "to") : maximumTo;
+    const normalizedTo = requestedTo < maximumTo ? requestedTo : maximumTo;
+    if (normalizedFrom >= normalizedTo)
+        throw new UsageAccountingError("Reconciliation range has no completed provider buckets.");
+    return { from: normalizedFrom, to: normalizedTo };
+}
+function utcDay(value) {
+    return `${value.toISOString().slice(0, 10)}T00:00:00.000Z`;
+}
+function iso(value, name) {
+    const date = new Date(value);
+    if (Number.isNaN(date.getTime()))
+        throw new UsageAccountingError(`${name} must be an ISO timestamp.`);
+    return date.toISOString();
+}
+//# sourceMappingURL=usage-reconciliation-support.js.map

package/dist/usage-reconciliation-support.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-reconciliation-support.js","sourceRoot":"","sources":["../src/usage-reconciliation-support.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEtE,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEhC,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,GAAQ,EAAE,GAAW,EAAE,SAAuB;IACjF,IAAI,SAAkB,CAAC;IACvB,KAAK,MAAM,OAAO,IAAI,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;QACpC,IAAI,OAAO;YAAE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC1E,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,gBAAgB,CAAC,CAAC;QACrE,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;QAChB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE;gBACpC,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,GAAG,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;gBACvE,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YACH,IAAI,QAAQ,CAAC,EAAE;gBAAE,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,CAAC;YACpE,MAAM,MAAM,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;YACrD,SAAS,GAAG,IAAI,oBAAoB,CAClC,4CAA4C,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,EAC5F,sCAAsC,CACvC,CAAC;YACF,IAAI,CAAC,SAAS;gBAAE,MAAM,SAAS,CAAC;QAClC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,SAAS,GAAG,KAAK,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IACD,MAAM,SAAS,YAAY,KAAK;QAC9B,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,IAAI,oBAAoB,CAAC,+CAA+C,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAY,EAAE,EAAsB,EAAE,GAAS,EAAE,QAAgB;IACjG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,QAAQ,GAAG,CAAC;QAAE,MAAM,IAAI,oBAAoB,CAAC,+CAA+C,CAAC,CAAC;IAChI,OAAO,oBAAoB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;AAChG,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,IAAY,EAAE,EAAsB,EAAE,GAAS;IACjF,OAAO,oBAAoB,CAAC,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAY,EAAE,EAAU,EAAE,OAAe;IAClE,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,IAAI,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC;QAC1G,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAChD,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,GAAG,OAAO;QAAE,MAAM,IAAI,oBAAoB,CAAC,qDAAqD,OAAO,sBAAsB,CAAC,CAAC;IAC/I,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,MAAc,EAAE,MAAuB,EAAE,UAAsB;IAC/G,OAAO,WAAW,QAAQ,IAAI,MAAM,IAAI,MAAM,IAAI,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,IAAgB,EAAE,GAAW;IACtD,OAAO,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;AAC9E,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,MAAM,MAAM,GAAG,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IACnG,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,KAAc;IACzC,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;IAClC,OAAO,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACnF,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAAc;IACxC,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAgC,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7H,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,KAAc;IACvC,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,EAAoC,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAC5H,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,KAA8B;IACxD,OAAO,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CACnE,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,CACrF,CAAe,CAAC;AACnB,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAY,EAAE,EAAsB,EAAE,SAAiB;IACnF,MAAM,cAAc,GAAG,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,MAAM,WAAW,GAAG,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACnD,MAAM,YAAY,GAAG,WAAW,GAAG,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;IACvE,IAAI,cAAc,IAAI,YAAY;QAAE,MAAM,IAAI,oBAAoB,CAAC,yDAAyD,CAAC,CAAC;IAC9H,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC;AACpD,CAAC;AAED,SAAS,MAAM,CAAC,KAAW;IACzB,OAAO,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,gBAAgB,CAAC;AAC7D,CAAC;AAED,SAAS,GAAG,CAAC,KAAa,EAAE,IAAY;IACtC,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;QAAE,MAAM,IAAI,oBAAoB,CAAC,GAAG,IAAI,4BAA4B,CAAC,CAAC;IACtG,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;AAC5B,CAAC"}

package/dist/usage-reconciliation.d.ts

@@ -1,4 +1,6 @@
+import type { JsonObject } from "@assemblyline-agents/core";
 import type { UsageAggregate, UsageStore } from "./state.js";
+export declare const USAGE_RECONCILER_EXPORT_NAME: "assemblyLineUsageReconciler";
 export interface UsageReconciliationReport {
     provider: string;
     from: string;
@@ -30,44 +32,21 @@ export interface ProviderReconciliationResult {
     reconciledTransactions: number;
     report: UsageReconciliationReport;
 }
-export interface OpenAIOrganizationReconciliationOptions {
+export interface UsageReconcilerContext {
     state: UsageStore;
-    adminKey: string;
-    from: string;
-    to?: string;
-    provider?: string;
-    projectIds?: string[];
-    apiKeyIds?: string[];
-    agentByProjectId?: Record<string, string>;
-    agentByApiKeyId?: Record<string, string>;
-    settlementLagHours?: number;
-    now?: Date;
-    fetch?: typeof fetch;
-    baseUrl?: string;
+    env: Record<string, string | undefined>;
+    input: JsonObject;
+    fetch: typeof fetch;
 }
-/**
- * Import settled OpenAI organization token and cash control totals.
- *
- * Control totals are deliberately separate from transaction records. A
- * project/API-key mapping is the only way they receive an agentId; Assembly Line
- * never invents per-run cost attribution from aggregate provider invoices.
- */
-export declare function reconcileOpenAIOrganization(options: OpenAIOrganizationReconciliationOptions): Promise<ProviderReconciliationResult>;
-export interface OpenRouterReconciliationOptions {
-    state: UsageStore;
-    apiKey?: string;
-    managementKey?: string;
-    from: string;
-    to?: string;
-    provider?: string;
-    apiKeyHash?: string;
-    agentId?: string;
-    now?: Date;
-    fetch?: typeof fetch;
-    baseUrl?: string;
+export interface ProviderUsageReconciler {
+    provider: string;
+    reconcile(context: UsageReconcilerContext): Promise<ProviderReconciliationResult>;
+}
+export interface UsageReconcilerModule {
+    providers: ProviderUsageReconciler[];
 }
-/** Retry missing generation receipts and import completed OpenRouter activity totals. */
-export declare function reconcileOpenRouterUsage(options: OpenRouterReconciliationOptions): Promise<ProviderReconciliationResult>;
+/** Read the optional usage surface exported by a selected model plugin. */
+export declare function usageReconcilerModuleFromImport(imported: unknown): UsageReconcilerModule | undefined;
 export declare function usageReconciliationReport(state: UsageStore, input: {
     provider: string;
     from: string;

package/dist/usage-reconciliation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"usage-reconciliation.d.ts","sourceRoot":"","sources":["../src/usage-reconciliation.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EACV,cAAc,EAGd,UAAU,EACX,MAAM,YAAY,CAAC;AAyBpB,MAAM,WAAW,yBAAyB;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,EAAE,cAAc,CAAC;IAC7B,aAAa,EAAE,cAAc,CAAC;IAC9B,QAAQ,EAAE;QACR,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,EAAE,MAAM,CAAC;QACxB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,QAAQ,EAAE;QACR,kBAAkB,EAAE,MAAM,CAAC;QAC3B,mCAAmC,EAAE,MAAM,CAAC;QAC5C,8BAA8B,EAAE,MAAM,CAAC;QACvC,mBAAmB,EAAE,MAAM,CAAC;QAC5B,qBAAqB,EAAE,OAAO,CAAC;QAC/B,oBAAoB,EAAE,OAAO,CAAC;QAC9B,8BAA8B,EAAE,OAAO,CAAC;KACzC,CAAC;CACH;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,qBAAqB,EAAE,MAAM,CAAC;IAC9B,sBAAsB,EAAE,MAAM,CAAC;IAC/B,MAAM,EAAE,yBAAyB,CAAC;CACnC;AAED,MAAM,WAAW,uCAAuC;IACtD,KAAK,EAAE,UAAU,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,eAAe,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACzC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,GAAG,CAAC,EAAE,IAAI,CAAC;IACX,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;;;;GAMG;AACH,wBAAsB,2BAA2B,CAC/C,OAAO,EAAE,uCAAuC,GAC/C,OAAO,CAAC,4BAA4B,CAAC,CA4GvC;AAED,MAAM,WAAW,+BAA+B;IAC9C,KAAK,EAAE,UAAU,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,IAAI,CAAC;IACX,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,yFAAyF;AACzF,wBAAsB,wBAAwB,CAC5C,OAAO,EAAE,+BAA+B,GACvC,OAAO,CAAC,4BAA4B,CAAC,CA0FvC;AAED,wBAAsB,yBAAyB,CAC7C,KAAK,EAAE,UAAU,EACjB,KAAK,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE,GACpD,OAAO,CAAC,yBAAyB,CAAC,CAsCpC"}
+{"version":3,"file":"usage-reconciliation.d.ts","sourceRoot":"","sources":["../src/usage-reconciliation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAkB7D,eAAO,MAAM,4BAA4B,EAAG,6BAAsC,CAAC;AAEnF,MAAM,WAAW,yBAAyB;IACxC,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,EAAE,cAAc,CAAC;IAC7B,aAAa,EAAE,cAAc,CAAC;IAC9B,QAAQ,EAAE;QACR,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,EAAE,MAAM,CAAC;QACxB,gBAAgB,EAAE,MAAM,CAAC;KAC1B,CAAC;IACF,QAAQ,EAAE;QACR,kBAAkB,EAAE,MAAM,CAAC;QAC3B,mCAAmC,EAAE,MAAM,CAAC;QAC5C,8BAA8B,EAAE,MAAM,CAAC;QACvC,mBAAmB,EAAE,MAAM,CAAC;QAC5B,qBAAqB,EAAE,OAAO,CAAC;QAC/B,oBAAoB,EAAE,OAAO,CAAC;QAC9B,8BAA8B,EAAE,OAAO,CAAC;KACzC,CAAC;CACH;AAED,MAAM,WAAW,4BAA4B;IAC3C,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,qBAAqB,EAAE,MAAM,CAAC;IAC9B,sBAAsB,EAAE,MAAM,CAAC;IAC/B,MAAM,EAAE,yBAAyB,CAAC;CACnC;AAED,MAAM,WAAW,sBAAsB;IACrC,KAAK,EAAE,UAAU,CAAC;IAClB,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IACxC,KAAK,EAAE,UAAU,CAAC;IAClB,KAAK,EAAE,OAAO,KAAK,CAAC;CACrB;AAED,MAAM,WAAW,uBAAuB;IACtC,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,4BAA4B,CAAC,CAAC;CACnF;AAED,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,uBAAuB,EAAE,CAAC;CACtC;AAED,2EAA2E;AAC3E,wBAAgB,+BAA+B,CAAC,QAAQ,EAAE,OAAO,GAAG,qBAAqB,GAAG,SAAS,CAQpG;AAED,wBAAsB,yBAAyB,CAC7C,KAAK,EAAE,UAAU,EACjB,KAAK,EAAE;IAAE,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE,GACpD,OAAO,CAAC,yBAAyB,CAAC,CAsCpC"}

package/dist/usage-reconciliation.js

@@ -1,7 +1,3 @@
-import { openRouterGenerationAccounting } from "@assemblyline-agents/pi";
-import { UsageAccountingError, sha256Json, usageRecordInput } from "./usage-records.js";
-const DEFAULT_PROVIDER_SETTLEMENT_LAG_HOURS = 48;
-const FETCH_TIMEOUT_MS = 10_000;
 const EMPTY_AGGREGATE = {
     dimensions: {},
     recordCount: "0",
@@ -17,194 +13,18 @@ const EMPTY_AGGREGATE = {
     unreconciledTokenRecords: "0",
     unreconciledRecords: "0"
 };
-/**
- * Import settled OpenAI organization token and cash control totals.
- *
- * Control totals are deliberately separate from transaction records. A
- * project/API-key mapping is the only way they receive an agentId; Assembly Line
- * never invents per-run cost attribution from aggregate provider invoices.
- */
-export async function reconcileOpenAIOrganization(options) {
-    const provider = clean(options.provider) ?? "openai";
-    const adminKey = requiredSecret(options.adminKey, "OpenAI admin key");
-    const range = settledDailyRange(options.from, options.to, options.now ?? new Date(), options.settlementLagHours ?? DEFAULT_PROVIDER_SETTLEMENT_LAG_HOURS);
-    const baseUrl = (clean(options.baseUrl) ?? "https://api.openai.com/v1").replace(/\/+$/u, "");
-    const common = {
-        start_time: String(Math.floor(new Date(range.from).getTime() / 1_000)),
-        end_time: String(Math.floor(new Date(range.to).getTime() / 1_000)),
-        bucket_width: "1d"
-    };
-    const usageBuckets = await fetchOpenAIPages(`${baseUrl}/organization/usage/completions`, adminKey, common, ["model", "project_id", "api_key_id"], options.projectIds, options.apiKeyIds, options.fetch);
-    const costBuckets = await fetchOpenAIPages(`${baseUrl}/organization/costs`, adminKey, common, ["project_id", "line_item"], options.projectIds, undefined, options.fetch);
-    let recordedControlTotals = 0;
-    for (const bucket of usageBuckets) {
-        const startTime = integer(bucket.start_time);
-        const endTime = integer(bucket.end_time);
-        if (startTime === undefined || endTime === undefined)
+export const USAGE_RECONCILER_EXPORT_NAME = "assemblyLineUsageReconciler";
+/** Read the optional usage surface exported by a selected model plugin. */
+export function usageReconcilerModuleFromImport(imported) {
+    const candidates = [imported, imported?.default];
+    for (const candidate of candidates) {
+        if (!candidate || typeof candidate !== "object")
             continue;
-        for (const result of arrayObjects(bucket.results)) {
-            const inputTokens = integer(result.input_tokens) ?? 0;
-            const outputTokens = integer(result.output_tokens) ?? 0;
-            const cachedTokens = integer(result.input_cached_tokens) ?? 0;
-            const cacheWriteTokens = integer(result.input_cache_write_tokens) ?? 0;
-            const dimensions = openAIDimensions(result);
-            const model = string(result.model);
-            await options.state.recordUsage({
-                idempotencyKey: controlKey(provider, "openai-usage", startTime, dimensions),
-                ...openAIAgentAttribution(result, options),
-                provider,
-                ...(model ? { model } : {}),
-                billingMode: "api_key",
-                recordKind: "control_total",
-                inputTokens,
-                outputTokens,
-                cachedTokens,
-                cacheWriteTokens,
-                tokenSource: "provider_reconciled",
-                costSource: "unavailable",
-                providerReceipt: providerReceipt("openai.organization.usage.completions", startTime, endTime, result),
-                metadata: compactJson({
-                    controlType: "token_usage",
-                    numModelRequests: integer(result.num_model_requests),
-                    ...dimensions
-                }),
-                occurredAt: epochIso(startTime)
-            });
-            recordedControlTotals += 1;
-        }
+        const value = candidate[USAGE_RECONCILER_EXPORT_NAME];
+        if (isUsageReconcilerModule(value))
+            return value;
     }
-    for (const bucket of costBuckets) {
-        const startTime = integer(bucket.start_time);
-        const endTime = integer(bucket.end_time);
-        if (startTime === undefined || endTime === undefined)
-            continue;
-        for (const result of arrayObjects(bucket.results)) {
-            const amount = object(result.amount);
-            const costUsd = number(amount?.value);
-            if (costUsd === undefined)
-                continue;
-            const currency = string(amount?.currency)?.toUpperCase() ?? "USD";
-            const dimensions = openAIDimensions(result);
-            await options.state.recordUsage({
-                idempotencyKey: controlKey(provider, "openai-costs", startTime, dimensions),
-                ...openAIAgentAttribution(result, options),
-                provider,
-                billingMode: "api_key",
-                recordKind: "control_total",
-                costUsd,
-                currency,
-                tokenSource: "unavailable",
-                costSource: "provider_reconciled",
-                providerReceipt: providerReceipt("openai.organization.costs", startTime, endTime, result),
-                metadata: compactJson({
-                    controlType: "cash_cost",
-                    quantity: number(result.quantity),
-                    ...dimensions
-                }),
-                occurredAt: epochIso(startTime)
-            });
-            recordedControlTotals += 1;
-        }
-    }
-    return {
-        provider,
-        ...range,
-        recordedControlTotals,
-        reconciledTransactions: 0,
-        report: await usageReconciliationReport(options.state, { provider, ...range })
-    };
-}
-/** Retry missing generation receipts and import completed OpenRouter activity totals. */
-export async function reconcileOpenRouterUsage(options) {
-    const provider = clean(options.provider) ?? "openrouter";
-    const range = completedUtcRange(options.from, options.to, options.now ?? new Date());
-    const baseUrl = (clean(options.baseUrl) ?? "https://openrouter.ai/api/v1").replace(/\/+$/u, "");
-    let reconciledTransactions = 0;
-    if (clean(options.apiKey)) {
-        const candidates = (options.state.queryUsage
-            ? await options.state.queryUsage({ provider, recordKind: "transaction", from: range.from, to: range.to, limit: 10_000 })
-            : (await options.state.listUsage()).filter((record) => record.provider === provider &&
-                record.recordKind === "transaction" &&
-                record.occurredAt >= range.from &&
-                record.occurredAt < range.to))
-            .filter((record) => Boolean(record.providerRequestId) &&
-            record.costSource === "unavailable");
-        for (const record of candidates) {
-            const accounting = await openRouterGenerationAccounting(record.providerRequestId, {
-                env: { OPENROUTER_API_KEY: options.apiKey },
-                ...(options.fetch ? { fetch: options.fetch } : {}),
-                fallback: harnessUsage(record)
-            });
-            if (!accounting || accounting.usage.costSource !== "provider_reconciled")
-                continue;
-            await options.state.recordUsage(reconciledTransactionInput(record, accounting));
-            reconciledTransactions += 1;
-        }
-    }
-    let recordedControlTotals = 0;
-    const managementKey = clean(options.managementKey);
-    if (managementKey) {
-        for (const date of utcDates(range.from, range.to)) {
-            const url = new URL(`${baseUrl}/activity`);
-            url.searchParams.set("date", date);
-            if (clean(options.apiKeyHash))
-                url.searchParams.set("api_key_hash", options.apiKeyHash);
-            const body = await fetchJsonWithRetry(url, managementKey, options.fetch);
-            for (const row of arrayObjects(object(body)?.data)) {
-                const dimensions = compactJson({
-                    endpointId: string(row.endpoint_id),
-                    model: string(row.model),
-                    modelPermaslug: string(row.model_permaslug),
-                    providerName: string(row.provider_name)
-                });
-                const inputTokens = integer(row.prompt_tokens) ?? 0;
-                const outputTokens = integer(row.completion_tokens) ?? 0;
-                const reasoningTokens = Math.min(integer(row.reasoning_tokens) ?? 0, outputTokens);
-                const costUsd = number(row.usage);
-                const upstreamCostUsd = number(row.byok_usage_inference);
-                const model = string(row.model);
-                const control = {
-                    idempotencyKey: controlKey(provider, "openrouter-activity", date, dimensions),
-                    provider,
-                    ...(model ? { model } : {}),
-                    billingMode: upstreamCostUsd && !costUsd ? "byok" : "unknown",
-                    recordKind: "control_total",
-                    inputTokens,
-                    outputTokens,
-                    reasoningTokens,
-                    tokenSource: "provider_reconciled",
-                    costSource: costUsd === undefined ? "unavailable" : "provider_reconciled",
-                    providerReceipt: {
-                        source: "openrouter.activity",
-                        date,
-                        result: row
-                    },
-                    metadata: compactJson({
-                        controlType: "activity",
-                        requests: integer(row.requests),
-                        apiKeyHash: clean(options.apiKeyHash),
-                        ...dimensions
-                    }),
-                    occurredAt: `${date}T00:00:00.000Z`
-                };
-                if (costUsd !== undefined)
-                    control.costUsd = costUsd;
-                if (upstreamCostUsd !== undefined)
-                    control.upstreamCostUsd = upstreamCostUsd;
-                if (clean(options.agentId))
-                    control.agentId = options.agentId;
-                await options.state.recordUsage(control);
-                recordedControlTotals += 1;
-            }
-        }
-    }
-    return {
-        provider,
-        ...range,
-        recordedControlTotals,
-        reconciledTransactions,
-        report: await usageReconciliationReport(options.state, { provider, ...range })
-    };
+    return undefined;
 }
 export async function usageReconciliationReport(state, input) {
     const [transactions] = await summarize(state, {
@@ -252,212 +72,15 @@ async function summarize(state, query) {
     const { summarizeUsageRecords } = await import("./usage-records.js");
     return summarizeUsageRecords(records, query);
 }
-async function fetchOpenAIPages(endpoint, key, common, groups, projectIds, apiKeyIds, fetchImpl) {
-    const buckets = [];
-    let page;
-    do {
-        const url = new URL(endpoint);
-        for (const [name, value] of Object.entries(common))
-            url.searchParams.set(name, value);
-        url.searchParams.set("limit", "31");
-        for (const group of groups)
-            url.searchParams.append("group_by", group);
-        for (const projectId of projectIds ?? [])
-            url.searchParams.append("project_ids", projectId);
-        for (const apiKeyId of apiKeyIds ?? [])
-            url.searchParams.append("api_key_ids", apiKeyId);
-        if (page)
-            url.searchParams.set("page", page);
-        const body = object(await fetchJsonWithRetry(url, key, fetchImpl));
-        buckets.push(...arrayObjects(body?.data));
-        page = body?.has_more === true ? string(body.next_page) : undefined;
-        if (body?.has_more === true && !page) {
-            throw new UsageAccountingError("Provider pagination reported has_more without next_page.");
-        }
-    } while (page);
-    return buckets;
-}
-async function fetchJsonWithRetry(url, key, fetchImpl) {
-    const request = fetchImpl ?? globalThis.fetch;
-    let lastError;
-    for (const delayMs of [0, 150, 750]) {
-        if (delayMs)
-            await delay(delayMs);
-        const controller = new AbortController();
-        const timer = setTimeout(() => controller.abort(), FETCH_TIMEOUT_MS);
-        timer.unref?.();
-        try {
-            const response = await request(url, {
-                headers: { authorization: `Bearer ${key}`, accept: "application/json" },
-                signal: controller.signal
-            });
-            if (response.ok)
-                return response.json();
-            const retryable = response.status === 429 || response.status >= 500;
-            const detail = (await response.text()).slice(0, 500);
-            lastError = new UsageAccountingError(`Provider reconciliation failed with HTTP ${response.status}${detail ? `: ${detail}` : "."}`, "usage_provider_reconciliation_failed");
-            if (!retryable)
-                throw lastError;
-        }
-        catch (error) {
-            lastError = error;
-        }
-        finally {
-            clearTimeout(timer);
-        }
-    }
-    throw lastError instanceof Error
-        ? lastError
-        : new UsageAccountingError("Provider reconciliation failed after retries.");
-}
-function reconciledTransactionInput(record, accounting) {
-    const input = usageRecordInput(record);
-    const usage = accounting.usage;
-    if (usage.inputTokens !== undefined)
-        input.inputTokens = usage.inputTokens;
-    if (usage.outputTokens !== undefined)
-        input.outputTokens = usage.outputTokens;
-    if (usage.cachedTokens !== undefined)
-        input.cachedTokens = usage.cachedTokens;
-    if (usage.cacheWriteTokens !== undefined)
-        input.cacheWriteTokens = usage.cacheWriteTokens;
-    if (usage.reasoningTokens !== undefined)
-        input.reasoningTokens = usage.reasoningTokens;
-    if (usage.costUsd !== undefined)
-        input.costUsd = usage.costUsd;
-    if (usage.upstreamCostUsd !== undefined)
-        input.upstreamCostUsd = usage.upstreamCostUsd;
-    if (usage.currency !== undefined)
-        input.currency = usage.currency;
-    if (usage.tokenSource !== undefined)
-        input.tokenSource = usage.tokenSource;
-    if (usage.costSource !== undefined)
-        input.costSource = usage.costSource;
-    if (accounting.providerReceipt)
-        input.providerReceipt = accounting.providerReceipt;
-    input.billingMode = accounting.billingMode;
-    return input;
-}
-function harnessUsage(record) {
-    return {
-        inputTokens: record.inputTokens,
-        outputTokens: record.outputTokens,
-        cachedTokens: record.cachedTokens,
-        cacheWriteTokens: record.cacheWriteTokens,
-        reasoningTokens: record.reasoningTokens,
-        tokenSource: record.tokenSource,
-        costSource: record.costSource
-    };
-}
-function openAIAgentAttribution(result, options) {
-    const apiKeyId = string(result.api_key_id);
-    const projectId = string(result.project_id);
-    const agentId = (apiKeyId ? options.agentByApiKeyId?.[apiKeyId] : undefined) ??
-        (projectId ? options.agentByProjectId?.[projectId] : undefined);
-    return agentId ? { agentId } : {};
-}
-function openAIDimensions(result) {
-    return compactJson({
-        apiKeyId: string(result.api_key_id),
-        projectId: string(result.project_id),
-        model: string(result.model),
-        lineItem: string(result.line_item),
-        serviceTier: string(result.service_tier),
-        batch: typeof result.batch === "boolean" ? result.batch : undefined
-    });
-}
-function providerReceipt(source, startTime, endTime, result) {
-    return {
-        source,
-        bucketStart: epochIso(startTime),
-        bucketEnd: epochIso(endTime),
-        result: result
-    };
-}
-function controlKey(provider, source, bucket, dimensions) {
-    return `control:${provider}:${source}:${bucket}:${sha256Json(dimensions)}`;
-}
-function settledDailyRange(from, to, now, lagHours) {
-    if (!Number.isFinite(lagHours) || lagHours < 0) {
-        throw new UsageAccountingError("Provider settlement lag must be non-negative.");
-    }
-    const cutoff = new Date(now.getTime() - lagHours * 60 * 60 * 1_000);
-    const settledTo = utcDay(cutoff);
-    return normalizedRange(from, to, settledTo);
-}
-function completedUtcRange(from, to, now) {
-    return normalizedRange(from, to, utcDay(now));
-}
-function normalizedRange(from, to, maximumTo) {
-    const normalizedFrom = iso(from, "from");
-    const requestedTo = to ? iso(to, "to") : maximumTo;
-    const normalizedTo = requestedTo < maximumTo ? requestedTo : maximumTo;
-    if (normalizedFrom >= normalizedTo) {
-        throw new UsageAccountingError("Reconciliation range has no completed provider buckets.");
-    }
-    return { from: normalizedFrom, to: normalizedTo };
-}
-function utcDay(value) {
-    return `${value.toISOString().slice(0, 10)}T00:00:00.000Z`;
-}
-function utcDates(from, to) {
-    const dates = [];
-    for (let cursor = new Date(from); cursor < new Date(to); cursor = new Date(cursor.getTime() + 86_400_000)) {
-        dates.push(cursor.toISOString().slice(0, 10));
-    }
-    if (dates.length > 30) {
-        throw new UsageAccountingError("OpenRouter activity reconciliation supports at most 30 completed UTC days.");
-    }
-    return dates;
-}
 function difference(left, right) {
     return (BigInt(left) - BigInt(right)).toString();
 }
-function epochIso(value) {
-    return new Date(value * 1_000).toISOString();
-}
-function iso(value, name) {
-    const date = new Date(value);
-    if (Number.isNaN(date.getTime()))
-        throw new UsageAccountingError(`${name} must be an ISO timestamp.`);
-    return date.toISOString();
-}
-function arrayObjects(value) {
-    return Array.isArray(value)
-        ? value.filter((entry) => Boolean(object(entry)))
-        : [];
-}
-function object(value) {
-    return Boolean(value) && typeof value === "object" && !Array.isArray(value)
-        ? value
-        : undefined;
-}
-function compactJson(value) {
-    return Object.fromEntries(Object.entries(value).filter(([, entry]) => typeof entry === "string" ||
-        typeof entry === "number" ||
-        typeof entry === "boolean"));
-}
-function string(value) {
-    return typeof value === "string" && value.trim() ? value.trim() : undefined;
-}
-function clean(value) {
-    return value?.trim() || undefined;
-}
-function number(value) {
-    const parsed = typeof value === "number" ? value : typeof value === "string" ? Number(value) : NaN;
-    return Number.isFinite(parsed) && parsed >= 0 ? parsed : undefined;
-}
-function integer(value) {
-    const parsed = number(value);
-    return parsed !== undefined && Number.isSafeInteger(parsed) ? parsed : undefined;
-}
-function requiredSecret(value, name) {
-    const secret = clean(value);
-    if (!secret)
-        throw new UsageAccountingError(`${name} is required.`);
-    return secret;
-}
-function delay(ms) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
+function isUsageReconcilerModule(value) {
+    if (!value || typeof value !== "object")
+        return false;
+    const providers = value.providers;
+    return Array.isArray(providers) && providers.every((provider) => Boolean(provider) && typeof provider === "object" &&
+        typeof provider.provider === "string" &&
+        typeof provider.reconcile === "function");
 }
 //# sourceMappingURL=usage-reconciliation.js.map