@cogitator-ai/core 0.1.0 → 0.3.0

package/dist/cost-routing/model-selector.js

@@ -0,0 +1,216 @@
+import { getModelRegistry } from '@cogitator-ai/models';
+const ADVANCED_MODELS = [
+    'gpt-4o',
+    'gpt-4-turbo',
+    'claude-3-opus',
+    'claude-sonnet-4',
+    'claude-3-5-sonnet',
+    'gemini-1.5-pro',
+    'llama3.1:70b',
+    'llama3.2:70b',
+    'qwen2.5:72b',
+    'mixtral',
+    'deepseek-r1',
+];
+const FAST_MODELS = [
+    'gpt-4o-mini',
+    'claude-3-5-haiku',
+    'gemini-1.5-flash',
+    'gemini-2.0-flash',
+    'llama3.2:1b',
+    'llama3.2:3b',
+    'phi3',
+    'phi4',
+    'qwen2.5:1.5b',
+    'qwen2.5:3b',
+];
+const CODE_MODELS = [
+    'claude-3-5-sonnet',
+    'claude-sonnet-4',
+    'gpt-4o',
+    'codellama',
+    'starcoder',
+    'deepseek-coder',
+    'qwen2.5-coder',
+];
+export class ModelSelector {
+    config;
+    constructor(config) {
+        this.config = config;
+    }
+    async selectModel(requirements) {
+        const registry = getModelRegistry();
+        await registry.initialize();
+        const candidates = registry.listModels({
+            supportsTools: requirements.needsToolCalling ? true : undefined,
+            supportsVision: requirements.needsVision ? true : undefined,
+            minContextWindow: requirements.needsLongContext ? 32000 : undefined,
+            excludeDeprecated: true,
+        });
+        if (candidates.length === 0) {
+            return this.fallbackRecommendation();
+        }
+        const scored = candidates
+            .map((m) => this.scoreModel(m, requirements))
+            .filter((s) => s.score >= (this.config.minCapabilityMatch ?? 0.3) * 100)
+            .sort((a, b) => b.score - a.score);
+        if (scored.length === 0) {
+            return this.fallbackRecommendation();
+        }
+        if (this.config.preferLocal) {
+            const local = scored.find((s) => this.isLocalModel(s.model.id));
+            if (local && local.score >= scored[0].score * 0.8) {
+                return this.toRecommendation(local, scored, requirements);
+            }
+        }
+        return this.toRecommendation(scored[0], scored, requirements);
+    }
+    scoreModel(model, requirements) {
+        let score = 100;
+        const reasons = [];
+        if (requirements.needsVision && !model.capabilities?.supportsVision) {
+            return { model, score: 0, reasons: ['Does not support vision (required)'] };
+        }
+        if (requirements.needsToolCalling && !model.capabilities?.supportsTools) {
+            return { model, score: 0, reasons: ['Does not support tool calling (required)'] };
+        }
+        if (requirements.needsLongContext && model.contextWindow < 32000) {
+            score -= 30;
+            reasons.push(`Limited context window (${model.contextWindow} tokens)`);
+        }
+        else if (model.contextWindow >= 100000) {
+            score += 5;
+            reasons.push('Large context window');
+        }
+        const isLocal = this.isLocalModel(model.id);
+        if (isLocal) {
+            score += 15;
+            reasons.push('Local model (no API cost)');
+        }
+        if (!isLocal) {
+            const avgCost = (model.pricing.input + model.pricing.output) / 2;
+            if (requirements.costSensitivity === 'high') {
+                if (avgCost > 10) {
+                    score -= 25;
+                    reasons.push('High cost model');
+                }
+                else if (avgCost < 1) {
+                    score += 10;
+                    reasons.push('Cost-effective');
+                }
+            }
+            else if (requirements.costSensitivity === 'low' && avgCost > 10) {
+                score += 5;
+                reasons.push('Premium model (quality prioritized)');
+            }
+        }
+        const isAdvanced = this.isAdvancedModel(model.id);
+        if (requirements.needsReasoning === 'advanced') {
+            if (isAdvanced) {
+                score += 15;
+                reasons.push('Strong reasoning capability');
+            }
+            else {
+                score -= 20;
+                reasons.push('May lack advanced reasoning');
+            }
+        }
+        else if (requirements.needsReasoning === 'basic' && !isAdvanced) {
+            score += 5;
+            reasons.push('Appropriate for basic tasks');
+        }
+        const isFast = this.isFastModel(model.id);
+        if (requirements.needsSpeed === 'fast') {
+            if (isFast || isLocal) {
+                score += 10;
+                reasons.push('Fast response time');
+            }
+            else {
+                score -= 10;
+                reasons.push('May be slower');
+            }
+        }
+        else if (requirements.needsSpeed === 'slow-ok' && isAdvanced) {
+            score += 5;
+            reasons.push('Quality over speed');
+        }
+        if (requirements.domains?.includes('code')) {
+            if (this.isCodeModel(model.id)) {
+                score += 10;
+                reasons.push('Strong coding capability');
+            }
+        }
+        if (requirements.complexity === 'complex' && isAdvanced) {
+            score += 10;
+            reasons.push('Suitable for complex tasks');
+        }
+        else if (requirements.complexity === 'simple' && isFast) {
+            score += 5;
+            reasons.push('Efficient for simple tasks');
+        }
+        return {
+            model,
+            score: Math.max(0, score),
+            reasons,
+        };
+    }
+    toRecommendation(scored, allScored, requirements) {
+        const fallbacks = allScored
+            .filter((s) => s.model.id !== scored.model.id)
+            .slice(0, 3)
+            .map((s) => s.model.id);
+        return {
+            modelId: scored.model.id,
+            provider: scored.model.provider,
+            score: scored.score,
+            reasons: scored.reasons,
+            estimatedCost: this.estimateCost(scored.model, requirements.complexity ?? 'moderate'),
+            fallbacks,
+        };
+    }
+    estimateCost(model, complexity) {
+        const tokenEstimates = {
+            simple: { input: 500, output: 200 },
+            moderate: { input: 2000, output: 1000 },
+            complex: { input: 8000, output: 4000 },
+        };
+        const estimate = tokenEstimates[complexity];
+        const inputCost = (model.pricing.input * estimate.input) / 1_000_000;
+        const outputCost = (model.pricing.output * estimate.output) / 1_000_000;
+        return inputCost + outputCost;
+    }
+    fallbackRecommendation() {
+        return {
+            modelId: 'gpt-4o-mini',
+            provider: 'openai',
+            score: 50,
+            reasons: ['Fallback model - no suitable candidates found'],
+            estimatedCost: 0.0005,
+            fallbacks: ['claude-3-5-haiku', 'gemini-1.5-flash'],
+        };
+    }
+    isLocalModel(modelId) {
+        const lower = modelId.toLowerCase();
+        return (lower.includes('ollama') ||
+            lower.includes('llama') ||
+            lower.includes('phi') ||
+            lower.includes('qwen') ||
+            lower.includes('mixtral') ||
+            lower.includes('codellama') ||
+            lower.includes('deepseek') ||
+            lower.includes('gemma'));
+    }
+    isAdvancedModel(modelId) {
+        const lower = modelId.toLowerCase();
+        return ADVANCED_MODELS.some((m) => lower.includes(m.toLowerCase()));
+    }
+    isFastModel(modelId) {
+        const lower = modelId.toLowerCase();
+        return FAST_MODELS.some((m) => lower.includes(m.toLowerCase()));
+    }
+    isCodeModel(modelId) {
+        const lower = modelId.toLowerCase();
+        return CODE_MODELS.some((m) => lower.includes(m.toLowerCase()));
+    }
+}
+//# sourceMappingURL=model-selector.js.map

