awb-agent-manager 1.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/LICENSE +21 -0
- package/README.md +246 -0
- package/dist/lib/agent-lockfile.d.ts +37 -0
- package/dist/lib/agent-lockfile.js +205 -0
- package/dist/lib/agent-lockfile.js.map +1 -0
- package/dist/lib/agent-manager-commands.d.ts +34 -0
- package/dist/lib/agent-manager-commands.js +531 -0
- package/dist/lib/agent-manager-commands.js.map +1 -0
- package/dist/lib/base-session-manager.d.ts +127 -0
- package/dist/lib/base-session-manager.js +667 -0
- package/dist/lib/base-session-manager.js.map +1 -0
- package/dist/lib/chat-attachment-prep.d.ts +35 -0
- package/dist/lib/chat-attachment-prep.js +180 -0
- package/dist/lib/chat-attachment-prep.js.map +1 -0
- package/dist/lib/chat-session-manager.d.ts +29 -0
- package/dist/lib/chat-session-manager.js +400 -0
- package/dist/lib/chat-session-manager.js.map +1 -0
- package/dist/lib/circuit-breaker.d.ts +34 -0
- package/dist/lib/circuit-breaker.js +146 -0
- package/dist/lib/circuit-breaker.js.map +1 -0
- package/dist/lib/cli-adapters/antigravity.d.ts +15 -0
- package/dist/lib/cli-adapters/antigravity.js +114 -0
- package/dist/lib/cli-adapters/antigravity.js.map +1 -0
- package/dist/lib/cli-adapters/base.d.ts +115 -0
- package/dist/lib/cli-adapters/base.js +111 -0
- package/dist/lib/cli-adapters/base.js.map +1 -0
- package/dist/lib/cli-adapters/claude.d.ts +19 -0
- package/dist/lib/cli-adapters/claude.js +233 -0
- package/dist/lib/cli-adapters/claude.js.map +1 -0
- package/dist/lib/cli-adapters/codex.d.ts +14 -0
- package/dist/lib/cli-adapters/codex.js +152 -0
- package/dist/lib/cli-adapters/codex.js.map +1 -0
- package/dist/lib/cli-adapters/deepseek.d.ts +15 -0
- package/dist/lib/cli-adapters/deepseek.js +70 -0
- package/dist/lib/cli-adapters/deepseek.js.map +1 -0
- package/dist/lib/cli-adapters/index.d.ts +8 -0
- package/dist/lib/cli-adapters/index.js +27 -0
- package/dist/lib/cli-adapters/index.js.map +1 -0
- package/dist/lib/cli-adapters/model-introspect.d.ts +3 -0
- package/dist/lib/cli-adapters/model-introspect.js +65 -0
- package/dist/lib/cli-adapters/model-introspect.js.map +1 -0
- package/dist/lib/cli-error-signatures.d.ts +9 -0
- package/dist/lib/cli-error-signatures.js +22 -0
- package/dist/lib/cli-error-signatures.js.map +1 -0
- package/dist/lib/cli-output-summary.d.ts +2 -0
- package/dist/lib/cli-output-summary.js +104 -0
- package/dist/lib/cli-output-summary.js.map +1 -0
- package/dist/lib/cli-resolver.d.ts +2 -0
- package/dist/lib/cli-resolver.js +177 -0
- package/dist/lib/cli-resolver.js.map +1 -0
- package/dist/lib/config.d.ts +13 -0
- package/dist/lib/config.js +124 -0
- package/dist/lib/config.js.map +1 -0
- package/dist/lib/constants.d.ts +36 -0
- package/dist/lib/constants.js +51 -0
- package/dist/lib/constants.js.map +1 -0
- package/dist/lib/environment-provisioner.d.ts +31 -0
- package/dist/lib/environment-provisioner.js +182 -0
- package/dist/lib/environment-provisioner.js.map +1 -0
- package/dist/lib/error-log-uploader.d.ts +19 -0
- package/dist/lib/error-log-uploader.js +138 -0
- package/dist/lib/error-log-uploader.js.map +1 -0
- package/dist/lib/event-dispatcher.d.ts +149 -0
- package/dist/lib/event-dispatcher.js +799 -0
- package/dist/lib/event-dispatcher.js.map +1 -0
- package/dist/lib/event-log-recorder.d.ts +11 -0
- package/dist/lib/event-log-recorder.js +77 -0
- package/dist/lib/event-log-recorder.js.map +1 -0
- package/dist/lib/event-stream.d.ts +26 -0
- package/dist/lib/event-stream.js +177 -0
- package/dist/lib/event-stream.js.map +1 -0
- package/dist/lib/fs-browser.d.ts +30 -0
- package/dist/lib/fs-browser.js +287 -0
- package/dist/lib/fs-browser.js.map +1 -0
- package/dist/lib/host-mcp/logs.d.ts +19 -0
- package/dist/lib/host-mcp/logs.js +102 -0
- package/dist/lib/host-mcp/logs.js.map +1 -0
- package/dist/lib/host-mcp/platform.d.ts +24 -0
- package/dist/lib/host-mcp/platform.js +133 -0
- package/dist/lib/host-mcp/platform.js.map +1 -0
- package/dist/lib/host-mcp/screens.d.ts +22 -0
- package/dist/lib/host-mcp/screens.js +172 -0
- package/dist/lib/host-mcp/screens.js.map +1 -0
- package/dist/lib/host-mcp/server.d.ts +3 -0
- package/dist/lib/host-mcp/server.js +233 -0
- package/dist/lib/host-mcp/server.js.map +1 -0
- package/dist/lib/host-mcp/tools.d.ts +172 -0
- package/dist/lib/host-mcp/tools.js +1241 -0
- package/dist/lib/host-mcp/tools.js.map +1 -0
- package/dist/lib/instance-heartbeat.d.ts +55 -0
- package/dist/lib/instance-heartbeat.js +118 -0
- package/dist/lib/instance-heartbeat.js.map +1 -0
- package/dist/lib/legacy-import.d.ts +17 -0
- package/dist/lib/legacy-import.js +70 -0
- package/dist/lib/legacy-import.js.map +1 -0
- package/dist/lib/logging.d.ts +3 -0
- package/dist/lib/logging.js +65 -0
- package/dist/lib/logging.js.map +1 -0
- package/dist/lib/managed-agent-context.d.ts +25 -0
- package/dist/lib/managed-agent-context.js +33 -0
- package/dist/lib/managed-agent-context.js.map +1 -0
- package/dist/lib/managed-agent-store.d.ts +34 -0
- package/dist/lib/managed-agent-store.js +160 -0
- package/dist/lib/managed-agent-store.js.map +1 -0
- package/dist/lib/managed-agents.d.ts +24 -0
- package/dist/lib/managed-agents.js +87 -0
- package/dist/lib/managed-agents.js.map +1 -0
- package/dist/lib/mcp-client.d.ts +8 -0
- package/dist/lib/mcp-client.js +122 -0
- package/dist/lib/mcp-client.js.map +1 -0
- package/dist/lib/orphan-cleanup.d.ts +6 -0
- package/dist/lib/orphan-cleanup.js +108 -0
- package/dist/lib/orphan-cleanup.js.map +1 -0
- package/dist/lib/presence-heartbeat.d.ts +8 -0
- package/dist/lib/presence-heartbeat.js +61 -0
- package/dist/lib/presence-heartbeat.js.map +1 -0
- package/dist/lib/prompts.d.ts +29 -0
- package/dist/lib/prompts.js +208 -0
- package/dist/lib/prompts.js.map +1 -0
- package/dist/lib/rest.d.ts +61 -0
- package/dist/lib/rest.js +307 -0
- package/dist/lib/rest.js.map +1 -0
- package/dist/lib/run-provisioner.d.ts +21 -0
- package/dist/lib/run-provisioner.js +115 -0
- package/dist/lib/run-provisioner.js.map +1 -0
- package/dist/lib/self-path.d.ts +7 -0
- package/dist/lib/self-path.js +29 -0
- package/dist/lib/self-path.js.map +1 -0
- package/dist/lib/self-update.d.ts +46 -0
- package/dist/lib/self-update.js +482 -0
- package/dist/lib/self-update.js.map +1 -0
- package/dist/lib/service-install.d.ts +23 -0
- package/dist/lib/service-install.js +673 -0
- package/dist/lib/service-install.js.map +1 -0
- package/dist/lib/setup.d.ts +16 -0
- package/dist/lib/setup.js +157 -0
- package/dist/lib/setup.js.map +1 -0
- package/dist/lib/subagent-manager.d.ts +84 -0
- package/dist/lib/subagent-manager.js +793 -0
- package/dist/lib/subagent-manager.js.map +1 -0
- package/dist/lib/subagent-monitor.d.ts +43 -0
- package/dist/lib/subagent-monitor.js +229 -0
- package/dist/lib/subagent-monitor.js.map +1 -0
- package/dist/lib/ticket-session-manager.d.ts +23 -0
- package/dist/lib/ticket-session-manager.js +471 -0
- package/dist/lib/ticket-session-manager.js.map +1 -0
- package/dist/lib/worktree-manager.d.ts +39 -0
- package/dist/lib/worktree-manager.js +226 -0
- package/dist/lib/worktree-manager.js.map +1 -0
- package/dist/main.d.ts +2 -0
- package/dist/main.js +670 -0
- package/dist/main.js.map +1 -0
- package/dist/package.json +57 -0
- package/package.json +57 -0
|
@@ -0,0 +1,400 @@
|
|
|
1
|
+
import { BaseSessionManager, } from './base-session-manager.js';
|
|
2
|
+
import { ADAPTER_CAPABILITIES } from './cli-adapters/base.js';
|
|
3
|
+
import { fetchChatRoomHistory, postChatRoomMessage } from './rest.js';
|
|
4
|
+
import { log } from './logging.js';
|
|
5
|
+
import { composeChatRoomPrompt } from './prompts.js';
|
|
6
|
+
import { approxBase64Bytes, prepareChatAttachments, renderAttachmentBlock, } from './chat-attachment-prep.js';
|
|
7
|
+
const { PERSISTENT_SESSION } = ADAPTER_CAPABILITIES;
|
|
8
|
+
const HISTORY_ATTACHMENT_SCAN = 20;
|
|
9
|
+
const HISTORY_IMAGE_MAX_COUNT = 4;
|
|
10
|
+
const HISTORY_IMAGE_MAX_BYTES = 10 * 1024 * 1024;
|
|
11
|
+
const FALLBACK_MAX_CHARS = 1500;
|
|
12
|
+
const PROGRESS_MIN_INTERVAL_MS = 1500;
|
|
13
|
+
const PROGRESS_MAX_PER_SESSION = 30;
|
|
14
|
+
const PROGRESS_SUMMARY_MAX = 80;
|
|
15
|
+
export class ChatSessionManager extends BaseSessionManager {
|
|
16
|
+
#historyRing = new Map();
|
|
17
|
+
#HISTORY_MAX = 30;
|
|
18
|
+
#ROOMS_MAX = 200;
|
|
19
|
+
#chatSent = new Set();
|
|
20
|
+
#progressMeta = new Map();
|
|
21
|
+
constructor(config) {
|
|
22
|
+
super(config, {
|
|
23
|
+
keyField: 'sessionKey',
|
|
24
|
+
logTag: '[chat-session]',
|
|
25
|
+
cfgPrefix: 'cfg-chat-',
|
|
26
|
+
kindLabel: 'chat_session',
|
|
27
|
+
});
|
|
28
|
+
}
|
|
29
|
+
#makeKey(roomId, agentId) {
|
|
30
|
+
return `${roomId}|${agentId || '_'}`;
|
|
31
|
+
}
|
|
32
|
+
#evictRoomsBeyondCap() {
|
|
33
|
+
while (this.#historyRing.size > this.#ROOMS_MAX) {
|
|
34
|
+
const oldest = this.#historyRing.keys().next().value;
|
|
35
|
+
if (oldest === undefined)
|
|
36
|
+
break;
|
|
37
|
+
this.#historyRing.delete(oldest);
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
_historyRooms() {
|
|
41
|
+
return Array.from(this.#historyRing.keys());
|
|
42
|
+
}
|
|
43
|
+
_historyEntries(roomId) {
|
|
44
|
+
return (this.#historyRing.get(roomId) || []).slice();
|
|
45
|
+
}
|
|
46
|
+
recordRoomMessage(payload) {
|
|
47
|
+
const rid = payload?.room_id;
|
|
48
|
+
if (!rid)
|
|
49
|
+
return;
|
|
50
|
+
const msgType = typeof payload?.type === 'string' ? payload.type : 'message';
|
|
51
|
+
if (msgType !== 'message')
|
|
52
|
+
return;
|
|
53
|
+
let buf = this.#historyRing.get(rid);
|
|
54
|
+
if (buf) {
|
|
55
|
+
this.#historyRing.delete(rid);
|
|
56
|
+
this.#historyRing.set(rid, buf);
|
|
57
|
+
}
|
|
58
|
+
else {
|
|
59
|
+
buf = [];
|
|
60
|
+
this.#historyRing.set(rid, buf);
|
|
61
|
+
this.#evictRoomsBeyondCap();
|
|
62
|
+
}
|
|
63
|
+
buf.push({
|
|
64
|
+
sender_type: payload.sender_type,
|
|
65
|
+
sender_id: payload.sender_id,
|
|
66
|
+
sender_name: payload.sender_name,
|
|
67
|
+
content: payload.content,
|
|
68
|
+
created_at: payload.created_at,
|
|
69
|
+
attachments: Array.isArray(payload.attachments) ? payload.attachments : undefined,
|
|
70
|
+
});
|
|
71
|
+
while (buf.length > this.#HISTORY_MAX)
|
|
72
|
+
buf.shift();
|
|
73
|
+
}
|
|
74
|
+
async dispatch(spec) {
|
|
75
|
+
if (!spec.roomId)
|
|
76
|
+
return { dispatched: false, reason: 'no_room' };
|
|
77
|
+
const dedupKey = `msg:${spec.agentId || '_'}:${spec.senderId || ''}:${spec.createdAt || ''}`;
|
|
78
|
+
if (!this._rememberDedup(dedupKey)) {
|
|
79
|
+
return { dispatched: false, reason: 'duplicate_chat' };
|
|
80
|
+
}
|
|
81
|
+
const sessionKey = this.#makeKey(spec.roomId, spec.agentId);
|
|
82
|
+
const sess = this._getLiveSession(sessionKey);
|
|
83
|
+
if (sess) {
|
|
84
|
+
log(`[chat-session] reused existing pid=${sess.pid} room=${spec.roomId.slice(0, 8)} agent=${(spec.agentId || '').slice(0, 8)} turn=${sess.turnCount + 1}`);
|
|
85
|
+
const canEmitImages = sess.adapter.has(PERSISTENT_SESSION) && sess.cli_type === 'claude';
|
|
86
|
+
const prepared = await prepareChatAttachments(this._config, spec.roomId, spec.attachments, {
|
|
87
|
+
fetchImages: canEmitImages,
|
|
88
|
+
});
|
|
89
|
+
const followupText = this.#followupTurnText(spec, prepared);
|
|
90
|
+
const images = canEmitImages ? this.#extractTurnImages(prepared) : undefined;
|
|
91
|
+
this._sendFollowUp(sess, followupText, { onProgress: spec.onProgress, images });
|
|
92
|
+
return { dispatched: true, pid: sess.pid };
|
|
93
|
+
}
|
|
94
|
+
if (this._inflight.has(sessionKey)) {
|
|
95
|
+
log(`[chat-session] dispatch dropped (spawn already in-flight for same key): room=${spec.roomId.slice(0, 8)} agent=${(spec.agentId || '').slice(0, 8)}`);
|
|
96
|
+
this._forgetDedup(dedupKey);
|
|
97
|
+
return { dispatched: false, reason: 'inflight_spawn' };
|
|
98
|
+
}
|
|
99
|
+
if (!this._ensureCapacity()) {
|
|
100
|
+
this._forgetDedup(dedupKey);
|
|
101
|
+
return { dispatched: false, reason: 'cap_busy' };
|
|
102
|
+
}
|
|
103
|
+
this._inflight.set(sessionKey, {
|
|
104
|
+
agentId: spec.agentId || '',
|
|
105
|
+
roomId: spec.roomId,
|
|
106
|
+
});
|
|
107
|
+
let history = (this.#historyRing.get(spec.roomId) || []).slice();
|
|
108
|
+
if (history.length === 0) {
|
|
109
|
+
try {
|
|
110
|
+
history = await fetchChatRoomHistory(this._config, spec.roomId);
|
|
111
|
+
}
|
|
112
|
+
catch {
|
|
113
|
+
history = [];
|
|
114
|
+
}
|
|
115
|
+
}
|
|
116
|
+
const cli = String(spec.agentContext?.cli || 'claude').toLowerCase();
|
|
117
|
+
const canEmitImages = cli === 'claude';
|
|
118
|
+
const preparedFirstTurn = await prepareChatAttachments(this._config, spec.roomId, spec.attachments, { fetchImages: canEmitImages });
|
|
119
|
+
const historyAttachments = await this.#prepareHistoryAttachments(spec.roomId, history, canEmitImages);
|
|
120
|
+
const firstTurnText = composeChatRoomPrompt(spec.roomId, history, {
|
|
121
|
+
content: spec.content || '',
|
|
122
|
+
sender_name: spec.senderName || '',
|
|
123
|
+
sender_id: spec.senderId || '',
|
|
124
|
+
}, preparedFirstTurn, true, historyAttachments);
|
|
125
|
+
const firstTurnImages = canEmitImages
|
|
126
|
+
? this.#mergeImages(this.#extractHistoryImages(history, historyAttachments), this.#extractTurnImages(preparedFirstTurn))
|
|
127
|
+
: undefined;
|
|
128
|
+
const monitorMeta = {};
|
|
129
|
+
let spawned = null;
|
|
130
|
+
try {
|
|
131
|
+
spawned = await this._spawnSession(sessionKey, spec.rolePrompt || '', firstTurnText, { onProgress: spec.onProgress, monitorMeta, agentContext: spec.agentContext, firstTurnImages });
|
|
132
|
+
if (spawned) {
|
|
133
|
+
spawned.roomId = spec.roomId;
|
|
134
|
+
spawned.agentId = spec.agentId;
|
|
135
|
+
spawned._effectiveApiKey = spec.agentContext?.api_key || this._config.apiKey;
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
finally {
|
|
139
|
+
this._inflight.delete(sessionKey);
|
|
140
|
+
}
|
|
141
|
+
if (!spawned) {
|
|
142
|
+
this._forgetDedup(dedupKey);
|
|
143
|
+
return { dispatched: false, reason: 'spawn_failed' };
|
|
144
|
+
}
|
|
145
|
+
return { dispatched: true, pid: spawned.pid, firstTurn: true };
|
|
146
|
+
}
|
|
147
|
+
_onStdoutParsed(sess, parsed, _rawLine) {
|
|
148
|
+
if (parsed.isResult) {
|
|
149
|
+
this.#progressMeta.delete(sess.pid);
|
|
150
|
+
}
|
|
151
|
+
if (parsed.raw?.type === 'assistant') {
|
|
152
|
+
const content = parsed.raw?.message?.content;
|
|
153
|
+
if (Array.isArray(content)) {
|
|
154
|
+
const progressTools = [];
|
|
155
|
+
let sawFinal = false;
|
|
156
|
+
for (const block of content) {
|
|
157
|
+
if (block?.type !== 'tool_use' || typeof block.name !== 'string')
|
|
158
|
+
continue;
|
|
159
|
+
if (block.name.includes('send_chat_room_message')) {
|
|
160
|
+
this.#chatSent.add(sess.pid);
|
|
161
|
+
sawFinal = true;
|
|
162
|
+
continue;
|
|
163
|
+
}
|
|
164
|
+
progressTools.push({ name: block.name, input: block.input });
|
|
165
|
+
}
|
|
166
|
+
if (progressTools.length > 0) {
|
|
167
|
+
this.#emitProgress(sess, progressTools);
|
|
168
|
+
}
|
|
169
|
+
if (sawFinal) {
|
|
170
|
+
const meta = this.#progressMeta.get(sess.pid);
|
|
171
|
+
if (meta)
|
|
172
|
+
meta.finalSeen = true;
|
|
173
|
+
else
|
|
174
|
+
this.#progressMeta.set(sess.pid, { lastEmitMs: 0, count: 0, finalSeen: true });
|
|
175
|
+
}
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
}
|
|
179
|
+
_onStderrLine(_sess, _line) {
|
|
180
|
+
}
|
|
181
|
+
async _onChildExit(sess, _code, _signal) {
|
|
182
|
+
const sent = this.#chatSent.has(sess.pid);
|
|
183
|
+
const body = sent ? '' : this._collectOutputTail(sess.pid, FALLBACK_MAX_CHARS);
|
|
184
|
+
this.#chatSent.delete(sess.pid);
|
|
185
|
+
this.#progressMeta.delete(sess.pid);
|
|
186
|
+
const exitRoomId = sess.roomId;
|
|
187
|
+
if (exitRoomId) {
|
|
188
|
+
const stillUsed = Array.from(this._sessions.values()).some((s) => s !== sess && s.roomId === exitRoomId);
|
|
189
|
+
if (!stillUsed)
|
|
190
|
+
this.#historyRing.delete(exitRoomId);
|
|
191
|
+
}
|
|
192
|
+
if (sent)
|
|
193
|
+
return;
|
|
194
|
+
const roomId = sess.roomId;
|
|
195
|
+
const agentId = sess.agentId;
|
|
196
|
+
if (!roomId || !agentId)
|
|
197
|
+
return;
|
|
198
|
+
const message = body
|
|
199
|
+
? `⚠️ Agent가 응답하지 못했습니다. CLI 출력:\n\`\`\`\n${body}\n\`\`\``
|
|
200
|
+
: '⚠️ Agent가 응답하지 못했습니다 (출력 없음).';
|
|
201
|
+
log(`[chat-session] fallback message for room=${roomId} agent=${agentId} pid=${sess.pid} outputLen=${body.length}`);
|
|
202
|
+
const cfg = { ...this._config, apiKey: sess._effectiveApiKey || this._config.apiKey };
|
|
203
|
+
await postChatRoomMessage(cfg, roomId, agentId, message);
|
|
204
|
+
}
|
|
205
|
+
#emitProgress(sess, tools) {
|
|
206
|
+
if (tools.length === 0)
|
|
207
|
+
return;
|
|
208
|
+
const roomId = sess.roomId;
|
|
209
|
+
const agentId = sess.agentId;
|
|
210
|
+
if (!roomId || !agentId)
|
|
211
|
+
return;
|
|
212
|
+
const now = Date.now();
|
|
213
|
+
let meta = this.#progressMeta.get(sess.pid);
|
|
214
|
+
if (!meta) {
|
|
215
|
+
meta = { lastEmitMs: 0, count: 0, finalSeen: false };
|
|
216
|
+
this.#progressMeta.set(sess.pid, meta);
|
|
217
|
+
}
|
|
218
|
+
if (meta.finalSeen)
|
|
219
|
+
return;
|
|
220
|
+
if (meta.count >= PROGRESS_MAX_PER_SESSION)
|
|
221
|
+
return;
|
|
222
|
+
if (now - meta.lastEmitMs < PROGRESS_MIN_INTERVAL_MS)
|
|
223
|
+
return;
|
|
224
|
+
const message = this.#formatProgressLine(tools);
|
|
225
|
+
if (!message)
|
|
226
|
+
return;
|
|
227
|
+
meta.lastEmitMs = now;
|
|
228
|
+
meta.count += 1;
|
|
229
|
+
const cfg = { ...this._config, apiKey: sess._effectiveApiKey || this._config.apiKey };
|
|
230
|
+
void postChatRoomMessage(cfg, roomId, agentId, message, { type: 'progress' });
|
|
231
|
+
}
|
|
232
|
+
#formatProgressLine(tools) {
|
|
233
|
+
const head = tools[0];
|
|
234
|
+
const headLabel = this.#renderToolLabel(head.name, head.input);
|
|
235
|
+
if (tools.length === 1)
|
|
236
|
+
return `_${headLabel}_`;
|
|
237
|
+
return `_${headLabel} (+${tools.length - 1} more)_`;
|
|
238
|
+
}
|
|
239
|
+
#renderToolLabel(name, input) {
|
|
240
|
+
const icon = this.#iconForTool(name);
|
|
241
|
+
const display = this.#displayToolName(name);
|
|
242
|
+
const summary = this.#summarizeToolInput(name, input);
|
|
243
|
+
return summary ? `${icon} ${display} — ${summary}` : `${icon} ${display}`;
|
|
244
|
+
}
|
|
245
|
+
#displayToolName(name) {
|
|
246
|
+
return name;
|
|
247
|
+
}
|
|
248
|
+
#iconForTool(name) {
|
|
249
|
+
const lower = name.toLowerCase();
|
|
250
|
+
if (lower.startsWith('mcp__'))
|
|
251
|
+
return '📋';
|
|
252
|
+
if (lower === 'read' || lower === 'glob' || lower === 'grep')
|
|
253
|
+
return '🔍';
|
|
254
|
+
if (lower === 'edit' || lower === 'write' || lower === 'notebookedit')
|
|
255
|
+
return '✏️';
|
|
256
|
+
if (lower === 'bash')
|
|
257
|
+
return '💻';
|
|
258
|
+
if (lower === 'webfetch' || lower === 'websearch')
|
|
259
|
+
return '🌐';
|
|
260
|
+
if (lower === 'task' || lower === 'agent')
|
|
261
|
+
return '🤖';
|
|
262
|
+
return '🔧';
|
|
263
|
+
}
|
|
264
|
+
#summarizeToolInput(name, input) {
|
|
265
|
+
if (!input || typeof input !== 'object')
|
|
266
|
+
return '';
|
|
267
|
+
const lower = name.toLowerCase();
|
|
268
|
+
let raw = '';
|
|
269
|
+
if (lower === 'read' || lower === 'edit' || lower === 'write' || lower === 'notebookedit') {
|
|
270
|
+
raw = input.file_path ?? input.notebook_path ?? '';
|
|
271
|
+
}
|
|
272
|
+
else if (lower === 'glob') {
|
|
273
|
+
raw = input.pattern ?? '';
|
|
274
|
+
}
|
|
275
|
+
else if (lower === 'grep') {
|
|
276
|
+
raw = input.pattern ?? '';
|
|
277
|
+
}
|
|
278
|
+
else if (lower === 'bash') {
|
|
279
|
+
raw = input.command ?? '';
|
|
280
|
+
}
|
|
281
|
+
else if (lower === 'webfetch' || lower === 'websearch') {
|
|
282
|
+
raw = input.url ?? input.query ?? '';
|
|
283
|
+
}
|
|
284
|
+
else if (lower === 'task' || lower === 'agent') {
|
|
285
|
+
raw = input.description ?? input.subagent_type ?? '';
|
|
286
|
+
}
|
|
287
|
+
else {
|
|
288
|
+
const preferred = ['ticket_id', 'room_id', 'content', 'query', 'pattern', 'path', 'url'];
|
|
289
|
+
for (const k of preferred) {
|
|
290
|
+
if (typeof input[k] === 'string' && input[k]) {
|
|
291
|
+
raw = input[k];
|
|
292
|
+
break;
|
|
293
|
+
}
|
|
294
|
+
}
|
|
295
|
+
if (!raw) {
|
|
296
|
+
for (const v of Object.values(input)) {
|
|
297
|
+
if (typeof v === 'string' && v) {
|
|
298
|
+
raw = v;
|
|
299
|
+
break;
|
|
300
|
+
}
|
|
301
|
+
}
|
|
302
|
+
}
|
|
303
|
+
}
|
|
304
|
+
let s = typeof raw === 'string' ? raw : String(raw ?? '');
|
|
305
|
+
s = s.replace(/\s+/g, ' ').trim();
|
|
306
|
+
if (!s)
|
|
307
|
+
return '';
|
|
308
|
+
if (s.length > PROGRESS_SUMMARY_MAX) {
|
|
309
|
+
s = s.slice(0, PROGRESS_SUMMARY_MAX - 1) + '…';
|
|
310
|
+
}
|
|
311
|
+
return s.replace(/[`_*]/g, (c) => `\\${c}`);
|
|
312
|
+
}
|
|
313
|
+
#followupTurnText(spec, attachments) {
|
|
314
|
+
const text = (spec.content || '').trim();
|
|
315
|
+
if (attachments.length === 0)
|
|
316
|
+
return text;
|
|
317
|
+
const lines = [];
|
|
318
|
+
if (text)
|
|
319
|
+
lines.push(text);
|
|
320
|
+
for (const ln of renderAttachmentBlock(attachments)) {
|
|
321
|
+
lines.push(ln);
|
|
322
|
+
}
|
|
323
|
+
return lines.join('\n');
|
|
324
|
+
}
|
|
325
|
+
async #prepareHistoryAttachments(roomId, history, canEmitImages) {
|
|
326
|
+
const out = new Map();
|
|
327
|
+
const real = (Array.isArray(history) ? history : []).filter((h) => h && (!h.type || h.type === 'message'));
|
|
328
|
+
const recent = real.slice(-HISTORY_ATTACHMENT_SCAN);
|
|
329
|
+
let imageCountBudget = canEmitImages ? HISTORY_IMAGE_MAX_COUNT : 0;
|
|
330
|
+
let imageByteBudget = HISTORY_IMAGE_MAX_BYTES;
|
|
331
|
+
for (let i = recent.length - 1; i >= 0; i--) {
|
|
332
|
+
const entry = recent[i];
|
|
333
|
+
const raws = Array.isArray(entry.attachments) ? entry.attachments : [];
|
|
334
|
+
if (raws.length === 0)
|
|
335
|
+
continue;
|
|
336
|
+
const wantImages = canEmitImages && imageCountBudget > 0;
|
|
337
|
+
const prepared = await prepareChatAttachments(this._config, roomId, raws, {
|
|
338
|
+
fetchImages: wantImages,
|
|
339
|
+
materialize: false,
|
|
340
|
+
});
|
|
341
|
+
for (const att of prepared) {
|
|
342
|
+
if (att.kind !== 'image_base64')
|
|
343
|
+
continue;
|
|
344
|
+
const sz = approxBase64Bytes(att.image_base64);
|
|
345
|
+
if (imageCountBudget <= 0 || sz > imageByteBudget) {
|
|
346
|
+
att.kind = 'metadata_only';
|
|
347
|
+
att.image_base64 = undefined;
|
|
348
|
+
att.note = att.note || 'history image not inlined (vision budget exceeded)';
|
|
349
|
+
continue;
|
|
350
|
+
}
|
|
351
|
+
imageCountBudget -= 1;
|
|
352
|
+
imageByteBudget -= sz;
|
|
353
|
+
}
|
|
354
|
+
out.set(entry, prepared);
|
|
355
|
+
}
|
|
356
|
+
return out;
|
|
357
|
+
}
|
|
358
|
+
#extractHistoryImages(history, prepared) {
|
|
359
|
+
const images = [];
|
|
360
|
+
const real = (Array.isArray(history) ? history : []).filter((h) => h && (!h.type || h.type === 'message'));
|
|
361
|
+
for (const entry of real.slice(-HISTORY_ATTACHMENT_SCAN)) {
|
|
362
|
+
const atts = prepared.get(entry);
|
|
363
|
+
if (!atts)
|
|
364
|
+
continue;
|
|
365
|
+
for (const att of atts) {
|
|
366
|
+
if (att.kind === 'image_base64' && att.image_base64) {
|
|
367
|
+
images.push({ media_type: att.mime_type || 'image/png', data: att.image_base64 });
|
|
368
|
+
}
|
|
369
|
+
}
|
|
370
|
+
}
|
|
371
|
+
return images;
|
|
372
|
+
}
|
|
373
|
+
#mergeImages(...lists) {
|
|
374
|
+
const merged = [];
|
|
375
|
+
for (const l of lists)
|
|
376
|
+
if (l && l.length)
|
|
377
|
+
merged.push(...l);
|
|
378
|
+
return merged.length > 0 ? merged : undefined;
|
|
379
|
+
}
|
|
380
|
+
#extractTurnImages(attachments) {
|
|
381
|
+
const images = [];
|
|
382
|
+
for (const att of attachments) {
|
|
383
|
+
if (att.kind === 'image_base64' && att.image_base64) {
|
|
384
|
+
images.push({ media_type: att.mime_type || 'image/png', data: att.image_base64 });
|
|
385
|
+
}
|
|
386
|
+
}
|
|
387
|
+
return images.length > 0 ? images : undefined;
|
|
388
|
+
}
|
|
389
|
+
_snapshot() {
|
|
390
|
+
return Array.from(this._sessions.values()).map((s) => ({
|
|
391
|
+
roomId: s.roomId,
|
|
392
|
+
agentId: s.agentId,
|
|
393
|
+
pid: s.pid,
|
|
394
|
+
turnCount: s.turnCount,
|
|
395
|
+
startedAt: s.startedAt,
|
|
396
|
+
lastTouchedAt: s.lastTouchedAt,
|
|
397
|
+
}));
|
|
398
|
+
}
|
|
399
|
+
}
|
|
400
|
+
//# sourceMappingURL=chat-session-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-session-manager.js","sourceRoot":"","sources":["../../src/lib/chat-session-manager.ts"],"names":[],"mappings":"AAOA,OAAO,EACL,kBAAkB,GAInB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,oBAAoB,EAAoC,MAAM,wBAAwB,CAAC;AAChG,OAAO,EAAE,oBAAoB,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AACtE,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AACnC,OAAO,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EACL,iBAAiB,EACjB,sBAAsB,EACtB,qBAAqB,GAGtB,MAAM,2BAA2B,CAAC;AAEnC,MAAM,EAAE,kBAAkB,EAAE,GAAG,oBAAoB,CAAC;AAwBpD,MAAM,uBAAuB,GAAG,EAAE,CAAC;AAInC,MAAM,uBAAuB,GAAG,CAAC,CAAC;AAGlC,MAAM,uBAAuB,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;AAGjD,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAGhC,MAAM,wBAAwB,GAAG,IAAI,CAAC;AAKtC,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAEpC,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAEhC,MAAM,OAAO,kBACX,SAAQ,kBAAkB;IAG1B,YAAY,GAAG,IAAI,GAAG,EAA8B,CAAC;IACrD,YAAY,GAAG,EAAE,CAAC;IAQlB,UAAU,GAAG,GAAG,CAAC;IAQjB,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAI9B,aAAa,GAAG,IAAI,GAAG,EAAqE,CAAC;IAE7F,YAAY,MAA0B;QACpC,KAAK,CAAC,MAAM,EAAE;YACZ,QAAQ,EAAE,YAAY;YACtB,MAAM,EAAE,gBAAgB;YACxB,SAAS,EAAE,WAAW;YACtB,SAAS,EAAE,cAAc;SAC1B,CAAC,CAAC;IACL,CAAC;IAKD,QAAQ,CAAC,MAAc,EAAE,OAAe;QACtC,OAAO,GAAG,MAAM,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;IACvC,CAAC;IAMD,oBAAoB;QAClB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YAChD,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;YACrD,IAAI,MAAM,KAAK,SAAS;gBAAE,MAAM;YAChC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAKD,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC;IAC9C,CAAC;IAMD,eAAe,CAAC,MAAc;QAC5B,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;IACvD,CAAC;IAED,iBAAiB,CAAC,OAAY;QAC5B,MAAM,GAAG,GAAG,OAAO,EAAE,OAAO,CAAC;QAC7B,IAAI,CAAC,GAAG;YAAE,OAAO;QAOjB,MAAM,OAAO,GAAG,OAAO,OAAO,EAAE,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7E,IAAI,OAAO,KAAK,SAAS;YAAE,OAAO;QAClC,IAAI,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACrC,IAAI,GAAG,EAAE,CAAC;YAGR,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC9B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,GAAG,GAAG,EAAE,CAAC;YACT,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAChC,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC9B,CAAC;QACD,GAAG,CAAC,IAAI,CAAC;YACP,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,UAAU,EAAE,OAAO,CAAC,UAAU;YAM9B,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAClF,CAAC,CAAC;QACH,OAAO,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,YAAY;YAAE,GAAG,CAAC,KAAK,EAAE,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,IAAsB;QACnC,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAMlE,MAAM,QAAQ,GAAG,OAAO,IAAI,CAAC,OAAO,IAAI,GAAG,IAAI,IAAI,CAAC,QAAQ,IAAI,EAAE,IAAI,IAAI,CAAC,SAAS,IAAI,EAAE,EAAE,CAAC;QAC7F,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC;QACzD,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAK5D,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QAE9C,IAAI,IAAI,EAAE,CAAC;YAGT,GAAG,CACD,sCAAsC,IAAI,CAAC,GAAG,SAAS,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,IAAI,CAAC,SAAS,GAAG,CAAC,EAAE,CACtJ,CAAC;YAIF,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,IAAI,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC;YACzF,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,EAAE;gBACzF,WAAW,EAAE,aAAa;aAC3B,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC7E,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,YAAY,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;YAChF,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7C,CAAC;QASD,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACnC,GAAG,CACD,gFAAgF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CACpJ,CAAC;YAIF,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC5B,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC;QACzD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;YAC5B,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC5B,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;QACnD,CAAC;QASD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE;YAC7B,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;YAC3B,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QAEH,IAAI,OAAO,GAAuB,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QACrF,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAClE,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,GAAG,EAAE,CAAC;YACf,CAAC;QACH,CAAC;QAID,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,GAAG,IAAI,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACrE,MAAM,aAAa,GAAG,GAAG,KAAK,QAAQ,CAAC;QACvC,MAAM,iBAAiB,GAAG,MAAM,sBAAsB,CACpD,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,WAAW,EAChB,EAAE,WAAW,EAAE,aAAa,EAAE,CAC/B,CAAC;QAMF,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAC9D,IAAI,CAAC,MAAM,EACX,OAAO,EACP,aAAa,CACd,CAAC;QACF,MAAM,aAAa,GAAG,qBAAqB,CACzC,IAAI,CAAC,MAAM,EACX,OAAO,EACP;YACE,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;YAC3B,WAAW,EAAE,IAAI,CAAC,UAAU,IAAI,EAAE;YAClC,SAAS,EAAE,IAAI,CAAC,QAAQ,IAAI,EAAE;SAC/B,EACD,iBAAiB,EACjB,IAAI,EACJ,kBAAkB,CACnB,CAAC;QAGF,MAAM,eAAe,GAAG,aAAa;YACnC,CAAC,CAAC,IAAI,CAAC,YAAY,CACf,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,kBAAkB,CAAC,EACvD,IAAI,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAC3C;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,MAAM,WAAW,GAAgB,EAAE,CAAC;QACpC,IAAI,OAAO,GAAyB,IAAI,CAAC;QACzC,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAChC,UAAU,EACV,IAAI,CAAC,UAAU,IAAI,EAAE,EACrB,aAAa,EACb,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,eAAe,EAAE,CAC/F,CAAC;YAMF,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;gBAC7B,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBAC/B,OAAO,CAAC,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;YAC/E,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACpC,CAAC;QACD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC5B,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;QACvD,CAAC;QACD,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IACjE,CAAC;IAIS,eAAe,CAAC,IAAmB,EAAE,MAAmB,EAAE,QAAgB;QAOlF,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,CAAC;QAGD,IAAI,MAAM,CAAC,GAAG,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,CAAC;YAC7C,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3B,MAAM,aAAa,GAAwC,EAAE,CAAC;gBAC9D,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;wBAAE,SAAS;oBAC3E,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,CAAC;wBAClD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;wBAC7B,QAAQ,GAAG,IAAI,CAAC;wBAChB,SAAS;oBACX,CAAC;oBACD,aAAa,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC/D,CAAC;gBACD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC7B,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;gBAC1C,CAAC;gBACD,IAAI,QAAQ,EAAE,CAAC;oBAIb,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC9C,IAAI,IAAI;wBAAE,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;;wBAC3B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACtF,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAES,aAAa,CAAC,KAAoB,EAAE,KAAa;IAE3D,CAAC;IAES,KAAK,CAAC,YAAY,CAC1B,IAAmB,EACnB,KAAoB,EACpB,OAA8B;QAE9B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAG1C,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;QAG/E,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAWpC,MAAM,UAAU,GAAuB,IAAI,CAAC,MAAM,CAAC;QACnD,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC,MAAM,KAAK,UAAU,CAC7C,CAAC;YACF,IAAI,CAAC,SAAS;gBAAE,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,IAAI;YAAE,OAAO;QAEjB,MAAM,MAAM,GAAuB,IAAI,CAAC,MAAM,CAAC;QAC/C,MAAM,OAAO,GAAuB,IAAI,CAAC,OAAO,CAAC;QACjD,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO;YAAE,OAAO;QAEhC,MAAM,OAAO,GAAG,IAAI;YAClB,CAAC,CAAC,0CAA0C,IAAI,UAAU;YAC1D,CAAC,CAAC,+BAA+B,CAAC;QAEpC,GAAG,CAAC,4CAA4C,MAAM,UAAU,OAAO,QAAQ,IAAI,CAAC,GAAG,cAAc,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QACpH,MAAM,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QACtF,MAAM,mBAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC3D,CAAC;IAOD,aAAa,CAAC,IAAmB,EAAE,KAA0C;QAC3E,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC/B,MAAM,MAAM,GAAuB,IAAI,CAAC,MAAM,CAAC;QAC/C,MAAM,OAAO,GAAuB,IAAI,CAAC,OAAO,CAAC;QAMjD,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO;YAAE,OAAO;QAEhC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;YACrD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC;QACD,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,IAAI,CAAC,KAAK,IAAI,wBAAwB;YAAE,OAAO;QACnD,IAAI,GAAG,GAAG,IAAI,CAAC,UAAU,GAAG,wBAAwB;YAAE,OAAO;QAE7D,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO;YAAE,OAAO;QAErB,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;QACtB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;QAEhB,MAAM,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QAKtF,KAAK,mBAAmB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC;IAChF,CAAC;IAED,mBAAmB,CAAC,KAA0C;QAC5D,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,SAAS,GAAG,CAAC;QAChD,OAAO,IAAI,SAAS,MAAM,KAAK,CAAC,MAAM,GAAG,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,gBAAgB,CAAC,IAAY,EAAE,KAAU;QACvC,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACtD,OAAO,OAAO,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,OAAO,MAAM,OAAO,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,OAAO,EAAE,CAAC;IAC5E,CAAC;IAED,gBAAgB,CAAC,IAAY;QAG3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,YAAY,CAAC,IAAY;QACvB,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC;YAAE,OAAO,IAAI,CAAC;QAC3C,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC;QAC1E,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO,IAAI,KAAK,KAAK,cAAc;YAAE,OAAO,IAAI,CAAC;QACnF,IAAI,KAAK,KAAK,MAAM;YAAE,OAAO,IAAI,CAAC;QAClC,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,WAAW;YAAE,OAAO,IAAI,CAAC;QAC/D,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO;YAAE,OAAO,IAAI,CAAC;QACvD,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,mBAAmB,CAAC,IAAY,EAAE,KAAU;QAC1C,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACjC,IAAI,GAAG,GAAY,EAAE,CAAC;QACtB,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO,IAAI,KAAK,KAAK,cAAc,EAAE,CAAC;YAC1F,GAAG,GAAG,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC;QACrD,CAAC;aAAM,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YAC5B,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,CAAC;aAAM,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YAC5B,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,CAAC;aAAM,IAAI,KAAK,KAAK,MAAM,EAAE,CAAC;YAC5B,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,EAAE,CAAC;QAC5B,CAAC;aAAM,IAAI,KAAK,KAAK,UAAU,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;YACzD,GAAG,GAAG,KAAK,CAAC,GAAG,IAAI,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;QACvC,CAAC;aAAM,IAAI,KAAK,KAAK,MAAM,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YACjD,GAAG,GAAG,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,aAAa,IAAI,EAAE,CAAC;QACvD,CAAC;aAAM,CAAC;YAGN,MAAM,SAAS,GAAG,CAAC,WAAW,EAAE,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;YACzF,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;gBAC1B,IAAI,OAAQ,KAAa,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAK,KAAa,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC/D,GAAG,GAAI,KAAa,CAAC,CAAC,CAAC,CAAC;oBACxB,MAAM;gBACR,CAAC;YACH,CAAC;YACD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,KAAK,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;oBACrC,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,EAAE,CAAC;wBAC/B,GAAG,GAAG,CAAC,CAAC;wBACR,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,CAAC,GAAG,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC,CAAC;QAG1D,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,IAAI,CAAC,CAAC;YAAE,OAAO,EAAE,CAAC;QAClB,IAAI,CAAC,CAAC,MAAM,GAAG,oBAAoB,EAAE,CAAC;YACpC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,oBAAoB,GAAG,CAAC,CAAC,GAAG,GAAG,CAAC;QACjD,CAAC;QAGD,OAAO,CAAC,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC9C,CAAC;IASD,iBAAiB,CACf,IAA0B,EAC1B,WAAiC;QAEjC,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACzC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAC1C,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,IAAI,IAAI;YAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3B,KAAK,MAAM,EAAE,IAAI,qBAAqB,CAAC,WAAW,CAAC,EAAE,CAAC;YACpD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;QACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAUD,KAAK,CAAC,0BAA0B,CAC9B,MAAc,EACd,OAA2B,EAC3B,aAAsB;QAEtB,MAAM,GAAG,GAAG,IAAI,GAAG,EAA0C,CAAC;QAC9D,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CACzD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAC9C,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,uBAAuB,CAAC,CAAC;QACpD,IAAI,gBAAgB,GAAG,aAAa,CAAC,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC,CAAC;QACnE,IAAI,eAAe,GAAG,uBAAuB,CAAC;QAE9C,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;YACxB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;YACvE,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YAChC,MAAM,UAAU,GAAG,aAAa,IAAI,gBAAgB,GAAG,CAAC,CAAC;YACzD,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE;gBACxE,WAAW,EAAE,UAAU;gBACvB,WAAW,EAAE,KAAK;aACnB,CAAC,CAAC;YACH,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc;oBAAE,SAAS;gBAC1C,MAAM,EAAE,GAAG,iBAAiB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBAC/C,IAAI,gBAAgB,IAAI,CAAC,IAAI,EAAE,GAAG,eAAe,EAAE,CAAC;oBAGlD,GAAG,CAAC,IAAI,GAAG,eAAe,CAAC;oBAC3B,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC;oBAC7B,GAAG,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,oDAAoD,CAAC;oBAC5E,SAAS;gBACX,CAAC;gBACD,gBAAgB,IAAI,CAAC,CAAC;gBACtB,eAAe,IAAI,EAAE,CAAC;YACxB,CAAC;YACD,GAAG,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC3B,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAID,qBAAqB,CACnB,OAA2B,EAC3B,QAAqD;QAErD,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM,CACzD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAC9C,CAAC;QACF,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,uBAAuB,CAAC,EAAE,CAAC;YACzD,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;gBACvB,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;oBACpD,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,SAAS,IAAI,WAAW,EAAE,IAAI,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;gBACpF,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAKD,YAAY,CAAC,GAAG,KAAqC;QACnD,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,CAAC,IAAI,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM;gBAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5D,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IAChD,CAAC;IAKD,kBAAkB,CAAC,WAAiC;QAClD,MAAM,MAAM,GAAgB,EAAE,CAAC;QAC/B,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;gBACpD,MAAM,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,SAAS,IAAI,WAAW,EAAE,IAAI,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC,CAAC;YACpF,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IAChD,CAAC;IAED,SAAS;QAIP,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrD,MAAM,EAAE,CAAC,CAAC,MAAM;YAChB,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,GAAG,EAAE,CAAC,CAAC,GAAG;YACV,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,SAAS,EAAE,CAAC,CAAC,SAAS;YACtB,aAAa,EAAE,CAAC,CAAC,aAAa;SAC/B,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
export interface CircuitBreakerEntry {
|
|
2
|
+
consecutiveFailures: number;
|
|
3
|
+
lastFailureAt: number;
|
|
4
|
+
lastExitCode: number | null;
|
|
5
|
+
lastExitTail: string;
|
|
6
|
+
open: boolean;
|
|
7
|
+
openedAt: number;
|
|
8
|
+
}
|
|
9
|
+
export declare class CircuitBreaker {
|
|
10
|
+
#private;
|
|
11
|
+
constructor(opts?: {
|
|
12
|
+
threshold?: number;
|
|
13
|
+
cooldownMs?: number;
|
|
14
|
+
maxKeys?: number;
|
|
15
|
+
staleMs?: number;
|
|
16
|
+
});
|
|
17
|
+
static key(agentId: string, ticketId: string, role: string): string;
|
|
18
|
+
static isTransientExit(code: number | null): boolean;
|
|
19
|
+
record(key: string, exitCode: number | null, tailSnippet?: string, opts?: {
|
|
20
|
+
forceOpen?: boolean;
|
|
21
|
+
}): {
|
|
22
|
+
justOpened: boolean;
|
|
23
|
+
entry: CircuitBreakerEntry;
|
|
24
|
+
};
|
|
25
|
+
shouldBlock(key: string): string | null;
|
|
26
|
+
reset(key: string): void;
|
|
27
|
+
resetAgent(agentId: string): void;
|
|
28
|
+
sweep(now?: number): number;
|
|
29
|
+
getOpenBreakers(): Array<{
|
|
30
|
+
key: string;
|
|
31
|
+
entry: CircuitBreakerEntry;
|
|
32
|
+
}>;
|
|
33
|
+
get size(): number;
|
|
34
|
+
}
|
|
@@ -0,0 +1,146 @@
|
|
|
1
|
+
import { log } from './logging.js';
|
|
2
|
+
const TRANSIENT_EXIT_CODES = new Set([
|
|
3
|
+
143,
|
|
4
|
+
137,
|
|
5
|
+
130,
|
|
6
|
+
]);
|
|
7
|
+
const DEFAULT_THRESHOLD = 5;
|
|
8
|
+
const BREAKER_COOLDOWN_MS = 60 * 60_000;
|
|
9
|
+
const DEFAULT_MAX_KEYS = 2000;
|
|
10
|
+
export class CircuitBreaker {
|
|
11
|
+
#state = new Map();
|
|
12
|
+
#threshold;
|
|
13
|
+
#cooldownMs;
|
|
14
|
+
#maxKeys;
|
|
15
|
+
#staleMs;
|
|
16
|
+
constructor(opts) {
|
|
17
|
+
this.#threshold = opts?.threshold ?? DEFAULT_THRESHOLD;
|
|
18
|
+
this.#cooldownMs = opts?.cooldownMs ?? BREAKER_COOLDOWN_MS;
|
|
19
|
+
this.#maxKeys = opts?.maxKeys ?? DEFAULT_MAX_KEYS;
|
|
20
|
+
this.#staleMs = opts?.staleMs ?? this.#cooldownMs;
|
|
21
|
+
}
|
|
22
|
+
static key(agentId, ticketId, role) {
|
|
23
|
+
return `${agentId}:${ticketId}:${role || '_'}`;
|
|
24
|
+
}
|
|
25
|
+
static isTransientExit(code) {
|
|
26
|
+
if (code === null)
|
|
27
|
+
return true;
|
|
28
|
+
if (code === 0)
|
|
29
|
+
return false;
|
|
30
|
+
return TRANSIENT_EXIT_CODES.has(code);
|
|
31
|
+
}
|
|
32
|
+
record(key, exitCode, tailSnippet, opts) {
|
|
33
|
+
const now = Date.now();
|
|
34
|
+
let entry = this.#state.get(key);
|
|
35
|
+
if (!entry) {
|
|
36
|
+
this.#sweepStale(now);
|
|
37
|
+
this.#enforceCap();
|
|
38
|
+
entry = {
|
|
39
|
+
consecutiveFailures: 0,
|
|
40
|
+
lastFailureAt: 0,
|
|
41
|
+
lastExitCode: null,
|
|
42
|
+
lastExitTail: '',
|
|
43
|
+
open: false,
|
|
44
|
+
openedAt: 0,
|
|
45
|
+
};
|
|
46
|
+
this.#state.set(key, entry);
|
|
47
|
+
}
|
|
48
|
+
entry.consecutiveFailures += 1;
|
|
49
|
+
entry.lastFailureAt = now;
|
|
50
|
+
entry.lastExitCode = exitCode;
|
|
51
|
+
entry.lastExitTail = (tailSnippet || '').slice(0, 200);
|
|
52
|
+
let justOpened = false;
|
|
53
|
+
const crossedThreshold = entry.consecutiveFailures >= this.#threshold;
|
|
54
|
+
if (!entry.open && (crossedThreshold || opts?.forceOpen)) {
|
|
55
|
+
entry.open = true;
|
|
56
|
+
entry.openedAt = now;
|
|
57
|
+
justOpened = true;
|
|
58
|
+
log(`[circuit-breaker] OPEN key=${key} failures=${entry.consecutiveFailures} ` +
|
|
59
|
+
`exit=${exitCode}${opts?.forceOpen ? ' (forced — non-retryable)' : ''} — ` +
|
|
60
|
+
`blocking re-dispatch until operator intervenes`);
|
|
61
|
+
}
|
|
62
|
+
else {
|
|
63
|
+
log(`[circuit-breaker] recorded failure key=${key} count=${entry.consecutiveFailures}/${this.#threshold} exit=${exitCode}`);
|
|
64
|
+
}
|
|
65
|
+
return { justOpened, entry };
|
|
66
|
+
}
|
|
67
|
+
shouldBlock(key) {
|
|
68
|
+
const entry = this.#state.get(key);
|
|
69
|
+
if (!entry || !entry.open)
|
|
70
|
+
return null;
|
|
71
|
+
const elapsed = Date.now() - entry.openedAt;
|
|
72
|
+
if (elapsed >= this.#cooldownMs) {
|
|
73
|
+
log(`[circuit-breaker] HALF-OPEN probe allowed key=${key} ` +
|
|
74
|
+
`(${Math.round(elapsed / 60_000)}min since open)`);
|
|
75
|
+
entry.openedAt = Date.now();
|
|
76
|
+
return null;
|
|
77
|
+
}
|
|
78
|
+
return (`circuit_breaker_open (${entry.consecutiveFailures} consecutive non-transient exits, ` +
|
|
79
|
+
`last exit=${entry.lastExitCode}, opened ${Math.round((Date.now() - entry.openedAt) / 1000)}s ago)`);
|
|
80
|
+
}
|
|
81
|
+
reset(key) {
|
|
82
|
+
if (this.#state.has(key)) {
|
|
83
|
+
log(`[circuit-breaker] RESET key=${key}`);
|
|
84
|
+
this.#state.delete(key);
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
resetAgent(agentId) {
|
|
88
|
+
for (const key of [...this.#state.keys()]) {
|
|
89
|
+
if (key.startsWith(agentId + ':')) {
|
|
90
|
+
log(`[circuit-breaker] RESET (agent restart) key=${key}`);
|
|
91
|
+
this.#state.delete(key);
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
sweep(now = Date.now()) {
|
|
96
|
+
return this.#sweepStale(now);
|
|
97
|
+
}
|
|
98
|
+
#sweepStale(now) {
|
|
99
|
+
let removed = 0;
|
|
100
|
+
for (const [key, entry] of this.#state) {
|
|
101
|
+
if (now - entry.lastFailureAt > this.#staleMs) {
|
|
102
|
+
this.#state.delete(key);
|
|
103
|
+
removed++;
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
if (removed > 0) {
|
|
107
|
+
log(`[circuit-breaker] swept ${removed} stale key(s) — ${this.#state.size} remaining`);
|
|
108
|
+
}
|
|
109
|
+
return removed;
|
|
110
|
+
}
|
|
111
|
+
#enforceCap() {
|
|
112
|
+
if (this.#state.size < this.#maxKeys)
|
|
113
|
+
return;
|
|
114
|
+
let oldestClosed = null;
|
|
115
|
+
let oldestClosedAt = Infinity;
|
|
116
|
+
let oldestAny = null;
|
|
117
|
+
let oldestAnyAt = Infinity;
|
|
118
|
+
for (const [key, entry] of this.#state) {
|
|
119
|
+
if (entry.lastFailureAt < oldestAnyAt) {
|
|
120
|
+
oldestAnyAt = entry.lastFailureAt;
|
|
121
|
+
oldestAny = key;
|
|
122
|
+
}
|
|
123
|
+
if (!entry.open && entry.lastFailureAt < oldestClosedAt) {
|
|
124
|
+
oldestClosedAt = entry.lastFailureAt;
|
|
125
|
+
oldestClosed = key;
|
|
126
|
+
}
|
|
127
|
+
}
|
|
128
|
+
const victim = oldestClosed ?? oldestAny;
|
|
129
|
+
if (victim !== null) {
|
|
130
|
+
this.#state.delete(victim);
|
|
131
|
+
log(`[circuit-breaker] cap reached (${this.#maxKeys}) — evicted key=${victim}`);
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
getOpenBreakers() {
|
|
135
|
+
const results = [];
|
|
136
|
+
for (const [key, entry] of this.#state) {
|
|
137
|
+
if (entry.open)
|
|
138
|
+
results.push({ key, entry });
|
|
139
|
+
}
|
|
140
|
+
return results;
|
|
141
|
+
}
|
|
142
|
+
get size() {
|
|
143
|
+
return this.#state.size;
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
//# sourceMappingURL=circuit-breaker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"circuit-breaker.js","sourceRoot":"","sources":["../../src/lib/circuit-breaker.ts"],"names":[],"mappings":"AAcA,OAAO,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAKnC,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAC;IACnC,GAAG;IACH,GAAG;IACH,GAAG;CACJ,CAAC,CAAC;AAMH,MAAM,iBAAiB,GAAG,CAAC,CAAC;AAK5B,MAAM,mBAAmB,GAAG,EAAE,GAAG,MAAM,CAAC;AASxC,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAY9B,MAAM,OAAO,cAAc;IAChB,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;IAChD,UAAU,CAAS;IACnB,WAAW,CAAS;IACpB,QAAQ,CAAS;IAMjB,QAAQ,CAAS;IAE1B,YAAY,IAKX;QACC,IAAI,CAAC,UAAU,GAAG,IAAI,EAAE,SAAS,IAAI,iBAAiB,CAAC;QACvD,IAAI,CAAC,WAAW,GAAG,IAAI,EAAE,UAAU,IAAI,mBAAmB,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,IAAI,EAAE,OAAO,IAAI,gBAAgB,CAAC;QAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,EAAE,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC;IACpD,CAAC;IAED,MAAM,CAAC,GAAG,CAAC,OAAe,EAAE,QAAgB,EAAE,IAAY;QACxD,OAAO,GAAG,OAAO,IAAI,QAAQ,IAAI,IAAI,IAAI,GAAG,EAAE,CAAC;IACjD,CAAC;IAID,MAAM,CAAC,eAAe,CAAC,IAAmB;QACxC,IAAI,IAAI,KAAK,IAAI;YAAE,OAAO,IAAI,CAAC;QAC/B,IAAI,IAAI,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAC7B,OAAO,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACxC,CAAC;IAYD,MAAM,CACJ,GAAW,EACX,QAAuB,EACvB,WAAoB,EACpB,IAA8B;QAE9B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,IAAI,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,CAAC,KAAK,EAAE,CAAC;YAKX,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;YACtB,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,KAAK,GAAG;gBACN,mBAAmB,EAAE,CAAC;gBACtB,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,IAAI;gBAClB,YAAY,EAAE,EAAE;gBAChB,IAAI,EAAE,KAAK;gBACX,QAAQ,EAAE,CAAC;aACZ,CAAC;YACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC9B,CAAC;QAED,KAAK,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC/B,KAAK,CAAC,aAAa,GAAG,GAAG,CAAC;QAC1B,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC;QAC9B,KAAK,CAAC,YAAY,GAAG,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAEvD,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,MAAM,gBAAgB,GAAG,KAAK,CAAC,mBAAmB,IAAI,IAAI,CAAC,UAAU,CAAC;QACtE,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,EAAE,SAAS,CAAC,EAAE,CAAC;YACzD,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;YAClB,KAAK,CAAC,QAAQ,GAAG,GAAG,CAAC;YACrB,UAAU,GAAG,IAAI,CAAC;YAClB,GAAG,CACD,8BAA8B,GAAG,aAAa,KAAK,CAAC,mBAAmB,GAAG;gBACxE,QAAQ,QAAQ,GAAG,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,EAAE,KAAK;gBAC1E,gDAAgD,CACnD,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,GAAG,CACD,0CAA0C,GAAG,UAAU,KAAK,CAAC,mBAAmB,IAAI,IAAI,CAAC,UAAU,SAAS,QAAQ,EAAE,CACvH,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;IAC/B,CAAC;IAMD,WAAW,CAAC,GAAW;QACrB,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnC,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QAGvC,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC5C,IAAI,OAAO,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YAChC,GAAG,CACD,iDAAiD,GAAG,GAAG;gBACrD,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,iBAAiB,CACpD,CAAC;YAGF,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,CACL,yBAAyB,KAAK,CAAC,mBAAmB,oCAAoC;YACtF,aAAa,KAAK,CAAC,YAAY,YAAY,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC,QAAQ,CACpG,CAAC;IACJ,CAAC;IAMD,KAAK,CAAC,GAAW;QACf,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,GAAG,CAAC,+BAA+B,GAAG,EAAE,CAAC,CAAC;YAC1C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAGD,UAAU,CAAC,OAAe;QACxB,KAAK,MAAM,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YAC1C,IAAI,GAAG,CAAC,UAAU,CAAC,OAAO,GAAG,GAAG,CAAC,EAAE,CAAC;gBAClC,GAAG,CAAC,+CAA+C,GAAG,EAAE,CAAC,CAAC;gBAC1D,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAUD,KAAK,CAAC,MAAc,IAAI,CAAC,GAAG,EAAE;QAC5B,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC/B,CAAC;IAED,WAAW,CAAC,GAAW;QACrB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACvC,IAAI,GAAG,GAAG,KAAK,CAAC,aAAa,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC9C,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACxB,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;QACD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,GAAG,CAAC,2BAA2B,OAAO,mBAAmB,IAAI,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,CAAC;QACzF,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAQD,WAAW;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,QAAQ;YAAE,OAAO;QAC7C,IAAI,YAAY,GAAkB,IAAI,CAAC;QACvC,IAAI,cAAc,GAAG,QAAQ,CAAC;QAC9B,IAAI,SAAS,GAAkB,IAAI,CAAC;QACpC,IAAI,WAAW,GAAG,QAAQ,CAAC;QAC3B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACvC,IAAI,KAAK,CAAC,aAAa,GAAG,WAAW,EAAE,CAAC;gBACtC,WAAW,GAAG,KAAK,CAAC,aAAa,CAAC;gBAClC,SAAS,GAAG,GAAG,CAAC;YAClB,CAAC;YACD,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,aAAa,GAAG,cAAc,EAAE,CAAC;gBACxD,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;gBACrC,YAAY,GAAG,GAAG,CAAC;YACrB,CAAC;QACH,CAAC;QACD,MAAM,MAAM,GAAG,YAAY,IAAI,SAAS,CAAC;QACzC,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;YACpB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC3B,GAAG,CAAC,kCAAkC,IAAI,CAAC,QAAQ,mBAAmB,MAAM,EAAE,CAAC,CAAC;QAClF,CAAC;IACH,CAAC;IAGD,eAAe;QACb,MAAM,OAAO,GAAuD,EAAE,CAAC;QACvE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACvC,IAAI,KAAK,CAAC,IAAI;gBAAE,OAAO,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAGD,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;IAC1B,CAAC;CACF"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { type AdapterCredential, type AdapterMcpContext, CliAdapter, type OneshotSpec, type ParseResult, type SpawnDescriptor } from './base.js';
|
|
2
|
+
export declare class AntigravityCliAdapter extends CliAdapter {
|
|
3
|
+
#private;
|
|
4
|
+
static cliType: string;
|
|
5
|
+
constructor();
|
|
6
|
+
resolveBin(configured?: string | null): string;
|
|
7
|
+
buildOneshotSpawn({ rolePrompt, taskText, model }: OneshotSpec): SpawnDescriptor;
|
|
8
|
+
parseStdoutLine(line: string): ParseResult;
|
|
9
|
+
collectOneshotResult(lines: string[]): string | null;
|
|
10
|
+
configDirEnv(): string;
|
|
11
|
+
authEnvKeys(): string[];
|
|
12
|
+
prepareCliHome(cliHomeDir: string, credential?: AdapterCredential | null, mcp?: AdapterMcpContext | null): Promise<{
|
|
13
|
+
extraEnv: Record<string, string>;
|
|
14
|
+
}>;
|
|
15
|
+
}
|