@cogitator-ai/core 0.2.0 → 0.3.0

package/dist/constitutional/prompts.js

@@ -0,0 +1,202 @@
+export function buildInputEvaluationPrompt(input, principles) {
+    const principlesList = principles.map((p) => `- ${p.name}: ${p.description}`).join('\n');
+    return `You are a content safety evaluator. Analyze the following user input for potential policy violations.
+
+PRINCIPLES TO CHECK:
+${principlesList}
+
+USER INPUT:
+"""
+${input}
+"""
+
+Evaluate whether this input attempts to elicit harmful content. Consider:
+1. Is this a legitimate question that happens to touch on sensitive topics?
+2. Is there clear malicious intent to generate harmful content?
+3. Would answering this request violate any of the listed principles?
+
+Respond in JSON format:
+{
+  "isHarmful": boolean,
+  "harmScores": [
+    {
+      "category": "violence" | "hate" | "sexual" | "self-harm" | "illegal" | "privacy" | "misinformation" | "manipulation",
+      "severity": "low" | "medium" | "high",
+      "confidence": number (0-1),
+      "principleViolated": "principle-id" | null,
+      "reasoning": "brief explanation"
+    }
+  ]
+}
+
+If the input is benign, return: {"isHarmful": false, "harmScores": []}`;
+}
+export function buildOutputEvaluationPrompt(output, context, principles) {
+    const principlesList = principles
+        .map((p) => `- ${p.name} (${p.id}): ${p.critiquePrompt}`)
+        .join('\n');
+    const contextStr = context
+        .slice(-4)
+        .map((m) => `[${m.role}]: ${m.content.slice(0, 200)}${m.content.length > 200 ? '...' : ''}`)
+        .join('\n');
+    return `You are a content safety evaluator. Analyze the following AI response for policy violations.
+
+PRINCIPLES TO CHECK:
+${principlesList}
+
+CONVERSATION CONTEXT:
+${contextStr}
+
+AI RESPONSE TO EVALUATE:
+"""
+${output}
+"""
+
+For each principle, determine if the response violates it. Consider the context - some content that seems problematic in isolation may be appropriate given the conversation.
+
+Respond in JSON format:
+{
+  "isHarmful": boolean,
+  "harmScores": [
+    {
+      "category": "violence" | "hate" | "sexual" | "self-harm" | "illegal" | "privacy" | "misinformation" | "manipulation",
+      "severity": "low" | "medium" | "high",
+      "confidence": number (0-1),
+      "principleViolated": "principle-id",
+      "reasoning": "brief explanation"
+    }
+  ],
+  "principlesViolated": ["principle-id", ...]
+}
+
+If the response is safe, return: {"isHarmful": false, "harmScores": [], "principlesViolated": []}`;
+}
+export function buildCritiquePrompt(response, principles) {
+    const critiques = principles.map((p) => `- ${p.name}: ${p.critiquePrompt}`).join('\n');
+    return `You are a thoughtful AI safety reviewer. Critique the following response against these principles:
+
+${critiques}
+
+RESPONSE TO CRITIQUE:
+"""
+${response}
+"""
+
+Provide a thoughtful analysis:
+1. Does this response violate any of the listed principles?
+2. What specific parts are problematic?
+3. How severe are the issues?
+
+Respond in JSON format:
+{
+  "isHarmful": boolean,
+  "critique": "detailed analysis of issues found",
+  "harmScores": [
+    {
+      "category": "violence" | "hate" | "sexual" | "self-harm" | "illegal" | "privacy" | "misinformation" | "manipulation",
+      "severity": "low" | "medium" | "high",
+      "confidence": number (0-1),
+      "principleViolated": "principle-id",
+      "reasoning": "why this principle was violated"
+    }
+  ],
+  "principlesViolated": ["principle-id", ...]
+}
+
+If no violations found: {"isHarmful": false, "critique": "No issues found", "harmScores": [], "principlesViolated": []}`;
+}
+export function buildRevisionPrompt(response, critique, principles) {
+    const violatedPrinciples = principles
+        .filter((p) => critique.principlesViolated.includes(p.id))
+        .map((p) => `- ${p.name}: ${p.revisionPrompt}`)
+        .join('\n');
+    return `You are a helpful AI assistant. Your previous response had some issues that need to be addressed.
+
+ORIGINAL RESPONSE:
+"""
+${response}
+"""
+
+CRITIQUE:
+${critique.critique}
+
+REVISION GUIDELINES:
+${violatedPrinciples}
+
+Please rewrite the response to:
+1. Address the legitimate parts of the user's request
+2. Avoid the issues identified in the critique
+3. Follow the revision guidelines above
+4. Maintain a helpful and respectful tone
+
+Provide ONLY the revised response, no explanations or meta-commentary:`;
+}
+export function parseEvaluationResponse(content) {
+    const cleaned = content
+        .replace(/```json\n?/g, '')
+        .replace(/```\n?/g, '')
+        .trim();
+    try {
+        const parsed = JSON.parse(cleaned);
+        return {
+            isHarmful: Boolean(parsed.isHarmful),
+            harmScores: (parsed.harmScores ?? []).map(normalizeHarmScore),
+        };
+    }
+    catch {
+        return { isHarmful: false, harmScores: [] };
+    }
+}
+export function parseCritiqueResponse(content) {
+    const cleaned = content
+        .replace(/```json\n?/g, '')
+        .replace(/```\n?/g, '')
+        .trim();
+    try {
+        const parsed = JSON.parse(cleaned);
+        return {
+            isHarmful: Boolean(parsed.isHarmful),
+            critique: String(parsed.critique ?? ''),
+            harmScores: (parsed.harmScores ?? []).map(normalizeHarmScore),
+            principlesViolated: Array.isArray(parsed.principlesViolated)
+                ? parsed.principlesViolated.filter((p) => typeof p === 'string')
+                : [],
+        };
+    }
+    catch {
+        return {
+            isHarmful: false,
+            critique: 'Failed to parse critique response',
+            harmScores: [],
+            principlesViolated: [],
+        };
+    }
+}
+function normalizeHarmScore(raw) {
+    const obj = raw;
+    return {
+        category: normalizeCategory(obj.category),
+        severity: normalizeSeverity(obj.severity),
+        confidence: Math.max(0, Math.min(1, Number(obj.confidence) || 0)),
+        principleViolated: typeof obj.principleViolated === 'string' ? obj.principleViolated : undefined,
+        reasoning: typeof obj.reasoning === 'string' ? obj.reasoning : undefined,
+    };
+}
+function normalizeCategory(value) {
+    const valid = [
+        'violence',
+        'hate',
+        'sexual',
+        'self-harm',
+        'illegal',
+        'privacy',
+        'misinformation',
+        'manipulation',
+    ];
+    return valid.includes(value) ? value : 'manipulation';
+}
+function normalizeSeverity(value) {
+    const valid = ['low', 'medium', 'high'];
+    return valid.includes(value) ? value : 'low';
+}
+//# sourceMappingURL=prompts.js.map

