@illuma-ai/agents 1.0.90 → 1.0.94

package/src/specs/cache.simple.test.js

@@ -0,0 +1,282 @@
+/* eslint-disable no-console */
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import { config } from 'dotenv';
+config();
+import { Calculator } from '@/tools/Calculator';
+import { HumanMessage, } from '@langchain/core/messages';
+import { ModelEndHandler, ToolEndHandler } from '@/events';
+import { capitalizeFirstLetter } from './spec.utils';
+import { createContentAggregator } from '@/stream';
+import { GraphEvents, Providers } from '@/common';
+import { getLLMConfig } from '@/utils/llmConfig';
+import { getArgs } from '@/scripts/args';
+import { Run } from '@/run';
+/**
+ * These tests verify that prompt caching works correctly across multi-turn
+ * conversations and that messages are not mutated in place.
+ */
+describe('Prompt Caching Integration Tests', () => {
+    jest.setTimeout(120000);
+    const setupTest = () => {
+        const collectedUsage = [];
+        const { contentParts, aggregateContent } = createContentAggregator();
+        const customHandlers = {
+            [GraphEvents.TOOL_END]: new ToolEndHandler(),
+            [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
+            [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+                handle: (event, data) => {
+                    aggregateContent({
+                        event,
+                        data: data,
+                    });
+                },
+            },
+            [GraphEvents.ON_RUN_STEP]: {
+                handle: (event, data) => {
+                    aggregateContent({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_RUN_STEP_DELTA]: {
+                handle: (event, data) => {
+                    aggregateContent({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_MESSAGE_DELTA]: {
+                handle: (event, data) => {
+                    aggregateContent({ event, data: data });
+                },
+            },
+        };
+        return { collectedUsage, contentParts, customHandlers };
+    };
+    const streamConfig = {
+        configurable: { thread_id: 'cache-test-thread' },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    describe('Anthropic Prompt Caching', () => {
+        const provider = Providers.ANTHROPIC;
+        test(`${capitalizeFirstLetter(provider)}: multi-turn conversation with caching should not corrupt messages`, async () => {
+            const { userName, location } = await getArgs();
+            const llmConfig = getLLMConfig(provider);
+            const { collectedUsage, customHandlers } = setupTest();
+            const run = await Run.create({
+                runId: 'cache-test-anthropic',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...llmConfig, promptCache: true },
+                    tools: [new Calculator()],
+                    instructions: 'You are a helpful assistant.',
+                    additional_instructions: `User: ${userName}, Location: ${location}`,
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers,
+            });
+            // Turn 1
+            const turn1Messages = [
+                new HumanMessage('Hello, what is 2+2?'),
+            ];
+            const turn1ContentSnapshot = JSON.stringify(turn1Messages[0].content);
+            const turn1Result = await run.processStream({ messages: turn1Messages }, streamConfig);
+            expect(turn1Result).toBeDefined();
+            // Verify original message was NOT mutated
+            expect(JSON.stringify(turn1Messages[0].content)).toBe(turn1ContentSnapshot);
+            expect(turn1Messages[0].content).not.toContain('cache_control');
+            const turn1RunMessages = run.getRunMessages();
+            expect(turn1RunMessages).toBeDefined();
+            expect(turn1RunMessages.length).toBeGreaterThan(0);
+            // Turn 2 - build on conversation
+            const turn2Messages = [
+                ...turn1Messages,
+                ...turn1RunMessages,
+                new HumanMessage('Now multiply that by 10'),
+            ];
+            const turn2HumanContentSnapshot = JSON.stringify(turn2Messages[turn2Messages.length - 1].content);
+            const run2 = await Run.create({
+                runId: 'cache-test-anthropic-2',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...llmConfig, promptCache: true },
+                    tools: [new Calculator()],
+                    instructions: 'You are a helpful assistant.',
+                    additional_instructions: `User: ${userName}, Location: ${location}`,
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers,
+            });
+            const turn2Result = await run2.processStream({ messages: turn2Messages }, streamConfig);
+            expect(turn2Result).toBeDefined();
+            // Verify messages were NOT mutated
+            expect(JSON.stringify(turn2Messages[turn2Messages.length - 1].content)).toBe(turn2HumanContentSnapshot);
+            // Check that we got cache read tokens (indicating caching worked)
+            console.log(`${provider} Usage:`, collectedUsage);
+            expect(collectedUsage.length).toBeGreaterThan(0);
+            console.log(`${capitalizeFirstLetter(provider)} multi-turn caching test passed - messages not mutated`);
+        });
+        test(`${capitalizeFirstLetter(provider)}: tool calls should work with caching enabled`, async () => {
+            const llmConfig = getLLMConfig(provider);
+            const { customHandlers } = setupTest();
+            const run = await Run.create({
+                runId: 'cache-test-anthropic-tools',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...llmConfig, promptCache: true },
+                    tools: [new Calculator()],
+                    instructions: 'You are a math assistant. Use the calculator tool for all calculations.',
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers,
+            });
+            const messages = [
+                new HumanMessage('Calculate 123 * 456 using the calculator'),
+            ];
+            const result = await run.processStream({ messages }, streamConfig);
+            expect(result).toBeDefined();
+            const runMessages = run.getRunMessages();
+            expect(runMessages).toBeDefined();
+            // Should have used the calculator tool
+            const hasToolUse = runMessages?.some((msg) => msg._getType() === 'ai' &&
+                (msg.tool_calls?.length ?? 0) > 0);
+            expect(hasToolUse).toBe(true);
+            console.log(`${capitalizeFirstLetter(provider)} tool call with caching test passed`);
+        });
+    });
+    describe('Bedrock Prompt Caching', () => {
+        const provider = Providers.BEDROCK;
+        test(`${capitalizeFirstLetter(provider)}: multi-turn conversation with caching should not corrupt messages`, async () => {
+            const { userName, location } = await getArgs();
+            const llmConfig = getLLMConfig(provider);
+            const { collectedUsage, customHandlers } = setupTest();
+            const run = await Run.create({
+                runId: 'cache-test-bedrock',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...llmConfig, promptCache: true },
+                    tools: [new Calculator()],
+                    instructions: 'You are a helpful assistant.',
+                    additional_instructions: `User: ${userName}, Location: ${location}`,
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers,
+            });
+            // Turn 1
+            const turn1Messages = [
+                new HumanMessage('Hello, what is 5+5?'),
+            ];
+            const turn1ContentSnapshot = JSON.stringify(turn1Messages[0].content);
+            const turn1Result = await run.processStream({ messages: turn1Messages }, streamConfig);
+            expect(turn1Result).toBeDefined();
+            // Verify original message was NOT mutated
+            expect(JSON.stringify(turn1Messages[0].content)).toBe(turn1ContentSnapshot);
+            const turn1RunMessages = run.getRunMessages();
+            expect(turn1RunMessages).toBeDefined();
+            expect(turn1RunMessages.length).toBeGreaterThan(0);
+            // Turn 2
+            const turn2Messages = [
+                ...turn1Messages,
+                ...turn1RunMessages,
+                new HumanMessage('Multiply that by 3'),
+            ];
+            const turn2HumanContentSnapshot = JSON.stringify(turn2Messages[turn2Messages.length - 1].content);
+            const run2 = await Run.create({
+                runId: 'cache-test-bedrock-2',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...llmConfig, promptCache: true },
+                    tools: [new Calculator()],
+                    instructions: 'You are a helpful assistant.',
+                    additional_instructions: `User: ${userName}, Location: ${location}`,
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers,
+            });
+            const turn2Result = await run2.processStream({ messages: turn2Messages }, streamConfig);
+            expect(turn2Result).toBeDefined();
+            // Verify messages were NOT mutated
+            expect(JSON.stringify(turn2Messages[turn2Messages.length - 1].content)).toBe(turn2HumanContentSnapshot);
+            console.log(`${provider} Usage:`, collectedUsage);
+            expect(collectedUsage.length).toBeGreaterThan(0);
+            console.log(`${capitalizeFirstLetter(provider)} multi-turn caching test passed - messages not mutated`);
+        });
+        test(`${capitalizeFirstLetter(provider)}: tool calls should work with caching enabled`, async () => {
+            const llmConfig = getLLMConfig(provider);
+            const { customHandlers } = setupTest();
+            const run = await Run.create({
+                runId: 'cache-test-bedrock-tools',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...llmConfig, promptCache: true },
+                    tools: [new Calculator()],
+                    instructions: 'You are a math assistant. Use the calculator tool for all calculations.',
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers,
+            });
+            const messages = [
+                new HumanMessage('Calculate 789 * 123 using the calculator'),
+            ];
+            const result = await run.processStream({ messages }, streamConfig);
+            expect(result).toBeDefined();
+            const runMessages = run.getRunMessages();
+            expect(runMessages).toBeDefined();
+            // Should have used the calculator tool
+            const hasToolUse = runMessages?.some((msg) => msg._getType() === 'ai' &&
+                (msg.tool_calls?.length ?? 0) > 0);
+            expect(hasToolUse).toBe(true);
+            console.log(`${capitalizeFirstLetter(provider)} tool call with caching test passed`);
+        });
+    });
+    describe('Cross-provider message isolation', () => {
+        test('Messages processed by Anthropic should not affect Bedrock processing', async () => {
+            const anthropicConfig = getLLMConfig(Providers.ANTHROPIC);
+            const bedrockConfig = getLLMConfig(Providers.BEDROCK);
+            const { customHandlers: handlers1 } = setupTest();
+            const { customHandlers: handlers2 } = setupTest();
+            // Create a shared message array
+            const sharedMessages = [
+                new HumanMessage('Hello, what is the capital of France?'),
+            ];
+            const originalContent = JSON.stringify(sharedMessages[0].content);
+            // Process with Anthropic first
+            const anthropicRun = await Run.create({
+                runId: 'cross-provider-anthropic',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...anthropicConfig, promptCache: true },
+                    instructions: 'You are a helpful assistant.',
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers: handlers1,
+            });
+            const anthropicResult = await anthropicRun.processStream({ messages: sharedMessages }, streamConfig);
+            expect(anthropicResult).toBeDefined();
+            // Verify message not mutated
+            expect(JSON.stringify(sharedMessages[0].content)).toBe(originalContent);
+            // Now process with Bedrock using the SAME messages
+            const bedrockRun = await Run.create({
+                runId: 'cross-provider-bedrock',
+                graphConfig: {
+                    type: 'standard',
+                    llmConfig: { ...bedrockConfig, promptCache: true },
+                    instructions: 'You are a helpful assistant.',
+                },
+                returnContent: true,
+                skipCleanup: true,
+                customHandlers: handlers2,
+            });
+            const bedrockResult = await bedrockRun.processStream({ messages: sharedMessages }, streamConfig);
+            expect(bedrockResult).toBeDefined();
+            // Verify message STILL not mutated after both providers processed
+            expect(JSON.stringify(sharedMessages[0].content)).toBe(originalContent);
+            console.log('Cross-provider message isolation test passed');
+        });
+    });
+});
+//# sourceMappingURL=cache.simple.test.js.map

package/src/specs/cache.simple.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.simple.test.js","sourceRoot":"","sources":["cache.simple.test.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,uDAAuD;AACvD,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,MAAM,EAAE,CAAC;AACT,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAGL,YAAY,GAEb,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AACrD,OAAO,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AACnD,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B;;;GAGG;AACH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAExB,MAAM,SAAS,GAAG,GAIhB,EAAE;QACF,MAAM,cAAc,GAAoB,EAAE,CAAC;QAC3C,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;QAErE,MAAM,cAAc,GAAiD;YACnE,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;YAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,CAAC,cAAc,CAAC;YACjE,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;gBACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACjB,EAAE;oBACR,gBAAgB,CAAC;wBACf,KAAK;wBACL,IAAI,EAAE,IAA6C;qBACpD,CAAC,CAAC;gBACL,CAAC;aACF;YACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBACzB,MAAM,EAAE,CACN,KAA8B,EAC9B,IAAuB,EACjB,EAAE;oBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;gBACvD,CAAC;aACF;YACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;gBAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACjB,EAAE;oBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;gBACjE,CAAC;aACF;YACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;gBAC9B,MAAM,EAAE,CACN,KAAmC,EACnC,IAAuB,EACjB,EAAE;oBACR,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;gBACjE,CAAC;aACF;SACF,CAAC;QAEF,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,cAAc,EAAE,CAAC;IAC1D,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG;QACnB,YAAY,EAAE,EAAE,SAAS,EAAE,mBAAmB,EAAE;QAChD,UAAU,EAAE,QAAQ;QACpB,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,MAAM,QAAQ,GAAG,SAAS,CAAC,SAAS,CAAC;QAErC,IAAI,CAAC,GAAG,qBAAqB,CAAC,QAAQ,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YACtH,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,SAAS,EAAE,CAAC;YAEvD,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBACrC,KAAK,EAAE,sBAAsB;gBAC7B,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,IAAI,EAAiB;oBAC7D,KAAK,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;oBACzB,YAAY,EAAE,8BAA8B;oBAC5C,uBAAuB,EAAE,SAAS,QAAQ,eAAe,QAAQ,EAAE;iBACpE;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc;aACf,CAAC,CAAC;YAEH,SAAS;YACT,MAAM,aAAa,GAAkB;gBACnC,IAAI,YAAY,CAAC,qBAAqB,CAAC;aACxC,CAAC;YACF,MAAM,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAEtE,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CACzC,EAAE,QAAQ,EAAE,aAAa,EAAE,EAC3B,YAAY,CACb,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAElC,0CAA0C;YAC1C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACnD,oBAAoB,CACrB,CAAC;YACF,MAAM,CAAE,aAAa,CAAC,CAAC,CAAS,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YAEzE,MAAM,gBAAgB,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YAC9C,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,CAAC,gBAAiB,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEpD,iCAAiC;YACjC,MAAM,aAAa,GAAkB;gBACnC,GAAG,aAAa;gBAChB,GAAG,gBAAiB;gBACpB,IAAI,YAAY,CAAC,yBAAyB,CAAC;aAC5C,CAAC;YACF,MAAM,yBAAyB,GAAG,IAAI,CAAC,SAAS,CAC9C,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,CAChD,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBACtC,KAAK,EAAE,wBAAwB;gBAC/B,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,IAAI,EAAiB;oBAC7D,KAAK,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;oBACzB,YAAY,EAAE,8BAA8B;oBAC5C,uBAAuB,EAAE,SAAS,QAAQ,eAAe,QAAQ,EAAE;iBACpE;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,aAAa,CAC1C,EAAE,QAAQ,EAAE,aAAa,EAAE,EAC3B,YAAY,CACb,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAElC,mCAAmC;YACnC,MAAM,CACJ,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAChE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAElC,kEAAkE;YAClE,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,SAAS,EAAE,cAAc,CAAC,CAAC;YAClD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEjD,OAAO,CAAC,GAAG,CACT,GAAG,qBAAqB,CAAC,QAAQ,CAAC,wDAAwD,CAC3F,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,qBAAqB,CAAC,QAAQ,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YACjG,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,EAAE,cAAc,EAAE,GAAG,SAAS,EAAE,CAAC;YAEvC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBACrC,KAAK,EAAE,4BAA4B;gBACnC,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,IAAI,EAAiB;oBAC7D,KAAK,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;oBACzB,YAAY,EACV,yEAAyE;iBAC5E;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAkB;gBAC9B,IAAI,YAAY,CAAC,0CAA0C,CAAC;aAC7D,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,YAAY,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,MAAM,WAAW,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YACzC,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAElC,uCAAuC;YACvC,MAAM,UAAU,GAAG,WAAW,EAAE,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,QAAQ,EAAE,KAAK,IAAI;gBACvB,CAAE,GAAiB,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CACnD,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE9B,OAAO,CAAC,GAAG,CACT,GAAG,qBAAqB,CAAC,QAAQ,CAAC,qCAAqC,CACxE,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,CAAC;QAEnC,IAAI,CAAC,GAAG,qBAAqB,CAAC,QAAQ,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YACtH,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,MAAM,OAAO,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,SAAS,EAAE,CAAC;YAEvD,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBACrC,KAAK,EAAE,oBAAoB;gBAC3B,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,IAAI,EAAiB;oBAC7D,KAAK,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;oBACzB,YAAY,EAAE,8BAA8B;oBAC5C,uBAAuB,EAAE,SAAS,QAAQ,eAAe,QAAQ,EAAE;iBACpE;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc;aACf,CAAC,CAAC;YAEH,SAAS;YACT,MAAM,aAAa,GAAkB;gBACnC,IAAI,YAAY,CAAC,qBAAqB,CAAC;aACxC,CAAC;YACF,MAAM,oBAAoB,GAAG,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAEtE,MAAM,WAAW,GAAG,MAAM,GAAG,CAAC,aAAa,CACzC,EAAE,QAAQ,EAAE,aAAa,EAAE,EAC3B,YAAY,CACb,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAElC,0CAA0C;YAC1C,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CACnD,oBAAoB,CACrB,CAAC;YAEF,MAAM,gBAAgB,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YAC9C,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,CAAC,gBAAiB,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEpD,SAAS;YACT,MAAM,aAAa,GAAkB;gBACnC,GAAG,aAAa;gBAChB,GAAG,gBAAiB;gBACpB,IAAI,YAAY,CAAC,oBAAoB,CAAC;aACvC,CAAC;YACF,MAAM,yBAAyB,GAAG,IAAI,CAAC,SAAS,CAC9C,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,CAChD,CAAC;YAEF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBACtC,KAAK,EAAE,sBAAsB;gBAC7B,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,IAAI,EAAiB;oBAC7D,KAAK,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;oBACzB,YAAY,EAAE,8BAA8B;oBAC5C,uBAAuB,EAAE,SAAS,QAAQ,eAAe,QAAQ,EAAE;iBACpE;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,aAAa,CAC1C,EAAE,QAAQ,EAAE,aAAa,EAAE,EAC3B,YAAY,CACb,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAElC,mCAAmC;YACnC,MAAM,CACJ,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAChE,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YAElC,OAAO,CAAC,GAAG,CAAC,GAAG,QAAQ,SAAS,EAAE,cAAc,CAAC,CAAC;YAClD,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEjD,OAAO,CAAC,GAAG,CACT,GAAG,qBAAqB,CAAC,QAAQ,CAAC,wDAAwD,CAC3F,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,qBAAqB,CAAC,QAAQ,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YACjG,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACzC,MAAM,EAAE,cAAc,EAAE,GAAG,SAAS,EAAE,CAAC;YAEvC,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBACrC,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,SAAS,EAAE,WAAW,EAAE,IAAI,EAAiB;oBAC7D,KAAK,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC;oBACzB,YAAY,EACV,yEAAyE;iBAC5E;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAkB;gBAC9B,IAAI,YAAY,CAAC,0CAA0C,CAAC;aAC7D,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,EAAE,YAAY,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YAE7B,MAAM,WAAW,GAAG,GAAG,CAAC,cAAc,EAAE,CAAC;YACzC,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAElC,uCAAuC;YACvC,MAAM,UAAU,GAAG,WAAW,EAAE,IAAI,CAClC,CAAC,GAAG,EAAE,EAAE,CACN,GAAG,CAAC,QAAQ,EAAE,KAAK,IAAI;gBACvB,CAAE,GAAiB,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CACnD,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE9B,OAAO,CAAC,GAAG,CACT,GAAG,qBAAqB,CAAC,QAAQ,CAAC,qCAAqC,CACxE,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,IAAI,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,eAAe,GAAG,YAAY,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC1D,MAAM,aAAa,GAAG,YAAY,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAAC;YAClD,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAAC;YAElD,gCAAgC;YAChC,MAAM,cAAc,GAAkB;gBACpC,IAAI,YAAY,CAAC,uCAAuC,CAAC;aAC1D,CAAC;YACF,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAElE,+BAA+B;YAC/B,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBAC9C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,eAAe,EAAE,WAAW,EAAE,IAAI,EAAiB;oBACnE,YAAY,EAAE,8BAA8B;iBAC7C;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,SAAS;aAC1B,CAAC,CAAC;YAEH,MAAM,eAAe,GAAG,MAAM,YAAY,CAAC,aAAa,CACtD,EAAE,QAAQ,EAAE,cAAc,EAAE,EAC5B,YAAY,CACb,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;YAEtC,6BAA6B;YAC7B,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAExE,mDAAmD;YACnD,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;gBAC5C,KAAK,EAAE,wBAAwB;gBAC/B,WAAW,EAAE;oBACX,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,EAAE,GAAG,aAAa,EAAE,WAAW,EAAE,IAAI,EAAiB;oBACjE,YAAY,EAAE,8BAA8B;iBAC7C;gBACD,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,SAAS;aAC1B,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,MAAM,UAAU,CAAC,aAAa,CAClD,EAAE,QAAQ,EAAE,cAAc,EAAE,EAC5B,YAAY,CACb,CAAC;YACF,MAAM,CAAC,aAAa,CAAC,CAAC,WAAW,EAAE,CAAC;YAEpC,kEAAkE;YAClE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAExE,OAAO,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/specs/cache.simple.test.ts

@@ -10,9 +10,9 @@ import {
   UsageMetadata,
 } from '@langchain/core/messages';
 import type * as t from '@/types';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
 import { ModelEndHandler, ToolEndHandler } from '@/events';
 import { capitalizeFirstLetter } from './spec.utils';
+import { createContentAggregator } from '@/stream';
 import { GraphEvents, Providers } from '@/common';
 import { getLLMConfig } from '@/utils/llmConfig';
 import { getArgs } from '@/scripts/args';
@@ -36,7 +36,6 @@ describe('Prompt Caching Integration Tests', () => {
     const customHandlers: Record<string | GraphEvents, t.EventHandler> = {
       [GraphEvents.TOOL_END]: new ToolEndHandler(),
       [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
-      [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
       [GraphEvents.ON_RUN_STEP_COMPLETED]: {
         handle: (
           event: GraphEvents.ON_RUN_STEP_COMPLETED,
@@ -101,6 +100,7 @@ describe('Prompt Caching Integration Tests', () => {
           additional_instructions: `User: ${userName}, Location: ${location}`,
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers,
       });
 
@@ -146,6 +146,7 @@ describe('Prompt Caching Integration Tests', () => {
           additional_instructions: `User: ${userName}, Location: ${location}`,
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers,
       });
 
@@ -183,6 +184,7 @@ describe('Prompt Caching Integration Tests', () => {
             'You are a math assistant. Use the calculator tool for all calculations.',
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers,
       });
 
@@ -228,6 +230,7 @@ describe('Prompt Caching Integration Tests', () => {
           additional_instructions: `User: ${userName}, Location: ${location}`,
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers,
       });
 
@@ -272,6 +275,7 @@ describe('Prompt Caching Integration Tests', () => {
           additional_instructions: `User: ${userName}, Location: ${location}`,
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers,
       });
 
@@ -308,6 +312,7 @@ describe('Prompt Caching Integration Tests', () => {
             'You are a math assistant. Use the calculator tool for all calculations.',
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers,
       });
 
@@ -357,6 +362,7 @@ describe('Prompt Caching Integration Tests', () => {
           instructions: 'You are a helpful assistant.',
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers: handlers1,
       });
 
@@ -378,6 +384,7 @@ describe('Prompt Caching Integration Tests', () => {
           instructions: 'You are a helpful assistant.',
         },
         returnContent: true,
+        skipCleanup: true,
         customHandlers: handlers2,
       });
 

package/src/specs/custom-event-await.test.js

@@ -0,0 +1,148 @@
+import { HumanMessage } from '@langchain/core/messages';
+import { ContentTypes, GraphEvents, Providers } from '@/common';
+import { ToolEndHandler, ModelEndHandler } from '@/events';
+import { createContentAggregator } from '@/stream';
+import { Run } from '@/run';
+describe('Custom event handler awaitHandlers behavior', () => {
+    jest.setTimeout(15000);
+    const llmConfig = {
+        provider: Providers.OPENAI,
+        streaming: true,
+        streamUsage: false,
+    };
+    const config = {
+        configurable: {
+            thread_id: 'test-thread',
+        },
+        streamMode: 'values',
+        version: 'v2',
+    };
+    it('should fully aggregate all content before processStream returns', async () => {
+        const longResponse = 'The quick brown fox jumps over the lazy dog and then runs across the field to find shelter from the rain';
+        let aggregateCallCount = 0;
+        const { contentParts, aggregateContent } = createContentAggregator();
+        const wrappedAggregate = (params) => {
+            aggregateCallCount++;
+            aggregateContent(params);
+        };
+        let messageDeltaCount = 0;
+        const customHandlers = {
+            [GraphEvents.TOOL_END]: new ToolEndHandler(),
+            [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+            [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+                handle: (event, data) => {
+                    wrappedAggregate({
+                        event,
+                        data: data,
+                    });
+                },
+            },
+            [GraphEvents.ON_RUN_STEP]: {
+                handle: (event, data) => {
+                    wrappedAggregate({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_RUN_STEP_DELTA]: {
+                handle: (event, data) => {
+                    wrappedAggregate({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_MESSAGE_DELTA]: {
+                handle: async (event, data) => {
+                    messageDeltaCount++;
+                    wrappedAggregate({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_REASONING_DELTA]: {
+                handle: (event, data) => {
+                    wrappedAggregate({ event, data: data });
+                },
+            },
+        };
+        const run = await Run.create({
+            runId: 'test-await-handlers',
+            graphConfig: {
+                type: 'standard',
+                llmConfig,
+            },
+            returnContent: true,
+            skipCleanup: true,
+            customHandlers,
+        });
+        run.Graph.overrideTestModel([longResponse]);
+        const inputs = { messages: [new HumanMessage('hello')] };
+        const finalContentParts = await run.processStream(inputs, config);
+        expect(finalContentParts).toBeDefined();
+        expect(finalContentParts.length).toBeGreaterThan(0);
+        expect(messageDeltaCount).toBeGreaterThan(0);
+        expect(aggregateCallCount).toBeGreaterThan(0);
+        const typedParts = contentParts;
+        const textParts = typedParts.filter((p) => p !== undefined && p.type === ContentTypes.TEXT);
+        expect(textParts.length).toBeGreaterThan(0);
+        const aggregatedText = textParts
+            .map((p) => p[ContentTypes.TEXT])
+            .join('');
+        expect(aggregatedText).toBe(longResponse);
+    });
+    it('should aggregate content from async handlers before processStream returns', async () => {
+        const response = 'This is a test of async handler aggregation with multiple tokens';
+        const { contentParts, aggregateContent } = createContentAggregator();
+        let asyncHandlerCompletions = 0;
+        const customHandlers = {
+            [GraphEvents.TOOL_END]: new ToolEndHandler(),
+            [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
+            [GraphEvents.ON_RUN_STEP_COMPLETED]: {
+                handle: (event, data) => {
+                    aggregateContent({
+                        event,
+                        data: data,
+                    });
+                },
+            },
+            [GraphEvents.ON_RUN_STEP]: {
+                handle: (event, data) => {
+                    aggregateContent({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_RUN_STEP_DELTA]: {
+                handle: (event, data) => {
+                    aggregateContent({ event, data: data });
+                },
+            },
+            [GraphEvents.ON_MESSAGE_DELTA]: {
+                handle: async (event, data) => {
+                    await new Promise((resolve) => setTimeout(resolve, 5));
+                    aggregateContent({ event, data: data });
+                    asyncHandlerCompletions++;
+                },
+            },
+            [GraphEvents.ON_REASONING_DELTA]: {
+                handle: (event, data) => {
+                    aggregateContent({ event, data: data });
+                },
+            },
+        };
+        const run = await Run.create({
+            runId: 'test-async-handlers',
+            graphConfig: {
+                type: 'standard',
+                llmConfig,
+            },
+            returnContent: true,
+            skipCleanup: true,
+            customHandlers,
+        });
+        run.Graph.overrideTestModel([response]);
+        const inputs = { messages: [new HumanMessage('hello')] };
+        await run.processStream(inputs, config);
+        expect(asyncHandlerCompletions).toBeGreaterThan(0);
+        const typedParts = contentParts;
+        const textParts = typedParts.filter((p) => p !== undefined && p.type === ContentTypes.TEXT);
+        expect(textParts.length).toBeGreaterThan(0);
+        const aggregatedText = textParts
+            .map((p) => p[ContentTypes.TEXT])
+            .join('');
+        expect(aggregatedText).toBe(response);
+    });
+});
+//# sourceMappingURL=custom-event-await.test.js.map

package/src/specs/custom-event-await.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom-event-await.test.js","sourceRoot":"","sources":["custom-event-await.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAChE,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC3D,OAAO,EAAE,uBAAuB,EAAE,MAAM,UAAU,CAAC;AACnD,OAAO,EAAE,GAAG,EAAE,MAAM,OAAO,CAAC;AAE5B,QAAQ,CAAC,6CAA6C,EAAE,GAAG,EAAE;IAC3D,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAEvB,MAAM,SAAS,GAAgB;QAC7B,QAAQ,EAAE,SAAS,CAAC,MAAM;QAC1B,SAAS,EAAE,IAAI;QACf,WAAW,EAAE,KAAK;KACnB,CAAC;IAEF,MAAM,MAAM,GAAG;QACb,YAAY,EAAE;YACZ,SAAS,EAAE,aAAa;SACzB;QACD,UAAU,EAAE,QAAiB;QAC7B,OAAO,EAAE,IAAa;KACvB,CAAC;IAEF,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,YAAY,GAChB,0GAA0G,CAAC;QAE7G,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAC3B,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;QAErE,MAAM,gBAAgB,GAAwB,CAAC,MAAM,EAAE,EAAE;YACvD,kBAAkB,EAAE,CAAC;YACrB,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC,CAAC;QAEF,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAE1B,MAAM,cAAc,GAAiD;YACnE,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;YAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;YACnD,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;gBACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACvB,EAAE;oBACF,gBAAgB,CAAC;wBACf,KAAK;wBACL,IAAI,EAAE,IAA6C;qBACpD,CAAC,CAAC;gBACL,CAAC;aACF;YACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBACzB,MAAM,EAAE,CAAC,KAA8B,EAAE,IAAuB,EAAE,EAAE;oBAClE,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;gBACvD,CAAC;aACF;YACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;gBAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACvB,EAAE;oBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;gBACjE,CAAC;aACF;YACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;gBAC9B,MAAM,EAAE,KAAK,EACX,KAAmC,EACnC,IAAuB,EACvB,EAAE;oBACF,iBAAiB,EAAE,CAAC;oBACpB,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;gBACjE,CAAC;aACF;YACD,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE;gBAChC,MAAM,EAAE,CACN,KAAqC,EACrC,IAAuB,EACvB,EAAE;oBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA6B,EAAE,CAAC,CAAC;gBACnE,CAAC;aACF;SACF,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;YACrC,KAAK,EAAE,qBAAqB;YAC5B,WAAW,EAAE;gBACX,IAAI,EAAE,UAAU;gBAChB,SAAS;aACV;YACD,aAAa,EAAE,IAAI;YACnB,WAAW,EAAE,IAAI;YACjB,cAAc;SACf,CAAC,CAAC;QAEH,GAAG,CAAC,KAAM,CAAC,iBAAiB,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QAE7C,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;QACzD,MAAM,iBAAiB,GAAG,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAElE,MAAM,CAAC,iBAAiB,CAAC,CAAC,WAAW,EAAE,CAAC;QACxC,MAAM,CAAC,iBAAkB,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAErD,MAAM,CAAC,iBAAiB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7C,MAAM,CAAC,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAE9C,MAAM,UAAU,GAAG,YAAyC,CAAC;QAC7D,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CACjC,CAAC,CAAsC,EAAE,EAAE,CACzC,CAAC,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,CAClD,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAE5C,MAAM,cAAc,GAAG,SAAS;aAC7B,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACH,CAAmD,CAClD,YAAY,CAAC,IAAI,CAClB,CACJ;aACA,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,QAAQ,GACZ,kEAAkE,CAAC;QAErE,MAAM,EAAE,YAAY,EAAE,gBAAgB,EAAE,GAAG,uBAAuB,EAAE,CAAC;QAErE,IAAI,uBAAuB,GAAG,CAAC,CAAC;QAEhC,MAAM,cAAc,GAAiD;YACnE,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,IAAI,cAAc,EAAE;YAC5C,CAAC,WAAW,CAAC,cAAc,CAAC,EAAE,IAAI,eAAe,EAAE;YACnD,CAAC,WAAW,CAAC,qBAAqB,CAAC,EAAE;gBACnC,MAAM,EAAE,CACN,KAAwC,EACxC,IAAuB,EACvB,EAAE;oBACF,gBAAgB,CAAC;wBACf,KAAK;wBACL,IAAI,EAAE,IAA6C;qBACpD,CAAC,CAAC;gBACL,CAAC;aACF;YACD,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBACzB,MAAM,EAAE,CAAC,KAA8B,EAAE,IAAuB,EAAE,EAAE;oBAClE,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAAiB,EAAE,CAAC,CAAC;gBACvD,CAAC;aACF;YACD,CAAC,WAAW,CAAC,iBAAiB,CAAC,EAAE;gBAC/B,MAAM,EAAE,CACN,KAAoC,EACpC,IAAuB,EACvB,EAAE;oBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;gBACjE,CAAC;aACF;YACD,CAAC,WAAW,CAAC,gBAAgB,CAAC,EAAE;gBAC9B,MAAM,EAAE,KAAK,EACX,KAAmC,EACnC,IAAuB,EACvB,EAAE;oBACF,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;oBAC7D,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA2B,EAAE,CAAC,CAAC;oBAC/D,uBAAuB,EAAE,CAAC;gBAC5B,CAAC;aACF;YACD,CAAC,WAAW,CAAC,kBAAkB,CAAC,EAAE;gBAChC,MAAM,EAAE,CACN,KAAqC,EACrC,IAAuB,EACvB,EAAE;oBACF,gBAAgB,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,IAA6B,EAAE,CAAC,CAAC;gBACnE,CAAC;aACF;SACF,CAAC;QAEF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,MAAM,CAAW;YACrC,KAAK,EAAE,qBAAqB;YAC5B,WAAW,EAAE;gBACX,IAAI,EAAE,UAAU;gBAChB,SAAS;aACV;YACD,aAAa,EAAE,IAAI;YACnB,WAAW,EAAE,IAAI;YACjB,cAAc;SACf,CAAC,CAAC;QAEH,GAAG,CAAC,KAAM,CAAC,iBAAiB,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEzC,MAAM,MAAM,GAAG,EAAE,QAAQ,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;QACzD,MAAM,GAAG,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAExC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAEnD,MAAM,UAAU,GAAG,YAAyC,CAAC;QAC7D,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CACjC,CAAC,CAAsC,EAAE,EAAE,CACzC,CAAC,KAAK,SAAS,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,CAClD,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAE5C,MAAM,cAAc,GAAG,SAAS;aAC7B,GAAG,CACF,CAAC,CAAC,EAAE,EAAE,CACH,CAAmD,CAClD,YAAY,CAAC,IAAI,CAClB,CACJ;aACA,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}