package/dist/cost-routing/model-selector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-selector.js","sourceRoot":"","sources":["../../src/cost-routing/model-selector.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAkB,MAAM,sBAAsB,CAAC;AAExE,MAAM,eAAe,GAAG;IACtB,QAAQ;IACR,aAAa;IACb,eAAe;IACf,iBAAiB;IACjB,mBAAmB;IACnB,gBAAgB;IAChB,cAAc;IACd,cAAc;IACd,aAAa;IACb,SAAS;IACT,aAAa;CACd,CAAC;AAEF,MAAM,WAAW,GAAG;IAClB,aAAa;IACb,kBAAkB;IAClB,kBAAkB;IAClB,kBAAkB;IAClB,aAAa;IACb,aAAa;IACb,MAAM;IACN,MAAM;IACN,cAAc;IACd,YAAY;CACb,CAAC;AAEF,MAAM,WAAW,GAAG;IAClB,mBAAmB;IACnB,iBAAiB;IACjB,QAAQ;IACR,WAAW;IACX,WAAW;IACX,gBAAgB;IAChB,eAAe;CAChB,CAAC;AAQF,MAAM,OAAO,aAAa;IAChB,MAAM,CAAoB;IAElC,YAAY,MAAyB;QACnC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,YAA8B;QAC9C,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;QACpC,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;QAE5B,MAAM,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC;YACrC,aAAa,EAAE,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YAC/D,cAAc,EAAE,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YAC3D,gBAAgB,EAAE,YAAY,CAAC,gBAAgB,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACnE,iBAAiB,EAAE,IAAI;SACxB,CAAC,CAAC;QAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC;QACvC,CAAC;QAED,MAAM,MAAM,GAAG,UAAU;aACtB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC;aAC5C,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,IAAI,GAAG,CAAC,GAAG,GAAG,CAAC;aACvE,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAErC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC;QACvC,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YAChE,IAAI,KAAK,IAAI,KAAK,CAAC,KAAK,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,GAAG,EAAE,CAAC;gBAClD,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IAChE,CAAC;IAEO,UAAU,CAAC,KAAgB,EAAE,YAA8B;QACjE,IAAI,KAAK,GAAG,GAAG,CAAC;QAChB,MAAM,OAAO,GAAa,EAAE,CAAC;QAE7B,IAAI,YAAY,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,cAAc,EAAE,CAAC;YACpE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,oCAAoC,CAAC,EAAE,CAAC;QAC9E,CAAC;QAED,IAAI,YAAY,CAAC,gBAAgB,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,aAAa,EAAE,CAAC;YACxE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,0CAA0C,CAAC,EAAE,CAAC;QACpF,CAAC;QAED,IAAI,YAAY,CAAC,gBAAgB,IAAI,KAAK,CAAC,aAAa,GAAG,KAAK,EAAE,CAAC;YACjE,KAAK,IAAI,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CAAC,2BAA2B,KAAK,CAAC,aAAa,UAAU,CAAC,CAAC;QACzE,CAAC;aAAM,IAAI,KAAK,CAAC,aAAa,IAAI,MAAM,EAAE,CAAC;YACzC,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACvC,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,OAAO,EAAE,CAAC;YACZ,KAAK,IAAI,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,IAAI,YAAY,CAAC,eAAe,KAAK,MAAM,EAAE,CAAC;gBAC5C,IAAI,OAAO,GAAG,EAAE,EAAE,CAAC;oBACjB,KAAK,IAAI,EAAE,CAAC;oBACZ,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBAClC,CAAC;qBAAM,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBACvB,KAAK,IAAI,EAAE,CAAC;oBACZ,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;iBAAM,IAAI,YAAY,CAAC,eAAe,KAAK,KAAK,IAAI,OAAO,GAAG,EAAE,EAAE,CAAC;gBAClE,KAAK,IAAI,CAAC,CAAC;gBACX,OAAO,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAClD,IAAI,YAAY,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;YAC/C,IAAI,UAAU,EAAE,CAAC;gBACf,KAAK,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC9C,CAAC;iBAAM,CAAC;gBACN,KAAK,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;aAAM,IAAI,YAAY,CAAC,cAAc,KAAK,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;YAClE,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC1C,IAAI,YAAY,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;YACvC,IAAI,MAAM,IAAI,OAAO,EAAE,CAAC;gBACtB,KAAK,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,KAAK,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;aAAM,IAAI,YAAY,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,EAAE,CAAC;YAC/D,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,YAAY,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,IAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC/B,KAAK,IAAI,EAAE,CAAC;gBACZ,OAAO,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC;QAED,IAAI,YAAY,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,EAAE,CAAC;YACxD,KAAK,IAAI,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,YAAY,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,EAAE,CAAC;YAC1D,KAAK,IAAI,CAAC,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QAC7C,CAAC;QAED,OAAO;YACL,KAAK;YACL,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC;YACzB,OAAO;SACR,CAAC;IACJ,CAAC;IAEO,gBAAgB,CACtB,MAAmB,EACnB,SAAwB,EACxB,YAA8B;QAE9B,MAAM,SAAS,GAAG,SAAS;aACxB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,KAAK,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;aAC7C,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAE1B,OAAO;YACL,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,EAAE;YACxB,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ;YAC/B,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,EAAE,YAAY,CAAC,UAAU,IAAI,UAAU,CAAC;YACrF,SAAS;SACV,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,KAAgB,EAAE,UAA6C;QAClF,MAAM,cAAc,GAAG;YACrB,MAAM,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;YACnC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;YACvC,OAAO,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;SACvC,CAAC;QAEF,MAAM,QAAQ,GAAG,cAAc,CAAC,UAAU,CAAC,CAAC;QAC5C,MAAM,SAAS,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,SAAS,CAAC;QACrE,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;QAExE,OAAO,SAAS,GAAG,UAAU,CAAC;IAChC,CAAC;IAEO,sBAAsB;QAC5B,OAAO;YACL,OAAO,EAAE,aAAa;YACtB,QAAQ,EAAE,QAAQ;YAClB,KAAK,EAAE,EAAE;YACT,OAAO,EAAE,CAAC,+CAA+C,CAAC;YAC1D,aAAa,EAAE,MAAM;YACrB,SAAS,EAAE,CAAC,kBAAkB,EAAE,kBAAkB,CAAC;SACpD,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,OAAe;QAClC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,CACL,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACxB,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;YACvB,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;YACrB,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YACtB,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;YACzB,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC;YAC3B,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;YAC1B,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CACxB,CAAC;IACJ,CAAC;IAEO,eAAe,CAAC,OAAe;QACrC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IACtE,CAAC;IAEO,WAAW,CAAC,OAAe;QACjC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC;IAEO,WAAW,CAAC,OAAe;QACjC,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;IAClE,CAAC;CACF"}

