@illuma-ai/agents 1.0.89 → 1.0.93

package/src/types/graph.test.js

@@ -0,0 +1,192 @@
+/**
+ * Type-level tests to ensure StructuredOutputConfig interface is correctly defined.
+ * These tests don't run at runtime but will fail at compile time if types are wrong.
+ */
+describe('StructuredOutputConfig type', () => {
+    describe('type compatibility', () => {
+        it('should accept minimal configuration', () => {
+            const config = {
+                schema: { type: 'object' },
+            };
+            expect(config.schema).toBeDefined();
+        });
+        it('should accept full configuration', () => {
+            const config = {
+                schema: {
+                    type: 'object',
+                    properties: {
+                        name: { type: 'string' },
+                        value: { type: 'number' },
+                    },
+                    required: ['name'],
+                },
+                name: 'TestSchema',
+                description: 'A test schema for structured output',
+                mode: 'auto',
+                strict: true,
+                handleErrors: true,
+                maxRetries: 3,
+                includeRaw: false,
+            };
+            expect(config.schema).toBeDefined();
+            expect(config.name).toBe('TestSchema');
+            expect(config.description).toBe('A test schema for structured output');
+            expect(config.mode).toBe('auto');
+            expect(config.strict).toBe(true);
+            expect(config.handleErrors).toBe(true);
+            expect(config.maxRetries).toBe(3);
+            expect(config.includeRaw).toBe(false);
+        });
+        it('should accept all valid mode values', () => {
+            const modes = ['auto', 'tool', 'provider'];
+            for (const mode of modes) {
+                const config = {
+                    schema: { type: 'string' },
+                    mode,
+                };
+                expect(config.mode).toBe(mode);
+            }
+        });
+        it('should make all fields except schema optional', () => {
+            const minimalConfig = {
+                schema: { type: 'boolean' },
+            };
+            // These should all be undefined for minimal config
+            expect(minimalConfig.name).toBeUndefined();
+            expect(minimalConfig.description).toBeUndefined();
+            expect(minimalConfig.mode).toBeUndefined();
+            expect(minimalConfig.strict).toBeUndefined();
+            expect(minimalConfig.handleErrors).toBeUndefined();
+            expect(minimalConfig.maxRetries).toBeUndefined();
+            expect(minimalConfig.includeRaw).toBeUndefined();
+        });
+    });
+    describe('schema property types', () => {
+        it('should accept object schema', () => {
+            const config = {
+                schema: {
+                    type: 'object',
+                    properties: {
+                        nested: {
+                            type: 'object',
+                            properties: {
+                                value: { type: 'string' },
+                            },
+                        },
+                    },
+                },
+            };
+            expect(config.schema.type).toBe('object');
+        });
+        it('should accept array schema', () => {
+            const config = {
+                schema: {
+                    type: 'array',
+                    items: {
+                        type: 'object',
+                        properties: {
+                            id: { type: 'number' },
+                        },
+                    },
+                },
+            };
+            expect(config.schema.type).toBe('array');
+        });
+        it('should accept schema with enum', () => {
+            const config = {
+                schema: {
+                    type: 'string',
+                    enum: ['option1', 'option2', 'option3'],
+                },
+            };
+            expect(config.schema.enum).toEqual(['option1', 'option2', 'option3']);
+        });
+        it('should accept schema with $ref', () => {
+            const config = {
+                schema: {
+                    $schema: 'http://json-schema.org/draft-07/schema#',
+                    definitions: {
+                        Address: {
+                            type: 'object',
+                            properties: {
+                                street: { type: 'string' },
+                                city: { type: 'string' },
+                            },
+                        },
+                    },
+                    type: 'object',
+                    properties: {
+                        address: { $ref: '#/definitions/Address' },
+                    },
+                },
+            };
+            expect(config.schema.$schema).toBeDefined();
+        });
+    });
+});
+describe('AgentInputs with structuredOutput', () => {
+    it('should accept structuredOutput as optional field', () => {
+        const inputsWithout = {
+            agentId: 'agent-1',
+            provider: 'openai',
+        };
+        expect(inputsWithout.structuredOutput).toBeUndefined();
+        const inputsWith = {
+            agentId: 'agent-2',
+            provider: 'anthropic',
+            structuredOutput: {
+                schema: { type: 'object' },
+            },
+        };
+        expect(inputsWith.structuredOutput).toBeDefined();
+    });
+});
+describe('BaseGraphState with structuredResponse', () => {
+    it('should support structuredResponse field', () => {
+        const state = {
+            messages: [],
+            structuredResponse: { result: 'success', data: [1, 2, 3] },
+        };
+        expect(state.structuredResponse).toEqual({
+            result: 'success',
+            data: [1, 2, 3],
+        });
+    });
+    it('should allow structuredResponse to be undefined', () => {
+        const state = {
+            messages: [],
+        };
+        expect(state.structuredResponse).toBeUndefined();
+    });
+});
+describe('AgentInputs with summarizeCallback', () => {
+    it('should accept summarizeCallback as optional field', () => {
+        const inputsWithout = {
+            agentId: 'agent-1',
+            provider: 'openai',
+        };
+        expect(inputsWithout.summarizeCallback).toBeUndefined();
+        const inputsWith = {
+            agentId: 'agent-2',
+            provider: 'anthropic',
+            summarizeCallback: async (_messages) => 'summary',
+        };
+        expect(inputsWith.summarizeCallback).toBeDefined();
+    });
+    it('should accept a valid callback function type', () => {
+        const mockCallback = async (messages) => {
+            if (messages.length === 0) {
+                return undefined;
+            }
+            return `Summary of ${messages.length} messages`;
+        };
+        const inputs = {
+            agentId: 'agent-summarize',
+            provider: 'openai',
+            summarizeCallback: mockCallback,
+        };
+        expect(inputs.summarizeCallback).toBe(mockCallback);
+    });
+});
+export {};
+//# sourceMappingURL=graph.test.js.map

