@octelium/console-agent 0.1.0-b-ca02.1.1
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 +267 -0
- package/README.md +207 -0
- package/apis/octelium.binpb +0 -0
- package/dist/agent/pi/auth.d.ts +32 -0
- package/dist/agent/pi/auth.js +279 -0
- package/dist/agent/pi/auth.js.map +1 -0
- package/dist/agent/pi/backend.d.ts +77 -0
- package/dist/agent/pi/backend.js +518 -0
- package/dist/agent/pi/backend.js.map +1 -0
- package/dist/agent/pi/mapper.d.ts +33 -0
- package/dist/agent/pi/mapper.js +386 -0
- package/dist/agent/pi/mapper.js.map +1 -0
- package/dist/agent/pi/model.d.ts +47 -0
- package/dist/agent/pi/model.js +199 -0
- package/dist/agent/pi/model.js.map +1 -0
- package/dist/agent/prompt.d.ts +22 -0
- package/dist/agent/prompt.js +75 -0
- package/dist/agent/prompt.js.map +1 -0
- package/dist/agent/resources.d.ts +13 -0
- package/dist/agent/resources.js +78 -0
- package/dist/agent/resources.js.map +1 -0
- package/dist/agent/tools/builtin.d.ts +13 -0
- package/dist/agent/tools/builtin.js +80 -0
- package/dist/agent/tools/builtin.js.map +1 -0
- package/dist/agent/tools/octelium.d.ts +28 -0
- package/dist/agent/tools/octelium.js +236 -0
- package/dist/agent/tools/octelium.js.map +1 -0
- package/dist/agent/tools/present.d.ts +61 -0
- package/dist/agent/tools/present.js +273 -0
- package/dist/agent/tools/present.js.map +1 -0
- package/dist/agent/tools/util.d.ts +19 -0
- package/dist/agent/tools/util.js +103 -0
- package/dist/agent/tools/util.js.map +1 -0
- package/dist/agent/tools/web.d.ts +24 -0
- package/dist/agent/tools/web.js +237 -0
- package/dist/agent/tools/web.js.map +1 -0
- package/dist/agent/types.d.ts +60 -0
- package/dist/agent/types.js +2 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/app.d.ts +36 -0
- package/dist/app.js +126 -0
- package/dist/app.js.map +1 -0
- package/dist/cli.d.ts +2 -0
- package/dist/cli.js +249 -0
- package/dist/cli.js.map +1 -0
- package/dist/config.d.ts +159 -0
- package/dist/config.js +251 -0
- package/dist/config.js.map +1 -0
- package/dist/errors.d.ts +13 -0
- package/dist/errors.js +18 -0
- package/dist/errors.js.map +1 -0
- package/dist/index.d.ts +6 -0
- package/dist/index.js +6 -0
- package/dist/index.js.map +1 -0
- package/dist/log.d.ts +11 -0
- package/dist/log.js +41 -0
- package/dist/log.js.map +1 -0
- package/dist/octelium/catalog.d.ts +61 -0
- package/dist/octelium/catalog.js +595 -0
- package/dist/octelium/catalog.js.map +1 -0
- package/dist/octelium/client.d.ts +53 -0
- package/dist/octelium/client.js +201 -0
- package/dist/octelium/client.js.map +1 -0
- package/dist/octelium/search.d.ts +24 -0
- package/dist/octelium/search.js +211 -0
- package/dist/octelium/search.js.map +1 -0
- package/dist/protocol/index.d.ts +432 -0
- package/dist/protocol/index.js +44 -0
- package/dist/protocol/index.js.map +1 -0
- package/dist/runs/builder.d.ts +17 -0
- package/dist/runs/builder.js +81 -0
- package/dist/runs/builder.js.map +1 -0
- package/dist/runs/manager.d.ts +52 -0
- package/dist/runs/manager.js +536 -0
- package/dist/runs/manager.js.map +1 -0
- package/dist/server/http.d.ts +9 -0
- package/dist/server/http.js +100 -0
- package/dist/server/http.js.map +1 -0
- package/dist/server/server.d.ts +48 -0
- package/dist/server/server.js +483 -0
- package/dist/server/server.js.map +1 -0
- package/dist/skills.d.ts +4 -0
- package/dist/skills.js +69 -0
- package/dist/skills.js.map +1 -0
- package/dist/store/conversations.d.ts +26 -0
- package/dist/store/conversations.js +183 -0
- package/dist/store/conversations.js.map +1 -0
- package/dist/store/files.d.ts +27 -0
- package/dist/store/files.js +161 -0
- package/dist/store/files.js.map +1 -0
- package/dist/store/settings.d.ts +14 -0
- package/dist/store/settings.js +48 -0
- package/dist/store/settings.js.map +1 -0
- package/dist/version.d.ts +1 -0
- package/dist/version.js +11 -0
- package/dist/version.js.map +1 -0
- package/package.json +57 -0
|
@@ -0,0 +1,518 @@
|
|
|
1
|
+
import * as fs from "node:fs";
|
|
2
|
+
import * as path from "node:path";
|
|
3
|
+
import { createAgentSession, DefaultResourceLoader, ModelRuntime, SessionManager, SettingsManager, } from "@earendil-works/pi-coding-agent";
|
|
4
|
+
import { THINKING_LEVELS } from "../../config.js";
|
|
5
|
+
import { badRequest } from "../../errors.js";
|
|
6
|
+
import { SettingsStore } from "../../store/settings.js";
|
|
7
|
+
import { buildSystemPrompt } from "../prompt.js";
|
|
8
|
+
import { createBuiltinTools, filterAvailableTools } from "../tools/builtin.js";
|
|
9
|
+
import { createOcteliumTools } from "../tools/octelium.js";
|
|
10
|
+
import { createPresentationTools, PRESENTATION_TOOL_NAMES, } from "../tools/present.js";
|
|
11
|
+
import { createWebTools } from "../tools/web.js";
|
|
12
|
+
import { AuthManager } from "./auth.js";
|
|
13
|
+
import { PiEventMapper } from "./mapper.js";
|
|
14
|
+
import { resolveModel, toModelInfo } from "./model.js";
|
|
15
|
+
const maxImageBytes = 5 * 1024 * 1024;
|
|
16
|
+
export const PREFERRED_MODELS = {
|
|
17
|
+
anthropic: "claude-opus-4-8",
|
|
18
|
+
openai: "gpt-5.5",
|
|
19
|
+
};
|
|
20
|
+
const imageMimeTypes = new Set([
|
|
21
|
+
"image/png",
|
|
22
|
+
"image/jpeg",
|
|
23
|
+
"image/gif",
|
|
24
|
+
"image/webp",
|
|
25
|
+
]);
|
|
26
|
+
export const formatBytes = (n) => {
|
|
27
|
+
if (n < 1024) {
|
|
28
|
+
return `${n} B`;
|
|
29
|
+
}
|
|
30
|
+
if (n < 1024 * 1024) {
|
|
31
|
+
return `${(n / 1024).toFixed(1)} KB`;
|
|
32
|
+
}
|
|
33
|
+
return `${(n / (1024 * 1024)).toFixed(1)} MB`;
|
|
34
|
+
};
|
|
35
|
+
export class PiBackend {
|
|
36
|
+
model;
|
|
37
|
+
thinkingLevel;
|
|
38
|
+
auth;
|
|
39
|
+
opts;
|
|
40
|
+
settings;
|
|
41
|
+
agentDir;
|
|
42
|
+
modelRuntime;
|
|
43
|
+
resolvedModel;
|
|
44
|
+
modelError;
|
|
45
|
+
resolving;
|
|
46
|
+
sessions = new Map();
|
|
47
|
+
pendingSessions = new Map();
|
|
48
|
+
systemPrompt = "";
|
|
49
|
+
toolNames = [];
|
|
50
|
+
builtinTools = [];
|
|
51
|
+
constructor(opts) {
|
|
52
|
+
this.opts = opts;
|
|
53
|
+
this.agentDir = path.join(opts.config.dataDir, "pi");
|
|
54
|
+
this.settings = new SettingsStore(opts.config.dataDir);
|
|
55
|
+
this.thinkingLevel =
|
|
56
|
+
this.settings.read().thinkingLevel ?? opts.config.llm.thinkingLevel;
|
|
57
|
+
}
|
|
58
|
+
get issues() {
|
|
59
|
+
return this.modelError ? [this.modelError] : [];
|
|
60
|
+
}
|
|
61
|
+
async init() {
|
|
62
|
+
process.env.PI_OFFLINE ??= "1";
|
|
63
|
+
process.env.PI_TELEMETRY ??= "0";
|
|
64
|
+
process.env.PI_SKIP_VERSION_CHECK ??= "1";
|
|
65
|
+
fs.mkdirSync(this.agentDir, { recursive: true, mode: 0o700 });
|
|
66
|
+
this.modelRuntime =
|
|
67
|
+
this.opts.modelRuntime ??
|
|
68
|
+
(await ModelRuntime.create({
|
|
69
|
+
authPath: path.join(this.agentDir, "auth.json"),
|
|
70
|
+
modelsPath: null,
|
|
71
|
+
allowModelNetwork: false,
|
|
72
|
+
refreshOnCreate: false,
|
|
73
|
+
}));
|
|
74
|
+
this.auth = new AuthManager({
|
|
75
|
+
modelRuntime: this.modelRuntime,
|
|
76
|
+
dataDir: this.opts.config.dataDir,
|
|
77
|
+
logger: this.opts.logger,
|
|
78
|
+
providers: this.opts.config.llm.loginProviders,
|
|
79
|
+
onLogin: (provider, selectModel) => this.onLogin(provider, selectModel),
|
|
80
|
+
onLogout: (provider) => this.onLogout(provider),
|
|
81
|
+
});
|
|
82
|
+
const { available, unavailable } = filterAvailableTools(this.opts.config.agent.tools);
|
|
83
|
+
this.builtinTools = available;
|
|
84
|
+
if (unavailable.length > 0) {
|
|
85
|
+
this.opts.logger.info("Some built-in tools are disabled since their binaries are not installed", { tools: unavailable.join(",") });
|
|
86
|
+
}
|
|
87
|
+
this.toolNames = this.createTools("init").map((t) => t.name);
|
|
88
|
+
this.systemPrompt = buildSystemPrompt(this.promptContext());
|
|
89
|
+
try {
|
|
90
|
+
await this.ensureModel();
|
|
91
|
+
}
|
|
92
|
+
catch (err) {
|
|
93
|
+
this.opts.logger.warn("Could not resolve the LLM model", {
|
|
94
|
+
error: err,
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
promptContext() {
|
|
99
|
+
const { config, identity, octelium, catalog } = this.opts;
|
|
100
|
+
return {
|
|
101
|
+
domain: config.octelium.domain,
|
|
102
|
+
user: identity?.user,
|
|
103
|
+
workspace: identity?.workspace,
|
|
104
|
+
workDir: config.workDir,
|
|
105
|
+
octeliumMode: octelium.mode,
|
|
106
|
+
octeliumUnavailableReason: this.opts.octeliumUnavailableReason,
|
|
107
|
+
apiIndex: catalog.formatIndex(),
|
|
108
|
+
tools: this.toolNames,
|
|
109
|
+
approvals: config.approvals,
|
|
110
|
+
append: config.agent.systemPromptAppend,
|
|
111
|
+
};
|
|
112
|
+
}
|
|
113
|
+
get prompt() {
|
|
114
|
+
return this.systemPrompt;
|
|
115
|
+
}
|
|
116
|
+
async ensureModel() {
|
|
117
|
+
if (this.resolvedModel) {
|
|
118
|
+
return this.resolvedModel;
|
|
119
|
+
}
|
|
120
|
+
if (this.opts.model) {
|
|
121
|
+
this.resolvedModel = this.opts.model;
|
|
122
|
+
this.model = toModelInfo(this.opts.model);
|
|
123
|
+
return this.resolvedModel;
|
|
124
|
+
}
|
|
125
|
+
this.resolving ??= this.resolveInitialModel()
|
|
126
|
+
.then((resolved) => {
|
|
127
|
+
this.resolvedModel = resolved.model;
|
|
128
|
+
this.model = resolved.info;
|
|
129
|
+
this.modelError = undefined;
|
|
130
|
+
this.opts.logger.info("Using the LLM model", {
|
|
131
|
+
provider: resolved.model.provider,
|
|
132
|
+
model: resolved.model.id,
|
|
133
|
+
api: resolved.model.api,
|
|
134
|
+
baseUrl: resolved.baseUrl,
|
|
135
|
+
});
|
|
136
|
+
return resolved.model;
|
|
137
|
+
})
|
|
138
|
+
.catch((err) => {
|
|
139
|
+
this.modelError = err.message;
|
|
140
|
+
throw err;
|
|
141
|
+
})
|
|
142
|
+
.finally(() => {
|
|
143
|
+
this.resolving = undefined;
|
|
144
|
+
});
|
|
145
|
+
return this.resolving;
|
|
146
|
+
}
|
|
147
|
+
async resolveInitialModel() {
|
|
148
|
+
let configured;
|
|
149
|
+
let configErr;
|
|
150
|
+
try {
|
|
151
|
+
configured = await resolveModel(this.opts.config.llm, {
|
|
152
|
+
modelRuntime: this.modelRuntime,
|
|
153
|
+
octelium: this.opts.octelium,
|
|
154
|
+
domain: this.opts.config.octelium.domain,
|
|
155
|
+
logger: this.opts.logger,
|
|
156
|
+
});
|
|
157
|
+
}
|
|
158
|
+
catch (err) {
|
|
159
|
+
configErr = err;
|
|
160
|
+
}
|
|
161
|
+
const saved = this.settings.read().model;
|
|
162
|
+
if (saved) {
|
|
163
|
+
const model = this.modelRuntime.getModel(saved.provider, saved.id);
|
|
164
|
+
if (model && this.modelRuntime.hasConfiguredAuth(model.provider)) {
|
|
165
|
+
return { model, info: toModelInfo(model) };
|
|
166
|
+
}
|
|
167
|
+
this.opts.logger.warn("The selected model is not available anymore", {
|
|
168
|
+
provider: saved.provider,
|
|
169
|
+
model: saved.id,
|
|
170
|
+
});
|
|
171
|
+
}
|
|
172
|
+
if (configured) {
|
|
173
|
+
return configured;
|
|
174
|
+
}
|
|
175
|
+
throw configErr;
|
|
176
|
+
}
|
|
177
|
+
async listModels() {
|
|
178
|
+
let error;
|
|
179
|
+
try {
|
|
180
|
+
await this.ensureModel();
|
|
181
|
+
}
|
|
182
|
+
catch (err) {
|
|
183
|
+
error = err;
|
|
184
|
+
}
|
|
185
|
+
const available = await this.modelRuntime.getAvailable();
|
|
186
|
+
return {
|
|
187
|
+
current: this.model,
|
|
188
|
+
thinkingLevel: this.thinkingLevel,
|
|
189
|
+
models: available.map(toModelInfo),
|
|
190
|
+
error: !this.model && error
|
|
191
|
+
? { code: "model_unavailable", message: error.message }
|
|
192
|
+
: undefined,
|
|
193
|
+
};
|
|
194
|
+
}
|
|
195
|
+
async setModel(req) {
|
|
196
|
+
if (typeof req?.provider !== "string" ||
|
|
197
|
+
typeof req.id !== "string" ||
|
|
198
|
+
req.provider === "" ||
|
|
199
|
+
req.id === "") {
|
|
200
|
+
throw badRequest('The "provider" and "id" fields are required');
|
|
201
|
+
}
|
|
202
|
+
if (req.thinkingLevel !== undefined &&
|
|
203
|
+
!THINKING_LEVELS.includes(req.thinkingLevel)) {
|
|
204
|
+
throw badRequest(`Invalid thinking level: ${req.thinkingLevel}`);
|
|
205
|
+
}
|
|
206
|
+
const model = this.modelRuntime.getModel(req.provider, req.id);
|
|
207
|
+
if (!model) {
|
|
208
|
+
throw badRequest(`Unknown model ${req.provider}/${req.id}`);
|
|
209
|
+
}
|
|
210
|
+
if (!this.modelRuntime.hasConfiguredAuth(model.provider)) {
|
|
211
|
+
throw badRequest(`The provider "${model.provider}" is not configured. Sign in to it first`);
|
|
212
|
+
}
|
|
213
|
+
this.applyModel(model, req.thinkingLevel);
|
|
214
|
+
return this.listModels();
|
|
215
|
+
}
|
|
216
|
+
applyModel(model, thinkingLevel) {
|
|
217
|
+
this.resolvedModel = model;
|
|
218
|
+
this.model = toModelInfo(model);
|
|
219
|
+
this.modelError = undefined;
|
|
220
|
+
if (thinkingLevel) {
|
|
221
|
+
this.thinkingLevel = thinkingLevel;
|
|
222
|
+
}
|
|
223
|
+
this.settings.update({
|
|
224
|
+
model: { provider: model.provider, id: model.id },
|
|
225
|
+
thinkingLevel: this.thinkingLevel,
|
|
226
|
+
});
|
|
227
|
+
this.opts.logger.info("Using the LLM model", {
|
|
228
|
+
provider: model.provider,
|
|
229
|
+
model: model.id,
|
|
230
|
+
});
|
|
231
|
+
for (const entry of this.sessions.values()) {
|
|
232
|
+
if (!entry.busy) {
|
|
233
|
+
this.syncSession(entry).catch((err) => this.opts.logger.warn("Could not update the session model", {
|
|
234
|
+
error: err,
|
|
235
|
+
}));
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
}
|
|
239
|
+
async syncSession(entry) {
|
|
240
|
+
const model = this.resolvedModel;
|
|
241
|
+
if (!model) {
|
|
242
|
+
return;
|
|
243
|
+
}
|
|
244
|
+
if (entry.session.model?.provider !== model.provider ||
|
|
245
|
+
entry.session.model?.id !== model.id) {
|
|
246
|
+
await entry.session.setModel(model);
|
|
247
|
+
}
|
|
248
|
+
if (this.thinkingLevel &&
|
|
249
|
+
entry.session.thinkingLevel !== this.thinkingLevel) {
|
|
250
|
+
entry.session.setThinkingLevel(this.thinkingLevel);
|
|
251
|
+
}
|
|
252
|
+
}
|
|
253
|
+
async onLogin(provider, selectModel) {
|
|
254
|
+
if (!selectModel && this.resolvedModel) {
|
|
255
|
+
return undefined;
|
|
256
|
+
}
|
|
257
|
+
const available = await this.modelRuntime.getAvailable(provider);
|
|
258
|
+
const model = available.find((m) => m.id === PREFERRED_MODELS[provider]) ??
|
|
259
|
+
available[0];
|
|
260
|
+
if (!model) {
|
|
261
|
+
return undefined;
|
|
262
|
+
}
|
|
263
|
+
this.applyModel(model);
|
|
264
|
+
return toModelInfo(model);
|
|
265
|
+
}
|
|
266
|
+
async onLogout(provider) {
|
|
267
|
+
if (this.settings.read().model?.provider === provider) {
|
|
268
|
+
this.settings.update({ model: undefined });
|
|
269
|
+
}
|
|
270
|
+
if (this.resolvedModel?.provider === provider) {
|
|
271
|
+
this.resolvedModel = undefined;
|
|
272
|
+
this.model = undefined;
|
|
273
|
+
this.modelError = `Signed out of the LLM provider "${provider}"`;
|
|
274
|
+
}
|
|
275
|
+
}
|
|
276
|
+
createTools(conversationId) {
|
|
277
|
+
const { config, catalog, octelium, files, runtime } = this.opts;
|
|
278
|
+
const base = { conversationId, workDir: config.workDir, runtime };
|
|
279
|
+
return [
|
|
280
|
+
...createBuiltinTools({
|
|
281
|
+
...base,
|
|
282
|
+
names: this.builtinTools,
|
|
283
|
+
approveBash: config.approvals.bash,
|
|
284
|
+
}),
|
|
285
|
+
...createOcteliumTools({
|
|
286
|
+
...base,
|
|
287
|
+
catalog,
|
|
288
|
+
client: octelium,
|
|
289
|
+
approvalPolicy: config.approvals.octeliumAPI,
|
|
290
|
+
}),
|
|
291
|
+
...createPresentationTools({ ...base, files }),
|
|
292
|
+
...createWebTools({ ...base, webSearch: config.webSearch }),
|
|
293
|
+
];
|
|
294
|
+
}
|
|
295
|
+
async getSession(conversationId) {
|
|
296
|
+
const existing = this.sessions.get(conversationId);
|
|
297
|
+
if (existing) {
|
|
298
|
+
if (existing.idleTimer) {
|
|
299
|
+
clearTimeout(existing.idleTimer);
|
|
300
|
+
existing.idleTimer = undefined;
|
|
301
|
+
}
|
|
302
|
+
return existing;
|
|
303
|
+
}
|
|
304
|
+
let pending = this.pendingSessions.get(conversationId);
|
|
305
|
+
if (!pending) {
|
|
306
|
+
pending = this.createSession(conversationId).finally(() => this.pendingSessions.delete(conversationId));
|
|
307
|
+
this.pendingSessions.set(conversationId, pending);
|
|
308
|
+
}
|
|
309
|
+
return pending;
|
|
310
|
+
}
|
|
311
|
+
async createSession(conversationId) {
|
|
312
|
+
const { config, store } = this.opts;
|
|
313
|
+
const model = await this.ensureModel();
|
|
314
|
+
const sessionDir = store.sessionDir(conversationId);
|
|
315
|
+
fs.mkdirSync(sessionDir, { recursive: true, mode: 0o700 });
|
|
316
|
+
const sessionFile = store.getSessionFile(conversationId);
|
|
317
|
+
const sessionManager = sessionFile && fs.existsSync(sessionFile)
|
|
318
|
+
? SessionManager.open(sessionFile, sessionDir, config.workDir)
|
|
319
|
+
: SessionManager.create(config.workDir, sessionDir, {
|
|
320
|
+
id: conversationId,
|
|
321
|
+
});
|
|
322
|
+
const loader = new DefaultResourceLoader({
|
|
323
|
+
cwd: config.workDir,
|
|
324
|
+
agentDir: this.agentDir,
|
|
325
|
+
noExtensions: true,
|
|
326
|
+
noPromptTemplates: true,
|
|
327
|
+
noThemes: true,
|
|
328
|
+
noContextFiles: !config.agent.loadContextFiles,
|
|
329
|
+
additionalSkillPaths: this.opts.skillPaths ?? [],
|
|
330
|
+
systemPrompt: buildSystemPrompt(this.promptContext()),
|
|
331
|
+
});
|
|
332
|
+
await loader.reload();
|
|
333
|
+
const { session } = await createAgentSession({
|
|
334
|
+
cwd: config.workDir,
|
|
335
|
+
agentDir: this.agentDir,
|
|
336
|
+
modelRuntime: this.modelRuntime,
|
|
337
|
+
model,
|
|
338
|
+
thinkingLevel: this.thinkingLevel,
|
|
339
|
+
noTools: "builtin",
|
|
340
|
+
customTools: this.createTools(conversationId),
|
|
341
|
+
resourceLoader: loader,
|
|
342
|
+
sessionManager,
|
|
343
|
+
settingsManager: SettingsManager.inMemory({
|
|
344
|
+
compaction: { enabled: config.agent.compaction },
|
|
345
|
+
retry: {
|
|
346
|
+
enabled: config.agent.maxRetries > 0,
|
|
347
|
+
maxRetries: config.agent.maxRetries,
|
|
348
|
+
},
|
|
349
|
+
enableInstallTelemetry: false,
|
|
350
|
+
enableAnalytics: false,
|
|
351
|
+
cacheWarming: "off",
|
|
352
|
+
defaultThinkingLevel: this.thinkingLevel,
|
|
353
|
+
}),
|
|
354
|
+
});
|
|
355
|
+
if (session.model?.provider !== model.provider ||
|
|
356
|
+
session.model?.id !== model.id) {
|
|
357
|
+
await session.setModel(model);
|
|
358
|
+
}
|
|
359
|
+
const entry = { session, busy: false };
|
|
360
|
+
this.sessions.set(conversationId, entry);
|
|
361
|
+
return entry;
|
|
362
|
+
}
|
|
363
|
+
scheduleIdle(conversationId, entry) {
|
|
364
|
+
if (entry.idleTimer) {
|
|
365
|
+
clearTimeout(entry.idleTimer);
|
|
366
|
+
}
|
|
367
|
+
entry.idleTimer = setTimeout(() => {
|
|
368
|
+
if (entry.busy || this.sessions.get(conversationId) !== entry) {
|
|
369
|
+
return;
|
|
370
|
+
}
|
|
371
|
+
this.sessions.delete(conversationId);
|
|
372
|
+
entry.session.dispose();
|
|
373
|
+
}, this.opts.config.agent.sessionIdleTimeoutSeconds * 1000);
|
|
374
|
+
entry.idleTimer.unref();
|
|
375
|
+
}
|
|
376
|
+
buildPrompt(input, model) {
|
|
377
|
+
const images = [];
|
|
378
|
+
const lines = [];
|
|
379
|
+
for (const file of input.attachments) {
|
|
380
|
+
lines.push(`- ${file.name} (${file.mimeType}, ${formatBytes(file.size)}): ${file.path}`);
|
|
381
|
+
if (model.input.includes("image") &&
|
|
382
|
+
imageMimeTypes.has(file.mimeType) &&
|
|
383
|
+
file.size <= maxImageBytes) {
|
|
384
|
+
try {
|
|
385
|
+
images.push({
|
|
386
|
+
type: "image",
|
|
387
|
+
data: fs.readFileSync(file.path).toString("base64"),
|
|
388
|
+
mimeType: file.mimeType,
|
|
389
|
+
});
|
|
390
|
+
}
|
|
391
|
+
catch (err) {
|
|
392
|
+
this.opts.logger.warn("Could not read an image attachment", {
|
|
393
|
+
path: file.path,
|
|
394
|
+
error: err,
|
|
395
|
+
});
|
|
396
|
+
}
|
|
397
|
+
}
|
|
398
|
+
}
|
|
399
|
+
const text = lines.length === 0
|
|
400
|
+
? input.text
|
|
401
|
+
: `${input.text}\n\n[The User attached the following files]\n${lines.join("\n")}`.trim();
|
|
402
|
+
return { text, images };
|
|
403
|
+
}
|
|
404
|
+
async run(input, ctx) {
|
|
405
|
+
const conversationId = ctx.conversation.id;
|
|
406
|
+
const model = await this.ensureModel();
|
|
407
|
+
const entry = await this.getSession(conversationId);
|
|
408
|
+
if (entry.busy) {
|
|
409
|
+
return {
|
|
410
|
+
status: "failed",
|
|
411
|
+
error: { message: "The conversation is busy" },
|
|
412
|
+
};
|
|
413
|
+
}
|
|
414
|
+
const mapper = new PiEventMapper(ctx.builder, { hiddenTools: new Set(PRESENTATION_TOOL_NAMES) }, ctx.setActivity);
|
|
415
|
+
const unsubscribe = entry.session.subscribe((event) => {
|
|
416
|
+
try {
|
|
417
|
+
mapper.handle(event);
|
|
418
|
+
}
|
|
419
|
+
catch (err) {
|
|
420
|
+
this.opts.logger.warn("Could not handle an agent event", {
|
|
421
|
+
type: event.type,
|
|
422
|
+
error: err,
|
|
423
|
+
});
|
|
424
|
+
}
|
|
425
|
+
});
|
|
426
|
+
const onAbort = () => {
|
|
427
|
+
void entry.session.abort();
|
|
428
|
+
};
|
|
429
|
+
ctx.signal.addEventListener("abort", onAbort, { once: true });
|
|
430
|
+
entry.busy = true;
|
|
431
|
+
try {
|
|
432
|
+
if (ctx.signal.aborted) {
|
|
433
|
+
return { status: "completed", usage: mapper.usage };
|
|
434
|
+
}
|
|
435
|
+
await this.syncSession(entry);
|
|
436
|
+
const { text, images } = this.buildPrompt(input, model);
|
|
437
|
+
await entry.session.prompt(text, {
|
|
438
|
+
images: images.length > 0 ? images : undefined,
|
|
439
|
+
expandPromptTemplates: false,
|
|
440
|
+
});
|
|
441
|
+
}
|
|
442
|
+
catch (err) {
|
|
443
|
+
if (ctx.signal.aborted) {
|
|
444
|
+
return { status: "completed", usage: mapper.usage };
|
|
445
|
+
}
|
|
446
|
+
return {
|
|
447
|
+
status: "failed",
|
|
448
|
+
error: { message: err.message ?? String(err) },
|
|
449
|
+
usage: mapper.usage,
|
|
450
|
+
};
|
|
451
|
+
}
|
|
452
|
+
finally {
|
|
453
|
+
unsubscribe();
|
|
454
|
+
ctx.signal.removeEventListener("abort", onAbort);
|
|
455
|
+
entry.busy = false;
|
|
456
|
+
if (entry.session.sessionFile) {
|
|
457
|
+
this.opts.store.setSessionFile(conversationId, entry.session.sessionFile);
|
|
458
|
+
}
|
|
459
|
+
this.scheduleIdle(conversationId, entry);
|
|
460
|
+
}
|
|
461
|
+
const last = mapper.lastAssistant;
|
|
462
|
+
if (last?.stopReason === "error") {
|
|
463
|
+
return {
|
|
464
|
+
status: "failed",
|
|
465
|
+
error: {
|
|
466
|
+
code: "model_error",
|
|
467
|
+
message: last.errorMessage ?? "The model request failed",
|
|
468
|
+
},
|
|
469
|
+
usage: mapper.usage,
|
|
470
|
+
};
|
|
471
|
+
}
|
|
472
|
+
return { status: "completed", usage: mapper.usage };
|
|
473
|
+
}
|
|
474
|
+
async generateTitle(text, signal) {
|
|
475
|
+
const model = await this.ensureModel();
|
|
476
|
+
const timeout = AbortSignal.timeout(30000);
|
|
477
|
+
const res = await this.modelRuntime.completeSimple(model, {
|
|
478
|
+
systemPrompt: "You generate short titles for chat conversations. Reply with a title of at most 6 words that summarizes the User's request. Reply with the title only, without quotes or trailing punctuation.",
|
|
479
|
+
messages: [
|
|
480
|
+
{ role: "user", content: text.slice(0, 4000), timestamp: Date.now() },
|
|
481
|
+
],
|
|
482
|
+
}, {
|
|
483
|
+
maxTokens: 512,
|
|
484
|
+
signal: signal ? AbortSignal.any([signal, timeout]) : timeout,
|
|
485
|
+
});
|
|
486
|
+
if (res.stopReason === "error" || res.stopReason === "aborted") {
|
|
487
|
+
return undefined;
|
|
488
|
+
}
|
|
489
|
+
const title = res.content
|
|
490
|
+
.filter((c) => c.type === "text")
|
|
491
|
+
.map((c) => c.text)
|
|
492
|
+
.join(" ")
|
|
493
|
+
.split("\n")
|
|
494
|
+
.map((line) => line.trim())
|
|
495
|
+
.find((line) => line !== "")
|
|
496
|
+
?.replace(/^["'`*#\s]+|["'`*.\s]+$/g, "")
|
|
497
|
+
.slice(0, 80);
|
|
498
|
+
return title || undefined;
|
|
499
|
+
}
|
|
500
|
+
forgetConversation(conversationId) {
|
|
501
|
+
const entry = this.sessions.get(conversationId);
|
|
502
|
+
if (!entry) {
|
|
503
|
+
return;
|
|
504
|
+
}
|
|
505
|
+
if (entry.idleTimer) {
|
|
506
|
+
clearTimeout(entry.idleTimer);
|
|
507
|
+
}
|
|
508
|
+
this.sessions.delete(conversationId);
|
|
509
|
+
entry.session.dispose();
|
|
510
|
+
}
|
|
511
|
+
async dispose() {
|
|
512
|
+
this.auth?.dispose();
|
|
513
|
+
for (const [id] of this.sessions) {
|
|
514
|
+
this.forgetConversation(id);
|
|
515
|
+
}
|
|
516
|
+
}
|
|
517
|
+
}
|
|
518
|
+
//# sourceMappingURL=backend.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"backend.js","sourceRoot":"","sources":["../../../src/agent/pi/backend.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAOlC,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,YAAY,EACZ,cAAc,EACd,eAAe,GAGhB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,eAAe,EAAe,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAW7C,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAsB,MAAM,cAAc,CAAC;AACrE,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC/E,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EACL,uBAAuB,EACvB,uBAAuB,GACxB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAQjD,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,WAAW,EAAsB,MAAM,YAAY,CAAC;AAE3E,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,CAAC;AAEtC,MAAM,CAAC,MAAM,gBAAgB,GAA2B;IACtD,SAAS,EAAE,iBAAiB;IAC5B,MAAM,EAAE,SAAS;CAClB,CAAC;AAEF,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC;IAC7B,WAAW;IACX,YAAY;IACZ,WAAW;IACX,YAAY;CACb,CAAC,CAAC;AAmCH,MAAM,CAAC,MAAM,WAAW,GAAG,CAAC,CAAS,EAAU,EAAE;IAC/C,IAAI,CAAC,GAAG,IAAI,EAAE,CAAC;QACb,OAAO,GAAG,CAAC,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,CAAC,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;QACpB,OAAO,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;IACvC,CAAC;IACD,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;AAChD,CAAC,CAAC;AAEF,MAAM,OAAO,SAAS;IACpB,KAAK,CAAa;IAClB,aAAa,CAAU;IACvB,IAAI,CAAe;IACX,IAAI,CAAmB;IACvB,QAAQ,CAAgB;IACxB,QAAQ,CAAS;IACjB,YAAY,CAAgB;IAC5B,aAAa,CAAc;IAC3B,UAAU,CAAU;IACpB,SAAS,CAAuB;IAChC,QAAQ,GAAG,IAAI,GAAG,EAAwB,CAAC;IAC3C,eAAe,GAAG,IAAI,GAAG,EAAiC,CAAC;IAC3D,YAAY,GAAG,EAAE,CAAC;IAClB,SAAS,GAAa,EAAE,CAAC;IACzB,YAAY,GAAa,EAAE,CAAC;IAEpC,YAAY,IAAsB;QAChC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACvD,IAAI,CAAC,aAAa;YAChB,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC;IACxE,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,CAAC,GAAG,CAAC,UAAU,KAAK,GAAG,CAAC;QAC/B,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK,GAAG,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,qBAAqB,KAAK,GAAG,CAAC;QAE1C,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAE9D,IAAI,CAAC,YAAY;YACf,IAAI,CAAC,IAAI,CAAC,YAAY;gBACtB,CAAC,MAAM,YAAY,CAAC,MAAM,CAAC;oBACzB,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC;oBAC/C,UAAU,EAAE,IAAI;oBAChB,iBAAiB,EAAE,KAAK;oBACxB,eAAe,EAAE,KAAK;iBACvB,CAAC,CAAC,CAAC;QAEN,IAAI,CAAC,IAAI,GAAG,IAAI,WAAW,CAAC;YAC1B,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO;YACjC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;YACxB,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc;YAC9C,OAAO,EAAE,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,WAAW,CAAC;YACvE,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;SAChD,CAAC,CAAC;QAEH,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,oBAAoB,CACrD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAC7B,CAAC;QACF,IAAI,CAAC,YAAY,GAAG,SAAS,CAAC;QAC9B,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CACnB,yEAAyE,EACzE,EAAE,KAAK,EAAE,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CACjC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC7D,IAAI,CAAC,YAAY,GAAG,iBAAiB,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QAE5D,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;gBACvD,KAAK,EAAE,GAAY;aACpB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEO,aAAa;QACnB,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;QAC1D,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM;YAC9B,IAAI,EAAE,QAAQ,EAAE,IAAI;YACpB,SAAS,EAAE,QAAQ,EAAE,SAAS;YAC9B,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,YAAY,EAAE,QAAQ,CAAC,IAAI;YAC3B,yBAAyB,EAAE,IAAI,CAAC,IAAI,CAAC,yBAAyB;YAC9D,QAAQ,EAAE,OAAO,CAAC,WAAW,EAAE;YAC/B,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,kBAAkB;SACxC,CAAC;IACJ,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,WAAW;QACvB,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC,aAAa,CAAC;QAC5B,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;YACrC,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,OAAO,IAAI,CAAC,aAAa,CAAC;QAC5B,CAAC;QACD,IAAI,CAAC,SAAS,KAAK,IAAI,CAAC,mBAAmB,EAAE;aAC1C,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;YACjB,IAAI,CAAC,aAAa,GAAG,QAAQ,CAAC,KAAK,CAAC;YACpC,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;YAC5B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBAC3C,QAAQ,EAAE,QAAQ,CAAC,KAAK,CAAC,QAAQ;gBACjC,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,EAAE;gBACxB,GAAG,EAAE,QAAQ,CAAC,KAAK,CAAC,GAAG;gBACvB,OAAO,EAAE,QAAQ,CAAC,OAAO;aAC1B,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,KAAK,CAAC;QACxB,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE;YACpB,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC;YAC9B,MAAM,GAAG,CAAC;QACZ,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACZ,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC7B,CAAC,CAAC,CAAC;QACL,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC/B,IAAI,UAAqC,CAAC;QAC1C,IAAI,SAA4B,CAAC;QACjC,IAAI,CAAC;YACH,UAAU,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;gBACpD,YAAY,EAAE,IAAI,CAAC,YAAa;gBAChC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAC5B,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM;gBACxC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;aACzB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAY,CAAC;QAC3B,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC;QACzC,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,IAAI,CAAC,YAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;YACpE,IAAI,KAAK,IAAI,IAAI,CAAC,YAAa,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAClE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7C,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6CAA6C,EAAE;gBACnE,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,KAAK,CAAC,EAAE;aAChB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,UAAU,CAAC;QACpB,CAAC;QACD,MAAM,SAAS,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,KAAwB,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,KAAK,GAAG,GAAY,CAAC;QACvB,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAa,CAAC,YAAY,EAAE,CAAC;QAC1D,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,KAAK;YACnB,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,MAAM,EAAE,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC;YAClC,KAAK,EACH,CAAC,IAAI,CAAC,KAAK,IAAI,KAAK;gBAClB,CAAC,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE;gBACvD,CAAC,CAAC,SAAS;SAChB,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,GAAoB;QACjC,IACE,OAAO,GAAG,EAAE,QAAQ,KAAK,QAAQ;YACjC,OAAO,GAAG,CAAC,EAAE,KAAK,QAAQ;YAC1B,GAAG,CAAC,QAAQ,KAAK,EAAE;YACnB,GAAG,CAAC,EAAE,KAAK,EAAE,EACb,CAAC;YACD,MAAM,UAAU,CAAC,6CAA6C,CAAC,CAAC;QAClE,CAAC;QACD,IACE,GAAG,CAAC,aAAa,KAAK,SAAS;YAC/B,CAAE,eAAqC,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,EACnE,CAAC;YACD,MAAM,UAAU,CAAC,2BAA2B,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,YAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QAChE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,UAAU,CAAC,iBAAiB,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,YAAa,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1D,MAAM,UAAU,CACd,iBAAiB,KAAK,CAAC,QAAQ,0CAA0C,CAC1E,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;QAC1C,OAAO,IAAI,CAAC,UAAU,EAAE,CAAC;IAC3B,CAAC;IAEO,UAAU,CAAC,KAAiB,EAAE,aAAsB;QAC1D,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC;QAChC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC5B,IAAI,aAAa,EAAE,CAAC;YAClB,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC;QACrC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE;YACjD,aAAa,EAAE,IAAI,CAAC,aAAa;SAClC,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;YAC3C,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,KAAK,EAAE,KAAK,CAAC,EAAE;SAChB,CAAC,CAAC;QAEH,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;gBAChB,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CACpC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE;oBAC1D,KAAK,EAAE,GAAY;iBACpB,CAAC,CACH,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,KAAmB;QAC3C,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC;QACjC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QACD,IACE,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,KAAK,KAAK,CAAC,QAAQ;YAChD,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,KAAK,KAAK,CAAC,EAAE,EACpC,CAAC;YACD,MAAM,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;QACD,IACE,IAAI,CAAC,aAAa;YAClB,KAAK,CAAC,OAAO,CAAC,aAAa,KAAK,IAAI,CAAC,aAAa,EAClD,CAAC;YACD,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,aAAmC,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,QAAgB,EAChB,WAAoB;QAEpB,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvC,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAa,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAClE,MAAM,KAAK,GACT,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAC1D,SAAS,CAAC,CAAC,CAAC,CAAC;QACf,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvB,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,QAAgB;QACrC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACtD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,IAAI,CAAC,aAAa,EAAE,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC9C,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;YAC/B,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;YACvB,IAAI,CAAC,UAAU,GAAG,mCAAmC,QAAQ,GAAG,CAAC;QACnE,CAAC;IACH,CAAC;IAEO,WAAW,CAAC,cAAsB;QACxC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;QAChE,MAAM,IAAI,GAAG,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;QAClE,OAAO;YACL,GAAG,kBAAkB,CAAC;gBACpB,GAAG,IAAI;gBACP,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,WAAW,EAAE,MAAM,CAAC,SAAS,CAAC,IAAI;aACnC,CAAC;YACF,GAAG,mBAAmB,CAAC;gBACrB,GAAG,IAAI;gBACP,OAAO;gBACP,MAAM,EAAE,QAAQ;gBAChB,cAAc,EAAE,MAAM,CAAC,SAAS,CAAC,WAAW;aAC7C,CAAC;YACF,GAAG,uBAAuB,CAAC,EAAE,GAAG,IAAI,EAAE,KAAK,EAAE,CAAC;YAC9C,GAAG,cAAc,CAAC,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC;SAC7B,CAAC;IACnC,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,cAAsB;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACnD,IAAI,QAAQ,EAAE,CAAC;YACb,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACvB,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;gBACjC,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;YACjC,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QACvD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CACxD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,cAAc,CAAC,CAC5C,CAAC;YACF,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACpD,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,cAAsB;QAChD,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC;QACpC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QAEvC,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QACpD,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,MAAM,WAAW,GAAG,KAAK,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QACzD,MAAM,cAAc,GAClB,WAAW,IAAI,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC;YACvC,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,MAAM,CAAC,OAAO,CAAC;YAC9D,CAAC,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE;gBAChD,EAAE,EAAE,cAAc;aACnB,CAAC,CAAC;QAET,MAAM,MAAM,GAAG,IAAI,qBAAqB,CAAC;YACvC,GAAG,EAAE,MAAM,CAAC,OAAO;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI;YAClB,iBAAiB,EAAE,IAAI;YACvB,QAAQ,EAAE,IAAI;YACd,cAAc,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB;YAC9C,oBAAoB,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE;YAChD,YAAY,EAAE,iBAAiB,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC;SACtD,CAAC,CAAC;QACH,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;QAEtB,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,kBAAkB,CAAC;YAC3C,GAAG,EAAE,MAAM,CAAC,OAAO;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK;YACL,aAAa,EAAE,IAAI,CAAC,aAAmC;YACvD,OAAO,EAAE,SAAS;YAClB,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC;YAC7C,cAAc,EAAE,MAAM;YACtB,cAAc;YACd,eAAe,EAAE,eAAe,CAAC,QAAQ,CAAC;gBACxC,UAAU,EAAE,EAAE,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE;gBAChD,KAAK,EAAE;oBACL,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC;oBACpC,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,UAAU;iBACpC;gBACD,sBAAsB,EAAE,KAAK;gBAC7B,eAAe,EAAE,KAAK;gBACtB,YAAY,EAAE,KAAK;gBACnB,oBAAoB,EAAE,IAAI,CAAC,aAAmC;aAC/D,CAAC;SACH,CAAC,CAAC;QAEH,IACE,OAAO,CAAC,KAAK,EAAE,QAAQ,KAAK,KAAK,CAAC,QAAQ;YAC1C,OAAO,CAAC,KAAK,EAAE,EAAE,KAAK,KAAK,CAAC,EAAE,EAC9B,CAAC;YACD,MAAM,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QAED,MAAM,KAAK,GAAiB,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACrD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QACzC,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,YAAY,CAAC,cAAsB,EAAE,KAAmB;QAC9D,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;QACD,KAAK,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAChC,IAAI,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,KAAK,KAAK,EAAE,CAAC;gBAC9D,OAAO;YACT,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;YACrC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;QAC1B,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,GAAG,IAAI,CAAC,CAAC;QAC5D,KAAK,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAEO,WAAW,CACjB,KAAsB,EACtB,KAAiB;QAEjB,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC;YACrC,KAAK,CAAC,IAAI,CACR,KAAK,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,QAAQ,KAAK,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAC7E,CAAC;YACF,IACE,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC7B,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC;gBACjC,IAAI,CAAC,IAAI,IAAI,aAAa,EAC1B,CAAC;gBACD,IAAI,CAAC;oBACH,MAAM,CAAC,IAAI,CAAC;wBACV,IAAI,EAAE,OAAO;wBACb,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;wBACnD,QAAQ,EAAE,IAAI,CAAC,QAAQ;qBACxB,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE;wBAC1D,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,KAAK,EAAE,GAAY;qBACpB,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,GACR,KAAK,CAAC,MAAM,KAAK,CAAC;YAChB,CAAC,CAAC,KAAK,CAAC,IAAI;YACZ,CAAC,CAAC,GAAG,KAAK,CAAC,IAAI,gDAAgD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QAC7F,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,GAAG,CACP,KAAsB,EACtB,GAAsB;QAEtB,MAAM,cAAc,GAAG,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC;QAC3C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QACpD,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;YACf,OAAO;gBACL,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,EAAE,OAAO,EAAE,0BAA0B,EAAE;aAC/C,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,aAAa,CAC9B,GAAG,CAAC,OAAO,EACX,EAAE,WAAW,EAAE,IAAI,GAAG,CAAC,uBAAuB,CAAC,EAAE,EACjD,GAAG,CAAC,WAAW,CAChB,CAAC;QACF,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;YACpD,IAAI,CAAC;gBACH,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;oBACvD,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,GAAY;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,KAAK,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC7B,CAAC,CAAC;QACF,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9D,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;QAElB,IAAI,CAAC;YACH,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACvB,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;YACtD,CAAC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE;gBAC/B,MAAM,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;gBAC9C,qBAAqB,EAAE,KAAK;aAC7B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACvB,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;YACtD,CAAC;YACD,OAAO;gBACL,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,EAAE,OAAO,EAAG,GAAa,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE;gBACzD,KAAK,EAAE,MAAM,CAAC,KAAK;aACpB,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,WAAW,EAAE,CAAC;YACd,GAAG,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACjD,KAAK,CAAC,IAAI,GAAG,KAAK,CAAC;YACnB,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;gBAC9B,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAC5B,cAAc,EACd,KAAK,CAAC,OAAO,CAAC,WAAW,CAC1B,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;QAC3C,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,aAAa,CAAC;QAClC,IAAI,IAAI,EAAE,UAAU,KAAK,OAAO,EAAE,CAAC;YACjC,OAAO;gBACL,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE;oBACL,IAAI,EAAE,aAAa;oBACnB,OAAO,EAAE,IAAI,CAAC,YAAY,IAAI,0BAA0B;iBACzD;gBACD,KAAK,EAAE,MAAM,CAAC,KAAK;aACpB,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,aAAa,CACjB,IAAY,EACZ,MAAoB;QAEpB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAa,CAAC,cAAc,CACjD,KAAK,EACL;YACE,YAAY,EACV,gMAAgM;YAClM,QAAQ,EAAE;gBACR,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;aACtE;SACF,EACD;YACE,SAAS,EAAE,GAAG;YACd,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO;SAC9D,CACF,CAAC;QACF,IAAI,GAAG,CAAC,UAAU,KAAK,OAAO,IAAI,GAAG,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;YAC/D,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO;aACtB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAsB,CAAC,IAAI,CAAC;aACxC,IAAI,CAAC,GAAG,CAAC;aACT,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;aAC1B,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC;YAC5B,EAAE,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC;aACxC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QAChB,OAAO,KAAK,IAAI,SAAS,CAAC;IAC5B,CAAC;IAED,kBAAkB,CAAC,cAAsB;QACvC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAChD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO;QACT,CAAC;QACD,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACpB,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACrC,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,CAAC;QACrB,KAAK,MAAM,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACjC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
import type { AssistantMessage } from "@earendil-works/pi-ai";
|
|
2
|
+
import type { AgentSessionEvent } from "@earendil-works/pi-coding-agent";
|
|
3
|
+
import type { RunActivity, ToolDetails, Usage } from "../../protocol/index.ts";
|
|
4
|
+
import type { MessageBuilder } from "../../runs/builder.ts";
|
|
5
|
+
export interface MapperOptions {
|
|
6
|
+
hiddenTools: Set<string>;
|
|
7
|
+
maxOutputChars?: number;
|
|
8
|
+
updateIntervalMs?: number;
|
|
9
|
+
}
|
|
10
|
+
export declare const toolTitle: (name: string, args: unknown) => string;
|
|
11
|
+
export declare const resultText: (result: unknown) => string;
|
|
12
|
+
export declare const toolDetails: (name: string, args: unknown, result: unknown) => ToolDetails | undefined;
|
|
13
|
+
export declare const emptyUsage: () => Usage;
|
|
14
|
+
export declare class PiEventMapper {
|
|
15
|
+
readonly usage: Usage;
|
|
16
|
+
lastAssistant?: AssistantMessage;
|
|
17
|
+
private builder;
|
|
18
|
+
private opts;
|
|
19
|
+
private setActivity;
|
|
20
|
+
private turn;
|
|
21
|
+
private contentBlocks;
|
|
22
|
+
private lastUpdates;
|
|
23
|
+
constructor(builder: MessageBuilder, opts: MapperOptions, setActivity: (activity: RunActivity | undefined) => void);
|
|
24
|
+
private truncate;
|
|
25
|
+
handle(event: AgentSessionEvent): void;
|
|
26
|
+
private key;
|
|
27
|
+
private handleAssistantEvent;
|
|
28
|
+
private handleAssistantEnd;
|
|
29
|
+
private ensureToolBlock;
|
|
30
|
+
private handleToolStart;
|
|
31
|
+
private handleToolUpdate;
|
|
32
|
+
private handleToolEnd;
|
|
33
|
+
}
|