@yuaone/core 0.9.7 → 0.9.9

package/dist/autonomous/task-memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-memory.js","sourceRoot":"","sources":["../../src/autonomous/task-memory.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,WAAW,EAAY,UAAU,EAAE,MAAM,SAAS,CAAC;AAChH,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAkC3C,gFAAgF;AAEhF,MAAM,OAAO,UAAW,SAAQ,YAAY;IACzB,QAAQ,CAAS;IAElC,YAAY,QAAiB;QAC3B,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC9D,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,kEAAkE;IAClE,MAAM,CAAC,MAAc,EAAE,IAAY;QACjC,MAAM,KAAK,GAAc;YACvB,MAAM;YACN,IAAI;YACJ,YAAY,EAAE,UAAU;YACxB,YAAY,EAAE,EAAE;YAChB,eAAe,EAAE,EAAE;YACnB,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,kDAAkD;IAClD,IAAI,CAAC,MAAc;QACjB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;YACzD,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAc,CAAC;QACtC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,iCAAiC;IACjC,QAAQ,CAAC,MAAc,EAAE,KAAgB;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QACxB,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC;QAC3B,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oCAAoC;IACpC,MAAM,CAAC,MAAc,EAAE,MAA8B,EAAE,OAAgB;QACrE,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QACxB,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;QACtB,KAAK,CAAC,OAAO,GAAG,OAAO,CAAC;QACxB,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,oDAAoD;IACpD,cAAc,CAAC,MAAc,EAAE,KAAoB;QACjD,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QACxB,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,yDAAyD;QACzD,IAAI,KAAK,CAAC,eAAe,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YACvC,KAAK,CAAC,eAAe,GAAG,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjD,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjB,OAAO,KAAK,CAAC;IACf,CAAC;IAED,sDAAsD;IACtD,IAAI;QACF,IAAI,CAAC;YACH,OAAO,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC;iBAC9B,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;iBAChC,GAAG,CAAC,CAAC,CAAC,EAAE;gBACP,IAAI,CAAC;oBACH,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAc,CAAC;gBAChF,CAAC;gBAAC,MAAM,CAAC;oBAAC,OAAO,IAAI,CAAC;gBAAC,CAAC;YAC1B,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,CAAC,EAAkB,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC;iBACzC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;QAC5D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED,0BAA0B;IAC1B,MAAM,CAAC,MAAc;QACnB,IAAI,CAAC;YACH,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;YAClC,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,8EAA8E;IAEtE,QAAQ,CAAC,MAAc;QAC7B,4CAA4C;QAC5C,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACjE,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,OAAO,CAAC,CAAC;IAC7C,CAAC;IAEO,IAAI,CAAC,KAAgB;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG,IAAI,GAAG,MAAM,CAAC;QAC1B,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAC5D,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACxB,CAAC;IAEO,UAAU,CAAC,KAAgB;QACjC,MAAM,KAAK,GAAe;YACxB,IAAI,EAAE,0BAA0B;YAChC,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,KAAK,EAAE,KAAK,CAAC,YAAY;YACzB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;CACF"}

package/dist/budget-governor-v2.d.ts

