@specforge/mcp 2.6.0 → 3.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +73 -0
- package/bin/{specforge-mcp → specforge} +0 -5
- package/dist/ai-provider/circuit-breaker.d.ts +63 -0
- package/dist/ai-provider/circuit-breaker.d.ts.map +1 -0
- package/dist/ai-provider/circuit-breaker.js +160 -0
- package/dist/ai-provider/circuit-breaker.js.map +1 -0
- package/dist/ai-provider/cli-version.d.ts +50 -0
- package/dist/ai-provider/cli-version.d.ts.map +1 -0
- package/dist/ai-provider/cli-version.js +141 -0
- package/dist/ai-provider/cli-version.js.map +1 -0
- package/dist/ai-provider/config-loader.d.ts +45 -0
- package/dist/ai-provider/config-loader.d.ts.map +1 -0
- package/dist/ai-provider/config-loader.js +106 -0
- package/dist/ai-provider/config-loader.js.map +1 -0
- package/dist/ai-provider/errors.d.ts +48 -0
- package/dist/ai-provider/errors.d.ts.map +1 -0
- package/dist/ai-provider/errors.js +102 -0
- package/dist/ai-provider/errors.js.map +1 -0
- package/dist/ai-provider/events.d.ts +73 -0
- package/dist/ai-provider/events.d.ts.map +1 -0
- package/dist/ai-provider/events.js +75 -0
- package/dist/ai-provider/events.js.map +1 -0
- package/dist/ai-provider/factory.d.ts +31 -0
- package/dist/ai-provider/factory.d.ts.map +1 -0
- package/dist/ai-provider/factory.js +100 -0
- package/dist/ai-provider/factory.js.map +1 -0
- package/dist/ai-provider/index.d.ts +24 -0
- package/dist/ai-provider/index.d.ts.map +1 -0
- package/dist/ai-provider/index.js +46 -0
- package/dist/ai-provider/index.js.map +1 -0
- package/dist/ai-provider/instance-coordinator.d.ts +54 -0
- package/dist/ai-provider/instance-coordinator.d.ts.map +1 -0
- package/dist/ai-provider/instance-coordinator.js +199 -0
- package/dist/ai-provider/instance-coordinator.js.map +1 -0
- package/dist/ai-provider/jsonl-parser.d.ts +43 -0
- package/dist/ai-provider/jsonl-parser.d.ts.map +1 -0
- package/dist/ai-provider/jsonl-parser.js +107 -0
- package/dist/ai-provider/jsonl-parser.js.map +1 -0
- package/dist/ai-provider/lifecycle.d.ts +50 -0
- package/dist/ai-provider/lifecycle.d.ts.map +1 -0
- package/dist/ai-provider/lifecycle.js +145 -0
- package/dist/ai-provider/lifecycle.js.map +1 -0
- package/dist/ai-provider/logger.d.ts +69 -0
- package/dist/ai-provider/logger.d.ts.map +1 -0
- package/dist/ai-provider/logger.js +161 -0
- package/dist/ai-provider/logger.js.map +1 -0
- package/dist/ai-provider/metrics.d.ts +91 -0
- package/dist/ai-provider/metrics.d.ts.map +1 -0
- package/dist/ai-provider/metrics.js +187 -0
- package/dist/ai-provider/metrics.js.map +1 -0
- package/dist/ai-provider/process-manager.d.ts +97 -0
- package/dist/ai-provider/process-manager.d.ts.map +1 -0
- package/dist/ai-provider/process-manager.js +477 -0
- package/dist/ai-provider/process-manager.js.map +1 -0
- package/dist/ai-provider/providers/claude-code.d.ts +64 -0
- package/dist/ai-provider/providers/claude-code.d.ts.map +1 -0
- package/dist/ai-provider/providers/claude-code.js +205 -0
- package/dist/ai-provider/providers/claude-code.js.map +1 -0
- package/dist/ai-provider/retry-executor.d.ts +52 -0
- package/dist/ai-provider/retry-executor.d.ts.map +1 -0
- package/dist/ai-provider/retry-executor.js +138 -0
- package/dist/ai-provider/retry-executor.js.map +1 -0
- package/dist/ai-provider/safe-args.d.ts +58 -0
- package/dist/ai-provider/safe-args.d.ts.map +1 -0
- package/dist/ai-provider/safe-args.js +176 -0
- package/dist/ai-provider/safe-args.js.map +1 -0
- package/dist/ai-provider/semaphore.d.ts +50 -0
- package/dist/ai-provider/semaphore.d.ts.map +1 -0
- package/dist/ai-provider/semaphore.js +97 -0
- package/dist/ai-provider/semaphore.js.map +1 -0
- package/dist/ai-provider/tracer.d.ts +67 -0
- package/dist/ai-provider/tracer.d.ts.map +1 -0
- package/dist/ai-provider/tracer.js +209 -0
- package/dist/ai-provider/tracer.js.map +1 -0
- package/dist/ai-provider/types.d.ts +181 -0
- package/dist/ai-provider/types.d.ts.map +1 -0
- package/dist/ai-provider/types.js +8 -0
- package/dist/ai-provider/types.js.map +1 -0
- package/dist/autopilot/agents/agent-runner.d.ts +109 -0
- package/dist/autopilot/agents/agent-runner.d.ts.map +1 -0
- package/dist/autopilot/agents/agent-runner.js +731 -0
- package/dist/autopilot/agents/agent-runner.js.map +1 -0
- package/dist/autopilot/agents/agent-selector.d.ts +59 -0
- package/dist/autopilot/agents/agent-selector.d.ts.map +1 -0
- package/dist/autopilot/agents/agent-selector.js +234 -0
- package/dist/autopilot/agents/agent-selector.js.map +1 -0
- package/dist/autopilot/agents/model-selector.d.ts +49 -0
- package/dist/autopilot/agents/model-selector.d.ts.map +1 -0
- package/dist/autopilot/agents/model-selector.js +62 -0
- package/dist/autopilot/agents/model-selector.js.map +1 -0
- package/dist/autopilot/agents/profiles/builtin.d.ts +55 -0
- package/dist/autopilot/agents/profiles/builtin.d.ts.map +1 -0
- package/dist/autopilot/agents/profiles/builtin.js +323 -0
- package/dist/autopilot/agents/profiles/builtin.js.map +1 -0
- package/dist/autopilot/agents/profiles/types.d.ts +98 -0
- package/dist/autopilot/agents/profiles/types.d.ts.map +1 -0
- package/dist/autopilot/agents/profiles/types.js +17 -0
- package/dist/autopilot/agents/profiles/types.js.map +1 -0
- package/dist/autopilot/api/autopilot-api-client.d.ts +217 -0
- package/dist/autopilot/api/autopilot-api-client.d.ts.map +1 -0
- package/dist/autopilot/api/autopilot-api-client.js +402 -0
- package/dist/autopilot/api/autopilot-api-client.js.map +1 -0
- package/dist/autopilot/cli/abort.d.ts +20 -0
- package/dist/autopilot/cli/abort.d.ts.map +1 -0
- package/dist/autopilot/cli/abort.js +201 -0
- package/dist/autopilot/cli/abort.js.map +1 -0
- package/dist/autopilot/cli/display.d.ts +63 -0
- package/dist/autopilot/cli/display.d.ts.map +1 -0
- package/dist/autopilot/cli/display.js +260 -0
- package/dist/autopilot/cli/display.js.map +1 -0
- package/dist/autopilot/cli/index.d.ts +24 -0
- package/dist/autopilot/cli/index.d.ts.map +1 -0
- package/dist/autopilot/cli/index.js +79 -0
- package/dist/autopilot/cli/index.js.map +1 -0
- package/dist/autopilot/cli/pause.d.ts +18 -0
- package/dist/autopilot/cli/pause.d.ts.map +1 -0
- package/dist/autopilot/cli/pause.js +110 -0
- package/dist/autopilot/cli/pause.js.map +1 -0
- package/dist/autopilot/cli/resume.d.ts +22 -0
- package/dist/autopilot/cli/resume.d.ts.map +1 -0
- package/dist/autopilot/cli/resume.js +172 -0
- package/dist/autopilot/cli/resume.js.map +1 -0
- package/dist/autopilot/cli/run.d.ts +25 -0
- package/dist/autopilot/cli/run.d.ts.map +1 -0
- package/dist/autopilot/cli/run.js +220 -0
- package/dist/autopilot/cli/run.js.map +1 -0
- package/dist/autopilot/cli/status.d.ts +20 -0
- package/dist/autopilot/cli/status.d.ts.map +1 -0
- package/dist/autopilot/cli/status.js +217 -0
- package/dist/autopilot/cli/status.js.map +1 -0
- package/dist/autopilot/config.d.ts +45 -0
- package/dist/autopilot/config.d.ts.map +1 -0
- package/dist/autopilot/config.js +269 -0
- package/dist/autopilot/config.js.map +1 -0
- package/dist/autopilot/core/dependency-resolver.d.ts +108 -0
- package/dist/autopilot/core/dependency-resolver.d.ts.map +1 -0
- package/dist/autopilot/core/dependency-resolver.js +394 -0
- package/dist/autopilot/core/dependency-resolver.js.map +1 -0
- package/dist/autopilot/core/dispatcher.d.ts +215 -0
- package/dist/autopilot/core/dispatcher.d.ts.map +1 -0
- package/dist/autopilot/core/dispatcher.js +594 -0
- package/dist/autopilot/core/dispatcher.js.map +1 -0
- package/dist/autopilot/core/failure-handler.d.ts +145 -0
- package/dist/autopilot/core/failure-handler.d.ts.map +1 -0
- package/dist/autopilot/core/failure-handler.js +308 -0
- package/dist/autopilot/core/failure-handler.js.map +1 -0
- package/dist/autopilot/core/rate-limit-handler.d.ts +108 -0
- package/dist/autopilot/core/rate-limit-handler.d.ts.map +1 -0
- package/dist/autopilot/core/rate-limit-handler.js +195 -0
- package/dist/autopilot/core/rate-limit-handler.js.map +1 -0
- package/dist/autopilot/core/state-manager.d.ts +160 -0
- package/dist/autopilot/core/state-manager.d.ts.map +1 -0
- package/dist/autopilot/core/state-manager.js +393 -0
- package/dist/autopilot/core/state-manager.js.map +1 -0
- package/dist/autopilot/core/timeout-manager.d.ts +95 -0
- package/dist/autopilot/core/timeout-manager.d.ts.map +1 -0
- package/dist/autopilot/core/timeout-manager.js +188 -0
- package/dist/autopilot/core/timeout-manager.js.map +1 -0
- package/dist/autopilot/git/branch-manager.d.ts +117 -0
- package/dist/autopilot/git/branch-manager.d.ts.map +1 -0
- package/dist/autopilot/git/branch-manager.js +238 -0
- package/dist/autopilot/git/branch-manager.js.map +1 -0
- package/dist/autopilot/git/index.d.ts +9 -0
- package/dist/autopilot/git/index.d.ts.map +1 -0
- package/dist/autopilot/git/index.js +9 -0
- package/dist/autopilot/git/index.js.map +1 -0
- package/dist/autopilot/git/merge-manager.d.ts +118 -0
- package/dist/autopilot/git/merge-manager.d.ts.map +1 -0
- package/dist/autopilot/git/merge-manager.js +304 -0
- package/dist/autopilot/git/merge-manager.js.map +1 -0
- package/dist/autopilot/git/worktree-manager.d.ts +128 -0
- package/dist/autopilot/git/worktree-manager.d.ts.map +1 -0
- package/dist/autopilot/git/worktree-manager.js +298 -0
- package/dist/autopilot/git/worktree-manager.js.map +1 -0
- package/dist/autopilot/index.d.ts +30 -0
- package/dist/autopilot/index.d.ts.map +1 -0
- package/dist/autopilot/index.js +55 -0
- package/dist/autopilot/index.js.map +1 -0
- package/dist/autopilot/sync/index.d.ts +7 -0
- package/dist/autopilot/sync/index.d.ts.map +1 -0
- package/dist/autopilot/sync/index.js +7 -0
- package/dist/autopilot/sync/index.js.map +1 -0
- package/dist/autopilot/sync/sync-manager.d.ts +168 -0
- package/dist/autopilot/sync/sync-manager.d.ts.map +1 -0
- package/dist/autopilot/sync/sync-manager.js +303 -0
- package/dist/autopilot/sync/sync-manager.js.map +1 -0
- package/dist/autopilot/types.d.ts +454 -0
- package/dist/autopilot/types.d.ts.map +1 -0
- package/dist/autopilot/types.js +26 -0
- package/dist/autopilot/types.js.map +1 -0
- package/dist/autopilot/utils/audit-logger.d.ts +176 -0
- package/dist/autopilot/utils/audit-logger.d.ts.map +1 -0
- package/dist/autopilot/utils/audit-logger.js +308 -0
- package/dist/autopilot/utils/audit-logger.js.map +1 -0
- package/dist/autopilot/utils/cost-tracker.d.ts +162 -0
- package/dist/autopilot/utils/cost-tracker.d.ts.map +1 -0
- package/dist/autopilot/utils/cost-tracker.js +269 -0
- package/dist/autopilot/utils/cost-tracker.js.map +1 -0
- package/dist/autopilot/utils/index.d.ts +9 -0
- package/dist/autopilot/utils/index.d.ts.map +1 -0
- package/dist/autopilot/utils/index.js +9 -0
- package/dist/autopilot/utils/index.js.map +1 -0
- package/dist/autopilot/utils/progress-reporter.d.ts +132 -0
- package/dist/autopilot/utils/progress-reporter.d.ts.map +1 -0
- package/dist/autopilot/utils/progress-reporter.js +290 -0
- package/dist/autopilot/utils/progress-reporter.js.map +1 -0
- package/dist/autopilot/worker/worker-pool.d.ts +179 -0
- package/dist/autopilot/worker/worker-pool.d.ts.map +1 -0
- package/dist/autopilot/worker/worker-pool.js +331 -0
- package/dist/autopilot/worker/worker-pool.js.map +1 -0
- package/dist/autopilot/worker/worker-session.d.ts +171 -0
- package/dist/autopilot/worker/worker-session.d.ts.map +1 -0
- package/dist/autopilot/worker/worker-session.js +295 -0
- package/dist/autopilot/worker/worker-session.js.map +1 -0
- package/dist/cli/index.d.ts +1 -1
- package/dist/cli/index.d.ts.map +1 -1
- package/dist/cli/index.js +4 -1
- package/dist/cli/index.js.map +1 -1
- package/dist/index.js +0 -1
- package/dist/index.js.map +1 -1
- package/dist/tools/core/epic.js +1 -1
- package/dist/tools/core/epic.js.map +1 -1
- package/dist/tools/core/lookup.d.ts.map +1 -1
- package/dist/tools/core/lookup.js +3 -2
- package/dist/tools/core/lookup.js.map +1 -1
- package/dist/tools/core/specification.js +1 -1
- package/dist/tools/core/specification.js.map +1 -1
- package/dist/tools/core/ticket.d.ts.map +1 -1
- package/dist/tools/core/ticket.js +4 -6
- package/dist/tools/core/ticket.js.map +1 -1
- package/dist/tools/index.d.ts.map +1 -1
- package/dist/tools/index.js +60 -0
- package/dist/tools/index.js.map +1 -1
- package/dist/validation/index.d.ts.map +1 -1
- package/dist/validation/index.js +4 -1
- package/dist/validation/index.js.map +1 -1
- package/package.json +8 -4
|
@@ -0,0 +1,331 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Worker Pool
|
|
3
|
+
*
|
|
4
|
+
* Manages multiple worker sessions with shared AI Provider.
|
|
5
|
+
*/
|
|
6
|
+
import { EventEmitter } from 'events';
|
|
7
|
+
import { createAIProvider } from '../../ai-provider/index.js';
|
|
8
|
+
import { WorkerSession } from './worker-session.js';
|
|
9
|
+
import { createAgentRunner } from '../agents/agent-runner.js';
|
|
10
|
+
import { createAgentSelector } from '../agents/agent-selector.js';
|
|
11
|
+
/**
|
|
12
|
+
* Worker Pool
|
|
13
|
+
*
|
|
14
|
+
* Manages worker sessions with a shared AI Provider instance.
|
|
15
|
+
*/
|
|
16
|
+
export class WorkerPool extends EventEmitter {
|
|
17
|
+
config;
|
|
18
|
+
workingDirectory;
|
|
19
|
+
worktreeManager;
|
|
20
|
+
aiProvider;
|
|
21
|
+
agentSelector;
|
|
22
|
+
workers = new Map();
|
|
23
|
+
initialized = false;
|
|
24
|
+
shuttingDown = false;
|
|
25
|
+
constructor(options) {
|
|
26
|
+
super();
|
|
27
|
+
this.config = options.config;
|
|
28
|
+
this.workingDirectory = options.workingDirectory;
|
|
29
|
+
this.worktreeManager = options.worktreeManager;
|
|
30
|
+
this.agentSelector = createAgentSelector(options.config.agentProfiles);
|
|
31
|
+
}
|
|
32
|
+
/**
|
|
33
|
+
* Initialize the worker pool
|
|
34
|
+
*/
|
|
35
|
+
async initialize() {
|
|
36
|
+
if (this.initialized)
|
|
37
|
+
return;
|
|
38
|
+
// Create shared AI Provider
|
|
39
|
+
const aiConfig = {
|
|
40
|
+
mode: 'claude_code',
|
|
41
|
+
maxParallelProcesses: this.config.aiProvider.maxParallelProcesses,
|
|
42
|
+
requestTimeoutSeconds: this.config.aiProvider.requestTimeoutSeconds,
|
|
43
|
+
maxRetries: this.config.aiProvider.maxRetries,
|
|
44
|
+
workingDirectory: this.workingDirectory,
|
|
45
|
+
};
|
|
46
|
+
this.aiProvider = await createAIProvider(aiConfig);
|
|
47
|
+
// Create workers
|
|
48
|
+
for (let i = 0; i < this.config.maxParallelWorkers; i++) {
|
|
49
|
+
await this.createWorker(i);
|
|
50
|
+
}
|
|
51
|
+
this.initialized = true;
|
|
52
|
+
this.emit('pool:ready', { workerCount: this.workers.size });
|
|
53
|
+
}
|
|
54
|
+
/**
|
|
55
|
+
* Create a single worker
|
|
56
|
+
*/
|
|
57
|
+
async createWorker(workerId) {
|
|
58
|
+
// Determine worktree path and branch
|
|
59
|
+
const worktreePath = this.worktreeManager
|
|
60
|
+
? await this.worktreeManager.getWorkerPath(workerId)
|
|
61
|
+
: `${this.workingDirectory}/.worktrees/worker-${workerId}`;
|
|
62
|
+
const branchName = `autopilot/worker-${workerId}`;
|
|
63
|
+
// Create worker session
|
|
64
|
+
const session = new WorkerSession(workerId, worktreePath, branchName);
|
|
65
|
+
// Create agent runner for this worker
|
|
66
|
+
const runner = createAgentRunner({
|
|
67
|
+
aiProvider: this.aiProvider,
|
|
68
|
+
workingDirectory: worktreePath,
|
|
69
|
+
agentSelector: this.agentSelector,
|
|
70
|
+
autoCommit: this.config.autoCommit,
|
|
71
|
+
timeout: this.config.ticketTimeoutMs,
|
|
72
|
+
});
|
|
73
|
+
session.setAgentRunner(runner);
|
|
74
|
+
// Forward events
|
|
75
|
+
session.on('status:changed', (data) => {
|
|
76
|
+
if (data.status === 'idle') {
|
|
77
|
+
this.emit('worker:idle', { workerId: data.workerId });
|
|
78
|
+
}
|
|
79
|
+
});
|
|
80
|
+
session.on('ticket:started', (data) => {
|
|
81
|
+
this.emit('worker:started', data);
|
|
82
|
+
});
|
|
83
|
+
session.on('ticket:completed', (data) => {
|
|
84
|
+
this.emit('worker:completed', data);
|
|
85
|
+
});
|
|
86
|
+
session.on('ticket:failed', (data) => {
|
|
87
|
+
this.emit('worker:failed', data);
|
|
88
|
+
});
|
|
89
|
+
this.workers.set(workerId, session);
|
|
90
|
+
this.emit('worker:created', { workerId });
|
|
91
|
+
return session;
|
|
92
|
+
}
|
|
93
|
+
/**
|
|
94
|
+
* Get an idle worker
|
|
95
|
+
*/
|
|
96
|
+
getIdleWorker() {
|
|
97
|
+
for (const worker of this.workers.values()) {
|
|
98
|
+
if (worker.isIdle()) {
|
|
99
|
+
return worker;
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
return undefined;
|
|
103
|
+
}
|
|
104
|
+
/**
|
|
105
|
+
* Get all idle workers
|
|
106
|
+
*/
|
|
107
|
+
getIdleWorkers() {
|
|
108
|
+
return [...this.workers.values()].filter(w => w.isIdle());
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* Get all busy workers
|
|
112
|
+
*/
|
|
113
|
+
getBusyWorkers() {
|
|
114
|
+
return [...this.workers.values()].filter(w => w.isBusy());
|
|
115
|
+
}
|
|
116
|
+
/**
|
|
117
|
+
* Get worker by ID
|
|
118
|
+
*/
|
|
119
|
+
getWorker(workerId) {
|
|
120
|
+
return this.workers.get(workerId);
|
|
121
|
+
}
|
|
122
|
+
/**
|
|
123
|
+
* Get all workers
|
|
124
|
+
*/
|
|
125
|
+
getAllWorkers() {
|
|
126
|
+
return [...this.workers.values()];
|
|
127
|
+
}
|
|
128
|
+
/**
|
|
129
|
+
* Get worker count
|
|
130
|
+
*/
|
|
131
|
+
getWorkerCount() {
|
|
132
|
+
return this.workers.size;
|
|
133
|
+
}
|
|
134
|
+
/**
|
|
135
|
+
* Get idle worker count
|
|
136
|
+
*/
|
|
137
|
+
getIdleCount() {
|
|
138
|
+
return this.getIdleWorkers().length;
|
|
139
|
+
}
|
|
140
|
+
/**
|
|
141
|
+
* Get busy worker count
|
|
142
|
+
*/
|
|
143
|
+
getBusyCount() {
|
|
144
|
+
return this.getBusyWorkers().length;
|
|
145
|
+
}
|
|
146
|
+
/**
|
|
147
|
+
* Assign ticket to idle worker
|
|
148
|
+
*/
|
|
149
|
+
async assignTicket(context) {
|
|
150
|
+
const worker = this.getIdleWorker();
|
|
151
|
+
if (!worker) {
|
|
152
|
+
return null;
|
|
153
|
+
}
|
|
154
|
+
const workerId = worker.getWorkerId();
|
|
155
|
+
const promise = worker.process(context);
|
|
156
|
+
return { workerId, promise };
|
|
157
|
+
}
|
|
158
|
+
/**
|
|
159
|
+
* Process ticket with specific worker
|
|
160
|
+
*/
|
|
161
|
+
async processWithWorker(workerId, context) {
|
|
162
|
+
const worker = this.workers.get(workerId);
|
|
163
|
+
if (!worker) {
|
|
164
|
+
throw new Error(`Worker ${workerId} not found`);
|
|
165
|
+
}
|
|
166
|
+
return worker.process(context);
|
|
167
|
+
}
|
|
168
|
+
/**
|
|
169
|
+
* Pause all workers
|
|
170
|
+
*/
|
|
171
|
+
pauseAll() {
|
|
172
|
+
for (const worker of this.workers.values()) {
|
|
173
|
+
worker.pause();
|
|
174
|
+
}
|
|
175
|
+
}
|
|
176
|
+
/**
|
|
177
|
+
* Resume all workers
|
|
178
|
+
*/
|
|
179
|
+
resumeAll() {
|
|
180
|
+
for (const worker of this.workers.values()) {
|
|
181
|
+
worker.resume();
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
/**
|
|
185
|
+
* Cancel all active work
|
|
186
|
+
*/
|
|
187
|
+
cancelAll() {
|
|
188
|
+
for (const worker of this.workers.values()) {
|
|
189
|
+
worker.cancel();
|
|
190
|
+
}
|
|
191
|
+
}
|
|
192
|
+
/**
|
|
193
|
+
* Get worker states
|
|
194
|
+
*/
|
|
195
|
+
getWorkerStates() {
|
|
196
|
+
return [...this.workers.values()].map(w => w.getState());
|
|
197
|
+
}
|
|
198
|
+
/**
|
|
199
|
+
* Get pool statistics
|
|
200
|
+
*/
|
|
201
|
+
getStats() {
|
|
202
|
+
let idle = 0;
|
|
203
|
+
let busy = 0;
|
|
204
|
+
let paused = 0;
|
|
205
|
+
let completed = 0;
|
|
206
|
+
let failed = 0;
|
|
207
|
+
let totalCost = 0;
|
|
208
|
+
for (const worker of this.workers.values()) {
|
|
209
|
+
const stats = worker.getStats();
|
|
210
|
+
completed += stats.completed;
|
|
211
|
+
failed += stats.failed;
|
|
212
|
+
totalCost += stats.cost;
|
|
213
|
+
switch (stats.status) {
|
|
214
|
+
case 'idle':
|
|
215
|
+
idle++;
|
|
216
|
+
break;
|
|
217
|
+
case 'paused':
|
|
218
|
+
paused++;
|
|
219
|
+
break;
|
|
220
|
+
default:
|
|
221
|
+
busy++;
|
|
222
|
+
}
|
|
223
|
+
}
|
|
224
|
+
return {
|
|
225
|
+
total: this.workers.size,
|
|
226
|
+
idle,
|
|
227
|
+
busy,
|
|
228
|
+
paused,
|
|
229
|
+
completed,
|
|
230
|
+
failed,
|
|
231
|
+
totalCost,
|
|
232
|
+
};
|
|
233
|
+
}
|
|
234
|
+
/**
|
|
235
|
+
* Wait for all workers to complete
|
|
236
|
+
*/
|
|
237
|
+
async waitForAll() {
|
|
238
|
+
const busyWorkers = this.getBusyWorkers();
|
|
239
|
+
if (busyWorkers.length === 0)
|
|
240
|
+
return;
|
|
241
|
+
await Promise.all(busyWorkers.map(worker => new Promise((resolve) => {
|
|
242
|
+
if (worker.isIdle()) {
|
|
243
|
+
resolve();
|
|
244
|
+
return;
|
|
245
|
+
}
|
|
246
|
+
const handler = (data) => {
|
|
247
|
+
if (data.workerId === worker.getWorkerId()) {
|
|
248
|
+
worker.off('status:changed', handler);
|
|
249
|
+
resolve();
|
|
250
|
+
}
|
|
251
|
+
};
|
|
252
|
+
worker.on('status:changed', handler);
|
|
253
|
+
})));
|
|
254
|
+
}
|
|
255
|
+
/**
|
|
256
|
+
* Wait for any worker to become idle
|
|
257
|
+
*/
|
|
258
|
+
async waitForIdle() {
|
|
259
|
+
const idle = this.getIdleWorker();
|
|
260
|
+
if (idle)
|
|
261
|
+
return idle;
|
|
262
|
+
return new Promise((resolve) => {
|
|
263
|
+
const handler = (data) => {
|
|
264
|
+
const worker = this.workers.get(data.workerId);
|
|
265
|
+
if (worker && worker.isIdle()) {
|
|
266
|
+
this.off('worker:idle', handler);
|
|
267
|
+
resolve(worker);
|
|
268
|
+
}
|
|
269
|
+
};
|
|
270
|
+
this.on('worker:idle', handler);
|
|
271
|
+
});
|
|
272
|
+
}
|
|
273
|
+
/**
|
|
274
|
+
* Shutdown the worker pool
|
|
275
|
+
*/
|
|
276
|
+
async shutdown() {
|
|
277
|
+
if (this.shuttingDown)
|
|
278
|
+
return;
|
|
279
|
+
this.shuttingDown = true;
|
|
280
|
+
// Cancel all active work
|
|
281
|
+
this.cancelAll();
|
|
282
|
+
// Wait for workers to finish
|
|
283
|
+
await this.waitForAll();
|
|
284
|
+
// Cleanup workers
|
|
285
|
+
for (const [workerId, worker] of this.workers) {
|
|
286
|
+
worker.reset();
|
|
287
|
+
worker.removeAllListeners();
|
|
288
|
+
this.emit('worker:destroyed', { workerId });
|
|
289
|
+
}
|
|
290
|
+
this.workers.clear();
|
|
291
|
+
// Shutdown AI Provider
|
|
292
|
+
if (this.aiProvider) {
|
|
293
|
+
await this.aiProvider.shutdown();
|
|
294
|
+
this.aiProvider = undefined;
|
|
295
|
+
}
|
|
296
|
+
this.initialized = false;
|
|
297
|
+
this.shuttingDown = false;
|
|
298
|
+
this.emit('pool:shutdown');
|
|
299
|
+
}
|
|
300
|
+
/**
|
|
301
|
+
* Get AI Provider
|
|
302
|
+
*/
|
|
303
|
+
getAIProvider() {
|
|
304
|
+
return this.aiProvider;
|
|
305
|
+
}
|
|
306
|
+
/**
|
|
307
|
+
* Get agent selector
|
|
308
|
+
*/
|
|
309
|
+
getAgentSelector() {
|
|
310
|
+
return this.agentSelector;
|
|
311
|
+
}
|
|
312
|
+
/**
|
|
313
|
+
* Check if initialized
|
|
314
|
+
*/
|
|
315
|
+
isInitialized() {
|
|
316
|
+
return this.initialized;
|
|
317
|
+
}
|
|
318
|
+
/**
|
|
319
|
+
* Check if shutting down
|
|
320
|
+
*/
|
|
321
|
+
isShuttingDown() {
|
|
322
|
+
return this.shuttingDown;
|
|
323
|
+
}
|
|
324
|
+
}
|
|
325
|
+
/**
|
|
326
|
+
* Create a worker pool
|
|
327
|
+
*/
|
|
328
|
+
export function createWorkerPool(options) {
|
|
329
|
+
return new WorkerPool(options);
|
|
330
|
+
}
|
|
331
|
+
//# sourceMappingURL=worker-pool.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker-pool.js","sourceRoot":"","sources":["../../../src/autopilot/worker/worker-pool.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAC;AAO9D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAe,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAC3E,OAAO,EAAiB,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AA0BjF;;;;GAIG;AACH,MAAM,OAAO,UAAW,SAAQ,YAAY;IAClC,MAAM,CAAkB;IACxB,gBAAgB,CAAS;IACzB,eAAe,CAAmB;IAClC,UAAU,CAAc;IACxB,aAAa,CAAgB;IAC7B,OAAO,GAA+B,IAAI,GAAG,EAAE,CAAC;IAChD,WAAW,GAAY,KAAK,CAAC;IAC7B,YAAY,GAAY,KAAK,CAAC;IAEtC,YAAY,OAA0B;QACpC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QACjD,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;QAC/C,IAAI,CAAC,aAAa,GAAG,mBAAmB,CAAC,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAE7B,4BAA4B;QAC5B,MAAM,QAAQ,GAAqB;YACjC,IAAI,EAAE,aAAa;YACnB,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,oBAAoB;YACjE,qBAAqB,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,qBAAqB;YACnE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU;YAC7C,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;SACxC,CAAC;QAEF,IAAI,CAAC,UAAU,GAAG,MAAM,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAEnD,iBAAiB;QACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC,EAAE,EAAE,CAAC;YACxD,MAAM,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAExB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9D,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CAAC,QAAgB;QACzC,qCAAqC;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe;YACvC,CAAC,CAAC,MAAM,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,QAAQ,CAAC;YACpD,CAAC,CAAC,GAAG,IAAI,CAAC,gBAAgB,sBAAsB,QAAQ,EAAE,CAAC;QAE7D,MAAM,UAAU,GAAG,oBAAoB,QAAQ,EAAE,CAAC;QAElD,wBAAwB;QACxB,MAAM,OAAO,GAAG,IAAI,aAAa,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;QAEtE,sCAAsC;QACtC,MAAM,MAAM,GAAG,iBAAiB,CAAC;YAC/B,UAAU,EAAE,IAAI,CAAC,UAAW;YAC5B,gBAAgB,EAAE,YAAY;YAC9B,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;YAClC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;SACrC,CAAC,CAAC;QAEH,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAE/B,iBAAiB;QACjB,OAAO,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,IAAI,EAAE,EAAE;YACpC,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;gBAC3B,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,gBAAgB,EAAE,CAAC,IAAI,EAAE,EAAE;YACpC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,kBAAkB,EAAE,CAAC,IAAI,EAAE,EAAE;YACtC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,IAAI,EAAE,EAAE;YACnC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEpC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE1C,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,aAAa;QACX,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBACpB,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAAgB;QACxB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACpC,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC,MAAM,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,YAAY;QACV,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC,MAAM,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,OAAsB;QAIvC,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QACpC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAExC,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CACrB,QAAgB,EAChB,OAAsB;QAEtB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,UAAU,QAAQ,YAAY,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS;QACP,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS;QACP,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,QAAQ;QASN,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YAChC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;YACvB,SAAS,IAAI,KAAK,CAAC,IAAI,CAAC;YAExB,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;gBACrB,KAAK,MAAM;oBACT,IAAI,EAAE,CAAC;oBACP,MAAM;gBACR,KAAK,QAAQ;oBACX,MAAM,EAAE,CAAC;oBACT,MAAM;gBACR;oBACE,IAAI,EAAE,CAAC;YACX,CAAC;QACH,CAAC;QAED,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YACxB,IAAI;YACJ,IAAI;YACJ,MAAM;YACN,SAAS;YACT,MAAM;YACN,SAAS;SACV,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAC1C,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAErC,MAAM,OAAO,CAAC,GAAG,CACf,WAAW,CAAC,GAAG,CACb,MAAM,CAAC,EAAE,CACP,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAC5B,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;gBACpB,OAAO,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,CAAC,IAA0B,EAAE,EAAE;gBAC7C,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;oBAC3C,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;oBACtC,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,CAAC,EAAE,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;QACvC,CAAC,CAAC,CACL,CACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,EAAE,CAAC;QAClC,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;QAEtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,OAAO,GAAG,CAAC,IAA0B,EAAE,EAAE;gBAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC/C,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC;oBAC9B,IAAI,CAAC,GAAG,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;oBACjC,OAAO,CAAC,MAAM,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC,CAAC;YAEF,IAAI,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,yBAAyB;QACzB,IAAI,CAAC,SAAS,EAAE,CAAC;QAEjB,6BAA6B;QAC7B,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,kBAAkB;QAClB,KAAK,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC9C,MAAM,CAAC,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAErB,uBAAuB;QACvB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC9B,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAE1B,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,OAA0B;IACzD,OAAO,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC;AACjC,CAAC"}
|
|
@@ -0,0 +1,171 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Worker Session
|
|
3
|
+
*
|
|
4
|
+
* Per-worker state and execution context.
|
|
5
|
+
*/
|
|
6
|
+
import { EventEmitter } from 'events';
|
|
7
|
+
import type { WorkerSession as WorkerSessionState, WorkerStatus, TokenUsage, TicketContext, ImplementationResult } from '../types.js';
|
|
8
|
+
import type { AgentRunner } from '../agents/agent-runner.js';
|
|
9
|
+
/**
|
|
10
|
+
* Worker Session Events
|
|
11
|
+
*/
|
|
12
|
+
export interface WorkerSessionEvents {
|
|
13
|
+
'status:changed': {
|
|
14
|
+
workerId: number;
|
|
15
|
+
status: WorkerStatus;
|
|
16
|
+
};
|
|
17
|
+
'ticket:started': {
|
|
18
|
+
workerId: number;
|
|
19
|
+
ticketId: string;
|
|
20
|
+
};
|
|
21
|
+
'ticket:completed': {
|
|
22
|
+
workerId: number;
|
|
23
|
+
ticketId: string;
|
|
24
|
+
result: ImplementationResult;
|
|
25
|
+
};
|
|
26
|
+
'ticket:failed': {
|
|
27
|
+
workerId: number;
|
|
28
|
+
ticketId: string;
|
|
29
|
+
error: string;
|
|
30
|
+
};
|
|
31
|
+
'cost:updated': {
|
|
32
|
+
workerId: number;
|
|
33
|
+
totalCost: number;
|
|
34
|
+
};
|
|
35
|
+
}
|
|
36
|
+
/**
|
|
37
|
+
* Worker Session
|
|
38
|
+
*
|
|
39
|
+
* Manages state and execution for a single worker.
|
|
40
|
+
*/
|
|
41
|
+
export declare class WorkerSession extends EventEmitter {
|
|
42
|
+
private state;
|
|
43
|
+
private agentRunner?;
|
|
44
|
+
private abortController?;
|
|
45
|
+
constructor(workerId: number, worktreePath: string, branchName: string);
|
|
46
|
+
/**
|
|
47
|
+
* Get current state
|
|
48
|
+
*/
|
|
49
|
+
getState(): WorkerSessionState;
|
|
50
|
+
/**
|
|
51
|
+
* Get worker ID
|
|
52
|
+
*/
|
|
53
|
+
getWorkerId(): number;
|
|
54
|
+
/**
|
|
55
|
+
* Get current status
|
|
56
|
+
*/
|
|
57
|
+
getStatus(): WorkerStatus;
|
|
58
|
+
/**
|
|
59
|
+
* Set status
|
|
60
|
+
*/
|
|
61
|
+
setStatus(status: WorkerStatus): void;
|
|
62
|
+
/**
|
|
63
|
+
* Check if worker is idle
|
|
64
|
+
*/
|
|
65
|
+
isIdle(): boolean;
|
|
66
|
+
/**
|
|
67
|
+
* Check if worker is busy
|
|
68
|
+
*/
|
|
69
|
+
isBusy(): boolean;
|
|
70
|
+
/**
|
|
71
|
+
* Set agent runner
|
|
72
|
+
*/
|
|
73
|
+
setAgentRunner(runner: AgentRunner): void;
|
|
74
|
+
/**
|
|
75
|
+
* Get agent runner
|
|
76
|
+
*/
|
|
77
|
+
getAgentRunner(): AgentRunner | undefined;
|
|
78
|
+
/**
|
|
79
|
+
* Get worktree path
|
|
80
|
+
*/
|
|
81
|
+
getWorktreePath(): string;
|
|
82
|
+
/**
|
|
83
|
+
* Get branch name
|
|
84
|
+
*/
|
|
85
|
+
getBranchName(): string;
|
|
86
|
+
/**
|
|
87
|
+
* Start processing a ticket
|
|
88
|
+
*/
|
|
89
|
+
process(context: TicketContext): Promise<ImplementationResult>;
|
|
90
|
+
/**
|
|
91
|
+
* Internal process implementation
|
|
92
|
+
*/
|
|
93
|
+
private processInternal;
|
|
94
|
+
/**
|
|
95
|
+
* Retry current ticket
|
|
96
|
+
*/
|
|
97
|
+
retry(context: TicketContext): Promise<ImplementationResult>;
|
|
98
|
+
/**
|
|
99
|
+
* Cancel current processing
|
|
100
|
+
*/
|
|
101
|
+
cancel(): void;
|
|
102
|
+
/**
|
|
103
|
+
* Pause the worker
|
|
104
|
+
*/
|
|
105
|
+
pause(): void;
|
|
106
|
+
/**
|
|
107
|
+
* Resume the worker
|
|
108
|
+
*/
|
|
109
|
+
resume(): void;
|
|
110
|
+
/**
|
|
111
|
+
* Add cost
|
|
112
|
+
*/
|
|
113
|
+
addCost(cost: number): void;
|
|
114
|
+
/**
|
|
115
|
+
* Get total cost
|
|
116
|
+
*/
|
|
117
|
+
getTotalCost(): number;
|
|
118
|
+
/**
|
|
119
|
+
* Get completed count
|
|
120
|
+
*/
|
|
121
|
+
getCompletedCount(): number;
|
|
122
|
+
/**
|
|
123
|
+
* Get failed count
|
|
124
|
+
*/
|
|
125
|
+
getFailedCount(): number;
|
|
126
|
+
/**
|
|
127
|
+
* Get retry count for current ticket
|
|
128
|
+
*/
|
|
129
|
+
getRetryCount(): number;
|
|
130
|
+
/**
|
|
131
|
+
* Get current ticket ID
|
|
132
|
+
*/
|
|
133
|
+
getCurrentTicketId(): string | undefined;
|
|
134
|
+
/**
|
|
135
|
+
* Get last error
|
|
136
|
+
*/
|
|
137
|
+
getLastError(): string | undefined;
|
|
138
|
+
/**
|
|
139
|
+
* Update token usage for current ticket
|
|
140
|
+
*/
|
|
141
|
+
updateTokenUsage(tokens: TokenUsage): void;
|
|
142
|
+
/**
|
|
143
|
+
* Get current token usage
|
|
144
|
+
*/
|
|
145
|
+
getCurrentTokenUsage(): TokenUsage | undefined;
|
|
146
|
+
/**
|
|
147
|
+
* Get elapsed time for current ticket
|
|
148
|
+
*/
|
|
149
|
+
getElapsedTime(): number;
|
|
150
|
+
/**
|
|
151
|
+
* Reset worker state
|
|
152
|
+
*/
|
|
153
|
+
reset(): void;
|
|
154
|
+
/**
|
|
155
|
+
* Get statistics
|
|
156
|
+
*/
|
|
157
|
+
getStats(): {
|
|
158
|
+
workerId: number;
|
|
159
|
+
status: WorkerStatus;
|
|
160
|
+
completed: number;
|
|
161
|
+
failed: number;
|
|
162
|
+
cost: number;
|
|
163
|
+
currentTicket?: string;
|
|
164
|
+
elapsedMs?: number;
|
|
165
|
+
};
|
|
166
|
+
}
|
|
167
|
+
/**
|
|
168
|
+
* Create a worker session
|
|
169
|
+
*/
|
|
170
|
+
export declare function createWorkerSession(workerId: number, worktreePath: string, branchName: string): WorkerSession;
|
|
171
|
+
//# sourceMappingURL=worker-session.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"worker-session.d.ts","sourceRoot":"","sources":["../../../src/autopilot/worker/worker-session.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,KAAK,EACV,aAAa,IAAI,kBAAkB,EACnC,YAAY,EACZ,UAAU,EACV,aAAa,EACb,oBAAoB,EACrB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAE7D;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,gBAAgB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,YAAY,CAAA;KAAE,CAAC;IAC7D,gBAAgB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IACzD,kBAAkB,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,oBAAoB,CAAA;KAAE,CAAC;IACzF,eAAe,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IACvE,cAAc,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC;CACzD;AAED;;;;GAIG;AACH,qBAAa,aAAc,SAAQ,YAAY;IAC7C,OAAO,CAAC,KAAK,CAAqB;IAClC,OAAO,CAAC,WAAW,CAAC,CAAc;IAClC,OAAO,CAAC,eAAe,CAAC,CAAkB;gBAGxC,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM;IAgBpB;;OAEG;IACH,QAAQ,IAAI,kBAAkB;IAI9B;;OAEG;IACH,WAAW,IAAI,MAAM;IAIrB;;OAEG;IACH,SAAS,IAAI,YAAY;IAIzB;;OAEG;IACH,SAAS,CAAC,MAAM,EAAE,YAAY,GAAG,IAAI;IAYrC;;OAEG;IACH,MAAM,IAAI,OAAO;IAIjB;;OAEG;IACH,MAAM,IAAI,OAAO;IAIjB;;OAEG;IACH,cAAc,CAAC,MAAM,EAAE,WAAW,GAAG,IAAI;IAMzC;;OAEG;IACH,cAAc,IAAI,WAAW,GAAG,SAAS;IAIzC;;OAEG;IACH,eAAe,IAAI,MAAM;IAIzB;;OAEG;IACH,aAAa,IAAI,MAAM;IAIvB;;OAEG;IACG,OAAO,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAIpE;;OAEG;YACW,eAAe;IA6D7B;;OAEG;IACG,KAAK,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAMlE;;OAEG;IACH,MAAM,IAAI,IAAI;IAMd;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;OAEG;IACH,MAAM,IAAI,IAAI;IAMd;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAQ3B;;OAEG;IACH,YAAY,IAAI,MAAM;IAItB;;OAEG;IACH,iBAAiB,IAAI,MAAM;IAI3B;;OAEG;IACH,cAAc,IAAI,MAAM;IAIxB;;OAEG;IACH,aAAa,IAAI,MAAM;IAIvB;;OAEG;IACH,kBAAkB,IAAI,MAAM,GAAG,SAAS;IAIxC;;OAEG;IACH,YAAY,IAAI,MAAM,GAAG,SAAS;IAIlC;;OAEG;IACH,gBAAgB,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI;IAI1C;;OAEG;IACH,oBAAoB,IAAI,UAAU,GAAG,SAAS;IAI9C;;OAEG;IACH,cAAc,IAAI,MAAM;IAKxB;;OAEG;IACH,KAAK,IAAI,IAAI;IAYb;;OAEG;IACH,QAAQ,IAAI;QACV,QAAQ,EAAE,MAAM,CAAC;QACjB,MAAM,EAAE,YAAY,CAAC;QACrB,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB;CAaF;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,MAAM,EAChB,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM,GACjB,aAAa,CAEf"}
|