@cogitator-ai/core 0.2.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.
Files changed (157) hide show
  1. package/README.md +920 -15
  2. package/dist/cogitator.d.ts +31 -1
  3. package/dist/cogitator.d.ts.map +1 -1
  4. package/dist/cogitator.js +127 -6
  5. package/dist/cogitator.js.map +1 -1
  6. package/dist/constitutional/constitution.d.ts +9 -0
  7. package/dist/constitutional/constitution.d.ts.map +1 -0
  8. package/dist/constitutional/constitution.js +215 -0
  9. package/dist/constitutional/constitution.js.map +1 -0
  10. package/dist/constitutional/constitutional-ai.d.ts +36 -0
  11. package/dist/constitutional/constitutional-ai.d.ts.map +1 -0
  12. package/dist/constitutional/constitutional-ai.js +163 -0
  13. package/dist/constitutional/constitutional-ai.js.map +1 -0
  14. package/dist/constitutional/critique-reviser.d.ts +20 -0
  15. package/dist/constitutional/critique-reviser.d.ts.map +1 -0
  16. package/dist/constitutional/critique-reviser.js +98 -0
  17. package/dist/constitutional/critique-reviser.js.map +1 -0
  18. package/dist/constitutional/index.d.ts +13 -0
  19. package/dist/constitutional/index.d.ts.map +1 -0
  20. package/dist/constitutional/index.js +8 -0
  21. package/dist/constitutional/index.js.map +1 -0
  22. package/dist/constitutional/input-filter.d.ts +19 -0
  23. package/dist/constitutional/input-filter.d.ts.map +1 -0
  24. package/dist/constitutional/input-filter.js +88 -0
  25. package/dist/constitutional/input-filter.js.map +1 -0
  26. package/dist/constitutional/output-filter.d.ts +19 -0
  27. package/dist/constitutional/output-filter.d.ts.map +1 -0
  28. package/dist/constitutional/output-filter.js +86 -0
  29. package/dist/constitutional/output-filter.js.map +1 -0
  30. package/dist/constitutional/prompts.d.ts +11 -0
  31. package/dist/constitutional/prompts.d.ts.map +1 -0
  32. package/dist/constitutional/prompts.js +202 -0
  33. package/dist/constitutional/prompts.js.map +1 -0
  34. package/dist/constitutional/tool-guard.d.ts +18 -0
  35. package/dist/constitutional/tool-guard.d.ts.map +1 -0
  36. package/dist/constitutional/tool-guard.js +125 -0
  37. package/dist/constitutional/tool-guard.js.map +1 -0
  38. package/dist/cost-routing/budget-enforcer.d.ts +26 -0
  39. package/dist/cost-routing/budget-enforcer.d.ts.map +1 -0
  40. package/dist/cost-routing/budget-enforcer.js +86 -0
  41. package/dist/cost-routing/budget-enforcer.js.map +1 -0
  42. package/dist/cost-routing/cost-router.d.ts +34 -0
  43. package/dist/cost-routing/cost-router.d.ts.map +1 -0
  44. package/dist/cost-routing/cost-router.js +80 -0
  45. package/dist/cost-routing/cost-router.js.map +1 -0
  46. package/dist/cost-routing/cost-tracker.d.ts +20 -0
  47. package/dist/cost-routing/cost-tracker.d.ts.map +1 -0
  48. package/dist/cost-routing/cost-tracker.js +85 -0
  49. package/dist/cost-routing/cost-tracker.js.map +1 -0
  50. package/dist/cost-routing/index.d.ts +6 -0
  51. package/dist/cost-routing/index.d.ts.map +1 -0
  52. package/dist/cost-routing/index.js +6 -0
  53. package/dist/cost-routing/index.js.map +1 -0
  54. package/dist/cost-routing/model-selector.d.ts +15 -0
  55. package/dist/cost-routing/model-selector.d.ts.map +1 -0
  56. package/dist/cost-routing/model-selector.js +216 -0
  57. package/dist/cost-routing/model-selector.js.map +1 -0
  58. package/dist/cost-routing/task-analyzer.d.ts +13 -0
  59. package/dist/cost-routing/task-analyzer.d.ts.map +1 -0
  60. package/dist/cost-routing/task-analyzer.js +185 -0
  61. package/dist/cost-routing/task-analyzer.js.map +1 -0
  62. package/dist/index.d.ts +13 -2
  63. package/dist/index.d.ts.map +1 -1
  64. package/dist/index.js +7 -2
  65. package/dist/index.js.map +1 -1
  66. package/dist/learning/ab-testing.d.ts +45 -0
  67. package/dist/learning/ab-testing.d.ts.map +1 -0
  68. package/dist/learning/ab-testing.js +267 -0
  69. package/dist/learning/ab-testing.js.map +1 -0
  70. package/dist/learning/agent-optimizer.d.ts.map +1 -1
  71. package/dist/learning/agent-optimizer.js +26 -21
  72. package/dist/learning/agent-optimizer.js.map +1 -1
  73. package/dist/learning/auto-optimizer.d.ts +38 -0
  74. package/dist/learning/auto-optimizer.d.ts.map +1 -0
  75. package/dist/learning/auto-optimizer.js +229 -0
  76. package/dist/learning/auto-optimizer.js.map +1 -0
  77. package/dist/learning/demo-selector.d.ts.map +1 -1
  78. package/dist/learning/demo-selector.js +7 -7
  79. package/dist/learning/demo-selector.js.map +1 -1
  80. package/dist/learning/index.d.ts +13 -1
  81. package/dist/learning/index.d.ts.map +1 -1
  82. package/dist/learning/index.js +7 -1
  83. package/dist/learning/index.js.map +1 -1
  84. package/dist/learning/instruction-optimizer.d.ts.map +1 -1
  85. package/dist/learning/instruction-optimizer.js +7 -11
  86. package/dist/learning/instruction-optimizer.js.map +1 -1
  87. package/dist/learning/metrics.d.ts.map +1 -1
  88. package/dist/learning/metrics.js +26 -16
  89. package/dist/learning/metrics.js.map +1 -1
  90. package/dist/learning/postgres-trace-store.d.ts +53 -0
  91. package/dist/learning/postgres-trace-store.d.ts.map +1 -0
  92. package/dist/learning/postgres-trace-store.js +692 -0
  93. package/dist/learning/postgres-trace-store.js.map +1 -0
  94. package/dist/learning/prompt-logger.d.ts +29 -0
  95. package/dist/learning/prompt-logger.d.ts.map +1 -0
  96. package/dist/learning/prompt-logger.js +157 -0
  97. package/dist/learning/prompt-logger.js.map +1 -0
  98. package/dist/learning/prompt-monitor.d.ts +29 -0
  99. package/dist/learning/prompt-monitor.d.ts.map +1 -0
  100. package/dist/learning/prompt-monitor.js +243 -0
  101. package/dist/learning/prompt-monitor.js.map +1 -0
  102. package/dist/learning/prompts.d.ts.map +1 -1
  103. package/dist/learning/prompts.js +24 -13
  104. package/dist/learning/prompts.js.map +1 -1
  105. package/dist/learning/rollback-manager.d.ts +36 -0
  106. package/dist/learning/rollback-manager.d.ts.map +1 -0
  107. package/dist/learning/rollback-manager.js +177 -0
  108. package/dist/learning/rollback-manager.js.map +1 -0
  109. package/dist/learning/trace-store.d.ts.map +1 -1
  110. package/dist/learning/trace-store.js +8 -10
  111. package/dist/learning/trace-store.js.map +1 -1
  112. package/dist/reasoning/branch-evaluator.d.ts.map +1 -1
  113. package/dist/reasoning/branch-evaluator.js +14 -8
  114. package/dist/reasoning/branch-evaluator.js.map +1 -1
  115. package/dist/reasoning/branch-generator.d.ts.map +1 -1
  116. package/dist/reasoning/branch-generator.js +5 -3
  117. package/dist/reasoning/branch-generator.js.map +1 -1
  118. package/dist/reasoning/prompts.d.ts.map +1 -1
  119. package/dist/reasoning/prompts.js +7 -5
  120. package/dist/reasoning/prompts.js.map +1 -1
  121. package/dist/reasoning/thought-tree.d.ts.map +1 -1
  122. package/dist/reasoning/thought-tree.js +9 -11
  123. package/dist/reasoning/thought-tree.js.map +1 -1
  124. package/dist/reflection/insight-store.d.ts.map +1 -1
  125. package/dist/reflection/insight-store.js +8 -6
  126. package/dist/reflection/insight-store.js.map +1 -1
  127. package/dist/reflection/prompts.d.ts.map +1 -1
  128. package/dist/reflection/prompts.js +11 -6
  129. package/dist/reflection/prompts.js.map +1 -1
  130. package/dist/reflection/reflection-engine.d.ts.map +1 -1
  131. package/dist/reflection/reflection-engine.js +8 -10
  132. package/dist/reflection/reflection-engine.js.map +1 -1
  133. package/dist/time-travel/checkpoint-store.d.ts +34 -0
  134. package/dist/time-travel/checkpoint-store.d.ts.map +1 -0
  135. package/dist/time-travel/checkpoint-store.js +240 -0
  136. package/dist/time-travel/checkpoint-store.js.map +1 -0
  137. package/dist/time-travel/comparator.d.ts +26 -0
  138. package/dist/time-travel/comparator.d.ts.map +1 -0
  139. package/dist/time-travel/comparator.js +253 -0
  140. package/dist/time-travel/comparator.js.map +1 -0
  141. package/dist/time-travel/forker.d.ts +22 -0
  142. package/dist/time-travel/forker.d.ts.map +1 -0
  143. package/dist/time-travel/forker.js +118 -0
  144. package/dist/time-travel/forker.js.map +1 -0
  145. package/dist/time-travel/index.d.ts +6 -0
  146. package/dist/time-travel/index.d.ts.map +1 -0
  147. package/dist/time-travel/index.js +6 -0
  148. package/dist/time-travel/index.js.map +1 -0
  149. package/dist/time-travel/replayer.d.ts +20 -0
  150. package/dist/time-travel/replayer.d.ts.map +1 -0
  151. package/dist/time-travel/replayer.js +147 -0
  152. package/dist/time-travel/replayer.js.map +1 -0
  153. package/dist/time-travel/time-travel.d.ts +41 -0
  154. package/dist/time-travel/time-travel.d.ts.map +1 -0
  155. package/dist/time-travel/time-travel.js +127 -0
  156. package/dist/time-travel/time-travel.js.map +1 -0
  157. package/package.json +13 -5