package/src/types/graph.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph.test.js","sourceRoot":"","sources":["graph.test.ts"],"names":[],"mappings":"AASA;;;GAGG;AACH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,MAAM,GAA2B;gBACrC,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC3B,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,MAAM,GAA2B;gBACrC,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;wBACxB,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;qBAC1B;oBACD,QAAQ,EAAE,CAAC,MAAM,CAAC;iBACnB;gBACD,IAAI,EAAE,YAAY;gBAClB,WAAW,EAAE,qCAAqC;gBAClD,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,IAAI;gBACZ,YAAY,EAAE,IAAI;gBAClB,UAAU,EAAE,CAAC;gBACb,UAAU,EAAE,KAAK;aAClB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACjC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,KAAK,GAA2B,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;YAEnE,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAA2B;oBACrC,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;oBAC1B,IAAI;iBACL,CAAC;gBACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,aAAa,GAA2B;gBAC5C,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;aAC5B,CAAC;YAEF,mDAAmD;YACnD,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YAC3C,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC;YAClD,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;YAC3C,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YAC7C,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC,aAAa,EAAE,CAAC;YACnD,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAC;YACjD,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,aAAa,EAAE,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,MAAM,GAA2B;gBACrC,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,MAAM,EAAE;4BACN,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC1B;yBACF;qBACF;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,MAAM,GAA2B;gBACrC,MAAM,EAAE;oBACN,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;yBACvB;qBACF;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAA2B;gBACrC,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC;iBACxC;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,MAAM,GAA2B;gBACrC,MAAM,EAAE;oBACN,OAAO,EAAE,yCAAyC;oBAClD,WAAW,EAAE;wBACX,OAAO,EAAE;4BACP,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;gCAC1B,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BACzB;yBACF;qBACF;oBACD,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,OAAO,EAAE,EAAE,IAAI,EAAE,uBAAuB,EAAE;qBAC3C;iBACF;aACF,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,aAAa,GAAyB;YAC1C,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,QAAmC;SAC9C,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,aAAa,EAAE,CAAC;QAEvD,MAAM,UAAU,GAAyB;YACvC,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,WAAsC;YAChD,gBAAgB,EAAE;gBAChB,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aAC3B;SACF,CAAC;QACF,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,WAAW,EAAE,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,KAAK,GAA4B;YACrC,QAAQ,EAAE,EAAE;YACZ,kBAAkB,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;SAC3D,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC;YACvC,MAAM,EAAE,SAAS;YACjB,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;SAChB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,KAAK,GAA4B;YACrC,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,aAAa,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,aAAa,GAAyB;YAC1C,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,QAAmC;SAC9C,CAAC;QACF,MAAM,CAAC,aAAa,CAAC,iBAAiB,CAAC,CAAC,aAAa,EAAE,CAAC;QAExD,MAAM,UAAU,GAAyB;YACvC,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,WAAsC;YAChD,iBAAiB,EAAE,KAAK,EAAE,SAAwB,EAAE,EAAE,CAAC,SAAS;SACjE,CAAC;QACF,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,WAAW,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,YAAY,GAAG,KAAK,EAAE,QAAuB,EAA+B,EAAE;YAClF,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,OAAO,cAAc,QAAQ,CAAC,MAAM,WAAW,CAAC;QAClD,CAAC,CAAC;QAEF,MAAM,MAAM,GAAyB;YACnC,OAAO,EAAE,iBAAiB;YAC1B,QAAQ,EAAE,QAAmC;YAC7C,iBAAiB,EAAE,YAAY;SAChC,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/src/types/graph.ts

