@gotgenes/pi-subagents 6.18.8 → 6.19.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/CHANGELOG.md +16 -0
- package/docs/architecture/architecture.md +8 -4
- package/docs/plans/0172-extract-turn-formatting.md +206 -0
- package/docs/retro/0171-reduce-render-result-complexity.md +34 -0
- package/docs/retro/0172-extract-turn-formatting.md +40 -0
- package/package.json +1 -1
- package/src/lifecycle/agent-runner.ts +17 -39
- package/src/session/content-items.ts +53 -0
- package/src/ui/message-formatters.ts +5 -21
package/CHANGELOG.md
CHANGED
|
@@ -5,6 +5,22 @@ All notable changes to this project will be documented in this file.
|
|
|
5
5
|
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
|
|
6
6
|
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
|
|
7
7
|
|
|
8
|
+
## [6.19.0](https://github.com/gotgenes/pi-packages/compare/pi-subagents-v6.18.8...pi-subagents-v6.19.0) (2026-05-24)
|
|
9
|
+
|
|
10
|
+
|
|
11
|
+
### Features
|
|
12
|
+
|
|
13
|
+
* extract shared content-item parsing into session/content-items ([5ed0d1c](https://github.com/gotgenes/pi-packages/commit/5ed0d1c6291d9044e1ab85c637b1e5f0051789f3))
|
|
14
|
+
* extract shared content-item parsing into session/content-items ([413fda0](https://github.com/gotgenes/pi-packages/commit/413fda0cc8bb496a79285d0ec97c97d9b0b6cc6d))
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
### Documentation
|
|
18
|
+
|
|
19
|
+
* mark step 9 (extract turn-formatting) done in architecture ([04a0b55](https://github.com/gotgenes/pi-packages/commit/04a0b554b8848adc0f43b8939fa866086282a6af))
|
|
20
|
+
* plan extract shared turn-formatting logic ([#172](https://github.com/gotgenes/pi-packages/issues/172)) ([818affe](https://github.com/gotgenes/pi-packages/commit/818affe22457cfbc1cabc5d4e7477e9391b3ed46))
|
|
21
|
+
* **retro:** add planning stage notes for issue [#172](https://github.com/gotgenes/pi-packages/issues/172) ([809b4cf](https://github.com/gotgenes/pi-packages/commit/809b4cf4dd9f59f1c57eb0776835af88e0cef8f4))
|
|
22
|
+
* **retro:** add retro notes for issue [#171](https://github.com/gotgenes/pi-packages/issues/171) ([2b50b37](https://github.com/gotgenes/pi-packages/commit/2b50b374f9d99305144bc6227eb48e3a9d68efb3))
|
|
23
|
+
|
|
8
24
|
## [6.18.8](https://github.com/gotgenes/pi-packages/compare/pi-subagents-v6.18.7...pi-subagents-v6.18.8) (2026-05-24)
|
|
9
25
|
|
|
10
26
|
|
|
@@ -244,6 +244,7 @@ src/
|
|
|
244
244
|
├── session/ session assembly and preparation
|
|
245
245
|
│ ├── session-config.ts pure assembler (main entry)
|
|
246
246
|
│ ├── prompts.ts system prompt building
|
|
247
|
+
│ ├── content-items.ts shared message content parsing (tool-call names, assistant content)
|
|
247
248
|
│ ├── context.ts parent conversation extraction
|
|
248
249
|
│ ├── memory.ts persistent MEMORY.md per agent
|
|
249
250
|
│ ├── skill-loader.ts skill preloading
|
|
@@ -484,8 +485,9 @@ Files with highest commit frequency × complexity (accelerating trend):
|
|
|
484
485
|
|
|
485
486
|
### Production duplication
|
|
486
487
|
|
|
487
|
-
|
|
488
|
-
|
|
488
|
+
The 18-line clone group between `agent-runner.ts` and `message-formatters.ts` was resolved in #172.
|
|
489
|
+
`ToolCallContent`, `getToolCallName`, and `extractAssistantContent` now live in `session/content-items.ts`.
|
|
490
|
+
No known production duplication remains.
|
|
489
491
|
|
|
490
492
|
### Proposed bag decompositions
|
|
491
493
|
|
|
@@ -648,9 +650,11 @@ Extracted per-status result formatting from `renderResult` in `agent-tool.ts` in
|
|
|
648
650
|
`renderResult` reduced from ~80 lines (cognitive complexity 43) to a 10-line guard + `renderAgentResult` dispatcher.
|
|
649
651
|
The inline `stats()` closure became the exported `renderStats` helper, shared by all status renderers.
|
|
650
652
|
|
|
651
|
-
### Step 9: Extract shared turn-formatting logic ([#172][172])
|
|
653
|
+
### Step 9: Extract shared turn-formatting logic ([#172][172]) ✓ Done
|
|
652
654
|
|
|
653
|
-
|
|
655
|
+
Extracted `ToolCallContent`, `getToolCallName`, and `extractAssistantContent` into `session/content-items.ts`.
|
|
656
|
+
Both `lifecycle/agent-runner.ts` (`getAgentConversation`) and `ui/message-formatters.ts` (`formatAssistantMessage`) now import from the shared module.
|
|
657
|
+
Eliminates the 18-line production-duplication finding.
|
|
654
658
|
|
|
655
659
|
### Step dependencies
|
|
656
660
|
|
|
@@ -0,0 +1,206 @@
|
|
|
1
|
+
---
|
|
2
|
+
issue: 172
|
|
3
|
+
issue_title: "refactor(pi-subagents): extract shared turn-formatting logic"
|
|
4
|
+
---
|
|
5
|
+
|
|
6
|
+
# Extract shared turn-formatting logic
|
|
7
|
+
|
|
8
|
+
## Problem Statement
|
|
9
|
+
|
|
10
|
+
Fallow identified 18 lines of duplicated production code between `lifecycle/agent-runner.ts` and `ui/message-formatters.ts` (originally `conversation-viewer.ts` before #170 extracted formatters).
|
|
11
|
+
Both sites iterate over assistant message content items, extracting tool names and text parts for display.
|
|
12
|
+
The `ToolCallContent` interface and `getToolCallName` helper are duplicated verbatim in both files.
|
|
13
|
+
|
|
14
|
+
## Goals
|
|
15
|
+
|
|
16
|
+
- Extract the duplicated `ToolCallContent` type and `getToolCallName` function into a single shared module.
|
|
17
|
+
- Extract the content-iteration pattern (collecting text parts and tool names from assistant message content) into a reusable `extractAssistantContent` function.
|
|
18
|
+
- Both consumers (`getAgentConversation` in `agent-runner.ts` and `formatAssistantMessage` in `message-formatters.ts`) import from the shared module.
|
|
19
|
+
- Eliminate the fallow production-duplication finding.
|
|
20
|
+
|
|
21
|
+
## Non-Goals
|
|
22
|
+
|
|
23
|
+
- Moving `extractText` from `session/context.ts` to the new module — same concern (content parsing) but out of scope; note as a follow-up.
|
|
24
|
+
- Refactoring `getAgentConversation` itself (it has no tests and its full-loop structure mixes user/assistant/toolResult formatting) — separate concern.
|
|
25
|
+
- Changing `buildParentContext` in `session/context.ts`, which has a similar but simpler iteration pattern (no tool calls, different data source).
|
|
26
|
+
|
|
27
|
+
## Background
|
|
28
|
+
|
|
29
|
+
### Current duplication sites
|
|
30
|
+
|
|
31
|
+
`lifecycle/agent-runner.ts` (private scope):
|
|
32
|
+
|
|
33
|
+
```typescript
|
|
34
|
+
interface ToolCallContent {
|
|
35
|
+
type: "toolCall";
|
|
36
|
+
name?: string;
|
|
37
|
+
toolName?: string;
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
function getToolCallName(c: { type: string }): string {
|
|
41
|
+
if (c.type !== "toolCall") return "unknown";
|
|
42
|
+
const tc = c as ToolCallContent;
|
|
43
|
+
return tc.name ?? tc.toolName ?? "unknown";
|
|
44
|
+
}
|
|
45
|
+
```
|
|
46
|
+
|
|
47
|
+
`ui/message-formatters.ts` (exported):
|
|
48
|
+
|
|
49
|
+
```typescript
|
|
50
|
+
interface ToolCallContent { /* identical */ }
|
|
51
|
+
export function getToolCallName(c: { type: string }): string { /* identical */ }
|
|
52
|
+
```
|
|
53
|
+
|
|
54
|
+
The content-iteration pattern appears in:
|
|
55
|
+
|
|
56
|
+
1. `getAgentConversation` (agent-runner.ts lines 480–486) — plain-text output for LLM consumption
|
|
57
|
+
2. `formatAssistantMessage` (message-formatters.ts lines 144–148) — themed display lines for TUI
|
|
58
|
+
|
|
59
|
+
Both collect `textParts: string[]` and tool names via the same `for (const c of content)` loop with identical guards.
|
|
60
|
+
|
|
61
|
+
### Structural analysis
|
|
62
|
+
|
|
63
|
+
Per the code-design skill's "structural reasons before extracting duplication" check: the two consumers differ only in *presentation* (plain text vs. themed TUI lines).
|
|
64
|
+
The *data extraction* — identifying text items and tool-call items, extracting tool names — is the same logical operation.
|
|
65
|
+
This is incidental duplication suitable for extraction.
|
|
66
|
+
|
|
67
|
+
### Dependencies
|
|
68
|
+
|
|
69
|
+
- Issue #164 (domain directory reorganization): ✓ closed — files are already in `lifecycle/` and `ui/`.
|
|
70
|
+
- Issue #170 (buildContentLines complexity reduction): ✓ closed — formatting logic is already in `message-formatters.ts`.
|
|
71
|
+
- Issue #170 is related: the extraction may simplify `formatAssistantMessage` slightly (the issue body predicted this).
|
|
72
|
+
|
|
73
|
+
### Placement
|
|
74
|
+
|
|
75
|
+
The architecture doc (Phase 10, Step 9) says: "extracts into a shared function in the session domain."
|
|
76
|
+
The `session/` directory already hosts `context.ts` which exports the related `extractText` function.
|
|
77
|
+
A new `session/content-items.ts` module keeps the concern focused without overloading `context.ts`.
|
|
78
|
+
|
|
79
|
+
## Design Overview
|
|
80
|
+
|
|
81
|
+
### New module: `session/content-items.ts`
|
|
82
|
+
|
|
83
|
+
```typescript
|
|
84
|
+
/** Tool-call content item — SDK exposes this at runtime but doesn't export the type. */
|
|
85
|
+
export interface ToolCallContent {
|
|
86
|
+
type: "toolCall";
|
|
87
|
+
name?: string;
|
|
88
|
+
toolName?: string;
|
|
89
|
+
}
|
|
90
|
+
|
|
91
|
+
/** Extracts the display name from a tool-call content item. */
|
|
92
|
+
export function getToolCallName(c: { type: string }): string {
|
|
93
|
+
if (c.type !== "toolCall") return "unknown";
|
|
94
|
+
const tc = c as ToolCallContent;
|
|
95
|
+
return tc.name ?? tc.toolName ?? "unknown";
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
/** Extracted text parts and tool names from assistant message content. */
|
|
99
|
+
export interface AssistantContentParts {
|
|
100
|
+
textParts: string[];
|
|
101
|
+
toolNames: string[];
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
/**
|
|
105
|
+
* Extract text and tool-call names from assistant message content items.
|
|
106
|
+
* Pure data extraction — consumers apply their own formatting.
|
|
107
|
+
*/
|
|
108
|
+
export function extractAssistantContent(
|
|
109
|
+
content: { type: string; [key: string]: unknown }[],
|
|
110
|
+
): AssistantContentParts {
|
|
111
|
+
const textParts: string[] = [];
|
|
112
|
+
const toolNames: string[] = [];
|
|
113
|
+
for (const c of content) {
|
|
114
|
+
if (c.type === "text" && c.text) textParts.push(c.text as string);
|
|
115
|
+
else if (c.type === "toolCall") toolNames.push(getToolCallName(c));
|
|
116
|
+
}
|
|
117
|
+
return { textParts, toolNames };
|
|
118
|
+
}
|
|
119
|
+
```
|
|
120
|
+
|
|
121
|
+
### Consumer call sites (pseudocode)
|
|
122
|
+
|
|
123
|
+
`getAgentConversation` (agent-runner.ts):
|
|
124
|
+
|
|
125
|
+
```typescript
|
|
126
|
+
const { textParts, toolNames } = extractAssistantContent(msg.content);
|
|
127
|
+
if (textParts.length > 0) parts.push(`[Assistant]: ${textParts.join("\n")}`);
|
|
128
|
+
if (toolNames.length > 0) parts.push(`[Tool Calls]:\n${toolNames.map(n => ` Tool: ${n}`).join("\n")}`);
|
|
129
|
+
```
|
|
130
|
+
|
|
131
|
+
`formatAssistantMessage` (message-formatters.ts):
|
|
132
|
+
|
|
133
|
+
```typescript
|
|
134
|
+
const { textParts, toolNames } = extractAssistantContent(content);
|
|
135
|
+
const lines: string[] = [theme.bold("[Assistant]")];
|
|
136
|
+
if (textParts.length > 0) lines.push(...wrapText(textParts.join("\n").trim(), width));
|
|
137
|
+
for (const name of toolNames) lines.push(truncateToWidth(theme.fg("muted", ` [Tool: ${name}]`), width));
|
|
138
|
+
```
|
|
139
|
+
|
|
140
|
+
Both consumers call the same extraction, then apply their own presentation.
|
|
141
|
+
This follows Tell-Don't-Ask: the shared function returns structured data, not formatted strings.
|
|
142
|
+
|
|
143
|
+
## Module-Level Changes
|
|
144
|
+
|
|
145
|
+
### New files
|
|
146
|
+
|
|
147
|
+
| File | Description |
|
|
148
|
+
| ------------------------------------ | ---------------------------------------------------------------------------------------- |
|
|
149
|
+
| `src/session/content-items.ts` | `ToolCallContent`, `getToolCallName`, `AssistantContentParts`, `extractAssistantContent` |
|
|
150
|
+
| `test/session/content-items.test.ts` | Unit tests for `getToolCallName` and `extractAssistantContent` |
|
|
151
|
+
|
|
152
|
+
### Changed files
|
|
153
|
+
|
|
154
|
+
| File | Change |
|
|
155
|
+
| ----------------------------------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
|
|
156
|
+
| `src/lifecycle/agent-runner.ts` | Remove `ToolCallContent` and `getToolCallName`; import `extractAssistantContent` from `session/content-items`; refactor `getAgentConversation` assistant branch |
|
|
157
|
+
| `src/ui/message-formatters.ts` | Remove `ToolCallContent` and `getToolCallName`; import `extractAssistantContent` and `getToolCallName` from `session/content-items`; refactor `formatAssistantMessage` |
|
|
158
|
+
| `docs/architecture/architecture.md` | Update Step 9 status to "✓ Done"; update `session/` module listing to include `content-items.ts`; update production duplication section |
|
|
159
|
+
|
|
160
|
+
### Architecture doc updates
|
|
161
|
+
|
|
162
|
+
The architecture doc references this issue in three places:
|
|
163
|
+
|
|
164
|
+
1. Line 487 — "Production duplication" subsection: update to note the duplication is resolved.
|
|
165
|
+
2. Line 247 — `session/` module listing: add `content-items.ts`.
|
|
166
|
+
3. Line 653 — Phase 10, Step 9: mark "✓ Done".
|
|
167
|
+
|
|
168
|
+
## Test Impact Analysis
|
|
169
|
+
|
|
170
|
+
1. **New unit tests enabled**: `getToolCallName` and `extractAssistantContent` get direct unit tests for the first time.
|
|
171
|
+
Previously, `getToolCallName` was only exercised indirectly through `formatAssistantMessage` tests.
|
|
172
|
+
2. **Existing tests that stay as-is**: `message-formatters.test.ts` tests for `formatAssistantMessage` continue to exercise the full pipeline (extraction + formatting).
|
|
173
|
+
They become integration-level tests relative to the new extraction layer — they should not be simplified or removed.
|
|
174
|
+
3. **No existing tests become redundant**: `getAgentConversation` has no tests today, so nothing to deduplicate.
|
|
175
|
+
|
|
176
|
+
## TDD Order
|
|
177
|
+
|
|
178
|
+
1. `test:` Write tests for `getToolCallName` in `test/session/content-items.test.ts` — covers `toolCall` with `name`, `toolName`, both (prefers `name`), neither (returns "unknown"), and non-toolCall type.
|
|
179
|
+
Commit: `test: add getToolCallName unit tests`
|
|
180
|
+
2. `test:` Write tests for `extractAssistantContent` in the same file — covers empty array, text-only items, toolCall-only items, mixed items, and items with other types (e.g., `image`).
|
|
181
|
+
Commit: `test: add extractAssistantContent unit tests`
|
|
182
|
+
3. `feat:` Create `src/session/content-items.ts` with `ToolCallContent`, `getToolCallName`, `AssistantContentParts`, and `extractAssistantContent`.
|
|
183
|
+
All tests go green.
|
|
184
|
+
Commit: `feat: extract shared content-item parsing into session/content-items`
|
|
185
|
+
4. `refactor:` Update `message-formatters.ts` — remove local `ToolCallContent` and `getToolCallName`; import `getToolCallName` and `extractAssistantContent` from `session/content-items`; refactor `formatAssistantMessage` to use `extractAssistantContent`.
|
|
186
|
+
Existing `message-formatters.test.ts` stays green.
|
|
187
|
+
Commit: `refactor: use shared content-items in message-formatters`
|
|
188
|
+
5. `refactor:` Update `agent-runner.ts` — remove local `ToolCallContent` and `getToolCallName`; import `extractAssistantContent` from `session/content-items`; refactor `getAgentConversation` assistant branch.
|
|
189
|
+
Existing `agent-runner.test.ts` stays green.
|
|
190
|
+
Commit: `refactor: use shared content-items in agent-runner`
|
|
191
|
+
6. `docs:` Update `docs/architecture/architecture.md` — mark Step 9 done, update module listing, update duplication section.
|
|
192
|
+
Commit: `docs: mark step 9 (extract turn-formatting) done in architecture`
|
|
193
|
+
|
|
194
|
+
## Risks and Mitigations
|
|
195
|
+
|
|
196
|
+
| Risk | Mitigation |
|
|
197
|
+
| ----------------------------------------------------------------------------------------------------------------- | -------------------------------------------------------------------------------------------------------------- |
|
|
198
|
+
| `getToolCallName` signature change breaking callers | Signature is identical — no change needed. Import path changes are the only difference. |
|
|
199
|
+
| `extractAssistantContent` return shape mismatch with consumers | Consumer pseudocode verified above; both sites restructure trivially around `{ textParts, toolNames }`. |
|
|
200
|
+
| `message-formatters.ts` re-exports `getToolCallName` — removing the local definition could break external imports | Grep confirms no external consumers import `getToolCallName` from `message-formatters.ts`. Drop the re-export. |
|
|
201
|
+
| `isBashExecution` in `message-formatters.ts` also uses a local type — could be confused with this extraction | `isBashExecution` and `BashExecutionMessage` are unrelated to the tool-call duplication; leave them in place. |
|
|
202
|
+
|
|
203
|
+
## Open Questions
|
|
204
|
+
|
|
205
|
+
- Should `extractText` (currently in `session/context.ts`) move to `session/content-items.ts` for consistency?
|
|
206
|
+
Deferred — it works fine where it is, and moving it means updating all importers for no functional benefit.
|
|
@@ -41,3 +41,37 @@ A docs commit updated the architecture file to remove `renderResult` from the co
|
|
|
41
41
|
## Stage: User Note (2026-05-24T21:30:00Z)
|
|
42
42
|
|
|
43
43
|
We need to update `/plan-issue` so the plan includes updating any architecture documents.
|
|
44
|
+
|
|
45
|
+
## Stage: Final Retrospective (2026-05-24T22:00:00Z)
|
|
46
|
+
|
|
47
|
+
### Session summary
|
|
48
|
+
|
|
49
|
+
Issue #171 shipped cleanly across three sessions: planning produced a 9-step TDD plan, implementation added 43 tests and extracted `renderResult` into `tools/result-renderer.ts` (cognitive complexity 43 → dispatcher), and shipping released `pi-subagents-v6.18.8` with no deviations.
|
|
50
|
+
The only friction was an incomplete architecture doc update that required a second manual pass.
|
|
51
|
+
|
|
52
|
+
### Observations
|
|
53
|
+
|
|
54
|
+
#### What went well
|
|
55
|
+
|
|
56
|
+
- The plan-to-implementation pipeline worked smoothly for a mechanical extraction refactor — zero deviations from the plan across all 9 TDD steps.
|
|
57
|
+
- The `Theme` type from `display.ts` and the `widget-renderer.ts` pattern provided a proven template, making the extraction straightforward.
|
|
58
|
+
- ESLint pre-commit hooks caught a stale `eslint-disable` rule (`no-unsafe-return`) automatically during the refactor commit.
|
|
59
|
+
|
|
60
|
+
#### What caused friction (agent side)
|
|
61
|
+
|
|
62
|
+
- `missing-context` — The TDD session's initial architecture update (commit `1183522`) only updated the complexity hotspot table and layout listing, missing health metrics, the Mermaid domain diagram, the Phase 10 structural refactoring table, and the SKILL.md domain table.
|
|
63
|
+
The user had to explicitly request the fuller update, which landed as a second commit (`1510dc7`).
|
|
64
|
+
Impact: extra user round-trip and a second commit for what should have been one complete update.
|
|
65
|
+
- `missing-context` — The `git add ../../.pi/skills/...` path triggered a `pi-permission-system` permission denial because the relative path escaped the package directory.
|
|
66
|
+
Switched to repo-root-relative path to resolve.
|
|
67
|
+
Impact: minor — one failed commit attempt, immediately retried.
|
|
68
|
+
|
|
69
|
+
#### What caused friction (user side)
|
|
70
|
+
|
|
71
|
+
- The `/plan-issue` prompt did not instruct the planner to check architecture docs for sections affected by the change.
|
|
72
|
+
Because the plan's Module-Level Changes omitted `docs/architecture/architecture.md`, the TDD session treated architecture updates as an afterthought rather than a planned deliverable.
|
|
73
|
+
Impact: the user had to intervene with an explicit request and a retro note.
|
|
74
|
+
|
|
75
|
+
### Changes made
|
|
76
|
+
|
|
77
|
+
1. `.pi/prompts/plan-issue.md` — Added architecture-doc check to the Module-Level Changes bullet: "When the change adds, removes, or moves a module, check `packages/<PKG>/docs/architecture/` for layout listings, complexity tables, health metrics, or domain diagrams that reference the affected files and list them as doc updates."
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
---
|
|
2
|
+
issue: 172
|
|
3
|
+
issue_title: "refactor(pi-subagents): extract shared turn-formatting logic"
|
|
4
|
+
---
|
|
5
|
+
|
|
6
|
+
# Retro: #172 — Extract shared turn-formatting logic
|
|
7
|
+
|
|
8
|
+
## Stage: Planning (2026-05-24T18:00:00Z)
|
|
9
|
+
|
|
10
|
+
### Session summary
|
|
11
|
+
|
|
12
|
+
Planned the extraction of duplicated turn-formatting logic from `lifecycle/agent-runner.ts` and `ui/message-formatters.ts` into a new shared module `session/content-items.ts`.
|
|
13
|
+
The plan covers extracting `ToolCallContent`, `getToolCallName`, and a new `extractAssistantContent` function, with a 6-step TDD order.
|
|
14
|
+
|
|
15
|
+
### Observations
|
|
16
|
+
|
|
17
|
+
- Issue #170 (completed) shifted the duplication target from `conversation-viewer.ts` to `message-formatters.ts` — the issue body's line references are stale but the duplication still exists in the same form.
|
|
18
|
+
- Both dependencies (#164 and #170) are closed, so this is unblocked.
|
|
19
|
+
- The duplication is clearly incidental (same data extraction, different presentation) — safe to extract per the code-design skill's structural-reasons check.
|
|
20
|
+
- `getToolCallName` has no direct unit tests today; the extraction enables testing it for the first time.
|
|
21
|
+
- `getAgentConversation` also has no tests — noted as out of scope but worth a follow-up.
|
|
22
|
+
- Considered adding `extractText` to the new module for consistency but deferred to keep scope tight.
|
|
23
|
+
|
|
24
|
+
## Stage: Implementation — TDD (2026-05-24T19:05:00Z)
|
|
25
|
+
|
|
26
|
+
### Session summary
|
|
27
|
+
|
|
28
|
+
Completed all 6 TDD steps from the plan.
|
|
29
|
+
Created `session/content-items.ts` with `getToolCallName` and `extractAssistantContent`, added 11 unit tests, then refactored both `message-formatters.ts` and `agent-runner.ts` to use the shared module.
|
|
30
|
+
Test count went from 896 to 907 (+11).
|
|
31
|
+
|
|
32
|
+
### Observations
|
|
33
|
+
|
|
34
|
+
- Steps 1 and 2 (test-only commits) were folded into step 3's feat commit per the plan's intent — all three land together.
|
|
35
|
+
- The `getToolCallName` parameter type needed widening from `{ type: string }` to `{ type: string; [key: string]: unknown }` to allow test object literals to pass excess-property checking.
|
|
36
|
+
This in turn required an `as unknown as` double cast at the `agent-runner.ts` call site, because the SDK's `TextContent | ThinkingContent | ToolCall` union lacks an index signature.
|
|
37
|
+
Same pattern already present in `conversation-viewer.ts`.
|
|
38
|
+
- `message-formatters.ts` had both an import and a re-export of `getToolCallName`; simplified to a pure re-export only.
|
|
39
|
+
- The lint fixup (unused import) was amended into the same refactor commit before pushing.
|
|
40
|
+
- Architecture doc updated: `content-items.ts` added to session module listing, production-duplication section updated, Step 9 marked Done.
|
package/package.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
/**
|
|
2
|
-
* agent-runner.ts
|
|
2
|
+
* agent-runner.ts - Core execution engine: creates sessions, runs agents, collects results.
|
|
3
3
|
*/
|
|
4
4
|
|
|
5
5
|
import type { Model } from "@earendil-works/pi-ai";
|
|
@@ -11,6 +11,7 @@ import {
|
|
|
11
11
|
import type { AgentConfigLookup } from "#src/config/agent-types";
|
|
12
12
|
import type { ParentSessionInfo } from "#src/lifecycle/agent-manager";
|
|
13
13
|
import type { ParentSnapshot } from "#src/lifecycle/parent-snapshot";
|
|
14
|
+
import { extractAssistantContent } from "#src/session/content-items";
|
|
14
15
|
import { extractText } from "#src/session/context";
|
|
15
16
|
import type { EnvInfo } from "#src/session/env";
|
|
16
17
|
import { type AssemblerIO, assembleSessionConfig, type ToolFilterConfig } from "#src/session/session-config";
|
|
@@ -19,27 +20,10 @@ import type { ShellExec, SubagentType, ThinkingLevel } from "#src/types";
|
|
|
19
20
|
/** Names of tools registered by this extension that subagents must NOT inherit. */
|
|
20
21
|
const EXCLUDED_TOOL_NAMES = ["Agent", "get_subagent_result", "steer_subagent"];
|
|
21
22
|
|
|
22
|
-
// ── Local message-shape types ───────────────────────────────────────────────
|
|
23
|
-
// The Pi SDK does not export a narrow type for tool-call content variants.
|
|
24
|
-
|
|
25
|
-
/** Tool-call content item — SDK exposes this variant at runtime but doesn’t export the narrow type. */
|
|
26
|
-
interface ToolCallContent {
|
|
27
|
-
type: "toolCall";
|
|
28
|
-
name?: string;
|
|
29
|
-
toolName?: string;
|
|
30
|
-
}
|
|
31
|
-
|
|
32
|
-
/** Extracts the display name from a tool-call content item. */
|
|
33
|
-
function getToolCallName(c: { type: string }): string {
|
|
34
|
-
if (c.type !== "toolCall") return "unknown";
|
|
35
|
-
const tc = c as ToolCallContent;
|
|
36
|
-
return tc.name ?? tc.toolName ?? "unknown";
|
|
37
|
-
}
|
|
38
|
-
|
|
39
23
|
/**
|
|
40
24
|
* Filter the session's active tool names according to extension/denylist rules.
|
|
41
25
|
*
|
|
42
|
-
* Run twice
|
|
26
|
+
* Run twice - once before `bindExtensions` (filters built-in tools) and once after
|
|
43
27
|
* (filters extension-registered tools, which only join the active set during
|
|
44
28
|
* `bindExtensions`). Extracting this keeps the two callsites consistent and makes
|
|
45
29
|
* the post-bind re-filter trivial.
|
|
@@ -116,7 +100,7 @@ export interface CreateSessionOptions {
|
|
|
116
100
|
}
|
|
117
101
|
|
|
118
102
|
/**
|
|
119
|
-
* Environment discovery
|
|
103
|
+
* Environment discovery - detect runtime context and resolve directories.
|
|
120
104
|
*
|
|
121
105
|
* Decouples the runner from direct process/SDK reads so each can be stubbed
|
|
122
106
|
* independently in tests.
|
|
@@ -128,7 +112,7 @@ export interface EnvironmentIO {
|
|
|
128
112
|
}
|
|
129
113
|
|
|
130
114
|
/**
|
|
131
|
-
* Session factory
|
|
115
|
+
* Session factory - create SDK objects for a child agent session.
|
|
132
116
|
*
|
|
133
117
|
* Decouples the runner from direct Pi SDK imports and sibling-module IO,
|
|
134
118
|
* making it testable via plain stub objects without vi.mock().
|
|
@@ -153,15 +137,15 @@ export type RunnerIO = EnvironmentIO & SessionFactoryIO;
|
|
|
153
137
|
// ── Public interfaces ─────────────────────────────────────────────────────────
|
|
154
138
|
|
|
155
139
|
/**
|
|
156
|
-
* Parent execution context
|
|
140
|
+
* Parent execution context - where/who is running.
|
|
157
141
|
*
|
|
158
142
|
* Groups the four fields that describe the parent environment and identity,
|
|
159
143
|
* separating them from the per-call execution parameters in RunOptions.
|
|
160
144
|
*/
|
|
161
145
|
export interface RunContext {
|
|
162
|
-
/** Shell-exec callback for detectEnv
|
|
146
|
+
/** Shell-exec callback for detectEnv - injected from pi.exec(). */
|
|
163
147
|
exec: ShellExec;
|
|
164
|
-
/** Agent config lookup
|
|
148
|
+
/** Agent config lookup - provides resolveAgentConfig and getToolNamesForType. */
|
|
165
149
|
registry: AgentConfigLookup;
|
|
166
150
|
/** Override working directory (e.g. for worktree isolation). */
|
|
167
151
|
cwd?: string;
|
|
@@ -170,14 +154,14 @@ export interface RunContext {
|
|
|
170
154
|
}
|
|
171
155
|
|
|
172
156
|
export interface RunOptions {
|
|
173
|
-
/** Parent execution context
|
|
157
|
+
/** Parent execution context - where/who is running. */
|
|
174
158
|
context: RunContext;
|
|
175
159
|
model?: Model<any>;
|
|
176
160
|
maxTurns?: number;
|
|
177
161
|
signal?: AbortSignal;
|
|
178
162
|
isolated?: boolean;
|
|
179
163
|
thinkingLevel?: ThinkingLevel;
|
|
180
|
-
/** Called once after session creation
|
|
164
|
+
/** Called once after session creation - session delivery mechanism. */
|
|
181
165
|
onSessionCreated?: (session: AgentSession) => void;
|
|
182
166
|
/**
|
|
183
167
|
* Default max turns from runtime config. Falls back to the module-scope
|
|
@@ -285,7 +269,7 @@ export async function runAgent(
|
|
|
285
269
|
options: RunOptions,
|
|
286
270
|
io: RunnerIO,
|
|
287
271
|
): Promise<RunResult> {
|
|
288
|
-
// Resolve working directory upfront
|
|
272
|
+
// Resolve working directory upfront - needed for detectEnv before assembly.
|
|
289
273
|
const effectiveCwd = options.context.cwd ?? snapshot.cwd;
|
|
290
274
|
const env = await io.detectEnv(options.context.exec, effectiveCwd);
|
|
291
275
|
|
|
@@ -312,7 +296,7 @@ export async function runAgent(
|
|
|
312
296
|
const agentDir = io.getAgentDir();
|
|
313
297
|
|
|
314
298
|
// Load extensions/skills: true or string[] → load; false → don't.
|
|
315
|
-
// Suppress AGENTS.md/CLAUDE.md and APPEND_SYSTEM.md
|
|
299
|
+
// Suppress AGENTS.md/CLAUDE.md and APPEND_SYSTEM.md - upstream's
|
|
316
300
|
// buildSystemPrompt() re-appends both AFTER systemPromptOverride, which
|
|
317
301
|
// would defeat prompt_mode: replace and isolated: true. Parent context, if
|
|
318
302
|
// wanted, reaches the subagent via prompt_mode: append (parentSystemPrompt
|
|
@@ -351,7 +335,7 @@ export async function runAgent(
|
|
|
351
335
|
|
|
352
336
|
// Filter active tools: remove our own tools to prevent nesting,
|
|
353
337
|
// apply extension allowlist if specified, and apply disallowedTools denylist.
|
|
354
|
-
// First pass
|
|
338
|
+
// First pass - over built-in tools, before bindExtensions registers extension tools.
|
|
355
339
|
if (cfg.toolFilter.extensions !== false || cfg.toolFilter.disallowedSet) {
|
|
356
340
|
const filtered = filterActiveTools(session.getActiveToolNames(), cfg.toolFilter);
|
|
357
341
|
session.setActiveToolsByName(filtered);
|
|
@@ -391,7 +375,7 @@ export async function runAgent(
|
|
|
391
375
|
if (!softLimitReached && turnCount >= maxTurns) {
|
|
392
376
|
softLimitReached = true;
|
|
393
377
|
void session.steer(
|
|
394
|
-
"You have reached your turn limit. Wrap up immediately
|
|
378
|
+
"You have reached your turn limit. Wrap up immediately - provide your final answer now.",
|
|
395
379
|
);
|
|
396
380
|
} else if (softLimitReached && turnCount >= maxTurns + (options.graceTurns ?? 5)) {
|
|
397
381
|
aborted = true;
|
|
@@ -475,17 +459,11 @@ export function getAgentConversation(session: AgentSession): string {
|
|
|
475
459
|
: extractText(msg.content);
|
|
476
460
|
if (text.trim()) parts.push(`[User]: ${text.trim()}`);
|
|
477
461
|
} else if (msg.role === "assistant") {
|
|
478
|
-
const textParts
|
|
479
|
-
const toolCalls: string[] = [];
|
|
480
|
-
for (const c of msg.content) {
|
|
481
|
-
if (c.type === "text" && c.text) textParts.push(c.text);
|
|
482
|
-
else if (c.type === "toolCall")
|
|
483
|
-
toolCalls.push(` Tool: ${getToolCallName(c)}`);
|
|
484
|
-
}
|
|
462
|
+
const { textParts, toolNames } = extractAssistantContent(msg.content);
|
|
485
463
|
if (textParts.length > 0)
|
|
486
464
|
parts.push(`[Assistant]: ${textParts.join("\n")}`);
|
|
487
|
-
if (
|
|
488
|
-
parts.push(`[Tool Calls]:\n${
|
|
465
|
+
if (toolNames.length > 0)
|
|
466
|
+
parts.push(`[Tool Calls]:\n${toolNames.map((n) => ` Tool: ${n}`).join("\n")}`);
|
|
489
467
|
} else if (msg.role === "toolResult") {
|
|
490
468
|
const text = extractText(msg.content);
|
|
491
469
|
const truncated = text.length > 200 ? text.slice(0, 200) + "..." : text;
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* content-items.ts — Shared parsing utilities for Pi SDK message content items.
|
|
3
|
+
*
|
|
4
|
+
* Provides type-safe extraction of text parts and tool-call names from
|
|
5
|
+
* assistant message content arrays. Pure functions — no IO.
|
|
6
|
+
*/
|
|
7
|
+
|
|
8
|
+
import type { TextContent, ToolCall } from "@earendil-works/pi-ai";
|
|
9
|
+
|
|
10
|
+
// ── Types ─────────────────────────────────────────────────────────────────────
|
|
11
|
+
|
|
12
|
+
/** Extracted text parts and tool names from assistant message content. */
|
|
13
|
+
export interface AssistantContentParts {
|
|
14
|
+
textParts: string[];
|
|
15
|
+
toolNames: string[];
|
|
16
|
+
}
|
|
17
|
+
|
|
18
|
+
// ── Functions ─────────────────────────────────────────────────────────────────
|
|
19
|
+
|
|
20
|
+
/**
|
|
21
|
+
* Extracts the display name from a tool-call content item.
|
|
22
|
+
*
|
|
23
|
+
* Returns 'unknown' for non-toolCall items.
|
|
24
|
+
* The Pi SDK's ToolCall.name is always present — no fallback chain needed.
|
|
25
|
+
*/
|
|
26
|
+
export function getToolCallName(c: { type: string }): string {
|
|
27
|
+
if (c.type !== "toolCall") return "unknown";
|
|
28
|
+
return (c as ToolCall).name;
|
|
29
|
+
}
|
|
30
|
+
|
|
31
|
+
/**
|
|
32
|
+
* Extract text parts and tool-call names from assistant message content items.
|
|
33
|
+
*
|
|
34
|
+
* Accepts any array whose elements carry a `type` discriminant — all Pi SDK
|
|
35
|
+
* content types (TextContent, ThinkingContent, ToolCall) satisfy this constraint.
|
|
36
|
+
* Pure data extraction — consumers apply their own presentation formatting.
|
|
37
|
+
* Skips items of unknown types (e.g. thinking blocks, images) and empty text.
|
|
38
|
+
*/
|
|
39
|
+
export function extractAssistantContent(
|
|
40
|
+
content: ReadonlyArray<{ type: string }>,
|
|
41
|
+
): AssistantContentParts {
|
|
42
|
+
const textParts: string[] = [];
|
|
43
|
+
const toolNames: string[] = [];
|
|
44
|
+
for (const c of content) {
|
|
45
|
+
if (c.type === "text") {
|
|
46
|
+
const text = (c as TextContent).text;
|
|
47
|
+
if (text) textParts.push(text);
|
|
48
|
+
} else if (c.type === "toolCall") {
|
|
49
|
+
toolNames.push(getToolCallName(c));
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
return { textParts, toolNames };
|
|
53
|
+
}
|
|
@@ -6,6 +6,7 @@
|
|
|
6
6
|
*/
|
|
7
7
|
|
|
8
8
|
import { truncateToWidth } from "@earendil-works/pi-tui";
|
|
9
|
+
import { extractAssistantContent } from "#src/session/content-items";
|
|
9
10
|
import { extractText } from "#src/session/context";
|
|
10
11
|
import type { Theme } from "#src/ui/display";
|
|
11
12
|
import { describeActivity } from "#src/ui/display";
|
|
@@ -20,6 +21,8 @@ export interface FormatterContext {
|
|
|
20
21
|
|
|
21
22
|
// ── File-local types and guards ─────────────────────────────────────────────
|
|
22
23
|
|
|
24
|
+
export { getToolCallName } from "#src/session/content-items";
|
|
25
|
+
|
|
23
26
|
/** Bash execution message — 'bashExecution' role is not in the SDK's AgentSession message role union. */
|
|
24
27
|
export interface BashExecutionMessage {
|
|
25
28
|
role: "bashExecution";
|
|
@@ -32,20 +35,6 @@ export function isBashExecution(msg: { role: string }): msg is BashExecutionMess
|
|
|
32
35
|
return msg.role === "bashExecution";
|
|
33
36
|
}
|
|
34
37
|
|
|
35
|
-
/** Tool-call content item — SDK exposes this variant at runtime but doesn't export the narrow type. */
|
|
36
|
-
interface ToolCallContent {
|
|
37
|
-
type: "toolCall";
|
|
38
|
-
name?: string;
|
|
39
|
-
toolName?: string;
|
|
40
|
-
}
|
|
41
|
-
|
|
42
|
-
/** Extracts the tool name from a toolCall content item, falling back to 'unknown'. */
|
|
43
|
-
export function getToolCallName(c: { type: string }): string {
|
|
44
|
-
if (c.type !== "toolCall") return "unknown";
|
|
45
|
-
const tc = c as ToolCallContent;
|
|
46
|
-
return tc.name ?? tc.toolName ?? "unknown";
|
|
47
|
-
}
|
|
48
|
-
|
|
49
38
|
// ── formatUserMessage ─────────────────────────────────────────────────────────
|
|
50
39
|
|
|
51
40
|
/**
|
|
@@ -141,17 +130,12 @@ export function formatAssistantMessage(
|
|
|
141
130
|
ctx: FormatterContext,
|
|
142
131
|
): string[] {
|
|
143
132
|
const { theme, wrapText } = ctx;
|
|
144
|
-
const textParts
|
|
145
|
-
const toolCalls: string[] = [];
|
|
146
|
-
for (const c of content) {
|
|
147
|
-
if (c.type === "text" && c.text) textParts.push(c.text as string);
|
|
148
|
-
else if (c.type === "toolCall") toolCalls.push(getToolCallName(c));
|
|
149
|
-
}
|
|
133
|
+
const { textParts, toolNames } = extractAssistantContent(content);
|
|
150
134
|
const lines: string[] = [theme.bold("[Assistant]")];
|
|
151
135
|
if (textParts.length > 0) {
|
|
152
136
|
lines.push(...wrapText(textParts.join("\n").trim(), width));
|
|
153
137
|
}
|
|
154
|
-
for (const name of
|
|
138
|
+
for (const name of toolNames) {
|
|
155
139
|
lines.push(truncateToWidth(theme.fg("muted", ` [Tool: ${name}]`), width));
|
|
156
140
|
}
|
|
157
141
|
return lines;
|