shariq-pi-extensions 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +94 -0
- package/THIRD_PARTY_NOTICES.md +11 -0
- package/docs/ARCHITECTURE.md +54 -0
- package/docs/DEVELOPMENT.md +88 -0
- package/docs/EXTENSIONS.md +89 -0
- package/extensions/antigravity-provider/LICENSE +21 -0
- package/extensions/antigravity-provider/README.md +23 -0
- package/extensions/antigravity-provider/antigravity/cloud-code-assist.ts +519 -0
- package/extensions/antigravity-provider/antigravity/index.ts +67 -0
- package/extensions/antigravity-provider/antigravity/models.ts +184 -0
- package/extensions/antigravity-provider/antigravity/oauth.ts +499 -0
- package/extensions/antigravity-provider/index.ts +3 -0
- package/extensions/ask-user/README.md +21 -0
- package/extensions/ask-user/index.ts +417 -0
- package/extensions/background-terminals/README.md +43 -0
- package/extensions/background-terminals/index.ts +400 -0
- package/extensions/background-terminals/src/manager.ts +480 -0
- package/extensions/background-terminals/src/output-buffer.ts +110 -0
- package/extensions/background-terminals/src/presentation.ts +74 -0
- package/extensions/background-terminals/src/types.ts +47 -0
- package/extensions/background-terminals/src/ui.ts +478 -0
- package/extensions/context-usage/LICENSE +21 -0
- package/extensions/context-usage/README.md +20 -0
- package/extensions/context-usage/index.ts +2006 -0
- package/extensions/context-usage/lib/ansi.ts +60 -0
- package/extensions/context-usage/lib/boundary.ts +27 -0
- package/extensions/context-usage/lib/chat.ts +190 -0
- package/extensions/context-usage/lib/config.ts +30 -0
- package/extensions/context-usage/lib/fmt.ts +59 -0
- package/extensions/context-usage/lib/style.ts +260 -0
- package/extensions/copy-all/README.md +15 -0
- package/extensions/copy-all/index.ts +72 -0
- package/extensions/factory-provider/API_KEYS.md +17 -0
- package/extensions/factory-provider/AUTH.md +15 -0
- package/extensions/factory-provider/README.md +90 -0
- package/extensions/factory-provider/REFRESH_MODELS.md +185 -0
- package/extensions/factory-provider/UPDATING.md +97 -0
- package/extensions/factory-provider/factory/api-keys.ts +407 -0
- package/extensions/factory-provider/factory/auth.ts +132 -0
- package/extensions/factory-provider/factory/constants.ts +15 -0
- package/extensions/factory-provider/factory/dashboard.ts +363 -0
- package/extensions/factory-provider/factory/droid.ts +84 -0
- package/extensions/factory-provider/factory/limits.ts +342 -0
- package/extensions/factory-provider/factory/models.ts +283 -0
- package/extensions/factory-provider/factory/responses.ts +331 -0
- package/extensions/factory-provider/factory/warm-theme.ts +53 -0
- package/extensions/factory-provider/factory/websocket.ts +130 -0
- package/extensions/factory-provider/index.ts +204 -0
- package/extensions/factory-provider/scripts/audit-reasoning-efforts.py +84 -0
- package/extensions/firecrawl-web/README.md +31 -0
- package/extensions/firecrawl-web/auth.ts +92 -0
- package/extensions/firecrawl-web/client.ts +238 -0
- package/extensions/firecrawl-web/index.ts +96 -0
- package/extensions/firecrawl-web/output.ts +125 -0
- package/extensions/git-info/README.md +25 -0
- package/extensions/git-info/index.ts +162 -0
- package/extensions/git-info/src/git.ts +206 -0
- package/extensions/git-info/src/ui.ts +291 -0
- package/extensions/goal/README.md +44 -0
- package/extensions/goal/constants.ts +4 -0
- package/extensions/goal/extension.ts +1064 -0
- package/extensions/goal/index.ts +2 -0
- package/extensions/goal/types.ts +56 -0
- package/extensions/goal/ui.ts +334 -0
- package/extensions/orchestration/README.md +58 -0
- package/extensions/orchestration/dashboard.ts +192 -0
- package/extensions/orchestration/engine.ts +706 -0
- package/extensions/orchestration/index.ts +243 -0
- package/extensions/orchestration/prompts.ts +124 -0
- package/extensions/orchestration/protocol.ts +116 -0
- package/extensions/orchestration/settings-ui.ts +41 -0
- package/extensions/orchestration/settings.ts +59 -0
- package/extensions/orchestration/storage.ts +49 -0
- package/extensions/orchestration/types.ts +105 -0
- package/extensions/performance-status/README.md +5 -0
- package/extensions/performance-status/index.ts +171 -0
- package/extensions/pi-memory/README.md +53 -0
- package/extensions/pi-memory/index.ts +326 -0
- package/extensions/pi-memory/src/config.ts +75 -0
- package/extensions/pi-memory/src/database.ts +617 -0
- package/extensions/pi-memory/src/extraction.ts +178 -0
- package/extensions/pi-memory/src/project.ts +80 -0
- package/extensions/pi-memory/src/projection.ts +59 -0
- package/extensions/pi-memory/src/redaction.ts +60 -0
- package/extensions/pi-memory/src/retrieval.ts +30 -0
- package/extensions/pi-memory/src/service.ts +194 -0
- package/extensions/pi-memory/src/session.ts +146 -0
- package/extensions/pi-memory/src/types.ts +82 -0
- package/extensions/shared/README.md +18 -0
- package/extensions/shared/activity-status.ts +31 -0
- package/extensions/shared/child-session.ts +150 -0
- package/extensions/shared/context-utilization.ts +47 -0
- package/extensions/shared/dashboard-state.ts +99 -0
- package/extensions/shared/tool-call-timeout.ts +104 -0
- package/extensions/shared/tui-dashboard.ts +139 -0
- package/extensions/shell-shortcuts/README.md +17 -0
- package/extensions/shell-shortcuts/features/exit-alias.ts +10 -0
- package/extensions/shell-shortcuts/index.ts +6 -0
- package/extensions/subagents/README.md +89 -0
- package/extensions/subagents/index.ts +1386 -0
- package/extensions/subagents/src/backend.ts +69 -0
- package/extensions/subagents/src/backends/pi.ts +781 -0
- package/extensions/subagents/src/catalog.ts +85 -0
- package/extensions/subagents/src/config.ts +241 -0
- package/extensions/subagents/src/context.ts +57 -0
- package/extensions/subagents/src/coordinator.ts +64 -0
- package/extensions/subagents/src/domain.ts +294 -0
- package/extensions/subagents/src/format.ts +74 -0
- package/extensions/subagents/src/manager.ts +813 -0
- package/extensions/subagents/src/prompt.ts +94 -0
- package/extensions/subagents/src/result-delivery.ts +20 -0
- package/extensions/subagents/src/runtime.ts +48 -0
- package/extensions/subagents/src/ui/takeover.ts +695 -0
- package/extensions/subagents/src/ui/transcript.ts +182 -0
- package/extensions/subagents/src/worktree.ts +175 -0
- package/extensions/web-fetch/README.md +13 -0
- package/extensions/web-fetch/index.ts +441 -0
- package/package.json +117 -0
- package/skills/background-terminals/SKILL.md +42 -0
- package/skills/orchestration/SKILL.md +33 -0
- package/skills/subagents/SKILL.md +89 -0
- package/themes/ember-warm-dark.json +85 -0
|
@@ -0,0 +1,781 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* pi backend — real implementation over the pi SDK.
|
|
3
|
+
*
|
|
4
|
+
* Each subagent is an in-process `AgentSession` (a port of v1
|
|
5
|
+
* subagents/manager.ts + shared/child-session.ts):
|
|
6
|
+
* - real session files visible in /resume, child resources loaded per-cwd
|
|
7
|
+
* with trust gating, and the child tool denylist;
|
|
8
|
+
* - `session.subscribe()` events translated to normalized SubagentEvents;
|
|
9
|
+
* - send() steers a streaming run or starts a fresh prompt() when idle;
|
|
10
|
+
* - interrupt clears the queue and aborts; closing the session scope emits
|
|
11
|
+
* the child session_shutdown hook and disposes the session.
|
|
12
|
+
*/
|
|
13
|
+
|
|
14
|
+
import type { AssistantMessage, Message, Model } from "@earendil-works/pi-ai";
|
|
15
|
+
import type {
|
|
16
|
+
AgentSession,
|
|
17
|
+
AgentSessionEvent,
|
|
18
|
+
ModelRegistry,
|
|
19
|
+
ToolDefinition,
|
|
20
|
+
} from "@earendil-works/pi-coding-agent";
|
|
21
|
+
import {
|
|
22
|
+
createAgentSession,
|
|
23
|
+
DefaultResourceLoader,
|
|
24
|
+
getAgentDir,
|
|
25
|
+
SessionManager,
|
|
26
|
+
SettingsManager,
|
|
27
|
+
} from "@earendil-works/pi-coding-agent";
|
|
28
|
+
import type { Cause, Scope } from "effect";
|
|
29
|
+
import { Effect, Queue, Stream } from "effect";
|
|
30
|
+
import { Type } from "typebox";
|
|
31
|
+
import type { SubagentBackend, SubagentSession } from "../backend.ts";
|
|
32
|
+
import type {
|
|
33
|
+
SpawnTask,
|
|
34
|
+
SubagentEvent,
|
|
35
|
+
SubagentMeta,
|
|
36
|
+
TranscriptPart,
|
|
37
|
+
} from "../domain.ts";
|
|
38
|
+
import { SendError, SpawnError } from "../domain.ts";
|
|
39
|
+
|
|
40
|
+
const CHILD_SHUTDOWN_TIMEOUT_MS = 5_000;
|
|
41
|
+
const CHILD_TOOL_CALL_TIMEOUT_MS = 3 * 60 * 1_000;
|
|
42
|
+
|
|
43
|
+
function capabilityExclusions(capability: SpawnTask["capability"]): string[] {
|
|
44
|
+
switch (capability) {
|
|
45
|
+
case "read-only":
|
|
46
|
+
return ["bash", "edit", "write"];
|
|
47
|
+
case "read-write":
|
|
48
|
+
return ["bash"];
|
|
49
|
+
case "execute":
|
|
50
|
+
return ["edit", "write"];
|
|
51
|
+
case "all":
|
|
52
|
+
return [];
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
const READ_CAPABILITY_TOOLS = new Set([
|
|
57
|
+
"read",
|
|
58
|
+
"grep",
|
|
59
|
+
"find",
|
|
60
|
+
"ls",
|
|
61
|
+
"web_search",
|
|
62
|
+
"web_scrape",
|
|
63
|
+
"web_fetch",
|
|
64
|
+
"pi_memory_search",
|
|
65
|
+
"pi_memory_read",
|
|
66
|
+
"pi_memory_status",
|
|
67
|
+
"get_goal",
|
|
68
|
+
"read_terminal",
|
|
69
|
+
"list_terminals",
|
|
70
|
+
"message_parent",
|
|
71
|
+
"ask_parent",
|
|
72
|
+
"list_peers",
|
|
73
|
+
"message_peer",
|
|
74
|
+
]);
|
|
75
|
+
|
|
76
|
+
const WRITE_CAPABILITY_TOOLS = new Set(["write", "edit"]);
|
|
77
|
+
const EXECUTE_CAPABILITY_TOOLS = new Set([
|
|
78
|
+
"bash",
|
|
79
|
+
"start_terminal",
|
|
80
|
+
"read_terminal",
|
|
81
|
+
"write_terminal",
|
|
82
|
+
"list_terminals",
|
|
83
|
+
"stop_terminal",
|
|
84
|
+
]);
|
|
85
|
+
|
|
86
|
+
/** Restrictive modes fail closed when a newly registered extension tool has no classification. */
|
|
87
|
+
export function filterToolsForCapability(
|
|
88
|
+
toolNames: ReadonlyArray<string>,
|
|
89
|
+
capability: SpawnTask["capability"],
|
|
90
|
+
): string[] {
|
|
91
|
+
if (capability === "all") return [...toolNames];
|
|
92
|
+
return toolNames.filter((name) => {
|
|
93
|
+
if (READ_CAPABILITY_TOOLS.has(name)) return true;
|
|
94
|
+
if (capability === "read-write" && WRITE_CAPABILITY_TOOLS.has(name)) return true;
|
|
95
|
+
if (capability === "execute" && EXECUTE_CAPABILITY_TOOLS.has(name)) return true;
|
|
96
|
+
return false;
|
|
97
|
+
});
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
/** Tools that headless children must not receive. Everything else stays enabled. */
|
|
101
|
+
const CHILD_EXCLUDED_TOOL_NAMES = [
|
|
102
|
+
"spawn_agent",
|
|
103
|
+
"wait_agent",
|
|
104
|
+
"close_agent",
|
|
105
|
+
"check_agent",
|
|
106
|
+
"list_agents",
|
|
107
|
+
"send_message",
|
|
108
|
+
"apply_agent_changes",
|
|
109
|
+
"task",
|
|
110
|
+
"reply_question",
|
|
111
|
+
"workflow",
|
|
112
|
+
"create_orchestration",
|
|
113
|
+
"get_orchestration",
|
|
114
|
+
"ask_user",
|
|
115
|
+
] as const;
|
|
116
|
+
|
|
117
|
+
// --- Model + effort resolution -----------------------------------------------
|
|
118
|
+
|
|
119
|
+
type ThinkingLevel = NonNullable<
|
|
120
|
+
NonNullable<Parameters<typeof createAgentSession>[0]>["thinkingLevel"]
|
|
121
|
+
>;
|
|
122
|
+
|
|
123
|
+
function createParentBridgeTools(task: SpawnTask): ToolDefinition[] {
|
|
124
|
+
const bridge = task.parent.bridge;
|
|
125
|
+
if (!bridge) return [];
|
|
126
|
+
return [
|
|
127
|
+
{
|
|
128
|
+
name: "message_parent",
|
|
129
|
+
label: "Message Parent",
|
|
130
|
+
description: "Send the parent a concise non-blocking update or warning.",
|
|
131
|
+
parameters: Type.Object({
|
|
132
|
+
message: Type.String({ description: "Parent update" }),
|
|
133
|
+
}),
|
|
134
|
+
async execute(_toolCallId, params) {
|
|
135
|
+
const { message } = params as { message: string };
|
|
136
|
+
bridge.notify(message);
|
|
137
|
+
return { content: [{ type: "text", text: "Update sent to parent." }], details: {} };
|
|
138
|
+
},
|
|
139
|
+
},
|
|
140
|
+
{
|
|
141
|
+
name: "ask_parent",
|
|
142
|
+
label: "Ask Parent",
|
|
143
|
+
description: "Ask the parent a blocking question required to continue.",
|
|
144
|
+
parameters: Type.Object({
|
|
145
|
+
question: Type.String({ description: "Parent question" }),
|
|
146
|
+
}),
|
|
147
|
+
async execute(_toolCallId, params, signal) {
|
|
148
|
+
const { question } = params as { question: string };
|
|
149
|
+
const reply = await bridge.ask(question, signal);
|
|
150
|
+
return { content: [{ type: "text", text: `Parent reply:\n${reply}` }], details: {} };
|
|
151
|
+
},
|
|
152
|
+
},
|
|
153
|
+
{
|
|
154
|
+
name: "list_peers",
|
|
155
|
+
label: "List Peer Agents",
|
|
156
|
+
description: "List sibling agents managed by the main thread. This cannot create agents.",
|
|
157
|
+
parameters: Type.Object({}),
|
|
158
|
+
async execute() {
|
|
159
|
+
const peers = bridge.listPeers();
|
|
160
|
+
const text = peers.length
|
|
161
|
+
? peers.map((peer) => `${peer.id}\t${peer.status}\t${peer.title}`).join("\n")
|
|
162
|
+
: "No peer agents are currently tracked.";
|
|
163
|
+
return { content: [{ type: "text", text }], details: { peers } };
|
|
164
|
+
},
|
|
165
|
+
},
|
|
166
|
+
{
|
|
167
|
+
name: "message_peer",
|
|
168
|
+
label: "Message Peer Agent",
|
|
169
|
+
description: "Route a message to an existing sibling agent through the main-thread manager. This cannot create agents.",
|
|
170
|
+
parameters: Type.Object({
|
|
171
|
+
id: Type.String({ description: "Target peer id from list_peers" }),
|
|
172
|
+
message: Type.String({ description: "Message for the peer" }),
|
|
173
|
+
}),
|
|
174
|
+
async execute(_toolCallId, params) {
|
|
175
|
+
const { id, message } = params as { id: string; message: string };
|
|
176
|
+
await bridge.messagePeer(id, message);
|
|
177
|
+
return { content: [{ type: "text", text: `Message sent to ${id}.` }], details: { id } };
|
|
178
|
+
},
|
|
179
|
+
},
|
|
180
|
+
] as ToolDefinition[];
|
|
181
|
+
}
|
|
182
|
+
|
|
183
|
+
/**
|
|
184
|
+
* Resolve the generic model hint against the parent registry (v1 semantics):
|
|
185
|
+
* "provider/model-id" is exact; a bare id prefers the inherited provider,
|
|
186
|
+
* then must be unambiguous across providers. No hint inherits the parent
|
|
187
|
+
* model; with nothing to inherit, the SDK default applies.
|
|
188
|
+
*/
|
|
189
|
+
function resolvePiModel(
|
|
190
|
+
registry: ModelRegistry,
|
|
191
|
+
hint: string | undefined,
|
|
192
|
+
inherited: { provider: string; id: string } | undefined,
|
|
193
|
+
): Model<any> | undefined {
|
|
194
|
+
if (!hint) {
|
|
195
|
+
if (!inherited) return undefined;
|
|
196
|
+
return registry.find(inherited.provider, inherited.id) ?? undefined;
|
|
197
|
+
}
|
|
198
|
+
const slash = hint.indexOf("/");
|
|
199
|
+
if (slash > 0) {
|
|
200
|
+
const provider = hint.slice(0, slash);
|
|
201
|
+
const id = hint.slice(slash + 1);
|
|
202
|
+
const found = registry.find(provider, id);
|
|
203
|
+
if (found) return found;
|
|
204
|
+
throw new Error(`Unknown model "${hint}".`);
|
|
205
|
+
}
|
|
206
|
+
if (inherited) {
|
|
207
|
+
const found = registry.find(inherited.provider, hint);
|
|
208
|
+
if (found) return found;
|
|
209
|
+
}
|
|
210
|
+
const matches = registry.getAll().filter((m) => m.id === hint);
|
|
211
|
+
if (matches.length === 1) return matches[0];
|
|
212
|
+
if (matches.length > 1) {
|
|
213
|
+
throw new Error(
|
|
214
|
+
`Model "${hint}" exists in multiple providers (${matches.map((m) => m.provider).join(", ")}). Use "provider/${hint}".`,
|
|
215
|
+
);
|
|
216
|
+
}
|
|
217
|
+
throw new Error(`Unknown model "${hint}".`);
|
|
218
|
+
}
|
|
219
|
+
|
|
220
|
+
// --- Child session helpers (ported from v1 shared/child-session.ts) -----------
|
|
221
|
+
|
|
222
|
+
/** Load normal global/package resources and trust-gated project resources. */
|
|
223
|
+
async function createChildResources(cwd: string, projectTrusted: boolean) {
|
|
224
|
+
const agentDir = getAgentDir();
|
|
225
|
+
const settingsManager = SettingsManager.create(cwd, agentDir, {
|
|
226
|
+
projectTrusted,
|
|
227
|
+
});
|
|
228
|
+
const loader = new DefaultResourceLoader({ cwd, agentDir, settingsManager });
|
|
229
|
+
await loader.reload();
|
|
230
|
+
return { loader, settingsManager };
|
|
231
|
+
}
|
|
232
|
+
|
|
233
|
+
function waitBounded(operation: Promise<unknown>, timeoutMs: number) {
|
|
234
|
+
let timer: ReturnType<typeof setTimeout> | undefined;
|
|
235
|
+
const timeout = new Promise<void>((resolve) => {
|
|
236
|
+
timer = setTimeout(resolve, timeoutMs);
|
|
237
|
+
});
|
|
238
|
+
return Promise.race([
|
|
239
|
+
operation.then(
|
|
240
|
+
() => undefined,
|
|
241
|
+
() => undefined,
|
|
242
|
+
),
|
|
243
|
+
timeout,
|
|
244
|
+
])
|
|
245
|
+
.catch(() => {})
|
|
246
|
+
.finally(() => {
|
|
247
|
+
if (timer) clearTimeout(timer);
|
|
248
|
+
});
|
|
249
|
+
}
|
|
250
|
+
|
|
251
|
+
/** Emit child session_shutdown (bounded), then dispose. Never throws. */
|
|
252
|
+
async function shutdownAndDisposeChildSession(session: AgentSession) {
|
|
253
|
+
try {
|
|
254
|
+
if (session.extensionRunner.hasHandlers("session_shutdown")) {
|
|
255
|
+
await waitBounded(
|
|
256
|
+
session.extensionRunner.emit({
|
|
257
|
+
type: "session_shutdown",
|
|
258
|
+
reason: "quit",
|
|
259
|
+
}),
|
|
260
|
+
CHILD_SHUTDOWN_TIMEOUT_MS,
|
|
261
|
+
);
|
|
262
|
+
}
|
|
263
|
+
} catch {
|
|
264
|
+
// Extension runner inspection/emission is best-effort during teardown.
|
|
265
|
+
} finally {
|
|
266
|
+
try {
|
|
267
|
+
session.dispose();
|
|
268
|
+
} catch {
|
|
269
|
+
// Disposal is terminal and must remain idempotent for callers.
|
|
270
|
+
}
|
|
271
|
+
}
|
|
272
|
+
}
|
|
273
|
+
|
|
274
|
+
// --- Tool-call timeout guard (ported from v1 shared/tool-call-timeout.ts) -----
|
|
275
|
+
|
|
276
|
+
/**
|
|
277
|
+
* Wrap every registered child tool with an independent execution timeout so a
|
|
278
|
+
* hung tool cannot wedge a headless child forever. apply() is idempotent and
|
|
279
|
+
* re-applied on agent_start to pick up tools registered between runs.
|
|
280
|
+
*/
|
|
281
|
+
function createToolCallTimeoutGuard(timeoutMs = CHILD_TOOL_CALL_TIMEOUT_MS) {
|
|
282
|
+
const wrapped = new WeakSet<ToolDefinition>();
|
|
283
|
+
|
|
284
|
+
const wrap = (definition: ToolDefinition) => {
|
|
285
|
+
if (wrapped.has(definition)) return;
|
|
286
|
+
wrapped.add(definition);
|
|
287
|
+
const execute = definition.execute;
|
|
288
|
+
definition.execute = async (toolCallId, params, signal, onUpdate, ctx) => {
|
|
289
|
+
const timeoutController = new AbortController();
|
|
290
|
+
const executionSignal = signal
|
|
291
|
+
? AbortSignal.any([signal, timeoutController.signal])
|
|
292
|
+
: timeoutController.signal;
|
|
293
|
+
let timer: ReturnType<typeof setTimeout> | undefined;
|
|
294
|
+
const timeout = new Promise<never>((_resolve, reject) => {
|
|
295
|
+
timer = setTimeout(() => {
|
|
296
|
+
const error = new Error(
|
|
297
|
+
`Tool call "${definition.name}" timed out after ${Math.round(timeoutMs / 60_000)} minutes.`,
|
|
298
|
+
);
|
|
299
|
+
reject(error);
|
|
300
|
+
timeoutController.abort(error);
|
|
301
|
+
}, timeoutMs);
|
|
302
|
+
});
|
|
303
|
+
try {
|
|
304
|
+
return await Promise.race([
|
|
305
|
+
execute.call(
|
|
306
|
+
definition,
|
|
307
|
+
toolCallId,
|
|
308
|
+
params,
|
|
309
|
+
executionSignal,
|
|
310
|
+
onUpdate,
|
|
311
|
+
ctx,
|
|
312
|
+
),
|
|
313
|
+
timeout,
|
|
314
|
+
]);
|
|
315
|
+
} finally {
|
|
316
|
+
if (timer) clearTimeout(timer);
|
|
317
|
+
}
|
|
318
|
+
};
|
|
319
|
+
};
|
|
320
|
+
|
|
321
|
+
return {
|
|
322
|
+
apply(session: AgentSession) {
|
|
323
|
+
for (const { name } of session.getAllTools()) {
|
|
324
|
+
const definition = session.getToolDefinition(name);
|
|
325
|
+
if (definition) wrap(definition);
|
|
326
|
+
}
|
|
327
|
+
},
|
|
328
|
+
};
|
|
329
|
+
}
|
|
330
|
+
|
|
331
|
+
// --- Event translation ----------------------------------------------------------
|
|
332
|
+
|
|
333
|
+
function messageRole(msg: unknown): Message["role"] | undefined {
|
|
334
|
+
const role = (msg as { role?: string } | undefined)?.role;
|
|
335
|
+
if (role === "user" || role === "assistant" || role === "toolResult")
|
|
336
|
+
return role;
|
|
337
|
+
return undefined;
|
|
338
|
+
}
|
|
339
|
+
|
|
340
|
+
function lastAssistantMessage(
|
|
341
|
+
session: AgentSession,
|
|
342
|
+
): AssistantMessage | undefined {
|
|
343
|
+
const messages = session.messages;
|
|
344
|
+
for (let i = messages.length - 1; i >= 0; i--) {
|
|
345
|
+
const msg = messages[i];
|
|
346
|
+
if (messageRole(msg) === "assistant") return msg as AssistantMessage;
|
|
347
|
+
}
|
|
348
|
+
return undefined;
|
|
349
|
+
}
|
|
350
|
+
|
|
351
|
+
/** Final assistant text output (last assistant message with text), v1 semantics. */
|
|
352
|
+
function finalOutput(session: AgentSession): string {
|
|
353
|
+
const messages = session.messages;
|
|
354
|
+
for (let i = messages.length - 1; i >= 0; i--) {
|
|
355
|
+
const msg = messages[i];
|
|
356
|
+
if (messageRole(msg) !== "assistant") continue;
|
|
357
|
+
const text = (msg as AssistantMessage).content
|
|
358
|
+
.filter((part) => part.type === "text")
|
|
359
|
+
.map((part) => part.text)
|
|
360
|
+
.join("\n")
|
|
361
|
+
.trim();
|
|
362
|
+
if (text) return text;
|
|
363
|
+
}
|
|
364
|
+
return "";
|
|
365
|
+
}
|
|
366
|
+
|
|
367
|
+
function safeJson(value: unknown): string | undefined {
|
|
368
|
+
try {
|
|
369
|
+
const text = JSON.stringify(value);
|
|
370
|
+
return text === "{}" ? undefined : text;
|
|
371
|
+
} catch {
|
|
372
|
+
return undefined;
|
|
373
|
+
}
|
|
374
|
+
}
|
|
375
|
+
|
|
376
|
+
/** First non-empty line of a tool result-ish value (v1 liveToolPreview). */
|
|
377
|
+
function toolPreview(value: unknown): string | undefined {
|
|
378
|
+
if (typeof value === "string") {
|
|
379
|
+
return value
|
|
380
|
+
.split("\n")
|
|
381
|
+
.find((line) => line.trim())
|
|
382
|
+
?.trim();
|
|
383
|
+
}
|
|
384
|
+
if (!value || typeof value !== "object") return undefined;
|
|
385
|
+
const content = (value as { content?: unknown }).content;
|
|
386
|
+
if (!Array.isArray(content)) return undefined;
|
|
387
|
+
for (const part of content) {
|
|
388
|
+
if (!part || typeof part !== "object") continue;
|
|
389
|
+
const record = part as { type?: unknown; text?: unknown };
|
|
390
|
+
if (record.type !== "text" || typeof record.text !== "string") continue;
|
|
391
|
+
const firstLine = record.text.split("\n").find((line) => line.trim());
|
|
392
|
+
if (firstLine) return firstLine.trim();
|
|
393
|
+
}
|
|
394
|
+
return undefined;
|
|
395
|
+
}
|
|
396
|
+
|
|
397
|
+
function assistantParts(msg: AssistantMessage): TranscriptPart[] {
|
|
398
|
+
const parts: TranscriptPart[] = [];
|
|
399
|
+
for (const part of msg.content) {
|
|
400
|
+
if (part.type === "text") {
|
|
401
|
+
parts.push({ type: "text", text: part.text });
|
|
402
|
+
} else if (part.type === "thinking") {
|
|
403
|
+
parts.push({
|
|
404
|
+
type: "thinking",
|
|
405
|
+
text: part.redacted ? "" : part.thinking,
|
|
406
|
+
redacted: part.redacted,
|
|
407
|
+
});
|
|
408
|
+
} else if (part.type === "toolCall") {
|
|
409
|
+
parts.push({
|
|
410
|
+
type: "toolCall",
|
|
411
|
+
toolId: part.id,
|
|
412
|
+
name: part.name,
|
|
413
|
+
argsPreview: safeJson(part.arguments),
|
|
414
|
+
});
|
|
415
|
+
}
|
|
416
|
+
}
|
|
417
|
+
return parts;
|
|
418
|
+
}
|
|
419
|
+
|
|
420
|
+
function userText(msg: Message): string {
|
|
421
|
+
const content = (msg as { content: unknown }).content;
|
|
422
|
+
if (typeof content === "string") return content;
|
|
423
|
+
if (!Array.isArray(content)) return "";
|
|
424
|
+
return content
|
|
425
|
+
.filter(
|
|
426
|
+
(part): part is { type: "text"; text: string } =>
|
|
427
|
+
!!part &&
|
|
428
|
+
typeof part === "object" &&
|
|
429
|
+
(part as { type?: unknown }).type === "text",
|
|
430
|
+
)
|
|
431
|
+
.map((part) => part.text)
|
|
432
|
+
.join("\n");
|
|
433
|
+
}
|
|
434
|
+
|
|
435
|
+
// --- The session ------------------------------------------------------------------
|
|
436
|
+
|
|
437
|
+
function boundedError(error: unknown) {
|
|
438
|
+
return (error instanceof Error ? error.message : String(error)).slice(
|
|
439
|
+
0,
|
|
440
|
+
4096,
|
|
441
|
+
);
|
|
442
|
+
}
|
|
443
|
+
|
|
444
|
+
const makePiSession = (
|
|
445
|
+
task: SpawnTask,
|
|
446
|
+
): Effect.Effect<SubagentSession, SpawnError, Scope.Scope> =>
|
|
447
|
+
Effect.gen(function* () {
|
|
448
|
+
const registry = task.parent.modelRegistry;
|
|
449
|
+
if (!registry) {
|
|
450
|
+
return yield* new SpawnError({
|
|
451
|
+
message: "pi backend requires the parent session's model registry.",
|
|
452
|
+
});
|
|
453
|
+
}
|
|
454
|
+
|
|
455
|
+
const model = yield* Effect.try({
|
|
456
|
+
try: () =>
|
|
457
|
+
resolvePiModel(registry, task.model, task.parent.inheritedModel),
|
|
458
|
+
catch: (error) => new SpawnError({ message: boundedError(error) }),
|
|
459
|
+
});
|
|
460
|
+
// pi's thinking levels ARE the shared reasoning-effort scale.
|
|
461
|
+
const thinkingLevel = (task.reasoningEffort ??
|
|
462
|
+
task.parent.inheritedThinkingLevel) as ThinkingLevel | undefined;
|
|
463
|
+
|
|
464
|
+
const session = yield* Effect.tryPromise({
|
|
465
|
+
try: async () => {
|
|
466
|
+
const { loader, settingsManager } = await createChildResources(
|
|
467
|
+
task.cwd,
|
|
468
|
+
task.parent.projectTrusted,
|
|
469
|
+
);
|
|
470
|
+
const sessionManager = task.resumeSessionFile
|
|
471
|
+
? SessionManager.open(task.resumeSessionFile, undefined, task.cwd)
|
|
472
|
+
: SessionManager.create(task.cwd, undefined, {
|
|
473
|
+
parentSession: task.parent.parentSessionFile,
|
|
474
|
+
});
|
|
475
|
+
if (!task.resumeSessionFile) {
|
|
476
|
+
for (const message of task.parent.inheritedMessages ?? []) {
|
|
477
|
+
sessionManager.appendMessage(message);
|
|
478
|
+
}
|
|
479
|
+
}
|
|
480
|
+
const { session } = await createAgentSession({
|
|
481
|
+
cwd: task.cwd,
|
|
482
|
+
sessionManager,
|
|
483
|
+
settingsManager,
|
|
484
|
+
resourceLoader: loader,
|
|
485
|
+
model,
|
|
486
|
+
thinkingLevel,
|
|
487
|
+
customTools: createParentBridgeTools(task),
|
|
488
|
+
excludeTools: [
|
|
489
|
+
...CHILD_EXCLUDED_TOOL_NAMES,
|
|
490
|
+
...capabilityExclusions(task.capability),
|
|
491
|
+
],
|
|
492
|
+
});
|
|
493
|
+
// Start child extension session hooks/resources in headless mode.
|
|
494
|
+
// A rejection here would otherwise leak the freshly created session:
|
|
495
|
+
// the scope finalizer that owns cleanup is only registered later.
|
|
496
|
+
try {
|
|
497
|
+
await session.bindExtensions({ mode: "print" });
|
|
498
|
+
session.setActiveToolsByName(
|
|
499
|
+
filterToolsForCapability(session.getActiveToolNames(), task.capability),
|
|
500
|
+
);
|
|
501
|
+
} catch (error) {
|
|
502
|
+
await shutdownAndDisposeChildSession(session);
|
|
503
|
+
throw error;
|
|
504
|
+
}
|
|
505
|
+
return session;
|
|
506
|
+
},
|
|
507
|
+
catch: (error) => new SpawnError({ message: boundedError(error) }),
|
|
508
|
+
});
|
|
509
|
+
|
|
510
|
+
const state = {
|
|
511
|
+
closed: false,
|
|
512
|
+
/** prompt() rejection for the active run; folded into RunSettled. */
|
|
513
|
+
runError: undefined as string | undefined,
|
|
514
|
+
/** One terminal event per run: lifecycle, prompt-rejection, and abort
|
|
515
|
+
* fallbacks can all race to settle; the first wins. */
|
|
516
|
+
settled: false,
|
|
517
|
+
};
|
|
518
|
+
|
|
519
|
+
const events = yield* Queue.make<SubagentEvent, Cause.Done>();
|
|
520
|
+
const emit = (event: SubagentEvent) => {
|
|
521
|
+
Queue.offerUnsafe(events, event);
|
|
522
|
+
};
|
|
523
|
+
|
|
524
|
+
const toolTimeout = createToolCallTimeoutGuard();
|
|
525
|
+
toolTimeout.apply(session);
|
|
526
|
+
|
|
527
|
+
const activeModel = (): Model<any> | undefined => {
|
|
528
|
+
const sessionModel = session.model;
|
|
529
|
+
const last = lastAssistantMessage(session);
|
|
530
|
+
if (!last) return sessionModel;
|
|
531
|
+
if (
|
|
532
|
+
sessionModel &&
|
|
533
|
+
(last.provider !== sessionModel.provider ||
|
|
534
|
+
last.model !== sessionModel.id)
|
|
535
|
+
) {
|
|
536
|
+
// The session changed models after this assistant response.
|
|
537
|
+
return sessionModel;
|
|
538
|
+
}
|
|
539
|
+
return (
|
|
540
|
+
registry.find(last.provider, last.responseModel ?? last.model) ??
|
|
541
|
+
sessionModel
|
|
542
|
+
);
|
|
543
|
+
};
|
|
544
|
+
|
|
545
|
+
const currentMeta = (): SubagentMeta => {
|
|
546
|
+
const m = activeModel();
|
|
547
|
+
return {
|
|
548
|
+
backend: "pi",
|
|
549
|
+
origin: task.origin,
|
|
550
|
+
modelLabel: m ? `${m.provider}/${m.id}` : undefined,
|
|
551
|
+
contextWindow: m?.contextWindow,
|
|
552
|
+
sessionFilePath: session.sessionFile,
|
|
553
|
+
nativeSessionId: session.sessionId,
|
|
554
|
+
agentType: task.agentType,
|
|
555
|
+
persona: task.persona,
|
|
556
|
+
capability: task.capability,
|
|
557
|
+
isolation: task.isolation,
|
|
558
|
+
worktree: task.worktree,
|
|
559
|
+
concurrencyGroup: task.concurrencyGroup,
|
|
560
|
+
resumedFrom: task.resumeSessionFile,
|
|
561
|
+
};
|
|
562
|
+
};
|
|
563
|
+
|
|
564
|
+
const emitUsage = () => {
|
|
565
|
+
const usage = session.getContextUsage();
|
|
566
|
+
emit({
|
|
567
|
+
_tag: "UsageChanged",
|
|
568
|
+
tokens: usage?.tokens ?? undefined,
|
|
569
|
+
contextWindow: activeModel()?.contextWindow ?? usage?.contextWindow,
|
|
570
|
+
});
|
|
571
|
+
};
|
|
572
|
+
|
|
573
|
+
const settle = () => {
|
|
574
|
+
if (state.settled) return;
|
|
575
|
+
state.settled = true;
|
|
576
|
+
const last = lastAssistantMessage(session);
|
|
577
|
+
const partialText = finalOutput(session) || undefined;
|
|
578
|
+
if (last?.stopReason === "aborted") {
|
|
579
|
+
emit({
|
|
580
|
+
_tag: "RunSettled",
|
|
581
|
+
outcome: { _tag: "Interrupted", partialText },
|
|
582
|
+
});
|
|
583
|
+
return;
|
|
584
|
+
}
|
|
585
|
+
const errorText =
|
|
586
|
+
state.runError ??
|
|
587
|
+
(last?.stopReason === "error"
|
|
588
|
+
? (last.errorMessage ?? "Run failed")
|
|
589
|
+
: undefined);
|
|
590
|
+
if (errorText !== undefined) {
|
|
591
|
+
emit({
|
|
592
|
+
_tag: "RunSettled",
|
|
593
|
+
outcome: {
|
|
594
|
+
_tag: "Failed",
|
|
595
|
+
errorText: boundedError(errorText),
|
|
596
|
+
partialText,
|
|
597
|
+
},
|
|
598
|
+
});
|
|
599
|
+
return;
|
|
600
|
+
}
|
|
601
|
+
emit({
|
|
602
|
+
_tag: "RunSettled",
|
|
603
|
+
outcome: { _tag: "Completed", finalText: finalOutput(session) },
|
|
604
|
+
});
|
|
605
|
+
};
|
|
606
|
+
|
|
607
|
+
const handleEvent = (event: AgentSessionEvent) => {
|
|
608
|
+
if (state.closed) return;
|
|
609
|
+
switch (event.type) {
|
|
610
|
+
case "agent_start":
|
|
611
|
+
// Extensions may register tools between runs; guard new ones too.
|
|
612
|
+
toolTimeout.apply(session);
|
|
613
|
+
state.settled = false;
|
|
614
|
+
emit({ _tag: "RunStarted" });
|
|
615
|
+
break;
|
|
616
|
+
case "message_update": {
|
|
617
|
+
const streamEvent = event.assistantMessageEvent;
|
|
618
|
+
if (streamEvent.type === "text_delta") {
|
|
619
|
+
emit({
|
|
620
|
+
_tag: "AssistantDelta",
|
|
621
|
+
kind: "text",
|
|
622
|
+
delta: streamEvent.delta,
|
|
623
|
+
});
|
|
624
|
+
} else if (streamEvent.type === "thinking_delta") {
|
|
625
|
+
emit({
|
|
626
|
+
_tag: "AssistantDelta",
|
|
627
|
+
kind: "thinking",
|
|
628
|
+
delta: streamEvent.delta,
|
|
629
|
+
});
|
|
630
|
+
}
|
|
631
|
+
break;
|
|
632
|
+
}
|
|
633
|
+
case "message_end": {
|
|
634
|
+
const role = messageRole(event.message);
|
|
635
|
+
if (role === "user") {
|
|
636
|
+
const text = userText(event.message as Message);
|
|
637
|
+
if (text.trim()) emit({ _tag: "UserMessage", text });
|
|
638
|
+
} else if (role === "assistant") {
|
|
639
|
+
emit({
|
|
640
|
+
_tag: "AssistantMessage",
|
|
641
|
+
parts: assistantParts(event.message as AssistantMessage),
|
|
642
|
+
});
|
|
643
|
+
emitUsage();
|
|
644
|
+
emit({ _tag: "MetaChanged", meta: currentMeta() });
|
|
645
|
+
}
|
|
646
|
+
// toolResult messages are covered by tool_execution_end.
|
|
647
|
+
break;
|
|
648
|
+
}
|
|
649
|
+
case "tool_execution_start":
|
|
650
|
+
emit({
|
|
651
|
+
_tag: "ToolStart",
|
|
652
|
+
toolId: event.toolCallId,
|
|
653
|
+
name: event.toolName,
|
|
654
|
+
argsPreview: safeJson(event.args),
|
|
655
|
+
});
|
|
656
|
+
break;
|
|
657
|
+
case "tool_execution_update":
|
|
658
|
+
emit({
|
|
659
|
+
_tag: "ToolUpdate",
|
|
660
|
+
toolId: event.toolCallId,
|
|
661
|
+
outputPreview: toolPreview(event.partialResult),
|
|
662
|
+
});
|
|
663
|
+
break;
|
|
664
|
+
case "tool_execution_end":
|
|
665
|
+
emit({
|
|
666
|
+
_tag: "ToolEnd",
|
|
667
|
+
toolId: event.toolCallId,
|
|
668
|
+
name: event.toolName,
|
|
669
|
+
isError: event.isError,
|
|
670
|
+
outputPreview: toolPreview(event.result),
|
|
671
|
+
});
|
|
672
|
+
break;
|
|
673
|
+
case "queue_update":
|
|
674
|
+
emit({
|
|
675
|
+
_tag: "QueueChanged",
|
|
676
|
+
queued: [
|
|
677
|
+
...event.steering.map((text) => ({
|
|
678
|
+
text,
|
|
679
|
+
kind: "steer" as const,
|
|
680
|
+
})),
|
|
681
|
+
...event.followUp.map((text) => ({
|
|
682
|
+
text,
|
|
683
|
+
kind: "follow-up" as const,
|
|
684
|
+
})),
|
|
685
|
+
],
|
|
686
|
+
});
|
|
687
|
+
break;
|
|
688
|
+
case "agent_settled":
|
|
689
|
+
settle();
|
|
690
|
+
break;
|
|
691
|
+
}
|
|
692
|
+
};
|
|
693
|
+
const unsubscribe = session.subscribe(handleEvent);
|
|
694
|
+
|
|
695
|
+
yield* Effect.addFinalizer(() =>
|
|
696
|
+
Effect.promise(async () => {
|
|
697
|
+
state.closed = true;
|
|
698
|
+
unsubscribe();
|
|
699
|
+
try {
|
|
700
|
+
session.clearQueue();
|
|
701
|
+
} catch {
|
|
702
|
+
// Continue with abort/dispose.
|
|
703
|
+
}
|
|
704
|
+
await waitBounded(session.abort(), CHILD_SHUTDOWN_TIMEOUT_MS);
|
|
705
|
+
await shutdownAndDisposeChildSession(session);
|
|
706
|
+
Queue.endUnsafe(events);
|
|
707
|
+
}),
|
|
708
|
+
);
|
|
709
|
+
|
|
710
|
+
/** Start a fresh run (v1 manager.run): fire-and-forget, errors -> events. */
|
|
711
|
+
const startRun = (text: string) => {
|
|
712
|
+
state.runError = undefined;
|
|
713
|
+
state.settled = false;
|
|
714
|
+
emit({ _tag: "RunStarted" });
|
|
715
|
+
void session.prompt(text).catch((error) => {
|
|
716
|
+
state.runError = boundedError(error);
|
|
717
|
+
// Preflight failures may never start the agent lifecycle, so no
|
|
718
|
+
// agent_settled will arrive for them.
|
|
719
|
+
if (!session.isStreaming) settle();
|
|
720
|
+
});
|
|
721
|
+
};
|
|
722
|
+
|
|
723
|
+
// Session naming is best-effort.
|
|
724
|
+
yield* Effect.try(() =>
|
|
725
|
+
session.sessionManager.appendSessionInfo(`subagent: ${task.title}`),
|
|
726
|
+
).pipe(Effect.ignore);
|
|
727
|
+
|
|
728
|
+
emit({ _tag: "MetaChanged", meta: currentMeta() });
|
|
729
|
+
startRun(task.prompt);
|
|
730
|
+
|
|
731
|
+
return {
|
|
732
|
+
meta: Effect.sync(currentMeta),
|
|
733
|
+
events: Stream.fromQueue(events),
|
|
734
|
+
send: (text) =>
|
|
735
|
+
Effect.suspend((): Effect.Effect<void, SendError> => {
|
|
736
|
+
if (state.closed) {
|
|
737
|
+
return new SendError({ message: "Subagent session is closed." });
|
|
738
|
+
}
|
|
739
|
+
if (session.isStreaming) {
|
|
740
|
+
// Steer the active run via the SDK's queue; queue_update events
|
|
741
|
+
// render it, message_end(user) lands it in the transcript. A
|
|
742
|
+
// rejected steer is a real send failure, not a diagnostic.
|
|
743
|
+
return Effect.tryPromise({
|
|
744
|
+
try: () => session.steer(text),
|
|
745
|
+
catch: (error) => new SendError({ message: boundedError(error) }),
|
|
746
|
+
}).pipe(Effect.asVoid);
|
|
747
|
+
}
|
|
748
|
+
return Effect.sync(() => startRun(text));
|
|
749
|
+
}),
|
|
750
|
+
interrupt: Effect.promise(async () => {
|
|
751
|
+
if (state.closed) return;
|
|
752
|
+
try {
|
|
753
|
+
session.clearQueue();
|
|
754
|
+
} catch {
|
|
755
|
+
// Abort regardless.
|
|
756
|
+
}
|
|
757
|
+
await session.abort().catch(() => undefined);
|
|
758
|
+
// Only resolve once streaming has actually stopped: reporting the
|
|
759
|
+
// interrupt as complete while the run keeps working would let the
|
|
760
|
+
// manager settle a run that is still mutating the workspace. The
|
|
761
|
+
// manager bounds this effect at 5s and force-disposes on timeout.
|
|
762
|
+
while (!state.closed && session.isStreaming) {
|
|
763
|
+
await new Promise((resolve) => setTimeout(resolve, 50));
|
|
764
|
+
}
|
|
765
|
+
// No streaming run means no agent_settled will arrive; emit the
|
|
766
|
+
// terminal event (once) so the run cannot look running forever.
|
|
767
|
+
if (!state.closed && !state.settled) {
|
|
768
|
+
state.settled = true;
|
|
769
|
+
emit({ _tag: "RunSettled", outcome: { _tag: "Interrupted" } });
|
|
770
|
+
}
|
|
771
|
+
}),
|
|
772
|
+
} satisfies SubagentSession;
|
|
773
|
+
});
|
|
774
|
+
|
|
775
|
+
export const piBackend: SubagentBackend = {
|
|
776
|
+
name: "pi",
|
|
777
|
+
capabilities: { steering: true, modelSelection: true, reasoningEffort: true },
|
|
778
|
+
// In-process SDK: always available.
|
|
779
|
+
available: Effect.succeed(true),
|
|
780
|
+
spawn: makePiSession,
|
|
781
|
+
};
|