@felan-ai/ext-rtk-optimizer 0.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +23 -0
- package/NOTICE +14 -0
- package/README.md +115 -0
- package/dist/command-rewriter.d.ts +27 -0
- package/dist/command-rewriter.d.ts.map +1 -0
- package/dist/command-rewriter.js +126 -0
- package/dist/command-rewriter.js.map +1 -0
- package/dist/command.d.ts +12 -0
- package/dist/command.d.ts.map +1 -0
- package/dist/command.js +240 -0
- package/dist/command.js.map +1 -0
- package/dist/config.d.ts +13 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +173 -0
- package/dist/config.js.map +1 -0
- package/dist/index.d.ts +18 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +306 -0
- package/dist/index.js.map +1 -0
- package/dist/output-compactor.d.ts +31 -0
- package/dist/output-compactor.d.ts.map +1 -0
- package/dist/output-compactor.js +455 -0
- package/dist/output-compactor.js.map +1 -0
- package/dist/output-metrics.d.ts +15 -0
- package/dist/output-metrics.d.ts.map +1 -0
- package/dist/output-metrics.js +50 -0
- package/dist/output-metrics.js.map +1 -0
- package/dist/record-utils.d.ts +12 -0
- package/dist/record-utils.d.ts.map +1 -0
- package/dist/record-utils.js +23 -0
- package/dist/record-utils.js.map +1 -0
- package/dist/runtime-path.d.ts +4 -0
- package/dist/runtime-path.d.ts.map +1 -0
- package/dist/runtime-path.js +73 -0
- package/dist/runtime-path.js.map +1 -0
- package/dist/techniques/ansi.d.ts +3 -0
- package/dist/techniques/ansi.d.ts.map +1 -0
- package/dist/techniques/ansi.js +13 -0
- package/dist/techniques/ansi.js.map +1 -0
- package/dist/techniques/build.d.ts +3 -0
- package/dist/techniques/build.d.ts.map +1 -0
- package/dist/techniques/build.js +127 -0
- package/dist/techniques/build.js.map +1 -0
- package/dist/techniques/command-detection.d.ts +3 -0
- package/dist/techniques/command-detection.d.ts.map +1 -0
- package/dist/techniques/command-detection.js +44 -0
- package/dist/techniques/command-detection.js.map +1 -0
- package/dist/techniques/git.d.ts +6 -0
- package/dist/techniques/git.d.ts.map +1 -0
- package/dist/techniques/git.js +188 -0
- package/dist/techniques/git.js.map +1 -0
- package/dist/techniques/index.d.ts +9 -0
- package/dist/techniques/index.d.ts.map +1 -0
- package/dist/techniques/index.js +9 -0
- package/dist/techniques/index.js.map +1 -0
- package/dist/techniques/linter.d.ts +3 -0
- package/dist/techniques/linter.d.ts.map +1 -0
- package/dist/techniques/linter.js +133 -0
- package/dist/techniques/linter.js.map +1 -0
- package/dist/techniques/path-utils.d.ts +2 -0
- package/dist/techniques/path-utils.d.ts.map +1 -0
- package/dist/techniques/path-utils.js +55 -0
- package/dist/techniques/path-utils.js.map +1 -0
- package/dist/techniques/search.d.ts +2 -0
- package/dist/techniques/search.d.ts.map +1 -0
- package/dist/techniques/search.js +53 -0
- package/dist/techniques/search.js.map +1 -0
- package/dist/techniques/source.d.ts +7 -0
- package/dist/techniques/source.d.ts.map +1 -0
- package/dist/techniques/source.js +250 -0
- package/dist/techniques/source.js.map +1 -0
- package/dist/techniques/test-output.d.ts +3 -0
- package/dist/techniques/test-output.d.ts.map +1 -0
- package/dist/techniques/test-output.js +157 -0
- package/dist/techniques/test-output.js.map +1 -0
- package/dist/techniques/truncate.d.ts +2 -0
- package/dist/techniques/truncate.d.ts.map +1 -0
- package/dist/techniques/truncate.js +10 -0
- package/dist/techniques/truncate.js.map +1 -0
- package/dist/tool-execution-sanitizer.d.ts +6 -0
- package/dist/tool-execution-sanitizer.d.ts.map +1 -0
- package/dist/tool-execution-sanitizer.js +20 -0
- package/dist/tool-execution-sanitizer.js.map +1 -0
- package/dist/tool-shapes.d.ts +11 -0
- package/dist/tool-shapes.d.ts.map +1 -0
- package/dist/tool-shapes.js +41 -0
- package/dist/tool-shapes.js.map +1 -0
- package/dist/types.d.ts +43 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +33 -0
- package/dist/types.js.map +1 -0
- package/package.json +42 -0
package/dist/config.js
ADDED
|
@@ -0,0 +1,173 @@
|
|
|
1
|
+
import { joinRuntimePath } from './runtime-path.js';
|
|
2
|
+
import { DEFAULT_RTK_OPTIMIZER_CONFIG, RTK_MODES, RTK_SOURCE_FILTER_LEVELS, } from './types.js';
|
|
3
|
+
export const RTK_OPTIMIZER_CONFIG_DIRECTORY = 'rtk-optimizer';
|
|
4
|
+
export const RTK_OPTIMIZER_CONFIG_FILE = `${RTK_OPTIMIZER_CONFIG_DIRECTORY}/config.json`;
|
|
5
|
+
const encoder = new TextEncoder();
|
|
6
|
+
const decoder = new TextDecoder();
|
|
7
|
+
export async function loadRtkOptimizerConfig(runtime) {
|
|
8
|
+
const storage = runtime.storage('agent');
|
|
9
|
+
let content;
|
|
10
|
+
try {
|
|
11
|
+
content = await storage.readFile(RTK_OPTIMIZER_CONFIG_FILE);
|
|
12
|
+
}
|
|
13
|
+
catch (error) {
|
|
14
|
+
if (!isNotFoundError(error)) {
|
|
15
|
+
return {
|
|
16
|
+
config: cloneDefaultConfig(),
|
|
17
|
+
warning: `Failed to read ${getRtkOptimizerConfigPath(runtime)}: ${errorMessage(error)}`,
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
const config = cloneDefaultConfig();
|
|
21
|
+
try {
|
|
22
|
+
await writeConfig(runtime, config);
|
|
23
|
+
return { config };
|
|
24
|
+
}
|
|
25
|
+
catch (writeError) {
|
|
26
|
+
return {
|
|
27
|
+
config,
|
|
28
|
+
warning: `Failed to create ${getRtkOptimizerConfigPath(runtime)}: ${errorMessage(writeError)}`,
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
try {
|
|
33
|
+
const parsed = JSON.parse(decoder.decode(content));
|
|
34
|
+
return { config: validateRtkOptimizerConfig(parsed, getRtkOptimizerConfigPath(runtime)) };
|
|
35
|
+
}
|
|
36
|
+
catch (error) {
|
|
37
|
+
return {
|
|
38
|
+
config: cloneDefaultConfig(),
|
|
39
|
+
warning: `Invalid ${getRtkOptimizerConfigPath(runtime)}: ${errorMessage(error)}`,
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
export async function saveRtkOptimizerConfig(runtime, config) {
|
|
44
|
+
const normalized = validateRtkOptimizerConfig(config);
|
|
45
|
+
await writeConfig(runtime, normalized);
|
|
46
|
+
return normalized;
|
|
47
|
+
}
|
|
48
|
+
export function getRtkOptimizerConfigPath(runtime) {
|
|
49
|
+
return joinRuntimePath(runtime.storage('agent').root, RTK_OPTIMIZER_CONFIG_FILE);
|
|
50
|
+
}
|
|
51
|
+
export function validateRtkOptimizerConfig(value, source = 'RTK optimizer config') {
|
|
52
|
+
const config = optionalObject(value, source);
|
|
53
|
+
assertKnownKeys(config, source, [
|
|
54
|
+
'enabled',
|
|
55
|
+
'mode',
|
|
56
|
+
'guardWhenRtkMissing',
|
|
57
|
+
'showRewriteNotifications',
|
|
58
|
+
'outputCompaction',
|
|
59
|
+
]);
|
|
60
|
+
const outputSource = optionalObject(config.outputCompaction, `${source}.outputCompaction`);
|
|
61
|
+
assertKnownKeys(outputSource, `${source}.outputCompaction`, [
|
|
62
|
+
'enabled',
|
|
63
|
+
'stripAnsi',
|
|
64
|
+
'readCompaction',
|
|
65
|
+
'truncate',
|
|
66
|
+
'sourceCodeFilteringEnabled',
|
|
67
|
+
'preserveExactSkillReads',
|
|
68
|
+
'sourceCodeFiltering',
|
|
69
|
+
'smartTruncate',
|
|
70
|
+
'aggregateTestOutput',
|
|
71
|
+
'filterBuildOutput',
|
|
72
|
+
'compactGitOutput',
|
|
73
|
+
'aggregateLinterOutput',
|
|
74
|
+
'groupSearchOutput',
|
|
75
|
+
'trackSavings',
|
|
76
|
+
]);
|
|
77
|
+
const readSource = optionalObject(outputSource.readCompaction, `${source}.outputCompaction.readCompaction`);
|
|
78
|
+
assertKnownKeys(readSource, `${source}.outputCompaction.readCompaction`, ['enabled']);
|
|
79
|
+
const truncateSource = optionalObject(outputSource.truncate, `${source}.outputCompaction.truncate`);
|
|
80
|
+
assertKnownKeys(truncateSource, `${source}.outputCompaction.truncate`, ['enabled', 'maxChars']);
|
|
81
|
+
const smartTruncateSource = optionalObject(outputSource.smartTruncate, `${source}.outputCompaction.smartTruncate`);
|
|
82
|
+
assertKnownKeys(smartTruncateSource, `${source}.outputCompaction.smartTruncate`, ['enabled', 'maxLines']);
|
|
83
|
+
const defaults = DEFAULT_RTK_OPTIMIZER_CONFIG;
|
|
84
|
+
const outputDefaults = defaults.outputCompaction;
|
|
85
|
+
const outputCompaction = {
|
|
86
|
+
enabled: optionalBoolean(outputSource.enabled, outputDefaults.enabled, `${source}.outputCompaction.enabled`),
|
|
87
|
+
stripAnsi: optionalBoolean(outputSource.stripAnsi, outputDefaults.stripAnsi, `${source}.outputCompaction.stripAnsi`),
|
|
88
|
+
readCompaction: {
|
|
89
|
+
enabled: optionalBoolean(readSource.enabled, outputDefaults.readCompaction.enabled, `${source}.outputCompaction.readCompaction.enabled`),
|
|
90
|
+
},
|
|
91
|
+
truncate: {
|
|
92
|
+
enabled: optionalBoolean(truncateSource.enabled, outputDefaults.truncate.enabled, `${source}.outputCompaction.truncate.enabled`),
|
|
93
|
+
maxChars: optionalInteger(truncateSource.maxChars, outputDefaults.truncate.maxChars, 1_000, 200_000, `${source}.outputCompaction.truncate.maxChars`),
|
|
94
|
+
},
|
|
95
|
+
sourceCodeFilteringEnabled: optionalBoolean(outputSource.sourceCodeFilteringEnabled, outputDefaults.sourceCodeFilteringEnabled, `${source}.outputCompaction.sourceCodeFilteringEnabled`),
|
|
96
|
+
preserveExactSkillReads: optionalBoolean(outputSource.preserveExactSkillReads, outputDefaults.preserveExactSkillReads, `${source}.outputCompaction.preserveExactSkillReads`),
|
|
97
|
+
sourceCodeFiltering: optionalEnum(outputSource.sourceCodeFiltering, outputDefaults.sourceCodeFiltering, RTK_SOURCE_FILTER_LEVELS, `${source}.outputCompaction.sourceCodeFiltering`),
|
|
98
|
+
smartTruncate: {
|
|
99
|
+
enabled: optionalBoolean(smartTruncateSource.enabled, outputDefaults.smartTruncate.enabled, `${source}.outputCompaction.smartTruncate.enabled`),
|
|
100
|
+
maxLines: optionalInteger(smartTruncateSource.maxLines, outputDefaults.smartTruncate.maxLines, 40, 4_000, `${source}.outputCompaction.smartTruncate.maxLines`),
|
|
101
|
+
},
|
|
102
|
+
aggregateTestOutput: optionalBoolean(outputSource.aggregateTestOutput, outputDefaults.aggregateTestOutput, `${source}.outputCompaction.aggregateTestOutput`),
|
|
103
|
+
filterBuildOutput: optionalBoolean(outputSource.filterBuildOutput, outputDefaults.filterBuildOutput, `${source}.outputCompaction.filterBuildOutput`),
|
|
104
|
+
compactGitOutput: optionalBoolean(outputSource.compactGitOutput, outputDefaults.compactGitOutput, `${source}.outputCompaction.compactGitOutput`),
|
|
105
|
+
aggregateLinterOutput: optionalBoolean(outputSource.aggregateLinterOutput, outputDefaults.aggregateLinterOutput, `${source}.outputCompaction.aggregateLinterOutput`),
|
|
106
|
+
groupSearchOutput: optionalBoolean(outputSource.groupSearchOutput, outputDefaults.groupSearchOutput, `${source}.outputCompaction.groupSearchOutput`),
|
|
107
|
+
trackSavings: optionalBoolean(outputSource.trackSavings, outputDefaults.trackSavings, `${source}.outputCompaction.trackSavings`),
|
|
108
|
+
};
|
|
109
|
+
return {
|
|
110
|
+
enabled: optionalBoolean(config.enabled, defaults.enabled, `${source}.enabled`),
|
|
111
|
+
mode: optionalEnum(config.mode, defaults.mode, RTK_MODES, `${source}.mode`),
|
|
112
|
+
guardWhenRtkMissing: optionalBoolean(config.guardWhenRtkMissing, defaults.guardWhenRtkMissing, `${source}.guardWhenRtkMissing`),
|
|
113
|
+
showRewriteNotifications: optionalBoolean(config.showRewriteNotifications, defaults.showRewriteNotifications, `${source}.showRewriteNotifications`),
|
|
114
|
+
outputCompaction,
|
|
115
|
+
};
|
|
116
|
+
}
|
|
117
|
+
async function writeConfig(runtime, config) {
|
|
118
|
+
const storage = runtime.storage('agent');
|
|
119
|
+
await storage.mkdir(RTK_OPTIMIZER_CONFIG_DIRECTORY, { recursive: true });
|
|
120
|
+
await storage.writeFile(RTK_OPTIMIZER_CONFIG_FILE, encoder.encode(`${JSON.stringify(config, null, 2)}\n`));
|
|
121
|
+
}
|
|
122
|
+
function optionalObject(value, path) {
|
|
123
|
+
if (value === undefined)
|
|
124
|
+
return {};
|
|
125
|
+
if (typeof value !== 'object' || value === null || Array.isArray(value)) {
|
|
126
|
+
throw new Error(`${path} must be an object`);
|
|
127
|
+
}
|
|
128
|
+
return value;
|
|
129
|
+
}
|
|
130
|
+
function optionalBoolean(value, fallback, path) {
|
|
131
|
+
if (value === undefined)
|
|
132
|
+
return fallback;
|
|
133
|
+
if (typeof value !== 'boolean')
|
|
134
|
+
throw new Error(`${path} must be a boolean`);
|
|
135
|
+
return value;
|
|
136
|
+
}
|
|
137
|
+
function optionalInteger(value, fallback, minimum, maximum, path) {
|
|
138
|
+
if (value === undefined)
|
|
139
|
+
return fallback;
|
|
140
|
+
if (!Number.isSafeInteger(value) || value < minimum || value > maximum) {
|
|
141
|
+
throw new Error(`${path} must be an integer from ${minimum} to ${maximum}`);
|
|
142
|
+
}
|
|
143
|
+
return value;
|
|
144
|
+
}
|
|
145
|
+
function optionalEnum(value, fallback, allowed, path) {
|
|
146
|
+
if (value === undefined)
|
|
147
|
+
return fallback;
|
|
148
|
+
if (typeof value !== 'string' || !allowed.includes(value)) {
|
|
149
|
+
throw new Error(`${path} must be ${formatAlternatives(allowed)}`);
|
|
150
|
+
}
|
|
151
|
+
return value;
|
|
152
|
+
}
|
|
153
|
+
function assertKnownKeys(value, path, allowed) {
|
|
154
|
+
const allowedSet = new Set(allowed);
|
|
155
|
+
const unknown = Object.keys(value).find((key) => !allowedSet.has(key));
|
|
156
|
+
if (unknown)
|
|
157
|
+
throw new Error(`${path} contains unknown field: ${unknown}`);
|
|
158
|
+
}
|
|
159
|
+
function formatAlternatives(values) {
|
|
160
|
+
if (values.length < 2)
|
|
161
|
+
return values[0] ?? 'a supported value';
|
|
162
|
+
return `${values.slice(0, -1).join(', ')}, or ${values.at(-1)}`;
|
|
163
|
+
}
|
|
164
|
+
function isNotFoundError(error) {
|
|
165
|
+
return typeof error === 'object' && error !== null && 'code' in error && error.code === 'ENOENT';
|
|
166
|
+
}
|
|
167
|
+
function cloneDefaultConfig() {
|
|
168
|
+
return structuredClone(DEFAULT_RTK_OPTIMIZER_CONFIG);
|
|
169
|
+
}
|
|
170
|
+
function errorMessage(error) {
|
|
171
|
+
return error instanceof Error ? error.message : String(error);
|
|
172
|
+
}
|
|
173
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EACL,4BAA4B,EAC5B,SAAS,EACT,wBAAwB,GAGzB,MAAM,YAAY,CAAC;AAEpB,MAAM,CAAC,MAAM,8BAA8B,GAAG,eAAe,CAAC;AAC9D,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,8BAA8B,cAAc,CAAC;AAEzF,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAClC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;AAOlC,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,OAAqB;IAChE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACzC,IAAI,OAAmB,CAAC;IACxB,IAAI,CAAC;QACH,OAAO,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO;gBACL,MAAM,EAAE,kBAAkB,EAAE;gBAC5B,OAAO,EAAE,kBAAkB,yBAAyB,CAAC,OAAO,CAAC,KAAK,YAAY,CAAC,KAAK,CAAC,EAAE;aACxF,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,kBAAkB,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YACnC,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,UAAU,EAAE,CAAC;YACpB,OAAO;gBACL,MAAM;gBACN,OAAO,EAAE,oBAAoB,yBAAyB,CAAC,OAAO,CAAC,KAAK,YAAY,CAAC,UAAU,CAAC,EAAE;aAC/F,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAY,CAAC;QAC9D,OAAO,EAAE,MAAM,EAAE,0BAA0B,CAAC,MAAM,EAAE,yBAAyB,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;IAC5F,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,MAAM,EAAE,kBAAkB,EAAE;YAC5B,OAAO,EAAE,WAAW,yBAAyB,CAAC,OAAO,CAAC,KAAK,YAAY,CAAC,KAAK,CAAC,EAAE;SACjF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,OAAqB,EACrB,MAA0B;IAE1B,MAAM,UAAU,GAAG,0BAA0B,CAAC,MAAM,CAAC,CAAC;IACtD,MAAM,WAAW,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACvC,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,OAAqB;IAC7D,OAAO,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;AACnF,CAAC;AAED,MAAM,UAAU,0BAA0B,CAAC,KAAc,EAAE,MAAM,GAAG,sBAAsB;IACxF,MAAM,MAAM,GAAG,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC7C,eAAe,CAAC,MAAM,EAAE,MAAM,EAAE;QAC9B,SAAS;QACT,MAAM;QACN,qBAAqB;QACrB,0BAA0B;QAC1B,kBAAkB;KACnB,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,cAAc,CAAC,MAAM,CAAC,gBAAgB,EAAE,GAAG,MAAM,mBAAmB,CAAC,CAAC;IAC3F,eAAe,CAAC,YAAY,EAAE,GAAG,MAAM,mBAAmB,EAAE;QAC1D,SAAS;QACT,WAAW;QACX,gBAAgB;QAChB,UAAU;QACV,4BAA4B;QAC5B,yBAAyB;QACzB,qBAAqB;QACrB,eAAe;QACf,qBAAqB;QACrB,mBAAmB;QACnB,kBAAkB;QAClB,uBAAuB;QACvB,mBAAmB;QACnB,cAAc;KACf,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,MAAM,kCAAkC,CAAC,CAAC;IAC5G,eAAe,CAAC,UAAU,EAAE,GAAG,MAAM,kCAAkC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEtF,MAAM,cAAc,GAAG,cAAc,CAAC,YAAY,CAAC,QAAQ,EAAE,GAAG,MAAM,4BAA4B,CAAC,CAAC;IACpG,eAAe,CAAC,cAAc,EAAE,GAAG,MAAM,4BAA4B,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;IAEhG,MAAM,mBAAmB,GAAG,cAAc,CAAC,YAAY,CAAC,aAAa,EAAE,GAAG,MAAM,iCAAiC,CAAC,CAAC;IACnH,eAAe,CAAC,mBAAmB,EAAE,GAAG,MAAM,iCAAiC,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC;IAE1G,MAAM,QAAQ,GAAG,4BAA4B,CAAC;IAC9C,MAAM,cAAc,GAAG,QAAQ,CAAC,gBAAgB,CAAC;IACjD,MAAM,gBAAgB,GAA8B;QAClD,OAAO,EAAE,eAAe,CAAC,YAAY,CAAC,OAAO,EAAE,cAAc,CAAC,OAAO,EAAE,GAAG,MAAM,2BAA2B,CAAC;QAC5G,SAAS,EAAE,eAAe,CACxB,YAAY,CAAC,SAAS,EACtB,cAAc,CAAC,SAAS,EACxB,GAAG,MAAM,6BAA6B,CACvC;QACD,cAAc,EAAE;YACd,OAAO,EAAE,eAAe,CACtB,UAAU,CAAC,OAAO,EAClB,cAAc,CAAC,cAAc,CAAC,OAAO,EACrC,GAAG,MAAM,0CAA0C,CACpD;SACF;QACD,QAAQ,EAAE;YACR,OAAO,EAAE,eAAe,CACtB,cAAc,CAAC,OAAO,EACtB,cAAc,CAAC,QAAQ,CAAC,OAAO,EAC/B,GAAG,MAAM,oCAAoC,CAC9C;YACD,QAAQ,EAAE,eAAe,CACvB,cAAc,CAAC,QAAQ,EACvB,cAAc,CAAC,QAAQ,CAAC,QAAQ,EAChC,KAAK,EACL,OAAO,EACP,GAAG,MAAM,qCAAqC,CAC/C;SACF;QACD,0BAA0B,EAAE,eAAe,CACzC,YAAY,CAAC,0BAA0B,EACvC,cAAc,CAAC,0BAA0B,EACzC,GAAG,MAAM,8CAA8C,CACxD;QACD,uBAAuB,EAAE,eAAe,CACtC,YAAY,CAAC,uBAAuB,EACpC,cAAc,CAAC,uBAAuB,EACtC,GAAG,MAAM,2CAA2C,CACrD;QACD,mBAAmB,EAAE,YAAY,CAC/B,YAAY,CAAC,mBAAmB,EAChC,cAAc,CAAC,mBAAmB,EAClC,wBAAwB,EACxB,GAAG,MAAM,uCAAuC,CACjD;QACD,aAAa,EAAE;YACb,OAAO,EAAE,eAAe,CACtB,mBAAmB,CAAC,OAAO,EAC3B,cAAc,CAAC,aAAa,CAAC,OAAO,EACpC,GAAG,MAAM,yCAAyC,CACnD;YACD,QAAQ,EAAE,eAAe,CACvB,mBAAmB,CAAC,QAAQ,EAC5B,cAAc,CAAC,aAAa,CAAC,QAAQ,EACrC,EAAE,EACF,KAAK,EACL,GAAG,MAAM,0CAA0C,CACpD;SACF;QACD,mBAAmB,EAAE,eAAe,CAClC,YAAY,CAAC,mBAAmB,EAChC,cAAc,CAAC,mBAAmB,EAClC,GAAG,MAAM,uCAAuC,CACjD;QACD,iBAAiB,EAAE,eAAe,CAChC,YAAY,CAAC,iBAAiB,EAC9B,cAAc,CAAC,iBAAiB,EAChC,GAAG,MAAM,qCAAqC,CAC/C;QACD,gBAAgB,EAAE,eAAe,CAC/B,YAAY,CAAC,gBAAgB,EAC7B,cAAc,CAAC,gBAAgB,EAC/B,GAAG,MAAM,oCAAoC,CAC9C;QACD,qBAAqB,EAAE,eAAe,CACpC,YAAY,CAAC,qBAAqB,EAClC,cAAc,CAAC,qBAAqB,EACpC,GAAG,MAAM,yCAAyC,CACnD;QACD,iBAAiB,EAAE,eAAe,CAChC,YAAY,CAAC,iBAAiB,EAC9B,cAAc,CAAC,iBAAiB,EAChC,GAAG,MAAM,qCAAqC,CAC/C;QACD,YAAY,EAAE,eAAe,CAC3B,YAAY,CAAC,YAAY,EACzB,cAAc,CAAC,YAAY,EAC3B,GAAG,MAAM,gCAAgC,CAC1C;KACF,CAAC;IAEF,OAAO;QACL,OAAO,EAAE,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,GAAG,MAAM,UAAU,CAAC;QAC/E,IAAI,EAAE,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,GAAG,MAAM,OAAO,CAAC;QAC3E,mBAAmB,EAAE,eAAe,CAClC,MAAM,CAAC,mBAAmB,EAC1B,QAAQ,CAAC,mBAAmB,EAC5B,GAAG,MAAM,sBAAsB,CAChC;QACD,wBAAwB,EAAE,eAAe,CACvC,MAAM,CAAC,wBAAwB,EAC/B,QAAQ,CAAC,wBAAwB,EACjC,GAAG,MAAM,2BAA2B,CACrC;QACD,gBAAgB;KACjB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,OAAqB,EAAE,MAA0B;IAC1E,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACzC,MAAM,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzE,MAAM,OAAO,CAAC,SAAS,CAAC,yBAAyB,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;AAC7G,CAAC;AAED,SAAS,cAAc,CAAC,KAAc,EAAE,IAAY;IAClD,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,EAAE,CAAC;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,oBAAoB,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAgC,CAAC;AAC1C,CAAC;AAED,SAAS,eAAe,CAAC,KAAc,EAAE,QAAiB,EAAE,IAAY;IACtE,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,QAAQ,CAAC;IACzC,IAAI,OAAO,KAAK,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,oBAAoB,CAAC,CAAC;IAC7E,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,KAAc,EAAE,QAAgB,EAAE,OAAe,EAAE,OAAe,EAAE,IAAY;IACvG,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,QAAQ,CAAC;IACzC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAK,KAAgB,GAAG,OAAO,IAAK,KAAgB,GAAG,OAAO,EAAE,CAAC;QAC/F,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,4BAA4B,OAAO,OAAO,OAAO,EAAE,CAAC,CAAC;IAC9E,CAAC;IACD,OAAO,KAAe,CAAC;AACzB,CAAC;AAED,SAAS,YAAY,CAAmB,KAAc,EAAE,QAAW,EAAE,OAAqB,EAAE,IAAY;IACtG,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,QAAQ,CAAC;IACzC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAU,CAAC,EAAE,CAAC;QAC/D,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,YAAY,kBAAkB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACpE,CAAC;IACD,OAAO,KAAU,CAAC;AACpB,CAAC;AAED,SAAS,eAAe,CAAC,KAA8B,EAAE,IAAY,EAAE,OAA0B;IAC/F,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;IACpC,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACvE,IAAI,OAAO;QAAE,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,4BAA4B,OAAO,EAAE,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,kBAAkB,CAAC,MAAyB;IACnD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,MAAM,CAAC,CAAC,CAAC,IAAI,mBAAmB,CAAC;IAC/D,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAClE,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC;AACnG,CAAC;AAED,SAAS,kBAAkB;IACzB,OAAO,eAAe,CAAC,4BAA4B,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC"}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { FelanExtension } from '@felan-ai/agent-core';
|
|
2
|
+
import type { RtkOptimizerConfig } from './types.js';
|
|
3
|
+
declare const rtkOptimizerExtension: FelanExtension;
|
|
4
|
+
export declare function shouldInjectSourceFilterTroubleshootingNote(config: RtkOptimizerConfig): boolean;
|
|
5
|
+
export declare function injectGuidelineIntoPrompt(systemPrompt: string, guideline: string): string;
|
|
6
|
+
export interface BoundedNoticeTracker {
|
|
7
|
+
remember(key: string): boolean;
|
|
8
|
+
reset(): void;
|
|
9
|
+
}
|
|
10
|
+
export declare function createBoundedNoticeTracker(maxEntries: number): BoundedNoticeTracker;
|
|
11
|
+
export { getRtkOptimizerConfigPath, loadRtkOptimizerConfig, saveRtkOptimizerConfig, validateRtkOptimizerConfig, } from './config.js';
|
|
12
|
+
export { computeRewriteDecision, inspectRtkRuntime, resolveRtkRewrite } from './command-rewriter.js';
|
|
13
|
+
export { compactToolResult } from './output-compactor.js';
|
|
14
|
+
export { OutputMetrics } from './output-metrics.js';
|
|
15
|
+
export { DEFAULT_RTK_OPTIMIZER_CONFIG } from './types.js';
|
|
16
|
+
export type { RtkMode, RtkOptimizerConfig, RtkOutputCompactionConfig, RtkSourceFilterLevel, RuntimeStatus, } from './types.js';
|
|
17
|
+
export default rtkOptimizerExtension;
|
|
18
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAoB,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAiB7E,OAAO,KAAK,EAAE,kBAAkB,EAAiB,MAAM,YAAY,CAAC;AAOpE,QAAA,MAAM,qBAAqB,EAAE,cA4N5B,CAAC;AAEF,wBAAgB,2CAA2C,CAAC,MAAM,EAAE,kBAAkB,GAAG,OAAO,CAU/F;AAED,wBAAgB,yBAAyB,CAAC,YAAY,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAyBzF;AA4DD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;IAC/B,KAAK,IAAI,IAAI,CAAC;CACf;AAED,wBAAgB,0BAA0B,CAAC,UAAU,EAAE,MAAM,GAAG,oBAAoB,CAoBnF;AAWD,OAAO,EACL,yBAAyB,EACzB,sBAAsB,EACtB,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AACrG,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,4BAA4B,EAAE,MAAM,YAAY,CAAC;AAC1D,YAAY,EACV,OAAO,EACP,kBAAkB,EAClB,yBAAyB,EACzB,oBAAoB,EACpB,aAAa,GACd,MAAM,YAAY,CAAC;AACpB,eAAe,qBAAqB,CAAC"}
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,306 @@
|
|
|
1
|
+
import { registerRtkCommand } from './command.js';
|
|
2
|
+
import { computeRewriteDecision, inspectRtkRuntime } from './command-rewriter.js';
|
|
3
|
+
import { loadRtkOptimizerConfig, saveRtkOptimizerConfig } from './config.js';
|
|
4
|
+
import { compactToolResult } from './output-compactor.js';
|
|
5
|
+
import { OutputMetrics } from './output-metrics.js';
|
|
6
|
+
import { toRecord } from './record-utils.js';
|
|
7
|
+
import { sanitizeStreamingCommandResult } from './tool-execution-sanitizer.js';
|
|
8
|
+
import { codexResultHasExited, isCommandToolName, isStreamingCommandToolName, readCodexSessionId, readRunningCodexSessionId, readToolCommand, writeToolCommand, } from './tool-shapes.js';
|
|
9
|
+
const EXTENSION_NAME = 'RTK optimizer';
|
|
10
|
+
const RUNTIME_STATUS_MAX_AGE_MS = 30_000;
|
|
11
|
+
const SOURCE_FILTER_TROUBLESHOOTING_NOTE = "RTK note: If file edits repeatedly fail because old text does not match, run '/rtk', disable lossy read compaction, re-read the file, apply the edit, and then re-enable compaction if desired.";
|
|
12
|
+
const rtkOptimizerExtension = async (pi) => {
|
|
13
|
+
const initialLoad = await loadRtkOptimizerConfig(pi.runtime);
|
|
14
|
+
let config = initialLoad.config;
|
|
15
|
+
let pendingConfigWarning = initialLoad.warning;
|
|
16
|
+
let runtimeStatus = { rtkAvailable: false };
|
|
17
|
+
let statusRefresh;
|
|
18
|
+
let missingRtkWarningShown = false;
|
|
19
|
+
const metrics = new OutputMetrics();
|
|
20
|
+
const warnedMessages = createBoundedNoticeTracker(100);
|
|
21
|
+
const suggestionNotices = createBoundedNoticeTracker(200);
|
|
22
|
+
const codexSessionCommands = new Map();
|
|
23
|
+
const activeCodexCommands = new Map();
|
|
24
|
+
const refreshRuntimeStatus = () => {
|
|
25
|
+
statusRefresh ??= inspectRtkRuntime(pi.runtime)
|
|
26
|
+
.then((status) => {
|
|
27
|
+
runtimeStatus = status;
|
|
28
|
+
if (status.rtkAvailable)
|
|
29
|
+
missingRtkWarningShown = false;
|
|
30
|
+
return status;
|
|
31
|
+
})
|
|
32
|
+
.finally(() => {
|
|
33
|
+
statusRefresh = undefined;
|
|
34
|
+
});
|
|
35
|
+
return statusRefresh;
|
|
36
|
+
};
|
|
37
|
+
const ensureRuntimeStatusFresh = async () => {
|
|
38
|
+
if (!config.guardWhenRtkMissing)
|
|
39
|
+
return;
|
|
40
|
+
if (runtimeStatus.lastCheckedAt === undefined ||
|
|
41
|
+
Date.now() - runtimeStatus.lastCheckedAt > RUNTIME_STATUS_MAX_AGE_MS) {
|
|
42
|
+
await refreshRuntimeStatus();
|
|
43
|
+
}
|
|
44
|
+
};
|
|
45
|
+
const warnOnce = (ctx, message, level = 'warning') => {
|
|
46
|
+
if (!warnedMessages.remember(message))
|
|
47
|
+
return;
|
|
48
|
+
ctx.ui.notify(message, level);
|
|
49
|
+
};
|
|
50
|
+
const maybeWarnRtkMissing = (ctx) => {
|
|
51
|
+
if (!config.enabled || !config.guardWhenRtkMissing || runtimeStatus.rtkAvailable || missingRtkWarningShown)
|
|
52
|
+
return;
|
|
53
|
+
missingRtkWarningShown = true;
|
|
54
|
+
const reason = runtimeStatus.lastError ? ` (${runtimeStatus.lastError})` : '';
|
|
55
|
+
warnOnce(ctx, `${EXTENSION_NAME}: rtk is unavailable; command rewriting is bypassed${reason}.`);
|
|
56
|
+
};
|
|
57
|
+
registerRtkCommand(pi, {
|
|
58
|
+
getConfig: () => config,
|
|
59
|
+
setConfig: async (next) => {
|
|
60
|
+
config = await saveRtkOptimizerConfig(pi.runtime, next);
|
|
61
|
+
},
|
|
62
|
+
getRuntimeStatus: () => runtimeStatus,
|
|
63
|
+
refreshRuntimeStatus,
|
|
64
|
+
getMetricsSummary: () => metrics.summary(),
|
|
65
|
+
clearMetrics: () => metrics.clear(),
|
|
66
|
+
});
|
|
67
|
+
pi.on('session_start', async (_event, ctx) => {
|
|
68
|
+
warnedMessages.reset();
|
|
69
|
+
suggestionNotices.reset();
|
|
70
|
+
codexSessionCommands.clear();
|
|
71
|
+
activeCodexCommands.clear();
|
|
72
|
+
metrics.clear();
|
|
73
|
+
missingRtkWarningShown = false;
|
|
74
|
+
const loaded = await loadRtkOptimizerConfig(pi.runtime);
|
|
75
|
+
config = loaded.config;
|
|
76
|
+
pendingConfigWarning = loaded.warning ?? pendingConfigWarning;
|
|
77
|
+
if (pendingConfigWarning) {
|
|
78
|
+
warnOnce(ctx, pendingConfigWarning);
|
|
79
|
+
pendingConfigWarning = undefined;
|
|
80
|
+
}
|
|
81
|
+
if (config.enabled) {
|
|
82
|
+
await refreshRuntimeStatus();
|
|
83
|
+
maybeWarnRtkMissing(ctx);
|
|
84
|
+
}
|
|
85
|
+
});
|
|
86
|
+
pi.on('session_shutdown', () => {
|
|
87
|
+
codexSessionCommands.clear();
|
|
88
|
+
activeCodexCommands.clear();
|
|
89
|
+
});
|
|
90
|
+
pi.on('agent_end', () => {
|
|
91
|
+
activeCodexCommands.clear();
|
|
92
|
+
});
|
|
93
|
+
pi.on('before_agent_start', async (event, ctx) => {
|
|
94
|
+
if (!config.enabled)
|
|
95
|
+
return undefined;
|
|
96
|
+
await ensureRuntimeStatusFresh();
|
|
97
|
+
maybeWarnRtkMissing(ctx);
|
|
98
|
+
if (!shouldInjectSourceFilterTroubleshootingNote(config))
|
|
99
|
+
return undefined;
|
|
100
|
+
const systemPrompt = injectGuidelineIntoPrompt(event.systemPrompt, SOURCE_FILTER_TROUBLESHOOTING_NOTE);
|
|
101
|
+
return systemPrompt === event.systemPrompt ? undefined : { systemPrompt };
|
|
102
|
+
});
|
|
103
|
+
pi.on('tool_call', async (event, ctx) => {
|
|
104
|
+
if (!config.enabled || !isCommandToolName(event.toolName))
|
|
105
|
+
return undefined;
|
|
106
|
+
const command = readToolCommand(event.toolName, event.input);
|
|
107
|
+
if (!command)
|
|
108
|
+
return undefined;
|
|
109
|
+
await ensureRuntimeStatusFresh();
|
|
110
|
+
if (config.guardWhenRtkMissing && !runtimeStatus.rtkAvailable) {
|
|
111
|
+
maybeWarnRtkMissing(ctx);
|
|
112
|
+
return undefined;
|
|
113
|
+
}
|
|
114
|
+
const decision = await computeRewriteDecision(pi.runtime, command);
|
|
115
|
+
if (!decision.changed) {
|
|
116
|
+
if (decision.warning) {
|
|
117
|
+
warnOnce(ctx, `${EXTENSION_NAME}: rtk rewrite skipped for '${trimMessage(command, 100)}' (${trimMessage(decision.warning)}).`);
|
|
118
|
+
}
|
|
119
|
+
return undefined;
|
|
120
|
+
}
|
|
121
|
+
if (config.mode === 'rewrite') {
|
|
122
|
+
writeToolCommand(event.toolName, event.input, decision.rewrittenCommand);
|
|
123
|
+
if (config.showRewriteNotifications && ctx.hasUI) {
|
|
124
|
+
ctx.ui.notify(`RTK rewrite: ${trimMessage(command, 100)} -> ${trimMessage(decision.rewrittenCommand, 120)}`, 'info');
|
|
125
|
+
}
|
|
126
|
+
return undefined;
|
|
127
|
+
}
|
|
128
|
+
const suggestionKey = `${command}:${decision.rewrittenCommand}`;
|
|
129
|
+
if (suggestionNotices.remember(suggestionKey) && ctx.hasUI) {
|
|
130
|
+
ctx.ui.notify(`RTK suggestion: ${decision.rewrittenCommand}`, 'info');
|
|
131
|
+
}
|
|
132
|
+
return undefined;
|
|
133
|
+
});
|
|
134
|
+
pi.on('tool_execution_start', (event) => {
|
|
135
|
+
if (event.toolName !== 'exec_command')
|
|
136
|
+
return;
|
|
137
|
+
const command = readToolCommand(event.toolName, event.args);
|
|
138
|
+
if (command)
|
|
139
|
+
activeCodexCommands.set(event.toolCallId, command);
|
|
140
|
+
});
|
|
141
|
+
pi.on('tool_execution_update', (event) => {
|
|
142
|
+
recordCodexSessionFromExecution(event.toolName, event.toolCallId, event.partialResult, activeCodexCommands, codexSessionCommands);
|
|
143
|
+
if (!config.enabled ||
|
|
144
|
+
!config.outputCompaction.enabled ||
|
|
145
|
+
!config.outputCompaction.stripAnsi ||
|
|
146
|
+
!isStreamingCommandToolName(event.toolName))
|
|
147
|
+
return;
|
|
148
|
+
sanitizeStreamingCommandResult(event.partialResult);
|
|
149
|
+
});
|
|
150
|
+
pi.on('tool_execution_end', (event) => {
|
|
151
|
+
recordCodexSessionFromExecution(event.toolName, event.toolCallId, event.result, activeCodexCommands, codexSessionCommands);
|
|
152
|
+
activeCodexCommands.delete(event.toolCallId);
|
|
153
|
+
if (!config.enabled ||
|
|
154
|
+
!config.outputCompaction.enabled ||
|
|
155
|
+
!config.outputCompaction.stripAnsi ||
|
|
156
|
+
!isStreamingCommandToolName(event.toolName))
|
|
157
|
+
return;
|
|
158
|
+
sanitizeStreamingCommandResult(event.result);
|
|
159
|
+
});
|
|
160
|
+
pi.on('tool_result', async (event, ctx) => {
|
|
161
|
+
const originatingCommand = commandForResult(event.toolName, event.input, codexSessionCommands);
|
|
162
|
+
updateCodexSessionCommands(event.toolName, event.input, event.details, originatingCommand, codexSessionCommands);
|
|
163
|
+
if (!config.enabled || !config.outputCompaction.enabled)
|
|
164
|
+
return undefined;
|
|
165
|
+
try {
|
|
166
|
+
const outcome = compactToolResult({
|
|
167
|
+
toolName: event.toolName,
|
|
168
|
+
input: event.input,
|
|
169
|
+
content: event.content,
|
|
170
|
+
}, config, {
|
|
171
|
+
cwd: ctx.cwd,
|
|
172
|
+
agentDir: pi.agentDir,
|
|
173
|
+
...(originatingCommand === undefined ? {} : { command: originatingCommand }),
|
|
174
|
+
metrics,
|
|
175
|
+
});
|
|
176
|
+
if (!outcome.changed || !outcome.content)
|
|
177
|
+
return undefined;
|
|
178
|
+
return {
|
|
179
|
+
content: outcome.content,
|
|
180
|
+
...(outcome.metadata === undefined ? {} : { details: mergeCompactionDetails(event.details, outcome.metadata) }),
|
|
181
|
+
};
|
|
182
|
+
}
|
|
183
|
+
catch (error) {
|
|
184
|
+
warnOnce(ctx, `${EXTENSION_NAME}: output compaction failed; using raw output (${trimMessage(errorMessage(error))}).`);
|
|
185
|
+
return undefined;
|
|
186
|
+
}
|
|
187
|
+
});
|
|
188
|
+
};
|
|
189
|
+
export function shouldInjectSourceFilterTroubleshootingNote(config) {
|
|
190
|
+
const compaction = config.outputCompaction;
|
|
191
|
+
return (config.enabled &&
|
|
192
|
+
compaction.enabled &&
|
|
193
|
+
compaction.readCompaction.enabled &&
|
|
194
|
+
compaction.sourceCodeFilteringEnabled &&
|
|
195
|
+
compaction.sourceCodeFiltering !== 'none' &&
|
|
196
|
+
(compaction.smartTruncate.enabled || compaction.truncate.enabled));
|
|
197
|
+
}
|
|
198
|
+
export function injectGuidelineIntoPrompt(systemPrompt, guideline) {
|
|
199
|
+
if (!systemPrompt || systemPrompt.includes(guideline))
|
|
200
|
+
return systemPrompt;
|
|
201
|
+
const bullet = `- ${guideline}`;
|
|
202
|
+
const header = 'Guidelines:\n';
|
|
203
|
+
let headerIndex = systemPrompt.indexOf(`\n${header}`);
|
|
204
|
+
let contentStart;
|
|
205
|
+
if (headerIndex >= 0) {
|
|
206
|
+
contentStart = headerIndex + header.length + 1;
|
|
207
|
+
}
|
|
208
|
+
else if (systemPrompt.startsWith(header)) {
|
|
209
|
+
headerIndex = 0;
|
|
210
|
+
contentStart = header.length;
|
|
211
|
+
}
|
|
212
|
+
else {
|
|
213
|
+
return `${systemPrompt}\n\n${guideline}`;
|
|
214
|
+
}
|
|
215
|
+
const remainder = systemPrompt.slice(contentStart);
|
|
216
|
+
let consumed = 0;
|
|
217
|
+
for (const line of remainder.split('\n')) {
|
|
218
|
+
if (!line || !/^[-*+\s]/u.test(line))
|
|
219
|
+
break;
|
|
220
|
+
consumed += line.length + 1;
|
|
221
|
+
}
|
|
222
|
+
const insertAt = consumed === 0 ? contentStart : contentStart + consumed - 1;
|
|
223
|
+
const before = systemPrompt.slice(0, insertAt);
|
|
224
|
+
const after = systemPrompt.slice(insertAt);
|
|
225
|
+
return `${before}${before.endsWith('\n') ? '' : '\n'}${bullet}${after.startsWith('\n') ? '' : '\n'}${after}`;
|
|
226
|
+
}
|
|
227
|
+
function commandForResult(toolName, input, codexSessionCommands) {
|
|
228
|
+
if (isCommandToolName(toolName))
|
|
229
|
+
return readToolCommand(toolName, input);
|
|
230
|
+
if (toolName !== 'write_stdin')
|
|
231
|
+
return undefined;
|
|
232
|
+
const sessionId = readCodexSessionId(input);
|
|
233
|
+
return sessionId === undefined ? undefined : codexSessionCommands.get(sessionId);
|
|
234
|
+
}
|
|
235
|
+
function recordCodexSessionFromExecution(toolName, toolCallId, result, activeCodexCommands, codexSessionCommands) {
|
|
236
|
+
if (toolName !== 'exec_command')
|
|
237
|
+
return;
|
|
238
|
+
const command = activeCodexCommands.get(toolCallId);
|
|
239
|
+
if (!command)
|
|
240
|
+
return;
|
|
241
|
+
const sessionId = readRunningCodexSessionId(toRecord(result).details);
|
|
242
|
+
if (sessionId !== undefined)
|
|
243
|
+
codexSessionCommands.set(sessionId, command);
|
|
244
|
+
}
|
|
245
|
+
function updateCodexSessionCommands(toolName, input, details, command, codexSessionCommands) {
|
|
246
|
+
if (toolName === 'exec_command' && command) {
|
|
247
|
+
const runningSessionId = readRunningCodexSessionId(details);
|
|
248
|
+
if (runningSessionId !== undefined)
|
|
249
|
+
codexSessionCommands.set(runningSessionId, command);
|
|
250
|
+
return;
|
|
251
|
+
}
|
|
252
|
+
if (toolName !== 'write_stdin')
|
|
253
|
+
return;
|
|
254
|
+
const requestedSessionId = readCodexSessionId(input);
|
|
255
|
+
if (requestedSessionId === undefined)
|
|
256
|
+
return;
|
|
257
|
+
if (codexResultHasExited(details))
|
|
258
|
+
codexSessionCommands.delete(requestedSessionId);
|
|
259
|
+
}
|
|
260
|
+
function mergeCompactionDetails(existingDetails, compaction) {
|
|
261
|
+
const base = toRecord(existingDetails);
|
|
262
|
+
const metadata = toRecord(base.metadata);
|
|
263
|
+
return {
|
|
264
|
+
...base,
|
|
265
|
+
rtkCompaction: compaction,
|
|
266
|
+
metadata: { ...metadata, rtkCompaction: compaction },
|
|
267
|
+
...(Object.keys(base).length === 0 && existingDetails !== undefined ? { rawDetails: existingDetails } : {}),
|
|
268
|
+
};
|
|
269
|
+
}
|
|
270
|
+
export function createBoundedNoticeTracker(maxEntries) {
|
|
271
|
+
const limit = Math.max(1, Math.floor(maxEntries));
|
|
272
|
+
const seen = new Set();
|
|
273
|
+
const order = [];
|
|
274
|
+
return {
|
|
275
|
+
remember(key) {
|
|
276
|
+
if (seen.has(key))
|
|
277
|
+
return false;
|
|
278
|
+
seen.add(key);
|
|
279
|
+
order.push(key);
|
|
280
|
+
while (order.length > limit) {
|
|
281
|
+
const evicted = order.shift();
|
|
282
|
+
if (evicted !== undefined)
|
|
283
|
+
seen.delete(evicted);
|
|
284
|
+
}
|
|
285
|
+
return true;
|
|
286
|
+
},
|
|
287
|
+
reset() {
|
|
288
|
+
seen.clear();
|
|
289
|
+
order.length = 0;
|
|
290
|
+
},
|
|
291
|
+
};
|
|
292
|
+
}
|
|
293
|
+
function trimMessage(value, maxLength = 220) {
|
|
294
|
+
const clean = value.replace(/\s+/g, ' ').trim();
|
|
295
|
+
return clean.length <= maxLength ? clean : `${clean.slice(0, maxLength - 1)}…`;
|
|
296
|
+
}
|
|
297
|
+
function errorMessage(error) {
|
|
298
|
+
return error instanceof Error ? error.message : String(error);
|
|
299
|
+
}
|
|
300
|
+
export { getRtkOptimizerConfigPath, loadRtkOptimizerConfig, saveRtkOptimizerConfig, validateRtkOptimizerConfig, } from './config.js';
|
|
301
|
+
export { computeRewriteDecision, inspectRtkRuntime, resolveRtkRewrite } from './command-rewriter.js';
|
|
302
|
+
export { compactToolResult } from './output-compactor.js';
|
|
303
|
+
export { OutputMetrics } from './output-metrics.js';
|
|
304
|
+
export { DEFAULT_RTK_OPTIMIZER_CONFIG } from './types.js';
|
|
305
|
+
export default rtkOptimizerExtension;
|
|
306
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAClF,OAAO,EAAE,sBAAsB,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAqC,MAAM,uBAAuB,CAAC;AAC7F,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAC7C,OAAO,EAAE,8BAA8B,EAAE,MAAM,+BAA+B,CAAC;AAC/E,OAAO,EACL,oBAAoB,EACpB,iBAAiB,EACjB,0BAA0B,EAC1B,kBAAkB,EAClB,yBAAyB,EACzB,eAAe,EACf,gBAAgB,GACjB,MAAM,kBAAkB,CAAC;AAG1B,MAAM,cAAc,GAAG,eAAe,CAAC;AACvC,MAAM,yBAAyB,GAAG,MAAM,CAAC;AACzC,MAAM,kCAAkC,GACtC,iMAAiM,CAAC;AAEpM,MAAM,qBAAqB,GAAmB,KAAK,EAAE,EAAE,EAAE,EAAE;IACzD,MAAM,WAAW,GAAG,MAAM,sBAAsB,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;IAC7D,IAAI,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;IAChC,IAAI,oBAAoB,GAAG,WAAW,CAAC,OAAO,CAAC;IAC/C,IAAI,aAAa,GAAkB,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC;IAC3D,IAAI,aAAiD,CAAC;IACtD,IAAI,sBAAsB,GAAG,KAAK,CAAC;IAEnC,MAAM,OAAO,GAAG,IAAI,aAAa,EAAE,CAAC;IACpC,MAAM,cAAc,GAAG,0BAA0B,CAAC,GAAG,CAAC,CAAC;IACvD,MAAM,iBAAiB,GAAG,0BAA0B,CAAC,GAAG,CAAC,CAAC;IAC1D,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACvD,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAAkB,CAAC;IAEtD,MAAM,oBAAoB,GAAG,GAA2B,EAAE;QACxD,aAAa,KAAK,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC;aAC5C,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE;YACf,aAAa,GAAG,MAAM,CAAC;YACvB,IAAI,MAAM,CAAC,YAAY;gBAAE,sBAAsB,GAAG,KAAK,CAAC;YACxD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;aACD,OAAO,CAAC,GAAG,EAAE;YACZ,aAAa,GAAG,SAAS,CAAC;QAC5B,CAAC,CAAC,CAAC;QACL,OAAO,aAAa,CAAC;IACvB,CAAC,CAAC;IAEF,MAAM,wBAAwB,GAAG,KAAK,IAAmB,EAAE;QACzD,IAAI,CAAC,MAAM,CAAC,mBAAmB;YAAE,OAAO;QACxC,IACE,aAAa,CAAC,aAAa,KAAK,SAAS;YACzC,IAAI,CAAC,GAAG,EAAE,GAAG,aAAa,CAAC,aAAa,GAAG,yBAAyB,EACpE,CAAC;YACD,MAAM,oBAAoB,EAAE,CAAC;QAC/B,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,QAAQ,GAAG,CAAC,GAAqB,EAAE,OAAe,EAAE,QAA6B,SAAS,EAAQ,EAAE;QACxG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO;QAC9C,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAChC,CAAC,CAAC;IAEF,MAAM,mBAAmB,GAAG,CAAC,GAAqB,EAAQ,EAAE;QAC1D,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,mBAAmB,IAAI,aAAa,CAAC,YAAY,IAAI,sBAAsB;YAAE,OAAO;QAEnH,sBAAsB,GAAG,IAAI,CAAC;QAC9B,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,aAAa,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;QAC9E,QAAQ,CAAC,GAAG,EAAE,GAAG,cAAc,sDAAsD,MAAM,GAAG,CAAC,CAAC;IAClG,CAAC,CAAC;IAEF,kBAAkB,CAAC,EAAE,EAAE;QACrB,SAAS,EAAE,GAAG,EAAE,CAAC,MAAM;QACvB,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YACxB,MAAM,GAAG,MAAM,sBAAsB,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAC1D,CAAC;QACD,gBAAgB,EAAE,GAAG,EAAE,CAAC,aAAa;QACrC,oBAAoB;QACpB,iBAAiB,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE;QAC1C,YAAY,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE;KACpC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC3C,cAAc,CAAC,KAAK,EAAE,CAAC;QACvB,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC1B,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAC7B,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAC5B,OAAO,CAAC,KAAK,EAAE,CAAC;QAChB,sBAAsB,GAAG,KAAK,CAAC;QAE/B,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QACvB,oBAAoB,GAAG,MAAM,CAAC,OAAO,IAAI,oBAAoB,CAAC;QAC9D,IAAI,oBAAoB,EAAE,CAAC;YACzB,QAAQ,CAAC,GAAG,EAAE,oBAAoB,CAAC,CAAC;YACpC,oBAAoB,GAAG,SAAS,CAAC;QACnC,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,MAAM,oBAAoB,EAAE,CAAC;YAC7B,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAC7B,oBAAoB,CAAC,KAAK,EAAE,CAAC;QAC7B,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,GAAG,EAAE;QACtB,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QAC/C,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QACtC,MAAM,wBAAwB,EAAE,CAAC;QACjC,mBAAmB,CAAC,GAAG,CAAC,CAAC;QAEzB,IAAI,CAAC,2CAA2C,CAAC,MAAM,CAAC;YAAE,OAAO,SAAS,CAAC;QAC3E,MAAM,YAAY,GAAG,yBAAyB,CAAC,KAAK,CAAC,YAAY,EAAE,kCAAkC,CAAC,CAAC;QACvG,OAAO,YAAY,KAAK,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QACtC,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC;YAAE,OAAO,SAAS,CAAC;QAC5E,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QAC7D,IAAI,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAE/B,MAAM,wBAAwB,EAAE,CAAC;QACjC,IAAI,MAAM,CAAC,mBAAmB,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC;YAC9D,mBAAmB,CAAC,GAAG,CAAC,CAAC;YACzB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,sBAAsB,CAAC,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACnE,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACtB,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrB,QAAQ,CACN,GAAG,EACH,GAAG,cAAc,8BAA8B,WAAW,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAChH,CAAC;YACJ,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;YAC9B,gBAAgB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,gBAAgB,CAAC,CAAC;YACzE,IAAI,MAAM,CAAC,wBAAwB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACjD,GAAG,CAAC,EAAE,CAAC,MAAM,CACX,gBAAgB,WAAW,CAAC,OAAO,EAAE,GAAG,CAAC,OAAO,WAAW,CAAC,QAAQ,CAAC,gBAAgB,EAAE,GAAG,CAAC,EAAE,EAC7F,MAAM,CACP,CAAC;YACJ,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,aAAa,GAAG,GAAG,OAAO,IAAI,QAAQ,CAAC,gBAAgB,EAAE,CAAC;QAChE,IAAI,iBAAiB,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YAC3D,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,mBAAmB,QAAQ,CAAC,gBAAgB,EAAE,EAAE,MAAM,CAAC,CAAC;QACxE,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,sBAAsB,EAAE,CAAC,KAAK,EAAE,EAAE;QACtC,IAAI,KAAK,CAAC,QAAQ,KAAK,cAAc;YAAE,OAAO;QAC9C,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAC5D,IAAI,OAAO;YAAE,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,uBAAuB,EAAE,CAAC,KAAK,EAAE,EAAE;QACvC,+BAA+B,CAC7B,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,aAAa,EACnB,mBAAmB,EACnB,oBAAoB,CACrB,CAAC;QACF,IACE,CAAC,MAAM,CAAC,OAAO;YACf,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO;YAChC,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS;YAClC,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC;YAE3C,OAAO;QAET,8BAA8B,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,CAAC,KAAK,EAAE,EAAE;QACpC,+BAA+B,CAC7B,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,UAAU,EAChB,KAAK,CAAC,MAAM,EACZ,mBAAmB,EACnB,oBAAoB,CACrB,CAAC;QACF,mBAAmB,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC7C,IACE,CAAC,MAAM,CAAC,OAAO;YACf,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO;YAChC,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS;YAClC,CAAC,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC;YAE3C,OAAO;QAET,8BAA8B,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QACxC,MAAM,kBAAkB,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,oBAAoB,CAAC,CAAC;QAC/F,0BAA0B,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,CAAC;QAEjH,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAE1E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,iBAAiB,CAC/B;gBACE,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,OAAO,EAAE,KAAK,CAAC,OAAO;aACvB,EACD,MAAM,EACN;gBACE,GAAG,EAAE,GAAG,CAAC,GAAG;gBACZ,QAAQ,EAAE,EAAE,CAAC,QAAQ;gBACrB,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC;gBAC5E,OAAO;aACR,CACF,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO;gBAAE,OAAO,SAAS,CAAC;YAC3D,OAAO;gBACL,OAAO,EAAE,OAAO,CAAC,OAA+B;gBAChD,GAAG,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,sBAAsB,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;aAChH,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,QAAQ,CACN,GAAG,EACH,GAAG,cAAc,iDAAiD,WAAW,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,IAAI,CACvG,CAAC;YACF,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC;AAEF,MAAM,UAAU,2CAA2C,CAAC,MAA0B;IACpF,MAAM,UAAU,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAC3C,OAAO,CACL,MAAM,CAAC,OAAO;QACd,UAAU,CAAC,OAAO;QAClB,UAAU,CAAC,cAAc,CAAC,OAAO;QACjC,UAAU,CAAC,0BAA0B;QACrC,UAAU,CAAC,mBAAmB,KAAK,MAAM;QACzC,CAAC,UAAU,CAAC,aAAa,CAAC,OAAO,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,CAClE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,YAAoB,EAAE,SAAiB;IAC/E,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,YAAY,CAAC;IAC3E,MAAM,MAAM,GAAG,KAAK,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAG,eAAe,CAAC;IAC/B,IAAI,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC;IACtD,IAAI,YAAoB,CAAC;IACzB,IAAI,WAAW,IAAI,CAAC,EAAE,CAAC;QACrB,YAAY,GAAG,WAAW,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;IACjD,CAAC;SAAM,IAAI,YAAY,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3C,WAAW,GAAG,CAAC,CAAC;QAChB,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC;IAC/B,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,YAAY,OAAO,SAAS,EAAE,CAAC;IAC3C,CAAC;IAED,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IACnD,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACzC,IAAI,CAAC,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,MAAM;QAC5C,QAAQ,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;IAC9B,CAAC;IACD,MAAM,QAAQ,GAAG,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,GAAG,QAAQ,GAAG,CAAC,CAAC;IAC7E,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC3C,OAAO,GAAG,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,KAAK,EAAE,CAAC;AAC/G,CAAC;AAED,SAAS,gBAAgB,CACvB,QAAgB,EAChB,KAAc,EACd,oBAAiD;IAEjD,IAAI,iBAAiB,CAAC,QAAQ,CAAC;QAAE,OAAO,eAAe,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACzE,IAAI,QAAQ,KAAK,aAAa;QAAE,OAAO,SAAS,CAAC;IACjD,MAAM,SAAS,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAC5C,OAAO,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,+BAA+B,CACtC,QAAgB,EAChB,UAAkB,EAClB,MAAe,EACf,mBAAgD,EAChD,oBAAyC;IAEzC,IAAI,QAAQ,KAAK,cAAc;QAAE,OAAO;IACxC,MAAM,OAAO,GAAG,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACpD,IAAI,CAAC,OAAO;QAAE,OAAO;IACrB,MAAM,SAAS,GAAG,yBAAyB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC;IACtE,IAAI,SAAS,KAAK,SAAS;QAAE,oBAAoB,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,0BAA0B,CACjC,QAAgB,EAChB,KAAc,EACd,OAAgB,EAChB,OAA2B,EAC3B,oBAAyC;IAEzC,IAAI,QAAQ,KAAK,cAAc,IAAI,OAAO,EAAE,CAAC;QAC3C,MAAM,gBAAgB,GAAG,yBAAyB,CAAC,OAAO,CAAC,CAAC;QAC5D,IAAI,gBAAgB,KAAK,SAAS;YAAE,oBAAoB,CAAC,GAAG,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;QACxF,OAAO;IACT,CAAC;IACD,IAAI,QAAQ,KAAK,aAAa;QAAE,OAAO;IAEvC,MAAM,kBAAkB,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IACrD,IAAI,kBAAkB,KAAK,SAAS;QAAE,OAAO;IAC7C,IAAI,oBAAoB,CAAC,OAAO,CAAC;QAAE,oBAAoB,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;AACrF,CAAC;AAED,SAAS,sBAAsB,CAC7B,eAAwB,EACxB,UAAwC;IAExC,MAAM,IAAI,GAAG,QAAQ,CAAC,eAAe,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACzC,OAAO;QACL,GAAG,IAAI;QACP,aAAa,EAAE,UAAU;QACzB,QAAQ,EAAE,EAAE,GAAG,QAAQ,EAAE,aAAa,EAAE,UAAU,EAAE;QACpD,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,eAAe,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC5G,CAAC;AACJ,CAAC;AAOD,MAAM,UAAU,0BAA0B,CAAC,UAAkB;IAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC;IAClD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,OAAO;QACL,QAAQ,CAAC,GAAG;YACV,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,OAAO,KAAK,CAAC;YAChC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACd,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAChB,OAAO,KAAK,CAAC,MAAM,GAAG,KAAK,EAAE,CAAC;gBAC5B,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC9B,IAAI,OAAO,KAAK,SAAS;oBAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAClD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK;YACH,IAAI,CAAC,KAAK,EAAE,CAAC;YACb,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;QACnB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,KAAa,EAAE,SAAS,GAAG,GAAG;IACjD,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IAChD,OAAO,KAAK,CAAC,MAAM,IAAI,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,GAAG,CAAC;AACjF,CAAC;AAED,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAChE,CAAC;AAED,OAAO,EACL,yBAAyB,EACzB,sBAAsB,EACtB,sBAAsB,EACtB,0BAA0B,GAC3B,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,sBAAsB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AACrG,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,4BAA4B,EAAE,MAAM,YAAY,CAAC;AAQ1D,eAAe,qBAAqB,CAAC"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import type { OutputMetrics } from './output-metrics.js';
|
|
2
|
+
import type { RtkOptimizerConfig } from './types.js';
|
|
3
|
+
interface ToolResultLikeEvent {
|
|
4
|
+
toolName: string;
|
|
5
|
+
input?: unknown;
|
|
6
|
+
content?: unknown;
|
|
7
|
+
}
|
|
8
|
+
export interface ToolResultCompactionOptions {
|
|
9
|
+
readonly cwd: string;
|
|
10
|
+
readonly agentDir: string;
|
|
11
|
+
readonly command?: string;
|
|
12
|
+
readonly metrics?: OutputMetrics;
|
|
13
|
+
}
|
|
14
|
+
export interface ToolResultCompactionMetadata {
|
|
15
|
+
applied: boolean;
|
|
16
|
+
techniques: string[];
|
|
17
|
+
truncated: boolean;
|
|
18
|
+
originalCharCount: number;
|
|
19
|
+
compactedCharCount: number;
|
|
20
|
+
originalLineCount: number;
|
|
21
|
+
compactedLineCount: number;
|
|
22
|
+
}
|
|
23
|
+
export interface ToolResultCompactionOutcome {
|
|
24
|
+
changed: boolean;
|
|
25
|
+
content?: unknown[];
|
|
26
|
+
techniques: string[];
|
|
27
|
+
metadata?: ToolResultCompactionMetadata;
|
|
28
|
+
}
|
|
29
|
+
export declare function compactToolResult(event: ToolResultLikeEvent, config: RtkOptimizerConfig, options: ToolResultCompactionOptions): ToolResultCompactionOutcome;
|
|
30
|
+
export {};
|
|
31
|
+
//# sourceMappingURL=output-compactor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"output-compactor.d.ts","sourceRoot":"","sources":["../src/output-compactor.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAIzD,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAErD,UAAU,mBAAmB;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,2BAA2B;IAC1C,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC;CAClC;AAED,MAAM,WAAW,4BAA4B;IAC3C,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,SAAS,EAAE,OAAO,CAAC;IACnB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,kBAAkB,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,2BAA2B;IAC1C,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,EAAE,CAAC;IACpB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,4BAA4B,CAAC;CACzC;AA+iBD,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,mBAAmB,EAC1B,MAAM,EAAE,kBAAkB,EAC1B,OAAO,EAAE,2BAA2B,GACnC,2BAA2B,CA2E7B"}
|