@namzu/sdk 1.4.0 → 2.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/CHANGELOG.md +19 -0
- package/dist/registry/tool/execute.d.ts.map +1 -1
- package/dist/registry/tool/execute.js +40 -10
- package/dist/registry/tool/execute.js.map +1 -1
- package/dist/registry/tool/execute.test.js +93 -7
- package/dist/registry/tool/execute.test.js.map +1 -1
- package/dist/registry/toolset/catalog.d.ts.map +1 -1
- package/dist/registry/toolset/catalog.js +17 -4
- package/dist/registry/toolset/catalog.js.map +1 -1
- package/dist/registry/toolset/catalog.test.js +101 -0
- package/dist/registry/toolset/catalog.test.js.map +1 -1
- package/dist/runtime/query/__tests__/capability-negotiation.test.js +79 -0
- package/dist/runtime/query/__tests__/capability-negotiation.test.js.map +1 -1
- package/dist/runtime/query/__tests__/deferred-tools.test.js +100 -0
- package/dist/runtime/query/__tests__/deferred-tools.test.js.map +1 -1
- package/dist/runtime/query/__tests__/long-document-flow.test.js +12 -12
- package/dist/runtime/query/__tests__/long-document-flow.test.js.map +1 -1
- package/dist/runtime/query/__tests__/stream-recovery.test.js +1 -1
- package/dist/runtime/query/__tests__/stream-recovery.test.js.map +1 -1
- package/dist/runtime/query/executor.js +2 -2
- package/dist/runtime/query/executor.js.map +1 -1
- package/dist/runtime/query/iteration/index.d.ts.map +1 -1
- package/dist/runtime/query/iteration/index.js +10 -0
- package/dist/runtime/query/iteration/index.js.map +1 -1
- package/dist/tools/builtins/__tests__/edit.test.js +206 -54
- package/dist/tools/builtins/__tests__/edit.test.js.map +1 -1
- package/dist/tools/builtins/__tests__/payload-budget.test.js +30 -16
- package/dist/tools/builtins/__tests__/payload-budget.test.js.map +1 -1
- package/dist/tools/builtins/__tests__/write-file.test.js +37 -13
- package/dist/tools/builtins/__tests__/write-file.test.js.map +1 -1
- package/dist/tools/builtins/atomic-write-file.d.ts +13 -0
- package/dist/tools/builtins/atomic-write-file.d.ts.map +1 -0
- package/dist/tools/builtins/atomic-write-file.js +46 -0
- package/dist/tools/builtins/atomic-write-file.js.map +1 -0
- package/dist/tools/builtins/bash.js +2 -2
- package/dist/tools/builtins/bash.js.map +1 -1
- package/dist/tools/builtins/edit.d.ts +2 -5
- package/dist/tools/builtins/edit.d.ts.map +1 -1
- package/dist/tools/builtins/edit.js +107 -125
- package/dist/tools/builtins/edit.js.map +1 -1
- package/dist/tools/builtins/file-mutation-lock.d.ts +7 -0
- package/dist/tools/builtins/file-mutation-lock.d.ts.map +1 -0
- package/dist/tools/builtins/file-mutation-lock.js +25 -0
- package/dist/tools/builtins/file-mutation-lock.js.map +1 -0
- package/dist/tools/builtins/index.js +2 -2
- package/dist/tools/builtins/index.js.map +1 -1
- package/dist/tools/builtins/write-file.d.ts +1 -2
- package/dist/tools/builtins/write-file.d.ts.map +1 -1
- package/dist/tools/builtins/write-file.js +66 -40
- package/dist/tools/builtins/write-file.js.map +1 -1
- package/dist/tools/defineTool.d.ts +2 -0
- package/dist/tools/defineTool.d.ts.map +1 -1
- package/dist/tools/defineTool.js +2 -0
- package/dist/tools/defineTool.js.map +1 -1
- package/dist/types/provider/chat.d.ts +9 -0
- package/dist/types/provider/chat.d.ts.map +1 -1
- package/dist/types/sandbox/index.d.ts +4 -0
- package/dist/types/sandbox/index.d.ts.map +1 -1
- package/dist/types/sandbox/index.js.map +1 -1
- package/dist/types/tool/index.d.ts +14 -0
- package/dist/types/tool/index.d.ts.map +1 -1
- package/package.json +2 -1
- package/src/registry/tool/execute.test.ts +107 -7
- package/src/registry/tool/execute.ts +45 -10
- package/src/registry/toolset/catalog.test.ts +118 -0
- package/src/registry/toolset/catalog.ts +26 -4
- package/src/runtime/query/__tests__/capability-negotiation.test.ts +88 -0
- package/src/runtime/query/__tests__/deferred-tools.test.ts +111 -0
- package/src/runtime/query/__tests__/long-document-flow.test.ts +12 -12
- package/src/runtime/query/__tests__/stream-recovery.test.ts +1 -1
- package/src/runtime/query/executor.ts +2 -2
- package/src/runtime/query/iteration/index.ts +15 -0
- package/src/tools/builtins/__tests__/edit.test.ts +248 -64
- package/src/tools/builtins/__tests__/payload-budget.test.ts +40 -20
- package/src/tools/builtins/__tests__/write-file.test.ts +52 -13
- package/src/tools/builtins/atomic-write-file.ts +58 -0
- package/src/tools/builtins/bash.ts +2 -2
- package/src/tools/builtins/edit.ts +117 -163
- package/src/tools/builtins/file-mutation-lock.ts +27 -0
- package/src/tools/builtins/index.ts +2 -2
- package/src/tools/builtins/write-file.ts +68 -44
- package/src/tools/defineTool.ts +4 -0
- package/src/types/provider/chat.ts +14 -1
- package/src/types/sandbox/index.ts +4 -0
- package/src/types/tool/index.ts +14 -0
|
@@ -12,6 +12,17 @@ function makeTool(name, description = `${name} tool`) {
|
|
|
12
12
|
},
|
|
13
13
|
};
|
|
14
14
|
}
|
|
15
|
+
function makeCatalog() {
|
|
16
|
+
const catalog = new ToolCatalog();
|
|
17
|
+
catalog.registerSource({ id: 'host', kind: 'host_tool', name: 'Host' });
|
|
18
|
+
catalog.registerToolset({
|
|
19
|
+
id: 'tools',
|
|
20
|
+
sourceId: 'host',
|
|
21
|
+
name: 'Tools',
|
|
22
|
+
defaultPolicy: { enabled: true, loading: 'eager' },
|
|
23
|
+
});
|
|
24
|
+
return catalog;
|
|
25
|
+
}
|
|
15
26
|
describe('ToolCatalog', () => {
|
|
16
27
|
it('keeps sources, toolsets, and tools as separate records', () => {
|
|
17
28
|
const catalog = new ToolCatalog();
|
|
@@ -143,5 +154,95 @@ describe('ToolCatalog', () => {
|
|
|
143
154
|
expect(catalog.getTool('write_file')?.policy.loading).toBe('suspended');
|
|
144
155
|
expect(catalog.toLLMTools({ loading: ['suspended'] })).toEqual([]);
|
|
145
156
|
});
|
|
157
|
+
it('rejects catalog, definition, and LLM schema name mismatches', () => {
|
|
158
|
+
const catalog = makeCatalog();
|
|
159
|
+
expect(() => catalog.registerTool({
|
|
160
|
+
name: 'edit',
|
|
161
|
+
description: 'Edit a file',
|
|
162
|
+
sourceId: 'host',
|
|
163
|
+
toolsetId: 'tools',
|
|
164
|
+
policy: { enabled: true, loading: 'eager' },
|
|
165
|
+
definition: makeTool('write'),
|
|
166
|
+
})).toThrow(/definition name mismatch/);
|
|
167
|
+
expect(() => catalog.registerTool({
|
|
168
|
+
name: 'edit',
|
|
169
|
+
description: 'Edit a file',
|
|
170
|
+
sourceId: 'host',
|
|
171
|
+
toolsetId: 'tools',
|
|
172
|
+
policy: { enabled: true, loading: 'eager' },
|
|
173
|
+
llmSchema: {
|
|
174
|
+
type: 'function',
|
|
175
|
+
function: {
|
|
176
|
+
name: 'write',
|
|
177
|
+
description: 'Write',
|
|
178
|
+
parameters: { type: 'object' },
|
|
179
|
+
},
|
|
180
|
+
},
|
|
181
|
+
})).toThrow(/LLM schema name mismatch/);
|
|
182
|
+
});
|
|
183
|
+
it('rejects an explicit LLM schema that could replace an enforced definition schema', () => {
|
|
184
|
+
const catalog = makeCatalog();
|
|
185
|
+
const definition = {
|
|
186
|
+
...makeTool('edit'),
|
|
187
|
+
modelInputSchema: {
|
|
188
|
+
type: 'object',
|
|
189
|
+
properties: { path: { type: 'string' } },
|
|
190
|
+
required: ['path'],
|
|
191
|
+
additionalProperties: false,
|
|
192
|
+
},
|
|
193
|
+
enforceModelInput: true,
|
|
194
|
+
};
|
|
195
|
+
expect(() => catalog.registerTool({
|
|
196
|
+
name: 'edit',
|
|
197
|
+
description: 'Edit a file',
|
|
198
|
+
sourceId: 'host',
|
|
199
|
+
toolsetId: 'tools',
|
|
200
|
+
policy: { enabled: true, loading: 'eager' },
|
|
201
|
+
definition,
|
|
202
|
+
llmSchema: {
|
|
203
|
+
type: 'function',
|
|
204
|
+
function: {
|
|
205
|
+
name: 'edit',
|
|
206
|
+
description: 'Unsafe override',
|
|
207
|
+
parameters: { type: 'object' },
|
|
208
|
+
},
|
|
209
|
+
},
|
|
210
|
+
})).toThrow(/cannot combine an explicit llmSchema with enforceModelInput/);
|
|
211
|
+
});
|
|
212
|
+
it('rejects an enforced catalog definition without an explicit model schema', () => {
|
|
213
|
+
const catalog = makeCatalog();
|
|
214
|
+
expect(() => catalog.registerTool({
|
|
215
|
+
name: 'edit',
|
|
216
|
+
description: 'Edit a file',
|
|
217
|
+
sourceId: 'host',
|
|
218
|
+
toolsetId: 'tools',
|
|
219
|
+
policy: { enabled: true, loading: 'eager' },
|
|
220
|
+
definition: {
|
|
221
|
+
...makeTool('edit'),
|
|
222
|
+
enforceModelInput: true,
|
|
223
|
+
},
|
|
224
|
+
})).toThrow(/enforceModelInput.*modelInputSchema/);
|
|
225
|
+
});
|
|
226
|
+
it('uses a definition model schema when no explicit catalog schema overrides it', () => {
|
|
227
|
+
const registry = new ToolRegistry();
|
|
228
|
+
const modelInputSchema = {
|
|
229
|
+
type: 'object',
|
|
230
|
+
properties: { new_string: { type: 'string' } },
|
|
231
|
+
required: ['new_string'],
|
|
232
|
+
additionalProperties: false,
|
|
233
|
+
};
|
|
234
|
+
registry.register({
|
|
235
|
+
...makeTool('edit'),
|
|
236
|
+
modelInputSchema,
|
|
237
|
+
});
|
|
238
|
+
const catalog = createToolCatalogFromRegistry(registry);
|
|
239
|
+
const firstSchema = catalog.toLLMTools()[0]?.function.parameters;
|
|
240
|
+
expect(firstSchema).toEqual(modelInputSchema);
|
|
241
|
+
expect(firstSchema).not.toBe(modelInputSchema);
|
|
242
|
+
(firstSchema?.properties).newStr = { type: 'string' };
|
|
243
|
+
const nextSchema = catalog.toLLMTools()[0]?.function.parameters;
|
|
244
|
+
expect(nextSchema).toEqual(modelInputSchema);
|
|
245
|
+
expect(JSON.stringify(nextSchema)).not.toContain('newStr');
|
|
246
|
+
});
|
|
146
247
|
});
|
|
147
248
|
//# sourceMappingURL=catalog.test.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"catalog.test.js","sourceRoot":"","sources":["../../../src/registry/toolset/catalog.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AACjD,OAAO,EAAE,WAAW,EAAE,6BAA6B,EAAE,MAAM,cAAc,CAAA;AAEzE,SAAS,QAAQ,CAAC,IAAY,EAAE,WAAW,GAAG,GAAG,IAAI,OAAO;IAC3D,OAAO;QACN,IAAI;QACJ,WAAW;QACX,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC;QACvD,KAAK,CAAC,OAAO;YACZ,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,MAAM,EAAE,CAAA;QAChD,CAAC;KACD,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACjE,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,OAAO,CAAC,cAAc,CAAC;YACtB,EAAE,EAAE,qBAAqB;YACzB,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE;gBACV,IAAI,EAAE,iBAAiB;gBACvB,GAAG,EAAE,qCAAqC;gBAC1C,SAAS,EAAE,iBAAiB;aAC5B;SACD,CAAC,CAAA;QACF,OAAO,CAAC,eAAe,CAAC;YACvB,EAAE,EAAE,6BAA6B;YACjC,QAAQ,EAAE,qBAAqB;YAC/B,IAAI,EAAE,yBAAyB;YAC/B,aAAa,EAAE;gBACd,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,IAAI;gBACf,gBAAgB,EAAE,cAAc;aAChC;SACD,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,uBAAuB;YAC7B,WAAW,EAAE,gCAAgC;YAC7C,QAAQ,EAAE,qBAAqB;YAC/B,SAAS,EAAE,6BAA6B;YACxC,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE;SAC/D,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC9C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1E,uBAAuB;SACvB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC5E,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAA;QACnC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,sBAAsB,EAAE,0BAA0B,CAAC,EAAE,UAAU,CAAC,CAAA;QAC3F,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,EAAE,UAAU,CAAC,CAAA;QACxF,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,CAAA;QAE1D,MAAM,OAAO,GAAG,6BAA6B,CAAC,QAAQ,EAAE;YACvD,MAAM,EAAE;gBACP,EAAE,EAAE,MAAM;gBACV,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,qCAAqC;aAClD;YACD,OAAO,EAAE;gBACR,EAAE,EAAE,cAAc;gBAClB,IAAI,EAAE,oBAAoB;gBAC1B,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;aAClD;SACD,CAAC,CAAA;QAEF,MAAM,CACL,OAAO,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,OAAO,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CACpF,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAA;QACnC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACzF,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,OAAO,CAAC,cAAc,CAAC;YACtB,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,uBAAuB;SACpC,CAAC,CAAA;QACF,OAAO,CAAC,eAAe,CAAC;YACvB,EAAE,EAAE,QAAQ;YACZ,QAAQ,EAAE,MAAM;YAChB,IAAI,EAAE,eAAe;YACrB,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAClD,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,QAAQ;YACnB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3C,CAAC,CAAA;QAEF,2EAA2E;QAC3E,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAA;QAE5E,sCAAsC;QACtC,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;QAEhD,+CAA+C;QAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;QACjD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC/E,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;QACvE,OAAO,CAAC,eAAe,CAAC;YACvB,EAAE,EAAE,IAAI;YACR,QAAQ,EAAE,MAAM;YAChB,IAAI,EAAE,OAAO;YACb,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAClD,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3C,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE;SAC5D,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC,CAAA;QAChF,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,EAAE,KAAK,IAAI,CAAC,CAAA;QAC3E,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC,EAAE,KAAK,IAAI,CAAC,CAAA;QAC7E,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACpE,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAA;QACnC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAA;QACxC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,UAAU,CAAC,CAAA;QACrD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,WAAW,CAAC,CAAA;QAEtD,MAAM,OAAO,GAAG,6BAA6B,CAAC,QAAQ,CAAC,CAAA;QAEvD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAClE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACtE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACvE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACnE,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
1
|
+
{"version":3,"file":"catalog.test.js","sourceRoot":"","sources":["../../../src/registry/toolset/catalog.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AACjD,OAAO,EAAE,WAAW,EAAE,6BAA6B,EAAE,MAAM,cAAc,CAAA;AAEzE,SAAS,QAAQ,CAAC,IAAY,EAAE,WAAW,GAAG,GAAG,IAAI,OAAO;IAC3D,OAAO;QACN,IAAI;QACJ,WAAW;QACX,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC;QACvD,KAAK,CAAC,OAAO;YACZ,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,MAAM,EAAE,CAAA;QAChD,CAAC;KACD,CAAA;AACF,CAAC;AAED,SAAS,WAAW;IACnB,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;IACjC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;IACvE,OAAO,CAAC,eAAe,CAAC;QACvB,EAAE,EAAE,OAAO;QACX,QAAQ,EAAE,MAAM;QAChB,IAAI,EAAE,OAAO;QACb,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;KAClD,CAAC,CAAA;IACF,OAAO,OAAO,CAAA;AACf,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACjE,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,OAAO,CAAC,cAAc,CAAC;YACtB,EAAE,EAAE,qBAAqB;YACzB,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE;gBACV,IAAI,EAAE,iBAAiB;gBACvB,GAAG,EAAE,qCAAqC;gBAC1C,SAAS,EAAE,iBAAiB;aAC5B;SACD,CAAC,CAAA;QACF,OAAO,CAAC,eAAe,CAAC;YACvB,EAAE,EAAE,6BAA6B;YACjC,QAAQ,EAAE,qBAAqB;YAC/B,IAAI,EAAE,yBAAyB;YAC/B,aAAa,EAAE;gBACd,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,UAAU;gBACnB,SAAS,EAAE,IAAI;gBACf,gBAAgB,EAAE,cAAc;aAChC;SACD,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,uBAAuB;YAC7B,WAAW,EAAE,gCAAgC;YAC7C,QAAQ,EAAE,qBAAqB;YAC/B,SAAS,EAAE,6BAA6B;YACxC,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE;SAC/D,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC9C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC1E,uBAAuB;SACvB,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC5E,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAA;QACnC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,sBAAsB,EAAE,0BAA0B,CAAC,EAAE,UAAU,CAAC,CAAA;QAC3F,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,EAAE,UAAU,CAAC,CAAA;QACxF,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC,CAAA;QAE1D,MAAM,OAAO,GAAG,6BAA6B,CAAC,QAAQ,EAAE;YACvD,MAAM,EAAE;gBACP,EAAE,EAAE,MAAM;gBACV,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,qCAAqC;aAClD;YACD,OAAO,EAAE;gBACR,EAAE,EAAE,cAAc;gBAClB,IAAI,EAAE,oBAAoB;gBAC1B,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;aAClD;SACD,CAAC,CAAA;QAEF,MAAM,CACL,OAAO,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,OAAO,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CACpF,CAAC,OAAO,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAA;QACnC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;QACzF,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,OAAO,CAAC,cAAc,CAAC;YACtB,EAAE,EAAE,MAAM;YACV,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,uBAAuB;SACpC,CAAC,CAAA;QACF,OAAO,CAAC,eAAe,CAAC;YACvB,EAAE,EAAE,QAAQ;YACZ,QAAQ,EAAE,MAAM;YAChB,IAAI,EAAE,eAAe;YACrB,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAClD,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,QAAQ;YACnB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3C,CAAC,CAAA;QAEF,2EAA2E;QAC3E,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,MAAM,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,CAAA;QAE5E,sCAAsC;QACtC,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;QAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;QAEhD,+CAA+C;QAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,cAAc,CAAC,CAAA;QACjD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC/E,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;QACjC,OAAO,CAAC,cAAc,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;QACvE,OAAO,CAAC,eAAe,CAAC;YACvB,EAAE,EAAE,IAAI;YACR,QAAQ,EAAE,MAAM;YAChB,IAAI,EAAE,OAAO;YACb,aAAa,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAClD,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3C,CAAC,CAAA;QACF,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,IAAI;YACf,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE;SAC5D,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC,CAAA;QAChF,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,EAAE,KAAK,IAAI,CAAC,CAAA;QAC3E,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CAAC,EAAE,KAAK,IAAI,CAAC,CAAA;QAC7E,MAAM,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACpE,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAA;QACnC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAA;QACxC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,UAAU,CAAC,CAAA;QACrD,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,WAAW,CAAC,CAAA;QAEtD,MAAM,OAAO,GAAG,6BAA6B,CAAC,QAAQ,CAAC,CAAA;QAEvD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAClE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACtE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACvE,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,OAAO,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACnE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACtE,MAAM,OAAO,GAAG,WAAW,EAAE,CAAA;QAC7B,MAAM,CAAC,GAAG,EAAE,CACX,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,OAAO;YAClB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;YAC3C,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC;SAC7B,CAAC,CACF,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAA;QAErC,MAAM,CAAC,GAAG,EAAE,CACX,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,OAAO;YAClB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;YAC3C,SAAS,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACT,IAAI,EAAE,OAAO;oBACb,WAAW,EAAE,OAAO;oBACpB,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAC9B;aACD;SACD,CAAC,CACF,CAAC,OAAO,CAAC,0BAA0B,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;QAC1F,MAAM,OAAO,GAAG,WAAW,EAAE,CAAA;QAC7B,MAAM,UAAU,GAAmB;YAClC,GAAG,QAAQ,CAAC,MAAM,CAAC;YACnB,gBAAgB,EAAE;gBACjB,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACxC,QAAQ,EAAE,CAAC,MAAM,CAAC;gBAClB,oBAAoB,EAAE,KAAK;aAC3B;YACD,iBAAiB,EAAE,IAAI;SACvB,CAAA;QAED,MAAM,CAAC,GAAG,EAAE,CACX,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,OAAO;YAClB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;YAC3C,UAAU;YACV,SAAS,EAAE;gBACV,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE;oBACT,IAAI,EAAE,MAAM;oBACZ,WAAW,EAAE,iBAAiB;oBAC9B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;iBAC9B;aACD;SACD,CAAC,CACF,CAAC,OAAO,CAAC,6DAA6D,CAAC,CAAA;IACzE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;QAClF,MAAM,OAAO,GAAG,WAAW,EAAE,CAAA;QAC7B,MAAM,CAAC,GAAG,EAAE,CACX,OAAO,CAAC,YAAY,CAAC;YACpB,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,aAAa;YAC1B,QAAQ,EAAE,MAAM;YAChB,SAAS,EAAE,OAAO;YAClB,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE;YAC3C,UAAU,EAAE;gBACX,GAAG,QAAQ,CAAC,MAAM,CAAC;gBACnB,iBAAiB,EAAE,IAAI;aACvB;SACD,CAAC,CACF,CAAC,OAAO,CAAC,qCAAqC,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAA;QACnC,MAAM,gBAAgB,GAAG;YACxB,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC9C,QAAQ,EAAE,CAAC,YAAY,CAAC;YACxB,oBAAoB,EAAE,KAAK;SAC3B,CAAA;QACD,QAAQ,CAAC,QAAQ,CAAC;YACjB,GAAG,QAAQ,CAAC,MAAM,CAAC;YACnB,gBAAgB;SAChB,CAAC,CAAA;QAEF,MAAM,OAAO,GAAG,6BAA6B,CAAC,QAAQ,CAAC,CAAA;QACvD,MAAM,WAAW,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,UAAU,CAAA;QAChE,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAA;QAC7C,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAC7C;QAAA,CAAC,WAAW,EAAE,UAAsC,CAAA,CAAC,MAAM,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAA;QAEjF,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,UAAU,CAAA;QAC/D,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAA;QAC5C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;IAC3D,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -28,6 +28,30 @@ class CapturingProvider {
|
|
|
28
28
|
yield { id: 'msg_1', delta: {}, finishReason: 'stop', usage: ZERO_USAGE };
|
|
29
29
|
}
|
|
30
30
|
}
|
|
31
|
+
class ClosingSummaryProvider {
|
|
32
|
+
id = 'closing-summary';
|
|
33
|
+
name = 'Closing Summary Provider';
|
|
34
|
+
calls = [];
|
|
35
|
+
async *chatStream(params) {
|
|
36
|
+
this.calls.push(params);
|
|
37
|
+
if (this.calls.length === 1) {
|
|
38
|
+
yield {
|
|
39
|
+
id: 'msg_empty',
|
|
40
|
+
delta: {},
|
|
41
|
+
finishReason: 'stop',
|
|
42
|
+
usage: ZERO_USAGE,
|
|
43
|
+
};
|
|
44
|
+
return;
|
|
45
|
+
}
|
|
46
|
+
yield { id: 'msg_summary', delta: { content: 'closing summary' } };
|
|
47
|
+
yield {
|
|
48
|
+
id: 'msg_summary',
|
|
49
|
+
delta: {},
|
|
50
|
+
finishReason: 'stop',
|
|
51
|
+
usage: ZERO_USAGE,
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
}
|
|
31
55
|
const NO_TOOLS_CAPABILITIES = {
|
|
32
56
|
supportsTools: false,
|
|
33
57
|
supportsStreaming: true,
|
|
@@ -48,6 +72,24 @@ function registerEchoTool(tools) {
|
|
|
48
72
|
execute: async () => ({ success: true, output: 'ok' }),
|
|
49
73
|
});
|
|
50
74
|
}
|
|
75
|
+
function registerEnforcedTool(tools, name) {
|
|
76
|
+
tools.register({
|
|
77
|
+
name,
|
|
78
|
+
description: `${name} tool`,
|
|
79
|
+
inputSchema: z.object({
|
|
80
|
+
new_string: z.string().optional(),
|
|
81
|
+
newStr: z.string().optional(),
|
|
82
|
+
}),
|
|
83
|
+
modelInputSchema: {
|
|
84
|
+
type: 'object',
|
|
85
|
+
properties: { new_string: { type: 'string' } },
|
|
86
|
+
required: ['new_string'],
|
|
87
|
+
additionalProperties: false,
|
|
88
|
+
},
|
|
89
|
+
enforceModelInput: true,
|
|
90
|
+
execute: async () => ({ success: true, output: 'ok' }),
|
|
91
|
+
});
|
|
92
|
+
}
|
|
51
93
|
function baseParams(provider, tools, workingDirectory) {
|
|
52
94
|
return {
|
|
53
95
|
provider,
|
|
@@ -133,6 +175,43 @@ describe('query() capability negotiation', () => {
|
|
|
133
175
|
expect(run.status).toBe('completed');
|
|
134
176
|
expect(provider.lastParams?.tools?.map((t) => t.function.name)).toContain('echo');
|
|
135
177
|
});
|
|
178
|
+
it('propagates enforcement names from the exact allowed and cache-stable tool schemas', async () => {
|
|
179
|
+
const provider = new CapturingProvider();
|
|
180
|
+
const tools = new ToolRegistry();
|
|
181
|
+
registerEnforcedTool(tools, 'edit');
|
|
182
|
+
registerEnforcedTool(tools, 'cached_edit');
|
|
183
|
+
registerEnforcedTool(tools, 'excluded_edit');
|
|
184
|
+
tools.suspendAll();
|
|
185
|
+
tools.activate(['edit']);
|
|
186
|
+
const run = await drainQuery({
|
|
187
|
+
...baseParams(provider, tools, await mkWorkdir()),
|
|
188
|
+
allowedTools: ['edit', 'cached_edit'],
|
|
189
|
+
messages: [createUserMessage('hello')],
|
|
190
|
+
});
|
|
191
|
+
expect(run.status).toBe('completed');
|
|
192
|
+
expect(provider.lastParams?.tools?.map((tool) => tool.function.name)).toEqual([
|
|
193
|
+
'edit',
|
|
194
|
+
'cached_edit',
|
|
195
|
+
]);
|
|
196
|
+
expect(provider.lastParams?.enforceToolInputSchema).toEqual(['edit', 'cached_edit']);
|
|
197
|
+
expect(JSON.stringify(provider.lastParams?.tools)).not.toContain('newStr');
|
|
198
|
+
});
|
|
199
|
+
it('keeps the same enforcement hint on the closing-summary provider call', async () => {
|
|
200
|
+
const provider = new ClosingSummaryProvider();
|
|
201
|
+
const tools = new ToolRegistry();
|
|
202
|
+
registerEnforcedTool(tools, 'edit');
|
|
203
|
+
const run = await drainQuery({
|
|
204
|
+
...baseParams(provider, tools, await mkWorkdir()),
|
|
205
|
+
messages: [createUserMessage('hello')],
|
|
206
|
+
});
|
|
207
|
+
expect(run.status).toBe('completed');
|
|
208
|
+
expect(provider.calls).toHaveLength(2);
|
|
209
|
+
for (const call of provider.calls) {
|
|
210
|
+
expect(call.tools?.map((tool) => tool.function.name)).toEqual(['edit']);
|
|
211
|
+
expect(call.enforceToolInputSchema).toEqual(['edit']);
|
|
212
|
+
}
|
|
213
|
+
expect(provider.calls[1]?.toolChoice).toBe('none');
|
|
214
|
+
});
|
|
136
215
|
it('emits a vision capability_warning when attachments hit a no-vision provider', async () => {
|
|
137
216
|
const provider = new CapturingProvider(NO_VISION_CAPABILITIES);
|
|
138
217
|
const events = [];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"capability-negotiation.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/capability-negotiation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACxD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EACN,gCAAgC,EAChC,2BAA2B,GAC3B,MAAM,mCAAmC,CAAA;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAEhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AASnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,iBAAiB;IACb,EAAE,GAAG,WAAW,CAAA;IAChB,IAAI,GAAG,oBAAoB,CAAA;IAC3B,YAAY,CAAuB;IAC5C,UAAU,CAAuB;IAEjC,YAAY,YAAmC;QAC9C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAA;IACjC,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAA;QACjD,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAA;IAC1E,CAAC;CACD;AAED,MAAM,qBAAqB,GAAyB;IACnD,aAAa,EAAE,KAAK;IACpB,iBAAiB,EAAE,IAAI;IACvB,uBAAuB,EAAE,KAAK;IAC9B,cAAc,EAAE,IAAI;CACpB,CAAA;AAED,MAAM,sBAAsB,GAAyB;IACpD,aAAa,EAAE,IAAI;IACnB,iBAAiB,EAAE,IAAI;IACvB,uBAAuB,EAAE,IAAI;IAC7B,cAAc,EAAE,KAAK;CACrB,CAAA;AAED,SAAS,gBAAgB,CAAC,KAAmB;IAC5C,KAAK,CAAC,QAAQ,CAAC;QACd,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,qBAAqB;QAClC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QAC3C,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;KACtD,CAAC,CAAA;AACH,CAAC;AAED,SAAS,UAAU,CAAC,QAAqB,EAAE,KAAmB,EAAE,gBAAwB;IACvF,OAAO;QACN,QAAQ;QACR,KAAK;QACL,SAAS,EAAE;YACV,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,KAAK;YAChB,WAAW,EAAE,OAAO;YACpB,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,GAAG;SACtB;QACD,OAAO,EAAE,YAAY;QACrB,SAAS,EAAE,YAAY;QACvB,gBAAgB;QAChB,SAAS,EAAE,gBAA6B;QACxC,QAAQ,EAAE,gBAA4B;QACtC,SAAS,EAAE,gBAA6B;QACxC,QAAQ,EAAE,gBAA4B;KACtC,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,2BAA2B,CAAC,IAAI,iBAAiB,EAAE,CAAC,CAAC,CAAC,OAAO,CACnE,gCAAgC,CAChC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;QACtF,MAAM,QAAQ,GAAG,2BAA2B,CAAC;YAC5C,YAAY,EAAE;gBACb,aAAa,EAAE,KAAK;gBACpB,iBAAiB,EAAE,IAAI;gBACvB,uBAAuB,EAAE,KAAK;aAC9B;SACD,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAC1C,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC/C,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,KAAK,UAAU,SAAS;QACvB,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAA;QAC9D,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAClB,OAAO,GAAG,CAAA;IACX,CAAC;IAED,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC5F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,qBAAqB,CAAC,CAAA;QAC7D,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,gBAAgB,CAAC,KAAK,CAAC,CAAA;QACvB,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;SACtC,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAEpC,oEAAoE;QACpE,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,aAAa,EAAE,CAAA;QAClD,6CAA6C;QAC7C,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;aACrB,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAA;QACvD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAE1C,6CAA6C;QAC7C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAC1B,CAAC,CAAC,EAA0D,EAAE,CAC7D,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAChC,CAAA;QACD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACzC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;IAC/C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAEvB,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;IAClF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC5F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,sBAAsB,CAAC,CAAA;QAC9D,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,YAAY,EAAE,EAAE,MAAM,SAAS,EAAE,CAAC;YAC9D,QAAQ,EAAE;gBACT,iBAAiB,CAAC,wBAAwB,EAAE;oBAC3C,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE;iBAC5C,CAAC;aACF;SACD,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAC1B,CAAC,CAAC,EAA0D,EAAE,CAC7D,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAChC,CAAA;QACD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC1C,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,sBAAsB,CAAC,CAAA;QAC9D,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,UAAU,CACf;YACC,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,YAAY,EAAE,EAAE,MAAM,SAAS,EAAE,CAAC;YAC9D,QAAQ,EAAE,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;SAC3C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,qBAAqB,CAAC,CAAA;QAC7D,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAEvB,MAAM,MAAM,CACX,UAAU,CAAC;YACV,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACtC,kBAAkB,EAAE,IAAI;SACxB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,sBAAsB,CAAC,CAAA;QAE9D,MAAM,MAAM,CACX,UAAU,CAAC;YACV,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,YAAY,EAAE,EAAE,MAAM,SAAS,EAAE,CAAC;YAC9D,QAAQ,EAAE,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YACzF,kBAAkB,EAAE,IAAI;SACxB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
1
|
+
{"version":3,"file":"capability-negotiation.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/capability-negotiation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACxD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EACN,gCAAgC,EAChC,2BAA2B,GAC3B,MAAM,mCAAmC,CAAA;AAC1C,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAEhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AASnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,iBAAiB;IACb,EAAE,GAAG,WAAW,CAAA;IAChB,IAAI,GAAG,oBAAoB,CAAA;IAC3B,YAAY,CAAuB;IAC5C,UAAU,CAAuB;IAEjC,YAAY,YAAmC;QAC9C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAA;IACjC,CAAC;IAED,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAA;QACjD,MAAM,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,CAAA;IAC1E,CAAC;CACD;AAED,MAAM,sBAAsB;IAClB,EAAE,GAAG,iBAAiB,CAAA;IACtB,IAAI,GAAG,0BAA0B,CAAA;IACjC,KAAK,GAA2B,EAAE,CAAA;IAE3C,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACvB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM;gBACL,EAAE,EAAE,WAAW;gBACf,KAAK,EAAE,EAAE;gBACT,YAAY,EAAE,MAAM;gBACpB,KAAK,EAAE,UAAU;aACjB,CAAA;YACD,OAAM;QACP,CAAC;QACD,MAAM,EAAE,EAAE,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,iBAAiB,EAAE,EAAE,CAAA;QAClE,MAAM;YACL,EAAE,EAAE,aAAa;YACjB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,MAAM,qBAAqB,GAAyB;IACnD,aAAa,EAAE,KAAK;IACpB,iBAAiB,EAAE,IAAI;IACvB,uBAAuB,EAAE,KAAK;IAC9B,cAAc,EAAE,IAAI;CACpB,CAAA;AAED,MAAM,sBAAsB,GAAyB;IACpD,aAAa,EAAE,IAAI;IACnB,iBAAiB,EAAE,IAAI;IACvB,uBAAuB,EAAE,IAAI;IAC7B,cAAc,EAAE,KAAK;CACrB,CAAA;AAED,SAAS,gBAAgB,CAAC,KAAmB;IAC5C,KAAK,CAAC,QAAQ,CAAC;QACd,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,qBAAqB;QAClC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QAC3C,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;KACtD,CAAC,CAAA;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAmB,EAAE,IAAY;IAC9D,KAAK,CAAC,QAAQ,CAAC;QACd,IAAI;QACJ,WAAW,EAAE,GAAG,IAAI,OAAO;QAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;YACjC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAC7B,CAAC;QACF,gBAAgB,EAAE;YACjB,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC9C,QAAQ,EAAE,CAAC,YAAY,CAAC;YACxB,oBAAoB,EAAE,KAAK;SAC3B;QACD,iBAAiB,EAAE,IAAI;QACvB,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;KACtD,CAAC,CAAA;AACH,CAAC;AAED,SAAS,UAAU,CAAC,QAAqB,EAAE,KAAmB,EAAE,gBAAwB;IACvF,OAAO;QACN,QAAQ;QACR,KAAK;QACL,SAAS,EAAE;YACV,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,KAAK;YAChB,WAAW,EAAE,OAAO;YACpB,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,GAAG;SACtB;QACD,OAAO,EAAE,YAAY;QACrB,SAAS,EAAE,YAAY;QACvB,gBAAgB;QAChB,SAAS,EAAE,gBAA6B;QACxC,QAAQ,EAAE,gBAA4B;QACtC,SAAS,EAAE,gBAA6B;QACxC,QAAQ,EAAE,gBAA4B;KACtC,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,2BAA2B,CAAC,IAAI,iBAAiB,EAAE,CAAC,CAAC,CAAC,OAAO,CACnE,gCAAgC,CAChC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;QACtF,MAAM,QAAQ,GAAG,2BAA2B,CAAC;YAC5C,YAAY,EAAE;gBACb,aAAa,EAAE,KAAK;gBACpB,iBAAiB,EAAE,IAAI;gBACvB,uBAAuB,EAAE,KAAK;aAC9B;SACD,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAC1C,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;IAC/C,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,KAAK,UAAU,SAAS;QACvB,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAA;QAC9D,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAClB,OAAO,GAAG,CAAA;IACX,CAAC;IAED,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC5F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,qBAAqB,CAAC,CAAA;QAC7D,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,gBAAgB,CAAC,KAAK,CAAC,CAAA;QACvB,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;SACtC,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAEpC,oEAAoE;QACpE,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,aAAa,EAAE,CAAA;QAClD,6CAA6C;QAC7C,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;aACrB,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAA;QACvD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAE1C,6CAA6C;QAC7C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAC1B,CAAC,CAAC,EAA0D,EAAE,CAC7D,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAChC,CAAA;QACD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACzC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QAC7C,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;IAC/C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC3F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAEvB,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;IAClF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QAClG,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,oBAAoB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;QACnC,oBAAoB,CAAC,KAAK,EAAE,aAAa,CAAC,CAAA;QAC1C,oBAAoB,CAAC,KAAK,EAAE,eAAe,CAAC,CAAA;QAC5C,KAAK,CAAC,UAAU,EAAE,CAAA;QAClB,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;QAExB,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,YAAY,EAAE,CAAC,MAAM,EAAE,aAAa,CAAC;YACrC,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC7E,MAAM;YACN,aAAa;SACb,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,sBAAsB,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAA;QACpF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,QAAQ,GAAG,IAAI,sBAAsB,EAAE,CAAA;QAC7C,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,oBAAoB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;QAEnC,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;SACtC,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACtC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;YACvE,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,CAAA;QACtD,CAAC;QACD,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;QAC5F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,sBAAsB,CAAC,CAAA;QAC9D,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,YAAY,EAAE,EAAE,MAAM,SAAS,EAAE,CAAC;YAC9D,QAAQ,EAAE;gBACT,iBAAiB,CAAC,wBAAwB,EAAE;oBAC3C,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE;iBAC5C,CAAC;aACF;SACD,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAC1B,CAAC,CAAC,EAA0D,EAAE,CAC7D,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAChC,CAAA;QACD,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC1C,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,sBAAsB,CAAC,CAAA;QAC9D,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,UAAU,CACf;YACC,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,YAAY,EAAE,EAAE,MAAM,SAAS,EAAE,CAAC;YAC9D,QAAQ,EAAE,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;SAC3C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,qBAAqB,CAAC,CAAA;QAC7D,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,gBAAgB,CAAC,KAAK,CAAC,CAAA;QAEvB,MAAM,MAAM,CACX,UAAU,CAAC;YACV,GAAG,UAAU,CAAC,QAAQ,EAAE,KAAK,EAAE,MAAM,SAAS,EAAE,CAAC;YACjD,QAAQ,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACtC,kBAAkB,EAAE,IAAI;SACxB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,QAAQ,GAAG,IAAI,iBAAiB,CAAC,sBAAsB,CAAC,CAAA;QAE9D,MAAM,MAAM,CACX,UAAU,CAAC;YACV,GAAG,UAAU,CAAC,QAAQ,EAAE,IAAI,YAAY,EAAE,EAAE,MAAM,SAAS,EAAE,CAAC;YAC9D,QAAQ,EAAE,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YACzF,kBAAkB,EAAE,IAAI;SACxB,CAAC,CACF,CAAC,OAAO,CAAC,OAAO,CAAC,uBAAuB,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -32,6 +32,55 @@ class CapturingProvider {
|
|
|
32
32
|
};
|
|
33
33
|
}
|
|
34
34
|
}
|
|
35
|
+
class DeferredActivationProvider {
|
|
36
|
+
id = 'deferred-activation';
|
|
37
|
+
name = 'Deferred Activation Provider';
|
|
38
|
+
calls = [];
|
|
39
|
+
async *chatStream(params) {
|
|
40
|
+
this.calls.push(params);
|
|
41
|
+
if (this.calls.length === 1) {
|
|
42
|
+
yield {
|
|
43
|
+
id: 'msg_search',
|
|
44
|
+
delta: {
|
|
45
|
+
toolCalls: [
|
|
46
|
+
{
|
|
47
|
+
index: 0,
|
|
48
|
+
id: 'toolu_search',
|
|
49
|
+
type: 'function',
|
|
50
|
+
function: { name: SearchToolsTool.name },
|
|
51
|
+
},
|
|
52
|
+
],
|
|
53
|
+
},
|
|
54
|
+
};
|
|
55
|
+
yield {
|
|
56
|
+
id: 'msg_search',
|
|
57
|
+
delta: {
|
|
58
|
+
toolCalls: [
|
|
59
|
+
{
|
|
60
|
+
index: 0,
|
|
61
|
+
id: 'toolu_search',
|
|
62
|
+
function: { arguments: '{"query":"canonical_key"}' },
|
|
63
|
+
},
|
|
64
|
+
],
|
|
65
|
+
},
|
|
66
|
+
};
|
|
67
|
+
yield {
|
|
68
|
+
id: 'msg_search',
|
|
69
|
+
delta: {},
|
|
70
|
+
finishReason: 'tool_calls',
|
|
71
|
+
usage: ZERO_USAGE,
|
|
72
|
+
};
|
|
73
|
+
return;
|
|
74
|
+
}
|
|
75
|
+
yield { id: 'msg_done', delta: { content: 'done' } };
|
|
76
|
+
yield {
|
|
77
|
+
id: 'msg_done',
|
|
78
|
+
delta: {},
|
|
79
|
+
finishReason: 'stop',
|
|
80
|
+
usage: ZERO_USAGE,
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
}
|
|
35
84
|
function registerDeferredDocumentTool(tools, name = 'generate_document') {
|
|
36
85
|
tools.register({
|
|
37
86
|
name,
|
|
@@ -124,6 +173,57 @@ describe('query deferred tool discovery', () => {
|
|
|
124
173
|
expect(systemPrompt).toContain('Use search_tools to load these before use:');
|
|
125
174
|
expect(systemPrompt).toContain('- generate_document');
|
|
126
175
|
});
|
|
176
|
+
it('adds enforcement only after a deferred model-schema tool is activated', async () => {
|
|
177
|
+
const provider = new DeferredActivationProvider();
|
|
178
|
+
const tools = new ToolRegistry();
|
|
179
|
+
tools.register({
|
|
180
|
+
name: 'deferred_edit',
|
|
181
|
+
description: 'Mutate a document using the provided content.',
|
|
182
|
+
inputSchema: z.object({ legacyKey: z.string() }),
|
|
183
|
+
modelInputSchema: {
|
|
184
|
+
type: 'object',
|
|
185
|
+
properties: { canonical_key: { type: 'string' } },
|
|
186
|
+
required: ['canonical_key'],
|
|
187
|
+
additionalProperties: false,
|
|
188
|
+
},
|
|
189
|
+
enforceModelInput: true,
|
|
190
|
+
execute: async () => ({ success: true, output: 'edited' }),
|
|
191
|
+
}, 'deferred');
|
|
192
|
+
const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-deferred-enforcement-'));
|
|
193
|
+
workdirs.push(workingDirectory);
|
|
194
|
+
const run = await drainQuery({
|
|
195
|
+
provider,
|
|
196
|
+
tools,
|
|
197
|
+
allowedTools: ['deferred_edit'],
|
|
198
|
+
runConfig: {
|
|
199
|
+
model: 'mock-model',
|
|
200
|
+
timeoutMs: 5_000,
|
|
201
|
+
tokenBudget: 100_000,
|
|
202
|
+
maxIterations: 3,
|
|
203
|
+
maxResponseTokens: 256,
|
|
204
|
+
},
|
|
205
|
+
agentId: 'agent_test',
|
|
206
|
+
agentName: 'Test Agent',
|
|
207
|
+
messages: [createUserMessage('find and load the right tool')],
|
|
208
|
+
workingDirectory,
|
|
209
|
+
sessionId: 'ses_deferred_enforcement',
|
|
210
|
+
threadId: 'thd_deferred_enforcement',
|
|
211
|
+
projectId: 'prj_deferred_enforcement',
|
|
212
|
+
tenantId: 'tnt_deferred_enforcement',
|
|
213
|
+
});
|
|
214
|
+
expect(run.status).toBe('completed');
|
|
215
|
+
expect(provider.calls).toHaveLength(2);
|
|
216
|
+
expect(provider.calls[0]?.tools?.map((tool) => tool.function.name)).toEqual([
|
|
217
|
+
SearchToolsTool.name,
|
|
218
|
+
]);
|
|
219
|
+
expect(provider.calls[0]?.enforceToolInputSchema).toBeUndefined();
|
|
220
|
+
expect(provider.calls[1]?.tools?.map((tool) => tool.function.name)).toEqual([
|
|
221
|
+
'deferred_edit',
|
|
222
|
+
SearchToolsTool.name,
|
|
223
|
+
]);
|
|
224
|
+
expect(provider.calls[1]?.enforceToolInputSchema).toEqual(['deferred_edit']);
|
|
225
|
+
expect(provider.calls[1]?.tools?.[0]?.function.parameters).toEqual(tools.get('deferred_edit')?.modelInputSchema);
|
|
226
|
+
});
|
|
127
227
|
it('does not let search_tools reveal or activate deferred tools outside allowedTools', async () => {
|
|
128
228
|
const tools = new ToolRegistry();
|
|
129
229
|
registerDeferredDocumentTool(tools);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"deferred-tools.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/deferred-tools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACxD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAA;AAEzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAOnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,iBAAiB;IACb,EAAE,GAAG,WAAW,CAAA;IAChB,IAAI,GAAG,oBAAoB,CAAA;IACpC,UAAU,CAAuB;IAEjC,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;SAC1B,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,SAAS,4BAA4B,CAAC,KAAmB,EAAE,IAAI,GAAG,mBAAmB;IACpF,KAAK,CAAC,QAAQ,CACb;QACC,IAAI;QACJ,WAAW,EAAE,6CAA6C;QAC1D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACtB,CAAC;QACF,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;KAC7D,EACD,UAAU,CACV,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,CAAC;YACxD,gBAAgB;YAChB,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;YAC1C,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;SAC1C,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAClD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAClE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CACjC,qFAAqF,CACrF,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,YAAY,EAAE,CAAC,mBAAmB,CAAC;YACnC,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAC9C,gBAAgB;YAChB,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;YAClD,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;SAClD,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QACnC,4BAA4B,CAAC,KAAK,EAAE,0BAA0B,CAAC,CAAA;QAE/D,qEAAqE;QACrE,sEAAsE;QACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,WAAW,EAAE,EACtB;YACC,KAAK,EAAE,4BAAqC;YAC5C,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;YACnB,YAAY,EAAE,CAAC,mBAAmB,EAAE,eAAe,CAAC,IAAI,CAAC;SACzD,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wCAAwC,CAAC,CAAA;QACzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAA;QAC/D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QACnG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,qEAAqE;QACrE,4DAA4D;QAC5D,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAA;QACjF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,KAAK,CAAC,QAAQ,CACb;gBACC,IAAI;gBACJ,WAAW,EAAE,kBAAkB,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG;gBAChD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACzC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aACtD,EACD,UAAU,CACV,CAAA;QACF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,SAAS,EAAE,EACpB;YACC,KAAK,EAAE,oBAA6B;YACpC,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;SACnB,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;QAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACjC,SAAS,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;YAC5E,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;SACnD,CAAC,CAAA;QACF,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACnD,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACrD,CAAC;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
1
|
+
{"version":3,"file":"deferred-tools.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/deferred-tools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AACxD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAA;AAEzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAOnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,iBAAiB;IACb,EAAE,GAAG,WAAW,CAAA;IAChB,IAAI,GAAG,oBAAoB,CAAA;IACpC,UAAU,CAAuB;IAEjC,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,UAAU,GAAG,MAAM,CAAA;QACxB,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE;SAC1B,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,MAAM,0BAA0B;IACtB,EAAE,GAAG,qBAAqB,CAAA;IAC1B,IAAI,GAAG,8BAA8B,CAAA;IACrC,KAAK,GAA2B,EAAE,CAAA;IAE3C,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QAC7C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACvB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM;gBACL,EAAE,EAAE,YAAY;gBAChB,KAAK,EAAE;oBACN,SAAS,EAAE;wBACV;4BACC,KAAK,EAAE,CAAC;4BACR,EAAE,EAAE,cAAc;4BAClB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,eAAe,CAAC,IAAI,EAAE;yBACxC;qBACD;iBACD;aACD,CAAA;YACD,MAAM;gBACL,EAAE,EAAE,YAAY;gBAChB,KAAK,EAAE;oBACN,SAAS,EAAE;wBACV;4BACC,KAAK,EAAE,CAAC;4BACR,EAAE,EAAE,cAAc;4BAClB,QAAQ,EAAE,EAAE,SAAS,EAAE,2BAA2B,EAAE;yBACpD;qBACD;iBACD;aACD,CAAA;YACD,MAAM;gBACL,EAAE,EAAE,YAAY;gBAChB,KAAK,EAAE,EAAE;gBACT,YAAY,EAAE,YAAY;gBAC1B,KAAK,EAAE,UAAU;aACjB,CAAA;YACD,OAAM;QACP,CAAC;QAED,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAA;QACpD,MAAM;YACL,EAAE,EAAE,UAAU;YACd,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,SAAS,4BAA4B,CAAC,KAAmB,EAAE,IAAI,GAAG,mBAAmB;IACpF,KAAK,CAAC,QAAQ,CACb;QACC,IAAI;QACJ,WAAW,EAAE,6CAA6C;QAC1D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;YACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;SACtB,CAAC;QACF,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC;KAC7D,EACD,UAAU,CACV,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,yBAAyB,CAAC,CAAC;YACxD,gBAAgB;YAChB,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;YAC1C,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;SAC1C,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAClD,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAClE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CACjC,qFAAqF,CACrF,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAA;QACxC,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QAEnC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,YAAY,EAAE,CAAC,mBAAmB,CAAC;YACnC,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;YAC9C,gBAAgB;YAChB,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;YAClD,SAAS,EAAE,4BAAyC;YACpD,QAAQ,EAAE,4BAAwC;SAClD,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEnE,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAA;QAC5F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAA;QAEjD,MAAM,YAAY,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;aACxD,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC;aAC9C,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;aACjC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,4CAA4C,CAAC,CAAA;QAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,QAAQ,GAAG,IAAI,0BAA0B,EAAE,CAAA;QACjD,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,KAAK,CAAC,QAAQ,CACb;YACC,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,+CAA+C;YAC5D,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAChD,gBAAgB,EAAE;gBACjB,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE,aAAa,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACjD,QAAQ,EAAE,CAAC,eAAe,CAAC;gBAC3B,oBAAoB,EAAE,KAAK;aAC3B;YACD,iBAAiB,EAAE,IAAI;YACvB,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;SAC1D,EACD,UAAU,CACV,CAAA;QAED,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,6BAA6B,CAAC,CAAC,CAAA;QACrF,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAC/B,MAAM,GAAG,GAAG,MAAM,UAAU,CAAC;YAC5B,QAAQ;YACR,KAAK;YACL,YAAY,EAAE,CAAC,eAAe,CAAC;YAC/B,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,8BAA8B,CAAC,CAAC;YAC7D,gBAAgB;YAChB,SAAS,EAAE,0BAAuC;YAClD,QAAQ,EAAE,0BAAsC;YAChD,SAAS,EAAE,0BAAuC;YAClD,QAAQ,EAAE,0BAAsC;SAChD,CAAC,CAAA;QAEF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACtC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3E,eAAe,CAAC,IAAI;SACpB,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC,aAAa,EAAE,CAAA;QACjE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YAC3E,eAAe;YACf,eAAe,CAAC,IAAI;SACpB,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC,OAAO,CAAC,CAAC,eAAe,CAAC,CAAC,CAAA;QAC5E,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC,OAAO,CACjE,KAAK,CAAC,GAAG,CAAC,eAAe,CAAC,EAAE,gBAAgB,CAC5C,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,4BAA4B,CAAC,KAAK,CAAC,CAAA;QACnC,4BAA4B,CAAC,KAAK,EAAE,0BAA0B,CAAC,CAAA;QAE/D,qEAAqE;QACrE,sEAAsE;QACtE,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,WAAW,EAAE,EACtB;YACC,KAAK,EAAE,4BAAqC;YAC5C,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;YACnB,YAAY,EAAE,CAAC,mBAAmB,EAAE,eAAe,CAAC,IAAI,CAAC;SACzD,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wCAAwC,CAAC,CAAA;QACzE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAA;QAC/D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACnE,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oFAAoF,EAAE,KAAK,IAAI,EAAE;QACnG,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,qEAAqE;QACrE,4DAA4D;QAC5D,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAA;QACjF,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,KAAK,CAAC,QAAQ,CACb;gBACC,IAAI;gBACJ,WAAW,EAAE,kBAAkB,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG;gBAChD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBACzC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aACtD,EACD,UAAU,CACV,CAAA;QACF,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,OAAO,CAC3C,EAAE,KAAK,EAAE,SAAS,EAAE,EACpB;YACC,KAAK,EAAE,oBAA6B;YACpC,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;YACzC,GAAG,EAAE,EAAE;YACP,GAAG,EAAE,GAAG,EAAE,CAAC,SAAS;YACpB,YAAY,EAAE,KAAK;SACnB,CACD,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAA;QACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;QAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,aAAa,CAAC;YACjC,SAAS,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;YAC5E,KAAK,EAAE,CAAC;YACR,UAAU,EAAE,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC;SACnD,CAAC,CAAA;QACF,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACnD,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACrD,CAAC;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -89,27 +89,27 @@ describe('query long-document tool flow', () => {
|
|
|
89
89
|
name: 'write',
|
|
90
90
|
input: {
|
|
91
91
|
path: 'outputs/long-document-flow.md',
|
|
92
|
-
content: '# Long document flow\n\n{{
|
|
92
|
+
content: '# Long document flow\n\n{{CHUNK_001}}\n',
|
|
93
93
|
},
|
|
94
94
|
},
|
|
95
|
-
{
|
|
95
|
+
...chunks.map((chunk, index) => ({
|
|
96
96
|
name: 'edit',
|
|
97
97
|
input: {
|
|
98
98
|
path: 'outputs/long-document-flow.md',
|
|
99
|
-
|
|
100
|
-
|
|
99
|
+
old_string: `{{CHUNK_${String(index + 1).padStart(3, '0')}}}`,
|
|
100
|
+
new_string: `${chunk}\n{{CHUNK_${String(index + 2).padStart(3, '0')}}}`,
|
|
101
101
|
replace_all: false,
|
|
102
102
|
},
|
|
103
|
-
},
|
|
104
|
-
|
|
103
|
+
})),
|
|
104
|
+
{
|
|
105
105
|
name: 'edit',
|
|
106
106
|
input: {
|
|
107
107
|
path: 'outputs/long-document-flow.md',
|
|
108
|
-
|
|
109
|
-
|
|
108
|
+
old_string: `{{CHUNK_${String(chunks.length + 1).padStart(3, '0')}}}`,
|
|
109
|
+
new_string: '',
|
|
110
110
|
replace_all: false,
|
|
111
111
|
},
|
|
112
|
-
}
|
|
112
|
+
},
|
|
113
113
|
]);
|
|
114
114
|
const tools = new ToolRegistry();
|
|
115
115
|
tools.register(WriteFileTool);
|
|
@@ -142,9 +142,9 @@ describe('query long-document tool flow', () => {
|
|
|
142
142
|
.map((event) => event.toolName);
|
|
143
143
|
expect(run.status).toBe('completed');
|
|
144
144
|
expect(run.result).toBe('Long document created and verified.');
|
|
145
|
-
expect(provider.calls).toBe(
|
|
146
|
-
expect(executingTools).toEqual(['write', 'edit', 'edit', 'edit', 'edit']);
|
|
147
|
-
expect(final).not.toContain('{{
|
|
145
|
+
expect(provider.calls).toBe(7);
|
|
146
|
+
expect(executingTools).toEqual(['write', 'edit', 'edit', 'edit', 'edit', 'edit']);
|
|
147
|
+
expect(final).not.toContain('{{CHUNK_');
|
|
148
148
|
expect(final.split('\n').length).toBeGreaterThan(160);
|
|
149
149
|
expect(final).toContain('## Section 1');
|
|
150
150
|
expect(final).toContain('## Section 4');
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"long-document-flow.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/long-document-flow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAA;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExD,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAE1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAOD,MAAM,4BAA4B;IAKJ;IAJpB,EAAE,GAAG,wBAAwB,CAAA;IAC7B,IAAI,GAAG,iCAAiC,CAAA;IACjD,KAAK,GAAG,CAAC,CAAA;IAET,YAA6B,KAA0B;QAA1B,UAAK,GAAL,KAAK,CAAqB;IAAG,CAAC;IAE3D,KAAK,CAAC,CAAC,UAAU;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAEf,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,MAAM;gBACL,EAAE,EAAE,UAAU;gBACd,KAAK,EAAE,EAAE,OAAO,EAAE,qCAAqC,EAAE;gBACzD,YAAY,EAAE,MAAM;gBACpB,KAAK,EAAE,UAAU;aACjB,CAAA;YACD,OAAM;QACP,CAAC;QAED,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;qBAC7B;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;qBACnD;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,YAAY;YAC1B,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,MAAM,QAAQ,GAAa,EAAE,CAAA;IAE7B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE;YAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE;gBACzD,OAAO,WAAW,YAAY,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,qFAAqF,CAAA;YACzI,CAAC,CAAC,CAAA;YACF,MAAM,KAAK,GAAG,CAAC,cAAc,YAAY,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACzE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;YACzC,OAAO,KAAK,CAAA;QACb,CAAC,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,IAAI,4BAA4B,CAAC;YACjD;gBACC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"long-document-flow.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/long-document-flow.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAA;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExD,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAChE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAE1E,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAOD,MAAM,4BAA4B;IAKJ;IAJpB,EAAE,GAAG,wBAAwB,CAAA;IAC7B,IAAI,GAAG,iCAAiC,CAAA;IACjD,KAAK,GAAG,CAAC,CAAA;IAET,YAA6B,KAA0B;QAA1B,UAAK,GAAL,KAAK,CAAqB;IAAG,CAAC;IAE3D,KAAK,CAAC,CAAC,UAAU;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnC,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAEf,IAAI,CAAC,IAAI,EAAE,CAAC;YACX,MAAM;gBACL,EAAE,EAAE,UAAU;gBACd,KAAK,EAAE,EAAE,OAAO,EAAE,qCAAqC,EAAE;gBACzD,YAAY,EAAE,MAAM;gBACpB,KAAK,EAAE,UAAU;aACjB,CAAA;YACD,OAAM;QACP,CAAC;QAED,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,IAAI,EAAE,UAAU;wBAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE;qBAC7B;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE;gBACN,SAAS,EAAE;oBACV;wBACC,KAAK,EAAE,CAAC;wBACR,EAAE,EAAE,SAAS,IAAI,CAAC,KAAK,EAAE;wBACzB,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;qBACnD;iBACD;aACD;SACD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,YAAY;YAC1B,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC9C,MAAM,QAAQ,GAAa,EAAE,CAAA;IAE7B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC/F,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAE/B,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,EAAE;YAC5D,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE;gBACzD,OAAO,WAAW,YAAY,GAAG,CAAC,IAAI,SAAS,GAAG,CAAC,qFAAqF,CAAA;YACzI,CAAC,CAAC,CAAA;YACF,MAAM,KAAK,GAAG,CAAC,cAAc,YAAY,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACzE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAA;YACzC,OAAO,KAAK,CAAA;QACb,CAAC,CAAC,CAAA;QAEF,MAAM,QAAQ,GAAG,IAAI,4BAA4B,CAAC;YACjD;gBACC,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,OAAO,EAAE,yCAAyC;iBAClD;aACD;YACD,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAChC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,UAAU,EAAE,WAAW,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI;oBAC7D,UAAU,EAAE,GAAG,KAAK,aAAa,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI;oBACvE,WAAW,EAAE,KAAK;iBAClB;aACD,CAAC,CAAC;YACH;gBACC,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACN,IAAI,EAAE,+BAA+B;oBACrC,UAAU,EAAE,WAAW,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI;oBACrE,UAAU,EAAE,EAAE;oBACd,WAAW,EAAE,KAAK;iBAClB;aACD;SACD,CAAC,CAAA;QACF,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAA;QAC7B,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAA;QACxB,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,gBAAgB;YACzB,SAAS,EAAE,qBAAqB;YAChC,QAAQ,EAAE,CAAC,iBAAiB,CAAC,wBAAwB,CAAC,CAAC;YACvD,gBAAgB;YAChB,SAAS,EAAE,wBAAqC;YAChD,QAAQ,EAAE,wBAAoC;YAC9C,SAAS,EAAE,wBAAqC;YAChD,QAAQ,EAAE,wBAAoC;SAC9C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,+BAA+B,CAAC,EAAE,OAAO,CAAC,CAAA;QAC9F,MAAM,cAAc,GAAG,MAAM;aAC3B,MAAM,CACN,CAAC,KAAK,EAA0D,EAAE,CACjE,KAAK,CAAC,IAAI,KAAK,gBAAgB,CAChC;aACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;QAEhC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAA;QAC9D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAA;QACjF,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;QACrD,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -136,7 +136,7 @@ describe('query stream recovery', () => {
|
|
|
136
136
|
isError: true,
|
|
137
137
|
});
|
|
138
138
|
expect(completedTool?.type === 'tool_completed' ? completedTool.result : '').toContain('call was cut off');
|
|
139
|
-
expect(completedTool?.type === 'tool_completed' ? completedTool.result : '').toContain('
|
|
139
|
+
expect(completedTool?.type === 'tool_completed' ? completedTool.result : '').toContain('advance that marker with bounded exact edit calls');
|
|
140
140
|
});
|
|
141
141
|
it('preserves classified provider metadata through the primary run boundary', async () => {
|
|
142
142
|
const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-provider-error-'));
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"stream-recovery.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/stream-recovery.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC5D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAA;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAEhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,2BAA2B;IACvB,EAAE,GAAG,wBAAwB,CAAA;IAC7B,IAAI,GAAG,iCAAiC,CAAA;IACjD,KAAK,GAAG,CAAC,CAAA;IAET,KAAK,CAAC,CAAC,UAAU;QAChB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAEf,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM;gBACL,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE;oBACN,SAAS,EAAE;wBACV;4BACC,KAAK,EAAE,CAAC;4BACR,EAAE,EAAE,eAAe;4BACnB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE;yBAChC;qBACD;iBACD;aACD,CAAA;YACD,MAAM;gBACL,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE;oBACN,SAAS,EAAE;wBACV;4BACC,KAAK,EAAE,CAAC;4BACR,EAAE,EAAE,eAAe;4BACnB,QAAQ,EAAE;gCACT,SAAS,EAAE,0CAA0C;6BACrD;yBACD;qBACD;iBACD;aACD,CAAA;YACD,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAA;QACjD,CAAC;QAED,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,iCAAiC,EAAE;SACrD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,MAAM,yBAAyB;IACrB,EAAE,GAAG,oBAAoB,CAAA;IACzB,IAAI,GAAG,6BAA6B,CAAA;IAE7C,KAAK,CAAC,CAAC,UAAU;QAChB,MAAM,MAAM,OAAO,CAAC,MAAM,CACzB,IAAI,oBAAoB,CAAC;YACxB,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,oBAAoB;YAChC,MAAM,EAAE,GAAG;YACX,YAAY,EAAE,IAAI;SAClB,CAAC,CACF,CAAA;IACF,CAAC;CACD;AAED,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACtC,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,QAAQ,GAAG,IAAI,2BAA2B,EAAE,CAAA;QAClD,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gBAAgB;SACxB,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,KAAK,CAAC,QAAQ,CAAC;YACd,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,cAAc;YAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACrB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;gBAChB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;aACnB,CAAC;YACF,OAAO,EAAE,WAAW;SACpB,CAAC,CAAA;QACF,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC,CAAA;QAChF,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAC/B,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAC/C,gBAAgB;YAChB,SAAS,EAAE,qBAAkC;YAC7C,QAAQ,EAAE,qBAAiC;YAC3C,SAAS,EAAE,qBAAkC;YAC7C,QAAQ,EAAE,qBAAiC;SAC3C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAA;QAC1D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAE1C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACvE,MAAM,CACL,MAAM,CAAC,IAAI,CACV,CAAC,KAAK,EAAE,EAAE,CACT,KAAK,CAAC,IAAI,KAAK,sBAAsB;YACrC,KAAK,CAAC,cAAc,KAAK,IAAI;YAC7B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,IAAI,CACrC,CACD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;QAEhE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAChC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,SAAS,KAAK,eAAe,CACjF,CAAA;QACD,MAAM,CAAC,aAAa,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,IAAI;SACb,CAAC,CAAA;QACF,MAAM,CAAC,aAAa,EAAE,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,CACrF,kBAAkB,CAClB,CAAA;QACD,MAAM,CAAC,aAAa,EAAE,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,CACrF,
|
|
1
|
+
{"version":3,"file":"stream-recovery.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/stream-recovery.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AAC9C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC5D,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAEvB,OAAO,EAAE,oBAAoB,EAAE,MAAM,6BAA6B,CAAA;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAEhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,MAAM,UAAU,GAAG;IAClB,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;IACnB,WAAW,EAAE,CAAC;IACd,YAAY,EAAE,CAAC;IACf,gBAAgB,EAAE,CAAC;CACnB,CAAA;AAED,MAAM,2BAA2B;IACvB,EAAE,GAAG,wBAAwB,CAAA;IAC7B,IAAI,GAAG,iCAAiC,CAAA;IACjD,KAAK,GAAG,CAAC,CAAA;IAET,KAAK,CAAC,CAAC,UAAU;QAChB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QAEf,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,EAAE,CAAC;YACtB,MAAM;gBACL,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE;oBACN,SAAS,EAAE;wBACV;4BACC,KAAK,EAAE,CAAC;4BACR,EAAE,EAAE,eAAe;4BACnB,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE;yBAChC;qBACD;iBACD;aACD,CAAA;YACD,MAAM;gBACL,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE;oBACN,SAAS,EAAE;wBACV;4BACC,KAAK,EAAE,CAAC;4BACR,EAAE,EAAE,eAAe;4BACnB,QAAQ,EAAE;gCACT,SAAS,EAAE,0CAA0C;6BACrD;yBACD;qBACD;iBACD;aACD,CAAA;YACD,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAA;QACjD,CAAC;QAED,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE,OAAO,EAAE,iCAAiC,EAAE;SACrD,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO;YACX,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED,MAAM,yBAAyB;IACrB,EAAE,GAAG,oBAAoB,CAAA;IACzB,IAAI,GAAG,6BAA6B,CAAA;IAE7C,KAAK,CAAC,CAAC,UAAU;QAChB,MAAM,MAAM,OAAO,CAAC,MAAM,CACzB,IAAI,oBAAoB,CAAC;YACxB,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,oBAAoB;YAChC,MAAM,EAAE,GAAG;YACX,YAAY,EAAE,IAAI;SAClB,CAAC,CACF,CAAA;IACF,CAAC;CACD;AAED,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACtC,IAAI,QAAQ,GAAa,EAAE,CAAA;IAE3B,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;QACnF,QAAQ,GAAG,EAAE,CAAA;IACd,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,QAAQ,GAAG,IAAI,2BAA2B,EAAE,CAAA;QAClD,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;YACtC,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gBAAgB;SACxB,CAAC,CAAC,CAAA;QACH,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;QAChC,KAAK,CAAC,QAAQ,CAAC;YACd,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,cAAc;YAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;gBACrB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;gBAChB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;aACnB,CAAC;YACF,OAAO,EAAE,WAAW;SACpB,CAAC,CAAA;QACF,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC,CAAA;QAChF,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAC/B,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ;YACR,KAAK;YACL,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;YAC/C,gBAAgB;YAChB,SAAS,EAAE,qBAAkC;YAC7C,QAAQ,EAAE,qBAAiC;YAC3C,SAAS,EAAE,qBAAkC;YAC7C,QAAQ,EAAE,qBAAiC;SAC3C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACpC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAA;QAC1D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;QAE1C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACvE,MAAM,CACL,MAAM,CAAC,IAAI,CACV,CAAC,KAAK,EAAE,EAAE,CACT,KAAK,CAAC,IAAI,KAAK,sBAAsB;YACrC,KAAK,CAAC,cAAc,KAAK,IAAI;YAC7B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,IAAI,CACrC,CACD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACZ,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAA;QAEhE,MAAM,aAAa,GAAG,MAAM,CAAC,IAAI,CAChC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,SAAS,KAAK,eAAe,CACjF,CAAA;QACD,MAAM,CAAC,aAAa,CAAC,CAAC,aAAa,CAAC;YACnC,IAAI,EAAE,gBAAgB;YACtB,QAAQ,EAAE,YAAY;YACtB,OAAO,EAAE,IAAI;SACb,CAAC,CAAA;QACF,MAAM,CAAC,aAAa,EAAE,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,CACrF,kBAAkB,CAClB,CAAA;QACD,MAAM,CAAC,aAAa,EAAE,IAAI,KAAK,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,SAAS,CACrF,mDAAmD,CACnD,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAA;QAC/E,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;QAC/B,MAAM,MAAM,GAAe,EAAE,CAAA;QAE7B,MAAM,GAAG,GAAG,MAAM,UAAU,CAC3B;YACC,QAAQ,EAAE,IAAI,yBAAyB,EAAE;YACzC,KAAK,EAAE,IAAI,YAAY,EAAE;YACzB,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,OAAO;gBACpB,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,GAAG;aACtB;YACD,OAAO,EAAE,YAAY;YACrB,SAAS,EAAE,YAAY;YACvB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,+BAA+B,CAAC,CAAC;YAC9D,gBAAgB;YAChB,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;YAC1C,SAAS,EAAE,oBAAiC;YAC5C,QAAQ,EAAE,oBAAgC;SAC1C,EACD,CAAC,KAAK,EAAE,EAAE;YACT,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACnB,CAAC,CACD,CAAA;QAED,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACjC,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC;YACrC,IAAI,EAAE,UAAU;YAChB,UAAU,EAAE,oBAAoB;YAChC,MAAM,EAAE,GAAG;YACX,YAAY,EAAE,IAAI;SAClB,CAAC,CAAA;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,aAAa,CAAC;YACzE,IAAI,EAAE,YAAY;YAClB,aAAa,EAAE,GAAG,CAAC,iBAAiB;SACpC,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
|
|
@@ -47,7 +47,7 @@ export class ToolExecutor {
|
|
|
47
47
|
const baseContext = this.buildToolContext();
|
|
48
48
|
// Respect each tool's `concurrencySafe` flag. Read-only tools
|
|
49
49
|
// (ls/grep/glob/…) run in parallel; tools that mutate shared state
|
|
50
|
-
// (edit/write/
|
|
50
|
+
// (edit/write/bash — `concurrencySafe: false`) are serialized in
|
|
51
51
|
// a single chain, so e.g. several `edit` calls to the SAME file in one
|
|
52
52
|
// turn apply one-after-another instead of racing read→modify→write
|
|
53
53
|
// (which let the last writer clobber the rest). Results are written by
|
|
@@ -390,6 +390,6 @@ function formatFailedToolOutput(output, error) {
|
|
|
390
390
|
return `${output}\n\n${errorText}`;
|
|
391
391
|
}
|
|
392
392
|
function truncatedToolInputMessage(toolName) {
|
|
393
|
-
return `Error: Tool "${toolName}" call was cut off while the model was streaming JSON arguments. The tool was NOT executed. Retry with a much shorter input. Self-budget
|
|
393
|
+
return `Error: Tool "${toolName}" call was cut off while the model was streaming JSON arguments. The tool was NOT executed. Retry with a much shorter input. Self-budget content/new_string under 12000 characters before calling file tools. For long files, create a short opening with write and a deterministic marker, then advance that marker with bounded exact edit calls; for delegated work, pass a shared workspace filename/reference instead of embedding the content in the tool call.`;
|
|
394
394
|
}
|
|
395
395
|
//# sourceMappingURL=executor.js.map
|