langchain 1.0.0-alpha.3 → 1.0.0-alpha.5

package/dist/agents/middlewareAgent/ReactAgent.js

@@ -0,0 +1,254 @@
+import { isClientTool, validateLLMHasNoBoundTools } from "../utils.js";
+import { ToolNode } from "../nodes/ToolNode.js";
+import { createAgentAnnotationConditional } from "./annotation.js";
+import { initializeMiddlewareStates } 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;
+		/**
+		* Check if the LLM already has bound tools and throw if it does.
+		*/
+		if (options.llm && typeof options.llm !== "function") validateLLMHasNoBoundTools(options.llm);
+		/**
+		* validate that model and llm options are not provided together
+		*/
+		if (options.llm && options.model) throw new Error("Cannot provide both `model` and `llm` options.");
+		/**
+		* validate that either model or llm option is provided
+		*/
+		if (!options.llm && !options.model) throw new Error("Either `model` or `llm` option must be provided to create an agent.");
+		const toolClasses = (Array.isArray(options.tools) ? options.tools : options.tools?.tools) ?? [];
+		/**
+		* 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.middlewares);
+		const workflow = new StateGraph(schema, this.options.contextSchema);
+		const allNodeWorkflows = workflow;
+		const beforeModelNodes = [];
+		const afterModelNodes = [];
+		const prepareModelRequestHookMiddlewares = [];
+		const middlewares = this.options.middlewares ?? [];
+		for (let i = 0; i < middlewares.length; i++) {
+			let beforeModelNode;
+			let afterModelNode;
+			const middleware = middlewares[i];
+			if (middleware.beforeModel) {
+				beforeModelNode = new BeforeModelNode(middleware);
+				const name = `before_model_${middleware.name}_${i}`;
+				beforeModelNodes.push({
+					index: i,
+					name
+				});
+				allNodeWorkflows.addNode(name, beforeModelNode, beforeModelNode.nodeOptions);
+			}
+			if (middleware.afterModel) {
+				afterModelNode = new AfterModelNode(middleware);
+				const name = `after_model_${middleware.name}_${i}`;
+				afterModelNodes.push({
+					index: i,
+					name
+				});
+				allNodeWorkflows.addNode(name, afterModelNode, afterModelNode.nodeOptions);
+			}
+			if (middleware.prepareModelRequest) prepareModelRequestHookMiddlewares.push([middleware, () => ({
+				...beforeModelNode?.getState(),
+				...afterModelNode?.getState()
+			})]);
+		}
+		/**
+		* Add Nodes
+		*/
+		allNodeWorkflows.addNode("model_request", new AgentNode({
+			llm: this.options.llm,
+			model: this.options.model,
+			prompt: this.options.prompt,
+			includeAgentName: this.options.includeAgentName,
+			name: this.options.name,
+			responseFormat: this.options.responseFormat,
+			middlewares: this.options.middlewares,
+			toolClasses,
+			shouldReturnDirect,
+			signal: this.options.signal,
+			prepareModelRequestHookMiddlewares
+		}), 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 - 1; i++) allNodeWorkflows.addEdge(beforeModelNodes[i].name, beforeModelNodes[i + 1].name);
+		const lastBeforeModelNode = beforeModelNodes.at(-1);
+		if (beforeModelNodes.length > 0 && lastBeforeModelNode) allNodeWorkflows.addEdge(lastBeforeModelNode.name, "model_request");
+		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--) allNodeWorkflows.addEdge(afterModelNodes[i].name, afterModelNodes[i - 1].name);
+		if (afterModelNodes.length > 0) {
+			const firstAfterModelNode = afterModelNodes[0].name;
+			const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));
+			if (modelPaths.length === 1) allNodeWorkflows.addEdge(firstAfterModelNode, modelPaths[0]);
+			else allNodeWorkflows.addConditionalEdges(firstAfterModelNode, this.#createModelRouter(), modelPaths);
+		}
+		/**
+		* 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,
+			interruptBefore: this.options.interruptBefore,
+			interruptAfter: this.options.interruptAfter,
+			store: this.options.store,
+			name: this.options.name,
+			description: this.options.description
+		});
+	}
+	/**
+	* Get the compiled graph.
+	*/
+	get graph() {
+		return this.#graph;
+	}
+	/**
+	* Get possible edge destinations from model node.
+	* @param toolClasses names of tools to call
+	* @returns list of possible edge destinations
+	*/
+	#getModelPaths(toolClasses) {
+		const paths = [];
+		if (toolClasses.length > 0) paths.push("tools");
+		paths.push(END);
+		return paths;
+	}
+	/**
+	* Create routing function for model node conditional edges.
+	*/
+	#createModelRouter() {
+		/**
+		* determine if the agent should continue or not
+		*/
+		/**
+		* ToDo: fix type
+		*/
+		return (state) => {
+			const messages = state.messages;
+			const lastMessage = messages.at(-1);
+			if (!(lastMessage instanceof AIMessage) || !lastMessage.tool_calls || lastMessage.tool_calls.length === 0) return END;
+			/**
+			* The tool node processes a single message.
+			*/
+			if (this.#toolBehaviorVersion === "v1") return "tools";
+			/**
+			* Route to tools node
+			*/
+			return lastMessage.tool_calls.map((toolCall) => new Send("tools", {
+				...state,
+				lg_tool_call: toolCall
+			}));
+		};
+	}
+	/**
+	* 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 (lastMessage instanceof ToolMessage && lastMessage.name && shouldReturnDirect.has(lastMessage.name)) return this.options.responseFormat ? "model_request" : END;
+			return "model_request";
+		};
+	}
+	/**
+	* Initialize middleware states if not already present in the input state.
+	*/
+	#initializeMiddlewareStates(state) {
+		if (!this.options.middlewares || this.options.middlewares.length === 0 || state instanceof Command || !state) return state;
+		const defaultStates = initializeMiddlewareStates(this.options.middlewares, 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;
+	}
+	/**
+	* @inheritdoc
+	*/
+	get invoke() {
+		const invokeFunc = async (state, config) => {
+			const initializedState = this.#initializeMiddlewareStates(state);
+			return this.#graph.invoke(initializedState, config);
+		};
+		return invokeFunc;
+	}
+	/**
+	* ToDo(@christian-bromann): Add stream and streamEvents methods
+	*/
+	/**
+	* 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

@@ -0,0 +1 @@
+{"version":3,"file":"ReactAgent.js","names":["options: CreateAgentParams<StructuredResponseFormat, ContextSchema>","#toolBehaviorVersion","beforeModelNodes: { index: number; name: string }[]","afterModelNodes: { index: number; name: string }[]","prepareModelRequestHookMiddlewares: [\n      AgentMiddleware,\n      /**\n       * ToDo: better type to get the state of middleware\n       */\n      () => any\n    ][]","beforeModelNode: BeforeModelNode | undefined","afterModelNode: AfterModelNode | undefined","#getModelPaths","#createModelRouter","toolReturnTarget: string","#createToolsRouter","#graph","toolClasses: (ClientTool | ServerTool)[]","paths: (\"tools\" | typeof END)[]","state: any","shouldReturnDirect: Set<string>","state: InvokeStateParameter<TMiddlewares>","invokeFunc: InvokeFunction","config?: LangGraphRunnableConfig<FullContext>","#initializeMiddlewareStates","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 prefer-destructuring, no-instanceof/no-instanceof, @typescript-eslint/ban-types */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\n\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  Command,\n  CompiledStateGraph,\n  type LangGraphRunnableConfig,\n} from \"@langchain/langgraph\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\n\nimport { createAgentAnnotationConditional } from \"./annotation.js\";\nimport { isClientTool, validateLLMHasNoBoundTools } 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 { initializeMiddlewareStates } from \"./nodes/utils.js\";\n\nimport type { ClientTool, ServerTool, WithStateGraphNodes } from \"../types.js\";\n\nimport {\n  CreateAgentParams,\n  AgentMiddleware,\n  InferMiddlewareStates,\n  InferMiddlewareInputStates,\n  BuiltInState,\n  InferMiddlewareContextInputs,\n  IsAllOptional,\n  InferContextInput,\n} from \"./types.js\";\n\nimport {\n  type AnyAnnotationRoot,\n  type ToAnnotationRoot,\n} from \"../annotation.js\";\nimport type { ResponseFormatUndefined } from \"../responses.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  TMiddlewares extends readonly AgentMiddleware<any, any, any>[]\n> = (StructuredResponseFormat extends ResponseFormatUndefined\n  ? BuiltInState\n  : BuiltInState & { structuredResponse: StructuredResponseFormat }) &\n  InferMiddlewareStates<TMiddlewares>;\n\ntype InvokeStateParameter<\n  TMiddlewares extends readonly AgentMiddleware<any, any, any>[]\n> =\n  | (BuiltInState & InferMiddlewareInputStates<TMiddlewares>)\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  TMiddlewares extends readonly AgentMiddleware<any, any, any>[] = []\n> = CompiledStateGraph<\n  any,\n  any,\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  any,\n  any,\n  MergedAgentState<StructuredResponseFormat, TMiddlewares>,\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  TMiddlewares extends readonly AgentMiddleware<any, any, any>[] = []\n> {\n  #graph: AgentGraph<StructuredResponseFormat, ContextSchema, TMiddlewares>;\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     * Check if the LLM already has bound tools and throw if it does.\n     */\n    if (options.llm && typeof options.llm !== \"function\") {\n      validateLLMHasNoBoundTools(options.llm);\n    }\n\n    /**\n     * validate that model and llm options are not provided together\n     */\n    if (options.llm && options.model) {\n      throw new Error(\"Cannot provide both `model` and `llm` options.\");\n    }\n\n    /**\n     * validate that either model or llm option is provided\n     */\n    if (!options.llm && !options.model) {\n      throw new Error(\n        \"Either `model` or `llm` option must be provided to create an agent.\"\n      );\n    }\n\n    const toolClasses =\n      (Array.isArray(options.tools) ? options.tools : options.tools?.tools) ??\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      TMiddlewares\n    >(\n      this.options.responseFormat !== undefined,\n      this.options.middlewares as TMiddlewares\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: { index: number; name: string }[] = [];\n    const afterModelNodes: { index: number; name: string }[] = [];\n    const prepareModelRequestHookMiddlewares: [\n      AgentMiddleware,\n      /**\n       * ToDo: better type to get the state of middleware\n       */\n      () => any\n    ][] = [];\n\n    const middlewares = this.options.middlewares ?? [];\n    for (let i = 0; i < middlewares.length; i++) {\n      let beforeModelNode: BeforeModelNode | undefined;\n      let afterModelNode: AfterModelNode | undefined;\n      const middleware = middlewares[i];\n      if (middleware.beforeModel) {\n        beforeModelNode = new BeforeModelNode(middleware);\n        const name = `before_model_${middleware.name}_${i}`;\n        beforeModelNodes.push({\n          index: i,\n          name,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          beforeModelNode,\n          beforeModelNode.nodeOptions\n        );\n      }\n      if (middleware.afterModel) {\n        afterModelNode = new AfterModelNode(middleware);\n        const name = `after_model_${middleware.name}_${i}`;\n        afterModelNodes.push({\n          index: i,\n          name,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          afterModelNode,\n          afterModelNode.nodeOptions\n        );\n      }\n\n      if (middleware.prepareModelRequest) {\n        prepareModelRequestHookMiddlewares.push([\n          middleware,\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        llm: this.options.llm,\n        model: this.options.model,\n        prompt: this.options.prompt,\n        includeAgentName: this.options.includeAgentName,\n        name: this.options.name,\n        responseFormat: this.options.responseFormat,\n        middlewares: this.options.middlewares,\n        toolClasses,\n        shouldReturnDirect,\n        signal: this.options.signal,\n        prepareModelRequestHookMiddlewares,\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 in sequence\n    for (let i = 0; i < beforeModelNodes.length - 1; i++) {\n      allNodeWorkflows.addEdge(\n        beforeModelNodes[i].name,\n        beforeModelNodes[i + 1].name\n      );\n    }\n\n    // Connect last beforeModel node to agent\n    const lastBeforeModelNode = beforeModelNodes.at(-1);\n    if (beforeModelNodes.length > 0 && lastBeforeModelNode) {\n      allNodeWorkflows.addEdge(lastBeforeModelNode.name, \"model_request\");\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      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\n    for (let i = afterModelNodes.length - 1; i > 0; i--) {\n      allNodeWorkflows.addEdge(\n        afterModelNodes[i].name,\n        afterModelNodes[i - 1].name\n      );\n    }\n\n    // Connect first afterModel node (last to execute) to model paths\n    if (afterModelNodes.length > 0) {\n      const firstAfterModelNode = afterModelNodes[0].name;\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      if (modelPaths.length === 1) {\n        allNodeWorkflows.addEdge(firstAfterModelNode, modelPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          firstAfterModelNode,\n          this.#createModelRouter(),\n          modelPaths\n        );\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      interruptBefore: this.options.interruptBefore,\n      interruptAfter: this.options.interruptAfter,\n      store: this.options.store,\n      name: this.options.name,\n      description: this.options.description,\n    }) as AgentGraph<StructuredResponseFormat, ContextSchema, TMiddlewares>;\n  }\n\n  /**\n   * Get the compiled graph.\n   */\n  get graph(): AgentGraph<\n    StructuredResponseFormat,\n    ContextSchema,\n    TMiddlewares\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   * @returns list of possible edge destinations\n   */\n  #getModelPaths(\n    toolClasses: (ClientTool | ServerTool)[]\n  ): (\"tools\" | typeof END)[] {\n    const paths: (\"tools\" | typeof END)[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(\"tools\");\n    }\n\n    paths.push(END);\n\n    return paths;\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    /**\n     * ToDo: fix type\n     */\n    return (state: any) => {\n      const messages = state.messages;\n      const lastMessage = messages.at(-1);\n\n      if (\n        !(lastMessage instanceof AIMessage) ||\n        !lastMessage.tool_calls ||\n        lastMessage.tool_calls.length === 0\n      ) {\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\n       */\n      return lastMessage.tool_calls.map(\n        (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n      );\n    };\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        lastMessage instanceof ToolMessage &&\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   * Initialize middleware states if not already present in the input state.\n   */\n  #initializeMiddlewareStates(\n    state: InvokeStateParameter<TMiddlewares>\n  ): InvokeStateParameter<TMiddlewares> {\n    if (\n      !this.options.middlewares ||\n      this.options.middlewares.length === 0 ||\n      state instanceof Command ||\n      !state\n    ) {\n      return state;\n    }\n\n    const defaultStates = initializeMiddlewareStates(\n      this.options.middlewares,\n      state\n    );\n    const updatedState = { ...state } as InvokeStateParameter<TMiddlewares>;\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        // @ts-expect-error - ToDo: fix type\n        updatedState[key as keyof InvokeStateParameter<TMiddlewares>] = value;\n      }\n    }\n\n    return updatedState;\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get invoke() {\n    type FullState = MergedAgentState<StructuredResponseFormat, TMiddlewares>;\n    type FullContext = InferContextInput<ContextSchema> &\n      InferMiddlewareContextInputs<TMiddlewares>;\n\n    // Create overloaded function type based on whether context has required fields\n    type InvokeFunction = IsAllOptional<FullContext> extends true\n      ? (\n          state: InvokeStateParameter<TMiddlewares>,\n          config?: LangGraphRunnableConfig<FullContext>\n        ) => Promise<FullState>\n      : (\n          state: InvokeStateParameter<TMiddlewares>,\n          config?: LangGraphRunnableConfig<FullContext>\n        ) => Promise<FullState>;\n\n    const invokeFunc: InvokeFunction = async (\n      state: InvokeStateParameter<TMiddlewares>,\n      config?: LangGraphRunnableConfig<FullContext>\n    ): Promise<FullState> => {\n      const initializedState = this.#initializeMiddlewareStates(state);\n      return this.#graph.invoke(\n        initializedState,\n        config as any\n      ) as Promise<FullState>;\n    };\n\n    return invokeFunc;\n  }\n\n  /**\n   * ToDo(@christian-bromann): Add stream and streamEvents methods\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":";;;;;;;;;;;AAgFA,IAAa,aAAb,MAQE;CACA;CAEA,uBAAoC;CAEpC,YACSA,SACP;EADO;EAEP,KAAKC,uBAAuB,QAAQ,WAAW,KAAKA;;;;AAKpD,MAAI,QAAQ,OAAO,OAAO,QAAQ,QAAQ,YACxC,2BAA2B,QAAQ,IAAI;;;;AAMzC,MAAI,QAAQ,OAAO,QAAQ,MACzB,OAAM,IAAI,MAAM;;;;AAMlB,MAAI,CAAC,QAAQ,OAAO,CAAC,QAAQ,MAC3B,OAAM,IAAI,MACR;EAIJ,MAAM,eACH,MAAM,QAAQ,QAAQ,MAAM,GAAG,QAAQ,QAAQ,QAAQ,OAAO,UAC/D,CAAE;;;;;EAMJ,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,YACd;EAED,MAAM,WAAW,IAAI,WACnB,QACA,KAAK,QAAQ;EAGf,MAAM,mBAAmB;EAMzB,MAAMC,mBAAsD,CAAE;EAC9D,MAAMC,kBAAqD,CAAE;EAC7D,MAAMC,qCAMA,CAAE;EAER,MAAM,cAAc,KAAK,QAAQ,eAAe,CAAE;AAClD,OAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,KAAK;GAC3C,IAAIC;GACJ,IAAIC;GACJ,MAAM,aAAa,YAAY;AAC/B,OAAI,WAAW,aAAa;IAC1B,kBAAkB,IAAI,gBAAgB;IACtC,MAAM,OAAO,CAAC,aAAa,EAAE,WAAW,KAAK,CAAC,EAAE,GAAG;IACnD,iBAAiB,KAAK;KACpB,OAAO;KACP;IACD,EAAC;IACF,iBAAiB,QACf,MACA,iBACA,gBAAgB,YACjB;GACF;AACD,OAAI,WAAW,YAAY;IACzB,iBAAiB,IAAI,eAAe;IACpC,MAAM,OAAO,CAAC,YAAY,EAAE,WAAW,KAAK,CAAC,EAAE,GAAG;IAClD,gBAAgB,KAAK;KACnB,OAAO;KACP;IACD,EAAC;IACF,iBAAiB,QACf,MACA,gBACA,eAAe,YAChB;GACF;AAED,OAAI,WAAW,qBACb,mCAAmC,KAAK,CACtC,YACA,OAAO;IACL,GAAG,iBAAiB,UAAU;IAC9B,GAAG,gBAAgB,UAAU;GAC9B,EACF,EAAC;EAEL;;;;EAKD,iBAAiB,QACf,iBACA,IAAI,UAAU;GACZ,KAAK,KAAK,QAAQ;GAClB,OAAO,KAAK,QAAQ;GACpB,QAAQ,KAAK,QAAQ;GACrB,kBAAkB,KAAK,QAAQ;GAC/B,MAAM,KAAK,QAAQ;GACnB,gBAAgB,KAAK,QAAQ;GAC7B,aAAa,KAAK,QAAQ;GAC1B;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,SAAS,GAAG,KAC/C,iBAAiB,QACf,iBAAiB,GAAG,MACpB,iBAAiB,IAAI,GAAG,KACzB;EAIH,MAAM,sBAAsB,iBAAiB,GAAG,GAAG;AACnD,MAAI,iBAAiB,SAAS,KAAK,qBACjC,iBAAiB,QAAQ,oBAAoB,MAAM,gBAAgB;EAIrE,MAAM,qBAAqB,gBAAgB,GAAG,GAAG;AACjD,MAAI,gBAAgB,SAAS,KAAK,oBAChC,iBAAiB,QAAQ,iBAAiB,mBAAmB,KAAK;OAC7D;GACL,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,KAC9C,iBAAiB,QACf,gBAAgB,GAAG,MACnB,gBAAgB,IAAI,GAAG,KACxB;AAIH,MAAI,gBAAgB,SAAS,GAAG;GAC9B,MAAM,sBAAsB,gBAAgB,GAAG;GAC/C,MAAM,aAAa,KAAKD,eAAe,YAAY,OAAO,aAAa,CAAC;AACxE,OAAI,WAAW,WAAW,GACxB,iBAAiB,QAAQ,qBAAqB,WAAW,GAAG;QAE5D,iBAAiB,oBACf,qBACA,KAAKC,oBAAoB,EACzB,WACD;EAEJ;;;;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,iBAAiB,KAAK,QAAQ;GAC9B,gBAAgB,KAAK,QAAQ;GAC7B,OAAO,KAAK,QAAQ;GACpB,MAAM,KAAK,QAAQ;GACnB,aAAa,KAAK,QAAQ;EAC3B,EAAC;CACH;;;;CAKD,IAAI,QAIF;AACA,SAAO,KAAKA;CACb;;;;;;CAOD,eACEC,aAC0B;EAC1B,MAAMC,QAAkC,CAAE;AAC1C,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;EAGrB,MAAM,KAAK,IAAI;AAEf,SAAO;CACR;;;;CAKD,qBAAqB;;;;;;;AAOnB,SAAO,CAACC,UAAe;GACrB,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,EAAE,uBAAuB,cACzB,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,EAElC,QAAO;;;;AAMT,OAAI,KAAKb,yBAAyB,KAChC,QAAO;;;;AAMT,UAAO,YAAY,WAAW,IAC5B,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;CAKD,mBAAmBc,oBAAiC;;;;AAIlD,SAAO,CAACD,UAAe;GACrB,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,SAAS,SAAS;AAG/C,OACE,uBAAuB,eACvB,YAAY,QACZ,mBAAmB,IAAI,YAAY,KAAK,CAIxC,QAAO,KAAK,QAAQ,iBAAiB,kBAAkB;AAIzD,UAAO;EACR;CACF;;;;CAKD,4BACEE,OACoC;AACpC,MACE,CAAC,KAAK,QAAQ,eACd,KAAK,QAAQ,YAAY,WAAW,KACpC,iBAAiB,WACjB,CAAC,MAED,QAAO;EAGT,MAAM,gBAAgB,2BACpB,KAAK,QAAQ,aACb,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,eAEX,aAAa,OAAmD;AAIpE,SAAO;CACR;;;;CAKD,IAAI,SAAS;EAgBX,MAAMC,aAA6B,OACjCD,OACAE,WACuB;GACvB,MAAM,mBAAmB,KAAKC,4BAA4B,MAAM;AAChE,UAAO,KAAKR,OAAO,OACjB,kBACA,OACD;EACF;AAED,SAAO;CACR;;;;;;;;;;;;;;CAgBD,MAAM,eAAeS,QAMlB;EACD,MAAM,iBAAiB,MAAM,KAAKT,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,YAAYS,QAMf;EACD,MAAM,iBAAiB,MAAM,KAAKT,OAAO,eAAe;AACxD,SAAO,eAAe,YAAY,OAAO;CAC1C;AACF"}

