@evalops/maestro 0.10.8 → 0.10.11

package/dist/services/usage-analytics/recorder.js

@@ -0,0 +1,128 @@
+import { createLogger } from "../../utils/logger.js";
+import { getUsageAnalyticsService } from "./service.js";
+const logger = createLogger("usage-analytics:recorder");
+const RECENT_USAGE_EVENT_TTL_MS = 10 * 60 * 1000;
+const RECENT_USAGE_EVENT_LIMIT = 2_000;
+const recentUsageEvents = new Map();
+function pruneRecentUsageEvents(now) {
+    for (const [key, seenAt] of recentUsageEvents) {
+        if (now - seenAt <= RECENT_USAGE_EVENT_TTL_MS) {
+            break;
+        }
+        recentUsageEvents.delete(key);
+    }
+    while (recentUsageEvents.size > RECENT_USAGE_EVENT_LIMIT) {
+        const oldestKey = recentUsageEvents.keys().next().value;
+        if (!oldestKey)
+            break;
+        recentUsageEvents.delete(oldestKey);
+    }
+}
+function rememberUsageEvent(key, now = Date.now()) {
+    pruneRecentUsageEvents(now);
+    if (recentUsageEvents.has(key)) {
+        return false;
+    }
+    recentUsageEvents.set(key, now);
+    return true;
+}
+function forgetUsageEvent(key) {
+    recentUsageEvents.delete(key);
+}
+function createUsageEventKey(params) {
+    const { message, occurredAt } = params;
+    const usage = message.usage;
+    return [
+        params.workspaceId,
+        params.agentId,
+        params.sessionId ?? "",
+        message.provider,
+        message.model,
+        occurredAt.getTime().toString(),
+        usage.input.toString(),
+        usage.output.toString(),
+        usage.cacheRead.toString(),
+        usage.cacheWrite.toString(),
+        typeof usage.cost?.total === "number" ? usage.cost.total.toString() : "",
+    ].join("\u001f");
+}
+function firstHeader(req, names) {
+    for (const name of names) {
+        const raw = req.headers[name.toLowerCase()];
+        const value = Array.isArray(raw) ? raw[0] : raw;
+        const trimmed = value?.trim();
+        if (trimmed)
+            return trimmed;
+    }
+    return undefined;
+}
+export function resolveUsageWorkspaceId(req) {
+    return (firstHeader(req, [
+        "x-maestro-workspace-id",
+        "x-composer-workspace-id",
+        "x-maestro-workspace",
+        "x-composer-workspace",
+    ]) ?? process.cwd());
+}
+export function resolveUsageAgentId(params) {
+    return (firstHeader(params.req, ["x-maestro-agent-id", "x-composer-agent-id"]) ??
+        params.sessionId ??
+        params.subject ??
+        "web");
+}
+export function recordAssistantUsageMetric(params) {
+    const { message } = params;
+    const usage = message.usage;
+    if (!usage) {
+        return;
+    }
+    const occurredAt = typeof message.timestamp === "number" && Number.isFinite(message.timestamp)
+        ? new Date(message.timestamp)
+        : new Date();
+    const workspaceId = resolveUsageWorkspaceId(params.req);
+    const agentId = resolveUsageAgentId(params);
+    const eventKey = createUsageEventKey({
+        workspaceId,
+        agentId,
+        sessionId: params.sessionId,
+        message,
+        occurredAt,
+    });
+    if (!rememberUsageEvent(eventKey)) {
+        logger.debug("Skipped duplicate usage analytics event", {
+            workspaceId,
+            agentId,
+            sessionId: params.sessionId,
+            provider: message.provider,
+            model: message.model,
+        });
+        return;
+    }
+    void getUsageAnalyticsService()
+        .recordLlmCall({
+        workspaceId,
+        agentId,
+        sessionId: params.sessionId,
+        provider: message.provider,
+        model: message.model,
+        inputTokens: usage.input,
+        outputTokens: usage.output,
+        cacheReadTokens: usage.cacheRead,
+        cacheWriteTokens: usage.cacheWrite,
+        costUsd: usage.cost?.total,
+        occurredAt,
+    })
+        .catch((error) => {
+        forgetUsageEvent(eventKey);
+        logger.warn("Usage analytics recording failed", {
+            error: error instanceof Error ? error.message : String(error),
+            sessionId: params.sessionId,
+            provider: message.provider,
+            model: message.model,
+        });
+    });
+}
+export function resetUsageAnalyticsRecorderForTest() {
+    recentUsageEvents.clear();
+}
+//# sourceMappingURL=recorder.js.map