package/dist/cost-routing/task-analyzer.d.ts

@@ -0,0 +1,13 @@
+import type { TaskRequirements } from '@cogitator-ai/types';
+export declare class TaskAnalyzer {
+    analyze(task: string): TaskRequirements;
+    private detectVision;
+    private detectToolNeeds;
+    private detectLongContext;
+    private detectReasoningLevel;
+    private detectSpeedNeeds;
+    private detectCostSensitivity;
+    private detectComplexity;
+    private detectDomains;
+}
+//# sourceMappingURL=task-analyzer.d.ts.map

package/dist/cost-routing/task-analyzer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-analyzer.d.ts","sourceRoot":"","sources":["../../src/cost-routing/task-analyzer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,gBAAgB,EAKjB,MAAM,qBAAqB,CAAC;AA4F7B,qBAAa,YAAY;IACvB,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,gBAAgB;IAevC,OAAO,CAAC,YAAY;IAIpB,OAAO,CAAC,eAAe;IAIvB,OAAO,CAAC,iBAAiB;IAsBzB,OAAO,CAAC,oBAAoB;IAU5B,OAAO,CAAC,gBAAgB;IAUxB,OAAO,CAAC,qBAAqB;IAW7B,OAAO,CAAC,gBAAgB;IAkBxB,OAAO,CAAC,aAAa;CAStB"}

package/dist/cost-routing/task-analyzer.js