package/dist/agents/middlewareAgent/annotation.cjs

@@ -0,0 +1,39 @@
+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, middlewares = []) {
+	const baseAnnotation = { messages: (0, __langchain_langgraph.Annotation)({
+		reducer: __langchain_langgraph.messagesStateReducer,
+		default: () => []
+	}) };
+	for (const middleware of middlewares) 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

@@ -0,0 +1 @@
+{"version":3,"file":"annotation.cjs","names":["middlewares: TMiddlewares","baseAnnotation: Record<string, any>","messagesStateReducer","parsedDefaults: Record<string, any>","x: any","y: any","Annotation"],"sources":["../../../src/agents/middlewareAgent/annotation.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\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 \"../responses.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  TMiddlewares extends readonly AgentMiddleware<any, any, any>[]\n> = {\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n} & (T extends ResponseFormatUndefined\n  ? {}\n  : { structuredResponse: LastValue<T> }) &\n  InferMiddlewareStates<TMiddlewares>;\n\nexport function createAgentAnnotationConditional<\n  T extends Record<string, any> | ResponseFormatUndefined,\n  TMiddlewares extends readonly AgentMiddleware<any, any, any>[] = []\n>(\n  hasStructuredResponse = true,\n  middlewares: TMiddlewares = [] as unknown as TMiddlewares\n): AnnotationRoot<MergedAnnotationSpec<T, TMiddlewares>> {\n  const baseAnnotation: Record<string, any> = {\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  };\n\n  // Add middleware state properties to the annotation\n  for (const middleware of middlewares) {\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, TMiddlewares>\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, TMiddlewares>>;\n}\n"],"mappings":";;;;AAyBA,SAAgB,iCAId,wBAAwB,MACxBA,cAA4B,CAAE,GACyB;CACvD,MAAMC,iBAAsC,EAC1C,gDAA8C;EAC5C,SAASC;EACT,SAAS,MAAM,CAAE;CAClB,EAAC,CACH;AAGD,MAAK,MAAM,cAAc,YACvB,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,GAAQC,MAAW,KAAK;KAClC,SAAS,MAAM;IAChB,EAAC;GACH;EACF;CACF;AAGH,KAAI,CAAC,sBACH,QAAOC,iCAAW,KAAK,eAAe;AAKxC,QAAOA,iCAAW,KAAK;EACrB,GAAG;EACH,2DAC6D;CAC9D,EAAC;AACH"}

package/dist/agents/middlewareAgent/annotation.js

@@ -0,0 +1,38 @@
+import { Annotation, messagesStateReducer } from "@langchain/langgraph";
+
+//#region src/agents/middlewareAgent/annotation.ts
+function createAgentAnnotationConditional(hasStructuredResponse = true, middlewares = []) {
+	const baseAnnotation = { messages: Annotation({
+		reducer: messagesStateReducer,
+		default: () => []
+	}) };
+	for (const middleware of middlewares) 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

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

@@ -0,0 +1 @@
+{"version":3,"file":"annotation.js","names":["middlewares: TMiddlewares","baseAnnotation: Record<string, any>","parsedDefaults: Record<string, any>","x: any","y: any"],"sources":["../../../src/agents/middlewareAgent/annotation.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\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 \"../responses.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  TMiddlewares extends readonly AgentMiddleware<any, any, any>[]\n> = {\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n} & (T extends ResponseFormatUndefined\n  ? {}\n  : { structuredResponse: LastValue<T> }) &\n  InferMiddlewareStates<TMiddlewares>;\n\nexport function createAgentAnnotationConditional<\n  T extends Record<string, any> | ResponseFormatUndefined,\n  TMiddlewares extends readonly AgentMiddleware<any, any, any>[] = []\n>(\n  hasStructuredResponse = true,\n  middlewares: TMiddlewares = [] as unknown as TMiddlewares\n): AnnotationRoot<MergedAnnotationSpec<T, TMiddlewares>> {\n  const baseAnnotation: Record<string, any> = {\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  };\n\n  // Add middleware state properties to the annotation\n  for (const middleware of middlewares) {\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, TMiddlewares>\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, TMiddlewares>>;\n}\n"],"mappings":";;;AAyBA,SAAgB,iCAId,wBAAwB,MACxBA,cAA4B,CAAE,GACyB;CACvD,MAAMC,iBAAsC,EAC1C,UAAU,WAAoC;EAC5C,SAAS;EACT,SAAS,MAAM,CAAE;CAClB,EAAC,CACH;AAGD,MAAK,MAAM,cAAc,YACvB,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,OAAO,WAAW;KAC/B,SAAS,CAACC,GAAQC,MAAW,KAAK;KAClC,SAAS,MAAM;IAChB,EAAC;GACH;EACF;CACF;AAGH,KAAI,CAAC,sBACH,QAAO,WAAW,KAAK,eAAe;AAKxC,QAAO,WAAW,KAAK;EACrB,GAAG;EACH,oBACE,YAA2D;CAC9D,EAAC;AACH"}

package/dist/agents/middlewareAgent/index.cjs

@@ -0,0 +1,11 @@
+const require_ReactAgent = require('./ReactAgent.cjs');
+const require_middleware = require('./middleware.cjs');
+
+//#region src/agents/middlewareAgent/index.ts
+function createAgent(params) {
+	return new require_ReactAgent.ReactAgent(params);
+}
+
+//#endregion
+exports.createAgent = createAgent;
+//# sourceMappingURL=index.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","names":["params: CreateAgentParams<StructuredResponseFormat, ContextSchema, any>","ReactAgent"],"sources":["../../../src/agents/middlewareAgent/index.ts"],"sourcesContent":["import type {\n  InteropZodObject,\n  InteropZodType,\n} from \"@langchain/core/utils/types\";\n\nimport { type AnyAnnotationRoot } from \"../annotation.js\";\nimport type { CreateAgentParams, AgentMiddleware } from \"./types.js\";\nimport type { ExtractZodArrayTypes } from \"../types.js\";\nimport type {\n  ToolStrategy,\n  TypedToolStrategy,\n  ProviderStrategy,\n  ResponseFormat,\n  ResponseFormatUndefined,\n  JsonSchemaFormat,\n} from \"../responses.js\";\nimport { ReactAgent } from \"./ReactAgent.js\";\n\n/**\n * Creates a StateGraph agent that relies on a chat model utilizing tool calling.\n *\n * @example\n * ```ts\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { createAgent, tool } from \"langchain\";\n * import { z } from \"zod\";\n *\n * const model = new ChatOpenAI({\n *   model: \"gpt-4o\",\n * });\n *\n * const getWeather = tool((input) => {\n *   if ([\"sf\", \"san francisco\"].includes(input.location.toLowerCase())) {\n *     return \"It's 60 degrees and foggy.\";\n *   } else {\n *     return \"It's 90 degrees and sunny.\";\n *   }\n * }, {\n *   name: \"get_weather\",\n *   description: \"Call to get the current weather.\",\n *   schema: z.object({\n *     location: z.string().describe(\"Location to get the weather for.\"),\n *   })\n * })\n *\n * const agent = createAgent({ llm: model, tools: [getWeather] });\n *\n * const inputs = {\n *   messages: [{ role: \"user\", content: \"what is the weather in SF?\" }],\n * };\n *\n * const stream = await agent.stream(inputs, { streamMode: \"values\" });\n *\n * for await (const { messages } of stream) {\n *   console.log(messages);\n * }\n * // Returns the messages in the state at each step of execution\n * ```\n */\n// Overload 1: With responseFormat as single InteropZodType\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, InteropZodType<T>> & {\n    responseFormat: InteropZodType<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 2: With responseFormat as array of InteropZodTypes (infers union type)\nexport function createAgent<\n  T extends readonly InteropZodType<any>[],\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    ExtractZodArrayTypes<T> extends Record<string, any>\n      ? ExtractZodArrayTypes<T>\n      : Record<string, any>,\n    ContextSchema,\n    T\n  > & {\n    responseFormat: T;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<\n  ExtractZodArrayTypes<T> extends Record<string, any>\n    ? ExtractZodArrayTypes<T>\n    : Record<string, any>,\n  ContextSchema,\n  TMiddlewares\n>;\n\n// Overload 3: With responseFormat as JsonSchemaFormat (JSON schema object)\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    Record<string, unknown>,\n    ContextSchema,\n    JsonSchemaFormat\n  > & {\n    responseFormat: JsonSchemaFormat;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<Record<string, unknown>, ContextSchema, TMiddlewares>;\n\n// Overload 4: With responseFormat as array of JsonSchemaFormat (JSON schema objects)\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    Record<string, unknown>,\n    ContextSchema,\n    JsonSchemaFormat[]\n  > & {\n    responseFormat: JsonSchemaFormat[];\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<Record<string, unknown>, ContextSchema, TMiddlewares>;\n\n// Overload 4.5: With responseFormat as union of JsonSchemaFormat | JsonSchemaFormat[]\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    Record<string, unknown>,\n    ContextSchema,\n    JsonSchemaFormat | JsonSchemaFormat[]\n  > & {\n    responseFormat: JsonSchemaFormat | JsonSchemaFormat[];\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<Record<string, unknown>, ContextSchema, TMiddlewares>;\n\n// Overload 5: With responseFormat as TypedToolStrategy (for union types from toolStrategy)\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, TypedToolStrategy<T>> & {\n    responseFormat: TypedToolStrategy<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 6: With responseFormat as single ToolStrategy instance\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, ToolStrategy<T>> & {\n    responseFormat: ToolStrategy<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 7: With responseFormat as ProviderStrategy\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, ProviderStrategy<T>> & {\n    responseFormat: ProviderStrategy<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 8: Without responseFormat property at all - with proper middleware state typing\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: Omit<\n    CreateAgentParams<ResponseFormatUndefined, ContextSchema, never>,\n    \"responseFormat\"\n  > & { middlewares?: TMiddlewares }\n): ReactAgent<ResponseFormatUndefined, ContextSchema, TMiddlewares>;\n\n// Overload 9: With responseFormat explicitly undefined\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: Omit<\n    CreateAgentParams<ResponseFormatUndefined, ContextSchema, never>,\n    \"responseFormat\"\n  > & {\n    responseFormat?: undefined;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<ResponseFormatUndefined, ContextSchema, TMiddlewares>;\n\n// Overload 10: For other ResponseFormat values (failsafe)\nexport function createAgent<\n  StructuredResponseFormat extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    StructuredResponseFormat,\n    ContextSchema,\n    ResponseFormat\n  > & {\n    responseFormat: ResponseFormat;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<StructuredResponseFormat, ContextSchema, TMiddlewares>;\n\n// Implementation\nexport function createAgent<\n  StructuredResponseFormat extends Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject,\n  TMiddlewares extends readonly AgentMiddleware<any, any, any>[] = []\n>(\n  params: CreateAgentParams<StructuredResponseFormat, ContextSchema, any>\n): ReactAgent<StructuredResponseFormat, ContextSchema, TMiddlewares> {\n  return new ReactAgent(params);\n}\n\nexport { createMiddleware } from \"./middleware.js\";\n"],"mappings":";;;;AA4RA,SAAgB,YAKdA,QACmE;AACnE,QAAO,IAAIC,8BAAW;AACvB"}

package/dist/agents/middlewareAgent/index.js

@@ -0,0 +1,11 @@
+import { ReactAgent } from "./ReactAgent.js";
+import { createMiddleware } from "./middleware.js";
+
+//#region src/agents/middlewareAgent/index.ts
+function createAgent(params) {
+	return new ReactAgent(params);
+}
+
+//#endregion
+export { createAgent };
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":["params: CreateAgentParams<StructuredResponseFormat, ContextSchema, any>"],"sources":["../../../src/agents/middlewareAgent/index.ts"],"sourcesContent":["import type {\n  InteropZodObject,\n  InteropZodType,\n} from \"@langchain/core/utils/types\";\n\nimport { type AnyAnnotationRoot } from \"../annotation.js\";\nimport type { CreateAgentParams, AgentMiddleware } from \"./types.js\";\nimport type { ExtractZodArrayTypes } from \"../types.js\";\nimport type {\n  ToolStrategy,\n  TypedToolStrategy,\n  ProviderStrategy,\n  ResponseFormat,\n  ResponseFormatUndefined,\n  JsonSchemaFormat,\n} from \"../responses.js\";\nimport { ReactAgent } from \"./ReactAgent.js\";\n\n/**\n * Creates a StateGraph agent that relies on a chat model utilizing tool calling.\n *\n * @example\n * ```ts\n * import { ChatOpenAI } from \"@langchain/openai\";\n * import { createAgent, tool } from \"langchain\";\n * import { z } from \"zod\";\n *\n * const model = new ChatOpenAI({\n *   model: \"gpt-4o\",\n * });\n *\n * const getWeather = tool((input) => {\n *   if ([\"sf\", \"san francisco\"].includes(input.location.toLowerCase())) {\n *     return \"It's 60 degrees and foggy.\";\n *   } else {\n *     return \"It's 90 degrees and sunny.\";\n *   }\n * }, {\n *   name: \"get_weather\",\n *   description: \"Call to get the current weather.\",\n *   schema: z.object({\n *     location: z.string().describe(\"Location to get the weather for.\"),\n *   })\n * })\n *\n * const agent = createAgent({ llm: model, tools: [getWeather] });\n *\n * const inputs = {\n *   messages: [{ role: \"user\", content: \"what is the weather in SF?\" }],\n * };\n *\n * const stream = await agent.stream(inputs, { streamMode: \"values\" });\n *\n * for await (const { messages } of stream) {\n *   console.log(messages);\n * }\n * // Returns the messages in the state at each step of execution\n * ```\n */\n// Overload 1: With responseFormat as single InteropZodType\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, InteropZodType<T>> & {\n    responseFormat: InteropZodType<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 2: With responseFormat as array of InteropZodTypes (infers union type)\nexport function createAgent<\n  T extends readonly InteropZodType<any>[],\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    ExtractZodArrayTypes<T> extends Record<string, any>\n      ? ExtractZodArrayTypes<T>\n      : Record<string, any>,\n    ContextSchema,\n    T\n  > & {\n    responseFormat: T;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<\n  ExtractZodArrayTypes<T> extends Record<string, any>\n    ? ExtractZodArrayTypes<T>\n    : Record<string, any>,\n  ContextSchema,\n  TMiddlewares\n>;\n\n// Overload 3: With responseFormat as JsonSchemaFormat (JSON schema object)\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    Record<string, unknown>,\n    ContextSchema,\n    JsonSchemaFormat\n  > & {\n    responseFormat: JsonSchemaFormat;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<Record<string, unknown>, ContextSchema, TMiddlewares>;\n\n// Overload 4: With responseFormat as array of JsonSchemaFormat (JSON schema objects)\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    Record<string, unknown>,\n    ContextSchema,\n    JsonSchemaFormat[]\n  > & {\n    responseFormat: JsonSchemaFormat[];\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<Record<string, unknown>, ContextSchema, TMiddlewares>;\n\n// Overload 4.5: With responseFormat as union of JsonSchemaFormat | JsonSchemaFormat[]\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    Record<string, unknown>,\n    ContextSchema,\n    JsonSchemaFormat | JsonSchemaFormat[]\n  > & {\n    responseFormat: JsonSchemaFormat | JsonSchemaFormat[];\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<Record<string, unknown>, ContextSchema, TMiddlewares>;\n\n// Overload 5: With responseFormat as TypedToolStrategy (for union types from toolStrategy)\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, TypedToolStrategy<T>> & {\n    responseFormat: TypedToolStrategy<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 6: With responseFormat as single ToolStrategy instance\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, ToolStrategy<T>> & {\n    responseFormat: ToolStrategy<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 7: With responseFormat as ProviderStrategy\nexport function createAgent<\n  T extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<T, ContextSchema, ProviderStrategy<T>> & {\n    responseFormat: ProviderStrategy<T>;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<T, ContextSchema, TMiddlewares>;\n\n// Overload 8: Without responseFormat property at all - with proper middleware state typing\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: Omit<\n    CreateAgentParams<ResponseFormatUndefined, ContextSchema, never>,\n    \"responseFormat\"\n  > & { middlewares?: TMiddlewares }\n): ReactAgent<ResponseFormatUndefined, ContextSchema, TMiddlewares>;\n\n// Overload 9: With responseFormat explicitly undefined\nexport function createAgent<\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: Omit<\n    CreateAgentParams<ResponseFormatUndefined, ContextSchema, never>,\n    \"responseFormat\"\n  > & {\n    responseFormat?: undefined;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<ResponseFormatUndefined, ContextSchema, TMiddlewares>;\n\n// Overload 10: For other ResponseFormat values (failsafe)\nexport function createAgent<\n  StructuredResponseFormat extends Record<string, any> = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddlewares extends readonly AgentMiddleware<\n    any,\n    any,\n    any\n  >[] = readonly AgentMiddleware<any, any, any>[]\n>(\n  params: CreateAgentParams<\n    StructuredResponseFormat,\n    ContextSchema,\n    ResponseFormat\n  > & {\n    responseFormat: ResponseFormat;\n    middlewares?: TMiddlewares;\n  }\n): ReactAgent<StructuredResponseFormat, ContextSchema, TMiddlewares>;\n\n// Implementation\nexport function createAgent<\n  StructuredResponseFormat extends Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject,\n  TMiddlewares extends readonly AgentMiddleware<any, any, any>[] = []\n>(\n  params: CreateAgentParams<StructuredResponseFormat, ContextSchema, any>\n): ReactAgent<StructuredResponseFormat, ContextSchema, TMiddlewares> {\n  return new ReactAgent(params);\n}\n\nexport { createMiddleware } from \"./middleware.js\";\n"],"mappings":";;;;AA4RA,SAAgB,YAKdA,QACmE;AACnE,QAAO,IAAI,WAAW;AACvB"}

