@matthias-hausberger/beige 0.0.1

package/dist/gateway/policy.spec.js

@@ -0,0 +1,121 @@
+import { describe, it, expect } from "vitest";
+import { PolicyEngine } from "./policy.js";
+import { createFullConfig, createMinimalConfig } from "../test/fixtures.js";
+describe("PolicyEngine", () => {
+    describe("with full config", () => {
+        const config = createFullConfig();
+        const engine = new PolicyEngine(config);
+        describe("isCoreTool", () => {
+            it("returns true for 'read'", () => {
+                expect(engine.isCoreTool("read")).toBe(true);
+            });
+            it("returns true for 'write'", () => {
+                expect(engine.isCoreTool("write")).toBe(true);
+            });
+            it("returns true for 'patch'", () => {
+                expect(engine.isCoreTool("patch")).toBe(true);
+            });
+            it("returns true for 'exec'", () => {
+                expect(engine.isCoreTool("exec")).toBe(true);
+            });
+            it("returns false for custom tools", () => {
+                expect(engine.isCoreTool("kv")).toBe(false);
+                expect(engine.isCoreTool("browser")).toBe(false);
+            });
+            it("returns false for unknown tools", () => {
+                expect(engine.isCoreTool("unknown")).toBe(false);
+                expect(engine.isCoreTool("")).toBe(false);
+            });
+        });
+        describe("isToolAllowed", () => {
+            it("allows configured tools for agent", () => {
+                expect(engine.isToolAllowed("assistant", "kv")).toBe(true);
+                expect(engine.isToolAllowed("researcher", "kv")).toBe(true);
+                expect(engine.isToolAllowed("researcher", "browser")).toBe(true);
+            });
+            it("denies unconfigured tools for agent", () => {
+                expect(engine.isToolAllowed("assistant", "browser")).toBe(false);
+                expect(engine.isToolAllowed("restricted", "kv")).toBe(false);
+            });
+            it("denies all tools for unknown agent", () => {
+                expect(engine.isToolAllowed("unknown-agent", "kv")).toBe(false);
+                expect(engine.isToolAllowed("unknown-agent", "browser")).toBe(false);
+            });
+            it("denies when agent has empty tools array", () => {
+                expect(engine.isToolAllowed("restricted", "kv")).toBe(false);
+                expect(engine.isToolAllowed("restricted", "browser")).toBe(false);
+            });
+        });
+        describe("isAgentValid", () => {
+            it("returns true for configured agents", () => {
+                expect(engine.isAgentValid("assistant")).toBe(true);
+                expect(engine.isAgentValid("researcher")).toBe(true);
+                expect(engine.isAgentValid("restricted")).toBe(true);
+            });
+            it("returns false for unknown agents", () => {
+                expect(engine.isAgentValid("unknown")).toBe(false);
+                expect(engine.isAgentValid("")).toBe(false);
+            });
+        });
+        describe("getToolTarget", () => {
+            it("returns 'gateway' for gateway-targeted tools", () => {
+                expect(engine.getToolTarget("kv")).toBe("gateway");
+            });
+            it("returns 'sandbox' for sandbox-targeted tools", () => {
+                expect(engine.getToolTarget("browser")).toBe("sandbox");
+            });
+            it("returns undefined for unknown tools", () => {
+                expect(engine.getToolTarget("unknown")).toBeUndefined();
+                expect(engine.getToolTarget("")).toBeUndefined();
+            });
+        });
+    });
+    describe("with minimal config (no tools)", () => {
+        const config = createMinimalConfig();
+        const engine = new PolicyEngine(config);
+        it("denies all custom tools when none are configured", () => {
+            expect(engine.isToolAllowed("assistant", "kv")).toBe(false);
+            expect(engine.isToolAllowed("assistant", "browser")).toBe(false);
+        });
+        it("still recognizes core tools", () => {
+            expect(engine.isCoreTool("read")).toBe(true);
+            expect(engine.isCoreTool("write")).toBe(true);
+            expect(engine.isCoreTool("patch")).toBe(true);
+            expect(engine.isCoreTool("exec")).toBe(true);
+        });
+        it("returns undefined for tool targets when no tools configured", () => {
+            expect(engine.getToolTarget("kv")).toBeUndefined();
+        });
+    });
+    describe("edge cases", () => {
+        it("handles agent with empty tools array", () => {
+            const config = createMinimalConfig({
+                agents: {
+                    empty: {
+                        model: { provider: "anthropic", model: "claude" },
+                        tools: [],
+                    },
+                },
+            });
+            const engine = new PolicyEngine(config);
+            expect(engine.isToolAllowed("empty", "any-tool")).toBe(false);
+        });
+        it("handles tool names with special characters", () => {
+            const config = createMinimalConfig({
+                tools: {
+                    "my-custom-tool": { path: "/tools/my-custom-tool", target: "gateway" },
+                },
+                agents: {
+                    assistant: {
+                        model: { provider: "anthropic", model: "claude" },
+                        tools: ["my-custom-tool"],
+                    },
+                },
+            });
+            const engine = new PolicyEngine(config);
+            expect(engine.isToolAllowed("assistant", "my-custom-tool")).toBe(true);
+            expect(engine.getToolTarget("my-custom-tool")).toBe("gateway");
+        });
+    });
+});
+//# sourceMappingURL=policy.spec.js.map

