@namzu/sdk 2.0.0 → 3.1.0

package/dist/runtime/query/__tests__/partial-batch-resume.test.js

@@ -0,0 +1,205 @@
+import { mkdtemp, rm, writeFile } from 'node:fs/promises';
+import { tmpdir } from 'node:os';
+import { join } from 'node:path';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { ActivityStore } from '../../../store/activity/memory.js';
+import { RunDiskStore } from '../../../store/run/disk.js';
+import { createAssistantMessage, createUserMessage, } from '../../../types/message/index.js';
+import { ToolExecutor } from '../executor.js';
+import { planCrashResume, unansweredToolCalls } from '../resume-pending.js';
+/**
+ * A batch's results reach the history only when the WHOLE batch settles,
+ * so a hard kill part-way through loses every result that had already come
+ * back — and the resumed run re-executes those calls. For a `write_file`
+ * that is waste; for a payment or an email it is a second one.
+ *
+ * Nothing new had to be recorded to fix it. The executor already awaits a
+ * `tool_completed` per tool, inline, carrying the id, the name, the result
+ * and the error flag, and the transcript already persists it. The record
+ * was durable all along and simply never read back.
+ */
+function makeLogger() {
+    const self = { info: vi.fn(), warn: vi.fn(), error: vi.fn(), debug: vi.fn() };
+    self.child = vi.fn(() => self);
+    return self;
+}
+const call = (id, name) => ({
+    id,
+    type: 'function',
+    function: { name, arguments: '{}' },
+});
+function checkpointWith(messages) {
+    return { id: 'cp_1', messages };
+}
+const parkedBatch = () => [
+    createUserMessage('charge the customer and email them'),
+    {
+        ...createAssistantMessage(''),
+        toolCalls: [call('t1', 'charge_card'), call('t2', 'send_email')],
+    },
+];
+describe('reading completed calls back out of the transcript', () => {
+    let dir;
+    let store;
+    beforeEach(async () => {
+        dir = await mkdtemp(join(tmpdir(), 'namzu-resume-'));
+        store = new RunDiskStore({ baseDir: dir });
+        await store.initRun('run_1');
+    });
+    afterEach(async () => {
+        await rm(dir, { recursive: true, force: true });
+    });
+    const append = async (lines) => {
+        await writeFile(join(dir, 'run_1', 'transcript.jsonl'), `${lines.map((l) => JSON.stringify(l)).join('\n')}\n`, 'utf-8');
+    };
+    it('recovers the calls that finished', async () => {
+        await append([
+            { type: 'tool_executing', toolUseId: 't1', toolName: 'charge_card' },
+            {
+                type: 'tool_completed',
+                toolUseId: 't1',
+                toolName: 'charge_card',
+                result: 'charged',
+                isError: false,
+            },
+        ]);
+        const completed = await store.readCompletedTools();
+        expect(completed.get('t1')).toEqual({
+            toolUseId: 't1',
+            toolName: 'charge_card',
+            result: 'charged',
+            isError: false,
+        });
+        // The one that never finished must NOT be invented — it still has to
+        // run, and claiming otherwise would drop the work silently.
+        expect(completed.has('t2')).toBe(false);
+    });
+    it('keeps the last result when a tool was retried', async () => {
+        await append([
+            {
+                type: 'tool_completed',
+                toolUseId: 't1',
+                toolName: 'fetch',
+                result: 'timeout',
+                isError: true,
+            },
+            { type: 'tool_completed', toolUseId: 't1', toolName: 'fetch', result: 'ok', isError: false },
+        ]);
+        expect((await store.readCompletedTools()).get('t1')?.result).toBe('ok');
+    });
+    it('survives the torn last line a killed process leaves behind', async () => {
+        // This is the ordinary shape of a file that was being appended to
+        // when the process died — which is the exact case being recovered.
+        await writeFile(join(dir, 'run_1', 'transcript.jsonl'), `${JSON.stringify({ type: 'tool_completed', toolUseId: 't1', toolName: 'a', result: 'r', isError: false })}\n{"type":"tool_com`, 'utf-8');
+        const completed = await store.readCompletedTools();
+        expect(completed.size).toBe(1);
+    });
+    it('returns nothing when the run never wrote a transcript', async () => {
+        expect((await store.readCompletedTools()).size).toBe(0);
+    });
+    it('ignores an event missing the fields that identify a call', async () => {
+        await append([{ type: 'tool_completed', result: 'orphan', isError: false }]);
+        expect((await store.readCompletedTools()).size).toBe(0);
+    });
+});
+describe('planning the resume of a part-executed batch', () => {
+    it('takes over when some calls already ran', () => {
+        const messages = parkedBatch();
+        const plan = planCrashResume(checkpointWith(messages), new Map([['t1', {}]]), makeLogger());
+        expect(plan).not.toBeNull();
+        // The assistant turn is kept, not stripped: the results about to be
+        // produced have to answer the `tool_use` blocks that are in it.
+        expect(plan?.response.message.toolCalls).toHaveLength(2);
+        expect(plan?.denials.size).toBe(0);
+    });
+    it('leaves an untouched batch to the ordinary repair', () => {
+        // A tool-review park records the checkpoint BEFORE any execution, so
+        // nothing completed and re-deciding costs only a round trip. Taking
+        // it over here would execute calls a human had not answered yet.
+        expect(planCrashResume(checkpointWith(parkedBatch()), new Map(), makeLogger())).toBeNull();
+    });
+    it('says which calls are about to run for the first time', () => {
+        const log = makeLogger();
+        planCrashResume(checkpointWith(parkedBatch()), new Map([['t1', {}]]), log);
+        const warned = log.warn.mock.calls[0];
+        expect(warned?.[1]).toMatchObject({ completed: 1, total: 2, remaining: ['send_email'] });
+    });
+    it('does nothing when the turn is fully answered', () => {
+        const messages = [
+            ...parkedBatch(),
+            { role: 'tool', content: 'charged', toolCallId: 't1' },
+            { role: 'tool', content: 'sent', toolCallId: 't2' },
+        ];
+        expect(planCrashResume(checkpointWith(messages), new Map([['t1', {}]]), makeLogger())).toBeNull();
+    });
+});
+describe('executing a batch that carries recovered results', () => {
+    const response = {
+        message: {
+            role: 'assistant',
+            content: null,
+            toolCalls: [call('t1', 'charge_card'), call('t2', 'send_email')],
+        },
+        finishReason: 'tool_calls',
+        usage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
+    };
+    async function runBatch(prior) {
+        const execute = vi.fn(async () => ({ success: true, output: 'freshly executed' }));
+        const tools = {
+            register: vi.fn(),
+            unregister: vi.fn(),
+            execute,
+            get: vi.fn(() => ({ isConcurrencySafe: () => true })),
+            has: vi.fn(() => true),
+            listNames: vi.fn(() => []),
+            getAvailability: vi.fn(),
+        };
+        const executor = new ToolExecutor({
+            tools,
+            runId: 'run_1',
+            workingDirectory: tmpdir(),
+            permissionMode: 'auto',
+            env: {},
+            abortSignal: new AbortController().signal,
+        }, new ActivityStore('run_1', {
+            enabled: false,
+            trackToolCalls: false,
+            trackLlmTurns: false,
+        }), async () => { }, makeLogger());
+        const batch = await executor.executeBatch(response, undefined, prior);
+        return { batch, execute };
+    }
+    it('does not run a tool that already ran', async () => {
+        const { batch, execute } = await runBatch(new Map([['t1', { result: 'charged $40', isError: false }]]));
+        // The whole point: charging the card twice is the failure.
+        expect(execute).toHaveBeenCalledTimes(1);
+        expect(batch.results.find((r) => r.toolCallId === 't1')?.output).toBe('charged $40');
+    });
+    it('still runs the calls that never completed', async () => {
+        const { batch } = await runBatch(new Map([['t1', { result: 'charged', isError: false }]]));
+        expect(batch.results.find((r) => r.toolCallId === 't2')?.output).toBe('freshly executed');
+        // Every `tool_use` block is answered, in the original order.
+        expect(batch.results.map((r) => r.toolCallId)).toEqual(['t1', 't2']);
+    });
+    it('preserves a recovered failure as a failure', async () => {
+        const { batch } = await runBatch(new Map([['t1', { result: 'declined', isError: true }]]));
+        expect(batch.results.find((r) => r.toolCallId === 't1')?.isError).toBe(true);
+    });
+    it('runs everything when nothing was recovered', async () => {
+        const { execute } = await runBatch();
+        expect(execute).toHaveBeenCalledTimes(2);
+    });
+});
+describe('which calls are worth asking about', () => {
+    it('is only the unanswered ones', () => {
+        const messages = [
+            ...parkedBatch(),
+            { role: 'tool', content: 'charged', toolCallId: 't1' },
+        ];
+        expect(unansweredToolCalls(messages).map((tc) => tc.id)).toEqual(['t2']);
+    });
+    it('is empty when the history holds no tool calls at all', () => {
+        expect(unansweredToolCalls([createUserMessage('hello')])).toEqual([]);
+    });
+});
+//# sourceMappingURL=partial-batch-resume.test.js.map