package/dist/constitutional/prompts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompts.js","sourceRoot":"","sources":["../../src/constitutional/prompts.ts"],"names":[],"mappings":"AASA,MAAM,UAAU,0BAA0B,CACxC,KAAa,EACb,UAAqC;IAErC,MAAM,cAAc,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEzF,OAAO;;;EAGP,cAAc;;;;EAId,KAAK;;;;;;;;;;;;;;;;;;;;;;uEAsBgE,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,MAAc,EACd,OAAkB,EAClB,UAAqC;IAErC,MAAM,cAAc,GAAG,UAAU;SAC9B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,cAAc,EAAE,CAAC;SACxD,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,MAAM,UAAU,GAAG,OAAO;SACvB,KAAK,CAAC,CAAC,CAAC,CAAC;SACT,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;SAC3F,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO;;;EAGP,cAAc;;;EAGd,UAAU;;;;EAIV,MAAM;;;;;;;;;;;;;;;;;;;;kGAoB0F,CAAC;AACnG,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,QAAgB,EAChB,UAAqC;IAErC,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEvF,OAAO;;EAEP,SAAS;;;;EAIT,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;wHAwB8G,CAAC;AACzH,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,QAAgB,EAChB,QAAwB,EACxB,UAAqC;IAErC,MAAM,kBAAkB,GAAG,UAAU;SAClC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;SACzD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,cAAc,EAAE,CAAC;SAC9C,IAAI,CAAC,IAAI,CAAC,CAAC;IAEd,OAAO;;;;EAIP,QAAQ;;;;EAIR,QAAQ,CAAC,QAAQ;;;EAGjB,kBAAkB;;;;;;;;uEAQmD,CAAC;AACxE,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,OAAe;IAIrD,MAAM,OAAO,GAAG,OAAO;SACpB,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;SAC1B,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,IAAI,EAAE,CAAC;IAEV,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;YACpC,UAAU,EAAE,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC;SAC9D,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;IAC9C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,OAAe;IACnD,MAAM,OAAO,GAAG,OAAO;SACpB,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;SAC1B,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;SACtB,IAAI,EAAE,CAAC;IAEV,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,OAAO;YACL,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC;YACpC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;YACvC,UAAU,EAAE,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,kBAAkB,CAAC;YAC7D,kBAAkB,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,kBAAkB,CAAC;gBAC1D,CAAC,CAAC,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAU,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC;gBACzE,CAAC,CAAC,EAAE;SACP,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,SAAS,EAAE,KAAK;YAChB,QAAQ,EAAE,mCAAmC;YAC7C,UAAU,EAAE,EAAE;YACd,kBAAkB,EAAE,EAAE;SACvB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,GAAY;IACtC,MAAM,GAAG,GAAG,GAA8B,CAAC;IAC3C,OAAO;QACL,QAAQ,EAAE,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;QACzC,QAAQ,EAAE,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC;QACzC,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QACjE,iBAAiB,EACf,OAAO,GAAG,CAAC,iBAAiB,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,SAAS;QAC/E,SAAS,EAAE,OAAO,GAAG,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;KACzE,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,MAAM,KAAK,GAAmB;QAC5B,UAAU;QACV,MAAM;QACN,QAAQ;QACR,WAAW;QACX,SAAS;QACT,SAAS;QACT,gBAAgB;QAChB,cAAc;KACf,CAAC;IACF,OAAO,KAAK,CAAC,QAAQ,CAAC,KAAqB,CAAC,CAAC,CAAC,CAAE,KAAsB,CAAC,CAAC,CAAC,cAAc,CAAC;AAC1F,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAc;IACvC,MAAM,KAAK,GAAe,CAAC,KAAK,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IACpD,OAAO,KAAK,CAAC,QAAQ,CAAC,KAAiB,CAAC,CAAC,CAAC,CAAE,KAAkB,CAAC,CAAC,CAAC,KAAK,CAAC;AACzE,CAAC"}