@@ -0,0 +1,185 @@
+const VISION_KEYWORDS = [
+    'image',
+    'photo',
+    'picture',
+    'screenshot',
+    'diagram',
+    'chart',
+    'graph',
+    'visual',
+    'see',
+    'look at',
+    'analyze this',
+    'show me',
+    'drawing',
+    'illustration',
+    'png',
+    'jpg',
+    'jpeg',
+    'gif',
+    'svg',
+];
+const TOOL_KEYWORDS = [
+    'search',
+    'calculate',
+    'execute',
+    'run',
+    'fetch',
+    'api',
+    'database',
+    'file',
+    'code',
+    'browse',
+    'query',
+    'call',
+    'invoke',
+    'request',
+    'download',
+    'upload',
+    'scrape',
+    'crawl',
+];
+const ADVANCED_REASONING_KEYWORDS = [
+    'analyze',
+    'compare',
+    'evaluate',
+    'synthesize',
+    'critique',
+    'design',
+    'architect',
+    'optimize',
+    'reason',
+    'deduce',
+    'infer',
+    'prove',
+    'derive',
+    'solve complex',
+    'multi-step',
+    'strategic',
+];
+const MODERATE_REASONING_KEYWORDS = [
+    'explain',
+    'summarize',
+    'describe',
+    'outline',
+    'list',
+    'categorize',
+    'classify',
+    'organize',
+    'plan',
+];
+const SPEED_KEYWORDS = {
+    fast: ['quick', 'fast', 'asap', 'urgent', 'immediately', 'brief', 'short'],
+    slow: ['detailed', 'thorough', 'comprehensive', 'in-depth', 'extensive', 'careful'],
+};
+const DOMAIN_PATTERNS = {
+    code: /\b(code|program|function|class|api|typescript|javascript|python|rust|go|java|c\+\+|implement|debug|refactor|test|compile|build)\b/i,
+    math: /\b(math|calculate|equation|formula|statistics|probability|algebra|calculus|numerical|compute)\b/i,
+    creative: /\b(write|story|article|creative|blog|content|poem|essay|narrative|fiction|script|dialogue)\b/i,
+    analysis: /\b(analyze|research|data|report|study|investigate|examine|assess|evaluate|review)\b/i,
+    legal: /\b(legal|law|contract|compliance|regulation|policy|statute|liability|terms)\b/i,
+    medical: /\b(medical|health|clinical|diagnosis|treatment|symptom|patient|doctor|healthcare)\b/i,
+    finance: /\b(finance|financial|investment|stock|trading|budget|accounting|revenue|profit)\b/i,
+};
+export class TaskAnalyzer {
+    analyze(task) {
+        const lower = task.toLowerCase();
+        return {
+            needsVision: this.detectVision(lower),
+            needsToolCalling: this.detectToolNeeds(lower),
+            needsLongContext: this.detectLongContext(task),
+            needsReasoning: this.detectReasoningLevel(lower),
+            needsSpeed: this.detectSpeedNeeds(lower),
+            costSensitivity: this.detectCostSensitivity(lower),
+            complexity: this.detectComplexity(task),
+            domains: this.detectDomains(task),
+        };
+    }
+    detectVision(task) {
+        return VISION_KEYWORDS.some((k) => task.includes(k));
+    }
+    detectToolNeeds(task) {
+        return TOOL_KEYWORDS.some((k) => task.includes(k));
+    }
+    detectLongContext(task) {
+        const wordCount = task.split(/\s+/).length;
+        if (wordCount > 500)
+            return true;
+        const longContextIndicators = [
+            'document',
+            'file',
+            'codebase',
+            'repository',
+            'large',
+            'entire',
+            'full',
+            'complete',
+            'all of',
+            'whole',
+            'multiple files',
+            'many pages',
+        ];
+        const lower = task.toLowerCase();
+        return longContextIndicators.some((k) => lower.includes(k));
+    }
+    detectReasoningLevel(task) {
+        if (ADVANCED_REASONING_KEYWORDS.some((k) => task.includes(k))) {
+            return 'advanced';
+        }
+        if (MODERATE_REASONING_KEYWORDS.some((k) => task.includes(k))) {
+            return 'moderate';
+        }
+        return 'basic';
+    }
+    detectSpeedNeeds(task) {
+        if (SPEED_KEYWORDS.fast.some((k) => task.includes(k))) {
+            return 'fast';
+        }
+        if (SPEED_KEYWORDS.slow.some((k) => task.includes(k))) {
+            return 'slow-ok';
+        }
+        return 'balanced';
+    }
+    detectCostSensitivity(task) {
+        const lower = task;
+        if (/\b(cheap|free|budget|low.?cost|economical)\b/.test(lower)) {
+            return 'high';
+        }
+        if (/\b(best|premium|quality|accurate|precise)\b/.test(lower)) {
+            return 'low';
+        }
+        return 'medium';
+    }
+    detectComplexity(task) {
+        const wordCount = task.split(/\s+/).length;
+        const sentenceCount = task.split(/[.!?]+/).filter((s) => s.trim()).length;
+        const hasMultipleSteps = /\b(then|after|next|finally|first|second|third|step)\b/i.test(task);
+        const hasConditions = /\b(if|when|unless|otherwise|either|or)\b/i.test(task);
+        let complexityScore = 0;
+        if (wordCount > 100)
+            complexityScore++;
+        if (wordCount > 300)
+            complexityScore++;
+        if (sentenceCount > 5)
+            complexityScore++;
+        if (hasMultipleSteps)
+            complexityScore++;
+        if (hasConditions)
+            complexityScore++;
+        if (complexityScore >= 3)
+            return 'complex';
+        if (complexityScore >= 1)
+            return 'moderate';
+        return 'simple';
+    }
+    detectDomains(task) {
+        const domains = [];
+        for (const [domain, pattern] of Object.entries(DOMAIN_PATTERNS)) {
+            if (pattern.test(task)) {
+                domains.push(domain);
+            }
+        }
+        return domains;
+    }
+}
+//# sourceMappingURL=task-analyzer.js.map