package/dist/services/usage-analytics/recorder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recorder.js","sourceRoot":"","sources":["../../../src/services/usage-analytics/recorder.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,wBAAwB,EAAE,MAAM,cAAc,CAAC;AAExD,MAAM,MAAM,GAAG,YAAY,CAAC,0BAA0B,CAAC,CAAC;AACxD,MAAM,yBAAyB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AACjD,MAAM,wBAAwB,GAAG,KAAK,CAAC;AACvC,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;AAEpD,SAAS,sBAAsB,CAAC,GAAW;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,IAAI,iBAAiB,EAAE,CAAC;QAC/C,IAAI,GAAG,GAAG,MAAM,IAAI,yBAAyB,EAAE,CAAC;YAC/C,MAAM;QACP,CAAC;QACD,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,OAAO,iBAAiB,CAAC,IAAI,GAAG,wBAAwB,EAAE,CAAC;QAC1D,MAAM,SAAS,GAAG,iBAAiB,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QACxD,IAAI,CAAC,SAAS;YAAE,MAAM;QACtB,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;AACF,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAW,EAAE,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE;IACxD,sBAAsB,CAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;QAChC,OAAO,KAAK,CAAC;IACd,CAAC;IACD,iBAAiB,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAChC,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAW;IACpC,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,mBAAmB,CAAC,MAM5B;IACA,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;IACvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,OAAO;QACN,MAAM,CAAC,WAAW;QAClB,MAAM,CAAC,OAAO;QACd,MAAM,CAAC,SAAS,IAAI,EAAE;QACtB,OAAO,CAAC,QAAQ;QAChB,OAAO,CAAC,KAAK;QACb,UAAU,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;QAC/B,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE;QACtB,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE;QACvB,KAAK,CAAC,SAAS,CAAC,QAAQ,EAAE;QAC1B,KAAK,CAAC,UAAU,CAAC,QAAQ,EAAE;QAC3B,OAAO,KAAK,CAAC,IAAI,EAAE,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE;KACxE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,WAAW,CACnB,GAAoB,EACpB,KAAe;IAEf,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;QAChD,MAAM,OAAO,GAAG,KAAK,EAAE,IAAI,EAAE,CAAC;QAC9B,IAAI,OAAO;YAAE,OAAO,OAAO,CAAC;IAC7B,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,GAAoB;IAC3D,OAAO,CACN,WAAW,CAAC,GAAG,EAAE;QAChB,wBAAwB;QACxB,yBAAyB;QACzB,qBAAqB;QACrB,sBAAsB;KACtB,CAAC,IAAI,OAAO,CAAC,GAAG,EAAE,CACnB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAInC;IACA,OAAO,CACN,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,oBAAoB,EAAE,qBAAqB,CAAC,CAAC;QACtE,MAAM,CAAC,SAAS;QAChB,MAAM,CAAC,OAAO;QACd,KAAK,CACL,CAAC;AACH,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,MAK1C;IACA,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;IAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IAC5B,IAAI,CAAC,KAAK,EAAE,CAAC;QACZ,OAAO;IACR,CAAC;IAED,MAAM,UAAU,GACf,OAAO,OAAO,CAAC,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC;QAC1E,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC;QAC7B,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;IACf,MAAM,WAAW,GAAG,uBAAuB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACxD,MAAM,OAAO,GAAG,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAC5C,MAAM,QAAQ,GAAG,mBAAmB,CAAC;QACpC,WAAW;QACX,OAAO;QACP,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,OAAO;QACP,UAAU;KACV,CAAC,CAAC;IAEH,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,yCAAyC,EAAE;YACvD,WAAW;YACX,OAAO;YACP,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;SACpB,CAAC,CAAC;QACH,OAAO;IACR,CAAC;IAED,KAAK,wBAAwB,EAAE;SAC7B,aAAa,CAAC;QACd,WAAW;QACX,OAAO;QACP,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,WAAW,EAAE,KAAK,CAAC,KAAK;QACxB,YAAY,EAAE,KAAK,CAAC,MAAM;QAC1B,eAAe,EAAE,KAAK,CAAC,SAAS;QAChC,gBAAgB,EAAE,KAAK,CAAC,UAAU;QAClC,OAAO,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK;QAC1B,UAAU;KACV,CAAC;SACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QAChB,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC3B,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;YAC/C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7D,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,KAAK,EAAE,OAAO,CAAC,KAAK;SACpB,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,kCAAkC;IACjD,iBAAiB,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC","sourcesContent":["import type { IncomingMessage } from \"node:http\";\nimport type { AssistantMessage } from \"../../agent/types.js\";\nimport { createLogger } from \"../../utils/logger.js\";\nimport { getUsageAnalyticsService } from \"./service.js\";\n\nconst logger = createLogger(\"usage-analytics:recorder\");\nconst RECENT_USAGE_EVENT_TTL_MS = 10 * 60 * 1000;\nconst RECENT_USAGE_EVENT_LIMIT = 2_000;\nconst recentUsageEvents = new Map<string, number>();\n\nfunction pruneRecentUsageEvents(now: number): void {\n\tfor (const [key, seenAt] of recentUsageEvents) {\n\t\tif (now - seenAt <= RECENT_USAGE_EVENT_TTL_MS) {\n\t\t\tbreak;\n\t\t}\n\t\trecentUsageEvents.delete(key);\n\t}\n\n\twhile (recentUsageEvents.size > RECENT_USAGE_EVENT_LIMIT) {\n\t\tconst oldestKey = recentUsageEvents.keys().next().value;\n\t\tif (!oldestKey) break;\n\t\trecentUsageEvents.delete(oldestKey);\n\t}\n}\n\nfunction rememberUsageEvent(key: string, now = Date.now()): boolean {\n\tpruneRecentUsageEvents(now);\n\tif (recentUsageEvents.has(key)) {\n\t\treturn false;\n\t}\n\trecentUsageEvents.set(key, now);\n\treturn true;\n}\n\nfunction forgetUsageEvent(key: string): void {\n\trecentUsageEvents.delete(key);\n}\n\nfunction createUsageEventKey(params: {\n\tworkspaceId: string;\n\tagentId: string;\n\tsessionId?: string;\n\tmessage: AssistantMessage;\n\toccurredAt: Date;\n}): string {\n\tconst { message, occurredAt } = params;\n\tconst usage = message.usage;\n\treturn [\n\t\tparams.workspaceId,\n\t\tparams.agentId,\n\t\tparams.sessionId ?? \"\",\n\t\tmessage.provider,\n\t\tmessage.model,\n\t\toccurredAt.getTime().toString(),\n\t\tusage.input.toString(),\n\t\tusage.output.toString(),\n\t\tusage.cacheRead.toString(),\n\t\tusage.cacheWrite.toString(),\n\t\ttypeof usage.cost?.total === \"number\" ? usage.cost.total.toString() : \"\",\n\t].join(\"\\u001f\");\n}\n\nfunction firstHeader(\n\treq: IncomingMessage,\n\tnames: string[],\n): string | undefined {\n\tfor (const name of names) {\n\t\tconst raw = req.headers[name.toLowerCase()];\n\t\tconst value = Array.isArray(raw) ? raw[0] : raw;\n\t\tconst trimmed = value?.trim();\n\t\tif (trimmed) return trimmed;\n\t}\n\treturn undefined;\n}\n\nexport function resolveUsageWorkspaceId(req: IncomingMessage): string {\n\treturn (\n\t\tfirstHeader(req, [\n\t\t\t\"x-maestro-workspace-id\",\n\t\t\t\"x-composer-workspace-id\",\n\t\t\t\"x-maestro-workspace\",\n\t\t\t\"x-composer-workspace\",\n\t\t]) ?? process.cwd()\n\t);\n}\n\nexport function resolveUsageAgentId(params: {\n\treq: IncomingMessage;\n\tsessionId?: string;\n\tsubject?: string;\n}): string {\n\treturn (\n\t\tfirstHeader(params.req, [\"x-maestro-agent-id\", \"x-composer-agent-id\"]) ??\n\t\tparams.sessionId ??\n\t\tparams.subject ??\n\t\t\"web\"\n\t);\n}\n\nexport function recordAssistantUsageMetric(params: {\n\treq: IncomingMessage;\n\tmessage: AssistantMessage;\n\tsessionId?: string;\n\tsubject?: string;\n}): void {\n\tconst { message } = params;\n\tconst usage = message.usage;\n\tif (!usage) {\n\t\treturn;\n\t}\n\n\tconst occurredAt =\n\t\ttypeof message.timestamp === \"number\" && Number.isFinite(message.timestamp)\n\t\t\t? new Date(message.timestamp)\n\t\t\t: new Date();\n\tconst workspaceId = resolveUsageWorkspaceId(params.req);\n\tconst agentId = resolveUsageAgentId(params);\n\tconst eventKey = createUsageEventKey({\n\t\tworkspaceId,\n\t\tagentId,\n\t\tsessionId: params.sessionId,\n\t\tmessage,\n\t\toccurredAt,\n\t});\n\n\tif (!rememberUsageEvent(eventKey)) {\n\t\tlogger.debug(\"Skipped duplicate usage analytics event\", {\n\t\t\tworkspaceId,\n\t\t\tagentId,\n\t\t\tsessionId: params.sessionId,\n\t\t\tprovider: message.provider,\n\t\t\tmodel: message.model,\n\t\t});\n\t\treturn;\n\t}\n\n\tvoid getUsageAnalyticsService()\n\t\t.recordLlmCall({\n\t\t\tworkspaceId,\n\t\t\tagentId,\n\t\t\tsessionId: params.sessionId,\n\t\t\tprovider: message.provider,\n\t\t\tmodel: message.model,\n\t\t\tinputTokens: usage.input,\n\t\t\toutputTokens: usage.output,\n\t\t\tcacheReadTokens: usage.cacheRead,\n\t\t\tcacheWriteTokens: usage.cacheWrite,\n\t\t\tcostUsd: usage.cost?.total,\n\t\t\toccurredAt,\n\t\t})\n\t\t.catch((error) => {\n\t\t\tforgetUsageEvent(eventKey);\n\t\t\tlogger.warn(\"Usage analytics recording failed\", {\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\tsessionId: params.sessionId,\n\t\t\t\tprovider: message.provider,\n\t\t\t\tmodel: message.model,\n\t\t\t});\n\t\t});\n}\n\nexport function resetUsageAnalyticsRecorderForTest(): void {\n\trecentUsageEvents.clear();\n}\n"]}

package/dist/services/usage-analytics/service.d.ts

@@ -0,0 +1,15 @@
+import { type DbClient } from "../../db/client.js";
+import type { UsageAnalyticsQuery, UsageAnalyticsRecordResult, UsageMetricInput } from "./types.js";
+export declare class UsageAnalyticsUnavailableError extends Error {
+    constructor();
+}
+export declare class UsageAnalyticsService {
+    private readonly getDatabase;
+    private readonly databaseConfigured;
+    constructor(getDatabase?: () => DbClient, databaseConfigured?: () => boolean);
+    recordLlmCall(input: UsageMetricInput): Promise<UsageAnalyticsRecordResult>;
+    queryUsage(query: UsageAnalyticsQuery): Promise<import("./types.js").UsageAnalyticsReport>;
+}
+export declare function getUsageAnalyticsService(): UsageAnalyticsService;
+export declare function setUsageAnalyticsServiceForTest(service: UsageAnalyticsService | null): void;
+//# sourceMappingURL=service.d.ts.map

