@assemblyline-agents/openai 6.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Assembly Line contributors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,21 @@
+# @assemblyline-agents/openai
+
+Official OpenAI API model-provider plugin for Assembly Line.
+
+```yaml
+model: openai/gpt-5.4-mini
+```
+
+The package owns Pi provider registration, model discovery, API-key
+authentication, semantic-memory embeddings, response accounting, and usage
+reconciliation. Set `OPENAI_API_KEY`; `OPENAI_BASE_URL` is optional.
+
+Install and select it with:
+
+```sh
+assembly-line add openai agent --role model
+```
+
+## License
+
+MIT

package/ai.assemblyline/index.cjs

@@ -0,0 +1,10 @@
+module.exports = {
+  "models": {
+    "openai": {
+      "definition": "@assemblyline-agents/openai",
+      "provider": "openai",
+      "defaultModel": "openai/gpt-5.4-mini",
+      "contract": "./ai.assemblyline/models/openai.json"
+    }
+  }
+};

package/ai.assemblyline/models/openai.json

@@ -0,0 +1,30 @@
+{
+  "provider": "openai",
+  "packageName": "@assemblyline-agents/openai",
+  "billingMode": "api_key",
+  "authentication": {
+    "kind": "api-key",
+    "interactive": true
+  },
+  "capabilities": [
+    "text",
+    "images",
+    "reasoning",
+    "tool-calling",
+    "usage-reconciliation",
+    "embeddings"
+  ],
+  "embedding": {
+    "model": "text-embedding-3-small",
+    "dimensions": 1536,
+    "maxInputTokens": 8191
+  },
+  "requiredConfig": [],
+  "optionalConfig": [
+    "OPENAI_BASE_URL"
+  ],
+  "requiredCredentials": [
+    "OPENAI_API_KEY"
+  ],
+  "optionalCredentials": []
+}

package/dist/embeddings.d.ts

@@ -0,0 +1,10 @@
+import type { EmbeddingProvider, ModelProviderFactoryContext } from "@assemblyline-agents/core";
+export declare function createOpenAIEmbeddingProvider(context: ModelProviderFactoryContext): EmbeddingProvider;
+export declare function resolveApiKey(context: ModelProviderFactoryContext, envName: string, providerId: string): Promise<string>;
+export declare function embeddingResponse(response: Response, model: string): Promise<{
+    embedding: number[];
+    model: string;
+    dimensions: number;
+}>;
+export declare function trimSlash(value: string | undefined): string | undefined;
+//# sourceMappingURL=embeddings.d.ts.map

package/dist/embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAIhG,wBAAgB,6BAA6B,CAAC,OAAO,EAAE,2BAA2B,GAAG,iBAAiB,CAgBrG;AAED,wBAAsB,aAAa,CACjC,OAAO,EAAE,2BAA2B,EACpC,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,MAAM,CAAC,CAOjB;AAED,wBAAsB,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC;IAAE,SAAS,EAAE,MAAM,EAAE,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAA;CAAE,CAAC,CAY9I;AAED,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,GAAG,SAAS,CAGvE"}

package/dist/embeddings.js

