@aexol/spectral 0.7.1 → 0.7.5
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 +5 -0
- package/dist/agent/agents.js +1 -1
- package/dist/agent/index.js +199 -184
- package/dist/commands/serve.js +0 -3
- package/dist/designer/data/systems/renault/DESIGN.md +1 -1
- package/dist/designer/philosophies.js +668 -0
- package/dist/mcp/sampling-handler.js +1 -1
- package/dist/memory/commands/status.js +1 -1
- package/dist/memory/compaction.js +2 -2
- package/dist/memory/config.js +1 -1
- package/dist/memory/debug-log.js +1 -1
- package/dist/memory/hooks/compaction-hook.js +29 -0
- package/dist/memory/index.js +2 -0
- package/dist/memory/observer.js +2 -2
- package/dist/memory/project-observations-store.js +14 -0
- package/dist/memory/tokens.js +1 -1
- package/dist/memory/tools/read-project-observations.js +82 -0
- package/dist/memory/tools/recall-observation.js +2 -2
- package/dist/pi/agent-core/agent-loop.js +501 -0
- package/dist/pi/agent-core/agent.js +401 -0
- package/dist/pi/agent-core/harness/agent-harness.js +899 -0
- package/dist/pi/agent-core/harness/compaction/branch-summarization.js +173 -0
- package/dist/pi/agent-core/harness/compaction/compaction.js +532 -0
- package/dist/pi/agent-core/harness/compaction/utils.js +130 -0
- package/dist/pi/agent-core/harness/env/nodejs.js +485 -0
- package/dist/pi/agent-core/harness/messages.js +101 -0
- package/dist/pi/agent-core/harness/prompt-templates.js +229 -0
- package/dist/pi/agent-core/harness/session/jsonl-repo.js +100 -0
- package/dist/pi/agent-core/harness/session/jsonl-storage.js +230 -0
- package/dist/pi/agent-core/harness/session/memory-repo.js +41 -0
- package/dist/pi/agent-core/harness/session/memory-storage.js +113 -0
- package/dist/pi/agent-core/harness/session/repo-utils.js +38 -0
- package/dist/pi/agent-core/harness/session/session.js +196 -0
- package/dist/pi/agent-core/harness/session/uuid.js +49 -0
- package/dist/pi/agent-core/harness/skills.js +310 -0
- package/dist/pi/agent-core/harness/system-prompt.js +29 -0
- package/dist/pi/agent-core/harness/types.js +93 -0
- package/dist/pi/agent-core/harness/utils/shell-output.js +125 -0
- package/dist/pi/agent-core/harness/utils/truncate.js +289 -0
- package/dist/pi/agent-core/index.js +24 -0
- package/dist/pi/agent-core/node.js +2 -0
- package/dist/pi/agent-core/proxy.js +277 -0
- package/dist/pi/agent-core/types.js +1 -0
- package/dist/pi/ai/api-registry.js +43 -0
- package/dist/pi/ai/cli.js +120 -0
- package/dist/pi/ai/env-api-keys.js +169 -0
- package/dist/pi/ai/image-models.generated.js +441 -0
- package/dist/pi/ai/image-models.js +22 -0
- package/dist/pi/ai/images-api-registry.js +21 -0
- package/dist/pi/ai/images.js +13 -0
- package/dist/pi/ai/index.js +18 -0
- package/dist/pi/ai/models.generated.js +16220 -0
- package/dist/pi/ai/models.js +70 -0
- package/dist/pi/ai/oauth.js +1 -0
- package/dist/pi/ai/providers/anthropic.js +945 -0
- package/dist/pi/ai/providers/faux.js +367 -0
- package/dist/pi/ai/providers/github-copilot-headers.js +28 -0
- package/dist/pi/ai/providers/openai-completions.js +945 -0
- package/dist/pi/ai/providers/openai-prompt-cache.js +9 -0
- package/dist/pi/ai/providers/register-builtins.js +97 -0
- package/dist/pi/ai/providers/simple-options.js +40 -0
- package/dist/pi/ai/providers/transform-messages.js +183 -0
- package/dist/pi/ai/session-resources.js +21 -0
- package/dist/pi/ai/stream.js +26 -0
- package/dist/pi/ai/types.js +1 -0
- package/dist/pi/ai/utils/diagnostics.js +24 -0
- package/dist/pi/ai/utils/event-stream.js +80 -0
- package/dist/pi/ai/utils/hash.js +13 -0
- package/dist/pi/ai/utils/headers.js +7 -0
- package/dist/pi/ai/utils/json-parse.js +112 -0
- package/dist/pi/ai/utils/node-http-proxy.js +96 -0
- package/dist/pi/ai/utils/oauth/anthropic.js +334 -0
- package/dist/pi/ai/utils/oauth/device-code.js +54 -0
- package/dist/pi/ai/utils/oauth/github-copilot.js +270 -0
- package/dist/pi/ai/utils/oauth/index.js +121 -0
- package/dist/pi/ai/utils/oauth/oauth-page.js +104 -0
- package/dist/pi/ai/utils/oauth/openai-codex.js +384 -0
- package/dist/pi/ai/utils/oauth/pkce.js +30 -0
- package/dist/pi/ai/utils/oauth/types.js +1 -0
- package/dist/pi/ai/utils/overflow.js +150 -0
- package/dist/pi/ai/utils/sanitize-unicode.js +25 -0
- package/dist/pi/ai/utils/typebox-helpers.js +20 -0
- package/dist/pi/ai/utils/validation.js +280 -0
- package/dist/pi/coding-agent/bun/cli.js +7 -0
- package/dist/pi/coding-agent/bun/restore-sandbox-env.js +31 -0
- package/dist/pi/coding-agent/cli/args.js +340 -0
- package/dist/pi/coding-agent/cli/file-processor.js +82 -0
- package/dist/pi/coding-agent/cli/initial-message.js +21 -0
- package/dist/pi/coding-agent/cli.js +17 -0
- package/dist/pi/coding-agent/config.js +414 -0
- package/dist/pi/coding-agent/core/agent-session-runtime.js +299 -0
- package/dist/pi/coding-agent/core/agent-session-services.js +117 -0
- package/dist/pi/coding-agent/core/agent-session.js +2498 -0
- package/dist/pi/coding-agent/core/auth-guidance.js +20 -0
- package/dist/pi/coding-agent/core/auth-storage.js +441 -0
- package/dist/pi/coding-agent/core/bash-executor.js +110 -0
- package/dist/pi/coding-agent/core/compaction/branch-summarization.js +242 -0
- package/dist/pi/coding-agent/core/compaction/compaction.js +624 -0
- package/dist/pi/coding-agent/core/compaction/index.js +6 -0
- package/dist/pi/coding-agent/core/compaction/utils.js +152 -0
- package/dist/pi/coding-agent/core/defaults.js +1 -0
- package/dist/pi/coding-agent/core/diagnostics.js +1 -0
- package/dist/pi/coding-agent/core/event-bus.js +24 -0
- package/dist/pi/coding-agent/core/exec.js +74 -0
- package/dist/pi/coding-agent/core/export-html/ansi-to-html.js +248 -0
- package/dist/pi/coding-agent/core/export-html/index.js +225 -0
- package/dist/pi/coding-agent/core/export-html/tool-renderer.js +107 -0
- package/dist/pi/coding-agent/core/extensions/index.js +8 -0
- package/dist/pi/coding-agent/core/extensions/loader.js +485 -0
- package/dist/pi/coding-agent/core/extensions/runner.js +824 -0
- package/dist/pi/coding-agent/core/extensions/types.js +44 -0
- package/dist/pi/coding-agent/core/extensions/wrapper.js +21 -0
- package/dist/pi/coding-agent/core/footer-data-provider.js +309 -0
- package/dist/pi/coding-agent/core/http-dispatcher.js +47 -0
- package/dist/pi/coding-agent/core/index.js +11 -0
- package/dist/pi/coding-agent/core/keybindings.js +294 -0
- package/dist/pi/coding-agent/core/messages.js +122 -0
- package/dist/pi/coding-agent/core/model-registry.js +728 -0
- package/dist/pi/coding-agent/core/model-resolver.js +494 -0
- package/dist/pi/coding-agent/core/output-guard.js +58 -0
- package/dist/pi/coding-agent/core/package-manager.js +2020 -0
- package/dist/pi/coding-agent/core/prompt-templates.js +237 -0
- package/dist/pi/coding-agent/core/provider-display-names.js +32 -0
- package/dist/pi/coding-agent/core/resolve-config-value.js +125 -0
- package/dist/pi/coding-agent/core/resource-loader.js +733 -0
- package/dist/pi/coding-agent/core/sdk.js +282 -0
- package/dist/pi/coding-agent/core/session-cwd.js +37 -0
- package/dist/pi/coding-agent/core/session-manager.js +1146 -0
- package/dist/pi/coding-agent/core/settings-manager.js +794 -0
- package/dist/pi/coding-agent/core/skills.js +386 -0
- package/dist/pi/coding-agent/core/slash-commands.js +24 -0
- package/dist/pi/coding-agent/core/source-info.js +18 -0
- package/dist/pi/coding-agent/core/system-prompt.js +122 -0
- package/dist/pi/coding-agent/core/telemetry.js +8 -0
- package/dist/pi/coding-agent/core/timings.js +30 -0
- package/dist/pi/coding-agent/core/tools/bash.js +341 -0
- package/dist/pi/coding-agent/core/tools/edit-diff.js +344 -0
- package/dist/pi/coding-agent/core/tools/edit.js +324 -0
- package/dist/pi/coding-agent/core/tools/file-mutation-queue.js +36 -0
- package/dist/pi/coding-agent/core/tools/find.js +297 -0
- package/dist/pi/coding-agent/core/tools/grep.js +303 -0
- package/dist/pi/coding-agent/core/tools/index.js +111 -0
- package/dist/pi/coding-agent/core/tools/ls.js +168 -0
- package/dist/pi/coding-agent/core/tools/output-accumulator.js +183 -0
- package/dist/pi/coding-agent/core/tools/path-utils.js +61 -0
- package/dist/pi/coding-agent/core/tools/read.js +288 -0
- package/dist/pi/coding-agent/core/tools/render-utils.js +48 -0
- package/dist/pi/coding-agent/core/tools/tool-definition-wrapper.js +33 -0
- package/dist/pi/coding-agent/core/tools/truncate.js +214 -0
- package/dist/pi/coding-agent/core/tools/write.js +212 -0
- package/dist/pi/coding-agent/index.js +41 -0
- package/dist/pi/coding-agent/main.js +5 -0
- package/dist/pi/coding-agent/migrations.js +280 -0
- package/dist/pi/coding-agent/modes/index.js +7 -0
- package/dist/pi/coding-agent/modes/interactive/components/diff.js +132 -0
- package/dist/pi/coding-agent/modes/interactive/components/keybinding-hints.js +35 -0
- package/dist/pi/coding-agent/modes/interactive/components/visual-truncate.js +32 -0
- package/dist/pi/coding-agent/modes/interactive/interactive-mode.js +3 -0
- package/dist/pi/coding-agent/modes/interactive/theme/theme.js +1023 -0
- package/dist/pi/coding-agent/modes/print-mode.js +130 -0
- package/dist/pi/coding-agent/modes/rpc/jsonl.js +48 -0
- package/dist/pi/coding-agent/modes/rpc/rpc-client.js +409 -0
- package/dist/pi/coding-agent/modes/rpc/rpc-mode.js +600 -0
- package/dist/pi/coding-agent/modes/rpc/rpc-types.js +7 -0
- package/dist/pi/coding-agent/utils/ansi.js +51 -0
- package/dist/pi/coding-agent/utils/changelog.js +86 -0
- package/dist/pi/coding-agent/utils/child-process.js +87 -0
- package/dist/pi/coding-agent/utils/clipboard-image.js +244 -0
- package/dist/pi/coding-agent/utils/clipboard-native.js +13 -0
- package/dist/pi/coding-agent/utils/clipboard.js +116 -0
- package/dist/pi/coding-agent/utils/exif-orientation.js +157 -0
- package/dist/pi/coding-agent/utils/frontmatter.js +25 -0
- package/dist/pi/coding-agent/utils/fs-watch.js +24 -0
- package/dist/pi/coding-agent/utils/git.js +162 -0
- package/dist/pi/coding-agent/utils/html.js +39 -0
- package/dist/pi/coding-agent/utils/image-convert.js +38 -0
- package/dist/pi/coding-agent/utils/image-resize.js +136 -0
- package/dist/pi/coding-agent/utils/mime.js +68 -0
- package/dist/pi/coding-agent/utils/paths.js +91 -0
- package/dist/pi/coding-agent/utils/photon.js +120 -0
- package/dist/pi/coding-agent/utils/pi-user-agent.js +4 -0
- package/dist/pi/coding-agent/utils/shell.js +194 -0
- package/dist/pi/coding-agent/utils/sleep.js +16 -0
- package/dist/pi/coding-agent/utils/syntax-highlight.js +117 -0
- package/dist/pi/coding-agent/utils/tools-manager.js +327 -0
- package/dist/pi/coding-agent/utils/version-check.js +81 -0
- package/dist/pi/coding-agent/utils/windows-self-update.js +76 -0
- package/dist/pi/tui/autocomplete.js +631 -0
- package/dist/pi/tui/components/box.js +103 -0
- package/dist/pi/tui/components/cancellable-loader.js +34 -0
- package/dist/pi/tui/components/editor.js +1915 -0
- package/dist/pi/tui/components/image.js +88 -0
- package/dist/pi/tui/components/input.js +425 -0
- package/dist/pi/tui/components/loader.js +68 -0
- package/dist/pi/tui/components/markdown.js +633 -0
- package/dist/pi/tui/components/select-list.js +158 -0
- package/dist/pi/tui/components/settings-list.js +184 -0
- package/dist/pi/tui/components/spacer.js +22 -0
- package/dist/pi/tui/components/text.js +88 -0
- package/dist/pi/tui/components/truncated-text.js +50 -0
- package/dist/pi/tui/editor-component.js +1 -0
- package/dist/pi/tui/fuzzy.js +109 -0
- package/dist/pi/tui/index.js +31 -0
- package/dist/pi/tui/keybindings.js +173 -0
- package/dist/pi/tui/keys.js +1172 -0
- package/dist/pi/tui/kill-ring.js +43 -0
- package/dist/pi/tui/stdin-buffer.js +360 -0
- package/dist/pi/tui/terminal-image.js +335 -0
- package/dist/pi/tui/terminal.js +324 -0
- package/dist/pi/tui/tui.js +1076 -0
- package/dist/pi/tui/undo-stack.js +24 -0
- package/dist/pi/tui/utils.js +1016 -0
- package/dist/relay/dispatcher.js +30 -0
- package/dist/server/handlers/queue.js +52 -0
- package/dist/server/pi-bridge.js +9 -1
- package/dist/server/session-stream.js +76 -111
- package/dist/server/storage.js +154 -2
- package/dist/server/title-generator.js +14 -153
- package/package.json +24 -6
|
@@ -0,0 +1,183 @@
|
|
|
1
|
+
import { randomBytes } from "node:crypto";
|
|
2
|
+
import { createWriteStream } from "node:fs";
|
|
3
|
+
import { tmpdir } from "node:os";
|
|
4
|
+
import { join } from "node:path";
|
|
5
|
+
import { DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES, truncateTail } from "./truncate.js";
|
|
6
|
+
function defaultTempFilePath(prefix) {
|
|
7
|
+
const id = randomBytes(8).toString("hex");
|
|
8
|
+
return join(tmpdir(), `${prefix}-${id}.log`);
|
|
9
|
+
}
|
|
10
|
+
function byteLength(text) {
|
|
11
|
+
return Buffer.byteLength(text, "utf-8");
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* Incrementally tracks streaming output with bounded memory.
|
|
15
|
+
*
|
|
16
|
+
* Appends decode chunks with a streaming UTF-8 decoder, keeps only a decoded
|
|
17
|
+
* tail for display snapshots, and opens a temp file when the full output needs
|
|
18
|
+
* to be preserved.
|
|
19
|
+
*/
|
|
20
|
+
export class OutputAccumulator {
|
|
21
|
+
maxLines;
|
|
22
|
+
maxBytes;
|
|
23
|
+
maxRollingBytes;
|
|
24
|
+
tempFilePrefix;
|
|
25
|
+
decoder = new TextDecoder();
|
|
26
|
+
rawChunks = [];
|
|
27
|
+
tailText = "";
|
|
28
|
+
tailBytes = 0;
|
|
29
|
+
tailStartsAtLineBoundary = true;
|
|
30
|
+
totalRawBytes = 0;
|
|
31
|
+
totalDecodedBytes = 0;
|
|
32
|
+
completedLines = 0;
|
|
33
|
+
totalLines = 0;
|
|
34
|
+
currentLineBytes = 0;
|
|
35
|
+
hasOpenLine = false;
|
|
36
|
+
finished = false;
|
|
37
|
+
tempFilePath;
|
|
38
|
+
tempFileStream;
|
|
39
|
+
constructor(options = {}) {
|
|
40
|
+
this.maxLines = options.maxLines ?? DEFAULT_MAX_LINES;
|
|
41
|
+
this.maxBytes = options.maxBytes ?? DEFAULT_MAX_BYTES;
|
|
42
|
+
this.maxRollingBytes = Math.max(this.maxBytes * 2, 1);
|
|
43
|
+
this.tempFilePrefix = options.tempFilePrefix ?? "pi-output";
|
|
44
|
+
}
|
|
45
|
+
append(data) {
|
|
46
|
+
if (this.finished) {
|
|
47
|
+
throw new Error("Cannot append to a finished output accumulator");
|
|
48
|
+
}
|
|
49
|
+
this.totalRawBytes += data.length;
|
|
50
|
+
this.appendDecodedText(this.decoder.decode(data, { stream: true }));
|
|
51
|
+
if (this.tempFileStream || this.shouldUseTempFile()) {
|
|
52
|
+
this.ensureTempFile();
|
|
53
|
+
this.tempFileStream?.write(data);
|
|
54
|
+
}
|
|
55
|
+
else if (data.length > 0) {
|
|
56
|
+
this.rawChunks.push(data);
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
finish() {
|
|
60
|
+
if (this.finished) {
|
|
61
|
+
return;
|
|
62
|
+
}
|
|
63
|
+
this.finished = true;
|
|
64
|
+
this.appendDecodedText(this.decoder.decode());
|
|
65
|
+
if (this.shouldUseTempFile()) {
|
|
66
|
+
this.ensureTempFile();
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
snapshot(options = {}) {
|
|
70
|
+
const tailTruncation = truncateTail(this.getSnapshotText(), {
|
|
71
|
+
maxLines: this.maxLines,
|
|
72
|
+
maxBytes: this.maxBytes,
|
|
73
|
+
});
|
|
74
|
+
const truncated = this.totalLines > this.maxLines || this.totalDecodedBytes > this.maxBytes;
|
|
75
|
+
const truncatedBy = truncated
|
|
76
|
+
? (tailTruncation.truncatedBy ?? (this.totalDecodedBytes > this.maxBytes ? "bytes" : "lines"))
|
|
77
|
+
: null;
|
|
78
|
+
const truncation = {
|
|
79
|
+
...tailTruncation,
|
|
80
|
+
truncated,
|
|
81
|
+
truncatedBy,
|
|
82
|
+
totalLines: this.totalLines,
|
|
83
|
+
totalBytes: this.totalDecodedBytes,
|
|
84
|
+
maxLines: this.maxLines,
|
|
85
|
+
maxBytes: this.maxBytes,
|
|
86
|
+
};
|
|
87
|
+
if (options.persistIfTruncated && truncation.truncated) {
|
|
88
|
+
this.ensureTempFile();
|
|
89
|
+
}
|
|
90
|
+
return {
|
|
91
|
+
content: truncation.content,
|
|
92
|
+
truncation,
|
|
93
|
+
fullOutputPath: this.tempFilePath,
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
async closeTempFile() {
|
|
97
|
+
if (!this.tempFileStream) {
|
|
98
|
+
return;
|
|
99
|
+
}
|
|
100
|
+
const stream = this.tempFileStream;
|
|
101
|
+
this.tempFileStream = undefined;
|
|
102
|
+
await new Promise((resolve, reject) => {
|
|
103
|
+
const onError = (error) => {
|
|
104
|
+
stream.off("finish", onFinish);
|
|
105
|
+
reject(error);
|
|
106
|
+
};
|
|
107
|
+
const onFinish = () => {
|
|
108
|
+
stream.off("error", onError);
|
|
109
|
+
resolve();
|
|
110
|
+
};
|
|
111
|
+
stream.once("error", onError);
|
|
112
|
+
stream.once("finish", onFinish);
|
|
113
|
+
stream.end();
|
|
114
|
+
});
|
|
115
|
+
}
|
|
116
|
+
getLastLineBytes() {
|
|
117
|
+
return this.currentLineBytes;
|
|
118
|
+
}
|
|
119
|
+
appendDecodedText(text) {
|
|
120
|
+
if (text.length === 0) {
|
|
121
|
+
return;
|
|
122
|
+
}
|
|
123
|
+
const bytes = byteLength(text);
|
|
124
|
+
this.totalDecodedBytes += bytes;
|
|
125
|
+
this.tailText += text;
|
|
126
|
+
this.tailBytes += bytes;
|
|
127
|
+
if (this.tailBytes > this.maxRollingBytes * 2) {
|
|
128
|
+
this.trimTail();
|
|
129
|
+
}
|
|
130
|
+
let newlines = 0;
|
|
131
|
+
let lastNewline = -1;
|
|
132
|
+
for (let i = text.indexOf("\n"); i !== -1; i = text.indexOf("\n", i + 1)) {
|
|
133
|
+
newlines++;
|
|
134
|
+
lastNewline = i;
|
|
135
|
+
}
|
|
136
|
+
if (newlines === 0) {
|
|
137
|
+
this.currentLineBytes += bytes;
|
|
138
|
+
this.hasOpenLine = true;
|
|
139
|
+
}
|
|
140
|
+
else {
|
|
141
|
+
this.completedLines += newlines;
|
|
142
|
+
const tail = text.slice(lastNewline + 1);
|
|
143
|
+
this.currentLineBytes = byteLength(tail);
|
|
144
|
+
this.hasOpenLine = tail.length > 0;
|
|
145
|
+
}
|
|
146
|
+
this.totalLines = this.completedLines + (this.hasOpenLine ? 1 : 0);
|
|
147
|
+
}
|
|
148
|
+
trimTail() {
|
|
149
|
+
const buffer = Buffer.from(this.tailText, "utf-8");
|
|
150
|
+
if (buffer.length <= this.maxRollingBytes) {
|
|
151
|
+
this.tailBytes = buffer.length;
|
|
152
|
+
return;
|
|
153
|
+
}
|
|
154
|
+
let start = buffer.length - this.maxRollingBytes;
|
|
155
|
+
while (start < buffer.length && (buffer[start] & 0xc0) === 0x80) {
|
|
156
|
+
start++;
|
|
157
|
+
}
|
|
158
|
+
this.tailStartsAtLineBoundary = start === 0 ? this.tailStartsAtLineBoundary : buffer[start - 1] === 0x0a;
|
|
159
|
+
this.tailText = buffer.subarray(start).toString("utf-8");
|
|
160
|
+
this.tailBytes = byteLength(this.tailText);
|
|
161
|
+
}
|
|
162
|
+
getSnapshotText() {
|
|
163
|
+
if (this.tailStartsAtLineBoundary) {
|
|
164
|
+
return this.tailText;
|
|
165
|
+
}
|
|
166
|
+
const firstNewline = this.tailText.indexOf("\n");
|
|
167
|
+
return firstNewline === -1 ? this.tailText : this.tailText.slice(firstNewline + 1);
|
|
168
|
+
}
|
|
169
|
+
shouldUseTempFile() {
|
|
170
|
+
return (this.totalRawBytes > this.maxBytes || this.totalDecodedBytes > this.maxBytes || this.totalLines > this.maxLines);
|
|
171
|
+
}
|
|
172
|
+
ensureTempFile() {
|
|
173
|
+
if (this.tempFilePath) {
|
|
174
|
+
return;
|
|
175
|
+
}
|
|
176
|
+
this.tempFilePath = defaultTempFilePath(this.tempFilePrefix);
|
|
177
|
+
this.tempFileStream = createWriteStream(this.tempFilePath);
|
|
178
|
+
for (const chunk of this.rawChunks) {
|
|
179
|
+
this.tempFileStream.write(chunk);
|
|
180
|
+
}
|
|
181
|
+
this.rawChunks = [];
|
|
182
|
+
}
|
|
183
|
+
}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { accessSync, constants } from "node:fs";
|
|
2
|
+
import { normalizePath, resolvePath } from "../../utils/paths.js";
|
|
3
|
+
const NARROW_NO_BREAK_SPACE = "\u202F";
|
|
4
|
+
function tryMacOSScreenshotPath(filePath) {
|
|
5
|
+
return filePath.replace(/ (AM|PM)\./gi, `${NARROW_NO_BREAK_SPACE}$1.`);
|
|
6
|
+
}
|
|
7
|
+
function tryNFDVariant(filePath) {
|
|
8
|
+
// macOS stores filenames in NFD (decomposed) form, try converting user input to NFD
|
|
9
|
+
return filePath.normalize("NFD");
|
|
10
|
+
}
|
|
11
|
+
function tryCurlyQuoteVariant(filePath) {
|
|
12
|
+
// macOS uses U+2019 (right single quotation mark) in screenshot names like "Capture d'écran"
|
|
13
|
+
// Users typically type U+0027 (straight apostrophe)
|
|
14
|
+
return filePath.replace(/'/g, "\u2019");
|
|
15
|
+
}
|
|
16
|
+
function fileExists(filePath) {
|
|
17
|
+
try {
|
|
18
|
+
accessSync(filePath, constants.F_OK);
|
|
19
|
+
return true;
|
|
20
|
+
}
|
|
21
|
+
catch {
|
|
22
|
+
return false;
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
export function expandPath(filePath) {
|
|
26
|
+
return normalizePath(filePath, { normalizeUnicodeSpaces: true, stripAtPrefix: true });
|
|
27
|
+
}
|
|
28
|
+
/**
|
|
29
|
+
* Resolve a path relative to the given cwd.
|
|
30
|
+
* Handles ~ expansion and absolute paths.
|
|
31
|
+
*/
|
|
32
|
+
export function resolveToCwd(filePath, cwd) {
|
|
33
|
+
return resolvePath(filePath, cwd, { normalizeUnicodeSpaces: true, stripAtPrefix: true });
|
|
34
|
+
}
|
|
35
|
+
export function resolveReadPath(filePath, cwd) {
|
|
36
|
+
const resolved = resolveToCwd(filePath, cwd);
|
|
37
|
+
if (fileExists(resolved)) {
|
|
38
|
+
return resolved;
|
|
39
|
+
}
|
|
40
|
+
// Try macOS AM/PM variant (narrow no-break space before AM/PM)
|
|
41
|
+
const amPmVariant = tryMacOSScreenshotPath(resolved);
|
|
42
|
+
if (amPmVariant !== resolved && fileExists(amPmVariant)) {
|
|
43
|
+
return amPmVariant;
|
|
44
|
+
}
|
|
45
|
+
// Try NFD variant (macOS stores filenames in NFD form)
|
|
46
|
+
const nfdVariant = tryNFDVariant(resolved);
|
|
47
|
+
if (nfdVariant !== resolved && fileExists(nfdVariant)) {
|
|
48
|
+
return nfdVariant;
|
|
49
|
+
}
|
|
50
|
+
// Try curly quote variant (macOS uses U+2019 in screenshot names)
|
|
51
|
+
const curlyVariant = tryCurlyQuoteVariant(resolved);
|
|
52
|
+
if (curlyVariant !== resolved && fileExists(curlyVariant)) {
|
|
53
|
+
return curlyVariant;
|
|
54
|
+
}
|
|
55
|
+
// Try combined NFD + curly quote (for French macOS screenshots like "Capture d'écran")
|
|
56
|
+
const nfdCurlyVariant = tryCurlyQuoteVariant(nfdVariant);
|
|
57
|
+
if (nfdCurlyVariant !== resolved && fileExists(nfdCurlyVariant)) {
|
|
58
|
+
return nfdCurlyVariant;
|
|
59
|
+
}
|
|
60
|
+
return resolved;
|
|
61
|
+
}
|
|
@@ -0,0 +1,288 @@
|
|
|
1
|
+
import { basename, dirname, isAbsolute, relative, resolve as resolvePath, sep } from "node:path";
|
|
2
|
+
import { Text } from "../../../tui/index.js";
|
|
3
|
+
import { constants } from "fs";
|
|
4
|
+
import { access as fsAccess, readFile as fsReadFile } from "fs/promises";
|
|
5
|
+
import { Type } from "typebox";
|
|
6
|
+
import { getReadmePath } from "../../config.js";
|
|
7
|
+
import { keyHint, keyText } from "../../modes/interactive/components/keybinding-hints.js";
|
|
8
|
+
import { getLanguageFromPath, highlightCode } from "../../modes/interactive/theme/theme.js";
|
|
9
|
+
import { formatDimensionNote, resizeImage } from "../../utils/image-resize.js";
|
|
10
|
+
import { detectSupportedImageMimeTypeFromFile } from "../../utils/mime.js";
|
|
11
|
+
import { formatPathRelativeToCwdOrAbsolute } from "../../utils/paths.js";
|
|
12
|
+
import { resolveReadPath } from "./path-utils.js";
|
|
13
|
+
import { getTextOutput, invalidArgText, replaceTabs, shortenPath, str } from "./render-utils.js";
|
|
14
|
+
import { wrapToolDefinition } from "./tool-definition-wrapper.js";
|
|
15
|
+
import { DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES, formatSize, truncateHead } from "./truncate.js";
|
|
16
|
+
const readSchema = Type.Object({
|
|
17
|
+
path: Type.String({ description: "Path to the file to read (relative or absolute)" }),
|
|
18
|
+
offset: Type.Optional(Type.Number({ description: "Line number to start reading from (1-indexed)" })),
|
|
19
|
+
limit: Type.Optional(Type.Number({ description: "Maximum number of lines to read" })),
|
|
20
|
+
});
|
|
21
|
+
const COMPACT_RESOURCE_FILE_NAMES = new Set(["AGENTS.md", "AGENTS.MD", "CLAUDE.md", "CLAUDE.MD"]);
|
|
22
|
+
const defaultReadOperations = {
|
|
23
|
+
readFile: (path) => fsReadFile(path),
|
|
24
|
+
access: (path) => fsAccess(path, constants.R_OK),
|
|
25
|
+
detectImageMimeType: detectSupportedImageMimeTypeFromFile,
|
|
26
|
+
};
|
|
27
|
+
function formatReadLineRange(args, theme) {
|
|
28
|
+
if (args?.offset === undefined && args?.limit === undefined)
|
|
29
|
+
return "";
|
|
30
|
+
const startLine = args.offset ?? 1;
|
|
31
|
+
const endLine = args.limit !== undefined ? startLine + args.limit - 1 : "";
|
|
32
|
+
return theme.fg("warning", `:${startLine}${endLine ? `-${endLine}` : ""}`);
|
|
33
|
+
}
|
|
34
|
+
function formatReadCall(args, theme) {
|
|
35
|
+
const rawPath = str(args?.file_path ?? args?.path);
|
|
36
|
+
const path = rawPath !== null ? shortenPath(rawPath) : null;
|
|
37
|
+
const invalidArg = invalidArgText(theme);
|
|
38
|
+
const pathDisplay = path === null ? invalidArg : path ? theme.fg("accent", path) : theme.fg("toolOutput", "...");
|
|
39
|
+
return `${theme.fg("toolTitle", theme.bold("read"))} ${pathDisplay}${formatReadLineRange(args, theme)}`;
|
|
40
|
+
}
|
|
41
|
+
function trimTrailingEmptyLines(lines) {
|
|
42
|
+
let end = lines.length;
|
|
43
|
+
while (end > 0 && lines[end - 1] === "") {
|
|
44
|
+
end--;
|
|
45
|
+
}
|
|
46
|
+
return lines.slice(0, end);
|
|
47
|
+
}
|
|
48
|
+
function getNonVisionImageNote(model) {
|
|
49
|
+
if (!model || model.input.includes("image")) {
|
|
50
|
+
return undefined;
|
|
51
|
+
}
|
|
52
|
+
return "[Current model does not support images. The image will be omitted from this request.]";
|
|
53
|
+
}
|
|
54
|
+
function toPosixPath(filePath) {
|
|
55
|
+
return filePath.split(sep).join("/");
|
|
56
|
+
}
|
|
57
|
+
function getPiDocsClassification(absolutePath) {
|
|
58
|
+
const packageRoot = dirname(getReadmePath());
|
|
59
|
+
const relativePath = relative(resolvePath(packageRoot), resolvePath(absolutePath));
|
|
60
|
+
if (relativePath === "" ||
|
|
61
|
+
relativePath === ".." ||
|
|
62
|
+
relativePath.startsWith(`..${sep}`) ||
|
|
63
|
+
isAbsolute(relativePath)) {
|
|
64
|
+
return undefined;
|
|
65
|
+
}
|
|
66
|
+
const label = toPosixPath(relativePath);
|
|
67
|
+
if (label === "README.md" || label.startsWith("docs/") || label.startsWith("examples/")) {
|
|
68
|
+
return { kind: "docs", label };
|
|
69
|
+
}
|
|
70
|
+
return undefined;
|
|
71
|
+
}
|
|
72
|
+
function getCompactReadClassification(args, cwd) {
|
|
73
|
+
const rawPath = str(args?.file_path ?? args?.path);
|
|
74
|
+
if (!rawPath)
|
|
75
|
+
return undefined;
|
|
76
|
+
const absolutePath = resolveReadPath(rawPath, cwd);
|
|
77
|
+
const fileName = basename(absolutePath);
|
|
78
|
+
if (fileName === "SKILL.md") {
|
|
79
|
+
return { kind: "skill", label: basename(dirname(absolutePath)) || fileName };
|
|
80
|
+
}
|
|
81
|
+
const docsClassification = getPiDocsClassification(absolutePath);
|
|
82
|
+
if (docsClassification)
|
|
83
|
+
return docsClassification;
|
|
84
|
+
if (COMPACT_RESOURCE_FILE_NAMES.has(fileName)) {
|
|
85
|
+
return { kind: "resource", label: formatPathRelativeToCwdOrAbsolute(absolutePath, cwd) };
|
|
86
|
+
}
|
|
87
|
+
return undefined;
|
|
88
|
+
}
|
|
89
|
+
function formatCompactReadCall(classification, args, theme) {
|
|
90
|
+
const expandHint = theme.fg("dim", ` (${keyText("app.tools.expand")} to expand)`);
|
|
91
|
+
if (classification.kind === "skill") {
|
|
92
|
+
return (theme.fg("customMessageLabel", `\x1b[1m[skill]\x1b[22m `) +
|
|
93
|
+
theme.fg("customMessageText", classification.label) +
|
|
94
|
+
formatReadLineRange(args, theme) +
|
|
95
|
+
expandHint);
|
|
96
|
+
}
|
|
97
|
+
return (theme.fg("toolTitle", theme.bold(`read ${classification.kind}`)) +
|
|
98
|
+
" " +
|
|
99
|
+
theme.fg("accent", classification.label) +
|
|
100
|
+
formatReadLineRange(args, theme) +
|
|
101
|
+
expandHint);
|
|
102
|
+
}
|
|
103
|
+
function formatReadResult(args, result, options, theme, showImages, cwd, isError) {
|
|
104
|
+
if (!options.expanded && !isError && getCompactReadClassification(args, cwd)) {
|
|
105
|
+
return "";
|
|
106
|
+
}
|
|
107
|
+
const rawPath = str(args?.file_path ?? args?.path);
|
|
108
|
+
const output = getTextOutput(result, showImages);
|
|
109
|
+
const lang = rawPath ? getLanguageFromPath(rawPath) : undefined;
|
|
110
|
+
const renderedLines = lang ? highlightCode(replaceTabs(output), lang) : output.split("\n");
|
|
111
|
+
const lines = trimTrailingEmptyLines(renderedLines);
|
|
112
|
+
const maxLines = options.expanded ? lines.length : 10;
|
|
113
|
+
const displayLines = lines.slice(0, maxLines);
|
|
114
|
+
const remaining = lines.length - maxLines;
|
|
115
|
+
let text = `\n${displayLines.map((line) => (lang ? replaceTabs(line) : theme.fg("toolOutput", replaceTabs(line)))).join("\n")}`;
|
|
116
|
+
if (remaining > 0) {
|
|
117
|
+
text += `${theme.fg("muted", `\n... (${remaining} more lines,`)} ${keyHint("app.tools.expand", "to expand")})`;
|
|
118
|
+
}
|
|
119
|
+
const truncation = result.details?.truncation;
|
|
120
|
+
if (truncation?.truncated) {
|
|
121
|
+
if (truncation.firstLineExceedsLimit) {
|
|
122
|
+
text += `\n${theme.fg("warning", `[First line exceeds ${formatSize(truncation.maxBytes ?? DEFAULT_MAX_BYTES)} limit]`)}`;
|
|
123
|
+
}
|
|
124
|
+
else if (truncation.truncatedBy === "lines") {
|
|
125
|
+
text += `\n${theme.fg("warning", `[Truncated: showing ${truncation.outputLines} of ${truncation.totalLines} lines (${truncation.maxLines ?? DEFAULT_MAX_LINES} line limit)]`)}`;
|
|
126
|
+
}
|
|
127
|
+
else {
|
|
128
|
+
text += `\n${theme.fg("warning", `[Truncated: ${truncation.outputLines} lines shown (${formatSize(truncation.maxBytes ?? DEFAULT_MAX_BYTES)} limit)]`)}`;
|
|
129
|
+
}
|
|
130
|
+
}
|
|
131
|
+
return text;
|
|
132
|
+
}
|
|
133
|
+
export function createReadToolDefinition(cwd, options) {
|
|
134
|
+
const autoResizeImages = options?.autoResizeImages ?? true;
|
|
135
|
+
const ops = options?.operations ?? defaultReadOperations;
|
|
136
|
+
return {
|
|
137
|
+
name: "read",
|
|
138
|
+
label: "read",
|
|
139
|
+
description: `Read the contents of a file. Supports text files and images (jpg, png, gif, webp). Images are sent as attachments. For text files, output is truncated to ${DEFAULT_MAX_LINES} lines or ${DEFAULT_MAX_BYTES / 1024}KB (whichever is hit first). Use offset/limit for large files. When you need the full file, continue with offset until complete.`,
|
|
140
|
+
promptSnippet: "Read file contents",
|
|
141
|
+
promptGuidelines: ["Use read to examine files instead of cat or sed."],
|
|
142
|
+
parameters: readSchema,
|
|
143
|
+
async execute(_toolCallId, { path, offset, limit }, signal, _onUpdate, ctx) {
|
|
144
|
+
const absolutePath = resolveReadPath(path, cwd);
|
|
145
|
+
return new Promise((resolve, reject) => {
|
|
146
|
+
if (signal?.aborted) {
|
|
147
|
+
reject(new Error("Operation aborted"));
|
|
148
|
+
return;
|
|
149
|
+
}
|
|
150
|
+
let aborted = false;
|
|
151
|
+
const onAbort = () => {
|
|
152
|
+
aborted = true;
|
|
153
|
+
reject(new Error("Operation aborted"));
|
|
154
|
+
};
|
|
155
|
+
signal?.addEventListener("abort", onAbort, { once: true });
|
|
156
|
+
(async () => {
|
|
157
|
+
try {
|
|
158
|
+
// Check if file exists and is readable.
|
|
159
|
+
await ops.access(absolutePath);
|
|
160
|
+
if (aborted)
|
|
161
|
+
return;
|
|
162
|
+
const mimeType = ops.detectImageMimeType ? await ops.detectImageMimeType(absolutePath) : undefined;
|
|
163
|
+
let content;
|
|
164
|
+
let details;
|
|
165
|
+
const nonVisionImageNote = getNonVisionImageNote(ctx?.model);
|
|
166
|
+
if (mimeType) {
|
|
167
|
+
// Read image as binary.
|
|
168
|
+
const buffer = await ops.readFile(absolutePath);
|
|
169
|
+
const base64 = buffer.toString("base64");
|
|
170
|
+
if (autoResizeImages) {
|
|
171
|
+
// Resize image if needed before sending it back to the model.
|
|
172
|
+
const resized = await resizeImage({ type: "image", data: base64, mimeType });
|
|
173
|
+
if (!resized) {
|
|
174
|
+
let textNote = `Read image file [${mimeType}]\n[Image omitted: could not be resized below the inline image size limit.]`;
|
|
175
|
+
if (nonVisionImageNote)
|
|
176
|
+
textNote += `\n${nonVisionImageNote}`;
|
|
177
|
+
content = [{ type: "text", text: textNote }];
|
|
178
|
+
}
|
|
179
|
+
else {
|
|
180
|
+
const dimensionNote = formatDimensionNote(resized);
|
|
181
|
+
let textNote = `Read image file [${resized.mimeType}]`;
|
|
182
|
+
if (dimensionNote)
|
|
183
|
+
textNote += `\n${dimensionNote}`;
|
|
184
|
+
if (nonVisionImageNote)
|
|
185
|
+
textNote += `\n${nonVisionImageNote}`;
|
|
186
|
+
content = [
|
|
187
|
+
{ type: "text", text: textNote },
|
|
188
|
+
{ type: "image", data: resized.data, mimeType: resized.mimeType },
|
|
189
|
+
];
|
|
190
|
+
}
|
|
191
|
+
}
|
|
192
|
+
else {
|
|
193
|
+
let textNote = `Read image file [${mimeType}]`;
|
|
194
|
+
if (nonVisionImageNote)
|
|
195
|
+
textNote += `\n${nonVisionImageNote}`;
|
|
196
|
+
content = [
|
|
197
|
+
{ type: "text", text: textNote },
|
|
198
|
+
{ type: "image", data: base64, mimeType },
|
|
199
|
+
];
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
else {
|
|
203
|
+
// Read text content.
|
|
204
|
+
const buffer = await ops.readFile(absolutePath);
|
|
205
|
+
const textContent = buffer.toString("utf-8");
|
|
206
|
+
const allLines = textContent.split("\n");
|
|
207
|
+
const totalFileLines = allLines.length;
|
|
208
|
+
// Apply offset if specified. Convert from 1-indexed input to 0-indexed array access.
|
|
209
|
+
const startLine = offset ? Math.max(0, offset - 1) : 0;
|
|
210
|
+
const startLineDisplay = startLine + 1;
|
|
211
|
+
// Check if offset is out of bounds.
|
|
212
|
+
if (startLine >= allLines.length) {
|
|
213
|
+
throw new Error(`Offset ${offset} is beyond end of file (${allLines.length} lines total)`);
|
|
214
|
+
}
|
|
215
|
+
let selectedContent;
|
|
216
|
+
let userLimitedLines;
|
|
217
|
+
// If limit is specified by the user, honor it first. Otherwise truncateHead decides.
|
|
218
|
+
if (limit !== undefined) {
|
|
219
|
+
const endLine = Math.min(startLine + limit, allLines.length);
|
|
220
|
+
selectedContent = allLines.slice(startLine, endLine).join("\n");
|
|
221
|
+
userLimitedLines = endLine - startLine;
|
|
222
|
+
}
|
|
223
|
+
else {
|
|
224
|
+
selectedContent = allLines.slice(startLine).join("\n");
|
|
225
|
+
}
|
|
226
|
+
// Apply truncation, respecting both line and byte limits.
|
|
227
|
+
const truncation = truncateHead(selectedContent);
|
|
228
|
+
let outputText;
|
|
229
|
+
if (truncation.firstLineExceedsLimit) {
|
|
230
|
+
// First line alone exceeds the byte limit. Point the model at a bash fallback.
|
|
231
|
+
const firstLineSize = formatSize(Buffer.byteLength(allLines[startLine], "utf-8"));
|
|
232
|
+
outputText = `[Line ${startLineDisplay} is ${firstLineSize}, exceeds ${formatSize(DEFAULT_MAX_BYTES)} limit. Use bash: sed -n '${startLineDisplay}p' ${path} | head -c ${DEFAULT_MAX_BYTES}]`;
|
|
233
|
+
details = { truncation };
|
|
234
|
+
}
|
|
235
|
+
else if (truncation.truncated) {
|
|
236
|
+
// Truncation occurred. Build an actionable continuation notice.
|
|
237
|
+
const endLineDisplay = startLineDisplay + truncation.outputLines - 1;
|
|
238
|
+
const nextOffset = endLineDisplay + 1;
|
|
239
|
+
outputText = truncation.content;
|
|
240
|
+
if (truncation.truncatedBy === "lines") {
|
|
241
|
+
outputText += `\n\n[Showing lines ${startLineDisplay}-${endLineDisplay} of ${totalFileLines}. Use offset=${nextOffset} to continue.]`;
|
|
242
|
+
}
|
|
243
|
+
else {
|
|
244
|
+
outputText += `\n\n[Showing lines ${startLineDisplay}-${endLineDisplay} of ${totalFileLines} (${formatSize(DEFAULT_MAX_BYTES)} limit). Use offset=${nextOffset} to continue.]`;
|
|
245
|
+
}
|
|
246
|
+
details = { truncation };
|
|
247
|
+
}
|
|
248
|
+
else if (userLimitedLines !== undefined && startLine + userLimitedLines < allLines.length) {
|
|
249
|
+
// User-specified limit stopped early, but the file still has more content.
|
|
250
|
+
const remaining = allLines.length - (startLine + userLimitedLines);
|
|
251
|
+
const nextOffset = startLine + userLimitedLines + 1;
|
|
252
|
+
outputText = `${truncation.content}\n\n[${remaining} more lines in file. Use offset=${nextOffset} to continue.]`;
|
|
253
|
+
}
|
|
254
|
+
else {
|
|
255
|
+
// No truncation and no remaining user-limited content.
|
|
256
|
+
outputText = truncation.content;
|
|
257
|
+
}
|
|
258
|
+
content = [{ type: "text", text: outputText }];
|
|
259
|
+
}
|
|
260
|
+
if (aborted)
|
|
261
|
+
return;
|
|
262
|
+
signal?.removeEventListener("abort", onAbort);
|
|
263
|
+
resolve({ content, details });
|
|
264
|
+
}
|
|
265
|
+
catch (error) {
|
|
266
|
+
signal?.removeEventListener("abort", onAbort);
|
|
267
|
+
if (!aborted)
|
|
268
|
+
reject(error);
|
|
269
|
+
}
|
|
270
|
+
})();
|
|
271
|
+
});
|
|
272
|
+
},
|
|
273
|
+
renderCall(args, theme, context) {
|
|
274
|
+
const text = context.lastComponent ?? new Text("", 0, 0);
|
|
275
|
+
const classification = !context.expanded ? getCompactReadClassification(args, context.cwd) : undefined;
|
|
276
|
+
text.setText(classification ? formatCompactReadCall(classification, args, theme) : formatReadCall(args, theme));
|
|
277
|
+
return text;
|
|
278
|
+
},
|
|
279
|
+
renderResult(result, options, theme, context) {
|
|
280
|
+
const text = context.lastComponent ?? new Text("", 0, 0);
|
|
281
|
+
text.setText(formatReadResult(context.args, result, options, theme, context.showImages, context.cwd, context.isError));
|
|
282
|
+
return text;
|
|
283
|
+
},
|
|
284
|
+
};
|
|
285
|
+
}
|
|
286
|
+
export function createReadTool(cwd, options) {
|
|
287
|
+
return wrapToolDefinition(createReadToolDefinition(cwd, options));
|
|
288
|
+
}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import * as os from "node:os";
|
|
2
|
+
import { getCapabilities, getImageDimensions, imageFallback } from "../../../tui/index.js";
|
|
3
|
+
import { stripAnsi } from "../../utils/ansi.js";
|
|
4
|
+
import { sanitizeBinaryOutput } from "../../utils/shell.js";
|
|
5
|
+
export function shortenPath(path) {
|
|
6
|
+
if (typeof path !== "string")
|
|
7
|
+
return "";
|
|
8
|
+
const home = os.homedir();
|
|
9
|
+
if (path.startsWith(home)) {
|
|
10
|
+
return `~${path.slice(home.length)}`;
|
|
11
|
+
}
|
|
12
|
+
return path;
|
|
13
|
+
}
|
|
14
|
+
export function str(value) {
|
|
15
|
+
if (typeof value === "string")
|
|
16
|
+
return value;
|
|
17
|
+
if (value == null)
|
|
18
|
+
return "";
|
|
19
|
+
return null;
|
|
20
|
+
}
|
|
21
|
+
export function replaceTabs(text) {
|
|
22
|
+
return text.replace(/\t/g, " ");
|
|
23
|
+
}
|
|
24
|
+
export function normalizeDisplayText(text) {
|
|
25
|
+
return text.replace(/\r/g, "");
|
|
26
|
+
}
|
|
27
|
+
export function getTextOutput(result, showImages) {
|
|
28
|
+
if (!result)
|
|
29
|
+
return "";
|
|
30
|
+
const textBlocks = result.content.filter((c) => c.type === "text");
|
|
31
|
+
const imageBlocks = result.content.filter((c) => c.type === "image");
|
|
32
|
+
let output = textBlocks.map((c) => sanitizeBinaryOutput(stripAnsi(c.text || "")).replace(/\r/g, "")).join("\n");
|
|
33
|
+
const caps = getCapabilities();
|
|
34
|
+
if (imageBlocks.length > 0 && (!caps.images || !showImages)) {
|
|
35
|
+
const imageIndicators = imageBlocks
|
|
36
|
+
.map((img) => {
|
|
37
|
+
const mimeType = img.mimeType ?? "image/unknown";
|
|
38
|
+
const dims = img.data && img.mimeType ? (getImageDimensions(img.data, img.mimeType) ?? undefined) : undefined;
|
|
39
|
+
return imageFallback(mimeType, dims);
|
|
40
|
+
})
|
|
41
|
+
.join("\n");
|
|
42
|
+
output = output ? `${output}\n${imageIndicators}` : imageIndicators;
|
|
43
|
+
}
|
|
44
|
+
return output;
|
|
45
|
+
}
|
|
46
|
+
export function invalidArgText(theme) {
|
|
47
|
+
return theme.fg("error", "[invalid arg]");
|
|
48
|
+
}
|
|
@@ -0,0 +1,33 @@
|
|
|
1
|
+
/** Wrap a ToolDefinition into an AgentTool for the core runtime. */
|
|
2
|
+
export function wrapToolDefinition(definition, ctxFactory) {
|
|
3
|
+
return {
|
|
4
|
+
name: definition.name,
|
|
5
|
+
label: definition.label,
|
|
6
|
+
description: definition.description,
|
|
7
|
+
parameters: definition.parameters,
|
|
8
|
+
prepareArguments: definition.prepareArguments,
|
|
9
|
+
executionMode: definition.executionMode,
|
|
10
|
+
execute: (toolCallId, params, signal, onUpdate) => definition.execute(toolCallId, params, signal, onUpdate, ctxFactory?.()),
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
/** Wrap multiple ToolDefinitions into AgentTools for the core runtime. */
|
|
14
|
+
export function wrapToolDefinitions(definitions, ctxFactory) {
|
|
15
|
+
return definitions.map((definition) => wrapToolDefinition(definition, ctxFactory));
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Synthesize a minimal ToolDefinition from an AgentTool.
|
|
19
|
+
*
|
|
20
|
+
* This keeps AgentSession's internal registry definition-first even when a caller
|
|
21
|
+
* provides plain AgentTool overrides that do not include prompt metadata or renderers.
|
|
22
|
+
*/
|
|
23
|
+
export function createToolDefinitionFromAgentTool(tool) {
|
|
24
|
+
return {
|
|
25
|
+
name: tool.name,
|
|
26
|
+
label: tool.label,
|
|
27
|
+
description: tool.description,
|
|
28
|
+
parameters: tool.parameters,
|
|
29
|
+
prepareArguments: tool.prepareArguments,
|
|
30
|
+
executionMode: tool.executionMode,
|
|
31
|
+
execute: async (toolCallId, params, signal, onUpdate) => tool.execute(toolCallId, params, signal, onUpdate),
|
|
32
|
+
};
|
|
33
|
+
}
|