package/dist/services/usage-analytics/service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../../../src/services/usage-analytics/service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,QAAQ,EAA+B,MAAM,oBAAoB,CAAC;AAOhF,OAAO,KAAK,EAEX,mBAAmB,EACnB,0BAA0B,EAC1B,gBAAgB,EAChB,MAAM,YAAY,CAAC;AAIpB,qBAAa,8BAA+B,SAAQ,KAAK;;CAKxD;AAuED,qBAAa,qBAAqB;IAEhC,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,kBAAkB;gBADlB,WAAW,GAAE,MAAM,QAAgB,EACnC,kBAAkB,GAAE,MAAM,OAA8B;IAGpE,aAAa,CAClB,KAAK,EAAE,gBAAgB,GACrB,OAAO,CAAC,0BAA0B,CAAC;IAmEhC,UAAU,CAAC,KAAK,EAAE,mBAAmB;CA0D3C;AAID,wBAAgB,wBAAwB,IAAI,qBAAqB,CAGhE;AAED,wBAAgB,+BAA+B,CAC9C,OAAO,EAAE,qBAAqB,GAAG,IAAI,GACnC,IAAI,CAEN"}

package/dist/services/usage-analytics/service.js

@@ -0,0 +1,210 @@
+import { sql } from "drizzle-orm";
+import { getDb, isDatabaseConfigured } from "../../db/client.js";
+import { usageMetrics } from "../../db/schema.js";
+import { createLogger } from "../../utils/logger.js";
+import { createUsageAnalyticsReport, normalizeUsageMetricInput, } from "./aggregation.js";
+const logger = createLogger("usage-analytics");
+export class UsageAnalyticsUnavailableError extends Error {
+    constructor() {
+        super("Usage analytics database is not configured.");
+        this.name = "UsageAnalyticsUnavailableError";
+    }
+}
+function costUsdFromMicros(micros) {
+    return micros / 1_000_000;
+}
+function readNumber(row, key) {
+    const value = row[key];
+    if (typeof value === "number" && Number.isFinite(value))
+        return value;
+    if (typeof value === "bigint")
+        return Number(value);
+    if (typeof value === "string") {
+        const parsed = Number(value);
+        if (Number.isFinite(parsed))
+            return parsed;
+    }
+    return 0;
+}
+function readString(row, key) {
+    const value = row[key];
+    return typeof value === "string" ? value : "";
+}
+function readDateIso(row, key) {
+    const value = row[key];
+    if (value instanceof Date)
+        return value.toISOString();
+    if (typeof value === "string")
+        return new Date(value).toISOString();
+    return new Date(0).toISOString();
+}
+function periodUnit(period) {
+    if (period === "weekly")
+        return "week";
+    if (period === "monthly")
+        return "month";
+    return "day";
+}
+function buildFilters(query) {
+    const filters = [];
+    if (query.workspaceId) {
+        filters.push(sql `workspace_id = ${query.workspaceId}`);
+    }
+    if (query.agentId) {
+        filters.push(sql `agent_id = ${query.agentId}`);
+    }
+    if (query.provider) {
+        filters.push(sql `provider = ${query.provider}`);
+    }
+    if (query.model) {
+        filters.push(sql `model = ${query.model}`);
+    }
+    if (query.from) {
+        filters.push(sql `bucket_start >= ${query.from}`);
+    }
+    if (query.to) {
+        filters.push(sql `bucket_start <= ${query.to}`);
+    }
+    return filters;
+}
+function queryFiltersForResponse(query) {
+    return {
+        ...(query.workspaceId ? { workspaceId: query.workspaceId } : {}),
+        ...(query.agentId ? { agentId: query.agentId } : {}),
+        ...(query.provider ? { provider: query.provider } : {}),
+        ...(query.model ? { model: query.model } : {}),
+        ...(query.from ? { from: query.from.toISOString() } : {}),
+        ...(query.to ? { to: query.to.toISOString() } : {}),
+    };
+}
+export class UsageAnalyticsService {
+    constructor(getDatabase = getDb, databaseConfigured = isDatabaseConfigured) {
+        this.getDatabase = getDatabase;
+        this.databaseConfigured = databaseConfigured;
+    }
+    async recordLlmCall(input) {
+        if (!this.databaseConfigured()) {
+            return { recorded: false, reason: "database_unconfigured" };
+        }
+        const normalized = normalizeUsageMetricInput(input);
+        const db = this.getDatabase();
+        const now = new Date();
+        try {
+            await db
+                .insert(usageMetrics)
+                .values({
+                workspaceId: normalized.workspaceId,
+                agentId: normalized.agentId,
+                lastSessionId: normalized.sessionId,
+                provider: normalized.provider,
+                model: normalized.model,
+                bucketStart: normalized.bucketStart,
+                callCount: normalized.callCount,
+                inputTokens: normalized.inputTokens,
+                outputTokens: normalized.outputTokens,
+                cacheReadTokens: normalized.cacheReadTokens,
+                cacheWriteTokens: normalized.cacheWriteTokens,
+                totalTokens: normalized.totalTokens,
+                costUsdMicros: normalized.costUsdMicros,
+                firstSeenAt: normalized.occurredAt,
+                lastSeenAt: normalized.occurredAt,
+                createdAt: now,
+                updatedAt: now,
+            })
+                .onConflictDoUpdate({
+                target: [
+                    usageMetrics.workspaceId,
+                    usageMetrics.agentId,
+                    usageMetrics.provider,
+                    usageMetrics.model,
+                    usageMetrics.bucketStart,
+                ],
+                set: {
+                    lastSessionId: normalized.sessionId ?? sql `${usageMetrics.lastSessionId}`,
+                    callCount: sql `${usageMetrics.callCount} + 1`,
+                    inputTokens: sql `${usageMetrics.inputTokens} + ${normalized.inputTokens}`,
+                    outputTokens: sql `${usageMetrics.outputTokens} + ${normalized.outputTokens}`,
+                    cacheReadTokens: sql `${usageMetrics.cacheReadTokens} + ${normalized.cacheReadTokens}`,
+                    cacheWriteTokens: sql `${usageMetrics.cacheWriteTokens} + ${normalized.cacheWriteTokens}`,
+                    totalTokens: sql `${usageMetrics.totalTokens} + ${normalized.totalTokens}`,
+                    costUsdMicros: sql `${usageMetrics.costUsdMicros} + ${normalized.costUsdMicros}`,
+                    firstSeenAt: sql `LEAST(${usageMetrics.firstSeenAt}, ${normalized.occurredAt})`,
+                    lastSeenAt: sql `GREATEST(${usageMetrics.lastSeenAt}, ${normalized.occurredAt})`,
+                    updatedAt: now,
+                },
+            });
+            return { recorded: true };
+        }
+        catch (error) {
+            logger.warn("Failed to record usage analytics", {
+                error: error instanceof Error ? error.message : String(error),
+                workspaceId: normalized.workspaceId,
+                agentId: normalized.agentId,
+                provider: normalized.provider,
+                model: normalized.model,
+            });
+            throw error;
+        }
+    }
+    async queryUsage(query) {
+        if (!this.databaseConfigured()) {
+            throw new UsageAnalyticsUnavailableError();
+        }
+        const filters = buildFilters(query);
+        const where = filters.length > 0 ? sql `WHERE ${sql.join(filters, sql ` AND `)}` : sql ``;
+        const unit = periodUnit(query.period);
+        const db = this.getDatabase();
+        const result = await db.execute(sql `
+			SELECT
+				date_trunc(${unit}, bucket_start) AS bucket_start,
+				workspace_id,
+				agent_id,
+				provider,
+				model,
+				COALESCE(SUM(call_count), 0) AS calls,
+				COALESCE(SUM(input_tokens), 0) AS input_tokens,
+				COALESCE(SUM(output_tokens), 0) AS output_tokens,
+				COALESCE(SUM(cache_read_tokens), 0) AS cache_read_tokens,
+				COALESCE(SUM(cache_write_tokens), 0) AS cache_write_tokens,
+				COALESCE(SUM(total_tokens), 0) AS total_tokens,
+				COALESCE(SUM(cost_usd_micros), 0) AS cost_usd_micros
+			FROM usage_metrics
+			${where}
+			GROUP BY 1, workspace_id, agent_id, provider, model
+			ORDER BY 1 ASC, workspace_id ASC, agent_id ASC, provider ASC, model ASC
+		`);
+        const buckets = Array.from(result).map((row) => {
+            const costUsdMicros = readNumber(row, "cost_usd_micros");
+            return {
+                bucketStart: readDateIso(row, "bucket_start"),
+                workspaceId: readString(row, "workspace_id"),
+                agentId: readString(row, "agent_id"),
+                provider: readString(row, "provider"),
+                model: readString(row, "model"),
+                calls: readNumber(row, "calls"),
+                tokens: {
+                    input: readNumber(row, "input_tokens"),
+                    output: readNumber(row, "output_tokens"),
+                    cacheRead: readNumber(row, "cache_read_tokens"),
+                    cacheWrite: readNumber(row, "cache_write_tokens"),
+                    total: readNumber(row, "total_tokens"),
+                },
+                costUsd: costUsdFromMicros(costUsdMicros),
+            };
+        });
+        return createUsageAnalyticsReport({
+            period: query.period,
+            filters: queryFiltersForResponse(query),
+            buckets,
+        });
+    }
+}
+let defaultUsageAnalyticsService = null;
+export function getUsageAnalyticsService() {
+    defaultUsageAnalyticsService ??= new UsageAnalyticsService();
+    return defaultUsageAnalyticsService;
+}
+export function setUsageAnalyticsServiceForTest(service) {
+    defaultUsageAnalyticsService = service;
+}
+//# sourceMappingURL=service.js.map