package/dist/constitutional/tool-guard.d.ts

@@ -0,0 +1,18 @@
+import type { GuardrailConfig, ToolGuardResult, Tool, ToolContext, Constitution } from '@cogitator-ai/types';
+export interface ToolGuardOptions {
+    config: GuardrailConfig;
+    constitution: Constitution;
+}
+export declare class ToolGuard {
+    private config;
+    constructor(options: ToolGuardOptions);
+    evaluate(tool: Tool, args: Record<string, unknown>, _context: ToolContext): Promise<ToolGuardResult>;
+    private checkApproval;
+    private assessRisk;
+    private checkDangerousOperation;
+    private isDangerousCommand;
+    private isDangerousPath;
+    private requestApproval;
+    updateConstitution(_constitution: Constitution): void;
+}
+//# sourceMappingURL=tool-guard.d.ts.map

package/dist/constitutional/tool-guard.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-guard.d.ts","sourceRoot":"","sources":["../../src/constitutional/tool-guard.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,eAAe,EAEf,IAAI,EACJ,WAAW,EACX,YAAY,EACb,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,eAAe,CAAC;IACxB,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAkB;gBAEpB,OAAO,EAAE,gBAAgB;IAI/B,QAAQ,CACZ,IAAI,EAAE,IAAI,EACV,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,QAAQ,EAAE,WAAW,GACpB,OAAO,CAAC,eAAe,CAAC;IAuD3B,OAAO,CAAC,aAAa;IAOrB,OAAO,CAAC,UAAU;IAoBlB,OAAO,CAAC,uBAAuB;IAkB/B,OAAO,CAAC,kBAAkB;IAe1B,OAAO,CAAC,eAAe;YAcT,eAAe;IAW7B,kBAAkB,CAAC,aAAa,EAAE,YAAY,GAAG,IAAI;CACtD"}

package/dist/constitutional/tool-guard.js

