@nocobase/app-plugin-agents 0.0.0-stage → 0.1.0-beta.2

package/dist/server/core/reports/report.service.js

@@ -0,0 +1,310 @@
+import { DEFAULT_REPORT_DAYS, MAX_REPORT_DAYS, USAGE_GROUP_BYS, } from '../../../shared/reports.js';
+import { invalid } from '../../kernel/errors.js';
+import { listAgents } from '../agents/index.js';
+import { runsRepo, usageRepo, } from '../runs/run.store.js';
+import { reliability } from './metrics.js';
+import { modelUsageRows } from './model-usage.js';
+import { listPrices } from './prices.js';
+import { aggregateUsage } from './usage.js';
+const DAY_MS = 24 * 3600 * 1000;
+const PAGE = 200;
+/** Every day of `range`, with the runs started that day by outcome and the day's cost by agent type. */
+export function runDays(range, runs, onlineCost, runnerCost) {
+    const days = new Map();
+    for (let at = range.start.getTime(); at < range.end.getTime(); at += DAY_MS) {
+        const day = new Date(at).toISOString().slice(0, 10);
+        days.set(day, {
+            day,
+            completed: 0,
+            failed: 0,
+            cancelled: 0,
+            open: 0,
+            onlineCost: onlineCost.get(day) ?? null,
+            runnerCost: runnerCost.get(day) ?? null,
+        });
+    }
+    for (const run of runs) {
+        const day = days.get(new Date(run.createdAt).toISOString().slice(0, 10));
+        if (!day)
+            continue;
+        const outcome = run.status === 'completed' ||
+            run.status === 'failed' ||
+            run.status === 'cancelled'
+            ? run.status
+            : 'open';
+        days.set(day.day, { ...day, [outcome]: day[outcome] + 1 });
+    }
+    return [...days.values()];
+}
+const DAY_PATTERN = /^\d{4}-\d{2}-\d{2}$/u;
+function day(value, name) {
+    if (value === undefined || value === '')
+        return null;
+    if (!DAY_PATTERN.test(value) ||
+        Number.isNaN(Date.parse(`${value}T00:00:00Z`)) ||
+        new Date(`${value}T00:00:00Z`).toISOString().slice(0, 10) !== value)
+        throw invalid(`${name} must be a date (YYYY-MM-DD).`);
+    return value;
+}
+/** `from` / `to` as UTC days, `to` inclusive; the last 30 days up to today by default, at most 366 days. */
+export function reportRange(query, now) {
+    const to = day(query.to, 'to') ?? now.toISOString().slice(0, 10);
+    const end = new Date(Date.parse(`${to}T00:00:00Z`) + DAY_MS);
+    const from = day(query.from, 'from') ??
+        new Date(end.getTime() - DEFAULT_REPORT_DAYS * DAY_MS)
+            .toISOString()
+            .slice(0, 10);
+    const start = new Date(`${from}T00:00:00Z`);
+    const days = (end.getTime() - start.getTime()) / DAY_MS;
+    if (days < 1 || days > MAX_REPORT_DAYS)
+        throw invalid(`from must not be after to, and a report covers at most ${MAX_REPORT_DAYS} days.`);
+    return { from, to, start, end };
+}
+const iso = (value) => value === null || value === undefined || value === ''
+    ? null
+    : new Date(value).toISOString();
+function anyOf(f, field, values) {
+    return f.or(values.map((value) => f.string(field).eq(value)));
+}
+async function inPages(ids, read) {
+    const result = [];
+    for (let at = 0; at < ids.length; at += PAGE)
+        result.push(...(await read(ids.slice(at, at + PAGE))));
+    return result;
+}
+const inRange = (field, range) => (f) => f.and([
+    f.date(field).notBefore(range.start),
+    f.date(field).before(range.end),
+]);
+export function createReportService(deps) {
+    const sees = (caller, run) => caller.allRuns ||
+        run.actorUserId === caller.userId ||
+        run.ownerUserId === caller.userId;
+    /** What the subjects' domains say of the runs' subjects, by `kind:id`; unknown kinds are visible and in no group. */
+    async function factsOf(conn, caller, runs) {
+        const byKind = new Map();
+        for (const run of runs) {
+            const ids = byKind.get(run.subjectKind) ?? new Set();
+            ids.add(run.subjectId);
+            byKind.set(run.subjectKind, ids);
+        }
+        const facts = new Map();
+        for (const [kind, ids] of byKind) {
+            const reports = deps.subjects.get(kind)?.reports;
+            const known = reports
+                ? await reports.describe(conn, caller.userId, [...ids])
+                : new Map();
+            for (const id of ids)
+                facts.set(`${kind}:${id}`, known.get(id) ??
+                    (reports
+                        ? { label: id, visible: false, group: null }
+                        : { label: id, visible: true, group: null }));
+        }
+        return facts;
+    }
+    /** The runs the caller counts: theirs or all, on visible subjects, in the group when one is asked. */
+    async function countable(conn, caller, runs, groupId) {
+        const mine = runs.filter((run) => sees(caller, run));
+        const facts = await factsOf(conn, caller, mine);
+        const kept = mine.filter((run) => {
+            const fact = facts.get(`${run.subjectKind}:${run.subjectId}`);
+            return (fact !== undefined &&
+                fact.visible &&
+                (!groupId || fact.group?.id === groupId));
+        });
+        return { runs: kept, facts };
+    }
+    async function usageRecords(conn, caller, range, filter) {
+        const usage = await usageRepo(conn).findMany({
+            filter: inRange('createdAt', range),
+        });
+        const runIds = [...new Set(usage.map((row) => row.runId))];
+        const runs = await inPages(runIds, async (page) => await runsRepo(conn).findMany({ filter: (f) => anyOf(f, 'id', page) }));
+        const { runs: kept, facts } = await countable(conn, caller, runs.filter((run) => (!filter.agentId || run.agentId === filter.agentId) &&
+            (!filter.userId || run.actorUserId === filter.userId)), filter.groupId);
+        const byId = new Map(kept.map((run) => [run.id, run]));
+        const records = [];
+        for (const row of usage) {
+            const run = byId.get(row.runId);
+            if (!run)
+                continue;
+            const started = iso(run.startedAt);
+            const finished = iso(run.finishedAt);
+            records.push({
+                runId: run.id,
+                agentId: run.agentId,
+                type: run.agentType,
+                userId: run.actorUserId,
+                tool: row.tool,
+                modelService: row.modelService ?? null,
+                model: row.model ?? null,
+                subjectKind: run.subjectKind,
+                subjectId: run.subjectId,
+                groupId: facts.get(`${run.subjectKind}:${run.subjectId}`)?.group?.id ?? null,
+                day: (iso(row.createdAt) ?? '').slice(0, 10),
+                durationMs: started && finished
+                    ? Math.max(0, Date.parse(finished) - Date.parse(started))
+                    : 0,
+                inputTokens: Number(row.inputTokens) || 0,
+                outputTokens: Number(row.outputTokens) || 0,
+                cacheReadTokens: Number(row.cacheReadTokens) || 0,
+                cacheWriteTokens: Number(row.cacheWriteTokens) || 0,
+                reasoningTokens: Number(row.reasoningTokens) || 0,
+            });
+        }
+        return { records, facts };
+    }
+    async function agentNames(conn) {
+        return new Map((await listAgents(conn, true)).map((agent) => [agent.id, agent.name]));
+    }
+    async function groupNames(conn, groupBy, keys, facts) {
+        switch (groupBy) {
+            case 'agent':
+                return agentNames(conn);
+            case 'person':
+                return new Map(await deps.people.names(conn, keys));
+            case 'group': {
+                const names = new Map();
+                for (const fact of facts.values())
+                    if (fact.group)
+                        names.set(fact.group.id, fact.group.name);
+                return names;
+            }
+            case 'subject':
+                return new Map(keys.map((key) => [key, facts.get(key)?.label ?? key]));
+            case 'type':
+                // `online` and `runner` are words the application's pages translate; no name of the server's.
+                return new Map();
+            default:
+                return new Map(keys.filter(Boolean).map((key) => [key, key]));
+        }
+    }
+    return {
+        range: (query) => reportRange(query, deps.clock.now()),
+        async modelUsage(caller, query) {
+            const range = reportRange(query, deps.clock.now());
+            if (!caller.allRuns)
+                return { from: range.from, to: range.to, rows: [] };
+            const conn = deps.tx.read();
+            return {
+                from: range.from,
+                to: range.to,
+                rows: await modelUsageRows(conn, range, await listPrices(conn)),
+            };
+        },
+        async usage(caller, query) {
+            const groupBy = query.groupBy ?? 'agent';
+            if (!USAGE_GROUP_BYS.includes(groupBy))
+                throw invalid(`groupBy must be one of ${USAGE_GROUP_BYS.join(', ')}.`);
+            const range = reportRange(query, deps.clock.now());
+            const conn = deps.tx.read();
+            const { records, facts } = await usageRecords(conn, caller, range, {
+                groupId: query.groupId ?? null,
+                agentId: query.agentId ?? null,
+                userId: query.userId ?? null,
+            });
+            const prices = await listPrices(conn);
+            const first = aggregateUsage(records, groupBy, prices, undefined, query.series === true);
+            const names = await groupNames(conn, groupBy, first.rows.map((row) => row.key), facts);
+            return {
+                from: range.from,
+                to: range.to,
+                groupBy,
+                rows: first.rows.map((row) => ({
+                    ...row,
+                    name: names.get(row.key) ?? null,
+                })),
+                totals: first.totals,
+                daily: first.daily,
+                unpricedModels: first.unpricedModels,
+            };
+        },
+        async runRecords(caller, query) {
+            if (query.subjects && query.subjects.ids.length === 0)
+                return [];
+            const conn = deps.tx.read();
+            const when = (f) => [
+                ...(query.start ? [f.date('createdAt').notBefore(query.start)] : []),
+                ...(query.end ? [f.date('createdAt').before(query.end)] : []),
+            ];
+            const found = query.subjects
+                ? await inPages([...new Set(query.subjects.ids)], async (page) => await runsRepo(conn).findMany({
+                    filter: (f) => f.and([
+                        f.string('subjectKind').eq(query.subjects?.kind ?? ''),
+                        anyOf(f, 'subjectId', page),
+                        ...when(f),
+                    ]),
+                }))
+                : await runsRepo(conn).findMany(query.start || query.end ? { filter: (f) => f.and(when(f)) } : {});
+            const { runs } = await countable(conn, caller, found, query.groupId);
+            return runs
+                .map((run) => ({
+                id: run.id,
+                agentId: run.agentId,
+                subjectKind: run.subjectKind,
+                subjectId: run.subjectId,
+                status: run.status,
+                failureReason: run.failureReason,
+                retryOfRunId: run.retryOfRunId,
+                createdAt: iso(run.createdAt) ?? '',
+                dispatchedAt: iso(run.dispatchedAt),
+                startedAt: iso(run.startedAt),
+                finishedAt: iso(run.finishedAt),
+            }))
+                .sort((a, b) => a.createdAt.localeCompare(b.createdAt));
+        },
+        async runFigures(caller, range) {
+            const now = deps.clock.now();
+            const conn = deps.tx.read();
+            // Runs started in the range.
+            const started = await runsRepo(conn).findMany({
+                filter: inRange('createdAt', range),
+            });
+            const { runs } = await countable(conn, caller, started, range.groupId);
+            // Cost of the usage reported in the range.
+            const { records } = await usageRecords(conn, caller, range, {
+                groupId: range.groupId,
+            });
+            const prices = await listPrices(conn);
+            const usage = aggregateUsage(records, 'agent', prices);
+            const byType = aggregateUsage(records, 'type', prices);
+            const dailyCost = (type) => new Map(aggregateUsage(records.filter((record) => type === 'online'
+                ? record.type === 'online'
+                : record.type !== 'online'), 'day', prices).daily.map((point) => [point.day, point.cost]));
+            const names = await agentNames(conn);
+            return {
+                ...reliability(runs.map((run) => ({
+                    status: run.status,
+                    failureReason: run.failureReason,
+                    createdAt: iso(run.createdAt) ?? '',
+                    dispatchedAt: iso(run.dispatchedAt),
+                    startedAt: iso(run.startedAt),
+                    finishedAt: iso(run.finishedAt),
+                    lastActivityAt: iso(run.lastActivityAt),
+                })), now),
+                inputTokens: usage.totals.inputTokens,
+                outputTokens: usage.totals.outputTokens,
+                estimatedCost: usage.totals.cost,
+                costByAgent: usage.rows.map((row) => ({
+                    agentId: row.key,
+                    name: names.get(row.key) ?? null,
+                    cost: row.cost,
+                })),
+                usageByType: byType.rows.map((row) => ({
+                    type: row.key === 'online' ? 'online' : 'runner',
+                    runs: row.runs,
+                    inputTokens: row.inputTokens,
+                    outputTokens: row.outputTokens,
+                    cacheReadTokens: row.cacheReadTokens,
+                    cost: row.cost,
+                })),
+                activityDays: [
+                    ...new Set(runs.map((run) => (iso(run.createdAt) ?? '').slice(0, 10))),
+                ],
+                actorIds: [...new Set(runs.map((run) => run.actorUserId))],
+                daily: runDays(range, runs, dailyCost('online'), dailyCost('runner')),
+            };
+        },
+    };
+}
+//# sourceMappingURL=report.service.js.map

