@crewx/sdk 0.8.0-rc.58
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 +201 -0
- package/README.md +1022 -0
- package/dist/adapters/MastraToolAdapter.d.ts +9 -0
- package/dist/adapters/MastraToolAdapter.js +66 -0
- package/dist/adapters/MastraToolAdapter.js.map +1 -0
- package/dist/api/index.d.ts +2 -0
- package/dist/api/index.js +8 -0
- package/dist/api/index.js.map +1 -0
- package/dist/boxing/box-storage.interface.d.ts +13 -0
- package/dist/boxing/box-storage.interface.js +3 -0
- package/dist/boxing/box-storage.interface.js.map +1 -0
- package/dist/boxing/box.service.d.ts +15 -0
- package/dist/boxing/box.service.js +70 -0
- package/dist/boxing/box.service.js.map +1 -0
- package/dist/boxing/box.types.d.ts +86 -0
- package/dist/boxing/box.types.js +3 -0
- package/dist/boxing/box.types.js.map +1 -0
- package/dist/boxing/context-builder.d.ts +8 -0
- package/dist/boxing/context-builder.js +76 -0
- package/dist/boxing/context-builder.js.map +1 -0
- package/dist/boxing/index.d.ts +6 -0
- package/dist/boxing/index.js +11 -0
- package/dist/boxing/index.js.map +1 -0
- package/dist/boxing/tokenizer.d.ts +3 -0
- package/dist/boxing/tokenizer.js +11 -0
- package/dist/boxing/tokenizer.js.map +1 -0
- package/dist/config/api-provider-parser.d.ts +58 -0
- package/dist/config/api-provider-parser.js +212 -0
- package/dist/config/api-provider-parser.js.map +1 -0
- package/dist/config/index.d.ts +3 -0
- package/dist/config/index.js +20 -0
- package/dist/config/index.js.map +1 -0
- package/dist/config/log.config.d.ts +7 -0
- package/dist/config/log.config.js +20 -0
- package/dist/config/log.config.js.map +1 -0
- package/dist/config/pricing.d.ts +10 -0
- package/dist/config/pricing.js +44 -0
- package/dist/config/pricing.js.map +1 -0
- package/dist/config/timeout.config.d.ts +14 -0
- package/dist/config/timeout.config.js +34 -0
- package/dist/config/timeout.config.js.map +1 -0
- package/dist/config/yaml-loader.d.ts +8 -0
- package/dist/config/yaml-loader.js +155 -0
- package/dist/config/yaml-loader.js.map +1 -0
- package/dist/constants/index.d.ts +4 -0
- package/dist/constants/index.js +8 -0
- package/dist/constants/index.js.map +1 -0
- package/dist/constants.d.ts +1 -0
- package/dist/constants.js +18 -0
- package/dist/constants.js.map +1 -0
- package/dist/conversation/conversation-config.d.ts +9 -0
- package/dist/conversation/conversation-config.js +22 -0
- package/dist/conversation/conversation-config.js.map +1 -0
- package/dist/conversation/conversation-history.interface.d.ts +36 -0
- package/dist/conversation/conversation-history.interface.js +3 -0
- package/dist/conversation/conversation-history.interface.js.map +1 -0
- package/dist/conversation/conversation-storage.service.d.ts +16 -0
- package/dist/conversation/conversation-storage.service.js +213 -0
- package/dist/conversation/conversation-storage.service.js.map +1 -0
- package/dist/conversation/index.d.ts +3 -0
- package/dist/conversation/index.js +20 -0
- package/dist/conversation/index.js.map +1 -0
- package/dist/core/agent/agent-factory.d.ts +37 -0
- package/dist/core/agent/agent-factory.js +68 -0
- package/dist/core/agent/agent-factory.js.map +1 -0
- package/dist/core/agent/agent-runtime.d.ts +52 -0
- package/dist/core/agent/agent-runtime.js +206 -0
- package/dist/core/agent/agent-runtime.js.map +1 -0
- package/dist/core/agent/event-bus.d.ts +44 -0
- package/dist/core/agent/event-bus.js +43 -0
- package/dist/core/agent/event-bus.js.map +1 -0
- package/dist/core/agent/index.d.ts +3 -0
- package/dist/core/agent/index.js +13 -0
- package/dist/core/agent/index.js.map +1 -0
- package/dist/core/env-defaults.d.ts +1 -0
- package/dist/core/env-defaults.js +7 -0
- package/dist/core/env-defaults.js.map +1 -0
- package/dist/core/parallel/helpers.d.ts +27 -0
- package/dist/core/parallel/helpers.js +252 -0
- package/dist/core/parallel/helpers.js.map +1 -0
- package/dist/core/parallel/index.d.ts +4 -0
- package/dist/core/parallel/index.js +11 -0
- package/dist/core/parallel/index.js.map +1 -0
- package/dist/core/parallel/parallel-runner.d.ts +16 -0
- package/dist/core/parallel/parallel-runner.js +230 -0
- package/dist/core/parallel/parallel-runner.js.map +1 -0
- package/dist/core/parallel/types.d.ts +41 -0
- package/dist/core/parallel/types.js +3 -0
- package/dist/core/parallel/types.js.map +1 -0
- package/dist/core/providers/MastraAPIProvider.d.ts +31 -0
- package/dist/core/providers/MastraAPIProvider.js +365 -0
- package/dist/core/providers/MastraAPIProvider.js.map +1 -0
- package/dist/core/providers/ai-provider.interface.d.ts +78 -0
- package/dist/core/providers/ai-provider.interface.js +23 -0
- package/dist/core/providers/ai-provider.interface.js.map +1 -0
- package/dist/core/providers/base-ai.provider.d.ts +80 -0
- package/dist/core/providers/base-ai.provider.js +1183 -0
- package/dist/core/providers/base-ai.provider.js.map +1 -0
- package/dist/core/providers/base-ai.types.d.ts +26 -0
- package/dist/core/providers/base-ai.types.js +3 -0
- package/dist/core/providers/base-ai.types.js.map +1 -0
- package/dist/core/providers/claude.provider.d.ts +19 -0
- package/dist/core/providers/claude.provider.js +170 -0
- package/dist/core/providers/claude.provider.js.map +1 -0
- package/dist/core/providers/codex.provider.d.ts +21 -0
- package/dist/core/providers/codex.provider.js +134 -0
- package/dist/core/providers/codex.provider.js.map +1 -0
- package/dist/core/providers/copilot.provider.d.ts +24 -0
- package/dist/core/providers/copilot.provider.js +146 -0
- package/dist/core/providers/copilot.provider.js.map +1 -0
- package/dist/core/providers/dynamic-provider.factory.d.ts +74 -0
- package/dist/core/providers/dynamic-provider.factory.js +643 -0
- package/dist/core/providers/dynamic-provider.factory.js.map +1 -0
- package/dist/core/providers/gemini.provider.d.ts +16 -0
- package/dist/core/providers/gemini.provider.js +101 -0
- package/dist/core/providers/gemini.provider.js.map +1 -0
- package/dist/core/providers/index.d.ts +8 -0
- package/dist/core/providers/index.js +20 -0
- package/dist/core/providers/index.js.map +1 -0
- package/dist/core/providers/mock.provider.d.ts +13 -0
- package/dist/core/providers/mock.provider.js +55 -0
- package/dist/core/providers/mock.provider.js.map +1 -0
- package/dist/core/providers/provider-factory.d.ts +3 -0
- package/dist/core/providers/provider-factory.js +65 -0
- package/dist/core/providers/provider-factory.js.map +1 -0
- package/dist/core/providers/tool-call.types.d.ts +39 -0
- package/dist/core/providers/tool-call.types.js +3 -0
- package/dist/core/providers/tool-call.types.js.map +1 -0
- package/dist/core/remote/index.d.ts +3 -0
- package/dist/core/remote/index.js +20 -0
- package/dist/core/remote/index.js.map +1 -0
- package/dist/core/remote/remote-agent-manager.d.ts +24 -0
- package/dist/core/remote/remote-agent-manager.js +195 -0
- package/dist/core/remote/remote-agent-manager.js.map +1 -0
- package/dist/core/remote/remote-transport.d.ts +15 -0
- package/dist/core/remote/remote-transport.js +70 -0
- package/dist/core/remote/remote-transport.js.map +1 -0
- package/dist/core/remote/types.d.ts +79 -0
- package/dist/core/remote/types.js +3 -0
- package/dist/core/remote/types.js.map +1 -0
- package/dist/index.d.ts +56 -0
- package/dist/index.js +133 -0
- package/dist/index.js.map +1 -0
- package/dist/internal/index.d.ts +1 -0
- package/dist/internal/index.js +6 -0
- package/dist/internal/index.js.map +1 -0
- package/dist/knowledge/DocumentManager.d.ts +4 -0
- package/dist/knowledge/DocumentManager.js +119 -0
- package/dist/knowledge/DocumentManager.js.map +1 -0
- package/dist/knowledge/index.d.ts +1 -0
- package/dist/knowledge/index.js +18 -0
- package/dist/knowledge/index.js.map +1 -0
- package/dist/parsers/agent-call.util.d.ts +3 -0
- package/dist/parsers/agent-call.util.js +17 -0
- package/dist/parsers/agent-call.util.js.map +1 -0
- package/dist/parsers/claude.parser.d.ts +2 -0
- package/dist/parsers/claude.parser.js +46 -0
- package/dist/parsers/claude.parser.js.map +1 -0
- package/dist/parsers/codex.parser.d.ts +2 -0
- package/dist/parsers/codex.parser.js +89 -0
- package/dist/parsers/codex.parser.js.map +1 -0
- package/dist/parsers/copilot.parser.d.ts +2 -0
- package/dist/parsers/copilot.parser.js +58 -0
- package/dist/parsers/copilot.parser.js.map +1 -0
- package/dist/parsers/gemini.parser.d.ts +2 -0
- package/dist/parsers/gemini.parser.js +36 -0
- package/dist/parsers/gemini.parser.js.map +1 -0
- package/dist/parsers/index.d.ts +7 -0
- package/dist/parsers/index.js +45 -0
- package/dist/parsers/index.js.map +1 -0
- package/dist/schema/skills-parser.d.ts +8 -0
- package/dist/schema/skills-parser.js +438 -0
- package/dist/schema/skills-parser.js.map +1 -0
- package/dist/schema/skills.types.d.ts +158 -0
- package/dist/schema/skills.types.js +41 -0
- package/dist/schema/skills.types.js.map +1 -0
- package/dist/schemas/api-provider.schema.d.ts +432 -0
- package/dist/schemas/api-provider.schema.js +50 -0
- package/dist/schemas/api-provider.schema.js.map +1 -0
- package/dist/services/index.d.ts +2 -0
- package/dist/services/index.js +19 -0
- package/dist/services/index.js.map +1 -0
- package/dist/services/layout-loader.service.d.ts +18 -0
- package/dist/services/layout-loader.service.js +271 -0
- package/dist/services/layout-loader.service.js.map +1 -0
- package/dist/services/layout-renderer.service.d.ts +34 -0
- package/dist/services/layout-renderer.service.js +325 -0
- package/dist/services/layout-renderer.service.js.map +1 -0
- package/dist/services/props-validator.service.d.ts +29 -0
- package/dist/services/props-validator.service.js +332 -0
- package/dist/services/props-validator.service.js.map +1 -0
- package/dist/skills/adapter/claude-skill-adapter.d.ts +11 -0
- package/dist/skills/adapter/claude-skill-adapter.js +222 -0
- package/dist/skills/adapter/claude-skill-adapter.js.map +1 -0
- package/dist/skills/index.d.ts +6 -0
- package/dist/skills/index.js +31 -0
- package/dist/skills/index.js.map +1 -0
- package/dist/skills/runtime/progressive-loader.d.ts +27 -0
- package/dist/skills/runtime/progressive-loader.js +186 -0
- package/dist/skills/runtime/progressive-loader.js.map +1 -0
- package/dist/skills/runtime/runtime-requirements-validator.d.ts +23 -0
- package/dist/skills/runtime/runtime-requirements-validator.js +248 -0
- package/dist/skills/runtime/runtime-requirements-validator.js.map +1 -0
- package/dist/skills/runtime/skill-runtime.service.d.ts +42 -0
- package/dist/skills/runtime/skill-runtime.service.js +434 -0
- package/dist/skills/runtime/skill-runtime.service.js.map +1 -0
- package/dist/tools/file-system.service.d.ts +10 -0
- package/dist/tools/file-system.service.js +33 -0
- package/dist/tools/file-system.service.js.map +1 -0
- package/dist/tools/find.tool.d.ts +85 -0
- package/dist/tools/find.tool.js +139 -0
- package/dist/tools/find.tool.js.map +1 -0
- package/dist/tools/glob.tool.d.ts +100 -0
- package/dist/tools/glob.tool.js +153 -0
- package/dist/tools/glob.tool.js.map +1 -0
- package/dist/tools/grep.tool.d.ts +30 -0
- package/dist/tools/grep.tool.js +137 -0
- package/dist/tools/grep.tool.js.map +1 -0
- package/dist/tools/index.d.ts +12 -0
- package/dist/tools/index.js +40 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/ls.tool.d.ts +24 -0
- package/dist/tools/ls.tool.js +94 -0
- package/dist/tools/ls.tool.js.map +1 -0
- package/dist/tools/read-file.tool.d.ts +30 -0
- package/dist/tools/read-file.tool.js +69 -0
- package/dist/tools/read-file.tool.js.map +1 -0
- package/dist/tools/replace.tool.d.ts +36 -0
- package/dist/tools/replace.tool.js +68 -0
- package/dist/tools/replace.tool.js.map +1 -0
- package/dist/tools/run-shell-command.tool.d.ts +24 -0
- package/dist/tools/run-shell-command.tool.js +64 -0
- package/dist/tools/run-shell-command.tool.js.map +1 -0
- package/dist/tools/tree.tool.d.ts +24 -0
- package/dist/tools/tree.tool.js +109 -0
- package/dist/tools/tree.tool.js.map +1 -0
- package/dist/tools/types.d.ts +42 -0
- package/dist/tools/types.js +13 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/tools/utils/file-utils.d.ts +5 -0
- package/dist/tools/utils/file-utils.js +221 -0
- package/dist/tools/utils/file-utils.js.map +1 -0
- package/dist/tools/write-file.tool.d.ts +24 -0
- package/dist/tools/write-file.tool.js +55 -0
- package/dist/tools/write-file.tool.js.map +1 -0
- package/dist/types/agent.types.d.ts +134 -0
- package/dist/types/agent.types.js +16 -0
- package/dist/types/agent.types.js.map +1 -0
- package/dist/types/api-provider.types.d.ts +85 -0
- package/dist/types/api-provider.types.js +65 -0
- package/dist/types/api-provider.types.js.map +1 -0
- package/dist/types/index.d.ts +6 -0
- package/dist/types/index.js +23 -0
- package/dist/types/index.js.map +1 -0
- package/dist/types/layout.types.d.ts +115 -0
- package/dist/types/layout.types.js +22 -0
- package/dist/types/layout.types.js.map +1 -0
- package/dist/types/provider.types.d.ts +12 -0
- package/dist/types/provider.types.js +3 -0
- package/dist/types/provider.types.js.map +1 -0
- package/dist/types/skill-runtime.types.d.ts +244 -0
- package/dist/types/skill-runtime.types.js +44 -0
- package/dist/types/skill-runtime.types.js.map +1 -0
- package/dist/types/structured-payload.types.d.ts +46 -0
- package/dist/types/structured-payload.types.js +65 -0
- package/dist/types/structured-payload.types.js.map +1 -0
- package/dist/types/task-log.types.d.ts +14 -0
- package/dist/types/task-log.types.js +3 -0
- package/dist/types/task-log.types.js.map +1 -0
- package/dist/types/template.types.d.ts +38 -0
- package/dist/types/template.types.js +3 -0
- package/dist/types/template.types.js.map +1 -0
- package/dist/types.d.ts +1 -0
- package/dist/types.js +18 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/api-provider-normalizer.d.ts +16 -0
- package/dist/utils/api-provider-normalizer.js +135 -0
- package/dist/utils/api-provider-normalizer.js.map +1 -0
- package/dist/utils/base-message-formatter.d.ts +32 -0
- package/dist/utils/base-message-formatter.js +170 -0
- package/dist/utils/base-message-formatter.js.map +1 -0
- package/dist/utils/error-utils.d.ts +3 -0
- package/dist/utils/error-utils.js +27 -0
- package/dist/utils/error-utils.js.map +1 -0
- package/dist/utils/index.d.ts +4 -0
- package/dist/utils/index.js +21 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/math-utils.d.ts +3 -0
- package/dist/utils/math-utils.js +10 -0
- package/dist/utils/math-utils.js.map +1 -0
- package/dist/utils/mention-parser.d.ts +18 -0
- package/dist/utils/mention-parser.js +136 -0
- package/dist/utils/mention-parser.js.map +1 -0
- package/dist/utils/string-utils.d.ts +1 -0
- package/dist/utils/string-utils.js +10 -0
- package/dist/utils/string-utils.js.map +1 -0
- package/dist/utils.d.ts +3 -0
- package/dist/utils.js +20 -0
- package/dist/utils.js.map +1 -0
- package/package.json +134 -0
- package/schema/api-provider-config.json +138 -0
- package/schema/crewx-config.json +224 -0
- package/schema/skills-config.json +306 -0
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.runExecutesParallel = exports.runQueriesParallel = exports.createDefaultParallelRunner = exports.ParallelRunnerTimeoutError = exports.ParallelRunner = void 0;
|
|
4
|
+
var parallel_runner_1 = require("./parallel-runner");
|
|
5
|
+
Object.defineProperty(exports, "ParallelRunner", { enumerable: true, get: function () { return parallel_runner_1.ParallelRunner; } });
|
|
6
|
+
Object.defineProperty(exports, "ParallelRunnerTimeoutError", { enumerable: true, get: function () { return parallel_runner_1.ParallelRunnerTimeoutError; } });
|
|
7
|
+
Object.defineProperty(exports, "createDefaultParallelRunner", { enumerable: true, get: function () { return parallel_runner_1.createDefaultParallelRunner; } });
|
|
8
|
+
var helpers_1 = require("./helpers");
|
|
9
|
+
Object.defineProperty(exports, "runQueriesParallel", { enumerable: true, get: function () { return helpers_1.runQueriesParallel; } });
|
|
10
|
+
Object.defineProperty(exports, "runExecutesParallel", { enumerable: true, get: function () { return helpers_1.runExecutesParallel; } });
|
|
11
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/parallel/index.ts"],"names":[],"mappings":";;;AAAA,qDAI2B;AAHzB,iHAAA,cAAc,OAAA;AACd,6HAAA,0BAA0B,OAAA;AAC1B,8HAAA,2BAA2B,OAAA;AAU7B,qCAGmB;AAFjB,6GAAA,kBAAkB,OAAA;AAClB,8GAAA,mBAAmB,OAAA"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { ParallelRunnerMetrics, ParallelRunnerOptions, Task, TaskResult } from './types';
|
|
2
|
+
export declare class ParallelRunnerTimeoutError extends Error {
|
|
3
|
+
readonly taskId: string;
|
|
4
|
+
readonly timeoutMs: number;
|
|
5
|
+
constructor(taskId: string, timeoutMs: number);
|
|
6
|
+
}
|
|
7
|
+
export declare class ParallelRunner {
|
|
8
|
+
private readonly defaults;
|
|
9
|
+
private metrics;
|
|
10
|
+
constructor(defaults?: ParallelRunnerOptions);
|
|
11
|
+
run<T>(tasks: Task<T>[], options?: ParallelRunnerOptions<T>): Promise<TaskResult<T>[]>;
|
|
12
|
+
getMetrics(): ParallelRunnerMetrics;
|
|
13
|
+
private mergeOptions;
|
|
14
|
+
private mergeCallbacks;
|
|
15
|
+
}
|
|
16
|
+
export declare const createDefaultParallelRunner: () => ParallelRunner;
|
|
@@ -0,0 +1,230 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.createDefaultParallelRunner = exports.ParallelRunner = exports.ParallelRunnerTimeoutError = void 0;
|
|
4
|
+
const node_perf_hooks_1 = require("node:perf_hooks");
|
|
5
|
+
class ParallelRunnerTimeoutError extends Error {
|
|
6
|
+
constructor(taskId, timeoutMs) {
|
|
7
|
+
super(`Task ${taskId} timed out after ${timeoutMs}ms`);
|
|
8
|
+
this.taskId = taskId;
|
|
9
|
+
this.timeoutMs = timeoutMs;
|
|
10
|
+
this.name = 'ParallelRunnerTimeoutError';
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
exports.ParallelRunnerTimeoutError = ParallelRunnerTimeoutError;
|
|
14
|
+
class ParallelRunner {
|
|
15
|
+
constructor(defaults = {}) {
|
|
16
|
+
this.defaults = defaults;
|
|
17
|
+
this.metrics = {
|
|
18
|
+
totalTasks: 0,
|
|
19
|
+
startedTasks: 0,
|
|
20
|
+
completedTasks: 0,
|
|
21
|
+
successCount: 0,
|
|
22
|
+
failureCount: 0,
|
|
23
|
+
totalDurationMs: 0,
|
|
24
|
+
averageDurationMs: 0,
|
|
25
|
+
throughput: 0,
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
async run(tasks, options = {}) {
|
|
29
|
+
if (!Array.isArray(tasks)) {
|
|
30
|
+
throw new TypeError('ParallelRunner.run expects an array of tasks');
|
|
31
|
+
}
|
|
32
|
+
if (tasks.length === 0) {
|
|
33
|
+
this.metrics = {
|
|
34
|
+
totalTasks: 0,
|
|
35
|
+
startedTasks: 0,
|
|
36
|
+
completedTasks: 0,
|
|
37
|
+
successCount: 0,
|
|
38
|
+
failureCount: 0,
|
|
39
|
+
totalDurationMs: 0,
|
|
40
|
+
averageDurationMs: 0,
|
|
41
|
+
throughput: 0,
|
|
42
|
+
};
|
|
43
|
+
return [];
|
|
44
|
+
}
|
|
45
|
+
const merged = this.mergeOptions(options);
|
|
46
|
+
const maxConcurrency = Math.max(1, merged.maxConcurrency ?? 5);
|
|
47
|
+
const timeoutMs = merged.timeoutMs;
|
|
48
|
+
const failFast = merged.failFast ?? false;
|
|
49
|
+
const callbacks = merged.callbacks;
|
|
50
|
+
const evaluateTaskSuccess = merged.evaluateTaskSuccess ?? (() => true);
|
|
51
|
+
const results = [];
|
|
52
|
+
let successCount = 0;
|
|
53
|
+
let failureCount = 0;
|
|
54
|
+
let startedTasks = 0;
|
|
55
|
+
let aborted = false;
|
|
56
|
+
let cursor = 0;
|
|
57
|
+
const startedAt = node_perf_hooks_1.performance.now();
|
|
58
|
+
const executeTask = async (task) => {
|
|
59
|
+
if (callbacks?.onTaskStart) {
|
|
60
|
+
await callbacks.onTaskStart(task);
|
|
61
|
+
}
|
|
62
|
+
startedTasks += 1;
|
|
63
|
+
const controller = new AbortController();
|
|
64
|
+
const context = { signal: controller.signal };
|
|
65
|
+
let timeoutId;
|
|
66
|
+
let timeoutPromise;
|
|
67
|
+
if (typeof timeoutMs === 'number' && Number.isFinite(timeoutMs) && timeoutMs > 0) {
|
|
68
|
+
timeoutPromise = new Promise((_, reject) => {
|
|
69
|
+
timeoutId = setTimeout(() => {
|
|
70
|
+
const timeoutError = new ParallelRunnerTimeoutError(task.id, timeoutMs);
|
|
71
|
+
controller.abort(timeoutError);
|
|
72
|
+
reject(timeoutError);
|
|
73
|
+
}, timeoutMs);
|
|
74
|
+
});
|
|
75
|
+
}
|
|
76
|
+
const startedAtTask = node_perf_hooks_1.performance.now();
|
|
77
|
+
const runPromise = task.run(context);
|
|
78
|
+
const combinedPromise = timeoutPromise
|
|
79
|
+
? Promise.race([runPromise, timeoutPromise])
|
|
80
|
+
: runPromise;
|
|
81
|
+
try {
|
|
82
|
+
const value = await combinedPromise;
|
|
83
|
+
const finishedAtTask = node_perf_hooks_1.performance.now();
|
|
84
|
+
const durationMs = finishedAtTask - startedAtTask;
|
|
85
|
+
const success = evaluateTaskSuccess(value, task);
|
|
86
|
+
const result = {
|
|
87
|
+
taskId: task.id,
|
|
88
|
+
success,
|
|
89
|
+
value,
|
|
90
|
+
durationMs,
|
|
91
|
+
startedAt: startedAtTask,
|
|
92
|
+
finishedAt: finishedAtTask,
|
|
93
|
+
metadata: task.metadata,
|
|
94
|
+
aborted: false,
|
|
95
|
+
};
|
|
96
|
+
results.push(result);
|
|
97
|
+
success ? (successCount += 1) : (failureCount += 1);
|
|
98
|
+
if (callbacks?.onTaskComplete) {
|
|
99
|
+
await callbacks.onTaskComplete(result);
|
|
100
|
+
}
|
|
101
|
+
if (failFast && !success) {
|
|
102
|
+
aborted = true;
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
catch (rawError) {
|
|
106
|
+
const finishedAtTask = node_perf_hooks_1.performance.now();
|
|
107
|
+
const durationMs = finishedAtTask - startedAtTask;
|
|
108
|
+
const error = rawError instanceof Error ? rawError : new Error(String(rawError));
|
|
109
|
+
const result = {
|
|
110
|
+
taskId: task.id,
|
|
111
|
+
success: false,
|
|
112
|
+
error,
|
|
113
|
+
durationMs,
|
|
114
|
+
startedAt: startedAtTask,
|
|
115
|
+
finishedAt: finishedAtTask,
|
|
116
|
+
metadata: task.metadata,
|
|
117
|
+
aborted: controller.signal.aborted,
|
|
118
|
+
};
|
|
119
|
+
results.push(result);
|
|
120
|
+
failureCount += 1;
|
|
121
|
+
if (callbacks?.onError) {
|
|
122
|
+
await callbacks.onError(task, error, durationMs);
|
|
123
|
+
}
|
|
124
|
+
if (failFast) {
|
|
125
|
+
aborted = true;
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
finally {
|
|
129
|
+
if (timeoutId) {
|
|
130
|
+
clearTimeout(timeoutId);
|
|
131
|
+
}
|
|
132
|
+
if (timeoutPromise) {
|
|
133
|
+
runPromise.catch(() => undefined);
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
};
|
|
137
|
+
const worker = async () => {
|
|
138
|
+
while (true) {
|
|
139
|
+
if (aborted) {
|
|
140
|
+
break;
|
|
141
|
+
}
|
|
142
|
+
const currentIndex = cursor;
|
|
143
|
+
if (currentIndex >= tasks.length) {
|
|
144
|
+
break;
|
|
145
|
+
}
|
|
146
|
+
cursor += 1;
|
|
147
|
+
const task = tasks[currentIndex];
|
|
148
|
+
if (!task) {
|
|
149
|
+
break;
|
|
150
|
+
}
|
|
151
|
+
await executeTask(task);
|
|
152
|
+
if (aborted) {
|
|
153
|
+
break;
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
};
|
|
157
|
+
const workers = [];
|
|
158
|
+
const workerCount = Math.min(maxConcurrency, tasks.length);
|
|
159
|
+
for (let i = 0; i < workerCount; i += 1) {
|
|
160
|
+
workers.push(worker());
|
|
161
|
+
}
|
|
162
|
+
await Promise.all(workers);
|
|
163
|
+
const finishedAt = node_perf_hooks_1.performance.now();
|
|
164
|
+
const totalDurationMs = finishedAt - startedAt;
|
|
165
|
+
const averageDurationMs = results.length
|
|
166
|
+
? results.reduce((sum, item) => sum + item.durationMs, 0) / results.length
|
|
167
|
+
: 0;
|
|
168
|
+
const throughput = totalDurationMs > 0
|
|
169
|
+
? (results.length / (totalDurationMs / 1000))
|
|
170
|
+
: results.length;
|
|
171
|
+
this.metrics = {
|
|
172
|
+
totalTasks: tasks.length,
|
|
173
|
+
startedTasks,
|
|
174
|
+
completedTasks: results.length,
|
|
175
|
+
successCount,
|
|
176
|
+
failureCount,
|
|
177
|
+
totalDurationMs,
|
|
178
|
+
averageDurationMs,
|
|
179
|
+
throughput,
|
|
180
|
+
};
|
|
181
|
+
return results;
|
|
182
|
+
}
|
|
183
|
+
getMetrics() {
|
|
184
|
+
return this.metrics;
|
|
185
|
+
}
|
|
186
|
+
mergeOptions(options) {
|
|
187
|
+
return {
|
|
188
|
+
maxConcurrency: options.maxConcurrency ?? this.defaults.maxConcurrency,
|
|
189
|
+
timeoutMs: options.timeoutMs ?? this.defaults.timeoutMs,
|
|
190
|
+
failFast: options.failFast ?? this.defaults.failFast,
|
|
191
|
+
evaluateTaskSuccess: options.evaluateTaskSuccess ?? this.defaults.evaluateTaskSuccess,
|
|
192
|
+
callbacks: this.mergeCallbacks(this.defaults.callbacks, options.callbacks),
|
|
193
|
+
};
|
|
194
|
+
}
|
|
195
|
+
mergeCallbacks(base, override) {
|
|
196
|
+
if (!base && !override) {
|
|
197
|
+
return undefined;
|
|
198
|
+
}
|
|
199
|
+
return {
|
|
200
|
+
onTaskStart: async (task) => {
|
|
201
|
+
if (base?.onTaskStart) {
|
|
202
|
+
await base.onTaskStart(task);
|
|
203
|
+
}
|
|
204
|
+
if (override?.onTaskStart) {
|
|
205
|
+
await override.onTaskStart(task);
|
|
206
|
+
}
|
|
207
|
+
},
|
|
208
|
+
onTaskComplete: async (result) => {
|
|
209
|
+
if (base?.onTaskComplete) {
|
|
210
|
+
await base.onTaskComplete(result);
|
|
211
|
+
}
|
|
212
|
+
if (override?.onTaskComplete) {
|
|
213
|
+
await override.onTaskComplete(result);
|
|
214
|
+
}
|
|
215
|
+
},
|
|
216
|
+
onError: async (task, error, durationMs) => {
|
|
217
|
+
if (base?.onError) {
|
|
218
|
+
await base.onError(task, error, durationMs);
|
|
219
|
+
}
|
|
220
|
+
if (override?.onError) {
|
|
221
|
+
await override.onError(task, error, durationMs);
|
|
222
|
+
}
|
|
223
|
+
},
|
|
224
|
+
};
|
|
225
|
+
}
|
|
226
|
+
}
|
|
227
|
+
exports.ParallelRunner = ParallelRunner;
|
|
228
|
+
const createDefaultParallelRunner = () => new ParallelRunner();
|
|
229
|
+
exports.createDefaultParallelRunner = createDefaultParallelRunner;
|
|
230
|
+
//# sourceMappingURL=parallel-runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"parallel-runner.js","sourceRoot":"","sources":["../../../src/core/parallel/parallel-runner.ts"],"names":[],"mappings":";;;AAAA,qDAA8C;AAc9C,MAAa,0BAA2B,SAAQ,KAAK;IACnD,YAA4B,MAAc,EAAkB,SAAiB;QAC3E,KAAK,CAAC,QAAQ,MAAM,oBAAoB,SAAS,IAAI,CAAC,CAAC;QAD7B,WAAM,GAAN,MAAM,CAAQ;QAAkB,cAAS,GAAT,SAAS,CAAQ;QAE3E,IAAI,CAAC,IAAI,GAAG,4BAA4B,CAAC;IAC3C,CAAC;CACF;AALD,gEAKC;AAOD,MAAa,cAAc;IAYzB,YAA6B,WAAkC,EAAE;QAApC,aAAQ,GAAR,QAAQ,CAA4B;QAXzD,YAAO,GAA0B;YACvC,UAAU,EAAE,CAAC;YACb,YAAY,EAAE,CAAC;YACf,cAAc,EAAE,CAAC;YACjB,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;YACf,eAAe,EAAE,CAAC;YAClB,iBAAiB,EAAE,CAAC;YACpB,UAAU,EAAE,CAAC;SACd,CAAC;IAEkE,CAAC;IAErE,KAAK,CAAC,GAAG,CAAI,KAAgB,EAAE,UAAoC,EAAE;QACnE,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,SAAS,CAAC,8CAA8C,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,OAAO,GAAG;gBACb,UAAU,EAAE,CAAC;gBACb,YAAY,EAAE,CAAC;gBACf,cAAc,EAAE,CAAC;gBACjB,YAAY,EAAE,CAAC;gBACf,YAAY,EAAE,CAAC;gBACf,eAAe,EAAE,CAAC;gBAClB,iBAAiB,EAAE,CAAC;gBACpB,UAAU,EAAE,CAAC;aACd,CAAC;YACF,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,cAAc,IAAI,CAAC,CAAC,CAAC;QAC/D,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACnC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC;QAC1C,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QACnC,MAAM,mBAAmB,GAAG,MAAM,CAAC,mBAAmB,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;QAEvE,MAAM,OAAO,GAAoB,EAAE,CAAC;QACpC,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,MAAM,GAAG,CAAC,CAAC;QAEf,MAAM,SAAS,GAAG,6BAAW,CAAC,GAAG,EAAE,CAAC;QAEpC,MAAM,WAAW,GAAG,KAAK,EAAE,IAAa,EAAiB,EAAE;YACzD,IAAI,SAAS,EAAE,WAAW,EAAE,CAAC;gBAC3B,MAAM,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC,CAAC;YAED,YAAY,IAAI,CAAC,CAAC;YAClB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;YACzC,MAAM,OAAO,GAAyB,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC;YAEpE,IAAI,SAAqC,CAAC;YAC1C,IAAI,cAA0C,CAAC;YAE/C,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBACjF,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;oBAChD,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;wBAC1B,MAAM,YAAY,GAAG,IAAI,0BAA0B,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;wBACxE,UAAU,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;wBAC/B,MAAM,CAAC,YAAY,CAAC,CAAC;oBACvB,CAAC,EAAE,SAAS,CAAC,CAAC;gBAChB,CAAC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,aAAa,GAAG,6BAAW,CAAC,GAAG,EAAE,CAAC;YACxC,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACrC,MAAM,eAAe,GAAG,cAAc;gBACpC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;gBAC5C,CAAC,CAAC,UAAU,CAAC;YAEf,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC;gBACpC,MAAM,cAAc,GAAG,6BAAW,CAAC,GAAG,EAAE,CAAC;gBACzC,MAAM,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;gBAClD,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAEjD,MAAM,MAAM,GAAkB;oBAC5B,MAAM,EAAE,IAAI,CAAC,EAAE;oBACf,OAAO;oBACP,KAAK;oBACL,UAAU;oBACV,SAAS,EAAE,aAAa;oBACxB,UAAU,EAAE,cAAc;oBAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrB,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC;gBAEpD,IAAI,SAAS,EAAE,cAAc,EAAE,CAAC;oBAC9B,MAAM,SAAS,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBACzC,CAAC;gBAED,IAAI,QAAQ,IAAI,CAAC,OAAO,EAAE,CAAC;oBACzB,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;YACH,CAAC;YAAC,OAAO,QAAQ,EAAE,CAAC;gBAClB,MAAM,cAAc,GAAG,6BAAW,CAAC,GAAG,EAAE,CAAC;gBACzC,MAAM,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;gBAClD,MAAM,KAAK,GAAG,QAAQ,YAAY,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;gBAEjF,MAAM,MAAM,GAAkB;oBAC5B,MAAM,EAAE,IAAI,CAAC,EAAE;oBACf,OAAO,EAAE,KAAK;oBACd,KAAK;oBACL,UAAU;oBACV,SAAS,EAAE,aAAa;oBACxB,UAAU,EAAE,cAAc;oBAC1B,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,OAAO,EAAE,UAAU,CAAC,MAAM,CAAC,OAAO;iBACnC,CAAC;gBAEF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACrB,YAAY,IAAI,CAAC,CAAC;gBAElB,IAAI,SAAS,EAAE,OAAO,EAAE,CAAC;oBACvB,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;gBACnD,CAAC;gBAED,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,GAAG,IAAI,CAAC;gBACjB,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,IAAI,SAAS,EAAE,CAAC;oBACd,YAAY,CAAC,SAAS,CAAC,CAAC;gBAC1B,CAAC;gBAGD,IAAI,cAAc,EAAE,CAAC;oBACnB,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,MAAM,GAAG,KAAK,IAAmB,EAAE;YACvC,OAAO,IAAI,EAAE,CAAC;gBACZ,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM;gBACR,CAAC;gBAED,MAAM,YAAY,GAAG,MAAM,CAAC;gBAC5B,IAAI,YAAY,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;oBACjC,MAAM;gBACR,CAAC;gBACD,MAAM,IAAI,CAAC,CAAC;gBAEZ,MAAM,IAAI,GAAG,KAAK,CAAC,YAAY,CAAC,CAAC;gBACjC,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,MAAM;gBACR,CAAC;gBAED,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC;gBAExB,IAAI,OAAO,EAAE,CAAC;oBACZ,MAAM;gBACR,CAAC;YACH,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,OAAO,GAAoB,EAAE,CAAC;QACpC,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;QAC3D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;YACxC,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACzB,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAE3B,MAAM,UAAU,GAAG,6BAAW,CAAC,GAAG,EAAE,CAAC;QACrC,MAAM,eAAe,GAAG,UAAU,GAAG,SAAS,CAAC;QAC/C,MAAM,iBAAiB,GAAG,OAAO,CAAC,MAAM;YACtC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC,MAAM;YAC1E,CAAC,CAAC,CAAC,CAAC;QACN,MAAM,UAAU,GAAG,eAAe,GAAG,CAAC;YACpC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC,CAAC;YAC7C,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;QAEnB,IAAI,CAAC,OAAO,GAAG;YACb,UAAU,EAAE,KAAK,CAAC,MAAM;YACxB,YAAY;YACZ,cAAc,EAAE,OAAO,CAAC,MAAM;YAC9B,YAAY;YACZ,YAAY;YACZ,eAAe;YACf,iBAAiB;YACjB,UAAU;SACX,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAEO,YAAY,CAAI,OAAiC;QACvD,OAAO;YACL,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc;YACtE,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS;YACvD,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ;YACpD,mBAAmB,EAAE,OAAO,CAAC,mBAAmB,IAAI,IAAI,CAAC,QAAQ,CAAC,mBAAmB;YACrF,SAAS,EAAE,IAAI,CAAC,cAAc,CAC5B,IAAI,CAAC,QAAQ,CAAC,SAAyC,EACvD,OAAO,CAAC,SAAS,CAClB;SACF,CAAC;IACJ,CAAC;IAEO,cAAc,CACpB,IAAuB,EACvB,QAA2B;QAE3B,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACvB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,WAAW,EAAE,KAAK,EAAE,IAAa,EAAE,EAAE;gBACnC,IAAI,IAAI,EAAE,WAAW,EAAE,CAAC;oBACtB,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBAC/B,CAAC;gBACD,IAAI,QAAQ,EAAE,WAAW,EAAE,CAAC;oBAC1B,MAAM,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;YACD,cAAc,EAAE,KAAK,EAAE,MAAqB,EAAE,EAAE;gBAC9C,IAAI,IAAI,EAAE,cAAc,EAAE,CAAC;oBACzB,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBACpC,CAAC;gBACD,IAAI,QAAQ,EAAE,cAAc,EAAE,CAAC;oBAC7B,MAAM,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBACxC,CAAC;YACH,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,IAAa,EAAE,KAAY,EAAE,UAAkB,EAAE,EAAE;gBACjE,IAAI,IAAI,EAAE,OAAO,EAAE,CAAC;oBAClB,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;gBAC9C,CAAC;gBACD,IAAI,QAAQ,EAAE,OAAO,EAAE,CAAC;oBACtB,MAAM,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;SACF,CAAC;IACJ,CAAC;CACF;AA1PD,wCA0PC;AAKM,MAAM,2BAA2B,GAAG,GAAmB,EAAE,CAAC,IAAI,cAAc,EAAE,CAAC;AAAzE,QAAA,2BAA2B,+BAA8C"}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
export interface Task<T> {
|
|
2
|
+
id: string;
|
|
3
|
+
metadata?: Record<string, unknown>;
|
|
4
|
+
run: (context: TaskExecutionContext) => Promise<T>;
|
|
5
|
+
}
|
|
6
|
+
export interface TaskExecutionContext {
|
|
7
|
+
signal: AbortSignal;
|
|
8
|
+
}
|
|
9
|
+
export interface TaskResult<T> {
|
|
10
|
+
taskId: string;
|
|
11
|
+
success: boolean;
|
|
12
|
+
value?: T;
|
|
13
|
+
error?: Error;
|
|
14
|
+
durationMs: number;
|
|
15
|
+
startedAt: number;
|
|
16
|
+
finishedAt: number;
|
|
17
|
+
metadata?: Record<string, unknown>;
|
|
18
|
+
aborted?: boolean;
|
|
19
|
+
}
|
|
20
|
+
export interface TaskCallbacks<T> {
|
|
21
|
+
onTaskStart?: (task: Task<T>) => void | Promise<void>;
|
|
22
|
+
onTaskComplete?: (result: TaskResult<T>) => void | Promise<void>;
|
|
23
|
+
onError?: (task: Task<T>, error: Error, durationMs: number) => void | Promise<void>;
|
|
24
|
+
}
|
|
25
|
+
export interface ParallelRunnerOptions<T = unknown> {
|
|
26
|
+
maxConcurrency?: number;
|
|
27
|
+
timeoutMs?: number;
|
|
28
|
+
failFast?: boolean;
|
|
29
|
+
callbacks?: TaskCallbacks<T>;
|
|
30
|
+
evaluateTaskSuccess?: (value: T, task: Task<T>) => boolean;
|
|
31
|
+
}
|
|
32
|
+
export interface ParallelRunnerMetrics {
|
|
33
|
+
totalTasks: number;
|
|
34
|
+
startedTasks: number;
|
|
35
|
+
completedTasks: number;
|
|
36
|
+
successCount: number;
|
|
37
|
+
failureCount: number;
|
|
38
|
+
totalDurationMs: number;
|
|
39
|
+
averageDurationMs: number;
|
|
40
|
+
throughput: number;
|
|
41
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/core/parallel/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import type { AIProvider, AIQueryOptions, AIResponse } from './ai-provider.interface';
|
|
2
|
+
import type { ProviderTaskLogHandler } from './base-ai.types';
|
|
3
|
+
import type { APIProviderConfig, ToolExecutionContext, ProviderExecutionMode } from '../../types/api-provider.types';
|
|
4
|
+
export declare class MastraAPIProvider implements AIProvider {
|
|
5
|
+
readonly name: string;
|
|
6
|
+
private config;
|
|
7
|
+
private toolContext?;
|
|
8
|
+
private filteredToolSets;
|
|
9
|
+
private readonly permissionsByMode;
|
|
10
|
+
private readonly defaultMode;
|
|
11
|
+
private readonly logConfig;
|
|
12
|
+
private taskLogHandler?;
|
|
13
|
+
constructor(config: APIProviderConfig, mode?: ProviderExecutionMode);
|
|
14
|
+
setTaskLogHandler(handler: ProviderTaskLogHandler): void;
|
|
15
|
+
private emitLog;
|
|
16
|
+
private createModel;
|
|
17
|
+
setTools(tools: any[], context: ToolExecutionContext): void;
|
|
18
|
+
private applyDefaultModeToContext;
|
|
19
|
+
private buildModeAwareToolSets;
|
|
20
|
+
private getToolIdentifier;
|
|
21
|
+
private createMastraToolsForMode;
|
|
22
|
+
query(prompt: string, options?: AIQueryOptions): Promise<AIResponse>;
|
|
23
|
+
execute(prompt: string, options?: AIQueryOptions): Promise<AIResponse>;
|
|
24
|
+
private runWithMode;
|
|
25
|
+
private extractToolName;
|
|
26
|
+
private extractToolArgs;
|
|
27
|
+
private extractToolResult;
|
|
28
|
+
private convertResponse;
|
|
29
|
+
isAvailable(): Promise<boolean>;
|
|
30
|
+
getToolPath(): Promise<string | null>;
|
|
31
|
+
}
|