package/dist/services/usage-analytics/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../../src/services/usage-analytics/service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAY,GAAG,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAiB,KAAK,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EACN,0BAA0B,EAC1B,yBAAyB,GACzB,MAAM,kBAAkB,CAAC;AAQ1B,MAAM,MAAM,GAAG,YAAY,CAAC,iBAAiB,CAAC,CAAC;AAE/C,MAAM,OAAO,8BAA+B,SAAQ,KAAK;IACxD;QACC,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,gCAAgC,CAAC;IAC9C,CAAC;CACD;AAED,SAAS,iBAAiB,CAAC,MAAc;IACxC,OAAO,MAAM,GAAG,SAAS,CAAC;AAC3B,CAAC;AAED,SAAS,UAAU,CAAC,GAA4B,EAAE,GAAW;IAC5D,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;IACvB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACtE,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACpD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC/B,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,MAAM,CAAC;IAC5C,CAAC;IACD,OAAO,CAAC,CAAC;AACV,CAAC;AAED,SAAS,UAAU,CAAC,GAA4B,EAAE,GAAW;IAC5D,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;IACvB,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAC/C,CAAC;AAED,SAAS,WAAW,CAAC,GAA4B,EAAE,GAAW;IAC7D,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;IACvB,IAAI,KAAK,YAAY,IAAI;QAAE,OAAO,KAAK,CAAC,WAAW,EAAE,CAAC;IACtD,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IACpE,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;AAClC,CAAC;AAED,SAAS,UAAU,CAClB,MAAqC;IAErC,IAAI,MAAM,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IACvC,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC;IACzC,OAAO,KAAK,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,KAA0B;IAC/C,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,kBAAkB,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,cAAc,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAChD,CAAC;IACD,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,cAAc,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjD,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,WAAW,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IAC3C,CAAC;IACD,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QAChB,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,mBAAmB,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IACD,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;QACd,OAAO,CAAC,IAAI,CAAC,GAAG,CAAA,mBAAmB,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,uBAAuB,CAAC,KAA0B;IAC1D,OAAO;QACN,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChE,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACpD,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACvD,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9C,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzD,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACnD,CAAC;AACH,CAAC;AAED,MAAM,OAAO,qBAAqB;IACjC,YACkB,cAA8B,KAAK,EACnC,qBAAoC,oBAAoB;QADxD,gBAAW,GAAX,WAAW,CAAwB;QACnC,uBAAkB,GAAlB,kBAAkB,CAAsC;IACvE,CAAC;IAEJ,KAAK,CAAC,aAAa,CAClB,KAAuB;QAEvB,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAChC,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,uBAAuB,EAAE,CAAC;QAC7D,CAAC;QAED,MAAM,UAAU,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QAEvB,IAAI,CAAC;YACJ,MAAM,EAAE;iBACN,MAAM,CAAC,YAAY,CAAC;iBACpB,MAAM,CAAC;gBACP,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,aAAa,EAAE,UAAU,CAAC,SAAS;gBACnC,QAAQ,EAAE,UAAU,CAAC,QAAQ;gBAC7B,KAAK,EAAE,UAAU,CAAC,KAAK;gBACvB,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,SAAS,EAAE,UAAU,CAAC,SAAS;gBAC/B,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,YAAY,EAAE,UAAU,CAAC,YAAY;gBACrC,eAAe,EAAE,UAAU,CAAC,eAAe;gBAC3C,gBAAgB,EAAE,UAAU,CAAC,gBAAgB;gBAC7C,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,aAAa,EAAE,UAAU,CAAC,aAAa;gBACvC,WAAW,EAAE,UAAU,CAAC,UAAU;gBAClC,UAAU,EAAE,UAAU,CAAC,UAAU;gBACjC,SAAS,EAAE,GAAG;gBACd,SAAS,EAAE,GAAG;aACd,CAAC;iBACD,kBAAkB,CAAC;gBACnB,MAAM,EAAE;oBACP,YAAY,CAAC,WAAW;oBACxB,YAAY,CAAC,OAAO;oBACpB,YAAY,CAAC,QAAQ;oBACrB,YAAY,CAAC,KAAK;oBAClB,YAAY,CAAC,WAAW;iBACxB;gBACD,GAAG,EAAE;oBACJ,aAAa,EACZ,UAAU,CAAC,SAAS,IAAI,GAAG,CAAA,GAAG,YAAY,CAAC,aAAa,EAAE;oBAC3D,SAAS,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,SAAS,MAAM;oBAC7C,WAAW,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,WAAW,MAAM,UAAU,CAAC,WAAW,EAAE;oBACzE,YAAY,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,YAAY,MAAM,UAAU,CAAC,YAAY,EAAE;oBAC5E,eAAe,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,eAAe,MAAM,UAAU,CAAC,eAAe,EAAE;oBACrF,gBAAgB,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,gBAAgB,MAAM,UAAU,CAAC,gBAAgB,EAAE;oBACxF,WAAW,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,WAAW,MAAM,UAAU,CAAC,WAAW,EAAE;oBACzE,aAAa,EAAE,GAAG,CAAA,GAAG,YAAY,CAAC,aAAa,MAAM,UAAU,CAAC,aAAa,EAAE;oBAC/E,WAAW,EAAE,GAAG,CAAA,SAAS,YAAY,CAAC,WAAW,KAAK,UAAU,CAAC,UAAU,GAAG;oBAC9E,UAAU,EAAE,GAAG,CAAA,YAAY,YAAY,CAAC,UAAU,KAAK,UAAU,CAAC,UAAU,GAAG;oBAC/E,SAAS,EAAE,GAAG;iBACd;aACD,CAAC,CAAC;YACJ,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE;gBAC/C,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC7D,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,QAAQ,EAAE,UAAU,CAAC,QAAQ;gBAC7B,KAAK,EAAE,UAAU,CAAC,KAAK;aACvB,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAA0B;QAC1C,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;YAChC,MAAM,IAAI,8BAA8B,EAAE,CAAC;QAC5C,CAAC;QAED,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,KAAK,GACV,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAA,SAAS,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAA,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAA,EAAE,CAAC;QAC1E,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,CAAA;;iBAEpB,IAAI;;;;;;;;;;;;;KAahB,KAAK;;;GAGP,CAAC,CAAC;QAEH,MAAM,OAAO,GACZ,KAAK,CAAC,IAAI,CAAC,MAAM,CACjB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE,iBAAiB,CAAC,CAAC;YACzD,OAAO;gBACN,WAAW,EAAE,WAAW,CAAC,GAAG,EAAE,cAAc,CAAC;gBAC7C,WAAW,EAAE,UAAU,CAAC,GAAG,EAAE,cAAc,CAAC;gBAC5C,OAAO,EAAE,UAAU,CAAC,GAAG,EAAE,UAAU,CAAC;gBACpC,QAAQ,EAAE,UAAU,CAAC,GAAG,EAAE,UAAU,CAAC;gBACrC,KAAK,EAAE,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC;gBAC/B,KAAK,EAAE,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC;gBAC/B,MAAM,EAAE;oBACP,KAAK,EAAE,UAAU,CAAC,GAAG,EAAE,cAAc,CAAC;oBACtC,MAAM,EAAE,UAAU,CAAC,GAAG,EAAE,eAAe,CAAC;oBACxC,SAAS,EAAE,UAAU,CAAC,GAAG,EAAE,mBAAmB,CAAC;oBAC/C,UAAU,EAAE,UAAU,CAAC,GAAG,EAAE,oBAAoB,CAAC;oBACjD,KAAK,EAAE,UAAU,CAAC,GAAG,EAAE,cAAc,CAAC;iBACtC;gBACD,OAAO,EAAE,iBAAiB,CAAC,aAAa,CAAC;aACzC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,0BAA0B,CAAC;YACjC,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,OAAO,EAAE,uBAAuB,CAAC,KAAK,CAAC;YACvC,OAAO;SACP,CAAC,CAAC;IACJ,CAAC;CACD;AAED,IAAI,4BAA4B,GAAiC,IAAI,CAAC;AAEtE,MAAM,UAAU,wBAAwB;IACvC,4BAA4B,KAAK,IAAI,qBAAqB,EAAE,CAAC;IAC7D,OAAO,4BAA4B,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC9C,OAAqC;IAErC,4BAA4B,GAAG,OAAO,CAAC;AACxC,CAAC","sourcesContent":["import { type SQL, sql } from \"drizzle-orm\";\nimport { type DbClient, getDb, isDatabaseConfigured } from \"../../db/client.js\";\nimport { usageMetrics } from \"../../db/schema.js\";\nimport { createLogger } from \"../../utils/logger.js\";\nimport {\n\tcreateUsageAnalyticsReport,\n\tnormalizeUsageMetricInput,\n} from \"./aggregation.js\";\nimport type {\n\tUsageAnalyticsBucket,\n\tUsageAnalyticsQuery,\n\tUsageAnalyticsRecordResult,\n\tUsageMetricInput,\n} from \"./types.js\";\n\nconst logger = createLogger(\"usage-analytics\");\n\nexport class UsageAnalyticsUnavailableError extends Error {\n\tconstructor() {\n\t\tsuper(\"Usage analytics database is not configured.\");\n\t\tthis.name = \"UsageAnalyticsUnavailableError\";\n\t}\n}\n\nfunction costUsdFromMicros(micros: number): number {\n\treturn micros / 1_000_000;\n}\n\nfunction readNumber(row: Record<string, unknown>, key: string): number {\n\tconst value = row[key];\n\tif (typeof value === \"number\" && Number.isFinite(value)) return value;\n\tif (typeof value === \"bigint\") return Number(value);\n\tif (typeof value === \"string\") {\n\t\tconst parsed = Number(value);\n\t\tif (Number.isFinite(parsed)) return parsed;\n\t}\n\treturn 0;\n}\n\nfunction readString(row: Record<string, unknown>, key: string): string {\n\tconst value = row[key];\n\treturn typeof value === \"string\" ? value : \"\";\n}\n\nfunction readDateIso(row: Record<string, unknown>, key: string): string {\n\tconst value = row[key];\n\tif (value instanceof Date) return value.toISOString();\n\tif (typeof value === \"string\") return new Date(value).toISOString();\n\treturn new Date(0).toISOString();\n}\n\nfunction periodUnit(\n\tperiod: UsageAnalyticsQuery[\"period\"],\n): \"day\" | \"week\" | \"month\" {\n\tif (period === \"weekly\") return \"week\";\n\tif (period === \"monthly\") return \"month\";\n\treturn \"day\";\n}\n\nfunction buildFilters(query: UsageAnalyticsQuery): SQL[] {\n\tconst filters: SQL[] = [];\n\tif (query.workspaceId) {\n\t\tfilters.push(sql`workspace_id = ${query.workspaceId}`);\n\t}\n\tif (query.agentId) {\n\t\tfilters.push(sql`agent_id = ${query.agentId}`);\n\t}\n\tif (query.provider) {\n\t\tfilters.push(sql`provider = ${query.provider}`);\n\t}\n\tif (query.model) {\n\t\tfilters.push(sql`model = ${query.model}`);\n\t}\n\tif (query.from) {\n\t\tfilters.push(sql`bucket_start >= ${query.from}`);\n\t}\n\tif (query.to) {\n\t\tfilters.push(sql`bucket_start <= ${query.to}`);\n\t}\n\treturn filters;\n}\n\nfunction queryFiltersForResponse(query: UsageAnalyticsQuery) {\n\treturn {\n\t\t...(query.workspaceId ? { workspaceId: query.workspaceId } : {}),\n\t\t...(query.agentId ? { agentId: query.agentId } : {}),\n\t\t...(query.provider ? { provider: query.provider } : {}),\n\t\t...(query.model ? { model: query.model } : {}),\n\t\t...(query.from ? { from: query.from.toISOString() } : {}),\n\t\t...(query.to ? { to: query.to.toISOString() } : {}),\n\t};\n}\n\nexport class UsageAnalyticsService {\n\tconstructor(\n\t\tprivate readonly getDatabase: () => DbClient = getDb,\n\t\tprivate readonly databaseConfigured: () => boolean = isDatabaseConfigured,\n\t) {}\n\n\tasync recordLlmCall(\n\t\tinput: UsageMetricInput,\n\t): Promise<UsageAnalyticsRecordResult> {\n\t\tif (!this.databaseConfigured()) {\n\t\t\treturn { recorded: false, reason: \"database_unconfigured\" };\n\t\t}\n\n\t\tconst normalized = normalizeUsageMetricInput(input);\n\t\tconst db = this.getDatabase();\n\t\tconst now = new Date();\n\n\t\ttry {\n\t\t\tawait db\n\t\t\t\t.insert(usageMetrics)\n\t\t\t\t.values({\n\t\t\t\t\tworkspaceId: normalized.workspaceId,\n\t\t\t\t\tagentId: normalized.agentId,\n\t\t\t\t\tlastSessionId: normalized.sessionId,\n\t\t\t\t\tprovider: normalized.provider,\n\t\t\t\t\tmodel: normalized.model,\n\t\t\t\t\tbucketStart: normalized.bucketStart,\n\t\t\t\t\tcallCount: normalized.callCount,\n\t\t\t\t\tinputTokens: normalized.inputTokens,\n\t\t\t\t\toutputTokens: normalized.outputTokens,\n\t\t\t\t\tcacheReadTokens: normalized.cacheReadTokens,\n\t\t\t\t\tcacheWriteTokens: normalized.cacheWriteTokens,\n\t\t\t\t\ttotalTokens: normalized.totalTokens,\n\t\t\t\t\tcostUsdMicros: normalized.costUsdMicros,\n\t\t\t\t\tfirstSeenAt: normalized.occurredAt,\n\t\t\t\t\tlastSeenAt: normalized.occurredAt,\n\t\t\t\t\tcreatedAt: now,\n\t\t\t\t\tupdatedAt: now,\n\t\t\t\t})\n\t\t\t\t.onConflictDoUpdate({\n\t\t\t\t\ttarget: [\n\t\t\t\t\t\tusageMetrics.workspaceId,\n\t\t\t\t\t\tusageMetrics.agentId,\n\t\t\t\t\t\tusageMetrics.provider,\n\t\t\t\t\t\tusageMetrics.model,\n\t\t\t\t\t\tusageMetrics.bucketStart,\n\t\t\t\t\t],\n\t\t\t\t\tset: {\n\t\t\t\t\t\tlastSessionId:\n\t\t\t\t\t\t\tnormalized.sessionId ?? sql`${usageMetrics.lastSessionId}`,\n\t\t\t\t\t\tcallCount: sql`${usageMetrics.callCount} + 1`,\n\t\t\t\t\t\tinputTokens: sql`${usageMetrics.inputTokens} + ${normalized.inputTokens}`,\n\t\t\t\t\t\toutputTokens: sql`${usageMetrics.outputTokens} + ${normalized.outputTokens}`,\n\t\t\t\t\t\tcacheReadTokens: sql`${usageMetrics.cacheReadTokens} + ${normalized.cacheReadTokens}`,\n\t\t\t\t\t\tcacheWriteTokens: sql`${usageMetrics.cacheWriteTokens} + ${normalized.cacheWriteTokens}`,\n\t\t\t\t\t\ttotalTokens: sql`${usageMetrics.totalTokens} + ${normalized.totalTokens}`,\n\t\t\t\t\t\tcostUsdMicros: sql`${usageMetrics.costUsdMicros} + ${normalized.costUsdMicros}`,\n\t\t\t\t\t\tfirstSeenAt: sql`LEAST(${usageMetrics.firstSeenAt}, ${normalized.occurredAt})`,\n\t\t\t\t\t\tlastSeenAt: sql`GREATEST(${usageMetrics.lastSeenAt}, ${normalized.occurredAt})`,\n\t\t\t\t\t\tupdatedAt: now,\n\t\t\t\t\t},\n\t\t\t\t});\n\t\t\treturn { recorded: true };\n\t\t} catch (error) {\n\t\t\tlogger.warn(\"Failed to record usage analytics\", {\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\tworkspaceId: normalized.workspaceId,\n\t\t\t\tagentId: normalized.agentId,\n\t\t\t\tprovider: normalized.provider,\n\t\t\t\tmodel: normalized.model,\n\t\t\t});\n\t\t\tthrow error;\n\t\t}\n\t}\n\n\tasync queryUsage(query: UsageAnalyticsQuery) {\n\t\tif (!this.databaseConfigured()) {\n\t\t\tthrow new UsageAnalyticsUnavailableError();\n\t\t}\n\n\t\tconst filters = buildFilters(query);\n\t\tconst where =\n\t\t\tfilters.length > 0 ? sql`WHERE ${sql.join(filters, sql` AND `)}` : sql``;\n\t\tconst unit = periodUnit(query.period);\n\t\tconst db = this.getDatabase();\n\t\tconst result = await db.execute(sql`\n\t\t\tSELECT\n\t\t\t\tdate_trunc(${unit}, bucket_start) AS bucket_start,\n\t\t\t\tworkspace_id,\n\t\t\t\tagent_id,\n\t\t\t\tprovider,\n\t\t\t\tmodel,\n\t\t\t\tCOALESCE(SUM(call_count), 0) AS calls,\n\t\t\t\tCOALESCE(SUM(input_tokens), 0) AS input_tokens,\n\t\t\t\tCOALESCE(SUM(output_tokens), 0) AS output_tokens,\n\t\t\t\tCOALESCE(SUM(cache_read_tokens), 0) AS cache_read_tokens,\n\t\t\t\tCOALESCE(SUM(cache_write_tokens), 0) AS cache_write_tokens,\n\t\t\t\tCOALESCE(SUM(total_tokens), 0) AS total_tokens,\n\t\t\t\tCOALESCE(SUM(cost_usd_micros), 0) AS cost_usd_micros\n\t\t\tFROM usage_metrics\n\t\t\t${where}\n\t\t\tGROUP BY 1, workspace_id, agent_id, provider, model\n\t\t\tORDER BY 1 ASC, workspace_id ASC, agent_id ASC, provider ASC, model ASC\n\t\t`);\n\n\t\tconst buckets: UsageAnalyticsBucket[] = (\n\t\t\tArray.from(result) as Array<Record<string, unknown>>\n\t\t).map((row) => {\n\t\t\tconst costUsdMicros = readNumber(row, \"cost_usd_micros\");\n\t\t\treturn {\n\t\t\t\tbucketStart: readDateIso(row, \"bucket_start\"),\n\t\t\t\tworkspaceId: readString(row, \"workspace_id\"),\n\t\t\t\tagentId: readString(row, \"agent_id\"),\n\t\t\t\tprovider: readString(row, \"provider\"),\n\t\t\t\tmodel: readString(row, \"model\"),\n\t\t\t\tcalls: readNumber(row, \"calls\"),\n\t\t\t\ttokens: {\n\t\t\t\t\tinput: readNumber(row, \"input_tokens\"),\n\t\t\t\t\toutput: readNumber(row, \"output_tokens\"),\n\t\t\t\t\tcacheRead: readNumber(row, \"cache_read_tokens\"),\n\t\t\t\t\tcacheWrite: readNumber(row, \"cache_write_tokens\"),\n\t\t\t\t\ttotal: readNumber(row, \"total_tokens\"),\n\t\t\t\t},\n\t\t\t\tcostUsd: costUsdFromMicros(costUsdMicros),\n\t\t\t};\n\t\t});\n\n\t\treturn createUsageAnalyticsReport({\n\t\t\tperiod: query.period,\n\t\t\tfilters: queryFiltersForResponse(query),\n\t\t\tbuckets,\n\t\t});\n\t}\n}\n\nlet defaultUsageAnalyticsService: UsageAnalyticsService | null = null;\n\nexport function getUsageAnalyticsService(): UsageAnalyticsService {\n\tdefaultUsageAnalyticsService ??= new UsageAnalyticsService();\n\treturn defaultUsageAnalyticsService;\n}\n\nexport function setUsageAnalyticsServiceForTest(\n\tservice: UsageAnalyticsService | null,\n): void {\n\tdefaultUsageAnalyticsService = service;\n}\n"]}

