langchain 1.0.0-alpha.7 → 1.0.0-alpha.9

package/dist/agents/middlewareAgent/ReactAgent.js

@@ -1,461 +0,0 @@
-import { isClientTool } from "../utils.js";
-import { ToolNode } from "../nodes/ToolNode.js";
-import { createAgentAnnotationConditional } from "./annotation.js";
-import { initializeMiddlewareStates, parseJumpToTarget } from "./nodes/utils.js";
-import { AgentNode } from "./nodes/AgentNode.js";
-import { BeforeModelNode } from "./nodes/BeforeModalNode.js";
-import { AfterModelNode } from "./nodes/AfterModalNode.js";
-import { AIMessage, ToolMessage } from "@langchain/core/messages";
-import { Command, END, START, Send, StateGraph } from "@langchain/langgraph";
-
-//#region src/agents/middlewareAgent/ReactAgent.ts
-var ReactAgent = class {
-	#graph;
-	#toolBehaviorVersion = "v2";
-	constructor(options) {
-		this.options = options;
-		this.#toolBehaviorVersion = options.version ?? this.#toolBehaviorVersion;
-		/**
-		* define complete list of tools based on options and middleware
-		*/
-		const middlewareTools = this.options.middleware?.filter((m) => m.tools).flatMap((m) => m.tools) ?? [];
-		const toolClasses = [...(Array.isArray(options.tools) ? options.tools : options.tools?.tools) ?? [], ...middlewareTools];
-		/**
-		* If any of the tools are configured to return_directly after running,
-		* our graph needs to check if these were called
-		*/
-		const shouldReturnDirect = new Set(toolClasses.filter(isClientTool).filter((tool) => "returnDirect" in tool && tool.returnDirect).map((tool) => tool.name));
-		const schema = createAgentAnnotationConditional(this.options.responseFormat !== void 0, this.options.middleware);
-		const workflow = new StateGraph(schema, this.options.contextSchema);
-		const allNodeWorkflows = workflow;
-		const beforeModelNodes = [];
-		const afterModelNodes = [];
-		const modifyModelRequestHookMiddleware = [];
-		const middlewareNames = /* @__PURE__ */ new Set();
-		const middleware = this.options.middleware ?? [];
-		for (let i = 0; i < middleware.length; i++) {
-			let beforeModelNode;
-			let afterModelNode;
-			const m = middleware[i];
-			if (middlewareNames.has(m.name)) throw new Error(`Middleware ${m.name} is defined multiple times`);
-			middlewareNames.add(m.name);
-			if (m.beforeModel) {
-				beforeModelNode = new BeforeModelNode(m);
-				const name = `${m.name}.before_model`;
-				beforeModelNodes.push({
-					index: i,
-					name,
-					allowed: m.beforeModelJumpTo
-				});
-				allNodeWorkflows.addNode(name, beforeModelNode, beforeModelNode.nodeOptions);
-			}
-			if (m.afterModel) {
-				afterModelNode = new AfterModelNode(m);
-				const name = `${m.name}.after_model`;
-				afterModelNodes.push({
-					index: i,
-					name,
-					allowed: m.afterModelJumpTo
-				});
-				allNodeWorkflows.addNode(name, afterModelNode, afterModelNode.nodeOptions);
-			}
-			if (m.modifyModelRequest) modifyModelRequestHookMiddleware.push([m, () => ({
-				...beforeModelNode?.getState(),
-				...afterModelNode?.getState()
-			})]);
-		}
-		/**
-		* Add Nodes
-		*/
-		allNodeWorkflows.addNode("model_request", new AgentNode({
-			model: this.options.model,
-			systemPrompt: this.options.systemPrompt,
-			includeAgentName: this.options.includeAgentName,
-			name: this.options.name,
-			responseFormat: this.options.responseFormat,
-			middleware: this.options.middleware,
-			toolClasses,
-			shouldReturnDirect,
-			signal: this.options.signal,
-			modifyModelRequestHookMiddleware
-		}), AgentNode.nodeOptions);
-		/**
-		* add single tool node for all tools
-		*/
-		if (toolClasses.length > 0) {
-			const toolNode = new ToolNode(toolClasses.filter(isClientTool), { signal: this.options.signal });
-			allNodeWorkflows.addNode("tools", toolNode);
-		}
-		/**
-		* Add Edges
-		*/
-		if (beforeModelNodes.length > 0) allNodeWorkflows.addEdge(START, beforeModelNodes[0].name);
-		else allNodeWorkflows.addEdge(START, "model_request");
-		for (let i = 0; i < beforeModelNodes.length; i++) {
-			const node = beforeModelNodes[i];
-			const current = node.name;
-			const isLast = i === beforeModelNodes.length - 1;
-			const nextDefault = isLast ? "model_request" : beforeModelNodes[i + 1].name;
-			if (node.allowed && node.allowed.length > 0) {
-				const hasTools = toolClasses.filter(isClientTool).length > 0;
-				const allowedMapped = node.allowed.map((t) => parseJumpToTarget(t)).filter((dest) => dest !== "tools" || hasTools);
-				const destinations = Array.from(new Set([nextDefault, ...allowedMapped]));
-				allNodeWorkflows.addConditionalEdges(current, this.#createBeforeModelRouter(toolClasses.filter(isClientTool), nextDefault), destinations);
-			} else allNodeWorkflows.addEdge(current, nextDefault);
-		}
-		const lastAfterModelNode = afterModelNodes.at(-1);
-		if (afterModelNodes.length > 0 && lastAfterModelNode) allNodeWorkflows.addEdge("model_request", lastAfterModelNode.name);
-		else {
-			const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));
-			if (modelPaths.length === 1) allNodeWorkflows.addEdge("model_request", modelPaths[0]);
-			else allNodeWorkflows.addConditionalEdges("model_request", this.#createModelRouter(), modelPaths);
-		}
-		for (let i = afterModelNodes.length - 1; i > 0; i--) {
-			const node = afterModelNodes[i];
-			const current = node.name;
-			const nextDefault = afterModelNodes[i - 1].name;
-			if (node.allowed && node.allowed.length > 0) {
-				const hasTools = toolClasses.filter(isClientTool).length > 0;
-				const allowedMapped = node.allowed.map((t) => parseJumpToTarget(t)).filter((dest) => dest !== "tools" || hasTools);
-				const destinations = Array.from(new Set([nextDefault, ...allowedMapped]));
-				allNodeWorkflows.addConditionalEdges(current, this.#createAfterModelSequenceRouter(toolClasses.filter(isClientTool), node.allowed, nextDefault), destinations);
-			} else allNodeWorkflows.addEdge(current, nextDefault);
-		}
-		if (afterModelNodes.length > 0) {
-			const firstAfterModel = afterModelNodes[0];
-			const firstAfterModelNode = firstAfterModel.name;
-			const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool), true).filter((p) => p !== "tools" || toolClasses.filter(isClientTool).length > 0);
-			const allowJump = Boolean(firstAfterModel.allowed && firstAfterModel.allowed.length > 0);
-			const destinations = modelPaths;
-			allNodeWorkflows.addConditionalEdges(firstAfterModelNode, this.#createAfterModelRouter(toolClasses.filter(isClientTool), allowJump), destinations);
-		}
-		/**
-		* add edges for tools node
-		*/
-		if (toolClasses.length > 0) {
-			let toolReturnTarget;
-			if (beforeModelNodes.length > 0) toolReturnTarget = beforeModelNodes[0].name;
-			else toolReturnTarget = "model_request";
-			if (shouldReturnDirect.size > 0) allNodeWorkflows.addConditionalEdges("tools", this.#createToolsRouter(shouldReturnDirect), [toolReturnTarget, END]);
-			else allNodeWorkflows.addEdge("tools", toolReturnTarget);
-		}
-		/**
-		* compile the graph
-		*/
-		this.#graph = allNodeWorkflows.compile({
-			checkpointer: this.options.checkpointer ?? this.options.checkpointSaver,
-			store: this.options.store,
-			name: this.options.name,
-			description: this.options.description
-		});
-	}
-	/**
-	* Get the compiled {@link https://docs.langchain.com/oss/javascript/langgraph/use-graph-api | StateGraph}.
-	*/
-	get graph() {
-		return this.#graph;
-	}
-	/**
-	* Get possible edge destinations from model node.
-	* @param toolClasses names of tools to call
-	* @param includeModelRequest whether to include "model_request" as a valid path (for jumpTo routing)
-	* @returns list of possible edge destinations
-	*/
-	#getModelPaths(toolClasses, includeModelRequest = false) {
-		const paths = [];
-		if (toolClasses.length > 0) paths.push("tools");
-		if (includeModelRequest) paths.push("model_request");
-		paths.push(END);
-		return paths;
-	}
-	/**
-	* Create routing function for tools node conditional edges.
-	*/
-	#createToolsRouter(shouldReturnDirect) {
-		/**
-		* ToDo: fix type
-		*/
-		return (state) => {
-			const messages = state.messages;
-			const lastMessage = messages[messages.length - 1];
-			if (ToolMessage.isInstance(lastMessage) && lastMessage.name && shouldReturnDirect.has(lastMessage.name)) return this.options.responseFormat ? "model_request" : END;
-			return "model_request";
-		};
-	}
-	/**
-	* Create routing function for model node conditional edges.
-	*/
-	#createModelRouter() {
-		/**
-		* determine if the agent should continue or not
-		*/
-		return (state) => {
-			const messages = state.messages;
-			const lastMessage = messages.at(-1);
-			if (!AIMessage.isInstance(lastMessage) || !lastMessage.tool_calls || lastMessage.tool_calls.length === 0) return END;
-			const hasOnlyStructuredResponseCalls = lastMessage.tool_calls.every((toolCall) => toolCall.name.startsWith("extract-"));
-			if (hasOnlyStructuredResponseCalls) return END;
-			/**
-			* The tool node processes a single message.
-			*/
-			if (this.#toolBehaviorVersion === "v1") return "tools";
-			/**
-			* Route to tools node (filter out any structured response tool calls)
-			*/
-			const regularToolCalls = lastMessage.tool_calls.filter((toolCall) => !toolCall.name.startsWith("extract-"));
-			if (regularToolCalls.length === 0) return END;
-			return regularToolCalls.map((toolCall) => new Send("tools", {
-				...state,
-				lg_tool_call: toolCall
-			}));
-		};
-	}
-	/**
-	* Create routing function for jumpTo functionality after afterModel hooks.
-	*
-	* This router checks if the `jumpTo` property is set in the state after afterModel middleware
-	* execution. If set, it routes to the specified target ("model_request" or "tools").
-	* If not set, it falls back to the normal model routing logic for afterModel context.
-	*
-	* The jumpTo property is automatically cleared after use to prevent infinite loops.
-	*
-	* @param toolClasses - Available tool classes for validation
-	* @returns Router function that handles jumpTo logic and normal routing
-	*/
-	#createAfterModelRouter(toolClasses, allowJump) {
-		const hasStructuredResponse = Boolean(this.options.responseFormat);
-		return (state) => {
-			const messages = state.messages;
-			const lastMessage = messages.at(-1);
-			if (AIMessage.isInstance(lastMessage) && (!lastMessage.tool_calls || lastMessage.tool_calls.length === 0)) return END;
-			if (allowJump && state.jumpTo) {
-				if (state.jumpTo === END) return END;
-				if (state.jumpTo === "tools") {
-					if (toolClasses.length === 0) return END;
-					return new Send("tools", {
-						...state,
-						jumpTo: void 0
-					});
-				}
-				return new Send("model_request", {
-					...state,
-					jumpTo: void 0
-				});
-			}
-			const toolMessages = messages.filter(ToolMessage.isInstance);
-			const lastAiMessage = messages.filter(AIMessage.isInstance).at(-1);
-			const pendingToolCalls = lastAiMessage?.tool_calls?.filter((call) => !toolMessages.some((m) => m.tool_call_id === call.id));
-			if (pendingToolCalls && pendingToolCalls.length > 0) return pendingToolCalls.map((toolCall) => new Send("tools", {
-				...state,
-				lg_tool_call: toolCall
-			}));
-			const hasStructuredResponseCalls = lastAiMessage?.tool_calls?.some((toolCall) => toolCall.name.startsWith("extract-"));
-			if (pendingToolCalls && pendingToolCalls.length === 0 && !hasStructuredResponseCalls && hasStructuredResponse) return "model_request";
-			if (!AIMessage.isInstance(lastMessage) || !lastMessage.tool_calls || lastMessage.tool_calls.length === 0) return END;
-			const hasOnlyStructuredResponseCalls = lastMessage.tool_calls.every((toolCall) => toolCall.name.startsWith("extract-"));
-			const hasRegularToolCalls = lastMessage.tool_calls.some((toolCall) => !toolCall.name.startsWith("extract-"));
-			if (hasOnlyStructuredResponseCalls || !hasRegularToolCalls) return END;
-			/**
-			* For routing from afterModel nodes, always use simple string paths
-			* The Send API is handled at the model_request node level
-			*/
-			return "tools";
-		};
-	}
-	/**
-	* Router for afterModel sequence nodes (connecting later middlewares to earlier ones),
-	* honoring allowed jump targets and defaulting to the next node.
-	*/
-	#createAfterModelSequenceRouter(toolClasses, allowed, nextDefault) {
-		const allowedSet = new Set(allowed.map((t) => parseJumpToTarget(t)));
-		return (state) => {
-			if (state.jumpTo) {
-				const dest = parseJumpToTarget(state.jumpTo);
-				if (dest === END && allowedSet.has(END)) return END;
-				if (dest === "tools" && allowedSet.has("tools")) {
-					if (toolClasses.length === 0) return END;
-					return new Send("tools", {
-						...state,
-						jumpTo: void 0
-					});
-				}
-				if (dest === "model_request" && allowedSet.has("model_request")) return new Send("model_request", {
-					...state,
-					jumpTo: void 0
-				});
-			}
-			return nextDefault;
-		};
-	}
-	/**
-	* Create routing function for jumpTo functionality after beforeModel hooks.
-	* Falls back to the default next node if no jumpTo is present.
-	*/
-	#createBeforeModelRouter(toolClasses, nextDefault) {
-		return (state) => {
-			if (!state.jumpTo) return nextDefault;
-			const destination = parseJumpToTarget(state.jumpTo);
-			if (destination === END) return END;
-			if (destination === "tools") {
-				if (toolClasses.length === 0) return END;
-				return new Send("tools", {
-					...state,
-					jumpTo: void 0
-				});
-			}
-			return new Send("model_request", {
-				...state,
-				jumpTo: void 0
-			});
-		};
-	}
-	/**
-	* Initialize middleware states if not already present in the input state.
-	*/
-	async #initializeMiddlewareStates(state) {
-		if (!this.options.middleware || this.options.middleware.length === 0 || state instanceof Command || !state) return state;
-		const defaultStates = await initializeMiddlewareStates(this.options.middleware, state);
-		const updatedState = { ...state };
-		if (!updatedState) return updatedState;
-		for (const [key, value] of Object.entries(defaultStates)) if (!(key in updatedState)) updatedState[key] = value;
-		return updatedState;
-	}
-	/**
-	* Executes the agent with the given state and returns the final state after all processing.
-	*
-	* This method runs the agent's entire workflow synchronously, including:
-	* - Processing the input messages through any configured middleware
-	* - Calling the language model to generate responses
-	* - Executing any tool calls made by the model
-	* - Running all middleware hooks (beforeModel, afterModel, etc.)
-	*
-	* @param state - The initial state for the agent execution. Can be:
-	*   - An object containing `messages` array and any middleware-specific state properties
-	*   - A Command object for more advanced control flow
-	*
-	* @param config - Optional runtime configuration including:
-	* @param config.context - The context for the agent execution.
-	* @param config.configurable - LangGraph configuration options like `thread_id`, `run_id`, etc.
-	* @param config.store - The store for the agent execution for persisting state, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Memory storage}.
-	* @param config.signal - An optional {@link https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal | `AbortSignal`} for the agent execution.
-	* @param config.recursionLimit - The recursion limit for the agent execution.
-	*
-	* @returns A Promise that resolves to the final agent state after execution completes.
-	*          The returned state includes:
-	*          - a `messages` property containing an array with all messages (input, AI responses, tool calls/results)
-	*          - a `structuredResponse` property containing the structured response (if configured)
-	*          - all state values defined in the middleware
-	*
-	* @example
-	* ```typescript
-	* const agent = new ReactAgent({
-	*   llm: myModel,
-	*   tools: [calculator, webSearch],
-	*   responseFormat: z.object({
-	*     weather: z.string(),
-	*   }),
-	* });
-	*
-	* const result = await agent.invoke({
-	*   messages: [{ role: "human", content: "What's the weather in Paris?" }]
-	* });
-	*
-	* console.log(result.structuredResponse.weather); // outputs: "It's sunny and 75°F."
-	* ```
-	*/
-	async invoke(state, config) {
-		const initializedState = await this.#initializeMiddlewareStates(state);
-		return this.#graph.invoke(initializedState, config);
-	}
-	/**
-	* Executes the agent with streaming, returning an async iterable of events as they occur.
-	*
-	* This method runs the agent's workflow similar to `invoke`, but instead of waiting for
-	* completion, it streams events in real-time. This allows you to:
-	* - Display intermediate results to users as they're generated
-	* - Monitor the agent's progress through each step
-	* - Handle tool calls and results as they happen
-	* - Update UI with streaming responses from the LLM
-	*
-	* @param state - The initial state for the agent execution. Can be:
-	*   - An object containing `messages` array and any middleware-specific state properties
-	*   - A Command object for more advanced control flow
-	*
-	* @param config - Optional runtime configuration including:
-	* @param config.context - The context for the agent execution.
-	* @param config.configurable - LangGraph configuration options like `thread_id`, `run_id`, etc.
-	* @param config.store - The store for the agent execution for persisting state, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Memory storage}.
-	* @param config.signal - An optional {@link https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal | `AbortSignal`} for the agent execution.
-	* @param config.streamMode - The streaming mode for the agent execution, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/streaming#supported-stream-modes | Supported stream modes}.
-	* @param config.recursionLimit - The recursion limit for the agent execution.
-	*
-	* @returns A Promise that resolves to an IterableReadableStream of events.
-	*          Events include:
-	*          - `on_chat_model_start`: When the LLM begins processing
-	*          - `on_chat_model_stream`: Streaming tokens from the LLM
-	*          - `on_chat_model_end`: When the LLM completes
-	*          - `on_tool_start`: When a tool execution begins
-	*          - `on_tool_end`: When a tool execution completes
-	*          - `on_chain_start`: When middleware chains begin
-	*          - `on_chain_end`: When middleware chains complete
-	*          - And other LangGraph v2 stream events
-	*
-	* @example
-	* ```typescript
-	* const agent = new ReactAgent({
-	*   llm: myModel,
-	*   tools: [calculator, webSearch]
-	* });
-	*
-	* const stream = await agent.stream({
-	*   messages: [{ role: "human", content: "What's 2+2 and the weather in NYC?" }]
-	* });
-	*
-	* for await (const event of stream) {
-	*   //
-	* }
-	* ```
-	*/
-	async stream(state, config) {
-		const initializedState = await this.#initializeMiddlewareStates(state);
-		return this.#graph.streamEvents(initializedState, {
-			...config,
-			version: "v2"
-		});
-	}
-	/**
-	* Visualize the graph as a PNG image.
-	* @param params - Parameters for the drawMermaidPng method.
-	* @param params.withStyles - Whether to include styles in the graph.
-	* @param params.curveStyle - The style of the graph's curves.
-	* @param params.nodeColors - The colors of the graph's nodes.
-	* @param params.wrapLabelNWords - The maximum number of words to wrap in a node's label.
-	* @param params.backgroundColor - The background color of the graph.
-	* @returns PNG image as a buffer
-	*/
-	async drawMermaidPng(params) {
-		const representation = await this.#graph.getGraphAsync();
-		const image = await representation.drawMermaidPng(params);
-		const arrayBuffer = await image.arrayBuffer();
-		const buffer = new Uint8Array(arrayBuffer);
-		return buffer;
-	}
-	/**
-	* Draw the graph as a Mermaid string.
-	* @param params - Parameters for the drawMermaid method.
-	* @param params.withStyles - Whether to include styles in the graph.
-	* @param params.curveStyle - The style of the graph's curves.
-	* @param params.nodeColors - The colors of the graph's nodes.
-	* @param params.wrapLabelNWords - The maximum number of words to wrap in a node's label.
-	* @param params.backgroundColor - The background color of the graph.
-	* @returns Mermaid string
-	*/
-	async drawMermaid(params) {
-		const representation = await this.#graph.getGraphAsync();
-		return representation.drawMermaid(params);
-	}
-};
-
-//#endregion
-export { ReactAgent };
-//# sourceMappingURL=ReactAgent.js.map

