@namzu/sdk 2.0.0 → 3.1.0

package/dist/runtime/query/__tests__/hook-model-visibility.test.d.ts

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

package/dist/runtime/query/__tests__/hook-model-visibility.test.d.ts.map

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

package/dist/runtime/query/__tests__/hook-model-visibility.test.js

@@ -0,0 +1,119 @@
+import { describe, expect, it } from 'vitest';
+import { z } from 'zod';
+import { MockLLMProvider, registerMock } from '../../../provider/index.js';
+import { ToolRegistry } from '../../../registry/index.js';
+import { generateProjectId, generateSessionId, generateTenantId, generateThreadId, } from '../../../utils/id.js';
+import { drainQuery } from '../index.js';
+/**
+ * Both model-call hooks fired directly beside the request and the reply and
+ * were handed neither — only a run id and an iteration number. An extension
+ * could observe THAT a call happened and nothing about what it was, so a
+ * prompt audit, a redaction pass, or a per-tenant token ledger had no way to
+ * do its job from a hook.
+ */
+registerMock();
+function recordingManager(seen) {
+    return {
+        executeHooks: async (event, ctx) => {
+            seen.push({ event, ctx });
+            return [];
+        },
+    };
+}
+async function runWithHooks(seen, toolNames = []) {
+    const tools = new ToolRegistry();
+    for (const name of toolNames) {
+        tools.register({
+            name,
+            description: `${name} tool`,
+            inputSchema: z.object({}),
+            execute: () => Promise.resolve({ success: true, output: 'ok' }),
+        });
+    }
+    return drainQuery({
+        provider: new MockLLMProvider({
+            turns: [
+                { text: 'the answer', usage: { promptTokens: 11, completionTokens: 4, totalTokens: 15 } },
+            ],
+        }),
+        tools,
+        agentId: 'a',
+        agentName: 'A',
+        messages: [{ role: 'user', content: 'what is the answer' }],
+        workingDirectory: process.cwd(),
+        runConfig: {
+            model: 'mock-model',
+            tokenBudget: 100_000,
+            timeoutMs: 30_000,
+            maxIterations: 2,
+            temperature: 0.25,
+            maxResponseTokens: 512,
+        },
+        projectId: generateProjectId(),
+        sessionId: generateSessionId(),
+        threadId: generateThreadId(),
+        tenantId: generateTenantId(),
+        pluginManager: recordingManager(seen),
+    });
+}
+const pick = (seen, event) => seen.find((s) => s.event === event)?.ctx;
+describe('what an extension is shown about a model call', () => {
+    it('shows the request the run is about to send', async () => {
+        const seen = [];
+        await runWithHooks(seen);
+        const request = pick(seen, 'pre_llm_call')?.request;
+        expect(request?.model).toBe('mock-model');
+        expect(request?.messages.map((m) => m.content)).toContain('what is the answer');
+        expect(request?.temperature).toBe(0.25);
+        expect(request?.maxTokens).toBe(512);
+    });
+    it('names the tools the model is being offered', async () => {
+        // The set, not the schemas: an audit asks which capabilities were
+        // exposed on this turn, and the schemas are the driver's business.
+        const seen = [];
+        await runWithHooks(seen, ['read_file', 'write_file']);
+        expect([...(pick(seen, 'pre_llm_call')?.request?.toolNames ?? [])].sort()).toEqual([
+            'read_file',
+            'write_file',
+        ]);
+    });
+    it('shows the reply and what it cost', async () => {
+        const seen = [];
+        await runWithHooks(seen);
+        const response = pick(seen, 'post_llm_call')?.response;
+        expect(response?.content).toBe('the answer');
+        expect(response?.finishReason).toBe('stop');
+        // A per-tenant token ledger is the whole point of carrying usage.
+        expect(response?.usage).toMatchObject({ promptTokens: 11, completionTokens: 4 });
+    });
+    it('refuses a hook that writes into the request', async () => {
+        // One hook shaping the request for the next would make the outcome
+        // depend on registration order, and the last one registered would
+        // silently win. Shaping stays with the single-slot host callback.
+        const seen = [];
+        await runWithHooks(seen);
+        const request = pick(seen, 'pre_llm_call')?.request;
+        // Asserted present first: a missing projection would also make the
+        // write below throw, and the test would pass for the wrong reason.
+        expect(request).toBeDefined();
+        expect(Object.isFrozen(request)).toBe(true);
+        expect(() => {
+            ;
+            request.model = 'something-else';
+        }).toThrow();
+        expect(request?.model).toBe('mock-model');
+    });
+    it('hands over copies, so a write cannot reach the run history', async () => {
+        const seen = [];
+        const run = await runWithHooks(seen);
+        const shown = pick(seen, 'pre_llm_call')?.request?.messages ?? [];
+        const ask = shown.find((m) => m.content === 'what is the answer');
+        expect(ask).toBeDefined();
+        expect(() => {
+            ;
+            ask.content = 'tampered';
+        }).toThrow();
+        expect(run.messages.some((m) => m.content === 'what is the answer')).toBe(true);
+    });
+});
+//# sourceMappingURL=hook-model-visibility.test.js.map