package/dist/server/core/reports/report.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"report.service.js","sourceRoot":"","sources":["../../../../server/core/reports/report.service.ts"],"names":[],"mappings":"AAmBA,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,eAAe,GAUhB,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAGjD,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAEhD,OAAO,EACL,QAAQ,EACR,SAAS,GAGV,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,cAAc,EAAoB,MAAM,YAAY,CAAC;AAE9D,MAAM,MAAM,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAChC,MAAM,IAAI,GAAG,GAAG,CAAC;AAEjB,wGAAwG;AACxG,MAAM,UAAU,OAAO,CACrB,KAAyC,EACzC,IAAwD,EACxD,UAA6C,EAC7C,UAA6C;IAE7C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB,CAAC;IACvC,KAAK,IAAI,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC;QAC5E,MAAM,GAAG,GAAG,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE;YACZ,GAAG;YACH,SAAS,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,IAAI,EAAE,CAAC;YACP,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI;YACvC,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI;SACxC,CAAC,CAAC;IACL,CAAC;IACD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACzE,IAAI,CAAC,GAAG;YAAE,SAAS;QACnB,MAAM,OAAO,GACX,GAAG,CAAC,MAAM,KAAK,WAAW;YAC1B,GAAG,CAAC,MAAM,KAAK,QAAQ;YACvB,GAAG,CAAC,MAAM,KAAK,WAAW;YACxB,CAAC,CAAC,GAAG,CAAC,MAAM;YACZ,CAAC,CAAC,MAAM,CAAC;QACb,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7D,CAAC;IACD,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5B,CAAC;AAwCD,MAAM,WAAW,GAAG,sBAAsB,CAAC;AAE3C,SAAS,GAAG,CAAC,KAAyB,EAAE,IAAY;IAClD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE;QAAE,OAAO,IAAI,CAAC;IACrD,IACE,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC;QACxB,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,KAAK,YAAY,CAAC,CAAC;QAC9C,IAAI,IAAI,CAAC,GAAG,KAAK,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK;QAEnE,MAAM,OAAO,CAAC,GAAG,IAAI,+BAA+B,CAAC,CAAC;IACxD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,4GAA4G;AAC5G,MAAM,UAAU,WAAW,CACzB,KAAuD,EACvD,GAAS;IAET,MAAM,EAAE,GAAG,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACjE,MAAM,GAAG,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,GAAG,MAAM,CAAC,CAAC;IAC7D,MAAM,IAAI,GACR,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC;QACvB,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,mBAAmB,GAAG,MAAM,CAAC;aACnD,WAAW,EAAE;aACb,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAClB,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,GAAG,IAAI,YAAY,CAAC,CAAC;IAC5C,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,GAAG,MAAM,CAAC;IACxD,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,eAAe;QACpC,MAAM,OAAO,CACX,0DAA0D,eAAe,QAAQ,CAClF,CAAC;IACJ,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;AAClC,CAAC;AAED,MAAM,GAAG,GAAG,CAAC,KAAc,EAAiB,EAAE,CAC5C,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,EAAE;IACnD,CAAC,CAAC,IAAI;IACN,CAAC,CAAC,IAAI,IAAI,CAAC,KAAe,CAAC,CAAC,WAAW,EAAE,CAAC;AAE9C,SAAS,KAAK,CAAC,CAAgB,EAAE,KAAa,EAAE,MAAyB;IACvE,OAAO,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,GAAsB,EACtB,IAAwD;IAExD,MAAM,MAAM,GAAQ,EAAE,CAAC;IACvB,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,IAAI;QAC1C,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACzD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,OAAO,GACX,CAAC,KAAa,EAAE,KAAkB,EAAE,EAAE,CACtC,CAAC,CAAgB,EAAc,EAAE,CAC/B,CAAC,CAAC,GAAG,CAAC;IACJ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC;IACpC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC;CAChC,CAAC,CAAC;AAEP,MAAM,UAAU,mBAAmB,CAAC,IAKnC;IACC,MAAM,IAAI,GAAG,CAAC,MAAoB,EAAE,GAAc,EAAE,EAAE,CACpD,MAAM,CAAC,OAAO;QACd,GAAG,CAAC,WAAW,KAAK,MAAM,CAAC,MAAM;QACjC,GAAG,CAAC,WAAW,KAAK,MAAM,CAAC,MAAM,CAAC;IAEpC,qHAAqH;IACrH,KAAK,UAAU,OAAO,CACpB,IAAwB,EACxB,MAAoB,EACpB,IAA0B;QAE1B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;QAC9C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,IAAI,GAAG,EAAU,CAAC;YAC7D,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACvB,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QACnC,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAwB,CAAC;QAC9C,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,MAAM,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC;YACjD,MAAM,KAAK,GAAG,OAAO;gBACnB,CAAC,CAAC,MAAM,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC;gBACvD,CAAC,CAAC,IAAI,GAAG,EAAwB,CAAC;YACpC,KAAK,MAAM,EAAE,IAAI,GAAG;gBAClB,KAAK,CAAC,GAAG,CACP,GAAG,IAAI,IAAI,EAAE,EAAE,EACf,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;oBACX,CAAC,OAAO;wBACN,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE;wBAC5C,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CACjD,CAAC;QACN,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,sGAAsG;IACtG,KAAK,UAAU,SAAS,CACtB,IAAwB,EACxB,MAAoB,EACpB,IAA0B,EAC1B,OAAsB;QAEtB,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;QACrD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE;YAC/B,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;YAC9D,OAAO,CACL,IAAI,KAAK,SAAS;gBAClB,IAAI,CAAC,OAAO;gBACZ,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,KAAK,EAAE,EAAE,KAAK,OAAO,CAAC,CACzC,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;IAC/B,CAAC;IAED,KAAK,UAAU,YAAY,CACzB,IAAwB,EACxB,MAAoB,EACpB,KAAkB,EAClB,MAIC;QAED,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;YAC3C,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC;SACpC,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3D,MAAM,IAAI,GAAG,MAAM,OAAO,CACxB,MAAM,EACN,KAAK,EAAE,IAAI,EAAE,EAAE,CACb,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CACzE,CAAC;QACF,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,SAAS,CAC3C,IAAI,EACJ,MAAM,EACN,IAAI,CAAC,MAAM,CACT,CAAC,GAAG,EAAE,EAAE,CACN,CAAC,CAAC,MAAM,CAAC,OAAO,IAAI,GAAG,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC;YACnD,CAAC,CAAC,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,WAAW,KAAK,MAAM,CAAC,MAAM,CAAC,CACxD,EACD,MAAM,CAAC,OAAO,CACf,CAAC;QACF,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QACvD,MAAM,OAAO,GAAkB,EAAE,CAAC;QAClC,KAAK,MAAM,GAAG,IAAI,KAA4B,EAAE,CAAC;YAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAChC,IAAI,CAAC,GAAG;gBAAE,SAAS;YACnB,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACnC,MAAM,QAAQ,GAAG,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACrC,OAAO,CAAC,IAAI,CAAC;gBACX,KAAK,EAAE,GAAG,CAAC,EAAE;gBACb,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,IAAI,EAAE,GAAG,CAAC,SAAS;gBACnB,MAAM,EAAE,GAAG,CAAC,WAAW;gBACvB,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,YAAY,EAAE,GAAG,CAAC,YAAY,IAAI,IAAI;gBACtC,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,IAAI;gBACxB,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,OAAO,EACL,KAAK,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,IAAI,IAAI;gBACrE,GAAG,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;gBAC5C,UAAU,EACR,OAAO,IAAI,QAAQ;oBACjB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACzD,CAAC,CAAC,CAAC;gBACP,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;gBACzC,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,CAAC;gBAC3C,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC;gBACjD,gBAAgB,EAAE,MAAM,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,CAAC;gBACnD,eAAe,EAAE,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC;aAClD,CAAC,CAAC;QACL,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,UAAU,UAAU,CACvB,IAAwB;QAExB,OAAO,IAAI,GAAG,CACZ,CAAC,MAAM,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CACtE,CAAC;IACJ,CAAC;IAED,KAAK,UAAU,UAAU,CACvB,IAAwB,EACxB,OAAqB,EACrB,IAAuB,EACvB,KAAwC;QAExC,QAAQ,OAAO,EAAE,CAAC;YAChB,KAAK,OAAO;gBACV,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC;YAC1B,KAAK,QAAQ;gBACX,OAAO,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;YACtD,KAAK,OAAO,CAAC,CAAC,CAAC;gBACb,MAAM,KAAK,GAAG,IAAI,GAAG,EAAkB,CAAC;gBACxC,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,MAAM,EAAE;oBAC/B,IAAI,IAAI,CAAC,KAAK;wBAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC5D,OAAO,KAAK,CAAC;YACf,CAAC;YACD,KAAK,SAAS;gBACZ,OAAO,IAAI,GAAG,CACZ,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,KAAK,IAAI,GAAG,CAAU,CAAC,CAChE,CAAC;YACJ,KAAK,MAAM;gBACT,8FAA8F;gBAC9F,OAAO,IAAI,GAAG,EAAE,CAAC;YACnB;gBACE,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAU,CAAC,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED,OAAO;QACL,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;QAEtD,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK;YAC5B,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YACnD,IAAI,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;YACzE,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAC5B,OAAO;gBACL,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,IAAI,EAAE,MAAM,cAAc,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;aAChE,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK;YACvB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC;YACzC,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC;gBACpC,MAAM,OAAO,CAAC,0BAA0B,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzE,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;gBACjE,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;gBAC9B,OAAO,EAAE,KAAK,CAAC,OAAO,IAAI,IAAI;gBAC9B,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,IAAI;aAC7B,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,cAAc,CAC1B,OAAO,EACP,OAAO,EACP,MAAM,EACN,SAAS,EACT,KAAK,CAAC,MAAM,KAAK,IAAI,CACtB,CAAC;YACF,MAAM,KAAK,GAAG,MAAM,UAAU,CAC5B,IAAI,EACJ,OAAO,EACP,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,EAChC,KAAK,CACN,CAAC;YACF,OAAO;gBACL,IAAI,EAAE,KAAK,CAAC,IAAI;gBAChB,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,OAAO;gBACP,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC7B,GAAG,GAAG;oBACN,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI;iBACjC,CAAC,CAAC;gBACH,MAAM,EAAE,KAAK,CAAC,MAAM;gBACpB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,cAAc,EAAE,KAAK,CAAC,cAAc;aACrC,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK;YAC5B,IAAI,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,EAAE,CAAC;YACjE,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,CAAC,CAAgB,EAAgB,EAAE,CAAC;gBAC/C,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;aAC9D,CAAC;YACF,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ;gBAC1B,CAAC,CAAC,MAAM,OAAO,CACX,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAChC,KAAK,EAAE,IAAI,EAAE,EAAE,CACb,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;oBAC5B,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CACZ,CAAC,CAAC,GAAG,CAAC;wBACJ,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,IAAI,EAAE,CAAC;wBACtD,KAAK,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC;wBAC3B,GAAG,IAAI,CAAC,CAAC,CAAC;qBACX,CAAC;iBACL,CAAC,CACL;gBACH,CAAC,CAAC,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,CAC3B,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAClE,CAAC;YACN,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACrE,OAAO,IAAI;iBACR,GAAG,CAAC,CAAC,GAAG,EAAa,EAAE,CAAC,CAAC;gBACxB,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,WAAW,EAAE,GAAG,CAAC,WAAW;gBAC5B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,aAAa,EAAE,GAAG,CAAC,aAAa;gBAChC,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,SAAS,EAAE,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;gBACnC,YAAY,EAAE,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC;gBACnC,SAAS,EAAE,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC;gBAC7B,UAAU,EAAE,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;aAChC,CAAC,CAAC;iBACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,KAAK;YAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YAC5B,6BAA6B;YAC7B,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC;gBAC5C,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC;aACpC,CAAC,CAAC;YACH,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACvE,2CAA2C;YAC3C,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE;gBAC1D,OAAO,EAAE,KAAK,CAAC,OAAO;aACvB,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACvD,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACvD,MAAM,SAAS,GAAG,CAAC,IAAyB,EAAE,EAAE,CAC9C,IAAI,GAAG,CACL,cAAc,CACZ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CACxB,IAAI,KAAK,QAAQ;gBACf,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ;gBAC1B,CAAC,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAC7B,EACD,KAAK,EACL,MAAM,CACP,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAU,CAAC,CACzD,CAAC;YACJ,MAAM,KAAK,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,CAAC;YACrC,OAAO;gBACL,GAAG,WAAW,CACZ,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACjB,MAAM,EAAE,GAAG,CAAC,MAAM;oBAClB,aAAa,EAAE,GAAG,CAAC,aAAa;oBAChC,SAAS,EAAE,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE;oBACnC,YAAY,EAAE,GAAG,CAAC,GAAG,CAAC,YAAY,CAAC;oBACnC,SAAS,EAAE,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC;oBAC7B,UAAU,EAAE,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC;oBAC/B,cAAc,EAAE,GAAG,CAAC,GAAG,CAAC,cAAc,CAAC;iBACxC,CAAC,CAAC,EACH,GAAG,CACJ;gBACD,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,WAAW;gBACrC,YAAY,EAAE,KAAK,CAAC,MAAM,CAAC,YAAY;gBACvC,aAAa,EAAE,KAAK,CAAC,MAAM,CAAC,IAAI;gBAChC,WAAW,EAAE,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACpC,OAAO,EAAE,GAAG,CAAC,GAAG;oBAChB,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI;oBAChC,IAAI,EAAE,GAAG,CAAC,IAAI;iBACf,CAAC,CAAC;gBACH,WAAW,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;oBACrC,IAAI,EAAE,GAAG,CAAC,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ;oBAChD,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,WAAW,EAAE,GAAG,CAAC,WAAW;oBAC5B,YAAY,EAAE,GAAG,CAAC,YAAY;oBAC9B,eAAe,EAAE,GAAG,CAAC,eAAe;oBACpC,IAAI,EAAE,GAAG,CAAC,IAAI;iBACf,CAAC,CAAC;gBACH,YAAY,EAAE;oBACZ,GAAG,IAAI,GAAG,CACR,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAC3D;iBACF;gBACD,QAAQ,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;gBAC1D,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,QAAQ,CAAC,EAAE,SAAS,CAAC,QAAQ,CAAC,CAAC;aACtE,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/server/core/reports/usage.d.ts