@@ -0,0 +1,93 @@
+/**
+ * @module budget-governor-v2
+ * @description Multi-dimensional Budget Governor v2.
+ *
+ * Tracks and enforces budgets across six dimensions simultaneously:
+ *   task, project, daily, branch, research, model_tier
+ *
+ * Features:
+ * - Graceful degradation: suggests LOCAL model tier when usage >= degradeAtPercent
+ * - Hard halt: emits budget_exhausted event when usage >= haltAtPercent
+ * - Atomic JSON storage to ~/.yuan/budget/budget-state.json
+ * - Daily budget auto-resets at UTC midnight
+ * - Never throws — all checks are non-blocking
+ */
+import { EventEmitter } from "node:events";
+export type BudgetType = "task" | "project" | "daily" | "branch" | "research" | "model_tier";
+export interface BudgetAllocationV2 {
+    type: BudgetType;
+    id: string;
+    limit: number;
+    used: number;
+    remaining: number;
+    exhaustedAt?: string;
+    degradedAt?: string;
+}
+export interface BudgetPolicy {
+    taskBudget: number;
+    projectBudget: number;
+    dailyBudget: number;
+    branchBudget: number;
+    researchBudget: number;
+    degradeAtPercent: number;
+    haltAtPercent: number;
+}
+export interface BudgetStatusV2 {
+    taskId?: string;
+    projectId?: string;
+    allocations: BudgetAllocationV2[];
+    isDegraded: boolean;
+    isHalted: boolean;
+    recommendation: "continue" | "degrade" | "halt";
+    suggestedModelTier?: "LOCAL" | "STANDARD" | "PREMIUM";
+}
+export declare class BudgetGovernorV2 extends EventEmitter {
+    private readonly policy;
+    private readonly storageDir;
+    private readonly storagePath;
+    /** All tracked allocations keyed by `${type}:${id}` */
+    private allocations;
+    /** Active task entries (taskId → metadata) */
+    private activeTasks;
+    /** Last known daily date for midnight detection */
+    private currentDailyDate;
+    constructor(policy?: Partial<BudgetPolicy>, storageDir?: string);
+    /**
+     * Start tracking a task.
+     * Creates allocation entries for task / project (if given) / daily / branch (if given).
+     */
+    startTask(taskId: string, projectId?: string, branchName?: string): void;
+    /**
+     * Record token usage for a specific budget dimension.
+     * Triggers degradation/exhaustion events as thresholds are crossed.
+     */
+    recordUsage(tokens: number, budgetType: BudgetType, id: string): void;
+    /**
+     * Check if the task is allowed to continue.
+     * Returns status with recommendation ("continue" | "degrade" | "halt").
+     */
+    check(taskId: string): BudgetStatusV2;
+    /**
+     * End task and finalize its budget tracking.
+     */
+    endTask(taskId: string): void;
+    /**
+     * Reset the daily budget allocation.
+     * Call this at day boundary or let it auto-detect via internal checks.
+     */
+    resetDaily(): void;
+    /**
+     * Get all current allocations.
+     */
+    getAllocations(): BudgetAllocationV2[];
+    private _key;
+    private _makeAllocation;
+    private _ensureAllocation;
+    private _defaultLimit;
+    /** Auto-detect UTC day rollover and reset daily budget. */
+    private _checkDailyReset;
+    private _saveState;
+    private _loadState;
+    private _isValidAllocation;
+}
+//# sourceMappingURL=budget-governor-v2.d.ts.map

package/dist/budget-governor-v2.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-governor-v2.d.ts","sourceRoot":"","sources":["../src/budget-governor-v2.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAO3C,MAAM,MAAM,UAAU,GAClB,MAAM,GACN,SAAS,GACT,OAAO,GACP,QAAQ,GACR,UAAU,GACV,YAAY,CAAC;AAEjB,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,UAAU,CAAC;IACjB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,cAAc,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,kBAAkB,EAAE,CAAC;IAClC,UAAU,EAAE,OAAO,CAAC;IACpB,QAAQ,EAAE,OAAO,CAAC;IAClB,cAAc,EAAE,UAAU,GAAG,SAAS,GAAG,MAAM,CAAC;IAChD,kBAAkB,CAAC,EAAE,OAAO,GAAG,UAAU,GAAG,SAAS,CAAC;CACvD;AAsCD,qBAAa,gBAAiB,SAAQ,YAAY;IAChD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAe;IACtC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IAErC,uDAAuD;IACvD,OAAO,CAAC,WAAW,CAAkC;IAErD,8CAA8C;IAC9C,OAAO,CAAC,WAAW,CAAyB;IAE5C,mDAAmD;IACnD,OAAO,CAAC,gBAAgB,CAAS;gBAErB,MAAM,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,EAAE,UAAU,CAAC,EAAE,MAAM;IAc/D;;;OAGG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI;IA4BxE;;;OAGG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,GAAG,IAAI;IA4DrE;;;OAGG;IACH,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,cAAc;IAmFrC;;OAEG;IACH,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAS7B;;;OAGG;IACH,UAAU,IAAI,IAAI;IAkBlB;;OAEG;IACH,cAAc,IAAI,kBAAkB,EAAE;IAMtC,OAAO,CAAC,IAAI;IAIZ,OAAO,CAAC,eAAe;IAQvB,OAAO,CAAC,iBAAiB;IAWzB,OAAO,CAAC,aAAa;IAWrB,2DAA2D;IAC3D,OAAO,CAAC,gBAAgB;IASxB,OAAO,CAAC,UAAU;IAgBlB,OAAO,CAAC,UAAU;IAmClB,OAAO,CAAC,kBAAkB;CAW3B"}

package/dist/budget-governor-v2.js