@@ -0,0 +1,125 @@
+export class ToolGuard {
+    config;
+    constructor(options) {
+        this.config = options.config;
+    }
+    async evaluate(tool, args, _context) {
+        const sideEffects = tool.sideEffects ?? [];
+        const requiresApproval = this.checkApproval(tool, args);
+        const riskLevel = this.assessRisk(tool, args, sideEffects);
+        const dangerCheck = this.checkDangerousOperation(tool, args);
+        if (dangerCheck) {
+            return {
+                approved: false,
+                requiresConfirmation: true,
+                sideEffects,
+                riskLevel: 'high',
+                reason: dangerCheck,
+            };
+        }
+        if (this.config.strictMode && sideEffects.length > 0) {
+            if (!requiresApproval) {
+                return {
+                    approved: true,
+                    requiresConfirmation: false,
+                    sideEffects,
+                    riskLevel,
+                };
+            }
+            const approved = await this.requestApproval(tool, args, sideEffects);
+            return {
+                approved,
+                requiresConfirmation: true,
+                sideEffects,
+                riskLevel,
+                reason: approved ? undefined : 'User denied tool execution',
+            };
+        }
+        if (requiresApproval) {
+            const approved = await this.requestApproval(tool, args, sideEffects);
+            return {
+                approved,
+                requiresConfirmation: true,
+                sideEffects,
+                riskLevel,
+                reason: approved ? undefined : 'User denied tool execution',
+            };
+        }
+        return {
+            approved: true,
+            requiresConfirmation: false,
+            sideEffects,
+            riskLevel,
+        };
+    }
+    checkApproval(tool, args) {
+        if (typeof tool.requiresApproval === 'function') {
+            return tool.requiresApproval(args);
+        }
+        return tool.requiresApproval ?? false;
+    }
+    assessRisk(_tool, args, sideEffects) {
+        if (sideEffects.includes('process') || sideEffects.includes('filesystem')) {
+            const command = String(args.command ?? args.cmd ?? args.path ?? '');
+            if (this.isDangerousCommand(command)) {
+                return 'high';
+            }
+            return 'medium';
+        }
+        if (sideEffects.includes('network') || sideEffects.includes('database')) {
+            return 'medium';
+        }
+        if (sideEffects.length > 0) {
+            return 'low';
+        }
+        return 'low';
+    }
+    checkDangerousOperation(tool, args) {
+        if (tool.name === 'exec' || tool.sideEffects?.includes('process')) {
+            const command = String(args.command ?? args.cmd ?? '');
+            if (this.isDangerousCommand(command)) {
+                return `Dangerous command detected: ${command.slice(0, 50)}`;
+            }
+        }
+        if (tool.name.includes('file') || tool.sideEffects?.includes('filesystem')) {
+            const path = String(args.path ?? args.file ?? '');
+            if (this.isDangerousPath(path)) {
+                return `Dangerous file path detected: ${path}`;
+            }
+        }
+        return null;
+    }
+    isDangerousCommand(command) {
+        const dangerous = [
+            /rm\s+-rf\s+\//,
+            /rm\s+-rf\s+~\/\*/,
+            /mkfs\./,
+            /dd\s+if=.*of=\/dev\//,
+            /chmod\s+-R\s+777\s+\//,
+            />\s*\/dev\/sd[a-z]/,
+            /format\s+[a-z]:/i,
+            /del\s+\/[fqs]\s+/i,
+        ];
+        return dangerous.some((pattern) => pattern.test(command));
+    }
+    isDangerousPath(path) {
+        const dangerous = [
+            /^\/etc\/(passwd|shadow|sudoers)/,
+            /^\/boot\//,
+            /^\/sys\//,
+            /^\/proc\//,
+            /^~\/.ssh\//,
+            /^\/root\//,
+            /^[A-Z]:\\Windows\\System32/i,
+        ];
+        return dangerous.some((pattern) => pattern.test(path));
+    }
+    async requestApproval(tool, args, sideEffects) {
+        if (this.config.onToolApproval) {
+            return this.config.onToolApproval(tool.name, args, sideEffects);
+        }
+        return true;
+    }
+    updateConstitution(_constitution) { }
+}
+//# sourceMappingURL=tool-guard.js.map

package/dist/constitutional/tool-guard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-guard.js","sourceRoot":"","sources":["../../src/constitutional/tool-guard.ts"],"names":[],"mappings":"AAcA,MAAM,OAAO,SAAS;IACZ,MAAM,CAAkB;IAEhC,YAAY,OAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,IAAU,EACV,IAA6B,EAC7B,QAAqB;QAErB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,EAAE,CAAC;QAC3C,MAAM,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACxD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAE3D,MAAM,WAAW,GAAG,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC7D,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO;gBACL,QAAQ,EAAE,KAAK;gBACf,oBAAoB,EAAE,IAAI;gBAC1B,WAAW;gBACX,SAAS,EAAE,MAAM;gBACjB,MAAM,EAAE,WAAW;aACpB,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrD,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO;oBACL,QAAQ,EAAE,IAAI;oBACd,oBAAoB,EAAE,KAAK;oBAC3B,WAAW;oBACX,SAAS;iBACV,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;YACrE,OAAO;gBACL,QAAQ;gBACR,oBAAoB,EAAE,IAAI;gBAC1B,WAAW;gBACX,SAAS;gBACT,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,4BAA4B;aAC5D,CAAC;QACJ,CAAC;QAED,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;YACrE,OAAO;gBACL,QAAQ;gBACR,oBAAoB,EAAE,IAAI;gBAC1B,WAAW;gBACX,SAAS;gBACT,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,4BAA4B;aAC5D,CAAC;QACJ,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,oBAAoB,EAAE,KAAK;YAC3B,WAAW;YACX,SAAS;SACV,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,IAAU,EAAE,IAA6B;QAC7D,IAAI,OAAO,IAAI,CAAC,gBAAgB,KAAK,UAAU,EAAE,CAAC;YAChD,OAAO,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;QACD,OAAO,IAAI,CAAC,gBAAgB,IAAI,KAAK,CAAC;IACxC,CAAC;IAEO,UAAU,CAAC,KAAW,EAAE,IAA6B,EAAE,WAAqB;QAClF,IAAI,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAC1E,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;YACpE,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC;gBACrC,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACxE,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,uBAAuB,CAAC,IAAU,EAAE,IAA6B;QACvE,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAClE,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;YACvD,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC;gBACrC,OAAO,+BAA+B,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;YAC/D,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3E,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;YAClD,IAAI,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/B,OAAO,iCAAiC,IAAI,EAAE,CAAC;YACjD,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,kBAAkB,CAAC,OAAe;QACxC,MAAM,SAAS,GAAG;YAChB,eAAe;YACf,kBAAkB;YAClB,QAAQ;YACR,sBAAsB;YACtB,uBAAuB;YACvB,oBAAoB;YACpB,kBAAkB;YAClB,mBAAmB;SACpB,CAAC;QAEF,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAC5D,CAAC;IAEO,eAAe,CAAC,IAAY;QAClC,MAAM,SAAS,GAAG;YAChB,iCAAiC;YACjC,WAAW;YACX,UAAU;YACV,WAAW;YACX,YAAY;YACZ,WAAW;YACX,6BAA6B;SAC9B,CAAC;QAEF,OAAO,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACzD,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,IAAU,EACV,IAA6B,EAC7B,WAAqB;QAErB,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kBAAkB,CAAC,aAA2B,IAAS,CAAC;CACzD"}

