@hheei/pi-ext-ui 0.6.0 → 0.7.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/README.md +2 -2
- package/dist/background.d.ts +11 -0
- package/dist/background.js +31 -0
- package/dist/extension.js +59 -7
- package/dist/host-transcript-patch.js +176 -58
- package/dist/index.d.ts +5 -3
- package/dist/index.js +5 -3
- package/dist/message-view.d.ts +4 -10
- package/dist/message-view.js +29 -16
- package/dist/native-tools.d.ts +69 -0
- package/dist/native-tools.js +414 -0
- package/dist/nested-calls-view.d.ts +6 -2
- package/dist/nested-calls-view.js +58 -73
- package/dist/payload-parser.d.ts +10 -28
- package/dist/payload-parser.js +9 -269
- package/dist/renderer.d.ts +6 -4
- package/dist/renderer.js +242 -181
- package/dist/runtime.d.ts +13 -0
- package/dist/runtime.js +16 -0
- package/dist/settings.d.ts +3 -0
- package/dist/settings.js +18 -1
- package/dist/tool-group-view.d.ts +52 -8
- package/dist/tool-group-view.js +351 -164
- package/dist/tool-view.d.ts +20 -6
- package/dist/tool-view.js +77 -30
- package/dist/tracker.d.ts +36 -17
- package/dist/tracker.js +157 -19
- package/package.json +2 -2
package/dist/tool-view.d.ts
CHANGED
|
@@ -11,8 +11,12 @@ export interface RailSection {
|
|
|
11
11
|
export interface ToolViewHeaderSpec {
|
|
12
12
|
readonly title: string;
|
|
13
13
|
readonly status?: ToolStatusKind | undefined;
|
|
14
|
+
/** The call's own text. Plain unless `summaryStyled` says the tool layer themed it already. */
|
|
14
15
|
readonly summary?: string | undefined;
|
|
15
|
-
|
|
16
|
+
/** The summary already carries its semantic tokens; the frame must not recolor it. */
|
|
17
|
+
readonly summaryStyled?: boolean | undefined;
|
|
18
|
+
/** Secondary facts after the summary: duration, timeout. Always dim. */
|
|
19
|
+
readonly meta?: string | undefined;
|
|
16
20
|
}
|
|
17
21
|
export interface ToolViewCopySpec {
|
|
18
22
|
readonly primary: string;
|
|
@@ -23,13 +27,12 @@ export interface Collapsible {
|
|
|
23
27
|
collapseAll(): void;
|
|
24
28
|
}
|
|
25
29
|
export declare function isCollapsible(value: unknown): value is Collapsible;
|
|
26
|
-
/** One nested call inside a container, reduced to what a row and its
|
|
30
|
+
/** One nested call inside a container, reduced to what a summary row and its count need. */
|
|
27
31
|
export interface ToolViewChild {
|
|
28
32
|
readonly title: string;
|
|
29
33
|
readonly summary?: string | undefined;
|
|
30
34
|
readonly status?: ToolStatusKind | undefined;
|
|
31
35
|
readonly durationMs?: number | undefined;
|
|
32
|
-
readonly detail?: string | undefined;
|
|
33
36
|
}
|
|
34
37
|
export interface ToolViewSpec {
|
|
35
38
|
readonly theme: Theme;
|
|
@@ -51,14 +54,24 @@ export interface ToolViewSpec {
|
|
|
51
54
|
readonly copyFeedbackDurationMs?: number | undefined;
|
|
52
55
|
readonly noRail?: boolean | undefined;
|
|
53
56
|
readonly railToken?: RailColorToken | undefined;
|
|
57
|
+
/**
|
|
58
|
+
* Pi-native surface: success/error fill plus one padding row, and no leading rail. Read per render,
|
|
59
|
+
* so a settings change reaches a view the host already holds.
|
|
60
|
+
*/
|
|
61
|
+
readonly background?: (() => boolean) | undefined;
|
|
54
62
|
}
|
|
55
63
|
/** Public composition entry point for extensions that need the shared view behavior. */
|
|
56
64
|
export declare function createToolView(spec: ToolViewSpec): ToolView;
|
|
57
65
|
export declare const MAX_LEVEL2_SECTION_LINES = 500;
|
|
58
66
|
export declare const MAX_LEVEL2_BYTES: number;
|
|
59
|
-
/** Internal transcript composition: original components keep their layout and interactions. */
|
|
60
|
-
export declare function renderWithoutToolRails(component: Component, width: number): string[];
|
|
61
67
|
export declare function transitionFoldLevel(current: FoldLevel, ctrlKey: boolean): FoldLevel;
|
|
68
|
+
export declare const TOOL_SPINNER_FRAMES: readonly ["·", "✢", "✳", "✶", "✻", "✽", "✻", "✶", "✳", "✢"];
|
|
69
|
+
export declare const TOOL_SPINNER_INTERVAL_MS = 100;
|
|
70
|
+
/**
|
|
71
|
+
* One clock for every live tool, so the phase is shared and never restarts: streaming args and a
|
|
72
|
+
* running execution draw the same frame and differ only in color.
|
|
73
|
+
*/
|
|
74
|
+
export declare function toolSpinnerFrame(now?: number): string;
|
|
62
75
|
export declare function formatStatusGlyph(status: ToolStatusKind | undefined, theme: Theme): string;
|
|
63
76
|
/**
|
|
64
77
|
* The glyph, name, and summary of one tool execution, colored the way every tool row colors them.
|
|
@@ -73,7 +86,8 @@ export declare function formatToolHeaderText(parts: {
|
|
|
73
86
|
readonly status?: ToolStatusKind | undefined;
|
|
74
87
|
readonly title: string;
|
|
75
88
|
readonly summary?: string | undefined;
|
|
76
|
-
readonly
|
|
89
|
+
readonly summaryStyled?: boolean | undefined;
|
|
90
|
+
readonly meta?: string | undefined;
|
|
77
91
|
}, theme: Theme): string;
|
|
78
92
|
/**
|
|
79
93
|
* Collapses text that must occupy one row: header titles and summaries, board rows, and inline
|
package/dist/tool-view.js
CHANGED
|
@@ -1,6 +1,15 @@
|
|
|
1
1
|
import { Buffer } from "node:buffer";
|
|
2
2
|
import { copyToClipboard as copyToSystemClipboard, } from "@earendil-works/pi-coding-agent";
|
|
3
3
|
import { truncateToWidth, visibleWidth, wrapTextWithAnsi, } from "@earendil-works/pi-tui";
|
|
4
|
+
import { backgroundFill } from "./background.js";
|
|
5
|
+
/** Pi native tool surfaces: a failure outranks work in progress, which outranks a settled tool. */
|
|
6
|
+
function nativeSurface(status) {
|
|
7
|
+
if (status === "error")
|
|
8
|
+
return "toolErrorBg";
|
|
9
|
+
if (status === "running" || status === "streaming")
|
|
10
|
+
return "toolPendingBg";
|
|
11
|
+
return "toolSuccessBg";
|
|
12
|
+
}
|
|
4
13
|
export function isCollapsible(value) {
|
|
5
14
|
return (typeof value === "object" &&
|
|
6
15
|
value !== null &&
|
|
@@ -19,17 +28,6 @@ const DEFAULT_MAX_SECTION_LINES = 9;
|
|
|
19
28
|
export const MAX_LEVEL2_SECTION_LINES = 500;
|
|
20
29
|
export const MAX_LEVEL2_BYTES = 64 * 1024;
|
|
21
30
|
const DEFAULT_COPY_FEEDBACK_MS = 1200;
|
|
22
|
-
let outerRailDepth = 0;
|
|
23
|
-
/** Internal transcript composition: original components keep their layout and interactions. */
|
|
24
|
-
export function renderWithoutToolRails(component, width) {
|
|
25
|
-
outerRailDepth++;
|
|
26
|
-
try {
|
|
27
|
-
return component.render(width);
|
|
28
|
-
}
|
|
29
|
-
finally {
|
|
30
|
-
outerRailDepth--;
|
|
31
|
-
}
|
|
32
|
-
}
|
|
33
31
|
function isRecord(value) {
|
|
34
32
|
return typeof value === "object" && value !== null && !Array.isArray(value);
|
|
35
33
|
}
|
|
@@ -68,12 +66,22 @@ function isCopyFailed(stateHolder, localFailed) {
|
|
|
68
66
|
return true;
|
|
69
67
|
return isRecord(stateHolder) && stateHolder[COPY_FAILED_STATE_KEY] === true;
|
|
70
68
|
}
|
|
69
|
+
export const TOOL_SPINNER_FRAMES = ["·", "✢", "✳", "✶", "✻", "✽", "✻", "✶", "✳", "✢"];
|
|
70
|
+
export const TOOL_SPINNER_INTERVAL_MS = 100;
|
|
71
|
+
/**
|
|
72
|
+
* One clock for every live tool, so the phase is shared and never restarts: streaming args and a
|
|
73
|
+
* running execution draw the same frame and differ only in color.
|
|
74
|
+
*/
|
|
75
|
+
export function toolSpinnerFrame(now = performance.now()) {
|
|
76
|
+
const at = Math.floor(now / TOOL_SPINNER_INTERVAL_MS) % TOOL_SPINNER_FRAMES.length;
|
|
77
|
+
return TOOL_SPINNER_FRAMES[at] ?? "·";
|
|
78
|
+
}
|
|
71
79
|
export function formatStatusGlyph(status, theme) {
|
|
72
80
|
switch (status ?? "ok") {
|
|
73
81
|
case "streaming":
|
|
74
|
-
return theme.fg("muted",
|
|
82
|
+
return theme.fg("muted", toolSpinnerFrame());
|
|
75
83
|
case "running":
|
|
76
|
-
return theme.fg("accent",
|
|
84
|
+
return theme.fg("accent", toolSpinnerFrame());
|
|
77
85
|
case "ok":
|
|
78
86
|
return theme.fg("success", "✓");
|
|
79
87
|
case "warn":
|
|
@@ -98,9 +106,20 @@ export function formatToolHeaderText(parts, theme) {
|
|
|
98
106
|
const title = toSingleRow(parts.title);
|
|
99
107
|
const titlePart = theme.fg("toolTitle", theme.bold(title));
|
|
100
108
|
const summaryRaw = parts.summary === undefined ? "" : toSingleRow(parts.summary);
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
109
|
+
// A command is the row's content, not its caption: bash states it the way the native shell row
|
|
110
|
+
// does. A summary the tool layer already themed keeps its own tokens.
|
|
111
|
+
const summaryPart = summaryRaw === ""
|
|
112
|
+
? ""
|
|
113
|
+
: parts.summaryStyled === true
|
|
114
|
+
? ` ${summaryRaw}`
|
|
115
|
+
: title === "bash"
|
|
116
|
+
? ` ${theme.fg("toolTitle", theme.bold(summaryRaw))}`
|
|
117
|
+
: ` ${theme.fg("dim", summaryRaw)}`;
|
|
118
|
+
const metaRaw = parts.meta === undefined ? "" : toSingleRow(parts.meta);
|
|
119
|
+
const metaPart = metaRaw === ""
|
|
120
|
+
? ""
|
|
121
|
+
: `${summaryPart === "" ? " " : theme.fg("dim", " · ")}${theme.fg("dim", metaRaw)}`;
|
|
122
|
+
return `${statusPart === "" ? "" : `${statusPart} `}${titlePart}${summaryPart}${metaPart}`;
|
|
104
123
|
}
|
|
105
124
|
const truncate = (text, width) => truncateToWidth(text, width, "…");
|
|
106
125
|
/**
|
|
@@ -229,6 +248,8 @@ export class ToolView {
|
|
|
229
248
|
#lastWidth = 80;
|
|
230
249
|
#lastHeight = 0;
|
|
231
250
|
#lastNoRail = false;
|
|
251
|
+
/** Rows the native surface adds above the header. */
|
|
252
|
+
#lastTopPad = 0;
|
|
232
253
|
#cache;
|
|
233
254
|
constructor(spec) {
|
|
234
255
|
this.#spec = spec;
|
|
@@ -408,7 +429,7 @@ export class ToolView {
|
|
|
408
429
|
const safeWidth = Math.max(1, width - pad * 2);
|
|
409
430
|
const level = this.effectiveLevel;
|
|
410
431
|
const { theme, collapsedRows, sections } = this.#spec;
|
|
411
|
-
const noRail = options?.noRail ?? (
|
|
432
|
+
const noRail = options?.noRail ?? (this.#spec.noRail === true || this.#spec.background?.() === true);
|
|
412
433
|
const innerWidth = noRail ? safeWidth : Math.max(1, safeWidth - 2);
|
|
413
434
|
const output = [];
|
|
414
435
|
if (level === 0) {
|
|
@@ -426,11 +447,14 @@ export class ToolView {
|
|
|
426
447
|
return output;
|
|
427
448
|
}
|
|
428
449
|
for (const section of sections) {
|
|
450
|
+
const sectionOutput = [];
|
|
429
451
|
const rail = noRail ? "" : `${theme.fg(section.railToken, "▎")} `;
|
|
430
452
|
const { rows: rawLines, isByteTruncated, totalBytes, } = resolveRows(section.content, innerWidth, toContentRows);
|
|
431
453
|
const cap = level === 1 ? (section.maxLines ?? DEFAULT_MAX_SECTION_LINES) : MAX_LEVEL2_SECTION_LINES;
|
|
432
454
|
const sectionRows = [];
|
|
433
455
|
let logicalLinesCount = 0;
|
|
456
|
+
// Wrapped rows the last shown line still has: a single long line is not "1 more line".
|
|
457
|
+
let cutChunks = 0;
|
|
434
458
|
for (const line of rawLines) {
|
|
435
459
|
const wrapped = wrapTextWithAnsi(line, innerWidth);
|
|
436
460
|
const chunks = wrapped.length > 0 ? wrapped : [""];
|
|
@@ -442,21 +466,36 @@ export class ToolView {
|
|
|
442
466
|
sectionRows.push(chunk);
|
|
443
467
|
}
|
|
444
468
|
}
|
|
469
|
+
// Only a line that started showing is "continues"; a fully omitted one is a whole line.
|
|
470
|
+
cutChunks = canTake > 0 ? chunks.length - canTake : 0;
|
|
445
471
|
break;
|
|
446
472
|
}
|
|
447
473
|
sectionRows.push(...chunks);
|
|
448
474
|
logicalLinesCount++;
|
|
449
475
|
}
|
|
450
|
-
|
|
476
|
+
// The line that broke the cap is not an omitted whole line: its hidden part is `cutChunks`.
|
|
477
|
+
const omittedCount = rawLines.length - logicalLinesCount - (cutChunks > 0 ? 1 : 0);
|
|
478
|
+
// The hint names the real way to the hidden text; without a copy it must not promise one.
|
|
479
|
+
const copyHint = this.#spec.copy?.full !== undefined && this.#spec.copy.full !== ""
|
|
480
|
+
? ", shift+click copies all"
|
|
481
|
+
: "";
|
|
451
482
|
for (const row of sectionRows) {
|
|
452
|
-
|
|
483
|
+
sectionOutput.push(truncate(rail + row, safeWidth));
|
|
453
484
|
}
|
|
454
|
-
if (omittedCount > 0 || isByteTruncated) {
|
|
455
|
-
|
|
456
|
-
|
|
457
|
-
|
|
458
|
-
|
|
485
|
+
if (omittedCount > 0 || cutChunks > 0 || isByteTruncated) {
|
|
486
|
+
let text;
|
|
487
|
+
if (isByteTruncated) {
|
|
488
|
+
text = `… [Output truncated: showing 64KB of ${Math.round(totalBytes / 1024)}KB${copyHint}]`;
|
|
489
|
+
}
|
|
490
|
+
else if (omittedCount > 0) {
|
|
491
|
+
text = `… (${String(omittedCount)} more lines${cutChunks > 0 ? ", current line continues" : ""}${copyHint})`;
|
|
492
|
+
}
|
|
493
|
+
else {
|
|
494
|
+
text = `… (current line continues${copyHint})`;
|
|
495
|
+
}
|
|
496
|
+
sectionOutput.push(truncate(rail + theme.fg("dim", text), safeWidth));
|
|
459
497
|
}
|
|
498
|
+
output.push(...sectionOutput);
|
|
460
499
|
}
|
|
461
500
|
return output;
|
|
462
501
|
}
|
|
@@ -465,11 +504,14 @@ export class ToolView {
|
|
|
465
504
|
return [];
|
|
466
505
|
const level = this.effectiveLevel;
|
|
467
506
|
const copied = isCopiedActive(this.#spec.stateHolder, this.#localCopiedUntil);
|
|
468
|
-
const
|
|
507
|
+
const native = this.#spec.background?.() === true;
|
|
508
|
+
const surface = native ? nativeSurface(this.#spec.header.status) : undefined;
|
|
509
|
+
const noRail = this.#spec.noRail === true || native;
|
|
469
510
|
if (this.#cache !== undefined &&
|
|
470
511
|
this.#cache.width === width &&
|
|
471
512
|
this.#cache.level === level &&
|
|
472
513
|
this.#cache.noRail === noRail &&
|
|
514
|
+
this.#cache.surface === surface &&
|
|
473
515
|
this.#cache.copied === copied) {
|
|
474
516
|
return this.#cache.lines;
|
|
475
517
|
}
|
|
@@ -494,9 +536,13 @@ export class ToolView {
|
|
|
494
536
|
safeLines.push(leftPad !== "" ? `${leftPad}${rawLine}` : rawLine);
|
|
495
537
|
}
|
|
496
538
|
}
|
|
497
|
-
this.#
|
|
498
|
-
|
|
499
|
-
|
|
539
|
+
this.#lastTopPad = surface === undefined ? 0 : 1;
|
|
540
|
+
const lines = surface === undefined
|
|
541
|
+
? safeLines
|
|
542
|
+
: ["", ...safeLines, ""].map((line) => backgroundFill(this.#spec.theme, surface)(line + " ".repeat(Math.max(0, width - visibleWidth(line)))));
|
|
543
|
+
this.#lastHeight = lines.length;
|
|
544
|
+
this.#cache = { width, level, copied, noRail, surface, lines };
|
|
545
|
+
return lines;
|
|
500
546
|
}
|
|
501
547
|
handleMouse(event) {
|
|
502
548
|
if (event.type === "wheel" || this.#hidden) {
|
|
@@ -517,7 +563,8 @@ export class ToolView {
|
|
|
517
563
|
if (this.#lastHeight > 0 && (event.y < 0 || event.y >= this.#lastHeight)) {
|
|
518
564
|
return undefined;
|
|
519
565
|
}
|
|
520
|
-
|
|
566
|
+
const row = event.y - this.#lastTopPad;
|
|
567
|
+
if (row === 0) {
|
|
521
568
|
if (this.#spec.copy !== undefined &&
|
|
522
569
|
innerWidth >= 6 &&
|
|
523
570
|
innerX >= innerWidth - 3 &&
|
|
@@ -539,7 +586,7 @@ export class ToolView {
|
|
|
539
586
|
const railOffset = this.#lastNoRail ? 0 : 2;
|
|
540
587
|
const subEvent = {
|
|
541
588
|
...event,
|
|
542
|
-
y:
|
|
589
|
+
y: row - 1,
|
|
543
590
|
x: Math.max(0, innerX - railOffset),
|
|
544
591
|
width: Math.max(1, innerWidth - railOffset),
|
|
545
592
|
};
|
package/dist/tracker.d.ts
CHANGED
|
@@ -1,8 +1,15 @@
|
|
|
1
|
-
import type { AgentToolResult } from "@earendil-works/pi-coding-agent";
|
|
1
|
+
import type { AgentToolResult, ExtensionContext } from "@earendil-works/pi-coding-agent";
|
|
2
|
+
/** The public read-only session view the lifecycle context hands an extension. */
|
|
3
|
+
type SessionReadSource = ExtensionContext["sessionManager"];
|
|
2
4
|
export interface CapturedToolCall {
|
|
3
5
|
readonly id: string;
|
|
4
6
|
readonly toolName: string;
|
|
7
|
+
/** Positive resolution fact, when the host's `tool_call` hook reported this call. */
|
|
8
|
+
readonly resolved?: boolean | undefined;
|
|
5
9
|
readonly args?: unknown;
|
|
10
|
+
/** The request existed but was too large to keep; the bytes are not recorded. */
|
|
11
|
+
readonly argsTruncated?: boolean | undefined;
|
|
12
|
+
readonly argsBytes?: number | undefined;
|
|
6
13
|
readonly result?: AgentToolResult<unknown> | undefined;
|
|
7
14
|
readonly isError?: boolean | undefined;
|
|
8
15
|
readonly durationMs?: number | undefined;
|
|
@@ -25,21 +32,33 @@ export declare class ToolExecutionTracker {
|
|
|
25
32
|
parentToolCallId?: string | undefined;
|
|
26
33
|
}): void;
|
|
27
34
|
getCaptured(parentToolCallId: string | undefined): readonly CapturedToolCall[] | undefined;
|
|
35
|
+
/**
|
|
36
|
+
* The host's `tool_call` hook runs after the tool resolved and its arguments validated, so a
|
|
37
|
+
* recorded id is a positive fact. Nothing is recorded for a call the hook never saw: an unknown
|
|
38
|
+
* tool, an invalid argument set, a blocked or cancelled call all stay unknown.
|
|
39
|
+
*/
|
|
40
|
+
recordResolved(event: {
|
|
41
|
+
toolCallId: string;
|
|
42
|
+
parentToolCallId?: string | undefined;
|
|
43
|
+
}): void;
|
|
44
|
+
getResolved(parentToolCallId: string | undefined): ReadonlySet<string> | undefined;
|
|
28
45
|
clear(): void;
|
|
29
46
|
}
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
47
|
+
/** What the session branch says about one nested call's request. */
|
|
48
|
+
export interface NestedCallArgsFact {
|
|
49
|
+
readonly args: unknown;
|
|
50
|
+
/** The host kept the size but not the arguments, so the request is not recoverable. */
|
|
51
|
+
readonly truncated: boolean;
|
|
52
|
+
readonly bytes?: number | undefined;
|
|
53
|
+
}
|
|
54
|
+
/**
|
|
55
|
+
* The arguments a resumed session saved for its nested calls. One index belongs to one runtime: it
|
|
56
|
+
* follows that runtime's session, and shutting the runtime down or switching sessions drops it.
|
|
57
|
+
*/
|
|
58
|
+
export declare class NestedCallArgsIndex {
|
|
59
|
+
#private;
|
|
60
|
+
/** The session owns the index: switching, reloading or shutting down replaces it. */
|
|
61
|
+
setSource(manager: SessionReadSource | undefined): void;
|
|
62
|
+
args(toolCallId: string | undefined): NestedCallArgsFact | undefined;
|
|
63
|
+
}
|
|
64
|
+
export {};
|
package/dist/tracker.js
CHANGED
|
@@ -1,15 +1,30 @@
|
|
|
1
1
|
const MAX_STORED_PARENTS = 200;
|
|
2
|
+
/** A live capture keeps the request whole or not at all; a partial request would be a lie. */
|
|
2
3
|
const MAX_CAPTURED_PAYLOAD_CHARS = 32_768;
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
|
|
4
|
+
/** Host records at most this many nested calls per model-issued call. */
|
|
5
|
+
const MAX_RESOLVED_CALLS = 256;
|
|
6
|
+
/** Recovered arguments for at most this many calls; beyond it a row says it is not recorded. */
|
|
7
|
+
const MAX_INDEXED_CALLS = 512;
|
|
8
|
+
/**
|
|
9
|
+
* The captured request, or the fact that it was too large to keep. Nothing is rewritten: a caller
|
|
10
|
+
* either has the bytes the model sent or knows they are not recorded.
|
|
11
|
+
*/
|
|
12
|
+
function capturedArgs(args) {
|
|
13
|
+
let serialized;
|
|
14
|
+
try {
|
|
15
|
+
serialized = JSON.stringify(args);
|
|
8
16
|
}
|
|
9
|
-
|
|
17
|
+
catch {
|
|
18
|
+
// A value the host could not serialize is kept as-is; its size is unknown, not zero.
|
|
19
|
+
return { args };
|
|
20
|
+
}
|
|
21
|
+
if (serialized === undefined || serialized.length <= MAX_CAPTURED_PAYLOAD_CHARS)
|
|
22
|
+
return { args };
|
|
23
|
+
return { argsTruncated: true, argsBytes: serialized.length };
|
|
10
24
|
}
|
|
11
25
|
export class ToolExecutionTracker {
|
|
12
26
|
#capturedByParentId = new Map();
|
|
27
|
+
#resolvedByParentId = new Map();
|
|
13
28
|
recordStart(event) {
|
|
14
29
|
if (!event.parentToolCallId)
|
|
15
30
|
return;
|
|
@@ -27,7 +42,7 @@ export class ToolExecutionTracker {
|
|
|
27
42
|
const entry = {
|
|
28
43
|
id: event.toolCallId,
|
|
29
44
|
toolName: event.toolName,
|
|
30
|
-
|
|
45
|
+
...capturedArgs(event.args),
|
|
31
46
|
};
|
|
32
47
|
if (idx >= 0) {
|
|
33
48
|
list[idx] = { ...list[idx], ...entry };
|
|
@@ -77,20 +92,143 @@ export class ToolExecutionTracker {
|
|
|
77
92
|
return undefined;
|
|
78
93
|
return this.#capturedByParentId.get(parentToolCallId);
|
|
79
94
|
}
|
|
95
|
+
/**
|
|
96
|
+
* The host's `tool_call` hook runs after the tool resolved and its arguments validated, so a
|
|
97
|
+
* recorded id is a positive fact. Nothing is recorded for a call the hook never saw: an unknown
|
|
98
|
+
* tool, an invalid argument set, a blocked or cancelled call all stay unknown.
|
|
99
|
+
*/
|
|
100
|
+
recordResolved(event) {
|
|
101
|
+
const parent = event.parentToolCallId;
|
|
102
|
+
if (!parent)
|
|
103
|
+
return;
|
|
104
|
+
let ids = this.#resolvedByParentId.get(parent);
|
|
105
|
+
if (!ids) {
|
|
106
|
+
if (this.#resolvedByParentId.size >= MAX_STORED_PARENTS) {
|
|
107
|
+
const firstKey = this.#resolvedByParentId.keys().next().value;
|
|
108
|
+
if (firstKey !== undefined)
|
|
109
|
+
this.#resolvedByParentId.delete(firstKey);
|
|
110
|
+
}
|
|
111
|
+
ids = new Set();
|
|
112
|
+
this.#resolvedByParentId.set(parent, ids);
|
|
113
|
+
}
|
|
114
|
+
if (ids.size >= MAX_RESOLVED_CALLS)
|
|
115
|
+
return;
|
|
116
|
+
ids.add(event.toolCallId);
|
|
117
|
+
}
|
|
118
|
+
getResolved(parentToolCallId) {
|
|
119
|
+
if (!parentToolCallId)
|
|
120
|
+
return undefined;
|
|
121
|
+
return this.#resolvedByParentId.get(parentToolCallId);
|
|
122
|
+
}
|
|
80
123
|
clear() {
|
|
81
124
|
this.#capturedByParentId.clear();
|
|
125
|
+
this.#resolvedByParentId.clear();
|
|
82
126
|
}
|
|
83
127
|
}
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
export
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
128
|
+
/**
|
|
129
|
+
* The arguments a resumed session saved for its nested calls. One index belongs to one runtime: it
|
|
130
|
+
* follows that runtime's session, and shutting the runtime down or switching sessions drops it.
|
|
131
|
+
*/
|
|
132
|
+
export class NestedCallArgsIndex {
|
|
133
|
+
#source;
|
|
134
|
+
/** The leaf the index is complete up to; undefined means no index exists yet. */
|
|
135
|
+
#indexedLeaf;
|
|
136
|
+
#argsById = new Map();
|
|
137
|
+
/** The session owns the index: switching, reloading or shutting down replaces it. */
|
|
138
|
+
setSource(manager) {
|
|
139
|
+
this.#source = manager;
|
|
140
|
+
this.#reset();
|
|
141
|
+
}
|
|
142
|
+
args(toolCallId) {
|
|
143
|
+
const manager = this.#source;
|
|
144
|
+
if (manager === undefined || toolCallId === undefined)
|
|
145
|
+
return undefined;
|
|
146
|
+
try {
|
|
147
|
+
this.#refresh(manager);
|
|
148
|
+
}
|
|
149
|
+
catch {
|
|
150
|
+
this.#reset();
|
|
151
|
+
return undefined;
|
|
152
|
+
}
|
|
153
|
+
return this.#argsById.get(toolCallId);
|
|
154
|
+
}
|
|
155
|
+
#reset() {
|
|
156
|
+
this.#argsById.clear();
|
|
157
|
+
this.#indexedLeaf = undefined;
|
|
158
|
+
}
|
|
159
|
+
#remember(id, fact) {
|
|
160
|
+
if (!this.#argsById.has(id) && this.#argsById.size >= MAX_INDEXED_CALLS) {
|
|
161
|
+
const firstKey = this.#argsById.keys().next().value;
|
|
162
|
+
if (firstKey !== undefined)
|
|
163
|
+
this.#argsById.delete(firstKey);
|
|
164
|
+
}
|
|
165
|
+
this.#argsById.set(id, fact);
|
|
166
|
+
}
|
|
167
|
+
/** The host saves a bounded record on the parent's tool result; this reads it back by call id. */
|
|
168
|
+
#scan(entry) {
|
|
169
|
+
if (typeof entry !== "object" || entry === null)
|
|
170
|
+
return;
|
|
171
|
+
const record = entry;
|
|
172
|
+
if (record.type !== "message")
|
|
173
|
+
return;
|
|
174
|
+
const message = record.message;
|
|
175
|
+
if (typeof message !== "object" || message === null)
|
|
176
|
+
return;
|
|
177
|
+
const toolResult = message;
|
|
178
|
+
if (toolResult.role !== "toolResult")
|
|
179
|
+
return;
|
|
180
|
+
const nested = toolResult.nestedCalls;
|
|
181
|
+
if (typeof nested !== "object" || nested === null)
|
|
182
|
+
return;
|
|
183
|
+
const calls = nested.calls;
|
|
184
|
+
if (!Array.isArray(calls))
|
|
185
|
+
return;
|
|
186
|
+
for (const call of calls) {
|
|
187
|
+
if (typeof call !== "object" || call === null)
|
|
188
|
+
continue;
|
|
189
|
+
const saved = call;
|
|
190
|
+
if (typeof saved.id !== "string")
|
|
191
|
+
continue;
|
|
192
|
+
if (saved.arguments !== undefined) {
|
|
193
|
+
this.#remember(saved.id, { args: saved.arguments, truncated: false });
|
|
194
|
+
}
|
|
195
|
+
else if (typeof saved.argumentsBytes === "number") {
|
|
196
|
+
this.#remember(saved.id, {
|
|
197
|
+
args: undefined,
|
|
198
|
+
truncated: true,
|
|
199
|
+
bytes: saved.argumentsBytes,
|
|
200
|
+
});
|
|
201
|
+
}
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
/**
|
|
205
|
+
* Extend the index to the current leaf. An append walks only the new entries up to the previous
|
|
206
|
+
* leaf; a fork, branch switch or reload has no such chain and rebuilds. A render never scans the
|
|
207
|
+
* whole branch once the index is current, and a read failure is not cached as an empty answer.
|
|
208
|
+
*/
|
|
209
|
+
#refresh(manager) {
|
|
210
|
+
const leaf = manager.getLeafId();
|
|
211
|
+
if (leaf === this.#indexedLeaf)
|
|
212
|
+
return;
|
|
213
|
+
if (leaf !== null && this.#indexedLeaf !== undefined) {
|
|
214
|
+
const chain = [];
|
|
215
|
+
let current = leaf;
|
|
216
|
+
while (current !== null &&
|
|
217
|
+
current !== this.#indexedLeaf &&
|
|
218
|
+
chain.length <= MAX_INDEXED_CALLS) {
|
|
219
|
+
chain.push(current);
|
|
220
|
+
current = manager.getEntry(current)?.parentId ?? null;
|
|
221
|
+
}
|
|
222
|
+
if (current === this.#indexedLeaf) {
|
|
223
|
+
for (const id of chain.reverse())
|
|
224
|
+
this.#scan(manager.getEntry(id));
|
|
225
|
+
this.#indexedLeaf = leaf;
|
|
226
|
+
return;
|
|
227
|
+
}
|
|
228
|
+
}
|
|
229
|
+
this.#argsById.clear();
|
|
230
|
+
for (const entry of manager.getBranch())
|
|
231
|
+
this.#scan(entry);
|
|
232
|
+
this.#indexedLeaf = leaf;
|
|
233
|
+
}
|
|
96
234
|
}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@hheei/pi-ext-ui",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.7.0",
|
|
4
4
|
"description": "Collapsible tool rendering frame and tool renderer pilot for Pi.",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"main": "dist/index.js",
|
|
@@ -36,7 +36,7 @@
|
|
|
36
36
|
"access": "public"
|
|
37
37
|
},
|
|
38
38
|
"dependencies": {
|
|
39
|
-
"@hheei/pi-ext-core": "^0.7.
|
|
39
|
+
"@hheei/pi-ext-core": "^0.7.1"
|
|
40
40
|
},
|
|
41
41
|
"peerDependencies": {
|
|
42
42
|
"@earendil-works/pi-coding-agent": ">=1.0.1",
|