@strands-agents/sdk 1.5.0 → 1.7.0

package/dist/src/middleware/__tests__/copy-on-input.test.js

@@ -0,0 +1,379 @@
+import { describe, expect, it } from 'vitest';
+import { Agent } from '../../agent/agent.js';
+import { MockMessageModel } from '../../__fixtures__/mock-message-model.js';
+import { createMockTool } from '../../__fixtures__/tool-helpers.js';
+import { InvokeModelStage, ExecuteToolStage, AgentStreamStage } from '../stages.js';
+import { TextBlock, ToolResultBlock, Message } from '../../types/messages.js';
+describe('InvokeModelStage copy-on-input isolation', () => {
+    describe('messages', () => {
+        it('middleware receives a deep copy of messages, not a reference to the agent array', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            let receivedMessages;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedMessages = context.messages;
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(receivedMessages).not.toBe(agent.messages);
+        });
+        it('mutating the context messages array does not affect agent.messages', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                // Force-cast to bypass readonly for this mutation test
+                ;
+                context.messages.push(new Message({ role: 'user', content: [new TextBlock('injected')] }));
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            // Agent messages should only contain the original user message + assistant response,
+            // not the injected one
+            const userMessages = agent.messages.filter((m) => m.role === 'user');
+            expect(userMessages).toHaveLength(1);
+            expect(userMessages[0].content[0]).toBeInstanceOf(TextBlock);
+            expect(userMessages[0].content[0].text).toBe('Hello');
+        });
+        it('message content blocks are deep copied (not shared references)', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            let receivedContent;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedContent = context.messages[0]?.content[0];
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            // The content block in middleware should be a different instance
+            expect(receivedContent).not.toBe(agent.messages[0]?.content[0]);
+        });
+        it('message metadata is deep copied (not shared reference)', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                const msg = context.messages[0];
+                if (msg?.metadata) {
+                    msg.metadata.custom = { ...(msg.metadata.custom ?? {}), injected: 'value' };
+                }
+                return yield* next(context);
+            });
+            // Pre-set metadata on the agent's message so the middleware has something to mutate
+            agent.messages.push(new Message({
+                role: 'user',
+                content: [new TextBlock('Hello')],
+                metadata: { custom: { original: 'data' } },
+            }));
+            await agent.invoke('Hello');
+            const userMsg = agent.messages.find((m) => m.role === 'user' && m.metadata?.custom?.['original'] === 'data');
+            expect(userMsg?.metadata?.custom).toEqual({ original: 'data' });
+            expect(userMsg?.metadata?.custom).not.toHaveProperty('injected');
+        });
+    });
+    describe('systemPrompt', () => {
+        it('string systemPrompt is passed through (immutable primitive)', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false, systemPrompt: 'Be helpful' });
+            let receivedPrompt;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedPrompt = context.systemPrompt;
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(receivedPrompt).toBe('Be helpful');
+        });
+        it('array systemPrompt is deep copied', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const systemBlocks = [new TextBlock('Be helpful')];
+            const agent = new Agent({ model, printer: false, systemPrompt: systemBlocks });
+            let receivedPrompt;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedPrompt = context.systemPrompt;
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(receivedPrompt).not.toBe(systemBlocks);
+            expect(Array.isArray(receivedPrompt)).toBe(true);
+            expect(receivedPrompt[0]).not.toBe(systemBlocks[0]);
+        });
+    });
+    describe('toolSpecs', () => {
+        it('toolSpecs are deep copied across invocations', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const tool = createMockTool('testTool', () => new ToolResultBlock({
+                toolUseId: 'tool-1',
+                status: 'success',
+                content: [new TextBlock('ok')],
+            }));
+            const agent = new Agent({ model, tools: [tool], printer: false });
+            const receivedSpecs = [];
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedSpecs.push(context.toolSpecs);
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            await agent.invoke('Hello again');
+            // Each invocation should get a fresh copy
+            expect(receivedSpecs[0]).not.toBe(receivedSpecs[1]);
+            expect(receivedSpecs[0]).toEqual(receivedSpecs[1]);
+            expect(receivedSpecs[0]).toEqual(expect.arrayContaining([expect.objectContaining({ name: 'testTool' })]));
+        });
+    });
+    describe('modelState', () => {
+        it('modelState is not exposed on the middleware context', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false, systemPrompt: 'hi', modelState: { existing: 'value' } });
+            let contextKeys = [];
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                contextKeys = Object.keys(context);
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(contextKeys.sort()).toEqual(['agent', 'invocationState', 'messages', 'projectedInputTokens', 'systemPrompt', 'toolSpecs'].sort());
+        });
+        it('model state changes are written back to agent.modelState after streaming', async () => {
+            const model = new MockMessageModel();
+            const originalStream = model.stream.bind(model);
+            model.stream = async function* (messages, options) {
+                options?.modelState?.set('responseId', 'resp_123');
+                yield* originalStream(messages, options);
+            };
+            model.addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            await agent.invoke('Hello');
+            expect(agent.modelState.getAll()).toEqual({ responseId: 'resp_123' });
+        });
+        it('model state deletions are synced back to agent.modelState', async () => {
+            const model = new MockMessageModel();
+            const originalStream = model.stream.bind(model);
+            model.stream = async function* (messages, options) {
+                options?.modelState?.delete('oldKey');
+                options?.modelState?.set('newKey', 'fresh');
+                yield* originalStream(messages, options);
+            };
+            model.addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false, modelState: { oldKey: 'stale' } });
+            await agent.invoke('Hello');
+            expect(agent.modelState.get('oldKey')).toBeUndefined();
+            expect(agent.modelState.get('newKey')).toBe('fresh');
+        });
+        it('middleware mutations to agent.modelState before next() do not affect the model call', async () => {
+            const model = new MockMessageModel();
+            const originalStream = model.stream.bind(model);
+            let modelReceivedState;
+            model.stream = async function* (messages, options) {
+                modelReceivedState = options?.modelState?.getAll();
+                yield* originalStream(messages, options);
+            };
+            model.addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false, modelState: { key: 'snapshotted' } });
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                agent.modelState.set('key', 'mutated-before-next');
+                agent.modelState.set('extra', 'injected');
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(modelReceivedState).toEqual({ key: 'snapshotted' });
+        });
+        it('middleware mutations to agent.modelState after next() do not persist', async () => {
+            const model = new MockMessageModel();
+            const originalStream = model.stream.bind(model);
+            model.stream = async function* (messages, options) {
+                options?.modelState?.set('fromModel', 'model-wrote-this');
+                yield* originalStream(messages, options);
+            };
+            model.addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false, modelState: { key: 'original' } });
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                const result = yield* next(context);
+                agent.modelState.set('sneaky', 'should-be-gone');
+                agent.modelState.set('fromModel', 'overwritten');
+                return result;
+            });
+            await agent.invoke('Hello');
+            // Writeback happens after the full middleware chain, discarding any mutations
+            expect(agent.modelState.get('fromModel')).toBe('model-wrote-this');
+            expect(agent.modelState.get('sneaky')).toBeUndefined();
+            // 'key' persists because it was in the snapshot that the model received
+            expect(agent.modelState.get('key')).toBe('original');
+        });
+    });
+    describe('invocationState', () => {
+        it('invocationState is shared by reference — mutations are visible on the result', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                ;
+                context.invocationState['middlewareTouched'] = true;
+                return yield* next(context);
+            });
+            const result = await agent.invoke('Hello', { invocationState: { myKey: 'myValue' } });
+            expect(result.invocationState).toEqual({ myKey: 'myValue', middlewareTouched: true });
+        });
+    });
+    describe('toolChoice', () => {
+        it('toolChoice is undefined when not explicitly configured', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            let receivedChoice;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedChoice = context.toolChoice;
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(receivedChoice).toBeUndefined();
+        });
+        it('toolChoice is deep copied when present', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'plain response' }).addTurn({
+                type: 'toolUseBlock',
+                name: 'strands_structured_output',
+                toolUseId: 'so-1',
+                input: { name: 'Alice' },
+            });
+            const { z } = await import('zod');
+            const agent = new Agent({
+                model,
+                printer: false,
+                structuredOutputSchema: z.object({ name: z.string() }),
+            });
+            const receivedChoices = [];
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                receivedChoices.push(context.toolChoice);
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            // Second call should have toolChoice forced by structured output
+            const definedChoice = receivedChoices.find((c) => c !== undefined);
+            expect(definedChoice).toEqual({ tool: { name: 'strands_structured_output' } });
+        });
+    });
+    describe('functional style — passing modified context to next()', () => {
+        it('middleware can pass a new context with modified messages to next()', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            let terminalMessages;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                const newContext = {
+                    ...context,
+                    messages: [...context.messages, new Message({ role: 'user', content: [new TextBlock('extra context')] })],
+                };
+                return yield* next(newContext);
+            });
+            // Add a second middleware closer to terminal to capture what actually arrives
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                terminalMessages = context.messages;
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(terminalMessages).toHaveLength(2);
+            expect(terminalMessages[1].content[0].text).toBe('extra context');
+        });
+        it('middleware can pass a new context with modified toolSpecs to next()', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const tool = createMockTool('myTool', () => new ToolResultBlock({
+                toolUseId: 'tool-1',
+                status: 'success',
+                content: [new TextBlock('ok')],
+            }));
+            const agent = new Agent({ model, tools: [tool], printer: false });
+            let terminalSpecs;
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                // Filter out all tools
+                const newContext = {
+                    ...context,
+                    toolSpecs: [],
+                };
+                return yield* next(newContext);
+            });
+            agent.addMiddleware(InvokeModelStage, async function* (context, next) {
+                terminalSpecs = context.toolSpecs;
+                return yield* next(context);
+            });
+            await agent.invoke('Hello');
+            expect(terminalSpecs).toEqual([]);
+        });
+    });
+});
+describe('ExecuteToolStage copy-on-input isolation', () => {
+    function createToolCallingModel() {
+        return new MockMessageModel()
+            .addTurn({
+            type: 'toolUseBlock',
+            name: 'testTool',
+            toolUseId: 'tool-1',
+            input: { key: 'value', nested: { arr: [1, 2, 3] } },
+        })
+            .addTurn({ type: 'textBlock', text: 'Done' });
+    }
+    describe('toolUse.input', () => {
+        it('toolUse.input is a deep copy of the model output', async () => {
+            const model = createToolCallingModel();
+            const tool = createMockTool('testTool', () => new ToolResultBlock({
+                toolUseId: 'tool-1',
+                status: 'success',
+                content: [new TextBlock('ok')],
+            }));
+            const agent = new Agent({ model, tools: [tool], printer: false });
+            let receivedInput;
+            agent.addMiddleware(ExecuteToolStage, async function* (context, next) {
+                receivedInput = context.toolUse.input;
+                return yield* next(context);
+            });
+            await agent.invoke('Call the tool');
+            expect(receivedInput).toEqual({ key: 'value', nested: { arr: [1, 2, 3] } });
+        });
+        it('mutating toolUse.input in middleware does not affect the original message', async () => {
+            const model = createToolCallingModel();
+            const tool = createMockTool('testTool', () => new ToolResultBlock({
+                toolUseId: 'tool-1',
+                status: 'success',
+                content: [new TextBlock('ok')],
+            }));
+            const agent = new Agent({ model, tools: [tool], printer: false });
+            agent.addMiddleware(ExecuteToolStage, async function* (context, next) {
+                ;
+                context.toolUse.input['injected'] = true;
+                return yield* next(context);
+            });
+            await agent.invoke('Call the tool');
+            // The original tool use block in the assistant message should be unaffected
+            const assistantMsg = agent.messages.find((m) => m.role === 'assistant');
+            const toolUseBlock = assistantMsg?.content.find((b) => b.type === 'toolUseBlock');
+            expect(toolUseBlock).toBeDefined();
+            expect(toolUseBlock.input).not.toHaveProperty('injected');
+        });
+    });
+    describe('invocationState', () => {
+        it('invocationState is shared by reference (intentionally mutable)', async () => {
+            const model = createToolCallingModel();
+            const tool = createMockTool('testTool', () => new ToolResultBlock({
+                toolUseId: 'tool-1',
+                status: 'success',
+                content: [new TextBlock('ok')],
+            }));
+            const agent = new Agent({ model, tools: [tool], printer: false });
+            agent.addMiddleware(ExecuteToolStage, async function* (context, next) {
+                ;
+                context.invocationState['middlewareTouched'] = true;
+                return yield* next(context);
+            });
+            const result = await agent.invoke('Call the tool', { invocationState: { original: true } });
+            expect(result.invocationState).toEqual({ original: true, middlewareTouched: true });
+        });
+    });
+});
+describe('AgentStreamStage shared references', () => {
+    describe('args', () => {
+        it('args is shared by reference with the caller', async () => {
+            const model = new MockMessageModel().addTurn({ type: 'textBlock', text: 'Hi' });
+            const agent = new Agent({ model, printer: false });
+            const inputMessages = [new Message({ role: 'user', content: [new TextBlock('Hello')] })];
+            let receivedArgs;
+            agent.addMiddleware(AgentStreamStage, async function* (context, next) {
+                receivedArgs = context.args;
+                return yield* next(context);
+            });
+            await agent.invoke(inputMessages);
+            expect(receivedArgs).toBe(inputMessages);
+        });
+    });
+});
+//# sourceMappingURL=copy-on-input.test.js.map