@@ -0,0 +1,50 @@
+/**
+ * Grouping usage records into report rows, with cost worked out from the prices as they are now. Pure: the service
+ * reads the records and names, this adds them up.
+ *
+ * A record is one model's tokens in one run. A run counts once in a group however many records it has; its working
+ * time (start to finish) counts once too. A record without a matching price adds no cost: a group with no priced record
+ * has `cost: null`, and `pricedRuns` counts the runs whose every record was priced.
+ */
+import { type Costs, type PriceBook, type UsageGroupBy, type UsagePoint, type UsageRow } from '../../../shared/reports.js';
+export interface UsageRecord {
+    readonly runId: string;
+    readonly agentId: string;
+    /** The agent's type, `online` or `runner`. */
+    readonly type: string;
+    /** Who woke the agent. */
+    readonly userId: string;
+    readonly tool: string;
+    /** The model service of an online model call; null for a coding tool's usage. */
+    readonly modelService: string | null;
+    readonly model: string | null;
+    readonly subjectKind: string;
+    readonly subjectId: string;
+    /** The group of the run's subject, when its domain names one. */
+    readonly groupId: string | null;
+    /** `YYYY-MM-DD` (UTC) the usage was reported. */
+    readonly day: string;
+    /** The run's start to finish, or 0 when it never started. */
+    readonly durationMs: number;
+    readonly inputTokens: number;
+    readonly outputTokens: number;
+    readonly cacheReadTokens: number;
+    readonly cacheWriteTokens: number;
+    readonly reasoningTokens: number;
+}
+/** Amounts rounded to a millionth, or null when nothing was priced. */
+export declare function costs(amounts: ReadonlyMap<string, number>): Costs | null;
+/** The group a record falls in; `''` for a record without a group (or model). */
+export declare function keyOf(record: UsageRecord, groupBy: UsageGroupBy): string;
+export interface AggregatedUsage {
+    readonly rows: UsageRow[];
+    readonly totals: UsageRow;
+    readonly daily: UsagePoint[];
+    readonly unpricedModels: string[];
+}
+/**
+ * Groups `records`; `names` gives a group key its display name. Rows come by cost (unpriced last) then tokens; by
+ * day, in date order. With `series`, each day is also broken down by group, in the rows' order.
+ */
+export declare function aggregateUsage(records: readonly UsageRecord[], groupBy: UsageGroupBy, book: PriceBook, names?: ReadonlyMap<string, string>, series?: boolean): AggregatedUsage;
+//# sourceMappingURL=usage.d.ts.map