package/dist/agents/middlewareAgent/ReactAgent.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ReactAgent.js","names":["options: CreateAgentParams<StructuredResponseFormat, ContextSchema>","#toolBehaviorVersion","beforeModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[]","afterModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[]","modifyModelRequestHookMiddleware: [\n      AgentMiddleware,\n      /**\n       * ToDo: better type to get the state of middleware\n       */\n      () => any\n    ][]","beforeModelNode: BeforeModelNode | undefined","afterModelNode: AfterModelNode | undefined","#createBeforeModelRouter","#getModelPaths","#createModelRouter","#createAfterModelSequenceRouter","#createAfterModelRouter","toolReturnTarget: string","#createToolsRouter","#graph","toolClasses: (ClientTool | ServerTool)[]","includeModelRequest: boolean","paths: (\"tools\" | \"model_request\" | typeof END)[]","shouldReturnDirect: Set<string>","state: any","state: BuiltInState","allowJump: boolean","state: Omit<BuiltInState, \"jumpTo\"> & { jumpTo?: JumpTo }","allowed: string[]","nextDefault: string","#initializeMiddlewareStates","state: InvokeStateParameter<TMiddleware>","config?: InvokeConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >","config?: StreamConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >","params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }"],"sources":["../../../src/agents/middlewareAgent/ReactAgent.ts"],"sourcesContent":["/* eslint-disable no-instanceof/no-instanceof */\n/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\n\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  Command,\n  CompiledStateGraph,\n} from \"@langchain/langgraph\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\nimport { IterableReadableStream } from \"@langchain/core/utils/stream\";\n\nimport { createAgentAnnotationConditional } from \"./annotation.js\";\nimport { isClientTool } from \"../utils.js\";\n\nimport { AgentNode } from \"./nodes/AgentNode.js\";\nimport { ToolNode } from \"../nodes/ToolNode.js\";\nimport { BeforeModelNode } from \"./nodes/BeforeModalNode.js\";\nimport { AfterModelNode } from \"./nodes/AfterModalNode.js\";\nimport {\n  initializeMiddlewareStates,\n  parseJumpToTarget,\n} from \"./nodes/utils.js\";\n\nimport type { ClientTool, ServerTool, WithStateGraphNodes } from \"../types.js\";\n\nimport type {\n  CreateAgentParams,\n  AgentMiddleware,\n  InferMiddlewareStates,\n  InferMiddlewareInputStates,\n  BuiltInState,\n  InferMiddlewareContextInputs,\n  InferContextInput,\n  InvokeConfiguration,\n  StreamConfiguration,\n  JumpTo,\n} from \"./types.js\";\n\nimport {\n  type AnyAnnotationRoot,\n  type ToAnnotationRoot,\n  type ResponseFormatUndefined,\n} from \"../annotation.js\";\n\n// Helper type to get the state definition with middleware states\ntype MergedAgentState<\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined,\n  TMiddleware extends readonly AgentMiddleware<any, any, any>[]\n> = (StructuredResponseFormat extends ResponseFormatUndefined\n  ? Omit<BuiltInState, \"jumpTo\">\n  : Omit<BuiltInState, \"jumpTo\"> & {\n      structuredResponse: StructuredResponseFormat;\n    }) &\n  InferMiddlewareStates<TMiddleware>;\n\ntype InvokeStateParameter<\n  TMiddleware extends readonly AgentMiddleware<any, any, any>[]\n> =\n  | (BuiltInState & InferMiddlewareInputStates<TMiddleware>)\n  | Command<any, any, any>\n  | null;\n\ntype AgentGraph<\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddleware extends readonly AgentMiddleware<any, any, any>[] = []\n> = CompiledStateGraph<\n  any,\n  any,\n  any,\n  any,\n  MergedAgentState<StructuredResponseFormat, TMiddleware>,\n  ToAnnotationRoot<ContextSchema>[\"spec\"],\n  unknown\n>;\n\nexport class ReactAgent<\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddleware extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n> {\n  #graph: AgentGraph<StructuredResponseFormat, ContextSchema, TMiddleware>;\n\n  #toolBehaviorVersion: \"v1\" | \"v2\" = \"v2\";\n\n  constructor(\n    public options: CreateAgentParams<StructuredResponseFormat, ContextSchema>\n  ) {\n    this.#toolBehaviorVersion = options.version ?? this.#toolBehaviorVersion;\n\n    /**\n     * define complete list of tools based on options and middleware\n     */\n    const middlewareTools = (this.options.middleware\n      ?.filter((m) => m.tools)\n      .flatMap((m) => m.tools) ?? []) as (ClientTool | ServerTool)[];\n    const toolClasses = [\n      ...((Array.isArray(options.tools)\n        ? options.tools\n        : options.tools?.tools) ?? []),\n      ...middlewareTools,\n    ];\n\n    /**\n     * If any of the tools are configured to return_directly after running,\n     * our graph needs to check if these were called\n     */\n    const shouldReturnDirect = new Set(\n      toolClasses\n        .filter(isClientTool)\n        .filter((tool) => \"returnDirect\" in tool && tool.returnDirect)\n        .map((tool) => tool.name)\n    );\n\n    // Create a schema that merges agent base schema with middleware state schemas\n    const schema = createAgentAnnotationConditional<\n      StructuredResponseFormat,\n      TMiddleware\n    >(\n      this.options.responseFormat !== undefined,\n      this.options.middleware as TMiddleware\n    );\n\n    const workflow = new StateGraph(\n      schema as AnnotationRoot<any>,\n      this.options.contextSchema\n    );\n\n    const allNodeWorkflows = workflow as WithStateGraphNodes<\n      \"tools\" | \"model_request\" | string,\n      typeof workflow\n    >;\n\n    // Generate node names for middleware nodes that have hooks\n    const beforeModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[] = [];\n    const afterModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[] = [];\n    const modifyModelRequestHookMiddleware: [\n      AgentMiddleware,\n      /**\n       * ToDo: better type to get the state of middleware\n       */\n      () => any\n    ][] = [];\n\n    const middlewareNames = new Set<string>();\n    const middleware = this.options.middleware ?? [];\n    for (let i = 0; i < middleware.length; i++) {\n      let beforeModelNode: BeforeModelNode | undefined;\n      let afterModelNode: AfterModelNode | undefined;\n      const m = middleware[i];\n      if (middlewareNames.has(m.name)) {\n        throw new Error(`Middleware ${m.name} is defined multiple times`);\n      }\n\n      middlewareNames.add(m.name);\n      if (m.beforeModel) {\n        beforeModelNode = new BeforeModelNode(m);\n        const name = `${m.name}.before_model`;\n        beforeModelNodes.push({\n          index: i,\n          name,\n          allowed: m.beforeModelJumpTo,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          beforeModelNode,\n          beforeModelNode.nodeOptions\n        );\n      }\n      if (m.afterModel) {\n        afterModelNode = new AfterModelNode(m);\n        const name = `${m.name}.after_model`;\n        afterModelNodes.push({\n          index: i,\n          name,\n          allowed: m.afterModelJumpTo,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          afterModelNode,\n          afterModelNode.nodeOptions\n        );\n      }\n\n      if (m.modifyModelRequest) {\n        modifyModelRequestHookMiddleware.push([\n          m,\n          () => ({\n            ...beforeModelNode?.getState(),\n            ...afterModelNode?.getState(),\n          }),\n        ]);\n      }\n    }\n\n    /**\n     * Add Nodes\n     */\n    allNodeWorkflows.addNode(\n      \"model_request\",\n      new AgentNode({\n        model: this.options.model,\n        systemPrompt: this.options.systemPrompt,\n        includeAgentName: this.options.includeAgentName,\n        name: this.options.name,\n        responseFormat: this.options.responseFormat,\n        middleware: this.options.middleware,\n        toolClasses,\n        shouldReturnDirect,\n        signal: this.options.signal,\n        modifyModelRequestHookMiddleware,\n      }),\n      AgentNode.nodeOptions\n    );\n\n    /**\n     * add single tool node for all tools\n     */\n    if (toolClasses.length > 0) {\n      const toolNode = new ToolNode(toolClasses.filter(isClientTool), {\n        signal: this.options.signal,\n      });\n      allNodeWorkflows.addNode(\"tools\", toolNode);\n    }\n\n    /**\n     * Add Edges\n     */\n    // Determine starting point based on what nodes exist\n    if (beforeModelNodes.length > 0) {\n      // If we have beforeModel nodes, start with the first one\n      allNodeWorkflows.addEdge(START, beforeModelNodes[0].name);\n    } else {\n      // If no beforeModel nodes, go directly to agent\n      allNodeWorkflows.addEdge(START, \"model_request\");\n    }\n\n    // Connect beforeModel nodes; add conditional routing ONLY if allowed jumps are specified\n    for (let i = 0; i < beforeModelNodes.length; i++) {\n      const node = beforeModelNodes[i];\n      const current = node.name;\n      const isLast = i === beforeModelNodes.length - 1;\n      const nextDefault = isLast\n        ? \"model_request\"\n        : beforeModelNodes[i + 1].name;\n\n      if (node.allowed && node.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = node.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n        const destinations = Array.from(\n          new Set([nextDefault, ...allowedMapped])\n        ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n        allNodeWorkflows.addConditionalEdges(\n          current,\n          this.#createBeforeModelRouter(\n            toolClasses.filter(isClientTool),\n            nextDefault\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(current, nextDefault);\n      }\n    }\n\n    // Connect agent to last afterModel node (for reverse order execution)\n    const lastAfterModelNode = afterModelNodes.at(-1);\n    if (afterModelNodes.length > 0 && lastAfterModelNode) {\n      allNodeWorkflows.addEdge(\"model_request\", lastAfterModelNode.name);\n    } else {\n      // If no afterModel nodes, connect model_request directly to model paths\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      if (modelPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"model_request\", modelPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"model_request\",\n          this.#createModelRouter(),\n          modelPaths\n        );\n      }\n    }\n\n    // Connect afterModel nodes in reverse sequence; add conditional routing ONLY if allowed jumps are specified per node\n    for (let i = afterModelNodes.length - 1; i > 0; i--) {\n      const node = afterModelNodes[i];\n      const current = node.name;\n      const nextDefault = afterModelNodes[i - 1].name;\n\n      if (node.allowed && node.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = node.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n        const destinations = Array.from(\n          new Set([nextDefault, ...allowedMapped])\n        ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n        allNodeWorkflows.addConditionalEdges(\n          current,\n          this.#createAfterModelSequenceRouter(\n            toolClasses.filter(isClientTool),\n            node.allowed,\n            nextDefault\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(current, nextDefault);\n      }\n    }\n\n    // Connect first afterModel node (last to execute) to model paths with jumpTo support\n    if (afterModelNodes.length > 0) {\n      const firstAfterModel = afterModelNodes[0];\n      const firstAfterModelNode = firstAfterModel.name;\n      const modelPaths = this.#getModelPaths(\n        toolClasses.filter(isClientTool),\n        true\n      ).filter(\n        (p) => p !== \"tools\" || toolClasses.filter(isClientTool).length > 0\n      );\n\n      const allowJump = Boolean(\n        firstAfterModel.allowed && firstAfterModel.allowed.length > 0\n      );\n\n      const destinations = modelPaths;\n\n      allNodeWorkflows.addConditionalEdges(\n        firstAfterModelNode,\n        this.#createAfterModelRouter(\n          toolClasses.filter(isClientTool),\n          allowJump\n        ),\n        destinations\n      );\n    }\n\n    /**\n     * add edges for tools node\n     */\n    if (toolClasses.length > 0) {\n      // Tools should return to first beforeModel node or agent\n      let toolReturnTarget: string;\n      if (beforeModelNodes.length > 0) {\n        toolReturnTarget = beforeModelNodes[0].name;\n      } else {\n        toolReturnTarget = \"model_request\";\n      }\n\n      if (shouldReturnDirect.size > 0) {\n        allNodeWorkflows.addConditionalEdges(\n          \"tools\",\n          this.#createToolsRouter(shouldReturnDirect),\n          [toolReturnTarget, END]\n        );\n      } else {\n        allNodeWorkflows.addEdge(\"tools\", toolReturnTarget);\n      }\n    }\n\n    /**\n     * compile the graph\n     */\n    this.#graph = allNodeWorkflows.compile({\n      checkpointer: this.options.checkpointer ?? this.options.checkpointSaver,\n      store: this.options.store,\n      name: this.options.name,\n      description: this.options.description,\n    }) as AgentGraph<StructuredResponseFormat, ContextSchema, TMiddleware>;\n  }\n\n  /**\n   * Get the compiled {@link https://docs.langchain.com/oss/javascript/langgraph/use-graph-api | StateGraph}.\n   */\n  get graph(): AgentGraph<\n    StructuredResponseFormat,\n    ContextSchema,\n    TMiddleware\n  > {\n    return this.#graph;\n  }\n\n  /**\n   * Get possible edge destinations from model node.\n   * @param toolClasses names of tools to call\n   * @param includeModelRequest whether to include \"model_request\" as a valid path (for jumpTo routing)\n   * @returns list of possible edge destinations\n   */\n  #getModelPaths(\n    toolClasses: (ClientTool | ServerTool)[],\n    includeModelRequest: boolean = false\n  ): (\"tools\" | \"model_request\" | typeof END)[] {\n    const paths: (\"tools\" | \"model_request\" | typeof END)[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(\"tools\");\n    }\n\n    if (includeModelRequest) {\n      paths.push(\"model_request\");\n    }\n\n    paths.push(END);\n\n    return paths;\n  }\n\n  /**\n   * Create routing function for tools node conditional edges.\n   */\n  #createToolsRouter(shouldReturnDirect: Set<string>) {\n    /**\n     * ToDo: fix type\n     */\n    return (state: any) => {\n      const messages = state.messages;\n      const lastMessage = messages[messages.length - 1];\n\n      // Check if we just executed a returnDirect tool\n      if (\n        ToolMessage.isInstance(lastMessage) &&\n        lastMessage.name &&\n        shouldReturnDirect.has(lastMessage.name)\n      ) {\n        // If we have a response format, route to agent to generate structured response\n        // Otherwise, return directly\n        return this.options.responseFormat ? \"model_request\" : END;\n      }\n\n      // For non-returnDirect tools, always route back to agent\n      return \"model_request\";\n    };\n  }\n\n  /**\n   * Create routing function for model node conditional edges.\n   */\n  #createModelRouter() {\n    /**\n     * determine if the agent should continue or not\n     */\n    return (state: BuiltInState) => {\n      const messages = state.messages;\n      const lastMessage = messages.at(-1);\n\n      if (\n        !AIMessage.isInstance(lastMessage) ||\n        !lastMessage.tool_calls ||\n        lastMessage.tool_calls.length === 0\n      ) {\n        return END;\n      }\n\n      // Check if all tool calls are for structured response extraction\n      const hasOnlyStructuredResponseCalls = lastMessage.tool_calls.every(\n        (toolCall) => toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (hasOnlyStructuredResponseCalls) {\n        // If all tool calls are for structured response extraction, go to END\n        // The AgentNode will handle these internally and return the structured response\n        return END;\n      }\n\n      /**\n       * The tool node processes a single message.\n       */\n      if (this.#toolBehaviorVersion === \"v1\") {\n        return \"tools\";\n      }\n\n      /**\n       * Route to tools node (filter out any structured response tool calls)\n       */\n      const regularToolCalls = lastMessage.tool_calls.filter(\n        (toolCall) => !toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (regularToolCalls.length === 0) {\n        return END;\n      }\n\n      return regularToolCalls.map(\n        (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n      );\n    };\n  }\n\n  /**\n   * Create routing function for jumpTo functionality after afterModel hooks.\n   *\n   * This router checks if the `jumpTo` property is set in the state after afterModel middleware\n   * execution. If set, it routes to the specified target (\"model_request\" or \"tools\").\n   * If not set, it falls back to the normal model routing logic for afterModel context.\n   *\n   * The jumpTo property is automatically cleared after use to prevent infinite loops.\n   *\n   * @param toolClasses - Available tool classes for validation\n   * @returns Router function that handles jumpTo logic and normal routing\n   */\n  #createAfterModelRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    allowJump: boolean\n  ) {\n    const hasStructuredResponse = Boolean(this.options.responseFormat);\n\n    return (state: Omit<BuiltInState, \"jumpTo\"> & { jumpTo?: JumpTo }) => {\n      // First, check if we just processed a structured response\n      // If so, ignore any existing jumpTo and go to END\n      const messages = state.messages;\n      const lastMessage = messages.at(-1);\n      if (\n        AIMessage.isInstance(lastMessage) &&\n        (!lastMessage.tool_calls || lastMessage.tool_calls.length === 0)\n      ) {\n        return END;\n      }\n\n      // Check if jumpTo is set in the state and allowed\n      if (allowJump && state.jumpTo) {\n        if (state.jumpTo === END) {\n          return END;\n        }\n        if (state.jumpTo === \"tools\") {\n          // If trying to jump to tools but no tools are available, go to END\n          if (toolClasses.length === 0) {\n            return END;\n          }\n          return new Send(\"tools\", { ...state, jumpTo: undefined });\n        }\n        // destination === \"model_request\"\n        return new Send(\"model_request\", { ...state, jumpTo: undefined });\n      }\n\n      // check if there are pending tool calls\n      const toolMessages = messages.filter(ToolMessage.isInstance);\n      const lastAiMessage = messages.filter(AIMessage.isInstance).at(-1);\n      const pendingToolCalls = lastAiMessage?.tool_calls?.filter(\n        (call) => !toolMessages.some((m) => m.tool_call_id === call.id)\n      );\n      if (pendingToolCalls && pendingToolCalls.length > 0) {\n        return pendingToolCalls.map(\n          (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n        );\n      }\n\n      // if we exhausted all tool calls, but still have no structured response tool calls,\n      // go back to model_request\n      const hasStructuredResponseCalls = lastAiMessage?.tool_calls?.some(\n        (toolCall) => toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (\n        pendingToolCalls &&\n        pendingToolCalls.length === 0 &&\n        !hasStructuredResponseCalls &&\n        hasStructuredResponse\n      ) {\n        return \"model_request\";\n      }\n\n      if (\n        !AIMessage.isInstance(lastMessage) ||\n        !lastMessage.tool_calls ||\n        lastMessage.tool_calls.length === 0\n      ) {\n        return END;\n      }\n\n      // Check if all tool calls are for structured response extraction\n      const hasOnlyStructuredResponseCalls = lastMessage.tool_calls.every(\n        (toolCall) => toolCall.name.startsWith(\"extract-\")\n      );\n\n      // Check if there are any regular tool calls (non-structured response)\n      const hasRegularToolCalls = lastMessage.tool_calls.some(\n        (toolCall) => !toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (hasOnlyStructuredResponseCalls || !hasRegularToolCalls) {\n        return END;\n      }\n\n      /**\n       * For routing from afterModel nodes, always use simple string paths\n       * The Send API is handled at the model_request node level\n       */\n      return \"tools\";\n    };\n  }\n\n  /**\n   * Router for afterModel sequence nodes (connecting later middlewares to earlier ones),\n   * honoring allowed jump targets and defaulting to the next node.\n   */\n  #createAfterModelSequenceRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    allowed: string[],\n    nextDefault: string\n  ) {\n    const allowedSet = new Set(allowed.map((t) => parseJumpToTarget(t)));\n    return (state: BuiltInState) => {\n      if (state.jumpTo) {\n        const dest = parseJumpToTarget(state.jumpTo);\n        if (dest === END && allowedSet.has(END)) {\n          return END;\n        }\n        if (dest === \"tools\" && allowedSet.has(\"tools\")) {\n          if (toolClasses.length === 0) return END;\n          return new Send(\"tools\", { ...state, jumpTo: undefined });\n        }\n        if (dest === \"model_request\" && allowedSet.has(\"model_request\")) {\n          return new Send(\"model_request\", { ...state, jumpTo: undefined });\n        }\n      }\n      return nextDefault as any;\n    };\n  }\n\n  /**\n   * Create routing function for jumpTo functionality after beforeModel hooks.\n   * Falls back to the default next node if no jumpTo is present.\n   */\n  #createBeforeModelRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    nextDefault: string\n  ) {\n    return (state: BuiltInState) => {\n      if (!state.jumpTo) {\n        return nextDefault;\n      }\n      const destination = parseJumpToTarget(state.jumpTo);\n      if (destination === END) {\n        return END;\n      }\n      if (destination === \"tools\") {\n        if (toolClasses.length === 0) {\n          return END;\n        }\n        return new Send(\"tools\", { ...state, jumpTo: undefined });\n      }\n      // destination === \"model_request\"\n      return new Send(\"model_request\", { ...state, jumpTo: undefined });\n    };\n  }\n\n  /**\n   * Initialize middleware states if not already present in the input state.\n   */\n  async #initializeMiddlewareStates(\n    state: InvokeStateParameter<TMiddleware>\n  ): Promise<InvokeStateParameter<TMiddleware>> {\n    if (\n      !this.options.middleware ||\n      this.options.middleware.length === 0 ||\n      state instanceof Command ||\n      !state\n    ) {\n      return state;\n    }\n\n    const defaultStates = await initializeMiddlewareStates(\n      this.options.middleware,\n      state\n    );\n    const updatedState = { ...state } as InvokeStateParameter<TMiddleware>;\n    if (!updatedState) {\n      return updatedState;\n    }\n\n    // Only add defaults for keys that don't exist in current state\n    for (const [key, value] of Object.entries(defaultStates)) {\n      if (!(key in updatedState)) {\n        updatedState[key as keyof typeof updatedState] = value;\n      }\n    }\n\n    return updatedState;\n  }\n\n  /**\n   * Executes the agent with the given state and returns the final state after all processing.\n   *\n   * This method runs the agent's entire workflow synchronously, including:\n   * - Processing the input messages through any configured middleware\n   * - Calling the language model to generate responses\n   * - Executing any tool calls made by the model\n   * - Running all middleware hooks (beforeModel, afterModel, etc.)\n   *\n   * @param state - The initial state for the agent execution. Can be:\n   *   - An object containing `messages` array and any middleware-specific state properties\n   *   - A Command object for more advanced control flow\n   *\n   * @param config - Optional runtime configuration including:\n   * @param config.context - The context for the agent execution.\n   * @param config.configurable - LangGraph configuration options like `thread_id`, `run_id`, etc.\n   * @param config.store - The store for the agent execution for persisting state, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Memory storage}.\n   * @param config.signal - An optional {@link https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal | `AbortSignal`} for the agent execution.\n   * @param config.recursionLimit - The recursion limit for the agent execution.\n   *\n   * @returns A Promise that resolves to the final agent state after execution completes.\n   *          The returned state includes:\n   *          - a `messages` property containing an array with all messages (input, AI responses, tool calls/results)\n   *          - a `structuredResponse` property containing the structured response (if configured)\n   *          - all state values defined in the middleware\n   *\n   * @example\n   * ```typescript\n   * const agent = new ReactAgent({\n   *   llm: myModel,\n   *   tools: [calculator, webSearch],\n   *   responseFormat: z.object({\n   *     weather: z.string(),\n   *   }),\n   * });\n   *\n   * const result = await agent.invoke({\n   *   messages: [{ role: \"human\", content: \"What's the weather in Paris?\" }]\n   * });\n   *\n   * console.log(result.structuredResponse.weather); // outputs: \"It's sunny and 75°F.\"\n   * ```\n   */\n  async invoke(\n    state: InvokeStateParameter<TMiddleware>,\n    config?: InvokeConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >\n  ) {\n    type FullState = MergedAgentState<StructuredResponseFormat, TMiddleware>;\n    const initializedState = await this.#initializeMiddlewareStates(state);\n    return this.#graph.invoke(\n      initializedState,\n      config as unknown as InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    ) as Promise<FullState>;\n  }\n\n  /**\n   * Executes the agent with streaming, returning an async iterable of events as they occur.\n   *\n   * This method runs the agent's workflow similar to `invoke`, but instead of waiting for\n   * completion, it streams events in real-time. This allows you to:\n   * - Display intermediate results to users as they're generated\n   * - Monitor the agent's progress through each step\n   * - Handle tool calls and results as they happen\n   * - Update UI with streaming responses from the LLM\n   *\n   * @param state - The initial state for the agent execution. Can be:\n   *   - An object containing `messages` array and any middleware-specific state properties\n   *   - A Command object for more advanced control flow\n   *\n   * @param config - Optional runtime configuration including:\n   * @param config.context - The context for the agent execution.\n   * @param config.configurable - LangGraph configuration options like `thread_id`, `run_id`, etc.\n   * @param config.store - The store for the agent execution for persisting state, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Memory storage}.\n   * @param config.signal - An optional {@link https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal | `AbortSignal`} for the agent execution.\n   * @param config.streamMode - The streaming mode for the agent execution, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/streaming#supported-stream-modes | Supported stream modes}.\n   * @param config.recursionLimit - The recursion limit for the agent execution.\n   *\n   * @returns A Promise that resolves to an IterableReadableStream of events.\n   *          Events include:\n   *          - `on_chat_model_start`: When the LLM begins processing\n   *          - `on_chat_model_stream`: Streaming tokens from the LLM\n   *          - `on_chat_model_end`: When the LLM completes\n   *          - `on_tool_start`: When a tool execution begins\n   *          - `on_tool_end`: When a tool execution completes\n   *          - `on_chain_start`: When middleware chains begin\n   *          - `on_chain_end`: When middleware chains complete\n   *          - And other LangGraph v2 stream events\n   *\n   * @example\n   * ```typescript\n   * const agent = new ReactAgent({\n   *   llm: myModel,\n   *   tools: [calculator, webSearch]\n   * });\n   *\n   * const stream = await agent.stream({\n   *   messages: [{ role: \"human\", content: \"What's 2+2 and the weather in NYC?\" }]\n   * });\n   *\n   * for await (const event of stream) {\n   *   //\n   * }\n   * ```\n   */\n  async stream(\n    state: InvokeStateParameter<TMiddleware>,\n    config?: StreamConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >\n  ): Promise<IterableReadableStream<any>> {\n    const initializedState = await this.#initializeMiddlewareStates(state);\n    return this.#graph.streamEvents(initializedState, {\n      ...config,\n      version: \"v2\",\n    } as any) as IterableReadableStream<any>;\n  }\n\n  /**\n   * Visualize the graph as a PNG image.\n   * @param params - Parameters for the drawMermaidPng method.\n   * @param params.withStyles - Whether to include styles in the graph.\n   * @param params.curveStyle - The style of the graph's curves.\n   * @param params.nodeColors - The colors of the graph's nodes.\n   * @param params.wrapLabelNWords - The maximum number of words to wrap in a node's label.\n   * @param params.backgroundColor - The background color of the graph.\n   * @returns PNG image as a buffer\n   */\n  async drawMermaidPng(params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }) {\n    const representation = await this.#graph.getGraphAsync();\n    const image = await representation.drawMermaidPng(params);\n    const arrayBuffer = await image.arrayBuffer();\n    const buffer = new Uint8Array(arrayBuffer);\n    return buffer;\n  }\n\n  /**\n   * Draw the graph as a Mermaid string.\n   * @param params - Parameters for the drawMermaid method.\n   * @param params.withStyles - Whether to include styles in the graph.\n   * @param params.curveStyle - The style of the graph's curves.\n   * @param params.nodeColors - The colors of the graph's nodes.\n   * @param params.wrapLabelNWords - The maximum number of words to wrap in a node's label.\n   * @param params.backgroundColor - The background color of the graph.\n   * @returns Mermaid string\n   */\n  async drawMermaid(params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }) {\n    const representation = await this.#graph.getGraphAsync();\n    return representation.drawMermaid(params);\n  }\n}\n"],"mappings":";;;;;;;;;;;AAuFA,IAAa,aAAb,MAYE;CACA;CAEA,uBAAoC;CAEpC,YACSA,SACP;EADO;EAEP,KAAKC,uBAAuB,QAAQ,WAAW,KAAKA;;;;EAKpD,MAAM,kBAAmB,KAAK,QAAQ,YAClC,OAAO,CAAC,MAAM,EAAE,MAAM,CACvB,QAAQ,CAAC,MAAM,EAAE,MAAM,IAAI,CAAE;EAChC,MAAM,cAAc,CAClB,IAAK,MAAM,QAAQ,QAAQ,MAAM,GAC7B,QAAQ,QACR,QAAQ,OAAO,UAAU,CAAE,GAC/B,GAAG,eACJ;;;;;EAMD,MAAM,qBAAqB,IAAI,IAC7B,YACG,OAAO,aAAa,CACpB,OAAO,CAAC,SAAS,kBAAkB,QAAQ,KAAK,aAAa,CAC7D,IAAI,CAAC,SAAS,KAAK,KAAK;EAI7B,MAAM,SAAS,iCAIb,KAAK,QAAQ,mBAAmB,QAChC,KAAK,QAAQ,WACd;EAED,MAAM,WAAW,IAAI,WACnB,QACA,KAAK,QAAQ;EAGf,MAAM,mBAAmB;EAMzB,MAAMC,mBAIA,CAAE;EACR,MAAMC,kBAIA,CAAE;EACR,MAAMC,mCAMA,CAAE;EAER,MAAM,kCAAkB,IAAI;EAC5B,MAAM,aAAa,KAAK,QAAQ,cAAc,CAAE;AAChD,OAAK,IAAI,IAAI,GAAG,IAAI,WAAW,QAAQ,KAAK;GAC1C,IAAIC;GACJ,IAAIC;GACJ,MAAM,IAAI,WAAW;AACrB,OAAI,gBAAgB,IAAI,EAAE,KAAK,CAC7B,OAAM,IAAI,MAAM,CAAC,WAAW,EAAE,EAAE,KAAK,0BAA0B,CAAC;GAGlE,gBAAgB,IAAI,EAAE,KAAK;AAC3B,OAAI,EAAE,aAAa;IACjB,kBAAkB,IAAI,gBAAgB;IACtC,MAAM,OAAO,GAAG,EAAE,KAAK,aAAa,CAAC;IACrC,iBAAiB,KAAK;KACpB,OAAO;KACP;KACA,SAAS,EAAE;IACZ,EAAC;IACF,iBAAiB,QACf,MACA,iBACA,gBAAgB,YACjB;GACF;AACD,OAAI,EAAE,YAAY;IAChB,iBAAiB,IAAI,eAAe;IACpC,MAAM,OAAO,GAAG,EAAE,KAAK,YAAY,CAAC;IACpC,gBAAgB,KAAK;KACnB,OAAO;KACP;KACA,SAAS,EAAE;IACZ,EAAC;IACF,iBAAiB,QACf,MACA,gBACA,eAAe,YAChB;GACF;AAED,OAAI,EAAE,oBACJ,iCAAiC,KAAK,CACpC,GACA,OAAO;IACL,GAAG,iBAAiB,UAAU;IAC9B,GAAG,gBAAgB,UAAU;GAC9B,EACF,EAAC;EAEL;;;;EAKD,iBAAiB,QACf,iBACA,IAAI,UAAU;GACZ,OAAO,KAAK,QAAQ;GACpB,cAAc,KAAK,QAAQ;GAC3B,kBAAkB,KAAK,QAAQ;GAC/B,MAAM,KAAK,QAAQ;GACnB,gBAAgB,KAAK,QAAQ;GAC7B,YAAY,KAAK,QAAQ;GACzB;GACA;GACA,QAAQ,KAAK,QAAQ;GACrB;EACD,IACD,UAAU,YACX;;;;AAKD,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,WAAW,IAAI,SAAS,YAAY,OAAO,aAAa,EAAE,EAC9D,QAAQ,KAAK,QAAQ,OACtB;GACD,iBAAiB,QAAQ,SAAS,SAAS;EAC5C;;;;AAMD,MAAI,iBAAiB,SAAS,GAE5B,iBAAiB,QAAQ,OAAO,iBAAiB,GAAG,KAAK;OAGzD,iBAAiB,QAAQ,OAAO,gBAAgB;AAIlD,OAAK,IAAI,IAAI,GAAG,IAAI,iBAAiB,QAAQ,KAAK;GAChD,MAAM,OAAO,iBAAiB;GAC9B,MAAM,UAAU,KAAK;GACrB,MAAM,SAAS,MAAM,iBAAiB,SAAS;GAC/C,MAAM,cAAc,SAChB,kBACA,iBAAiB,IAAI,GAAG;AAE5B,OAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;IAC3C,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,KAAK,QACxB,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;IACjD,MAAM,eAAe,MAAM,KACzB,IAAI,IAAI,CAAC,aAAa,GAAG,aAAc,GACxC;IAED,iBAAiB,oBACf,SACA,KAAKC,yBACH,YAAY,OAAO,aAAa,EAChC,YACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,SAAS,YAAY;EAEjD;EAGD,MAAM,qBAAqB,gBAAgB,GAAG,GAAG;AACjD,MAAI,gBAAgB,SAAS,KAAK,oBAChC,iBAAiB,QAAQ,iBAAiB,mBAAmB,KAAK;OAC7D;GAEL,MAAM,aAAa,KAAKC,eAAe,YAAY,OAAO,aAAa,CAAC;AACxE,OAAI,WAAW,WAAW,GACxB,iBAAiB,QAAQ,iBAAiB,WAAW,GAAG;QAExD,iBAAiB,oBACf,iBACA,KAAKC,oBAAoB,EACzB,WACD;EAEJ;AAGD,OAAK,IAAI,IAAI,gBAAgB,SAAS,GAAG,IAAI,GAAG,KAAK;GACnD,MAAM,OAAO,gBAAgB;GAC7B,MAAM,UAAU,KAAK;GACrB,MAAM,cAAc,gBAAgB,IAAI,GAAG;AAE3C,OAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;IAC3C,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,KAAK,QACxB,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;IACjD,MAAM,eAAe,MAAM,KACzB,IAAI,IAAI,CAAC,aAAa,GAAG,aAAc,GACxC;IAED,iBAAiB,oBACf,SACA,KAAKC,gCACH,YAAY,OAAO,aAAa,EAChC,KAAK,SACL,YACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,SAAS,YAAY;EAEjD;AAGD,MAAI,gBAAgB,SAAS,GAAG;GAC9B,MAAM,kBAAkB,gBAAgB;GACxC,MAAM,sBAAsB,gBAAgB;GAC5C,MAAM,aAAa,KAAKF,eACtB,YAAY,OAAO,aAAa,EAChC,KACD,CAAC,OACA,CAAC,MAAM,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS,EACnE;GAED,MAAM,YAAY,QAChB,gBAAgB,WAAW,gBAAgB,QAAQ,SAAS,EAC7D;GAED,MAAM,eAAe;GAErB,iBAAiB,oBACf,qBACA,KAAKG,wBACH,YAAY,OAAO,aAAa,EAChC,UACD,EACD,aACD;EACF;;;;AAKD,MAAI,YAAY,SAAS,GAAG;GAE1B,IAAIC;AACJ,OAAI,iBAAiB,SAAS,GAC5B,mBAAmB,iBAAiB,GAAG;QAEvC,mBAAmB;AAGrB,OAAI,mBAAmB,OAAO,GAC5B,iBAAiB,oBACf,SACA,KAAKC,mBAAmB,mBAAmB,EAC3C,CAAC,kBAAkB,GAAI,EACxB;QAED,iBAAiB,QAAQ,SAAS,iBAAiB;EAEtD;;;;EAKD,KAAKC,SAAS,iBAAiB,QAAQ;GACrC,cAAc,KAAK,QAAQ,gBAAgB,KAAK,QAAQ;GACxD,OAAO,KAAK,QAAQ;GACpB,MAAM,KAAK,QAAQ;GACnB,aAAa,KAAK,QAAQ;EAC3B,EAAC;CACH;;;;CAKD,IAAI,QAIF;AACA,SAAO,KAAKA;CACb;;;;;;;CAQD,eACEC,aACAC,sBAA+B,OACa;EAC5C,MAAMC,QAAoD,CAAE;AAC5D,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;AAGrB,MAAI,qBACF,MAAM,KAAK,gBAAgB;EAG7B,MAAM,KAAK,IAAI;AAEf,SAAO;CACR;;;;CAKD,mBAAmBC,oBAAiC;;;;AAIlD,SAAO,CAACC,UAAe;GACrB,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,SAAS,SAAS;AAG/C,OACE,YAAY,WAAW,YAAY,IACnC,YAAY,QACZ,mBAAmB,IAAI,YAAY,KAAK,CAIxC,QAAO,KAAK,QAAQ,iBAAiB,kBAAkB;AAIzD,UAAO;EACR;CACF;;;;CAKD,qBAAqB;;;;AAInB,SAAO,CAACC,UAAwB;GAC9B,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,CAAC,UAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,EAElC,QAAO;GAIT,MAAM,iCAAiC,YAAY,WAAW,MAC5D,CAAC,aAAa,SAAS,KAAK,WAAW,WAAW,CACnD;AAED,OAAI,+BAGF,QAAO;;;;AAMT,OAAI,KAAKnB,yBAAyB,KAChC,QAAO;;;;GAMT,MAAM,mBAAmB,YAAY,WAAW,OAC9C,CAAC,aAAa,CAAC,SAAS,KAAK,WAAW,WAAW,CACpD;AAED,OAAI,iBAAiB,WAAW,EAC9B,QAAO;AAGT,UAAO,iBAAiB,IACtB,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;;;;;;;;;;CAcD,wBACEc,aACAM,WACA;EACA,MAAM,wBAAwB,QAAQ,KAAK,QAAQ,eAAe;AAElE,SAAO,CAACC,UAA8D;GAGpE,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AACnC,OACE,UAAU,WAAW,YAAY,KAChC,CAAC,YAAY,cAAc,YAAY,WAAW,WAAW,GAE9D,QAAO;AAIT,OAAI,aAAa,MAAM,QAAQ;AAC7B,QAAI,MAAM,WAAW,IACnB,QAAO;AAET,QAAI,MAAM,WAAW,SAAS;AAE5B,SAAI,YAAY,WAAW,EACzB,QAAO;AAET,YAAO,IAAI,KAAK,SAAS;MAAE,GAAG;MAAO,QAAQ;KAAW;IACzD;AAED,WAAO,IAAI,KAAK,iBAAiB;KAAE,GAAG;KAAO,QAAQ;IAAW;GACjE;GAGD,MAAM,eAAe,SAAS,OAAO,YAAY,WAAW;GAC5D,MAAM,gBAAgB,SAAS,OAAO,UAAU,WAAW,CAAC,GAAG,GAAG;GAClE,MAAM,mBAAmB,eAAe,YAAY,OAClD,CAAC,SAAS,CAAC,aAAa,KAAK,CAAC,MAAM,EAAE,iBAAiB,KAAK,GAAG,CAChE;AACD,OAAI,oBAAoB,iBAAiB,SAAS,EAChD,QAAO,iBAAiB,IACtB,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;GAKH,MAAM,6BAA6B,eAAe,YAAY,KAC5D,CAAC,aAAa,SAAS,KAAK,WAAW,WAAW,CACnD;AAED,OACE,oBACA,iBAAiB,WAAW,KAC5B,CAAC,8BACD,sBAEA,QAAO;AAGT,OACE,CAAC,UAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,EAElC,QAAO;GAIT,MAAM,iCAAiC,YAAY,WAAW,MAC5D,CAAC,aAAa,SAAS,KAAK,WAAW,WAAW,CACnD;GAGD,MAAM,sBAAsB,YAAY,WAAW,KACjD,CAAC,aAAa,CAAC,SAAS,KAAK,WAAW,WAAW,CACpD;AAED,OAAI,kCAAkC,CAAC,oBACrC,QAAO;;;;;AAOT,UAAO;EACR;CACF;;;;;CAMD,gCACEP,aACAQ,SACAC,aACA;EACA,MAAM,aAAa,IAAI,IAAI,QAAQ,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC;AACnE,SAAO,CAACJ,UAAwB;AAC9B,OAAI,MAAM,QAAQ;IAChB,MAAM,OAAO,kBAAkB,MAAM,OAAO;AAC5C,QAAI,SAAS,OAAO,WAAW,IAAI,IAAI,CACrC,QAAO;AAET,QAAI,SAAS,WAAW,WAAW,IAAI,QAAQ,EAAE;AAC/C,SAAI,YAAY,WAAW,EAAG,QAAO;AACrC,YAAO,IAAI,KAAK,SAAS;MAAE,GAAG;MAAO,QAAQ;KAAW;IACzD;AACD,QAAI,SAAS,mBAAmB,WAAW,IAAI,gBAAgB,CAC7D,QAAO,IAAI,KAAK,iBAAiB;KAAE,GAAG;KAAO,QAAQ;IAAW;GAEnE;AACD,UAAO;EACR;CACF;;;;;CAMD,yBACEL,aACAS,aACA;AACA,SAAO,CAACJ,UAAwB;AAC9B,OAAI,CAAC,MAAM,OACT,QAAO;GAET,MAAM,cAAc,kBAAkB,MAAM,OAAO;AACnD,OAAI,gBAAgB,IAClB,QAAO;AAET,OAAI,gBAAgB,SAAS;AAC3B,QAAI,YAAY,WAAW,EACzB,QAAO;AAET,WAAO,IAAI,KAAK,SAAS;KAAE,GAAG;KAAO,QAAQ;IAAW;GACzD;AAED,UAAO,IAAI,KAAK,iBAAiB;IAAE,GAAG;IAAO,QAAQ;GAAW;EACjE;CACF;;;;CAKD,MAAMK,4BACJC,OAC4C;AAC5C,MACE,CAAC,KAAK,QAAQ,cACd,KAAK,QAAQ,WAAW,WAAW,KACnC,iBAAiB,WACjB,CAAC,MAED,QAAO;EAGT,MAAM,gBAAgB,MAAM,2BAC1B,KAAK,QAAQ,YACb,MACD;EACD,MAAM,eAAe,EAAE,GAAG,MAAO;AACjC,MAAI,CAAC,aACH,QAAO;AAIT,OAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,cAAc,CACtD,KAAI,EAAE,OAAO,eACX,aAAa,OAAoC;AAIrD,SAAO;CACR;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6CD,MAAM,OACJA,OACAC,QAIA;EAEA,MAAM,mBAAmB,MAAM,KAAKF,4BAA4B,MAAM;AACtE,SAAO,KAAKX,OAAO,OACjB,kBACA,OAED;CACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAmDD,MAAM,OACJY,OACAE,QAIsC;EACtC,MAAM,mBAAmB,MAAM,KAAKH,4BAA4B,MAAM;AACtE,SAAO,KAAKX,OAAO,aAAa,kBAAkB;GAChD,GAAG;GACH,SAAS;EACV,EAAQ;CACV;;;;;;;;;;;CAYD,MAAM,eAAee,QAMlB;EACD,MAAM,iBAAiB,MAAM,KAAKf,OAAO,eAAe;EACxD,MAAM,QAAQ,MAAM,eAAe,eAAe,OAAO;EACzD,MAAM,cAAc,MAAM,MAAM,aAAa;EAC7C,MAAM,SAAS,IAAI,WAAW;AAC9B,SAAO;CACR;;;;;;;;;;;CAYD,MAAM,YAAYe,QAMf;EACD,MAAM,iBAAiB,MAAM,KAAKf,OAAO,eAAe;AACxD,SAAO,eAAe,YAAY,OAAO;CAC1C;AACF"}