package/dist/gateway/policy.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy.spec.js","sourceRoot":"","sources":["../../src/gateway/policy.spec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAE5E,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,MAAM,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAExC,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;YAC1B,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;gBACjC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;gBAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;gBAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;gBACjC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;gBACxC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;gBACzC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACjD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;YAC7B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;gBAC3C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC3D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC5D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;gBAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACjE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;gBAC5C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAChE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACvE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;gBACjD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAC7D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;YAC5B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;gBAC5C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACpD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACrD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;gBAC1C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnD,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;YAC7B,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;gBACtD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACrD,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;gBACtD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;gBAC7C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;gBACxD,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YACnD,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;QAC9C,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;QAExC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,MAAM,GAAG,mBAAmB,CAAC;gBACjC,MAAM,EAAE;oBACN,KAAK,EAAE;wBACL,KAAK,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE;wBACjD,KAAK,EAAE,EAAE;qBACV;iBACF;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;YAExC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,MAAM,GAAG,mBAAmB,CAAC;gBACjC,KAAK,EAAE;oBACL,gBAAgB,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE,MAAM,EAAE,SAAS,EAAE;iBACvE;gBACD,MAAM,EAAE;oBACN,SAAS,EAAE;wBACT,KAAK,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE;wBACjD,KAAK,EAAE,CAAC,gBAAgB,CAAC;qBAC1B;iBACF;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;YAExC,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/gateway/provider-health.d.ts

@@ -0,0 +1,83 @@
+/**
+ * Provider health tracking for rate limit handling and fallback logic.
+ *
+ * Tracks per-provider rate limits and cooldown periods. When a provider
+ * hits a rate limit, it's marked as "cooling down" until the retry-after
+ * time expires. If no retry-after is provided, a default cooldown is used.
+ *
+ * Persisted to ~/.beige/data/provider-health.json so it survives restarts.
+ */
+export interface ProviderHealthEntry {
+    /** ISO timestamp when the provider hit a rate limit. */
+    rateLimitedAt?: string;
+    /** ISO timestamp when we can retry (from retry-after header or default). */
+    retryAfter?: string;
+    /** Number of consecutive failures (resets on success). */
+    consecutiveFailures: number;
+    /** Last error message (for debugging). */
+    lastError?: string;
+}
+export interface ProviderHealthData {
+    /** provider/model → health entry */
+    providers: Record<string, ProviderHealthEntry>;
+    /** When the data was last persisted */
+    lastUpdated: string;
+}
+/**
+ * Tracks provider health and rate limits.
+ */
+export declare class ProviderHealthTracker {
+    private filePath;
+    private data;
+    constructor();
+    /**
+     * Get the health entry for a provider/model.
+     */
+    get(provider: string, model: string): ProviderHealthEntry | undefined;
+    /**
+     * Check if a provider/model is currently in cooldown (rate limited).
+     */
+    isCoolingDown(provider: string, model: string): boolean;
+    /**
+     * Get the remaining cooldown time in milliseconds, or 0 if not cooling down.
+     */
+    getRemainingCooldown(provider: string, model: string): number;
+    /**
+     * Mark a provider/model as rate limited.
+     * @param retryAfterMs  Optional retry-after time in ms. If not provided, uses default.
+     * @param error         Optional error message for debugging.
+     */
+    markRateLimited(provider: string, model: string, retryAfterMs?: number, error?: string): void;
+    /**
+     * Mark a provider/model as failed (but not necessarily rate limited).
+     */
+    markFailed(provider: string, model: string, error?: string): void;
+    /**
+     * Mark a provider/model as healthy (successful response).
+     */
+    markHealthy(provider: string, model: string): void;
+    /**
+     * Clear the cooldown for a provider/model.
+     */
+    clearCooldown(provider: string, model: string): void;
+    /**
+     * Get all provider/model keys that are currently in cooldown.
+     */
+    getCoolingDown(): Array<{
+        provider: string;
+        model: string;
+        retryAfter: Date;
+    }>;
+    private makeKey;
+    private load;
+    private save;
+}
+/**
+ * Extract rate limit info from an error response.
+ * Checks for common patterns in API error responses.
+ */
+export declare function extractRateLimitInfo(error: unknown): {
+    isRateLimit: boolean;
+    retryAfterMs?: number;
+};
+//# sourceMappingURL=provider-health.d.ts.map