package/dist/server/core/reports/usage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../../../server/core/reports/usage.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAGL,KAAK,KAAK,EAEV,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,KAAK,QAAQ,EAEd,MAAM,4BAA4B,CAAC;AAGpC,MAAM,WAAW,WAAW;IAC1B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,8CAA8C;IAC9C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,0BAA0B;IAC1B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,iFAAiF;IACjF,QAAQ,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAC;IACrC,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,iEAAiE;IACjE,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,iDAAiD;IACjD,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,6DAA6D;IAC7D,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;IACjC,QAAQ,CAAC,gBAAgB,EAAE,MAAM,CAAC;IAClC,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;CAClC;AA6CD,uEAAuE;AACvE,wBAAgB,KAAK,CAAC,OAAO,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,KAAK,GAAG,IAAI,CAQxE;AAqBD,iFAAiF;AACjF,wBAAgB,KAAK,CAAC,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,YAAY,GAAG,MAAM,CAmBxE;AAYD,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC1B,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC;IAC1B,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;IAC7B,QAAQ,CAAC,cAAc,EAAE,MAAM,EAAE,CAAC;CACnC;AAUD;;;GAGG;AACH,wBAAgB,cAAc,CAC5B,OAAO,EAAE,SAAS,WAAW,EAAE,EAC/B,OAAO,EAAE,YAAY,EACrB,IAAI,EAAE,SAAS,EACf,KAAK,GAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAa,EAC9C,MAAM,UAAQ,GACb,eAAe,CA0DjB"}

