atabey 0.0.13 → 0.0.14
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/ATABEY.md +1 -1
- package/README.md +252 -52
- package/dist/framework-mcp/src/index.js +914 -62
- package/dist/framework-mcp/src/index.js.map +1 -1
- package/dist/framework-mcp/src/tools/compliance/mask_pii.d.ts +17 -0
- package/dist/framework-mcp/src/tools/compliance/mask_pii.js +56 -0
- package/dist/framework-mcp/src/tools/compliance/mask_pii.js.map +1 -0
- package/dist/framework-mcp/src/tools/definitions.js +10 -9
- package/dist/framework-mcp/src/tools/definitions.js.map +1 -1
- package/dist/framework-mcp/src/tools/framework/run_tests.js +25 -0
- package/dist/framework-mcp/src/tools/framework/run_tests.js.map +1 -1
- package/dist/framework-mcp/src/tools/index.js +7 -8
- package/dist/framework-mcp/src/tools/index.js.map +1 -1
- package/dist/framework-mcp/src/tools/messaging/approve_operation.d.ts +22 -0
- package/dist/framework-mcp/src/tools/messaging/approve_operation.js +81 -0
- package/dist/framework-mcp/src/tools/messaging/approve_operation.js.map +1 -0
- package/dist/framework-mcp/src/tools/messaging/ask_human.d.ts +15 -0
- package/dist/framework-mcp/src/tools/messaging/ask_human.js +89 -20
- package/dist/framework-mcp/src/tools/messaging/ask_human.js.map +1 -1
- package/dist/framework-mcp/src/tools/observability/check_ports.js +3 -1
- package/dist/framework-mcp/src/tools/observability/check_ports.js.map +1 -1
- package/dist/framework-mcp/src/tools/quality/analyze_code_quality.js +13 -64
- package/dist/framework-mcp/src/tools/quality/analyze_code_quality.js.map +1 -1
- package/dist/framework-mcp/src/tools/schemas.d.ts +48 -53
- package/dist/framework-mcp/src/tools/schemas.js +18 -22
- package/dist/framework-mcp/src/tools/schemas.js.map +1 -1
- package/dist/framework-mcp/src/tools/shell/run_command.js +10 -0
- package/dist/framework-mcp/src/tools/shell/run_command.js.map +1 -1
- package/dist/framework-mcp/src/tools/types.d.ts +11 -105
- package/dist/framework-mcp/src/tools/types.js +1 -0
- package/dist/framework-mcp/src/tools/types.js.map +1 -1
- package/dist/framework-mcp/src/utils/auth.d.ts +36 -0
- package/dist/framework-mcp/src/utils/auth.js +79 -0
- package/dist/framework-mcp/src/utils/auth.js.map +1 -0
- package/dist/framework-mcp/src/utils/auto-rollback.d.ts +71 -0
- package/dist/framework-mcp/src/utils/auto-rollback.js +266 -0
- package/dist/framework-mcp/src/utils/auto-rollback.js.map +1 -0
- package/dist/framework-mcp/src/utils/compliance.js +12 -4
- package/dist/framework-mcp/src/utils/compliance.js.map +1 -1
- package/dist/framework-mcp/src/utils/context-optimizer.d.ts +69 -0
- package/dist/framework-mcp/src/utils/context-optimizer.js +157 -0
- package/dist/framework-mcp/src/utils/context-optimizer.js.map +1 -0
- package/dist/framework-mcp/src/utils/discipline.d.ts +68 -0
- package/dist/framework-mcp/src/utils/discipline.js +218 -0
- package/dist/framework-mcp/src/utils/discipline.js.map +1 -0
- package/dist/framework-mcp/src/utils/finops.d.ts +143 -0
- package/dist/framework-mcp/src/utils/finops.js +326 -0
- package/dist/framework-mcp/src/utils/finops.js.map +1 -0
- package/dist/framework-mcp/src/utils/human-in-loop.d.ts +95 -0
- package/dist/framework-mcp/src/utils/human-in-loop.js +216 -0
- package/dist/framework-mcp/src/utils/human-in-loop.js.map +1 -0
- package/dist/framework-mcp/src/utils/license-scanner.d.ts +71 -0
- package/dist/framework-mcp/src/utils/license-scanner.js +284 -0
- package/dist/framework-mcp/src/utils/license-scanner.js.map +1 -0
- package/dist/framework-mcp/src/utils/loop-detector.d.ts +105 -0
- package/dist/framework-mcp/src/utils/loop-detector.js +362 -0
- package/dist/framework-mcp/src/utils/loop-detector.js.map +1 -0
- package/dist/framework-mcp/src/utils/quality.d.ts +19 -0
- package/dist/framework-mcp/src/utils/quality.js +124 -0
- package/dist/framework-mcp/src/utils/quality.js.map +1 -0
- package/dist/framework-mcp/src/utils/rules-engine.d.ts +78 -0
- package/dist/framework-mcp/src/utils/rules-engine.js +232 -0
- package/dist/framework-mcp/src/utils/rules-engine.js.map +1 -0
- package/dist/framework-mcp/src/utils/silent-router.d.ts +47 -0
- package/dist/framework-mcp/src/utils/silent-router.js +129 -0
- package/dist/framework-mcp/src/utils/silent-router.js.map +1 -0
- package/dist/framework-mcp/src/utils/storage.d.ts +1 -0
- package/dist/framework-mcp/src/utils/storage.js +2 -2
- package/dist/framework-mcp/src/utils/storage.js.map +1 -1
- package/dist/framework-mcp/src/utils/telemetry-streamer.d.ts +133 -0
- package/dist/framework-mcp/src/utils/telemetry-streamer.js +397 -0
- package/dist/framework-mcp/src/utils/telemetry-streamer.js.map +1 -0
- package/dist/framework-mcp/tests/tools/file_system/file_system_tools.test.js +2 -2
- package/dist/framework-mcp/tests/tools/file_system/file_system_tools.test.js.map +1 -1
- package/dist/framework-mcp/tests/tools/framework/run_tests.test.d.ts +1 -0
- package/dist/framework-mcp/tests/tools/framework/run_tests.test.js +49 -0
- package/dist/framework-mcp/tests/tools/framework/run_tests.test.js.map +1 -0
- package/dist/framework-mcp/tests/tools/messaging/send_message.test.js +14 -5
- package/dist/framework-mcp/tests/tools/messaging/send_message.test.js.map +1 -1
- package/dist/framework-mcp/tests/tools/observability/check_ports.test.d.ts +1 -0
- package/dist/framework-mcp/tests/tools/observability/check_ports.test.js +30 -0
- package/dist/framework-mcp/tests/tools/observability/check_ports.test.js.map +1 -0
- package/dist/framework-mcp/tests/tools/shell/run_command.test.js +7 -0
- package/dist/framework-mcp/tests/tools/shell/run_command.test.js.map +1 -1
- package/dist/framework-mcp/tests/utils/auto-rollback.test.d.ts +7 -0
- package/dist/framework-mcp/tests/utils/auto-rollback.test.js +178 -0
- package/dist/framework-mcp/tests/utils/auto-rollback.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/context-optimizer.test.d.ts +1 -0
- package/dist/framework-mcp/tests/utils/context-optimizer.test.js +62 -0
- package/dist/framework-mcp/tests/utils/context-optimizer.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/finops.test.d.ts +7 -0
- package/dist/framework-mcp/tests/utils/finops.test.js +157 -0
- package/dist/framework-mcp/tests/utils/finops.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/human-in-loop.test.d.ts +1 -0
- package/dist/framework-mcp/tests/utils/human-in-loop.test.js +59 -0
- package/dist/framework-mcp/tests/utils/human-in-loop.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/license-scanner.test.d.ts +7 -0
- package/dist/framework-mcp/tests/utils/license-scanner.test.js +131 -0
- package/dist/framework-mcp/tests/utils/license-scanner.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/loop-detector.test.d.ts +7 -0
- package/dist/framework-mcp/tests/utils/loop-detector.test.js +204 -0
- package/dist/framework-mcp/tests/utils/loop-detector.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/rules-engine.test.d.ts +1 -0
- package/dist/framework-mcp/tests/utils/rules-engine.test.js +106 -0
- package/dist/framework-mcp/tests/utils/rules-engine.test.js.map +1 -0
- package/dist/framework-mcp/tests/utils/telemetry-streamer.test.d.ts +7 -0
- package/dist/framework-mcp/tests/utils/telemetry-streamer.test.js +201 -0
- package/dist/framework-mcp/tests/utils/telemetry-streamer.test.js.map +1 -0
- package/dist/src/cli/commands/check.js +16 -1
- package/dist/src/cli/commands/check.js.map +1 -1
- package/dist/src/cli/commands/dashboard.js +188 -164
- package/dist/src/cli/commands/dashboard.js.map +1 -1
- package/dist/src/cli/commands/gateway.js +2 -2
- package/dist/src/cli/commands/gateway.js.map +1 -1
- package/dist/src/cli/commands/init/scaffold-core.d.ts +2 -0
- package/dist/src/cli/commands/init/scaffold-core.js +16 -1
- package/dist/src/cli/commands/init/scaffold-core.js.map +1 -1
- package/dist/src/cli/commands/init/scaffold-ops.js +2 -2
- package/dist/src/cli/commands/init/scaffold-ops.js.map +1 -1
- package/dist/src/cli/commands/init/scaffold-standards.js +13 -1
- package/dist/src/cli/commands/init/scaffold-standards.js.map +1 -1
- package/dist/src/cli/commands/init.d.ts +4 -0
- package/dist/src/cli/commands/init.js +80 -24
- package/dist/src/cli/commands/init.js.map +1 -1
- package/dist/src/cli/commands/quickstart.d.ts +1 -1
- package/dist/src/cli/commands/quickstart.js +12 -24
- package/dist/src/cli/commands/quickstart.js.map +1 -1
- package/dist/src/cli/index.js +24 -23
- package/dist/src/cli/index.js.map +1 -1
- package/dist/src/cli/shims.js +7 -7
- package/dist/src/cli/utils/pkg.js +2 -3
- package/dist/src/cli/utils/pkg.js.map +1 -1
- package/dist/src/cli/utils/quality.d.ts +10 -0
- package/dist/src/cli/utils/quality.js +10 -0
- package/dist/src/cli/utils/quality.js.map +1 -0
- package/dist/src/modules/agents/definitions.js +45 -40
- package/dist/src/modules/agents/definitions.js.map +1 -1
- package/dist/src/modules/agents/registry/manager.js +6 -1
- package/dist/src/modules/agents/registry/manager.js.map +1 -1
- package/dist/src/modules/engines/agent-executor.d.ts +6 -10
- package/dist/src/modules/engines/agent-executor.js +53 -21
- package/dist/src/modules/engines/agent-executor.js.map +1 -1
- package/dist/src/modules/engines/agent-loop.d.ts +34 -0
- package/dist/src/modules/engines/agent-loop.js +102 -0
- package/dist/src/modules/engines/agent-loop.js.map +1 -0
- package/dist/src/modules/engines/evaluation-engine.js +2 -2
- package/dist/src/modules/engines/evaluation-engine.js.map +1 -1
- package/dist/src/modules/engines/quality-gate.d.ts +13 -5
- package/dist/src/modules/engines/quality-gate.js +39 -31
- package/dist/src/modules/engines/quality-gate.js.map +1 -1
- package/dist/src/modules/engines/routing-engine.d.ts +5 -3
- package/dist/src/modules/engines/routing-engine.js +16 -9
- package/dist/src/modules/engines/routing-engine.js.map +1 -1
- package/dist/src/modules/gateway/cost-tracker.d.ts +5 -0
- package/dist/src/modules/gateway/cost-tracker.js +71 -0
- package/dist/src/modules/gateway/cost-tracker.js.map +1 -1
- package/dist/src/modules/gateway/gateway.d.ts +4 -0
- package/dist/src/modules/gateway/gateway.js +8 -0
- package/dist/src/modules/gateway/gateway.js.map +1 -1
- package/dist/src/shared/audit.js +3 -2
- package/dist/src/shared/audit.js.map +1 -1
- package/dist/src/shared/pii.d.ts +41 -9
- package/dist/src/shared/pii.js +122 -31
- package/dist/src/shared/pii.js.map +1 -1
- package/dist/src/shared/storage.d.ts +19 -0
- package/dist/src/shared/storage.js +46 -6
- package/dist/src/shared/storage.js.map +1 -1
- package/dist/tests/embedding.test.d.ts +1 -0
- package/dist/tests/embedding.test.js +82 -0
- package/dist/tests/embedding.test.js.map +1 -0
- package/dist/tests/engines/evaluation-engine.test.d.ts +1 -0
- package/dist/tests/engines/evaluation-engine.test.js +71 -0
- package/dist/tests/engines/evaluation-engine.test.js.map +1 -0
- package/dist/tests/engines/planning-engine.test.d.ts +1 -0
- package/dist/tests/engines/planning-engine.test.js +100 -0
- package/dist/tests/engines/planning-engine.test.js.map +1 -0
- package/dist/tests/engines/quality-gate.test.d.ts +1 -0
- package/dist/tests/engines/quality-gate.test.js +59 -0
- package/dist/tests/engines/quality-gate.test.js.map +1 -0
- package/dist/tests/engines/risk-engine.test.d.ts +1 -0
- package/dist/tests/engines/risk-engine.test.js +80 -0
- package/dist/tests/engines/risk-engine.test.js.map +1 -0
- package/dist/tests/engines/routing-engine.test.d.ts +1 -0
- package/dist/tests/engines/routing-engine.test.js +78 -0
- package/dist/tests/engines/routing-engine.test.js.map +1 -0
- package/dist/tests/integration/hermes_locking.test.js +1 -0
- package/dist/tests/integration/hermes_locking.test.js.map +1 -1
- package/dist/tests/modules/gateway/cost-tracker.test.js +20 -1
- package/dist/tests/modules/gateway/cost-tracker.test.js.map +1 -1
- package/dist/tests/shared/lock.test.d.ts +1 -0
- package/dist/tests/shared/lock.test.js +95 -0
- package/dist/tests/shared/lock.test.js.map +1 -0
- package/dist/tests/shared/pii.test.d.ts +1 -0
- package/dist/tests/shared/pii.test.js +183 -0
- package/dist/tests/shared/pii.test.js.map +1 -0
- package/framework-mcp/README.md +307 -35
- package/framework-mcp/dist/dashboard/assets/index-B2mYld0c.js +1221 -0
- package/framework-mcp/dist/dashboard/index.html +1 -1
- package/framework-mcp/dist/framework-mcp/src/index.js +914 -62
- package/framework-mcp/dist/framework-mcp/src/tools/compliance/mask_pii.js +55 -0
- package/framework-mcp/dist/framework-mcp/src/tools/definitions.js +10 -9
- package/framework-mcp/dist/framework-mcp/src/tools/framework/run_tests.js +25 -0
- package/framework-mcp/dist/framework-mcp/src/tools/index.js +7 -8
- package/framework-mcp/dist/framework-mcp/src/tools/messaging/approve_operation.js +80 -0
- package/framework-mcp/dist/framework-mcp/src/tools/messaging/ask_human.js +89 -20
- package/framework-mcp/dist/framework-mcp/src/tools/observability/check_ports.js +3 -1
- package/framework-mcp/dist/framework-mcp/src/tools/quality/analyze_code_quality.js +13 -64
- package/framework-mcp/dist/framework-mcp/src/tools/schemas.js +18 -22
- package/framework-mcp/dist/framework-mcp/src/tools/shell/run_command.js +10 -0
- package/framework-mcp/dist/framework-mcp/src/tools/types.js +1 -0
- package/framework-mcp/dist/framework-mcp/src/utils/auth.js +78 -0
- package/framework-mcp/dist/framework-mcp/src/utils/auto-rollback.js +265 -0
- package/framework-mcp/dist/framework-mcp/src/utils/compliance.js +12 -4
- package/framework-mcp/dist/framework-mcp/src/utils/context-optimizer.js +156 -0
- package/framework-mcp/dist/framework-mcp/src/utils/discipline.js +217 -0
- package/framework-mcp/dist/framework-mcp/src/utils/finops.js +325 -0
- package/framework-mcp/dist/framework-mcp/src/utils/human-in-loop.js +215 -0
- package/framework-mcp/dist/framework-mcp/src/utils/license-scanner.js +283 -0
- package/framework-mcp/dist/framework-mcp/src/utils/loop-detector.js +361 -0
- package/framework-mcp/dist/framework-mcp/src/utils/quality.js +123 -0
- package/framework-mcp/dist/framework-mcp/src/utils/rules-engine.js +231 -0
- package/framework-mcp/dist/framework-mcp/src/utils/silent-router.js +128 -0
- package/framework-mcp/dist/framework-mcp/src/utils/storage.js +2 -2
- package/framework-mcp/dist/framework-mcp/src/utils/telemetry-streamer.js +396 -0
- package/framework-mcp/dist/src/cli/platforms/core.js +72 -0
- package/framework-mcp/dist/src/cli/platforms/index.js +5 -0
- package/framework-mcp/dist/src/cli/platforms/paths.js +101 -0
- package/framework-mcp/dist/src/cli/platforms/scaffold.js +72 -0
- package/framework-mcp/dist/src/cli/platforms/utils.js +75 -0
- package/framework-mcp/dist/src/cli/utils/claude.js +57 -0
- package/framework-mcp/dist/src/cli/utils/compliance.js +12 -0
- package/framework-mcp/dist/src/cli/utils/fs.js +138 -0
- package/framework-mcp/dist/src/cli/utils/pkg.js +282 -0
- package/framework-mcp/dist/src/cli/utils/quality.js +9 -0
- package/framework-mcp/dist/src/modules/agents/definitions.js +563 -0
- package/framework-mcp/dist/src/modules/agents/registry/analyst.js +39 -0
- package/framework-mcp/dist/src/modules/agents/registry/architect.js +42 -0
- package/framework-mcp/dist/src/modules/agents/registry/backend.js +50 -0
- package/framework-mcp/dist/src/modules/agents/registry/database.js +45 -0
- package/framework-mcp/dist/src/modules/agents/registry/devops.js +45 -0
- package/framework-mcp/dist/src/modules/agents/registry/explorer.js +37 -0
- package/framework-mcp/dist/src/modules/agents/registry/frontend.js +52 -0
- package/framework-mcp/dist/src/modules/agents/registry/git.js +36 -0
- package/framework-mcp/dist/src/modules/agents/registry/manager.js +64 -0
- package/framework-mcp/dist/src/modules/agents/registry/mobile.js +41 -0
- package/framework-mcp/dist/src/modules/agents/registry/native.js +41 -0
- package/framework-mcp/dist/src/modules/agents/registry/quality.js +43 -0
- package/framework-mcp/dist/src/modules/agents/registry/security.js +42 -0
- package/framework-mcp/dist/src/modules/agents/types.js +1 -0
- package/framework-mcp/dist/src/modules/engines/quality-gate.js +92 -0
- package/framework-mcp/dist/src/modules/engines/risk-engine.js +105 -0
- package/framework-mcp/dist/src/modules/engines/routing-engine.js +408 -0
- package/framework-mcp/dist/src/modules/engines/types.js +1 -0
- package/framework-mcp/dist/src/modules/providers/definitions.js +140 -0
- package/framework-mcp/dist/src/modules/providers/registry.js +18 -0
- package/framework-mcp/dist/src/modules/providers/shared.js +104 -0
- package/framework-mcp/dist/src/modules/providers/types.js +1 -0
- package/framework-mcp/dist/src/modules/skills/adapter-skills.js +449 -0
- package/framework-mcp/dist/src/modules/skills/definitions.js +70 -0
- package/framework-mcp/dist/src/shared/audit.js +220 -0
- package/framework-mcp/dist/src/shared/errors.js +68 -0
- package/framework-mcp/dist/src/shared/lock.js +159 -0
- package/framework-mcp/dist/src/shared/pii.js +122 -31
- package/framework-mcp/dist/src/shared/retention.js +170 -0
- package/framework-mcp/dist/src/shared/storage.js +46 -6
- package/framework-mcp/dist/src/shared/string.js +29 -0
- package/framework-mcp/package.json +8 -2
- package/mcp.json +11 -6
- package/package.json +13 -6
- package/templates/full/ATABEY_FULL.md +1 -1
- package/framework-mcp/dist/dashboard/assets/index-BAPGLiWE.js +0 -267
|
@@ -0,0 +1,143 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ─── FINOPS / BUDGET ENFORCEMENT ──────────────────────────────────
|
|
3
|
+
*
|
|
4
|
+
* Enterprise budget management for AI token consumption.
|
|
5
|
+
* Enforces strict cost limits per team, agent, or project.
|
|
6
|
+
*
|
|
7
|
+
* Features:
|
|
8
|
+
* - Team-based budget allocation (monthly/quarterly)
|
|
9
|
+
* - Per-agent spending limits
|
|
10
|
+
* - Automatic shutdown when budget exceeded
|
|
11
|
+
* - Central server sync for budget policies
|
|
12
|
+
* - Real-time cost tracking
|
|
13
|
+
* - Budget alerts at configurable thresholds (50%, 80%, 90%, 100%)
|
|
14
|
+
*
|
|
15
|
+
* Architecture:
|
|
16
|
+
* [Local MCP] → Budget Check → [Enterprise Server] → Shutdown Signal
|
|
17
|
+
*
|
|
18
|
+
* Environment Variables:
|
|
19
|
+
* ATABEY_BUDGET_TEAM - Team name for budget grouping
|
|
20
|
+
* ATABEY_BUDGET_MONTHLY - Monthly budget in USD (local override)
|
|
21
|
+
* ATABEY_BUDGET_AGENT_MAX - Max spend per agent in USD
|
|
22
|
+
* ATABEY_BUDGET_SYNC_URL - Enterprise server URL for budget sync
|
|
23
|
+
*/
|
|
24
|
+
declare const CONFIG: {
|
|
25
|
+
/** Team name for budget grouping */
|
|
26
|
+
TEAM: string;
|
|
27
|
+
/** Monthly budget in USD (0 = unlimited) */
|
|
28
|
+
MONTHLY_BUDGET: number;
|
|
29
|
+
/** Max spend per agent in USD (0 = unlimited) */
|
|
30
|
+
AGENT_MAX_BUDGET: number;
|
|
31
|
+
/** Enterprise server URL for budget policy sync */
|
|
32
|
+
SYNC_URL: string;
|
|
33
|
+
/** Sync token */
|
|
34
|
+
SYNC_TOKEN: string;
|
|
35
|
+
/** Budget alert thresholds (percentages) */
|
|
36
|
+
ALERT_THRESHOLDS: number[];
|
|
37
|
+
/** Cost per 1K tokens (USD) – default: $0.003 for Claude */
|
|
38
|
+
COST_PER_1K_TOKENS: number;
|
|
39
|
+
/** Sync interval in ms */
|
|
40
|
+
SYNC_INTERVAL_MS: number;
|
|
41
|
+
/** Enable/disable budget enforcement */
|
|
42
|
+
ENABLED: boolean;
|
|
43
|
+
};
|
|
44
|
+
export interface BudgetState {
|
|
45
|
+
team: string;
|
|
46
|
+
/** Total spend this month in USD */
|
|
47
|
+
monthlySpend: number;
|
|
48
|
+
/** Per-agent spend this month */
|
|
49
|
+
agentSpend: Record<string, number>;
|
|
50
|
+
/** Budget period start (ISO date) */
|
|
51
|
+
periodStart: string;
|
|
52
|
+
/** Budget period end (ISO date) */
|
|
53
|
+
periodEnd: string;
|
|
54
|
+
/** Whether this team/agent is blocked */
|
|
55
|
+
blocked: boolean;
|
|
56
|
+
/** Block reason if blocked */
|
|
57
|
+
blockReason: string | null;
|
|
58
|
+
/** Last alert thresholds triggered */
|
|
59
|
+
triggeredAlerts: number[];
|
|
60
|
+
/** Last sync timestamp */
|
|
61
|
+
lastSync: string | null;
|
|
62
|
+
}
|
|
63
|
+
interface BudgetPolicy {
|
|
64
|
+
team: string;
|
|
65
|
+
monthlyBudget: number;
|
|
66
|
+
agentMaxBudget: number;
|
|
67
|
+
blocked: boolean;
|
|
68
|
+
blockReason: string | null;
|
|
69
|
+
updatedAt: string;
|
|
70
|
+
}
|
|
71
|
+
export declare class BudgetManager {
|
|
72
|
+
private static instance;
|
|
73
|
+
private state;
|
|
74
|
+
private syncTimer;
|
|
75
|
+
private localMetricsPath;
|
|
76
|
+
private constructor();
|
|
77
|
+
static getInstance(): BudgetManager;
|
|
78
|
+
/**
|
|
79
|
+
* Start the budget manager.
|
|
80
|
+
* Begins periodic sync with enterprise server.
|
|
81
|
+
*/
|
|
82
|
+
start(): void;
|
|
83
|
+
/**
|
|
84
|
+
* Stop the budget manager.
|
|
85
|
+
*/
|
|
86
|
+
stop(): void;
|
|
87
|
+
/**
|
|
88
|
+
* Record token usage and check budget.
|
|
89
|
+
* Returns error message if budget exceeded, null if allowed.
|
|
90
|
+
*/
|
|
91
|
+
recordUsage(agent: string, tokens: number): string | null;
|
|
92
|
+
/**
|
|
93
|
+
* Check if an agent is within budget without recording usage.
|
|
94
|
+
*/
|
|
95
|
+
checkBudget(agent: string): {
|
|
96
|
+
allowed: boolean;
|
|
97
|
+
monthlySpend: number;
|
|
98
|
+
monthlyBudget: number;
|
|
99
|
+
agentSpend: number;
|
|
100
|
+
agentMaxBudget: number;
|
|
101
|
+
usagePercent: number;
|
|
102
|
+
blocked: boolean;
|
|
103
|
+
blockReason: string | null;
|
|
104
|
+
};
|
|
105
|
+
/**
|
|
106
|
+
* Get full budget state.
|
|
107
|
+
*/
|
|
108
|
+
getState(): BudgetState;
|
|
109
|
+
/**
|
|
110
|
+
* Reset budget for new period.
|
|
111
|
+
*/
|
|
112
|
+
resetPeriod(): void;
|
|
113
|
+
/**
|
|
114
|
+
* Apply a budget policy from the enterprise server.
|
|
115
|
+
*/
|
|
116
|
+
applyPolicy(policy: BudgetPolicy): void;
|
|
117
|
+
/**
|
|
118
|
+
* Sync budget state with enterprise server.
|
|
119
|
+
*/
|
|
120
|
+
private syncWithServer;
|
|
121
|
+
/**
|
|
122
|
+
* Recalculate spend from local metrics file.
|
|
123
|
+
*/
|
|
124
|
+
private recalculateFromLocalMetrics;
|
|
125
|
+
/**
|
|
126
|
+
* Trigger a budget alert.
|
|
127
|
+
*/
|
|
128
|
+
private triggerAlert;
|
|
129
|
+
/**
|
|
130
|
+
* Save budget state to disk.
|
|
131
|
+
*/
|
|
132
|
+
private saveState;
|
|
133
|
+
/**
|
|
134
|
+
* Reset the singleton instance (for testing).
|
|
135
|
+
*/
|
|
136
|
+
static resetInstance(): void;
|
|
137
|
+
/**
|
|
138
|
+
* Load budget state from disk.
|
|
139
|
+
*/
|
|
140
|
+
private loadState;
|
|
141
|
+
}
|
|
142
|
+
export declare const budgetManager: BudgetManager;
|
|
143
|
+
export { CONFIG as FinOpsConfig };
|
|
@@ -0,0 +1,326 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ─── FINOPS / BUDGET ENFORCEMENT ──────────────────────────────────
|
|
3
|
+
*
|
|
4
|
+
* Enterprise budget management for AI token consumption.
|
|
5
|
+
* Enforces strict cost limits per team, agent, or project.
|
|
6
|
+
*
|
|
7
|
+
* Features:
|
|
8
|
+
* - Team-based budget allocation (monthly/quarterly)
|
|
9
|
+
* - Per-agent spending limits
|
|
10
|
+
* - Automatic shutdown when budget exceeded
|
|
11
|
+
* - Central server sync for budget policies
|
|
12
|
+
* - Real-time cost tracking
|
|
13
|
+
* - Budget alerts at configurable thresholds (50%, 80%, 90%, 100%)
|
|
14
|
+
*
|
|
15
|
+
* Architecture:
|
|
16
|
+
* [Local MCP] → Budget Check → [Enterprise Server] → Shutdown Signal
|
|
17
|
+
*
|
|
18
|
+
* Environment Variables:
|
|
19
|
+
* ATABEY_BUDGET_TEAM - Team name for budget grouping
|
|
20
|
+
* ATABEY_BUDGET_MONTHLY - Monthly budget in USD (local override)
|
|
21
|
+
* ATABEY_BUDGET_AGENT_MAX - Max spend per agent in USD
|
|
22
|
+
* ATABEY_BUDGET_SYNC_URL - Enterprise server URL for budget sync
|
|
23
|
+
*/
|
|
24
|
+
import fs from "fs";
|
|
25
|
+
import path from "path";
|
|
26
|
+
// ─── Configuration ────────────────────────────────────────────────
|
|
27
|
+
const CONFIG = {
|
|
28
|
+
/** Team name for budget grouping */
|
|
29
|
+
TEAM: process.env.ATABEY_BUDGET_TEAM || "default",
|
|
30
|
+
/** Monthly budget in USD (0 = unlimited) */
|
|
31
|
+
MONTHLY_BUDGET: parseFloat(process.env.ATABEY_BUDGET_MONTHLY || "0"),
|
|
32
|
+
/** Max spend per agent in USD (0 = unlimited) */
|
|
33
|
+
AGENT_MAX_BUDGET: parseFloat(process.env.ATABEY_BUDGET_AGENT_MAX || "0"),
|
|
34
|
+
/** Enterprise server URL for budget policy sync */
|
|
35
|
+
SYNC_URL: process.env.ATABEY_BUDGET_SYNC_URL || "",
|
|
36
|
+
/** Sync token */
|
|
37
|
+
SYNC_TOKEN: process.env.ATABEY_BUDGET_SYNC_TOKEN || "",
|
|
38
|
+
/** Budget alert thresholds (percentages) */
|
|
39
|
+
ALERT_THRESHOLDS: (process.env.ATABEY_BUDGET_ALERT_THRESHOLDS || "50,80,90,100")
|
|
40
|
+
.split(",").map(t => parseInt(t.trim(), 10)).filter(t => !isNaN(t)),
|
|
41
|
+
/** Cost per 1K tokens (USD) – default: $0.003 for Claude */
|
|
42
|
+
COST_PER_1K_TOKENS: parseFloat(process.env.ATABEY_COST_PER_1K_TOKENS || "0.003"),
|
|
43
|
+
/** Sync interval in ms */
|
|
44
|
+
SYNC_INTERVAL_MS: parseInt(process.env.ATABEY_BUDGET_SYNC_INTERVAL || "60000", 10),
|
|
45
|
+
/** Enable/disable budget enforcement */
|
|
46
|
+
ENABLED: process.env.ATABEY_BUDGET_ENABLED === "true",
|
|
47
|
+
};
|
|
48
|
+
// ─── Budget Manager ───────────────────────────────────────────────
|
|
49
|
+
export class BudgetManager {
|
|
50
|
+
static instance;
|
|
51
|
+
state;
|
|
52
|
+
syncTimer = null;
|
|
53
|
+
localMetricsPath;
|
|
54
|
+
constructor() {
|
|
55
|
+
const now = new Date();
|
|
56
|
+
const periodStart = new Date(now.getFullYear(), now.getMonth(), 1).toISOString();
|
|
57
|
+
const periodEnd = new Date(now.getFullYear(), now.getMonth() + 1, 0).toISOString();
|
|
58
|
+
this.state = {
|
|
59
|
+
team: CONFIG.TEAM,
|
|
60
|
+
monthlySpend: 0,
|
|
61
|
+
agentSpend: {},
|
|
62
|
+
periodStart,
|
|
63
|
+
periodEnd,
|
|
64
|
+
blocked: false,
|
|
65
|
+
blockReason: null,
|
|
66
|
+
triggeredAlerts: [],
|
|
67
|
+
lastSync: null,
|
|
68
|
+
};
|
|
69
|
+
const frameworkDir = process.env.ATABEY_FRAMEWORK_DIR || path.join(process.cwd(), ".atabey");
|
|
70
|
+
this.localMetricsPath = path.join(frameworkDir, "observability", "metrics.json");
|
|
71
|
+
// Load existing state
|
|
72
|
+
this.loadState();
|
|
73
|
+
}
|
|
74
|
+
static getInstance() {
|
|
75
|
+
if (!BudgetManager.instance) {
|
|
76
|
+
BudgetManager.instance = new BudgetManager();
|
|
77
|
+
}
|
|
78
|
+
return BudgetManager.instance;
|
|
79
|
+
}
|
|
80
|
+
/**
|
|
81
|
+
* Start the budget manager.
|
|
82
|
+
* Begins periodic sync with enterprise server.
|
|
83
|
+
*/
|
|
84
|
+
start() {
|
|
85
|
+
if (!CONFIG.ENABLED) {
|
|
86
|
+
process.stderr.write("[FINOPS] Budget enforcement disabled.\n");
|
|
87
|
+
return;
|
|
88
|
+
}
|
|
89
|
+
process.stderr.write(`[FINOPS] Starting budget manager for team "${CONFIG.TEAM}"\n`);
|
|
90
|
+
if (CONFIG.MONTHLY_BUDGET > 0) {
|
|
91
|
+
process.stderr.write(`[FINOPS] Monthly budget: $${CONFIG.MONTHLY_BUDGET.toFixed(2)}\n`);
|
|
92
|
+
}
|
|
93
|
+
// Periodic sync with enterprise server
|
|
94
|
+
if (CONFIG.SYNC_URL) {
|
|
95
|
+
this.syncTimer = setInterval(() => this.syncWithServer(), CONFIG.SYNC_INTERVAL_MS);
|
|
96
|
+
this.syncWithServer(); // Initial sync
|
|
97
|
+
}
|
|
98
|
+
// Recalculate from local metrics
|
|
99
|
+
this.recalculateFromLocalMetrics();
|
|
100
|
+
}
|
|
101
|
+
/**
|
|
102
|
+
* Stop the budget manager.
|
|
103
|
+
*/
|
|
104
|
+
stop() {
|
|
105
|
+
if (this.syncTimer) {
|
|
106
|
+
clearInterval(this.syncTimer);
|
|
107
|
+
this.syncTimer = null;
|
|
108
|
+
}
|
|
109
|
+
this.saveState();
|
|
110
|
+
}
|
|
111
|
+
/**
|
|
112
|
+
* Record token usage and check budget.
|
|
113
|
+
* Returns error message if budget exceeded, null if allowed.
|
|
114
|
+
*/
|
|
115
|
+
recordUsage(agent, tokens) {
|
|
116
|
+
if (!CONFIG.ENABLED)
|
|
117
|
+
return null;
|
|
118
|
+
const cost = (tokens / 1000) * CONFIG.COST_PER_1K_TOKENS;
|
|
119
|
+
// Check if team is blocked
|
|
120
|
+
if (this.state.blocked) {
|
|
121
|
+
return `[FINOPS] ⛔ Budget exceeded: ${this.state.blockReason}`;
|
|
122
|
+
}
|
|
123
|
+
// Update spend
|
|
124
|
+
this.state.monthlySpend += cost;
|
|
125
|
+
this.state.agentSpend[agent] = (this.state.agentSpend[agent] || 0) + cost;
|
|
126
|
+
// Check team budget
|
|
127
|
+
if (CONFIG.MONTHLY_BUDGET > 0 && this.state.monthlySpend > CONFIG.MONTHLY_BUDGET) {
|
|
128
|
+
this.state.blocked = true;
|
|
129
|
+
this.state.blockReason = `Team "${CONFIG.TEAM}" exceeded monthly budget of $${CONFIG.MONTHLY_BUDGET.toFixed(2)}`;
|
|
130
|
+
this.saveState();
|
|
131
|
+
this.triggerAlert(100);
|
|
132
|
+
return `[FINOPS] ⛔ ${this.state.blockReason}`;
|
|
133
|
+
}
|
|
134
|
+
// Check agent budget
|
|
135
|
+
if (CONFIG.AGENT_MAX_BUDGET > 0 && (this.state.agentSpend[agent] || 0) > CONFIG.AGENT_MAX_BUDGET) {
|
|
136
|
+
this.state.blocked = true;
|
|
137
|
+
this.state.blockReason = `Agent "${agent}" exceeded max budget of $${CONFIG.AGENT_MAX_BUDGET.toFixed(2)}`;
|
|
138
|
+
this.saveState();
|
|
139
|
+
this.triggerAlert(100);
|
|
140
|
+
return `[FINOPS] ⛔ ${this.state.blockReason}`;
|
|
141
|
+
}
|
|
142
|
+
// Check alert thresholds
|
|
143
|
+
if (CONFIG.MONTHLY_BUDGET > 0) {
|
|
144
|
+
const usagePercent = (this.state.monthlySpend / CONFIG.MONTHLY_BUDGET) * 100;
|
|
145
|
+
for (const threshold of CONFIG.ALERT_THRESHOLDS) {
|
|
146
|
+
if (usagePercent >= threshold && !this.state.triggeredAlerts.includes(threshold)) {
|
|
147
|
+
this.triggerAlert(threshold);
|
|
148
|
+
process.stderr.write(`[FINOPS] ⚠️ Budget alert: ${Math.round(usagePercent)}% used ($${this.state.monthlySpend.toFixed(2)} / $${CONFIG.MONTHLY_BUDGET.toFixed(2)})\n`);
|
|
149
|
+
}
|
|
150
|
+
}
|
|
151
|
+
}
|
|
152
|
+
this.saveState();
|
|
153
|
+
return null;
|
|
154
|
+
}
|
|
155
|
+
/**
|
|
156
|
+
* Check if an agent is within budget without recording usage.
|
|
157
|
+
*/
|
|
158
|
+
checkBudget(agent) {
|
|
159
|
+
const agentSpend = this.state.agentSpend[agent] || 0;
|
|
160
|
+
const usagePercent = CONFIG.MONTHLY_BUDGET > 0
|
|
161
|
+
? (this.state.monthlySpend / CONFIG.MONTHLY_BUDGET) * 100
|
|
162
|
+
: 0;
|
|
163
|
+
return {
|
|
164
|
+
allowed: !this.state.blocked,
|
|
165
|
+
monthlySpend: this.state.monthlySpend,
|
|
166
|
+
monthlyBudget: CONFIG.MONTHLY_BUDGET,
|
|
167
|
+
agentSpend,
|
|
168
|
+
agentMaxBudget: CONFIG.AGENT_MAX_BUDGET,
|
|
169
|
+
usagePercent: Math.round(usagePercent * 100) / 100,
|
|
170
|
+
blocked: this.state.blocked,
|
|
171
|
+
blockReason: this.state.blockReason,
|
|
172
|
+
};
|
|
173
|
+
}
|
|
174
|
+
/**
|
|
175
|
+
* Get full budget state.
|
|
176
|
+
*/
|
|
177
|
+
getState() {
|
|
178
|
+
return { ...this.state };
|
|
179
|
+
}
|
|
180
|
+
/**
|
|
181
|
+
* Reset budget for new period.
|
|
182
|
+
*/
|
|
183
|
+
resetPeriod() {
|
|
184
|
+
const now = new Date();
|
|
185
|
+
this.state.monthlySpend = 0;
|
|
186
|
+
this.state.agentSpend = {};
|
|
187
|
+
this.state.periodStart = new Date(now.getFullYear(), now.getMonth(), 1).toISOString();
|
|
188
|
+
this.state.periodEnd = new Date(now.getFullYear(), now.getMonth() + 1, 0).toISOString();
|
|
189
|
+
this.state.blocked = false;
|
|
190
|
+
this.state.blockReason = null;
|
|
191
|
+
this.state.triggeredAlerts = [];
|
|
192
|
+
this.saveState();
|
|
193
|
+
process.stderr.write("[FINOPS] Budget period reset.\n");
|
|
194
|
+
}
|
|
195
|
+
/**
|
|
196
|
+
* Apply a budget policy from the enterprise server.
|
|
197
|
+
*/
|
|
198
|
+
applyPolicy(policy) {
|
|
199
|
+
if (policy.team !== CONFIG.TEAM)
|
|
200
|
+
return;
|
|
201
|
+
if (policy.blocked) {
|
|
202
|
+
this.state.blocked = true;
|
|
203
|
+
this.state.blockReason = policy.blockReason || "Blocked by enterprise policy";
|
|
204
|
+
process.stderr.write(`[FINOPS] 🛑 Team "${CONFIG.TEAM}" blocked by enterprise policy: ${this.state.blockReason}\n`);
|
|
205
|
+
}
|
|
206
|
+
this.saveState();
|
|
207
|
+
}
|
|
208
|
+
// ─── Private Methods ───────────────────────────────────────
|
|
209
|
+
/**
|
|
210
|
+
* Sync budget state with enterprise server.
|
|
211
|
+
*/
|
|
212
|
+
async syncWithServer() {
|
|
213
|
+
if (!CONFIG.SYNC_URL)
|
|
214
|
+
return;
|
|
215
|
+
try {
|
|
216
|
+
// Send local state to server
|
|
217
|
+
const url = `${CONFIG.SYNC_URL.replace(/\/$/, "")}/api/finops/sync`;
|
|
218
|
+
const response = await fetch(url, {
|
|
219
|
+
method: "POST",
|
|
220
|
+
headers: {
|
|
221
|
+
"Content-Type": "application/json",
|
|
222
|
+
"Authorization": `Bearer ${CONFIG.SYNC_TOKEN}`,
|
|
223
|
+
"X-Team": CONFIG.TEAM,
|
|
224
|
+
},
|
|
225
|
+
body: JSON.stringify({
|
|
226
|
+
team: this.state.team,
|
|
227
|
+
monthlySpend: this.state.monthlySpend,
|
|
228
|
+
agentSpend: this.state.agentSpend,
|
|
229
|
+
periodStart: this.state.periodStart,
|
|
230
|
+
periodEnd: this.state.periodEnd,
|
|
231
|
+
}),
|
|
232
|
+
});
|
|
233
|
+
if (response.ok) {
|
|
234
|
+
const policy = (await response.json());
|
|
235
|
+
this.applyPolicy(policy);
|
|
236
|
+
this.state.lastSync = new Date().toISOString();
|
|
237
|
+
}
|
|
238
|
+
else if (response.status === 402) {
|
|
239
|
+
// Payment required – budget exceeded on server side
|
|
240
|
+
const data = (await response.json());
|
|
241
|
+
this.state.blocked = true;
|
|
242
|
+
this.state.blockReason = data.reason || "Budget exceeded on enterprise server";
|
|
243
|
+
process.stderr.write(`[FINOPS] 🛑 Enterprise budget exceeded: ${this.state.blockReason}\n`);
|
|
244
|
+
}
|
|
245
|
+
}
|
|
246
|
+
catch (error) {
|
|
247
|
+
// Server unreachable – continue with local budget
|
|
248
|
+
process.stderr.write(`[FINOPS] Sync failed (server unreachable): ${error}\n`);
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
/**
|
|
252
|
+
* Recalculate spend from local metrics file.
|
|
253
|
+
*/
|
|
254
|
+
recalculateFromLocalMetrics() {
|
|
255
|
+
try {
|
|
256
|
+
if (!fs.existsSync(this.localMetricsPath))
|
|
257
|
+
return;
|
|
258
|
+
const metrics = JSON.parse(fs.readFileSync(this.localMetricsPath, "utf8"));
|
|
259
|
+
const periodStart = new Date(this.state.periodStart);
|
|
260
|
+
const periodEnd = new Date(this.state.periodEnd);
|
|
261
|
+
let totalSpend = 0;
|
|
262
|
+
const agentSpend = {};
|
|
263
|
+
for (const entry of metrics) {
|
|
264
|
+
const entryDate = new Date(entry.timestamp);
|
|
265
|
+
if (entryDate >= periodStart && entryDate <= periodEnd) {
|
|
266
|
+
const cost = (entry.estimatedTokens / 1000) * CONFIG.COST_PER_1K_TOKENS;
|
|
267
|
+
totalSpend += cost;
|
|
268
|
+
agentSpend[entry.agent] = (agentSpend[entry.agent] || 0) + cost;
|
|
269
|
+
}
|
|
270
|
+
}
|
|
271
|
+
this.state.monthlySpend = totalSpend;
|
|
272
|
+
this.state.agentSpend = agentSpend;
|
|
273
|
+
}
|
|
274
|
+
catch { /* ignore */ }
|
|
275
|
+
}
|
|
276
|
+
/**
|
|
277
|
+
* Trigger a budget alert.
|
|
278
|
+
*/
|
|
279
|
+
triggerAlert(threshold) {
|
|
280
|
+
if (!this.state.triggeredAlerts.includes(threshold)) {
|
|
281
|
+
this.state.triggeredAlerts.push(threshold);
|
|
282
|
+
}
|
|
283
|
+
}
|
|
284
|
+
/**
|
|
285
|
+
* Save budget state to disk.
|
|
286
|
+
*/
|
|
287
|
+
saveState() {
|
|
288
|
+
try {
|
|
289
|
+
const frameworkDir = process.env.ATABEY_FRAMEWORK_DIR || path.join(process.cwd(), ".atabey");
|
|
290
|
+
const finopsDir = path.join(frameworkDir, "finops");
|
|
291
|
+
if (!fs.existsSync(finopsDir)) {
|
|
292
|
+
fs.mkdirSync(finopsDir, { recursive: true });
|
|
293
|
+
}
|
|
294
|
+
const statePath = path.join(finopsDir, "budget-state.json");
|
|
295
|
+
fs.writeFileSync(statePath, JSON.stringify(this.state, null, 2));
|
|
296
|
+
}
|
|
297
|
+
catch { /* ignore */ }
|
|
298
|
+
}
|
|
299
|
+
/**
|
|
300
|
+
* Reset the singleton instance (for testing).
|
|
301
|
+
*/
|
|
302
|
+
static resetInstance() {
|
|
303
|
+
BudgetManager.instance = undefined;
|
|
304
|
+
}
|
|
305
|
+
/**
|
|
306
|
+
* Load budget state from disk.
|
|
307
|
+
*/
|
|
308
|
+
loadState() {
|
|
309
|
+
try {
|
|
310
|
+
const frameworkDir = process.env.ATABEY_FRAMEWORK_DIR || path.join(process.cwd(), ".atabey");
|
|
311
|
+
const statePath = path.join(frameworkDir, "finops", "budget-state.json");
|
|
312
|
+
if (fs.existsSync(statePath)) {
|
|
313
|
+
const saved = JSON.parse(fs.readFileSync(statePath, "utf8"));
|
|
314
|
+
// Only load if same period
|
|
315
|
+
if (saved.periodStart === this.state.periodStart) {
|
|
316
|
+
this.state = saved;
|
|
317
|
+
}
|
|
318
|
+
}
|
|
319
|
+
}
|
|
320
|
+
catch { /* ignore */ }
|
|
321
|
+
}
|
|
322
|
+
}
|
|
323
|
+
// ─── Singleton Export ─────────────────────────────────────────────
|
|
324
|
+
export const budgetManager = BudgetManager.getInstance();
|
|
325
|
+
export { CONFIG as FinOpsConfig };
|
|
326
|
+
//# sourceMappingURL=finops.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"finops.js","sourceRoot":"","sources":["../../../../framework-mcp/src/utils/finops.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;GAsBG;AAEH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,qEAAqE;AAErE,MAAM,MAAM,GAAG;IACX,oCAAoC;IACpC,IAAI,EAAE,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,SAAS;IACjD,4CAA4C;IAC5C,cAAc,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,GAAG,CAAC;IACpE,iDAAiD;IACjD,gBAAgB,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,GAAG,CAAC;IACxE,mDAAmD;IACnD,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,EAAE;IAClD,iBAAiB;IACjB,UAAU,EAAE,OAAO,CAAC,GAAG,CAAC,wBAAwB,IAAI,EAAE;IACtD,4CAA4C;IAC5C,gBAAgB,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,cAAc,CAAC;SAC3E,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACvE,4DAA4D;IAC5D,kBAAkB,EAAE,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,OAAO,CAAC;IAChF,0BAA0B;IAC1B,gBAAgB,EAAE,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,OAAO,EAAE,EAAE,CAAC;IAClF,wCAAwC;IACxC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,MAAM;CACxD,CAAC;AAiCF,qEAAqE;AAErE,MAAM,OAAO,aAAa;IACd,MAAM,CAAC,QAAQ,CAAgB;IAC/B,KAAK,CAAc;IACnB,SAAS,GAA0C,IAAI,CAAC;IACxD,gBAAgB,CAAS;IAEjC;QACI,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACjF,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QAEnF,IAAI,CAAC,KAAK,GAAG;YACT,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,YAAY,EAAE,CAAC;YACf,UAAU,EAAE,EAAE;YACd,WAAW;YACX,SAAS;YACT,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,IAAI;YACjB,eAAe,EAAE,EAAE;YACnB,QAAQ,EAAE,IAAI;SACjB,CAAC;QAEF,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,CAAC,CAAC;QAC7F,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC;QAEjF,sBAAsB;QACtB,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAEM,MAAM,CAAC,WAAW;QACrB,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;YAC1B,aAAa,CAAC,QAAQ,GAAG,IAAI,aAAa,EAAE,CAAC;QACjD,CAAC;QACD,OAAO,aAAa,CAAC,QAAQ,CAAC;IAClC,CAAC;IAED;;;OAGG;IACI,KAAK;QACR,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAChE,OAAO;QACX,CAAC;QAED,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,8CAA8C,MAAM,CAAC,IAAI,KAAK,CAAC,CAAC;QACrF,IAAI,MAAM,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC5F,CAAC;QAED,uCAAuC;QACvC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACnF,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,eAAe;QAC1C,CAAC;QAED,iCAAiC;QACjC,IAAI,CAAC,2BAA2B,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACI,IAAI;QACP,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC9B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAED;;;OAGG;IACI,WAAW,CAAC,KAAa,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QAEjC,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,MAAM,CAAC,kBAAkB,CAAC;QAEzD,2BAA2B;QAC3B,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACrB,OAAO,+BAA+B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;QACnE,CAAC;QAED,eAAe;QACf,IAAI,CAAC,KAAK,CAAC,YAAY,IAAI,IAAI,CAAC;QAChC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC;QAE1E,oBAAoB;QACpB,IAAI,MAAM,CAAC,cAAc,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;YAC/E,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,SAAS,MAAM,CAAC,IAAI,iCAAiC,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YACjH,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACvB,OAAO,cAAc,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;QAClD,CAAC;QAED,qBAAqB;QACrB,IAAI,MAAM,CAAC,gBAAgB,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC/F,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,UAAU,KAAK,6BAA6B,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;YAC1G,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACvB,OAAO,cAAc,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;QAClD,CAAC;QAED,yBAAyB;QACzB,IAAI,MAAM,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,YAAY,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,cAAc,CAAC,GAAG,GAAG,CAAC;YAC7E,KAAK,MAAM,SAAS,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;gBAC9C,IAAI,YAAY,IAAI,SAAS,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;oBAC/E,IAAI,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;oBAC7B,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,YAAY,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;gBAC1K,CAAC;YACL,CAAC;QACL,CAAC;QAED,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,OAAO,IAAI,CAAC;IAChB,CAAC;IAED;;OAEG;IACI,WAAW,CAAC,KAAa;QAU5B,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACrD,MAAM,YAAY,GAAG,MAAM,CAAC,cAAc,GAAG,CAAC;YAC1C,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,cAAc,CAAC,GAAG,GAAG;YACzD,CAAC,CAAC,CAAC,CAAC;QAER,OAAO;YACH,OAAO,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO;YAC5B,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;YACrC,aAAa,EAAE,MAAM,CAAC,cAAc;YACpC,UAAU;YACV,cAAc,EAAE,MAAM,CAAC,gBAAgB;YACvC,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,GAAG,CAAC,GAAG,GAAG;YAClD,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YAC3B,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;SACtC,CAAC;IACN,CAAC;IAED;;OAEG;IACI,QAAQ;QACX,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACI,WAAW;QACd,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC;QAC5B,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACtF,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACxF,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACI,WAAW,CAAC,MAAoB;QACnC,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI;YAAE,OAAO;QAExC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,8BAA8B,CAAC;YAC9E,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,MAAM,CAAC,IAAI,mCAAmC,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;QACxH,CAAC;QAED,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAED,8DAA8D;IAE9D;;OAEG;IACK,KAAK,CAAC,cAAc;QACxB,IAAI,CAAC,MAAM,CAAC,QAAQ;YAAE,OAAO;QAE7B,IAAI,CAAC;YACD,6BAA6B;YAC7B,MAAM,GAAG,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,kBAAkB,CAAC;YACpE,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC9B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,MAAM,CAAC,UAAU,EAAE;oBAC9C,QAAQ,EAAE,MAAM,CAAC,IAAI;iBACxB;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACjB,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;oBACrB,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;oBACrC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU;oBACjC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;oBACnC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS;iBAClC,CAAC;aACL,CAAC,CAAC;YAEH,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACd,MAAM,MAAM,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAiB,CAAC;gBACvD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBACzB,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACnD,CAAC;iBAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACjC,oDAAoD;gBACpD,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAwB,CAAC;gBAC5D,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC;gBAC1B,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,MAAM,IAAI,sCAAsC,CAAC;gBAC/E,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;YAChG,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,kDAAkD;YAClD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,8CAA8C,KAAK,IAAI,CAAC,CAAC;QAClF,CAAC;IACL,CAAC;IAED;;OAEG;IACK,2BAA2B;QAC/B,IAAI,CAAC;YACD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAAE,OAAO;YAElD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAIvE,CAAC;YAEH,MAAM,WAAW,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YACrD,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAEjD,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,MAAM,UAAU,GAA2B,EAAE,CAAC;YAE9C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC1B,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC5C,IAAI,SAAS,IAAI,WAAW,IAAI,SAAS,IAAI,SAAS,EAAE,CAAC;oBACrD,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,MAAM,CAAC,kBAAkB,CAAC;oBACxE,UAAU,IAAI,IAAI,CAAC;oBACnB,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC;gBACpE,CAAC;YACL,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,UAAU,CAAC;YACrC,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,UAAU,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,SAAiB;QAClC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAClD,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC;IAED;;OAEG;IACK,SAAS;QACb,IAAI,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,CAAC,CAAC;YAC7F,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;YACpD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC5B,EAAE,CAAC,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;YAC5D,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACrE,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,aAAa;QACvB,aAAa,CAAC,QAAQ,GAAG,SAAqC,CAAC;IACnE,CAAC;IAED;;OAEG;IACK,SAAS;QACb,IAAI,CAAC;YACD,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,CAAC,CAAC;YAC7F,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC;YACzE,IAAI,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAgB,CAAC;gBAC5E,2BAA2B;gBAC3B,IAAI,KAAK,CAAC,WAAW,KAAK,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;oBAC/C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;gBACvB,CAAC;YACL,CAAC;QACL,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IAC5B,CAAC;CACJ;AAED,qEAAqE;AAErE,MAAM,CAAC,MAAM,aAAa,GAAG,aAAa,CAAC,WAAW,EAAE,CAAC;AACzD,OAAO,EAAE,MAAM,IAAI,YAAY,EAAE,CAAC"}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* ─── HUMAN-IN-THE-LOOP (In-CLI Approval) ──────────────────────────
|
|
3
|
+
*
|
|
4
|
+
* Allows risk-gated approvals without the developer leaving their AI CLI chat.
|
|
5
|
+
*
|
|
6
|
+
* Key Innovation: Instead of forcing developers to switch to a web dashboard
|
|
7
|
+
* or another terminal, approval requests are sent as MCP notifications into
|
|
8
|
+
* the developer's existing chat interface.
|
|
9
|
+
*
|
|
10
|
+
* Flow:
|
|
11
|
+
* 1. Tool call triggers risk detection
|
|
12
|
+
* 2. If risk >= threshold, tool is blocked with a structured error
|
|
13
|
+
* 3. Error message includes the traceId and approval command
|
|
14
|
+
* 4. Developer types "atabey approve <traceId>" in the same chat
|
|
15
|
+
* 5. Hermes message status updates → retry succeeds
|
|
16
|
+
*
|
|
17
|
+
* Risk Levels:
|
|
18
|
+
* - LOW (0-30): Silent pass (no approval needed)
|
|
19
|
+
* - MEDIUM (31-59): Stealth warning to stderr only
|
|
20
|
+
* - HIGH (60-100): Blocked until approval
|
|
21
|
+
*/
|
|
22
|
+
declare const CONFIG: {
|
|
23
|
+
/** Risk threshold for blocking (0-100) */
|
|
24
|
+
HIGH_RISK_THRESHOLD: number;
|
|
25
|
+
/** Risk threshold for medium warning */
|
|
26
|
+
MEDIUM_RISK_THRESHOLD: number;
|
|
27
|
+
/** Auto-approve low risk operations */
|
|
28
|
+
AUTO_APPROVE_LOW_RISK: boolean;
|
|
29
|
+
/** Approval timeout in seconds */
|
|
30
|
+
APPROVAL_TIMEOUT_SECONDS: number;
|
|
31
|
+
};
|
|
32
|
+
export type RiskLevel = "LOW" | "MEDIUM" | "HIGH";
|
|
33
|
+
export interface ApprovalRequest {
|
|
34
|
+
traceId: string;
|
|
35
|
+
toolName: string;
|
|
36
|
+
agent: string;
|
|
37
|
+
riskScore: number;
|
|
38
|
+
riskLevel: RiskLevel;
|
|
39
|
+
reason: string;
|
|
40
|
+
args: Record<string, unknown>;
|
|
41
|
+
createdAt: number;
|
|
42
|
+
expiresAt: number;
|
|
43
|
+
status: "PENDING" | "APPROVED" | "REJECTED" | "EXPIRED";
|
|
44
|
+
}
|
|
45
|
+
/**
|
|
46
|
+
* Determine risk level from risk score.
|
|
47
|
+
*/
|
|
48
|
+
export declare function getRiskLevel(score: number): RiskLevel;
|
|
49
|
+
/**
|
|
50
|
+
* Create an approval request for a high-risk operation.
|
|
51
|
+
* Returns the approval request and a message for the AI/developer.
|
|
52
|
+
*/
|
|
53
|
+
export declare function createApprovalRequest(traceId: string, toolName: string, agent: string, riskScore: number, reason: string, args: Record<string, unknown>): {
|
|
54
|
+
request: ApprovalRequest;
|
|
55
|
+
message: string;
|
|
56
|
+
};
|
|
57
|
+
/**
|
|
58
|
+
* Check if an operation should be blocked based on risk score.
|
|
59
|
+
* Returns:
|
|
60
|
+
* - null if allowed (LOW risk or auto-approved)
|
|
61
|
+
* - { blocked: true, message } if blocked (HIGH risk)
|
|
62
|
+
* - { blocked: false, warning } if warning only (MEDIUM risk)
|
|
63
|
+
*/
|
|
64
|
+
export declare function checkRiskGate(traceId: string, toolName: string, agent: string, riskScore: number, reason: string, args: Record<string, unknown>): {
|
|
65
|
+
blocked: boolean;
|
|
66
|
+
message: string | null;
|
|
67
|
+
warning?: string;
|
|
68
|
+
} | null;
|
|
69
|
+
/**
|
|
70
|
+
* Check if an approval request exists and is still valid.
|
|
71
|
+
*/
|
|
72
|
+
export declare function getApprovalStatus(traceId: string): ApprovalRequest | null;
|
|
73
|
+
/**
|
|
74
|
+
* Approve a pending operation.
|
|
75
|
+
*/
|
|
76
|
+
export declare function approveOperation(traceId: string): {
|
|
77
|
+
success: boolean;
|
|
78
|
+
message: string;
|
|
79
|
+
};
|
|
80
|
+
/**
|
|
81
|
+
* Reject a pending operation.
|
|
82
|
+
*/
|
|
83
|
+
export declare function rejectOperation(traceId: string): {
|
|
84
|
+
success: boolean;
|
|
85
|
+
message: string;
|
|
86
|
+
};
|
|
87
|
+
/**
|
|
88
|
+
* Get all pending approval requests.
|
|
89
|
+
*/
|
|
90
|
+
export declare function getPendingApprovals(): ApprovalRequest[];
|
|
91
|
+
/**
|
|
92
|
+
* Get HITL config.
|
|
93
|
+
*/
|
|
94
|
+
export declare function getHumanInLoopConfig(): typeof CONFIG;
|
|
95
|
+
export { CONFIG as HumanInLoopConfig };
|