@bf-meishijian/aip 0.0.22 → 0.0.23

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.
@@ -0,0 +1,356 @@
1
+ import { createHash, randomUUID } from "node:crypto";
2
+ import { mkdir } from "node:fs/promises";
3
+ import { join } from "node:path";
4
+ import { openPiSession } from "./coding-run-manager.js";
5
+ import { generalRunTools } from "./general-run-tools.js";
6
+ import { logHostEvent, redactHostText } from "./host-log.js";
7
+ import { paths } from "./paths.js";
8
+ const LEASE_SECONDS = 30;
9
+ /** 普通会话 Run 的 Pi 执行入口;不创建工单、不续接未知结果、不重新执行已领取任务。 */
10
+ export class GeneralRunManager {
11
+ pi;
12
+ active = new Map();
13
+ workerProcessId = `general-host:${randomUUID()}`;
14
+ constructor(pi) {
15
+ this.pi = pi;
16
+ }
17
+ isBusy() {
18
+ return this.active.size > 0;
19
+ }
20
+ async poll(client, host, once = false) {
21
+ if (this.active.size >= 4)
22
+ return;
23
+ const spec = await client.claim({
24
+ hostId: host.hostId,
25
+ generation: host.generation,
26
+ workerProcessId: this.workerProcessId,
27
+ leaseSeconds: LEASE_SECONDS,
28
+ });
29
+ if (!spec || this.active.has(spec.runId))
30
+ return;
31
+ const active = { stopping: false, done: Promise.resolve() };
32
+ this.active.set(spec.runId, active);
33
+ active.done = this.execute(client, host, spec, active).finally(() => this.active.delete(spec.runId));
34
+ if (once)
35
+ await active.done;
36
+ else
37
+ void active.done.catch((error) => {
38
+ logHostEvent("general_run_result_unknown", {
39
+ runId: spec.runId,
40
+ hostId: host.hostId,
41
+ error: redactHostText(error instanceof Error ? error.message : String(error)),
42
+ });
43
+ });
44
+ }
45
+ async stop() {
46
+ const runs = [...this.active.values()];
47
+ for (const active of runs)
48
+ active.stopping = true;
49
+ await Promise.allSettled(runs.map((active) => active.session?.abort()));
50
+ await Promise.allSettled(runs.map((active) => active.done));
51
+ }
52
+ async execute(client, host, spec, active) {
53
+ const binding = {
54
+ hostId: host.hostId,
55
+ generation: host.generation,
56
+ runAttemptId: spec.attempt.attemptId,
57
+ attemptNo: spec.attempt.attemptNo,
58
+ executionToken: spec.attempt.executionToken,
59
+ taskToken: spec.taskToken,
60
+ };
61
+ let protocolLost = false;
62
+ let terminalStarted = false;
63
+ let cancelled = false;
64
+ let modelError;
65
+ let failureCode = "GENERAL_RUN_FAILED";
66
+ let heartbeatTimer;
67
+ let deadline;
68
+ let unsubscribe;
69
+ let textTimer;
70
+ let pending = Promise.resolve();
71
+ let sequence = spec.attempt.lastEventSequence;
72
+ let answer = "";
73
+ let text = "";
74
+ let secrets = [];
75
+ let redact = (value) => value;
76
+ const loseProtocol = () => {
77
+ protocolLost = true;
78
+ void active.session?.abort();
79
+ };
80
+ const emit = (kind, payload) => {
81
+ pending = pending.then(async () => {
82
+ await client.event(spec.runId, {
83
+ ...binding,
84
+ eventId: randomUUID(),
85
+ sequence: ++sequence,
86
+ kind,
87
+ occurredAt: new Date().toISOString(),
88
+ payload,
89
+ });
90
+ });
91
+ void pending.catch(loseProtocol);
92
+ return pending;
93
+ };
94
+ const flushText = (final = false) => {
95
+ if (textTimer)
96
+ clearTimeout(textTimer);
97
+ textTimer = undefined;
98
+ if (!text)
99
+ return;
100
+ text = redact(text);
101
+ let held = 0;
102
+ if (!final)
103
+ for (const secret of secrets) {
104
+ for (let length = Math.min(secret.length - 1, text.length); length > held; length--) {
105
+ if (text.endsWith(secret.slice(0, length))) {
106
+ held = length;
107
+ break;
108
+ }
109
+ }
110
+ }
111
+ const safe = text.slice(0, text.length - held);
112
+ text = held ? text.slice(-held) : "";
113
+ // 服务端正文事件有界;按字符分段,避免拆开 UTF-16 代理对。
114
+ let chunk = "";
115
+ for (const character of safe) {
116
+ if (Buffer.byteLength(chunk + character, "utf8") > 1024) {
117
+ void emit("TEXT_DELTA", { text: chunk });
118
+ chunk = "";
119
+ }
120
+ chunk += character;
121
+ }
122
+ if (chunk)
123
+ void emit("TEXT_DELTA", { text: chunk });
124
+ };
125
+ try {
126
+ if (spec.runKind !== "GENERAL_CHAT" || spec.input.kind !== "GENERAL_CHAT")
127
+ throw new Error("GENERAL_RUN_KIND_INVALID");
128
+ const input = spec.input;
129
+ if (spec.attempt.resume || spec.attempt.attemptNo !== 1) {
130
+ failureCode = "GENERAL_EXECUTION_INTERRUPTED";
131
+ throw new Error("本地运行结果未知,不自动重放已领取的任务");
132
+ }
133
+ let heartbeatPending = false;
134
+ const heartbeat = async () => {
135
+ if (heartbeatPending || terminalStarted)
136
+ return;
137
+ heartbeatPending = true;
138
+ try {
139
+ const status = await client.heartbeat(spec.runId, {
140
+ ...binding,
141
+ leaseSeconds: LEASE_SECONDS,
142
+ });
143
+ if (status.cancelRequested) {
144
+ cancelled = true;
145
+ await active.session?.abort();
146
+ }
147
+ }
148
+ catch {
149
+ loseProtocol();
150
+ }
151
+ finally {
152
+ heartbeatPending = false;
153
+ }
154
+ };
155
+ await heartbeat();
156
+ if (cancelled || protocolLost)
157
+ throw new Error("GENERAL_RUN_STOPPED");
158
+ heartbeatTimer = setInterval(() => void heartbeat(), 10_000);
159
+ deadline = setTimeout(() => {
160
+ failureCode = "RUN_TIMEOUT";
161
+ active.stopping = true;
162
+ void active.session?.abort();
163
+ }, spec.resolvedAgentConfig.runTimeoutSeconds * 1000);
164
+ const context = await client.context(spec.runId, binding);
165
+ if (context.executionIdentity.hostId !== host.hostId ||
166
+ context.executionIdentity.conversationId !== input.conversationId)
167
+ throw new Error("GENERAL_RUN_BINDING_INVALID");
168
+ const variables = context.environmentVariables;
169
+ secrets = Object.entries(variables)
170
+ .filter(([key, value]) => /TOKEN|PASSWORD|SECRET|KEY|AUTH/iu.test(key) && value.length >= 4)
171
+ .map(([, value]) => value);
172
+ redact = (value) => {
173
+ let result = value;
174
+ for (const secret of secrets)
175
+ result = result.split(secret).join("[凭证已脱敏]");
176
+ return result;
177
+ };
178
+ const conversationKey = createHash("sha256")
179
+ .update(input.conversationId)
180
+ .digest("hex")
181
+ .slice(0, 24);
182
+ const cwd = join(paths().workspaceRoot, "general", conversationKey);
183
+ await mkdir(cwd, { recursive: true, mode: 0o700 });
184
+ if (!this.pi.resolveRuntimeModel)
185
+ throw new Error("MODEL_RUNTIME_UNAVAILABLE");
186
+ const resolved = await this.pi.resolveRuntimeModel(context.execution.providerId, context.execution.modelId, {
187
+ ...binding,
188
+ runId: spec.runId,
189
+ kind: "GENERAL",
190
+ });
191
+ let actions = 0;
192
+ const tools = generalRunTools({
193
+ spec,
194
+ binding,
195
+ client,
196
+ cwd,
197
+ environmentVariables: variables,
198
+ redact,
199
+ protocolLost: loseProtocol,
200
+ beforeTool: async () => {
201
+ flushText();
202
+ await pending;
203
+ if (protocolLost || active.stopping || cancelled)
204
+ throw new Error("GENERAL_RUN_STOPPED");
205
+ if (++actions > spec.resolvedAgentConfig.actionBudget) {
206
+ failureCode = "ACTION_BUDGET_EXCEEDED";
207
+ active.stopping = true;
208
+ void active.session?.abort();
209
+ throw new Error("本轮工具调用次数已达到配置上限");
210
+ }
211
+ },
212
+ });
213
+ const opened = await openPiSession({
214
+ cwd,
215
+ sessionDir: join(paths().piSessions, "general", conversationKey),
216
+ agentDir: paths().piAgent,
217
+ modelRuntime: resolved.runtime,
218
+ model: {
219
+ ...resolved.model,
220
+ maxTokens: Math.min(resolved.model.maxTokens, spec.resolvedAgentConfig.maxOutputTokens),
221
+ },
222
+ thinkingLevel: context.execution.thinkingLevel,
223
+ extensions: [tools.extension],
224
+ systemPrompt: `${spec.resolvedAgentConfig.systemPrompt}\n\n当前为通用员工会话,工作目录:${cwd}。本地命令工具使用 ${tools.shell.name},按实际工具反馈操作;连接中断或结果未知不表示执行成功。`,
225
+ runId: spec.runId,
226
+ });
227
+ active.session = opened.session;
228
+ opened.session.setAutoRetryEnabled(false);
229
+ opened.session.setActiveToolsByName(tools.names);
230
+ if (active.stopping || cancelled || protocolLost)
231
+ throw new Error("GENERAL_RUN_STOPPED");
232
+ const currentMessage = context.messages.find((message) => message.id === input.userMessageId);
233
+ if (currentMessage?.role !== "USER")
234
+ throw new Error("GENERAL_RUN_INPUT_MISSING");
235
+ const history = context.messages.filter((message) => message.messageNo < currentMessage.messageNo);
236
+ const readImages = async (message) => {
237
+ if (!message.attachments?.length)
238
+ return [];
239
+ if (!resolved.model.input.includes("image"))
240
+ throw new Error("MODEL_IMAGES_UNSUPPORTED");
241
+ return Promise.all(message.attachments.map((attachment) => client.attachment(spec.runId, attachment.id, binding)));
242
+ };
243
+ const historyContent = [];
244
+ for (const message of history) {
245
+ historyContent.push({
246
+ type: "text",
247
+ text: `历史消息(${message.role}):\n${message.contentMarkdown}`,
248
+ });
249
+ historyContent.push(...(await readImages(message)));
250
+ }
251
+ if (history.length)
252
+ await opened.session.sendCustomMessage({
253
+ customType: "platform-conversation-history",
254
+ content: [
255
+ { type: "text", text: "以下是当前私人会话已经发生的消息,作为历史上下文使用:" },
256
+ ...historyContent,
257
+ ],
258
+ display: false,
259
+ details: { conversationId: input.conversationId },
260
+ });
261
+ unsubscribe = opened.session.subscribe((event) => {
262
+ if (event.type === "message_update" && event.assistantMessageEvent.type === "text_delta") {
263
+ text += event.assistantMessageEvent.delta;
264
+ if (Buffer.byteLength(text, "utf8") >= 1024)
265
+ flushText();
266
+ else
267
+ textTimer ??= setTimeout(() => flushText(), 50);
268
+ }
269
+ else if (event.type === "message_end" && event.message.role === "assistant") {
270
+ flushText(true);
271
+ answer = event.message.content
272
+ .filter((part) => part.type === "text")
273
+ .map((part) => part.text)
274
+ .join("\n");
275
+ if (event.message.stopReason === "error")
276
+ modelError = event.message.errorMessage ?? "模型调用失败";
277
+ if (event.message.stopReason === "aborted")
278
+ modelError = "模型调用已中断";
279
+ }
280
+ else if (event.type === "tool_execution_start" || event.type === "tool_execution_end") {
281
+ flushText();
282
+ void emit(event.type === "tool_execution_start" ? "TOOL_STARTED" : "TOOL_FINISHED", {
283
+ toolCallId: tools.toolCallId(event.toolCallId),
284
+ toolName: tools.platformNames.get(event.toolName) ?? event.toolName,
285
+ });
286
+ }
287
+ });
288
+ await emit("PHASE_STARTED", { phase: "GENERAL_CHAT", message: "本地员工开始处理任务" });
289
+ const images = await readImages(currentMessage);
290
+ if (!cancelled && !active.stopping && !protocolLost)
291
+ await opened.session.prompt(currentMessage.contentMarkdown, { images });
292
+ flushText(true);
293
+ await pending;
294
+ if (protocolLost)
295
+ throw new Error("GENERAL_EXECUTION_INTERRUPTED");
296
+ if (cancelled || active.stopping) {
297
+ failureCode = cancelled
298
+ ? "RUN_CANCELLED"
299
+ : failureCode === "GENERAL_RUN_FAILED"
300
+ ? "HOST_STOPPED"
301
+ : failureCode;
302
+ throw new Error(cancelled ? "运行已停止" : "本地执行已结束");
303
+ }
304
+ if (modelError)
305
+ throw new Error(modelError);
306
+ if (!answer.trim())
307
+ throw new Error("GENERAL_RUN_EMPTY_RESPONSE");
308
+ terminalStarted = true;
309
+ const completed = await client.complete(spec.runId, {
310
+ ...binding,
311
+ idempotencyKey: `${spec.runId}:complete`,
312
+ output: {
313
+ kind: "GENERAL_CHAT",
314
+ answerMarkdown: redact(answer),
315
+ citations: [],
316
+ ...(input.employee ? { employee: input.employee } : {}),
317
+ },
318
+ usage: {},
319
+ contextManifest: {
320
+ sessionId: opened.sessionId,
321
+ conversationId: input.conversationId,
322
+ includedMessageIds: context.messages.map((message) => message.id),
323
+ },
324
+ });
325
+ if (completed.status === "VALIDATION_FAILED") {
326
+ terminalStarted = false;
327
+ failureCode = "GENERAL_RUN_OUTPUT_INVALID";
328
+ throw new Error(completed.violations.join(";"));
329
+ }
330
+ }
331
+ catch (error) {
332
+ // 协议/终态回执丢失时由服务端租约记录未知结果,不能重试执行或宣称失败已回读。
333
+ if (protocolLost || terminalStarted)
334
+ throw error;
335
+ terminalStarted = true;
336
+ await client.fail(spec.runId, {
337
+ ...binding,
338
+ idempotencyKey: `${spec.runId}:fail`,
339
+ code: cancelled ? "RUN_CANCELLED" : failureCode,
340
+ message: redactHostText(redact(error instanceof Error ? error.message : String(error))),
341
+ retryable: false,
342
+ });
343
+ }
344
+ finally {
345
+ if (heartbeatTimer)
346
+ clearInterval(heartbeatTimer);
347
+ if (deadline)
348
+ clearTimeout(deadline);
349
+ if (textTimer)
350
+ clearTimeout(textTimer);
351
+ unsubscribe?.();
352
+ active.session?.dispose();
353
+ }
354
+ }
355
+ }
356
+ //# sourceMappingURL=general-run-manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"general-run-manager.js","sourceRoot":"","sources":["../src/general-run-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACrD,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAIjC,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAA0B,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACjF,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAkBnC,MAAM,aAAa,GAAG,EAAE,CAAC;AAEzB,mDAAmD;AACnD,MAAM,OAAO,iBAAiB;IAIC;IAHZ,MAAM,GAAG,IAAI,GAAG,EAAqB,CAAC;IACtC,eAAe,GAAG,gBAAgB,UAAU,EAAE,EAAE,CAAC;IAElE,YAA6B,EAAkB;QAAlB,OAAE,GAAF,EAAE,CAAgB;IAAG,CAAC;IAEnD,MAAM;QACJ,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,IAAU,EAAE,IAAI,GAAG,KAAK;QACjD,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC;YAAE,OAAO;QAClC,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,YAAY,EAAE,aAAa;SAC5B,CAAC,CAAC;QACH,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC;YAAE,OAAO;QACjD,MAAM,MAAM,GAAc,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;QACvE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAClE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAC/B,CAAC;QACF,IAAI,IAAI;YAAE,MAAM,MAAM,CAAC,IAAI,CAAC;;YAE1B,KAAK,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;gBACxC,YAAY,CAAC,4BAA4B,EAAE;oBACzC,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,KAAK,EAAE,cAAc,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;iBAC9E,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;IACP,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,KAAK,MAAM,MAAM,IAAI,IAAI;YAAE,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;QAClD,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QACxE,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;IAC9D,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,MAAc,EAAE,IAAU,EAAE,IAAa,EAAE,MAAiB;QAChF,MAAM,OAAO,GAAsB;YACjC,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YACpC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS;YACjC,cAAc,EAAE,IAAI,CAAC,OAAO,CAAC,cAAc;YAC3C,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC;QACF,IAAI,YAAY,GAAG,KAAK,CAAC;QACzB,IAAI,eAAe,GAAG,KAAK,CAAC;QAC5B,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,UAA8B,CAAC;QACnC,IAAI,WAAW,GAAG,oBAAoB,CAAC;QACvC,IAAI,cAA0D,CAAC;QAC/D,IAAI,QAAmD,CAAC;QACxD,IAAI,WAAqC,CAAC;QAC1C,IAAI,SAAoD,CAAC;QACzD,IAAI,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;QAChC,IAAI,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC;QAC9C,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,OAAO,GAAa,EAAE,CAAC;QAC3B,IAAI,MAAM,GAAG,CAAC,KAAa,EAAE,EAAE,CAAC,KAAK,CAAC;QACtC,MAAM,YAAY,GAAG,GAAG,EAAE;YACxB,YAAY,GAAG,IAAI,CAAC;YACpB,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;QAC/B,CAAC,CAAC;QACF,MAAM,IAAI,GAAG,CACX,IAA2D,EAC3D,OAAgC,EAChC,EAAE;YACF,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE;gBAChC,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE;oBAC7B,GAAG,OAAO;oBACV,OAAO,EAAE,UAAU,EAAE;oBACrB,QAAQ,EAAE,EAAE,QAAQ;oBACpB,IAAI;oBACJ,UAAU,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACpC,OAAO;iBACR,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YACH,KAAK,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YACjC,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,CAAC,KAAK,GAAG,KAAK,EAAE,EAAE;YAClC,IAAI,SAAS;gBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;YACvC,SAAS,GAAG,SAAS,CAAC;YACtB,IAAI,CAAC,IAAI;gBAAE,OAAO;YAClB,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YACpB,IAAI,IAAI,GAAG,CAAC,CAAC;YACb,IAAI,CAAC,KAAK;gBACR,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;oBAC7B,KAAK,IAAI,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC;wBACpF,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,EAAE,CAAC;4BAC3C,IAAI,GAAG,MAAM,CAAC;4BACd,MAAM;wBACR,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;YAC/C,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACrC,mCAAmC;YACnC,IAAI,KAAK,GAAG,EAAE,CAAC;YACf,KAAK,MAAM,SAAS,IAAI,IAAI,EAAE,CAAC;gBAC7B,IAAI,MAAM,CAAC,UAAU,CAAC,KAAK,GAAG,SAAS,EAAE,MAAM,CAAC,GAAG,IAAI,EAAE,CAAC;oBACxD,KAAK,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;oBACzC,KAAK,GAAG,EAAE,CAAC;gBACb,CAAC;gBACD,KAAK,IAAI,SAAS,CAAC;YACrB,CAAC;YACD,IAAI,KAAK;gBAAE,KAAK,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC;QACF,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,OAAO,KAAK,cAAc,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,cAAc;gBACvE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;YACzB,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;gBACxD,WAAW,GAAG,+BAA+B,CAAC;gBAC9C,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAC1C,CAAC;YACD,IAAI,gBAAgB,GAAG,KAAK,CAAC;YAC7B,MAAM,SAAS,GAAG,KAAK,IAAI,EAAE;gBAC3B,IAAI,gBAAgB,IAAI,eAAe;oBAAE,OAAO;gBAChD,gBAAgB,GAAG,IAAI,CAAC;gBACxB,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE;wBAChD,GAAG,OAAO;wBACV,YAAY,EAAE,aAAa;qBAC5B,CAAC,CAAC;oBACH,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;wBAC3B,SAAS,GAAG,IAAI,CAAC;wBACjB,MAAM,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;oBAChC,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,YAAY,EAAE,CAAC;gBACjB,CAAC;wBAAS,CAAC;oBACT,gBAAgB,GAAG,KAAK,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC;YACF,MAAM,SAAS,EAAE,CAAC;YAClB,IAAI,SAAS,IAAI,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACtE,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,KAAK,SAAS,EAAE,EAAE,MAAM,CAAC,CAAC;YAC7D,QAAQ,GAAG,UAAU,CAAC,GAAG,EAAE;gBACzB,WAAW,GAAG,aAAa,CAAC;gBAC5B,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;gBACvB,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;YAC/B,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,iBAAiB,GAAG,IAAI,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC1D,IACE,OAAO,CAAC,iBAAiB,CAAC,MAAM,KAAK,IAAI,CAAC,MAAM;gBAChD,OAAO,CAAC,iBAAiB,CAAC,cAAc,KAAK,KAAK,CAAC,cAAc;gBAEjE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;YACjD,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,CAAC;YAC/C,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC;iBAChC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,kCAAkC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;iBAC3F,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;YAC7B,MAAM,GAAG,CAAC,KAAK,EAAE,EAAE;gBACjB,IAAI,MAAM,GAAG,KAAK,CAAC;gBACnB,KAAK,MAAM,MAAM,IAAI,OAAO;oBAAE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC5E,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YACF,MAAM,eAAe,GAAG,UAAU,CAAC,QAAQ,CAAC;iBACzC,MAAM,CAAC,KAAK,CAAC,cAAc,CAAC;iBAC5B,MAAM,CAAC,KAAK,CAAC;iBACb,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAChB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC,aAAa,EAAE,SAAS,EAAE,eAAe,CAAC,CAAC;YACpE,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACnD,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,mBAAmB;gBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAC/E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,mBAAmB,CAChD,OAAO,CAAC,SAAS,CAAC,UAAU,EAC5B,OAAO,CAAC,SAAS,CAAC,OAAO,EACzB;gBACE,GAAG,OAAO;gBACV,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,IAAI,EAAE,SAAS;aAChB,CACF,CAAC;YACF,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,MAAM,KAAK,GAAG,eAAe,CAAC;gBAC5B,IAAI;gBACJ,OAAO;gBACP,MAAM;gBACN,GAAG;gBACH,oBAAoB,EAAE,SAAS;gBAC/B,MAAM;gBACN,YAAY,EAAE,YAAY;gBAC1B,UAAU,EAAE,KAAK,IAAI,EAAE;oBACrB,SAAS,EAAE,CAAC;oBACZ,MAAM,OAAO,CAAC;oBACd,IAAI,YAAY,IAAI,MAAM,CAAC,QAAQ,IAAI,SAAS;wBAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;oBACzF,IAAI,EAAE,OAAO,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,CAAC;wBACtD,WAAW,GAAG,wBAAwB,CAAC;wBACvC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;wBACvB,KAAK,MAAM,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;wBAC7B,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;oBACrC,CAAC;gBACH,CAAC;aACF,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;gBACjC,GAAG;gBACH,UAAU,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,eAAe,CAAC;gBAChE,QAAQ,EAAE,KAAK,EAAE,CAAC,OAAO;gBACzB,YAAY,EAAE,QAAQ,CAAC,OAAO;gBAC9B,KAAK,EAAE;oBACL,GAAG,QAAQ,CAAC,KAAK;oBACjB,SAAS,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,mBAAmB,CAAC,eAAe,CAAC;iBACxF;gBACD,aAAa,EAAE,OAAO,CAAC,SAAS,CAAC,aAAa;gBAC9C,UAAU,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC;gBAC7B,YAAY,EAAE,GAAG,IAAI,CAAC,mBAAmB,CAAC,YAAY,sBAAsB,GAAG,aAAa,KAAK,CAAC,KAAK,CAAC,IAAI,8BAA8B;gBAC1I,KAAK,EAAE,IAAI,CAAC,KAAK;aAClB,CAAC,CAAC;YACH,MAAM,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACjD,IAAI,MAAM,CAAC,QAAQ,IAAI,SAAS,IAAI,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzF,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,KAAK,CAAC,aAAa,CAAC,CAAC;YAC9F,IAAI,cAAc,EAAE,IAAI,KAAK,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAClF,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,MAAM,CACrC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS,GAAG,cAAc,CAAC,SAAS,CAC1D,CAAC;YACF,MAAM,UAAU,GAAG,KAAK,EAAE,OAA0C,EAAE,EAAE;gBACtE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM;oBAAE,OAAO,EAAE,CAAC;gBAC5C,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBACzF,OAAO,OAAO,CAAC,GAAG,CAChB,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACrC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,UAAU,CAAC,EAAE,EAAE,OAAO,CAAC,CACtD,CACF,CAAC;YACJ,CAAC,CAAC;YACF,MAAM,cAAc,GAAsC,EAAE,CAAC;YAC7D,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE,CAAC;gBAC9B,cAAc,CAAC,IAAI,CAAC;oBAClB,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,QAAQ,OAAO,CAAC,IAAI,OAAO,OAAO,CAAC,eAAe,EAAE;iBAC3D,CAAC,CAAC;gBACH,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;YACtD,CAAC;YACD,IAAI,OAAO,CAAC,MAAM;gBAChB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC;oBACrC,UAAU,EAAE,+BAA+B;oBAC3C,OAAO,EAAE;wBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,6BAA6B,EAAE;wBACrD,GAAG,cAAc;qBAClB;oBACD,OAAO,EAAE,KAAK;oBACd,OAAO,EAAE,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE;iBAClD,CAAC,CAAC;YACL,WAAW,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC/C,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,qBAAqB,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBACzF,IAAI,IAAI,KAAK,CAAC,qBAAqB,CAAC,KAAK,CAAC;oBAC1C,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI;wBAAE,SAAS,EAAE,CAAC;;wBACpD,SAAS,KAAK,UAAU,CAAC,GAAG,EAAE,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;gBACvD,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC9E,SAAS,CAAC,IAAI,CAAC,CAAC;oBAChB,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO;yBAC3B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;yBACtC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;yBACxB,IAAI,CAAC,IAAI,CAAC,CAAC;oBACd,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,KAAK,OAAO;wBACtC,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,QAAQ,CAAC;oBACtD,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,KAAK,SAAS;wBAAE,UAAU,GAAG,SAAS,CAAC;gBACrE,CAAC;qBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;oBACxF,SAAS,EAAE,CAAC;oBACZ,KAAK,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,sBAAsB,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,eAAe,EAAE;wBAClF,UAAU,EAAE,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,CAAC;wBAC9C,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,CAAC,QAAQ;qBACpE,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;YACH,MAAM,IAAI,CAAC,eAAe,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;YAC9E,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,cAAc,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,CAAC,YAAY;gBACjD,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,eAAe,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC1E,SAAS,CAAC,IAAI,CAAC,CAAC;YAChB,MAAM,OAAO,CAAC;YACd,IAAI,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;YACnE,IAAI,SAAS,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACjC,WAAW,GAAG,SAAS;oBACrB,CAAC,CAAC,eAAe;oBACjB,CAAC,CAAC,WAAW,KAAK,oBAAoB;wBACpC,CAAC,CAAC,cAAc;wBAChB,CAAC,CAAC,WAAW,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;YACnD,CAAC;YACD,IAAI,UAAU;gBAAE,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;YAClE,eAAe,GAAG,IAAI,CAAC;YACvB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE;gBAClD,GAAG,OAAO;gBACV,cAAc,EAAE,GAAG,IAAI,CAAC,KAAK,WAAW;gBACxC,MAAM,EAAE;oBACN,IAAI,EAAE,cAAc;oBACpB,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC;oBAC9B,SAAS,EAAE,EAAE;oBACb,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACxD;gBACD,KAAK,EAAE,EAAE;gBACT,eAAe,EAAE;oBACf,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,cAAc,EAAE,KAAK,CAAC,cAAc;oBACpC,kBAAkB,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;iBAClE;aACF,CAAC,CAAC;YACH,IAAI,SAAS,CAAC,MAAM,KAAK,mBAAmB,EAAE,CAAC;gBAC7C,eAAe,GAAG,KAAK,CAAC;gBACxB,WAAW,GAAG,4BAA4B,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yCAAyC;YACzC,IAAI,YAAY,IAAI,eAAe;gBAAE,MAAM,KAAK,CAAC;YACjD,eAAe,GAAG,IAAI,CAAC;YACvB,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE;gBAC5B,GAAG,OAAO;gBACV,cAAc,EAAE,GAAG,IAAI,CAAC,KAAK,OAAO;gBACpC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,WAAW;gBAC/C,OAAO,EAAE,cAAc,CAAC,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBACvF,SAAS,EAAE,KAAK;aACjB,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,cAAc;gBAAE,aAAa,CAAC,cAAc,CAAC,CAAC;YAClD,IAAI,QAAQ;gBAAE,YAAY,CAAC,QAAQ,CAAC,CAAC;YACrC,IAAI,SAAS;gBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;YACvC,WAAW,EAAE,EAAE,CAAC;YAChB,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;CACF"}
@@ -0,0 +1,24 @@
1
+ import { type components } from "@ai-platform/api-contracts";
2
+ import { type ExtensionFactory } from "@earendil-works/pi-coding-agent";
3
+ import type { GeneralRunClient } from "./general-run-client.js";
4
+ export type GeneralRunBinding = components["schemas"]["RunAttemptCredentials"] & {
5
+ hostId: string;
6
+ generation: number;
7
+ };
8
+ export declare function generalRunTools(options: {
9
+ spec: components["schemas"]["AgentRunSpec"];
10
+ binding: GeneralRunBinding;
11
+ client: Pick<GeneralRunClient, "tool" | "recordToolFacts">;
12
+ cwd: string;
13
+ environmentVariables: Record<string, string>;
14
+ beforeTool: () => Promise<void>;
15
+ redact: (text: string) => string;
16
+ protocolLost: () => void;
17
+ }): {
18
+ extension: ExtensionFactory;
19
+ names: string[];
20
+ platformNames: Map<string, string>;
21
+ toolCallId: (sourceId: string) => string;
22
+ shell: import("@ai-platform/api-contracts").HostShell;
23
+ };
24
+ //# sourceMappingURL=general-run-tools.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"general-run-tools.d.ts","sourceRoot":"","sources":["../src/general-run-tools.ts"],"names":[],"mappings":"AACA,OAAO,EACL,KAAK,UAAU,EAKhB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAOL,KAAK,gBAAgB,EAEtB,MAAM,iCAAiC,CAAC;AAEzC,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAKhE,MAAM,MAAM,iBAAiB,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC,uBAAuB,CAAC,GAAG;IAC/E,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,wBAAgB,eAAe,CAAC,OAAO,EAAE;IACvC,IAAI,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,cAAc,CAAC,CAAC;IAC5C,OAAO,EAAE,iBAAiB,CAAC;IAC3B,MAAM,EAAE,IAAI,CAAC,gBAAgB,EAAE,MAAM,GAAG,iBAAiB,CAAC,CAAC;IAC3D,GAAG,EAAE,MAAM,CAAC;IACZ,oBAAoB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC7C,UAAU,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;IAChC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,CAAC;IACjC,YAAY,EAAE,MAAM,IAAI,CAAC;CAC1B;;;;2BAU+B,MAAM;;EA8HrC"}
@@ -0,0 +1,144 @@
1
+ import { randomUUID } from "node:crypto";
2
+ import { managedAgentManifest, resolveHostShell, validateManagedAgentToolSnapshot, withExternalMcpTools, } from "@ai-platform/api-contracts";
3
+ import { createEditToolDefinition, createFindToolDefinition, createGrepToolDefinition, createLsToolDefinition, createReadToolDefinition, createWriteToolDefinition, } from "@earendil-works/pi-coding-agent";
4
+ import { redactHostText } from "./host-log.js";
5
+ import { createHostShellTool } from "./host-tools.js";
6
+ import { PlatformClientError } from "./platform-client.js";
7
+ export function generalRunTools(options) {
8
+ const { spec, binding, client, cwd, environmentVariables } = options;
9
+ const manifest = withExternalMcpTools(managedAgentManifest(spec.agentDefinition.id, spec.agentDefinition.version), spec.resolvedAgentConfig.externalTools);
10
+ const tools = validateManagedAgentToolSnapshot(manifest, spec.resolvedAgentConfig.tools);
11
+ const names = [];
12
+ const platformNames = new Map();
13
+ const callIds = new Map();
14
+ const toolCallId = (sourceId) => {
15
+ let id = callIds.get(sourceId);
16
+ if (!id) {
17
+ id = randomUUID();
18
+ callIds.set(sourceId, id);
19
+ }
20
+ return id;
21
+ };
22
+ const shell = resolveHostShell(process.platform, environmentVariables);
23
+ const definitions = [];
24
+ for (const tool of tools) {
25
+ if (tool.id === "workspace.command") {
26
+ // 显式配置的本地文件与命令能力复用 Pi 原生参数及各操作系统的执行后端。
27
+ for (const native of [
28
+ createHostShellTool(cwd, shell, environmentVariables),
29
+ createReadToolDefinition(cwd),
30
+ createWriteToolDefinition(cwd),
31
+ createEditToolDefinition(cwd),
32
+ createFindToolDefinition(cwd),
33
+ createGrepToolDefinition(cwd),
34
+ createLsToolDefinition(cwd),
35
+ ]) {
36
+ definitions.push(native);
37
+ platformNames.set(native.name, tool.id);
38
+ }
39
+ }
40
+ else {
41
+ definitions.push({
42
+ name: tool.providerName,
43
+ label: tool.providerName,
44
+ description: tool.description,
45
+ parameters: tool.parameters,
46
+ async execute(sourceId, args, signal) {
47
+ const id = toolCallId(sourceId);
48
+ let response;
49
+ try {
50
+ response = await client.tool(spec.runId, tool.id, {
51
+ ...binding,
52
+ toolCallId: id,
53
+ idempotencyKey: id,
54
+ graphNodeId: "general-pi",
55
+ input: args,
56
+ }, signal);
57
+ }
58
+ catch (error) {
59
+ if (!(error instanceof PlatformClientError) || error.status >= 500)
60
+ options.protocolLost();
61
+ throw error;
62
+ }
63
+ return {
64
+ content: [{ type: "text", text: JSON.stringify(response.result) }],
65
+ details: {},
66
+ };
67
+ },
68
+ });
69
+ platformNames.set(tool.providerName, tool.id);
70
+ }
71
+ }
72
+ names.push(...definitions.map((definition) => definition.name));
73
+ const extension = (api) => {
74
+ for (const definition of definitions) {
75
+ api.registerTool({
76
+ ...definition,
77
+ async execute(id, args, signal, update, context) {
78
+ await options.beforeTool();
79
+ let result;
80
+ const native = platformNames.get(definition.name) === "workspace.command";
81
+ const record = async (outcome) => {
82
+ if (!native)
83
+ return;
84
+ const factId = toolCallId(id);
85
+ const input = args;
86
+ try {
87
+ await client.recordToolFacts(spec.runId, {
88
+ ...binding,
89
+ facts: [
90
+ {
91
+ toolCallId: factId,
92
+ idempotencyKey: factId,
93
+ graphNodeId: "general-pi",
94
+ toolName: "workspace.command",
95
+ outcome,
96
+ input: {
97
+ command: redactHostText(options.redact(typeof input.command === "string"
98
+ ? input.command
99
+ : `${definition.name} ${typeof input.path === "string" ? input.path : cwd}`)),
100
+ },
101
+ ...(outcome === "FAILED" ? { errorCode: "LOCAL_TOOL_FAILED" } : {}),
102
+ },
103
+ ],
104
+ });
105
+ }
106
+ catch (error) {
107
+ options.protocolLost();
108
+ throw error;
109
+ }
110
+ };
111
+ try {
112
+ result = await definition.execute(id, args, signal, update, context);
113
+ }
114
+ catch (error) {
115
+ await record("FAILED");
116
+ throw new Error(options.redact(error instanceof Error ? error.message : String(error)));
117
+ }
118
+ await record(result.isError === true ? "FAILED" : "SUCCEEDED");
119
+ return {
120
+ ...result,
121
+ ...(result.details !== undefined
122
+ ? { details: redactValues(result.details, options.redact) }
123
+ : {}),
124
+ ...(result.structuredContent !== undefined
125
+ ? { structuredContent: redactValues(result.structuredContent, options.redact) }
126
+ : {}),
127
+ content: result.content.map((part) => part.type === "text" ? { ...part, text: options.redact(part.text) } : part),
128
+ };
129
+ },
130
+ });
131
+ }
132
+ };
133
+ return { extension, names, platformNames, toolCallId, shell };
134
+ }
135
+ function redactValues(value, redact) {
136
+ if (typeof value === "string")
137
+ return redact(value);
138
+ if (Array.isArray(value))
139
+ return value.map((item) => redactValues(item, redact));
140
+ if (value && typeof value === "object")
141
+ return Object.fromEntries(Object.entries(value).map(([key, item]) => [key, redactValues(item, redact)]));
142
+ return value;
143
+ }
144
+ //# sourceMappingURL=general-run-tools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"general-run-tools.js","sourceRoot":"","sources":["../src/general-run-tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAEL,oBAAoB,EACpB,gBAAgB,EAChB,gCAAgC,EAChC,oBAAoB,GACrB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,wBAAwB,EACxB,wBAAwB,EACxB,wBAAwB,EACxB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,GAG1B,MAAM,iCAAiC,CAAC;AAGzC,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAO3D,MAAM,UAAU,eAAe,CAAC,OAS/B;IACC,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,oBAAoB,EAAE,GAAG,OAAO,CAAC;IACrE,MAAM,QAAQ,GAAG,oBAAoB,CACnC,oBAAoB,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,EAC3E,IAAI,CAAC,mBAAmB,CAAC,aAAa,CACvC,CAAC;IACF,MAAM,KAAK,GAAG,gCAAgC,CAAC,QAAQ,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACzF,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC;IAChD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC1C,MAAM,UAAU,GAAG,CAAC,QAAgB,EAAE,EAAE;QACtC,IAAI,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC/B,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,EAAE,GAAG,UAAU,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC;IACF,MAAM,KAAK,GAAG,gBAAgB,CAAC,OAAO,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;IACvE,MAAM,WAAW,GAAqB,EAAE,CAAC;IACzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,EAAE,KAAK,mBAAmB,EAAE,CAAC;YACpC,uCAAuC;YACvC,KAAK,MAAM,MAAM,IAAI;gBACnB,mBAAmB,CAAC,GAAG,EAAE,KAAK,EAAE,oBAAoB,CAAC;gBACrD,wBAAwB,CAAC,GAAG,CAAC;gBAC7B,yBAAyB,CAAC,GAAG,CAAC;gBAC9B,wBAAwB,CAAC,GAAG,CAAC;gBAC7B,wBAAwB,CAAC,GAAG,CAAC;gBAC7B,wBAAwB,CAAC,GAAG,CAAC;gBAC7B,sBAAsB,CAAC,GAAG,CAAC;aAC5B,EAAE,CAAC;gBACF,WAAW,CAAC,IAAI,CAAC,MAAwB,CAAC,CAAC;gBAC3C,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,IAAI,CAAC,YAAY;gBACvB,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAqB;gBACtC,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM;oBAClC,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;oBAChC,IAAI,QAAuD,CAAC;oBAC5D,IAAI,CAAC;wBACH,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAC1B,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,EAAE,EACP;4BACE,GAAG,OAAO;4BACV,UAAU,EAAE,EAAE;4BACd,cAAc,EAAE,EAAE;4BAClB,WAAW,EAAE,YAAY;4BACzB,KAAK,EAAE,IAA+B;yBACvC,EACD,MAAM,CACP,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,CAAC,CAAC,KAAK,YAAY,mBAAmB,CAAC,IAAI,KAAK,CAAC,MAAM,IAAI,GAAG;4BAChE,OAAO,CAAC,YAAY,EAAE,CAAC;wBACzB,MAAM,KAAK,CAAC;oBACd,CAAC;oBACD,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;wBAClE,OAAO,EAAE,EAAE;qBACZ,CAAC;gBACJ,CAAC;aACF,CAAC,CAAC;YACH,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;IAChE,MAAM,SAAS,GAAqB,CAAC,GAAG,EAAE,EAAE;QAC1C,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,GAAG,CAAC,YAAY,CAAC;gBACf,GAAG,UAAU;gBACb,KAAK,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO;oBAC7C,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;oBAC3B,IAAI,MAAsD,CAAC;oBAC3D,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,mBAAmB,CAAC;oBAC1E,MAAM,MAAM,GAAG,KAAK,EAAE,OAA+B,EAAE,EAAE;wBACvD,IAAI,CAAC,MAAM;4BAAE,OAAO;wBACpB,MAAM,MAAM,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC;wBAC9B,MAAM,KAAK,GAAG,IAA+B,CAAC;wBAC9C,IAAI,CAAC;4BACH,MAAM,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,EAAE;gCACvC,GAAG,OAAO;gCACV,KAAK,EAAE;oCACL;wCACE,UAAU,EAAE,MAAM;wCAClB,cAAc,EAAE,MAAM;wCACtB,WAAW,EAAE,YAAY;wCACzB,QAAQ,EAAE,mBAAmB;wCAC7B,OAAO;wCACP,KAAK,EAAE;4CACL,OAAO,EAAE,cAAc,CACrB,OAAO,CAAC,MAAM,CACZ,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;gDAC/B,CAAC,CAAC,KAAK,CAAC,OAAO;gDACf,CAAC,CAAC,GAAG,UAAU,CAAC,IAAI,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,CAC9E,CACF;yCACF;wCACD,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qCACpE;iCACF;6BACF,CAAC,CAAC;wBACL,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,OAAO,CAAC,YAAY,EAAE,CAAC;4BACvB,MAAM,KAAK,CAAC;wBACd,CAAC;oBACH,CAAC,CAAC;oBACF,IAAI,CAAC;wBACH,MAAM,GAAG,MAAM,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;oBACvE,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACvB,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAC1F,CAAC;oBACD,MAAM,MAAM,CAAC,MAAM,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBAC/D,OAAO;wBACL,GAAG,MAAM;wBACT,GAAG,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS;4BAC9B,CAAC,CAAC,EAAE,OAAO,EAAE,YAAY,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE;4BAC3D,CAAC,CAAC,EAAE,CAAC;wBACP,GAAG,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS;4BACxC,CAAC,CAAC,EAAE,iBAAiB,EAAE,YAAY,CAAC,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,MAAM,CAAC,EAAE;4BAC/E,CAAC,CAAC,EAAE,CAAC;wBACP,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACnC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAC3E;qBACF,CAAC;gBACJ,CAAC;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC;IACF,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AAChE,CAAC;AAED,SAAS,YAAY,CAAI,KAAQ,EAAE,MAAgC;IACjE,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC,KAAK,CAAM,CAAC;IACzD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAM,CAAC;IACtF,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ;QACpC,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,CACzE,CAAC;IACT,OAAO,KAAK,CAAC;AACf,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"host-status.d.ts","sourceRoot":"","sources":["../src/host-status.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAIjF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAiC,KAAK,SAAS,EAAE,MAAM,YAAY,CAAC;AAG3E,eAAO,MAAM,sBAAsB,QAAe,CAAC;AACnD,KAAK,EAAE,GAAG,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAClD,KAAK,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;AAClE,KAAK,MAAM,GAAG;IACZ,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC,CAAC;IACtD,YAAY,EAAE,YAAY,GAAG,SAAS,CAAC;IACvC,UAAU,EAAE,8BAA8B,EAAE,GAAG,SAAS,CAAC;CAC1D,CAAC;AAEF,oFAAoF;AACpF,qBAAa,kBAAkB;IAQ3B,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,EAAE;IACnB,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;IAV1B,OAAO,CAAC,QAAQ,CAA8B;IAC9C,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,eAAe,CAAC,CAAS;IACjC,OAAO,CAAC,YAAY,CAAqB;gBAGtB,MAAM,EAAE,cAAc,EACtB,EAAE,EAAE,EAAE,EACN,IAAI,EAAE,SAAS,EACf,MAAM,CAAC,EAAE,MAAM,YAAA;IAGlC,IAAI,OAAO,YAEV;IAED,SAAS;IAIT,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM;IAQpB,MAAM,CAAC,UAAU,UAAQ,GAAG,OAAO,CAAC,MAAM,CAAC;YAgBnC,aAAa;CA+C5B;AAED,wBAAsB,kBAAkB,CAAC,EAAE,EAAE,EAAE;;;;iBAWuh4P,0DAAsB;qBAA0D,0DAAsB;iBAAsD,0DAAsB;oBAAyD,0DAAsB;kBAAuD,0DAAsB;mBAAwD,0DAAsB;iBAAsD,0DAAsB;;;oBAA8G,0DAAsB;;;;+BAA+J,0DAAsB;;;GADt25P"}
1
+ {"version":3,"file":"host-status.d.ts","sourceRoot":"","sources":["../src/host-status.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,4BAA4B,CAAC;AAIjF,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAClE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,EAAE,KAAK,SAAS,EAAiC,MAAM,YAAY,CAAC;AAG3E,eAAO,MAAM,sBAAsB,QAAe,CAAC;AACnD,KAAK,EAAE,GAAG,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAClD,KAAK,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;AAClE,KAAK,MAAM,GAAG;IACZ,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,kBAAkB,CAAC,CAAC,CAAC;IACtD,YAAY,EAAE,YAAY,GAAG,SAAS,CAAC;IACvC,UAAU,EAAE,8BAA8B,EAAE,GAAG,SAAS,CAAC;CAC1D,CAAC;AAEF,oFAAoF;AACpF,qBAAa,kBAAkB;IAQ3B,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,EAAE;IACnB,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC;IAV1B,OAAO,CAAC,QAAQ,CAA8B;IAC9C,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,YAAY,CAAC,CAAS;IAC9B,OAAO,CAAC,eAAe,CAAC,CAAS;IACjC,OAAO,CAAC,YAAY,CAAqB;gBAGtB,MAAM,EAAE,cAAc,EACtB,EAAE,EAAE,EAAE,EACN,IAAI,EAAE,SAAS,EACf,MAAM,CAAC,EAAE,MAAM,YAAA;IAGlC,IAAI,OAAO,YAEV;IAED,SAAS;IAIT,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM;IAQpB,MAAM,CAAC,UAAU,UAAQ,GAAG,OAAO,CAAC,MAAM,CAAC;YAgBnC,aAAa;CAgD5B;AAED,wBAAsB,kBAAkB,CAAC,EAAE,EAAE,EAAE;;;;iBAWu+vQ,0DAAsB;qBAA0D,0DAAsB;iBAAsD,0DAAsB;oBAAyD,0DAAsB;kBAAuD,0DAAsB;mBAAwD,0DAAsB;iBAAsD,0DAAsB;;;oBAA8G,0DAAsB;;;;+BAA+J,0DAAsB;;;GADtzxQ"}
@@ -77,6 +77,7 @@ export class HostStatusReporter {
77
77
  generation: this.host.generation,
78
78
  ...(this.bootId ? { bootId: this.bootId } : {}),
79
79
  ...local,
80
+ generalRunsSupported: true,
80
81
  catalogRevision: this.host.catalogRevision,
81
82
  ...(toolChecks && verification?.toolConfigurationRevision !== undefined
82
83
  ? { toolChecks, toolConfigurationRevision: verification.toolConfigurationRevision }
@@ -1 +1 @@
1
- {"version":3,"file":"host-status.js","sourceRoot":"","sources":["../src/host-status.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAE,aAAa,EAAE,cAAc,EAAkB,MAAM,YAAY,CAAC;AAE3E,MAAM,kBAAkB,GAAG,CAAC,GAAG,MAAM,CAAC;AACtC,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,GAAG,MAAM,CAAC;AASnD,oFAAoF;AACpF,MAAM,OAAO,kBAAkB;IAQV;IACA;IACA;IACA;IAVX,QAAQ,CAA8B;IACtC,aAAa,GAAG,KAAK,CAAC;IACtB,YAAY,CAAU;IACtB,eAAe,CAAU;IACzB,YAAY,CAAqB;IAEzC,YACmB,MAAsB,EACtB,EAAM,EACN,IAAe,EACf,MAAe;QAHf,WAAM,GAAN,MAAM,CAAgB;QACtB,OAAE,GAAF,EAAE,CAAI;QACN,SAAI,GAAJ,IAAI,CAAW;QACf,WAAM,GAAN,MAAM,CAAS;IAC/B,CAAC;IAEJ,IAAI,OAAO;QACT,OAAO,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,YAAY,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,YAAY,IAAI,kBAAkB,CAAC;IACjG,CAAC;IAED,QAAQ,CAAC,QAAiB;QACxB,OAAO,CACL,IAAI,CAAC,eAAe,KAAK,SAAS;YAClC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,sBAAsB;YAC3D,CAAC,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI,CAAC,YAAY,CAAC,CAC3D,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,KAAK;QAC7B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,aAAa;gBAAE,OAAO,IAAI,CAAC,QAAQ,CAAC;YAC5D,mFAAmF;YACnF,MAAM,IAAI,CAAC,QAAQ,CAAC;YACpB,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACvD,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;YAC1B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,UAAmB;QAC7C,MAAM,KAAK,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChD,IAAI,YAAsC,CAAC;QAC3C,IAAI,UAAwD,CAAC;QAC7D,IAAI,UAAU,EAAE,CAAC;YACf,YAAY,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACtC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;gBACxB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;aACjC,CAAC,CAAC;YACH,IACE,CAAC,YAAY,CAAC,eAAe;gBAC7B,CAAC,YAAY,CAAC,KAAK;gBACnB,CAAC,YAAY,CAAC,yBAAyB;gBAEvC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,UAAU,GAAG,MAAM,cAAc,CAC/B,YAAY,CAAC,KAAK,EAClB,YAAY,CAAC,oBAAoB,IAAI,EAAE,EACvC;gBACE,MAAM,EAAE,eAAe;gBACvB,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;gBACxB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;aACjC,CACF,CAAC;QACJ,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,aAAa,EAAE,CAAC;QACrC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI,CAAC,IAAI,CAAC,UAAU;YAC7F,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC5F,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;YACxB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;YAChC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/C,GAAG,KAAK;YACR,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,eAAe;YAC1C,GAAG,CAAC,UAAU,IAAI,YAAY,EAAE,yBAAyB,KAAK,SAAS;gBACrE,CAAC,CAAC,EAAE,UAAU,EAAE,yBAAyB,EAAE,YAAY,CAAC,yBAAyB,EAAE;gBACnF,CAAC,CAAC,EAAE,CAAC;SACR,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,GAAG,YAAY,EAAE,yBAAyB,CAAC;QAC9D,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC;IAC7C,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,EAAM;IAC7C,uFAAuF;IACvF,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,kBAAkB,EAAE,CAAC;IAChD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,mBAAmB,EAAE,CAAC;IAC9C,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC,aAAa,CAAC,CAAC;IAC9D,OAAO;QACL,SAAS;QACT,MAAM;QACN,mBAAmB,EAAE,SAAS,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAE,MAAgB,CAAC,CAAC,CAAE,MAAgB;KACzF,CAAC;AACJ,CAAC"}
1
+ {"version":3,"file":"host-status.js","sourceRoot":"","sources":["../src/host-status.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAGnC,OAAO,EAAkB,aAAa,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE3E,MAAM,kBAAkB,GAAG,CAAC,GAAG,MAAM,CAAC;AACtC,MAAM,CAAC,MAAM,sBAAsB,GAAG,GAAG,GAAG,MAAM,CAAC;AASnD,oFAAoF;AACpF,MAAM,OAAO,kBAAkB;IAQV;IACA;IACA;IACA;IAVX,QAAQ,CAA8B;IACtC,aAAa,GAAG,KAAK,CAAC;IACtB,YAAY,CAAU;IACtB,eAAe,CAAU;IACzB,YAAY,CAAqB;IAEzC,YACmB,MAAsB,EACtB,EAAM,EACN,IAAe,EACf,MAAe;QAHf,WAAM,GAAN,MAAM,CAAgB;QACtB,OAAE,GAAF,EAAE,CAAI;QACN,SAAI,GAAJ,IAAI,CAAW;QACf,WAAM,GAAN,MAAM,CAAS;IAC/B,CAAC;IAEJ,IAAI,OAAO;QACT,OAAO,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChC,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,YAAY,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,YAAY,IAAI,kBAAkB,CAAC;IACjG,CAAC;IAED,QAAQ,CAAC,QAAiB;QACxB,OAAO,CACL,IAAI,CAAC,eAAe,KAAK,SAAS;YAClC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,eAAe,IAAI,sBAAsB;YAC3D,CAAC,QAAQ,KAAK,SAAS,IAAI,QAAQ,KAAK,IAAI,CAAC,YAAY,CAAC,CAC3D,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,UAAU,GAAG,KAAK;QAC7B,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,aAAa;gBAAE,OAAO,IAAI,CAAC,QAAQ,CAAC;YAC5D,mFAAmF;YACnF,MAAM,IAAI,CAAC,QAAQ,CAAC;YACpB,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QACD,IAAI,CAAC,aAAa,GAAG,UAAU,CAAC;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACvD,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC;YAC1B,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,UAAmB;QAC7C,MAAM,KAAK,GAAG,MAAM,kBAAkB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAChD,IAAI,YAAsC,CAAC;QAC3C,IAAI,UAAwD,CAAC;QAC7D,IAAI,UAAU,EAAE,CAAC;YACf,YAAY,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;gBACtC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;gBACxB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;aACjC,CAAC,CAAC;YACH,IACE,CAAC,YAAY,CAAC,eAAe;gBAC7B,CAAC,YAAY,CAAC,KAAK;gBACnB,CAAC,YAAY,CAAC,yBAAyB;gBAEvC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,UAAU,GAAG,MAAM,cAAc,CAC/B,YAAY,CAAC,KAAK,EAClB,YAAY,CAAC,oBAAoB,IAAI,EAAE,EACvC;gBACE,MAAM,EAAE,eAAe;gBACvB,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;gBACxB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;aACjC,CACF,CAAC;QACJ,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,aAAa,EAAE,CAAC;QACrC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI,CAAC,IAAI,CAAC,UAAU;YAC7F,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAC5F,MAAM,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChC,MAAM,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;YAC7B,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM;YACxB,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU;YAChC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC/C,GAAG,KAAK;YACR,oBAAoB,EAAE,IAAI;YAC1B,eAAe,EAAE,IAAI,CAAC,IAAI,CAAC,eAAe;YAC1C,GAAG,CAAC,UAAU,IAAI,YAAY,EAAE,yBAAyB,KAAK,SAAS;gBACrE,CAAC,CAAC,EAAE,UAAU,EAAE,yBAAyB,EAAE,YAAY,CAAC,yBAAyB,EAAE;gBACnF,CAAC,CAAC,EAAE,CAAC;SACR,CAAC,CAAC;QACH,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC/B,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,GAAG,YAAY,EAAE,yBAAyB,CAAC;QAC9D,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC;IAC7C,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,EAAM;IAC7C,uFAAuF;IACvF,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC,kBAAkB,EAAE,CAAC;IAChD,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,mBAAmB,EAAE,CAAC;IAC9C,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,CAAC,aAAa,CAAC,CAAC;IAC9D,OAAO;QACL,SAAS;QACT,MAAM;QACN,mBAAmB,EAAE,SAAS,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAE,MAAgB,CAAC,CAAC,CAAE,MAAgB;KACzF,CAAC;AACJ,CAAC"}
@@ -13,6 +13,16 @@ export interface RuntimeModelBinding {
13
13
  hostId: string;
14
14
  generation: number;
15
15
  }
16
+ export interface GeneralRuntimeModelBinding {
17
+ kind: "GENERAL";
18
+ runId: string;
19
+ hostId: string;
20
+ generation: number;
21
+ runAttemptId: string;
22
+ attemptNo: number;
23
+ executionToken: string;
24
+ taskToken: string;
25
+ }
16
26
  export interface PiModelService {
17
27
  login(providerId: LocalCodingHostProviderId, mode: "local" | "remote"): Promise<{
18
28
  code: string;
@@ -20,7 +30,7 @@ export interface PiModelService {
20
30
  logout(providerId: LocalCodingHostProviderId): Promise<void>;
21
31
  readProviderStates(): Promise<ProviderState[]>;
22
32
  readAvailableModels(): Promise<LocalCodingHostModel[]>;
23
- resolveRuntimeModel?(providerId: LocalCodingHostProviderId, modelId: string, binding?: RuntimeModelBinding): Promise<{
33
+ resolveRuntimeModel?(providerId: LocalCodingHostProviderId, modelId: string, binding?: RuntimeModelBinding | GeneralRuntimeModelBinding): Promise<{
24
34
  runtime: ModelRuntime;
25
35
  model: Awaited<ReturnType<ModelRuntime["getAvailable"]>>[number];
26
36
  }>;
@@ -43,7 +53,7 @@ export declare class ProductionPiModelService implements PiModelService {
43
53
  thinkingLevelAdaptations: Record<"off" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max", "off" | "minimal" | "low" | "medium" | "high" | "xhigh" | "max">;
44
54
  allowedThinkingLevels: LocalCodingHostThinkingLevel[];
45
55
  }[]>;
46
- resolveRuntimeModel(providerId: LocalCodingHostProviderId, modelId: string, binding?: RuntimeModelBinding): Promise<{
56
+ resolveRuntimeModel(providerId: LocalCodingHostProviderId, modelId: string, binding?: RuntimeModelBinding | GeneralRuntimeModelBinding): Promise<{
47
57
  runtime: ModelRuntime;
48
58
  model: Model<import("@earendil-works/pi-ai").Api>;
49
59
  }>;