package/dist/server/core/reports/usage.js

@@ -0,0 +1,159 @@
+/**
+ * Grouping usage records into report rows, with cost worked out from the prices as they are now. Pure: the service
+ * reads the records and names, this adds them up.
+ *
+ * A record is one model's tokens in one run. A run counts once in a group however many records it has; its working
+ * time (start to finish) counts once too. A record without a matching price adds no cost: a group with no priced record
+ * has `cost: null`, and `pricedRuns` counts the runs whose every record was priced.
+ */
+import { priceFor, sourceLabel, } from '../../../shared/reports.js';
+import { costOf } from './prices.js';
+function accumulator() {
+    return {
+        runs: new Map(),
+        unpriced: new Set(),
+        inputTokens: 0,
+        outputTokens: 0,
+        cacheReadTokens: 0,
+        cacheWriteTokens: 0,
+        reasoningTokens: 0,
+        cost: new Map(),
+    };
+}
+function add(acc, record, price) {
+    acc.runs.set(record.runId, record.durationMs);
+    acc.inputTokens += record.inputTokens;
+    acc.outputTokens += record.outputTokens;
+    acc.cacheReadTokens += record.cacheReadTokens;
+    acc.cacheWriteTokens += record.cacheWriteTokens;
+    acc.reasoningTokens += record.reasoningTokens;
+    if (price)
+        acc.cost.set(price.currency, (acc.cost.get(price.currency) ?? 0) + costOf(price, record));
+    else
+        acc.unpriced.add(record.runId);
+}
+/** Amounts rounded to a millionth, or null when nothing was priced. */
+export function costs(amounts) {
+    if (amounts.size === 0)
+        return null;
+    return Object.fromEntries([...amounts].map(([currency, value]) => [
+        currency,
+        Math.round(value * 1e6) / 1e6,
+    ]));
+}
+function toRow(key, name, acc) {
+    let durationMs = 0;
+    for (const duration of acc.runs.values())
+        durationMs += duration;
+    return {
+        key,
+        name,
+        runs: acc.runs.size,
+        durationMs,
+        inputTokens: acc.inputTokens,
+        outputTokens: acc.outputTokens,
+        cacheReadTokens: acc.cacheReadTokens,
+        cacheWriteTokens: acc.cacheWriteTokens,
+        reasoningTokens: acc.reasoningTokens,
+        cost: costs(acc.cost),
+        pricedRuns: [...acc.runs.keys()].filter((id) => !acc.unpriced.has(id))
+            .length,
+    };
+}
+/** The group a record falls in; `''` for a record without a group (or model). */
+export function keyOf(record, groupBy) {
+    switch (groupBy) {
+        case 'agent':
+            return record.agentId;
+        case 'person':
+            return record.userId;
+        case 'group':
+            return record.groupId ?? '';
+        case 'subject':
+            return `${record.subjectKind}:${record.subjectId}`;
+        case 'day':
+            return record.day;
+        case 'model':
+            return record.model ?? '';
+        case 'tool':
+            return record.tool;
+        case 'type':
+            return record.type;
+    }
+}
+/** The first cost of a row, for ordering (rows of one currency compare as expected). */
+const firstCost = (row) => row.cost ? (Object.values(row.cost)[0] ?? 0) : -1;
+const tokensOf = (row) => row.inputTokens +
+    row.outputTokens +
+    row.cacheReadTokens +
+    row.cacheWriteTokens;
+const tokensAndCost = (acc) => ({
+    inputTokens: acc.inputTokens,
+    outputTokens: acc.outputTokens,
+    cacheReadTokens: acc.cacheReadTokens,
+    cacheWriteTokens: acc.cacheWriteTokens,
+    cost: costs(acc.cost),
+});
+/**
+ * Groups `records`; `names` gives a group key its display name. Rows come by cost (unpriced last) then tokens; by
+ * day, in date order. With `series`, each day is also broken down by group, in the rows' order.
+ */
+export function aggregateUsage(records, groupBy, book, names = new Map(), series = false) {
+    const groups = new Map();
+    const days = new Map();
+    const dayGroups = new Map();
+    const total = accumulator();
+    const unpricedModels = new Set();
+    const priceOf = new Map();
+    for (const record of records) {
+        const source = sourceLabel(record);
+        if (!priceOf.has(source))
+            priceOf.set(source, priceFor(book, record));
+        const price = priceOf.get(source) ?? null;
+        if (!price)
+            unpricedModels.add(record.model ? source : '');
+        const key = keyOf(record, groupBy);
+        const group = groups.get(key) ?? accumulator();
+        groups.set(key, group);
+        add(group, record, price);
+        const day = days.get(record.day) ?? accumulator();
+        days.set(record.day, day);
+        add(day, record, price);
+        if (series) {
+            const byKey = dayGroups.get(record.day) ?? new Map();
+            dayGroups.set(record.day, byKey);
+            const cell = byKey.get(key) ?? accumulator();
+            byKey.set(key, cell);
+            add(cell, record, price);
+        }
+        add(total, record, price);
+    }
+    const rows = [...groups].map(([key, acc]) => toRow(key, names.get(key) ?? null, acc));
+    rows.sort((a, b) => groupBy === 'day'
+        ? a.key.localeCompare(b.key)
+        : firstCost(b) - firstCost(a) || tokensOf(b) - tokensOf(a));
+    const daily = [...days]
+        .sort(([a], [b]) => a.localeCompare(b))
+        .map(([day, acc]) => {
+        const point = { day, ...tokensAndCost(acc) };
+        if (!series)
+            return point;
+        const byKey = dayGroups.get(day) ?? new Map();
+        return {
+            ...point,
+            series: rows
+                .filter((row) => byKey.has(row.key))
+                .map((row) => ({
+                key: row.key,
+                ...tokensAndCost(byKey.get(row.key) ?? accumulator()),
+            })),
+        };
+    });
+    return {
+        rows,
+        totals: toRow('total', null, total),
+        daily,
+        unpricedModels: [...unpricedModels].sort(),
+    };
+}
+//# sourceMappingURL=usage.js.map