package/dist/gateway/provider-health.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-health.d.ts","sourceRoot":"","sources":["../../src/gateway/provider-health.ts"],"names":[],"mappings":"AAIA;;;;;;;;GAQG;AAEH,MAAM,WAAW,mBAAmB;IAClC,wDAAwD;IACxD,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,4EAA4E;IAC5E,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,0DAA0D;IAC1D,mBAAmB,EAAE,MAAM,CAAC;IAC5B,0CAA0C;IAC1C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,kBAAkB;IACjC,oCAAoC;IACpC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;IAC/C,uCAAuC;IACvC,WAAW,EAAE,MAAM,CAAC;CACrB;AAQD;;GAEG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,IAAI,CAAqB;;IASjC;;OAEG;IACH,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,mBAAmB,GAAG,SAAS;IAKrE;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO;IAgBvD;;OAEG;IACH,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM;IAW7D;;;;OAIG;IACH,eAAe,CACb,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,MAAM,EACb,YAAY,CAAC,EAAE,MAAM,EACrB,KAAK,CAAC,EAAE,MAAM,GACb,IAAI;IAuBP;;OAEG;IACH,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI;IAajE;;OAEG;IACH,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAiBlD;;OAEG;IACH,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAcpD;;OAEG;IACH,cAAc,IAAI,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,IAAI,CAAA;KAAE,CAAC;IAmB9E,OAAO,CAAC,OAAO;IAIf,OAAO,CAAC,IAAI;IAaZ,OAAO,CAAC,IAAI;CAIb;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,OAAO,GAAG;IACpD,WAAW,EAAE,OAAO,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAyCA"}

package/dist/gateway/provider-health.js