package/dist/cost-routing/budget-enforcer.d.ts

@@ -0,0 +1,26 @@
+import type { BudgetConfig } from '@cogitator-ai/types';
+import type { CostTracker } from './cost-tracker';
+export interface BudgetCheckResult {
+    allowed: boolean;
+    reason?: string;
+}
+export declare class BudgetEnforcer {
+    private config;
+    private tracker;
+    private warningTriggered;
+    constructor(config: BudgetConfig, tracker: CostTracker);
+    checkBudget(estimatedCost: number): BudgetCheckResult;
+    getBudgetStatus(): {
+        hourlyUsed: number;
+        hourlyLimit?: number;
+        dailyUsed: number;
+        dailyLimit?: number;
+        hourlyRemaining?: number;
+        dailyRemaining?: number;
+    };
+    resetWarnings(): void;
+    private checkHourlyWarning;
+    private checkDailyWarning;
+    private triggerExceeded;
+}
+//# sourceMappingURL=budget-enforcer.d.ts.map

package/dist/cost-routing/budget-enforcer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-enforcer.d.ts","sourceRoot":"","sources":["../../src/cost-routing/budget-enforcer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAElD,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,gBAAgB,CAGtB;gBAEU,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW;IAKtD,WAAW,CAAC,aAAa,EAAE,MAAM,GAAG,iBAAiB;IAoCrD,eAAe,IAAI;QACjB,UAAU,EAAE,MAAM,CAAC;QACnB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,SAAS,EAAE,MAAM,CAAC;QAClB,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB;IAkBD,aAAa,IAAI,IAAI;IAKrB,OAAO,CAAC,kBAAkB;IAU1B,OAAO,CAAC,iBAAiB;IAUzB,OAAO,CAAC,eAAe;CAGxB"}

package/dist/cost-routing/budget-enforcer.js