package/dist/server/core/reports/usage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../../../server/core/reports/usage.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EACL,QAAQ,EACR,WAAW,GAQZ,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAuCrC,SAAS,WAAW;IAClB,OAAO;QACL,IAAI,EAAE,IAAI,GAAG,EAAE;QACf,QAAQ,EAAE,IAAI,GAAG,EAAE;QACnB,WAAW,EAAE,CAAC;QACd,YAAY,EAAE,CAAC;QACf,eAAe,EAAE,CAAC;QAClB,gBAAgB,EAAE,CAAC;QACnB,eAAe,EAAE,CAAC;QAClB,IAAI,EAAE,IAAI,GAAG,EAAE;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,GAAG,CACV,GAAgB,EAChB,MAAmB,EACnB,KAAwB;IAExB,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;IAC9C,GAAG,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;IACtC,GAAG,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC;IACxC,GAAG,CAAC,eAAe,IAAI,MAAM,CAAC,eAAe,CAAC;IAC9C,GAAG,CAAC,gBAAgB,IAAI,MAAM,CAAC,gBAAgB,CAAC;IAChD,GAAG,CAAC,eAAe,IAAI,MAAM,CAAC,eAAe,CAAC;IAC9C,IAAI,KAAK;QACP,GAAG,CAAC,IAAI,CAAC,GAAG,CACV,KAAK,CAAC,QAAQ,EACd,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAC5D,CAAC;;QACC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACtC,CAAC;AAED,uEAAuE;AACvE,MAAM,UAAU,KAAK,CAAC,OAAoC;IACxD,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACpC,OAAO,MAAM,CAAC,WAAW,CACvB,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC;QACtC,QAAQ;QACR,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,GAAG,GAAG;KAC9B,CAAC,CACH,CAAC;AACJ,CAAC;AAED,SAAS,KAAK,CAAC,GAAW,EAAE,IAAmB,EAAE,GAAgB;IAC/D,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,KAAK,MAAM,QAAQ,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE;QAAE,UAAU,IAAI,QAAQ,CAAC;IACjE,OAAO;QACL,GAAG;QACH,IAAI;QACJ,IAAI,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI;QACnB,UAAU;QACV,WAAW,EAAE,GAAG,CAAC,WAAW;QAC5B,YAAY,EAAE,GAAG,CAAC,YAAY;QAC9B,eAAe,EAAE,GAAG,CAAC,eAAe;QACpC,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;QACtC,eAAe,EAAE,GAAG,CAAC,eAAe;QACpC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;QACrB,UAAU,EAAE,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aACnE,MAAM;KACV,CAAC;AACJ,CAAC;AAED,iFAAiF;AACjF,MAAM,UAAU,KAAK,CAAC,MAAmB,EAAE,OAAqB;IAC9D,QAAQ,OAAO,EAAE,CAAC;QAChB,KAAK,OAAO;YACV,OAAO,MAAM,CAAC,OAAO,CAAC;QACxB,KAAK,QAAQ;YACX,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,KAAK,OAAO;YACV,OAAO,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QAC9B,KAAK,SAAS;YACZ,OAAO,GAAG,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;QACrD,KAAK,KAAK;YACR,OAAO,MAAM,CAAC,GAAG,CAAC;QACpB,KAAK,OAAO;YACV,OAAO,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;QAC5B,KAAK,MAAM;YACT,OAAO,MAAM,CAAC,IAAI,CAAC;QACrB,KAAK,MAAM;YACT,OAAO,MAAM,CAAC,IAAI,CAAC;IACvB,CAAC;AACH,CAAC;AAED,wFAAwF;AACxF,MAAM,SAAS,GAAG,CAAC,GAAa,EAAU,EAAE,CAC1C,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAEpD,MAAM,QAAQ,GAAG,CAAC,GAAa,EAAU,EAAE,CACzC,GAAG,CAAC,WAAW;IACf,GAAG,CAAC,YAAY;IAChB,GAAG,CAAC,eAAe;IACnB,GAAG,CAAC,gBAAgB,CAAC;AASvB,MAAM,aAAa,GAAG,CAAC,GAAgB,EAAiC,EAAE,CAAC,CAAC;IAC1E,WAAW,EAAE,GAAG,CAAC,WAAW;IAC5B,YAAY,EAAE,GAAG,CAAC,YAAY;IAC9B,eAAe,EAAE,GAAG,CAAC,eAAe;IACpC,gBAAgB,EAAE,GAAG,CAAC,gBAAgB;IACtC,IAAI,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;CACtB,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,UAAU,cAAc,CAC5B,OAA+B,EAC/B,OAAqB,EACrB,IAAe,EACf,QAAqC,IAAI,GAAG,EAAE,EAC9C,MAAM,GAAG,KAAK;IAEd,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC9C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC5C,MAAM,SAAS,GAAG,IAAI,GAAG,EAAoC,CAAC;IAC9D,MAAM,KAAK,GAAG,WAAW,EAAE,CAAC;IAC5B,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;IACzC,MAAM,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;IACrD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;QACtE,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC;QAC1C,IAAI,CAAC,KAAK;YAAE,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,WAAW,EAAE,CAAC;QAC/C,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QACvB,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,WAAW,EAAE,CAAC;QAClD,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC1B,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QACxB,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,EAAuB,CAAC;YAC1E,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACjC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,WAAW,EAAE,CAAC;YAC7C,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACrB,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QAC3B,CAAC;QACD,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IACD,MAAM,IAAI,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,CAC1C,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,GAAG,CAAC,CACxC,CAAC;IACF,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACjB,OAAO,KAAK,KAAK;QACf,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC;QAC5B,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAC7D,CAAC;IACF,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC;SACpB,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;SACtC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,EAAc,EAAE;QAC9B,MAAM,KAAK,GAAG,EAAE,GAAG,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC;QAC7C,IAAI,CAAC,MAAM;YAAE,OAAO,KAAK,CAAC;QAC1B,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,IAAI,GAAG,EAAuB,CAAC;QACnE,OAAO;YACL,GAAG,KAAK;YACR,MAAM,EAAE,IAAI;iBACT,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;iBACnC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACb,GAAG,EAAE,GAAG,CAAC,GAAG;gBACZ,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,WAAW,EAAE,CAAC;aACtD,CAAC,CAAC;SACN,CAAC;IACJ,CAAC,CAAC,CAAC;IACL,OAAO;QACL,IAAI;QACJ,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC;QACnC,KAAK;QACL,cAAc,EAAE,CAAC,GAAG,cAAc,CAAC,CAAC,IAAI,EAAE;KAC3C,CAAC;AACJ,CAAC"}