@@ -0,0 +1,219 @@
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from "fs";
+import { resolve } from "path";
+import { homedir } from "os";
+/** Default cooldown when no retry-after header is provided (30 minutes) */
+const DEFAULT_COOLDOWN_MS = 30 * 60 * 1000;
+/**
+ * Tracks provider health and rate limits.
+ */
+export class ProviderHealthTracker {
+    filePath;
+    data;
+    constructor() {
+        const dir = resolve(homedir(), ".beige", "data");
+        mkdirSync(dir, { recursive: true });
+        this.filePath = resolve(dir, "provider-health.json");
+        this.data = this.load();
+    }
+    /**
+     * Get the health entry for a provider/model.
+     */
+    get(provider, model) {
+        const key = this.makeKey(provider, model);
+        return this.data.providers[key];
+    }
+    /**
+     * Check if a provider/model is currently in cooldown (rate limited).
+     */
+    isCoolingDown(provider, model) {
+        const entry = this.get(provider, model);
+        if (!entry?.retryAfter)
+            return false;
+        const retryAfter = new Date(entry.retryAfter).getTime();
+        const now = Date.now();
+        if (now >= retryAfter) {
+            // Cooldown expired — clear it
+            this.clearCooldown(provider, model);
+            return false;
+        }
+        return true;
+    }
+    /**
+     * Get the remaining cooldown time in milliseconds, or 0 if not cooling down.
+     */
+    getRemainingCooldown(provider, model) {
+        const entry = this.get(provider, model);
+        if (!entry?.retryAfter)
+            return 0;
+        const retryAfter = new Date(entry.retryAfter).getTime();
+        const now = Date.now();
+        const remaining = retryAfter - now;
+        return Math.max(0, remaining);
+    }
+    /**
+     * Mark a provider/model as rate limited.
+     * @param retryAfterMs  Optional retry-after time in ms. If not provided, uses default.
+     * @param error         Optional error message for debugging.
+     */
+    markRateLimited(provider, model, retryAfterMs, error) {
+        const key = this.makeKey(provider, model);
+        const now = new Date();
+        const cooldownMs = retryAfterMs ?? DEFAULT_COOLDOWN_MS;
+        const retryAfter = new Date(now.getTime() + cooldownMs);
+        const existing = this.data.providers[key] ?? { consecutiveFailures: 0 };
+        this.data.providers[key] = {
+            ...existing,
+            rateLimitedAt: now.toISOString(),
+            retryAfter: retryAfter.toISOString(),
+            consecutiveFailures: existing.consecutiveFailures + 1,
+            lastError: error,
+        };
+        this.save();
+        console.log(`[PROVIDER_HEALTH] ${key} rate limited until ${retryAfter.toISOString()}` +
+            (error ? ` (${error})` : ""));
+    }
+    /**
+     * Mark a provider/model as failed (but not necessarily rate limited).
+     */
+    markFailed(provider, model, error) {
+        const key = this.makeKey(provider, model);
+        const existing = this.data.providers[key] ?? { consecutiveFailures: 0 };
+        this.data.providers[key] = {
+            ...existing,
+            consecutiveFailures: existing.consecutiveFailures + 1,
+            lastError: error,
+        };
+        this.save();
+    }
+    /**
+     * Mark a provider/model as healthy (successful response).
+     */
+    markHealthy(provider, model) {
+        const key = this.makeKey(provider, model);
+        const existing = this.data.providers[key];
+        if (existing && (existing.consecutiveFailures > 0 || existing.retryAfter)) {
+            this.data.providers[key] = {
+                consecutiveFailures: 0,
+                // Clear rate limit info on success
+                rateLimitedAt: undefined,
+                retryAfter: undefined,
+                lastError: undefined,
+            };
+            this.save();
+            console.log(`[PROVIDER_HEALTH] ${key} marked healthy`);
+        }
+    }
+    /**
+     * Clear the cooldown for a provider/model.
+     */
+    clearCooldown(provider, model) {
+        const key = this.makeKey(provider, model);
+        const existing = this.data.providers[key];
+        if (existing) {
+            this.data.providers[key] = {
+                ...existing,
+                rateLimitedAt: undefined,
+                retryAfter: undefined,
+            };
+            this.save();
+        }
+    }
+    /**
+     * Get all provider/model keys that are currently in cooldown.
+     */
+    getCoolingDown() {
+        const now = Date.now();
+        const result = [];
+        for (const [key, entry] of Object.entries(this.data.providers)) {
+            if (entry.retryAfter) {
+                const retryAfter = new Date(entry.retryAfter);
+                if (retryAfter.getTime() > now) {
+                    const [provider, model] = key.split("/");
+                    result.push({ provider, model, retryAfter });
+                }
+            }
+        }
+        return result;
+    }
+    // ── Private ────────────────────────────────────────────────────────
+    makeKey(provider, model) {
+        return `${provider}/${model}`;
+    }
+    load() {
+        if (!existsSync(this.filePath)) {
+            return { providers: {}, lastUpdated: new Date().toISOString() };
+        }
+        try {
+            const raw = readFileSync(this.filePath, "utf-8");
+            return JSON.parse(raw);
+        }
+        catch {
+            return { providers: {}, lastUpdated: new Date().toISOString() };
+        }
+    }
+    save() {
+        this.data.lastUpdated = new Date().toISOString();
+        writeFileSync(this.filePath, JSON.stringify(this.data, null, 2));
+    }
+}
+/**
+ * Extract rate limit info from an error response.
+ * Checks for common patterns in API error responses.
+ */
+export function extractRateLimitInfo(error) {
+    if (!error || typeof error !== "object") {
+        return { isRateLimit: false };
+    }
+    const err = error;
+    // Check for HTTP status 429
+    if (err.status === 429 || err.statusCode === 429) {
+        const retryAfter = err.headers?.["retry-after"] ?? err.headers?.["Retry-After"];
+        return {
+            isRateLimit: true,
+            retryAfterMs: parseRetryAfter(retryAfter),
+        };
+    }
+    // Check for error type/message patterns
+    const errorType = err.error?.type ?? err.type ?? "";
+    const errorMessage = err.message ?? err.error?.message ?? "";
+    const rateLimitPatterns = [
+        "rate_limit",
+        "rate limit",
+        "too many requests",
+        "overloaded",
+        "capacity",
+        "temporarily unavailable",
+    ];
+    const combined = `${errorType} ${errorMessage}`.toLowerCase();
+    const isRateLimit = rateLimitPatterns.some((p) => combined.includes(p));
+    if (isRateLimit) {
+        const retryAfter = err.headers?.["retry-after"] ?? err.headers?.["Retry-After"];
+        return {
+            isRateLimit: true,
+            retryAfterMs: parseRetryAfter(retryAfter),
+        };
+    }
+    return { isRateLimit: false };
+}
+/**
+ * Parse a retry-after header value.
+ * Can be either a number of seconds or an HTTP date.
+ */
+function parseRetryAfter(value) {
+    if (value === undefined || value === null)
+        return undefined;
+    if (typeof value === "number") {
+        return value * 1000; // seconds to ms
+    }
+    const num = parseInt(value, 10);
+    if (!isNaN(num)) {
+        return num * 1000; // seconds to ms
+    }
+    // Try parsing as HTTP date
+    const date = new Date(value);
+    if (!isNaN(date.getTime())) {
+        return Math.max(0, date.getTime() - Date.now());
+    }
+    return undefined;
+}
+//# sourceMappingURL=provider-health.js.map

