adaptive-memory-multi-model-router 1.2.2

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 (95) hide show
  1. package/README.md +114 -0
  2. package/demo/research-demo.js +266 -0
  3. package/dist/cache/prefixCache.d.ts +114 -0
  4. package/dist/cache/prefixCache.d.ts.map +1 -0
  5. package/dist/cache/prefixCache.js +285 -0
  6. package/dist/cache/prefixCache.js.map +1 -0
  7. package/dist/cache/responseCache.d.ts +58 -0
  8. package/dist/cache/responseCache.d.ts.map +1 -0
  9. package/dist/cache/responseCache.js +153 -0
  10. package/dist/cache/responseCache.js.map +1 -0
  11. package/dist/cli.js +59 -0
  12. package/dist/cost/costTracker.d.ts +95 -0
  13. package/dist/cost/costTracker.d.ts.map +1 -0
  14. package/dist/cost/costTracker.js +240 -0
  15. package/dist/cost/costTracker.js.map +1 -0
  16. package/dist/index.d.ts +723 -0
  17. package/dist/index.d.ts.map +1 -0
  18. package/dist/index.js +239 -0
  19. package/dist/index.js.map +1 -0
  20. package/dist/memory/episodicMemory.d.ts +82 -0
  21. package/dist/memory/episodicMemory.d.ts.map +1 -0
  22. package/dist/memory/episodicMemory.js +145 -0
  23. package/dist/memory/episodicMemory.js.map +1 -0
  24. package/dist/orchestration/haloOrchestrator.d.ts +102 -0
  25. package/dist/orchestration/haloOrchestrator.d.ts.map +1 -0
  26. package/dist/orchestration/haloOrchestrator.js +207 -0
  27. package/dist/orchestration/haloOrchestrator.js.map +1 -0
  28. package/dist/orchestration/mctsWorkflow.d.ts +85 -0
  29. package/dist/orchestration/mctsWorkflow.d.ts.map +1 -0
  30. package/dist/orchestration/mctsWorkflow.js +210 -0
  31. package/dist/orchestration/mctsWorkflow.js.map +1 -0
  32. package/dist/providers/localProvider.d.ts +102 -0
  33. package/dist/providers/localProvider.d.ts.map +1 -0
  34. package/dist/providers/localProvider.js +338 -0
  35. package/dist/providers/localProvider.js.map +1 -0
  36. package/dist/providers/registry.d.ts +55 -0
  37. package/dist/providers/registry.d.ts.map +1 -0
  38. package/dist/providers/registry.js +138 -0
  39. package/dist/providers/registry.js.map +1 -0
  40. package/dist/routing/advancedRouter.d.ts +68 -0
  41. package/dist/routing/advancedRouter.d.ts.map +1 -0
  42. package/dist/routing/advancedRouter.js +332 -0
  43. package/dist/routing/advancedRouter.js.map +1 -0
  44. package/dist/tools/tmlpdTools.d.ts +101 -0
  45. package/dist/tools/tmlpdTools.d.ts.map +1 -0
  46. package/dist/tools/tmlpdTools.js +368 -0
  47. package/dist/tools/tmlpdTools.js.map +1 -0
  48. package/dist/utils/batchProcessor.d.ts +96 -0
  49. package/dist/utils/batchProcessor.d.ts.map +1 -0
  50. package/dist/utils/batchProcessor.js +170 -0
  51. package/dist/utils/batchProcessor.js.map +1 -0
  52. package/dist/utils/compression.d.ts +61 -0
  53. package/dist/utils/compression.d.ts.map +1 -0
  54. package/dist/utils/compression.js +281 -0
  55. package/dist/utils/compression.js.map +1 -0
  56. package/dist/utils/reliability.d.ts +74 -0
  57. package/dist/utils/reliability.d.ts.map +1 -0
  58. package/dist/utils/reliability.js +177 -0
  59. package/dist/utils/reliability.js.map +1 -0
  60. package/dist/utils/speculativeDecoding.d.ts +117 -0
  61. package/dist/utils/speculativeDecoding.d.ts.map +1 -0
  62. package/dist/utils/speculativeDecoding.js +246 -0
  63. package/dist/utils/speculativeDecoding.js.map +1 -0
  64. package/dist/utils/tokenUtils.d.ts +50 -0
  65. package/dist/utils/tokenUtils.d.ts.map +1 -0
  66. package/dist/utils/tokenUtils.js +124 -0
  67. package/dist/utils/tokenUtils.js.map +1 -0
  68. package/examples/QUICKSTART.md +183 -0
  69. package/notebooks/quickstart.ipynb +157 -0
  70. package/package.json +83 -0
  71. package/python/examples.py +53 -0
  72. package/python/integrations.py +330 -0
  73. package/python/setup.py +28 -0
  74. package/python/tmlpd.py +369 -0
  75. package/qna/REDDIT_GAP_ANALYSIS.md +299 -0
  76. package/qna/TMLPD_QNA.md +751 -0
  77. package/rust/tmlpd.h +268 -0
  78. package/skill/SKILL.md +238 -0
  79. package/src/cache/prefixCache.ts +365 -0
  80. package/src/cache/responseCache.ts +147 -0
  81. package/src/cost/costTracker.ts +302 -0
  82. package/src/index.ts +224 -0
  83. package/src/memory/episodicMemory.ts +185 -0
  84. package/src/orchestration/haloOrchestrator.ts +266 -0
  85. package/src/orchestration/mctsWorkflow.ts +262 -0
  86. package/src/providers/localProvider.ts +406 -0
  87. package/src/providers/registry.ts +164 -0
  88. package/src/routing/advancedRouter.ts +406 -0
  89. package/src/tools/tmlpdTools.ts +433 -0
  90. package/src/utils/batchProcessor.ts +232 -0
  91. package/src/utils/compression.ts +325 -0
  92. package/src/utils/reliability.ts +221 -0
  93. package/src/utils/speculativeDecoding.ts +344 -0
  94. package/src/utils/tokenUtils.ts +145 -0
  95. package/tsconfig.json +18 -0