package/dist/src/middleware/__tests__/copy-on-input.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"copy-on-input.test.js","sourceRoot":"","sources":["../../../../src/middleware/__tests__/copy-on-input.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,sBAAsB,CAAA;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAA;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAA;AACnE,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAA;AAEnF,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAA;AAG7E,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,IAAI,gBAAgD,CAAA;YAEpD,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,gBAAgB,GAAG,OAAO,CAAC,QAAQ,CAAA;gBACnC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,uDAAuD;gBACvD,CAAC;gBAAC,OAAO,CAAC,QAAsB,CAAC,IAAI,CAAC,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;gBAC1G,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,qFAAqF;YACrF,uBAAuB;YACvB,MAAM,YAAY,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAA;YACpE,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACpC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,SAAS,CAAC,CAAA;YAC7D,MAAM,CAAE,YAAY,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,IAAI,eAAwB,CAAA;YAE5B,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,eAAe,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;gBACjD,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,iEAAiE;YACjE,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAA;gBAC/B,IAAI,GAAG,EAAE,QAAQ,EAAE,CAAC;oBAClB,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,IAAI,EAAE,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAA;gBAC7E,CAAC;gBACD,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,oFAAoF;YACpF,KAAK,CAAC,QAAQ,CAAC,IAAI,CACjB,IAAI,OAAO,CAAC;gBACV,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC;gBACjC,QAAQ,EAAE,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE;aAC3C,CAAC,CACH,CAAA;YAED,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC,UAAU,CAAC,KAAK,MAAM,CAAC,CAAA;YAC5G,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAA;YAC/D,MAAM,CAAC,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;QAClE,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAA;YAE9E,IAAI,cAAuB,CAAA;YAE3B,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,cAAc,GAAG,OAAO,CAAC,YAAY,CAAA;gBACrC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;QAC3C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,YAAY,GAAG,CAAC,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC,CAAA;YAClD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,CAAC,CAAA;YAE9E,IAAI,cAAuB,CAAA;YAE3B,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,cAAc,GAAG,OAAO,CAAC,YAAY,CAAA;gBACrC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;YAC7C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAChD,MAAM,CAAE,cAA8B,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAA;QACtE,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,IAAI,GAAG,cAAc,CACzB,UAAU,EACV,GAAG,EAAE,CACH,IAAI,eAAe,CAAC;gBAClB,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;aAC/B,CAAC,CACL,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAEjE,MAAM,aAAa,GAAc,EAAE,CAAA;YAEnC,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;gBACrC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAC3B,MAAM,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;YAEjC,0CAA0C;YAC1C,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAA;YACnD,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAA;YAClD,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC,CAAA;QAC3G,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,CAAC,CAAA;YAEzG,IAAI,WAAW,GAAa,EAAE,CAAA;YAE9B,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;gBAClC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAChC,CAAC,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,sBAAsB,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,EAAE,CACrG,CAAA;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAA;YACpC,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC/C,KAAK,CAAC,MAAM,GAAG,KAAK,SAAS,CAAC,EAAE,QAAQ,EAAE,OAAO;gBAC/C,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,YAAY,EAAE,UAAU,CAAC,CAAA;gBAClD,KAAK,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;YAC1C,CAAC,CAAA;YACD,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAEhD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAClD,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,CAAC,CAAA;QACvE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAA;YACpC,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC/C,KAAK,CAAC,MAAM,GAAG,KAAK,SAAS,CAAC,EAAE,QAAQ,EAAE,OAAO;gBAC/C,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAA;gBACrC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;gBAC3C,KAAK,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;YAC1C,CAAC,CAAA;YACD,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAEhD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,CAAA;YACnF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,EAAE,CAAA;YACtD,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qFAAqF,EAAE,KAAK,IAAI,EAAE;YACnG,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAA;YACpC,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC/C,IAAI,kBAAuD,CAAA;YAC3D,KAAK,CAAC,MAAM,GAAG,KAAK,SAAS,CAAC,EAAE,QAAQ,EAAE,OAAO;gBAC/C,kBAAkB,GAAG,OAAO,EAAE,UAAU,EAAE,MAAM,EAAE,CAAA;gBAClD,KAAK,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;YAC1C,CAAC,CAAA;YACD,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAEhD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,aAAa,EAAE,EAAE,CAAC,CAAA;YAEtF,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,CAAC,CAAA;gBAClD,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAA;gBACzC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,aAAa,EAAE,CAAC,CAAA;QAC5D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAA;YACpC,MAAM,cAAc,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC/C,KAAK,CAAC,MAAM,GAAG,KAAK,SAAS,CAAC,EAAE,QAAQ,EAAE,OAAO;gBAC/C,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAA;gBACzD,KAAK,CAAC,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAA;YAC1C,CAAC,CAAA;YACD,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAEhD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,EAAE,CAAC,CAAA;YAEnF,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;gBACnC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAA;gBAChD,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,WAAW,EAAE,aAAa,CAAC,CAAA;gBAChD,OAAO,MAAM,CAAA;YACf,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,8EAA8E;YAC9E,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;YAClE,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,aAAa,EAAE,CAAA;YACtD,wEAAwE;YACxE,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,CAAC;gBAAC,OAAO,CAAC,eAA2C,CAAC,mBAAmB,CAAC,GAAG,IAAI,CAAA;gBACjF,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,eAAe,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC,CAAA;YAErF,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAA;QACvF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,IAAI,cAAuB,CAAA;YAE3B,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,cAAc,GAAG,OAAO,CAAC,UAAU,CAAA;gBACnC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,cAAc,CAAC,CAAC,aAAa,EAAE,CAAA;QACxC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,gBAAgB,EAAE,CAAC,CAAC,OAAO,CAAC;gBAClG,IAAI,EAAE,cAAc;gBACpB,IAAI,EAAE,2BAA2B;gBACjC,SAAS,EAAE,MAAM;gBACjB,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE;aACzB,CAAC,CAAA;YACF,MAAM,EAAE,CAAC,EAAE,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,CAAA;YACjC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC;gBACtB,KAAK;gBACL,OAAO,EAAE,KAAK;gBACd,sBAAsB,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;aACvD,CAAC,CAAA;YAEF,MAAM,eAAe,GAAc,EAAE,CAAA;YAErC,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;gBACxC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,iEAAiE;YACjE,MAAM,aAAa,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC,CAAA;YAClE,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,2BAA2B,EAAE,EAAE,CAAC,CAAA;QAChF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,uDAAuD,EAAE,GAAG,EAAE;QACrE,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YAClF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,IAAI,gBAAgD,CAAA;YAEpD,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,MAAM,UAAU,GAAuB;oBACrC,GAAG,OAAO;oBACV,QAAQ,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,EAAE,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,eAAe,CAAC,CAAC,EAAE,CAAC,CAAC;iBAC1G,CAAA;gBACD,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;YAChC,CAAC,CAAC,CAAA;YAEF,8EAA8E;YAC9E,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,gBAAgB,GAAG,OAAO,CAAC,QAAQ,CAAA;gBACnC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,gBAAgB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YACxC,MAAM,CAAE,gBAAiB,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,CAAC,CAAe,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QACpF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,IAAI,GAAG,cAAc,CACzB,QAAQ,EACR,GAAG,EAAE,CACH,IAAI,eAAe,CAAC;gBAClB,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;aAC/B,CAAC,CACL,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAEjE,IAAI,aAAsB,CAAA;YAE1B,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,uBAAuB;gBACvB,MAAM,UAAU,GAAuB;oBACrC,GAAG,OAAO;oBACV,SAAS,EAAE,EAAE;iBACd,CAAA;gBACD,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;YAChC,CAAC,CAAC,CAAA;YAEF,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,aAAa,GAAG,OAAO,CAAC,SAAS,CAAA;gBACjC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;YAE3B,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAA;QACnC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,SAAS,sBAAsB;QAC7B,OAAO,IAAI,gBAAgB,EAAE;aAC1B,OAAO,CAAC;YACP,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,QAAQ;YACnB,KAAK,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE;SACpD,CAAC;aACD,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;IACjD,CAAC;IAED,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,KAAK,GAAG,sBAAsB,EAAE,CAAA;YACtC,MAAM,IAAI,GAAG,cAAc,CACzB,UAAU,EACV,GAAG,EAAE,CACH,IAAI,eAAe,CAAC;gBAClB,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;aAC/B,CAAC,CACL,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAEjE,IAAI,aAAoC,CAAA;YAExC,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CAAA;gBACrC,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;YAEnC,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAC7E,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,KAAK,GAAG,sBAAsB,EAAE,CAAA;YACtC,MAAM,IAAI,GAAG,cAAc,CACzB,UAAU,EACV,GAAG,EAAE,CACH,IAAI,eAAe,CAAC;gBAClB,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;aAC/B,CAAC,CACL,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAEjE,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,CAAC;gBAAC,OAAO,CAAC,OAAO,CAAC,KAAiC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAA;gBACtE,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;YAEnC,4EAA4E;YAC5E,MAAM,YAAY,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAA;YACvE,MAAM,YAAY,GAAG,YAAY,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAA;YACjF,MAAM,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAA;YAClC,MAAM,CAAE,YAAqC,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,KAAK,GAAG,sBAAsB,EAAE,CAAA;YACtC,MAAM,IAAI,GAAG,cAAc,CACzB,UAAU,EACV,GAAG,EAAE,CACH,IAAI,eAAe,CAAC;gBAClB,SAAS,EAAE,QAAQ;gBACnB,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC;aAC/B,CAAC,CACL,CAAA;YACD,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAEjE,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,CAAC;gBAAC,OAAO,CAAC,eAA2C,CAAC,mBAAmB,CAAC,GAAG,IAAI,CAAA;gBACjF,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,eAAe,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC,CAAA;YAE3F,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAA;YAC/E,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAA;YAElD,MAAM,aAAa,GAAG,CAAC,IAAI,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;YACxF,IAAI,YAAqB,CAAA;YAEzB,KAAK,CAAC,aAAa,CAAC,gBAAgB,EAAE,KAAK,SAAS,CAAC,EAAE,OAAO,EAAE,IAAI;gBAClE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAA;gBAC3B,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YAC7B,CAAC,CAAC,CAAA;YAEF,MAAM,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAA;YAEjC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;QAC1C,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/middleware/stages.d.ts

@@ -2,7 +2,6 @@ import type { MiddlewareStage } from './types.js';
 import type { LocalAgent, AgentStreamEvent, InvocationState, InvokeArgs, InvokeOptions, AgentResult } from '../types/agent.js';
 import type { Message, SystemPrompt, ToolResultBlock } from '../types/messages.js';
 import type { ToolSpec, ToolChoice } from '../tools/types.js';
-import type { StateStore } from '../state-store.js';
 import type { StreamAggregatedResult } from '../models/model.js';
 import type { ToolUseData } from '../hooks/events.js';
 import type { Tool } from '../tools/tool.js';
@@ -65,10 +64,10 @@ export interface InvokeModelContext {
     readonly toolSpecs: readonly ToolSpec[];
     /** Controls how the model selects tools. */
     readonly toolChoice?: ToolChoice;
-    /** Runtime state for stateful model providers. */
-    readonly modelState: StateStore;
-    /** Per-invocation state shared across hooks and tools. */
+    /** Per-invocation state. Shared by reference — mutations are visible to hooks, tools, and AgentResult. */
     readonly invocationState: InvocationState;
+    /** Estimated input token count for this model call, or undefined if estimation failed. */
+    readonly projectedInputTokens?: number;
 }
 /**
  * Result from model-stage middleware.
@@ -89,7 +88,7 @@ export interface ExecuteToolContext extends MiddlewareInterruptible {
     readonly tool: Tool | undefined;
     /** The tool use request (name, toolUseId, input). */
     readonly toolUse: ToolUseData;
-    /** Per-invocation state shared across hooks and tools. */
+    /** Per-invocation state. Shared by reference — mutations are visible to hooks, tools, and AgentResult. */
     readonly invocationState: InvocationState;
 }
 /**
@@ -103,18 +102,23 @@ export interface ExecuteToolResult {
 /**
  * Context passed to agent-stream-stage middleware.
  * Wraps the entire agent output stream at the outermost interception point.
+ *
+ * @internal Not part of the public API. The contract for this context (particularly
+ * around copy vs. reference semantics for `args` and `options`) is not yet finalized.
  */
 export interface AgentStreamContext extends MiddlewareInterruptible {
     /** The agent instance (escape hatch for advanced use cases). */
     readonly agent: LocalAgent;
-    /** The invocation arguments passed to agent.stream(). */
+    /** The invocation arguments passed to agent.stream(). Shared by reference. */
     readonly args: InvokeArgs;
-    /** Per-invocation options (cancel signal, structured output, etc.). */
+    /** Per-invocation options (cancel signal, structured output, etc.). Shared by reference. */
     readonly options?: InvokeOptions;
 }
 /**
  * Result from agent-stream-stage middleware.
  * The return value of the async generator.
+ *
+ * @internal Not part of the public API.
  */
 export interface AgentStreamResult {
     /** The final agent result from the stream. */
@@ -133,6 +137,9 @@ export declare const ExecuteToolStage: MiddlewareStage<ExecuteToolContext, Execu
 /**
  * Built-in stage wrapping the entire agent output stream.
  * Middleware registered for this stage can filter, transform, or inject events.
+ *
+ * @internal Not part of the public API. The context contract for this stage is not
+ * yet finalized — particularly around copy vs. reference semantics for args/options.
  */
 export declare const AgentStreamStage: MiddlewareStage<AgentStreamContext, AgentStreamResult, AgentStreamEvent>;
 //# sourceMappingURL=stages.d.ts.map

package/dist/src/middleware/stages.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stages.d.ts","sourceRoot":"","sources":["../../../src/middleware/stages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AACjD,OAAO,KAAK,EACV,UAAU,EACV,gBAAgB,EAChB,eAAe,EACf,UAAU,EACV,aAAa,EACb,WAAW,EACZ,MAAM,mBAAmB,CAAA;AAC1B,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAA;AAClF,OAAO,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAC7D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AACnD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAA;AAChE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AACrD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AAC5C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAEjD;;;;GAIG;AACH,MAAM,WAAW,yBAAyB,CAAC,CAAC,GAAG,SAAS;IACtD;;;;;OAKG;IACH,QAAQ,EAAE,CAAC,CAAA;CACZ;AAED;;;;GAIG;AACH,MAAM,WAAW,uBAAuB;IACtC;;;;;;;;;;OAUG;IACH,SAAS,CAAC,CAAC,GAAG,SAAS,EAAE,MAAM,EAAE,eAAe,GAAG,yBAAyB,CAAC,CAAC,CAAC,CAAA;CAChF;AAED;;;;;;GAMG;AACH,wBAAgB,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAO/G;AAED;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IACjC,gEAAgE;IAChE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAA;IAC1B,yCAAyC;IACzC,QAAQ,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,CAAA;IACrC,mDAAmD;IACnD,QAAQ,CAAC,YAAY,CAAC,EAAE,YAAY,CAAA;IACpC,kDAAkD;IAClD,QAAQ,CAAC,SAAS,EAAE,SAAS,QAAQ,EAAE,CAAA;IACvC,4CAA4C;IAC5C,QAAQ,CAAC,UAAU,CAAC,EAAE,UAAU,CAAA;IAChC,kDAAkD;IAClD,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAA;IAC/B,0DAA0D;IAC1D,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;CAC1C;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,mDAAmD;IACnD,QAAQ,CAAC,MAAM,EAAE,sBAAsB,CAAA;CACxC;AAED;;;GAGG;AACH,MAAM,WAAW,kBAAmB,SAAQ,uBAAuB;IACjE,gEAAgE;IAChE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAA;IAC1B,mEAAmE;IACnE,QAAQ,CAAC,IAAI,EAAE,IAAI,GAAG,SAAS,CAAA;IAC/B,qDAAqD;IACrD,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAA;IAC7B,0DAA0D;IAC1D,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;CAC1C;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,4CAA4C;IAC5C,QAAQ,CAAC,MAAM,EAAE,eAAe,CAAA;CACjC;AAED;;;GAGG;AACH,MAAM,WAAW,kBAAmB,SAAQ,uBAAuB;IACjE,gEAAgE;IAChE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAA;IAC1B,yDAAyD;IACzD,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAA;IACzB,uEAAuE;IACvE,QAAQ,CAAC,OAAO,CAAC,EAAE,aAAa,CAAA;CACjC;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,8CAA8C;IAC9C,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAA;CAC7B;AAED;;;GAGG;AACH,eAAO,MAAM,gBAAgB,0EAAsF,CAAA;AAEnH;;;GAGG;AACH,eAAO,MAAM,gBAAgB,0EAAsF,CAAA;AAEnH;;;GAGG;AACH,eAAO,MAAM,gBAAgB,0EAAsF,CAAA"}
+{"version":3,"file":"stages.d.ts","sourceRoot":"","sources":["../../../src/middleware/stages.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AACjD,OAAO,KAAK,EACV,UAAU,EACV,gBAAgB,EAChB,eAAe,EACf,UAAU,EACV,aAAa,EACb,WAAW,EACZ,MAAM,mBAAmB,CAAA;AAC1B,OAAO,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAA;AAClF,OAAO,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAA;AAC7D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAA;AAChE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAA;AACrD,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAA;AAC5C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAA;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAEjD;;;;GAIG;AACH,MAAM,WAAW,yBAAyB,CAAC,CAAC,GAAG,SAAS;IACtD;;;;;OAKG;IACH,QAAQ,EAAE,CAAC,CAAA;CACZ;AAED;;;;GAIG;AACH,MAAM,WAAW,uBAAuB;IACtC;;;;;;;;;;OAUG;IACH,SAAS,CAAC,CAAC,GAAG,SAAS,EAAE,MAAM,EAAE,eAAe,GAAG,yBAAyB,CAAC,CAAC,CAAC,CAAA;CAChF;AAED;;;;;;GAMG;AACH,wBAAgB,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,CAO/G;AAED;;;;GAIG;AACH,MAAM,WAAW,kBAAkB;IACjC,gEAAgE;IAChE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAA;IAC1B,yCAAyC;IACzC,QAAQ,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,CAAA;IACrC,mDAAmD;IACnD,QAAQ,CAAC,YAAY,CAAC,EAAE,YAAY,CAAA;IACpC,kDAAkD;IAClD,QAAQ,CAAC,SAAS,EAAE,SAAS,QAAQ,EAAE,CAAA;IACvC,4CAA4C;IAC5C,QAAQ,CAAC,UAAU,CAAC,EAAE,UAAU,CAAA;IAChC,0GAA0G;IAC1G,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;IACzC,0FAA0F;IAC1F,QAAQ,CAAC,oBAAoB,CAAC,EAAE,MAAM,CAAA;CACvC;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,mDAAmD;IACnD,QAAQ,CAAC,MAAM,EAAE,sBAAsB,CAAA;CACxC;AAED;;;GAGG;AACH,MAAM,WAAW,kBAAmB,SAAQ,uBAAuB;IACjE,gEAAgE;IAChE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAA;IAC1B,mEAAmE;IACnE,QAAQ,CAAC,IAAI,EAAE,IAAI,GAAG,SAAS,CAAA;IAC/B,qDAAqD;IACrD,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAA;IAC7B,0GAA0G;IAC1G,QAAQ,CAAC,eAAe,EAAE,eAAe,CAAA;CAC1C;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC,4CAA4C;IAC5C,QAAQ,CAAC,MAAM,EAAE,eAAe,CAAA;CACjC;AAED;;;;;;GAMG;AACH,MAAM,WAAW,kBAAmB,SAAQ,uBAAuB;IACjE,gEAAgE;IAChE,QAAQ,CAAC,KAAK,EAAE,UAAU,CAAA;IAC1B,8EAA8E;IAC9E,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAA;IACzB,4FAA4F;IAC5F,QAAQ,CAAC,OAAO,CAAC,EAAE,aAAa,CAAA;CACjC;AAED;;;;;GAKG;AACH,MAAM,WAAW,iBAAiB;IAChC,8CAA8C;IAC9C,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAA;CAC7B;AAED;;;GAGG;AACH,eAAO,MAAM,gBAAgB,0EAAsF,CAAA;AAEnH;;;GAGG;AACH,eAAO,MAAM,gBAAgB,0EAAsF,CAAA;AAEnH;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB,0EAAsF,CAAA"}

package/dist/src/middleware/stages.js

@@ -26,6 +26,9 @@ export const ExecuteToolStage = createStage('executeTool');
 /**
  * Built-in stage wrapping the entire agent output stream.
  * Middleware registered for this stage can filter, transform, or inject events.
+ *
+ * @internal Not part of the public API. The context contract for this stage is not
+ * yet finalized — particularly around copy vs. reference semantics for args/options.
  */
 export const AgentStreamStage = createStage('agentStream');
 //# sourceMappingURL=stages.js.map

package/dist/src/middleware/stages.js.map

@@ -1 +1 @@
-{"version":3,"file":"stages.js","sourceRoot":"","sources":["../../../src/middleware/stages.ts"],"names":[],"mappings":"AAqDA;;;;;;GAMG;AACH,MAAM,UAAU,WAAW,CAA4B,IAAY;IACjE,wGAAwG;IACxG,MAAM,KAAK,GAAG,EAAE,IAAI,EAAgD,CAAA;IACpE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAgB,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;IACxE,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,MAAe,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;IACtE,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,QAAiB,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;IAC1E,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;AAC1F,CAAC;AA+ED;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAA0D,aAAa,CAAC,CAAA;AAEnH;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAA0D,aAAa,CAAC,CAAA;AAEnH;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAA0D,aAAa,CAAC,CAAA"}
+{"version":3,"file":"stages.js","sourceRoot":"","sources":["../../../src/middleware/stages.ts"],"names":[],"mappings":"AAoDA;;;;;;GAMG;AACH,MAAM,UAAU,WAAW,CAA4B,IAAY;IACjE,wGAAwG;IACxG,MAAM,KAAK,GAAG,EAAE,IAAI,EAAgD,CAAA;IACpE,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAgB,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;IACxE,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,MAAe,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;IACtE,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,QAAiB,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAA;IAC1E,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAA;AAC1F,CAAC;AAoFD;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAA0D,aAAa,CAAC,CAAA;AAEnH;;;GAGG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAA0D,aAAa,CAAC,CAAA;AAEnH;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,WAAW,CAA0D,aAAa,CAAC,CAAA"}

package/dist/src/models/__tests__/google.test.js

@@ -779,6 +779,47 @@ describe('GoogleModel', () => {
                 { inlineData: { data: 'iVBORw==', mimeType: 'image/png', displayName: 'image.png' } },
             ]);
         });