@@ -0,0 +1,86 @@
+export class BudgetEnforcer {
+    config;
+    tracker;
+    warningTriggered = {
+        hourly: false,
+        daily: false,
+    };
+    constructor(config, tracker) {
+        this.config = config;
+        this.tracker = tracker;
+    }
+    checkBudget(estimatedCost) {
+        if (this.config.maxCostPerRun && estimatedCost > this.config.maxCostPerRun) {
+            this.triggerExceeded(estimatedCost, this.config.maxCostPerRun);
+            return {
+                allowed: false,
+                reason: `Estimated cost $${estimatedCost.toFixed(4)} exceeds per-run limit $${this.config.maxCostPerRun}`,
+            };
+        }
+        if (this.config.maxCostPerHour) {
+            const hourly = this.tracker.getHourlyCost();
+            if (hourly + estimatedCost > this.config.maxCostPerHour) {
+                this.triggerExceeded(hourly + estimatedCost, this.config.maxCostPerHour);
+                return {
+                    allowed: false,
+                    reason: `Would exceed hourly budget ($${hourly.toFixed(2)} + $${estimatedCost.toFixed(4)} > $${this.config.maxCostPerHour})`,
+                };
+            }
+            this.checkHourlyWarning(hourly);
+        }
+        if (this.config.maxCostPerDay) {
+            const daily = this.tracker.getDailyCost();
+            if (daily + estimatedCost > this.config.maxCostPerDay) {
+                this.triggerExceeded(daily + estimatedCost, this.config.maxCostPerDay);
+                return {
+                    allowed: false,
+                    reason: `Would exceed daily budget ($${daily.toFixed(2)} + $${estimatedCost.toFixed(4)} > $${this.config.maxCostPerDay})`,
+                };
+            }
+            this.checkDailyWarning(daily);
+        }
+        return { allowed: true };
+    }
+    getBudgetStatus() {
+        const hourlyUsed = this.tracker.getHourlyCost();
+        const dailyUsed = this.tracker.getDailyCost();
+        return {
+            hourlyUsed,
+            hourlyLimit: this.config.maxCostPerHour,
+            dailyUsed,
+            dailyLimit: this.config.maxCostPerDay,
+            hourlyRemaining: this.config.maxCostPerHour
+                ? Math.max(0, this.config.maxCostPerHour - hourlyUsed)
+                : undefined,
+            dailyRemaining: this.config.maxCostPerDay
+                ? Math.max(0, this.config.maxCostPerDay - dailyUsed)
+                : undefined,
+        };
+    }
+    resetWarnings() {
+        this.warningTriggered.hourly = false;
+        this.warningTriggered.daily = false;
+    }
+    checkHourlyWarning(current) {
+        if (!this.config.maxCostPerHour || this.warningTriggered.hourly)
+            return;
+        const threshold = this.config.warningThreshold ?? 0.8;
+        if (current / this.config.maxCostPerHour >= threshold) {
+            this.warningTriggered.hourly = true;
+            this.config.onBudgetWarning?.(current, this.config.maxCostPerHour);
+        }
+    }
+    checkDailyWarning(current) {
+        if (!this.config.maxCostPerDay || this.warningTriggered.daily)
+            return;
+        const threshold = this.config.warningThreshold ?? 0.8;
+        if (current / this.config.maxCostPerDay >= threshold) {
+            this.warningTriggered.daily = true;
+            this.config.onBudgetWarning?.(current, this.config.maxCostPerDay);
+        }
+    }
+    triggerExceeded(current, limit) {
+        this.config.onBudgetExceeded?.(current, limit);
+    }
+}
+//# sourceMappingURL=budget-enforcer.js.map

package/dist/cost-routing/budget-enforcer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-enforcer.js","sourceRoot":"","sources":["../../src/cost-routing/budget-enforcer.ts"],"names":[],"mappings":"AAQA,MAAM,OAAO,cAAc;IACjB,MAAM,CAAe;IACrB,OAAO,CAAc;IACrB,gBAAgB,GAAG;QACzB,MAAM,EAAE,KAAK;QACb,KAAK,EAAE,KAAK;KACb,CAAC;IAEF,YAAY,MAAoB,EAAE,OAAoB;QACpD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,WAAW,CAAC,aAAqB;QAC/B,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC3E,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YAC/D,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,mBAAmB,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,2BAA2B,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE;aAC1G,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YAC5C,IAAI,MAAM,GAAG,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;gBACxD,IAAI,CAAC,eAAe,CAAC,MAAM,GAAG,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;gBACzE,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,gCAAgC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG;iBAC7H,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC1C,IAAI,KAAK,GAAG,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBACtD,IAAI,CAAC,eAAe,CAAC,KAAK,GAAG,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;gBACvE,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,+BAA+B,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG;iBAC1H,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,eAAe;QAQb,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QAE9C,OAAO;YACL,UAAU;YACV,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;YACvC,SAAS;YACT,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YACrC,eAAe,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc;gBACzC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,GAAG,UAAU,CAAC;gBACtD,CAAC,CAAC,SAAS;YACb,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;gBACvC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,SAAS,CAAC;gBACpD,CAAC,CAAC,SAAS;SACd,CAAC;IACJ,CAAC;IAED,aAAa;QACX,IAAI,CAAC,gBAAgB,CAAC,MAAM,GAAG,KAAK,CAAC;QACrC,IAAI,CAAC,gBAAgB,CAAC,KAAK,GAAG,KAAK,CAAC;IACtC,CAAC;IAEO,kBAAkB,CAAC,OAAe;QACxC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,IAAI,CAAC,gBAAgB,CAAC,MAAM;YAAE,OAAO;QAExE,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,GAAG,CAAC;QACtD,IAAI,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,SAAS,EAAE,CAAC;YACtD,IAAI,CAAC,gBAAgB,CAAC,MAAM,GAAG,IAAI,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,OAAe;QACvC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,IAAI,CAAC,gBAAgB,CAAC,KAAK;YAAE,OAAO;QAEtE,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,GAAG,CAAC;QACtD,IAAI,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,SAAS,EAAE,CAAC;YACrD,IAAI,CAAC,gBAAgB,CAAC,KAAK,GAAG,IAAI,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,OAAe,EAAE,KAAa;QACpD,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACjD,CAAC;CACF"}