package/dist/gateway/provider-health.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-health.js","sourceRoot":"","sources":["../../src/gateway/provider-health.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,OAAO,EAAW,MAAM,MAAM,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AA8B7B,2EAA2E;AAC3E,MAAM,mBAAmB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAK3C;;GAEG;AACH,MAAM,OAAO,qBAAqB;IACxB,QAAQ,CAAS;IACjB,IAAI,CAAqB;IAEjC;QACE,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QACjD,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACpC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE,sBAAsB,CAAC,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,QAAgB,EAAE,KAAa;QACjC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB,EAAE,KAAa;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,EAAE,UAAU;YAAE,OAAO,KAAK,CAAC;QAErC,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,IAAI,GAAG,IAAI,UAAU,EAAE,CAAC;YACtB,8BAA8B;YAC9B,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;YACpC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,oBAAoB,CAAC,QAAgB,EAAE,KAAa;QAClD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,EAAE,UAAU;YAAE,OAAO,CAAC,CAAC;QAEjC,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC;QACxD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,UAAU,GAAG,GAAG,CAAC;QAEnC,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;IAChC,CAAC;IAED;;;;OAIG;IACH,eAAe,CACb,QAAgB,EAChB,KAAa,EACb,YAAqB,EACrB,KAAc;QAEd,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC1C,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,YAAY,IAAI,mBAAmB,CAAC;QACvD,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,UAAU,CAAC,CAAC;QAExD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,mBAAmB,EAAE,CAAC,EAAE,CAAC;QAExE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG;YACzB,GAAG,QAAQ;YACX,aAAa,EAAE,GAAG,CAAC,WAAW,EAAE;YAChC,UAAU,EAAE,UAAU,CAAC,WAAW,EAAE;YACpC,mBAAmB,EAAE,QAAQ,CAAC,mBAAmB,GAAG,CAAC;YACrD,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,IAAI,EAAE,CAAC;QACZ,OAAO,CAAC,GAAG,CACT,qBAAqB,GAAG,uBAAuB,UAAU,CAAC,WAAW,EAAE,EAAE;YACzE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,QAAgB,EAAE,KAAa,EAAE,KAAc;QACxD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,mBAAmB,EAAE,CAAC,EAAE,CAAC;QAExE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG;YACzB,GAAG,QAAQ;YACX,mBAAmB,EAAE,QAAQ,CAAC,mBAAmB,GAAG,CAAC;YACrD,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,IAAI,EAAE,CAAC;IACd,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,QAAgB,EAAE,KAAa;QACzC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAE1C,IAAI,QAAQ,IAAI,CAAC,QAAQ,CAAC,mBAAmB,GAAG,CAAC,IAAI,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC1E,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG;gBACzB,mBAAmB,EAAE,CAAC;gBACtB,mCAAmC;gBACnC,aAAa,EAAE,SAAS;gBACxB,UAAU,EAAE,SAAS;gBACrB,SAAS,EAAE,SAAS;aACrB,CAAC;YACF,IAAI,CAAC,IAAI,EAAE,CAAC;YACZ,OAAO,CAAC,GAAG,CAAC,qBAAqB,GAAG,iBAAiB,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB,EAAE,KAAa;QAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAE1C,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG;gBACzB,GAAG,QAAQ;gBACX,aAAa,EAAE,SAAS;gBACxB,UAAU,EAAE,SAAS;aACtB,CAAC;YACF,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAiE,EAAE,CAAC;QAEhF,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/D,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACrB,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBAC9C,IAAI,UAAU,CAAC,OAAO,EAAE,GAAG,GAAG,EAAE,CAAC;oBAC/B,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;oBACzC,MAAM,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;gBAC/C,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,sEAAsE;IAE9D,OAAO,CAAC,QAAgB,EAAE,KAAa;QAC7C,OAAO,GAAG,QAAQ,IAAI,KAAK,EAAE,CAAC;IAChC,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;QAClE,CAAC;QAED,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACjD,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAuB,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC;QAClE,CAAC;IACH,CAAC;IAEO,IAAI;QACV,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACjD,aAAa,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACnE,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAAc;IAIjD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;IAChC,CAAC;IAED,MAAM,GAAG,GAAG,KAA4B,CAAC;IAEzC,4BAA4B;IAC5B,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;QACjD,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC;QAChF,OAAO;YACL,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,eAAe,CAAC,UAAU,CAAC;SAC1C,CAAC;IACJ,CAAC;IAED,wCAAwC;IACxC,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,EAAE,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;IACpD,MAAM,YAAY,GAAG,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;IAE7D,MAAM,iBAAiB,GAAG;QACxB,YAAY;QACZ,YAAY;QACZ,mBAAmB;QACnB,YAAY;QACZ,UAAU;QACV,yBAAyB;KAC1B,CAAC;IAEF,MAAM,QAAQ,GAAG,GAAG,SAAS,IAAI,YAAY,EAAE,CAAC,WAAW,EAAE,CAAC;IAC9D,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAExE,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,GAAG,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC,aAAa,CAAC,CAAC;QAChF,OAAO;YACL,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,eAAe,CAAC,UAAU,CAAC;SAC1C,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;AAChC,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,KAAkC;IACzD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI;QAAE,OAAO,SAAS,CAAC;IAE5D,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,GAAG,IAAI,CAAC,CAAC,gBAAgB;IACvC,CAAC;IAED,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAChC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAChB,OAAO,GAAG,GAAG,IAAI,CAAC,CAAC,gBAAgB;IACrC,CAAC;IAED,2BAA2B;IAC3B,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/gateway/provider-health.spec.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=provider-health.spec.d.ts.map

package/dist/gateway/provider-health.spec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider-health.spec.d.ts","sourceRoot":"","sources":["../../src/gateway/provider-health.spec.ts"],"names":[],"mappings":""}