@@ -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"}
@@ -0,0 +1,216 @@
1
+ import { getModelRegistry } from '@cogitator-ai/models';
2
+ const ADVANCED_MODELS = [
3
+ 'gpt-4o',
4
+ 'gpt-4-turbo',
5
+ 'claude-3-opus',
6
+ 'claude-sonnet-4',
7
+ 'claude-3-5-sonnet',
8
+ 'gemini-1.5-pro',
9
+ 'llama3.1:70b',
10
+ 'llama3.2:70b',
11
+ 'qwen2.5:72b',
12
+ 'mixtral',
13
+ 'deepseek-r1',
14
+ ];
15
+ const FAST_MODELS = [
16
+ 'gpt-4o-mini',
17
+ 'claude-3-5-haiku',
18
+ 'gemini-1.5-flash',
19
+ 'gemini-2.0-flash',
20
+ 'llama3.2:1b',
21
+ 'llama3.2:3b',
22
+ 'phi3',
23
+ 'phi4',
24
+ 'qwen2.5:1.5b',
25
+ 'qwen2.5:3b',
26
+ ];
27
+ const CODE_MODELS = [
28
+ 'claude-3-5-sonnet',
29
+ 'claude-sonnet-4',
30
+ 'gpt-4o',
31
+ 'codellama',
32
+ 'starcoder',
33
+ 'deepseek-coder',
34
+ 'qwen2.5-coder',
35
+ ];
36
+ export class ModelSelector {
37
+ config;
38
+ constructor(config) {
39
+ this.config = config;
40
+ }
41
+ async selectModel(requirements) {
42
+ const registry = getModelRegistry();
43
+ await registry.initialize();
44
+ const candidates = registry.listModels({
45
+ supportsTools: requirements.needsToolCalling ? true : undefined,
46
+ supportsVision: requirements.needsVision ? true : undefined,
47
+ minContextWindow: requirements.needsLongContext ? 32000 : undefined,
48
+ excludeDeprecated: true,
49
+ });
50
+ if (candidates.length === 0) {
51
+ return this.fallbackRecommendation();
52
+ }
53
+ const scored = candidates
54
+ .map((m) => this.scoreModel(m, requirements))
55
+ .filter((s) => s.score >= (this.config.minCapabilityMatch ?? 0.3) * 100)
56
+ .sort((a, b) => b.score - a.score);
57
+ if (scored.length === 0) {
58
+ return this.fallbackRecommendation();
59
+ }
60
+ if (this.config.preferLocal) {
61
+ const local = scored.find((s) => this.isLocalModel(s.model.id));
62
+ if (local && local.score >= scored[0].score * 0.8) {
63
+ return this.toRecommendation(local, scored, requirements);
64
+ }
65
+ }
66
+ return this.toRecommendation(scored[0], scored, requirements);
67
+ }
68
+ scoreModel(model, requirements) {
69
+ let score = 100;
70
+ const reasons = [];
71
+ if (requirements.needsVision && !model.capabilities?.supportsVision) {
72
+ return { model, score: 0, reasons: ['Does not support vision (required)'] };
73
+ }
74
+ if (requirements.needsToolCalling && !model.capabilities?.supportsTools) {
75
+ return { model, score: 0, reasons: ['Does not support tool calling (required)'] };
76
+ }
77
+ if (requirements.needsLongContext && model.contextWindow < 32000) {
78
+ score -= 30;
79
+ reasons.push(`Limited context window (${model.contextWindow} tokens)`);
80
+ }
81
+ else if (model.contextWindow >= 100000) {
82
+ score += 5;
83
+ reasons.push('Large context window');
84
+ }
85
+ const isLocal = this.isLocalModel(model.id);
86
+ if (isLocal) {
87
+ score += 15;
88
+ reasons.push('Local model (no API cost)');
89
+ }
90
+ if (!isLocal) {
91
+ const avgCost = (model.pricing.input + model.pricing.output) / 2;
92
+ if (requirements.costSensitivity === 'high') {
93
+ if (avgCost > 10) {
94
+ score -= 25;
95
+ reasons.push('High cost model');
96
+ }
97
+ else if (avgCost < 1) {
98
+ score += 10;
99
+ reasons.push('Cost-effective');
100
+ }
101
+ }
102
+ else if (requirements.costSensitivity === 'low' && avgCost > 10) {
103
+ score += 5;
104
+ reasons.push('Premium model (quality prioritized)');
105
+ }
106
+ }
107
+ const isAdvanced = this.isAdvancedModel(model.id);
108
+ if (requirements.needsReasoning === 'advanced') {
109
+ if (isAdvanced) {
110
+ score += 15;
111
+ reasons.push('Strong reasoning capability');
112
+ }
113
+ else {
114
+ score -= 20;
115
+ reasons.push('May lack advanced reasoning');
116
+ }
117
+ }
118
+ else if (requirements.needsReasoning === 'basic' && !isAdvanced) {
119
+ score += 5;
120
+ reasons.push('Appropriate for basic tasks');
121
+ }
122
+ const isFast = this.isFastModel(model.id);
123
+ if (requirements.needsSpeed === 'fast') {
124
+ if (isFast || isLocal) {
125
+ score += 10;
126
+ reasons.push('Fast response time');
127
+ }
128
+ else {
129
+ score -= 10;
130
+ reasons.push('May be slower');
131
+ }
132
+ }
133
+ else if (requirements.needsSpeed === 'slow-ok' && isAdvanced) {
134
+ score += 5;
135
+ reasons.push('Quality over speed');
136
+ }
137
+ if (requirements.domains?.includes('code')) {
138
+ if (this.isCodeModel(model.id)) {
139
+ score += 10;
140
+ reasons.push('Strong coding capability');
141
+ }
142
+ }
143
+ if (requirements.complexity === 'complex' && isAdvanced) {
144
+ score += 10;
145
+ reasons.push('Suitable for complex tasks');
146
+ }
147
+ else if (requirements.complexity === 'simple' && isFast) {
148
+ score += 5;
149
+ reasons.push('Efficient for simple tasks');
150
+ }
151
+ return {
152
+ model,
153
+ score: Math.max(0, score),
154
+ reasons,
155
+ };
156
+ }
157
+ toRecommendation(scored, allScored, requirements) {
158
+ const fallbacks = allScored
159
+ .filter((s) => s.model.id !== scored.model.id)
160
+ .slice(0, 3)
161
+ .map((s) => s.model.id);
162
+ return {
163
+ modelId: scored.model.id,
164
+ provider: scored.model.provider,
165
+ score: scored.score,
166
+ reasons: scored.reasons,
167
+ estimatedCost: this.estimateCost(scored.model, requirements.complexity ?? 'moderate'),
168
+ fallbacks,
169
+ };
170
+ }
171
+ estimateCost(model, complexity) {
172
+ const tokenEstimates = {
173
+ simple: { input: 500, output: 200 },
174
+ moderate: { input: 2000, output: 1000 },
175
+ complex: { input: 8000, output: 4000 },
176
+ };
177
+ const estimate = tokenEstimates[complexity];
178
+ const inputCost = (model.pricing.input * estimate.input) / 1_000_000;
179
+ const outputCost = (model.pricing.output * estimate.output) / 1_000_000;
180
+ return inputCost + outputCost;
181
+ }
182
+ fallbackRecommendation() {
183
+ return {
184
+ modelId: 'gpt-4o-mini',
185
+ provider: 'openai',
186
+ score: 50,
187
+ reasons: ['Fallback model - no suitable candidates found'],
188
+ estimatedCost: 0.0005,
189
+ fallbacks: ['claude-3-5-haiku', 'gemini-1.5-flash'],
190
+ };
191
+ }
192
+ isLocalModel(modelId) {
193
+ const lower = modelId.toLowerCase();
194
+ return (lower.includes('ollama') ||
195
+ lower.includes('llama') ||
196
+ lower.includes('phi') ||
197
+ lower.includes('qwen') ||
198
+ lower.includes('mixtral') ||
199
+ lower.includes('codellama') ||
200
+ lower.includes('deepseek') ||
201
+ lower.includes('gemma'));
202
+ }
203
+ isAdvancedModel(modelId) {
204
+ const lower = modelId.toLowerCase();
205
+ return ADVANCED_MODELS.some((m) => lower.includes(m.toLowerCase()));
206
+ }
207
+ isFastModel(modelId) {
208
+ const lower = modelId.toLowerCase();
209
+ return FAST_MODELS.some((m) => lower.includes(m.toLowerCase()));
210
+ }
211
+ isCodeModel(modelId) {
212
+ const lower = modelId.toLowerCase();
213
+ return CODE_MODELS.some((m) => lower.includes(m.toLowerCase()));
214
+ }
215
+ }
216
+ //# sourceMappingURL=model-selector.js.map
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,13 @@
1
+ import type { TaskRequirements } from '@cogitator-ai/types';
2
+ export declare class TaskAnalyzer {
3
+ analyze(task: string): TaskRequirements;
4
+ private detectVision;
5
+ private detectToolNeeds;
6
+ private detectLongContext;
7
+ private detectReasoningLevel;
8
+ private detectSpeedNeeds;
9
+ private detectCostSensitivity;
10
+ private detectComplexity;
11
+ private detectDomains;
12
+ }
13
+ //# sourceMappingURL=task-analyzer.d.ts.map
@@ -0,0 +1 @@
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"}
@@ -0,0 +1,185 @@
1
+ const VISION_KEYWORDS = [
2
+ 'image',
3
+ 'photo',
4
+ 'picture',
5
+ 'screenshot',
6
+ 'diagram',
7
+ 'chart',
8
+ 'graph',
9
+ 'visual',
10
+ 'see',
11
+ 'look at',
12
+ 'analyze this',
13
+ 'show me',
14
+ 'drawing',
15
+ 'illustration',
16
+ 'png',
17
+ 'jpg',
18
+ 'jpeg',
19
+ 'gif',
20
+ 'svg',
21
+ ];
22
+ const TOOL_KEYWORDS = [
23
+ 'search',
24
+ 'calculate',
25
+ 'execute',
26
+ 'run',
27
+ 'fetch',
28
+ 'api',
29
+ 'database',
30
+ 'file',
31
+ 'code',
32
+ 'browse',
33
+ 'query',
34
+ 'call',
35
+ 'invoke',
36
+ 'request',
37
+ 'download',
38
+ 'upload',
39
+ 'scrape',
40
+ 'crawl',
41
+ ];
42
+ const ADVANCED_REASONING_KEYWORDS = [
43
+ 'analyze',
44
+ 'compare',
45
+ 'evaluate',
46
+ 'synthesize',
47
+ 'critique',
48
+ 'design',
49
+ 'architect',
50
+ 'optimize',
51
+ 'reason',
52
+ 'deduce',
53
+ 'infer',
54
+ 'prove',
55
+ 'derive',
56
+ 'solve complex',
57
+ 'multi-step',
58
+ 'strategic',
59
+ ];
60
+ const MODERATE_REASONING_KEYWORDS = [
61
+ 'explain',
62
+ 'summarize',
63
+ 'describe',
64
+ 'outline',
65
+ 'list',
66
+ 'categorize',
67
+ 'classify',
68
+ 'organize',
69
+ 'plan',
70
+ ];
71
+ const SPEED_KEYWORDS = {
72
+ fast: ['quick', 'fast', 'asap', 'urgent', 'immediately', 'brief', 'short'],
73
+ slow: ['detailed', 'thorough', 'comprehensive', 'in-depth', 'extensive', 'careful'],
74
+ };
75
+ const DOMAIN_PATTERNS = {
76
+ code: /\b(code|program|function|class|api|typescript|javascript|python|rust|go|java|c\+\+|implement|debug|refactor|test|compile|build)\b/i,
77
+ math: /\b(math|calculate|equation|formula|statistics|probability|algebra|calculus|numerical|compute)\b/i,
78
+ creative: /\b(write|story|article|creative|blog|content|poem|essay|narrative|fiction|script|dialogue)\b/i,
79
+ analysis: /\b(analyze|research|data|report|study|investigate|examine|assess|evaluate|review)\b/i,
80
+ legal: /\b(legal|law|contract|compliance|regulation|policy|statute|liability|terms)\b/i,
81
+ medical: /\b(medical|health|clinical|diagnosis|treatment|symptom|patient|doctor|healthcare)\b/i,
82
+ finance: /\b(finance|financial|investment|stock|trading|budget|accounting|revenue|profit)\b/i,
83
+ };
84
+ export class TaskAnalyzer {
85
+ analyze(task) {
86
+ const lower = task.toLowerCase();
87
+ return {
88
+ needsVision: this.detectVision(lower),
89
+ needsToolCalling: this.detectToolNeeds(lower),
90
+ needsLongContext: this.detectLongContext(task),
91
+ needsReasoning: this.detectReasoningLevel(lower),
92
+ needsSpeed: this.detectSpeedNeeds(lower),
93
+ costSensitivity: this.detectCostSensitivity(lower),
94
+ complexity: this.detectComplexity(task),
95
+ domains: this.detectDomains(task),
96
+ };
97
+ }
98
+ detectVision(task) {
99
+ return VISION_KEYWORDS.some((k) => task.includes(k));
100
+ }
101
+ detectToolNeeds(task) {
102
+ return TOOL_KEYWORDS.some((k) => task.includes(k));
103
+ }
104
+ detectLongContext(task) {
105
+ const wordCount = task.split(/\s+/).length;
106
+ if (wordCount > 500)
107
+ return true;
108
+ const longContextIndicators = [
109
+ 'document',
110
+ 'file',
111
+ 'codebase',
112
+ 'repository',
113
+ 'large',
114
+ 'entire',
115
+ 'full',
116
+ 'complete',
117
+ 'all of',
118
+ 'whole',
119
+ 'multiple files',
120
+ 'many pages',
121
+ ];
122
+ const lower = task.toLowerCase();
123
+ return longContextIndicators.some((k) => lower.includes(k));
124
+ }
125
+ detectReasoningLevel(task) {
126
+ if (ADVANCED_REASONING_KEYWORDS.some((k) => task.includes(k))) {
127
+ return 'advanced';
128
+ }
129
+ if (MODERATE_REASONING_KEYWORDS.some((k) => task.includes(k))) {
130
+ return 'moderate';
131
+ }
132
+ return 'basic';
133
+ }
134
+ detectSpeedNeeds(task) {
135
+ if (SPEED_KEYWORDS.fast.some((k) => task.includes(k))) {
136
+ return 'fast';
137
+ }
138
+ if (SPEED_KEYWORDS.slow.some((k) => task.includes(k))) {
139
+ return 'slow-ok';
140
+ }
141
+ return 'balanced';
142
+ }
143
+ detectCostSensitivity(task) {
144
+ const lower = task;
145
+ if (/\b(cheap|free|budget|low.?cost|economical)\b/.test(lower)) {
146
+ return 'high';
147
+ }
148
+ if (/\b(best|premium|quality|accurate|precise)\b/.test(lower)) {
149
+ return 'low';
150
+ }
151
+ return 'medium';
152
+ }
153
+ detectComplexity(task) {
154
+ const wordCount = task.split(/\s+/).length;
155
+ const sentenceCount = task.split(/[.!?]+/).filter((s) => s.trim()).length;
156
+ const hasMultipleSteps = /\b(then|after|next|finally|first|second|third|step)\b/i.test(task);
157
+ const hasConditions = /\b(if|when|unless|otherwise|either|or)\b/i.test(task);
158
+ let complexityScore = 0;
159
+ if (wordCount > 100)
160
+ complexityScore++;
161
+ if (wordCount > 300)
162
+ complexityScore++;
163
+ if (sentenceCount > 5)
164
+ complexityScore++;
165
+ if (hasMultipleSteps)
166
+ complexityScore++;
167
+ if (hasConditions)
168
+ complexityScore++;
169
+ if (complexityScore >= 3)
170
+ return 'complex';
171
+ if (complexityScore >= 1)
172
+ return 'moderate';
173
+ return 'simple';
174
+ }
175
+ detectDomains(task) {
176
+ const domains = [];
177
+ for (const [domain, pattern] of Object.entries(DOMAIN_PATTERNS)) {
178
+ if (pattern.test(task)) {
179
+ domains.push(domain);
180
+ }
181
+ }
182
+ return domains;
183
+ }
184
+ }
185
+ //# sourceMappingURL=task-analyzer.js.map
@@ -0,0 +1 @@
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 CHANGED
@@ -12,15 +12,26 @@ export { Logger, getLogger, setLogger, createLogger } from './logger';
12
12
  export type { LogLevel, LogContext, LogEntry, LoggerOptions } from './logger';
