openhorse 0.1.8 → 0.1.11
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/dist/agents/coordinator.d.ts +86 -0
- package/dist/agents/coordinator.d.ts.map +1 -0
- package/dist/agents/coordinator.js +251 -0
- package/dist/agents/coordinator.js.map +1 -0
- package/dist/agents/fork.d.ts +61 -0
- package/dist/agents/fork.d.ts.map +1 -0
- package/dist/agents/fork.js +169 -0
- package/dist/agents/fork.js.map +1 -0
- package/dist/agents/worker-pool.d.ts +72 -0
- package/dist/agents/worker-pool.d.ts.map +1 -0
- package/dist/agents/worker-pool.js +213 -0
- package/dist/agents/worker-pool.js.map +1 -0
- package/dist/cli.js +68 -4
- package/dist/cli.js.map +1 -1
- package/dist/commands/index.d.ts.map +1 -1
- package/dist/commands/index.js +57 -0
- package/dist/commands/index.js.map +1 -1
- package/dist/core/brain.d.ts +2 -1
- package/dist/core/brain.d.ts.map +1 -1
- package/dist/core/brain.js +3 -2
- package/dist/core/brain.js.map +1 -1
- package/dist/core/security-warning.d.ts +59 -0
- package/dist/core/security-warning.d.ts.map +1 -0
- package/dist/core/security-warning.js +163 -0
- package/dist/core/security-warning.js.map +1 -0
- package/dist/core/state.d.ts +94 -0
- package/dist/core/state.d.ts.map +1 -0
- package/dist/core/state.js +178 -0
- package/dist/core/state.js.map +1 -0
- package/dist/framework/query.d.ts +4 -0
- package/dist/framework/query.d.ts.map +1 -1
- package/dist/framework/query.js +49 -3
- package/dist/framework/query.js.map +1 -1
- package/dist/harness/harness.d.ts.map +1 -1
- package/dist/harness/harness.js +9 -1
- package/dist/harness/harness.js.map +1 -1
- package/dist/harness/safety.d.ts.map +1 -1
- package/dist/harness/safety.js +8 -0
- package/dist/harness/safety.js.map +1 -1
- package/dist/memory/embeddings.d.ts +2 -2
- package/dist/memory/embeddings.d.ts.map +1 -1
- package/dist/memory/embeddings.js +20 -6
- package/dist/memory/embeddings.js.map +1 -1
- package/dist/memory/entrypoint.d.ts +55 -0
- package/dist/memory/entrypoint.d.ts.map +1 -0
- package/dist/memory/entrypoint.js +151 -0
- package/dist/memory/entrypoint.js.map +1 -0
- package/dist/memory/relevant-finder.d.ts +54 -0
- package/dist/memory/relevant-finder.d.ts.map +1 -0
- package/dist/memory/relevant-finder.js +157 -0
- package/dist/memory/relevant-finder.js.map +1 -0
- package/dist/memory/storage.d.ts.map +1 -1
- package/dist/memory/storage.js +4 -3
- package/dist/memory/storage.js.map +1 -1
- package/dist/memory/team-paths.d.ts +63 -0
- package/dist/memory/team-paths.d.ts.map +1 -0
- package/dist/memory/team-paths.js +151 -0
- package/dist/memory/team-paths.js.map +1 -0
- package/dist/memory/vector-store.d.ts +2 -2
- package/dist/memory/vector-store.d.ts.map +1 -1
- package/dist/memory/vector-store.js +39 -12
- package/dist/memory/vector-store.js.map +1 -1
- package/dist/sdk/index.d.ts +10 -0
- package/dist/sdk/index.d.ts.map +1 -0
- package/dist/sdk/index.js +23 -0
- package/dist/sdk/index.js.map +1 -0
- package/dist/sdk/init.d.ts +24 -0
- package/dist/sdk/init.d.ts.map +1 -0
- package/dist/sdk/init.js +48 -0
- package/dist/sdk/init.js.map +1 -0
- package/dist/sdk/query.d.ts +19 -0
- package/dist/sdk/query.d.ts.map +1 -0
- package/dist/sdk/query.js +43 -0
- package/dist/sdk/query.js.map +1 -0
- package/dist/sdk/sessions.d.ts +19 -0
- package/dist/sdk/sessions.d.ts.map +1 -0
- package/dist/sdk/sessions.js +57 -0
- package/dist/sdk/sessions.js.map +1 -0
- package/dist/sdk/types.d.ts +47 -0
- package/dist/sdk/types.d.ts.map +1 -0
- package/dist/sdk/types.js +8 -0
- package/dist/sdk/types.js.map +1 -0
- package/dist/services/auth/auth.d.ts +121 -0
- package/dist/services/auth/auth.d.ts.map +1 -0
- package/dist/services/auth/auth.js +233 -0
- package/dist/services/auth/auth.js.map +1 -0
- package/dist/services/auth/aws.d.ts +27 -0
- package/dist/services/auth/aws.d.ts.map +1 -0
- package/dist/services/auth/aws.js +109 -0
- package/dist/services/auth/aws.js.map +1 -0
- package/dist/services/auth/index.d.ts +6 -0
- package/dist/services/auth/index.d.ts.map +1 -0
- package/dist/services/auth/index.js +17 -0
- package/dist/services/auth/index.js.map +1 -0
- package/dist/services/auto-fix/autoFixConfig.d.ts +33 -0
- package/dist/services/auto-fix/autoFixConfig.d.ts.map +1 -0
- package/dist/services/auto-fix/autoFixConfig.js +61 -0
- package/dist/services/auto-fix/autoFixConfig.js.map +1 -0
- package/dist/services/auto-fix/autoFixHook.d.ts +20 -0
- package/dist/services/auto-fix/autoFixHook.d.ts.map +1 -0
- package/dist/services/auto-fix/autoFixHook.js +89 -0
- package/dist/services/auto-fix/autoFixHook.js.map +1 -0
- package/dist/services/auto-fix/autoFixRunner.d.ts +60 -0
- package/dist/services/auto-fix/autoFixRunner.d.ts.map +1 -0
- package/dist/services/auto-fix/autoFixRunner.js +207 -0
- package/dist/services/auto-fix/autoFixRunner.js.map +1 -0
- package/dist/services/auto-fix/index.d.ts +7 -0
- package/dist/services/auto-fix/index.d.ts.map +1 -0
- package/dist/services/auto-fix/index.js +17 -0
- package/dist/services/auto-fix/index.js.map +1 -0
- package/dist/services/compact/auto-compact.d.ts +53 -0
- package/dist/services/compact/auto-compact.d.ts.map +1 -0
- package/dist/services/compact/auto-compact.js +114 -0
- package/dist/services/compact/auto-compact.js.map +1 -0
- package/dist/services/compact/compact.d.ts +51 -0
- package/dist/services/compact/compact.d.ts.map +1 -0
- package/dist/services/compact/compact.js +120 -0
- package/dist/services/compact/compact.js.map +1 -0
- package/dist/services/compact/index.d.ts +10 -0
- package/dist/services/compact/index.d.ts.map +1 -0
- package/dist/services/compact/index.js +23 -0
- package/dist/services/compact/index.js.map +1 -0
- package/dist/services/compact/micro-compact.d.ts +21 -0
- package/dist/services/compact/micro-compact.d.ts.map +1 -0
- package/dist/services/compact/micro-compact.js +99 -0
- package/dist/services/compact/micro-compact.js.map +1 -0
- package/dist/services/compact/summary-generator.d.ts +25 -0
- package/dist/services/compact/summary-generator.d.ts.map +1 -0
- package/dist/services/compact/summary-generator.js +128 -0
- package/dist/services/compact/summary-generator.js.map +1 -0
- package/dist/services/concurrent-sessions.d.ts +97 -0
- package/dist/services/concurrent-sessions.d.ts.map +1 -0
- package/dist/services/concurrent-sessions.js +263 -0
- package/dist/services/concurrent-sessions.js.map +1 -0
- package/dist/services/diagnostic-tracking/diagnosticTracking.d.ts +58 -0
- package/dist/services/diagnostic-tracking/diagnosticTracking.d.ts.map +1 -0
- package/dist/services/diagnostic-tracking/diagnosticTracking.js +167 -0
- package/dist/services/diagnostic-tracking/diagnosticTracking.js.map +1 -0
- package/dist/services/diagnostic-tracking/formatter.d.ts +17 -0
- package/dist/services/diagnostic-tracking/formatter.d.ts.map +1 -0
- package/dist/services/diagnostic-tracking/formatter.js +78 -0
- package/dist/services/diagnostic-tracking/formatter.js.map +1 -0
- package/dist/services/diagnostic-tracking/index.d.ts +7 -0
- package/dist/services/diagnostic-tracking/index.d.ts.map +1 -0
- package/dist/services/diagnostic-tracking/index.js +19 -0
- package/dist/services/diagnostic-tracking/index.js.map +1 -0
- package/dist/services/diagnostic-tracking/types.d.ts +52 -0
- package/dist/services/diagnostic-tracking/types.d.ts.map +1 -0
- package/dist/services/diagnostic-tracking/types.js +27 -0
- package/dist/services/diagnostic-tracking/types.js.map +1 -0
- package/dist/services/mcp/index.d.ts +5 -0
- package/dist/services/mcp/index.d.ts.map +1 -0
- package/dist/services/mcp/index.js +16 -0
- package/dist/services/mcp/index.js.map +1 -0
- package/dist/services/mcp/transports.d.ts +104 -0
- package/dist/services/mcp/transports.d.ts.map +1 -0
- package/dist/services/mcp/transports.js +354 -0
- package/dist/services/mcp/transports.js.map +1 -0
- package/dist/services/session-memory/index.d.ts +6 -0
- package/dist/services/session-memory/index.d.ts.map +1 -0
- package/dist/services/session-memory/index.js +15 -0
- package/dist/services/session-memory/index.js.map +1 -0
- package/dist/services/session-memory/prompts.d.ts +23 -0
- package/dist/services/session-memory/prompts.d.ts.map +1 -0
- package/dist/services/session-memory/prompts.js +70 -0
- package/dist/services/session-memory/prompts.js.map +1 -0
- package/dist/services/session-memory/sessionMemory.d.ts +64 -0
- package/dist/services/session-memory/sessionMemory.d.ts.map +1 -0
- package/dist/services/session-memory/sessionMemory.js +243 -0
- package/dist/services/session-memory/sessionMemory.js.map +1 -0
- package/dist/services/smart-routing.d.ts +41 -0
- package/dist/services/smart-routing.d.ts.map +1 -0
- package/dist/services/smart-routing.js +134 -0
- package/dist/services/smart-routing.js.map +1 -0
- package/dist/tools/git.d.ts +15 -0
- package/dist/tools/git.d.ts.map +1 -0
- package/dist/tools/git.js +282 -0
- package/dist/tools/git.js.map +1 -0
- package/dist/tools/index.d.ts.map +1 -1
- package/dist/tools/index.js +83 -14
- package/dist/tools/index.js.map +1 -1
- package/dist/tools/lsp.d.ts +18 -0
- package/dist/tools/lsp.d.ts.map +1 -0
- package/dist/tools/lsp.js +450 -0
- package/dist/tools/lsp.js.map +1 -0
- package/dist/tools/web.d.ts +3 -0
- package/dist/tools/web.d.ts.map +1 -1
- package/dist/tools/web.js +242 -69
- package/dist/tools/web.js.map +1 -1
- package/dist/ui/command-panel.d.ts +9 -0
- package/dist/ui/command-panel.d.ts.map +1 -1
- package/dist/ui/command-panel.js +105 -24
- package/dist/ui/command-panel.js.map +1 -1
- package/dist/ui/markdown.d.ts +10 -4
- package/dist/ui/markdown.d.ts.map +1 -1
- package/dist/ui/markdown.js +91 -4
- package/dist/ui/markdown.js.map +1 -1
- package/dist/ui/progress.d.ts +34 -0
- package/dist/ui/progress.d.ts.map +1 -0
- package/dist/ui/progress.js +88 -0
- package/dist/ui/progress.js.map +1 -0
- package/dist/ui/status-bar.d.ts.map +1 -1
- package/dist/ui/status-bar.js +10 -0
- package/dist/ui/status-bar.js.map +1 -1
- package/package.json +3 -7
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* openhorse - Coordinator
|
|
3
|
+
*
|
|
4
|
+
* 多 Agent 协调器,负责任务分配和结果聚合。
|
|
5
|
+
* 与 Fork/Worker Pool 模式配合使用。
|
|
6
|
+
*/
|
|
7
|
+
import type { Task } from '../core/agent';
|
|
8
|
+
import type { ForkResult } from './fork';
|
|
9
|
+
export interface CoordinatorConfig {
|
|
10
|
+
/** 协调模式 */
|
|
11
|
+
mode?: 'distribute' | 'pipeline' | 'parallel';
|
|
12
|
+
/** 最大并行任务 */
|
|
13
|
+
maxParallel?: number;
|
|
14
|
+
/** 结果聚合策略 */
|
|
15
|
+
aggregationStrategy?: 'first' | 'best' | 'all';
|
|
16
|
+
}
|
|
17
|
+
export interface TaskAssignment {
|
|
18
|
+
taskId: string;
|
|
19
|
+
agentId: string;
|
|
20
|
+
reason: string;
|
|
21
|
+
confidence: number;
|
|
22
|
+
}
|
|
23
|
+
export interface CoordinatorResult {
|
|
24
|
+
success: boolean;
|
|
25
|
+
assignments: TaskAssignment[];
|
|
26
|
+
results: Map<string, ForkResult>;
|
|
27
|
+
summary: string;
|
|
28
|
+
duration: number;
|
|
29
|
+
}
|
|
30
|
+
export declare class Coordinator {
|
|
31
|
+
private agents;
|
|
32
|
+
private taskAssignments;
|
|
33
|
+
private pendingTasks;
|
|
34
|
+
private completedTasks;
|
|
35
|
+
private config;
|
|
36
|
+
private workerPool;
|
|
37
|
+
constructor(config?: CoordinatorConfig);
|
|
38
|
+
/**
|
|
39
|
+
* 注册 Agent
|
|
40
|
+
*/
|
|
41
|
+
registerAgent(profile: AgentProfile): void;
|
|
42
|
+
/**
|
|
43
|
+
* 分配任务到最合适的 Agent
|
|
44
|
+
*/
|
|
45
|
+
assignTask(task: Task): TaskAssignment;
|
|
46
|
+
/**
|
|
47
|
+
* 执行任务(协调模式)
|
|
48
|
+
*/
|
|
49
|
+
execute(task: Task): Promise<CoordinatorResult>;
|
|
50
|
+
/**
|
|
51
|
+
* 收集所有已完成结果
|
|
52
|
+
*/
|
|
53
|
+
collectResults(): Map<string, ForkResult>;
|
|
54
|
+
/**
|
|
55
|
+
* 获取协调器状态
|
|
56
|
+
*/
|
|
57
|
+
getStatus(): {
|
|
58
|
+
registeredAgents: number;
|
|
59
|
+
pendingTasks: number;
|
|
60
|
+
completedTasks: number;
|
|
61
|
+
mode: string;
|
|
62
|
+
};
|
|
63
|
+
private registerDefaultAgents;
|
|
64
|
+
/**
|
|
65
|
+
* 任务分解(将大任务拆分为子任务)
|
|
66
|
+
*/
|
|
67
|
+
private decomposeTask;
|
|
68
|
+
/**
|
|
69
|
+
* 创建流水线任务序列
|
|
70
|
+
*/
|
|
71
|
+
private createPipeline;
|
|
72
|
+
/**
|
|
73
|
+
* 聚合结果
|
|
74
|
+
*/
|
|
75
|
+
private aggregateResults;
|
|
76
|
+
}
|
|
77
|
+
interface AgentProfile {
|
|
78
|
+
id: string;
|
|
79
|
+
name: string;
|
|
80
|
+
capabilities: string[];
|
|
81
|
+
priority: number;
|
|
82
|
+
}
|
|
83
|
+
export declare function getCoordinator(config?: CoordinatorConfig): Coordinator;
|
|
84
|
+
export declare function resetCoordinator(): void;
|
|
85
|
+
export {};
|
|
86
|
+
//# sourceMappingURL=coordinator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"coordinator.d.ts","sourceRoot":"","sources":["../../src/agents/coordinator.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAc,MAAM,eAAe,CAAC;AACtD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAQzC,MAAM,WAAW,iBAAiB;IAChC,WAAW;IACX,IAAI,CAAC,EAAE,YAAY,GAAG,UAAU,GAAG,UAAU,CAAC;IAC9C,aAAa;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,aAAa;IACb,mBAAmB,CAAC,EAAE,OAAO,GAAG,MAAM,GAAG,KAAK,CAAC;CAChD;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,WAAW,EAAE,cAAc,EAAE,CAAC;IAC9B,OAAO,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACjC,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAMD,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAwC;IACtD,OAAO,CAAC,eAAe,CAA0C;IACjE,OAAO,CAAC,YAAY,CAAc;IAClC,OAAO,CAAC,cAAc,CAAsC;IAC5D,OAAO,CAAC,MAAM,CAAoB;IAClC,OAAO,CAAC,UAAU,CAAa;gBAEnB,MAAM,CAAC,EAAE,iBAAiB;IAYtC;;OAEG;IACH,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAI1C;;OAEG;IACH,UAAU,CAAC,IAAI,EAAE,IAAI,GAAG,cAAc;IAetC;;OAEG;IACG,OAAO,CAAC,IAAI,EAAE,IAAI,GAAG,OAAO,CAAC,iBAAiB,CAAC;IA8FrD;;OAEG;IACH,cAAc,IAAI,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;IAIzC;;OAEG;IACH,SAAS,IAAI;QACX,gBAAgB,EAAE,MAAM,CAAC;QACzB,YAAY,EAAE,MAAM,CAAC;QACrB,cAAc,EAAE,MAAM,CAAC;QACvB,IAAI,EAAE,MAAM,CAAC;KACd;IAaD,OAAO,CAAC,qBAAqB;IA8B7B;;OAEG;IACH,OAAO,CAAC,aAAa;IAiBrB;;OAEG;IACH,OAAO,CAAC,cAAc;IAQtB;;OAEG;IACH,OAAO,CAAC,gBAAgB;CA6BzB;AAMD,UAAU,YAAY;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAQD,wBAAgB,cAAc,CAAC,MAAM,CAAC,EAAE,iBAAiB,GAAG,WAAW,CAKtE;AAED,wBAAgB,gBAAgB,IAAI,IAAI,CAEvC"}
|
|
@@ -0,0 +1,251 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* openhorse - Coordinator
|
|
4
|
+
*
|
|
5
|
+
* 多 Agent 协调器,负责任务分配和结果聚合。
|
|
6
|
+
* 与 Fork/Worker Pool 模式配合使用。
|
|
7
|
+
*/
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.Coordinator = void 0;
|
|
10
|
+
exports.getCoordinator = getCoordinator;
|
|
11
|
+
exports.resetCoordinator = resetCoordinator;
|
|
12
|
+
const router_1 = require("./router");
|
|
13
|
+
const worker_pool_1 = require("./worker-pool");
|
|
14
|
+
// ============================================================================
|
|
15
|
+
// Coordinator 实现
|
|
16
|
+
// ============================================================================
|
|
17
|
+
class Coordinator {
|
|
18
|
+
constructor(config) {
|
|
19
|
+
this.agents = new Map();
|
|
20
|
+
this.taskAssignments = new Map();
|
|
21
|
+
this.pendingTasks = [];
|
|
22
|
+
this.completedTasks = new Map();
|
|
23
|
+
this.config = {
|
|
24
|
+
mode: config?.mode || 'distribute',
|
|
25
|
+
maxParallel: config?.maxParallel || 3,
|
|
26
|
+
aggregationStrategy: config?.aggregationStrategy || 'best',
|
|
27
|
+
};
|
|
28
|
+
this.workerPool = (0, worker_pool_1.getWorkerPool)({ maxWorkers: this.config.maxParallel });
|
|
29
|
+
// 注册默认 Agent
|
|
30
|
+
this.registerDefaultAgents();
|
|
31
|
+
}
|
|
32
|
+
/**
|
|
33
|
+
* 注册 Agent
|
|
34
|
+
*/
|
|
35
|
+
registerAgent(profile) {
|
|
36
|
+
this.agents.set(profile.id, profile);
|
|
37
|
+
}
|
|
38
|
+
/**
|
|
39
|
+
* 分配任务到最合适的 Agent
|
|
40
|
+
*/
|
|
41
|
+
assignTask(task) {
|
|
42
|
+
const router = (0, router_1.getAgentRouter)();
|
|
43
|
+
const routeResult = router.route(task);
|
|
44
|
+
const assignment = {
|
|
45
|
+
taskId: task.id,
|
|
46
|
+
agentId: routeResult.agentId,
|
|
47
|
+
reason: routeResult.reason,
|
|
48
|
+
confidence: routeResult.confidence,
|
|
49
|
+
};
|
|
50
|
+
this.taskAssignments.set(task.id, assignment);
|
|
51
|
+
return assignment;
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* 执行任务(协调模式)
|
|
55
|
+
*/
|
|
56
|
+
async execute(task) {
|
|
57
|
+
const startTime = Date.now();
|
|
58
|
+
const assignments = [];
|
|
59
|
+
switch (this.config.mode) {
|
|
60
|
+
case 'distribute':
|
|
61
|
+
// 分发模式:单任务分配到最合适 Agent
|
|
62
|
+
const assignment = this.assignTask(task);
|
|
63
|
+
assignments.push(assignment);
|
|
64
|
+
const result = await this.workerPool.submit(task, {
|
|
65
|
+
taskDescription: task.description,
|
|
66
|
+
maxTurns: 5,
|
|
67
|
+
});
|
|
68
|
+
this.completedTasks.set(task.id, result);
|
|
69
|
+
return {
|
|
70
|
+
success: result.success,
|
|
71
|
+
assignments,
|
|
72
|
+
results: new Map([[task.id, result]]),
|
|
73
|
+
summary: result.content.slice(0, 200),
|
|
74
|
+
duration: Date.now() - startTime,
|
|
75
|
+
};
|
|
76
|
+
case 'parallel':
|
|
77
|
+
// 并行模式:任务分解后并行执行
|
|
78
|
+
const subtasks = this.decomposeTask(task);
|
|
79
|
+
const batchAssignments = [];
|
|
80
|
+
for (const sub of subtasks) {
|
|
81
|
+
const subAssignment = this.assignTask(sub);
|
|
82
|
+
batchAssignments.push(subAssignment);
|
|
83
|
+
}
|
|
84
|
+
const batchResults = await this.workerPool.submitBatch(subtasks);
|
|
85
|
+
const aggregated = this.aggregateResults(batchResults);
|
|
86
|
+
return {
|
|
87
|
+
success: aggregated.success,
|
|
88
|
+
assignments: batchAssignments,
|
|
89
|
+
results: batchResults,
|
|
90
|
+
summary: aggregated.summary,
|
|
91
|
+
duration: Date.now() - startTime,
|
|
92
|
+
};
|
|
93
|
+
case 'pipeline':
|
|
94
|
+
// 流水线模式:任务按顺序执行
|
|
95
|
+
const pipelineTasks = this.createPipeline(task);
|
|
96
|
+
const pipelineAssignments = [];
|
|
97
|
+
const pipelineResults = new Map();
|
|
98
|
+
for (const stage of pipelineTasks) {
|
|
99
|
+
const stageAssignment = this.assignTask(stage);
|
|
100
|
+
pipelineAssignments.push(stageAssignment);
|
|
101
|
+
const stageResult = await this.workerPool.submit(stage, {
|
|
102
|
+
taskDescription: stage.description,
|
|
103
|
+
maxTurns: 3,
|
|
104
|
+
});
|
|
105
|
+
pipelineResults.set(stage.id, stageResult);
|
|
106
|
+
if (!stageResult.success) {
|
|
107
|
+
// 流水线失败:停止后续阶段
|
|
108
|
+
return {
|
|
109
|
+
success: false,
|
|
110
|
+
assignments: pipelineAssignments,
|
|
111
|
+
results: pipelineResults,
|
|
112
|
+
summary: `Pipeline failed at stage ${stage.name}: ${stageResult.error}`,
|
|
113
|
+
duration: Date.now() - startTime,
|
|
114
|
+
};
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
return {
|
|
118
|
+
success: true,
|
|
119
|
+
assignments: pipelineAssignments,
|
|
120
|
+
results: pipelineResults,
|
|
121
|
+
summary: 'Pipeline completed successfully',
|
|
122
|
+
duration: Date.now() - startTime,
|
|
123
|
+
};
|
|
124
|
+
}
|
|
125
|
+
// 默认返回(防止 TypeScript 报错)
|
|
126
|
+
return {
|
|
127
|
+
success: false,
|
|
128
|
+
assignments: [],
|
|
129
|
+
results: new Map(),
|
|
130
|
+
summary: 'Unknown mode',
|
|
131
|
+
duration: Date.now() - startTime,
|
|
132
|
+
};
|
|
133
|
+
}
|
|
134
|
+
/**
|
|
135
|
+
* 收集所有已完成结果
|
|
136
|
+
*/
|
|
137
|
+
collectResults() {
|
|
138
|
+
return new Map(this.completedTasks);
|
|
139
|
+
}
|
|
140
|
+
/**
|
|
141
|
+
* 获取协调器状态
|
|
142
|
+
*/
|
|
143
|
+
getStatus() {
|
|
144
|
+
return {
|
|
145
|
+
registeredAgents: this.agents.size,
|
|
146
|
+
pendingTasks: this.pendingTasks.length,
|
|
147
|
+
completedTasks: this.completedTasks.size,
|
|
148
|
+
mode: this.config.mode || 'distribute',
|
|
149
|
+
};
|
|
150
|
+
}
|
|
151
|
+
// ============================================================================
|
|
152
|
+
// 内部方法
|
|
153
|
+
// ============================================================================
|
|
154
|
+
registerDefaultAgents() {
|
|
155
|
+
this.registerAgent({
|
|
156
|
+
id: 'coder',
|
|
157
|
+
name: 'Coder',
|
|
158
|
+
capabilities: ['implement', 'write', 'fix', 'refactor'],
|
|
159
|
+
priority: 60,
|
|
160
|
+
});
|
|
161
|
+
this.registerAgent({
|
|
162
|
+
id: 'leader',
|
|
163
|
+
name: 'Leader',
|
|
164
|
+
capabilities: ['coordinate', 'plan', 'organize'],
|
|
165
|
+
priority: 50,
|
|
166
|
+
});
|
|
167
|
+
this.registerAgent({
|
|
168
|
+
id: 'reviewer',
|
|
169
|
+
name: 'Reviewer',
|
|
170
|
+
capabilities: ['review', 'check', 'audit', 'analyze'],
|
|
171
|
+
priority: 55,
|
|
172
|
+
});
|
|
173
|
+
this.registerAgent({
|
|
174
|
+
id: 'tester',
|
|
175
|
+
name: 'Tester',
|
|
176
|
+
capabilities: ['test', 'verify', 'validate'],
|
|
177
|
+
priority: 55,
|
|
178
|
+
});
|
|
179
|
+
}
|
|
180
|
+
/**
|
|
181
|
+
* 任务分解(将大任务拆分为子任务)
|
|
182
|
+
*/
|
|
183
|
+
decomposeTask(task) {
|
|
184
|
+
// 简化实现:基于关键词分解
|
|
185
|
+
const classification = (0, router_1.classifyTask)(task.name, task.description);
|
|
186
|
+
// 根据类别分解
|
|
187
|
+
if (classification.category === 'coding') {
|
|
188
|
+
return [
|
|
189
|
+
{ ...task, id: `${task.id}-impl`, name: `${task.name}-impl`, description: `Implement: ${task.description}` },
|
|
190
|
+
{ ...task, id: `${task.id}-test`, name: `${task.name}-test`, description: `Write tests for: ${task.name}`, assignedTo: 'tester' },
|
|
191
|
+
{ ...task, id: `${task.id}-review`, name: `${task.name}-review`, description: `Review implementation`, assignedTo: 'reviewer' },
|
|
192
|
+
];
|
|
193
|
+
}
|
|
194
|
+
// 默认:不分解
|
|
195
|
+
return [task];
|
|
196
|
+
}
|
|
197
|
+
/**
|
|
198
|
+
* 创建流水线任务序列
|
|
199
|
+
*/
|
|
200
|
+
createPipeline(task) {
|
|
201
|
+
return [
|
|
202
|
+
{ ...task, id: `${task.id}-plan`, name: `Plan`, description: `Plan approach for: ${task.description}`, assignedTo: 'leader' },
|
|
203
|
+
{ ...task, id: `${task.id}-impl`, name: `Implement`, description: `Execute: ${task.description}`, assignedTo: 'coder' },
|
|
204
|
+
{ ...task, id: `${task.id}-verify`, name: `Verify`, description: `Verify results`, assignedTo: 'reviewer' },
|
|
205
|
+
];
|
|
206
|
+
}
|
|
207
|
+
/**
|
|
208
|
+
* 聚合结果
|
|
209
|
+
*/
|
|
210
|
+
aggregateResults(results) {
|
|
211
|
+
switch (this.config.aggregationStrategy) {
|
|
212
|
+
case 'first':
|
|
213
|
+
const firstResult = results.values().next().value;
|
|
214
|
+
return { success: firstResult?.success || false, summary: firstResult?.content || '' };
|
|
215
|
+
case 'best':
|
|
216
|
+
// 找到最好的结果
|
|
217
|
+
let best = null;
|
|
218
|
+
for (const result of results.values()) {
|
|
219
|
+
if (result.success && (!best || result.content.length > best.content.length)) {
|
|
220
|
+
best = result;
|
|
221
|
+
}
|
|
222
|
+
}
|
|
223
|
+
return { success: best?.success || false, summary: best?.content || 'No successful results' };
|
|
224
|
+
case 'all':
|
|
225
|
+
// 合并所有结果
|
|
226
|
+
const allContent = Array.from(results.values())
|
|
227
|
+
.map(r => r.content)
|
|
228
|
+
.filter(c => c)
|
|
229
|
+
.join('\n\n---\n\n');
|
|
230
|
+
const allSuccess = Array.from(results.values()).every(r => r.success);
|
|
231
|
+
return { success: allSuccess, summary: allContent.slice(0, 500) };
|
|
232
|
+
}
|
|
233
|
+
// 默认返回
|
|
234
|
+
return { success: false, summary: 'Unknown aggregation strategy' };
|
|
235
|
+
}
|
|
236
|
+
}
|
|
237
|
+
exports.Coordinator = Coordinator;
|
|
238
|
+
// ============================================================================
|
|
239
|
+
// 单例
|
|
240
|
+
// ============================================================================
|
|
241
|
+
let defaultCoordinator = null;
|
|
242
|
+
function getCoordinator(config) {
|
|
243
|
+
if (!defaultCoordinator) {
|
|
244
|
+
defaultCoordinator = new Coordinator(config);
|
|
245
|
+
}
|
|
246
|
+
return defaultCoordinator;
|
|
247
|
+
}
|
|
248
|
+
function resetCoordinator() {
|
|
249
|
+
defaultCoordinator = null;
|
|
250
|
+
}
|
|
251
|
+
//# sourceMappingURL=coordinator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"coordinator.js","sourceRoot":"","sources":["../../src/agents/coordinator.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAiUH,wCAKC;AAED,4CAEC;AAtUD,qCAAwD;AACxD,+CAA+D;AA8B/D,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,MAAa,WAAW;IAQtB,YAAY,MAA0B;QAP9B,WAAM,GAA8B,IAAI,GAAG,EAAE,CAAC;QAC9C,oBAAe,GAAgC,IAAI,GAAG,EAAE,CAAC;QACzD,iBAAY,GAAW,EAAE,CAAC;QAC1B,mBAAc,GAA4B,IAAI,GAAG,EAAE,CAAC;QAK1D,IAAI,CAAC,MAAM,GAAG;YACZ,IAAI,EAAE,MAAM,EAAE,IAAI,IAAI,YAAY;YAClC,WAAW,EAAE,MAAM,EAAE,WAAW,IAAI,CAAC;YACrC,mBAAmB,EAAE,MAAM,EAAE,mBAAmB,IAAI,MAAM;SAC3D,CAAC;QACF,IAAI,CAAC,UAAU,GAAG,IAAA,2BAAa,EAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC;QAEzE,aAAa;QACb,IAAI,CAAC,qBAAqB,EAAE,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,OAAqB;QACjC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,IAAU;QACnB,MAAM,MAAM,GAAG,IAAA,uBAAc,GAAE,CAAC;QAChC,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAEvC,MAAM,UAAU,GAAmB;YACjC,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,OAAO,EAAE,WAAW,CAAC,OAAO;YAC5B,MAAM,EAAE,WAAW,CAAC,MAAM;YAC1B,UAAU,EAAE,WAAW,CAAC,UAAU;SACnC,CAAC;QAEF,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAC9C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,IAAU;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,WAAW,GAAqB,EAAE,CAAC;QAEzC,QAAQ,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACzB,KAAK,YAAY;gBACf,uBAAuB;gBACvB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBACzC,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAE7B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE;oBAChD,eAAe,EAAE,IAAI,CAAC,WAAW;oBACjC,QAAQ,EAAE,CAAC;iBACZ,CAAC,CAAC;gBAEH,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;gBAEzC,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,WAAW;oBACX,OAAO,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC;oBACrC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBACrC,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBACjC,CAAC;YAEJ,KAAK,UAAU;gBACb,iBAAiB;gBACjB,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;gBAC1C,MAAM,gBAAgB,GAAqB,EAAE,CAAC;gBAE9C,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;oBAC3B,MAAM,aAAa,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;oBAC3C,gBAAgB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBACvC,CAAC;gBAED,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;gBACjE,MAAM,UAAU,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;gBAEvD,OAAO;oBACL,OAAO,EAAE,UAAU,CAAC,OAAO;oBAC3B,WAAW,EAAE,gBAAgB;oBAC7B,OAAO,EAAE,YAAY;oBACrB,OAAO,EAAE,UAAU,CAAC,OAAO;oBAC3B,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBACjC,CAAC;YAEJ,KAAK,UAAU;gBACb,gBAAgB;gBAChB,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;gBAChD,MAAM,mBAAmB,GAAqB,EAAE,CAAC;gBACjD,MAAM,eAAe,GAA4B,IAAI,GAAG,EAAE,CAAC;gBAE3D,KAAK,MAAM,KAAK,IAAI,aAAa,EAAE,CAAC;oBAClC,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;oBAC/C,mBAAmB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;oBAE1C,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE;wBACtD,eAAe,EAAE,KAAK,CAAC,WAAW;wBAClC,QAAQ,EAAE,CAAC;qBACZ,CAAC,CAAC;oBAEH,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;oBAE3C,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;wBACzB,eAAe;wBACf,OAAO;4BACL,OAAO,EAAE,KAAK;4BACd,WAAW,EAAE,mBAAmB;4BAChC,OAAO,EAAE,eAAe;4BACxB,OAAO,EAAE,4BAA4B,KAAK,CAAC,IAAI,KAAK,WAAW,CAAC,KAAK,EAAE;4BACvE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;yBACjC,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,OAAO,EAAE,IAAI;oBACb,WAAW,EAAE,mBAAmB;oBAChC,OAAO,EAAE,eAAe;oBACxB,OAAO,EAAE,iCAAiC;oBAC1C,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;iBACjC,CAAC;QACN,CAAC;QAED,yBAAyB;QACzB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,WAAW,EAAE,EAAE;YACf,OAAO,EAAE,IAAI,GAAG,EAAE;YAClB,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SACjC,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,SAAS;QAMP,OAAO;YACL,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YAClC,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,MAAM;YACtC,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI;YACxC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,YAAY;SACvC,CAAC;IACJ,CAAC;IAED,+EAA+E;IAC/E,OAAO;IACP,+EAA+E;IAEvE,qBAAqB;QAC3B,IAAI,CAAC,aAAa,CAAC;YACjB,EAAE,EAAE,OAAO;YACX,IAAI,EAAE,OAAO;YACb,YAAY,EAAE,CAAC,WAAW,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,CAAC;YACvD,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC;YACjB,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,UAAU,CAAC;YAChD,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC;YACjB,EAAE,EAAE,UAAU;YACd,IAAI,EAAE,UAAU;YAChB,YAAY,EAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC;YACrD,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;QAEH,IAAI,CAAC,aAAa,CAAC;YACjB,EAAE,EAAE,QAAQ;YACZ,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC;YAC5C,QAAQ,EAAE,EAAE;SACb,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,IAAU;QAC9B,eAAe;QACf,MAAM,cAAc,GAAG,IAAA,qBAAY,EAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAEjE,SAAS;QACT,IAAI,cAAc,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACzC,OAAO;gBACL,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,OAAO,EAAE,WAAW,EAAE,cAAc,IAAI,CAAC,WAAW,EAAE,EAAE;gBAC5G,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,OAAO,EAAE,WAAW,EAAE,oBAAoB,IAAI,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE;gBACjI,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,SAAS,EAAE,WAAW,EAAE,uBAAuB,EAAE,UAAU,EAAE,UAAU,EAAE;aAChI,CAAC;QACJ,CAAC;QAED,SAAS;QACT,OAAO,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,IAAU;QAC/B,OAAO;YACL,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,sBAAsB,IAAI,CAAC,WAAW,EAAE,EAAE,UAAU,EAAE,QAAQ,EAAE;YAC7H,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,WAAW,EAAE,YAAY,IAAI,CAAC,WAAW,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE;YACvH,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,gBAAgB,EAAE,UAAU,EAAE,UAAU,EAAE;SAC5G,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,OAAgC;QACvD,QAAQ,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;YACxC,KAAK,OAAO;gBACV,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;gBAClD,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,IAAI,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,IAAI,EAAE,EAAE,CAAC;YAEzF,KAAK,MAAM;gBACT,UAAU;gBACV,IAAI,IAAI,GAAsB,IAAI,CAAC;gBACnC,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;oBACtC,IAAI,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC7E,IAAI,GAAG,MAAM,CAAC;oBAChB,CAAC;gBACH,CAAC;gBACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,IAAI,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,IAAI,uBAAuB,EAAE,CAAC;YAEhG,KAAK,KAAK;gBACR,SAAS;gBACT,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;qBAC5C,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;qBACnB,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;qBACd,IAAI,CAAC,aAAa,CAAC,CAAC;gBACvB,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBACtE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;QACtE,CAAC;QAED,OAAO;QACP,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,8BAA8B,EAAE,CAAC;IACrE,CAAC;CACF;AAvQD,kCAuQC;AAaD,+EAA+E;AAC/E,KAAK;AACL,+EAA+E;AAE/E,IAAI,kBAAkB,GAAuB,IAAI,CAAC;AAElD,SAAgB,cAAc,CAAC,MAA0B;IACvD,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxB,kBAAkB,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,kBAAkB,CAAC;AAC5B,CAAC;AAED,SAAgB,gBAAgB;IAC9B,kBAAkB,GAAG,IAAI,CAAC;AAC5B,CAAC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* openhorse - Fork Subagent
|
|
3
|
+
*
|
|
4
|
+
* Fork 一个子 Agent 继承父上下文,执行独立任务后返回结果。
|
|
5
|
+
* 参考 OpenClaude 的 forkSubagent.ts 实现。
|
|
6
|
+
*/
|
|
7
|
+
import type { Message } from '../services/llm';
|
|
8
|
+
import type { OpenHorseTool } from '../framework/tool';
|
|
9
|
+
import type { PermissionMode } from '../commands/types';
|
|
10
|
+
export interface ForkOptions {
|
|
11
|
+
/** 继承父上下文(messages + systemPrompt) */
|
|
12
|
+
inheritContext: boolean;
|
|
13
|
+
/** 权限模式 */
|
|
14
|
+
permissionMode?: PermissionMode;
|
|
15
|
+
/** 可用工具(默认 TOOLS) */
|
|
16
|
+
tools?: OpenHorseTool[];
|
|
17
|
+
/** 最大轮次 */
|
|
18
|
+
maxTurns?: number;
|
|
19
|
+
/** 后台执行(不阻塞父 Agent) */
|
|
20
|
+
background?: boolean;
|
|
21
|
+
/** 任务描述 */
|
|
22
|
+
taskDescription: string;
|
|
23
|
+
/** 父消息历史 */
|
|
24
|
+
parentMessages?: Message[];
|
|
25
|
+
/** 父系统提示 */
|
|
26
|
+
parentSystemPrompt?: string;
|
|
27
|
+
/** 父工作目录 */
|
|
28
|
+
cwd?: string;
|
|
29
|
+
/** 父 memory/skills 内容 */
|
|
30
|
+
memoryContent?: string;
|
|
31
|
+
skillsContent?: string;
|
|
32
|
+
}
|
|
33
|
+
export interface ForkResult {
|
|
34
|
+
success: boolean;
|
|
35
|
+
content: string;
|
|
36
|
+
toolCalls?: Array<{
|
|
37
|
+
name: string;
|
|
38
|
+
args: Record<string, unknown>;
|
|
39
|
+
result: string;
|
|
40
|
+
}>;
|
|
41
|
+
error?: string;
|
|
42
|
+
duration: number;
|
|
43
|
+
tokenUsage?: {
|
|
44
|
+
promptTokens: number;
|
|
45
|
+
completionTokens: number;
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
/**
|
|
49
|
+
* Fork 子 Agent 执行任务
|
|
50
|
+
*/
|
|
51
|
+
export declare function forkSubagent(options: ForkOptions): Promise<ForkResult>;
|
|
52
|
+
/**
|
|
53
|
+
* 检测消息是否来自 Fork Subagent(防止递归 Fork)
|
|
54
|
+
*/
|
|
55
|
+
export declare function isForkSubagentMessage(content: string): boolean;
|
|
56
|
+
/**
|
|
57
|
+
* 标记 Fork Subagent 消息
|
|
58
|
+
*/
|
|
59
|
+
export declare function markForkSubagentMessage(content: string): string;
|
|
60
|
+
export { ForkOptions as ForkSubagentOptions, ForkResult as ForkSubagentResult };
|
|
61
|
+
//# sourceMappingURL=fork.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fork.d.ts","sourceRoot":"","sources":["../../src/agents/fork.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAQxD,MAAM,WAAW,WAAW;IAC1B,sCAAsC;IACtC,cAAc,EAAE,OAAO,CAAC;IACxB,WAAW;IACX,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,qBAAqB;IACrB,KAAK,CAAC,EAAE,aAAa,EAAE,CAAC;IACxB,WAAW;IACX,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,uBAAuB;IACvB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,WAAW;IACX,eAAe,EAAE,MAAM,CAAC;IACxB,YAAY;IACZ,cAAc,CAAC,EAAE,OAAO,EAAE,CAAC;IAC3B,YAAY;IACZ,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,YAAY;IACZ,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,yBAAyB;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IACnF,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAA;KAAE,CAAC;CACjE;AAMD;;GAEG;AACH,wBAAsB,YAAY,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,CAgI5E;AA6BD;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAE9D;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE/D;AAMD,OAAO,EAAE,WAAW,IAAI,mBAAmB,EAAE,UAAU,IAAI,kBAAkB,EAAE,CAAC"}
|
|
@@ -0,0 +1,169 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* openhorse - Fork Subagent
|
|
4
|
+
*
|
|
5
|
+
* Fork 一个子 Agent 继承父上下文,执行独立任务后返回结果。
|
|
6
|
+
* 参考 OpenClaude 的 forkSubagent.ts 实现。
|
|
7
|
+
*/
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.forkSubagent = forkSubagent;
|
|
10
|
+
exports.isForkSubagentMessage = isForkSubagentMessage;
|
|
11
|
+
exports.markForkSubagentMessage = markForkSubagentMessage;
|
|
12
|
+
const framework_1 = require("../framework");
|
|
13
|
+
const tools_1 = require("../tools");
|
|
14
|
+
// ============================================================================
|
|
15
|
+
// Fork 实现
|
|
16
|
+
// ============================================================================
|
|
17
|
+
/**
|
|
18
|
+
* Fork 子 Agent 执行任务
|
|
19
|
+
*/
|
|
20
|
+
async function forkSubagent(options) {
|
|
21
|
+
const { inheritContext = true, tools = tools_1.TOOLS, maxTurns = 5, background = false, taskDescription, parentMessages = [], parentSystemPrompt, cwd = process.cwd(), memoryContent = '', skillsContent = '', } = options;
|
|
22
|
+
const startTime = Date.now();
|
|
23
|
+
const forkId = `fork-${String(Date.now()).slice(-6)}`;
|
|
24
|
+
// 构建子 Agent 的消息历史
|
|
25
|
+
let messages;
|
|
26
|
+
if (inheritContext && parentMessages.length > 0) {
|
|
27
|
+
// 继承父上下文,但添加任务消息
|
|
28
|
+
messages = [...parentMessages];
|
|
29
|
+
// 添加 Fork 任务提示
|
|
30
|
+
messages.push({
|
|
31
|
+
role: 'user',
|
|
32
|
+
content: `[Fork Task] ${taskDescription}\n\nYou are a forked subagent. Complete this task independently and report results. Do not ask for clarification - make reasonable assumptions.`,
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
else {
|
|
36
|
+
// 创建独立上下文
|
|
37
|
+
const promptCtx = {
|
|
38
|
+
cwd,
|
|
39
|
+
platform: process.platform,
|
|
40
|
+
nodeVersion: process.version,
|
|
41
|
+
tools,
|
|
42
|
+
memoryContent,
|
|
43
|
+
skillsContent,
|
|
44
|
+
};
|
|
45
|
+
const systemPrompt = parentSystemPrompt || (0, framework_1.getSystemPrompt)(promptCtx);
|
|
46
|
+
messages = [
|
|
47
|
+
{ role: 'system', content: systemPrompt },
|
|
48
|
+
{ role: 'user', content: taskDescription },
|
|
49
|
+
];
|
|
50
|
+
}
|
|
51
|
+
// 工具执行器
|
|
52
|
+
const toolExecutor = async (name, args) => {
|
|
53
|
+
// 简化版:直接执行工具
|
|
54
|
+
const tool = tools.find(t => t.name === name);
|
|
55
|
+
if (!tool) {
|
|
56
|
+
return JSON.stringify({ success: false, error: `Unknown tool: ${name}` });
|
|
57
|
+
}
|
|
58
|
+
// 调用工具的 execute
|
|
59
|
+
const context = { cwd, config: { name: 'openhorse', mode: 'development' } };
|
|
60
|
+
try {
|
|
61
|
+
const result = await tool.execute(args, context);
|
|
62
|
+
return JSON.stringify(result);
|
|
63
|
+
}
|
|
64
|
+
catch (err) {
|
|
65
|
+
return JSON.stringify({ success: false, error: err.message });
|
|
66
|
+
}
|
|
67
|
+
};
|
|
68
|
+
// 简化的 LLM(不使用完整 LLMService,避免依赖)
|
|
69
|
+
// 这里假设有全局的 llm 实例可用
|
|
70
|
+
const llm = getGlobalLLM();
|
|
71
|
+
if (!llm) {
|
|
72
|
+
return {
|
|
73
|
+
success: false,
|
|
74
|
+
content: '',
|
|
75
|
+
error: 'No LLM available for fork subagent',
|
|
76
|
+
duration: Date.now() - startTime,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
// 执行查询
|
|
80
|
+
let finalContent = '';
|
|
81
|
+
let finalUsage;
|
|
82
|
+
const toolCalls = [];
|
|
83
|
+
try {
|
|
84
|
+
for await (const event of (0, framework_1.query)({
|
|
85
|
+
messages,
|
|
86
|
+
tools,
|
|
87
|
+
toolExecutor,
|
|
88
|
+
llm,
|
|
89
|
+
maxTurns,
|
|
90
|
+
streamCallbacks: {
|
|
91
|
+
onChunk: (chunk) => {
|
|
92
|
+
if (!background) {
|
|
93
|
+
// 非后台模式:输出到 stdout
|
|
94
|
+
process.stdout.write(chunk);
|
|
95
|
+
}
|
|
96
|
+
},
|
|
97
|
+
},
|
|
98
|
+
})) {
|
|
99
|
+
switch (event.type) {
|
|
100
|
+
case 'tool_call':
|
|
101
|
+
// 收集工具调用
|
|
102
|
+
break;
|
|
103
|
+
case 'tool_result':
|
|
104
|
+
toolCalls.push({
|
|
105
|
+
name: event.name,
|
|
106
|
+
args: {}, // 简化
|
|
107
|
+
result: event.result,
|
|
108
|
+
});
|
|
109
|
+
break;
|
|
110
|
+
case 'complete':
|
|
111
|
+
finalContent = event.content;
|
|
112
|
+
finalUsage = event.usage;
|
|
113
|
+
break;
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
return {
|
|
117
|
+
success: true,
|
|
118
|
+
content: finalContent,
|
|
119
|
+
toolCalls,
|
|
120
|
+
duration: Date.now() - startTime,
|
|
121
|
+
tokenUsage: finalUsage,
|
|
122
|
+
};
|
|
123
|
+
}
|
|
124
|
+
catch (err) {
|
|
125
|
+
return {
|
|
126
|
+
success: false,
|
|
127
|
+
content: '',
|
|
128
|
+
error: err.message,
|
|
129
|
+
duration: Date.now() - startTime,
|
|
130
|
+
};
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
// ============================================================================
|
|
134
|
+
// 辅助函数
|
|
135
|
+
// ============================================================================
|
|
136
|
+
/**
|
|
137
|
+
* 获取全局 LLM 实例(从 CLI 模块)
|
|
138
|
+
* 注意:这是一个简化实现,实际应该通过参数传递 LLM
|
|
139
|
+
*/
|
|
140
|
+
function getGlobalLLM() {
|
|
141
|
+
// 尝试从全局获取
|
|
142
|
+
try {
|
|
143
|
+
const cliModule = require('../cli');
|
|
144
|
+
if (cliModule.llm) {
|
|
145
|
+
return cliModule.llm;
|
|
146
|
+
}
|
|
147
|
+
}
|
|
148
|
+
catch {
|
|
149
|
+
// ignore
|
|
150
|
+
}
|
|
151
|
+
return null;
|
|
152
|
+
}
|
|
153
|
+
// ============================================================================
|
|
154
|
+
// 防递归保护
|
|
155
|
+
// ============================================================================
|
|
156
|
+
const FORK_BOILERPLATE_TAG = '<FORK_SUBAGENT>';
|
|
157
|
+
/**
|
|
158
|
+
* 检测消息是否来自 Fork Subagent(防止递归 Fork)
|
|
159
|
+
*/
|
|
160
|
+
function isForkSubagentMessage(content) {
|
|
161
|
+
return content.includes(FORK_BOILERPLATE_TAG);
|
|
162
|
+
}
|
|
163
|
+
/**
|
|
164
|
+
* 标记 Fork Subagent 消息
|
|
165
|
+
*/
|
|
166
|
+
function markForkSubagentMessage(content) {
|
|
167
|
+
return `${FORK_BOILERPLATE_TAG}\n${content}`;
|
|
168
|
+
}
|
|
169
|
+
//# sourceMappingURL=fork.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fork.js","sourceRoot":"","sources":["../../src/agents/fork.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;AAoDH,oCAgIC;AAgCD,sDAEC;AAKD,0DAEC;AAxND,4CAA2F;AAC3F,oCAAiC;AAuCjC,+EAA+E;AAC/E,UAAU;AACV,+EAA+E;AAE/E;;GAEG;AACI,KAAK,UAAU,YAAY,CAAC,OAAoB;IACrD,MAAM,EACJ,cAAc,GAAG,IAAI,EACrB,KAAK,GAAG,aAAK,EACb,QAAQ,GAAG,CAAC,EACZ,UAAU,GAAG,KAAK,EAClB,eAAe,EACf,cAAc,GAAG,EAAE,EACnB,kBAAkB,EAClB,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,EACnB,aAAa,GAAG,EAAE,EAClB,aAAa,GAAG,EAAE,GACnB,GAAG,OAAO,CAAC;IAEZ,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,QAAQ,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAEtD,kBAAkB;IAClB,IAAI,QAAmB,CAAC;IAExB,IAAI,cAAc,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,iBAAiB;QACjB,QAAQ,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC;QAC/B,eAAe;QACf,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,eAAe,eAAe,iJAAiJ;SACzL,CAAC,CAAC;IACL,CAAC;SAAM,CAAC;QACN,UAAU;QACV,MAAM,SAAS,GAAkB;YAC/B,GAAG;YACH,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,WAAW,EAAE,OAAO,CAAC,OAAO;YAC5B,KAAK;YACL,aAAa;YACb,aAAa;SACd,CAAC;QACF,MAAM,YAAY,GAAG,kBAAkB,IAAI,IAAA,2BAAe,EAAC,SAAS,CAAC,CAAC;QACtE,QAAQ,GAAG;YACT,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;YACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,EAAE;SAC3C,CAAC;IACJ,CAAC;IAED,QAAQ;IACR,MAAM,YAAY,GAAG,KAAK,EAAE,IAAY,EAAE,IAA6B,EAAE,EAAE;QACzE,aAAa;QACb,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,IAAI,EAAE,EAAE,CAAC,CAAC;QAC5E,CAAC;QACD,gBAAgB;QAChB,MAAM,OAAO,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,CAAC;QAC5E,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YACjD,OAAO,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;IACH,CAAC,CAAC;IAEF,iCAAiC;IACjC,oBAAoB;IACpB,MAAM,GAAG,GAAG,YAAY,EAAE,CAAC;IAC3B,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,OAAO;YACL,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,EAAE;YACX,KAAK,EAAE,oCAAoC;YAC3C,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SACjC,CAAC;IACJ,CAAC;IAED,OAAO;IACP,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,IAAI,UAA0E,CAAC;IAC/E,MAAM,SAAS,GAA2E,EAAE,CAAC;IAE7F,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAA,iBAAK,EAAC;YAC9B,QAAQ;YACR,KAAK;YACL,YAAY;YACZ,GAAG;YACH,QAAQ;YACR,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,IAAI,CAAC,UAAU,EAAE,CAAC;wBAChB,mBAAmB;wBACnB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC9B,CAAC;gBACH,CAAC;aACF;SACF,CAAC,EAAE,CAAC;YACH,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,WAAW;oBACd,SAAS;oBACT,MAAM;gBACR,KAAK,aAAa;oBAChB,SAAS,CAAC,IAAI,CAAC;wBACb,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,IAAI,EAAE,EAAE,EAAG,KAAK;wBAChB,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;oBACH,MAAM;gBACR,KAAK,UAAU;oBACb,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC;oBAC7B,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC;oBACzB,MAAM;YACV,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,YAAY;YACrB,SAAS;YACT,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAChC,UAAU,EAAE,UAAU;SACvB,CAAC;IACJ,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,OAAO,EAAE,EAAE;YACX,KAAK,EAAE,GAAG,CAAC,OAAO;YAClB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;SACjC,CAAC;IACJ,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,OAAO;AACP,+EAA+E;AAE/E;;;GAGG;AACH,SAAS,YAAY;IACnB,UAAU;IACV,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;QACpC,IAAI,SAAS,CAAC,GAAG,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC,GAAG,CAAC;QACvB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,+EAA+E;AAC/E,QAAQ;AACR,+EAA+E;AAE/E,MAAM,oBAAoB,GAAG,iBAAiB,CAAC;AAE/C;;GAEG;AACH,SAAgB,qBAAqB,CAAC,OAAe;IACnD,OAAO,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,SAAgB,uBAAuB,CAAC,OAAe;IACrD,OAAO,GAAG,oBAAoB,KAAK,OAAO,EAAE,CAAC;AAC/C,CAAC"}
|
|
@@ -0,0 +1,72 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* openhorse - Worker Pool
|
|
3
|
+
*
|
|
4
|
+
* 管理多个并发 Subagent,支持任务队列和结果收集。
|
|
5
|
+
*/
|
|
6
|
+
import type { Task } from '../core/agent';
|
|
7
|
+
import { type ForkOptions, type ForkResult } from './fork';
|
|
8
|
+
export interface WorkerInfo {
|
|
9
|
+
id: string;
|
|
10
|
+
status: 'idle' | 'running' | 'completed' | 'failed';
|
|
11
|
+
task?: Task;
|
|
12
|
+
startTime?: number;
|
|
13
|
+
}
|
|
14
|
+
export interface WorkerPoolConfig {
|
|
15
|
+
/** 最大 Worker 数量 */
|
|
16
|
+
maxWorkers?: number;
|
|
17
|
+
/** 单任务超时(ms) */
|
|
18
|
+
taskTimeout?: number;
|
|
19
|
+
/** 默认 ForkOptions */
|
|
20
|
+
defaultForkOptions?: Partial<ForkOptions>;
|
|
21
|
+
}
|
|
22
|
+
export declare class WorkerPool {
|
|
23
|
+
private workers;
|
|
24
|
+
private taskQueue;
|
|
25
|
+
private results;
|
|
26
|
+
private maxWorkers;
|
|
27
|
+
private taskTimeout;
|
|
28
|
+
private defaultForkOptions;
|
|
29
|
+
private nextWorkerId;
|
|
30
|
+
constructor(config?: WorkerPoolConfig);
|
|
31
|
+
/**
|
|
32
|
+
* 提交任务到 Pool
|
|
33
|
+
* @returns Worker ID
|
|
34
|
+
*/
|
|
35
|
+
submit(task: Task, forkOptions?: Partial<ForkOptions>): Promise<ForkResult>;
|
|
36
|
+
/**
|
|
37
|
+
* 批量提交任务
|
|
38
|
+
*/
|
|
39
|
+
submitBatch(tasks: Task[], forkOptions?: Partial<ForkOptions>): Promise<Map<string, ForkResult>>;
|
|
40
|
+
/**
|
|
41
|
+
* 广播消息到所有 Worker
|
|
42
|
+
*/
|
|
43
|
+
broadcast(message: string): void;
|
|
44
|
+
/**
|
|
45
|
+
* 收集所有结果
|
|
46
|
+
*/
|
|
47
|
+
collectResults(): Map<string, ForkResult>;
|
|
48
|
+
/**
|
|
49
|
+
* 清除已完成的结果
|
|
50
|
+
*/
|
|
51
|
+
clearResults(): void;
|
|
52
|
+
/**
|
|
53
|
+
* 获取 Pool 状态
|
|
54
|
+
*/
|
|
55
|
+
getStatus(): {
|
|
56
|
+
totalWorkers: number;
|
|
57
|
+
runningWorkers: number;
|
|
58
|
+
idleWorkers: number;
|
|
59
|
+
queueLength: number;
|
|
60
|
+
completedTasks: number;
|
|
61
|
+
};
|
|
62
|
+
/**
|
|
63
|
+
* 停止所有 Worker
|
|
64
|
+
*/
|
|
65
|
+
stopAll(): void;
|
|
66
|
+
private getIdleWorkers;
|
|
67
|
+
private allocateWorker;
|
|
68
|
+
private executeTask;
|
|
69
|
+
}
|
|
70
|
+
export declare function getWorkerPool(config?: WorkerPoolConfig): WorkerPool;
|
|
71
|
+
export declare function resetWorkerPool(): void;
|
|
72
|
+
//# sourceMappingURL=worker-pool.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker-pool.d.ts","sourceRoot":"","sources":["../../src/agents/worker-pool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAC1C,OAAO,EAAgB,KAAK,WAAW,EAAE,KAAK,UAAU,EAAE,MAAM,QAAQ,CAAC;AAMzE,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;IACpD,IAAI,CAAC,EAAE,IAAI,CAAC;IACZ,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,gBAAgB;IAC/B,mBAAmB;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,qBAAqB;IACrB,kBAAkB,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;CAC3C;AAMD,qBAAa,UAAU;IACrB,OAAO,CAAC,OAAO,CAAsC;IACrD,OAAO,CAAC,SAAS,CAAc;IAC/B,OAAO,CAAC,OAAO,CAAsC;IACrD,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,kBAAkB,CAAuB;IACjD,OAAO,CAAC,YAAY,CAAK;gBAEb,MAAM,CAAC,EAAE,gBAAgB;IAMrC;;;OAGG;IACG,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC;IAqCjF;;OAEG;IACG,WAAW,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,WAAW,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAetG;;OAEG;IACH,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAShC;;OAEG;IACH,cAAc,IAAI,GAAG,CAAC,MAAM,EAAE,UAAU,CAAC;IAIzC;;OAEG;IACH,YAAY,IAAI,IAAI;IAQpB;;OAEG;IACH,SAAS,IAAI;QACX,YAAY,EAAE,MAAM,CAAC;QACrB,cAAc,EAAE,MAAM,CAAC;QACvB,WAAW,EAAE,MAAM,CAAC;QACpB,WAAW,EAAE,MAAM,CAAC;QACpB,cAAc,EAAE,MAAM,CAAC;KACxB;IAUD;;OAEG;IACH,OAAO,IAAI,IAAI;IAaf,OAAO,CAAC,cAAc;IAItB,OAAO,CAAC,cAAc;YAyBR,WAAW;CAsD1B;AAQD,wBAAgB,aAAa,CAAC,MAAM,CAAC,EAAE,gBAAgB,GAAG,UAAU,CAKnE;AAED,wBAAgB,eAAe,IAAI,IAAI,CAEtC"}
|