package/dist/runtime/query/__tests__/hook-model-visibility.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hook-model-visibility.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/hook-model-visibility.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAE7C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAEzD,OAAO,EACN,iBAAiB,EACjB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,GAChB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC;;;;;;GAMG;AAEH,YAAY,EAAE,CAAA;AAId,SAAS,gBAAgB,CAAC,IAAY;IACrC,OAAO;QACN,YAAY,EAAE,KAAK,EAAE,KAAsB,EAAE,GAAgB,EAAE,EAAE;YAChE,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;YACzB,OAAO,EAAE,CAAA;QACV,CAAC;KACoC,CAAA;AACvC,CAAC;AAED,KAAK,UAAU,YAAY,CAAC,IAAY,EAAE,YAA+B,EAAE;IAC1E,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAA;IAChC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC9B,KAAK,CAAC,QAAQ,CAAC;YACd,IAAI;YACJ,WAAW,EAAE,GAAG,IAAI,OAAO;YAC3B,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;SACtD,CAAC,CAAA;IACZ,CAAC;IAED,OAAO,UAAU,CAAC;QACjB,QAAQ,EAAE,IAAI,eAAe,CAAC;YAC7B,KAAK,EAAE;gBACN,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,EAAE;aACzF;SACD,CAAC;QACF,KAAK;QACL,OAAO,EAAE,GAAG;QACZ,SAAS,EAAE,GAAG;QACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC;QAC3D,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE;QAC/B,SAAS,EAAE;YACV,KAAK,EAAE,YAAY;YACnB,WAAW,EAAE,OAAO;YACpB,SAAS,EAAE,MAAM;YACjB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,IAAI;YACjB,iBAAiB,EAAE,GAAG;SACtB;QACD,SAAS,EAAE,iBAAiB,EAAE;QAC9B,SAAS,EAAE,iBAAiB,EAAE;QAC9B,QAAQ,EAAE,gBAAgB,EAAE;QAC5B,QAAQ,EAAE,gBAAgB,EAAE;QAC5B,aAAa,EAAE,gBAAgB,CAAC,IAAI,CAAC;KAC5B,CAAC,CAAA;AACZ,CAAC;AAED,MAAM,IAAI,GAAG,CAAC,IAAY,EAAE,KAAsB,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,EAAE,GAAG,CAAA;AAE/F,QAAQ,CAAC,+CAA+C,EAAE,GAAG,EAAE;IAC9D,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,IAAI,GAAW,EAAE,CAAA;QACvB,MAAM,YAAY,CAAC,IAAI,CAAC,CAAA;QAExB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,EAAE,OAAO,CAAA;QACnD,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QACzC,MAAM,CAAC,OAAO,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAA;QAC/E,MAAM,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACvC,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC3D,kEAAkE;QAClE,mEAAmE;QACnE,MAAM,IAAI,GAAW,EAAE,CAAA;QACvB,MAAM,YAAY,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC,CAAA;QAErD,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,EAAE,OAAO,EAAE,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC;YAClF,WAAW;YACX,YAAY;SACZ,CAAC,CAAA;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,IAAI,GAAW,EAAE,CAAA;QACvB,MAAM,YAAY,CAAC,IAAI,CAAC,CAAA;QAExB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,EAAE,eAAe,CAAC,EAAE,QAAQ,CAAA;QACtD,MAAM,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QAC5C,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QAC3C,kEAAkE;QAClE,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,aAAa,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC,CAAA;IACjF,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC5D,mEAAmE;QACnE,kEAAkE;QAClE,kEAAkE;QAClE,MAAM,IAAI,GAAW,EAAE,CAAA;QACvB,MAAM,YAAY,CAAC,IAAI,CAAC,CAAA;QAExB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,EAAE,OAAO,CAAA;QACnD,mEAAmE;QACnE,mEAAmE;QACnE,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAA;QAC7B,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3C,MAAM,CAAC,GAAG,EAAE;YACX,CAAC;YAAC,OAAwC,CAAC,KAAK,GAAG,gBAAgB,CAAA;QACpE,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;QACZ,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,IAAI,GAAW,EAAE,CAAA;QACvB,MAAM,GAAG,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAA;QAEpC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,EAAE,cAAc,CAAC,EAAE,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAA;QACjE,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,oBAAoB,CAAC,CAAA;QACjE,MAAM,CAAC,GAAG,CAAC,CAAC,WAAW,EAAE,CAAA;QACzB,MAAM,CAAC,GAAG,EAAE;YACX,CAAC;YAAC,GAAsC,CAAC,OAAO,GAAG,UAAU,CAAA;QAC9D,CAAC,CAAC,CAAC,OAAO,EAAE,CAAA;QACZ,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAChF,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/long-document-flow.test.js