13
13
  export { ReflectionEngine, InMemoryInsightStore } from './reflection/index';
14
14
  export type { ReflectionEngineOptions } from './reflection/index';
15
- export { ThoughtTreeExecutor, BranchGenerator, BranchEvaluator, } from './reasoning/index';
15
+ export { ThoughtTreeExecutor, BranchGenerator, BranchEvaluator } from './reasoning/index';
16
16
  export type { BranchEvaluatorOptions } from './reasoning/index';
17
17
  export { AgentOptimizer, InMemoryTraceStore, MetricEvaluator, DemoSelector, InstructionOptimizer, createSuccessMetric, createExactMatchMetric, createContainsMetric, } from './learning/index';
18
18
  export type { AgentOptimizerOptions, MetricEvaluatorOptions, DemoSelectorOptions, InstructionOptimizerOptions, } from './learning/index';
19
+ export { TimeTravel, InMemoryCheckpointStore, ExecutionReplayer, ExecutionForker, TraceComparator, } from './time-travel/index';
20
+ export type { TimeTravelOptions, ExecutionReplayerOptions, ExecutionForkerOptions, TraceComparatorOptions, } from './time-travel/index';
21
+ export { ConstitutionalAI, InputFilter, OutputFilter, ToolGuard, CritiqueReviser, DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './constitutional/index';
22
+ export type { ConstitutionalAIOptions, InputFilterOptions, OutputFilterOptions, ToolGuardOptions, CritiqueReviserOptions, } from './constitutional/index';
23
+ export { CostAwareRouter, TaskAnalyzer, ModelSelector, CostTracker, BudgetEnforcer, } from './cost-routing/index';
24
+ export type { CostAwareRouterOptions, CostFilter, BudgetCheckResult } from './cost-routing/index';
19
25
  export { BaseLLMBackend, OllamaBackend, OpenAIBackend, AnthropicBackend, createLLMBackend, parseModel, } from './llm/index';
20
26
  export { withRetry, retryable, CircuitBreaker, CircuitBreakerRegistry, withFallback, withGracefulDegradation, createLLMFallbackExecutor, } from './utils/index';
21
27
  export type { RetryOptions, CircuitBreakerOptions, CircuitBreakerStats, CircuitState, FallbackConfig, LLMFallbackConfig, } from './utils/index';
22
28
  export { CogitatorError, ErrorCode, ERROR_STATUS_CODES, isRetryableError, getRetryDelay, } from '@cogitator-ai/types';
23
29
  export type { ErrorDetails, CogitatorErrorOptions } from '@cogitator-ai/types';
24
30
  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';
25
- export { DEFAULT_TOT_CONFIG, DEFAULT_LEARNING_CONFIG, DEFAULT_OPTIMIZER_CONFIG } from '@cogitator-ai/types';
31
+ export { DEFAULT_TOT_CONFIG, DEFAULT_LEARNING_CONFIG, DEFAULT_OPTIMIZER_CONFIG, DEFAULT_TIME_TRAVEL_CONFIG, } from '@cogitator-ai/types';
32
+ export type { ExecutionCheckpoint, TimeTravelCheckpointStore, CheckpointQuery, ReplayOptions, ReplayResult, ReplayMode, ForkOptions, ForkResult, TraceDiff, StepDiff, StepDiffStatus, TimeTravelConfig, } from '@cogitator-ai/types';
33
+ export type { HarmCategory, Severity, FilterLayer, PrincipleCategory, ConstitutionalPrinciple, Constitution, HarmScore, FilterResult, CritiqueResult, RevisionResult, GuardrailConfig, ToolGuardResult, } from '@cogitator-ai/types';
34
+ export { DEFAULT_GUARDRAIL_CONFIG } from '@cogitator-ai/types';
35
+ export type { CostRoutingConfig, BudgetConfig, CostRecord, CostSummary, ModelRecommendation, TaskRequirements, TaskComplexity, ReasoningLevel, SpeedPreference, CostSensitivity, } from '@cogitator-ai/types';
36
+ export { DEFAULT_COST_ROUTING_CONFIG } from '@cogitator-ai/types';
26
37
  //# sourceMappingURL=index.d.ts.map
@@ -1 +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,EAAE,gBAAgB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC5E,YAAY,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAElE,OAAO,EACL,mBAAmB,EACnB,eAAe,EACf,eAAe,GAChB,MAAM,mBAAmB,CAAC;AAC3B,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,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,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,MAAM,qBAAqB,CAAC"}
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,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 CHANGED
@@ -10,10 +10,15 @@ export { ToolRegistry } from './registry';
10
10
  export { calculator, datetime, builtinTools } from './tools/index';
11
11
  export { Logger, getLogger, setLogger, createLogger } from './logger';
12
12
  export { ReflectionEngine, InMemoryInsightStore } from './reflection/index';
13
- export { ThoughtTreeExecutor, BranchGenerator, BranchEvaluator, } from './reasoning/index';
13
+ export { ThoughtTreeExecutor, BranchGenerator, BranchEvaluator } from './reasoning/index';
14
14
  export { AgentOptimizer, InMemoryTraceStore, MetricEvaluator, DemoSelector, InstructionOptimizer, createSuccessMetric, createExactMatchMetric, createContainsMetric, } from './learning/index';
15
+ export { TimeTravel, InMemoryCheckpointStore, ExecutionReplayer, ExecutionForker, TraceComparator, } from './time-travel/index';
16
+ export { ConstitutionalAI, InputFilter, OutputFilter, ToolGuard, CritiqueReviser, DEFAULT_CONSTITUTION, DEFAULT_PRINCIPLES, createConstitution, extendConstitution, filterPrinciplesByLayer, getPrinciplesByCategory, getPrinciplesBySeverity, buildInputEvaluationPrompt, buildOutputEvaluationPrompt, buildCritiquePrompt, buildRevisionPrompt, parseEvaluationResponse, parseCritiqueResponse, } from './constitutional/index';
17
+ export { CostAwareRouter, TaskAnalyzer, ModelSelector, CostTracker, BudgetEnforcer, } from './cost-routing/index';
15
18
  export { BaseLLMBackend, OllamaBackend, OpenAIBackend, AnthropicBackend, createLLMBackend, parseModel, } from './llm/index';
16
19
  export { withRetry, retryable, CircuitBreaker, CircuitBreakerRegistry, withFallback, withGracefulDegradation, createLLMFallbackExecutor, } from './utils/index';
17
20
  export { CogitatorError, ErrorCode, ERROR_STATUS_CODES, isRetryableError, getRetryDelay, } from '@cogitator-ai/types';
18
- export { DEFAULT_TOT_CONFIG, DEFAULT_LEARNING_CONFIG, DEFAULT_OPTIMIZER_CONFIG } from '@cogitator-ai/types';
21
+ export { DEFAULT_TOT_CONFIG, DEFAULT_LEARNING_CONFIG, DEFAULT_OPTIMIZER_CONFIG, DEFAULT_TIME_TRAVEL_CONFIG, } from '@cogitator-ai/types';
22
+ export { DEFAULT_GUARDRAIL_CONFIG } from '@cogitator-ai/types';
23
+ export { DEFAULT_COST_ROUTING_CONFIG } from '@cogitator-ai/types';
19
24
  //# sourceMappingURL=index.js.map