package/dist/cost-routing/task-analyzer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-analyzer.js","sourceRoot":"","sources":["../../src/cost-routing/task-analyzer.ts"],"names":[],"mappings":"AAQA,MAAM,eAAe,GAAG;IACtB,OAAO;IACP,OAAO;IACP,SAAS;IACT,YAAY;IACZ,SAAS;IACT,OAAO;IACP,OAAO;IACP,QAAQ;IACR,KAAK;IACL,SAAS;IACT,cAAc;IACd,SAAS;IACT,SAAS;IACT,cAAc;IACd,KAAK;IACL,KAAK;IACL,MAAM;IACN,KAAK;IACL,KAAK;CACN,CAAC;AAEF,MAAM,aAAa,GAAG;IACpB,QAAQ;IACR,WAAW;IACX,SAAS;IACT,KAAK;IACL,OAAO;IACP,KAAK;IACL,UAAU;IACV,MAAM;IACN,MAAM;IACN,QAAQ;IACR,OAAO;IACP,MAAM;IACN,QAAQ;IACR,SAAS;IACT,UAAU;IACV,QAAQ;IACR,QAAQ;IACR,OAAO;CACR,CAAC;AAEF,MAAM,2BAA2B,GAAG;IAClC,SAAS;IACT,SAAS;IACT,UAAU;IACV,YAAY;IACZ,UAAU;IACV,QAAQ;IACR,WAAW;IACX,UAAU;IACV,QAAQ;IACR,QAAQ;IACR,OAAO;IACP,OAAO;IACP,QAAQ;IACR,eAAe;IACf,YAAY;IACZ,WAAW;CACZ,CAAC;AAEF,MAAM,2BAA2B,GAAG;IAClC,SAAS;IACT,WAAW;IACX,UAAU;IACV,SAAS;IACT,MAAM;IACN,YAAY;IACZ,UAAU;IACV,UAAU;IACV,MAAM;CACP,CAAC;AAEF,MAAM,cAAc,GAAG;IACrB,IAAI,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,CAAC;IAC1E,IAAI,EAAE,CAAC,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,CAAC;CACpF,CAAC;AAEF,MAAM,eAAe,GAA2B;IAC9C,IAAI,EAAE,oIAAoI;IAC1I,IAAI,EAAE,kGAAkG;IACxG,QAAQ,EACN,+FAA+F;IACjG,QAAQ,EAAE,sFAAsF;IAChG,KAAK,EAAE,gFAAgF;IACvF,OAAO,EAAE,sFAAsF;IAC/F,OAAO,EAAE,oFAAoF;CAC9F,CAAC;AAEF,MAAM,OAAO,YAAY;IACvB,OAAO,CAAC,IAAY;QAClB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAEjC,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;YACrC,gBAAgB,EAAE,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;YAC7C,gBAAgB,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC;YAC9C,cAAc,EAAE,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC;YAChD,UAAU,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACxC,eAAe,EAAE,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC;YAClD,UAAU,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC;YACvC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;SAClC,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,IAAY;QAC/B,OAAO,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC;IAEO,eAAe,CAAC,IAAY;QAClC,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACrD,CAAC;IAEO,iBAAiB,CAAC,IAAY;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;QAC3C,IAAI,SAAS,GAAG,GAAG;YAAE,OAAO,IAAI,CAAC;QAEjC,MAAM,qBAAqB,GAAG;YAC5B,UAAU;YACV,MAAM;YACN,UAAU;YACV,YAAY;YACZ,OAAO;YACP,QAAQ;YACR,MAAM;YACN,UAAU;YACV,QAAQ;YACR,OAAO;YACP,gBAAgB;YAChB,YAAY;SACb,CAAC;QACF,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACjC,OAAO,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9D,CAAC;IAEO,oBAAoB,CAAC,IAAY;QACvC,IAAI,2BAA2B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,IAAI,2BAA2B,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9D,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB,CAAC,IAAY;QACnC,IAAI,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACtD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,qBAAqB,CAAC,IAAY;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC;QACnB,IAAI,8CAA8C,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/D,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,IAAI,6CAA6C,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9D,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,gBAAgB,CAAC,IAAY;QACnC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;QAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QAC1E,MAAM,gBAAgB,GAAG,wDAAwD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7F,MAAM,aAAa,GAAG,2CAA2C,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE7E,IAAI,eAAe,GAAG,CAAC,CAAC;QACxB,IAAI,SAAS,GAAG,GAAG;YAAE,eAAe,EAAE,CAAC;QACvC,IAAI,SAAS,GAAG,GAAG;YAAE,eAAe,EAAE,CAAC;QACvC,IAAI,aAAa,GAAG,CAAC;YAAE,eAAe,EAAE,CAAC;QACzC,IAAI,gBAAgB;YAAE,eAAe,EAAE,CAAC;QACxC,IAAI,aAAa;YAAE,eAAe,EAAE,CAAC;QAErC,IAAI,eAAe,IAAI,CAAC;YAAE,OAAO,SAAS,CAAC;QAC3C,IAAI,eAAe,IAAI,CAAC;YAAE,OAAO,UAAU,CAAC;QAC5C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,aAAa,CAAC,IAAY;QAChC,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;YAChE,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBACvB,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;CACF"}

package/dist/index.d.ts

@@ -10,10 +10,28 @@ export { ToolRegistry } from './registry';
 export { calculator, datetime, builtinTools } from './tools/index';
 export { Logger, getLogger, setLogger, createLogger } from './logger';
 export type { LogLevel, LogContext, LogEntry, LoggerOptions } from './logger';