package/dist/services/usage-analytics/types.d.ts

@@ -0,0 +1,80 @@
+export declare const USAGE_ANALYTICS_PERIODS: readonly ["daily", "weekly", "monthly"];
+export type UsageAnalyticsPeriod = (typeof USAGE_ANALYTICS_PERIODS)[number];
+export interface UsageTokenTotals {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+    total: number;
+}
+export interface UsageMetricInput {
+    workspaceId: string;
+    agentId: string;
+    sessionId?: string;
+    provider: string;
+    model: string;
+    inputTokens?: number;
+    outputTokens?: number;
+    cacheReadTokens?: number;
+    cacheWriteTokens?: number;
+    costUsd?: number;
+    occurredAt?: Date;
+}
+export interface NormalizedUsageMetric {
+    workspaceId: string;
+    agentId: string;
+    sessionId?: string;
+    provider: string;
+    model: string;
+    bucketStart: Date;
+    occurredAt: Date;
+    callCount: number;
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens: number;
+    cacheWriteTokens: number;
+    totalTokens: number;
+    costUsdMicros: number;
+}
+export interface UsageAnalyticsQuery {
+    period: UsageAnalyticsPeriod;
+    workspaceId?: string;
+    agentId?: string;
+    provider?: string;
+    model?: string;
+    from?: Date;
+    to?: Date;
+}
+export interface UsageAnalyticsBucket {
+    bucketStart: string;
+    workspaceId: string;
+    agentId: string;
+    provider: string;
+    model: string;
+    calls: number;
+    tokens: UsageTokenTotals;
+    costUsd: number;
+}
+export interface UsageAnalyticsTotals {
+    calls: number;
+    tokens: UsageTokenTotals;
+    costUsd: number;
+}
+export interface UsageAnalyticsReport {
+    period: UsageAnalyticsPeriod;
+    filters: {
+        workspaceId?: string;
+        agentId?: string;
+        provider?: string;
+        model?: string;
+        from?: string;
+        to?: string;
+    };
+    totals: UsageAnalyticsTotals;
+    buckets: UsageAnalyticsBucket[];
+}
+export interface UsageAnalyticsRecordResult {
+    recorded: boolean;
+    reason?: "database_unconfigured";
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/services/usage-analytics/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/services/usage-analytics/types.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,uBAAuB,yCAA0C,CAAC;AAE/E,MAAM,MAAM,oBAAoB,GAAG,CAAC,OAAO,uBAAuB,CAAC,CAAC,MAAM,CAAC,CAAC;AAE5E,MAAM,WAAW,gBAAgB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,gBAAgB;IAChC,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,IAAI,CAAC;CAClB;AAED,MAAM,WAAW,qBAAqB;IACrC,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,IAAI,CAAC;IAClB,UAAU,EAAE,IAAI,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;IACxB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,mBAAmB;IACnC,MAAM,EAAE,oBAAoB,CAAC;IAC7B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE,IAAI,CAAC;IACZ,EAAE,CAAC,EAAE,IAAI,CAAC;CACV;AAED,MAAM,WAAW,oBAAoB;IACpC,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,gBAAgB,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,oBAAoB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,gBAAgB,CAAC;IACzB,OAAO,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,oBAAoB;IACpC,MAAM,EAAE,oBAAoB,CAAC;IAC7B,OAAO,EAAE;QACR,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,EAAE,CAAC,EAAE,MAAM,CAAC;KACZ,CAAC;IACF,MAAM,EAAE,oBAAoB,CAAC;IAC7B,OAAO,EAAE,oBAAoB,EAAE,CAAC;CAChC;AAED,MAAM,WAAW,0BAA0B;IAC1C,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,CAAC,EAAE,uBAAuB,CAAC;CACjC"}

package/dist/services/usage-analytics/types.js

@@ -0,0 +1,2 @@
+export const USAGE_ANALYTICS_PERIODS = ["daily", "weekly", "monthly"];
+//# sourceMappingURL=types.js.map

package/dist/services/usage-analytics/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/services/usage-analytics/types.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAU,CAAC","sourcesContent":["export const USAGE_ANALYTICS_PERIODS = [\"daily\", \"weekly\", \"monthly\"] as const;\n\nexport type UsageAnalyticsPeriod = (typeof USAGE_ANALYTICS_PERIODS)[number];\n\nexport interface UsageTokenTotals {\n\tinput: number;\n\toutput: number;\n\tcacheRead: number;\n\tcacheWrite: number;\n\ttotal: number;\n}\n\nexport interface UsageMetricInput {\n\tworkspaceId: string;\n\tagentId: string;\n\tsessionId?: string;\n\tprovider: string;\n\tmodel: string;\n\tinputTokens?: number;\n\toutputTokens?: number;\n\tcacheReadTokens?: number;\n\tcacheWriteTokens?: number;\n\tcostUsd?: number;\n\toccurredAt?: Date;\n}\n\nexport interface NormalizedUsageMetric {\n\tworkspaceId: string;\n\tagentId: string;\n\tsessionId?: string;\n\tprovider: string;\n\tmodel: string;\n\tbucketStart: Date;\n\toccurredAt: Date;\n\tcallCount: number;\n\tinputTokens: number;\n\toutputTokens: number;\n\tcacheReadTokens: number;\n\tcacheWriteTokens: number;\n\ttotalTokens: number;\n\tcostUsdMicros: number;\n}\n\nexport interface UsageAnalyticsQuery {\n\tperiod: UsageAnalyticsPeriod;\n\tworkspaceId?: string;\n\tagentId?: string;\n\tprovider?: string;\n\tmodel?: string;\n\tfrom?: Date;\n\tto?: Date;\n}\n\nexport interface UsageAnalyticsBucket {\n\tbucketStart: string;\n\tworkspaceId: string;\n\tagentId: string;\n\tprovider: string;\n\tmodel: string;\n\tcalls: number;\n\ttokens: UsageTokenTotals;\n\tcostUsd: number;\n}\n\nexport interface UsageAnalyticsTotals {\n\tcalls: number;\n\ttokens: UsageTokenTotals;\n\tcostUsd: number;\n}\n\nexport interface UsageAnalyticsReport {\n\tperiod: UsageAnalyticsPeriod;\n\tfilters: {\n\t\tworkspaceId?: string;\n\t\tagentId?: string;\n\t\tprovider?: string;\n\t\tmodel?: string;\n\t\tfrom?: string;\n\t\tto?: string;\n\t};\n\ttotals: UsageAnalyticsTotals;\n\tbuckets: UsageAnalyticsBucket[];\n}\n\nexport interface UsageAnalyticsRecordResult {\n\trecorded: boolean;\n\treason?: \"database_unconfigured\";\n}\n"]}

