@machina.ai/cell-cli-core 1.41.1-rc2 → 1.45.1-rc2

package/dist/src/context/graph/render.test.js

@@ -0,0 +1,280 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { render } from './render.js';
+import { NodeType } from './types.js';
+import { ContextWorkingBufferImpl } from '../pipeline/contextWorkingBuffer.js';
+describe('render', () => {
+    it('should render all provided nodes', async () => {
+        const mockNodes = [
+            {
+                id: '1',
+                type: NodeType.USER_PROMPT,
+                payload: {},
+            },
+            {
+                id: '2',
+                type: NodeType.AGENT_THOUGHT,
+                payload: {},
+            },
+            {
+                id: 'preview-1',
+                type: NodeType.USER_PROMPT,
+                payload: {},
+            },
+        ];
+        const orchestrator = {};
+        const sidecar = { config: {} }; // No budget
+        const mockAdvancedTokenCalculator = {
+            calculateTokensAndBaseUnits: vi.fn().mockReturnValue({
+                tokens: 100,
+                baseUnits: 100,
+            }),
+            getRawBaseUnits: vi.fn().mockReturnValue(100),
+            calculateConcreteListTokens: vi.fn().mockReturnValue(100),
+            getRawBaseUnitsForContent: vi.fn().mockReturnValue(0),
+        };
+        const env = {
+            tokenCalculator: {
+                calculateConcreteListTokens: vi.fn().mockReturnValue(100),
+                calculateTokenBreakdown: vi.fn().mockReturnValue({}),
+            },
+            graphMapper: {
+                fromGraph: vi.fn((nodes) => nodes.map((n) => ({ text: n.id }))),
+            },
+        };
+        const tracer = {
+            logEvent: vi.fn(),
+        };
+        const result = await render(mockNodes, orchestrator, sidecar, tracer, env, mockAdvancedTokenCalculator, {
+            protectionReasons: new Map(),
+            header: undefined,
+        });
+        expect(result.history).toEqual([
+            { text: '1' },
+            { text: '2' },
+            { text: 'preview-1' },
+        ]);
+        expect(result.baseUnits).toBe(100);
+    });
+    it('simulates the boundary knapsack problem (loose boundary policy)', async () => {
+        // 10k, 20k, 40k, 5k
+        const mockNodes = [
+            {
+                id: 'D',
+                type: NodeType.USER_PROMPT,
+                payload: {},
+            },
+            {
+                id: 'C',
+                type: NodeType.AGENT_THOUGHT,
+                payload: {},
+            },
+            {
+                id: 'B',
+                type: NodeType.USER_PROMPT,
+                payload: {},
+            },
+            {
+                id: 'A',
+                type: NodeType.AGENT_THOUGHT,
+                payload: {},
+            },
+        ];
+        const tokenMap = {
+            D: 5000,
+            C: 40000,
+            B: 20000,
+            A: 10000,
+        };
+        const orchestrator = {
+            executeTriggerSync: vi.fn(async (trigger, buffer, agedOutNodes) => {
+                const filteredNodes = buffer.nodes.filter((n) => !agedOutNodes.has(n.id));
+                return ContextWorkingBufferImpl.initialize(filteredNodes);
+            }),
+        };
+        const sidecar = {
+            config: {
+                budget: { maxTokens: 150000, retainedTokens: 65000 },
+            },
+        };
+        const currentTokens = 160000;
+        const mockAdvancedTokenCalculator = {
+            calculateTokensAndBaseUnits: vi.fn((nodes) => {
+                const tokens = nodes.length === 1 ? tokenMap[nodes[0].id] : currentTokens;
+                return { tokens, baseUnits: tokens };
+            }),
+            getRawBaseUnits: vi.fn((nodes) => {
+                if (nodes.length === 1)
+                    return tokenMap[nodes[0].id];
+                return currentTokens;
+            }),
+            calculateConcreteListTokens: vi.fn((nodes) => {
+                if (nodes.length === 1)
+                    return tokenMap[nodes[0].id];
+                return currentTokens;
+            }),
+        };
+        const env = {
+            llmClient: {
+                countTokens: vi.fn().mockResolvedValue({ totalTokens: 1000 }),
+            },
+            tokenCalculator: {
+                calculateConcreteListTokens: vi.fn((nodes) => {
+                    if (nodes.length === 1)
+                        return tokenMap[nodes[0].id];
+                    return currentTokens;
+                }),
+                calculateTokenBreakdown: vi.fn(() => ({})),
+            },
+            graphMapper: {
+                fromGraph: vi.fn((nodes) => nodes.map((n) => ({ text: n.id }))),
+            },
+        };
+        const tracer = {
+            logEvent: vi.fn(),
+        };
+        const result = await render(mockNodes, orchestrator, sidecar, tracer, env, mockAdvancedTokenCalculator, {
+            protectionReasons: new Map(),
+            header: undefined,
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const surviving = result.history.map((c) => c.text);
+        // Loose Boundary: A (10k), B (20k), C (40k). Total = 70k.
+        // Adding C pushes rolling total (70k) above retainedTokens (65k).
+        // Under loose policy, C survives. D is strictly older and drops.
+        expect(surviving).toEqual(['C', 'B', 'A']); // D is dropped
+        expect(result.baseUnits).toBe(160000);
+    });
+    it('drops nodes that are STRICTLY older than the boundary node', async () => {
+        const mockNodes = [
+            {
+                id: 'A',
+                type: NodeType.USER_PROMPT,
+                payload: {},
+            },
+            {
+                id: 'B',
+                type: NodeType.AGENT_THOUGHT,
+                payload: {},
+            },
+            {
+                id: 'C',
+                type: NodeType.USER_PROMPT,
+                payload: {},
+            },
+        ];
+        const tokenMap = {
+            C: 40000,
+            B: 40000,
+            A: 10000,
+        };
+        const orchestrator = {
+            executeTriggerSync: vi.fn(async (trigger, buffer, agedOutNodes) => {
+                const filteredNodes = buffer.nodes.filter((n) => !agedOutNodes.has(n.id));
+                return ContextWorkingBufferImpl.initialize(filteredNodes);
+            }),
+        };
+        const sidecar = {
+            config: {
+                budget: { maxTokens: 150000, retainedTokens: 65000 },
+            },
+        };
+        const currentTokens = 160000;
+        const mockAdvancedTokenCalculator = {
+            calculateTokensAndBaseUnits: vi.fn((nodes) => {
+                const tokens = nodes.length === 1 ? tokenMap[nodes[0].id] : currentTokens;
+                return { tokens, baseUnits: tokens };
+            }),
+            getRawBaseUnits: vi.fn((nodes) => {
+                if (nodes.length === 1)
+                    return tokenMap[nodes[0].id];
+                return currentTokens;
+            }),
+            calculateConcreteListTokens: vi.fn((nodes) => {
+                if (nodes.length === 1)
+                    return tokenMap[nodes[0].id];
+                return currentTokens;
+            }),
+        };
+        const env = {
+            llmClient: {
+                countTokens: vi.fn().mockResolvedValue({ totalTokens: 1000 }),
+            },
+            tokenCalculator: {
+                calculateConcreteListTokens: vi.fn((nodes) => {
+                    if (nodes.length === 1)
+                        return tokenMap[nodes[0].id];
+                    return currentTokens;
+                }),
+                calculateTokenBreakdown: vi.fn(() => ({})),
+            },
+            graphMapper: {
+                fromGraph: vi.fn((nodes) => nodes.map((n) => ({ text: n.id }))),
+            },
+        };
+        const tracer = {
+            logEvent: vi.fn(),
+        };
+        const result = await render(mockNodes, orchestrator, sidecar, tracer, env, mockAdvancedTokenCalculator, {
+            protectionReasons: new Map(),
+            header: undefined,
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const surviving = result.history.map((c) => c.text);
+        // C(40k), B(40k). Adding B pushes total to 80k. B is the boundary node and survives. A drops.
+        expect(surviving).toEqual(['B', 'C']); // A is dropped
+        expect(result.baseUnits).toBe(160000);
+    });
+    it('should exclude the last turn when lateBindPrompt is true', async () => {
+        const mockNodes = [
+            {
+                id: '1',
+                type: NodeType.USER_PROMPT,
+                turnId: 'turn-1',
+                payload: {},
+            },
+            {
+                id: '2',
+                type: NodeType.AGENT_THOUGHT,
+                turnId: 'turn-2',
+                payload: {},
+            },
+        ];
+        const orchestrator = {
+            executeTriggerSync: vi.fn(async (trigger, buffer) => buffer),
+        };
+        const sidecar = { config: {} }; // No budget
+        const mockAdvancedTokenCalculator = {
+            calculateTokensAndBaseUnits: vi.fn().mockReturnValue({
+                tokens: 100,
+                baseUnits: 100,
+            }),
+            getRawBaseUnits: vi.fn().mockReturnValue(50),
+            calculateConcreteListTokens: vi.fn().mockReturnValue(100),
+            getRawBaseUnitsForContent: vi.fn().mockReturnValue(0),
+        };
+        const env = {
+            tokenCalculator: {
+                calculateConcreteListTokens: vi.fn().mockReturnValue(100),
+                calculateTokenBreakdown: vi.fn().mockReturnValue({}),
+            },
+            graphMapper: {
+                fromGraph: vi.fn((nodes) => nodes.map((n) => ({ text: n.id }))),
+            },
+        };
+        const tracer = {
+            logEvent: vi.fn(),
+        };
+        const result = await render(mockNodes, orchestrator, sidecar, tracer, env, mockAdvancedTokenCalculator, {
+            lateBindPrompt: true,
+        });
+        expect(result.history).toEqual([{ text: '1' }]); // Turn 2 (node 2) is excluded
+        expect(result.pendingHistory).toEqual([{ text: '2' }]); // Turn 2 is included here
+        expect(result.baseUnits).toBe(50);
+    });
+});
+//# sourceMappingURL=render.test.js.map

package/dist/src/context/graph/render.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"render.test.js","sourceRoot":"","sources":["../../../../src/context/graph/render.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AAQtC,OAAO,EAAE,wBAAwB,EAAE,MAAM,qCAAqC,CAAC;AAE/E,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;IACtB,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,SAAS,GAAmB;YAChC;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,aAAa;gBAC5B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,WAAW;gBACf,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,OAAO,EAAE,EAAU;aACO;SAC7B,CAAC;QAEF,MAAM,YAAY,GAAG,EAA0B,CAAC;QAChD,MAAM,OAAO,GAAG,EAAE,MAAM,EAAE,EAAE,EAAoB,CAAC,CAAC,YAAY;QAC9D,MAAM,2BAA2B,GAAG;YAClC,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACnD,MAAM,EAAE,GAAG;gBACX,SAAS,EAAE,GAAG;aACf,CAAC;YACF,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC;YAC7C,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC;YACzD,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;SACtD,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,eAAe,EAAE;gBACf,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC;gBACzD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;aACrD;YACD,WAAW,EAAE;gBACX,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE,CAClD,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CACnC;aACF;SAC+B,CAAC;QACnC,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SACU,CAAC;QAE9B,MAAM,MAAM,GAAG,MAAM,MAAM,CACzB,SAAS,EACT,YAAY,EACZ,OAAO,EACP,MAAM,EACN,GAAG,EACH,2BAAiE,EACjE;YACE,iBAAiB,EAAE,IAAI,GAAG,EAAE;YAC5B,MAAM,EAAE,SAAS;SAClB,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC;YAC7B,EAAE,IAAI,EAAE,GAAG,EAAE;YACb,EAAE,IAAI,EAAE,GAAG,EAAE;YACb,EAAE,IAAI,EAAE,WAAW,EAAE;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,oBAAoB;QACpB,MAAM,SAAS,GAAmB;YAChC;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,aAAa;gBAC5B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,aAAa;gBAC5B,OAAO,EAAE,EAAU;aACO;SAC7B,CAAC;QAEF,MAAM,QAAQ,GAA2B;YACvC,CAAC,EAAE,IAAI;YACP,CAAC,EAAE,KAAK;YACR,CAAC,EAAE,KAAK;YACR,CAAC,EAAE,KAAK;SACT,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE;gBAChE,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CACvC,CAAC,CAAe,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAC7C,CAAC;gBACF,OAAO,wBAAwB,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YAC5D,CAAC,CAAC;SACgC,CAAC;QAErC,MAAM,OAAO,GAAG;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE;aACrD;SAC2B,CAAC;QAE/B,MAAM,aAAa,GAAG,MAAM,CAAC;QAE7B,MAAM,2BAA2B,GAAG;YAClC,2BAA2B,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;gBACpE,MAAM,MAAM,GACV,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC7D,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;YACvC,CAAC,CAAC;YACF,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;gBACxD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC;YACF,2BAA2B,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;gBACpE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC;SACH,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,SAAS,EAAE;gBACT,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;aAC9D;YACD,eAAe,EAAE;gBACf,2BAA2B,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;oBACpE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;wBAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACrD,OAAO,aAAa,CAAC;gBACvB,CAAC,CAAC;gBACF,uBAAuB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;aAC3C;YACD,WAAW,EAAE;gBACX,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE,CAClD,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CACnC;aACF;SAC+B,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SACU,CAAC;QAE9B,MAAM,MAAM,GAAG,MAAM,MAAM,CACzB,SAAS,EACT,YAAY,EACZ,OAAO,EACP,MAAM,EACN,GAAG,EACH,2BAAiE,EACjE;YACE,iBAAiB,EAAE,IAAI,GAAG,EAAE;YAC5B,MAAM,EAAE,SAAS;SAClB,CACF,CAAC;QAEF,8DAA8D;QAC9D,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACzD,0DAA0D;QAC1D,kEAAkE;QAClE,iEAAiE;QACjE,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe;QAC3D,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,SAAS,GAAmB;YAChC;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,aAAa;gBAC5B,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,OAAO,EAAE,EAAU;aACO;SAC7B,CAAC;QAEF,MAAM,QAAQ,GAA2B;YACvC,CAAC,EAAE,KAAK;YACR,CAAC,EAAE,KAAK;YACR,CAAC,EAAE,KAAK;SACT,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE;gBAChE,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CACvC,CAAC,CAAe,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAC7C,CAAC;gBACF,OAAO,wBAAwB,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;YAC5D,CAAC,CAAC;SACgC,CAAC;QAErC,MAAM,OAAO,GAAG;YACd,MAAM,EAAE;gBACN,MAAM,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE;aACrD;SAC2B,CAAC;QAE/B,MAAM,aAAa,GAAG,MAAM,CAAC;QAE7B,MAAM,2BAA2B,GAAG;YAClC,2BAA2B,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;gBACpE,MAAM,MAAM,GACV,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;gBAC7D,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;YACvC,CAAC,CAAC;YACF,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;gBACxD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC;YACF,2BAA2B,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;gBACpE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACrD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC;SACH,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,SAAS,EAAE;gBACT,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;aAC9D;YACD,eAAe,EAAE;gBACf,2BAA2B,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE;oBACpE,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;wBAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACrD,OAAO,aAAa,CAAC;gBACvB,CAAC,CAAC;gBACF,uBAAuB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;aAC3C;YACD,WAAW,EAAE;gBACX,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE,CAClD,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CACnC;aACF;SAC+B,CAAC;QAEnC,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SACU,CAAC;QAE9B,MAAM,MAAM,GAAG,MAAM,MAAM,CACzB,SAAS,EACT,YAAY,EACZ,OAAO,EACP,MAAM,EACN,GAAG,EACH,2BAAiE,EACjE;YACE,iBAAiB,EAAE,IAAI,GAAG,EAAE;YAC5B,MAAM,EAAE,SAAS;SAClB,CACF,CAAC;QAEF,8DAA8D;QAC9D,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACzD,8FAA8F;QAC9F,MAAM,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe;QACtD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,SAAS,GAAmB;YAChC;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,WAAW;gBAC1B,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,EAAU;aACO;YAC5B;gBACE,EAAE,EAAE,GAAG;gBACP,IAAI,EAAE,QAAQ,CAAC,aAAa;gBAC5B,MAAM,EAAE,QAAQ;gBAChB,OAAO,EAAE,EAAU;aACO;SAC7B,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,kBAAkB,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC;SAC1B,CAAC;QACrC,MAAM,OAAO,GAAG,EAAE,MAAM,EAAE,EAAE,EAAoB,CAAC,CAAC,YAAY;QAC9D,MAAM,2BAA2B,GAAG;YAClC,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACnD,MAAM,EAAE,GAAG;gBACX,SAAS,EAAE,GAAG;aACf,CAAC;YACF,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;YAC5C,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC;YACzD,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;SACtD,CAAC;QAEF,MAAM,GAAG,GAAG;YACV,eAAe,EAAE;gBACf,2BAA2B,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC;gBACzD,uBAAuB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;aACrD;YACD,WAAW,EAAE;gBACX,SAAS,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA8B,EAAE,EAAE,CAClD,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CACnC;aACF;SAC+B,CAAC;QACnC,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;SACU,CAAC;QAE9B,MAAM,MAAM,GAAG,MAAM,MAAM,CACzB,SAAS,EACT,YAAY,EACZ,OAAO,EACP,MAAM,EACN,GAAG,EACH,2BAAiE,EACjE;YACE,cAAc,EAAE,IAAI;SACrB,CACF,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,8BAA8B;QAC/E,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,0BAA0B;QAClF,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/context/graph/toGraph.d.ts

@@ -3,19 +3,21 @@
  * Copyright 2026 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import type { Content } from '@google/genai';
-import type { ConcreteNode } from './types.js';
-import type { ContextTokenCalculator } from '../utils/contextTokenCalculator.js';
-export declare function getStableId(obj: object, nodeIdentityMap: WeakMap<object, string>): string;
+import { type ConcreteNode } from './types.js';
+import type { NodeIdService } from './nodeIdService.js';
+import type { HistoryTurn } from '../../core/agentChatHistory.js';
+/**
+ * Generates a stable ID for an object reference using a NodeIdService.
+ * Falls back to content-based hashing for Part-like objects to ensure
+ * stability across object re-creations (e.g. during history mapping).
+ */
+export declare function getStableId(obj: object, idService: NodeIdService, turnSalt?: string, partIdx?: number): string;
+/**
+ * Builds a 1:1 Mirror Graph from Chat History.
+ * Every Part in history is mapped to exactly one ConcreteNode.
+ */
 export declare class ContextGraphBuilder {
-    private readonly tokenCalculator;
-    private readonly nodeIdentityMap;
-    private episodes;
-    private currentEpisode;
-    private pendingCallParts;
-    private pendingCallPartsWithoutId;
-    constructor(tokenCalculator: ContextTokenCalculator, nodeIdentityMap?: WeakMap<object, string>);
-    clear(): void;
-    processHistory(history: readonly Content[]): void;
-    getNodes(): ConcreteNode[];
+    private readonly idService;
+    constructor(idService: NodeIdService);
+    processHistory(history: readonly HistoryTurn[]): ConcreteNode[];
 }