+        it('disambiguates displayName for multiple same-format images in one tool result', () => {
+            const toolUseBlock = new ToolUseBlock({ toolUseId: 'test-id', name: 'two_images', input: {} });
+            const toolResultBlock = new ToolResultBlock({
+                toolUseId: 'test-id',
+                status: 'success',
+                content: [
+                    new ImageBlock({ format: 'png', source: { bytes: new Uint8Array([0x89, 0x50, 0x4e, 0x47]) } }),
+                    new ImageBlock({ format: 'png', source: { bytes: new Uint8Array([0x89, 0x50, 0x4e, 0x48]) } }),
+                ],
+            });
+            const messages = [
+                new Message({ role: 'assistant', content: [toolUseBlock] }),
+                new Message({ role: 'user', content: [toolResultBlock] }),
+            ];
+            const contents = formatMessages(messages);
+            const resultPart = contents[1].parts[0];
+            const displayNames = resultPart.functionResponse.parts.map((p) => p.inlineData.displayName);
+            // Gemini rejects duplicate part names within one function_response.parts.
+            expect(new Set(displayNames).size).toBe(displayNames.length);
+            expect(displayNames).toEqual(['image.png', 'image-1.png']);
+        });
+        it('disambiguates displayName for documents sharing a name in one tool result', () => {
+            const toolUseBlock = new ToolUseBlock({ toolUseId: 'test-id', name: 'two_docs', input: {} });
+            const toolResultBlock = new ToolResultBlock({
+                toolUseId: 'test-id',
+                status: 'success',
+                content: [
+                    new DocumentBlock({ name: 'report.pdf', format: 'pdf', source: { bytes: new Uint8Array([0x25, 0x50]) } }),
+                    new DocumentBlock({ name: 'report.pdf', format: 'pdf', source: { bytes: new Uint8Array([0x25, 0x51]) } }),
+                ],
+            });
+            const messages = [
+                new Message({ role: 'assistant', content: [toolUseBlock] }),
+                new Message({ role: 'user', content: [toolResultBlock] }),
+            ];
+            const contents = formatMessages(messages);
+            const resultPart = contents[1].parts[0];
+            const displayNames = resultPart.functionResponse.parts.map((p) => p.inlineData.displayName);
+            expect(new Set(displayNames).size).toBe(displayNames.length);
+            expect(displayNames).toEqual(['report.pdf', 'report-1.pdf']);
+        });
         it('formats document block with bytes source in tool result as inlineData', () => {
             const docBytes = new Uint8Array([0x25, 0x50, 0x44, 0x46]);
             const toolUseBlock = new ToolUseBlock({ toolUseId: 'test-id', name: 'read_doc', input: {} });