package/dist/agents/middlewareAgent/middleware.cjs

@@ -0,0 +1,47 @@
+
+//#region src/agents/middlewareAgent/middleware.ts
+/**
+* Creates a middleware instance with automatic schema inference.
+*
+* @param config - Middleware configuration
+* @param config.name - The name of the middleware
+* @param config.stateSchema - The schema of the middleware state
+* @param config.contextSchema - The schema of the middleware context
+* @param config.prepareModelRequest - The function to prepare the model request
+* @param config.beforeModel - The function to run before the model call
+* @param config.afterModel - The function to run after the model call
+* @returns A middleware instance
+*
+* @example
+* ```ts
+* const authMiddleware = createMiddleware({
+*   name: "AuthMiddleware",
+*   stateSchema: z.object({
+*     isAuthenticated: z.boolean().default(false),
+*   }),
+*   contextSchema: z.object({
+*     userId: z.string(),
+*   }),
+*   beforeModel: async (state, runtime, controls) => {
+*     if (!state.isAuthenticated) {
+*       return controls.terminate(new Error("Not authenticated"));
+*     }
+*   },
+* });
+* ```
+*/
+function createMiddleware(config) {
+	const middleware = {
+		name: config.name,
+		stateSchema: config.stateSchema,
+		contextSchema: config.contextSchema
+	};
+	if (config.prepareModelRequest) middleware.prepareModelRequest = async (options, state, runtime) => Promise.resolve(config.prepareModelRequest(options, state, runtime));
+	if (config.beforeModel) middleware.beforeModel = async (state, runtime, controls) => Promise.resolve(config.beforeModel(state, runtime, controls));
+	if (config.afterModel) middleware.afterModel = async (state, runtime, controls) => Promise.resolve(config.afterModel(state, runtime, controls));
+	return middleware;
+}
+
+//#endregion
+exports.createMiddleware = createMiddleware;
+//# sourceMappingURL=middleware.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.cjs","names":["config: {\n  name: string;\n  stateSchema?: TSchema;\n  contextSchema?: TContextSchema;\n  prepareModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  beforeModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n  afterModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n}","middleware: AgentMiddleware<TSchema, TContextSchema, any>"],"sources":["../../../src/agents/middlewareAgent/middleware.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\nimport { z } from \"zod\";\nimport type {\n  AgentMiddleware,\n  Runtime,\n  Controls,\n  MiddlewareResult,\n  AgentBuiltInState,\n  ModelRequest,\n} from \"./types.js\";\n\n/**\n * Creates a middleware instance with automatic schema inference.\n *\n * @param config - Middleware configuration\n * @param config.name - The name of the middleware\n * @param config.stateSchema - The schema of the middleware state\n * @param config.contextSchema - The schema of the middleware context\n * @param config.prepareModelRequest - The function to prepare the model request\n * @param config.beforeModel - The function to run before the model call\n * @param config.afterModel - The function to run after the model call\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * const authMiddleware = createMiddleware({\n *   name: \"AuthMiddleware\",\n *   stateSchema: z.object({\n *     isAuthenticated: z.boolean().default(false),\n *   }),\n *   contextSchema: z.object({\n *     userId: z.string(),\n *   }),\n *   beforeModel: async (state, runtime, controls) => {\n *     if (!state.isAuthenticated) {\n *       return controls.terminate(new Error(\"Not authenticated\"));\n *     }\n *   },\n * });\n * ```\n */\nexport function createMiddleware<\n  TSchema extends z.ZodObject<any> | undefined = undefined,\n  TContextSchema extends z.ZodObject<any> | undefined = undefined\n>(config: {\n  name: string;\n  stateSchema?: TSchema;\n  contextSchema?: TContextSchema;\n  prepareModelRequest?: (\n    options: ModelRequest,\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >\n  ) => Promise<ModelRequest | void> | ModelRequest | void;\n  beforeModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n  afterModel?: (\n    state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n      AgentBuiltInState,\n    runtime: Runtime<\n      TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}\n    >,\n    controls: Controls<\n      (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) &\n        AgentBuiltInState\n    >\n  ) =>\n    | Promise<\n        MiddlewareResult<\n          Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n        >\n      >\n    | MiddlewareResult<\n        Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>\n      >;\n}): AgentMiddleware<TSchema, TContextSchema, any> {\n  const middleware: AgentMiddleware<TSchema, TContextSchema, any> = {\n    name: config.name,\n    stateSchema: config.stateSchema,\n    contextSchema: config.contextSchema,\n  };\n\n  if (config.prepareModelRequest) {\n    middleware.prepareModelRequest = async (options, state, runtime) =>\n      Promise.resolve(config.prepareModelRequest!(options, state, runtime));\n  }\n\n  if (config.beforeModel) {\n    middleware.beforeModel = async (state, runtime, controls) =>\n      Promise.resolve(config.beforeModel!(state, runtime, controls));\n  }\n\n  if (config.afterModel) {\n    middleware.afterModel = async (state, runtime, controls) =>\n      Promise.resolve(config.afterModel!(state, runtime, controls));\n  }\n\n  return middleware;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAyCA,SAAgB,iBAGdA,QAkDgD;CAChD,MAAMC,aAA4D;EAChE,MAAM,OAAO;EACb,aAAa,OAAO;EACpB,eAAe,OAAO;CACvB;AAED,KAAI,OAAO,qBACT,WAAW,sBAAsB,OAAO,SAAS,OAAO,YACtD,QAAQ,QAAQ,OAAO,oBAAqB,SAAS,OAAO,QAAQ,CAAC;AAGzE,KAAI,OAAO,aACT,WAAW,cAAc,OAAO,OAAO,SAAS,aAC9C,QAAQ,QAAQ,OAAO,YAAa,OAAO,SAAS,SAAS,CAAC;AAGlE,KAAI,OAAO,YACT,WAAW,aAAa,OAAO,OAAO,SAAS,aAC7C,QAAQ,QAAQ,OAAO,WAAY,OAAO,SAAS,SAAS,CAAC;AAGjE,QAAO;AACR"}