package/dist/server/core/runs/availability.d.ts

@@ -0,0 +1,23 @@
+import type { RunnerService } from '../../runners/index.js';
+import type { DatabaseConnection } from '@nocobase/db';
+import type { ClaimEligibility } from './eligibility.js';
+import type { Agent } from '../../../shared/agents.js';
+import type { ModelGateway } from '../../online/gateway.js';
+/** Whether an agent could start work now, and how busy it is. */
+export interface AgentAvailability {
+    readonly online: boolean;
+    /** Its runs not finished (queued or held). */
+    readonly busy: number;
+}
+/** Per agent, read on `conn`; for a person's work with `actorUserId`. */
+export type Availability = (conn: DatabaseConnection, agents: readonly Agent[], actorUserId?: string) => Promise<Map<string, AgentAvailability>>;
+export declare function createAvailability(deps: {
+    readonly runners: Pick<RunnerService, 'online'>;
+    /** Whether a runner would take a person's work. */
+    readonly eligibility: Pick<ClaimEligibility, 'canClaim'>;
+    /** The models online agents use: an online agent is online while its service offers its model. */
+    readonly models: Pick<ModelGateway, 'catalog'>;
+    /** Per agent, the runs it has not finished. */
+    readonly openRuns: (conn: DatabaseConnection, agentIds: readonly string[]) => Promise<ReadonlyMap<string, number>>;
+}): Availability;
+//# sourceMappingURL=availability.d.ts.map

