poe-code 0.1.26
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 +64 -0
- package/dist/cli/api-client.d.ts +14 -0
- package/dist/cli/api-client.js +140 -0
- package/dist/cli/api-client.js.map +1 -0
- package/dist/cli/chat.d.ts +32 -0
- package/dist/cli/chat.js +2 -0
- package/dist/cli/chat.js.map +1 -0
- package/dist/cli/command-runner.d.ts +2 -0
- package/dist/cli/command-runner.js +37 -0
- package/dist/cli/command-runner.js.map +1 -0
- package/dist/cli/commands/agent.d.ts +18 -0
- package/dist/cli/commands/agent.js +84 -0
- package/dist/cli/commands/agent.js.map +1 -0
- package/dist/cli/commands/configure-agents.d.ts +3 -0
- package/dist/cli/commands/configure-agents.js +72 -0
- package/dist/cli/commands/configure-agents.js.map +1 -0
- package/dist/cli/commands/configure.d.ts +11 -0
- package/dist/cli/commands/configure.js +150 -0
- package/dist/cli/commands/configure.js.map +1 -0
- package/dist/cli/commands/init.d.ts +8 -0
- package/dist/cli/commands/init.js +45 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/interactive.d.ts +3 -0
- package/dist/cli/commands/interactive.js +17 -0
- package/dist/cli/commands/interactive.js.map +1 -0
- package/dist/cli/commands/login.d.ts +6 -0
- package/dist/cli/commands/login.js +51 -0
- package/dist/cli/commands/login.js.map +1 -0
- package/dist/cli/commands/logout.d.ts +3 -0
- package/dist/cli/commands/logout.js +31 -0
- package/dist/cli/commands/logout.js.map +1 -0
- package/dist/cli/commands/prerequisites.d.ts +3 -0
- package/dist/cli/commands/prerequisites.js +23 -0
- package/dist/cli/commands/prerequisites.js.map +1 -0
- package/dist/cli/commands/query.d.ts +7 -0
- package/dist/cli/commands/query.js +46 -0
- package/dist/cli/commands/query.js.map +1 -0
- package/dist/cli/commands/remove.d.ts +6 -0
- package/dist/cli/commands/remove.js +80 -0
- package/dist/cli/commands/remove.js.map +1 -0
- package/dist/cli/commands/root.d.ts +3 -0
- package/dist/cli/commands/root.js +7 -0
- package/dist/cli/commands/root.js.map +1 -0
- package/dist/cli/commands/shared.d.ts +24 -0
- package/dist/cli/commands/shared.js +61 -0
- package/dist/cli/commands/shared.js.map +1 -0
- package/dist/cli/commands/spawn-worktree.d.ts +6 -0
- package/dist/cli/commands/spawn-worktree.js +91 -0
- package/dist/cli/commands/spawn-worktree.js.map +1 -0
- package/dist/cli/commands/spawn.d.ts +7 -0
- package/dist/cli/commands/spawn.js +112 -0
- package/dist/cli/commands/spawn.js.map +1 -0
- package/dist/cli/commands/test.d.ts +6 -0
- package/dist/cli/commands/test.js +26 -0
- package/dist/cli/commands/test.js.map +1 -0
- package/dist/cli/constants.d.ts +10 -0
- package/dist/cli/constants.js +11 -0
- package/dist/cli/constants.js.map +1 -0
- package/dist/cli/container.d.ts +48 -0
- package/dist/cli/container.js +94 -0
- package/dist/cli/container.js.map +1 -0
- package/dist/cli/context.d.ts +37 -0
- package/dist/cli/context.js +198 -0
- package/dist/cli/context.js.map +1 -0
- package/dist/cli/environment.d.ts +19 -0
- package/dist/cli/environment.js +26 -0
- package/dist/cli/environment.js.map +1 -0
- package/dist/cli/error-logger.d.ts +46 -0
- package/dist/cli/error-logger.js +137 -0
- package/dist/cli/error-logger.js.map +1 -0
- package/dist/cli/errors.d.ts +91 -0
- package/dist/cli/errors.js +128 -0
- package/dist/cli/errors.js.map +1 -0
- package/dist/cli/file-mentions.d.ts +13 -0
- package/dist/cli/file-mentions.js +87 -0
- package/dist/cli/file-mentions.js.map +1 -0
- package/dist/cli/file-picker-state.d.ts +14 -0
- package/dist/cli/file-picker-state.js +23 -0
- package/dist/cli/file-picker-state.js.map +1 -0
- package/dist/cli/http.d.ts +11 -0
- package/dist/cli/http.js +2 -0
- package/dist/cli/http.js.map +1 -0
- package/dist/cli/interactive-command-runner.d.ts +11 -0
- package/dist/cli/interactive-command-runner.js +142 -0
- package/dist/cli/interactive-command-runner.js.map +1 -0
- package/dist/cli/interactive-launcher.d.ts +2 -0
- package/dist/cli/interactive-launcher.js +404 -0
- package/dist/cli/interactive-launcher.js.map +1 -0
- package/dist/cli/interactive-tasks.d.ts +7 -0
- package/dist/cli/interactive-tasks.js +199 -0
- package/dist/cli/interactive-tasks.js.map +1 -0
- package/dist/cli/interactive.d.ts +19 -0
- package/dist/cli/interactive.js +479 -0
- package/dist/cli/interactive.js.map +1 -0
- package/dist/cli/logger.d.ts +26 -0
- package/dist/cli/logger.js +87 -0
- package/dist/cli/logger.js.map +1 -0
- package/dist/cli/markdown-renderer.d.ts +27 -0
- package/dist/cli/markdown-renderer.js +178 -0
- package/dist/cli/markdown-renderer.js.map +1 -0
- package/dist/cli/markdown-renderer.test.d.ts +1 -0
- package/dist/cli/markdown-renderer.test.js +105 -0
- package/dist/cli/markdown-renderer.test.js.map +1 -0
- package/dist/cli/options.d.ts +30 -0
- package/dist/cli/options.js +84 -0
- package/dist/cli/options.js.map +1 -0
- package/dist/cli/program.d.ts +4 -0
- package/dist/cli/program.js +64 -0
- package/dist/cli/program.js.map +1 -0
- package/dist/cli/prompts.d.ts +23 -0
- package/dist/cli/prompts.js +58 -0
- package/dist/cli/prompts.js.map +1 -0
- package/dist/cli/service-registry.d.ts +33 -0
- package/dist/cli/service-registry.js +35 -0
- package/dist/cli/service-registry.js.map +1 -0
- package/dist/cli/telemetry.d.ts +15 -0
- package/dist/cli/telemetry.js +45 -0
- package/dist/cli/telemetry.js.map +1 -0
- package/dist/cli/types.d.ts +2 -0
- package/dist/cli/types.js +2 -0
- package/dist/cli/types.js.map +1 -0
- package/dist/commands/init.d.ts +9 -0
- package/dist/commands/init.js +43 -0
- package/dist/commands/init.js.map +1 -0
- package/dist/commands/publish-placeholder.d.ts +8 -0
- package/dist/commands/publish-placeholder.js +50 -0
- package/dist/commands/publish-placeholder.js.map +1 -0
- package/dist/commands/spawn-worktree.d.ts +25 -0
- package/dist/commands/spawn-worktree.js +139 -0
- package/dist/commands/spawn-worktree.js.map +1 -0
- package/dist/index.d.ts +4 -0
- package/dist/index.js +81 -0
- package/dist/index.js.map +1 -0
- package/dist/providers/claude-code-adapter.d.ts +20 -0
- package/dist/providers/claude-code-adapter.js +48 -0
- package/dist/providers/claude-code-adapter.js.map +1 -0
- package/dist/providers/codex-adapter.d.ts +19 -0
- package/dist/providers/codex-adapter.js +49 -0
- package/dist/providers/codex-adapter.js.map +1 -0
- package/dist/providers/index.d.ts +2 -0
- package/dist/providers/index.js +13 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/opencode-adapter.d.ts +18 -0
- package/dist/providers/opencode-adapter.js +45 -0
- package/dist/providers/opencode-adapter.js.map +1 -0
- package/dist/providers/roo-code-adapter.d.ts +19 -0
- package/dist/providers/roo-code-adapter.js +70 -0
- package/dist/providers/roo-code-adapter.js.map +1 -0
- package/dist/services/agent-config-manager.d.ts +32 -0
- package/dist/services/agent-config-manager.js +130 -0
- package/dist/services/agent-config-manager.js.map +1 -0
- package/dist/services/agent-registry.d.ts +27 -0
- package/dist/services/agent-registry.js +89 -0
- package/dist/services/agent-registry.js.map +1 -0
- package/dist/services/agent-session.d.ts +24 -0
- package/dist/services/agent-session.js +107 -0
- package/dist/services/agent-session.js.map +1 -0
- package/dist/services/agent-task-registry.d.ts +131 -0
- package/dist/services/agent-task-registry.js +491 -0
- package/dist/services/agent-task-registry.js.map +1 -0
- package/dist/services/chat.d.ts +94 -0
- package/dist/services/chat.js +241 -0
- package/dist/services/chat.js.map +1 -0
- package/dist/services/claude-code.d.ts +28 -0
- package/dist/services/claude-code.js +155 -0
- package/dist/services/claude-code.js.map +1 -0
- package/dist/services/codex.d.ts +27 -0
- package/dist/services/codex.js +231 -0
- package/dist/services/codex.js.map +1 -0
- package/dist/services/credentials.d.ts +11 -0
- package/dist/services/credentials.js +44 -0
- package/dist/services/credentials.js.map +1 -0
- package/dist/services/mcp-client.d.ts +38 -0
- package/dist/services/mcp-client.js +170 -0
- package/dist/services/mcp-client.js.map +1 -0
- package/dist/services/mcp-manager.d.ts +28 -0
- package/dist/services/mcp-manager.js +157 -0
- package/dist/services/mcp-manager.js.map +1 -0
- package/dist/services/model-strategy.d.ts +99 -0
- package/dist/services/model-strategy.js +182 -0
- package/dist/services/model-strategy.js.map +1 -0
- package/dist/services/opencode.d.ts +25 -0
- package/dist/services/opencode.js +161 -0
- package/dist/services/opencode.js.map +1 -0
- package/dist/services/poe-code.d.ts +7 -0
- package/dist/services/poe-code.js +5 -0
- package/dist/services/poe-code.js.map +1 -0
- package/dist/services/roo-code.d.ts +22 -0
- package/dist/services/roo-code.js +134 -0
- package/dist/services/roo-code.js.map +1 -0
- package/dist/services/service-install.d.ts +21 -0
- package/dist/services/service-install.js +47 -0
- package/dist/services/service-install.js.map +1 -0
- package/dist/services/service-manifest.d.ts +122 -0
- package/dist/services/service-manifest.js +368 -0
- package/dist/services/service-manifest.js.map +1 -0
- package/dist/services/task-logger.d.ts +24 -0
- package/dist/services/task-logger.js +73 -0
- package/dist/services/task-logger.js.map +1 -0
- package/dist/services/task-runner.d.ts +19 -0
- package/dist/services/task-runner.js +157 -0
- package/dist/services/task-runner.js.map +1 -0
- package/dist/services/tools.d.ts +76 -0
- package/dist/services/tools.js +870 -0
- package/dist/services/tools.js.map +1 -0
- package/dist/templates/claude-code/anthropic_key.sh.hbs +2 -0
- package/dist/templates/codex/config.toml.hbs +9 -0
- package/dist/templates/python/env.hbs +3 -0
- package/dist/templates/python/main.py.hbs +17 -0
- package/dist/templates/python/requirements.txt.hbs +2 -0
- package/dist/utils/backup.d.ts +5 -0
- package/dist/utils/backup.js +61 -0
- package/dist/utils/backup.js.map +1 -0
- package/dist/utils/command-line.d.ts +1 -0
- package/dist/utils/command-line.js +54 -0
- package/dist/utils/command-line.js.map +1 -0
- package/dist/utils/dry-run.d.ts +38 -0
- package/dist/utils/dry-run.js +211 -0
- package/dist/utils/dry-run.js.map +1 -0
- package/dist/utils/e2e-flags.d.ts +1 -0
- package/dist/utils/e2e-flags.js +9 -0
- package/dist/utils/e2e-flags.js.map +1 -0
- package/dist/utils/file-system.d.ts +21 -0
- package/dist/utils/file-system.js +2 -0
- package/dist/utils/file-system.js.map +1 -0
- package/dist/utils/json.d.ts +12 -0
- package/dist/utils/json.js +42 -0
- package/dist/utils/json.js.map +1 -0
- package/dist/utils/prerequisites.d.ts +41 -0
- package/dist/utils/prerequisites.js +92 -0
- package/dist/utils/prerequisites.js.map +1 -0
- package/dist/utils/templates.d.ts +4 -0
- package/dist/utils/templates.js +22 -0
- package/dist/utils/templates.js.map +1 -0
- package/dist/utils/toml.d.ts +8 -0
- package/dist/utils/toml.js +30 -0
- package/dist/utils/toml.js.map +1 -0
- package/dist/utils/worktree.d.ts +40 -0
- package/dist/utils/worktree.js +179 -0
- package/dist/utils/worktree.js.map +1 -0
- package/package.json +84 -0
- package/shared/conversation-layout.cjs +11 -0
- package/shared/conversation-layout.d.ts +7 -0
- package/shared/conversation-layout.js +11 -0
- package/shared/package.json +13 -0
|
@@ -0,0 +1,491 @@
|
|
|
1
|
+
import path from "node:path";
|
|
2
|
+
const DEFAULT_MAX_TASKS = 100;
|
|
3
|
+
const DAY_IN_MS = 24 * 60 * 60 * 1000;
|
|
4
|
+
export class AgentTaskRegistry {
|
|
5
|
+
constructor(options) {
|
|
6
|
+
this.tasks = new Map();
|
|
7
|
+
this.taskOrder = [];
|
|
8
|
+
this.completionCallbacks = [];
|
|
9
|
+
this.progressCallbacks = [];
|
|
10
|
+
this.completedQueue = [];
|
|
11
|
+
this.completedSet = new Set();
|
|
12
|
+
this.pendingDebounce = new Map();
|
|
13
|
+
this.fs = options.fs;
|
|
14
|
+
this.tasksDir = options.tasksDir;
|
|
15
|
+
this.logsDir = options.logsDir;
|
|
16
|
+
this.now = options.now ?? Date.now;
|
|
17
|
+
this.maxTasks = options.maxTasks ?? DEFAULT_MAX_TASKS;
|
|
18
|
+
this.archiveAfterMs = options.archiveAfterMs ?? DAY_IN_MS;
|
|
19
|
+
this.logRetentionMs = options.logRetentionMs ?? 7 * DAY_IN_MS;
|
|
20
|
+
this.completionRetentionMs = options.completionRetentionMs ?? 30 * DAY_IN_MS;
|
|
21
|
+
this.watchFactory = options.watchFactory;
|
|
22
|
+
this.debounceMs = options.debounceMs ?? 100;
|
|
23
|
+
this.logger = options.logger;
|
|
24
|
+
this.ensureDirectories();
|
|
25
|
+
this.bootstrapFromDisk();
|
|
26
|
+
// Start watching AFTER loading existing tasks to avoid race conditions
|
|
27
|
+
this.startWatching();
|
|
28
|
+
}
|
|
29
|
+
get size() {
|
|
30
|
+
return this.tasks.size;
|
|
31
|
+
}
|
|
32
|
+
getTasksDirectory() {
|
|
33
|
+
return this.tasksDir;
|
|
34
|
+
}
|
|
35
|
+
getLogsDirectory() {
|
|
36
|
+
return this.logsDir;
|
|
37
|
+
}
|
|
38
|
+
registerTask(input) {
|
|
39
|
+
const taskId = this.createTaskId();
|
|
40
|
+
const startTime = this.now();
|
|
41
|
+
const task = {
|
|
42
|
+
id: taskId,
|
|
43
|
+
toolName: input.toolName,
|
|
44
|
+
args: { ...input.args },
|
|
45
|
+
status: "running",
|
|
46
|
+
startTime,
|
|
47
|
+
logFile: path.join(this.logsDir, `${taskId}.log`),
|
|
48
|
+
progressFile: path.join(this.tasksDir, `${taskId}.progress.jsonl`)
|
|
49
|
+
};
|
|
50
|
+
this.tasks.set(taskId, task);
|
|
51
|
+
this.taskOrder.push(taskId);
|
|
52
|
+
this.trimCache();
|
|
53
|
+
this.persistTask(task);
|
|
54
|
+
this.ensureFile(task.progressFile);
|
|
55
|
+
this.logger?.("task_registered", {
|
|
56
|
+
id: taskId,
|
|
57
|
+
tool: input.toolName
|
|
58
|
+
});
|
|
59
|
+
return taskId;
|
|
60
|
+
}
|
|
61
|
+
updateTask(id, patch) {
|
|
62
|
+
const current = this.tasks.get(id) ?? this.loadTask(id);
|
|
63
|
+
if (!current) {
|
|
64
|
+
throw new Error(`Unknown task "${id}"`);
|
|
65
|
+
}
|
|
66
|
+
const updated = {
|
|
67
|
+
...current,
|
|
68
|
+
...patch,
|
|
69
|
+
args: current.args
|
|
70
|
+
};
|
|
71
|
+
this.tasks.set(id, updated);
|
|
72
|
+
this.persistTask(updated);
|
|
73
|
+
if (this.isTerminal(updated.status)) {
|
|
74
|
+
this.enqueueCompletion(updated);
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
getTask(id) {
|
|
78
|
+
return this.tasks.get(id) ?? this.loadTask(id);
|
|
79
|
+
}
|
|
80
|
+
getAllTasks() {
|
|
81
|
+
const tasks = Array.from(this.tasks.values());
|
|
82
|
+
return tasks.sort((a, b) => b.startTime - a.startTime);
|
|
83
|
+
}
|
|
84
|
+
getRunningTasks() {
|
|
85
|
+
const running = this.getAllTasks().filter((task) => task.status === "running");
|
|
86
|
+
// Check for zombie tasks (processes that are no longer alive)
|
|
87
|
+
for (const task of running) {
|
|
88
|
+
if (task.pid && !this.isProcessAlive(task.pid)) {
|
|
89
|
+
this.logger?.("zombie_task_detected", {
|
|
90
|
+
id: task.id,
|
|
91
|
+
pid: task.pid
|
|
92
|
+
});
|
|
93
|
+
this.updateTask(task.id, {
|
|
94
|
+
status: "failed",
|
|
95
|
+
error: "Process terminated unexpectedly (zombie task)",
|
|
96
|
+
endTime: this.now()
|
|
97
|
+
});
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
// Return only tasks that are still actually running
|
|
101
|
+
return this.getAllTasks().filter((task) => task.status === "running");
|
|
102
|
+
}
|
|
103
|
+
getCompletedTasks() {
|
|
104
|
+
return this.completedQueue.map((task) => ({ ...task }));
|
|
105
|
+
}
|
|
106
|
+
clearCompleted() {
|
|
107
|
+
this.completedQueue.length = 0;
|
|
108
|
+
this.completedSet.clear();
|
|
109
|
+
}
|
|
110
|
+
onTaskComplete(callback) {
|
|
111
|
+
this.completionCallbacks.push(callback);
|
|
112
|
+
}
|
|
113
|
+
onTaskProgress(callback) {
|
|
114
|
+
this.progressCallbacks.push(callback);
|
|
115
|
+
}
|
|
116
|
+
killTask(id) {
|
|
117
|
+
const task = this.tasks.get(id);
|
|
118
|
+
if (!task?.pid) {
|
|
119
|
+
return false;
|
|
120
|
+
}
|
|
121
|
+
try {
|
|
122
|
+
process.kill(task.pid);
|
|
123
|
+
const endTime = this.now();
|
|
124
|
+
const killedTask = {
|
|
125
|
+
...task,
|
|
126
|
+
status: "killed",
|
|
127
|
+
endTime
|
|
128
|
+
};
|
|
129
|
+
this.tasks.set(id, killedTask);
|
|
130
|
+
this.persistTask(killedTask);
|
|
131
|
+
this.enqueueCompletion(killedTask);
|
|
132
|
+
this.logger?.("task_killed", { id });
|
|
133
|
+
return true;
|
|
134
|
+
}
|
|
135
|
+
catch (error) {
|
|
136
|
+
this.logger?.("task_kill_failed", {
|
|
137
|
+
id,
|
|
138
|
+
message: error instanceof Error ? error.message : String(error)
|
|
139
|
+
});
|
|
140
|
+
return false;
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
readProgress(taskId) {
|
|
144
|
+
const task = this.tasks.get(taskId);
|
|
145
|
+
if (!task) {
|
|
146
|
+
return [];
|
|
147
|
+
}
|
|
148
|
+
if (!this.fs.existsSync(task.progressFile)) {
|
|
149
|
+
return [];
|
|
150
|
+
}
|
|
151
|
+
const content = this.fs.readFileSync(task.progressFile, "utf8");
|
|
152
|
+
const lines = content.split("\n").filter((line) => line.trim().length > 0);
|
|
153
|
+
const updates = [];
|
|
154
|
+
for (const line of lines) {
|
|
155
|
+
try {
|
|
156
|
+
const parsed = JSON.parse(line);
|
|
157
|
+
updates.push(parsed);
|
|
158
|
+
}
|
|
159
|
+
catch {
|
|
160
|
+
// Skip malformed lines
|
|
161
|
+
}
|
|
162
|
+
}
|
|
163
|
+
return updates;
|
|
164
|
+
}
|
|
165
|
+
async waitForTask(id, onUpdate) {
|
|
166
|
+
const existing = this.getTask(id);
|
|
167
|
+
if (!existing) {
|
|
168
|
+
return undefined;
|
|
169
|
+
}
|
|
170
|
+
if (this.isTerminal(existing.status)) {
|
|
171
|
+
return existing;
|
|
172
|
+
}
|
|
173
|
+
return new Promise((resolve) => {
|
|
174
|
+
const completion = (task) => {
|
|
175
|
+
if (task.id === id) {
|
|
176
|
+
cleanup();
|
|
177
|
+
resolve(task);
|
|
178
|
+
}
|
|
179
|
+
};
|
|
180
|
+
const progress = (taskId, update) => {
|
|
181
|
+
if (taskId === id) {
|
|
182
|
+
onUpdate(update);
|
|
183
|
+
}
|
|
184
|
+
};
|
|
185
|
+
const cleanup = () => {
|
|
186
|
+
const completionIndex = this.completionCallbacks.indexOf(completion);
|
|
187
|
+
if (completionIndex !== -1) {
|
|
188
|
+
this.completionCallbacks.splice(completionIndex, 1);
|
|
189
|
+
}
|
|
190
|
+
const progressIndex = this.progressCallbacks.indexOf(progress);
|
|
191
|
+
if (progressIndex !== -1) {
|
|
192
|
+
this.progressCallbacks.splice(progressIndex, 1);
|
|
193
|
+
}
|
|
194
|
+
};
|
|
195
|
+
this.onTaskComplete(completion);
|
|
196
|
+
this.onTaskProgress(progress);
|
|
197
|
+
});
|
|
198
|
+
}
|
|
199
|
+
dispose() {
|
|
200
|
+
this.watcher?.close();
|
|
201
|
+
for (const pending of this.pendingDebounce.values()) {
|
|
202
|
+
clearTimeout(pending);
|
|
203
|
+
}
|
|
204
|
+
this.pendingDebounce.clear();
|
|
205
|
+
this.completionCallbacks.length = 0;
|
|
206
|
+
this.progressCallbacks.length = 0;
|
|
207
|
+
this.tasks.clear();
|
|
208
|
+
this.taskOrder.length = 0;
|
|
209
|
+
this.completedQueue.length = 0;
|
|
210
|
+
this.completedSet.clear();
|
|
211
|
+
}
|
|
212
|
+
async waitForAllTasks() {
|
|
213
|
+
const running = this.getRunningTasks();
|
|
214
|
+
if (running.length === 0) {
|
|
215
|
+
return;
|
|
216
|
+
}
|
|
217
|
+
await Promise.all(running.map((task) => this.waitForTask(task.id, () => {
|
|
218
|
+
// No-op progress handler for waiting callers
|
|
219
|
+
}).then(() => undefined)));
|
|
220
|
+
}
|
|
221
|
+
ensureDirectories() {
|
|
222
|
+
if (!this.fs.existsSync(this.tasksDir)) {
|
|
223
|
+
this.fs.mkdirSync(this.tasksDir, { recursive: true });
|
|
224
|
+
}
|
|
225
|
+
if (!this.fs.existsSync(this.logsDir)) {
|
|
226
|
+
this.fs.mkdirSync(this.logsDir, { recursive: true });
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
bootstrapFromDisk() {
|
|
230
|
+
this.loadExistingTasks();
|
|
231
|
+
this.cleanupCompletedTasks();
|
|
232
|
+
this.cleanupArchives();
|
|
233
|
+
void this.cleanupLogs();
|
|
234
|
+
}
|
|
235
|
+
loadExistingTasks() {
|
|
236
|
+
let entries = [];
|
|
237
|
+
try {
|
|
238
|
+
entries = this.fs.readdirSync(this.tasksDir);
|
|
239
|
+
}
|
|
240
|
+
catch {
|
|
241
|
+
return;
|
|
242
|
+
}
|
|
243
|
+
for (const entry of entries) {
|
|
244
|
+
if (!entry.endsWith(".json")) {
|
|
245
|
+
continue;
|
|
246
|
+
}
|
|
247
|
+
if (entry.includes("archive")) {
|
|
248
|
+
continue;
|
|
249
|
+
}
|
|
250
|
+
const id = entry.slice(0, entry.length - 5);
|
|
251
|
+
const task = this.loadTask(id);
|
|
252
|
+
if (!task) {
|
|
253
|
+
continue;
|
|
254
|
+
}
|
|
255
|
+
this.tasks.set(id, task);
|
|
256
|
+
this.taskOrder.push(id);
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
cleanupCompletedTasks() {
|
|
260
|
+
const cutoff = this.now() - this.archiveAfterMs;
|
|
261
|
+
const tasks = Array.from(this.tasks.values());
|
|
262
|
+
for (const task of tasks) {
|
|
263
|
+
if (task.endTime !== undefined &&
|
|
264
|
+
task.endTime < cutoff &&
|
|
265
|
+
(task.status === "completed" || task.status === "failed")) {
|
|
266
|
+
this.archiveTask(task.id);
|
|
267
|
+
}
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
cleanupArchives() {
|
|
271
|
+
const archiveDir = path.join(this.tasksDir, "archive");
|
|
272
|
+
if (!this.fs.existsSync(archiveDir)) {
|
|
273
|
+
return;
|
|
274
|
+
}
|
|
275
|
+
let files = [];
|
|
276
|
+
try {
|
|
277
|
+
files = this.fs.readdirSync(archiveDir);
|
|
278
|
+
}
|
|
279
|
+
catch {
|
|
280
|
+
return;
|
|
281
|
+
}
|
|
282
|
+
const cutoff = this.now() - this.completionRetentionMs;
|
|
283
|
+
for (const file of files) {
|
|
284
|
+
const filePath = path.join(archiveDir, file);
|
|
285
|
+
try {
|
|
286
|
+
const stats = this.fs.statSync(filePath);
|
|
287
|
+
if (stats.mtimeMs < cutoff) {
|
|
288
|
+
this.fs.unlinkSync(filePath);
|
|
289
|
+
}
|
|
290
|
+
}
|
|
291
|
+
catch {
|
|
292
|
+
// Ignore failures
|
|
293
|
+
}
|
|
294
|
+
}
|
|
295
|
+
}
|
|
296
|
+
async cleanupLogs() {
|
|
297
|
+
let files = [];
|
|
298
|
+
try {
|
|
299
|
+
files = await this.fs.promises.readdir(this.logsDir);
|
|
300
|
+
}
|
|
301
|
+
catch {
|
|
302
|
+
return;
|
|
303
|
+
}
|
|
304
|
+
const cutoff = this.now() - this.logRetentionMs;
|
|
305
|
+
for (const file of files) {
|
|
306
|
+
if (!file.includes(".log.")) {
|
|
307
|
+
continue;
|
|
308
|
+
}
|
|
309
|
+
const filePath = path.join(this.logsDir, file);
|
|
310
|
+
try {
|
|
311
|
+
const stats = await this.fs.promises.stat(filePath);
|
|
312
|
+
if (stats.mtimeMs < cutoff) {
|
|
313
|
+
await this.fs.promises.unlink(filePath);
|
|
314
|
+
}
|
|
315
|
+
}
|
|
316
|
+
catch {
|
|
317
|
+
// Ignore cleanup errors
|
|
318
|
+
}
|
|
319
|
+
}
|
|
320
|
+
}
|
|
321
|
+
startWatching() {
|
|
322
|
+
const factory = this.watchFactory ??
|
|
323
|
+
((dir, listener) => this.fs.watch(dir, listener));
|
|
324
|
+
this.watcher = factory(this.tasksDir, (event, filename) => {
|
|
325
|
+
if (!filename) {
|
|
326
|
+
return;
|
|
327
|
+
}
|
|
328
|
+
const existing = this.pendingDebounce.get(filename);
|
|
329
|
+
if (existing) {
|
|
330
|
+
clearTimeout(existing);
|
|
331
|
+
}
|
|
332
|
+
if (this.debounceMs <= 0) {
|
|
333
|
+
this.pendingDebounce.delete(filename);
|
|
334
|
+
this.handleFileChange(filename);
|
|
335
|
+
return;
|
|
336
|
+
}
|
|
337
|
+
const timeout = setTimeout(() => {
|
|
338
|
+
this.pendingDebounce.delete(filename);
|
|
339
|
+
this.handleFileChange(filename);
|
|
340
|
+
}, this.debounceMs);
|
|
341
|
+
this.pendingDebounce.set(filename, timeout);
|
|
342
|
+
});
|
|
343
|
+
}
|
|
344
|
+
handleFileChange(filename) {
|
|
345
|
+
if (filename.endsWith(".json") && !filename.includes("archive")) {
|
|
346
|
+
const id = filename.slice(0, filename.length - 5);
|
|
347
|
+
const task = this.loadTask(id);
|
|
348
|
+
if (!task) {
|
|
349
|
+
return;
|
|
350
|
+
}
|
|
351
|
+
this.tasks.set(id, task);
|
|
352
|
+
if (this.isTerminal(task.status)) {
|
|
353
|
+
this.enqueueCompletion(task);
|
|
354
|
+
}
|
|
355
|
+
return;
|
|
356
|
+
}
|
|
357
|
+
if (filename.endsWith(".progress.jsonl")) {
|
|
358
|
+
const id = filename.slice(0, filename.length - ".progress.jsonl".length);
|
|
359
|
+
const update = this.readLatestProgress(id);
|
|
360
|
+
if (!update) {
|
|
361
|
+
return;
|
|
362
|
+
}
|
|
363
|
+
for (const callback of this.progressCallbacks) {
|
|
364
|
+
callback(id, update);
|
|
365
|
+
}
|
|
366
|
+
}
|
|
367
|
+
}
|
|
368
|
+
enqueueCompletion(task) {
|
|
369
|
+
if (!this.completedSet.has(task.id)) {
|
|
370
|
+
this.completedSet.add(task.id);
|
|
371
|
+
this.completedQueue.push(task);
|
|
372
|
+
}
|
|
373
|
+
else {
|
|
374
|
+
const index = this.completedQueue.findIndex((item) => item.id === task.id);
|
|
375
|
+
if (index !== -1) {
|
|
376
|
+
this.completedQueue[index] = task;
|
|
377
|
+
}
|
|
378
|
+
}
|
|
379
|
+
for (const callback of this.completionCallbacks) {
|
|
380
|
+
callback(task);
|
|
381
|
+
}
|
|
382
|
+
}
|
|
383
|
+
readLatestProgress(taskId) {
|
|
384
|
+
const task = this.tasks.get(taskId);
|
|
385
|
+
if (!task) {
|
|
386
|
+
return undefined;
|
|
387
|
+
}
|
|
388
|
+
if (!this.fs.existsSync(task.progressFile)) {
|
|
389
|
+
return undefined;
|
|
390
|
+
}
|
|
391
|
+
const content = this.fs.readFileSync(task.progressFile, "utf8");
|
|
392
|
+
const lines = content.split("\n").filter((line) => line.trim().length > 0);
|
|
393
|
+
if (lines.length === 0) {
|
|
394
|
+
return undefined;
|
|
395
|
+
}
|
|
396
|
+
const lastLine = lines[lines.length - 1];
|
|
397
|
+
try {
|
|
398
|
+
return JSON.parse(lastLine);
|
|
399
|
+
}
|
|
400
|
+
catch {
|
|
401
|
+
return undefined;
|
|
402
|
+
}
|
|
403
|
+
}
|
|
404
|
+
archiveTask(id) {
|
|
405
|
+
const archiveDir = path.join(this.tasksDir, "archive");
|
|
406
|
+
if (!this.fs.existsSync(archiveDir)) {
|
|
407
|
+
this.fs.mkdirSync(archiveDir, { recursive: true });
|
|
408
|
+
}
|
|
409
|
+
const source = path.join(this.tasksDir, `${id}.json`);
|
|
410
|
+
const destination = path.join(archiveDir, `${id}.json`);
|
|
411
|
+
if (this.fs.existsSync(source)) {
|
|
412
|
+
try {
|
|
413
|
+
this.fs.renameSync(source, destination);
|
|
414
|
+
}
|
|
415
|
+
catch {
|
|
416
|
+
return;
|
|
417
|
+
}
|
|
418
|
+
}
|
|
419
|
+
const progress = path.join(this.tasksDir, `${id}.progress.jsonl`);
|
|
420
|
+
if (this.fs.existsSync(progress)) {
|
|
421
|
+
try {
|
|
422
|
+
this.fs.unlinkSync(progress);
|
|
423
|
+
}
|
|
424
|
+
catch {
|
|
425
|
+
// ignore
|
|
426
|
+
}
|
|
427
|
+
}
|
|
428
|
+
this.tasks.delete(id);
|
|
429
|
+
const orderIndex = this.taskOrder.indexOf(id);
|
|
430
|
+
if (orderIndex !== -1) {
|
|
431
|
+
this.taskOrder.splice(orderIndex, 1);
|
|
432
|
+
}
|
|
433
|
+
}
|
|
434
|
+
trimCache() {
|
|
435
|
+
while (this.tasks.size > this.maxTasks) {
|
|
436
|
+
const oldest = this.taskOrder.shift();
|
|
437
|
+
if (!oldest) {
|
|
438
|
+
break;
|
|
439
|
+
}
|
|
440
|
+
this.tasks.delete(oldest);
|
|
441
|
+
}
|
|
442
|
+
}
|
|
443
|
+
ensureFile(filePath) {
|
|
444
|
+
if (!this.fs.existsSync(filePath)) {
|
|
445
|
+
this.fs.writeFileSync(filePath, "", { encoding: "utf8" });
|
|
446
|
+
}
|
|
447
|
+
}
|
|
448
|
+
persistTask(task) {
|
|
449
|
+
const filePath = path.join(this.tasksDir, `${task.id}.json`);
|
|
450
|
+
const serialized = JSON.stringify(task, null, 2);
|
|
451
|
+
this.fs.writeFileSync(filePath, serialized, { encoding: "utf8" });
|
|
452
|
+
}
|
|
453
|
+
loadTask(id) {
|
|
454
|
+
const filePath = path.join(this.tasksDir, `${id}.json`);
|
|
455
|
+
if (!this.fs.existsSync(filePath)) {
|
|
456
|
+
return undefined;
|
|
457
|
+
}
|
|
458
|
+
try {
|
|
459
|
+
const raw = this.fs.readFileSync(filePath, "utf8");
|
|
460
|
+
const parsed = JSON.parse(raw);
|
|
461
|
+
return parsed;
|
|
462
|
+
}
|
|
463
|
+
catch {
|
|
464
|
+
return undefined;
|
|
465
|
+
}
|
|
466
|
+
}
|
|
467
|
+
createTaskId() {
|
|
468
|
+
const timestamp = this.now();
|
|
469
|
+
const random = Math.random().toString(36).slice(2, 10);
|
|
470
|
+
return `task_${timestamp}_${random}`;
|
|
471
|
+
}
|
|
472
|
+
isTerminal(status) {
|
|
473
|
+
return status === "completed" || status === "failed" || status === "killed";
|
|
474
|
+
}
|
|
475
|
+
isProcessAlive(pid) {
|
|
476
|
+
try {
|
|
477
|
+
// Signal 0 doesn't kill the process, just checks if it exists
|
|
478
|
+
process.kill(pid, 0);
|
|
479
|
+
return true;
|
|
480
|
+
}
|
|
481
|
+
catch (error) {
|
|
482
|
+
// ESRCH means process doesn't exist
|
|
483
|
+
if (error && typeof error === "object" && "code" in error && error.code === "ESRCH") {
|
|
484
|
+
return false;
|
|
485
|
+
}
|
|
486
|
+
// EPERM means process exists but we don't have permission (still alive)
|
|
487
|
+
return true;
|
|
488
|
+
}
|
|
489
|
+
}
|
|
490
|
+
}
|
|
491
|
+
//# sourceMappingURL=agent-task-registry.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent-task-registry.js","sourceRoot":"","sources":["../../src/services/agent-task-registry.ts"],"names":[],"mappings":"AACA,OAAO,IAAI,MAAM,WAAW,CAAC;AAiF7B,MAAM,iBAAiB,GAAG,GAAG,CAAC;AAC9B,MAAM,SAAS,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAEtC,MAAM,OAAO,iBAAiB;IA0B5B,YAAY,OAAiC;QAT5B,UAAK,GAAG,IAAI,GAAG,EAAqB,CAAC;QACrC,cAAS,GAAa,EAAE,CAAC;QACzB,wBAAmB,GAAqC,EAAE,CAAC;QAC3D,sBAAiB,GAChC,EAAE,CAAC;QACY,mBAAc,GAAgB,EAAE,CAAC;QACjC,iBAAY,GAAG,IAAI,GAAG,EAAU,CAAC;QACjC,oBAAe,GAAG,IAAI,GAAG,EAA0B,CAAC;QAGnE,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;QACnC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,iBAAiB,CAAC;QACtD,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,SAAS,CAAC;QAC1D,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,CAAC,GAAG,SAAS,CAAC;QAC9D,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,IAAI,EAAE,GAAG,SAAS,CAAC;QAC7E,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,GAAG,CAAC;QAC5C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAE7B,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,uEAAuE;QACvE,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,gBAAgB;QACd,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,YAAY,CAAC,KAGZ;QACC,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACnC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAc;YACtB,EAAE,EAAE,MAAM;YACV,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,EAAE,GAAG,KAAK,CAAC,IAAI,EAAE;YACvB,MAAM,EAAE,SAAS;YACjB,SAAS;YACT,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC;YACjD,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,MAAM,iBAAiB,CAAC;SACnE,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,IAAI,CAAC,SAAS,EAAE,CAAC;QACjB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACvB,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,EAAE,CAAC,iBAAiB,EAAE;YAC/B,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,KAAK,CAAC,QAAQ;SACrB,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,UAAU,CAAC,EAAU,EAAE,KAAyB;QAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;QAC1C,CAAC;QACD,MAAM,OAAO,GAAc;YACzB,GAAG,OAAO;YACV,GAAG,KAAK;YACR,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC5B,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAE1B,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED,OAAO,CAAC,EAAU;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,WAAW;QACT,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9C,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC;IAED,eAAe;QACb,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;QAE/E,8DAA8D;QAC9D,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC/C,IAAI,CAAC,MAAM,EAAE,CAAC,sBAAsB,EAAE;oBACpC,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,GAAG,EAAE,IAAI,CAAC,GAAG;iBACd,CAAC,CAAC;gBACH,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE;oBACvB,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,+CAA+C;oBACtD,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,oDAAoD;QACpD,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IACxE,CAAC;IAED,iBAAiB;QACf,OAAO,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,cAAc;QACZ,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED,cAAc,CAAC,QAAmC;QAChD,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED,cAAc,CAAC,QAA0D;QACvE,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,QAAQ,CAAC,EAAU;QACjB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChC,IAAI,CAAC,IAAI,EAAE,GAAG,EAAE,CAAC;YACf,OAAO,KAAK,CAAC;QACf,CAAC;QAED,IAAI,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,UAAU,GAAc;gBAC5B,GAAG,IAAI;gBACP,MAAM,EAAE,QAAQ;gBAChB,OAAO;aACR,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC/B,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;YAC7B,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACnC,IAAI,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,EAAE,CAAC,kBAAkB,EAAE;gBAChC,EAAE;gBACF,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAChE,CAAC,CAAC;YACH,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,YAAY,CAAC,MAAc;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3C,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC3E,MAAM,OAAO,GAAqB,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAmB,CAAC;gBAClD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,uBAAuB;YACzB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,WAAW,CACf,EAAU,EACV,QAA0C;QAE1C,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACrC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,IAAI,OAAO,CAAwB,CAAC,OAAO,EAAE,EAAE;YACpD,MAAM,UAAU,GAAG,CAAC,IAAe,EAAE,EAAE;gBACrC,IAAI,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;oBACnB,OAAO,EAAE,CAAC;oBACV,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC,CAAC;YACF,MAAM,QAAQ,GAAG,CAAC,MAAc,EAAE,MAAsB,EAAE,EAAE;gBAC1D,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;oBAClB,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACnB,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,MAAM,eAAe,GAAG,IAAI,CAAC,mBAAmB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBACrE,IAAI,eAAe,KAAK,CAAC,CAAC,EAAE,CAAC;oBAC3B,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;gBACtD,CAAC;gBACD,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBAC/D,IAAI,aAAa,KAAK,CAAC,CAAC,EAAE,CAAC;oBACzB,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC,CAAC;YAEF,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAChC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;QACtB,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,EAAE,CAAC;YACpD,YAAY,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;QACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,mBAAmB,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;QAC1B,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC;IAEH,KAAK,CAAC,eAAe;QACnB,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACvC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,OAAO;QACT,CAAC;QACD,MAAM,OAAO,CAAC,GAAG,CACf,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACnB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,EAAE,GAAG,EAAE;YAC7B,6CAA6C;QAC/C,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CACzB,CACF,CAAC;IACJ,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;YACtC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACzB,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;IAC1B,CAAC;IAEO,iBAAiB;QACvB,IAAI,OAAO,GAAa,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7B,SAAS;YACX,CAAC;YACD,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC9B,SAAS;YACX,CAAC;YACD,MAAM,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,SAAS;YACX,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACzB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAEO,qBAAqB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC;QAChD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IACE,IAAI,CAAC,OAAO,KAAK,SAAS;gBAC1B,IAAI,CAAC,OAAO,GAAG,MAAM;gBACrB,CAAC,IAAI,CAAC,MAAM,KAAK,WAAW,IAAI,IAAI,CAAC,MAAM,KAAK,QAAQ,CAAC,EACzD,CAAC;gBACD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,eAAe;QACrB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QACvD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,OAAO;QACT,CAAC;QACD,IAAI,KAAK,GAAa,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC;QACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAC7C,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBACzC,IAAI,KAAK,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC;oBAC3B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,kBAAkB;YACpB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW;QACvB,IAAI,KAAK,GAAa,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC;QAChD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC/C,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACpD,IAAI,KAAK,CAAC,OAAO,GAAG,MAAM,EAAE,CAAC;oBAC3B,MAAM,IAAI,CAAC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,wBAAwB;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,aAAa;QACnB,MAAM,OAAO,GACX,IAAI,CAAC,YAAY;YACjB,CAAC,CAAC,GAAW,EAAE,QAA0D,EAAE,EAAE,CAC3E,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE;YACxD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;YACT,CAAC;YACD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACpD,IAAI,QAAQ,EAAE,CAAC;gBACb,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzB,CAAC;YACD,IAAI,IAAI,CAAC,UAAU,IAAI,CAAC,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACtC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;gBAChC,OAAO;YACT,CAAC;YACD,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACtC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;YACpB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,gBAAgB,CAAC,QAAgB;QACvC,IAAI,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAChE,MAAM,EAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAClD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;YACzB,IAAI,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;gBACjC,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,QAAQ,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACzC,MAAM,EAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,MAAM,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACzE,MAAM,MAAM,GAAG,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAC3C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;YACT,CAAC;YACD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC9C,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;IACH,CAAC;IAEO,iBAAiB,CAAC,IAAe;QACvC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC/B,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3E,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjB,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;YACpC,CAAC;QACH,CAAC;QACD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAChD,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,MAAc;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YAC3C,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAChE,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC3E,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACzC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAmB,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAED,WAAW,CAAC,EAAU;QACpB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;QACvD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QACtD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QACxD,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YAC1C,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO;YACT,CAAC;QACH,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,iBAAiB,CAAC,CAAC;QAClE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC/B,CAAC;YAAC,MAAM,CAAC;gBACP,SAAS;YACX,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACtB,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAEO,SAAS;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YACvC,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;YACtC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM;YACR,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,QAAgB;QACjC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,IAAe;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;QAC7D,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QACjD,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;IACpE,CAAC;IAEO,QAAQ,CAAC,EAAU;QACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QACxD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YACnD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAc,CAAC;YAC5C,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAEO,YAAY;QAClB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACvD,OAAO,QAAQ,SAAS,IAAI,MAAM,EAAE,CAAC;IACvC,CAAC;IAEO,UAAU,CAAC,MAAmC;QACpD,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,QAAQ,CAAC;IAC9E,CAAC;IAEO,cAAc,CAAC,GAAW;QAChC,IAAI,CAAC;YACH,8DAA8D;YAC9D,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YACrB,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,oCAAoC;YACpC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACpF,OAAO,KAAK,CAAC;YACf,CAAC;YACD,wEAAwE;YACxE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,94 @@
|
|
|
1
|
+
import { ModelStrategy, StrategyConfig } from "./model-strategy.js";
|
|
2
|
+
import type { AgentTaskRegistry } from "./agent-task-registry.js";
|
|
3
|
+
export interface ChatMessage {
|
|
4
|
+
role: "system" | "user" | "assistant" | "tool";
|
|
5
|
+
content: string;
|
|
6
|
+
tool_calls?: ToolCall[];
|
|
7
|
+
tool_call_id?: string;
|
|
8
|
+
name?: string;
|
|
9
|
+
}
|
|
10
|
+
export interface ToolCall {
|
|
11
|
+
id: string;
|
|
12
|
+
type: "function";
|
|
13
|
+
function: {
|
|
14
|
+
name: string;
|
|
15
|
+
arguments: string;
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
export interface Tool {
|
|
19
|
+
type: "function";
|
|
20
|
+
function: {
|
|
21
|
+
name: string;
|
|
22
|
+
description: string;
|
|
23
|
+
parameters: {
|
|
24
|
+
type: "object";
|
|
25
|
+
properties: Record<string, unknown>;
|
|
26
|
+
required?: string[];
|
|
27
|
+
};
|
|
28
|
+
};
|
|
29
|
+
}
|
|
30
|
+
export interface ChatCompletionRequest {
|
|
31
|
+
model: string;
|
|
32
|
+
messages: ChatMessage[];
|
|
33
|
+
tools?: Tool[];
|
|
34
|
+
temperature?: number;
|
|
35
|
+
max_tokens?: number;
|
|
36
|
+
stream?: boolean;
|
|
37
|
+
}
|
|
38
|
+
export interface ChatCompletionResponse {
|
|
39
|
+
id: string;
|
|
40
|
+
object: string;
|
|
41
|
+
created: number;
|
|
42
|
+
model: string;
|
|
43
|
+
choices: Array<{
|
|
44
|
+
index: number;
|
|
45
|
+
message: ChatMessage;
|
|
46
|
+
finish_reason: string;
|
|
47
|
+
}>;
|
|
48
|
+
usage?: {
|
|
49
|
+
prompt_tokens: number;
|
|
50
|
+
completion_tokens: number;
|
|
51
|
+
total_tokens: number;
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
export interface ToolExecutor {
|
|
55
|
+
executeTool(name: string, args: Record<string, unknown>): Promise<string>;
|
|
56
|
+
}
|
|
57
|
+
export interface ToolCallEvent {
|
|
58
|
+
toolName: string;
|
|
59
|
+
args: Record<string, unknown>;
|
|
60
|
+
result?: string;
|
|
61
|
+
error?: string;
|
|
62
|
+
}
|
|
63
|
+
export type ToolCallCallback = (event: ToolCallEvent) => void;
|
|
64
|
+
export declare class PoeChatService {
|
|
65
|
+
private apiKey;
|
|
66
|
+
private baseUrl;
|
|
67
|
+
private conversationHistory;
|
|
68
|
+
private currentModel;
|
|
69
|
+
private toolExecutor?;
|
|
70
|
+
private onToolCall?;
|
|
71
|
+
private modelStrategy?;
|
|
72
|
+
private strategyEnabled;
|
|
73
|
+
private taskRegistry?;
|
|
74
|
+
constructor(apiKey: string, model?: string, toolExecutor?: ToolExecutor, onToolCall?: ToolCallCallback, systemPrompt?: string, taskRegistry?: AgentTaskRegistry);
|
|
75
|
+
setToolCallCallback(callback: ToolCallCallback): void;
|
|
76
|
+
setModel(model: string): void;
|
|
77
|
+
getModel(): string;
|
|
78
|
+
setStrategy(config: StrategyConfig): void;
|
|
79
|
+
getStrategy(): ModelStrategy | undefined;
|
|
80
|
+
getStrategyInfo(): string;
|
|
81
|
+
disableStrategy(): void;
|
|
82
|
+
enableStrategy(): void;
|
|
83
|
+
isStrategyEnabled(): boolean;
|
|
84
|
+
getHistory(): ChatMessage[];
|
|
85
|
+
clearHistory(): void;
|
|
86
|
+
addSystemMessage(content: string): void;
|
|
87
|
+
sendMessage(userMessage: string, tools?: Tool[], options?: {
|
|
88
|
+
signal?: AbortSignal;
|
|
89
|
+
onChunk?: (chunk: string) => void;
|
|
90
|
+
}): Promise<ChatMessage>;
|
|
91
|
+
private injectCompletedTasks;
|
|
92
|
+
private detectMessageContext;
|
|
93
|
+
private makeApiRequest;
|
|
94
|
+
}
|