package/dist/agents/middlewareAgent/middleware.d.cts

@@ -0,0 +1,46 @@
+import { AgentBuiltInState, AgentMiddleware, Controls, MiddlewareResult, ModelRequest, Runtime } from "./types.cjs";
+import { z } from "zod";
+
+//#region src/agents/middlewareAgent/middleware.d.ts
+
+/**
+ * Creates a middleware instance with automatic schema inference.
+ *
+ * @param config - Middleware configuration
+ * @param config.name - The name of the middleware
+ * @param config.stateSchema - The schema of the middleware state
+ * @param config.contextSchema - The schema of the middleware context
+ * @param config.prepareModelRequest - The function to prepare the model request
+ * @param config.beforeModel - The function to run before the model call
+ * @param config.afterModel - The function to run after the model call
+ * @returns A middleware instance
+ *
+ * @example
+ * ```ts
+ * const authMiddleware = createMiddleware({
+ *   name: "AuthMiddleware",
+ *   stateSchema: z.object({
+ *     isAuthenticated: z.boolean().default(false),
+ *   }),
+ *   contextSchema: z.object({
+ *     userId: z.string(),
+ *   }),
+ *   beforeModel: async (state, runtime, controls) => {
+ *     if (!state.isAuthenticated) {
+ *       return controls.terminate(new Error("Not authenticated"));
+ *     }
+ *   },
+ * });
+ * ```
+ */
+declare function createMiddleware<TSchema extends z.ZodObject<any> | undefined = undefined, TContextSchema extends z.ZodObject<any> | undefined = undefined>(config: {
+  name: string;
+  stateSchema?: TSchema;
+  contextSchema?: TContextSchema;
+  prepareModelRequest?: (options: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>) => Promise<ModelRequest | void> | ModelRequest | void;
+  beforeModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;
+  afterModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;
+}): AgentMiddleware<TSchema, TContextSchema, any>;
+//#endregion
+export { createMiddleware };
+//# sourceMappingURL=middleware.d.cts.map