@@ -0,0 +1,56 @@
+const DEFAULT_MODEL = "text-embedding-3-small";
+export function createOpenAIEmbeddingProvider(context) {
+    return {
+        model: DEFAULT_MODEL,
+        dimensions: 1536,
+        maxInputTokens: 8191,
+        async embed(input) {
+            const apiKey = await resolveApiKey(context, "OPENAI_API_KEY", "openai");
+            const model = input.model ?? DEFAULT_MODEL;
+            const response = await context.fetch(`${trimSlash(context.env.OPENAI_BASE_URL) ?? "https://api.openai.com/v1"}/embeddings`, {
+                method: "POST",
+                headers: { authorization: `Bearer ${apiKey}`, "content-type": "application/json" },
+                body: JSON.stringify({ input: input.text, model })
+            });
+            return embeddingResponse(response, model);
+        }
+    };
+}
+export async function resolveApiKey(context, envName, providerId) {
+    const ambient = context.env[envName]?.trim();
+    if (ambient)
+        return ambient;
+    const credential = await context.credentials?.read(providerId);
+    const stored = typeof credential?.key === "string" ? credential.key.trim() : "";
+    if (stored)
+        return stored;
+    throw new Error(`Embedding provider ${providerId} requires ${envName} or a stored API-key credential.`);
+}
+export async function embeddingResponse(response, model) {
+    const body = await response.json().catch(() => undefined);
+    if (!response.ok) {
+        throw new Error(`Embedding request failed (${response.status}): ${errorMessage(body) ?? response.statusText}`);
+    }
+    const embedding = isRecord(body) && Array.isArray(body.data) && isRecord(body.data[0])
+        ? body.data[0].embedding
+        : undefined;
+    if (!Array.isArray(embedding) || embedding.length === 0 || embedding.some((value) => typeof value !== "number" || !Number.isFinite(value))) {
+        throw new Error("Embedding provider returned an invalid vector.");
+    }
+    return { embedding: [...embedding], model, dimensions: embedding.length };
+}
+export function trimSlash(value) {
+    const normalized = value?.trim().replace(/\/+$/u, "");
+    return normalized || undefined;
+}
+function errorMessage(value) {
+    if (!isRecord(value))
+        return undefined;
+    if (typeof value.message === "string")
+        return value.message;
+    return isRecord(value.error) && typeof value.error.message === "string" ? value.error.message : undefined;
+}
+function isRecord(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}
+//# sourceMappingURL=embeddings.js.map

package/dist/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../src/embeddings.ts"],"names":[],"mappings":"AAEA,MAAM,aAAa,GAAG,wBAAwB,CAAC;AAE/C,MAAM,UAAU,6BAA6B,CAAC,OAAoC;IAChF,OAAO;QACL,KAAK,EAAE,aAAa;QACpB,UAAU,EAAE,IAAI;QAChB,cAAc,EAAE,IAAI;QACpB,KAAK,CAAC,KAAK,CAAC,KAAK;YACf,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,QAAQ,CAAC,CAAC;YACxE,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,aAAa,CAAC;YAC3C,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,2BAA2B,aAAa,EAAE;gBAC1H,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAClF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC;aACnD,CAAC,CAAC;YACH,OAAO,iBAAiB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAAoC,EACpC,OAAe,EACf,UAAkB;IAElB,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC;IAC7C,IAAI,OAAO;QAAE,OAAO,OAAO,CAAC;IAC5B,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;IAC/D,MAAM,MAAM,GAAG,OAAO,UAAU,EAAE,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAChF,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC;IAC1B,MAAM,IAAI,KAAK,CAAC,sBAAsB,UAAU,aAAa,OAAO,kCAAkC,CAAC,CAAC;AAC1G,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,QAAkB,EAAE,KAAa;IACvE,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAY,CAAC;IACrE,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,6BAA6B,QAAQ,CAAC,MAAM,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;IACjH,CAAC;IACD,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpF,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;QACxB,CAAC,CAAC,SAAS,CAAC;IACd,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;QAC3I,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,EAAE,SAAS,EAAE,CAAC,GAAG,SAAS,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC;AAC5E,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,KAAyB;IACjD,MAAM,UAAU,GAAG,KAAK,EAAE,IAAI,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACtD,OAAO,UAAU,IAAI,SAAS,CAAC;AACjC,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACvC,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,KAAK,CAAC,OAAO,CAAC;IAC5D,OAAO,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,CAAC,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AAC5G,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,6 @@
+import type { ModelProviderMetadata, ModelProviderModule } from "@assemblyline-agents/core";
+import type { UsageReconcilerModule } from "@assemblyline-agents/runtime";
+export declare const OPENAI_MODEL_PROVIDER_METADATA: ModelProviderMetadata;
+export declare const assemblyLineModelProvider: ModelProviderModule;
+export declare const assemblyLineUsageReconciler: UsageReconcilerModule;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAE5F,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AAI1E,eAAO,MAAM,8BAA8B,EAAuC,qBAAqB,CAAC;AAExG,eAAO,MAAM,yBAAyB,EAAE,mBAMvC,CAAC;AAEF,eAAO,MAAM,2BAA2B,EAAE,qBAEzC,CAAC"}