package/dist/services/workspace-config/index.d.ts

@@ -0,0 +1,6 @@
+export { createWorkspaceConfigMiddleware, resolveWorkspaceConfigId, } from "./middleware.js";
+export { WorkspaceConfigValidationError, normalizeModelPreferences, normalizeRateLimits, normalizeSafetyRules, normalizeWorkspaceConfigId, normalizeWorkspaceConfigInput, normalizeWorkspaceConfigListQuery, normalizeWorkspaceConfigPatchInput, parseWorkspaceConfigLimit, parseWorkspaceConfigOffset, } from "./normalize.js";
+export { WorkspacePolicyViolationError, assertModelAllowedByWorkspacePolicy, evaluateModelPolicy, evaluateSessionTokenPolicy, evaluateToolPolicy, } from "./policy.js";
+export { WorkspaceConfigService, WorkspaceConfigUnavailableError, getWorkspaceConfigService, setWorkspaceConfigServiceForTest, } from "./service.js";
+export type { WorkspaceConfig, WorkspaceConfigInput, WorkspaceConfigListQuery, WorkspaceConfigListResult, WorkspaceConfigPatchInput, WorkspaceConfigRequestContext, WorkspaceConfigSummary, WorkspaceModelPreferences, WorkspacePolicyViolation, WorkspaceRateLimits, WorkspaceSafetyRules, } from "./types.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/services/workspace-config/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/services/workspace-config/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,+BAA+B,EAC/B,wBAAwB,GACxB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACN,8BAA8B,EAC9B,yBAAyB,EACzB,mBAAmB,EACnB,oBAAoB,EACpB,0BAA0B,EAC1B,6BAA6B,EAC7B,iCAAiC,EACjC,kCAAkC,EAClC,yBAAyB,EACzB,0BAA0B,GAC1B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACN,6BAA6B,EAC7B,mCAAmC,EACnC,mBAAmB,EACnB,0BAA0B,EAC1B,kBAAkB,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EACN,sBAAsB,EACtB,+BAA+B,EAC/B,yBAAyB,EACzB,gCAAgC,GAChC,MAAM,cAAc,CAAC;AACtB,YAAY,EACX,eAAe,EACf,oBAAoB,EACpB,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,EACzB,6BAA6B,EAC7B,sBAAsB,EACtB,yBAAyB,EACzB,wBAAwB,EACxB,mBAAmB,EACnB,oBAAoB,GACpB,MAAM,YAAY,CAAC"}