+export { ReflectionEngine, InMemoryInsightStore } from './reflection/index';
+export type { ReflectionEngineOptions } from './reflection/index';
+export { ThoughtTreeExecutor, BranchGenerator, BranchEvaluator } from './reasoning/index';
+export type { BranchEvaluatorOptions } from './reasoning/index';
+export { AgentOptimizer, InMemoryTraceStore, MetricEvaluator, DemoSelector, InstructionOptimizer, createSuccessMetric, createExactMatchMetric, createContainsMetric, } from './learning/index';
+export type { AgentOptimizerOptions, MetricEvaluatorOptions, DemoSelectorOptions, InstructionOptimizerOptions, } from './learning/index';
+export { TimeTravel, InMemoryCheckpointStore, ExecutionReplayer, ExecutionForker, TraceComparator, } from './time-travel/index';
+export type { TimeTravelOptions, ExecutionReplayerOptions, ExecutionForkerOptions, TraceComparatorOptions, } from './time-travel/index';
+export { ConstitutionalAI, InputFilter, OutputFilter, ToolGuard, CritiqueReviser, DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './constitutional/index';
+export type { ConstitutionalAIOptions, InputFilterOptions, OutputFilterOptions, ToolGuardOptions, CritiqueReviserOptions, } from './constitutional/index';
+export { CostAwareRouter, TaskAnalyzer, ModelSelector, CostTracker, BudgetEnforcer, } from './cost-routing/index';
+export type { CostAwareRouterOptions, CostFilter, BudgetCheckResult } from './cost-routing/index';
 export { BaseLLMBackend, OllamaBackend, OpenAIBackend, AnthropicBackend, createLLMBackend, parseModel, } from './llm/index';
 export { withRetry, retryable, CircuitBreaker, CircuitBreakerRegistry, withFallback, withGracefulDegradation, createLLMFallbackExecutor, } from './utils/index';
 export type { RetryOptions, CircuitBreakerOptions, CircuitBreakerStats, CircuitState, FallbackConfig, LLMFallbackConfig, } from './utils/index';
 export { CogitatorError, ErrorCode, ERROR_STATUS_CODES, isRetryableError, getRetryDelay, } from '@cogitator-ai/types';
 export type { ErrorDetails, CogitatorErrorOptions } from '@cogitator-ai/types';
-export type { AgentConfig, ResponseFormat, Tool, ToolConfig, ToolContext, ToolSchema, Message, MessageRole, ToolCall, ToolResult, LLMBackend, LLMProvider, LLMConfig, ChatRequest, ChatResponse, ChatStreamChunk, CogitatorConfig, RunOptions, RunResult, Span, } from '@cogitator-ai/types';
+export type { AgentConfig, ResponseFormat, Tool, ToolConfig, ToolContext, ToolSchema, Message, MessageRole, ToolCall, ToolResult, LLMBackend, LLMProvider, LLMConfig, ChatRequest, ChatResponse, ChatStreamChunk, CogitatorConfig, RunOptions, RunResult, Span, ToTConfig, ToTResult, ToTStats, ToTRunOptions, ThoughtTree, ThoughtNode, ThoughtBranch, BranchScore, ProposedAction, ExplorationStrategy, ExecutionTrace, ExecutionStep, TraceStore, TraceMetrics, TraceQuery, TraceStoreStats, Demo, DemoStep, DemoStats, MetricResult, MetricFn, MetricDefinition, MetricEvaluatorConfig, BuiltinMetric, InstructionGap, InstructionOptimizationResult, OptimizerConfig, OptimizationResult, LearningConfig, LearningRunOptions, LearningRunResult, CompileOptions, LearningStats, } from '@cogitator-ai/types';
+export { DEFAULT_TOT_CONFIG, DEFAULT_LEARNING_CONFIG, DEFAULT_OPTIMIZER_CONFIG, DEFAULT_TIME_TRAVEL_CONFIG, } from '@cogitator-ai/types';
+export type { ExecutionCheckpoint, TimeTravelCheckpointStore, CheckpointQuery, ReplayOptions, ReplayResult, ReplayMode, ForkOptions, ForkResult, TraceDiff, StepDiff, StepDiffStatus, TimeTravelConfig, } from '@cogitator-ai/types';
+export type { HarmCategory, Severity, FilterLayer, PrincipleCategory, ConstitutionalPrinciple, Constitution, HarmScore, FilterResult, CritiqueResult, RevisionResult, GuardrailConfig, ToolGuardResult, } from '@cogitator-ai/types';
+export { DEFAULT_GUARDRAIL_CONFIG } from '@cogitator-ai/types';
+export type { CostRoutingConfig, BudgetConfig, CostRecord, CostSummary, ModelRecommendation, TaskRequirements, TaskComplexity, ReasoningLevel, SpeedPreference, CostSensitivity, } from '@cogitator-ai/types';
+export { DEFAULT_COST_ROUTING_CONFIG } from '@cogitator-ai/types';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEnE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACtE,YAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAE9E,OAAO,EACL,cAAc,EACd,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,UAAU,GACX,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,SAAS,EACT,SAAS,EACT,cAAc,EACd,sBAAsB,EACtB,YAAY,EACZ,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,eAAe,CAAC;AACvB,YAAY,EACV,YAAY,EACZ,qBAAqB,EACrB,mBAAmB,EACnB,YAAY,EACZ,cAAc,EACd,iBAAiB,GAClB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,cAAc,EACd,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,GACd,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,YAAY,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAE/E,YAAY,EACV,WAAW,EACX,cAAc,EACd,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,OAAO,EACP,WAAW,EACX,QAAQ,EACR,UAAU,EACV,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,YAAY,EACZ,eAAe,EACf,eAAe,EACf,UAAU,EACV,SAAS,EACT,IAAI,GACL,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEnE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACtE,YAAY,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAE9E,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC5E,YAAY,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAElE,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAC1F,YAAY,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAEhE,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,eAAe,EACf,YAAY,EACZ,oBAAoB,EACpB,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,GACrB,MAAM,kBAAkB,CAAC;AAC1B,YAAY,EACV,qBAAqB,EACrB,sBAAsB,EACtB,mBAAmB,EACnB,2BAA2B,GAC5B,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EACL,UAAU,EACV,uBAAuB,EACvB,iBAAiB,EACjB,eAAe,EACf,eAAe,GAChB,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EACV,iBAAiB,EACjB,wBAAwB,EACxB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,gBAAgB,EAChB,WAAW,EACX,YAAY,EACZ,SAAS,EACT,eAAe,EACf,oBAAoB,EACpB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,uBAAuB,EACvB,uBAAuB,EACvB,uBAAuB,EACvB,0BAA0B,EAC1B,2BAA2B,EAC3B,mBAAmB,EACnB,mBAAmB,EACnB,uBAAuB,EACvB,qBAAqB,GACtB,MAAM,wBAAwB,CAAC;AAChC,YAAY,EACV,uBAAuB,EACvB,kBAAkB,EAClB,mBAAmB,EACnB,gBAAgB,EAChB,sBAAsB,GACvB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EACL,eAAe,EACf,YAAY,EACZ,aAAa,EACb,WAAW,EACX,cAAc,GACf,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EAAE,sBAAsB,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAElG,OAAO,EACL,cAAc,EACd,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,UAAU,GACX,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,SAAS,EACT,SAAS,EACT,cAAc,EACd,sBAAsB,EACtB,YAAY,EACZ,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,eAAe,CAAC;AACvB,YAAY,EACV,YAAY,EACZ,qBAAqB,EACrB,mBAAmB,EACnB,YAAY,EACZ,cAAc,EACd,iBAAiB,GAClB,MAAM,eAAe,CAAC;AAEvB,OAAO,EACL,cAAc,EACd,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,GACd,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EAAE,YAAY,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAE/E,YAAY,EACV,WAAW,EACX,cAAc,EACd,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,OAAO,EACP,WAAW,EACX,QAAQ,EACR,UAAU,EACV,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,YAAY,EACZ,eAAe,EACf,eAAe,EACf,UAAU,EACV,SAAS,EACT,IAAI,EACJ,SAAS,EACT,SAAS,EACT,QAAQ,EACR,aAAa,EACb,WAAW,EACX,WAAW,EACX,aAAa,EACb,WAAW,EACX,cAAc,EACd,mBAAmB,EACnB,cAAc,EACd,aAAa,EACb,UAAU,EACV,YAAY,EACZ,UAAU,EACV,eAAe,EACf,IAAI,EACJ,QAAQ,EACR,SAAS,EACT,YAAY,EACZ,QAAQ,EACR,gBAAgB,EAChB,qBAAqB,EACrB,aAAa,EACb,cAAc,EACd,6BAA6B,EAC7B,eAAe,EACf,kBAAkB,EAClB,cAAc,EACd,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,aAAa,GACd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,kBAAkB,EAClB,uBAAuB,EACvB,wBAAwB,EACxB,0BAA0B,GAC3B,MAAM,qBAAqB,CAAC;AAE7B,YAAY,EACV,mBAAmB,EACnB,yBAAyB,EACzB,eAAe,EACf,aAAa,EACb,YAAY,EACZ,UAAU,EACV,WAAW,EACX,UAAU,EACV,SAAS,EACT,QAAQ,EACR,cAAc,EACd,gBAAgB,GACjB,MAAM,qBAAqB,CAAC;AAE7B,YAAY,EACV,YAAY,EACZ,QAAQ,EACR,WAAW,EACX,iBAAiB,EACjB,uBAAuB,EACvB,YAAY,EACZ,SAAS,EACT,YAAY,EACZ,cAAc,EACd,cAAc,EACd,eAAe,EACf,eAAe,GAChB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAE/D,YAAY,EACV,iBAAiB,EACjB,YAAY,EACZ,UAAU,EACV,WAAW,EACX,mBAAmB,EACnB,gBAAgB,EAChB,cAAc,EACd,cAAc,EACd,eAAe,EACf,eAAe,GAChB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/index.js

@@ -9,7 +9,16 @@ export { tool, toolToSchema } from './tool';
 export { ToolRegistry } from './registry';
 export { calculator, datetime, builtinTools } from './tools/index';
 export { Logger, getLogger, setLogger, createLogger } from './logger';
+export { ReflectionEngine, InMemoryInsightStore } from './reflection/index';
+export { ThoughtTreeExecutor, BranchGenerator, BranchEvaluator } from './reasoning/index';
+export { AgentOptimizer, InMemoryTraceStore, MetricEvaluator, DemoSelector, InstructionOptimizer, createSuccessMetric, createExactMatchMetric, createContainsMetric, } from './learning/index';
+export { TimeTravel, InMemoryCheckpointStore, ExecutionReplayer, ExecutionForker, TraceComparator, } from './time-travel/index';
+export { ConstitutionalAI, InputFilter, OutputFilter, ToolGuard, CritiqueReviser, DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './constitutional/index';
+export { CostAwareRouter, TaskAnalyzer, ModelSelector, CostTracker, BudgetEnforcer, } from './cost-routing/index';
 export { BaseLLMBackend, OllamaBackend, OpenAIBackend, AnthropicBackend, createLLMBackend, parseModel, } from './llm/index';
 export { withRetry, retryable, CircuitBreaker, CircuitBreakerRegistry, withFallback, withGracefulDegradation, createLLMFallbackExecutor, } from './utils/index';
 export { CogitatorError, ErrorCode, ERROR_STATUS_CODES, isRetryableError, getRetryDelay, } from '@cogitator-ai/types';
+export { DEFAULT_TOT_CONFIG, DEFAULT_LEARNING_CONFIG, DEFAULT_OPTIMIZER_CONFIG, DEFAULT_TIME_TRAVEL_CONFIG, } from '@cogitator-ai/types';
+export { DEFAULT_GUARDRAIL_CONFIG } from '@cogitator-ai/types';
+export { DEFAULT_COST_ROUTING_CONFIG } from '@cogitator-ai/types';
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEnE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAGtE,OAAO,EACL,cAAc,EACd,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,UAAU,GACX,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,SAAS,EACT,SAAS,EACT,cAAc,EACd,sBAAsB,EACtB,YAAY,EACZ,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,eAAe,CAAC;AAUvB,OAAO,EACL,cAAc,EACd,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,GACd,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE1C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEnE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAGtE,OAAO,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAG5E,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,eAAe,EACf,YAAY,EACZ,oBAAoB,EACpB,mBAAmB,EACnB,sBAAsB,EACtB,oBAAoB,GACrB,MAAM,kBAAkB,CAAC;AAQ1B,OAAO,EACL,UAAU,EACV,uBAAuB,EACvB,iBAAiB,EACjB,eAAe,EACf,eAAe,GAChB,MAAM,qBAAqB,CAAC;AAQ7B,OAAO,EACL,gBAAgB,EAChB,WAAW,EACX,YAAY,EACZ,SAAS,EACT,eAAe,EACf,oBAAoB,EACpB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,EAClB,uBAAuB,EACvB,uBAAuB,EACvB,uBAAuB,EACvB,0BAA0B,EAC1B,2BAA2B,EAC3B,mBAAmB,EACnB,mBAAmB,EACnB,uBAAuB,EACvB,qBAAqB,GACtB,MAAM,wBAAwB,CAAC;AAShC,OAAO,EACL,eAAe,EACf,YAAY,EACZ,aAAa,EACb,WAAW,EACX,cAAc,GACf,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACL,cAAc,EACd,aAAa,EACb,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,UAAU,GACX,MAAM,aAAa,CAAC;AAErB,OAAO,EACL,SAAS,EACT,SAAS,EACT,cAAc,EACd,sBAAsB,EACtB,YAAY,EACZ,uBAAuB,EACvB,yBAAyB,GAC1B,MAAM,eAAe,CAAC;AAUvB,OAAO,EACL,cAAc,EACd,SAAS,EACT,kBAAkB,EAClB,gBAAgB,EAChB,aAAa,GACd,MAAM,qBAAqB,CAAC;AA0D7B,OAAO,EACL,kBAAkB,EAClB,uBAAuB,EACvB,wBAAwB,EACxB,0BAA0B,GAC3B,MAAM,qBAAqB,CAAC;AA+B7B,OAAO,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC;AAc/D,OAAO,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/learning/ab-testing.d.ts

@@ -0,0 +1,45 @@
+import type { ABTest, ABTestStore, ABTestVariant, ABTestOutcome, ABTestingConfig } from '@cogitator-ai/types';
+export interface ABTestingFrameworkConfig extends ABTestingConfig {
+    store: ABTestStore;
+}
+export declare class ABTestingFramework {
+    private store;
+    private config;
+    private activeTests;
+    constructor(config: ABTestingFrameworkConfig);
+    createTest(params: {
+        agentId: string;
+        name: string;
+        description?: string;
+        controlInstructions: string;
+        treatmentInstructions: string;
+        treatmentAllocation?: number;
+        minSampleSize?: number;
+        maxDuration?: number;
+        confidenceLevel?: number;
+        metricToOptimize?: string;
+    }): Promise<ABTest>;
+    startTest(testId: string): Promise<ABTest>;
+    pauseTest(testId: string): Promise<ABTest>;
+    resumeTest(testId: string): Promise<ABTest>;
+    completeTest(testId: string): Promise<{
+        test: ABTest;
+        outcome: ABTestOutcome;
+    }>;
+    cancelTest(testId: string): Promise<ABTest>;
+    getActiveTest(agentId: string): Promise<ABTest | null>;
+    selectVariant(test: ABTest): ABTestVariant;
+    getInstructionsForVariant(test: ABTest, variant: ABTestVariant): string;
+    recordResult(testId: string, variant: ABTestVariant, score: number, latency: number, cost: number): Promise<void>;
+    checkAndCompleteIfReady(testId: string): Promise<ABTestOutcome | null>;
+    analyzeResults(test: ABTest): ABTestOutcome;
+    private emptyResults;
+    private calculateStd;
+    private welchTTest;
+    private tDistributionPValue;
+    private incompleteBeta;
+    private gamma;
+    private tCriticalValue;
+    private tCDF;
+}
+//# sourceMappingURL=ab-testing.d.ts.map

package/dist/learning/ab-testing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ab-testing.d.ts","sourceRoot":"","sources":["../../src/learning/ab-testing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,MAAM,EACN,WAAW,EACX,aAAa,EACb,aAAa,EAEb,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,wBAAyB,SAAQ,eAAe;IAC/D,KAAK,EAAE,WAAW,CAAC;CACpB;AAgBD,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,KAAK,CAAc;IAC3B,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,WAAW,CAA6B;gBAEpC,MAAM,EAAE,wBAAwB;IAUtC,UAAU,CAAC,MAAM,EAAE;QACvB,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,mBAAmB,EAAE,MAAM,CAAC;QAC5B,qBAAqB,EAAE,MAAM,CAAC;QAC9B,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;KAC3B,GAAG,OAAO,CAAC,MAAM,CAAC;IAoBb,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAU1C,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAS1C,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAS3C,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,aAAa,CAAA;KAAE,CAAC;IAa/E,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAU3C,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IAa5D,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa;IAI1C,yBAAyB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,GAAG,MAAM;IAIjE,YAAY,CAChB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,aAAa,EACtB,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,GACX,OAAO,CAAC,IAAI,CAAC;IASV,uBAAuB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;IA4B5E,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,aAAa;IAyE3C,OAAO,CAAC,YAAY;IAWpB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,UAAU;IAqBlB,OAAO,CAAC,mBAAmB;IAM3B,OAAO,CAAC,cAAc;IAqBtB,OAAO,CAAC,KAAK;IAsBb,OAAO,CAAC,cAAc;IAoBtB,OAAO,CAAC,IAAI;CAKb"}