@@ -18,7 +18,13 @@ import type {
 import type { RunnableConfig, Runnable } from '@langchain/core/runnables';
 import type { ChatGenerationChunk } from '@langchain/core/outputs';
 import type { GoogleAIToolType } from '@langchain/google-common';
-import type { ToolMap, ToolEndEvent, GenericTool, LCTool, ToolApprovalConfig } from '@/types/tools';
+import type {
+  ToolMap,
+  ToolEndEvent,
+  GenericTool,
+  LCTool,
+  ToolApprovalConfig,
+} from '@/types/tools';
 import type { Providers, Callback, GraphNodeKeys } from '@/common';
 import type { StandardGraph, MultiAgentGraph } from '@/graphs';
 import type { ClientOptions } from '@/types/llm';
@@ -334,8 +340,8 @@ export type GraphEdge = {
   description?: string;
   /** Can return boolean or specific destination(s) */
   condition?: (state: BaseGraphState) => boolean | string | string[];
-  /** 'handoff' creates tools for dynamic routing, 'direct' creates direct edges, which also allow parallel execution */
-  edgeType?: 'handoff' | 'direct';
+  /** EdgeType.HANDOFF creates tools for dynamic routing, EdgeType.DIRECT creates direct edges with parallel execution */
+  edgeType?: import('@/common').EdgeType;
   /**
    * For direct edges: Optional prompt to add when transitioning through this edge.
    * String prompts can include variables like {results} which will be replaced with
@@ -381,7 +387,12 @@ export type StructuredOutputMode = 'tool' | 'provider' | 'native' | 'auto';
  * Resolved method used internally after mode resolution.
  * Maps to LangChain's withStructuredOutput method parameter plus our native path.
  */
-export type ResolvedStructuredOutputMethod = 'functionCalling' | 'jsonMode' | 'jsonSchema' | 'native' | undefined;
+export type ResolvedStructuredOutputMethod =
+  | 'functionCalling'
+  | 'jsonMode'
+  | 'jsonSchema'
+  | 'native'
+  | undefined;
 
 /**
  * Error thrown when the model refuses to produce structured output due to safety policies.
@@ -400,7 +411,7 @@ export class StructuredOutputTruncatedError extends Error {
   constructor(public stopReason: string) {
     super(
       `Structured output was truncated (stop_reason: ${stopReason}). ` +
-      `Increase max_tokens to allow the full JSON response to be generated.`
+        'Increase max_tokens to allow the full JSON response to be generated.'
     );
     this.name = 'StructuredOutputTruncatedError';
   }
@@ -538,5 +549,14 @@ export interface AgentInputs {
    * When provided, discarded messages are summarized by the caller (e.g., Ranger)
    * using a cheap LLM call, and the summary is prepended to the context.
    */
-  summarizeCallback?: (messagesToRefine: import('@langchain/core/messages').BaseMessage[]) => Promise<string | undefined>;
+  summarizeCallback?: (
+    messagesToRefine: import('@langchain/core/messages').BaseMessage[]
+  ) => Promise<string | undefined>;
+  /**
+   * Pre-existing summary text loaded from persistent storage (MongoDB/Redis).
+   * When provided, this summary is injected into the initial message context
+   * so the agent has prior conversation history even on new turns.
+   * Set by Ranger's SummaryStore when resuming a conversation.
+   */
+  persistedSummary?: string;
 }

package/src/types/index.js

@@ -0,0 +1,7 @@
+// src/types/index.ts
+export * from './graph';
+export * from './llm';
+export * from './run';
+export * from './stream';
+export * from './tools';
+//# sourceMappingURL=index.js.map

package/src/types/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,cAAc,SAAS,CAAC;AACxB,cAAc,OAAO,CAAC;AACtB,cAAc,OAAO,CAAC;AACtB,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC"}

package/src/types/llm.js

@@ -0,0 +1,2 @@
+import { Providers } from '@/common';
+//# sourceMappingURL=llm.js.map

package/src/types/llm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm.js","sourceRoot":"","sources":["llm.ts"],"names":[],"mappings":"AAoCA,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC"}

package/src/types/llm.ts

@@ -11,7 +11,6 @@ import type {
   ClientOptions as OAIClientOptions,
 } from '@langchain/openai';
 import type { GoogleGenerativeAIChatInput } from '@langchain/google-genai';
-import type { GeminiGenerationConfig } from '@langchain/google-common';
 import type { ChatVertexAIInput } from '@langchain/google-vertexai';
 import type { ChatDeepSeekCallOptions } from '@langchain/deepseek';
 import type { ChatOpenRouterCallOptions } from '@/llm/openrouter';
@@ -55,6 +54,11 @@ export type AnthropicReasoning = {
   thinking?: ThinkingConfig | boolean;
   thinkingBudget?: number;
 };
+export type GoogleThinkingConfig = {
+  thinkingBudget?: number;
+  includeThoughts?: boolean;
+  thinkingLevel?: string;
+};
 export type OpenAIClientOptions = ChatOpenAIFields;
 export type AnthropicClientOptions = AnthropicInput & {
   promptCache?: boolean;
@@ -62,6 +66,7 @@ export type AnthropicClientOptions = AnthropicInput & {
 export type MistralAIClientOptions = ChatMistralAIInput;
 export type VertexAIClientOptions = ChatVertexAIInput & {
   includeThoughts?: boolean;
+  thinkingConfig?: GoogleThinkingConfig;
 };
 export type BedrockAnthropicInput = ChatBedrockConverseInput & {
   additionalModelRequestFields?: ChatBedrockConverseInput['additionalModelRequestFields'] &
@@ -72,7 +77,7 @@ export type BedrockConverseClientOptions = ChatBedrockConverseInput;
 export type BedrockAnthropicClientOptions = BedrockAnthropicInput;
 export type GoogleClientOptions = GoogleGenerativeAIChatInput & {
   customHeaders?: RequestOptions['customHeaders'];
-  thinkingConfig?: GeminiGenerationConfig['thinkingConfig'];
+  thinkingConfig?: GoogleThinkingConfig;
 };
 export type DeepSeekClientOptions = ChatDeepSeekCallOptions;
 export type XAIClientOptions = ChatXAIInput;
@@ -124,7 +129,7 @@ export type ChatModelMap = {
   [Providers.MISTRALAI]: ChatMistralAI;
   [Providers.MISTRAL]: ChatMistralAI;
   [Providers.OPENROUTER]: ChatOpenRouter;
-  [Providers.BEDROCK]: CustomChatBedrockConverse;
+  [Providers.BEDROCK]: InstanceType<typeof CustomChatBedrockConverse>;
   [Providers.GOOGLE]: CustomChatGoogleGenerativeAI;
   [Providers.MOONSHOT]: ChatMoonshot;
 };

package/src/types/messages.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=messages.js.map

package/src/types/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"messages.js","sourceRoot":"","sources":["messages.ts"],"names":[],"mappings":""}

package/src/types/run.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=run.js.map

package/src/types/run.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run.js","sourceRoot":"","sources":["run.ts"],"names":[],"mappings":""}

package/src/types/run.ts

@@ -115,6 +115,8 @@ export type RunConfig = {
   returnContent?: boolean;
   tokenCounter?: TokenCounter;
   indexTokenCountMap?: Record<string, number>;
+  /** Skip post-stream cleanup (clearHeavyState) — useful for tests that inspect graph state after processStream */
+  skipCleanup?: boolean;
 };
 
 export type ProvidedCallbacks =

package/src/types/stream.js

@@ -0,0 +1,2 @@
+import { GraphEvents } from '@/common/enum';
+//# sourceMappingURL=stream.js.map

package/src/types/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["stream.ts"],"names":[],"mappings":"AAaA,OAAO,EAA2B,WAAW,EAAE,MAAM,eAAe,CAAC"}

package/src/types/tools.js

@@ -0,0 +1,2 @@
+import { EnvVar } from '@/common';
+//# sourceMappingURL=tools.js.map

package/src/types/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["tools.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC"}

package/src/types/tools.ts

@@ -47,6 +47,14 @@ export type ToolNodeOptions = {
   toolDefinitions?: Map<string, LCTool>;
   /** Agent ID for event-driven mode (used to identify which agent's context to use) */
   agentId?: string;
+  /** Tool names that must be executed directly (via runTool) even in event-driven mode (e.g., graph-managed handoff tools) */
+  directToolNames?: Set<string>;
+  /**
+   * StreamingToolCallBuffer for recovering truncated tool call arguments.
+   * When parsePartialJson loses content due to max_tokens truncation,
+   * the ToolNode can fall back to extracting raw field values from the buffer.
+   */
+  streamingToolCallBuffer?: import('@/tools/StreamingToolCallBuffer').StreamingToolCallBuffer;
   /**
    * HITL tool approval configuration.
    * When provided, tools matching the policy will call interrupt() before execution.
@@ -63,6 +71,7 @@ export type ToolEndEvent = {
   tool_call: ToolCall;
   /** The content index of the tool call */
   index: number;
+  type?: 'tool_call';
 };
 
 export type CodeEnvFile = {
@@ -156,6 +165,11 @@ export type ToolCallRequest = {
   stepId?: string;
   /** Usage turn count for this tool */
   turn?: number;
+  /** Code execution session context for session continuity in event-driven mode */
+  codeSessionContext?: {
+    session_id: string;
+    files?: CodeEnvFile[];
+  };
 };
 
 /** Batch request containing ALL tool calls for a graph step */
@@ -204,7 +218,7 @@ export type ProgrammaticCache = { toolMap: ToolMap; toolDefs: LCTool[] };
  * - 'interactive': User is present in the chat, approval prompts are shown.
  * - 'scheduled': Automated/scheduled execution, approval is auto-granted.
  */
-export type ExecutionContext = 'interactive' | 'scheduled';
+export type ExecutionContext = 'interactive' | 'scheduled' | 'browser';
 
 /**
  * Policy for when a tool requires human approval.
@@ -294,6 +308,9 @@ export type ToolApprovalEvent = ToolApprovalRequest & {
 /** Search mode: code_interpreter uses external sandbox, local uses safe substring matching */
 export type ToolSearchMode = 'code_interpreter' | 'local';
 
+/** Format for MCP tool names in search results */
+export type McpNameFormat = 'full' | 'base';
+
 /** Parameters for creating a Tool Search tool */
 export type ToolSearchParams = {
   apiKey?: string;
@@ -304,6 +321,8 @@ export type ToolSearchParams = {
   mode?: ToolSearchMode;
   /** Filter tools to only those from specific MCP server(s). Can be a single name or array of names. */
   mcpServer?: string | string[];
+  /** Format for MCP tool names: 'full' (tool_mcp_server) or 'base' (tool only). Default: 'full' */
+  mcpNameFormat?: McpNameFormat;
   [key: string]: unknown;
 };
 
@@ -429,7 +448,7 @@ export type CodeSessionContext = {
   /** Session ID from the code execution environment */
   session_id: string;
   /** Files generated in this session (for context/tracking) */
-  files: FileRefs;
+  files?: FileRefs;
   /** Timestamp of last update */
   lastUpdated: number;
 };

package/src/utils/contextAnalytics.js

@@ -0,0 +1,79 @@
+/**
+ * Context Analytics Utility
+ *
+ * Provides context analytics data for observability/traces.
+ * No console logging - just data structures for event emission.
+ */
+/**
+ * Resolve the message type string from a BaseMessage or message-like object.
+ *
+ * Uses the same duck-typing contract as LangChain's `isBaseMessage`:
+ * prefer `_getType()` (internal), fallback to `getType()` (public).
+ * This handles cross-package BaseMessage instances where `instanceof`
+ * checks may fail but prototype methods are still present.
+ */
+function resolveMessageType(msg) {
+    if (typeof msg._getType === 'function') {
+        return msg._getType();
+    }
+    if (typeof msg.getType === 'function') {
+        return msg.getType();
+    }
+    // Last resort: infer from known LangChain serialization fields
+    return msg.type ?? 'unknown';
+}
+/**
+ * Build context analytics for traces (no logging).
+ *
+ * @param messages - Array of BaseMessage (may originate from different @langchain/core installations)
+ * @param options  - Token counting and context budget info
+ */
+export function buildContextAnalytics(messages, options) {
+    const { tokenCounter, maxContextTokens, instructionTokens, indexTokenCountMap, } = options;
+    let totalTokens = 0;
+    const breakdown = {};
+    for (let i = 0; i < messages.length; i++) {
+        const msg = messages[i];
+        const type = resolveMessageType(msg);
+        let tokens = 0;
+        if (indexTokenCountMap && indexTokenCountMap[i] != null) {
+            tokens = indexTokenCountMap[i];
+        }
+        else if (tokenCounter) {
+            try {
+                tokens = tokenCounter(msg);
+            }
+            catch {
+                const content = typeof msg.content === 'string'
+                    ? msg.content
+                    : JSON.stringify(msg.content);
+                tokens = Math.ceil(content.length / 4);
+            }
+        }
+        totalTokens += tokens;
+        if (!breakdown[type]) {
+            breakdown[type] = { tokens: 0, percent: 0 };
+        }
+        breakdown[type].tokens += tokens;
+    }
+    for (const type of Object.keys(breakdown)) {
+        breakdown[type].percent =
+            totalTokens > 0
+                ? Math.round((breakdown[type].tokens / totalTokens) * 1000) / 10
+                : 0;
+    }
+    let utilizationPercent;
+    if (maxContextTokens && maxContextTokens > 0) {
+        utilizationPercent =
+            Math.round((totalTokens / maxContextTokens) * 1000) / 10;
+    }
+    return {
+        messageCount: messages.length,
+        totalTokens,
+        maxContextTokens,
+        instructionTokens,
+        utilizationPercent,
+        breakdown,
+    };
+}
+//# sourceMappingURL=contextAnalytics.js.map

package/src/utils/contextAnalytics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"contextAnalytics.js","sourceRoot":"","sources":["contextAnalytics.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAqBH;;;;;;;GAOG;AACH,SAAS,kBAAkB,CAAC,GAAgB;IAC1C,IAAI,OAAO,GAAG,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;QACvC,OAAO,GAAG,CAAC,QAAQ,EAAE,CAAC;IACxB,CAAC;IACD,IAAI,OAAO,GAAG,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;QACtC,OAAO,GAAG,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;IACD,+DAA+D;IAC/D,OAAQ,GAA0C,CAAC,IAAc,IAAI,SAAS,CAAC;AACjF,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,qBAAqB,CACnC,QAAuB,EACvB,OAKC;IAED,MAAM,EACJ,YAAY,EACZ,gBAAgB,EAChB,iBAAiB,EACjB,kBAAkB,GACnB,GAAG,OAAO,CAAC;IAEZ,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,MAAM,SAAS,GAAwD,EAAE,CAAC;IAE1E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,IAAI,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;QAErC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,kBAAkB,IAAI,kBAAkB,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;YACxD,MAAM,GAAG,kBAAkB,CAAC,CAAC,CAAE,CAAC;QAClC,CAAC;aAAM,IAAI,YAAY,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;YAC7B,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,OAAO,GACX,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ;oBAC7B,CAAC,CAAC,GAAG,CAAC,OAAO;oBACb,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAClC,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YACzC,CAAC;QACH,CAAC;QAED,WAAW,IAAI,MAAM,CAAC;QAEtB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;YACrB,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QAC9C,CAAC;QACD,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC;IACnC,CAAC;IAED,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1C,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO;YACrB,WAAW,GAAG,CAAC;gBACb,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,WAAW,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE;gBAChE,CAAC,CAAC,CAAC,CAAC;IACV,CAAC;IAED,IAAI,kBAAsC,CAAC;IAC3C,IAAI,gBAAgB,IAAI,gBAAgB,GAAG,CAAC,EAAE,CAAC;QAC7C,kBAAkB;YAChB,IAAI,CAAC,KAAK,CAAC,CAAC,WAAW,GAAG,gBAAgB,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7D,CAAC;IAED,OAAO;QACL,YAAY,EAAE,QAAQ,CAAC,MAAM;QAC7B,WAAW;QACX,gBAAgB;QAChB,iBAAiB;QACjB,kBAAkB;QAClB,SAAS;KACV,CAAC;AACJ,CAAC"}