package/dist/services/workspace-config/index.js

@@ -0,0 +1,5 @@
+export { createWorkspaceConfigMiddleware, resolveWorkspaceConfigId, } from "./middleware.js";
+export { WorkspaceConfigValidationError, normalizeModelPreferences, normalizeRateLimits, normalizeSafetyRules, normalizeWorkspaceConfigId, normalizeWorkspaceConfigInput, normalizeWorkspaceConfigListQuery, normalizeWorkspaceConfigPatchInput, parseWorkspaceConfigLimit, parseWorkspaceConfigOffset, } from "./normalize.js";
+export { WorkspacePolicyViolationError, assertModelAllowedByWorkspacePolicy, evaluateModelPolicy, evaluateSessionTokenPolicy, evaluateToolPolicy, } from "./policy.js";
+export { WorkspaceConfigService, WorkspaceConfigUnavailableError, getWorkspaceConfigService, setWorkspaceConfigServiceForTest, } from "./service.js";
+//# sourceMappingURL=index.js.map

package/dist/services/workspace-config/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/services/workspace-config/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,+BAA+B,EAC/B,wBAAwB,GACxB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACN,8BAA8B,EAC9B,yBAAyB,EACzB,mBAAmB,EACnB,oBAAoB,EACpB,0BAA0B,EAC1B,6BAA6B,EAC7B,iCAAiC,EACjC,kCAAkC,EAClC,yBAAyB,EACzB,0BAA0B,GAC1B,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACN,6BAA6B,EAC7B,mCAAmC,EACnC,mBAAmB,EACnB,0BAA0B,EAC1B,kBAAkB,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EACN,sBAAsB,EACtB,+BAA+B,EAC/B,yBAAyB,EACzB,gCAAgC,GAChC,MAAM,cAAc,CAAC","sourcesContent":["export {\n\tcreateWorkspaceConfigMiddleware,\n\tresolveWorkspaceConfigId,\n} from \"./middleware.js\";\nexport {\n\tWorkspaceConfigValidationError,\n\tnormalizeModelPreferences,\n\tnormalizeRateLimits,\n\tnormalizeSafetyRules,\n\tnormalizeWorkspaceConfigId,\n\tnormalizeWorkspaceConfigInput,\n\tnormalizeWorkspaceConfigListQuery,\n\tnormalizeWorkspaceConfigPatchInput,\n\tparseWorkspaceConfigLimit,\n\tparseWorkspaceConfigOffset,\n} from \"./normalize.js\";\nexport {\n\tWorkspacePolicyViolationError,\n\tassertModelAllowedByWorkspacePolicy,\n\tevaluateModelPolicy,\n\tevaluateSessionTokenPolicy,\n\tevaluateToolPolicy,\n} from \"./policy.js\";\nexport {\n\tWorkspaceConfigService,\n\tWorkspaceConfigUnavailableError,\n\tgetWorkspaceConfigService,\n\tsetWorkspaceConfigServiceForTest,\n} from \"./service.js\";\nexport type {\n\tWorkspaceConfig,\n\tWorkspaceConfigInput,\n\tWorkspaceConfigListQuery,\n\tWorkspaceConfigListResult,\n\tWorkspaceConfigPatchInput,\n\tWorkspaceConfigRequestContext,\n\tWorkspaceConfigSummary,\n\tWorkspaceModelPreferences,\n\tWorkspacePolicyViolation,\n\tWorkspaceRateLimits,\n\tWorkspaceSafetyRules,\n} from \"./types.js\";\n"]}