package/dist/index.js

@@ -0,0 +1,16 @@
+import modelProviderContract from "../ai.assemblyline/models/openai.json" with { type: "json" };
+import { openaiProvider } from "@earendil-works/pi-ai/providers/openai";
+import { reconcileOpenAIUsage } from "./usage-reconciliation.js";
+import { createOpenAIEmbeddingProvider } from "./embeddings.js";
+export const OPENAI_MODEL_PROVIDER_METADATA = modelProviderContract;
+export const assemblyLineModelProvider = {
+    providers: [{
+            metadata: OPENAI_MODEL_PROVIDER_METADATA,
+            create: () => openaiProvider(),
+            createEmbedding: createOpenAIEmbeddingProvider
+        }]
+};
+export const assemblyLineUsageReconciler = {
+    providers: [{ provider: "openai", reconcile: reconcileOpenAIUsage }]
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,qBAAqB,MAAM,uCAAuC,CAAC,OAAO,IAAI,EAAE,MAAM,EAAE,CAAC;AAEhG,OAAO,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAExE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AACjE,OAAO,EAAE,6BAA6B,EAAE,MAAM,iBAAiB,CAAC;AAEhE,MAAM,CAAC,MAAM,8BAA8B,GAAG,qBAAyD,CAAC;AAExG,MAAM,CAAC,MAAM,yBAAyB,GAAwB;IAC5D,SAAS,EAAE,CAAC;YACV,QAAQ,EAAE,8BAA8B;YACxC,MAAM,EAAE,GAAG,EAAE,CAAC,cAAc,EAAE;YAC9B,eAAe,EAAE,6BAA6B;SAC/C,CAAC;CACH,CAAC;AAEF,MAAM,CAAC,MAAM,2BAA2B,GAA0B;IAChE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,oBAAoB,EAAE,CAAC;CACrE,CAAC"}

package/dist/usage-reconciliation.d.ts

@@ -0,0 +1,3 @@
+import { type ProviderReconciliationResult, type UsageReconcilerContext } from "@assemblyline-agents/runtime";
+export declare function reconcileOpenAIUsage(context: UsageReconcilerContext): Promise<ProviderReconciliationResult>;
+//# sourceMappingURL=usage-reconciliation.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"usage-reconciliation.d.ts","sourceRoot":"","sources":["../src/usage-reconciliation.ts"],"names":[],"mappings":"AACA,OAAO,EAYL,KAAK,4BAA4B,EACjC,KAAK,sBAAsB,EAC5B,MAAM,8BAA8B,CAAC;AAItC,wBAAsB,oBAAoB,CAAC,OAAO,EAAE,sBAAsB,GAAG,OAAO,CAAC,4BAA4B,CAAC,CAsFjH"}

package/dist/usage-reconciliation.js

@@ -0,0 +1,143 @@
+import { compactUsageJson, fetchUsageJson, settledUsageRange, usageBodyString, usageControlKey, usageInteger, usageNumber, usageObject, usageObjectList, usageReconciliationReport, usageString } from "@assemblyline-agents/runtime";
+const DEFAULT_SETTLEMENT_LAG_HOURS = 48;
+export async function reconcileOpenAIUsage(context) {
+    const from = usageBodyString(context.input, "from");
+    if (!from)
+        throw new Error("Usage reconciliation requires an ISO `from` timestamp.");
+    const provider = usageBodyString(context.input, "providerLabel") ?? "openai";
+    const adminKey = context.env.ASSEMBLY_LINE_OPENAI_ADMIN_KEY ?? context.env.OPENAI_ADMIN_KEY;
+    if (!adminKey?.trim())
+        throw new Error("ASSEMBLY_LINE_OPENAI_ADMIN_KEY or OPENAI_ADMIN_KEY is required for OpenAI reconciliation.");
+    const range = settledUsageRange(from, usageBodyString(context.input, "to"), new Date(), optionalNumber(context.env.ASSEMBLY_LINE_USAGE_RECONCILIATION_LAG_HOURS) ?? DEFAULT_SETTLEMENT_LAG_HOURS);
+    const baseUrl = (context.env.ASSEMBLY_LINE_OPENAI_USAGE_BASE_URL?.trim() || "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 projectIds = commaList(context.env.ASSEMBLY_LINE_OPENAI_USAGE_PROJECT_IDS);
+    const apiKeyIds = commaList(context.env.ASSEMBLY_LINE_OPENAI_USAGE_API_KEY_IDS);
+    const agentByProjectId = jsonStringMap(context.env.ASSEMBLY_LINE_OPENAI_AGENT_BY_PROJECT_ID);
+    const agentByApiKeyId = jsonStringMap(context.env.ASSEMBLY_LINE_OPENAI_AGENT_BY_API_KEY_ID);
+    const usageBuckets = await fetchPages(`${baseUrl}/organization/usage/completions`, adminKey, common, ["model", "project_id", "api_key_id"], projectIds, apiKeyIds, context.fetch);
+    const costBuckets = await fetchPages(`${baseUrl}/organization/costs`, adminKey, common, ["project_id", "line_item"], projectIds, undefined, context.fetch);
+    let recordedControlTotals = 0;
+    for (const bucket of usageBuckets) {
+        const startTime = usageInteger(bucket.start_time);
+        const endTime = usageInteger(bucket.end_time);
+        if (startTime === undefined || endTime === undefined)
+            continue;
+        for (const result of usageObjectList(bucket.results)) {
+            const dimensions = dimensionsFor(result);
+            const model = usageString(result.model);
+            await context.state.recordUsage({
+                idempotencyKey: usageControlKey(provider, "openai-usage", startTime, dimensions),
+                ...agentAttribution(result, agentByProjectId, agentByApiKeyId),
+                provider,
+                ...(model ? { model } : {}),
+                billingMode: "api_key",
+                recordKind: "control_total",
+                inputTokens: usageInteger(result.input_tokens) ?? 0,
+                outputTokens: usageInteger(result.output_tokens) ?? 0,
+                cachedTokens: usageInteger(result.input_cached_tokens) ?? 0,
+                cacheWriteTokens: usageInteger(result.input_cache_write_tokens) ?? 0,
+                tokenSource: "provider_reconciled",
+                costSource: "unavailable",
+                providerReceipt: receipt("openai.organization.usage.completions", startTime, endTime, result),
+                metadata: compactUsageJson({ controlType: "token_usage", numModelRequests: usageInteger(result.num_model_requests), ...dimensions }),
+                occurredAt: epochIso(startTime)
+            });
+            recordedControlTotals += 1;
+        }
+    }
+    for (const bucket of costBuckets) {
+        const startTime = usageInteger(bucket.start_time);
+        const endTime = usageInteger(bucket.end_time);
+        if (startTime === undefined || endTime === undefined)
+            continue;
+        for (const result of usageObjectList(bucket.results)) {
+            const amount = usageObject(result.amount);
+            const costUsd = usageNumber(amount?.value);
+            if (costUsd === undefined)
+                continue;
+            const dimensions = dimensionsFor(result);
+            await context.state.recordUsage({
+                idempotencyKey: usageControlKey(provider, "openai-costs", startTime, dimensions),
+                ...agentAttribution(result, agentByProjectId, agentByApiKeyId),
+                provider,
+                billingMode: "api_key",
+                recordKind: "control_total",
+                costUsd,
+                currency: usageString(amount?.currency)?.toUpperCase() ?? "USD",
+                tokenSource: "unavailable",
+                costSource: "provider_reconciled",
+                providerReceipt: receipt("openai.organization.costs", startTime, endTime, result),
+                metadata: compactUsageJson({ controlType: "cash_cost", quantity: usageNumber(result.quantity), ...dimensions }),
+                occurredAt: epochIso(startTime)
+            });
+            recordedControlTotals += 1;
+        }
+    }
+    return {
+        provider,
+        ...range,
+        recordedControlTotals,
+        reconciledTransactions: 0,
+        report: await usageReconciliationReport(context.state, { provider, ...range })
+    };
+}
+async function fetchPages(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 = usageObject(await fetchUsageJson(url, key, fetchImpl));
+        buckets.push(...usageObjectList(body?.data));
+        page = body?.has_more === true ? usageString(body.next_page) : undefined;
+        if (body?.has_more === true && !page)
+            throw new Error("Provider pagination reported has_more without next_page.");
+    } while (page);
+    return buckets;
+}
+function agentAttribution(result, byProject, byKey) {
+    const apiKeyId = usageString(result.api_key_id);
+    const projectId = usageString(result.project_id);
+    const agentId = (apiKeyId ? byKey?.[apiKeyId] : undefined) ?? (projectId ? byProject?.[projectId] : undefined);
+    return agentId ? { agentId } : {};
+}
+function dimensionsFor(result) {
+    return compactUsageJson({
+        apiKeyId: usageString(result.api_key_id), projectId: usageString(result.project_id), model: usageString(result.model),
+        lineItem: usageString(result.line_item), serviceTier: usageString(result.service_tier),
+        batch: typeof result.batch === "boolean" ? result.batch : undefined
+    });
+}
+function receipt(source, start, end, result) {
+    return { source, bucketStart: epochIso(start), bucketEnd: epochIso(end), result: result };
+}
+function epochIso(value) { return new Date(value * 1_000).toISOString(); }
+function commaList(value) { const values = value?.split(",").map((entry) => entry.trim()).filter(Boolean); return values?.length ? values : undefined; }
+function optionalNumber(value) { if (!value?.trim())
+    return undefined; const parsed = Number(value); if (!Number.isFinite(parsed) || parsed < 0)
+    throw new Error("Usage reconciliation lag must be non-negative."); return parsed; }
+function jsonStringMap(value) {
+    if (!value?.trim())
+        return undefined;
+    const parsed = JSON.parse(value);
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed) || !Object.values(parsed).every((entry) => typeof entry === "string" && entry.trim())) {
+        throw new Error("Usage attribution maps must be JSON objects with string values.");
+    }
+    return Object.fromEntries(Object.entries(parsed).map(([key, entry]) => [key, entry.trim()]));
+}
+//# sourceMappingURL=usage-reconciliation.js.map

