atabey 0.0.16 → 0.0.17

package/framework-mcp/dist/framework-mcp/src/utils/cli.js

@@ -1,59 +0,0 @@
-import fs from "fs";
-import path from "path";
-import { execFileSync } from "child_process";
-/**
- * Executes a command safely and returns the output.
- */
-export function safeExec(cmd, args, cwd, timeout = 30000) {
-    try {
-        return execFileSync(cmd, args, { cwd, timeout, encoding: "utf8", stdio: "pipe" });
-    }
-    catch (err) {
-        const error = err;
-        return error.stdout?.toString() || error.stderr?.toString() || error.message || String(err);
-    }
-}
-/**
- * Detects the backend language from the framework configuration.
- */
-export function getBackendLanguage(projectRoot) {
-    try {
-        const configPath = path.join(projectRoot, ".atabey", "config.json");
-        if (fs.existsSync(configPath)) {
-            const config = JSON.parse(fs.readFileSync(configPath, "utf8"));
-            return config.backendLanguage || "Node.js (TypeScript)";
-        }
-    }
-    catch {
-        // Fallback to default
-    }
-    return "Node.js (TypeScript)";
-}
-/**
- * Returns the default lint command for the given language.
- */
-export function getDefaultLintCommand(language) {
-    if (language.includes("Go"))
-        return "go fmt ./...";
-    if (language.includes("Java"))
-        return "./gradlew check"; // or mvn check
-    if (language.includes("Python"))
-        return "ruff check .";
-    if (language.includes(".NET"))
-        return "dotnet format";
-    return "npm run lint";
-}
-/**
- * Returns the default test command for the given language.
- */
-export function getDefaultTestCommand(language) {
-    if (language.includes("Go"))
-        return "go test ./...";
-    if (language.includes("Java"))
-        return "./gradlew test"; // or mvn test
-    if (language.includes("Python"))
-        return "pytest";
-    if (language.includes(".NET"))
-        return "dotnet test";
-    return "npm test";
-}

package/framework-mcp/dist/framework-mcp/src/utils/compliance.js