package/dist/agents/middlewareAgent/annotation.cjs

@@ -1,45 +0,0 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const __langchain_langgraph = require_rolldown_runtime.__toESM(require("@langchain/langgraph"));
-
-//#region src/agents/middlewareAgent/annotation.ts
-function createAgentAnnotationConditional(hasStructuredResponse = true, middlewareList = []) {
-	const baseAnnotation = {
-		messages: (0, __langchain_langgraph.Annotation)({
-			reducer: __langchain_langgraph.messagesStateReducer,
-			default: () => []
-		}),
-		jumpTo: (0, __langchain_langgraph.Annotation)({
-			reducer: (_x, y) => y,
-			default: () => void 0
-		})
-	};
-	for (const middleware of middlewareList) if (middleware.stateSchema) {
-		let parsedDefaults = {};
-		try {
-			parsedDefaults = middleware.stateSchema.parse({});
-		} catch {}
-		const { shape } = middleware.stateSchema;
-		for (const [key] of Object.entries(shape)) {
-			/**
-			* Skip private state properties
-			*/
-			if (key.startsWith("_")) continue;
-			if (!(key in baseAnnotation)) {
-				const defaultValue = parsedDefaults[key] ?? void 0;
-				baseAnnotation[key] = (0, __langchain_langgraph.Annotation)({
-					reducer: (x, y) => y ?? x,
-					default: () => defaultValue
-				});
-			}
-		}
-	}
-	if (!hasStructuredResponse) return __langchain_langgraph.Annotation.Root(baseAnnotation);
-	return __langchain_langgraph.Annotation.Root({
-		...baseAnnotation,
-		structuredResponse: (0, __langchain_langgraph.Annotation)()
-	});
-}
-
-//#endregion
-exports.createAgentAnnotationConditional = createAgentAnnotationConditional;
-//# sourceMappingURL=annotation.cjs.map

