@cogitator-ai/core 0.1.0 → 0.3.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +920 -15
- package/dist/__tests__/agent.test.js +2 -2
- package/dist/__tests__/agent.test.js.map +1 -1
- package/dist/__tests__/base64.test.js +1 -1
- package/dist/__tests__/base64.test.js.map +1 -1
- package/dist/__tests__/calculator.test.js +1 -1
- package/dist/__tests__/calculator.test.js.map +1 -1
- package/dist/__tests__/cogitator-memory.test.js +2 -2
- package/dist/__tests__/cogitator-memory.test.js.map +1 -1
- package/dist/__tests__/datetime.test.js +1 -1
- package/dist/__tests__/datetime.test.js.map +1 -1
- package/dist/__tests__/exec.test.js +1 -1
- package/dist/__tests__/exec.test.js.map +1 -1
- package/dist/__tests__/filesystem.test.js +1 -1
- package/dist/__tests__/filesystem.test.js.map +1 -1
- package/dist/__tests__/google-backend.test.js +1 -1
- package/dist/__tests__/google-backend.test.js.map +1 -1
- package/dist/__tests__/hash.test.js +1 -1
- package/dist/__tests__/hash.test.js.map +1 -1
- package/dist/__tests__/http.test.js +1 -1
- package/dist/__tests__/http.test.js.map +1 -1
- package/dist/__tests__/json.test.js +1 -1
- package/dist/__tests__/json.test.js.map +1 -1
- package/dist/__tests__/logger.test.js +1 -1
- package/dist/__tests__/logger.test.js.map +1 -1
- package/dist/__tests__/random.test.js +1 -1
- package/dist/__tests__/random.test.js.map +1 -1
- package/dist/__tests__/regex.test.js +1 -1
- package/dist/__tests__/regex.test.js.map +1 -1
- package/dist/__tests__/registry.test.js +2 -2
- package/dist/__tests__/registry.test.js.map +1 -1
- package/dist/__tests__/sleep.test.js +1 -1
- package/dist/__tests__/sleep.test.js.map +1 -1
- package/dist/__tests__/tool.test.js +1 -1
- package/dist/__tests__/tool.test.js.map +1 -1
- package/dist/__tests__/uuid.test.js +1 -1
- package/dist/__tests__/uuid.test.js.map +1 -1
- package/dist/cogitator.d.ts +46 -1
- package/dist/cogitator.d.ts.map +1 -1
- package/dist/cogitator.js +274 -17
- package/dist/cogitator.js.map +1 -1
- package/dist/constitutional/constitution.d.ts +9 -0
- package/dist/constitutional/constitution.d.ts.map +1 -0
- package/dist/constitutional/constitution.js +215 -0
- package/dist/constitutional/constitution.js.map +1 -0
- package/dist/constitutional/constitutional-ai.d.ts +36 -0
- package/dist/constitutional/constitutional-ai.d.ts.map +1 -0
- package/dist/constitutional/constitutional-ai.js +163 -0
- package/dist/constitutional/constitutional-ai.js.map +1 -0
- package/dist/constitutional/critique-reviser.d.ts +20 -0
- package/dist/constitutional/critique-reviser.d.ts.map +1 -0
- package/dist/constitutional/critique-reviser.js +98 -0
- package/dist/constitutional/critique-reviser.js.map +1 -0
- package/dist/constitutional/index.d.ts +13 -0
- package/dist/constitutional/index.d.ts.map +1 -0
- package/dist/constitutional/index.js +8 -0
- package/dist/constitutional/index.js.map +1 -0
- package/dist/constitutional/input-filter.d.ts +19 -0
- package/dist/constitutional/input-filter.d.ts.map +1 -0
- package/dist/constitutional/input-filter.js +88 -0
- package/dist/constitutional/input-filter.js.map +1 -0
- package/dist/constitutional/output-filter.d.ts +19 -0
- package/dist/constitutional/output-filter.d.ts.map +1 -0
- package/dist/constitutional/output-filter.js +86 -0
- package/dist/constitutional/output-filter.js.map +1 -0
- package/dist/constitutional/prompts.d.ts +11 -0
- package/dist/constitutional/prompts.d.ts.map +1 -0
- package/dist/constitutional/prompts.js +202 -0
- package/dist/constitutional/prompts.js.map +1 -0
- package/dist/constitutional/tool-guard.d.ts +18 -0
- package/dist/constitutional/tool-guard.d.ts.map +1 -0
- package/dist/constitutional/tool-guard.js +125 -0
- package/dist/constitutional/tool-guard.js.map +1 -0
- package/dist/cost-routing/budget-enforcer.d.ts +26 -0
- package/dist/cost-routing/budget-enforcer.d.ts.map +1 -0
- package/dist/cost-routing/budget-enforcer.js +86 -0
- package/dist/cost-routing/budget-enforcer.js.map +1 -0
- package/dist/cost-routing/cost-router.d.ts +34 -0
- package/dist/cost-routing/cost-router.d.ts.map +1 -0
- package/dist/cost-routing/cost-router.js +80 -0
- package/dist/cost-routing/cost-router.js.map +1 -0
- package/dist/cost-routing/cost-tracker.d.ts +20 -0
- package/dist/cost-routing/cost-tracker.d.ts.map +1 -0
- package/dist/cost-routing/cost-tracker.js +85 -0
- package/dist/cost-routing/cost-tracker.js.map +1 -0
- package/dist/cost-routing/index.d.ts +6 -0
- package/dist/cost-routing/index.d.ts.map +1 -0
- package/dist/cost-routing/index.js +6 -0
- package/dist/cost-routing/index.js.map +1 -0
- package/dist/cost-routing/model-selector.d.ts +15 -0
- package/dist/cost-routing/model-selector.d.ts.map +1 -0
- package/dist/cost-routing/model-selector.js +216 -0
- package/dist/cost-routing/model-selector.js.map +1 -0
- package/dist/cost-routing/task-analyzer.d.ts +13 -0
- package/dist/cost-routing/task-analyzer.d.ts.map +1 -0
- package/dist/cost-routing/task-analyzer.js +185 -0
- package/dist/cost-routing/task-analyzer.js.map +1 -0
- package/dist/index.d.ts +19 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +9 -0
- package/dist/index.js.map +1 -1
- package/dist/learning/ab-testing.d.ts +45 -0
- package/dist/learning/ab-testing.d.ts.map +1 -0
- package/dist/learning/ab-testing.js +267 -0
- package/dist/learning/ab-testing.js.map +1 -0
- package/dist/learning/agent-optimizer.d.ts +42 -0
- package/dist/learning/agent-optimizer.d.ts.map +1 -0
- package/dist/learning/agent-optimizer.js +273 -0
- package/dist/learning/agent-optimizer.js.map +1 -0
- package/dist/learning/auto-optimizer.d.ts +38 -0
- package/dist/learning/auto-optimizer.d.ts.map +1 -0
- package/dist/learning/auto-optimizer.js +229 -0
- package/dist/learning/auto-optimizer.js.map +1 -0
- package/dist/learning/demo-selector.d.ts +29 -0
- package/dist/learning/demo-selector.d.ts.map +1 -0
- package/dist/learning/demo-selector.js +235 -0
- package/dist/learning/demo-selector.js.map +1 -0
- package/dist/learning/index.d.ts +24 -0
- package/dist/learning/index.d.ts.map +1 -0
- package/dist/learning/index.js +13 -0
- package/dist/learning/index.js.map +1 -0
- package/dist/learning/instruction-optimizer.d.ts +29 -0
- package/dist/learning/instruction-optimizer.d.ts.map +1 -0
- package/dist/learning/instruction-optimizer.js +175 -0
- package/dist/learning/instruction-optimizer.js.map +1 -0
- package/dist/learning/metrics.d.ts +37 -0
- package/dist/learning/metrics.d.ts.map +1 -0
- package/dist/learning/metrics.js +310 -0
- package/dist/learning/metrics.js.map +1 -0
- package/dist/learning/postgres-trace-store.d.ts +53 -0
- package/dist/learning/postgres-trace-store.d.ts.map +1 -0
- package/dist/learning/postgres-trace-store.js +692 -0
- package/dist/learning/postgres-trace-store.js.map +1 -0
- package/dist/learning/prompt-logger.d.ts +29 -0
- package/dist/learning/prompt-logger.d.ts.map +1 -0
- package/dist/learning/prompt-logger.js +157 -0
- package/dist/learning/prompt-logger.js.map +1 -0
- package/dist/learning/prompt-monitor.d.ts +29 -0
- package/dist/learning/prompt-monitor.d.ts.map +1 -0
- package/dist/learning/prompt-monitor.js +243 -0
- package/dist/learning/prompt-monitor.js.map +1 -0
- package/dist/learning/prompts.d.ts +28 -0
- package/dist/learning/prompts.d.ts.map +1 -0
- package/dist/learning/prompts.js +195 -0
- package/dist/learning/prompts.js.map +1 -0
- package/dist/learning/rollback-manager.d.ts +36 -0
- package/dist/learning/rollback-manager.d.ts.map +1 -0
- package/dist/learning/rollback-manager.js +177 -0
- package/dist/learning/rollback-manager.js.map +1 -0
- package/dist/learning/trace-store.d.ts +26 -0
- package/dist/learning/trace-store.d.ts.map +1 -0
- package/dist/learning/trace-store.js +218 -0
- package/dist/learning/trace-store.js.map +1 -0
- package/dist/llm/google.d.ts.map +1 -1
- package/dist/llm/google.js +1 -2
- package/dist/llm/google.js.map +1 -1
- package/dist/reasoning/branch-evaluator.d.ts +28 -0
- package/dist/reasoning/branch-evaluator.d.ts.map +1 -0
- package/dist/reasoning/branch-evaluator.js +143 -0
- package/dist/reasoning/branch-evaluator.js.map +1 -0
- package/dist/reasoning/branch-generator.d.ts +9 -0
- package/dist/reasoning/branch-generator.d.ts.map +1 -0
- package/dist/reasoning/branch-generator.js +60 -0
- package/dist/reasoning/branch-generator.js.map +1 -0
- package/dist/reasoning/index.d.ts +5 -0
- package/dist/reasoning/index.d.ts.map +1 -0
- package/dist/reasoning/index.js +5 -0
- package/dist/reasoning/index.js.map +1 -0
- package/dist/reasoning/prompts.d.ts +19 -0
- package/dist/reasoning/prompts.d.ts.map +1 -0
- package/dist/reasoning/prompts.js +161 -0
- package/dist/reasoning/prompts.js.map +1 -0
- package/dist/reasoning/thought-tree.d.ts +32 -0
- package/dist/reasoning/thought-tree.d.ts.map +1 -0
- package/dist/reasoning/thought-tree.js +352 -0
- package/dist/reasoning/thought-tree.js.map +1 -0
- package/dist/reflection/index.d.ts +4 -0
- package/dist/reflection/index.d.ts.map +1 -0
- package/dist/reflection/index.js +4 -0
- package/dist/reflection/index.js.map +1 -0
- package/dist/reflection/insight-store.d.ts +19 -0
- package/dist/reflection/insight-store.d.ts.map +1 -0
- package/dist/reflection/insight-store.js +129 -0
- package/dist/reflection/insight-store.js.map +1 -0
- package/dist/reflection/prompts.d.ts +18 -0
- package/dist/reflection/prompts.d.ts.map +1 -0
- package/dist/reflection/prompts.js +157 -0
- package/dist/reflection/prompts.js.map +1 -0
- package/dist/reflection/reflection-engine.d.ts +25 -0
- package/dist/reflection/reflection-engine.d.ts.map +1 -0
- package/dist/reflection/reflection-engine.js +202 -0
- package/dist/reflection/reflection-engine.js.map +1 -0
- package/dist/registry.d.ts +1 -0
- package/dist/registry.d.ts.map +1 -1
- package/dist/registry.js +3 -0
- package/dist/registry.js.map +1 -1
- package/dist/time-travel/checkpoint-store.d.ts +34 -0
- package/dist/time-travel/checkpoint-store.d.ts.map +1 -0
- package/dist/time-travel/checkpoint-store.js +240 -0
- package/dist/time-travel/checkpoint-store.js.map +1 -0
- package/dist/time-travel/comparator.d.ts +26 -0
- package/dist/time-travel/comparator.d.ts.map +1 -0
- package/dist/time-travel/comparator.js +253 -0
- package/dist/time-travel/comparator.js.map +1 -0
- package/dist/time-travel/forker.d.ts +22 -0
- package/dist/time-travel/forker.d.ts.map +1 -0
- package/dist/time-travel/forker.js +118 -0
- package/dist/time-travel/forker.js.map +1 -0
- package/dist/time-travel/index.d.ts +6 -0
- package/dist/time-travel/index.d.ts.map +1 -0
- package/dist/time-travel/index.js +6 -0
- package/dist/time-travel/index.js.map +1 -0
- package/dist/time-travel/replayer.d.ts +20 -0
- package/dist/time-travel/replayer.d.ts.map +1 -0
- package/dist/time-travel/replayer.js +147 -0
- package/dist/time-travel/replayer.js.map +1 -0
- package/dist/time-travel/time-travel.d.ts +41 -0
- package/dist/time-travel/time-travel.d.ts.map +1 -0
- package/dist/time-travel/time-travel.js +127 -0
- package/dist/time-travel/time-travel.js.map +1 -0
- package/dist/tool.d.ts.map +1 -1
- package/dist/tool.js +2 -0
- package/dist/tool.js.map +1 -1
- package/dist/tools/base64.d.ts.map +1 -1
- package/dist/tools/base64.js +2 -8
- package/dist/tools/base64.js.map +1 -1
- package/dist/tools/datetime.d.ts.map +1 -1
- package/dist/tools/datetime.js.map +1 -1
- package/dist/tools/exec.d.ts.map +1 -1
- package/dist/tools/exec.js +1 -4
- package/dist/tools/exec.js.map +1 -1
- package/dist/tools/filesystem.d.ts.map +1 -1
- package/dist/tools/filesystem.js +4 -1
- package/dist/tools/filesystem.js.map +1 -1
- package/dist/tools/hash.d.ts.map +1 -1
- package/dist/tools/hash.js +1 -4
- package/dist/tools/hash.js.map +1 -1
- package/dist/tools/http.d.ts.map +1 -1
- package/dist/tools/http.js +1 -4
- package/dist/tools/http.js.map +1 -1
- package/dist/tools/regex.d.ts.map +1 -1
- package/dist/tools/regex.js +4 -1
- package/dist/tools/regex.js.map +1 -1
- package/dist/utils/circuit-breaker.d.ts.map +1 -1
- package/dist/utils/circuit-breaker.js.map +1 -1
- package/dist/utils/fallback.d.ts.map +1 -1
- package/dist/utils/fallback.js +1 -4
- package/dist/utils/fallback.js.map +1 -1
- package/dist/utils/retry.d.ts.map +1 -1
- package/dist/utils/retry.js +8 -13
- package/dist/utils/retry.js.map +1 -1
- package/package.json +17 -8
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
export class ToolGuard {
|
|
2
|
+
config;
|
|
3
|
+
constructor(options) {
|
|
4
|
+
this.config = options.config;
|
|
5
|
+
}
|
|
6
|
+
async evaluate(tool, args, _context) {
|
|
7
|
+
const sideEffects = tool.sideEffects ?? [];
|
|
8
|
+
const requiresApproval = this.checkApproval(tool, args);
|
|
9
|
+
const riskLevel = this.assessRisk(tool, args, sideEffects);
|
|
10
|
+
const dangerCheck = this.checkDangerousOperation(tool, args);
|
|
11
|
+
if (dangerCheck) {
|
|
12
|
+
return {
|
|
13
|
+
approved: false,
|
|
14
|
+
requiresConfirmation: true,
|
|
15
|
+
sideEffects,
|
|
16
|
+
riskLevel: 'high',
|
|
17
|
+
reason: dangerCheck,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
if (this.config.strictMode && sideEffects.length > 0) {
|
|
21
|
+
if (!requiresApproval) {
|
|
22
|
+
return {
|
|
23
|
+
approved: true,
|
|
24
|
+
requiresConfirmation: false,
|
|
25
|
+
sideEffects,
|
|
26
|
+
riskLevel,
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
const approved = await this.requestApproval(tool, args, sideEffects);
|
|
30
|
+
return {
|
|
31
|
+
approved,
|
|
32
|
+
requiresConfirmation: true,
|
|
33
|
+
sideEffects,
|
|
34
|
+
riskLevel,
|
|
35
|
+
reason: approved ? undefined : 'User denied tool execution',
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
if (requiresApproval) {
|
|
39
|
+
const approved = await this.requestApproval(tool, args, sideEffects);
|
|
40
|
+
return {
|
|
41
|
+
approved,
|
|
42
|
+
requiresConfirmation: true,
|
|
43
|
+
sideEffects,
|
|
44
|
+
riskLevel,
|
|
45
|
+
reason: approved ? undefined : 'User denied tool execution',
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
return {
|
|
49
|
+
approved: true,
|
|
50
|
+
requiresConfirmation: false,
|
|
51
|
+
sideEffects,
|
|
52
|
+
riskLevel,
|
|
53
|
+
};
|
|
54
|
+
}
|
|
55
|
+
checkApproval(tool, args) {
|
|
56
|
+
if (typeof tool.requiresApproval === 'function') {
|
|
57
|
+
return tool.requiresApproval(args);
|
|
58
|
+
}
|
|
59
|
+
return tool.requiresApproval ?? false;
|
|
60
|
+
}
|
|
61
|
+
assessRisk(_tool, args, sideEffects) {
|
|
62
|
+
if (sideEffects.includes('process') || sideEffects.includes('filesystem')) {
|
|
63
|
+
const command = String(args.command ?? args.cmd ?? args.path ?? '');
|
|
64
|
+
if (this.isDangerousCommand(command)) {
|
|
65
|
+
return 'high';
|
|
66
|
+
}
|
|
67
|
+
return 'medium';
|
|
68
|
+
}
|
|
69
|
+
if (sideEffects.includes('network') || sideEffects.includes('database')) {
|
|
70
|
+
return 'medium';
|
|
71
|
+
}
|
|
72
|
+
if (sideEffects.length > 0) {
|
|
73
|
+
return 'low';
|
|
74
|
+
}
|
|
75
|
+
return 'low';
|
|
76
|
+
}
|
|
77
|
+
checkDangerousOperation(tool, args) {
|
|
78
|
+
if (tool.name === 'exec' || tool.sideEffects?.includes('process')) {
|
|
79
|
+
const command = String(args.command ?? args.cmd ?? '');
|
|
80
|
+
if (this.isDangerousCommand(command)) {
|
|
81
|
+
return `Dangerous command detected: ${command.slice(0, 50)}`;
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
if (tool.name.includes('file') || tool.sideEffects?.includes('filesystem')) {
|
|
85
|
+
const path = String(args.path ?? args.file ?? '');
|
|
86
|
+
if (this.isDangerousPath(path)) {
|
|
87
|
+
return `Dangerous file path detected: ${path}`;
|
|
88
|
+
}
|
|
89
|
+
}
|
|
90
|
+
return null;
|
|
91
|
+
}
|
|
92
|
+
isDangerousCommand(command) {
|
|
93
|
+
const dangerous = [
|
|
94
|
+
/rm\s+-rf\s+\//,
|
|
95
|
+
/rm\s+-rf\s+~\/\*/,
|
|
96
|
+
/mkfs\./,
|
|
97
|
+
/dd\s+if=.*of=\/dev\//,
|
|
98
|
+
/chmod\s+-R\s+777\s+\//,
|
|
99
|
+
/>\s*\/dev\/sd[a-z]/,
|
|
100
|
+
/format\s+[a-z]:/i,
|
|
101
|
+
/del\s+\/[fqs]\s+/i,
|
|
102
|
+
];
|
|
103
|
+
return dangerous.some((pattern) => pattern.test(command));
|
|
104
|
+
}
|
|
105
|
+
isDangerousPath(path) {
|
|
106
|
+
const dangerous = [
|
|
107
|
+
/^\/etc\/(passwd|shadow|sudoers)/,
|
|
108
|
+
/^\/boot\//,
|
|
109
|
+
/^\/sys\//,
|
|
110
|
+
/^\/proc\//,
|
|
111
|
+
/^~\/.ssh\//,
|
|
112
|
+
/^\/root\//,
|
|
113
|
+
/^[A-Z]:\\Windows\\System32/i,
|
|
114
|
+
];
|
|
115
|
+
return dangerous.some((pattern) => pattern.test(path));
|
|
116
|
+
}
|
|
117
|
+
async requestApproval(tool, args, sideEffects) {
|
|
118
|
+
if (this.config.onToolApproval) {
|
|
119
|
+
return this.config.onToolApproval(tool.name, args, sideEffects);
|
|
120
|
+
}
|
|
121
|
+
return true;
|
|
122
|
+
}
|
|
123
|
+
updateConstitution(_constitution) { }
|
|
124
|
+
}
|
|
125
|
+
//# sourceMappingURL=tool-guard.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import type { BudgetConfig } from '@cogitator-ai/types';
|
|
2
|
+
import type { CostTracker } from './cost-tracker';
|
|
3
|
+
export interface BudgetCheckResult {
|
|
4
|
+
allowed: boolean;
|
|
5
|
+
reason?: string;
|
|
6
|
+
}
|
|
7
|
+
export declare class BudgetEnforcer {
|
|
8
|
+
private config;
|
|
9
|
+
private tracker;
|
|
10
|
+
private warningTriggered;
|
|
11
|
+
constructor(config: BudgetConfig, tracker: CostTracker);
|
|
12
|
+
checkBudget(estimatedCost: number): BudgetCheckResult;
|
|
13
|
+
getBudgetStatus(): {
|
|
14
|
+
hourlyUsed: number;
|
|
15
|
+
hourlyLimit?: number;
|
|
16
|
+
dailyUsed: number;
|
|
17
|
+
dailyLimit?: number;
|
|
18
|
+
hourlyRemaining?: number;
|
|
19
|
+
dailyRemaining?: number;
|
|
20
|
+
};
|
|
21
|
+
resetWarnings(): void;
|
|
22
|
+
private checkHourlyWarning;
|
|
23
|
+
private checkDailyWarning;
|
|
24
|
+
private triggerExceeded;
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=budget-enforcer.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
export class BudgetEnforcer {
|
|
2
|
+
config;
|
|
3
|
+
tracker;
|
|
4
|
+
warningTriggered = {
|
|
5
|
+
hourly: false,
|
|
6
|
+
daily: false,
|
|
7
|
+
};
|
|
8
|
+
constructor(config, tracker) {
|
|
9
|
+
this.config = config;
|
|
10
|
+
this.tracker = tracker;
|
|
11
|
+
}
|
|
12
|
+
checkBudget(estimatedCost) {
|
|
13
|
+
if (this.config.maxCostPerRun && estimatedCost > this.config.maxCostPerRun) {
|
|
14
|
+
this.triggerExceeded(estimatedCost, this.config.maxCostPerRun);
|
|
15
|
+
return {
|
|
16
|
+
allowed: false,
|
|
17
|
+
reason: `Estimated cost $${estimatedCost.toFixed(4)} exceeds per-run limit $${this.config.maxCostPerRun}`,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
if (this.config.maxCostPerHour) {
|
|
21
|
+
const hourly = this.tracker.getHourlyCost();
|
|
22
|
+
if (hourly + estimatedCost > this.config.maxCostPerHour) {
|
|
23
|
+
this.triggerExceeded(hourly + estimatedCost, this.config.maxCostPerHour);
|
|
24
|
+
return {
|
|
25
|
+
allowed: false,
|
|
26
|
+
reason: `Would exceed hourly budget ($${hourly.toFixed(2)} + $${estimatedCost.toFixed(4)} > $${this.config.maxCostPerHour})`,
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
this.checkHourlyWarning(hourly);
|
|
30
|
+
}
|
|
31
|
+
if (this.config.maxCostPerDay) {
|
|
32
|
+
const daily = this.tracker.getDailyCost();
|
|
33
|
+
if (daily + estimatedCost > this.config.maxCostPerDay) {
|
|
34
|
+
this.triggerExceeded(daily + estimatedCost, this.config.maxCostPerDay);
|
|
35
|
+
return {
|
|
36
|
+
allowed: false,
|
|
37
|
+
reason: `Would exceed daily budget ($${daily.toFixed(2)} + $${estimatedCost.toFixed(4)} > $${this.config.maxCostPerDay})`,
|
|
38
|
+
};
|
|
39
|
+
}
|
|
40
|
+
this.checkDailyWarning(daily);
|
|
41
|
+
}
|
|
42
|
+
return { allowed: true };
|
|
43
|
+
}
|
|
44
|
+
getBudgetStatus() {
|
|
45
|
+
const hourlyUsed = this.tracker.getHourlyCost();
|
|
46
|
+
const dailyUsed = this.tracker.getDailyCost();
|
|
47
|
+
return {
|
|
48
|
+
hourlyUsed,
|
|
49
|
+
hourlyLimit: this.config.maxCostPerHour,
|
|
50
|
+
dailyUsed,
|
|
51
|
+
dailyLimit: this.config.maxCostPerDay,
|
|
52
|
+
hourlyRemaining: this.config.maxCostPerHour
|
|
53
|
+
? Math.max(0, this.config.maxCostPerHour - hourlyUsed)
|
|
54
|
+
: undefined,
|
|
55
|
+
dailyRemaining: this.config.maxCostPerDay
|
|
56
|
+
? Math.max(0, this.config.maxCostPerDay - dailyUsed)
|
|
57
|
+
: undefined,
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
resetWarnings() {
|
|
61
|
+
this.warningTriggered.hourly = false;
|
|
62
|
+
this.warningTriggered.daily = false;
|
|
63
|
+
}
|
|
64
|
+
checkHourlyWarning(current) {
|
|
65
|
+
if (!this.config.maxCostPerHour || this.warningTriggered.hourly)
|
|
66
|
+
return;
|
|
67
|
+
const threshold = this.config.warningThreshold ?? 0.8;
|
|
68
|
+
if (current / this.config.maxCostPerHour >= threshold) {
|
|
69
|
+
this.warningTriggered.hourly = true;
|
|
70
|
+
this.config.onBudgetWarning?.(current, this.config.maxCostPerHour);
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
checkDailyWarning(current) {
|
|
74
|
+
if (!this.config.maxCostPerDay || this.warningTriggered.daily)
|
|
75
|
+
return;
|
|
76
|
+
const threshold = this.config.warningThreshold ?? 0.8;
|
|
77
|
+
if (current / this.config.maxCostPerDay >= threshold) {
|
|
78
|
+
this.warningTriggered.daily = true;
|
|
79
|
+
this.config.onBudgetWarning?.(current, this.config.maxCostPerDay);
|
|
80
|
+
}
|
|
81
|
+
}
|
|
82
|
+
triggerExceeded(current, limit) {
|
|
83
|
+
this.config.onBudgetExceeded?.(current, limit);
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
//# sourceMappingURL=budget-enforcer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import type { CostRoutingConfig, TaskRequirements, ModelRecommendation, CostRecord, CostSummary } from '@cogitator-ai/types';
|
|
2
|
+
import { type BudgetCheckResult } from './budget-enforcer';
|
|
3
|
+
export interface CostAwareRouterOptions {
|
|
4
|
+
config?: Partial<CostRoutingConfig>;
|
|
5
|
+
}
|
|
6
|
+
export declare class CostAwareRouter {
|
|
7
|
+
private taskAnalyzer;
|
|
8
|
+
private modelSelector;
|
|
9
|
+
private costTracker;
|
|
10
|
+
private budgetEnforcer?;
|
|
11
|
+
private config;
|
|
12
|
+
constructor(options?: CostAwareRouterOptions);
|
|
13
|
+
analyzeTask(input: string): TaskRequirements;
|
|
14
|
+
recommendModel(input: string): Promise<ModelRecommendation>;
|
|
15
|
+
recommendModelForRequirements(requirements: TaskRequirements): Promise<ModelRecommendation>;
|
|
16
|
+
checkBudget(estimatedCost: number): BudgetCheckResult;
|
|
17
|
+
recordCost(record: Omit<CostRecord, 'timestamp'>): void;
|
|
18
|
+
getRunCost(runId: string): number;
|
|
19
|
+
getHourlyCost(): number;
|
|
20
|
+
getDailyCost(): number;
|
|
21
|
+
getCostSummary(): CostSummary;
|
|
22
|
+
getBudgetStatus(): {
|
|
23
|
+
hourlyUsed: number;
|
|
24
|
+
hourlyLimit?: number;
|
|
25
|
+
dailyUsed: number;
|
|
26
|
+
dailyLimit?: number;
|
|
27
|
+
hourlyRemaining?: number;
|
|
28
|
+
dailyRemaining?: number;
|
|
29
|
+
} | undefined;
|
|
30
|
+
getConfig(): CostRoutingConfig;
|
|
31
|
+
updateConfig(config: Partial<CostRoutingConfig>): void;
|
|
32
|
+
clearCostHistory(): void;
|
|
33
|
+
}
|
|
34
|
+
//# sourceMappingURL=cost-router.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,80 @@
|
|
|
1
|
+
import { TaskAnalyzer } from './task-analyzer';
|
|
2
|
+
import { ModelSelector } from './model-selector';
|
|
3
|
+
import { CostTracker } from './cost-tracker';
|
|
4
|
+
import { BudgetEnforcer } from './budget-enforcer';
|
|
5
|
+
const DEFAULT_CONFIG = {
|
|
6
|
+
enabled: true,
|
|
7
|
+
autoSelectModel: false,
|
|
8
|
+
preferLocal: true,
|
|
9
|
+
minCapabilityMatch: 0.3,
|
|
10
|
+
trackCosts: true,
|
|
11
|
+
};
|
|
12
|
+
export class CostAwareRouter {
|
|
13
|
+
taskAnalyzer;
|
|
14
|
+
modelSelector;
|
|
15
|
+
costTracker;
|
|
16
|
+
budgetEnforcer;
|
|
17
|
+
config;
|
|
18
|
+
constructor(options = {}) {
|
|
19
|
+
this.config = { ...DEFAULT_CONFIG, ...options.config };
|
|
20
|
+
this.taskAnalyzer = new TaskAnalyzer();
|
|
21
|
+
this.modelSelector = new ModelSelector(this.config);
|
|
22
|
+
this.costTracker = new CostTracker();
|
|
23
|
+
if (this.config.budget) {
|
|
24
|
+
this.budgetEnforcer = new BudgetEnforcer(this.config.budget, this.costTracker);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
analyzeTask(input) {
|
|
28
|
+
return this.taskAnalyzer.analyze(input);
|
|
29
|
+
}
|
|
30
|
+
async recommendModel(input) {
|
|
31
|
+
const requirements = this.analyzeTask(input);
|
|
32
|
+
return this.modelSelector.selectModel(requirements);
|
|
33
|
+
}
|
|
34
|
+
async recommendModelForRequirements(requirements) {
|
|
35
|
+
return this.modelSelector.selectModel(requirements);
|
|
36
|
+
}
|
|
37
|
+
checkBudget(estimatedCost) {
|
|
38
|
+
if (!this.budgetEnforcer)
|
|
39
|
+
return { allowed: true };
|
|
40
|
+
return this.budgetEnforcer.checkBudget(estimatedCost);
|
|
41
|
+
}
|
|
42
|
+
recordCost(record) {
|
|
43
|
+
if (this.config.trackCosts) {
|
|
44
|
+
this.costTracker.record(record);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
getRunCost(runId) {
|
|
48
|
+
return this.costTracker.getRunCost(runId);
|
|
49
|
+
}
|
|
50
|
+
getHourlyCost() {
|
|
51
|
+
return this.costTracker.getHourlyCost();
|
|
52
|
+
}
|
|
53
|
+
getDailyCost() {
|
|
54
|
+
return this.costTracker.getDailyCost();
|
|
55
|
+
}
|
|
56
|
+
getCostSummary() {
|
|
57
|
+
return this.costTracker.getSummary();
|
|
58
|
+
}
|
|
59
|
+
getBudgetStatus() {
|
|
60
|
+
return this.budgetEnforcer?.getBudgetStatus();
|
|
61
|
+
}
|
|
62
|
+
getConfig() {
|
|
63
|
+
return { ...this.config };
|
|
64
|
+
}
|
|
65
|
+
updateConfig(config) {
|
|
66
|
+
this.config = { ...this.config, ...config };
|
|
67
|
+
this.modelSelector = new ModelSelector(this.config);
|
|
68
|
+
if (this.config.budget) {
|
|
69
|
+
this.budgetEnforcer = new BudgetEnforcer(this.config.budget, this.costTracker);
|
|
70
|
+
}
|
|
71
|
+
else {
|
|
72
|
+
this.budgetEnforcer = undefined;
|
|
73
|
+
}
|
|
74
|
+
}
|
|
75
|
+
clearCostHistory() {
|
|
76
|
+
this.costTracker.clear();
|
|
77
|
+
this.budgetEnforcer?.resetWarnings();
|
|
78
|
+
}
|
|
79
|
+
}
|
|
80
|
+
//# sourceMappingURL=cost-router.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { CostRecord, CostSummary } from '@cogitator-ai/types';
|
|
2
|
+
export interface CostFilter {
|
|
3
|
+
agentId?: string;
|
|
4
|
+
runId?: string;
|
|
5
|
+
since?: Date;
|
|
6
|
+
}
|
|
7
|
+
export declare class CostTracker {
|
|
8
|
+
private records;
|
|
9
|
+
private hourlyWindow;
|
|
10
|
+
private dailyWindow;
|
|
11
|
+
record(record: Omit<CostRecord, 'timestamp'>): void;
|
|
12
|
+
getRunCost(runId: string): number;
|
|
13
|
+
getHourlyCost(): number;
|
|
14
|
+
getDailyCost(): number;
|
|
15
|
+
getSummary(filter?: CostFilter): CostSummary;
|
|
16
|
+
getRecords(filter?: CostFilter): CostRecord[];
|
|
17
|
+
clear(): void;
|
|
18
|
+
private pruneWindows;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=cost-tracker.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,85 @@
|
|
|
1
|
+
const HOUR_MS = 60 * 60 * 1000;
|
|
2
|
+
const DAY_MS = 24 * HOUR_MS;
|
|
3
|
+
export class CostTracker {
|
|
4
|
+
records = [];
|
|
5
|
+
hourlyWindow = [];
|
|
6
|
+
dailyWindow = [];
|
|
7
|
+
record(record) {
|
|
8
|
+
const entry = { ...record, timestamp: new Date() };
|
|
9
|
+
this.records.push(entry);
|
|
10
|
+
this.hourlyWindow.push(entry);
|
|
11
|
+
this.dailyWindow.push(entry);
|
|
12
|
+
this.pruneWindows();
|
|
13
|
+
}
|
|
14
|
+
getRunCost(runId) {
|
|
15
|
+
return this.records.filter((r) => r.runId === runId).reduce((sum, r) => sum + r.cost, 0);
|
|
16
|
+
}
|
|
17
|
+
getHourlyCost() {
|
|
18
|
+
this.pruneWindows();
|
|
19
|
+
return this.hourlyWindow.reduce((sum, r) => sum + r.cost, 0);
|
|
20
|
+
}
|
|
21
|
+
getDailyCost() {
|
|
22
|
+
this.pruneWindows();
|
|
23
|
+
return this.dailyWindow.reduce((sum, r) => sum + r.cost, 0);
|
|
24
|
+
}
|
|
25
|
+
getSummary(filter) {
|
|
26
|
+
let filtered = this.records;
|
|
27
|
+
if (filter?.agentId) {
|
|
28
|
+
filtered = filtered.filter((r) => r.agentId === filter.agentId);
|
|
29
|
+
}
|
|
30
|
+
if (filter?.runId) {
|
|
31
|
+
filtered = filtered.filter((r) => r.runId === filter.runId);
|
|
32
|
+
}
|
|
33
|
+
if (filter?.since) {
|
|
34
|
+
const since = filter.since.getTime();
|
|
35
|
+
filtered = filtered.filter((r) => r.timestamp.getTime() >= since);
|
|
36
|
+
}
|
|
37
|
+
const byModel = {};
|
|
38
|
+
const byAgent = {};
|
|
39
|
+
const runIds = new Set();
|
|
40
|
+
let totalCost = 0;
|
|
41
|
+
let totalInputTokens = 0;
|
|
42
|
+
let totalOutputTokens = 0;
|
|
43
|
+
for (const record of filtered) {
|
|
44
|
+
totalCost += record.cost;
|
|
45
|
+
totalInputTokens += record.inputTokens;
|
|
46
|
+
totalOutputTokens += record.outputTokens;
|
|
47
|
+
runIds.add(record.runId);
|
|
48
|
+
byModel[record.model] = (byModel[record.model] ?? 0) + record.cost;
|
|
49
|
+
byAgent[record.agentId] = (byAgent[record.agentId] ?? 0) + record.cost;
|
|
50
|
+
}
|
|
51
|
+
return {
|
|
52
|
+
totalCost,
|
|
53
|
+
totalInputTokens,
|
|
54
|
+
totalOutputTokens,
|
|
55
|
+
byModel,
|
|
56
|
+
byAgent,
|
|
57
|
+
runCount: runIds.size,
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
getRecords(filter) {
|
|
61
|
+
let filtered = [...this.records];
|
|
62
|
+
if (filter?.agentId) {
|
|
63
|
+
filtered = filtered.filter((r) => r.agentId === filter.agentId);
|
|
64
|
+
}
|
|
65
|
+
if (filter?.runId) {
|
|
66
|
+
filtered = filtered.filter((r) => r.runId === filter.runId);
|
|
67
|
+
}
|
|
68
|
+
if (filter?.since) {
|
|
69
|
+
const since = filter.since.getTime();
|
|
70
|
+
filtered = filtered.filter((r) => r.timestamp.getTime() >= since);
|
|
71
|
+
}
|
|
72
|
+
return filtered;
|
|
73
|
+
}
|
|
74
|
+
clear() {
|
|
75
|
+
this.records = [];
|
|
76
|
+
this.hourlyWindow = [];
|
|
77
|
+
this.dailyWindow = [];
|
|
78
|
+
}
|
|
79
|
+
pruneWindows() {
|
|
80
|
+
const now = Date.now();
|
|
81
|
+
this.hourlyWindow = this.hourlyWindow.filter((r) => now - r.timestamp.getTime() < HOUR_MS);
|
|
82
|
+
this.dailyWindow = this.dailyWindow.filter((r) => now - r.timestamp.getTime() < DAY_MS);
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
//# sourceMappingURL=cost-tracker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-tracker.js","sourceRoot":"","sources":["../../src/cost-routing/cost-tracker.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAC/B,MAAM,MAAM,GAAG,EAAE,GAAG,OAAO,CAAC;AAQ5B,MAAM,OAAO,WAAW;IACd,OAAO,GAAiB,EAAE,CAAC;IAC3B,YAAY,GAAiB,EAAE,CAAC;IAChC,WAAW,GAAiB,EAAE,CAAC;IAEvC,MAAM,CAAC,MAAqC;QAC1C,MAAM,KAAK,GAAe,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,CAAC;QAC/D,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9B,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,CAAC,YAAY,EAAE,CAAC;IACtB,CAAC;IAED,UAAU,CAAC,KAAa;QACtB,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED,aAAa;QACX,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,YAAY;QACV,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED,UAAU,CAAC,MAAmB;QAC5B,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;QAE5B,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YAClB,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YAClB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACrC,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,CAAC;QACpE,CAAC;QAED,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;QAEjC,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,gBAAgB,GAAG,CAAC,CAAC;QACzB,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAE1B,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC;YACzB,gBAAgB,IAAI,MAAM,CAAC,WAAW,CAAC;YACvC,iBAAiB,IAAI,MAAM,CAAC,YAAY,CAAC;YACzC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAEzB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;YACnE,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC;QACzE,CAAC;QAED,OAAO;YACL,SAAS;YACT,gBAAgB;YAChB,iBAAiB;YACjB,OAAO;YACP,OAAO;YACP,QAAQ,EAAE,MAAM,CAAC,IAAI;SACtB,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,MAAmB;QAC5B,IAAI,QAAQ,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;QAEjC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YAClB,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,MAAM,CAAC,KAAK,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,MAAM,EAAE,KAAK,EAAE,CAAC;YAClB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;YACrC,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,CAAC;QACpE,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,YAAY,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;IACxB,CAAC;IAEO,YAAY;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,OAAO,CAAC,CAAC;QAC3F,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,CAAC,CAAC;IAC1F,CAAC;CACF"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { TaskAnalyzer } from './task-analyzer';
|
|
2
|
+
export { ModelSelector } from './model-selector';
|
|
3
|
+
export { CostTracker, type CostFilter } from './cost-tracker';
|
|
4
|
+
export { BudgetEnforcer, type BudgetCheckResult } from './budget-enforcer';
|
|
5
|
+
export { CostAwareRouter, type CostAwareRouterOptions } from './cost-router';
|
|
6
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/cost-routing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,KAAK,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC9D,OAAO,EAAE,cAAc,EAAE,KAAK,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EAAE,eAAe,EAAE,KAAK,sBAAsB,EAAE,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { TaskAnalyzer } from './task-analyzer';
|
|
2
|
+
export { ModelSelector } from './model-selector';
|
|
3
|
+
export { CostTracker } from './cost-tracker';
|
|
4
|
+
export { BudgetEnforcer } from './budget-enforcer';
|
|
5
|
+
export { CostAwareRouter } from './cost-router';
|
|
6
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cost-routing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EAAE,WAAW,EAAmB,MAAM,gBAAgB,CAAC;AAC9D,OAAO,EAAE,cAAc,EAA0B,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EAAE,eAAe,EAA+B,MAAM,eAAe,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import type { TaskRequirements, ModelRecommendation, CostRoutingConfig } from '@cogitator-ai/types';
|
|
2
|
+
export declare class ModelSelector {
|
|
3
|
+
private config;
|
|
4
|
+
constructor(config: CostRoutingConfig);
|
|
5
|
+
selectModel(requirements: TaskRequirements): Promise<ModelRecommendation>;
|
|
6
|
+
private scoreModel;
|
|
7
|
+
private toRecommendation;
|
|
8
|
+
private estimateCost;
|
|
9
|
+
private fallbackRecommendation;
|
|
10
|
+
private isLocalModel;
|
|
11
|
+
private isAdvancedModel;
|
|
12
|
+
private isFastModel;
|
|
13
|
+
private isCodeModel;
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=model-selector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"model-selector.d.ts","sourceRoot":"","sources":["../../src/cost-routing/model-selector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AA8CpG,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAoB;gBAEtB,MAAM,EAAE,iBAAiB;IAI/B,WAAW,CAAC,YAAY,EAAE,gBAAgB,GAAG,OAAO,CAAC,mBAAmB,CAAC;IAkC/E,OAAO,CAAC,UAAU;IA4FlB,OAAO,CAAC,gBAAgB;IAoBxB,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,sBAAsB;IAW9B,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,eAAe;IAKvB,OAAO,CAAC,WAAW;IAKnB,OAAO,CAAC,WAAW;CAIpB"}
|