package/dist/agents/middlewareAgent/middleware.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.cts","names":["z","AgentMiddleware","Runtime","Controls","MiddlewareResult","AgentBuiltInState","ModelRequest","createMiddleware","ZodObject","TSchema","TContextSchema","infer","Promise","Partial"],"sources":["../../../src/agents/middlewareAgent/middleware.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/ban-types */\nimport { z } from \"zod\";\nimport type { AgentMiddleware, Runtime, Controls, MiddlewareResult, AgentBuiltInState, ModelRequest } from \"./types.js\";\n/**\n * Creates a middleware instance with automatic schema inference.\n *\n * @param config - Middleware configuration\n * @param config.name - The name of the middleware\n * @param config.stateSchema - The schema of the middleware state\n * @param config.contextSchema - The schema of the middleware context\n * @param config.prepareModelRequest - The function to prepare the model request\n * @param config.beforeModel - The function to run before the model call\n * @param config.afterModel - The function to run after the model call\n * @returns A middleware instance\n *\n * @example\n * ```ts\n * const authMiddleware = createMiddleware({\n *   name: \"AuthMiddleware\",\n *   stateSchema: z.object({\n *     isAuthenticated: z.boolean().default(false),\n *   }),\n *   contextSchema: z.object({\n *     userId: z.string(),\n *   }),\n *   beforeModel: async (state, runtime, controls) => {\n *     if (!state.isAuthenticated) {\n *       return controls.terminate(new Error(\"Not authenticated\"));\n *     }\n *   },\n * });\n * ```\n */\nexport declare function createMiddleware<TSchema extends z.ZodObject<any> | undefined = undefined, TContextSchema extends z.ZodObject<any> | undefined = undefined>(config: {\n    name: string;\n    stateSchema?: TSchema;\n    contextSchema?: TContextSchema;\n    prepareModelRequest?: (options: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>) => Promise<ModelRequest | void> | ModelRequest | void;\n    beforeModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;\n    afterModel?: (state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TContextSchema extends z.ZodObject<any> ? z.infer<TContextSchema> : {}>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>) => Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>> | MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>;\n}): AgentMiddleware<TSchema, TContextSchema, any>;\n"],"mappings":";;;;;;AAiCA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAKmHE,iBAL3FK,gBAK2FL,CAAAA,gBAL1DF,CAAAA,CAAEQ,SAKwDN,CAAAA,GAAAA,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,EAAAA,uBALOF,CAAAA,CAAEQ,SAKTN,CAAAA,GAAAA,CAAAA,GAAAA,SAAAA,GAAAA,SAAAA,CAAAA,CAAAA,MAAAA,EAAAA;EAAO,IAA8FO,EAAAA,MAAAA;EAAO,WAAWD,CAAAA,EAHxNC,OAGwND;EAAS,aAAgBC,CAAAA,EAF/OC,cAE+OD;EAAO,mBAAbE,CAAAA,EAAAA,CAAAA,OAAAA,EADzNL,YACyNK,EAAAA,KAAAA,EAAAA,CADnMF,OACmME,SADnLX,CAAAA,CAAEQ,SACiLG,CAAAA,GAAAA,CAAAA,GADhKX,CAAAA,CAAEW,KAC8JA,CADxJF,OACwJE,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GADvIN,iBACuIM,EAAAA,OAAAA,EAD3GT,OAC2GS,CADnGD,cACmGC,SAD5EX,CAAAA,CAAEQ,SAC0EG,CAAAA,GAAAA,CAAAA,GADzDX,CAAAA,CAAEW,KACuDA,CADjDD,cACiDC,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,GADvBC,OACuBD,CADfL,YACeK,GAAAA,IAAAA,CAAAA,GADQL,YACRK,GAAAA,IAAAA;EAAK,WAAkBN,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CAAzPI,OAAyPJ,SAAzOL,CAAAA,CAAEQ,SAAuOH,CAAAA,GAAAA,CAAAA,GAAtNL,CAAAA,CAAEW,KAAoNN,CAA9MI,OAA8MJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAA7LA,iBAA6LA,EAAAA,OAAAA,EAAjKH,OAAiKG,CAAzJK,cAAyJL,SAAlIL,CAAAA,CAAEQ,SAAgIH,CAAAA,GAAAA,CAAAA,GAA/GL,CAAAA,CAAEW,KAA6GN,CAAvGK,cAAuGL,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,QAAAA,EAAtEF,QAAsEE,CAAAA,CAA5DI,OAA4DJ,SAA5CL,CAAAA,CAAEQ,SAA0CH,CAAAA,GAAAA,CAAAA,GAAzBL,CAAAA,CAAEW,KAAuBN,CAAjBI,OAAiBJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GAAAA,iBAAAA,CAAAA,EAAAA,GAAuBO,OAAvBP,CAA+BD,gBAA/BC,CAAgDQ,OAAhDR,CAAwDI,OAAxDJ,SAAwEL,CAAAA,CAAEQ,SAA1EH,CAAAA,GAAAA,CAAAA,GAA2FL,CAAAA,CAAEW,KAA7FN,CAAmGI,OAAnGJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAAsHD,gBAAtHC,CAAuIQ,OAAvIR,CAA+II,OAA/IJ,SAA+JL,CAAAA,CAAEQ,SAAjKH,CAAAA,GAAAA,CAAAA,GAAkLL,CAAAA,CAAEW,KAApLN,CAA0LI,OAA1LJ,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;EAAiB,UAAvFF,CAAAA,EAAAA,CAAAA,KAAAA,EAAAA,CACpLM,OADoLN,SACpKH,CAAAA,CAAEQ,SADkKL,CAAAA,GAAAA,CAAAA,GACjJH,CAAAA,CAAEW,KAD+IR,CACzIM,OADyIN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GACxHE,iBADwHF,EAAAA,OAAAA,EAC5FD,OAD4FC,CACpFO,cADoFP,SAC7DH,CAAAA,CAAEQ,SAD2DL,CAAAA,GAAAA,CAAAA,GAC1CH,CAAAA,CAAEW,KADwCR,CAClCO,cADkCP,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,EAAAA,QAAAA,EACDA,QADCA,CAAAA,CACSM,OADTN,SACyBH,CAAAA,CAAEQ,SAD3BL,CAAAA,GAAAA,CAAAA,GAC4CH,CAAAA,CAAEW,KAD9CR,CACoDM,OADpDN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,GACqEE,iBADrEF,CAAAA,EAAAA,GAC4FS,OAD5FT,CACoGC,gBADpGD,CACqHU,OADrHV,CAC6HM,OAD7HN,SAC6IH,CAAAA,CAAEQ,SAD/IL,CAAAA,GAAAA,CAAAA,GACgKH,CAAAA,CAAEW,KADlKR,CACwKM,OADxKN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,GAC2LC,gBAD3LD,CAC4MU,OAD5MV,CACoNM,OADpNN,SACoOH,CAAAA,CAAEQ,SADtOL,CAAAA,GAAAA,CAAAA,GACuPH,CAAAA,CAAEW,KADzPR,CAC+PM,OAD/PN,CAAAA,GAAAA,CAAAA,CAAAA,CAAAA,CAAAA;CAAQ,CAAA,EAElNF,eAFwUQ,CAExTA,OAFwTA,EAE/SC,cAF+SD,EAAAA,GAAAA,CAAAA"}