package/dist/agents/middlewareAgent/annotation.cjs.map

@@ -1 +0,0 @@
-{"version":3,"file":"annotation.cjs","names":["middlewareList: TMiddleware","baseAnnotation: Record<string, any>","messagesStateReducer","_x: any","y: any","parsedDefaults: Record<string, any>","x: any","Annotation"],"sources":["../../../src/agents/middlewareAgent/annotation.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport {\n  Annotation,\n  Messages,\n  AnnotationRoot,\n  messagesStateReducer,\n  type BinaryOperatorAggregate,\n  type LastValue,\n} from \"@langchain/langgraph\";\nimport type { ResponseFormatUndefined } from \"../annotation.js\";\nimport type { AgentMiddleware, InferMiddlewareStates } from \"./types.js\";\n\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\n// Helper type for the merged annotation\ntype MergedAnnotationSpec<\n  T extends Record<string, any> | ResponseFormatUndefined,\n  TMiddleware extends readonly AgentMiddleware<any, any, any>[]\n> = {\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  jumpTo: LastValue<\"model_request\" | \"tools\" | undefined>;\n} & (T extends ResponseFormatUndefined\n  ? {}\n  : { structuredResponse: LastValue<T> }) &\n  InferMiddlewareStates<TMiddleware>;\n\nexport function createAgentAnnotationConditional<\n  T extends Record<string, any> | ResponseFormatUndefined,\n  TMiddleware extends readonly AgentMiddleware<any, any, any>[] = []\n>(\n  hasStructuredResponse = true,\n  middlewareList: TMiddleware = [] as unknown as TMiddleware\n): AnnotationRoot<MergedAnnotationSpec<T, TMiddleware>> {\n  const baseAnnotation: Record<string, any> = {\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n    jumpTo: Annotation<\"model_request\" | \"tools\" | undefined>({\n      /**\n       * Since `jumpTo` acts as a control command, we only want\n       * to apply it if explicitly set.\n       */\n      reducer: (_x: any, y: any) => y,\n      default: () => undefined,\n    }),\n  };\n\n  // Add middleware state properties to the annotation\n  for (const middleware of middlewareList) {\n    if (middleware.stateSchema) {\n      // Parse empty object to get default values\n      let parsedDefaults: Record<string, any> = {};\n      try {\n        parsedDefaults = middleware.stateSchema.parse({});\n      } catch {\n        // If parsing fails, we'll use undefined as defaults\n      }\n\n      const { shape } = middleware.stateSchema;\n      for (const [key] of Object.entries(shape)) {\n        /**\n         * Skip private state properties\n         */\n        if (key.startsWith(\"_\")) {\n          continue;\n        }\n\n        if (!(key in baseAnnotation)) {\n          const defaultValue = parsedDefaults[key] ?? undefined;\n          baseAnnotation[key] = Annotation({\n            reducer: (x: any, y: any) => y ?? x,\n            default: () => defaultValue,\n          });\n        }\n      }\n    }\n  }\n\n  if (!hasStructuredResponse) {\n    return Annotation.Root(baseAnnotation) as AnnotationRoot<\n      MergedAnnotationSpec<T, TMiddleware>\n    >;\n  }\n\n  return Annotation.Root({\n    ...baseAnnotation,\n    structuredResponse:\n      Annotation<T extends ResponseFormatUndefined ? never : T>(),\n  }) as unknown as AnnotationRoot<MergedAnnotationSpec<T, TMiddleware>>;\n}\n"],"mappings":";;;;AA0BA,SAAgB,iCAId,wBAAwB,MACxBA,iBAA8B,CAAE,GACsB;CACtD,MAAMC,iBAAsC;EAC1C,gDAA8C;GAC5C,SAASC;GACT,SAAS,MAAM,CAAE;EAClB,EAAC;EACF,8CAA0D;GAKxD,SAAS,CAACC,IAASC,MAAW;GAC9B,SAAS,MAAM;EAChB,EAAC;CACH;AAGD,MAAK,MAAM,cAAc,eACvB,KAAI,WAAW,aAAa;EAE1B,IAAIC,iBAAsC,CAAE;AAC5C,MAAI;GACF,iBAAiB,WAAW,YAAY,MAAM,CAAE,EAAC;EAClD,QAAO,CAEP;EAED,MAAM,EAAE,OAAO,GAAG,WAAW;AAC7B,OAAK,MAAM,CAAC,IAAI,IAAI,OAAO,QAAQ,MAAM,EAAE;;;;AAIzC,OAAI,IAAI,WAAW,IAAI,CACrB;AAGF,OAAI,EAAE,OAAO,iBAAiB;IAC5B,MAAM,eAAe,eAAe,QAAQ;IAC5C,eAAe,6CAAkB;KAC/B,SAAS,CAACC,GAAQF,MAAW,KAAK;KAClC,SAAS,MAAM;IAChB,EAAC;GACH;EACF;CACF;AAGH,KAAI,CAAC,sBACH,QAAOG,iCAAW,KAAK,eAAe;AAKxC,QAAOA,iCAAW,KAAK;EACrB,GAAG;EACH,2DAC6D;CAC9D,EAAC;AACH"}