package/dist/cost-routing/cost-router.d.ts

@@ -0,0 +1,34 @@
+import type { CostRoutingConfig, TaskRequirements, ModelRecommendation, CostRecord, CostSummary } from '@cogitator-ai/types';
+import { type BudgetCheckResult } from './budget-enforcer';
+export interface CostAwareRouterOptions {
+    config?: Partial<CostRoutingConfig>;
+}
+export declare class CostAwareRouter {
+    private taskAnalyzer;
+    private modelSelector;
+    private costTracker;
+    private budgetEnforcer?;
+    private config;
+    constructor(options?: CostAwareRouterOptions);
+    analyzeTask(input: string): TaskRequirements;
+    recommendModel(input: string): Promise<ModelRecommendation>;
+    recommendModelForRequirements(requirements: TaskRequirements): Promise<ModelRecommendation>;
+    checkBudget(estimatedCost: number): BudgetCheckResult;
+    recordCost(record: Omit<CostRecord, 'timestamp'>): void;
+    getRunCost(runId: string): number;
+    getHourlyCost(): number;
+    getDailyCost(): number;
+    getCostSummary(): CostSummary;
+    getBudgetStatus(): {
+        hourlyUsed: number;
+        hourlyLimit?: number;
+        dailyUsed: number;
+        dailyLimit?: number;
+        hourlyRemaining?: number;
+        dailyRemaining?: number;
+    } | undefined;
+    getConfig(): CostRoutingConfig;
+    updateConfig(config: Partial<CostRoutingConfig>): void;
+    clearCostHistory(): void;
+}
+//# sourceMappingURL=cost-router.d.ts.map

package/dist/cost-routing/cost-router.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-router.d.ts","sourceRoot":"","sources":["../../src/cost-routing/cost-router.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,mBAAmB,EACnB,UAAU,EACV,WAAW,EACZ,MAAM,qBAAqB,CAAC;AAI7B,OAAO,EAAkB,KAAK,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAE3E,MAAM,WAAW,sBAAsB;IACrC,MAAM,CAAC,EAAE,OAAO,CAAC,iBAAiB,CAAC,CAAC;CACrC;AAUD,qBAAa,eAAe;IAC1B,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,MAAM,CAAoB;gBAEtB,OAAO,GAAE,sBAA2B;IAWhD,WAAW,CAAC,KAAK,EAAE,MAAM,GAAG,gBAAgB;IAItC,cAAc,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAK3D,6BAA6B,CACjC,YAAY,EAAE,gBAAgB,GAC7B,OAAO,CAAC,mBAAmB,CAAC;IAI/B,WAAW,CAAC,aAAa,EAAE,MAAM,GAAG,iBAAiB;IAKrD,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,GAAG,IAAI;IAMvD,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAIjC,aAAa,IAAI,MAAM;IAIvB,YAAY,IAAI,MAAM;IAItB,cAAc,IAAI,WAAW;IAI7B,eAAe;;;;;;;;IAIf,SAAS,IAAI,iBAAiB;IAI9B,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,iBAAiB,CAAC,GAAG,IAAI;IAWtD,gBAAgB,IAAI,IAAI;CAIzB"}

package/dist/cost-routing/cost-router.js

@@ -0,0 +1,80 @@
+import { TaskAnalyzer } from './task-analyzer';
+import { ModelSelector } from './model-selector';
+import { CostTracker } from './cost-tracker';
+import { BudgetEnforcer } from './budget-enforcer';
+const DEFAULT_CONFIG = {
+    enabled: true,
+    autoSelectModel: false,
+    preferLocal: true,
+    minCapabilityMatch: 0.3,
+    trackCosts: true,
+};
+export class CostAwareRouter {
+    taskAnalyzer;
+    modelSelector;
+    costTracker;
+    budgetEnforcer;
+    config;
+    constructor(options = {}) {
+        this.config = { ...DEFAULT_CONFIG, ...options.config };
+        this.taskAnalyzer = new TaskAnalyzer();
+        this.modelSelector = new ModelSelector(this.config);
+        this.costTracker = new CostTracker();
+        if (this.config.budget) {
+            this.budgetEnforcer = new BudgetEnforcer(this.config.budget, this.costTracker);
+        }
+    }
+    analyzeTask(input) {
+        return this.taskAnalyzer.analyze(input);
+    }
+    async recommendModel(input) {
+        const requirements = this.analyzeTask(input);
+        return this.modelSelector.selectModel(requirements);
+    }
+    async recommendModelForRequirements(requirements) {
+        return this.modelSelector.selectModel(requirements);
+    }
+    checkBudget(estimatedCost) {
+        if (!this.budgetEnforcer)
+            return { allowed: true };
+        return this.budgetEnforcer.checkBudget(estimatedCost);
+    }
+    recordCost(record) {
+        if (this.config.trackCosts) {
+            this.costTracker.record(record);
+        }
+    }
+    getRunCost(runId) {
+        return this.costTracker.getRunCost(runId);
+    }
+    getHourlyCost() {
+        return this.costTracker.getHourlyCost();
+    }
+    getDailyCost() {
+        return this.costTracker.getDailyCost();
+    }
+    getCostSummary() {
+        return this.costTracker.getSummary();
+    }
+    getBudgetStatus() {
+        return this.budgetEnforcer?.getBudgetStatus();
+    }
+    getConfig() {
+        return { ...this.config };
+    }
+    updateConfig(config) {
+        this.config = { ...this.config, ...config };
+        this.modelSelector = new ModelSelector(this.config);
+        if (this.config.budget) {
+            this.budgetEnforcer = new BudgetEnforcer(this.config.budget, this.costTracker);
+        }
+        else {
+            this.budgetEnforcer = undefined;
+        }
+    }
+    clearCostHistory() {
+        this.costTracker.clear();
+        this.budgetEnforcer?.resetWarnings();
+    }
+}
+//# sourceMappingURL=cost-router.js.map