@@ -0,0 +1,368 @@
1
+ "use strict";
2
+ /**
3
+ * TMLPD PI Tools
4
+ *
5
+ * Main tools exposed to the PI agent via the MCP bridge.
6
+ * Features: streaming, caching, cost tracking, reliability.
7
+ */
8
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
9
+ if (k2 === undefined) k2 = k;
10
+ var desc = Object.getOwnPropertyDescriptor(m, k);
11
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
12
+ desc = { enumerable: true, get: function() { return m[k]; } };
13
+ }
14
+ Object.defineProperty(o, k2, desc);
15
+ }) : (function(o, m, k, k2) {
16
+ if (k2 === undefined) k2 = k;
17
+ o[k2] = m[k];
18
+ }));
19
+ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
20
+ Object.defineProperty(o, "default", { enumerable: true, value: v });
21
+ }) : function(o, v) {
22
+ o["default"] = v;
23
+ });
24
+ var __importStar = (this && this.__importStar) || (function () {
25
+ var ownKeys = function(o) {
26
+ ownKeys = Object.getOwnPropertyNames || function (o) {
27
+ var ar = [];
28
+ for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
29
+ return ar;
30
+ };
31
+ return ownKeys(o);
32
+ };
33
+ return function (mod) {
34
+ if (mod && mod.__esModule) return mod;
35
+ var result = {};
36
+ if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
37
+ __setModuleDefault(result, mod);
38
+ return result;
39
+ };
40
+ })();
41
+ Object.defineProperty(exports, "__esModule", { value: true });
42
+ exports.TMLPDTools = void 0;
43
+ exports.createTMLPD = createTMLPD;
44
+ exports.getDefault = getDefault;
45
+ const responseCache_1 = require("../cache/responseCache");
46
+ const costTracker_1 = require("../cost/costTracker");
47
+ const registry_1 = require("../providers/registry");
48
+ const reliability_1 = require("../utils/reliability");
49
+ const https = __importStar(require("https"));
50
+ const http = __importStar(require("http"));
51
+ class TMLPDTools {
52
+ cache;
53
+ costTracker;
54
+ registry;
55
+ circuitBreakers = new Map();
56
+ retryConfig;
57
+ maxConcurrent;
58
+ constructor(config = {}) {
59
+ this.cache = new responseCache_1.ResponseCache(config.cache);
60
+ this.costTracker = new costTracker_1.CostTracker(config.budget);
61
+ this.registry = new registry_1.ProviderRegistry();
62
+ this.retryConfig = { ...reliability_1.DEFAULT_RETRY_CONFIG, ...config.retry };
63
+ this.maxConcurrent = config.maxConcurrent || 5;
64
+ }
65
+ /**
66
+ * Get cost summary
67
+ */
68
+ getCostSummary() {
69
+ return this.costTracker.getSummary();
70
+ }
71
+ /**
72
+ * Get remaining budget
73
+ */
74
+ getRemainingBudget() {
75
+ return this.costTracker.getRemainingBudget();
76
+ }
77
+ /**
78
+ * Get cache stats
79
+ */
80
+ getCacheStats() {
81
+ return this.cache.getStats();
82
+ }
83
+ /**
84
+ * Get provider status
85
+ */
86
+ getProviderStatus() {
87
+ return this.registry.getStatus();
88
+ }
89
+ /**
90
+ * Execute single prompt with optional streaming
91
+ */
92
+ async execute(prompt, model, streaming) {
93
+ const selectedModel = model || this.registry.selectModel();
94
+ if (!selectedModel) {
95
+ return {
96
+ success: false,
97
+ error: "No providers available",
98
+ model: "unknown",
99
+ provider: "unknown",
100
+ };
101
+ }
102
+ // Check cache first
103
+ const cached = this.cache.get(prompt, selectedModel);
104
+ if (cached) {
105
+ return {
106
+ success: true,
107
+ content: cached.content,
108
+ model: cached.model,
109
+ provider: cached.provider,
110
+ tokens: cached.tokens,
111
+ cost: 0, // Cache hit = no cost
112
+ cached: true,
113
+ duration_ms: 0,
114
+ };
115
+ }
116
+ const provider = selectedModel.split("/")[0];
117
+ const breaker = this.getCircuitBreaker(provider);
118
+ const startTime = Date.now();
119
+ const { result, error, attempts, circuit_tripped } = await (0, reliability_1.withRetry)(async () => {
120
+ return this.executeRequest(selectedModel, streaming);
121
+ }, this.retryConfig, breaker);
122
+ const duration_ms = Date.now() - startTime;
123
+ if (!result) {
124
+ return {
125
+ success: false,
126
+ error: error?.message || "Execution failed",
127
+ model: selectedModel,
128
+ provider,
129
+ attempts,
130
+ };
131
+ }
132
+ // Record cost
133
+ const costSnapshot = this.costTracker.record(provider, selectedModel, result.input_tokens || 0, result.output_tokens || 0);
134
+ // Cache successful response
135
+ if (result.content) {
136
+ this.cache.set(prompt, selectedModel, {
137
+ content: result.content,
138
+ model: selectedModel,
139
+ provider,
140
+ tokens: costSnapshot.input_tokens + costSnapshot.output_tokens,
141
+ cost: costSnapshot.total_cost,
142
+ });
143
+ }
144
+ return {
145
+ success: true,
146
+ content: result.content,
147
+ model: selectedModel,
148
+ provider,
149
+ tokens: costSnapshot.input_tokens + costSnapshot.output_tokens,
150
+ cost: costSnapshot.total_cost,
151
+ duration_ms,
152
+ attempts,
153
+ };
154
+ }
155
+ /**
156
+ * Execute parallel across multiple models
157
+ */
158
+ async executeParallel(prompt, models, streaming) {
159
+ const selectedModels = models || this.registry.getReadyProviders().slice(0, 3).map(p => `${p}/default`);
160
+ const startTime = Date.now();
161
+ const results = [];
162
+ let total_cost = 0;
163
+ // Execute with concurrency limit
164
+ for (let i = 0; i < selectedModels.length; i += this.maxConcurrent) {
165
+ const batch = selectedModels.slice(i, i + this.maxConcurrent);
166
+ const batchResults = await Promise.all(batch.map((model) => this.execute(prompt, model, streaming)));
167
+ results.push(...batchResults);
168
+ for (const r of batchResults) {
169
+ if (r.success && r.cost) {
170
+ total_cost += r.cost;
171
+ }
172
+ }
173
+ }
174
+ const duration_ms = Date.now() - startTime;
175
+ return {
176
+ responses: results,
177
+ total_models: selectedModels.length,
178
+ successful_models: results.filter((r) => r.success).length,
179
+ total_cost,
180
+ duration_ms,
181
+ };
182
+ }
183
+ /**
184
+ * Get circuit breaker for provider
185
+ */
186
+ getCircuitBreaker(provider) {
187
+ if (!this.circuitBreakers.has(provider)) {
188
+ this.circuitBreakers.set(provider, new reliability_1.CircuitBreaker());
189
+ }
190
+ return this.circuitBreakers.get(provider);
191
+ }
192
+ /**
193
+ * Execute HTTP request to provider
194
+ */
195
+ async executeRequest(model, streaming) {
196
+ const provider = model.split("/")[0];
197
+ const providerConfig = this.registry.getStatus().providers[provider];
198
+ if (!providerConfig?.ready) {
199
+ throw new Error(`Provider ${provider} is not available`);
200
+ }
201
+ // Build request based on provider mode
202
+ if (providerConfig.mode === "gemini") {
203
+ return this.executeGemini(provider, model);
204
+ }
205
+ else if (providerConfig.mode === "anthropic") {
206
+ return this.executeAnthropic(provider, model);
207
+ }
208
+ else {
209
+ return this.executeOpenAI(provider, model);
210
+ }
211
+ }
212
+ async executeOpenAI(provider, model) {
213
+ return new Promise((resolve, reject) => {
214
+ const status = this.registry.getStatus().providers[provider];
215
+ const apiKey = process.env[`${provider.toUpperCase()}_API_KEY`] || "";
216
+ const baseUrl = process.env[`${provider.toUpperCase()}_OPENAI_BASE_URL`] || "";
217
+ const payload = JSON.stringify({
218
+ model: model.split("/")[1],
219
+ messages: [{ role: "user", content: "Placeholder" }], // Will be replaced by actual prompt via closure
220
+ });
221
+ const url = new URL(`${baseUrl}/chat/completions`);
222
+ const options = {
223
+ hostname: url.hostname,
224
+ port: url.port || (url.protocol === "https:" ? 443 : 80),
225
+ path: url.pathname,
226
+ method: "POST",
227
+ headers: {
228
+ "Content-Type": "application/json",
229
+ "Authorization": `Bearer ${apiKey}`,
230
+ "User-Agent": "TMLPD-Pi/1.0",
231
+ },
232
+ };
233
+ const protocol = url.protocol === "https:" ? https : http;
234
+ const req = protocol.request(options, (res) => {
235
+ let data = "";
236
+ res.on("data", (chunk) => (data += chunk));
237
+ res.on("end", () => {
238
+ if (res.statusCode && res.statusCode >= 400) {
239
+ reject(new Error(`HTTP ${res.statusCode}`));
240
+ return;
241
+ }
242
+ try {
243
+ const json = JSON.parse(data);
244
+ const content = json.choices?.[0]?.message?.content || "";
245
+ const tokens = json.usage?.total_tokens || 0;
246
+ resolve({ content, input_tokens: Math.floor(tokens * 0.3), output_tokens: Math.floor(tokens * 0.7) });
247
+ }
248
+ catch {
249
+ reject(new Error("Invalid JSON response"));
250
+ }
251
+ });
252
+ });
253
+ req.on("error", reject);
254
+ req.write(payload);
255
+ req.end();
256
+ });
257
+ }
258
+ async executeAnthropic(provider, model) {
259
+ return new Promise((resolve, reject) => {
260
+ const apiKey = process.env[`${provider.toUpperCase()}_API_KEY`] || "";
261
+ const baseUrl = process.env[`${provider.toUpperCase()}_BASE_URL`] || "";
262
+ const payload = JSON.stringify({
263
+ model: model.split("/")[1],
264
+ max_tokens: 4096,
265
+ messages: [{ role: "user", content: "Placeholder" }],
266
+ });
267
+ const url = new URL(baseUrl);
268
+ const options = {
269
+ hostname: url.hostname,
270
+ port: url.port || 443,
271
+ path: url.pathname,
272
+ method: "POST",
273
+ headers: {
274
+ "Content-Type": "application/json",
275
+ "x-api-key": apiKey,
276
+ "anthropic-version": "2023-06-01",
277
+ "User-Agent": "TMLPD-Pi/1.0",
278
+ },
279
+ };
280
+ const req = https.request(options, (res) => {
281
+ let data = "";
282
+ res.on("data", (chunk) => (data += chunk));
283
+ res.on("end", () => {
284
+ if (res.statusCode && res.statusCode >= 400) {
285
+ reject(new Error(`HTTP ${res.statusCode}`));
286
+ return;
287
+ }
288
+ try {
289
+ const json = JSON.parse(data);
290
+ const content = json.content?.[0]?.text || "";
291
+ const tokens = json.usage?.output_tokens || 0;
292
+ resolve({ content, input_tokens: Math.floor(tokens * 0.3), output_tokens: Math.floor(tokens * 0.7) });
293
+ }
294
+ catch {
295
+ reject(new Error("Invalid JSON response"));
296
+ }
297
+ });
298
+ });
299
+ req.on("error", reject);
300
+ req.write(payload);
301
+ req.end();
302
+ });
303
+ }
304
+ async executeGemini(provider, model) {
305
+ return new Promise((resolve, reject) => {
306
+ const apiKey = process.env.GOOGLE_API_KEY || "";
307
+ const baseUrl = "https://generativelanguage.googleapis.com/v1beta";
308
+ const payload = JSON.stringify({
309
+ contents: [{ parts: [{ text: "Placeholder" }] }],
310
+ generationConfig: { maxOutputTokens: 4096 },
311
+ });
312
+ const url = new URL(`${baseUrl}/models/${model.split("/")[1]}:generateContent`);
313
+ url.searchParams.set("key", apiKey);
314
+ const options = {
315
+ hostname: url.hostname,
316
+ port: url.port || 443,
317
+ path: url.pathname + url.search,
318
+ method: "POST",
319
+ headers: {
320
+ "Content-Type": "application/json",
321
+ "User-Agent": "TMLPD-Pi/1.0",
322
+ },
323
+ };
324
+ const req = https.request(options, (res) => {
325
+ let data = "";
326
+ res.on("data", (chunk) => (data += chunk));
327
+ res.on("end", () => {
328
+ if (res.statusCode && res.statusCode >= 400) {
329
+ reject(new Error(`HTTP ${res.statusCode}`));
330
+ return;
331
+ }
332
+ try {
333
+ const json = JSON.parse(data);
334
+ const content = json.candidates?.[0]?.content?.parts?.[0]?.text || "";
335
+ resolve({ content, input_tokens: 100, output_tokens: 100 });
336
+ }
337
+ catch {
338
+ reject(new Error("Invalid JSON response"));
339
+ }
340
+ });
341
+ });
342
+ req.on("error", reject);
343
+ req.write(payload);
344
+ req.end();
345
+ });
346
+ }
347
+ }
348
+ exports.TMLPDTools = TMLPDTools;
349
+ // ============================================================================
350
+ // Factory Function
351
+ // ============================================================================
352
+ let _defaultInstance = null;
353
+ /**
354
+ * Create a TMLPD instance
355
+ */
356
+ function createTMLPD(config) {
357
+ return new TMLPDTools(config);
358
+ }
359
+ /**
360
+ * Get default singleton instance
361
+ */
362
+ function getDefault() {
363
+ if (!_defaultInstance) {
364
+ _defaultInstance = new TMLPDTools();
365
+ }
366
+ return _defaultInstance;
367
+ }
368
+ //# sourceMappingURL=tmlpdTools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tmlpdTools.js","sourceRoot":"","sources":["../../src/tools/tmlpdTools.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+ZH,kCAEC;AAKD,gCAKC;AAzaD,0DAAoE;AACpE,qDAA6E;AAC7E,oDAAyD;AACzD,sDAAoG;AACpG,6CAA+B;AAC/B,2CAA6B;AAoC7B,MAAa,UAAU;IACb,KAAK,CAAgB;IACrB,WAAW,CAAc;IACzB,QAAQ,CAAmB;IAC3B,eAAe,GAAgC,IAAI,GAAG,EAAE,CAAC;IACzD,WAAW,CAAc;IACzB,aAAa,CAAS;IAE9B,YAAY,SAAsB,EAAE;QAClC,IAAI,CAAC,KAAK,GAAG,IAAI,6BAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,WAAW,GAAG,IAAI,yBAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,2BAAgB,EAAE,CAAC;QACvC,IAAI,CAAC,WAAW,GAAG,EAAE,GAAG,kCAAoB,EAAE,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;QAChE,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,kBAAkB;QAChB,OAAO,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,MAAc,EACd,KAAc,EACd,SAA2B;QAE3B,MAAM,aAAa,GAAG,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC;QAC3D,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,wBAAwB;gBAC/B,KAAK,EAAE,SAAS;gBAChB,QAAQ,EAAE,SAAS;aACpB,CAAC;QACJ,CAAC;QAED,oBAAoB;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QACrD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,IAAI,EAAE,CAAC,EAAE,sBAAsB;gBAC/B,MAAM,EAAE,IAAI;gBACZ,WAAW,EAAE,CAAC;aACf,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAEjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,GAAG,MAAM,IAAA,uBAAS,EAClE,KAAK,IAAI,EAAE;YACT,OAAO,IAAI,CAAC,cAAc,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC,EACD,IAAI,CAAC,WAAW,EAChB,OAAO,CACR,CAAC;QAEF,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAE3C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,EAAE,OAAO,IAAI,kBAAkB;gBAC3C,KAAK,EAAE,aAAa;gBACpB,QAAQ;gBACR,QAAQ;aACT,CAAC;QACJ,CAAC;QAED,cAAc;QACd,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAC1C,QAAQ,EACR,aAAa,EACb,MAAM,CAAC,YAAY,IAAI,CAAC,EACxB,MAAM,CAAC,aAAa,IAAI,CAAC,CAC1B,CAAC;QAEF,4BAA4B;QAC5B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,aAAa,EAAE;gBACpC,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,aAAa;gBACpB,QAAQ;gBACR,MAAM,EAAE,YAAY,CAAC,YAAY,GAAG,YAAY,CAAC,aAAa;gBAC9D,IAAI,EAAE,YAAY,CAAC,UAAU;aAC9B,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,KAAK,EAAE,aAAa;YACpB,QAAQ;YACR,MAAM,EAAE,YAAY,CAAC,YAAY,GAAG,YAAY,CAAC,aAAa;YAC9D,IAAI,EAAE,YAAY,CAAC,UAAU;YAC7B,WAAW;YACX,QAAQ;SACT,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,MAAiB,EACjB,SAA2B;QAE3B,MAAM,cAAc,GAAG,MAAM,IAAI,IAAI,CAAC,QAAQ,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACxG,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,MAAM,OAAO,GAAoB,EAAE,CAAC;QACpC,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,iCAAiC;QACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACnE,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9D,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CACpC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC,CAC7D,CAAC;YACF,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;YAE9B,KAAK,MAAM,CAAC,IAAI,YAAY,EAAE,CAAC;gBAC7B,IAAI,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;oBACxB,UAAU,IAAI,CAAC,CAAC,IAAI,CAAC;gBACvB,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAE3C,OAAO;YACL,SAAS,EAAE,OAAO;YAClB,YAAY,EAAE,cAAc,CAAC,MAAM;YACnC,iBAAiB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM;YAC1D,UAAU;YACV,WAAW;SACZ,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,QAAgB;QACxC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,4BAAc,EAAE,CAAC,CAAC;QAC3D,CAAC;QACD,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;IAC7C,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAC1B,KAAa,EACb,SAA2B;QAE3B,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAErE,IAAI,CAAC,cAAc,EAAE,KAAK,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,mBAAmB,CAAC,CAAC;QAC3D,CAAC;QAED,uCAAuC;QACvC,IAAI,cAAc,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACrC,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,cAAc,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/C,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAChD,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,QAAgB,EAChB,KAAa;QAEb,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;YACtE,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,kBAAkB,CAAC,IAAI,EAAE,CAAC;YAE/E,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC7B,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC1B,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,EAAE,gDAAgD;aACvG,CAAC,CAAC;YAEH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,OAAO,mBAAmB,CAAC,CAAC;YACnD,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACxD,IAAI,EAAE,GAAG,CAAC,QAAQ;gBAClB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,MAAM,EAAE;oBACnC,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;YAC1D,MAAM,GAAG,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC5C,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC;gBAC3C,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACjB,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,EAAE,CAAC;wBAC5C,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;wBAC5C,OAAO;oBACT,CAAC;oBACD,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;wBAC1D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;wBAC7C,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxG,CAAC;oBAAC,MAAM,CAAC;wBACP,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;oBAC7C,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACxB,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnB,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,QAAgB,EAChB,KAAa;QAEb,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;YACtE,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;YAExE,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC7B,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC1B,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;aACrD,CAAC,CAAC;YAEH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;YAC7B,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG;gBACrB,IAAI,EAAE,GAAG,CAAC,QAAQ;gBAClB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,WAAW,EAAE,MAAM;oBACnB,mBAAmB,EAAE,YAAY;oBACjC,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC;YAEF,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACzC,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC;gBAC3C,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACjB,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,EAAE,CAAC;wBAC5C,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;wBAC5C,OAAO;oBACT,CAAC;oBACD,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;wBAC9C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;wBAC9C,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,EAAE,CAAC,CAAC;oBACxG,CAAC;oBAAC,MAAM,CAAC;wBACP,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;oBAC7C,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACxB,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnB,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,QAAgB,EAChB,KAAa;QAEb,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,kDAAkD,CAAC;YAEnE,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC7B,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;gBAChD,gBAAgB,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE;aAC5C,CAAC,CAAC;YAEH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,OAAO,WAAW,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC;YAChF,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAEpC,MAAM,OAAO,GAAG;gBACd,QAAQ,EAAE,GAAG,CAAC,QAAQ;gBACtB,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,GAAG;gBACrB,IAAI,EAAE,GAAG,CAAC,QAAQ,GAAG,GAAG,CAAC,MAAM;gBAC/B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC;YAEF,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACzC,IAAI,IAAI,GAAG,EAAE,CAAC;gBACd,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC;gBAC3C,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACjB,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,IAAI,GAAG,EAAE,CAAC;wBAC5C,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;wBAC5C,OAAO;oBACT,CAAC;oBACD,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC;wBACtE,OAAO,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC9D,CAAC;oBAAC,MAAM,CAAC;wBACP,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC,CAAC;oBAC7C,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACxB,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACnB,GAAG,CAAC,GAAG,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AA1WD,gCA0WC;AACD,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E,IAAI,gBAAgB,GAAsB,IAAI,CAAC;AAE/C;;GAEG;AACH,SAAgB,WAAW,CAAC,MAAoB;IAC9C,OAAO,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;AAChC,CAAC;AAED;;GAEG;AACH,SAAgB,UAAU;IACxB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACtB,gBAAgB,GAAG,IAAI,UAAU,EAAE,CAAC;IACtC,CAAC;IACD,OAAO,gBAAgB,CAAC;AAC1B,CAAC"}
@@ -0,0 +1,96 @@
1
+ /**
2
+ * TMLPD Batch Processing Utilities
3
+ *
4
+ * Queue and process multiple prompts with:
5
+ * - Concurrency control
6
+ * - Priority scheduling
7
+ * - Progress callbacks
8
+ * - Rate limiting
9
+ */
10
+ export interface BatchItem {
11
+ id: string;
12
+ prompt: string;
13
+ model?: string;
14
+ priority: "high" | "normal" | "low";
15
+ callback?: (result: BatchResult) => void;
16
+ metadata?: Record<string, any>;
17
+ }
18
+ export interface BatchResult {
19
+ id: string;
20
+ success: boolean;
21
+ content?: string;
22
+ error?: string;
23
+ model: string;
24
+ cost: number;
25
+ duration_ms: number;
26
+ cached: boolean;
27
+ }
28
+ export interface BatchOptions {
29
+ concurrency?: number;
30
+ model?: string;
31
+ stop_on_error?: boolean;
32
+ rate_limit?: {
33
+ requests_per_minute?: number;
34
+ tokens_per_minute?: number;
35
+ };
36
+ }
37
+ export interface BatchProgress {
38
+ total: number;
39
+ completed: number;
40
+ failed: number;
41
+ in_progress: number;
42
+ total_cost: number;
43
+ }
44
+ export type ProgressCallback = (progress: BatchProgress, result?: BatchResult) => void;
45
+ export declare class BatchProcessor {
46
+ private queue;
47
+ private results;
48
+ private options;
49
+ private executing;
50
+ private progressCallbacks;
51
+ constructor(options?: BatchOptions);
52
+ /**
53
+ * Add item to batch queue.
54
+ */
55
+ add(item: Omit<BatchItem, "id">): string;
56
+ /**
57
+ * Add multiple items.
58
+ */
59
+ addBatch(items: Array<Omit<BatchItem, "id">>): string[];
60
+ /**
61
+ * Register progress callback.
62
+ */
63
+ onProgress(callback: ProgressCallback): void;
64
+ /**
65
+ * Get current progress.
66
+ */
67
+ getProgress(): BatchProgress;
68
+ /**
69
+ * Execute batch with concurrency control.
70
+ */
71
+ execute(executor: (item: BatchItem) => Promise<BatchResult>): Promise<BatchResult[]>;
72
+ /**
73
+ * Clear queue and results.
74
+ */
75
+ reset(): void;
76
+ /**
77
+ * Get queue size.
78
+ */
79
+ size(): number;
80
+ private sleep;
81
+ private waitForCompletion;
82
+ }
83
+ /**
84
+ * Helper function for simple batch execution.
85
+ */
86
+ export declare function executeBatch(items: Array<{
87
+ prompt: string;
88
+ model?: string;
89
+ priority?: "high" | "normal" | "low";
90
+ }>, executor: (prompt: string, model?: string) => Promise<BatchResult>, options?: BatchOptions): Promise<BatchResult[]>;
91
+ declare const _default: {
92
+ BatchProcessor: typeof BatchProcessor;
93
+ executeBatch: typeof executeBatch;
94
+ };
95
+ export default _default;
96
+ //# sourceMappingURL=batchProcessor.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"batchProcessor.d.ts","sourceRoot":"","sources":["../../src/utils/batchProcessor.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAIH,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,GAAG,QAAQ,GAAG,KAAK,CAAC;IACpC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,IAAI,CAAC;IACzC,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,YAAY;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,UAAU,CAAC,EAAE;QACX,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC5B,CAAC;CACH;AAED,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,MAAM,gBAAgB,GAAG,CAAC,QAAQ,EAAE,aAAa,EAAE,MAAM,CAAC,EAAE,WAAW,KAAK,IAAI,CAAC;AAEvF,qBAAa,cAAc;IACzB,OAAO,CAAC,KAAK,CAAmB;IAChC,OAAO,CAAC,OAAO,CAAqB;IACpC,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,SAAS,CAA0B;IAC3C,OAAO,CAAC,iBAAiB,CAA0B;gBAEvC,OAAO,GAAE,YAAiB;IAStC;;OAEG;IACH,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,GAAG,MAAM;IAUxC;;OAEG;IACH,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE;IAIvD;;OAEG;IACH,UAAU,CAAC,QAAQ,EAAE,gBAAgB,GAAG,IAAI;IAI5C;;OAEG;IACH,WAAW,IAAI,aAAa;IAU5B;;OAEG;IACG,OAAO,CACX,QAAQ,EAAE,CAAC,IAAI,EAAE,SAAS,KAAK,OAAO,CAAC,WAAW,CAAC,GAClD,OAAO,CAAC,WAAW,EAAE,CAAC;IAoEzB;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;OAEG;IACH,IAAI,IAAI,MAAM;IAId,OAAO,CAAC,KAAK;YAIC,iBAAiB;CAYhC;AAED;;GAEG;AACH,wBAAsB,YAAY,CAChC,KAAK,EAAE,KAAK,CAAC;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,QAAQ,GAAG,KAAK,CAAA;CAAE,CAAC,EACtF,QAAQ,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,WAAW,CAAC,EAClE,OAAO,GAAE,YAAiB,GACzB,OAAO,CAAC,WAAW,EAAE,CAAC,CAIxB;;;;;AAED,wBAGE"}
@@ -0,0 +1,170 @@
1
+ "use strict";
2
+ /**
3
+ * TMLPD Batch Processing Utilities
4
+ *
5
+ * Queue and process multiple prompts with:
6
+ * - Concurrency control
7
+ * - Priority scheduling
8
+ * - Progress callbacks
9
+ * - Rate limiting
10
+ */
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ exports.BatchProcessor = void 0;
13
+ exports.executeBatch = executeBatch;
14
+ const nanoid_1 = require("nanoid");
15
+ class BatchProcessor {
16
+ queue = [];
17
+ results = [];
18
+ options;
19
+ executing = new Set();
20
+ progressCallbacks = [];
21
+ constructor(options = {}) {
22
+ this.options = {
23
+ concurrency: options.concurrency || 5,
24
+ model: options.model || "gpt-4o",
25
+ stop_on_error: options.stop_on_error || false,
26
+ rate_limit: options.rate_limit || {}
27
+ };
28
+ }
29
+ /**
30
+ * Add item to batch queue.
31
+ */
32
+ add(item) {
33
+ const id = (0, nanoid_1.nanoid)(8);
34
+ this.queue.push({
35
+ ...item,
36
+ id,
37
+ priority: item.priority || "normal"
38
+ });
39
+ return id;
40
+ }
41
+ /**
42
+ * Add multiple items.
43
+ */
44
+ addBatch(items) {
45
+ return items.map(item => this.add(item));
46
+ }
47
+ /**
48
+ * Register progress callback.
49
+ */
50
+ onProgress(callback) {
51
+ this.progressCallbacks.push(callback);
52
+ }
53
+ /**
54
+ * Get current progress.
55
+ */
56
+ getProgress() {
57
+ return {
58
+ total: this.results.length + this.queue.length + this.executing.size,
59
+ completed: this.results.filter(r => r.success).length,
60
+ failed: this.results.filter(r => !r.success).length,
61
+ in_progress: this.executing.size,
62
+ total_cost: this.results.reduce((sum, r) => sum + r.cost, 0)
63
+ };
64
+ }
65
+ /**
66
+ * Execute batch with concurrency control.
67
+ */
68
+ async execute(executor) {
69
+ // Sort by priority
70
+ this.queue.sort((a, b) => {
71
+ const priorityOrder = { high: 0, normal: 1, low: 2 };
72
+ return priorityOrder[a.priority] - priorityOrder[b.priority];
73
+ });
74
+ const total = this.queue.length;
75
+ let index = 0;
76
+ // Process with concurrency limit
77
+ while (index < this.queue.length || this.executing.size > 0) {
78
+ // Launch up to concurrency limit
79
+ while (index < this.queue.length && this.executing.size < this.options.concurrency) {
80
+ const item = this.queue[index++];
81
+ this.executing.add(item.id);
82
+ // Execute with promise tracking
83
+ executor(item).then(result => {
84
+ this.results.push(result);
85
+ this.executing.delete(item.id);
86
+ // Notify progress
87
+ const progress = this.getProgress();
88
+ for (const cb of this.progressCallbacks) {
89
+ cb(progress, result);
90
+ }
91
+ }).catch(error => {
92
+ const result = {
93
+ id: item.id,
94
+ success: false,
95
+ error: error.message,
96
+ model: item.model || this.options.model,
97
+ cost: 0,
98
+ duration_ms: 0,
99
+ cached: false
100
+ };
101
+ this.results.push(result);
102
+ this.executing.delete(item.id);
103
+ const progress = this.getProgress();
104
+ for (const cb of this.progressCallbacks) {
105
+ cb(progress, result);
106
+ }
107
+ // Check stop_on_error
108
+ if (this.options.stop_on_error) {
109
+ // Cancel remaining items
110
+ this.queue = [];
111
+ }
112
+ });
113
+ // Rate limiting: wait between launches
114
+ if (this.options.rate_limit.requests_per_minute) {
115
+ const delay = 60000 / this.options.rate_limit.requests_per_minute;
116
+ await this.sleep(delay);
117
+ }
118
+ }
119
+ // Wait for at least one to complete
120
+ if (this.executing.size > 0) {
121
+ await this.waitForCompletion();
122
+ }
123
+ }
124
+ return this.results;
125
+ }
126
+ /**
127
+ * Clear queue and results.
128
+ */
129
+ reset() {
130
+ this.queue = [];
131
+ this.results = [];
132
+ this.executing.clear();
133
+ }
134
+ /**
135
+ * Get queue size.
136
+ */
137
+ size() {
138
+ return this.queue.length + this.executing.size + this.results.length;
139
+ }
140
+ sleep(ms) {
141
+ return new Promise(resolve => setTimeout(resolve, ms));
142
+ }
143
+ async waitForCompletion() {
144
+ await new Promise(resolve => {
145
+ const check = () => {
146
+ if (this.executing.size === 0) {
147
+ resolve(null);
148
+ }
149
+ else {
150
+ setTimeout(check, 50);
151
+ }
152
+ };
153
+ check();
154
+ });
155
+ }
156
+ }
157
+ exports.BatchProcessor = BatchProcessor;
158
+ /**
159
+ * Helper function for simple batch execution.
160
+ */
161
+ async function executeBatch(items, executor, options = {}) {
162
+ const processor = new BatchProcessor(options);
163
+ processor.addBatch(items.map(i => ({ ...i, priority: i.priority || "normal" })));
164
+ return processor.execute(async (item) => executor(item.prompt, item.model));
165
+ }
166
+ exports.default = {
167
+ BatchProcessor,
168
+ executeBatch
169
+ };
170
+ //# sourceMappingURL=batchProcessor.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"batchProcessor.js","sourceRoot":"","sources":["../../src/utils/batchProcessor.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AAkNH,oCAQC;AAxND,mCAAgC;AA0ChC,MAAa,cAAc;IACjB,KAAK,GAAgB,EAAE,CAAC;IACxB,OAAO,GAAkB,EAAE,CAAC;IAC5B,OAAO,CAAyB;IAChC,SAAS,GAAgB,IAAI,GAAG,EAAE,CAAC;IACnC,iBAAiB,GAAuB,EAAE,CAAC;IAEnD,YAAY,UAAwB,EAAE;QACpC,IAAI,CAAC,OAAO,GAAG;YACb,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,CAAC;YACrC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,QAAQ;YAChC,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,KAAK;YAC7C,UAAU,EAAE,OAAO,CAAC,UAAU,IAAI,EAAE;SACrC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAA2B;QAC7B,MAAM,EAAE,GAAG,IAAA,eAAM,EAAC,CAAC,CAAC,CAAC;QACrB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YACd,GAAG,IAAI;YACP,EAAE;YACF,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,QAAQ;SACpC,CAAC,CAAC;QACH,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,KAAmC;QAC1C,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,QAA0B;QACnC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI;YACpE,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM;YACrD,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM;YACnD,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI;YAChC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;SAC7D,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAmD;QAEnD,mBAAmB;QACnB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACvB,MAAM,aAAa,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC;YACrD,OAAO,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC;QAChC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,iCAAiC;QACjC,OAAO,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC5D,iCAAiC;YACjC,OAAO,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBACnF,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;gBACjC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAE5B,gCAAgC;gBAChC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE;oBAC3B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC1B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAE/B,kBAAkB;oBAClB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;oBACpC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;wBACxC,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACvB,CAAC;gBACH,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;oBACf,MAAM,MAAM,GAAgB;wBAC1B,EAAE,EAAE,IAAI,CAAC,EAAE;wBACX,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,KAAK,CAAC,OAAO;wBACpB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK;wBACvC,IAAI,EAAE,CAAC;wBACP,WAAW,EAAE,CAAC;wBACd,MAAM,EAAE,KAAK;qBACd,CAAC;oBACF,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC1B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAE/B,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;oBACpC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;wBACxC,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBACvB,CAAC;oBAED,sBAAsB;oBACtB,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;wBAC/B,yBAAyB;wBACzB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;oBAClB,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,uCAAuC;gBACvC,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,mBAAmB,EAAE,CAAC;oBAChD,MAAM,KAAK,GAAG,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC;oBAClE,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC1B,CAAC;YACH,CAAC;YAED,oCAAoC;YACpC,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACjC,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,IAAI;QACF,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;IACvE,CAAC;IAEO,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YAC1B,MAAM,KAAK,GAAG,GAAG,EAAE;gBACjB,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;oBAC9B,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC;qBAAM,CAAC;oBACN,UAAU,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC,CAAC;YACF,KAAK,EAAE,CAAC;QACV,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAjKD,wCAiKC;AAED;;GAEG;AACI,KAAK,UAAU,YAAY,CAChC,KAAsF,EACtF,QAAkE,EAClE,UAAwB,EAAE;IAE1B,MAAM,SAAS,GAAG,IAAI,cAAc,CAAC,OAAO,CAAC,CAAC;IAC9C,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;IACjF,OAAO,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;AAC9E,CAAC;AAED,kBAAe;IACb,cAAc;IACd,YAAY;CACb,CAAC"}