package/dist/services/workspace-config/middleware.d.ts

@@ -0,0 +1,5 @@
+import type { IncomingMessage } from "node:http";
+import type { Middleware } from "../../server/middleware.js";
+export declare function resolveWorkspaceConfigId(req: IncomingMessage): string;
+export declare function createWorkspaceConfigMiddleware(corsHeaders: Record<string, string>): Middleware;
+//# sourceMappingURL=middleware.d.ts.map

package/dist/services/workspace-config/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../../src/services/workspace-config/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AACjD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AA+C7D,wBAAgB,wBAAwB,CAAC,GAAG,EAAE,eAAe,GAAG,MAAM,CAarE;AAED,wBAAgB,+BAA+B,CAC9C,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACjC,UAAU,CA0DZ"}

package/dist/services/workspace-config/middleware.js

@@ -0,0 +1,98 @@
+import { setWorkspaceConfigContext } from "../../server/request-context.js";
+import { sendJson } from "../../server/server-utils.js";
+import { createLogger } from "../../utils/logger.js";
+import { WorkspaceConfigValidationError } from "./normalize.js";
+import { WorkspaceConfigUnavailableError, getWorkspaceConfigService, } from "./service.js";
+const logger = createLogger("workspace-config:middleware");
+function firstHeader(req, names) {
+    for (const name of names) {
+        const raw = req.headers[name.toLowerCase()];
+        const value = Array.isArray(raw) ? raw[0] : raw;
+        const trimmed = value?.trim();
+        if (trimmed)
+            return trimmed;
+    }
+    return undefined;
+}
+function safeUrl(req) {
+    return new URL(req.url || "/", `http://${req.headers.host || "localhost"}`);
+}
+function workspaceIdFromWorkspaceConfigRoute(pathname) {
+    const match = pathname.match(/^\/api\/workspace-configs\/([^/]+)$/);
+    if (!match?.[1])
+        return undefined;
+    const decoded = decodeURIComponent(match[1]).trim();
+    return decoded ? decoded : undefined;
+}
+function firstSearchParam(url, names) {
+    for (const name of names) {
+        const value = url.searchParams.get(name)?.trim();
+        if (value)
+            return value;
+    }
+    return undefined;
+}
+export function resolveWorkspaceConfigId(req) {
+    const url = safeUrl(req);
+    return (firstHeader(req, [
+        "x-maestro-workspace-id",
+        "x-composer-workspace-id",
+        "x-maestro-workspace",
+        "x-composer-workspace",
+    ]) ??
+        firstSearchParam(url, ["workspace_id", "workspaceId"]) ??
+        workspaceIdFromWorkspaceConfigRoute(url.pathname) ??
+        process.cwd());
+}
+export function createWorkspaceConfigMiddleware(corsHeaders) {
+    return async (req, res, next) => {
+        const url = safeUrl(req);
+        if (!url.pathname.startsWith("/api")) {
+            await next();
+            return;
+        }
+        const workspaceId = resolveWorkspaceConfigId(req);
+        const service = getWorkspaceConfigService();
+        if (!service.isConfigured()) {
+            setWorkspaceConfigContext({
+                workspaceId,
+                config: null,
+                source: "unconfigured",
+            });
+            await next();
+            return;
+        }
+        let config;
+        try {
+            config = await service.getConfig(workspaceId);
+        }
+        catch (error) {
+            if (error instanceof WorkspaceConfigUnavailableError) {
+                setWorkspaceConfigContext({
+                    workspaceId,
+                    config: null,
+                    source: "unconfigured",
+                });
+                await next();
+                return;
+            }
+            if (error instanceof WorkspaceConfigValidationError) {
+                sendJson(res, 400, { error: error.message }, corsHeaders, req);
+                return;
+            }
+            logger.warn("Failed to load workspace config", {
+                error: error instanceof Error ? error.message : String(error),
+                workspaceId,
+            });
+            sendJson(res, 503, { error: "Workspace config could not be loaded." }, corsHeaders, req);
+            return;
+        }
+        setWorkspaceConfigContext({
+            workspaceId,
+            config,
+            source: config ? "database" : "missing",
+        });
+        await next();
+    };
+}
+//# sourceMappingURL=middleware.js.map