package/dist/cost-routing/cost-router.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-router.js","sourceRoot":"","sources":["../../src/cost-routing/cost-router.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,cAAc,EAA0B,MAAM,mBAAmB,CAAC;AAM3E,MAAM,cAAc,GAAsB;IACxC,OAAO,EAAE,IAAI;IACb,eAAe,EAAE,KAAK;IACtB,WAAW,EAAE,IAAI;IACjB,kBAAkB,EAAE,GAAG;IACvB,UAAU,EAAE,IAAI;CACjB,CAAC;AAEF,MAAM,OAAO,eAAe;IAClB,YAAY,CAAe;IAC3B,aAAa,CAAgB;IAC7B,WAAW,CAAc;IACzB,cAAc,CAAkB;IAChC,MAAM,CAAoB;IAElC,YAAY,UAAkC,EAAE;QAC9C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACvD,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;QACvC,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;QAErC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;IAED,WAAW,CAAC,KAAa;QACvB,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAa;QAChC,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAC7C,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,6BAA6B,CACjC,YAA8B;QAE9B,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IACtD,CAAC;IAED,WAAW,CAAC,aAAqB;QAC/B,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QACnD,OAAO,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC;IACxD,CAAC;IAED,UAAU,CAAC,MAAqC;QAC9C,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;YAC3B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;IAC1C,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;IACzC,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;IACvC,CAAC;IAED,eAAe;QACb,OAAO,IAAI,CAAC,cAAc,EAAE,eAAe,EAAE,CAAC;IAChD,CAAC;IAED,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED,YAAY,CAAC,MAAkC;QAC7C,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QAC5C,IAAI,CAAC,aAAa,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEpD,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACjF,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAClC,CAAC;IACH,CAAC;IAED,gBAAgB;QACd,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,cAAc,EAAE,aAAa,EAAE,CAAC;IACvC,CAAC;CACF"}

package/dist/cost-routing/cost-tracker.d.ts

@@ -0,0 +1,20 @@
+import type { CostRecord, CostSummary } from '@cogitator-ai/types';
+export interface CostFilter {
+    agentId?: string;
+    runId?: string;
+    since?: Date;
+}
+export declare class CostTracker {
+    private records;
+    private hourlyWindow;
+    private dailyWindow;
+    record(record: Omit<CostRecord, 'timestamp'>): void;
+    getRunCost(runId: string): number;
+    getHourlyCost(): number;
+    getDailyCost(): number;
+    getSummary(filter?: CostFilter): CostSummary;
+    getRecords(filter?: CostFilter): CostRecord[];
+    clear(): void;
+    private pruneWindows;
+}
+//# sourceMappingURL=cost-tracker.d.ts.map

package/dist/cost-routing/cost-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cost-tracker.d.ts","sourceRoot":"","sources":["../../src/cost-routing/cost-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAKnE,MAAM,WAAW,UAAU;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,IAAI,CAAC;CACd;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,YAAY,CAAoB;IACxC,OAAO,CAAC,WAAW,CAAoB;IAEvC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,UAAU,EAAE,WAAW,CAAC,GAAG,IAAI;IAQnD,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM;IAIjC,aAAa,IAAI,MAAM;IAKvB,YAAY,IAAI,MAAM;IAKtB,UAAU,CAAC,MAAM,CAAC,EAAE,UAAU,GAAG,WAAW;IA0C5C,UAAU,CAAC,MAAM,CAAC,EAAE,UAAU,GAAG,UAAU,EAAE;IAiB7C,KAAK,IAAI,IAAI;IAMb,OAAO,CAAC,YAAY;CAKrB"}