adaptive-memory-multi-model-router 1.2.2 → 1.3.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +146 -66
- package/dist/index.d.ts +1 -1
- package/dist/index.js +1 -1
- package/dist/integrations/airtable.js +20 -0
- package/dist/integrations/discord.js +18 -0
- package/dist/integrations/github.js +23 -0
- package/dist/integrations/gmail.js +19 -0
- package/dist/integrations/google-calendar.js +18 -0
- package/dist/integrations/index.js +61 -0
- package/dist/integrations/jira.js +21 -0
- package/dist/integrations/linear.js +19 -0
- package/dist/integrations/notion.js +19 -0
- package/dist/integrations/slack.js +18 -0
- package/dist/integrations/telegram.js +19 -0
- package/dist/providers/registry.js +7 -3
- package/docs/ARCHITECTURAL-IMPROVEMENTS-2025.md +1391 -0
- package/docs/ARCHITECTURAL-IMPROVEMENTS-REVISED-2025.md +1051 -0
- package/docs/CONFIGURATION.md +476 -0
- package/docs/COUNCIL_DECISION.json +308 -0
- package/docs/COUNCIL_SUMMARY.md +265 -0
- package/docs/COUNCIL_V2.2_DECISION.md +416 -0
- package/docs/IMPROVEMENT_ROADMAP.md +515 -0
- package/docs/LLM_COUNCIL_DECISION.md +508 -0
- package/docs/QUICK_START_VISIBILITY.md +782 -0
- package/docs/REDDIT_GAP_ANALYSIS.md +299 -0
- package/docs/RESEARCH_BACKED_IMPROVEMENTS.md +1180 -0
- package/docs/TMLPD_QNA.md +751 -0
- package/docs/TMLPD_V2.1_COMPLETE.md +763 -0
- package/docs/TMLPD_V2.2_RESEARCH_ROADMAP.md +754 -0
- package/docs/V2.2_IMPLEMENTATION_COMPLETE.md +446 -0
- package/docs/V2_IMPLEMENTATION_GUIDE.md +388 -0
- package/docs/VISIBILITY_ADOPTION_PLAN.md +1005 -0
- package/docs/launch-content/LAUNCH_EXECUTION_CHECKLIST.md +421 -0
- package/docs/launch-content/README.md +457 -0
- package/docs/launch-content/assets/cost_comparison_100_tasks.png +0 -0
- package/docs/launch-content/assets/cumulative_savings.png +0 -0
- package/docs/launch-content/assets/parallel_speedup.png +0 -0
- package/docs/launch-content/assets/provider_pricing_comparison.png +0 -0
- package/docs/launch-content/assets/task_breakdown_comparison.png +0 -0
- package/docs/launch-content/generate_charts.py +313 -0
- package/docs/launch-content/hn_show_post.md +139 -0
- package/docs/launch-content/partner_outreach_templates.md +745 -0
- package/docs/launch-content/reddit_posts.md +467 -0
- package/docs/launch-content/twitter_thread.txt +460 -0
- package/examples/QUICKSTART.md +1 -1
- package/openclaw-alexa-bridge/ALL_REMAINING_FIXES_PLAN.md +313 -0
- package/openclaw-alexa-bridge/REMAINING_FIXES_SUMMARY.md +277 -0
- package/openclaw-alexa-bridge/src/alexa_handler_no_tmlpd.js +1234 -0
- package/openclaw-alexa-bridge/test_fixes.js +77 -0
- package/package.json +120 -29
- package/package.json.tmp +0 -0
- package/qna/TMLPD_QNA.md +3 -3
- package/skill/SKILL.md +2 -2
- package/src/__tests__/integration/tmpld_integration.test.py +540 -0
- package/src/agents/skill_enhanced_agent.py +318 -0
- package/src/memory/__init__.py +15 -0
- package/src/memory/agentic_memory.py +353 -0
- package/src/memory/semantic_memory.py +444 -0
- package/src/memory/simple_memory.py +466 -0
- package/src/memory/working_memory.py +447 -0
- package/src/orchestration/__init__.py +52 -0
- package/src/orchestration/execution_engine.py +353 -0
- package/src/orchestration/halo_orchestrator.py +367 -0
- package/src/orchestration/mcts_workflow.py +498 -0
- package/src/orchestration/role_assigner.py +473 -0
- package/src/orchestration/task_planner.py +522 -0
- package/src/providers/__init__.py +67 -0
- package/src/providers/anthropic.py +304 -0
- package/src/providers/base.py +241 -0
- package/src/providers/cerebras.py +373 -0
- package/src/providers/registry.py +476 -0
- package/src/routing/__init__.py +30 -0
- package/src/routing/universal_router.py +621 -0
- package/src/skills/TMLPD-QUICKREF.md +210 -0
- package/src/skills/TMLPD-SETUP-SUMMARY.md +157 -0
- package/src/skills/TMLPD.md +540 -0
- package/src/skills/__tests__/skill_manager.test.ts +328 -0
- package/src/skills/skill_manager.py +385 -0
- package/src/skills/test-tmlpd.sh +108 -0
- package/src/skills/tmlpd-category.yaml +67 -0
- package/src/skills/tmlpd-monitoring.yaml +188 -0
- package/src/skills/tmlpd-phase.yaml +132 -0
- package/src/state/__init__.py +17 -0
- package/src/state/simple_checkpoint.py +508 -0
- package/src/tmlpd_agent.py +464 -0
- package/src/tmpld_v2.py +427 -0
- package/src/workflows/__init__.py +18 -0
- package/src/workflows/advanced_difficulty_classifier.py +377 -0
- package/src/workflows/chaining_executor.py +417 -0
- package/src/workflows/difficulty_integration.py +209 -0
- package/src/workflows/orchestrator.py +469 -0
- package/src/workflows/orchestrator_executor.py +456 -0
- package/src/workflows/parallelization_executor.py +382 -0
- package/src/workflows/router.py +311 -0
- package/test_integration_simple.py +86 -0
- package/test_mcts_workflow.py +150 -0
- package/test_templd_integration.py +262 -0
- package/test_universal_router.py +275 -0
- package/tmlpd-pi-extension/README.md +36 -0
- package/tmlpd-pi-extension/dist/cache/prefixCache.d.ts +114 -0
- package/tmlpd-pi-extension/dist/cache/prefixCache.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/cache/prefixCache.js +285 -0
- package/tmlpd-pi-extension/dist/cache/prefixCache.js.map +1 -0
- package/tmlpd-pi-extension/dist/cache/responseCache.d.ts +58 -0
- package/tmlpd-pi-extension/dist/cache/responseCache.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/cache/responseCache.js +153 -0
- package/tmlpd-pi-extension/dist/cache/responseCache.js.map +1 -0
- package/tmlpd-pi-extension/dist/cli.js +59 -0
- package/tmlpd-pi-extension/dist/cost/costTracker.d.ts +95 -0
- package/tmlpd-pi-extension/dist/cost/costTracker.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/cost/costTracker.js +240 -0
- package/tmlpd-pi-extension/dist/cost/costTracker.js.map +1 -0
- package/tmlpd-pi-extension/dist/index.d.ts +723 -0
- package/tmlpd-pi-extension/dist/index.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/index.js +239 -0
- package/tmlpd-pi-extension/dist/index.js.map +1 -0
- package/tmlpd-pi-extension/dist/memory/episodicMemory.d.ts +82 -0
- package/tmlpd-pi-extension/dist/memory/episodicMemory.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/memory/episodicMemory.js +145 -0
- package/tmlpd-pi-extension/dist/memory/episodicMemory.js.map +1 -0
- package/tmlpd-pi-extension/dist/orchestration/haloOrchestrator.d.ts +102 -0
- package/tmlpd-pi-extension/dist/orchestration/haloOrchestrator.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/orchestration/haloOrchestrator.js +207 -0
- package/tmlpd-pi-extension/dist/orchestration/haloOrchestrator.js.map +1 -0
- package/tmlpd-pi-extension/dist/orchestration/mctsWorkflow.d.ts +85 -0
- package/tmlpd-pi-extension/dist/orchestration/mctsWorkflow.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/orchestration/mctsWorkflow.js +210 -0
- package/tmlpd-pi-extension/dist/orchestration/mctsWorkflow.js.map +1 -0
- package/tmlpd-pi-extension/dist/providers/localProvider.d.ts +102 -0
- package/tmlpd-pi-extension/dist/providers/localProvider.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/providers/localProvider.js +338 -0
- package/tmlpd-pi-extension/dist/providers/localProvider.js.map +1 -0
- package/tmlpd-pi-extension/dist/providers/registry.d.ts +55 -0
- package/tmlpd-pi-extension/dist/providers/registry.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/providers/registry.js +138 -0
- package/tmlpd-pi-extension/dist/providers/registry.js.map +1 -0
- package/tmlpd-pi-extension/dist/routing/advancedRouter.d.ts +68 -0
- package/tmlpd-pi-extension/dist/routing/advancedRouter.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/routing/advancedRouter.js +332 -0
- package/tmlpd-pi-extension/dist/routing/advancedRouter.js.map +1 -0
- package/tmlpd-pi-extension/dist/tools/tmlpdTools.d.ts +101 -0
- package/tmlpd-pi-extension/dist/tools/tmlpdTools.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/tools/tmlpdTools.js +368 -0
- package/tmlpd-pi-extension/dist/tools/tmlpdTools.js.map +1 -0
- package/tmlpd-pi-extension/dist/utils/batchProcessor.d.ts +96 -0
- package/tmlpd-pi-extension/dist/utils/batchProcessor.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/utils/batchProcessor.js +170 -0
- package/tmlpd-pi-extension/dist/utils/batchProcessor.js.map +1 -0
- package/tmlpd-pi-extension/dist/utils/compression.d.ts +61 -0
- package/tmlpd-pi-extension/dist/utils/compression.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/utils/compression.js +281 -0
- package/tmlpd-pi-extension/dist/utils/compression.js.map +1 -0
- package/tmlpd-pi-extension/dist/utils/reliability.d.ts +74 -0
- package/tmlpd-pi-extension/dist/utils/reliability.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/utils/reliability.js +177 -0
- package/tmlpd-pi-extension/dist/utils/reliability.js.map +1 -0
- package/tmlpd-pi-extension/dist/utils/speculativeDecoding.d.ts +117 -0
- package/tmlpd-pi-extension/dist/utils/speculativeDecoding.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/utils/speculativeDecoding.js +246 -0
- package/tmlpd-pi-extension/dist/utils/speculativeDecoding.js.map +1 -0
- package/tmlpd-pi-extension/dist/utils/tokenUtils.d.ts +50 -0
- package/tmlpd-pi-extension/dist/utils/tokenUtils.d.ts.map +1 -0
- package/tmlpd-pi-extension/dist/utils/tokenUtils.js +124 -0
- package/tmlpd-pi-extension/dist/utils/tokenUtils.js.map +1 -0
- package/tmlpd-pi-extension/examples/QUICKSTART.md +183 -0
- package/tmlpd-pi-extension/package-lock.json +75 -0
- package/tmlpd-pi-extension/package.json +172 -0
- package/tmlpd-pi-extension/python/examples.py +53 -0
- package/tmlpd-pi-extension/python/integrations.py +330 -0
- package/tmlpd-pi-extension/python/setup.py +28 -0
- package/tmlpd-pi-extension/python/tmlpd.py +369 -0
- package/tmlpd-pi-extension/qna/REDDIT_GAP_ANALYSIS.md +299 -0
- package/tmlpd-pi-extension/qna/TMLPD_QNA.md +751 -0
- package/tmlpd-pi-extension/skill/SKILL.md +238 -0
- package/{src → tmlpd-pi-extension/src}/index.ts +1 -1
- package/tmlpd-pi-extension/tsconfig.json +18 -0
- package/demo/research-demo.js +0 -266
- package/notebooks/quickstart.ipynb +0 -157
- package/rust/tmlpd.h +0 -268
- package/src/cache/prefixCache.ts +0 -365
- package/src/routing/advancedRouter.ts +0 -406
- package/src/utils/speculativeDecoding.ts +0 -344
- /package/{src → tmlpd-pi-extension/src}/cache/responseCache.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/cost/costTracker.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/memory/episodicMemory.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/orchestration/haloOrchestrator.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/orchestration/mctsWorkflow.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/providers/localProvider.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/providers/registry.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/tools/tmlpdTools.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/utils/batchProcessor.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/utils/compression.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/utils/reliability.ts +0 -0
- /package/{src → tmlpd-pi-extension/src}/utils/tokenUtils.ts +0 -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"}
|