package/dist/runtime/query/__tests__/partial-batch-resume.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"partial-batch-resume.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/partial-batch-resume.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAChC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAA;AAChC,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAExE,OAAO,EAAE,aAAa,EAAE,MAAM,mCAAmC,CAAA;AACjE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAGzD,OAAO,EAEN,sBAAsB,EACtB,iBAAiB,GACjB,MAAM,iCAAiC,CAAA;AAGxC,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAC7C,OAAO,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAA;AAE3E;;;;;;;;;;GAUG;AAEH,SAAS,UAAU;IAClB,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,EAAuB,CACjG;IAAC,IAA4C,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAA;IACxE,OAAO,IAAI,CAAA;AACZ,CAAC;AAED,MAAM,IAAI,GAAG,CAAC,EAAU,EAAE,IAAY,EAAE,EAAE,CAAC,CAAC;IAC3C,EAAE;IACF,IAAI,EAAE,UAAmB;IACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;CACnC,CAAC,CAAA;AAEF,SAAS,cAAc,CAAC,QAAmB;IAC1C,OAAO,EAAE,EAAE,EAAE,MAAsB,EAAE,QAAQ,EAAyB,CAAA;AACvE,CAAC;AAED,MAAM,WAAW,GAAG,GAAc,EAAE,CAAC;IACpC,iBAAiB,CAAC,oCAAoC,CAAC;IACvD;QACC,GAAG,sBAAsB,CAAC,EAAE,CAAC;QAC7B,SAAS,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;KACrD;CACZ,CAAA;AAED,QAAQ,CAAC,oDAAoD,EAAE,GAAG,EAAE;IACnE,IAAI,GAAW,CAAA;IACf,IAAI,KAAmB,CAAA;IAEvB,UAAU,CAAC,KAAK,IAAI,EAAE;QACrB,GAAG,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC,CAAC,CAAA;QACpD,KAAK,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAA;QAC1C,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;IAEF,MAAM,MAAM,GAAG,KAAK,EAAE,KAAgB,EAAE,EAAE;QACzC,MAAM,SAAS,CACd,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,kBAAkB,CAAC,EACtC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EACrD,OAAO,CACP,CAAA;IACF,CAAC,CAAA;IAED,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,MAAM,CAAC;YACZ,EAAE,IAAI,EAAE,gBAAgB,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,aAAa,EAAE;YACpE;gBACC,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE,aAAa;gBACvB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,KAAK;aACd;SACD,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAA;QAClD,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YACnC,SAAS,EAAE,IAAI;YACf,QAAQ,EAAE,aAAa;YACvB,MAAM,EAAE,SAAS;YACjB,OAAO,EAAE,KAAK;SACd,CAAC,CAAA;QACF,qEAAqE;QACrE,4DAA4D;QAC5D,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,MAAM,CAAC;YACZ;gBACC,IAAI,EAAE,gBAAgB;gBACtB,SAAS,EAAE,IAAI;gBACf,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,IAAI;aACb;YACD,EAAE,IAAI,EAAE,gBAAgB,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE;SAC5F,CAAC,CAAA;QAEF,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC3E,kEAAkE;QAClE,mEAAmE;QACnE,MAAM,SAAS,CACd,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,kBAAkB,CAAC,EACtC,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,qBAAqB,EAC/H,OAAO,CACP,CAAA;QAED,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAA;QAClD,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC/B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,MAAM,CAAC,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;QAC5E,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,kBAAkB,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,8CAA8C,EAAE,GAAG,EAAE;IAC7D,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACjD,MAAM,QAAQ,GAAG,WAAW,EAAE,CAAA;QAC9B,MAAM,IAAI,GAAG,eAAe,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAA;QAE3F,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAA;QAC3B,oEAAoE;QACpE,gEAAgE;QAChE,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACxD,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IACnC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC3D,qEAAqE;QACrE,oEAAoE;QACpE,iEAAiE;QACjE,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,EAAE,IAAI,GAAG,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAA;IAC3F,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC/D,MAAM,GAAG,GAAG,UAAU,EAAE,CAAA;QACxB,eAAe,CAAC,cAAc,CAAC,WAAW,EAAE,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA;QAE1E,MAAM,MAAM,GAAI,GAAG,CAAC,IAAoD,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAA;QACtF,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACvD,MAAM,QAAQ,GAAc;YAC3B,GAAG,WAAW,EAAE;YAChB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,EAAa;YACjE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAa;SAC9D,CAAA;QACD,MAAM,CACL,eAAe,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAC9E,CAAC,QAAQ,EAAE,CAAA;IACb,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IACjE,MAAM,QAAQ,GAAG;QAChB,OAAO,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI;YACb,SAAS,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;SAChE;QACD,YAAY,EAAE,YAAY;QAC1B,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE;KACtD,CAAA;IAEV,KAAK,UAAU,QAAQ,CAAC,KAAiE;QACxF,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAA;QAClF,MAAM,KAAK,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,OAAO;YACP,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,iBAAiB,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;YACrD,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC;YACtB,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC;YAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACW,CAAA;QAEpC,MAAM,QAAQ,GAAG,IAAI,YAAY,CAChC;YACC,KAAK;YACL,KAAK,EAAE,OAAgB;YACvB,gBAAgB,EAAE,MAAM,EAAE;YAC1B,cAAc,EAAE,MAAM;YACtB,GAAG,EAAE,EAAE;YACP,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;SACzC,EACD,IAAI,aAAa,CAAC,OAAgB,EAAE;YACnC,OAAO,EAAE,KAAK;YACd,cAAc,EAAE,KAAK;YACrB,aAAa,EAAE,KAAK;SACpB,CAAC,EACF,KAAK,IAAI,EAAE,GAAE,CAAC,EACd,UAAU,EAAE,CACZ,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,CAAC,CAAA;QACrE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,CAAA;IAC1B,CAAC;IAED,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,QAAQ,CACxC,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAC5D,CAAA;QAED,2DAA2D;QAC3D,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACxC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACrF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;QAC1F,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;QACzF,6DAA6D;QAC7D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;QAC1F,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,QAAQ,EAAE,CAAA;QACpC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IACnD,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACtC,MAAM,QAAQ,GAAc;YAC3B,GAAG,WAAW,EAAE;YAChB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,EAAa;SACjE,CAAA;QACD,MAAM,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAA;IACzE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC/D,MAAM,CAAC,mBAAmB,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/plugin-hooks.test.js