package/dist/server/core/runs/availability.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"availability.d.ts","sourceRoot":"","sources":["../../../../server/core/runs/availability.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAC5D,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAGvD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AAEzD,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,2BAA2B,CAAC;AAEvD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAE5D,iEAAiE;AACjE,MAAM,WAAW,iBAAiB;IAChC,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC;IACzB,8CAA8C;IAC9C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;CACvB;AAED,yEAAyE;AACzE,MAAM,MAAM,YAAY,GAAG,CACzB,IAAI,EAAE,kBAAkB,EACxB,MAAM,EAAE,SAAS,KAAK,EAAE,EACxB,WAAW,CAAC,EAAE,MAAM,KACjB,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC,CAAC;AAE7C,wBAAgB,kBAAkB,CAAC,IAAI,EAAE;IACvC,QAAQ,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;IAChD,mDAAmD;IACnD,QAAQ,CAAC,WAAW,EAAE,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;IACzD,kGAAkG;IAClG,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAC/C,+CAA+C;IAC/C,QAAQ,CAAC,QAAQ,EAAE,CACjB,IAAI,EAAE,kBAAkB,EACxB,QAAQ,EAAE,SAAS,MAAM,EAAE,KACxB,OAAO,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;CAC3C,GAAG,YAAY,CA4Cf"}

package/dist/server/core/runs/availability.js

@@ -0,0 +1,42 @@
+/**
+ * Whether each agent could start work now, and how busy it is: a runner agent while an online runner of the runners
+ * plugin has one of its tools enabled, installed and signed in, is among the runners it may use, and its owner's local
+ * policy lets it take the agent's runs; an online agent while the service of its first entry offers its model (the
+ * system default chat model's, for one that lists none). Asked for a person (`actorUserId`), a runner agent counts only
+ * the runners that would take that person's work by the claim's rules (`eligibility.ts`): trust, features and the
+ * agent's variables included. Ask outside a transaction then.
+ */
+import { policyAllowsAgent } from '@nocobase/agent-protocol';
+import { hasTool, onlineEntryOf } from './claim.js';
+import { offersEntry } from '../../../shared/models.js';
+export function createAvailability(deps) {
+    return async (conn, agents, actorUserId) => {
+        const runners = await deps.runners.online(conn);
+        const claimable = new Map();
+        if (actorUserId !== undefined)
+            for (const agent of agents)
+                if (agent.type === 'runner')
+                    claimable.set(agent.id, await deps.eligibility.canClaim(conn, agent, {
+                        actorUserId,
+                        runners,
+                    }));
+        const busy = await deps.openRuns(conn, agents.map((agent) => agent.id));
+        const catalog = agents.some((agent) => agent.type === 'online')
+            ? await deps.models.catalog()
+            : { services: [], defaultModel: null };
+        return new Map(agents.map((agent) => [
+            agent.id,
+            {
+                online: agent.type === 'online'
+                    ? offersEntry(catalog, onlineEntryOf(agent, undefined, catalog.defaultModel ?? null))
+                    : (claimable.get(agent.id) ??
+                        runners.some((runner) => hasTool(runner, agent) &&
+                            policyAllowsAgent(runner.policy, agent) &&
+                            (agent.runnerIds.length === 0 ||
+                                agent.runnerIds.includes(runner.id)))),
+                busy: busy.get(agent.id) ?? 0,
+            },
+        ]));
+    };
+}
+//# sourceMappingURL=availability.js.map

package/dist/server/core/runs/availability.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"availability.js","sourceRoot":"","sources":["../../../../server/core/runs/availability.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AACH,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAI7D,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAIpD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAiBxD,MAAM,UAAU,kBAAkB,CAAC,IAWlC;IACC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE;QACzC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAmB,CAAC;QAC7C,IAAI,WAAW,KAAK,SAAS;YAC3B,KAAK,MAAM,KAAK,IAAI,MAAM;gBACxB,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ;oBACzB,SAAS,CAAC,GAAG,CACX,KAAK,CAAC,EAAE,EACR,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE;wBAC3C,WAAW;wBACX,OAAO;qBACR,CAAC,CACH,CAAC;QACR,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAC9B,IAAI,EACJ,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAChC,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC;YAC7D,CAAC,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE;YAC7B,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC;QACzC,OAAO,IAAI,GAAG,CACZ,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;YACpB,KAAK,CAAC,EAAE;YACR;gBACE,MAAM,EACJ,KAAK,CAAC,IAAI,KAAK,QAAQ;oBACrB,CAAC,CAAC,WAAW,CACT,OAAO,EACP,aAAa,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,YAAY,IAAI,IAAI,CAAC,CAC9D;oBACH,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;wBACxB,OAAO,CAAC,IAAI,CACV,CAAC,MAAM,EAAE,EAAE,CACT,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC;4BACtB,iBAAiB,CAAC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC;4BACvC,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC;gCAC3B,KAAK,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CACzC,CAAC;gBACR,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC;aAC9B;SACF,CAAC,CACH,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}