@@ -120,7 +120,14 @@ describe('query long-document tool flow', () => {
             tools,
             runConfig: {
                 model: 'mock-model',
-                timeoutMs: 5_000,
+                // The RUN's own deadline, not the test's. This test asserts a
+                // flow, never a latency, so the deadline must sit far above
+                // the work: six model turns and five file edits take ~3s
+                // alone and longer under a parallel suite. The outer vitest
+                // timeout was raised for exactly that reason and this one was
+                // missed, so the run kept aborting itself at 5s and the
+                // failure read as a broken flow rather than a slow machine.
+                timeoutMs: 120_000,
                 tokenBudget: 100_000,
                 maxIterations: 8,
                 maxResponseTokens: 256,
@@ -149,6 +156,6 @@ describe('query long-document tool flow', () => {
         expect(final).toContain('## Section 1');
         expect(final).toContain('## Section 4');
         expect(final).toContain('Section 4.40');
-    });
+    }, 60_000);
 });
 //# sourceMappingURL=long-document-flow.test.js.map

package/dist/runtime/query/__tests__/long-document-flow.test.js.map

@@ -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,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"}
+{"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,8DAA8D;gBAC9D,4DAA4D;gBAC5D,yDAAyD;gBACzD,4DAA4D;gBAC5D,8DAA8D;gBAC9D,wDAAwD;gBACxD,4DAA4D;gBAC5D,SAAS,EAAE,OAAO;gBAClB,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,EAAE,MAAM,CAAC,CAAA;AACX,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/memory-promotion.test.d.ts

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

package/dist/runtime/query/__tests__/memory-promotion.test.d.ts.map

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

package/dist/runtime/query/__tests__/memory-promotion.test.js

@@ -0,0 +1,119 @@
+import { describe, expect, it, vi } from 'vitest';
+import { MockLLMProvider, registerMock } from '../../../provider/index.js';
+import { ToolRegistry } from '../../../registry/index.js';
+import { memoryCandidateFor } from '../../../types/run/memory-promotion.js';
+import { generateProjectId, generateSessionId, generateTenantId, generateThreadId, } from '../../../utils/id.js';
+import { drainQuery } from '../index.js';
+/**
+ * namzu could STORE a memory and could not FORM one. `MemoryStore` and
+ * its disk implementation have been here all along, and the only path
+ * into them was the model calling `save_memory` — so a run that worked
+ * out a durable fact and never thought to write it down lost it at
+ * settle, along with everything the compaction pass had already
+ * extracted and structured on the way.
+ */
+registerMock();
+function run(opts) {
+    return drainQuery({
+        provider: new MockLLMProvider({
+            turns: opts.failing ? [{ error: { message: 'provider is down' } }] : [{ text: 'done' }],
+        }),
+        tools: new ToolRegistry(),
+        agentId: 'a',
+        agentName: 'A',
+        messages: [{ role: 'user', content: 'ship the invoice job' }],
+        workingDirectory: process.cwd(),
+        runConfig: { model: 'mock', tokenBudget: 100_000, timeoutMs: 30_000, maxIterations: 2 },
+        projectId: generateProjectId(),
+        sessionId: generateSessionId(),
+        threadId: generateThreadId(),
+        tenantId: generateTenantId(),
+        ...(opts.compaction === false ? {} : { compactionConfig: { strategy: 'structured' } }),
+        ...(opts.promoteMemory ? { promoteMemory: opts.promoteMemory } : {}),
+    });
+}
+describe('what a finished run leaves behind', () => {
+    it('offers the extracted state when the run settles', async () => {
+        const promote = vi.fn();
+        await run({ promoteMemory: promote });
+        expect(promote).toHaveBeenCalledTimes(1);
+        const candidate = promote.mock.calls[0]?.[0];
+        expect(candidate.runId).toMatch(/^run_/);
+        expect(candidate.task).toContain('invoice');
+    });
+    it('offers it after a FAILED run too', async () => {
+        // A run that fell over still discovered things, and the approach that
+        // failed is exactly what a later run should not pay for twice.
+        const promote = vi.fn();
+        await run({ promoteMemory: promote, failing: true }).catch(() => { });
+        expect(promote).toHaveBeenCalledTimes(1);
+    });
+    it('does not retract an answer when the host throws', async () => {
+        // A memory that failed to form must not fail a run that already
+        // produced its answer.
+        const settled = await run({
+            promoteMemory: () => {
+                throw new Error('the memory store is unreachable');
+            },
+        });
+        expect(settled.status).toBe('completed');
+        expect(settled.result).toBe('done');
+    });
+    it('awaits an async host before the run returns', async () => {
+        // Fire-and-forget would race a one-shot process exiting, and the
+        // write would be lost precisely on the runs that are shortest.
+        // Asserted as an ORDER, not a flag: a flag checked after the run
+        // passes whenever the run happens to be slower than the write, which
+        // makes the test a race rather than a check.
+        const order = [];
+        await run({
+            promoteMemory: async () => {
+                await new Promise((resolve) => setTimeout(resolve, 50));
+                order.push('promoted');
+            },
+        });
+        order.push('returned');
+        expect(order).toEqual(['promoted', 'returned']);
+    });
+    it('is not consulted when no host asked for it', async () => {
+        const settled = await run({});
+        expect(settled.status).toBe('completed');
+    });
+    it('is not consulted when nothing extracted the state', async () => {
+        const promote = vi.fn();
+        await run({ promoteMemory: promote, compaction: false });
+        expect(promote).not.toHaveBeenCalled();
+    });
+});
+describe('whether there is anything to offer', () => {
+    // Tested apart from the run because inside the `try` the decision and
+    // the catch that swallows its failure look identical from outside: drop
+    // the guard and the candidate throws, the catch logs, and the run-level
+    // assertion still holds for the wrong reason.
+    it('is nothing at all without an extractor', () => {
+        // Inventing an empty candidate would ask a host to store a record of
+        // nothing.
+        expect(memoryCandidateFor('run_1', undefined)).toBeUndefined();
+    });
+    it('projects the state rather than handing it over', () => {
+        const state = {
+            task: 'ship it',
+            decisions: ['use the queue'],
+            discoveries: [],
+            userRequirements: ['never bill twice'],
+            failures: [],
+            environment: [],
+            files: new Map([['src/a.ts', {}]]),
+            evicted: { decisions: 2 },
+        };
+        const candidate = memoryCandidateFor('run_1', { getState: () => state });
+        expect(candidate?.files).toEqual(['src/a.ts']);
+        expect(candidate?.userRequirements).toEqual(['never bill twice']);
+        // Copied, so a host holding the candidate cannot edit the run's state.
+        expect(candidate?.decisions).not.toBe(state.decisions);
+        // Carried rather than hidden: a host deciding whether this is worth
+        // storing should know it is looking at a truncated record.
+        expect(candidate?.evicted).toEqual({ decisions: 2 });
+    });
+});
+//# sourceMappingURL=memory-promotion.test.js.map

package/dist/runtime/query/__tests__/memory-promotion.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-promotion.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/memory-promotion.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEjD,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAC1E,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAEzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,wCAAwC,CAAA;AAC3E,OAAO,EACN,iBAAiB,EACjB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,GAChB,MAAM,sBAAsB,CAAA;AAC7B,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC;;;;;;;GAOG;AAEH,YAAY,EAAE,CAAA;AAEd,SAAS,GAAG,CAAC,IAIZ;IACA,OAAO,UAAU,CAAC;QACjB,QAAQ,EAAE,IAAI,eAAe,CAAC;YAC7B,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,kBAAkB,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;SACvF,CAAC;QACF,KAAK,EAAE,IAAI,YAAY,EAAE;QACzB,OAAO,EAAE,GAAG;QACZ,SAAS,EAAE,GAAG;QACd,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,sBAAsB,EAAE,CAAC;QAC7D,gBAAgB,EAAE,OAAO,CAAC,GAAG,EAAE;QAC/B,SAAS,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,EAAE;QACvF,SAAS,EAAE,iBAAiB,EAAE;QAC9B,SAAS,EAAE,iBAAiB,EAAE;QAC9B,QAAQ,EAAE,gBAAgB,EAAE;QAC5B,QAAQ,EAAE,gBAAgB,EAAE;QAC5B,GAAG,CAAC,IAAI,CAAC,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,EAAE,CAAC;QACtF,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC3D,CAAC,CAAA;AACZ,CAAC;AAED,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QACvB,MAAM,GAAG,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAA;QAErC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QACxC,MAAM,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAuB,CAAA;QAClE,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACxC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QACjD,sEAAsE;QACtE,+DAA+D;QAC/D,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QACvB,MAAM,GAAG,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAA;QAEpE,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAChE,gEAAgE;QAChE,uBAAuB;QACvB,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC;YACzB,aAAa,EAAE,GAAG,EAAE;gBACnB,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAA;YACnD,CAAC;SACD,CAAC,CAAA;QAEF,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;QACxC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IACpC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC5D,iEAAiE;QACjE,+DAA+D;QAC/D,iEAAiE;QACjE,qEAAqE;QACrE,6CAA6C;QAC7C,MAAM,KAAK,GAAa,EAAE,CAAA;QAC1B,MAAM,GAAG,CAAC;YACT,aAAa,EAAE,KAAK,IAAI,EAAE;gBACzB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAA;gBACvD,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;YACvB,CAAC;SACD,CAAC,CAAA;QACF,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAEtB,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,CAAA;QAE7B,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;IACzC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QACvB,MAAM,GAAG,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAA;QAExD,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAA;IACvC,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IACnD,sEAAsE;IACtE,wEAAwE;IACxE,wEAAwE;IACxE,8CAA8C;IAE9C,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACjD,qEAAqE;QACrE,WAAW;QACX,MAAM,CAAC,kBAAkB,CAAC,OAAgB,EAAE,SAAS,CAAC,CAAC,CAAC,aAAa,EAAE,CAAA;IACxE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACzD,MAAM,KAAK,GAAG;YACb,IAAI,EAAE,SAAS;YACf,SAAS,EAAE,CAAC,eAAe,CAAC;YAC5B,WAAW,EAAE,EAAE;YACf,gBAAgB,EAAE,CAAC,kBAAkB,CAAC;YACtC,QAAQ,EAAE,EAAE;YACZ,WAAW,EAAE,EAAE;YACf,KAAK,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;YAClC,OAAO,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE;SACzB,CAAA;QACD,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAgB,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,CAAC,CAAA;QAEjF,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,CAAA;QAC9C,MAAM,CAAC,SAAS,EAAE,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAA;QACjE,uEAAuE;QACvE,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAA;QACtD,oEAAoE;QACpE,2DAA2D;QAC3D,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAA;IACrD,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/orphan-task-turn-end.test.js

@@ -3,35 +3,11 @@ import { mkdtemp } 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';
-const ZERO_USAGE = {
-    promptTokens: 0,
-    completionTokens: 0,
-    totalTokens: 0,
-    cachedTokens: 0,
-    cacheWriteTokens: 0,
-};
 /** Ends its turn with plain text on the first (and only) call. */
-class SingleTurnProvider {
-    id = 'single-turn';
-    name = 'Single Turn Provider';
-    calls = 0;
-    async *chatStream() {
-        this.calls += 1;
-        yield {
-            id: `msg_${this.calls}`,
-            delta: { content: 'Final answer.' },
-        };
-        yield {
-            id: `msg_${this.calls}`,
-            delta: {},
-            finishReason: 'stop',
-            usage: ZERO_USAGE,
-        };
-    }
-}
 /**
  * Gateway that permanently reports one running task. Every dispatch
  * tool is blocking, so a running task at end-of-turn is an orphan —
@@ -71,7 +47,8 @@ describe('end of turn with running agent tasks', () => {
     // hang detector here: with the busy-wait present this test times
     // out instead of completing.
     it('ends the run promptly instead of busy-waiting on orphan tasks', async () => {
-        const provider = new SingleTurnProvider();
+        // One text turn, no tools — the mock's default script shape.
+        const provider = new MockLLMProvider({ turns: [{ text: 'Final answer.' }] });
         const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-orphan-task-'));
         workdirs.push(workingDirectory);
         const events = [];
@@ -103,7 +80,8 @@ describe('end of turn with running agent tasks', () => {
         expect(run.stopReason).toBe('end_turn');
         expect(run.result).toBe('Final answer.');
         // One turn only — no futile re-invocation loop on the orphan.
-        expect(provider.calls).toBe(1);
+        // Exactly one model call: the run must NOT busy-wait on the orphan.
+        expect(provider.requests).toHaveLength(1);
         expect(events.some((event) => event.type === 'run_failed')).toBe(false);
     }, 10_000);
 });

package/dist/runtime/query/__tests__/orphan-task-turn-end.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"orphan-task-turn-end.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/orphan-task-turn-end.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AACrC,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;AAGhE,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,kEAAkE;AAClE,MAAM,kBAAkB;IACd,EAAE,GAAG,aAAa,CAAA;IAClB,IAAI,GAAG,sBAAsB,CAAA;IACtC,KAAK,GAAG,CAAC,CAAA;IAET,KAAK,CAAC,CAAC,UAAU;QAChB,IAAI,CAAC,KAAK,IAAI,CAAC,CAAA;QACf,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE,OAAO,EAAE,eAAe,EAAE;SACnC,CAAA;QACD,MAAM;YACL,EAAE,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE;YACvB,KAAK,EAAE,EAAE;YACT,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE,UAAU;SACjB,CAAA;IACF,CAAC;CACD;AAED;;;;;;GAMG;AACH,SAAS,iBAAiB;IACzB,MAAM,MAAM,GAAe;QAC1B,MAAM,EAAE,aAAuB;QAC/B,OAAO,EAAE,cAAc;QACvB,KAAK,EAAE,SAAS;QAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,CAAA;IACD,OAAO;QACN,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;QAC9B,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,CAAa,GAAG,EAAE,GAAE,CAAC,CAAC;QACpD,YAAY,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC5B,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;QACpB,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM;QACrB,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;QACzB,eAAe,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAE,CAAC;KAC/B,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACrD,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,iEAAiE;IACjE,oEAAoE;IACpE,iEAAiE;IACjE,+DAA+D;IAC/D,oEAAoE;IACpE,+DAA+D;IAC/D,iEAAiE;IACjE,6BAA6B;IAC7B,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC9E,MAAM,QAAQ,GAAG,IAAI,kBAAkB,EAAE,CAAA;QACzC,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAA;QAC5E,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,EAAE,IAAI,YAAY,EAAE;YACzB,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,uDAAuD;gBACvD,wDAAwD;gBACxD,SAAS,EAAE,OAAO;gBAClB,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,cAAc,CAAC,CAAC;YAC7C,gBAAgB;YAChB,WAAW,EAAE,iBAAiB,EAAE;YAChC,SAAS,EAAE,iBAA8B;YACzC,QAAQ,EAAE,iBAA6B;YACvC,SAAS,EAAE,iBAA8B;YACzC,QAAQ,EAAE,iBAA6B;SACvC,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,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACvC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACxC,8DAA8D;QAC9D,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC9B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACxE,CAAC,EAAE,MAAM,CAAC,CAAA;AACX,CAAC,CAAC,CAAA"}
+{"version":3,"file":"orphan-task-turn-end.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/orphan-task-turn-end.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAA;AACrC,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,eAAe,EAAE,MAAM,2BAA2B,CAAA;AAC3D,OAAO,EAAE,YAAY,EAAE,MAAM,mCAAmC,CAAA;AAGhE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iCAAiC,CAAA;AAGnE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAA;AAExC,kEAAkE;AAClE;;;;;;GAMG;AACH,SAAS,iBAAiB;IACzB,MAAM,MAAM,GAAe;QAC1B,MAAM,EAAE,aAAuB;QAC/B,OAAO,EAAE,cAAc;QACvB,KAAK,EAAE,SAAS;QAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,CAAA;IACD,OAAO;QACN,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,MAAM;QAC9B,WAAW,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,CAAa,GAAG,EAAE,GAAE,CAAC,CAAC;QACpD,YAAY,EAAE,KAAK,IAAI,EAAE,GAAE,CAAC;QAC5B,UAAU,EAAE,GAAG,EAAE,GAAE,CAAC;QACpB,OAAO,EAAE,GAAG,EAAE,CAAC,MAAM;QACrB,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;QACzB,eAAe,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAE,CAAC;KAC/B,CAAA;AACF,CAAC;AAED,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACrD,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,iEAAiE;IACjE,oEAAoE;IACpE,iEAAiE;IACjE,+DAA+D;IAC/D,oEAAoE;IACpE,+DAA+D;IAC/D,iEAAiE;IACjE,6BAA6B;IAC7B,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC9E,6DAA6D;QAC7D,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC,CAAA;QAC5E,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAA;QAC5E,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,EAAE,IAAI,YAAY,EAAE;YACzB,SAAS,EAAE;gBACV,KAAK,EAAE,YAAY;gBACnB,uDAAuD;gBACvD,wDAAwD;gBACxD,SAAS,EAAE,OAAO;gBAClB,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,cAAc,CAAC,CAAC;YAC7C,gBAAgB;YAChB,WAAW,EAAE,iBAAiB,EAAE;YAChC,SAAS,EAAE,iBAA8B;YACzC,QAAQ,EAAE,iBAA6B;YACvC,SAAS,EAAE,iBAA8B;YACzC,QAAQ,EAAE,iBAA6B;SACvC,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,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACvC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACxC,8DAA8D;QAC9D,oEAAoE;QACpE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACzC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IACxE,CAAC,EAAE,MAAM,CAAC,CAAA;AACX,CAAC,CAAC,CAAA"}

package/dist/runtime/query/__tests__/overflow-relief.test.d.ts

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

package/dist/runtime/query/__tests__/overflow-relief.test.d.ts.map

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

package/dist/runtime/query/__tests__/overflow-relief.test.js

@@ -0,0 +1,119 @@
+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';
+/**
+ * `context_length_exceeded` is the one provider failure the kernel can act
+ * on. It is correctly non-retryable — resending the identical prompt
+ * cannot help — and the kernel classified it precisely and then let the
+ * run die, while holding a compaction subsystem that could have made room.
+ *
+ * The threshold path guesses when to compact and can guess low, so this is
+ * not a hypothetical: a run carrying images, or a language the
+ * chars-per-token ratio does not fit, reaches the real window while still
+ * reading as comfortable.
+ */
+const dirs = [];
+afterEach(async () => {
+    await Promise.all(dirs.map((d) => rm(d, { recursive: true, force: true })));
+    dirs.length = 0;
+});
+/** Overflows until the prompt shrinks, then answers. */
+class OverflowingProvider extends MockLLMProvider {
+    limitChars;
+    seen = [];
+    constructor(limitChars) {
+        super({ turns: [{ text: 'answered after relief' }] });
+        this.limitChars = limitChars;
+    }
+    async *chatStream(params) {
+        const size = params.messages.reduce((n, m) => n + (typeof m.content === 'string' ? m.content.length : 0), 0);
+        this.seen.push(size);
+        if (size > this.limitChars) {
+            throw Object.assign(new Error('prompt is too long for this model: context_length_exceeded'), {
+                status: 400,
+            });
+        }
+        yield* super.chatStream(params);
+    }
+}
+async function run(provider, compaction) {
+    const workingDirectory = await mkdtemp(join(tmpdir(), 'namzu-overflow-'));
+    dirs.push(workingDirectory);
+    // A history big enough that shedding it changes the prompt size.
+    const messages = Array.from({ length: 12 }, (_, i) => createUserMessage(`turn ${i}: ${'context '.repeat(400)}`));
+    return drainQuery({
+        provider,
+        tools: new ToolRegistry(),
+        runConfig: {
+            model: 'mock-model',
+            timeoutMs: 20_000,
+            tokenBudget: 500_000,
+            maxIterations: 6,
+            maxResponseTokens: 256,
+        },
+        agentId: 'agent_o',
+        agentName: 'Overflow',
+        messages,
+        workingDirectory,
+        sessionId: 'ses_o',
+        threadId: 'thd_o',
+        projectId: 'prj_o',
+        tenantId: 'tnt_o',
+        retry: false,
+        ...(compaction
+            ? {
+                compactionConfig: {
+                    strategy: 'structured',
+                    triggerThreshold: 0.7,
+                    resetThreshold: 0.4,
+                    keepRecentMessages: 2,
+                    clearToolResults: true,
+                    keepRecentToolResults: 3,
+                    minToolResultCharsToClear: 1_000,
+                    maxToolResults: 30,
+                    maxListSize: 25,
+                    keepFirstEntries: 3,
+                    llmVerification: false,
+                    llmVerificationMaxTokens: 2048,
+                    richStateThreshold: 15,
+                    convoTextBudget: 12_000,
+                    maxSentencesPerTurn: 5,
+                    maxCharsPerNote: 500,
+                    maxCharsPerRequirement: 300,
+                    maxCharsPerTask: 400,
+                },
+            }
+            : {}),
+    });
+}
+describe('a context overflow is relieved rather than fatal', () => {
+    it('compacts and completes instead of dying', async () => {
+        const provider = new OverflowingProvider(20_000);
+        const result = await run(provider, true);
+        // Two attempts: the one that overflowed, and the one after relief.
+        expect(provider.seen.length).toBeGreaterThanOrEqual(2);
+        expect(provider.seen[1]).toBeLessThan(provider.seen[0]);
+        expect(result.result).toBe('answered after relief');
+        expect(result.stopReason).toBe('end_turn');
+    });
+    it('still fails when there is no compaction configured to shed with', async () => {
+        // Relief must not be claimed when nothing can be shed — retrying
+        // would send the identical prompt and arrive at the identical error.
+        const provider = new OverflowingProvider(20_000);
+        const result = await run(provider, false);
+        expect(result.status).toBe('failed');
+        expect(provider.seen).toHaveLength(1);
+    });
+    it('does not interfere with a run that never overflows', async () => {
+        const provider = new OverflowingProvider(10_000_000);
+        const result = await run(provider, true);
+        expect(provider.seen).toHaveLength(1);
+        expect(result.result).toBe('answered after relief');
+    });
+});
+//# sourceMappingURL=overflow-relief.test.js.map

package/dist/runtime/query/__tests__/overflow-relief.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"overflow-relief.test.js","sourceRoot":"","sources":["../../../../src/runtime/query/__tests__/overflow-relief.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;;;;;;;;;;GAUG;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,wDAAwD;AACxD,MAAM,mBAAoB,SAAQ,eAAe;IAEnB;IAD7B,IAAI,GAAa,EAAE,CAAA;IACnB,YAA6B,UAAkB;QAC9C,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,uBAAuB,EAAE,CAAC,EAAE,CAAC,CAAA;QADzB,eAAU,GAAV,UAAU,CAAQ;IAE/C,CAAC;IAEQ,KAAK,CAAC,CAAC,UAAU,CAAC,MAA4B;QACtD,MAAM,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAClC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EACpE,CAAC,CACD,CAAA;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpB,IAAI,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,4DAA4D,CAAC,EAAE;gBAC5F,MAAM,EAAE,GAAG;aACX,CAAC,CAAA;QACH,CAAC;QACD,KAAK,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAA;IAChC,CAAC;CACD;AAED,KAAK,UAAU,GAAG,CAAC,QAAyB,EAAE,UAAmB;IAChE,MAAM,gBAAgB,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAA;IACzE,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAA;IAE3B,iEAAiE;IACjE,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACpD,iBAAiB,CAAC,QAAQ,CAAC,KAAK,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CACzD,CAAA;IAED,OAAO,UAAU,CAAC;QACjB,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,iBAAiB,EAAE,GAAG;SACtB;QACD,OAAO,EAAE,SAAS;QAClB,SAAS,EAAE,UAAU;QACrB,QAAQ;QACR,gBAAgB;QAChB,SAAS,EAAE,OAAoB;QAC/B,QAAQ,EAAE,OAAmB;QAC7B,SAAS,EAAE,OAAoB;QAC/B,QAAQ,EAAE,OAAmB;QAC7B,KAAK,EAAE,KAAK;QACZ,GAAG,CAAC,UAAU;YACb,CAAC,CAAC;gBACA,gBAAgB,EAAE;oBACjB,QAAQ,EAAE,YAAqB;oBAC/B,gBAAgB,EAAE,GAAG;oBACrB,cAAc,EAAE,GAAG;oBACnB,kBAAkB,EAAE,CAAC;oBACrB,gBAAgB,EAAE,IAAI;oBACtB,qBAAqB,EAAE,CAAC;oBACxB,yBAAyB,EAAE,KAAK;oBAChC,cAAc,EAAE,EAAE;oBAClB,WAAW,EAAE,EAAE;oBACf,gBAAgB,EAAE,CAAC;oBACnB,eAAe,EAAE,KAAK;oBACtB,wBAAwB,EAAE,IAAI;oBAC9B,kBAAkB,EAAE,EAAE;oBACtB,eAAe,EAAE,MAAM;oBACvB,mBAAmB,EAAE,CAAC;oBACtB,eAAe,EAAE,GAAG;oBACpB,sBAAsB,EAAE,GAAG;oBAC3B,eAAe,EAAE,GAAG;iBACX;aACV;YACF,CAAC,CAAC,EAAE,CAAC;KACN,CAAC,CAAA;AACH,CAAC;AAED,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IACjE,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,QAAQ,GAAG,IAAI,mBAAmB,CAAC,MAAM,CAAC,CAAA;QAChD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QAExC,mEAAmE;QACnE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAA;QACtD,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAW,CAAC,CAAA;QACjE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAA;QACnD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;IAC3C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAChF,iEAAiE;QACjE,qEAAqE;QACrE,MAAM,QAAQ,GAAG,IAAI,mBAAmB,CAAC,MAAM,CAAC,CAAA;QAChD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAA;QAEzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QACpC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,QAAQ,GAAG,IAAI,mBAAmB,CAAC,UAAU,CAAC,CAAA;QACpD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAA;QAExC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;AACH,CAAC,CAAC,CAAA"}

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

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

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

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