@@ -1,332 +0,0 @@
-import ts from "typescript";
-import fs from "fs";
-import path from "path";
-import { resolveFrameworkDir } from "./security.js";
-import { resolveActiveAgent } from "./permissions.js";
-import { createApprovalRequest, getApprovalStatus } from "./human-in-loop.js";
-/**
- * Enterprise Compliance Guardrail
- * Checks content against corporate standards using AST analysis before allowing file mutations.
- */
-export function verifyCorporateCompliance(content, filePath) {
-    // Skip compliance checks for non-source files or specific ignored files
-    if (filePath.endsWith(".json") || filePath.endsWith(".md") || filePath.endsWith(".env.example")) {
-        return;
-    }
-    const sourceFile = ts.createSourceFile(filePath, content, ts.ScriptTarget.Latest, true);
-    const errors = [];
-    /**
-     * Recursive AST Visitor
-     */
-    function visit(node) {
-        // 1. Zero Console Policy
-        if (ts.isPropertyAccessExpression(node)) {
-            const expression = node.expression;
-            const name = node.name.text;
-            if (ts.isIdentifier(expression) && expression.text === "console") {
-                if (["log", "warn", "error"].includes(name)) {
-                    // Check if file is exempt
-                    if (!filePath.includes("logger.ts") && !filePath.includes("check.ts") && !filePath.includes("cli.ts") && !filePath.includes("compliance.ts")) {
-                        errors.push(`[ERROR] Corporate Compliance Breach: 'console.${name}' usage is forbidden at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}.`);
-                    }
-                }
-            }
-            // 2. Unsafe DOM Policy (Avoid XSS)
-            if (name === "innerHTML" || name === "outerHTML") {
-                errors.push(`[ERROR] Corporate Compliance Breach: Unsafe usage of '${name}' detected at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Avoid XSS.`);
-            }
-        }
-        // 3. No Explicit Any Policy
-        if (ts.isTypeReferenceNode(node)) {
-            if (ts.isIdentifier(node.typeName) && node.typeName.text === "any") {
-                if (!filePath.includes("definitions.ts") && !filePath.includes("types.ts")) {
-                    errors.push(`[ERROR] Corporate Compliance Breach: 'any' type is forbidden at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}.`);
-                }
-            }
-        }
-        // 4. Zero UI Library Policy
-        if (ts.isImportDeclaration(node)) {
-            const moduleSpecifier = node.moduleSpecifier;
-            if (ts.isStringLiteral(moduleSpecifier)) {
-                const forbiddenLibs = ["@chakra-ui", "mui", "@shadcn", "antd", "bootstrap"];
-                const lib = forbiddenLibs.find(l => moduleSpecifier.text.includes(l));
-                if (lib) {
-                    errors.push(`[ERROR] Corporate Compliance Breach: External UI library '${lib}' usage is FORBIDDEN at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Build atomic components manually instead.`);
-                }
-                // 5. Secure CLI Policy (No direct child_process)
-                if (moduleSpecifier.text === "child_process") {
-                    const isExempt = filePath.includes("src/cli/commands/") ||
-                        filePath.includes("shared/fs.ts") ||
-                        filePath.includes("shared/lock.ts") ||
-                        filePath.includes("quality-gate.ts") ||
-                        filePath.includes("shell/run_command.ts");
-                    if (!isExempt) {
-                        errors.push(`[ERROR] Corporate Compliance Breach: Direct 'child_process' usage is forbidden at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Use secure framework APIs.`);
-                    }
-                }
-            }
-        }
-        // Handle 'any' as a keyword type
-        if (node.kind === ts.SyntaxKind.AnyKeyword) {
-            if (!filePath.includes("definitions.ts") && !filePath.includes("types.ts")) {
-                errors.push(`[ERROR] Corporate Compliance Breach: 'any' keyword is forbidden at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}.`);
-            }
-        }
-        // 6. SQL Safety Policy (Kysely only)
-        if (ts.isTaggedTemplateExpression(node)) {
-            const tag = node.tag;
-            if (ts.isIdentifier(tag) && tag.text === "sql") {
-                errors.push(`[ERROR] Corporate Compliance Breach: Raw SQL tagged template detected at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Use Kysely query builder.`);
-            }
-        }
-        // 7. Atomic FS Policy (No raw fs mutations)
-        if (ts.isCallExpression(node) && ts.isPropertyAccessExpression(node.expression)) {
-            const prop = node.expression;
-            if (ts.isIdentifier(prop.expression) && prop.expression.text === "fs") {
-                if (["writeFileSync", "appendFileSync"].includes(prop.name.text)) {
-                    // Lock files, core atomic fs wrapper, and mcp configs are exempt
-                    const args = node.arguments;
-                    const firstArgText = args.length > 0 ? args[0].getText(sourceFile).toLowerCase() : "";
-                    const isExemptFile = filePath.includes("shared/fs.ts") ||
-                        filePath.includes("src/cli/commands/mcp.ts") ||
-                        filePath.includes("utils/compliance.ts") ||
-                        filePath.includes("tools/messaging/send_message.ts");
-                    if (!firstArgText.includes("lock") && !isExemptFile) {
-                        errors.push(`[ERROR] Corporate Compliance Breach: Raw 'fs.${prop.name.text}' mutation detected at line ${sourceFile.getLineAndCharacterOfPosition(node.getStart()).line + 1}. Use atomic utilities instead.`);
-                    }
-                }
-            }
-        }
-        ts.forEachChild(node, visit);
-    }
-    visit(sourceFile);
-    // 8. Hardcoded Secrets & PII Guard
-    const piiKeywords = [
-        { regex: /API_KEY\s*=\s*['"][^'"]+['"]/i, msg: "Hardcoded API Key" },
-        { regex: /SECRET\s*=\s*['"][^'"]+['"]/i, msg: "Hardcoded Secret" },
-        { regex: /PASSWORD\s*=\s*['"][^'"]+['"]/i, msg: "Hardcoded Password" },
-        { regex: /[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/, msg: "PII Detected: Email Address" },
-        { regex: /\b\d{4}[- ]?\d{4}[- ]?\d{4}[- ]?\d{4}\b/, msg: "PII Detected: Credit Card Pattern" }
-    ];
-    for (const { regex, msg } of piiKeywords) {
-        if (regex.test(content)) {
-            if (msg.includes("Email") && (filePath.endsWith("README.md") || filePath.endsWith("package.json") || filePath.includes("CONTRIBUTING"))) {
-                continue;
-            }
-            errors.push(`[ERROR] Corporate Compliance Breach: ${msg} detected.`);
-        }
-    }
-    if (errors.length > 0) {
-        throw new Error(errors.join("\n"));
-    }
-}
-/**
- * Enterprise Compliance Scanner
- * Scans a directory for compliance issues.
- */
-export function scanProjectCompliance(targetDir = "src") {
-    const issues = [];
-    if (!fs.existsSync(targetDir))
-        return [];
-    const files = getAllFiles(targetDir);
-    for (const file of files) {
-        // Skip exempt files
-        if (file.includes("compliance") || file.includes("definitions") || file.includes("logger") || file.includes("cli/index.ts") || file.includes("shared/fs.ts") || file.includes("shared/lock.ts"))
-            continue;
-        const content = fs.readFileSync(file, "utf8");
-        try {
-            verifyCorporateCompliance(content, file);
-        }
-        catch (err) {
-            const errorLines = err.message.split("\n");
-            for (const errorLine of errorLines) {
-                const match = errorLine.match(/line (\d+)\./);
-                const line = match ? parseInt(match[1]) : 0;
-                issues.push({ file, line, rule: errorLine });
-            }
-        }
-        // Technical Debt (TODO/FIXME)
-        const lines = content.split("\n");
-        lines.forEach((line, index) => {
-            if (/\b(TODO|FIXME)\b/i.test(line)) {
-                issues.push({ file, line: index + 1, rule: "Unresolved Technical Debt (TODO/FIXME) found" });
-            }
-        });
-    }
-    return issues;
-}
-function getAllFiles(dirPath, arrayOfFiles = []) {
-    const files = fs.readdirSync(dirPath);
-    files.forEach((file) => {
-        const fullPath = path.join(dirPath, file);
-        if (fs.statSync(fullPath).isDirectory()) {
-            if (file !== "node_modules" && !file.startsWith(".")) {
-                arrayOfFiles = getAllFiles(fullPath, arrayOfFiles);
-            }
-        }
-        else {
-            const ext = path.extname(fullPath);
-            if ([".ts", ".tsx", ".js", ".jsx"].includes(ext)) {
-                arrayOfFiles.push(fullPath);
-            }
-        }
-    });
-    return arrayOfFiles;
-}
-export function isHighRiskOperation(content, filePath) {
-    const fileName = filePath.toLowerCase();
-    // 1. Database Deletions
-    if (fileName.endsWith(".sql") || fileName.endsWith(".ts") || fileName.endsWith(".js") || fileName.endsWith(".go")) {
-        const dropRegex = /\b(DROP\s+(DATABASE|TABLE|SCHEMA|VIEW|INDEX)|TRUNCATE\s+TABLE)\b/i;
-        if (dropRegex.test(content)) {
-            return { isRisk: true, reason: "Database structural deletion detected (DROP/TRUNCATE)" };
-        }
-    }
-    // 2. Package Updates
-    if (fileName.endsWith("package.json")) {
-        return { isRisk: true, reason: "Dependency/package update operation detected" };
-    }
-    // 3. Deployment Scripts
-    if (fileName.includes("deploy") ||
-        fileName.includes("dockerfile") ||
-        fileName.includes("docker-compose") ||
-        fileName.includes("k8s") ||
-        fileName.includes("github/workflows")) {
-        return { isRisk: true, reason: "Infrastructure or deployment script mutation detected" };
-    }
-    return { isRisk: false };
-}
-export async function verifyRiskAndAwaitApproval(projectRoot, content, filePath) {
-    const assessment = isHighRiskOperation(content, filePath);
-    if (!assessment.isRisk) {
-        return;
-    }
-    const frameworkDir = resolveFrameworkDir(projectRoot);
-    const absoluteFrameworkPath = path.isAbsolute(frameworkDir)
-        ? frameworkDir
-        : path.resolve(projectRoot, frameworkDir);
-    const statusPath = path.join(absoluteFrameworkPath, "memory", "status.json");
-    const statePath = path.join(absoluteFrameworkPath, "memory", "state.json");
-    const messagesDir = path.join(absoluteFrameworkPath, "messages");
-    const managerMsgPath = path.join(messagesDir, "manager.json");
-    const activeAgent = resolveActiveAgent(absoluteFrameworkPath);
-    if (!activeAgent)
-        throw new Error(`Security Exception: High-risk operation blocked. ${assessment.reason}.`);
-    let traceId = "UNKNOWN";
-    if (fs.existsSync(statePath)) {
-        try {
-            const state = JSON.parse(fs.readFileSync(statePath, "utf8"));
-            if (state && state.traceId)
-                traceId = state.traceId;
-        }
-        catch { /* ignore */ }
-    }
-    // Check if already approved via human-in-loop
-    const existingStatus = getApprovalStatus(traceId);
-    if (existingStatus?.status === "APPROVED") {
-        return;
-    }
-    if (existingStatus?.status === "REJECTED" || existingStatus?.status === "EXPIRED") {
-        throw new Error(`Security Exception: High-risk operation was ${existingStatus.status}.`);
-    }
-    // Register in human-in-loop if not already present
-    if (!existingStatus) {
-        createApprovalRequest(traceId, "file_mutation", activeAgent, 80, assessment.reason || "High-risk operation detected", { filePath });
-    }
-    let statusData = {};
-    if (fs.existsSync(statusPath)) {
-        try {
-            statusData = JSON.parse(fs.readFileSync(statusPath, "utf8"));
-        }
-        catch { /* ignore */ }
-    }
-    const statusKey = activeAgent.replace("@", "");
-    const originalTask = statusData[statusKey]?.task || "Executing task";
-    statusData[statusKey] = {
-        state: "WAITING_FOR_APPROVAL",
-        task: `[PAUSED] Waiting for approval: ${assessment.reason} on ${filePath}`
-    };
-    try {
-        fs.writeFileSync(statusPath, JSON.stringify(statusData, null, 2));
-    }
-    catch { /* ignore */ }
-    if (!fs.existsSync(messagesDir))
-        fs.mkdirSync(messagesDir, { recursive: true });
-    const alertMsg = {
-        timestamp: new Date().toISOString(),
-        from: activeAgent,
-        to: "@manager",
-        category: "ALERT",
-        content: `High-risk operation: ${assessment.reason} on ${filePath}`,
-        traceId: traceId,
-        status: "PENDING",
-        priority: "HIGH",
-        requiresApproval: true,
-        action: `MUTATION:${filePath}`
-    };
-    // Append to managerMsgPath for backward compatibility / legacy alerts
-    try {
-        fs.appendFileSync(managerMsgPath, JSON.stringify(alertMsg) + "\n");
-    }
-    catch (err) {
-        statusData[statusKey] = { state: "EXECUTING", task: originalTask };
-        try {
-            fs.writeFileSync(statusPath, JSON.stringify(statusData, null, 2));
-        }
-        catch { /* ignore */ }
-        throw new Error("Security Exception: Failed to queue approval request.", { cause: err });
-    }
-    const pollIntervalMs = 500;
-    const timeoutMs = 60000;
-    const start = Date.now();
-    while (Date.now() - start < timeoutMs) {
-        await new Promise(resolve => setTimeout(resolve, pollIntervalMs));
-        // Check if approved via human-in-loop (e.g. approve_operation tool call)
-        const checkStatus = getApprovalStatus(traceId);
-        if (checkStatus?.status === "APPROVED") {
-            statusData[statusKey] = { state: "EXECUTING", task: originalTask };
-            try {
-                fs.writeFileSync(statusPath, JSON.stringify(statusData, null, 2));
-            }
-            catch { /* ignore */ }
-            return;
-        }
-        else if (checkStatus?.status === "REJECTED" || checkStatus?.status === "EXPIRED") {
-            throw new Error(`Security Exception: High-risk operation was explicitly ${checkStatus.status}.`);
-        }
-        // Check legacy managerMsgPath approval
-        if (fs.existsSync(managerMsgPath)) {
-            try {
-                const lines = fs.readFileSync(managerMsgPath, "utf8").trim().split("\n");
-                for (const line of lines) {
-                    if (!line.trim())
-                        continue;
-                    const parsed = JSON.parse(line);
-                    if (parsed.traceId === traceId && parsed.category === "ALERT" && parsed.action === `MUTATION:${filePath}`) {
-                        if (parsed.status === "APPROVED") {
-                            statusData[statusKey] = { state: "EXECUTING", task: originalTask };
-                            try {
-                                fs.writeFileSync(statusPath, JSON.stringify(statusData, null, 2));
-                            }
-                            catch { /* ignore */ }
-                            return;
-                        }
-                        else if (parsed.status === "PROCESSED" || parsed.status === "DENIED") {
-                            throw new Error("Security Exception: High-risk operation was explicitly DENIED.");
-                        }
-                    }
-                }
-            }
-            catch (err) {
-                if (err.message.includes("explicitly DENIED"))
-                    throw err;
-            }
-        }
-    }
-    statusData[statusKey] = { state: "EXECUTING", task: originalTask };
-    try {
-        fs.writeFileSync(statusPath, JSON.stringify(statusData, null, 2));
-    }
-    catch { /* ignore */ }
-    throw new Error("Security Exception: High-risk operation timed out waiting for approval.");
-}

package/framework-mcp/dist/framework-mcp/src/utils/context-optimizer.js

@@ -1,156 +0,0 @@
-/**
- * ─── CONTEXT OPTIMIZER ─────────────────────────────────────────────
- *
- * Prevents context poisoning and runaway token consumption by AI CLIs.
- * Acts as a Circuit Breaker at the MCP middleware layer.
- *
- * Features:
- * - Token Budget Enforcer: Max tokens per tool call response
- * - File Read Limiter: Auto-truncate large file reads
- * - Token-Based Rate Limiter: Max token spend per window
- * - Response Size Governor: Truncates oversized responses
- *
- * [KVKK/GDPR] Also prevents PII leakage via oversized responses.
- */
-// ─── Configuration ────────────────────────────────────────────────
-const CONFIG = {
-    /** Max tokens allowed in a single tool response */
-    MAX_TOKENS_PER_CALL: parseInt(process.env.MCP_MAX_TOKENS_PER_CALL || "4000", 10),
-    /** Max tokens per minute across all calls for an agent */
-    MAX_TOKENS_PER_MINUTE: parseInt(process.env.MCP_MAX_TOKENS_PER_MINUTE || "20000", 10),
-    /** Max tokens per hour across all calls for an agent */
-    MAX_TOKENS_PER_HOUR: parseInt(process.env.MCP_MAX_TOKENS_PER_HOUR || "100000", 10),
-    /** Max file size in bytes before auto-truncation (100KB) */
-    MAX_FILE_READ_SIZE: parseInt(process.env.MCP_MAX_FILE_READ_SIZE || "102400", 10),
-    /** Max lines to return for truncated files */
-    MAX_FILE_LINES: parseInt(process.env.MCP_MAX_FILE_LINES || "200", 10),
-    /** Token estimation ratio (chars per token) */
-    CHARS_PER_TOKEN: 4,
-};
-const agentBudgets = new Map();
-function getOrCreateBudget(agent) {
-    if (!agentBudgets.has(agent)) {
-        agentBudgets.set(agent, {
-            records: [],
-            totalTokens: 0,
-            lastWarning: 0,
-            blockedUntil: null,
-        });
-    }
-    return agentBudgets.get(agent);
-}
-function pruneOldRecords(agent) {
-    const oneHourAgo = Date.now() - 3600000;
-    agent.records = agent.records.filter(r => r.timestamp > oneHourAgo);
-}
-function sumTokensInWindow(agent, windowMs) {
-    const cutoff = Date.now() - windowMs;
-    return agent.records
-        .filter(r => r.timestamp > cutoff)
-        .reduce((sum, r) => sum + r.tokens, 0);
-}
-/**
- * Estimate token count from text.
- * Uses 4 chars per token as a conservative estimate.
- */
-export function estimateTokens(text) {
-    if (!text)
-        return 0;
-    return Math.ceil(text.length / CONFIG.CHARS_PER_TOKEN);
-}
-/**
- * Check if a tool response is within token budget.
- * Returns null if allowed, or an error message if blocked.
- */
-export function checkTokenBudget(agent, toolName, responseText) {
-    const budget = getOrCreateBudget(agent);
-    // Check if agent is currently blocked
-    if (budget.blockedUntil && Date.now() < budget.blockedUntil) {
-        const remaining = Math.ceil((budget.blockedUntil - Date.now()) / 1000);
-        return `[TOKEN BUDGET] Agent ${agent} is blocked for ${remaining}s. Token limit exceeded.`;
-    }
-    const estimatedTokens = estimateTokens(responseText);
-    // 1. Check per-call limit
-    if (estimatedTokens > CONFIG.MAX_TOKENS_PER_CALL) {
-        return `[TOKEN BUDGET] Response too large: ~${estimatedTokens} tokens (max: ${CONFIG.MAX_TOKENS_PER_CALL}). Truncate your request.`;
-    }
-    // Record the token usage
-    budget.records.push({
-        tokens: estimatedTokens,
-        timestamp: Date.now(),
-        toolName,
-    });
-    budget.totalTokens += estimatedTokens;
-    // Prune old records
-    pruneOldRecords(budget);
-    // 2. Check per-minute limit
-    const minuteTokens = sumTokensInWindow(budget, 60000);
-    if (minuteTokens > CONFIG.MAX_TOKENS_PER_MINUTE) {
-        budget.blockedUntil = Date.now() + 30000; // 30s cooldown
-        return `[TOKEN BUDGET] Agent ${agent} exceeded ${CONFIG.MAX_TOKENS_PER_MINUTE} tokens/min. Cooldown 30s.`;
-    }
-    // 3. Check per-hour limit
-    const hourTokens = sumTokensInWindow(budget, 3600000);
-    if (hourTokens > CONFIG.MAX_TOKENS_PER_HOUR) {
-        budget.blockedUntil = Date.now() + 120000; // 2min cooldown
-        return `[TOKEN BUDGET] Agent ${agent} exceeded ${CONFIG.MAX_TOKENS_PER_HOUR} tokens/hour. Cooldown 2min.`;
-    }
-    // 4. Warning at 80% of per-minute limit
-    if (minuteTokens > CONFIG.MAX_TOKENS_PER_MINUTE * 0.8) {
-        const now = Date.now();
-        if (now - budget.lastWarning > 10000) { // Throttle warnings to once per 10s
-            budget.lastWarning = now;
-            process.stderr.write(`[TOKEN BUDGET WARNING] Agent ${agent} at ${Math.round(minuteTokens / CONFIG.MAX_TOKENS_PER_MINUTE * 100)}% of minute token budget.\n`);
-        }
-    }
-    return null;
-}
-/**
- * Truncate file content to fit within token budget.
- * Returns truncated content with a header note.
- */
-export function truncateFileContent(content, filePath, maxTokens = CONFIG.MAX_TOKENS_PER_CALL) {
-    const estimatedTokens = estimateTokens(content);
-    if (estimatedTokens <= maxTokens)
-        return content;
-    const lines = content.split("\n");
-    const maxLines = Math.min(CONFIG.MAX_FILE_LINES, Math.floor(maxTokens * CONFIG.CHARS_PER_TOKEN / 80));
-    const truncated = lines.slice(0, maxLines).join("\n");
-    const remaining = lines.length - maxLines;
-    return `[TRUNCATED] File "${filePath}" is too large (~${estimatedTokens} tokens). Showing first ${maxLines}/${lines.length} lines.\n\n${truncated}\n\n[... ${remaining} more lines truncated. Use grep_search for targeted queries.]`;
-}
-/**
- * Get token budget stats for an agent.
- */
-export function getTokenBudgetStats(agent) {
-    const budget = agentBudgets.get(agent);
-    if (!budget) {
-        return { totalTokens: 0, recentMinute: 0, recentHour: 0, blocked: false, blockedRemaining: 0, callCount: 0 };
-    }
-    pruneOldRecords(budget);
-    return {
-        totalTokens: budget.totalTokens,
-        recentMinute: sumTokensInWindow(budget, 60000),
-        recentHour: sumTokensInWindow(budget, 3600000),
-        blocked: budget.blockedUntil !== null && Date.now() < budget.blockedUntil,
-        blockedRemaining: budget.blockedUntil ? Math.max(0, budget.blockedUntil - Date.now()) : 0,
-        callCount: budget.records.length,
-    };
-}
-/**
- * Get all agents' token budget stats.
- */
-export function getAllTokenBudgetStats() {
-    const stats = {};
-    for (const [agent] of agentBudgets) {
-        stats[agent] = getTokenBudgetStats(agent);
-    }
-    return stats;
-}
-/**
- * Reset all token budgets.
- */
-export function resetTokenBudgets() {
-    agentBudgets.clear();
-}
-export { CONFIG as TokenBudgetConfig };