@@ -0,0 +1,345 @@
+/**
+ * @module budget-governor-v2
+ * @description Multi-dimensional Budget Governor v2.
+ *
+ * Tracks and enforces budgets across six dimensions simultaneously:
+ *   task, project, daily, branch, research, model_tier
+ *
+ * Features:
+ * - Graceful degradation: suggests LOCAL model tier when usage >= degradeAtPercent
+ * - Hard halt: emits budget_exhausted event when usage >= haltAtPercent
+ * - Atomic JSON storage to ~/.yuan/budget/budget-state.json
+ * - Daily budget auto-resets at UTC midnight
+ * - Never throws — all checks are non-blocking
+ */
+import { EventEmitter } from "node:events";
+import { writeFileSync, readFileSync, mkdirSync, renameSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+// ─── Defaults ───
+const DEFAULT_POLICY = {
+    taskBudget: 50_000,
+    projectBudget: 500_000,
+    dailyBudget: 200_000,
+    branchBudget: 100_000,
+    researchBudget: 20_000,
+    degradeAtPercent: 80,
+    haltAtPercent: 100,
+};
+function todayUtc() {
+    return new Date().toISOString().slice(0, 10); // "YYYY-MM-DD"
+}
+function nowIso() {
+    return new Date().toISOString();
+}
+// ─── BudgetGovernorV2 ───
+export class BudgetGovernorV2 extends EventEmitter {
+    policy;
+    storageDir;
+    storagePath;
+    /** All tracked allocations keyed by `${type}:${id}` */
+    allocations;
+    /** Active task entries (taskId → metadata) */
+    activeTasks;
+    /** Last known daily date for midnight detection */
+    currentDailyDate;
+    constructor(policy, storageDir) {
+        super();
+        this.policy = { ...DEFAULT_POLICY, ...policy };
+        this.storageDir =
+            storageDir ?? join(homedir(), ".yuan", "budget");
+        this.storagePath = join(this.storageDir, "budget-state.json");
+        this.allocations = new Map();
+        this.activeTasks = new Map();
+        this.currentDailyDate = todayUtc();
+        this._loadState();
+    }
+    // ─── Public API ───
+    /**
+     * Start tracking a task.
+     * Creates allocation entries for task / project (if given) / daily / branch (if given).
+     */
+    startTask(taskId, projectId, branchName) {
+        try {
+            this._checkDailyReset();
+            const entry = {
+                taskId,
+                projectId,
+                branchName,
+                startedAt: nowIso(),
+            };
+            this.activeTasks.set(taskId, entry);
+            // Ensure allocation entries exist
+            this._ensureAllocation("task", taskId, this.policy.taskBudget);
+            if (projectId) {
+                this._ensureAllocation("project", projectId, this.policy.projectBudget);
+            }
+            this._ensureAllocation("daily", this.currentDailyDate, this.policy.dailyBudget);
+            if (branchName) {
+                this._ensureAllocation("branch", branchName, this.policy.branchBudget);
+            }
+            this._saveState();
+        }
+        catch {
+            // never crash main loop
+        }
+    }
+    /**
+     * Record token usage for a specific budget dimension.
+     * Triggers degradation/exhaustion events as thresholds are crossed.
+     */
+    recordUsage(tokens, budgetType, id) {
+        try {
+            this._checkDailyReset();
+            const key = this._key(budgetType, id);
+            let alloc = this.allocations.get(key);
+            if (!alloc) {
+                // Auto-create allocation for unknown id using policy default
+                const limit = this._defaultLimit(budgetType);
+                alloc = this._makeAllocation(budgetType, id, limit);
+                this.allocations.set(key, alloc);
+            }
+            const previousUsed = alloc.used;
+            alloc.used = Math.min(alloc.used + tokens, Number.MAX_SAFE_INTEGER);
+            alloc.remaining = Math.max(0, alloc.limit - alloc.used);
+            const percentUsed = alloc.limit > 0 ? (alloc.used / alloc.limit) * 100 : 100;
+            const previousPercent = alloc.limit > 0 ? (previousUsed / alloc.limit) * 100 : 0;
+            const haltThreshold = this.policy.haltAtPercent;
+            const degradeThreshold = this.policy.degradeAtPercent;
+            // Halt threshold crossed
+            if (percentUsed >= haltThreshold && !alloc.exhaustedAt) {
+                alloc.exhaustedAt = nowIso();
+                this.emit("agent:budget_exhausted", {
+                    kind: "agent:budget_exhausted",
+                    budgetType,
+                    id,
+                    percentUsed: Math.round(percentUsed * 100) / 100,
+                    timestamp: alloc.exhaustedAt,
+                });
+            }
+            // Degrade threshold crossed (only once)
+            if (percentUsed >= degradeThreshold &&
+                previousPercent < degradeThreshold &&
+                !alloc.degradedAt) {
+                alloc.degradedAt = nowIso();
+                this.emit("agent:budget_degraded", {
+                    kind: "agent:budget_degraded",
+                    budgetType,
+                    id,
+                    percentUsed: Math.round(percentUsed * 100) / 100,
+                    suggestedTier: "LOCAL",
+                    timestamp: alloc.degradedAt,
+                });
+            }
+            this._saveState();
+        }
+        catch {
+            // never crash main loop
+        }
+    }
+    /**
+     * Check if the task is allowed to continue.
+     * Returns status with recommendation ("continue" | "degrade" | "halt").
+     */
+    check(taskId) {
+        try {
+            this._checkDailyReset();
+            const entry = this.activeTasks.get(taskId);
+            const relevantAllocs = [];
+            // Collect allocations relevant to this task
+            const taskAlloc = this.allocations.get(this._key("task", taskId));
+            if (taskAlloc)
+                relevantAllocs.push(taskAlloc);
+            if (entry?.projectId) {
+                const pAlloc = this.allocations.get(this._key("project", entry.projectId));
+                if (pAlloc)
+                    relevantAllocs.push(pAlloc);
+            }
+            const dailyAlloc = this.allocations.get(this._key("daily", this.currentDailyDate));
+            if (dailyAlloc)
+                relevantAllocs.push(dailyAlloc);
+            if (entry?.branchName) {
+                const bAlloc = this.allocations.get(this._key("branch", entry.branchName));
+                if (bAlloc)
+                    relevantAllocs.push(bAlloc);
+            }
+            const haltThreshold = this.policy.haltAtPercent;
+            const degradeThreshold = this.policy.degradeAtPercent;
+            let isHalted = false;
+            let isDegraded = false;
+            for (const alloc of relevantAllocs) {
+                const percent = alloc.limit > 0 ? (alloc.used / alloc.limit) * 100 : 0;
+                if (percent >= haltThreshold)
+                    isHalted = true;
+                if (percent >= degradeThreshold)
+                    isDegraded = true;
+            }
+            const recommendation = isHalted
+                ? "halt"
+                : isDegraded
+                    ? "degrade"
+                    : "continue";
+            const status = {
+                taskId,
+                projectId: entry?.projectId,
+                allocations: relevantAllocs,
+                isDegraded,
+                isHalted,
+                recommendation,
+                suggestedModelTier: isHalted
+                    ? "LOCAL"
+                    : isDegraded
+                        ? "LOCAL"
+                        : undefined,
+            };
+            this.emit("agent:budget_status", {
+                kind: "agent:budget_status",
+                taskId,
+                recommendation,
+                timestamp: nowIso(),
+            });
+            return status;
+        }
+        catch {
+            // fallback: safe continue
+            return {
+                taskId,
+                allocations: [],
+                isDegraded: false,
+                isHalted: false,
+                recommendation: "continue",
+            };
+        }
+    }
+    /**
+     * End task and finalize its budget tracking.
+     */
+    endTask(taskId) {
+        try {
+            this.activeTasks.delete(taskId);
+            this._saveState();
+        }
+        catch {
+            // never crash main loop
+        }
+    }
+    /**
+     * Reset the daily budget allocation.
+     * Call this at day boundary or let it auto-detect via internal checks.
+     */
+    resetDaily() {
+        try {
+            const today = todayUtc();
+            this.currentDailyDate = today;
+            // Remove old daily allocations and create fresh one
+            for (const [key] of this.allocations) {
+                if (key.startsWith("daily:")) {
+                    this.allocations.delete(key);
+                }
+            }
+            this._ensureAllocation("daily", today, this.policy.dailyBudget);
+            this._saveState();
+        }
+        catch {
+            // never crash main loop
+        }
+    }
+    /**
+     * Get all current allocations.
+     */
+    getAllocations() {
+        return Array.from(this.allocations.values());
+    }
+    // ─── Private Helpers ───
+    _key(type, id) {
+        return `${type}:${id}`;
+    }
+    _makeAllocation(type, id, limit) {
+        return { type, id, limit, used: 0, remaining: limit };
+    }
+    _ensureAllocation(type, id, limit) {
+        const key = this._key(type, id);
+        if (!this.allocations.has(key)) {
+            this.allocations.set(key, this._makeAllocation(type, id, limit));
+        }
+    }
+    _defaultLimit(type) {
+        switch (type) {
+            case "task": return this.policy.taskBudget;
+            case "project": return this.policy.projectBudget;
+            case "daily": return this.policy.dailyBudget;
+            case "branch": return this.policy.branchBudget;
+            case "research": return this.policy.researchBudget;
+            case "model_tier": return this.policy.taskBudget; // sensible fallback
+        }
+    }
+    /** Auto-detect UTC day rollover and reset daily budget. */
+    _checkDailyReset() {
+        const today = todayUtc();
+        if (today !== this.currentDailyDate) {
+            this.resetDaily();
+        }
+    }
+    // ─── Atomic Storage ───
+    _saveState() {
+        try {
+            mkdirSync(this.storageDir, { recursive: true });
+            const state = {
+                allocations: Array.from(this.allocations.values()),
+                dailyDate: this.currentDailyDate,
+            };
+            const tmp = this.storagePath + ".tmp";
+            writeFileSync(tmp, JSON.stringify(state, null, 2), "utf-8");
+            // Atomic rename on POSIX
+            renameSync(tmp, this.storagePath);
+        }
+        catch {
+            // storage failure is non-fatal — in-memory state remains valid
+        }
+    }
+    _loadState() {
+        try {
+            const raw = readFileSync(this.storagePath, "utf-8");
+            const state = JSON.parse(raw);
+            // Restore allocations
+            this.allocations = new Map();
+            if (Array.isArray(state.allocations)) {
+                for (const alloc of state.allocations) {
+                    if (this._isValidAllocation(alloc)) {
+                        this.allocations.set(this._key(alloc.type, alloc.id), alloc);
+                    }
+                }
+            }
+            // Restore or reset daily date
+            if (typeof state.dailyDate === "string") {
+                const today = todayUtc();
+                if (state.dailyDate === today) {
+                    this.currentDailyDate = today;
+                }
+                else {
+                    // Day has changed since last run — reset daily
+                    this.currentDailyDate = today;
+                    for (const [key] of this.allocations) {
+                        if (key.startsWith("daily:")) {
+                            this.allocations.delete(key);
+                        }
+                    }
+                }
+            }
+        }
+        catch {
+            // File doesn't exist or is corrupt — start fresh
+        }
+    }
+    _isValidAllocation(v) {
+        if (typeof v !== "object" || v === null)
+            return false;
+        const obj = v;
+        return (typeof obj["type"] === "string" &&
+            typeof obj["id"] === "string" &&
+            typeof obj["limit"] === "number" &&
+            typeof obj["used"] === "number" &&
+            typeof obj["remaining"] === "number");
+    }
+}
+//# sourceMappingURL=budget-governor-v2.js.map

package/dist/budget-governor-v2.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-governor-v2.js","sourceRoot":"","sources":["../src/budget-governor-v2.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAwDjC,mBAAmB;AAEnB,MAAM,cAAc,GAAiB;IACnC,UAAU,EAAE,MAAM;IAClB,aAAa,EAAE,OAAO;IACtB,WAAW,EAAE,OAAO;IACpB,YAAY,EAAE,OAAO;IACrB,cAAc,EAAE,MAAM;IACtB,gBAAgB,EAAE,EAAE;IACpB,aAAa,EAAE,GAAG;CACnB,CAAC;AAEF,SAAS,QAAQ;IACf,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe;AAC/D,CAAC;AAED,SAAS,MAAM;IACb,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;AAClC,CAAC;AAED,2BAA2B;AAE3B,MAAM,OAAO,gBAAiB,SAAQ,YAAY;IAC/B,MAAM,CAAe;IACrB,UAAU,CAAS;IACnB,WAAW,CAAS;IAErC,uDAAuD;IAC/C,WAAW,CAAkC;IAErD,8CAA8C;IACtC,WAAW,CAAyB;IAE5C,mDAAmD;IAC3C,gBAAgB,CAAS;IAEjC,YAAY,MAA8B,EAAE,UAAmB;QAC7D,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAC/C,IAAI,CAAC,UAAU;YACb,UAAU,IAAI,IAAI,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QACnD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,mBAAmB,CAAC,CAAC;QAC9D,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,QAAQ,EAAE,CAAC;QACnC,IAAI,CAAC,UAAU,EAAE,CAAC;IACpB,CAAC;IAED,qBAAqB;IAErB;;;OAGG;IACH,SAAS,CAAC,MAAc,EAAE,SAAkB,EAAE,UAAmB;QAC/D,IAAI,CAAC;YACH,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAExB,MAAM,KAAK,GAAc;gBACvB,MAAM;gBACN,SAAS;gBACT,UAAU;gBACV,SAAS,EAAE,MAAM,EAAE;aACpB,CAAC;YACF,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;YAEpC,kCAAkC;YAClC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YAC/D,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC1E,CAAC;YACD,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAChF,IAAI,UAAU,EAAE,CAAC;gBACf,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;YACzE,CAAC;YAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,WAAW,CAAC,MAAc,EAAE,UAAsB,EAAE,EAAU;QAC5D,IAAI,CAAC;YACH,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAExB,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACtC,IAAI,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAEtC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,6DAA6D;gBAC7D,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;gBAC7C,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;gBACpD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YACnC,CAAC;YAED,MAAM,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC;YAChC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,GAAG,MAAM,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACpE,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC;YAExD,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;YAC7E,MAAM,eAAe,GACnB,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3D,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YAChD,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAEtD,yBAAyB;YACzB,IAAI,WAAW,IAAI,aAAa,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;gBACvD,KAAK,CAAC,WAAW,GAAG,MAAM,EAAE,CAAC;gBAC7B,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE;oBAClC,IAAI,EAAE,wBAAwB;oBAC9B,UAAU;oBACV,EAAE;oBACF,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,GAAG,CAAC,GAAG,GAAG;oBAChD,SAAS,EAAE,KAAK,CAAC,WAAW;iBAC7B,CAAC,CAAC;YACL,CAAC;YAED,wCAAwC;YACxC,IACE,WAAW,IAAI,gBAAgB;gBAC/B,eAAe,GAAG,gBAAgB;gBAClC,CAAC,KAAK,CAAC,UAAU,EACjB,CAAC;gBACD,KAAK,CAAC,UAAU,GAAG,MAAM,EAAE,CAAC;gBAC5B,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE;oBACjC,IAAI,EAAE,uBAAuB;oBAC7B,UAAU;oBACV,EAAE;oBACF,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,GAAG,CAAC,GAAG,GAAG;oBAChD,aAAa,EAAE,OAAO;oBACtB,SAAS,EAAE,KAAK,CAAC,UAAU;iBAC5B,CAAC,CAAC;YACL,CAAC;YAED,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,MAAc;QAClB,IAAI,CAAC;YACH,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAExB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC3C,MAAM,cAAc,GAAyB,EAAE,CAAC;YAEhD,4CAA4C;YAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAClE,IAAI,SAAS;gBAAE,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAE9C,IAAI,KAAK,EAAE,SAAS,EAAE,CAAC;gBACrB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CACjC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,CACtC,CAAC;gBACF,IAAI,MAAM;oBAAE,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CACrC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAC1C,CAAC;YACF,IAAI,UAAU;gBAAE,cAAc,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEhD,IAAI,KAAK,EAAE,UAAU,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CACjC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,UAAU,CAAC,CACtC,CAAC;gBACF,IAAI,MAAM;oBAAE,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YAChD,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;YAEtD,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;gBACnC,MAAM,OAAO,GACX,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzD,IAAI,OAAO,IAAI,aAAa;oBAAE,QAAQ,GAAG,IAAI,CAAC;gBAC9C,IAAI,OAAO,IAAI,gBAAgB;oBAAE,UAAU,GAAG,IAAI,CAAC;YACrD,CAAC;YAED,MAAM,cAAc,GAAoC,QAAQ;gBAC9D,CAAC,CAAC,MAAM;gBACR,CAAC,CAAC,UAAU;oBACZ,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,UAAU,CAAC;YAEf,MAAM,MAAM,GAAmB;gBAC7B,MAAM;gBACN,SAAS,EAAE,KAAK,EAAE,SAAS;gBAC3B,WAAW,EAAE,cAAc;gBAC3B,UAAU;gBACV,QAAQ;gBACR,cAAc;gBACd,kBAAkB,EAAE,QAAQ;oBAC1B,CAAC,CAAC,OAAO;oBACT,CAAC,CAAC,UAAU;wBACZ,CAAC,CAAC,OAAO;wBACT,CAAC,CAAC,SAAS;aACd,CAAC;YAEF,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBAC/B,IAAI,EAAE,qBAAqB;gBAC3B,MAAM;gBACN,cAAc;gBACd,SAAS,EAAE,MAAM,EAAE;aACpB,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,0BAA0B;YAC1B,OAAO;gBACL,MAAM;gBACN,WAAW,EAAE,EAAE;gBACf,UAAU,EAAE,KAAK;gBACjB,QAAQ,EAAE,KAAK;gBACf,cAAc,EAAE,UAAU;aAC3B,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,MAAc;QACpB,IAAI,CAAC;YACH,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAChC,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;YACzB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;YAE9B,oDAAoD;YACpD,KAAK,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBACrC,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC7B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;YACD,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;QAAC,MAAM,CAAC;YACP,wBAAwB;QAC1B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,0BAA0B;IAElB,IAAI,CAAC,IAAgB,EAAE,EAAU;QACvC,OAAO,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC;IACzB,CAAC;IAEO,eAAe,CACrB,IAAgB,EAChB,EAAU,EACV,KAAa;QAEb,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;IACxD,CAAC;IAEO,iBAAiB,CACvB,IAAgB,EAChB,EAAU,EACV,KAAa;QAEb,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAChC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,IAAgB;QACpC,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM,CAAC,CAAO,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;YACjD,KAAK,SAAS,CAAC,CAAI,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;YACpD,KAAK,OAAO,CAAC,CAAM,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YAClD,KAAK,QAAQ,CAAC,CAAK,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;YACnD,KAAK,UAAU,CAAC,CAAG,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;YACrD,KAAK,YAAY,CAAC,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB;QACxE,CAAC;IACH,CAAC;IAED,2DAA2D;IACnD,gBAAgB;QACtB,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;QACzB,IAAI,KAAK,KAAK,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACpC,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC;IACH,CAAC;IAED,yBAAyB;IAEjB,UAAU;QAChB,IAAI,CAAC;YACH,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAChD,MAAM,KAAK,GAAmB;gBAC5B,WAAW,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;gBAClD,SAAS,EAAE,IAAI,CAAC,gBAAgB;aACjC,CAAC;YACF,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC;YACtC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAC5D,yBAAyB;YACzB,UAAU,CAAC,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACpC,CAAC;QAAC,MAAM,CAAC;YACP,+DAA+D;QACjE,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACpD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAmB,CAAC;YAEhD,sBAAsB;YACtB,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;gBACrC,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;oBACtC,IAAI,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;wBACnC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;oBAC/D,CAAC;gBACH,CAAC;YACH,CAAC;YAED,8BAA8B;YAC9B,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBACxC,MAAM,KAAK,GAAG,QAAQ,EAAE,CAAC;gBACzB,IAAI,KAAK,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;oBAC9B,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;gBAChC,CAAC;qBAAM,CAAC;oBACN,+CAA+C;oBAC/C,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;oBAC9B,KAAK,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;wBACrC,IAAI,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;4BAC7B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;wBAC/B,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iDAAiD;QACnD,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,CAAU;QACnC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI;YAAE,OAAO,KAAK,CAAC;QACtD,MAAM,GAAG,GAAG,CAA4B,CAAC;QACzC,OAAO,CACL,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,QAAQ;YAC/B,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,QAAQ;YAC7B,OAAO,GAAG,CAAC,OAAO,CAAC,KAAK,QAAQ;YAChC,OAAO,GAAG,CAAC,MAAM,CAAC,KAAK,QAAQ;YAC/B,OAAO,GAAG,CAAC,WAAW,CAAC,KAAK,QAAQ,CACrC,CAAC;IACJ,CAAC;CACF"}

package/dist/capability-graph.d.ts

@@ -0,0 +1,102 @@
+/**
+ * @module capability-graph
+ * @description Capability Graph — represents tools, skills, playbooks, agents, and evidence
+ * as interconnected graph nodes. Enables skill composition, smarter tool selection, and
+ * reasoning about what the agent can/cannot do.
+ *
+ * Storage: ~/.yuan/capability-graph.json
+ * Atomic writes (.tmp + renameSync).
+ */
+import { EventEmitter } from "node:events";
+export type CapabilityNodeType = "tool" | "skill" | "playbook" | "agent" | "evidence";
+export interface CapabilityNode {
+    id: string;
+    type: CapabilityNodeType;
+    name: string;
+    description: string;
+    successRate: number;
+    usageCount: number;
+    tags: string[];
+    metadata: Record<string, unknown>;
+    addedAt: string;
+}
+export type CapabilityEdgeType = "uses" | "requires" | "produces" | "verified_by" | "learned_from" | "composed_of" | "conflicts_with";
+export interface CapabilityEdge {
+    fromId: string;
+    toId: string;
+    type: CapabilityEdgeType;
+    weight: number;
+    addedAt: string;
+}
+export interface CapabilityPath {
+    nodes: CapabilityNode[];
+    edges: CapabilityEdge[];
+    totalWeight: number;
+}
+export declare class CapabilityGraph extends EventEmitter {
+    private readonly storageFile;
+    private readonly storageDir;
+    private nodes;
+    private edges;
+    constructor(storageDir?: string);
+    /**
+     * Add or update a node. Emits agent:capability_graph_updated.
+     */
+    upsertNode(node: Omit<CapabilityNode, "addedAt"> & {
+        addedAt?: string;
+    }): void;
+    /**
+     * Add a directed edge between two nodes.
+     * If the edge (fromId, toId, type) already exists, it is replaced.
+     */
+    addEdge(fromId: string, toId: string, type: CapabilityEdgeType, weight?: number): void;
+    /**
+     * Find nodes matching the given filter criteria.
+     */
+    findNodes(filter: {
+        type?: CapabilityNodeType;
+        tags?: string[];
+        minSuccessRate?: number;
+    }): CapabilityNode[];
+    /**
+     * Get all edges originating from a node, optionally filtered by edge type.
+     */
+    getEdges(nodeId: string, edgeType?: CapabilityEdgeType): CapabilityEdge[];
+    /**
+     * BFS path search between two nodes. Max depth 5. Returns shortest path by hop count.
+     */
+    findPath(fromId: string, toId: string): CapabilityPath | null;
+    /**
+     * Get composed capabilities: all tool/skill nodes reachable from nodes
+     * tagged with taskType via "uses" or "composed_of" edges (max depth 3).
+     */
+    getComposition(taskType: string): CapabilityNode[];
+    /**
+     * Update success rate for a node using exponential moving average.
+     * newRate = (oldRate * (n-1) + (success ? 1 : 0)) / n
+     */
+    recordOutcome(nodeId: string, success: boolean): void;
+    /**
+     * Export the full graph as a plain JSON object.
+     */
+    export(): {
+        nodes: CapabilityNode[];
+        edges: CapabilityEdge[];
+    };
+    /**
+     * Import skills from SkillRegistry as graph nodes.
+     * Each skill is added as a "skill" type node; existing nodes are updated.
+     */
+    importFromSkillRegistry(skills: Array<{
+        id: string;
+        name: string;
+        taskType: string;
+        pattern: string;
+        successRate: number;
+        usageCount: number;
+    }>): void;
+    private _load;
+    private _populateDefaults;
+    private _save;
+}
+//# sourceMappingURL=capability-graph.d.ts.map

package/dist/capability-graph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"capability-graph.d.ts","sourceRoot":"","sources":["../src/capability-graph.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAa3C,MAAM,MAAM,kBAAkB,GAC1B,MAAM,GACN,OAAO,GACP,UAAU,GACV,OAAO,GACP,UAAU,CAAC;AAEf,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,kBAAkB,CAAC;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,MAAM,kBAAkB,GAC1B,MAAM,GACN,UAAU,GACV,UAAU,GACV,aAAa,GACb,cAAc,GACd,aAAa,GACb,gBAAgB,CAAC;AAErB,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,kBAAkB,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;CACrB;AAiJD,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAS;IACrC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IAEpC,OAAO,CAAC,KAAK,CAA0C;IAEvD,OAAO,CAAC,KAAK,CAA4C;gBAE7C,UAAU,CAAC,EAAE,MAAM;IAU/B;;OAEG;IACH,UAAU,CACR,IAAI,EAAE,IAAI,CAAC,cAAc,EAAE,SAAS,CAAC,GAAG;QAAE,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,GAC3D,IAAI;IAkBP;;;OAGG;IACH,OAAO,CACL,MAAM,EAAE,MAAM,EACd,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,kBAAkB,EACxB,MAAM,GAAE,MAAY,GACnB,IAAI;IAuBP;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE;QAChB,IAAI,CAAC,EAAE,kBAAkB,CAAC;QAC1B,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;QAChB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB,GAAG,cAAc,EAAE;IAkBpB;;OAEG;IACH,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,kBAAkB,GAAG,cAAc,EAAE;IAMzE;;OAEG;IACH,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,IAAI;IAwD7D;;;OAGG;IACH,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc,EAAE;IA4BlD;;;OAGG;IACH,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAUrD;;OAEG;IACH,MAAM,IAAI;QAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QAAC,KAAK,EAAE,cAAc,EAAE,CAAA;KAAE;IAS9D;;;OAGG;IACH,uBAAuB,CACrB,MAAM,EAAE,KAAK,CAAC;QACZ,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,GACD,IAAI;IAiBP,OAAO,CAAC,KAAK;IA6Bb,OAAO,CAAC,iBAAiB;IAWzB,OAAO,CAAC,KAAK;CAad"}