langchain 1.0.0-alpha.5 → 1.0.0-alpha.7

package/dist/agents/middlewareAgent/nodes/middleware.cjs

@@ -1,5 +1,7 @@
+const require_rolldown_runtime = require('../../../_virtual/rolldown_runtime.cjs');
 const require_RunnableCallable = require('../../RunnableCallable.cjs');
 const require_utils = require('./utils.cjs');
+const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));
 
 //#region src/agents/middlewareAgent/nodes/middleware.ts
 var MiddlewareNode = class extends require_RunnableCallable.RunnableCallable {
@@ -9,20 +11,22 @@ var MiddlewareNode = class extends require_RunnableCallable.RunnableCallable {
 		*/
 		let filteredContext = {};
 		/**
-		* Check both config.context and config.configurable.context
+		* Parse context using middleware's contextSchema to apply defaults and validation
 		*/
-		if (this.middleware.contextSchema && config?.context) {
+		if (this.middleware.contextSchema) {
 			/**
 			* Extract only the fields relevant to this middleware's schema
 			*/
 			const schemaShape = this.middleware.contextSchema?.shape;
 			if (schemaShape) {
 				const relevantContext = {};
-				for (const key of Object.keys(schemaShape)) if (key in config.context) relevantContext[key] = config.context[key];
+				const invokeContext = config?.context || {};
+				for (const key of Object.keys(schemaShape)) if (key in invokeContext) relevantContext[key] = invokeContext[key];
 				/**
-				* Parse to apply defaults and validation
+				* Parse to apply defaults and validation, even if relevantContext is empty
+				* This will throw if required fields are missing and no defaults exist
 				*/
-				filteredContext = this.middleware.contextSchema.parse(relevantContext);
+				filteredContext = (0, __langchain_core_utils_types.interopParse)(this.middleware.contextSchema, relevantContext);
 			}
 		}
 		/**
@@ -30,15 +34,11 @@ var MiddlewareNode = class extends require_RunnableCallable.RunnableCallable {
 		*/
 		const runtime = {
 			toolCalls: require_utils.parseToolCalls(state.messages),
-			toolResults: require_utils.parseToolResults(state.messages),
-			context: filteredContext
-		};
-		const controls = {
-			jumpTo: (target, stateUpdate) => ({
-				type: "jump",
-				target,
-				stateUpdate
-			}),
+			context: filteredContext,
+			writer: config?.writer,
+			interrupt: config?.interrupt,
+			signal: config?.signal,
+			tools: this.middleware.tools ?? [],
 			terminate: (result$1) => {
 				if (result$1 instanceof Error) throw result$1;
 				return {
@@ -47,20 +47,45 @@ var MiddlewareNode = class extends require_RunnableCallable.RunnableCallable {
 				};
 			}
 		};
-		const result = await this.runHook(state, runtime, controls);
+		const result = await this.runHook(state, runtime);
 		/**
 		* If result is undefined, return current state
 		*/
-		if (!result) return state;
+		if (!result) return {
+			...state,
+			jumpTo: void 0
+		};
+		/**
+		* Verify that the jump target is allowed for the middleware
+		*/
+		const jumpToConstraint = this.name?.startsWith("BeforeModelNode_") ? this.middleware.beforeModelJumpTo : this.middleware.afterModelJumpTo;
+		if (typeof result.jumpTo === "string" && !jumpToConstraint?.includes(result.jumpTo)) {
+			const constraint = this.name?.startsWith("BeforeModelNode_") ? "beforeModelJumpTo" : "afterModelJumpTo";
+			const suggestion = jumpToConstraint && jumpToConstraint.length > 0 ? `must be one of: ${jumpToConstraint?.join(", ")}.` : `no ${constraint} defined in middleware ${this.middleware.name}.`;
+			throw new Error(`Invalid jump target: ${result.jumpTo}, ${suggestion}.`);
+		}
+		const jumpTo = require_utils.parseJumpToTarget(result.jumpTo);
 		/**
 		* If result is a control action, handle it
 		*/
+		if (typeof result === "object" && "type" in result) {
+			if (result.type === "terminate") {
+				if (result.error) throw result.error;
+				return {
+					...state,
+					...result.result || {},
+					jumpTo
+				};
+			}
+			throw new Error(`Invalid control action: ${JSON.stringify(result)}`);
+		}
 		/**
 		* If result is a state update, merge it with current state
 		*/
 		return {
 			...state,
-			...result
+			...result,
+			jumpTo
 		};
 	}
 	get nodeOptions() {

package/dist/agents/middlewareAgent/nodes/middleware.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.cjs","names":["RunnableCallable","state: TStateSchema","config?: LangGraphRunnableConfig","relevantContext: Record<string, unknown>","runtime: Runtime<TContextSchema>","parseToolCalls","parseToolResults","controls: Controls<TStateSchema>","target: \"model\" | \"tools\"","stateUpdate?: Partial<TStateSchema>","result?: Partial<TStateSchema> | Error","result","derivePrivateState"],"sources":["../../../../src/agents/middlewareAgent/nodes/middleware.ts"],"sourcesContent":["/* eslint-disable no-instanceof/no-instanceof */\nimport { z } from \"zod\";\nimport { LangGraphRunnableConfig, Command } from \"@langchain/langgraph\";\n\nimport { RunnableCallable } from \"../../RunnableCallable.js\";\nimport type {\n  Runtime,\n  Controls,\n  ControlAction,\n  AgentMiddleware,\n  MiddlewareResult,\n} from \"../types.js\";\nimport {\n  derivePrivateState,\n  parseToolCalls,\n  parseToolResults,\n} from \"./utils.js\";\n\ntype NodeOutput<TStateSchema extends Record<string, any>> =\n  | TStateSchema\n  | Command<any, TStateSchema, string>;\n\nexport abstract class MiddlewareNode<\n  TStateSchema extends Record<string, any>,\n  TContextSchema extends Record<string, any>\n> extends RunnableCallable<TStateSchema, NodeOutput<TStateSchema>> {\n  abstract middleware: AgentMiddleware<\n    z.ZodObject<z.ZodRawShape>,\n    z.ZodObject<z.ZodRawShape>\n  >;\n\n  abstract runHook(\n    state: TStateSchema,\n    config?: Runtime<TContextSchema>,\n    controls?: Controls<TStateSchema>\n  ): Promise<MiddlewareResult<TStateSchema>>;\n\n  async invokeMiddleware(\n    state: TStateSchema,\n    config?: LangGraphRunnableConfig\n  ): Promise<NodeOutput<TStateSchema>> {\n    /**\n     * Filter context based on middleware's contextSchema\n     */\n    let filteredContext = {} as TContextSchema;\n    /**\n     * Check both config.context and config.configurable.context\n     */\n    if (this.middleware.contextSchema && config?.context) {\n      /**\n       * Extract only the fields relevant to this middleware's schema\n       */\n      const schemaShape = this.middleware.contextSchema?.shape;\n      if (schemaShape) {\n        const relevantContext: Record<string, unknown> = {};\n        for (const key of Object.keys(schemaShape)) {\n          if (key in config.context) {\n            relevantContext[key] = config.context[key];\n          }\n        }\n        /**\n         * Parse to apply defaults and validation\n         */\n        filteredContext = this.middleware.contextSchema.parse(\n          relevantContext\n        ) as TContextSchema;\n      }\n    }\n\n    /**\n     * ToDo: implement later\n     */\n    const runtime: Runtime<TContextSchema> = {\n      toolCalls: parseToolCalls(state.messages),\n      toolResults: parseToolResults(state.messages),\n      context: filteredContext,\n    };\n\n    const controls: Controls<TStateSchema> = {\n      jumpTo: (\n        target: \"model\" | \"tools\",\n        stateUpdate?: Partial<TStateSchema>\n      ): ControlAction<TStateSchema> => ({\n        type: \"jump\",\n        target,\n        stateUpdate,\n      }),\n      terminate: (\n        result?: Partial<TStateSchema> | Error\n      ): ControlAction<TStateSchema> => {\n        if (result instanceof Error) {\n          throw result;\n        }\n        return { type: \"terminate\", result };\n      },\n    };\n\n    const result = await this.runHook(state, runtime, controls);\n\n    /**\n     * If result is undefined, return current state\n     */\n    if (!result) {\n      return state;\n    }\n\n    /**\n     * If result is a control action, handle it\n     */\n    // if (typeof result === \"object\" && \"type\" in result) {\n    //   // Handle control actions\n    //   const action = result as ControlAction<TStateSchema>;\n    //   if (action.type === \"terminate\") {\n    //     if (action.error) {\n    //       throw action.error;\n    //     }\n    //     return { ...state, ...(action.result || {}) };\n    //   }\n\n    //   if (action.type === \"jump\") {\n    //     return new Command<any, TStateSchema, string>({\n    //       goto: action.target,\n    //       update: { ...state, ...(action.stateUpdate || {}) },\n    //     });\n    //   }\n\n    //   throw new Error(`Invalid control action: ${JSON.stringify(action)}`);\n    // }\n\n    /**\n     * If result is a state update, merge it with current state\n     */\n    return { ...state, ...result };\n  }\n\n  get nodeOptions(): {\n    input: z.ZodObject<TStateSchema>;\n  } {\n    return {\n      input: derivePrivateState(\n        this.middleware.stateSchema\n      ) as z.ZodObject<TStateSchema>,\n    };\n  }\n}\n"],"mappings":";;;;AAsBA,IAAsB,iBAAtB,cAGUA,0CAAyD;CAYjE,MAAM,iBACJC,OACAC,QACmC;;;;EAInC,IAAI,kBAAkB,CAAE;;;;AAIxB,MAAI,KAAK,WAAW,iBAAiB,QAAQ,SAAS;;;;GAIpD,MAAM,cAAc,KAAK,WAAW,eAAe;AACnD,OAAI,aAAa;IACf,MAAMC,kBAA2C,CAAE;AACnD,SAAK,MAAM,OAAO,OAAO,KAAK,YAAY,CACxC,KAAI,OAAO,OAAO,SAChB,gBAAgB,OAAO,OAAO,QAAQ;;;;IAM1C,kBAAkB,KAAK,WAAW,cAAc,MAC9C,gBACD;GACF;EACF;;;;EAKD,MAAMC,UAAmC;GACvC,WAAWC,6BAAe,MAAM,SAAS;GACzC,aAAaC,+BAAiB,MAAM,SAAS;GAC7C,SAAS;EACV;EAED,MAAMC,WAAmC;GACvC,QAAQ,CACNC,QACAC,iBACiC;IACjC,MAAM;IACN;IACA;GACD;GACD,WAAW,CACTC,aACgC;AAChC,QAAIC,oBAAkB,MACpB,OAAMA;AAER,WAAO;KAAE,MAAM;KAAa;IAAQ;GACrC;EACF;EAED,MAAM,SAAS,MAAM,KAAK,QAAQ,OAAO,SAAS,SAAS;;;;AAK3D,MAAI,CAAC,OACH,QAAO;;;;;;;AA6BT,SAAO;GAAE,GAAG;GAAO,GAAG;EAAQ;CAC/B;CAED,IAAI,cAEF;AACA,SAAO,EACL,OAAOC,iCACL,KAAK,WAAW,YACjB,CACF;CACF;AACF"}
+{"version":3,"file":"middleware.cjs","names":["RunnableCallable","state: TStateSchema","config?: LangGraphRunnableConfig","relevantContext: Record<string, unknown>","runtime: Runtime<TStateSchema, TContextSchema>","parseToolCalls","result?: Partial<TStateSchema> | Error","result","parseJumpToTarget","derivePrivateState"],"sources":["../../../../src/agents/middlewareAgent/nodes/middleware.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\n/* eslint-disable no-instanceof/no-instanceof */\nimport { z } from \"zod/v3\";\nimport { LangGraphRunnableConfig, Command } from \"@langchain/langgraph\";\nimport { interopParse } from \"@langchain/core/utils/types\";\n\nimport { RunnableCallable } from \"../../RunnableCallable.js\";\nimport type {\n  Runtime,\n  ControlAction,\n  AgentMiddleware,\n  MiddlewareResult,\n  JumpToTarget,\n} from \"../types.js\";\nimport {\n  derivePrivateState,\n  parseToolCalls,\n  parseJumpToTarget,\n} from \"./utils.js\";\n\ntype NodeOutput<TStateSchema extends Record<string, any>> =\n  | TStateSchema\n  | Command<any, TStateSchema, string>;\n\nexport abstract class MiddlewareNode<\n  TStateSchema extends Record<string, any>,\n  TContextSchema extends Record<string, any>\n> extends RunnableCallable<TStateSchema, NodeOutput<TStateSchema>> {\n  abstract middleware: AgentMiddleware<\n    z.ZodObject<z.ZodRawShape>,\n    z.ZodObject<z.ZodRawShape>\n  >;\n\n  abstract runHook(\n    state: TStateSchema,\n    config?: Runtime<TStateSchema, TContextSchema>\n  ): Promise<MiddlewareResult<TStateSchema>>;\n\n  async invokeMiddleware(\n    state: TStateSchema,\n    config?: LangGraphRunnableConfig\n  ): Promise<NodeOutput<TStateSchema>> {\n    /**\n     * Filter context based on middleware's contextSchema\n     */\n    let filteredContext = {} as TContextSchema;\n    /**\n     * Parse context using middleware's contextSchema to apply defaults and validation\n     */\n    if (this.middleware.contextSchema) {\n      /**\n       * Extract only the fields relevant to this middleware's schema\n       */\n      const schemaShape = this.middleware.contextSchema?.shape;\n      if (schemaShape) {\n        const relevantContext: Record<string, unknown> = {};\n        const invokeContext = config?.context || {};\n        for (const key of Object.keys(schemaShape)) {\n          if (key in invokeContext) {\n            relevantContext[key] = invokeContext[key];\n          }\n        }\n        /**\n         * Parse to apply defaults and validation, even if relevantContext is empty\n         * This will throw if required fields are missing and no defaults exist\n         */\n        filteredContext = interopParse(\n          this.middleware.contextSchema,\n          relevantContext\n        ) as TContextSchema;\n      }\n    }\n\n    /**\n     * ToDo: implement later\n     */\n    const runtime: Runtime<TStateSchema, TContextSchema> = {\n      toolCalls: parseToolCalls(state.messages),\n      context: filteredContext,\n      writer: config?.writer,\n      interrupt: config?.interrupt,\n      signal: config?.signal,\n      tools: this.middleware.tools ?? [],\n      terminate: (\n        result?: Partial<TStateSchema> | Error\n      ): ControlAction<TStateSchema> => {\n        if (result instanceof Error) {\n          throw result;\n        }\n        return { type: \"terminate\", result };\n      },\n    };\n\n    const result = await this.runHook(state, runtime);\n\n    /**\n     * If result is undefined, return current state\n     */\n    if (!result) {\n      return { ...state, jumpTo: undefined };\n    }\n\n    /**\n     * Verify that the jump target is allowed for the middleware\n     */\n    const jumpToConstraint = this.name?.startsWith(\"BeforeModelNode_\")\n      ? this.middleware.beforeModelJumpTo\n      : this.middleware.afterModelJumpTo;\n    if (\n      typeof result.jumpTo === \"string\" &&\n      !jumpToConstraint?.includes(result.jumpTo as JumpToTarget)\n    ) {\n      const constraint = this.name?.startsWith(\"BeforeModelNode_\")\n        ? \"beforeModelJumpTo\"\n        : \"afterModelJumpTo\";\n      const suggestion =\n        jumpToConstraint && jumpToConstraint.length > 0\n          ? `must be one of: ${jumpToConstraint?.join(\", \")}.`\n          : `no ${constraint} defined in middleware ${this.middleware.name}.`;\n      throw new Error(`Invalid jump target: ${result.jumpTo}, ${suggestion}.`);\n    }\n\n    const jumpTo = parseJumpToTarget(result.jumpTo as string);\n\n    /**\n     * If result is a control action, handle it\n     */\n    if (typeof result === \"object\" && \"type\" in result) {\n      // Handle control actions\n      if (result.type === \"terminate\") {\n        if (result.error) {\n          throw result.error;\n        }\n        return {\n          ...state,\n          ...(result.result || {}),\n          jumpTo,\n        };\n      }\n\n      throw new Error(`Invalid control action: ${JSON.stringify(result)}`);\n    }\n\n    /**\n     * If result is a state update, merge it with current state\n     */\n    return { ...state, ...result, jumpTo };\n  }\n\n  get nodeOptions(): {\n    input: z.ZodObject<TStateSchema>;\n  } {\n    return {\n      input: derivePrivateState(\n        this.middleware.stateSchema\n      ) as z.ZodObject<TStateSchema>,\n    };\n  }\n}\n"],"mappings":";;;;;;AAwBA,IAAsB,iBAAtB,cAGUA,0CAAyD;CAWjE,MAAM,iBACJC,OACAC,QACmC;;;;EAInC,IAAI,kBAAkB,CAAE;;;;AAIxB,MAAI,KAAK,WAAW,eAAe;;;;GAIjC,MAAM,cAAc,KAAK,WAAW,eAAe;AACnD,OAAI,aAAa;IACf,MAAMC,kBAA2C,CAAE;IACnD,MAAM,gBAAgB,QAAQ,WAAW,CAAE;AAC3C,SAAK,MAAM,OAAO,OAAO,KAAK,YAAY,CACxC,KAAI,OAAO,eACT,gBAAgB,OAAO,cAAc;;;;;IAOzC,iEACE,KAAK,WAAW,eAChB,gBACD;GACF;EACF;;;;EAKD,MAAMC,UAAiD;GACrD,WAAWC,6BAAe,MAAM,SAAS;GACzC,SAAS;GACT,QAAQ,QAAQ;GAChB,WAAW,QAAQ;GACnB,QAAQ,QAAQ;GAChB,OAAO,KAAK,WAAW,SAAS,CAAE;GAClC,WAAW,CACTC,aACgC;AAChC,QAAIC,oBAAkB,MACpB,OAAMA;AAER,WAAO;KAAE,MAAM;KAAa;IAAQ;GACrC;EACF;EAED,MAAM,SAAS,MAAM,KAAK,QAAQ,OAAO,QAAQ;;;;AAKjD,MAAI,CAAC,OACH,QAAO;GAAE,GAAG;GAAO,QAAQ;EAAW;;;;EAMxC,MAAM,mBAAmB,KAAK,MAAM,WAAW,mBAAmB,GAC9D,KAAK,WAAW,oBAChB,KAAK,WAAW;AACpB,MACE,OAAO,OAAO,WAAW,YACzB,CAAC,kBAAkB,SAAS,OAAO,OAAuB,EAC1D;GACA,MAAM,aAAa,KAAK,MAAM,WAAW,mBAAmB,GACxD,sBACA;GACJ,MAAM,aACJ,oBAAoB,iBAAiB,SAAS,IAC1C,CAAC,gBAAgB,EAAE,kBAAkB,KAAK,KAAK,CAAC,CAAC,CAAC,GAClD,CAAC,GAAG,EAAE,WAAW,uBAAuB,EAAE,KAAK,WAAW,KAAK,CAAC,CAAC;AACvE,SAAM,IAAI,MAAM,CAAC,qBAAqB,EAAE,OAAO,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;EACxE;EAED,MAAM,SAASC,gCAAkB,OAAO,OAAiB;;;;AAKzD,MAAI,OAAO,WAAW,YAAY,UAAU,QAAQ;AAElD,OAAI,OAAO,SAAS,aAAa;AAC/B,QAAI,OAAO,MACT,OAAM,OAAO;AAEf,WAAO;KACL,GAAG;KACH,GAAI,OAAO,UAAU,CAAE;KACvB;IACD;GACF;AAED,SAAM,IAAI,MAAM,CAAC,wBAAwB,EAAE,KAAK,UAAU,OAAO,EAAE;EACpE;;;;AAKD,SAAO;GAAE,GAAG;GAAO,GAAG;GAAQ;EAAQ;CACvC;CAED,IAAI,cAEF;AACA,SAAO,EACL,OAAOC,iCACL,KAAK,WAAW,YACjB,CACF;CACF;AACF"}

package/dist/agents/middlewareAgent/nodes/middleware.js

@@ -1,5 +1,6 @@
 import { RunnableCallable } from "../../RunnableCallable.js";
-import { derivePrivateState, parseToolCalls, parseToolResults } from "./utils.js";
+import { derivePrivateState, parseJumpToTarget, parseToolCalls } from "./utils.js";
+import { interopParse } from "@langchain/core/utils/types";
 
 //#region src/agents/middlewareAgent/nodes/middleware.ts
 var MiddlewareNode = class extends RunnableCallable {
@@ -9,20 +10,22 @@ var MiddlewareNode = class extends RunnableCallable {
 		*/
 		let filteredContext = {};
 		/**
-		* Check both config.context and config.configurable.context
+		* Parse context using middleware's contextSchema to apply defaults and validation
 		*/
-		if (this.middleware.contextSchema && config?.context) {
+		if (this.middleware.contextSchema) {
 			/**
 			* Extract only the fields relevant to this middleware's schema
 			*/
 			const schemaShape = this.middleware.contextSchema?.shape;
 			if (schemaShape) {
 				const relevantContext = {};
-				for (const key of Object.keys(schemaShape)) if (key in config.context) relevantContext[key] = config.context[key];
+				const invokeContext = config?.context || {};
+				for (const key of Object.keys(schemaShape)) if (key in invokeContext) relevantContext[key] = invokeContext[key];
 				/**
-				* Parse to apply defaults and validation
+				* Parse to apply defaults and validation, even if relevantContext is empty
+				* This will throw if required fields are missing and no defaults exist
 				*/
-				filteredContext = this.middleware.contextSchema.parse(relevantContext);
+				filteredContext = interopParse(this.middleware.contextSchema, relevantContext);
 			}
 		}
 		/**
@@ -30,15 +33,11 @@ var MiddlewareNode = class extends RunnableCallable {
 		*/
 		const runtime = {
 			toolCalls: parseToolCalls(state.messages),
-			toolResults: parseToolResults(state.messages),
-			context: filteredContext
-		};
-		const controls = {
-			jumpTo: (target, stateUpdate) => ({
-				type: "jump",
-				target,
-				stateUpdate
-			}),
+			context: filteredContext,
+			writer: config?.writer,
+			interrupt: config?.interrupt,
+			signal: config?.signal,
+			tools: this.middleware.tools ?? [],
 			terminate: (result$1) => {
 				if (result$1 instanceof Error) throw result$1;
 				return {
@@ -47,20 +46,45 @@ var MiddlewareNode = class extends RunnableCallable {
 				};
 			}
 		};
-		const result = await this.runHook(state, runtime, controls);
+		const result = await this.runHook(state, runtime);
 		/**
 		* If result is undefined, return current state
 		*/
-		if (!result) return state;
+		if (!result) return {
+			...state,
+			jumpTo: void 0
+		};
+		/**
+		* Verify that the jump target is allowed for the middleware
+		*/
+		const jumpToConstraint = this.name?.startsWith("BeforeModelNode_") ? this.middleware.beforeModelJumpTo : this.middleware.afterModelJumpTo;
+		if (typeof result.jumpTo === "string" && !jumpToConstraint?.includes(result.jumpTo)) {
+			const constraint = this.name?.startsWith("BeforeModelNode_") ? "beforeModelJumpTo" : "afterModelJumpTo";
+			const suggestion = jumpToConstraint && jumpToConstraint.length > 0 ? `must be one of: ${jumpToConstraint?.join(", ")}.` : `no ${constraint} defined in middleware ${this.middleware.name}.`;
+			throw new Error(`Invalid jump target: ${result.jumpTo}, ${suggestion}.`);
+		}
+		const jumpTo = parseJumpToTarget(result.jumpTo);
 		/**
 		* If result is a control action, handle it
 		*/
+		if (typeof result === "object" && "type" in result) {
+			if (result.type === "terminate") {
+				if (result.error) throw result.error;
+				return {
+					...state,
+					...result.result || {},
+					jumpTo
+				};
+			}
+			throw new Error(`Invalid control action: ${JSON.stringify(result)}`);
+		}
 		/**
 		* If result is a state update, merge it with current state
 		*/
 		return {
 			...state,
-			...result
+			...result,
+			jumpTo
 		};
 	}
 	get nodeOptions() {

package/dist/agents/middlewareAgent/nodes/middleware.js.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.js","names":["state: TStateSchema","config?: LangGraphRunnableConfig","relevantContext: Record<string, unknown>","runtime: Runtime<TContextSchema>","controls: Controls<TStateSchema>","target: \"model\" | \"tools\"","stateUpdate?: Partial<TStateSchema>","result?: Partial<TStateSchema> | Error","result"],"sources":["../../../../src/agents/middlewareAgent/nodes/middleware.ts"],"sourcesContent":["/* eslint-disable no-instanceof/no-instanceof */\nimport { z } from \"zod\";\nimport { LangGraphRunnableConfig, Command } from \"@langchain/langgraph\";\n\nimport { RunnableCallable } from \"../../RunnableCallable.js\";\nimport type {\n  Runtime,\n  Controls,\n  ControlAction,\n  AgentMiddleware,\n  MiddlewareResult,\n} from \"../types.js\";\nimport {\n  derivePrivateState,\n  parseToolCalls,\n  parseToolResults,\n} from \"./utils.js\";\n\ntype NodeOutput<TStateSchema extends Record<string, any>> =\n  | TStateSchema\n  | Command<any, TStateSchema, string>;\n\nexport abstract class MiddlewareNode<\n  TStateSchema extends Record<string, any>,\n  TContextSchema extends Record<string, any>\n> extends RunnableCallable<TStateSchema, NodeOutput<TStateSchema>> {\n  abstract middleware: AgentMiddleware<\n    z.ZodObject<z.ZodRawShape>,\n    z.ZodObject<z.ZodRawShape>\n  >;\n\n  abstract runHook(\n    state: TStateSchema,\n    config?: Runtime<TContextSchema>,\n    controls?: Controls<TStateSchema>\n  ): Promise<MiddlewareResult<TStateSchema>>;\n\n  async invokeMiddleware(\n    state: TStateSchema,\n    config?: LangGraphRunnableConfig\n  ): Promise<NodeOutput<TStateSchema>> {\n    /**\n     * Filter context based on middleware's contextSchema\n     */\n    let filteredContext = {} as TContextSchema;\n    /**\n     * Check both config.context and config.configurable.context\n     */\n    if (this.middleware.contextSchema && config?.context) {\n      /**\n       * Extract only the fields relevant to this middleware's schema\n       */\n      const schemaShape = this.middleware.contextSchema?.shape;\n      if (schemaShape) {\n        const relevantContext: Record<string, unknown> = {};\n        for (const key of Object.keys(schemaShape)) {\n          if (key in config.context) {\n            relevantContext[key] = config.context[key];\n          }\n        }\n        /**\n         * Parse to apply defaults and validation\n         */\n        filteredContext = this.middleware.contextSchema.parse(\n          relevantContext\n        ) as TContextSchema;\n      }\n    }\n\n    /**\n     * ToDo: implement later\n     */\n    const runtime: Runtime<TContextSchema> = {\n      toolCalls: parseToolCalls(state.messages),\n      toolResults: parseToolResults(state.messages),\n      context: filteredContext,\n    };\n\n    const controls: Controls<TStateSchema> = {\n      jumpTo: (\n        target: \"model\" | \"tools\",\n        stateUpdate?: Partial<TStateSchema>\n      ): ControlAction<TStateSchema> => ({\n        type: \"jump\",\n        target,\n        stateUpdate,\n      }),\n      terminate: (\n        result?: Partial<TStateSchema> | Error\n      ): ControlAction<TStateSchema> => {\n        if (result instanceof Error) {\n          throw result;\n        }\n        return { type: \"terminate\", result };\n      },\n    };\n\n    const result = await this.runHook(state, runtime, controls);\n\n    /**\n     * If result is undefined, return current state\n     */\n    if (!result) {\n      return state;\n    }\n\n    /**\n     * If result is a control action, handle it\n     */\n    // if (typeof result === \"object\" && \"type\" in result) {\n    //   // Handle control actions\n    //   const action = result as ControlAction<TStateSchema>;\n    //   if (action.type === \"terminate\") {\n    //     if (action.error) {\n    //       throw action.error;\n    //     }\n    //     return { ...state, ...(action.result || {}) };\n    //   }\n\n    //   if (action.type === \"jump\") {\n    //     return new Command<any, TStateSchema, string>({\n    //       goto: action.target,\n    //       update: { ...state, ...(action.stateUpdate || {}) },\n    //     });\n    //   }\n\n    //   throw new Error(`Invalid control action: ${JSON.stringify(action)}`);\n    // }\n\n    /**\n     * If result is a state update, merge it with current state\n     */\n    return { ...state, ...result };\n  }\n\n  get nodeOptions(): {\n    input: z.ZodObject<TStateSchema>;\n  } {\n    return {\n      input: derivePrivateState(\n        this.middleware.stateSchema\n      ) as z.ZodObject<TStateSchema>,\n    };\n  }\n}\n"],"mappings":";;;;AAsBA,IAAsB,iBAAtB,cAGU,iBAAyD;CAYjE,MAAM,iBACJA,OACAC,QACmC;;;;EAInC,IAAI,kBAAkB,CAAE;;;;AAIxB,MAAI,KAAK,WAAW,iBAAiB,QAAQ,SAAS;;;;GAIpD,MAAM,cAAc,KAAK,WAAW,eAAe;AACnD,OAAI,aAAa;IACf,MAAMC,kBAA2C,CAAE;AACnD,SAAK,MAAM,OAAO,OAAO,KAAK,YAAY,CACxC,KAAI,OAAO,OAAO,SAChB,gBAAgB,OAAO,OAAO,QAAQ;;;;IAM1C,kBAAkB,KAAK,WAAW,cAAc,MAC9C,gBACD;GACF;EACF;;;;EAKD,MAAMC,UAAmC;GACvC,WAAW,eAAe,MAAM,SAAS;GACzC,aAAa,iBAAiB,MAAM,SAAS;GAC7C,SAAS;EACV;EAED,MAAMC,WAAmC;GACvC,QAAQ,CACNC,QACAC,iBACiC;IACjC,MAAM;IACN;IACA;GACD;GACD,WAAW,CACTC,aACgC;AAChC,QAAIC,oBAAkB,MACpB,OAAMA;AAER,WAAO;KAAE,MAAM;KAAa;IAAQ;GACrC;EACF;EAED,MAAM,SAAS,MAAM,KAAK,QAAQ,OAAO,SAAS,SAAS;;;;AAK3D,MAAI,CAAC,OACH,QAAO;;;;;;;AA6BT,SAAO;GAAE,GAAG;GAAO,GAAG;EAAQ;CAC/B;CAED,IAAI,cAEF;AACA,SAAO,EACL,OAAO,mBACL,KAAK,WAAW,YACjB,CACF;CACF;AACF"}
+{"version":3,"file":"middleware.js","names":["state: TStateSchema","config?: LangGraphRunnableConfig","relevantContext: Record<string, unknown>","runtime: Runtime<TStateSchema, TContextSchema>","result?: Partial<TStateSchema> | Error","result"],"sources":["../../../../src/agents/middlewareAgent/nodes/middleware.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\n/* eslint-disable no-instanceof/no-instanceof */\nimport { z } from \"zod/v3\";\nimport { LangGraphRunnableConfig, Command } from \"@langchain/langgraph\";\nimport { interopParse } from \"@langchain/core/utils/types\";\n\nimport { RunnableCallable } from \"../../RunnableCallable.js\";\nimport type {\n  Runtime,\n  ControlAction,\n  AgentMiddleware,\n  MiddlewareResult,\n  JumpToTarget,\n} from \"../types.js\";\nimport {\n  derivePrivateState,\n  parseToolCalls,\n  parseJumpToTarget,\n} from \"./utils.js\";\n\ntype NodeOutput<TStateSchema extends Record<string, any>> =\n  | TStateSchema\n  | Command<any, TStateSchema, string>;\n\nexport abstract class MiddlewareNode<\n  TStateSchema extends Record<string, any>,\n  TContextSchema extends Record<string, any>\n> extends RunnableCallable<TStateSchema, NodeOutput<TStateSchema>> {\n  abstract middleware: AgentMiddleware<\n    z.ZodObject<z.ZodRawShape>,\n    z.ZodObject<z.ZodRawShape>\n  >;\n\n  abstract runHook(\n    state: TStateSchema,\n    config?: Runtime<TStateSchema, TContextSchema>\n  ): Promise<MiddlewareResult<TStateSchema>>;\n\n  async invokeMiddleware(\n    state: TStateSchema,\n    config?: LangGraphRunnableConfig\n  ): Promise<NodeOutput<TStateSchema>> {\n    /**\n     * Filter context based on middleware's contextSchema\n     */\n    let filteredContext = {} as TContextSchema;\n    /**\n     * Parse context using middleware's contextSchema to apply defaults and validation\n     */\n    if (this.middleware.contextSchema) {\n      /**\n       * Extract only the fields relevant to this middleware's schema\n       */\n      const schemaShape = this.middleware.contextSchema?.shape;\n      if (schemaShape) {\n        const relevantContext: Record<string, unknown> = {};\n        const invokeContext = config?.context || {};\n        for (const key of Object.keys(schemaShape)) {\n          if (key in invokeContext) {\n            relevantContext[key] = invokeContext[key];\n          }\n        }\n        /**\n         * Parse to apply defaults and validation, even if relevantContext is empty\n         * This will throw if required fields are missing and no defaults exist\n         */\n        filteredContext = interopParse(\n          this.middleware.contextSchema,\n          relevantContext\n        ) as TContextSchema;\n      }\n    }\n\n    /**\n     * ToDo: implement later\n     */\n    const runtime: Runtime<TStateSchema, TContextSchema> = {\n      toolCalls: parseToolCalls(state.messages),\n      context: filteredContext,\n      writer: config?.writer,\n      interrupt: config?.interrupt,\n      signal: config?.signal,\n      tools: this.middleware.tools ?? [],\n      terminate: (\n        result?: Partial<TStateSchema> | Error\n      ): ControlAction<TStateSchema> => {\n        if (result instanceof Error) {\n          throw result;\n        }\n        return { type: \"terminate\", result };\n      },\n    };\n\n    const result = await this.runHook(state, runtime);\n\n    /**\n     * If result is undefined, return current state\n     */\n    if (!result) {\n      return { ...state, jumpTo: undefined };\n    }\n\n    /**\n     * Verify that the jump target is allowed for the middleware\n     */\n    const jumpToConstraint = this.name?.startsWith(\"BeforeModelNode_\")\n      ? this.middleware.beforeModelJumpTo\n      : this.middleware.afterModelJumpTo;\n    if (\n      typeof result.jumpTo === \"string\" &&\n      !jumpToConstraint?.includes(result.jumpTo as JumpToTarget)\n    ) {\n      const constraint = this.name?.startsWith(\"BeforeModelNode_\")\n        ? \"beforeModelJumpTo\"\n        : \"afterModelJumpTo\";\n      const suggestion =\n        jumpToConstraint && jumpToConstraint.length > 0\n          ? `must be one of: ${jumpToConstraint?.join(\", \")}.`\n          : `no ${constraint} defined in middleware ${this.middleware.name}.`;\n      throw new Error(`Invalid jump target: ${result.jumpTo}, ${suggestion}.`);\n    }\n\n    const jumpTo = parseJumpToTarget(result.jumpTo as string);\n\n    /**\n     * If result is a control action, handle it\n     */\n    if (typeof result === \"object\" && \"type\" in result) {\n      // Handle control actions\n      if (result.type === \"terminate\") {\n        if (result.error) {\n          throw result.error;\n        }\n        return {\n          ...state,\n          ...(result.result || {}),\n          jumpTo,\n        };\n      }\n\n      throw new Error(`Invalid control action: ${JSON.stringify(result)}`);\n    }\n\n    /**\n     * If result is a state update, merge it with current state\n     */\n    return { ...state, ...result, jumpTo };\n  }\n\n  get nodeOptions(): {\n    input: z.ZodObject<TStateSchema>;\n  } {\n    return {\n      input: derivePrivateState(\n        this.middleware.stateSchema\n      ) as z.ZodObject<TStateSchema>,\n    };\n  }\n}\n"],"mappings":";;;;;AAwBA,IAAsB,iBAAtB,cAGU,iBAAyD;CAWjE,MAAM,iBACJA,OACAC,QACmC;;;;EAInC,IAAI,kBAAkB,CAAE;;;;AAIxB,MAAI,KAAK,WAAW,eAAe;;;;GAIjC,MAAM,cAAc,KAAK,WAAW,eAAe;AACnD,OAAI,aAAa;IACf,MAAMC,kBAA2C,CAAE;IACnD,MAAM,gBAAgB,QAAQ,WAAW,CAAE;AAC3C,SAAK,MAAM,OAAO,OAAO,KAAK,YAAY,CACxC,KAAI,OAAO,eACT,gBAAgB,OAAO,cAAc;;;;;IAOzC,kBAAkB,aAChB,KAAK,WAAW,eAChB,gBACD;GACF;EACF;;;;EAKD,MAAMC,UAAiD;GACrD,WAAW,eAAe,MAAM,SAAS;GACzC,SAAS;GACT,QAAQ,QAAQ;GAChB,WAAW,QAAQ;GACnB,QAAQ,QAAQ;GAChB,OAAO,KAAK,WAAW,SAAS,CAAE;GAClC,WAAW,CACTC,aACgC;AAChC,QAAIC,oBAAkB,MACpB,OAAMA;AAER,WAAO;KAAE,MAAM;KAAa;IAAQ;GACrC;EACF;EAED,MAAM,SAAS,MAAM,KAAK,QAAQ,OAAO,QAAQ;;;;AAKjD,MAAI,CAAC,OACH,QAAO;GAAE,GAAG;GAAO,QAAQ;EAAW;;;;EAMxC,MAAM,mBAAmB,KAAK,MAAM,WAAW,mBAAmB,GAC9D,KAAK,WAAW,oBAChB,KAAK,WAAW;AACpB,MACE,OAAO,OAAO,WAAW,YACzB,CAAC,kBAAkB,SAAS,OAAO,OAAuB,EAC1D;GACA,MAAM,aAAa,KAAK,MAAM,WAAW,mBAAmB,GACxD,sBACA;GACJ,MAAM,aACJ,oBAAoB,iBAAiB,SAAS,IAC1C,CAAC,gBAAgB,EAAE,kBAAkB,KAAK,KAAK,CAAC,CAAC,CAAC,GAClD,CAAC,GAAG,EAAE,WAAW,uBAAuB,EAAE,KAAK,WAAW,KAAK,CAAC,CAAC;AACvE,SAAM,IAAI,MAAM,CAAC,qBAAqB,EAAE,OAAO,OAAO,EAAE,EAAE,WAAW,CAAC,CAAC;EACxE;EAED,MAAM,SAAS,kBAAkB,OAAO,OAAiB;;;;AAKzD,MAAI,OAAO,WAAW,YAAY,UAAU,QAAQ;AAElD,OAAI,OAAO,SAAS,aAAa;AAC/B,QAAI,OAAO,MACT,OAAM,OAAO;AAEf,WAAO;KACL,GAAG;KACH,GAAI,OAAO,UAAU,CAAE;KACvB;IACD;GACF;AAED,SAAM,IAAI,MAAM,CAAC,wBAAwB,EAAE,KAAK,UAAU,OAAO,EAAE;EACpE;;;;AAKD,SAAO;GAAE,GAAG;GAAO,GAAG;GAAQ;EAAQ;CACvC;CAED,IAAI,cAEF;AACA,SAAO,EACL,OAAO,mBACL,KAAK,WAAW,YACjB,CACF;CACF;AACF"}

package/dist/agents/middlewareAgent/nodes/utils.cjs

@@ -1,6 +1,8 @@
 const require_rolldown_runtime = require('../../../_virtual/rolldown_runtime.cjs');
 const __langchain_core_messages = require_rolldown_runtime.__toESM(require("@langchain/core/messages"));
-const zod = require_rolldown_runtime.__toESM(require("zod"));
+const __langchain_langgraph = require_rolldown_runtime.__toESM(require("@langchain/langgraph"));
+const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));
+const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
 
 //#region src/agents/middlewareAgent/nodes/utils.ts
 /**
@@ -10,15 +12,11 @@ const zod = require_rolldown_runtime.__toESM(require("zod"));
 * Private properties (starting with _) are automatically made optional since
 * users cannot provide them when invoking the agent.
 */
-function initializeMiddlewareStates(middlewares, state) {
+async function initializeMiddlewareStates(middlewareList, state) {
 	const middlewareStates = {};
-	for (const middleware of middlewares) if (middleware.stateSchema) {
-		const { shape } = middleware.stateSchema;
-		const modifiedShape = {};
-		for (const [key, value] of Object.entries(shape)) if (key.startsWith("_")) modifiedShape[key] = value.optional();
-		else modifiedShape[key] = value;
-		const modifiedSchema = zod.z.object(modifiedShape);
-		const parseResult = modifiedSchema.safeParse(state);
+	for (const middleware of middlewareList) if (middleware.stateSchema) {
+		const modifiedSchema = (0, __langchain_core_utils_types.interopZodObjectMakeFieldsOptional)(middleware.stateSchema, (key) => key.startsWith("_"));
+		const parseResult = await (0, __langchain_core_utils_types.interopSafeParseAsync)(modifiedSchema, state);
 		if (parseResult.success) {
 			Object.assign(middlewareStates, parseResult.data);
 			continue;
@@ -38,21 +36,30 @@ function initializeMiddlewareStates(middlewares, state) {
 * @returns A new schema containing only the private properties (underscore-prefixed), all made optional
 */
 function derivePrivateState(stateSchema) {
-	const builtInStateSchema = { messages: zod.z.custom(() => []) };
-	if (!stateSchema) return zod.z.object(builtInStateSchema);
+	const builtInStateSchema = { messages: zod_v3.z.custom(() => []) };
+	if (!stateSchema) return zod_v3.z.object(builtInStateSchema);
 	const { shape } = stateSchema;
 	const privateShape = builtInStateSchema;
 	for (const [key, value] of Object.entries(shape)) if (key.startsWith("_")) privateShape[key] = value.optional();
 	else privateShape[key] = value;
-	return zod.z.object(privateShape);
+	return zod_v3.z.object(privateShape);
 }
 /**
-* Parse out all tool calls from the messages
+* Parse out all tool calls from the messages and populate the results
 * @param messages - The messages to parse
 * @returns The tool calls
 */
 function parseToolCalls(messages) {
-	return messages.filter((message) => (0, __langchain_core_messages.isAIMessage)(message) && message.tool_calls).map((message) => message.tool_calls).flat() || [];
+	const calls = messages.filter((message) => __langchain_core_messages.AIMessage.isInstance(message) && message.tool_calls).map((message) => message.tool_calls).flat() || [];
+	const results = parseToolResults(messages);
+	return calls.map((call) => {
+		const callResult = results.find((result) => result.id === call.id);
+		if (callResult) return {
+			...call,
+			result: callResult.result
+		};
+		return call;
+	});
 }
 /**
 * Parse out all tool results from the messages
@@ -60,15 +67,22 @@ function parseToolCalls(messages) {
 * @returns The tool results
 */
 function parseToolResults(messages) {
-	return messages.filter((message) => (0, __langchain_core_messages.isToolMessage)(message)).map((message) => ({
+	return messages.filter((message) => __langchain_core_messages.ToolMessage.isInstance(message)).map((message) => ({
 		id: message.tool_call_id,
 		result: message.content
 	}));
 }
+function parseJumpToTarget(target) {
+	if (!target) return void 0;
+	if (target === "model") return "model_request";
+	if (target === "tools") return "tools";
+	if (target === "end") return __langchain_langgraph.END;
+	throw new Error(`Invalid jump target: ${target}, must be "model", "tools" or "end".`);
+}
 
 //#endregion
 exports.derivePrivateState = derivePrivateState;
 exports.initializeMiddlewareStates = initializeMiddlewareStates;
+exports.parseJumpToTarget = parseJumpToTarget;
 exports.parseToolCalls = parseToolCalls;
-exports.parseToolResults = parseToolResults;
 //# sourceMappingURL=utils.cjs.map

package/dist/agents/middlewareAgent/nodes/utils.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"utils.cjs","names":["middlewares: readonly AgentMiddleware<any, any, any>[]","state: unknown","middlewareStates: Record<string, any>","modifiedShape: Record<string, any>","z","issue: ZodIssue","stateSchema?: z.ZodObject<z.ZodRawShape>","privateShape: Record<string, any>","messages: BaseMessage[]"],"sources":["../../../../src/agents/middlewareAgent/nodes/utils.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  type ToolMessage,\n  type AIMessage,\n  isToolMessage,\n  isAIMessage,\n} from \"@langchain/core/messages\";\nimport { z, type ZodIssue, type ZodTypeAny } from \"zod\";\n\nimport type { AgentMiddleware, ToolCall, ToolResult } from \"../types.js\";\n\n/**\n * Helper function to initialize middleware state defaults.\n * This is used to ensure all middleware state properties are initialized.\n *\n * Private properties (starting with _) are automatically made optional since\n * users cannot provide them when invoking the agent.\n */\nexport function initializeMiddlewareStates(\n  middlewares: readonly AgentMiddleware<any, any, any>[],\n  state: unknown\n): Record<string, any> {\n  const middlewareStates: Record<string, any> = {};\n\n  for (const middleware of middlewares) {\n    if (middleware.stateSchema) {\n      // Create a modified schema where private properties are optional\n      const { shape } = middleware.stateSchema;\n      const modifiedShape: Record<string, any> = {};\n\n      for (const [key, value] of Object.entries(shape)) {\n        if (key.startsWith(\"_\")) {\n          // Make private properties optional\n          modifiedShape[key] = (value as ZodTypeAny).optional();\n        } else {\n          // Keep public properties as-is\n          modifiedShape[key] = value;\n        }\n      }\n\n      const modifiedSchema = z.object(modifiedShape);\n\n      // Use safeParse with the modified schema\n      const parseResult = modifiedSchema.safeParse(state);\n\n      if (parseResult.success) {\n        Object.assign(middlewareStates, parseResult.data);\n        continue;\n      }\n\n      /**\n       * If safeParse fails, there are required public fields missing\n       */\n      const requiredFields = parseResult.error.issues\n        .filter(\n          (issue: ZodIssue) =>\n            issue.code === \"invalid_type\" && issue.message === \"Required\"\n        )\n        .map(\n          (issue: ZodIssue) => `  - ${issue.path.join(\".\")}: ${issue.message}`\n        )\n        .join(\"\\n\");\n\n      throw new Error(\n        `Middleware \"${middleware.name}\" has required state fields that must be initialized:\\n` +\n          `${requiredFields}\\n\\n` +\n          `To fix this, either:\\n` +\n          `1. Provide default values in your middleware's state schema using .default():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().default(\"default value\")\\n` +\n          `   })\\n\\n` +\n          `2. Or make the fields optional using .optional():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().optional()\\n` +\n          `   })\\n\\n` +\n          `3. Or ensure you pass these values when invoking the agent:\\n` +\n          `   agent.invoke({\\n` +\n          `     messages: [...],\\n` +\n          `     ${parseResult.error.issues[0]?.path.join(\".\")}: \"value\"\\n` +\n          `   })`\n      );\n    }\n  }\n\n  return middlewareStates;\n}\n\n/**\n * Users can define private and public state for a middleware. Private state properties start with an underscore.\n * This function will return the private state properties from the state schema, making all of them optional.\n * @param stateSchema - The middleware state schema\n * @returns A new schema containing only the private properties (underscore-prefixed), all made optional\n */\nexport function derivePrivateState(stateSchema?: z.ZodObject<z.ZodRawShape>) {\n  const builtInStateSchema = {\n    messages: z.custom<BaseMessage[]>(() => []),\n  };\n\n  if (!stateSchema) {\n    return z.object(builtInStateSchema);\n  }\n\n  const { shape } = stateSchema;\n  const privateShape: Record<string, any> = builtInStateSchema;\n\n  // Filter properties that start with underscore and make them optional\n  for (const [key, value] of Object.entries(shape)) {\n    if (key.startsWith(\"_\")) {\n      // Make the private property optional\n      privateShape[key] = value.optional();\n    } else {\n      privateShape[key] = value;\n    }\n  }\n\n  // Return a new schema with only private properties (all optional)\n  return z.object(privateShape);\n}\n\n/**\n * Parse out all tool calls from the messages\n * @param messages - The messages to parse\n * @returns The tool calls\n */\nexport function parseToolCalls(messages: BaseMessage[]): ToolCall[] {\n  return (\n    messages\n      .filter(\n        (message) => isAIMessage(message) && (message as AIMessage).tool_calls\n      )\n      .map((message) => (message as AIMessage).tool_calls as ToolCall[])\n      .flat() || []\n  );\n}\n\n/**\n * Parse out all tool results from the messages\n * @param messages - The messages to parse\n * @returns The tool results\n */\nexport function parseToolResults(messages: BaseMessage[]): ToolResult[] {\n  return messages\n    .filter((message) => isToolMessage(message))\n    .map((message) => ({\n      id: (message as ToolMessage).tool_call_id,\n      result: (message as ToolMessage).content,\n    }));\n}\n"],"mappings":";;;;;;;;;;;;AAkBA,SAAgB,2BACdA,aACAC,OACqB;CACrB,MAAMC,mBAAwC,CAAE;AAEhD,MAAK,MAAM,cAAc,YACvB,KAAI,WAAW,aAAa;EAE1B,MAAM,EAAE,OAAO,GAAG,WAAW;EAC7B,MAAMC,gBAAqC,CAAE;AAE7C,OAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,MAAM,CAC9C,KAAI,IAAI,WAAW,IAAI,EAErB,cAAc,OAAQ,MAAqB,UAAU;OAGrD,cAAc,OAAO;EAIzB,MAAM,iBAAiBC,MAAE,OAAO,cAAc;EAG9C,MAAM,cAAc,eAAe,UAAU,MAAM;AAEnD,MAAI,YAAY,SAAS;GACvB,OAAO,OAAO,kBAAkB,YAAY,KAAK;AACjD;EACD;;;;EAKD,MAAM,iBAAiB,YAAY,MAAM,OACtC,OACC,CAACC,UACC,MAAM,SAAS,kBAAkB,MAAM,YAAY,WACtD,CACA,IACC,CAACA,UAAoB,CAAC,IAAI,EAAE,MAAM,KAAK,KAAK,IAAI,CAAC,EAAE,EAAE,MAAM,SAAS,CACrE,CACA,KAAK,KAAK;AAEb,QAAM,IAAI,MACR,CAAC,YAAY,EAAE,WAAW,KAAK,uDAAuD,EACjF,eAAe,0aAAI,EAad,YAAY,MAAM,OAAO,IAAI,KAAK,KAAK,IAAI,CAAC,gBAAW,CACxD;CAEZ;AAGH,QAAO;AACR;;;;;;;AAQD,SAAgB,mBAAmBC,aAA0C;CAC3E,MAAM,qBAAqB,EACzB,UAAUF,MAAE,OAAsB,MAAM,CAAE,EAAC,CAC5C;AAED,KAAI,CAAC,YACH,QAAOA,MAAE,OAAO,mBAAmB;CAGrC,MAAM,EAAE,OAAO,GAAG;CAClB,MAAMG,eAAoC;AAG1C,MAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,MAAM,CAC9C,KAAI,IAAI,WAAW,IAAI,EAErB,aAAa,OAAO,MAAM,UAAU;MAEpC,aAAa,OAAO;AAKxB,QAAOH,MAAE,OAAO,aAAa;AAC9B;;;;;;AAOD,SAAgB,eAAeI,UAAqC;AAClE,QACE,SACG,OACC,CAAC,uDAAwB,QAAQ,IAAK,QAAsB,WAC7D,CACA,IAAI,CAAC,YAAa,QAAsB,WAAyB,CACjE,MAAM,IAAI,CAAE;AAElB;;;;;;AAOD,SAAgB,iBAAiBA,UAAuC;AACtE,QAAO,SACJ,OAAO,CAAC,yDAA0B,QAAQ,CAAC,CAC3C,IAAI,CAAC,aAAa;EACjB,IAAK,QAAwB;EAC7B,QAAS,QAAwB;CAClC,GAAE;AACN"}
+{"version":3,"file":"utils.cjs","names":["middlewareList: readonly AgentMiddleware<any, any, any>[]","state: unknown","middlewareStates: Record<string, any>","issue: ZodIssue","stateSchema?: z.ZodObject<z.ZodRawShape>","z","privateShape: Record<string, any>","messages: BaseMessage[]","AIMessage","ToolMessage","target?: string","END"],"sources":["../../../../src/agents/middlewareAgent/nodes/utils.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { z } from \"zod/v3\";\nimport {\n  type BaseMessage,\n  ToolMessage,\n  AIMessage,\n} from \"@langchain/core/messages\";\nimport {\n  interopSafeParseAsync,\n  interopZodObjectMakeFieldsOptional,\n} from \"@langchain/core/utils/types\";\nimport { type ZodIssue } from \"zod/v3\";\nimport { END } from \"@langchain/langgraph\";\n\nimport type {\n  AgentMiddleware,\n  ToolCall,\n  ToolResult,\n  JumpTo,\n} from \"../types.js\";\n\n/**\n * Helper function to initialize middleware state defaults.\n * This is used to ensure all middleware state properties are initialized.\n *\n * Private properties (starting with _) are automatically made optional since\n * users cannot provide them when invoking the agent.\n */\nexport async function initializeMiddlewareStates(\n  middlewareList: readonly AgentMiddleware<any, any, any>[],\n  state: unknown\n): Promise<Record<string, any>> {\n  const middlewareStates: Record<string, any> = {};\n\n  for (const middleware of middlewareList) {\n    if (middleware.stateSchema) {\n      // Create a modified schema where private properties are optional\n      const modifiedSchema = interopZodObjectMakeFieldsOptional(\n        middleware.stateSchema,\n        (key) => key.startsWith(\"_\")\n      );\n\n      // Use safeParse with the modified schema\n      const parseResult = await interopSafeParseAsync(modifiedSchema, state);\n\n      if (parseResult.success) {\n        Object.assign(middlewareStates, parseResult.data);\n        continue;\n      }\n\n      /**\n       * If safeParse fails, there are required public fields missing\n       */\n      const requiredFields = parseResult.error.issues\n        .filter(\n          (issue: ZodIssue) =>\n            issue.code === \"invalid_type\" && issue.message === \"Required\"\n        )\n        .map(\n          (issue: ZodIssue) => `  - ${issue.path.join(\".\")}: ${issue.message}`\n        )\n        .join(\"\\n\");\n\n      throw new Error(\n        `Middleware \"${middleware.name}\" has required state fields that must be initialized:\\n` +\n          `${requiredFields}\\n\\n` +\n          `To fix this, either:\\n` +\n          `1. Provide default values in your middleware's state schema using .default():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().default(\"default value\")\\n` +\n          `   })\\n\\n` +\n          `2. Or make the fields optional using .optional():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().optional()\\n` +\n          `   })\\n\\n` +\n          `3. Or ensure you pass these values when invoking the agent:\\n` +\n          `   agent.invoke({\\n` +\n          `     messages: [...],\\n` +\n          `     ${parseResult.error.issues[0]?.path.join(\".\")}: \"value\"\\n` +\n          `   })`\n      );\n    }\n  }\n\n  return middlewareStates;\n}\n\n/**\n * Users can define private and public state for a middleware. Private state properties start with an underscore.\n * This function will return the private state properties from the state schema, making all of them optional.\n * @param stateSchema - The middleware state schema\n * @returns A new schema containing only the private properties (underscore-prefixed), all made optional\n */\nexport function derivePrivateState(\n  stateSchema?: z.ZodObject<z.ZodRawShape>\n): z.ZodObject<z.ZodRawShape> {\n  const builtInStateSchema = {\n    messages: z.custom<BaseMessage[]>(() => []),\n  };\n\n  if (!stateSchema) {\n    return z.object(builtInStateSchema);\n  }\n\n  const { shape } = stateSchema;\n  const privateShape: Record<string, any> = builtInStateSchema;\n\n  // Filter properties that start with underscore and make them optional\n  for (const [key, value] of Object.entries(shape)) {\n    if (key.startsWith(\"_\")) {\n      // Make the private property optional\n      privateShape[key] = value.optional();\n    } else {\n      privateShape[key] = value;\n    }\n  }\n\n  // Return a new schema with only private properties (all optional)\n  return z.object(privateShape);\n}\n\n/**\n * Parse out all tool calls from the messages and populate the results\n * @param messages - The messages to parse\n * @returns The tool calls\n */\nexport function parseToolCalls(messages: BaseMessage[]): ToolCall[] {\n  const calls =\n    messages\n      .filter(\n        (message) =>\n          AIMessage.isInstance(message) && (message as AIMessage).tool_calls\n      )\n      .map((message) => (message as AIMessage).tool_calls as ToolCall[])\n      .flat() || [];\n\n  const results = parseToolResults(messages);\n  return calls.map((call) => {\n    const callResult = results.find((result) => result.id === call.id);\n    if (callResult) {\n      return {\n        ...call,\n        result: callResult.result,\n      };\n    }\n    return call;\n  });\n}\n\n/**\n * Parse out all tool results from the messages\n * @param messages - The messages to parse\n * @returns The tool results\n */\nfunction parseToolResults(messages: BaseMessage[]): ToolResult[] {\n  return messages\n    .filter((message) => ToolMessage.isInstance(message))\n    .map((message) => ({\n      id: (message as ToolMessage).tool_call_id,\n      result: (message as ToolMessage).content,\n    }));\n}\n\n/**\n * Parse `jumpTo` target from user facing labels to a LangGraph node names\n */\nexport function parseJumpToTarget(target: string): JumpTo;\nexport function parseJumpToTarget(target?: string): JumpTo | undefined {\n  if (!target) {\n    return undefined;\n  }\n  if (target === \"model\") {\n    return \"model_request\";\n  }\n  if (target === \"tools\") {\n    return \"tools\";\n  }\n  if (target === \"end\") {\n    return END;\n  }\n\n  throw new Error(\n    `Invalid jump target: ${target}, must be \"model\", \"tools\" or \"end\".`\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;AA4BA,eAAsB,2BACpBA,gBACAC,OAC8B;CAC9B,MAAMC,mBAAwC,CAAE;AAEhD,MAAK,MAAM,cAAc,eACvB,KAAI,WAAW,aAAa;EAE1B,MAAM,sFACJ,WAAW,aACX,CAAC,QAAQ,IAAI,WAAW,IAAI,CAC7B;EAGD,MAAM,cAAc,8DAA4B,gBAAgB,MAAM;AAEtE,MAAI,YAAY,SAAS;GACvB,OAAO,OAAO,kBAAkB,YAAY,KAAK;AACjD;EACD;;;;EAKD,MAAM,iBAAiB,YAAY,MAAM,OACtC,OACC,CAACC,UACC,MAAM,SAAS,kBAAkB,MAAM,YAAY,WACtD,CACA,IACC,CAACA,UAAoB,CAAC,IAAI,EAAE,MAAM,KAAK,KAAK,IAAI,CAAC,EAAE,EAAE,MAAM,SAAS,CACrE,CACA,KAAK,KAAK;AAEb,QAAM,IAAI,MACR,CAAC,YAAY,EAAE,WAAW,KAAK,uDAAuD,EACjF,eAAe,0aAAI,EAad,YAAY,MAAM,OAAO,IAAI,KAAK,KAAK,IAAI,CAAC,gBAAW,CACxD;CAEZ;AAGH,QAAO;AACR;;;;;;;AAQD,SAAgB,mBACdC,aAC4B;CAC5B,MAAM,qBAAqB,EACzB,UAAUC,SAAE,OAAsB,MAAM,CAAE,EAAC,CAC5C;AAED,KAAI,CAAC,YACH,QAAOA,SAAE,OAAO,mBAAmB;CAGrC,MAAM,EAAE,OAAO,GAAG;CAClB,MAAMC,eAAoC;AAG1C,MAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,MAAM,CAC9C,KAAI,IAAI,WAAW,IAAI,EAErB,aAAa,OAAO,MAAM,UAAU;MAEpC,aAAa,OAAO;AAKxB,QAAOD,SAAE,OAAO,aAAa;AAC9B;;;;;;AAOD,SAAgB,eAAeE,UAAqC;CAClE,MAAM,QACJ,SACG,OACC,CAAC,YACCC,oCAAU,WAAW,QAAQ,IAAK,QAAsB,WAC3D,CACA,IAAI,CAAC,YAAa,QAAsB,WAAyB,CACjE,MAAM,IAAI,CAAE;CAEjB,MAAM,UAAU,iBAAiB,SAAS;AAC1C,QAAO,MAAM,IAAI,CAAC,SAAS;EACzB,MAAM,aAAa,QAAQ,KAAK,CAAC,WAAW,OAAO,OAAO,KAAK,GAAG;AAClE,MAAI,WACF,QAAO;GACL,GAAG;GACH,QAAQ,WAAW;EACpB;AAEH,SAAO;CACR,EAAC;AACH;;;;;;AAOD,SAAS,iBAAiBD,UAAuC;AAC/D,QAAO,SACJ,OAAO,CAAC,YAAYE,sCAAY,WAAW,QAAQ,CAAC,CACpD,IAAI,CAAC,aAAa;EACjB,IAAK,QAAwB;EAC7B,QAAS,QAAwB;CAClC,GAAE;AACN;AAMD,SAAgB,kBAAkBC,QAAqC;AACrE,KAAI,CAAC,OACH,QAAO;AAET,KAAI,WAAW,QACb,QAAO;AAET,KAAI,WAAW,QACb,QAAO;AAET,KAAI,WAAW,MACb,QAAOC;AAGT,OAAM,IAAI,MACR,CAAC,qBAAqB,EAAE,OAAO,oCAAoC,CAAC;AAEvE"}

package/dist/agents/middlewareAgent/nodes/utils.js

@@ -1,5 +1,7 @@
-import { isAIMessage, isToolMessage } from "@langchain/core/messages";
-import { z } from "zod";
+import { AIMessage, ToolMessage } from "@langchain/core/messages";
+import { END } from "@langchain/langgraph";
+import { interopSafeParseAsync, interopZodObjectMakeFieldsOptional } from "@langchain/core/utils/types";
+import { z } from "zod/v3";
 
 //#region src/agents/middlewareAgent/nodes/utils.ts
 /**
@@ -9,15 +11,11 @@ import { z } from "zod";
 * Private properties (starting with _) are automatically made optional since
 * users cannot provide them when invoking the agent.
 */
-function initializeMiddlewareStates(middlewares, state) {
+async function initializeMiddlewareStates(middlewareList, state) {
 	const middlewareStates = {};
-	for (const middleware of middlewares) if (middleware.stateSchema) {
-		const { shape } = middleware.stateSchema;
-		const modifiedShape = {};
-		for (const [key, value] of Object.entries(shape)) if (key.startsWith("_")) modifiedShape[key] = value.optional();
-		else modifiedShape[key] = value;
-		const modifiedSchema = z.object(modifiedShape);
-		const parseResult = modifiedSchema.safeParse(state);
+	for (const middleware of middlewareList) if (middleware.stateSchema) {
+		const modifiedSchema = interopZodObjectMakeFieldsOptional(middleware.stateSchema, (key) => key.startsWith("_"));
+		const parseResult = await interopSafeParseAsync(modifiedSchema, state);
 		if (parseResult.success) {
 			Object.assign(middlewareStates, parseResult.data);
 			continue;
@@ -46,12 +44,21 @@ function derivePrivateState(stateSchema) {
 	return z.object(privateShape);
 }
 /**
-* Parse out all tool calls from the messages
+* Parse out all tool calls from the messages and populate the results
 * @param messages - The messages to parse
 * @returns The tool calls
 */
 function parseToolCalls(messages) {
-	return messages.filter((message) => isAIMessage(message) && message.tool_calls).map((message) => message.tool_calls).flat() || [];
+	const calls = messages.filter((message) => AIMessage.isInstance(message) && message.tool_calls).map((message) => message.tool_calls).flat() || [];
+	const results = parseToolResults(messages);
+	return calls.map((call) => {
+		const callResult = results.find((result) => result.id === call.id);
+		if (callResult) return {
+			...call,
+			result: callResult.result
+		};
+		return call;
+	});
 }
 /**
 * Parse out all tool results from the messages
@@ -59,12 +66,19 @@ function parseToolCalls(messages) {
 * @returns The tool results
 */
 function parseToolResults(messages) {
-	return messages.filter((message) => isToolMessage(message)).map((message) => ({
+	return messages.filter((message) => ToolMessage.isInstance(message)).map((message) => ({
 		id: message.tool_call_id,
 		result: message.content
 	}));
 }
+function parseJumpToTarget(target) {
+	if (!target) return void 0;
+	if (target === "model") return "model_request";
+	if (target === "tools") return "tools";
+	if (target === "end") return END;
+	throw new Error(`Invalid jump target: ${target}, must be "model", "tools" or "end".`);
+}
 
 //#endregion
-export { derivePrivateState, initializeMiddlewareStates, parseToolCalls, parseToolResults };
+export { derivePrivateState, initializeMiddlewareStates, parseJumpToTarget, parseToolCalls };
 //# sourceMappingURL=utils.js.map

package/dist/agents/middlewareAgent/nodes/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","names":["middlewares: readonly AgentMiddleware<any, any, any>[]","state: unknown","middlewareStates: Record<string, any>","modifiedShape: Record<string, any>","issue: ZodIssue","stateSchema?: z.ZodObject<z.ZodRawShape>","privateShape: Record<string, any>","messages: BaseMessage[]"],"sources":["../../../../src/agents/middlewareAgent/nodes/utils.ts"],"sourcesContent":["import {\n  type BaseMessage,\n  type ToolMessage,\n  type AIMessage,\n  isToolMessage,\n  isAIMessage,\n} from \"@langchain/core/messages\";\nimport { z, type ZodIssue, type ZodTypeAny } from \"zod\";\n\nimport type { AgentMiddleware, ToolCall, ToolResult } from \"../types.js\";\n\n/**\n * Helper function to initialize middleware state defaults.\n * This is used to ensure all middleware state properties are initialized.\n *\n * Private properties (starting with _) are automatically made optional since\n * users cannot provide them when invoking the agent.\n */\nexport function initializeMiddlewareStates(\n  middlewares: readonly AgentMiddleware<any, any, any>[],\n  state: unknown\n): Record<string, any> {\n  const middlewareStates: Record<string, any> = {};\n\n  for (const middleware of middlewares) {\n    if (middleware.stateSchema) {\n      // Create a modified schema where private properties are optional\n      const { shape } = middleware.stateSchema;\n      const modifiedShape: Record<string, any> = {};\n\n      for (const [key, value] of Object.entries(shape)) {\n        if (key.startsWith(\"_\")) {\n          // Make private properties optional\n          modifiedShape[key] = (value as ZodTypeAny).optional();\n        } else {\n          // Keep public properties as-is\n          modifiedShape[key] = value;\n        }\n      }\n\n      const modifiedSchema = z.object(modifiedShape);\n\n      // Use safeParse with the modified schema\n      const parseResult = modifiedSchema.safeParse(state);\n\n      if (parseResult.success) {\n        Object.assign(middlewareStates, parseResult.data);\n        continue;\n      }\n\n      /**\n       * If safeParse fails, there are required public fields missing\n       */\n      const requiredFields = parseResult.error.issues\n        .filter(\n          (issue: ZodIssue) =>\n            issue.code === \"invalid_type\" && issue.message === \"Required\"\n        )\n        .map(\n          (issue: ZodIssue) => `  - ${issue.path.join(\".\")}: ${issue.message}`\n        )\n        .join(\"\\n\");\n\n      throw new Error(\n        `Middleware \"${middleware.name}\" has required state fields that must be initialized:\\n` +\n          `${requiredFields}\\n\\n` +\n          `To fix this, either:\\n` +\n          `1. Provide default values in your middleware's state schema using .default():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().default(\"default value\")\\n` +\n          `   })\\n\\n` +\n          `2. Or make the fields optional using .optional():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().optional()\\n` +\n          `   })\\n\\n` +\n          `3. Or ensure you pass these values when invoking the agent:\\n` +\n          `   agent.invoke({\\n` +\n          `     messages: [...],\\n` +\n          `     ${parseResult.error.issues[0]?.path.join(\".\")}: \"value\"\\n` +\n          `   })`\n      );\n    }\n  }\n\n  return middlewareStates;\n}\n\n/**\n * Users can define private and public state for a middleware. Private state properties start with an underscore.\n * This function will return the private state properties from the state schema, making all of them optional.\n * @param stateSchema - The middleware state schema\n * @returns A new schema containing only the private properties (underscore-prefixed), all made optional\n */\nexport function derivePrivateState(stateSchema?: z.ZodObject<z.ZodRawShape>) {\n  const builtInStateSchema = {\n    messages: z.custom<BaseMessage[]>(() => []),\n  };\n\n  if (!stateSchema) {\n    return z.object(builtInStateSchema);\n  }\n\n  const { shape } = stateSchema;\n  const privateShape: Record<string, any> = builtInStateSchema;\n\n  // Filter properties that start with underscore and make them optional\n  for (const [key, value] of Object.entries(shape)) {\n    if (key.startsWith(\"_\")) {\n      // Make the private property optional\n      privateShape[key] = value.optional();\n    } else {\n      privateShape[key] = value;\n    }\n  }\n\n  // Return a new schema with only private properties (all optional)\n  return z.object(privateShape);\n}\n\n/**\n * Parse out all tool calls from the messages\n * @param messages - The messages to parse\n * @returns The tool calls\n */\nexport function parseToolCalls(messages: BaseMessage[]): ToolCall[] {\n  return (\n    messages\n      .filter(\n        (message) => isAIMessage(message) && (message as AIMessage).tool_calls\n      )\n      .map((message) => (message as AIMessage).tool_calls as ToolCall[])\n      .flat() || []\n  );\n}\n\n/**\n * Parse out all tool results from the messages\n * @param messages - The messages to parse\n * @returns The tool results\n */\nexport function parseToolResults(messages: BaseMessage[]): ToolResult[] {\n  return messages\n    .filter((message) => isToolMessage(message))\n    .map((message) => ({\n      id: (message as ToolMessage).tool_call_id,\n      result: (message as ToolMessage).content,\n    }));\n}\n"],"mappings":";;;;;;;;;;;AAkBA,SAAgB,2BACdA,aACAC,OACqB;CACrB,MAAMC,mBAAwC,CAAE;AAEhD,MAAK,MAAM,cAAc,YACvB,KAAI,WAAW,aAAa;EAE1B,MAAM,EAAE,OAAO,GAAG,WAAW;EAC7B,MAAMC,gBAAqC,CAAE;AAE7C,OAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,MAAM,CAC9C,KAAI,IAAI,WAAW,IAAI,EAErB,cAAc,OAAQ,MAAqB,UAAU;OAGrD,cAAc,OAAO;EAIzB,MAAM,iBAAiB,EAAE,OAAO,cAAc;EAG9C,MAAM,cAAc,eAAe,UAAU,MAAM;AAEnD,MAAI,YAAY,SAAS;GACvB,OAAO,OAAO,kBAAkB,YAAY,KAAK;AACjD;EACD;;;;EAKD,MAAM,iBAAiB,YAAY,MAAM,OACtC,OACC,CAACC,UACC,MAAM,SAAS,kBAAkB,MAAM,YAAY,WACtD,CACA,IACC,CAACA,UAAoB,CAAC,IAAI,EAAE,MAAM,KAAK,KAAK,IAAI,CAAC,EAAE,EAAE,MAAM,SAAS,CACrE,CACA,KAAK,KAAK;AAEb,QAAM,IAAI,MACR,CAAC,YAAY,EAAE,WAAW,KAAK,uDAAuD,EACjF,eAAe,0aAAI,EAad,YAAY,MAAM,OAAO,IAAI,KAAK,KAAK,IAAI,CAAC,gBAAW,CACxD;CAEZ;AAGH,QAAO;AACR;;;;;;;AAQD,SAAgB,mBAAmBC,aAA0C;CAC3E,MAAM,qBAAqB,EACzB,UAAU,EAAE,OAAsB,MAAM,CAAE,EAAC,CAC5C;AAED,KAAI,CAAC,YACH,QAAO,EAAE,OAAO,mBAAmB;CAGrC,MAAM,EAAE,OAAO,GAAG;CAClB,MAAMC,eAAoC;AAG1C,MAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,MAAM,CAC9C,KAAI,IAAI,WAAW,IAAI,EAErB,aAAa,OAAO,MAAM,UAAU;MAEpC,aAAa,OAAO;AAKxB,QAAO,EAAE,OAAO,aAAa;AAC9B;;;;;;AAOD,SAAgB,eAAeC,UAAqC;AAClE,QACE,SACG,OACC,CAAC,YAAY,YAAY,QAAQ,IAAK,QAAsB,WAC7D,CACA,IAAI,CAAC,YAAa,QAAsB,WAAyB,CACjE,MAAM,IAAI,CAAE;AAElB;;;;;;AAOD,SAAgB,iBAAiBA,UAAuC;AACtE,QAAO,SACJ,OAAO,CAAC,YAAY,cAAc,QAAQ,CAAC,CAC3C,IAAI,CAAC,aAAa;EACjB,IAAK,QAAwB;EAC7B,QAAS,QAAwB;CAClC,GAAE;AACN"}
+{"version":3,"file":"utils.js","names":["middlewareList: readonly AgentMiddleware<any, any, any>[]","state: unknown","middlewareStates: Record<string, any>","issue: ZodIssue","stateSchema?: z.ZodObject<z.ZodRawShape>","privateShape: Record<string, any>","messages: BaseMessage[]","target?: string"],"sources":["../../../../src/agents/middlewareAgent/nodes/utils.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { z } from \"zod/v3\";\nimport {\n  type BaseMessage,\n  ToolMessage,\n  AIMessage,\n} from \"@langchain/core/messages\";\nimport {\n  interopSafeParseAsync,\n  interopZodObjectMakeFieldsOptional,\n} from \"@langchain/core/utils/types\";\nimport { type ZodIssue } from \"zod/v3\";\nimport { END } from \"@langchain/langgraph\";\n\nimport type {\n  AgentMiddleware,\n  ToolCall,\n  ToolResult,\n  JumpTo,\n} from \"../types.js\";\n\n/**\n * Helper function to initialize middleware state defaults.\n * This is used to ensure all middleware state properties are initialized.\n *\n * Private properties (starting with _) are automatically made optional since\n * users cannot provide them when invoking the agent.\n */\nexport async function initializeMiddlewareStates(\n  middlewareList: readonly AgentMiddleware<any, any, any>[],\n  state: unknown\n): Promise<Record<string, any>> {\n  const middlewareStates: Record<string, any> = {};\n\n  for (const middleware of middlewareList) {\n    if (middleware.stateSchema) {\n      // Create a modified schema where private properties are optional\n      const modifiedSchema = interopZodObjectMakeFieldsOptional(\n        middleware.stateSchema,\n        (key) => key.startsWith(\"_\")\n      );\n\n      // Use safeParse with the modified schema\n      const parseResult = await interopSafeParseAsync(modifiedSchema, state);\n\n      if (parseResult.success) {\n        Object.assign(middlewareStates, parseResult.data);\n        continue;\n      }\n\n      /**\n       * If safeParse fails, there are required public fields missing\n       */\n      const requiredFields = parseResult.error.issues\n        .filter(\n          (issue: ZodIssue) =>\n            issue.code === \"invalid_type\" && issue.message === \"Required\"\n        )\n        .map(\n          (issue: ZodIssue) => `  - ${issue.path.join(\".\")}: ${issue.message}`\n        )\n        .join(\"\\n\");\n\n      throw new Error(\n        `Middleware \"${middleware.name}\" has required state fields that must be initialized:\\n` +\n          `${requiredFields}\\n\\n` +\n          `To fix this, either:\\n` +\n          `1. Provide default values in your middleware's state schema using .default():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().default(\"default value\")\\n` +\n          `   })\\n\\n` +\n          `2. Or make the fields optional using .optional():\\n` +\n          `   stateSchema: z.object({\\n` +\n          `     myField: z.string().optional()\\n` +\n          `   })\\n\\n` +\n          `3. Or ensure you pass these values when invoking the agent:\\n` +\n          `   agent.invoke({\\n` +\n          `     messages: [...],\\n` +\n          `     ${parseResult.error.issues[0]?.path.join(\".\")}: \"value\"\\n` +\n          `   })`\n      );\n    }\n  }\n\n  return middlewareStates;\n}\n\n/**\n * Users can define private and public state for a middleware. Private state properties start with an underscore.\n * This function will return the private state properties from the state schema, making all of them optional.\n * @param stateSchema - The middleware state schema\n * @returns A new schema containing only the private properties (underscore-prefixed), all made optional\n */\nexport function derivePrivateState(\n  stateSchema?: z.ZodObject<z.ZodRawShape>\n): z.ZodObject<z.ZodRawShape> {\n  const builtInStateSchema = {\n    messages: z.custom<BaseMessage[]>(() => []),\n  };\n\n  if (!stateSchema) {\n    return z.object(builtInStateSchema);\n  }\n\n  const { shape } = stateSchema;\n  const privateShape: Record<string, any> = builtInStateSchema;\n\n  // Filter properties that start with underscore and make them optional\n  for (const [key, value] of Object.entries(shape)) {\n    if (key.startsWith(\"_\")) {\n      // Make the private property optional\n      privateShape[key] = value.optional();\n    } else {\n      privateShape[key] = value;\n    }\n  }\n\n  // Return a new schema with only private properties (all optional)\n  return z.object(privateShape);\n}\n\n/**\n * Parse out all tool calls from the messages and populate the results\n * @param messages - The messages to parse\n * @returns The tool calls\n */\nexport function parseToolCalls(messages: BaseMessage[]): ToolCall[] {\n  const calls =\n    messages\n      .filter(\n        (message) =>\n          AIMessage.isInstance(message) && (message as AIMessage).tool_calls\n      )\n      .map((message) => (message as AIMessage).tool_calls as ToolCall[])\n      .flat() || [];\n\n  const results = parseToolResults(messages);\n  return calls.map((call) => {\n    const callResult = results.find((result) => result.id === call.id);\n    if (callResult) {\n      return {\n        ...call,\n        result: callResult.result,\n      };\n    }\n    return call;\n  });\n}\n\n/**\n * Parse out all tool results from the messages\n * @param messages - The messages to parse\n * @returns The tool results\n */\nfunction parseToolResults(messages: BaseMessage[]): ToolResult[] {\n  return messages\n    .filter((message) => ToolMessage.isInstance(message))\n    .map((message) => ({\n      id: (message as ToolMessage).tool_call_id,\n      result: (message as ToolMessage).content,\n    }));\n}\n\n/**\n * Parse `jumpTo` target from user facing labels to a LangGraph node names\n */\nexport function parseJumpToTarget(target: string): JumpTo;\nexport function parseJumpToTarget(target?: string): JumpTo | undefined {\n  if (!target) {\n    return undefined;\n  }\n  if (target === \"model\") {\n    return \"model_request\";\n  }\n  if (target === \"tools\") {\n    return \"tools\";\n  }\n  if (target === \"end\") {\n    return END;\n  }\n\n  throw new Error(\n    `Invalid jump target: ${target}, must be \"model\", \"tools\" or \"end\".`\n  );\n}\n"],"mappings":";;;;;;;;;;;;;AA4BA,eAAsB,2BACpBA,gBACAC,OAC8B;CAC9B,MAAMC,mBAAwC,CAAE;AAEhD,MAAK,MAAM,cAAc,eACvB,KAAI,WAAW,aAAa;EAE1B,MAAM,iBAAiB,mCACrB,WAAW,aACX,CAAC,QAAQ,IAAI,WAAW,IAAI,CAC7B;EAGD,MAAM,cAAc,MAAM,sBAAsB,gBAAgB,MAAM;AAEtE,MAAI,YAAY,SAAS;GACvB,OAAO,OAAO,kBAAkB,YAAY,KAAK;AACjD;EACD;;;;EAKD,MAAM,iBAAiB,YAAY,MAAM,OACtC,OACC,CAACC,UACC,MAAM,SAAS,kBAAkB,MAAM,YAAY,WACtD,CACA,IACC,CAACA,UAAoB,CAAC,IAAI,EAAE,MAAM,KAAK,KAAK,IAAI,CAAC,EAAE,EAAE,MAAM,SAAS,CACrE,CACA,KAAK,KAAK;AAEb,QAAM,IAAI,MACR,CAAC,YAAY,EAAE,WAAW,KAAK,uDAAuD,EACjF,eAAe,0aAAI,EAad,YAAY,MAAM,OAAO,IAAI,KAAK,KAAK,IAAI,CAAC,gBAAW,CACxD;CAEZ;AAGH,QAAO;AACR;;;;;;;AAQD,SAAgB,mBACdC,aAC4B;CAC5B,MAAM,qBAAqB,EACzB,UAAU,EAAE,OAAsB,MAAM,CAAE,EAAC,CAC5C;AAED,KAAI,CAAC,YACH,QAAO,EAAE,OAAO,mBAAmB;CAGrC,MAAM,EAAE,OAAO,GAAG;CAClB,MAAMC,eAAoC;AAG1C,MAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,MAAM,CAC9C,KAAI,IAAI,WAAW,IAAI,EAErB,aAAa,OAAO,MAAM,UAAU;MAEpC,aAAa,OAAO;AAKxB,QAAO,EAAE,OAAO,aAAa;AAC9B;;;;;;AAOD,SAAgB,eAAeC,UAAqC;CAClE,MAAM,QACJ,SACG,OACC,CAAC,YACC,UAAU,WAAW,QAAQ,IAAK,QAAsB,WAC3D,CACA,IAAI,CAAC,YAAa,QAAsB,WAAyB,CACjE,MAAM,IAAI,CAAE;CAEjB,MAAM,UAAU,iBAAiB,SAAS;AAC1C,QAAO,MAAM,IAAI,CAAC,SAAS;EACzB,MAAM,aAAa,QAAQ,KAAK,CAAC,WAAW,OAAO,OAAO,KAAK,GAAG;AAClE,MAAI,WACF,QAAO;GACL,GAAG;GACH,QAAQ,WAAW;EACpB;AAEH,SAAO;CACR,EAAC;AACH;;;;;;AAOD,SAAS,iBAAiBA,UAAuC;AAC/D,QAAO,SACJ,OAAO,CAAC,YAAY,YAAY,WAAW,QAAQ,CAAC,CACpD,IAAI,CAAC,aAAa;EACjB,IAAK,QAAwB;EAC7B,QAAS,QAAwB;CAClC,GAAE;AACN;AAMD,SAAgB,kBAAkBC,QAAqC;AACrE,KAAI,CAAC,OACH,QAAO;AAET,KAAI,WAAW,QACb,QAAO;AAET,KAAI,WAAW,QACb,QAAO;AAET,KAAI,WAAW,MACb,QAAO;AAGT,OAAM,IAAI,MACR,CAAC,qBAAqB,EAAE,OAAO,oCAAoC,CAAC;AAEvE"}