package/dist/usage-reconciliation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-reconciliation.js","sourceRoot":"","sources":["../src/usage-reconciliation.ts"],"names":[],"mappings":"AACA,OAAO,EACL,gBAAgB,EAChB,cAAc,EACd,iBAAiB,EACjB,eAAe,EACf,eAAe,EACf,YAAY,EACZ,WAAW,EACX,WAAW,EACX,eAAe,EACf,yBAAyB,EACzB,WAAW,EAGZ,MAAM,8BAA8B,CAAC;AAEtC,MAAM,4BAA4B,GAAG,EAAE,CAAC;AAExC,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,OAA+B;IACxE,MAAM,IAAI,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACpD,IAAI,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;IACrF,MAAM,QAAQ,GAAG,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,eAAe,CAAC,IAAI,QAAQ,CAAC;IAC7E,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC5F,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,2FAA2F,CAAC,CAAC;IACpI,MAAM,KAAK,GAAG,iBAAiB,CAC7B,IAAI,EACJ,eAAe,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,EACpC,IAAI,IAAI,EAAE,EACV,cAAc,CAAC,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,IAAI,4BAA4B,CACzG,CAAC;IACF,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,mCAAmC,EAAE,IAAI,EAAE,IAAI,2BAA2B,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC9H,MAAM,MAAM,GAAG;QACb,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC;QACtE,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC;QAClE,YAAY,EAAE,IAAI;KACnB,CAAC;IACF,MAAM,UAAU,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IACjF,MAAM,SAAS,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,sCAAsC,CAAC,CAAC;IAChF,MAAM,gBAAgB,GAAG,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IAC7F,MAAM,eAAe,GAAG,aAAa,CAAC,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IAC5F,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC,GAAG,OAAO,iCAAiC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,YAAY,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAClL,MAAM,WAAW,GAAG,MAAM,UAAU,CAAC,GAAG,OAAO,qBAAqB,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;IAC3J,IAAI,qBAAqB,GAAG,CAAC,CAAC;IAE9B,KAAK,MAAM,MAAM,IAAI,YAAY,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,SAAS,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS;YAAE,SAAS;QAC/D,KAAK,MAAM,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YACrD,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YACzC,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;gBAC9B,cAAc,EAAE,eAAe,CAAC,QAAQ,EAAE,cAAc,EAAE,SAAS,EAAE,UAAU,CAAC;gBAChF,GAAG,gBAAgB,CAAC,MAAM,EAAE,gBAAgB,EAAE,eAAe,CAAC;gBAC9D,QAAQ;gBACR,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3B,WAAW,EAAE,SAAS;gBACtB,UAAU,EAAE,eAAe;gBAC3B,WAAW,EAAE,YAAY,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC;gBACnD,YAAY,EAAE,YAAY,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC;gBACrD,YAAY,EAAE,YAAY,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC3D,gBAAgB,EAAE,YAAY,CAAC,MAAM,CAAC,wBAAwB,CAAC,IAAI,CAAC;gBACpE,WAAW,EAAE,qBAAqB;gBAClC,UAAU,EAAE,aAAa;gBACzB,eAAe,EAAE,OAAO,CAAC,uCAAuC,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC;gBAC7F,QAAQ,EAAE,gBAAgB,CAAC,EAAE,WAAW,EAAE,aAAa,EAAE,gBAAgB,EAAE,YAAY,CAAC,MAAM,CAAC,kBAAkB,CAAC,EAAE,GAAG,UAAU,EAAE,CAAC;gBACpI,UAAU,EAAE,QAAQ,CAAC,SAAS,CAAC;aAChC,CAAC,CAAC;YACH,qBAAqB,IAAI,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;QACjC,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,SAAS,KAAK,SAAS,IAAI,OAAO,KAAK,SAAS;YAAE,SAAS;QAC/D,KAAK,MAAM,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;YACrD,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAC3C,IAAI,OAAO,KAAK,SAAS;gBAAE,SAAS;YACpC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YACzC,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;gBAC9B,cAAc,EAAE,eAAe,CAAC,QAAQ,EAAE,cAAc,EAAE,SAAS,EAAE,UAAU,CAAC;gBAChF,GAAG,gBAAgB,CAAC,MAAM,EAAE,gBAAgB,EAAE,eAAe,CAAC;gBAC9D,QAAQ;gBACR,WAAW,EAAE,SAAS;gBACtB,UAAU,EAAE,eAAe;gBAC3B,OAAO;gBACP,QAAQ,EAAE,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,WAAW,EAAE,IAAI,KAAK;gBAC/D,WAAW,EAAE,aAAa;gBAC1B,UAAU,EAAE,qBAAqB;gBACjC,eAAe,EAAE,OAAO,CAAC,2BAA2B,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC;gBACjF,QAAQ,EAAE,gBAAgB,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,GAAG,UAAU,EAAE,CAAC;gBAC/G,UAAU,EAAE,QAAQ,CAAC,SAAS,CAAC;aAChC,CAAC,CAAC;YACH,qBAAqB,IAAI,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IACD,OAAO;QACL,QAAQ;QACR,GAAG,KAAK;QACR,qBAAqB;QACrB,sBAAsB,EAAE,CAAC;QACzB,MAAM,EAAE,MAAM,yBAAyB,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,GAAG,KAAK,EAAE,CAAC;KAC/E,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,QAAgB,EAAE,GAAW,EAAE,MAA8B,EAAE,MAAgB,EAAE,UAAgC,EAAE,SAA+B,EAAE,SAAuB;IACnM,MAAM,OAAO,GAA8B,EAAE,CAAC;IAC9C,IAAI,IAAwB,CAAC;IAC7B,GAAG,CAAC;QACF,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC9B,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACtF,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACpC,KAAK,MAAM,KAAK,IAAI,MAAM;YAAE,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACvE,KAAK,MAAM,SAAS,IAAI,UAAU,IAAI,EAAE;YAAE,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QAC5F,KAAK,MAAM,QAAQ,IAAI,SAAS,IAAI,EAAE;YAAE,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QACzF,IAAI,IAAI;YAAE,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,cAAc,CAAC,GAAG,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC,CAAC;QACpE,OAAO,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAC7C,IAAI,GAAG,IAAI,EAAE,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACzE,IAAI,IAAI,EAAE,QAAQ,KAAK,IAAI,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IACpH,CAAC,QAAQ,IAAI,EAAE;IACf,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,gBAAgB,CAAC,MAA+B,EAAE,SAA6C,EAAE,KAAyC;IACjJ,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACjD,MAAM,OAAO,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC/G,OAAO,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;AACpC,CAAC;AAED,SAAS,aAAa,CAAC,MAA+B;IACpD,OAAO,gBAAgB,CAAC;QACtB,QAAQ,EAAE,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,SAAS,EAAE,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC;QACrH,QAAQ,EAAE,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,WAAW,EAAE,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC;QACtF,KAAK,EAAE,OAAO,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;KACpE,CAAC,CAAC;AACL,CAAC;AAED,SAAS,OAAO,CAAC,MAAc,EAAE,KAAa,EAAE,GAAW,EAAE,MAA+B;IAC1F,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,MAAoB,EAAE,CAAC;AAC1G,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa,IAAY,OAAO,IAAI,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;AAC1F,SAAS,SAAS,CAAC,KAAyB,IAA0B,MAAM,MAAM,GAAG,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;AAClM,SAAS,cAAc,CAAC,KAAyB,IAAwB,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE;IAAE,OAAO,SAAS,CAAC,CAAC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,MAAM,GAAG,CAAC;IAAE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,CAAC,CAAC;AAC5Q,SAAS,aAAa,CAAC,KAAyB;IAC9C,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE;QAAE,OAAO,SAAS,CAAC;IACrC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAY,CAAC;IAC5C,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;QACzJ,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;IACrF,CAAC;IACD,OAAO,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAG,KAAgB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AAC3G,CAAC"}

package/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "@assemblyline-agents/openai",
+  "version": "6.0.0",
+  "type": "module",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "dependencies": {
+    "@earendil-works/pi-ai": "0.80.7",
+    "@assemblyline-agents/core": "6.0.0",
+    "@assemblyline-agents/runtime": "6.0.0"
+  },
+  "description": "OpenAI model provider plugin for Assembly Line's Pi harness.",
+  "files": [
+    "dist",
+    "plugin.json",
+    "ai.assemblyline"
+  ],
+  "engines": {
+    "node": ">=22.19.0"
+  },
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "check": "tsc -p tsconfig.json --noEmit"
+  }
+}

package/plugin.json

@@ -0,0 +1,10 @@
+{
+  "$schema": "https://agent-plugins.org/schemas/1.0.0/plugin.schema.json",
+  "name": "openai",
+  "version": "6.0.0",
+  "extensions": {
+    "ai.assemblyline": {
+      "entry": "./ai.assemblyline/index.cjs"
+    }
+  }
+}