package/dist/agents/middlewareAgent/annotation.js

@@ -1,44 +0,0 @@
-import { Annotation, messagesStateReducer } from "@langchain/langgraph";
-
-//#region src/agents/middlewareAgent/annotation.ts
-function createAgentAnnotationConditional(hasStructuredResponse = true, middlewareList = []) {
-	const baseAnnotation = {
-		messages: Annotation({
-			reducer: messagesStateReducer,
-			default: () => []
-		}),
-		jumpTo: Annotation({
-			reducer: (_x, y) => y,
-			default: () => void 0
-		})
-	};
-	for (const middleware of middlewareList) if (middleware.stateSchema) {
-		let parsedDefaults = {};
-		try {
-			parsedDefaults = middleware.stateSchema.parse({});
-		} catch {}
-		const { shape } = middleware.stateSchema;
-		for (const [key] of Object.entries(shape)) {
-			/**
-			* Skip private state properties
-			*/
-			if (key.startsWith("_")) continue;
-			if (!(key in baseAnnotation)) {
-				const defaultValue = parsedDefaults[key] ?? void 0;
-				baseAnnotation[key] = Annotation({
-					reducer: (x, y) => y ?? x,
-					default: () => defaultValue
-				});
-			}
-		}
-	}
-	if (!hasStructuredResponse) return Annotation.Root(baseAnnotation);
-	return Annotation.Root({
-		...baseAnnotation,
-		structuredResponse: Annotation()
-	});
-}
-
-//#endregion
-export { createAgentAnnotationConditional };
-//# sourceMappingURL=annotation.js.map