@@ -13,7 +13,6 @@ describe('applyLifecycleHookResults', () => {
         ['skip', { action: 'skip', reason: 'no' }],
         ['modify', { action: 'modify', input: {} }],
         ['retry', { action: 'retry' }],
-        ['resume', { action: 'resume', input: 'x' }],
     ])('throws on unsupported %s action for lifecycle event', (_, result) => {
         expect(() => applyLifecycleHookResults('iteration_start', [result])).toThrow(/unsupported action/);
     });

package/dist/runtime/query/__tests__/plugin-hooks.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"plugin-hooks.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/plugin-hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7C,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAA;AAE9D,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACzD,MAAM,OAAO,GAAuB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAA;QACpF,MAAM,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAA;IAC5E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAuB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAA;QAC5E,MAAM,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CACvE,qCAAqC,CACrC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,IAAI,CAAC;QACP,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAsB,CAAC;QAC9D,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAsB,CAAC;QAC/D,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAsB,CAAC;QAClD,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,EAAsB,CAAC;KAChE,CAAC,CAAC,qDAAqD,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;QACvE,MAAM,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAC3E,oBAAoB,CACpB,CAAA;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}
+{"version":3,"file":"plugin-hooks.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/plugin-hooks.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7C,OAAO,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAA;AAE9D,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IAC1C,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACzD,MAAM,OAAO,GAAuB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAA;QACpF,MAAM,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAA;IAC5E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAuB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAA;QAC5E,MAAM,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CACvE,qCAAqC,CACrC,CAAA;IACF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,IAAI,CAAC;QACP,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAsB,CAAC;QAC9D,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,EAAsB,CAAC;QAC/D,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,OAAO,EAAsB,CAAC;KAClD,CAAC,CAAC,qDAAqD,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;QACvE,MAAM,CAAC,GAAG,EAAE,CAAC,yBAAyB,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAC3E,oBAAoB,CACpB,CAAA;IACF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/prepare-step-chain.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=prepare-step-chain.test.d.ts.map

package/dist/runtime/query/__tests__/prepare-step-chain.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepare-step-chain.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/prepare-step-chain.test.ts"],"names":[],"mappings":""}

package/dist/runtime/query/__tests__/prepare-step-chain.test.js

@@ -0,0 +1,104 @@
+import { mkdtemp, rm } from 'node:fs/promises';
+import { tmpdir } from 'node:os';
+import { join } from 'node:path';
+import { afterEach, describe, expect, it } from 'vitest';
+import { MockLLMProvider } from '../../../provider/mock.js';
+import { ToolRegistry } from '../../../registry/tool/execute.js';
+import { createUserMessage } from '../../../types/message/index.js';
+import { drainQuery } from '../index.js';
+/**
+ * A single slot is enough for one concern and no help with two. A host
+ * with a per-tenant system prefix AND a cost-based model downgrade had to
+ * hand-compose them into one callback — which puts the ordering in the
+ * host's own code where nothing can see it, and makes each concern's
+ * failure the other's problem.
+ */
+const dirs = [];
+afterEach(async () => {
+    await Promise.all(dirs.map((d) => rm(d, { recursive: true, force: true })));
+    dirs.length = 0;
+});
+async function run(prepareStep) {
+    const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-chain-'));
+    dirs.push(workingDirectory);
+    const provider = new MockLLMProvider({ turns: [{ text: 'done' }] });
+    await drainQuery({
+        provider,
+        tools: new ToolRegistry(),
+        runConfig: {
+            model: 'base-model',
+            timeoutMs: 10_000,
+            tokenBudget: 100_000,
+            maxIterations: 3,
+            temperature: 0.5,
+        },
+        agentId: 'agent_c',
+        agentName: 'Chained',
+        messages: [createUserMessage('do the work')],
+        workingDirectory,
+        sessionId: 'ses_c',
+        threadId: 'thd_c',
+        projectId: 'prj_c',
+        tenantId: 'tnt_c',
+        prepareStep,
+    });
+    return provider.requests[0];
+}
+describe('several shaping stages', () => {
+    it('applies every stage, so two concerns do not have to be one callback', async () => {
+        const sent = await run([() => ({ model: 'cheap-model' }), () => ({ temperature: 0.1 })]);
+        expect(sent?.model).toBe('cheap-model');
+        expect(sent?.temperature).toBe(0.1);
+    });
+    it('lets a later stage override an earlier one, in declaration order', async () => {
+        // Last writer wins — visibly, because the order is a line in the
+        // host's code rather than an accident of install history.
+        const sent = await run([() => ({ model: 'first' }), () => ({ model: 'second' })]);
+        expect(sent?.model).toBe('second');
+    });
+    it('shows a stage what the ones before it decided', async () => {
+        // Reading the accumulated decision is how a later stage REFINES an
+        // earlier one instead of guessing at it.
+        const seen = [];
+        const sent = await run([
+            () => ({ model: 'downgraded' }),
+            (context) => {
+                seen.push(context.prepared);
+                return context.prepared.model === 'downgraded' ? { temperature: 0 } : {};
+            },
+        ]);
+        expect(seen[0]).toMatchObject({ model: 'downgraded' });
+        expect(sent?.temperature).toBe(0);
+    });
+    it('starts the first stage with nothing decided', async () => {
+        const seen = [];
+        await run([
+            (context) => {
+                seen.push(context.prepared);
+                return {};
+            },
+        ]);
+        expect(seen[0]).toEqual({});
+    });
+    it('keeps running the rest when one stage throws', async () => {
+        // One broken concern must not silently disable the others it was
+        // declared beside.
+        const sent = await run([
+            () => {
+                throw new Error('this stage is broken');
+            },
+            () => ({ model: 'still-applied' }),
+        ]);
+        expect(sent?.model).toBe('still-applied');
+    });
+    it('still accepts a single function, unchanged', async () => {
+        const sent = await run(() => ({ model: 'solo' }));
+        expect(sent?.model).toBe('solo');
+    });
+    it('treats an empty list as no shaping at all', async () => {
+        const sent = await run([]);
+        expect(sent?.model).toBe('base-model');
+        expect(sent?.temperature).toBe(0.5);
+    });
+});
+//# sourceMappingURL=prepare-step-chain.test.js.map

package/dist/runtime/query/__tests__/prepare-step-chain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepare-step-chain.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/prepare-step-chain.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;AAExD,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAA;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAEhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAGnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC;;;;;;GAMG;AAEH,MAAM,IAAI,GAAa,EAAE,CAAA;AAEzB,SAAS,CAAC,KAAK,IAAI,EAAE;IACpB,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;IAC3E,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA;AAChB,CAAC,CAAC,CAAA;AAEF,KAAK,UAAU,GAAG,CAAC,WAA6B;IAC/C,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAA;IACtE,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;IAE3B,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAA;IACnE,MAAM,UAAU,CAAC;QAChB,QAAQ;QACR,KAAK,EAAE,IAAI,YAAY,EAAE;QACzB,SAAS,EAAE;YACV,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,MAAM;YACjB,WAAW,EAAE,OAAO;YACpB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,GAAG;SAChB;QACD,OAAO,EAAE,SAAS;QAClB,SAAS,EAAE,SAAS;QACpB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5C,gBAAgB;QAChB,SAAS,EAAE,OAAoB;QAC/B,QAAQ,EAAE,OAAmB;QAC7B,SAAS,EAAE,OAAoB;QAC/B,QAAQ,EAAE,OAAmB;QAC7B,WAAW;KACX,CAAC,CAAA;IAEF,OAAO,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;AAC5B,CAAC;AAED,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAA;QAExF,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QACvC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACpC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QACjF,iEAAiE;QACjE,0DAA0D;QAC1D,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAA;QAEjF,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACnC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC9D,mEAAmE;QACnE,yCAAyC;QACzC,MAAM,IAAI,GAAc,EAAE,CAAA;QAC1B,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC;YACtB,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;YAC/B,CAAC,OAAO,EAAE,EAAE;gBACX,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;gBAC3B,OAAO,OAAO,CAAC,QAAQ,CAAC,KAAK,KAAK,YAAY,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAA;YACzE,CAAC;SACD,CAAC,CAAA;QAEF,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAA;QACtD,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAClC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,IAAI,GAAc,EAAE,CAAA;QAC1B,MAAM,GAAG,CAAC;YACT,CAAC,OAAO,EAAE,EAAE;gBACX,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAA;gBAC3B,OAAO,EAAE,CAAA;YACV,CAAC;SACD,CAAC,CAAA;QAEF,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC7D,iEAAiE;QACjE,mBAAmB;QACnB,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC;YACtB,GAAG,EAAE;gBACJ,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;YACxC,CAAC;YACD,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC;SAClC,CAAC,CAAA;QAEF,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;QAEjD,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACjC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,CAAA;QAE1B,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACtC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACpC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/prepare-step.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=prepare-step.test.d.ts.map

package/dist/runtime/query/__tests__/prepare-step.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepare-step.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/prepare-step.test.ts"],"names":[],"mappings":""}

package/dist/runtime/query/__tests__/prepare-step.test.js

@@ -0,0 +1,194 @@
+import { mkdtemp, rm } from 'node:fs/promises';
+import { tmpdir } from 'node:os';
+import { join } from 'node:path';
+import { afterEach, describe, expect, it, vi } from 'vitest';
+import { z } from 'zod';
+import { MockLLMProvider } from '../../../provider/mock.js';
+import { ToolRegistry } from '../../../registry/tool/execute.js';
+import { createUserMessage } from '../../../types/message/index.js';
+import { drainQuery } from '../index.js';
+/**
+ * `stopWhen` let a run decide TO STOP from what its steps produced. This
+ * is the other half — deciding how the next step should be SHAPED.
+ *
+ * Without it the tool surface and the model are fixed at `query()` time,
+ * so a phased agent (research with search tools, then write with file
+ * tools, then verify with a cheaper model) had to be three separate runs,
+ * each starting blind to the last one's context.
+ */
+const dirs = [];
+afterEach(async () => {
+    await Promise.all(dirs.map((d) => rm(d, { recursive: true, force: true })));
+    dirs.length = 0;
+});
+function tool(name, calls) {
+    return {
+        name,
+        description: `${name} tool`,
+        inputSchema: z.object({}),
+        execute: () => {
+            calls.push(name);
+            return Promise.resolve({ success: true, output: `${name} ok` });
+        },
+    };
+}
+async function run(opts) {
+    const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-prepare-'));
+    dirs.push(workingDirectory);
+    const calls = [];
+    const tools = new ToolRegistry();
+    for (const name of opts.toolNames ?? ['search', 'write_file']) {
+        tools.register(tool(name, calls));
+    }
+    const provider = new MockLLMProvider({ turns: opts.turns });
+    const result = await drainQuery({
+        provider,
+        tools,
+        runConfig: {
+            model: 'base-model',
+            timeoutMs: 10_000,
+            tokenBudget: 100_000,
+            maxIterations: 6,
+            maxResponseTokens: 256,
+            temperature: 0.5,
+        },
+        agentId: 'agent_p',
+        agentName: 'Phased',
+        messages: [createUserMessage('do the work')],
+        workingDirectory,
+        sessionId: 'ses_p',
+        threadId: 'thd_p',
+        projectId: 'prj_p',
+        tenantId: 'tnt_p',
+        ...(opts.prepareStep ? { prepareStep: opts.prepareStep } : {}),
+    });
+    return { result, provider, calls };
+}
+describe('prepareStep shapes each step', () => {
+    it('narrows the tool surface for a phase', async () => {
+        const { provider } = await run({
+            turns: [{ text: 'done' }],
+            prepareStep: () => ({ activeTools: ['search'] }),
+        });
+        const sent = provider.requests[0]?.tools ?? [];
+        expect(sent.map((t) => t.function.name)).toEqual(['search']);
+    });
+    it('can change the surface between steps', async () => {
+        // The whole point: research first, then write.
+        const { provider } = await run({
+            turns: [{ toolCalls: [{ name: 'search', args: {} }] }, { text: 'written' }],
+            prepareStep: ({ stepNumber }) => ({
+                activeTools: stepNumber === 1 ? ['search'] : ['write_file'],
+            }),
+        });
+        expect(provider.requests[0]?.tools?.map((t) => t.function.name)).toEqual(['search']);
+        expect(provider.requests[1]?.tools?.map((t) => t.function.name)).toEqual(['write_file']);
+    });
+    it('does NOT touch tool_choice', async () => {
+        // No wire format has an `allowed_tools` parameter, and moving `tool_choice`
+        // invalidates cached MESSAGE blocks too — a strictly worse trade for
+        // the same effect.
+        const { provider } = await run({
+            turns: [{ text: 'done' }],
+            prepareStep: () => ({ activeTools: ['search'] }),
+        });
+        expect(provider.requests[0]?.toolChoice).toBeUndefined();
+    });
+    it('overrides the model for one step', async () => {
+        const { provider } = await run({
+            turns: [{ toolCalls: [{ name: 'search', args: {} }] }, { text: 'done' }],
+            prepareStep: ({ stepNumber }) => (stepNumber === 2 ? { model: 'cheap-model' } : undefined),
+        });
+        expect(provider.requests[0]?.model).toBe('base-model');
+        expect(provider.requests[1]?.model).toBe('cheap-model');
+    });
+    it('overrides sampling for one step', async () => {
+        const { provider } = await run({
+            turns: [{ text: 'done' }],
+            prepareStep: () => ({ temperature: 0, maxResponseTokens: 64 }),
+        });
+        expect(provider.requests[0]?.temperature).toBe(0);
+        expect(provider.requests[0]?.maxTokens).toBe(64);
+    });
+    it('adds step guidance to the REQUEST without retaining it in history', async () => {
+        // Pushing it onto the run would accumulate one stale instruction per
+        // iteration.
+        const { provider, result } = await run({
+            turns: [{ toolCalls: [{ name: 'search', args: {} }] }, { text: 'done' }],
+            prepareStep: ({ stepNumber }) => stepNumber === 1 ? { system: 'PHASE: research only' } : undefined,
+        });
+        const first = provider.requests[0]?.messages ?? [];
+        expect(JSON.stringify(first)).toContain('PHASE: research only');
+        // Not in the second request, and not in the run's history.
+        expect(JSON.stringify(provider.requests[1]?.messages ?? [])).not.toContain('PHASE:');
+        expect(JSON.stringify(result.messages)).not.toContain('PHASE:');
+    });
+    it('sees the steps produced so far', async () => {
+        const seen = [];
+        await run({
+            turns: [{ toolCalls: [{ name: 'search', args: {} }] }, { text: 'done' }],
+            prepareStep: ({ stepNumber, steps, messages }) => {
+                seen.push(steps.length);
+                expect(stepNumber).toBe(seen.length);
+                expect(messages.length).toBeGreaterThan(0);
+                return undefined;
+            },
+        });
+        // Step 1 has nothing behind it; step 2 can read step 1's result.
+        expect(seen).toEqual([0, 1]);
+    });
+});
+describe('prepareStep is safe to get wrong', () => {
+    it('fails OPEN — a throwing hook does not kill a healthy run', async () => {
+        // Same reasoning as `stopWhen`, and deliberately opposite to a
+        // guardrail: nothing unsafe gets through when step shaping is
+        // skipped.
+        const { result, provider } = await run({
+            turns: [{ text: 'still fine' }],
+            prepareStep: () => {
+                throw new Error('phase table missing');
+            },
+        });
+        expect(result.result).toBe('still fine');
+        expect(result.stopReason).toBe('end_turn');
+        // Fell back to the run's full surface.
+        expect(provider.requests[0]?.tools).toHaveLength(2);
+    });
+    it('ignores tools that are not registered rather than failing the run', async () => {
+        // A phase list that outlives a tool rename should narrow the
+        // surface, not kill the agent mid-run.
+        const { provider, result } = await run({
+            turns: [{ text: 'done' }],
+            prepareStep: () => ({ activeTools: ['search', 'renamed_away'] }),
+        });
+        expect(provider.requests[0]?.tools?.map((t) => t.function.name)).toEqual(['search']);
+        expect(result.stopReason).toBe('end_turn');
+    });
+    it('returning nothing is the same as having no hook', async () => {
+        const hook = vi.fn(() => undefined);
+        const { provider } = await run({ turns: [{ text: 'done' }], prepareStep: hook });
+        expect(hook).toHaveBeenCalled();
+        expect(provider.requests[0]?.model).toBe('base-model');
+        expect(provider.requests[0]?.temperature).toBe(0.5);
+        expect(provider.requests[0]?.tools).toHaveLength(2);
+    });
+    it('an empty activeTools list really does remove every tool', async () => {
+        // Distinguished from "no opinion" — `[]` is a decision.
+        const { provider } = await run({
+            turns: [{ text: 'done' }],
+            prepareStep: () => ({ activeTools: [] }),
+        });
+        expect(provider.requests[0]?.tools).toBeUndefined();
+    });
+    it('an async hook is awaited', async () => {
+        const { provider } = await run({
+            turns: [{ text: 'done' }],
+            prepareStep: async () => {
+                await Promise.resolve();
+                return { model: 'async-model' };
+            },
+        });
+        expect(provider.requests[0]?.model).toBe('async-model');
+    });
+});
+//# sourceMappingURL=prepare-step.test.js.map

package/dist/runtime/query/__tests__/prepare-step.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prepare-step.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/prepare-step.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,eAAe,EAAE,MAAM,2BAA2B,CAAA;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAEhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAInE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC;;;;;;;;GAQG;AAEH,MAAM,IAAI,GAAa,EAAE,CAAA;AAEzB,SAAS,CAAC,KAAK,IAAI,EAAE;IACpB,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAA;IAC3E,IAAI,CAAC,MAAM,GAAG,CAAC,CAAA;AAChB,CAAC,CAAC,CAAA;AAEF,SAAS,IAAI,CAAC,IAAY,EAAE,KAAe;IAC1C,OAAO;QACN,IAAI;QACJ,WAAW,EAAE,GAAG,IAAI,OAAO;QAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;QACzB,OAAO,EAAE,GAAG,EAAE;YACb,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAChB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,KAAK,EAAE,CAAC,CAAA;QAChE,CAAC;KAC4B,CAAA;AAC/B,CAAC;AAED,KAAK,UAAU,GAAG,CAAC,IAIlB;IACA,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,gBAAgB,CAAC,CAAC,CAAA;IACxE,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;IAE3B,MAAM,KAAK,GAAa,EAAE,CAAA;IAC1B,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;IAChC,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE,CAAC;QAC/D,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAA;IAClC,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;IAC3D,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC;QAC/B,QAAQ;QACR,KAAK;QACL,SAAS,EAAE;YACV,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,MAAM;YACjB,WAAW,EAAE,OAAO;YACpB,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,GAAG;YACtB,WAAW,EAAE,GAAG;SAChB;QACD,OAAO,EAAE,SAAS;QAClB,SAAS,EAAE,QAAQ;QACnB,QAAQ,EAAE,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5C,gBAAgB;QAChB,SAAS,EAAE,OAAoB;QAC/B,QAAQ,EAAE,OAAmB;QAC7B,SAAS,EAAE,OAAoB;QAC/B,QAAQ,EAAE,OAAmB;QAC7B,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC9D,CAAC,CAAA;IAEF,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAA;AACnC,CAAC;AAED,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC7C,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACzB,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;SAChD,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE,CAAA;QAC9C,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;IAC7D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACrD,+CAA+C;QAC/C,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YAC3E,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;gBACjC,WAAW,EAAE,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;aAC3D,CAAC;SACF,CAAC,CAAA;QAEF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;QACpF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAA;IACzF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC3C,4EAA4E;QAC5E,qEAAqE;QACrE,mBAAmB;QACnB,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACzB,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;SAChD,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,aAAa,EAAE,CAAA;IACzD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACxE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;SAC1F,CAAC,CAAA;QAEF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACtD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACzB,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE,CAAC;SAC9D,CAAC,CAAA;QAEF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QACjD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QAClF,qEAAqE;QACrE,aAAa;QACb,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC;YACtC,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACxE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAC/B,UAAU,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC,SAAS;SAClE,CAAC,CAAA;QAEF,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAA;QAClD,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAA;QAE/D,2DAA2D;QAC3D,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;QACpF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;IAChE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,IAAI,GAAa,EAAE,CAAA;QACzB,MAAM,GAAG,CAAC;YACT,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACxE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE;gBAChD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;gBACvB,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;gBACpC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;gBAC1C,OAAO,SAAS,CAAA;YACjB,CAAC;SACD,CAAC,CAAA;QAEF,iEAAiE;QACjE,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;IAC7B,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IACjD,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACzE,+DAA+D;QAC/D,8DAA8D;QAC9D,WAAW;QACX,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YACtC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;YAC/B,WAAW,EAAE,GAAG,EAAE;gBACjB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAA;YACvC,CAAC;SACD,CAAC,CAAA;QAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACxC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC1C,uCAAuC;QACvC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QAClF,6DAA6D;QAC7D,uCAAuC;QACvC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC;YACtC,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACzB,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,QAAQ,EAAE,cAAc,CAAC,EAAE,CAAC;SAChE,CAAC,CAAA;QAEF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAA;QACpF,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,IAAI,GAAG,EAAE,CAAC,EAAE,CAAc,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;QAChD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAA;QAEhF,MAAM,CAAC,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAA;QAC/B,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACtD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QACnD,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACxE,wDAAwD;QACxD,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACzB,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;SACxC,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,aAAa,EAAE,CAAA;IACpD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,GAAG,CAAC;YAC9B,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YACzB,WAAW,EAAE,KAAK,IAAI,EAAE;gBACvB,MAAM,OAAO,CAAC,OAAO,EAAE,CAAA;gBACvB,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,CAAA;YAChC,CAAC;SACD,CAAC,CAAA;QACF,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/probe-veto-is-error.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=probe-veto-is-error.test.d.ts.map

package/dist/runtime/query/__tests__/probe-veto-is-error.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"probe-veto-is-error.test.d.ts","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/probe-veto-is-error.test.ts"],"names":[],"mappings":""}