langchain 1.0.0-alpha.7 → 1.0.0-alpha.8

package/dist/agents/ReactAgent.cjs

@@ -52,7 +52,7 @@ var ReactAgent = class {
 		/**
 		* add single tool node for all tools
 		*/
-		if (toolClasses.length > 0) {
+		if (toolClasses.filter(require_utils.isClientTool).length > 0) {
 			const toolNode = new require_ToolNode.ToolNode(toolClasses.filter(require_utils.isClientTool), { signal: this.options.signal });
 			allNodeWorkflows.addNode("tools", toolNode);
 		}
@@ -82,7 +82,7 @@ var ReactAgent = class {
 		/**
 		* add edges for tools node
 		*/
-		if (toolClasses.length > 0) if (shouldReturnDirect.size > 0) allNodeWorkflows.addConditionalEdges("tools", this.#createToolsRouter(shouldReturnDirect), [this.#getEntryPoint(), __langchain_langgraph.END]);
+		if (toolClasses.filter(require_utils.isClientTool).length > 0) if (shouldReturnDirect.size > 0) allNodeWorkflows.addConditionalEdges("tools", this.#createToolsRouter(shouldReturnDirect), [this.#getEntryPoint(), __langchain_langgraph.END]);
 		else allNodeWorkflows.addEdge("tools", this.#getEntryPoint());
 		/**
 		* compile the graph

package/dist/agents/ReactAgent.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ReactAgent.cjs","names":["options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >","#toolBehaviorVersion","validateLLMHasNoBoundTools","isClientTool","enhanceStateSchemaWithMessageReducer","createAgentAnnotationConditional","StateGraph","AgentNode","#inputSchema","ToolNode","START","#getEntryPoint","#getPostModelHookPaths","#createPostModelHookRouter","#getModelPaths","#createModelRouter","#createToolsRouter","END","#graph","toolClasses: (ClientTool | ServerTool)[]","paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[]","state: InternalAgentState<StructuredResponseFormat>","ToolMessage","AIMessage","Send","paths: (\"tools\" | typeof END)[]","shouldReturnDirect: Set<string>","params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }"],"sources":["../../src/agents/ReactAgent.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  CompiledStateGraph,\n  MessagesAnnotation,\n} from \"@langchain/langgraph\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\n\nimport {\n  createAgentAnnotationConditional,\n  ReactAgentAnnotation,\n} from \"./annotation.js\";\nimport { isClientTool, validateLLMHasNoBoundTools } from \"./utils.js\";\nimport { AgentNode } from \"./nodes/AgentNode.js\";\nimport { ToolNode } from \"./nodes/ToolNode.js\";\nimport type {\n  CreateAgentParams,\n  ClientTool,\n  ServerTool,\n  InternalAgentState,\n  WithStateGraphNodes,\n} from \"./types.js\";\nimport {\n  enhanceStateSchemaWithMessageReducer,\n  type AnyAnnotationRoot,\n  type ToAnnotationRoot,\n  type ResponseFormatUndefined,\n} from \"./annotation.js\";\n\ntype AgentGraph<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> = CompiledStateGraph<\n  ToAnnotationRoot<StateSchema>[\"State\"],\n  ToAnnotationRoot<StateSchema>[\"Update\"],\n  any,\n  typeof MessagesAnnotation.spec & ToAnnotationRoot<StateSchema>[\"spec\"],\n  ReactAgentAnnotation<StructuredResponseFormat>[\"spec\"] &\n    ToAnnotationRoot<StateSchema>[\"spec\"],\n  ToAnnotationRoot<ContextSchema>[\"spec\"],\n  unknown\n>;\n\nexport class ReactAgent<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> {\n  #graph: AgentGraph<StateSchema, StructuredResponseFormat, ContextSchema>;\n\n  #inputSchema?: AnnotationRoot<ToAnnotationRoot<StateSchema>[\"spec\"]>;\n\n  #toolBehaviorVersion: \"v1\" | \"v2\" = \"v2\";\n\n  constructor(\n    public options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >\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 = Array.isArray(options.tools)\n      ? options.tools\n      : options.tools.tools;\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    const schema = this.options.stateSchema\n      ? enhanceStateSchemaWithMessageReducer(this.options.stateSchema)\n      : createAgentAnnotationConditional<StructuredResponseFormat>(\n          this.options.responseFormat !== undefined\n        );\n\n    const workflow = new StateGraph(schema, this.options.contextSchema);\n\n    const allNodeWorkflows = workflow as WithStateGraphNodes<\n      \"pre_model_hook\" | \"post_model_hook\" | \"tools\" | \"agent\",\n      typeof workflow\n    >;\n\n    /**\n     * Add Nodes\n     */\n    allNodeWorkflows.addNode(\n      \"agent\",\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        toolClasses,\n        shouldReturnDirect,\n        signal: this.options.signal,\n      }),\n      {\n        input: this.#inputSchema,\n      }\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     * setup preModelHook\n     */\n    if (options.preModelHook) {\n      allNodeWorkflows.addNode(\"pre_model_hook\", options.preModelHook);\n    }\n\n    /**\n     * setup postModelHook\n     */\n    if (options.postModelHook) {\n      allNodeWorkflows.addNode(\"post_model_hook\", options.postModelHook);\n    }\n\n    /**\n     * Add Edges\n     */\n    allNodeWorkflows.addEdge(START, this.#getEntryPoint());\n\n    if (this.options.preModelHook) {\n      allNodeWorkflows.addEdge(\"pre_model_hook\", \"agent\");\n    }\n\n    if (this.options.postModelHook) {\n      allNodeWorkflows.addEdge(\"agent\", \"post_model_hook\");\n      const postHookPaths = this.#getPostModelHookPaths(\n        toolClasses.filter(isClientTool)\n      );\n      if (postHookPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"post_model_hook\", postHookPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"post_model_hook\",\n          this.#createPostModelHookRouter(),\n          postHookPaths\n        );\n      }\n    } else {\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      if (modelPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"agent\", modelPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"agent\",\n          this.#createModelRouter(),\n          modelPaths\n        );\n      }\n    }\n\n    /**\n     * add edges for tools node\n     */\n    if (toolClasses.length > 0) {\n      if (shouldReturnDirect.size > 0) {\n        allNodeWorkflows.addConditionalEdges(\n          \"tools\",\n          this.#createToolsRouter(shouldReturnDirect),\n          [this.#getEntryPoint(), END]\n        );\n      } else {\n        allNodeWorkflows.addEdge(\"tools\", this.#getEntryPoint());\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    });\n  }\n\n  /**\n   * Get the compiled graph.\n   */\n  get graph(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat,\n    ContextSchema\n  > {\n    return this.#graph;\n  }\n\n  #getEntryPoint() {\n    const entryPoint = this.options.preModelHook ? \"pre_model_hook\" : \"agent\";\n    return entryPoint;\n  }\n\n  /**\n   * Get possible edge destinations from post_model_hook node.\n   */\n  #getPostModelHookPaths(toolClasses: (ClientTool | ServerTool)[]) {\n    const paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(this.#getEntryPoint(), \"tools\");\n    }\n    paths.push(END);\n    return paths;\n  }\n\n  #createPostModelHookRouter() {\n    return (state: InternalAgentState<StructuredResponseFormat>) => {\n      const messages = state.messages;\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\n      if (pendingToolCalls && pendingToolCalls.length > 0) {\n        /**\n         * The tool node processes a single message.\n         * All tool calls in the message are executed in parallel within the tool node.\n         * @deprecated likely to be removed in the next version of the agent\n         */\n        if (this.#toolBehaviorVersion === \"v1\") {\n          return \"tools\";\n        }\n\n        /**\n         * The tool node processes a single tool call. Tool calls are distributed across\n         * multiple instances of the tool node using the Send API.\n         */\n        return pendingToolCalls.map(\n          (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n        );\n      }\n\n      if (ToolMessage.isInstance(messages.at(-1))) {\n        return this.#getEntryPoint();\n      }\n\n      return END;\n    };\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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        if (this.options.postModelHook) {\n          return \"post_model_hook\";\n        }\n\n        return END;\n      }\n\n      if (this.options.postModelHook) {\n        return \"post_model_hook\";\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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 ? this.#getEntryPoint() : END;\n      }\n\n      // For non-returnDirect tools, always route back to agent\n      return this.#getEntryPoint();\n    };\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get invoke(): AgentGraph<StateSchema, StructuredResponseFormat>[\"invoke\"] {\n    return this.#graph.invoke.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get stream(): AgentGraph<StateSchema, StructuredResponseFormat>[\"stream\"] {\n    return this.#graph.stream.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get streamEvents(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat\n  >[\"streamEvents\"] {\n    return this.#graph.streamEvents.bind(this.#graph);\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":";;;;;;;;;AAmDA,IAAa,aAAb,MAME;CACA;CAEA;CAEA,uBAAoC;CAEpC,YACSA,SAKP;EALO;EAMP,KAAKC,uBAAuB,QAAQ,WAAW,KAAKA;;;;AAKpD,MAAI,QAAQ,OAAO,OAAO,QAAQ,QAAQ,YACxCC,yCAA2B,QAAQ,IAAI;;;;AAMzC,MAAI,QAAQ,OAAO,QAAQ,MACzB,OAAM,IAAI,MAAM;;;;AAMlB,MAAI,CAAC,QAAQ,OAAO,CAAC,QAAQ,MAC3B,OAAM,IAAI,MACR;EAIJ,MAAM,cAAc,MAAM,QAAQ,QAAQ,MAAM,GAC5C,QAAQ,QACR,QAAQ,MAAM;;;;;EAMlB,MAAM,qBAAqB,IAAI,IAC7B,YACG,OAAOC,2BAAa,CACpB,OAAO,CAAC,SAAS,kBAAkB,QAAQ,KAAK,aAAa,CAC7D,IAAI,CAAC,SAAS,KAAK,KAAK;EAG7B,MAAM,SAAS,KAAK,QAAQ,cACxBC,wDAAqC,KAAK,QAAQ,YAAY,GAC9DC,oDACE,KAAK,QAAQ,mBAAmB,OACjC;EAEL,MAAM,WAAW,IAAIC,iCAAW,QAAQ,KAAK,QAAQ;EAErD,MAAM,mBAAmB;;;;EAQzB,iBAAiB,QACf,SACA,IAAIC,4BAAU;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;GACA;GACA,QAAQ,KAAK,QAAQ;EACtB,IACD,EACE,OAAO,KAAKC,aACb,EACF;;;;AAKD,MAAI,YAAY,SAAS,GAAG;GAC1B,MAAM,WAAW,IAAIC,0BAAS,YAAY,OAAON,2BAAa,EAAE,EAC9D,QAAQ,KAAK,QAAQ,OACtB;GACD,iBAAiB,QAAQ,SAAS,SAAS;EAC5C;;;;AAKD,MAAI,QAAQ,cACV,iBAAiB,QAAQ,kBAAkB,QAAQ,aAAa;;;;AAMlE,MAAI,QAAQ,eACV,iBAAiB,QAAQ,mBAAmB,QAAQ,cAAc;;;;EAMpE,iBAAiB,QAAQO,6BAAO,KAAKC,gBAAgB,CAAC;AAEtD,MAAI,KAAK,QAAQ,cACf,iBAAiB,QAAQ,kBAAkB,QAAQ;AAGrD,MAAI,KAAK,QAAQ,eAAe;GAC9B,iBAAiB,QAAQ,SAAS,kBAAkB;GACpD,MAAM,gBAAgB,KAAKC,uBACzB,YAAY,OAAOT,2BAAa,CACjC;AACD,OAAI,cAAc,WAAW,GAC3B,iBAAiB,QAAQ,mBAAmB,cAAc,GAAG;QAE7D,iBAAiB,oBACf,mBACA,KAAKU,4BAA4B,EACjC,cACD;EAEJ,OAAM;GACL,MAAM,aAAa,KAAKC,eAAe,YAAY,OAAOX,2BAAa,CAAC;AACxE,OAAI,WAAW,WAAW,GACxB,iBAAiB,QAAQ,SAAS,WAAW,GAAG;QAEhD,iBAAiB,oBACf,SACA,KAAKY,oBAAoB,EACzB,WACD;EAEJ;;;;AAKD,MAAI,YAAY,SAAS,EACvB,KAAI,mBAAmB,OAAO,GAC5B,iBAAiB,oBACf,SACA,KAAKC,mBAAmB,mBAAmB,EAC3C,CAAC,KAAKL,gBAAgB,EAAEM,yBAAI,EAC7B;OAED,iBAAiB,QAAQ,SAAS,KAAKN,gBAAgB,CAAC;;;;EAO5D,KAAKO,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;CAED,iBAAiB;EACf,MAAM,aAAa,KAAK,QAAQ,eAAe,mBAAmB;AAClE,SAAO;CACR;;;;CAKD,uBAAuBC,aAA0C;EAC/D,MAAMC,QAA+D,CAAE;AACvE,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,KAAKT,gBAAgB,EAAE,QAAQ;EAE5C,MAAM,KAAKM,0BAAI;AACf,SAAO;CACR;CAED,6BAA6B;AAC3B,SAAO,CAACI,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,eAAe,SAAS,OAAOC,sCAAY,WAAW;GAC5D,MAAM,gBAAgB,SAAS,OAAOC,oCAAU,WAAW,CAAC,GAAG,GAAG;GAClE,MAAM,mBAAmB,eAAe,YAAY,OAClD,CAAC,SAAS,CAAC,aAAa,KAAK,CAAC,MAAM,EAAE,iBAAiB,KAAK,GAAG,CAChE;AAED,OAAI,oBAAoB,iBAAiB,SAAS,GAAG;;;;;;AAMnD,QAAI,KAAKtB,yBAAyB,KAChC,QAAO;;;;;AAOT,WAAO,iBAAiB,IACtB,CAAC,aAAa,IAAIuB,2BAAK,SAAS;KAAE,GAAG;KAAO,cAAc;IAAU,GACrE;GACF;AAED,OAAIF,sCAAY,WAAW,SAAS,GAAG,GAAG,CAAC,CACzC,QAAO,KAAKX,gBAAgB;AAG9B,UAAOM;EACR;CACF;;;;;;CAOD,eACEE,aAC0B;EAC1B,MAAMM,QAAkC,CAAE;AAC1C,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;EAGrB,MAAM,KAAKR,0BAAI;AAEf,SAAO;CACR;;;;CAKD,qBAAqB;;;;AAInB,SAAO,CAACI,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,CAACE,oCAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,GAClC;AACA,QAAI,KAAK,QAAQ,cACf,QAAO;AAGT,WAAON;GACR;AAED,OAAI,KAAK,QAAQ,cACf,QAAO;;;;AAMT,OAAI,KAAKhB,yBAAyB,KAChC,QAAO;;;;AAMT,UAAO,YAAY,WAAW,IAC5B,CAAC,aAAa,IAAIuB,2BAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;CAKD,mBAAmBE,oBAAiC;AAClD,SAAO,CAACL,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,SAAS,SAAS;AAG/C,OACEC,sCAAY,WAAW,YAAY,IACnC,YAAY,QACZ,mBAAmB,IAAI,YAAY,KAAK,CAIxC,QAAO,KAAK,QAAQ,iBAAiB,KAAKX,gBAAgB,GAAGM;AAI/D,UAAO,KAAKN,gBAAgB;EAC7B;CACF;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKO,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKA,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,eAGc;AAChB,SAAO,KAAKA,OAAO,aAAa,KAAK,KAAKA,OAAO;CAClD;;;;;;;;;;;CAYD,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"}
+{"version":3,"file":"ReactAgent.cjs","names":["options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >","#toolBehaviorVersion","validateLLMHasNoBoundTools","isClientTool","enhanceStateSchemaWithMessageReducer","createAgentAnnotationConditional","StateGraph","AgentNode","#inputSchema","ToolNode","START","#getEntryPoint","#getPostModelHookPaths","#createPostModelHookRouter","#getModelPaths","#createModelRouter","#createToolsRouter","END","#graph","toolClasses: (ClientTool | ServerTool)[]","paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[]","state: InternalAgentState<StructuredResponseFormat>","ToolMessage","AIMessage","Send","paths: (\"tools\" | typeof END)[]","shouldReturnDirect: Set<string>","params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }"],"sources":["../../src/agents/ReactAgent.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  CompiledStateGraph,\n  MessagesAnnotation,\n} from \"@langchain/langgraph\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\n\nimport {\n  createAgentAnnotationConditional,\n  ReactAgentAnnotation,\n} from \"./annotation.js\";\nimport { isClientTool, validateLLMHasNoBoundTools } from \"./utils.js\";\nimport { AgentNode } from \"./nodes/AgentNode.js\";\nimport { ToolNode } from \"./nodes/ToolNode.js\";\nimport type {\n  CreateAgentParams,\n  ClientTool,\n  ServerTool,\n  InternalAgentState,\n  WithStateGraphNodes,\n} from \"./types.js\";\nimport {\n  enhanceStateSchemaWithMessageReducer,\n  type AnyAnnotationRoot,\n  type ToAnnotationRoot,\n  type ResponseFormatUndefined,\n} from \"./annotation.js\";\n\ntype AgentGraph<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> = CompiledStateGraph<\n  ToAnnotationRoot<StateSchema>[\"State\"],\n  ToAnnotationRoot<StateSchema>[\"Update\"],\n  any,\n  typeof MessagesAnnotation.spec & ToAnnotationRoot<StateSchema>[\"spec\"],\n  ReactAgentAnnotation<StructuredResponseFormat>[\"spec\"] &\n    ToAnnotationRoot<StateSchema>[\"spec\"],\n  ToAnnotationRoot<ContextSchema>[\"spec\"],\n  unknown\n>;\n\nexport class ReactAgent<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> {\n  #graph: AgentGraph<StateSchema, StructuredResponseFormat, ContextSchema>;\n\n  #inputSchema?: AnnotationRoot<ToAnnotationRoot<StateSchema>[\"spec\"]>;\n\n  #toolBehaviorVersion: \"v1\" | \"v2\" = \"v2\";\n\n  constructor(\n    public options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >\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 = Array.isArray(options.tools)\n      ? options.tools\n      : options.tools.tools;\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    const schema = this.options.stateSchema\n      ? enhanceStateSchemaWithMessageReducer(this.options.stateSchema)\n      : createAgentAnnotationConditional<StructuredResponseFormat>(\n          this.options.responseFormat !== undefined\n        );\n\n    const workflow = new StateGraph(schema, this.options.contextSchema);\n\n    const allNodeWorkflows = workflow as WithStateGraphNodes<\n      \"pre_model_hook\" | \"post_model_hook\" | \"tools\" | \"agent\",\n      typeof workflow\n    >;\n\n    /**\n     * Add Nodes\n     */\n    allNodeWorkflows.addNode(\n      \"agent\",\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        toolClasses,\n        shouldReturnDirect,\n        signal: this.options.signal,\n      }),\n      {\n        input: this.#inputSchema,\n      }\n    );\n\n    /**\n     * add single tool node for all tools\n     */\n    if (toolClasses.filter(isClientTool).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     * setup preModelHook\n     */\n    if (options.preModelHook) {\n      allNodeWorkflows.addNode(\"pre_model_hook\", options.preModelHook);\n    }\n\n    /**\n     * setup postModelHook\n     */\n    if (options.postModelHook) {\n      allNodeWorkflows.addNode(\"post_model_hook\", options.postModelHook);\n    }\n\n    /**\n     * Add Edges\n     */\n    allNodeWorkflows.addEdge(START, this.#getEntryPoint());\n\n    if (this.options.preModelHook) {\n      allNodeWorkflows.addEdge(\"pre_model_hook\", \"agent\");\n    }\n\n    if (this.options.postModelHook) {\n      allNodeWorkflows.addEdge(\"agent\", \"post_model_hook\");\n      const postHookPaths = this.#getPostModelHookPaths(\n        toolClasses.filter(isClientTool)\n      );\n      if (postHookPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"post_model_hook\", postHookPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"post_model_hook\",\n          this.#createPostModelHookRouter(),\n          postHookPaths\n        );\n      }\n    } else {\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      if (modelPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"agent\", modelPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"agent\",\n          this.#createModelRouter(),\n          modelPaths\n        );\n      }\n    }\n\n    /**\n     * add edges for tools node\n     */\n    if (toolClasses.filter(isClientTool).length > 0) {\n      if (shouldReturnDirect.size > 0) {\n        allNodeWorkflows.addConditionalEdges(\n          \"tools\",\n          this.#createToolsRouter(shouldReturnDirect),\n          [this.#getEntryPoint(), END]\n        );\n      } else {\n        allNodeWorkflows.addEdge(\"tools\", this.#getEntryPoint());\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    });\n  }\n\n  /**\n   * Get the compiled graph.\n   */\n  get graph(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat,\n    ContextSchema\n  > {\n    return this.#graph;\n  }\n\n  #getEntryPoint() {\n    const entryPoint = this.options.preModelHook ? \"pre_model_hook\" : \"agent\";\n    return entryPoint;\n  }\n\n  /**\n   * Get possible edge destinations from post_model_hook node.\n   */\n  #getPostModelHookPaths(toolClasses: (ClientTool | ServerTool)[]) {\n    const paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(this.#getEntryPoint(), \"tools\");\n    }\n    paths.push(END);\n    return paths;\n  }\n\n  #createPostModelHookRouter() {\n    return (state: InternalAgentState<StructuredResponseFormat>) => {\n      const messages = state.messages;\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\n      if (pendingToolCalls && pendingToolCalls.length > 0) {\n        /**\n         * The tool node processes a single message.\n         * All tool calls in the message are executed in parallel within the tool node.\n         * @deprecated likely to be removed in the next version of the agent\n         */\n        if (this.#toolBehaviorVersion === \"v1\") {\n          return \"tools\";\n        }\n\n        /**\n         * The tool node processes a single tool call. Tool calls are distributed across\n         * multiple instances of the tool node using the Send API.\n         */\n        return pendingToolCalls.map(\n          (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n        );\n      }\n\n      if (ToolMessage.isInstance(messages.at(-1))) {\n        return this.#getEntryPoint();\n      }\n\n      return END;\n    };\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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        if (this.options.postModelHook) {\n          return \"post_model_hook\";\n        }\n\n        return END;\n      }\n\n      if (this.options.postModelHook) {\n        return \"post_model_hook\";\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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 ? this.#getEntryPoint() : END;\n      }\n\n      // For non-returnDirect tools, always route back to agent\n      return this.#getEntryPoint();\n    };\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get invoke(): AgentGraph<StateSchema, StructuredResponseFormat>[\"invoke\"] {\n    return this.#graph.invoke.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get stream(): AgentGraph<StateSchema, StructuredResponseFormat>[\"stream\"] {\n    return this.#graph.stream.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get streamEvents(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat\n  >[\"streamEvents\"] {\n    return this.#graph.streamEvents.bind(this.#graph);\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":";;;;;;;;;AAmDA,IAAa,aAAb,MAME;CACA;CAEA;CAEA,uBAAoC;CAEpC,YACSA,SAKP;EALO;EAMP,KAAKC,uBAAuB,QAAQ,WAAW,KAAKA;;;;AAKpD,MAAI,QAAQ,OAAO,OAAO,QAAQ,QAAQ,YACxCC,yCAA2B,QAAQ,IAAI;;;;AAMzC,MAAI,QAAQ,OAAO,QAAQ,MACzB,OAAM,IAAI,MAAM;;;;AAMlB,MAAI,CAAC,QAAQ,OAAO,CAAC,QAAQ,MAC3B,OAAM,IAAI,MACR;EAIJ,MAAM,cAAc,MAAM,QAAQ,QAAQ,MAAM,GAC5C,QAAQ,QACR,QAAQ,MAAM;;;;;EAMlB,MAAM,qBAAqB,IAAI,IAC7B,YACG,OAAOC,2BAAa,CACpB,OAAO,CAAC,SAAS,kBAAkB,QAAQ,KAAK,aAAa,CAC7D,IAAI,CAAC,SAAS,KAAK,KAAK;EAG7B,MAAM,SAAS,KAAK,QAAQ,cACxBC,wDAAqC,KAAK,QAAQ,YAAY,GAC9DC,oDACE,KAAK,QAAQ,mBAAmB,OACjC;EAEL,MAAM,WAAW,IAAIC,iCAAW,QAAQ,KAAK,QAAQ;EAErD,MAAM,mBAAmB;;;;EAQzB,iBAAiB,QACf,SACA,IAAIC,4BAAU;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;GACA;GACA,QAAQ,KAAK,QAAQ;EACtB,IACD,EACE,OAAO,KAAKC,aACb,EACF;;;;AAKD,MAAI,YAAY,OAAOL,2BAAa,CAAC,SAAS,GAAG;GAC/C,MAAM,WAAW,IAAIM,0BAAS,YAAY,OAAON,2BAAa,EAAE,EAC9D,QAAQ,KAAK,QAAQ,OACtB;GACD,iBAAiB,QAAQ,SAAS,SAAS;EAC5C;;;;AAKD,MAAI,QAAQ,cACV,iBAAiB,QAAQ,kBAAkB,QAAQ,aAAa;;;;AAMlE,MAAI,QAAQ,eACV,iBAAiB,QAAQ,mBAAmB,QAAQ,cAAc;;;;EAMpE,iBAAiB,QAAQO,6BAAO,KAAKC,gBAAgB,CAAC;AAEtD,MAAI,KAAK,QAAQ,cACf,iBAAiB,QAAQ,kBAAkB,QAAQ;AAGrD,MAAI,KAAK,QAAQ,eAAe;GAC9B,iBAAiB,QAAQ,SAAS,kBAAkB;GACpD,MAAM,gBAAgB,KAAKC,uBACzB,YAAY,OAAOT,2BAAa,CACjC;AACD,OAAI,cAAc,WAAW,GAC3B,iBAAiB,QAAQ,mBAAmB,cAAc,GAAG;QAE7D,iBAAiB,oBACf,mBACA,KAAKU,4BAA4B,EACjC,cACD;EAEJ,OAAM;GACL,MAAM,aAAa,KAAKC,eAAe,YAAY,OAAOX,2BAAa,CAAC;AACxE,OAAI,WAAW,WAAW,GACxB,iBAAiB,QAAQ,SAAS,WAAW,GAAG;QAEhD,iBAAiB,oBACf,SACA,KAAKY,oBAAoB,EACzB,WACD;EAEJ;;;;AAKD,MAAI,YAAY,OAAOZ,2BAAa,CAAC,SAAS,EAC5C,KAAI,mBAAmB,OAAO,GAC5B,iBAAiB,oBACf,SACA,KAAKa,mBAAmB,mBAAmB,EAC3C,CAAC,KAAKL,gBAAgB,EAAEM,yBAAI,EAC7B;OAED,iBAAiB,QAAQ,SAAS,KAAKN,gBAAgB,CAAC;;;;EAO5D,KAAKO,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;CAED,iBAAiB;EACf,MAAM,aAAa,KAAK,QAAQ,eAAe,mBAAmB;AAClE,SAAO;CACR;;;;CAKD,uBAAuBC,aAA0C;EAC/D,MAAMC,QAA+D,CAAE;AACvE,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,KAAKT,gBAAgB,EAAE,QAAQ;EAE5C,MAAM,KAAKM,0BAAI;AACf,SAAO;CACR;CAED,6BAA6B;AAC3B,SAAO,CAACI,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,eAAe,SAAS,OAAOC,sCAAY,WAAW;GAC5D,MAAM,gBAAgB,SAAS,OAAOC,oCAAU,WAAW,CAAC,GAAG,GAAG;GAClE,MAAM,mBAAmB,eAAe,YAAY,OAClD,CAAC,SAAS,CAAC,aAAa,KAAK,CAAC,MAAM,EAAE,iBAAiB,KAAK,GAAG,CAChE;AAED,OAAI,oBAAoB,iBAAiB,SAAS,GAAG;;;;;;AAMnD,QAAI,KAAKtB,yBAAyB,KAChC,QAAO;;;;;AAOT,WAAO,iBAAiB,IACtB,CAAC,aAAa,IAAIuB,2BAAK,SAAS;KAAE,GAAG;KAAO,cAAc;IAAU,GACrE;GACF;AAED,OAAIF,sCAAY,WAAW,SAAS,GAAG,GAAG,CAAC,CACzC,QAAO,KAAKX,gBAAgB;AAG9B,UAAOM;EACR;CACF;;;;;;CAOD,eACEE,aAC0B;EAC1B,MAAMM,QAAkC,CAAE;AAC1C,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;EAGrB,MAAM,KAAKR,0BAAI;AAEf,SAAO;CACR;;;;CAKD,qBAAqB;;;;AAInB,SAAO,CAACI,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,CAACE,oCAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,GAClC;AACA,QAAI,KAAK,QAAQ,cACf,QAAO;AAGT,WAAON;GACR;AAED,OAAI,KAAK,QAAQ,cACf,QAAO;;;;AAMT,OAAI,KAAKhB,yBAAyB,KAChC,QAAO;;;;AAMT,UAAO,YAAY,WAAW,IAC5B,CAAC,aAAa,IAAIuB,2BAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;CAKD,mBAAmBE,oBAAiC;AAClD,SAAO,CAACL,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,SAAS,SAAS;AAG/C,OACEC,sCAAY,WAAW,YAAY,IACnC,YAAY,QACZ,mBAAmB,IAAI,YAAY,KAAK,CAIxC,QAAO,KAAK,QAAQ,iBAAiB,KAAKX,gBAAgB,GAAGM;AAI/D,UAAO,KAAKN,gBAAgB;EAC7B;CACF;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKO,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKA,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,eAGc;AAChB,SAAO,KAAKA,OAAO,aAAa,KAAK,KAAKA,OAAO;CAClD;;;;;;;;;;;CAYD,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/ReactAgent.js

@@ -51,7 +51,7 @@ var ReactAgent = class {
 		/**
 		* add single tool node for all tools
 		*/
-		if (toolClasses.length > 0) {
+		if (toolClasses.filter(isClientTool).length > 0) {
 			const toolNode = new ToolNode(toolClasses.filter(isClientTool), { signal: this.options.signal });
 			allNodeWorkflows.addNode("tools", toolNode);
 		}
@@ -81,7 +81,7 @@ var ReactAgent = class {
 		/**
 		* add edges for tools node
 		*/
-		if (toolClasses.length > 0) if (shouldReturnDirect.size > 0) allNodeWorkflows.addConditionalEdges("tools", this.#createToolsRouter(shouldReturnDirect), [this.#getEntryPoint(), END]);
+		if (toolClasses.filter(isClientTool).length > 0) if (shouldReturnDirect.size > 0) allNodeWorkflows.addConditionalEdges("tools", this.#createToolsRouter(shouldReturnDirect), [this.#getEntryPoint(), END]);
 		else allNodeWorkflows.addEdge("tools", this.#getEntryPoint());
 		/**
 		* compile the graph

package/dist/agents/ReactAgent.js.map

@@ -1 +1 @@
-{"version":3,"file":"ReactAgent.js","names":["options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >","#toolBehaviorVersion","#inputSchema","#getEntryPoint","#getPostModelHookPaths","#createPostModelHookRouter","#getModelPaths","#createModelRouter","#createToolsRouter","#graph","toolClasses: (ClientTool | ServerTool)[]","paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[]","state: InternalAgentState<StructuredResponseFormat>","paths: (\"tools\" | typeof END)[]","shouldReturnDirect: Set<string>","params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }"],"sources":["../../src/agents/ReactAgent.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  CompiledStateGraph,\n  MessagesAnnotation,\n} from \"@langchain/langgraph\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\n\nimport {\n  createAgentAnnotationConditional,\n  ReactAgentAnnotation,\n} from \"./annotation.js\";\nimport { isClientTool, validateLLMHasNoBoundTools } from \"./utils.js\";\nimport { AgentNode } from \"./nodes/AgentNode.js\";\nimport { ToolNode } from \"./nodes/ToolNode.js\";\nimport type {\n  CreateAgentParams,\n  ClientTool,\n  ServerTool,\n  InternalAgentState,\n  WithStateGraphNodes,\n} from \"./types.js\";\nimport {\n  enhanceStateSchemaWithMessageReducer,\n  type AnyAnnotationRoot,\n  type ToAnnotationRoot,\n  type ResponseFormatUndefined,\n} from \"./annotation.js\";\n\ntype AgentGraph<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> = CompiledStateGraph<\n  ToAnnotationRoot<StateSchema>[\"State\"],\n  ToAnnotationRoot<StateSchema>[\"Update\"],\n  any,\n  typeof MessagesAnnotation.spec & ToAnnotationRoot<StateSchema>[\"spec\"],\n  ReactAgentAnnotation<StructuredResponseFormat>[\"spec\"] &\n    ToAnnotationRoot<StateSchema>[\"spec\"],\n  ToAnnotationRoot<ContextSchema>[\"spec\"],\n  unknown\n>;\n\nexport class ReactAgent<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> {\n  #graph: AgentGraph<StateSchema, StructuredResponseFormat, ContextSchema>;\n\n  #inputSchema?: AnnotationRoot<ToAnnotationRoot<StateSchema>[\"spec\"]>;\n\n  #toolBehaviorVersion: \"v1\" | \"v2\" = \"v2\";\n\n  constructor(\n    public options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >\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 = Array.isArray(options.tools)\n      ? options.tools\n      : options.tools.tools;\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    const schema = this.options.stateSchema\n      ? enhanceStateSchemaWithMessageReducer(this.options.stateSchema)\n      : createAgentAnnotationConditional<StructuredResponseFormat>(\n          this.options.responseFormat !== undefined\n        );\n\n    const workflow = new StateGraph(schema, this.options.contextSchema);\n\n    const allNodeWorkflows = workflow as WithStateGraphNodes<\n      \"pre_model_hook\" | \"post_model_hook\" | \"tools\" | \"agent\",\n      typeof workflow\n    >;\n\n    /**\n     * Add Nodes\n     */\n    allNodeWorkflows.addNode(\n      \"agent\",\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        toolClasses,\n        shouldReturnDirect,\n        signal: this.options.signal,\n      }),\n      {\n        input: this.#inputSchema,\n      }\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     * setup preModelHook\n     */\n    if (options.preModelHook) {\n      allNodeWorkflows.addNode(\"pre_model_hook\", options.preModelHook);\n    }\n\n    /**\n     * setup postModelHook\n     */\n    if (options.postModelHook) {\n      allNodeWorkflows.addNode(\"post_model_hook\", options.postModelHook);\n    }\n\n    /**\n     * Add Edges\n     */\n    allNodeWorkflows.addEdge(START, this.#getEntryPoint());\n\n    if (this.options.preModelHook) {\n      allNodeWorkflows.addEdge(\"pre_model_hook\", \"agent\");\n    }\n\n    if (this.options.postModelHook) {\n      allNodeWorkflows.addEdge(\"agent\", \"post_model_hook\");\n      const postHookPaths = this.#getPostModelHookPaths(\n        toolClasses.filter(isClientTool)\n      );\n      if (postHookPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"post_model_hook\", postHookPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"post_model_hook\",\n          this.#createPostModelHookRouter(),\n          postHookPaths\n        );\n      }\n    } else {\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      if (modelPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"agent\", modelPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"agent\",\n          this.#createModelRouter(),\n          modelPaths\n        );\n      }\n    }\n\n    /**\n     * add edges for tools node\n     */\n    if (toolClasses.length > 0) {\n      if (shouldReturnDirect.size > 0) {\n        allNodeWorkflows.addConditionalEdges(\n          \"tools\",\n          this.#createToolsRouter(shouldReturnDirect),\n          [this.#getEntryPoint(), END]\n        );\n      } else {\n        allNodeWorkflows.addEdge(\"tools\", this.#getEntryPoint());\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    });\n  }\n\n  /**\n   * Get the compiled graph.\n   */\n  get graph(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat,\n    ContextSchema\n  > {\n    return this.#graph;\n  }\n\n  #getEntryPoint() {\n    const entryPoint = this.options.preModelHook ? \"pre_model_hook\" : \"agent\";\n    return entryPoint;\n  }\n\n  /**\n   * Get possible edge destinations from post_model_hook node.\n   */\n  #getPostModelHookPaths(toolClasses: (ClientTool | ServerTool)[]) {\n    const paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(this.#getEntryPoint(), \"tools\");\n    }\n    paths.push(END);\n    return paths;\n  }\n\n  #createPostModelHookRouter() {\n    return (state: InternalAgentState<StructuredResponseFormat>) => {\n      const messages = state.messages;\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\n      if (pendingToolCalls && pendingToolCalls.length > 0) {\n        /**\n         * The tool node processes a single message.\n         * All tool calls in the message are executed in parallel within the tool node.\n         * @deprecated likely to be removed in the next version of the agent\n         */\n        if (this.#toolBehaviorVersion === \"v1\") {\n          return \"tools\";\n        }\n\n        /**\n         * The tool node processes a single tool call. Tool calls are distributed across\n         * multiple instances of the tool node using the Send API.\n         */\n        return pendingToolCalls.map(\n          (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n        );\n      }\n\n      if (ToolMessage.isInstance(messages.at(-1))) {\n        return this.#getEntryPoint();\n      }\n\n      return END;\n    };\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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        if (this.options.postModelHook) {\n          return \"post_model_hook\";\n        }\n\n        return END;\n      }\n\n      if (this.options.postModelHook) {\n        return \"post_model_hook\";\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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 ? this.#getEntryPoint() : END;\n      }\n\n      // For non-returnDirect tools, always route back to agent\n      return this.#getEntryPoint();\n    };\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get invoke(): AgentGraph<StateSchema, StructuredResponseFormat>[\"invoke\"] {\n    return this.#graph.invoke.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get stream(): AgentGraph<StateSchema, StructuredResponseFormat>[\"stream\"] {\n    return this.#graph.stream.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get streamEvents(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat\n  >[\"streamEvents\"] {\n    return this.#graph.streamEvents.bind(this.#graph);\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":";;;;;;;;AAmDA,IAAa,aAAb,MAME;CACA;CAEA;CAEA,uBAAoC;CAEpC,YACSA,SAKP;EALO;EAMP,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,cAAc,MAAM,QAAQ,QAAQ,MAAM,GAC5C,QAAQ,QACR,QAAQ,MAAM;;;;;EAMlB,MAAM,qBAAqB,IAAI,IAC7B,YACG,OAAO,aAAa,CACpB,OAAO,CAAC,SAAS,kBAAkB,QAAQ,KAAK,aAAa,CAC7D,IAAI,CAAC,SAAS,KAAK,KAAK;EAG7B,MAAM,SAAS,KAAK,QAAQ,cACxB,qCAAqC,KAAK,QAAQ,YAAY,GAC9D,iCACE,KAAK,QAAQ,mBAAmB,OACjC;EAEL,MAAM,WAAW,IAAI,WAAW,QAAQ,KAAK,QAAQ;EAErD,MAAM,mBAAmB;;;;EAQzB,iBAAiB,QACf,SACA,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;GACA;GACA,QAAQ,KAAK,QAAQ;EACtB,IACD,EACE,OAAO,KAAKC,aACb,EACF;;;;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;;;;AAKD,MAAI,QAAQ,cACV,iBAAiB,QAAQ,kBAAkB,QAAQ,aAAa;;;;AAMlE,MAAI,QAAQ,eACV,iBAAiB,QAAQ,mBAAmB,QAAQ,cAAc;;;;EAMpE,iBAAiB,QAAQ,OAAO,KAAKC,gBAAgB,CAAC;AAEtD,MAAI,KAAK,QAAQ,cACf,iBAAiB,QAAQ,kBAAkB,QAAQ;AAGrD,MAAI,KAAK,QAAQ,eAAe;GAC9B,iBAAiB,QAAQ,SAAS,kBAAkB;GACpD,MAAM,gBAAgB,KAAKC,uBACzB,YAAY,OAAO,aAAa,CACjC;AACD,OAAI,cAAc,WAAW,GAC3B,iBAAiB,QAAQ,mBAAmB,cAAc,GAAG;QAE7D,iBAAiB,oBACf,mBACA,KAAKC,4BAA4B,EACjC,cACD;EAEJ,OAAM;GACL,MAAM,aAAa,KAAKC,eAAe,YAAY,OAAO,aAAa,CAAC;AACxE,OAAI,WAAW,WAAW,GACxB,iBAAiB,QAAQ,SAAS,WAAW,GAAG;QAEhD,iBAAiB,oBACf,SACA,KAAKC,oBAAoB,EACzB,WACD;EAEJ;;;;AAKD,MAAI,YAAY,SAAS,EACvB,KAAI,mBAAmB,OAAO,GAC5B,iBAAiB,oBACf,SACA,KAAKC,mBAAmB,mBAAmB,EAC3C,CAAC,KAAKL,gBAAgB,EAAE,GAAI,EAC7B;OAED,iBAAiB,QAAQ,SAAS,KAAKA,gBAAgB,CAAC;;;;EAO5D,KAAKM,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;CAED,iBAAiB;EACf,MAAM,aAAa,KAAK,QAAQ,eAAe,mBAAmB;AAClE,SAAO;CACR;;;;CAKD,uBAAuBC,aAA0C;EAC/D,MAAMC,QAA+D,CAAE;AACvE,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,KAAKR,gBAAgB,EAAE,QAAQ;EAE5C,MAAM,KAAK,IAAI;AACf,SAAO;CACR;CAED,6BAA6B;AAC3B,SAAO,CAACS,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,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;AAED,OAAI,oBAAoB,iBAAiB,SAAS,GAAG;;;;;;AAMnD,QAAI,KAAKX,yBAAyB,KAChC,QAAO;;;;;AAOT,WAAO,iBAAiB,IACtB,CAAC,aAAa,IAAI,KAAK,SAAS;KAAE,GAAG;KAAO,cAAc;IAAU,GACrE;GACF;AAED,OAAI,YAAY,WAAW,SAAS,GAAG,GAAG,CAAC,CACzC,QAAO,KAAKE,gBAAgB;AAG9B,UAAO;EACR;CACF;;;;;;CAOD,eACEO,aAC0B;EAC1B,MAAMG,QAAkC,CAAE;AAC1C,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;EAGrB,MAAM,KAAK,IAAI;AAEf,SAAO;CACR;;;;CAKD,qBAAqB;;;;AAInB,SAAO,CAACD,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,CAAC,UAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,GAClC;AACA,QAAI,KAAK,QAAQ,cACf,QAAO;AAGT,WAAO;GACR;AAED,OAAI,KAAK,QAAQ,cACf,QAAO;;;;AAMT,OAAI,KAAKX,yBAAyB,KAChC,QAAO;;;;AAMT,UAAO,YAAY,WAAW,IAC5B,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;CAKD,mBAAmBa,oBAAiC;AAClD,SAAO,CAACF,UAAwD;GAC9D,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,KAAKT,gBAAgB,GAAG;AAI/D,UAAO,KAAKA,gBAAgB;EAC7B;CACF;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKM,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKA,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,eAGc;AAChB,SAAO,KAAKA,OAAO,aAAa,KAAK,KAAKA,OAAO;CAClD;;;;;;;;;;;CAYD,MAAM,eAAeM,QAMlB;EACD,MAAM,iBAAiB,MAAM,KAAKN,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,YAAYM,QAMf;EACD,MAAM,iBAAiB,MAAM,KAAKN,OAAO,eAAe;AACxD,SAAO,eAAe,YAAY,OAAO;CAC1C;AACF"}
+{"version":3,"file":"ReactAgent.js","names":["options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >","#toolBehaviorVersion","#inputSchema","#getEntryPoint","#getPostModelHookPaths","#createPostModelHookRouter","#getModelPaths","#createModelRouter","#createToolsRouter","#graph","toolClasses: (ClientTool | ServerTool)[]","paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[]","state: InternalAgentState<StructuredResponseFormat>","paths: (\"tools\" | typeof END)[]","shouldReturnDirect: Set<string>","params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }"],"sources":["../../src/agents/ReactAgent.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  CompiledStateGraph,\n  MessagesAnnotation,\n} from \"@langchain/langgraph\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\n\nimport {\n  createAgentAnnotationConditional,\n  ReactAgentAnnotation,\n} from \"./annotation.js\";\nimport { isClientTool, validateLLMHasNoBoundTools } from \"./utils.js\";\nimport { AgentNode } from \"./nodes/AgentNode.js\";\nimport { ToolNode } from \"./nodes/ToolNode.js\";\nimport type {\n  CreateAgentParams,\n  ClientTool,\n  ServerTool,\n  InternalAgentState,\n  WithStateGraphNodes,\n} from \"./types.js\";\nimport {\n  enhanceStateSchemaWithMessageReducer,\n  type AnyAnnotationRoot,\n  type ToAnnotationRoot,\n  type ResponseFormatUndefined,\n} from \"./annotation.js\";\n\ntype AgentGraph<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> = CompiledStateGraph<\n  ToAnnotationRoot<StateSchema>[\"State\"],\n  ToAnnotationRoot<StateSchema>[\"Update\"],\n  any,\n  typeof MessagesAnnotation.spec & ToAnnotationRoot<StateSchema>[\"spec\"],\n  ReactAgentAnnotation<StructuredResponseFormat>[\"spec\"] &\n    ToAnnotationRoot<StateSchema>[\"spec\"],\n  ToAnnotationRoot<ContextSchema>[\"spec\"],\n  unknown\n>;\n\nexport class ReactAgent<\n  StateSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot,\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot\n> {\n  #graph: AgentGraph<StateSchema, StructuredResponseFormat, ContextSchema>;\n\n  #inputSchema?: AnnotationRoot<ToAnnotationRoot<StateSchema>[\"spec\"]>;\n\n  #toolBehaviorVersion: \"v1\" | \"v2\" = \"v2\";\n\n  constructor(\n    public options: CreateAgentParams<\n      StateSchema,\n      StructuredResponseFormat,\n      ContextSchema\n    >\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 = Array.isArray(options.tools)\n      ? options.tools\n      : options.tools.tools;\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    const schema = this.options.stateSchema\n      ? enhanceStateSchemaWithMessageReducer(this.options.stateSchema)\n      : createAgentAnnotationConditional<StructuredResponseFormat>(\n          this.options.responseFormat !== undefined\n        );\n\n    const workflow = new StateGraph(schema, this.options.contextSchema);\n\n    const allNodeWorkflows = workflow as WithStateGraphNodes<\n      \"pre_model_hook\" | \"post_model_hook\" | \"tools\" | \"agent\",\n      typeof workflow\n    >;\n\n    /**\n     * Add Nodes\n     */\n    allNodeWorkflows.addNode(\n      \"agent\",\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        toolClasses,\n        shouldReturnDirect,\n        signal: this.options.signal,\n      }),\n      {\n        input: this.#inputSchema,\n      }\n    );\n\n    /**\n     * add single tool node for all tools\n     */\n    if (toolClasses.filter(isClientTool).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     * setup preModelHook\n     */\n    if (options.preModelHook) {\n      allNodeWorkflows.addNode(\"pre_model_hook\", options.preModelHook);\n    }\n\n    /**\n     * setup postModelHook\n     */\n    if (options.postModelHook) {\n      allNodeWorkflows.addNode(\"post_model_hook\", options.postModelHook);\n    }\n\n    /**\n     * Add Edges\n     */\n    allNodeWorkflows.addEdge(START, this.#getEntryPoint());\n\n    if (this.options.preModelHook) {\n      allNodeWorkflows.addEdge(\"pre_model_hook\", \"agent\");\n    }\n\n    if (this.options.postModelHook) {\n      allNodeWorkflows.addEdge(\"agent\", \"post_model_hook\");\n      const postHookPaths = this.#getPostModelHookPaths(\n        toolClasses.filter(isClientTool)\n      );\n      if (postHookPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"post_model_hook\", postHookPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"post_model_hook\",\n          this.#createPostModelHookRouter(),\n          postHookPaths\n        );\n      }\n    } else {\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      if (modelPaths.length === 1) {\n        allNodeWorkflows.addEdge(\"agent\", modelPaths[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"agent\",\n          this.#createModelRouter(),\n          modelPaths\n        );\n      }\n    }\n\n    /**\n     * add edges for tools node\n     */\n    if (toolClasses.filter(isClientTool).length > 0) {\n      if (shouldReturnDirect.size > 0) {\n        allNodeWorkflows.addConditionalEdges(\n          \"tools\",\n          this.#createToolsRouter(shouldReturnDirect),\n          [this.#getEntryPoint(), END]\n        );\n      } else {\n        allNodeWorkflows.addEdge(\"tools\", this.#getEntryPoint());\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    });\n  }\n\n  /**\n   * Get the compiled graph.\n   */\n  get graph(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat,\n    ContextSchema\n  > {\n    return this.#graph;\n  }\n\n  #getEntryPoint() {\n    const entryPoint = this.options.preModelHook ? \"pre_model_hook\" : \"agent\";\n    return entryPoint;\n  }\n\n  /**\n   * Get possible edge destinations from post_model_hook node.\n   */\n  #getPostModelHookPaths(toolClasses: (ClientTool | ServerTool)[]) {\n    const paths: (typeof END | \"agent\" | \"pre_model_hook\" | \"tools\")[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(this.#getEntryPoint(), \"tools\");\n    }\n    paths.push(END);\n    return paths;\n  }\n\n  #createPostModelHookRouter() {\n    return (state: InternalAgentState<StructuredResponseFormat>) => {\n      const messages = state.messages;\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\n      if (pendingToolCalls && pendingToolCalls.length > 0) {\n        /**\n         * The tool node processes a single message.\n         * All tool calls in the message are executed in parallel within the tool node.\n         * @deprecated likely to be removed in the next version of the agent\n         */\n        if (this.#toolBehaviorVersion === \"v1\") {\n          return \"tools\";\n        }\n\n        /**\n         * The tool node processes a single tool call. Tool calls are distributed across\n         * multiple instances of the tool node using the Send API.\n         */\n        return pendingToolCalls.map(\n          (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n        );\n      }\n\n      if (ToolMessage.isInstance(messages.at(-1))) {\n        return this.#getEntryPoint();\n      }\n\n      return END;\n    };\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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        if (this.options.postModelHook) {\n          return \"post_model_hook\";\n        }\n\n        return END;\n      }\n\n      if (this.options.postModelHook) {\n        return \"post_model_hook\";\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    return (state: InternalAgentState<StructuredResponseFormat>) => {\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 ? this.#getEntryPoint() : END;\n      }\n\n      // For non-returnDirect tools, always route back to agent\n      return this.#getEntryPoint();\n    };\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get invoke(): AgentGraph<StateSchema, StructuredResponseFormat>[\"invoke\"] {\n    return this.#graph.invoke.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get stream(): AgentGraph<StateSchema, StructuredResponseFormat>[\"stream\"] {\n    return this.#graph.stream.bind(this.#graph);\n  }\n\n  /**\n   * @inheritdoc\n   */\n  get streamEvents(): AgentGraph<\n    StateSchema,\n    StructuredResponseFormat\n  >[\"streamEvents\"] {\n    return this.#graph.streamEvents.bind(this.#graph);\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":";;;;;;;;AAmDA,IAAa,aAAb,MAME;CACA;CAEA;CAEA,uBAAoC;CAEpC,YACSA,SAKP;EALO;EAMP,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,cAAc,MAAM,QAAQ,QAAQ,MAAM,GAC5C,QAAQ,QACR,QAAQ,MAAM;;;;;EAMlB,MAAM,qBAAqB,IAAI,IAC7B,YACG,OAAO,aAAa,CACpB,OAAO,CAAC,SAAS,kBAAkB,QAAQ,KAAK,aAAa,CAC7D,IAAI,CAAC,SAAS,KAAK,KAAK;EAG7B,MAAM,SAAS,KAAK,QAAQ,cACxB,qCAAqC,KAAK,QAAQ,YAAY,GAC9D,iCACE,KAAK,QAAQ,mBAAmB,OACjC;EAEL,MAAM,WAAW,IAAI,WAAW,QAAQ,KAAK,QAAQ;EAErD,MAAM,mBAAmB;;;;EAQzB,iBAAiB,QACf,SACA,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;GACA;GACA,QAAQ,KAAK,QAAQ;EACtB,IACD,EACE,OAAO,KAAKC,aACb,EACF;;;;AAKD,MAAI,YAAY,OAAO,aAAa,CAAC,SAAS,GAAG;GAC/C,MAAM,WAAW,IAAI,SAAS,YAAY,OAAO,aAAa,EAAE,EAC9D,QAAQ,KAAK,QAAQ,OACtB;GACD,iBAAiB,QAAQ,SAAS,SAAS;EAC5C;;;;AAKD,MAAI,QAAQ,cACV,iBAAiB,QAAQ,kBAAkB,QAAQ,aAAa;;;;AAMlE,MAAI,QAAQ,eACV,iBAAiB,QAAQ,mBAAmB,QAAQ,cAAc;;;;EAMpE,iBAAiB,QAAQ,OAAO,KAAKC,gBAAgB,CAAC;AAEtD,MAAI,KAAK,QAAQ,cACf,iBAAiB,QAAQ,kBAAkB,QAAQ;AAGrD,MAAI,KAAK,QAAQ,eAAe;GAC9B,iBAAiB,QAAQ,SAAS,kBAAkB;GACpD,MAAM,gBAAgB,KAAKC,uBACzB,YAAY,OAAO,aAAa,CACjC;AACD,OAAI,cAAc,WAAW,GAC3B,iBAAiB,QAAQ,mBAAmB,cAAc,GAAG;QAE7D,iBAAiB,oBACf,mBACA,KAAKC,4BAA4B,EACjC,cACD;EAEJ,OAAM;GACL,MAAM,aAAa,KAAKC,eAAe,YAAY,OAAO,aAAa,CAAC;AACxE,OAAI,WAAW,WAAW,GACxB,iBAAiB,QAAQ,SAAS,WAAW,GAAG;QAEhD,iBAAiB,oBACf,SACA,KAAKC,oBAAoB,EACzB,WACD;EAEJ;;;;AAKD,MAAI,YAAY,OAAO,aAAa,CAAC,SAAS,EAC5C,KAAI,mBAAmB,OAAO,GAC5B,iBAAiB,oBACf,SACA,KAAKC,mBAAmB,mBAAmB,EAC3C,CAAC,KAAKL,gBAAgB,EAAE,GAAI,EAC7B;OAED,iBAAiB,QAAQ,SAAS,KAAKA,gBAAgB,CAAC;;;;EAO5D,KAAKM,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;CAED,iBAAiB;EACf,MAAM,aAAa,KAAK,QAAQ,eAAe,mBAAmB;AAClE,SAAO;CACR;;;;CAKD,uBAAuBC,aAA0C;EAC/D,MAAMC,QAA+D,CAAE;AACvE,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,KAAKR,gBAAgB,EAAE,QAAQ;EAE5C,MAAM,KAAK,IAAI;AACf,SAAO;CACR;CAED,6BAA6B;AAC3B,SAAO,CAACS,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,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;AAED,OAAI,oBAAoB,iBAAiB,SAAS,GAAG;;;;;;AAMnD,QAAI,KAAKX,yBAAyB,KAChC,QAAO;;;;;AAOT,WAAO,iBAAiB,IACtB,CAAC,aAAa,IAAI,KAAK,SAAS;KAAE,GAAG;KAAO,cAAc;IAAU,GACrE;GACF;AAED,OAAI,YAAY,WAAW,SAAS,GAAG,GAAG,CAAC,CACzC,QAAO,KAAKE,gBAAgB;AAG9B,UAAO;EACR;CACF;;;;;;CAOD,eACEO,aAC0B;EAC1B,MAAMG,QAAkC,CAAE;AAC1C,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;EAGrB,MAAM,KAAK,IAAI;AAEf,SAAO;CACR;;;;CAKD,qBAAqB;;;;AAInB,SAAO,CAACD,UAAwD;GAC9D,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,CAAC,UAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,GAClC;AACA,QAAI,KAAK,QAAQ,cACf,QAAO;AAGT,WAAO;GACR;AAED,OAAI,KAAK,QAAQ,cACf,QAAO;;;;AAMT,OAAI,KAAKX,yBAAyB,KAChC,QAAO;;;;AAMT,UAAO,YAAY,WAAW,IAC5B,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;CAKD,mBAAmBa,oBAAiC;AAClD,SAAO,CAACF,UAAwD;GAC9D,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,KAAKT,gBAAgB,GAAG;AAI/D,UAAO,KAAKA,gBAAgB;EAC7B;CACF;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKM,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,SAAsE;AACxE,SAAO,KAAKA,OAAO,OAAO,KAAK,KAAKA,OAAO;CAC5C;;;;CAKD,IAAI,eAGc;AAChB,SAAO,KAAKA,OAAO,aAAa,KAAK,KAAKA,OAAO;CAClD;;;;;;;;;;;CAYD,MAAM,eAAeM,QAMlB;EACD,MAAM,iBAAiB,MAAM,KAAKN,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,YAAYM,QAMf;EACD,MAAM,iBAAiB,MAAM,KAAKN,OAAO,eAAe;AACxD,SAAO,eAAe,YAAY,OAAO;CAC1C;AACF"}

package/dist/agents/RunnableCallable.cjs

@@ -21,6 +21,16 @@ var RunnableCallable = class extends __langchain_core_runnables.Runnable {
 	getState() {
 		return this.#state;
 	}
+	/**
+	* This allows us to set the state of the runnable, e.g. for model and middleware nodes.
+	* @internal
+	*/
+	setState(state) {
+		this.#state = {
+			...this.#state,
+			...state
+		};
+	}
 	async invoke(input, options) {
 		const mergedConfig = (0, __langchain_core_runnables.mergeConfigs)(this.config, options);
 		const returnValue = await __langchain_core_singletons.AsyncLocalStorageProviderSingleton.runWithConfig(mergedConfig, async () => this.func(input, mergedConfig));

package/dist/agents/RunnableCallable.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"RunnableCallable.cjs","names":["Runnable","fields: RunnableCallableArgs<I, O>","#state","input: I","options?: Partial<RunnableConfig> | undefined","AsyncLocalStorageProviderSingleton"],"sources":["../../src/agents/RunnableCallable.ts"],"sourcesContent":["import {\n  mergeConfigs,\n  Runnable,\n  RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\n\nexport interface RunnableCallableArgs<I, O> {\n  /**\n   * The name of the runnable.\n   */\n  name?: string;\n  /**\n   * The function to call.\n   */\n  func: (...args: I[]) => O | Promise<O>;\n  /**\n   * The tags to add to the runnable.\n   */\n  tags?: string[];\n  /**\n   * Whether to recurse the runnable.\n   */\n  recurse?: boolean;\n}\n\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n  lc_namespace: string[] = [\"langgraph\"];\n\n  func: RunnableCallableArgs<I, O>[\"func\"];\n\n  tags?: RunnableCallableArgs<I, O>[\"tags\"];\n\n  config?: RunnableConfig;\n\n  trace = true;\n\n  recurse = true;\n\n  #state: Awaited<O>;\n\n  constructor(fields: RunnableCallableArgs<I, O>) {\n    super();\n    this.name = fields.name ?? fields.func.name;\n    this.func = fields.func;\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\n    this.recurse = fields.recurse ?? this.recurse;\n  }\n\n  getState(): Awaited<O> {\n    return this.#state;\n  }\n\n  async invoke(\n    input: I,\n    options?: Partial<RunnableConfig> | undefined\n  ): Promise<O> {\n    const mergedConfig = mergeConfigs(this.config, options);\n\n    const returnValue = await AsyncLocalStorageProviderSingleton.runWithConfig(\n      mergedConfig,\n      async () => this.func(input, mergedConfig as I)\n    );\n\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\n      return await AsyncLocalStorageProviderSingleton.runWithConfig(\n        mergedConfig,\n        async () => (returnValue as Runnable<I, O>).invoke(input, mergedConfig)\n      );\n    }\n\n    this.#state = returnValue;\n    return returnValue;\n  }\n}\n"],"mappings":";;;;;AA0BA,IAAa,mBAAb,cAAgEA,oCAAe;CAC7E,eAAyB,CAAC,WAAY;CAEtC;CAEA;CAEA;CAEA,QAAQ;CAER,UAAU;CAEV;CAEA,YAAYC,QAAoC;EAC9C,OAAO;EACP,KAAK,OAAO,OAAO,QAAQ,OAAO,KAAK;EACvC,KAAK,OAAO,OAAO;EACnB,KAAK,SAAS,OAAO,OAAO,EAAE,MAAM,OAAO,KAAM,IAAG;EACpD,KAAK,UAAU,OAAO,WAAW,KAAK;CACvC;CAED,WAAuB;AACrB,SAAO,KAAKC;CACb;CAED,MAAM,OACJC,OACAC,SACY;EACZ,MAAM,4DAA4B,KAAK,QAAQ,QAAQ;EAEvD,MAAM,cAAc,MAAMC,+DAAmC,cAC3D,cACA,YAAY,KAAK,KAAK,OAAO,aAAkB,CAChD;AAED,MAAIL,oCAAS,WAAW,YAAY,IAAI,KAAK,QAC3C,QAAO,MAAMK,+DAAmC,cAC9C,cACA,YAAa,YAA+B,OAAO,OAAO,aAAa,CACxE;EAGH,KAAKH,SAAS;AACd,SAAO;CACR;AACF"}
+{"version":3,"file":"RunnableCallable.cjs","names":["Runnable","fields: RunnableCallableArgs<I, O>","#state","state: Awaited<O>","input: I","options?: Partial<RunnableConfig> | undefined","AsyncLocalStorageProviderSingleton"],"sources":["../../src/agents/RunnableCallable.ts"],"sourcesContent":["import {\n  mergeConfigs,\n  Runnable,\n  RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\n\nexport interface RunnableCallableArgs<I, O> {\n  /**\n   * The name of the runnable.\n   */\n  name?: string;\n  /**\n   * The function to call.\n   */\n  func: (...args: I[]) => O | Promise<O>;\n  /**\n   * The tags to add to the runnable.\n   */\n  tags?: string[];\n  /**\n   * Whether to recurse the runnable.\n   */\n  recurse?: boolean;\n}\n\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n  lc_namespace: string[] = [\"langgraph\"];\n\n  func: RunnableCallableArgs<I, O>[\"func\"];\n\n  tags?: RunnableCallableArgs<I, O>[\"tags\"];\n\n  config?: RunnableConfig;\n\n  trace = true;\n\n  recurse = true;\n\n  #state: Awaited<O>;\n\n  constructor(fields: RunnableCallableArgs<I, O>) {\n    super();\n    this.name = fields.name ?? fields.func.name;\n    this.func = fields.func;\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\n    this.recurse = fields.recurse ?? this.recurse;\n  }\n\n  getState(): Awaited<O> {\n    return this.#state;\n  }\n\n  /**\n   * This allows us to set the state of the runnable, e.g. for model and middleware nodes.\n   * @internal\n   */\n  setState(state: Awaited<O>) {\n    this.#state = {\n      ...this.#state,\n      ...state,\n    };\n  }\n\n  async invoke(\n    input: I,\n    options?: Partial<RunnableConfig> | undefined\n  ): Promise<O> {\n    const mergedConfig = mergeConfigs(this.config, options);\n\n    const returnValue = await AsyncLocalStorageProviderSingleton.runWithConfig(\n      mergedConfig,\n      async () => this.func(input, mergedConfig as I)\n    );\n\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\n      return await AsyncLocalStorageProviderSingleton.runWithConfig(\n        mergedConfig,\n        async () => (returnValue as Runnable<I, O>).invoke(input, mergedConfig)\n      );\n    }\n\n    this.#state = returnValue;\n    return returnValue;\n  }\n}\n"],"mappings":";;;;;AA0BA,IAAa,mBAAb,cAAgEA,oCAAe;CAC7E,eAAyB,CAAC,WAAY;CAEtC;CAEA;CAEA;CAEA,QAAQ;CAER,UAAU;CAEV;CAEA,YAAYC,QAAoC;EAC9C,OAAO;EACP,KAAK,OAAO,OAAO,QAAQ,OAAO,KAAK;EACvC,KAAK,OAAO,OAAO;EACnB,KAAK,SAAS,OAAO,OAAO,EAAE,MAAM,OAAO,KAAM,IAAG;EACpD,KAAK,UAAU,OAAO,WAAW,KAAK;CACvC;CAED,WAAuB;AACrB,SAAO,KAAKC;CACb;;;;;CAMD,SAASC,OAAmB;EAC1B,KAAKD,SAAS;GACZ,GAAG,KAAKA;GACR,GAAG;EACJ;CACF;CAED,MAAM,OACJE,OACAC,SACY;EACZ,MAAM,4DAA4B,KAAK,QAAQ,QAAQ;EAEvD,MAAM,cAAc,MAAMC,+DAAmC,cAC3D,cACA,YAAY,KAAK,KAAK,OAAO,aAAkB,CAChD;AAED,MAAIN,oCAAS,WAAW,YAAY,IAAI,KAAK,QAC3C,QAAO,MAAMM,+DAAmC,cAC9C,cACA,YAAa,YAA+B,OAAO,OAAO,aAAa,CACxE;EAGH,KAAKJ,SAAS;AACd,SAAO;CACR;AACF"}

package/dist/agents/RunnableCallable.d.cts

@@ -29,6 +29,11 @@ declare class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O>
   recurse: boolean;
   constructor(fields: RunnableCallableArgs<I, O>);
   getState(): Awaited<O>;
+  /**
+   * This allows us to set the state of the runnable, e.g. for model and middleware nodes.
+   * @internal
+   */
+  setState(state: Awaited<O>): void;
   invoke(input: I, options?: Partial<RunnableConfig> | undefined): Promise<O>;
 }
 //#endregion

package/dist/agents/RunnableCallable.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"RunnableCallable.d.cts","names":["Runnable","RunnableConfig","RunnableCallableArgs","I","O","Promise","RunnableCallable","Awaited","Partial"],"sources":["../../src/agents/RunnableCallable.d.ts"],"sourcesContent":["import { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nexport interface RunnableCallableArgs<I, O> {\n    /**\n     * The name of the runnable.\n     */\n    name?: string;\n    /**\n     * The function to call.\n     */\n    func: (...args: I[]) => O | Promise<O>;\n    /**\n     * The tags to add to the runnable.\n     */\n    tags?: string[];\n    /**\n     * Whether to recurse the runnable.\n     */\n    recurse?: boolean;\n}\nexport declare class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n    #private;\n    lc_namespace: string[];\n    func: RunnableCallableArgs<I, O>[\"func\"];\n    tags?: RunnableCallableArgs<I, O>[\"tags\"];\n    config?: RunnableConfig;\n    trace: boolean;\n    recurse: boolean;\n    constructor(fields: RunnableCallableArgs<I, O>);\n    getState(): Awaited<O>;\n    invoke(input: I, options?: Partial<RunnableConfig> | undefined): Promise<O>;\n}\n"],"mappings":";;;UACiBE;;AAAjB;;EAAqC,IAQjBC,CAAAA,EAAAA,MAAAA;EAAC;;;EAAkB,IAAA,EAAA,CAAA,GAAA,IAAA,EAAnBA,CAAmB,EAAA,EAAA,GAAXC,CAAW,GAAPC,OAAO,CAACD,CAAD,CAAA;EAUlBE;;;EAA6D,IAAEF,CAAAA,EAAAA,MAAAA,EAAAA;EAAC;;;EAGvD,OACED,CAAAA,EAAAA,OAAAA;;AAArBD,cAJUI,gBAIVJ,CAAAA,IAAAA,OAAAA,EAAAA,IAAAA,OAAAA,CAAAA,SAJ6DF,QAI7DE,CAJsEC,CAItED,EAJyEE,CAIzEF,CAAAA,CAAAA;EAAoB,CAAA,OAClBD;EAAc,YAGkBE,EAAAA,MAAAA,EAAAA;EAAC,IAAEC,EALtCF,oBAKsCE,CALjBD,CAKiBC,EALdA,CAKcA,CAAAA,CAAAA,MAAAA,CAAAA;EAAC,IAAzBF,CAAAA,EAJbA,oBAIaA,CAJQC,CAIRD,EAJWE,CAIXF,CAAAA,CAAAA,MAAAA,CAAAA;EAAoB,MACpBE,CAAAA,EAJXH,cAIWG;EAAC,KAATG,EAAAA,OAAAA;EAAO,OACLJ,EAAAA,OAAAA;EAAC,WAAoBF,CAAAA,MAAAA,EAFfC,oBAEeD,CAFME,CAENF,EAFSG,CAETH,CAAAA;EAAc,QAAtBO,CAAAA,CAAAA,EADfD,OACeC,CADPJ,CACOI,CAAAA;EAAO,MAAuCJ,CAAAA,KAAAA,EAA3DD,CAA2DC,EAAAA,OAAAA,CAAAA,EAA9CI,OAA8CJ,CAAtCH,cAAsCG,CAAAA,GAAAA,SAAAA,CAAAA,EAARC,OAAQD,CAAAA,CAAAA,CAAAA"}
+{"version":3,"file":"RunnableCallable.d.cts","names":["Runnable","RunnableConfig","RunnableCallableArgs","I","O","Promise","RunnableCallable","Awaited","Partial"],"sources":["../../src/agents/RunnableCallable.d.ts"],"sourcesContent":["import { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nexport interface RunnableCallableArgs<I, O> {\n    /**\n     * The name of the runnable.\n     */\n    name?: string;\n    /**\n     * The function to call.\n     */\n    func: (...args: I[]) => O | Promise<O>;\n    /**\n     * The tags to add to the runnable.\n     */\n    tags?: string[];\n    /**\n     * Whether to recurse the runnable.\n     */\n    recurse?: boolean;\n}\nexport declare class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n    #private;\n    lc_namespace: string[];\n    func: RunnableCallableArgs<I, O>[\"func\"];\n    tags?: RunnableCallableArgs<I, O>[\"tags\"];\n    config?: RunnableConfig;\n    trace: boolean;\n    recurse: boolean;\n    constructor(fields: RunnableCallableArgs<I, O>);\n    getState(): Awaited<O>;\n    /**\n     * This allows us to set the state of the runnable, e.g. for model and middleware nodes.\n     * @internal\n     */\n    setState(state: Awaited<O>): void;\n    invoke(input: I, options?: Partial<RunnableConfig> | undefined): Promise<O>;\n}\n"],"mappings":";;;UACiBE;;AAAjB;;EAAqC,IAQjBC,CAAAA,EAAAA,MAAAA;EAAC;;;EAAkB,IAAA,EAAA,CAAA,GAAA,IAAA,EAAnBA,CAAmB,EAAA,EAAA,GAAXC,CAAW,GAAPC,OAAO,CAACD,CAAD,CAAA;EAUlBE;;;EAA6D,IAAEF,CAAAA,EAAAA,MAAAA,EAAAA;EAAC;;;EAGvD,OACED,CAAAA,EAAAA,OAAAA;;AAArBD,cAJUI,gBAIVJ,CAAAA,IAAAA,OAAAA,EAAAA,IAAAA,OAAAA,CAAAA,SAJ6DF,QAI7DE,CAJsEC,CAItED,EAJyEE,CAIzEF,CAAAA,CAAAA;EAAoB,CAAA,OAClBD;EAAc,YAGkBE,EAAAA,MAAAA,EAAAA;EAAC,IAAEC,EALtCF,oBAKsCE,CALjBD,CAKiBC,EALdA,CAKcA,CAAAA,CAAAA,MAAAA,CAAAA;EAAC,IAAzBF,CAAAA,EAJbA,oBAIaA,CAJQC,CAIRD,EAJWE,CAIXF,CAAAA,CAAAA,MAAAA,CAAAA;EAAoB,MACpBE,CAAAA,EAJXH,cAIWG;EAAC,KAATG,EAAAA,OAAAA;EAAO,OAKKH,EAAAA,OAAAA;EAAC,WAATG,CAAAA,MAAAA,EANIL,oBAMJK,CANyBJ,CAMzBI,EAN4BH,CAM5BG,CAAAA;EAAO,QACTJ,CAAAA,CAAAA,EANFI,OAMEJ,CANMC,CAMND,CAAAA;EAAC;;;;EAAyD,QAfJH,CAAAA,KAAAA,EAcpDO,OAdoDP,CAc5CI,CAd4CJ,CAAAA,CAAAA,EAAAA,IAAAA;EAAQ,MAAA,CAAA,KAAA,EAe9DG,CAf8D,EAAA,OAAA,CAAA,EAejDK,OAfiD,CAezCP,cAfyC,CAAA,GAAA,SAAA,CAAA,EAeXI,OAfW,CAeHD,CAfG,CAAA"}

package/dist/agents/RunnableCallable.d.ts

@@ -29,6 +29,11 @@ declare class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O>
   recurse: boolean;
   constructor(fields: RunnableCallableArgs<I, O>);
   getState(): Awaited<O>;
+  /**
+   * This allows us to set the state of the runnable, e.g. for model and middleware nodes.
+   * @internal
+   */
+  setState(state: Awaited<O>): void;
   invoke(input: I, options?: Partial<RunnableConfig> | undefined): Promise<O>;
 }
 //#endregion

package/dist/agents/RunnableCallable.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RunnableCallable.d.ts","names":["Runnable","RunnableConfig","RunnableCallableArgs","I","O","Promise","RunnableCallable","Awaited","Partial"],"sources":["../../src/agents/RunnableCallable.d.ts"],"sourcesContent":["import { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nexport interface RunnableCallableArgs<I, O> {\n    /**\n     * The name of the runnable.\n     */\n    name?: string;\n    /**\n     * The function to call.\n     */\n    func: (...args: I[]) => O | Promise<O>;\n    /**\n     * The tags to add to the runnable.\n     */\n    tags?: string[];\n    /**\n     * Whether to recurse the runnable.\n     */\n    recurse?: boolean;\n}\nexport declare class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n    #private;\n    lc_namespace: string[];\n    func: RunnableCallableArgs<I, O>[\"func\"];\n    tags?: RunnableCallableArgs<I, O>[\"tags\"];\n    config?: RunnableConfig;\n    trace: boolean;\n    recurse: boolean;\n    constructor(fields: RunnableCallableArgs<I, O>);\n    getState(): Awaited<O>;\n    invoke(input: I, options?: Partial<RunnableConfig> | undefined): Promise<O>;\n}\n"],"mappings":";;;UACiBE;;AAAjB;;EAAqC,IAQjBC,CAAAA,EAAAA,MAAAA;EAAC;;;EAAkB,IAAA,EAAA,CAAA,GAAA,IAAA,EAAnBA,CAAmB,EAAA,EAAA,GAAXC,CAAW,GAAPC,OAAO,CAACD,CAAD,CAAA;EAUlBE;;;EAA6D,IAAEF,CAAAA,EAAAA,MAAAA,EAAAA;EAAC;;;EAGvD,OACED,CAAAA,EAAAA,OAAAA;;AAArBD,cAJUI,gBAIVJ,CAAAA,IAAAA,OAAAA,EAAAA,IAAAA,OAAAA,CAAAA,SAJ6DF,QAI7DE,CAJsEC,CAItED,EAJyEE,CAIzEF,CAAAA,CAAAA;EAAoB,CAAA,OAClBD;EAAc,YAGkBE,EAAAA,MAAAA,EAAAA;EAAC,IAAEC,EALtCF,oBAKsCE,CALjBD,CAKiBC,EALdA,CAKcA,CAAAA,CAAAA,MAAAA,CAAAA;EAAC,IAAzBF,CAAAA,EAJbA,oBAIaA,CAJQC,CAIRD,EAJWE,CAIXF,CAAAA,CAAAA,MAAAA,CAAAA;EAAoB,MACpBE,CAAAA,EAJXH,cAIWG;EAAC,KAATG,EAAAA,OAAAA;EAAO,OACLJ,EAAAA,OAAAA;EAAC,WAAoBF,CAAAA,MAAAA,EAFfC,oBAEeD,CAFME,CAENF,EAFSG,CAETH,CAAAA;EAAc,QAAtBO,CAAAA,CAAAA,EADfD,OACeC,CADPJ,CACOI,CAAAA;EAAO,MAAuCJ,CAAAA,KAAAA,EAA3DD,CAA2DC,EAAAA,OAAAA,CAAAA,EAA9CI,OAA8CJ,CAAtCH,cAAsCG,CAAAA,GAAAA,SAAAA,CAAAA,EAARC,OAAQD,CAAAA,CAAAA,CAAAA"}
+{"version":3,"file":"RunnableCallable.d.ts","names":["Runnable","RunnableConfig","RunnableCallableArgs","I","O","Promise","RunnableCallable","Awaited","Partial"],"sources":["../../src/agents/RunnableCallable.d.ts"],"sourcesContent":["import { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nexport interface RunnableCallableArgs<I, O> {\n    /**\n     * The name of the runnable.\n     */\n    name?: string;\n    /**\n     * The function to call.\n     */\n    func: (...args: I[]) => O | Promise<O>;\n    /**\n     * The tags to add to the runnable.\n     */\n    tags?: string[];\n    /**\n     * Whether to recurse the runnable.\n     */\n    recurse?: boolean;\n}\nexport declare class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n    #private;\n    lc_namespace: string[];\n    func: RunnableCallableArgs<I, O>[\"func\"];\n    tags?: RunnableCallableArgs<I, O>[\"tags\"];\n    config?: RunnableConfig;\n    trace: boolean;\n    recurse: boolean;\n    constructor(fields: RunnableCallableArgs<I, O>);\n    getState(): Awaited<O>;\n    /**\n     * This allows us to set the state of the runnable, e.g. for model and middleware nodes.\n     * @internal\n     */\n    setState(state: Awaited<O>): void;\n    invoke(input: I, options?: Partial<RunnableConfig> | undefined): Promise<O>;\n}\n"],"mappings":";;;UACiBE;;AAAjB;;EAAqC,IAQjBC,CAAAA,EAAAA,MAAAA;EAAC;;;EAAkB,IAAA,EAAA,CAAA,GAAA,IAAA,EAAnBA,CAAmB,EAAA,EAAA,GAAXC,CAAW,GAAPC,OAAO,CAACD,CAAD,CAAA;EAUlBE;;;EAA6D,IAAEF,CAAAA,EAAAA,MAAAA,EAAAA;EAAC;;;EAGvD,OACED,CAAAA,EAAAA,OAAAA;;AAArBD,cAJUI,gBAIVJ,CAAAA,IAAAA,OAAAA,EAAAA,IAAAA,OAAAA,CAAAA,SAJ6DF,QAI7DE,CAJsEC,CAItED,EAJyEE,CAIzEF,CAAAA,CAAAA;EAAoB,CAAA,OAClBD;EAAc,YAGkBE,EAAAA,MAAAA,EAAAA;EAAC,IAAEC,EALtCF,oBAKsCE,CALjBD,CAKiBC,EALdA,CAKcA,CAAAA,CAAAA,MAAAA,CAAAA;EAAC,IAAzBF,CAAAA,EAJbA,oBAIaA,CAJQC,CAIRD,EAJWE,CAIXF,CAAAA,CAAAA,MAAAA,CAAAA;EAAoB,MACpBE,CAAAA,EAJXH,cAIWG;EAAC,KAATG,EAAAA,OAAAA;EAAO,OAKKH,EAAAA,OAAAA;EAAC,WAATG,CAAAA,MAAAA,EANIL,oBAMJK,CANyBJ,CAMzBI,EAN4BH,CAM5BG,CAAAA;EAAO,QACTJ,CAAAA,CAAAA,EANFI,OAMEJ,CANMC,CAMND,CAAAA;EAAC;;;;EAAyD,QAfJH,CAAAA,KAAAA,EAcpDO,OAdoDP,CAc5CI,CAd4CJ,CAAAA,CAAAA,EAAAA,IAAAA;EAAQ,MAAA,CAAA,KAAA,EAe9DG,CAf8D,EAAA,OAAA,CAAA,EAejDK,OAfiD,CAezCP,cAfyC,CAAA,GAAA,SAAA,CAAA,EAeXI,OAfW,CAeHD,CAfG,CAAA"}

package/dist/agents/RunnableCallable.js

@@ -20,6 +20,16 @@ var RunnableCallable = class extends Runnable {
 	getState() {
 		return this.#state;
 	}
+	/**
+	* This allows us to set the state of the runnable, e.g. for model and middleware nodes.
+	* @internal
+	*/
+	setState(state) {
+		this.#state = {
+			...this.#state,
+			...state
+		};
+	}
 	async invoke(input, options) {
 		const mergedConfig = mergeConfigs(this.config, options);
 		const returnValue = await AsyncLocalStorageProviderSingleton.runWithConfig(mergedConfig, async () => this.func(input, mergedConfig));

package/dist/agents/RunnableCallable.js.map

@@ -1 +1 @@
-{"version":3,"file":"RunnableCallable.js","names":["fields: RunnableCallableArgs<I, O>","#state","input: I","options?: Partial<RunnableConfig> | undefined"],"sources":["../../src/agents/RunnableCallable.ts"],"sourcesContent":["import {\n  mergeConfigs,\n  Runnable,\n  RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\n\nexport interface RunnableCallableArgs<I, O> {\n  /**\n   * The name of the runnable.\n   */\n  name?: string;\n  /**\n   * The function to call.\n   */\n  func: (...args: I[]) => O | Promise<O>;\n  /**\n   * The tags to add to the runnable.\n   */\n  tags?: string[];\n  /**\n   * Whether to recurse the runnable.\n   */\n  recurse?: boolean;\n}\n\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n  lc_namespace: string[] = [\"langgraph\"];\n\n  func: RunnableCallableArgs<I, O>[\"func\"];\n\n  tags?: RunnableCallableArgs<I, O>[\"tags\"];\n\n  config?: RunnableConfig;\n\n  trace = true;\n\n  recurse = true;\n\n  #state: Awaited<O>;\n\n  constructor(fields: RunnableCallableArgs<I, O>) {\n    super();\n    this.name = fields.name ?? fields.func.name;\n    this.func = fields.func;\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\n    this.recurse = fields.recurse ?? this.recurse;\n  }\n\n  getState(): Awaited<O> {\n    return this.#state;\n  }\n\n  async invoke(\n    input: I,\n    options?: Partial<RunnableConfig> | undefined\n  ): Promise<O> {\n    const mergedConfig = mergeConfigs(this.config, options);\n\n    const returnValue = await AsyncLocalStorageProviderSingleton.runWithConfig(\n      mergedConfig,\n      async () => this.func(input, mergedConfig as I)\n    );\n\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\n      return await AsyncLocalStorageProviderSingleton.runWithConfig(\n        mergedConfig,\n        async () => (returnValue as Runnable<I, O>).invoke(input, mergedConfig)\n      );\n    }\n\n    this.#state = returnValue;\n    return returnValue;\n  }\n}\n"],"mappings":";;;;AA0BA,IAAa,mBAAb,cAAgE,SAAe;CAC7E,eAAyB,CAAC,WAAY;CAEtC;CAEA;CAEA;CAEA,QAAQ;CAER,UAAU;CAEV;CAEA,YAAYA,QAAoC;EAC9C,OAAO;EACP,KAAK,OAAO,OAAO,QAAQ,OAAO,KAAK;EACvC,KAAK,OAAO,OAAO;EACnB,KAAK,SAAS,OAAO,OAAO,EAAE,MAAM,OAAO,KAAM,IAAG;EACpD,KAAK,UAAU,OAAO,WAAW,KAAK;CACvC;CAED,WAAuB;AACrB,SAAO,KAAKC;CACb;CAED,MAAM,OACJC,OACAC,SACY;EACZ,MAAM,eAAe,aAAa,KAAK,QAAQ,QAAQ;EAEvD,MAAM,cAAc,MAAM,mCAAmC,cAC3D,cACA,YAAY,KAAK,KAAK,OAAO,aAAkB,CAChD;AAED,MAAI,SAAS,WAAW,YAAY,IAAI,KAAK,QAC3C,QAAO,MAAM,mCAAmC,cAC9C,cACA,YAAa,YAA+B,OAAO,OAAO,aAAa,CACxE;EAGH,KAAKF,SAAS;AACd,SAAO;CACR;AACF"}
+{"version":3,"file":"RunnableCallable.js","names":["fields: RunnableCallableArgs<I, O>","#state","state: Awaited<O>","input: I","options?: Partial<RunnableConfig> | undefined"],"sources":["../../src/agents/RunnableCallable.ts"],"sourcesContent":["import {\n  mergeConfigs,\n  Runnable,\n  RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\n\nexport interface RunnableCallableArgs<I, O> {\n  /**\n   * The name of the runnable.\n   */\n  name?: string;\n  /**\n   * The function to call.\n   */\n  func: (...args: I[]) => O | Promise<O>;\n  /**\n   * The tags to add to the runnable.\n   */\n  tags?: string[];\n  /**\n   * Whether to recurse the runnable.\n   */\n  recurse?: boolean;\n}\n\nexport class RunnableCallable<I = unknown, O = unknown> extends Runnable<I, O> {\n  lc_namespace: string[] = [\"langgraph\"];\n\n  func: RunnableCallableArgs<I, O>[\"func\"];\n\n  tags?: RunnableCallableArgs<I, O>[\"tags\"];\n\n  config?: RunnableConfig;\n\n  trace = true;\n\n  recurse = true;\n\n  #state: Awaited<O>;\n\n  constructor(fields: RunnableCallableArgs<I, O>) {\n    super();\n    this.name = fields.name ?? fields.func.name;\n    this.func = fields.func;\n    this.config = fields.tags ? { tags: fields.tags } : undefined;\n    this.recurse = fields.recurse ?? this.recurse;\n  }\n\n  getState(): Awaited<O> {\n    return this.#state;\n  }\n\n  /**\n   * This allows us to set the state of the runnable, e.g. for model and middleware nodes.\n   * @internal\n   */\n  setState(state: Awaited<O>) {\n    this.#state = {\n      ...this.#state,\n      ...state,\n    };\n  }\n\n  async invoke(\n    input: I,\n    options?: Partial<RunnableConfig> | undefined\n  ): Promise<O> {\n    const mergedConfig = mergeConfigs(this.config, options);\n\n    const returnValue = await AsyncLocalStorageProviderSingleton.runWithConfig(\n      mergedConfig,\n      async () => this.func(input, mergedConfig as I)\n    );\n\n    if (Runnable.isRunnable(returnValue) && this.recurse) {\n      return await AsyncLocalStorageProviderSingleton.runWithConfig(\n        mergedConfig,\n        async () => (returnValue as Runnable<I, O>).invoke(input, mergedConfig)\n      );\n    }\n\n    this.#state = returnValue;\n    return returnValue;\n  }\n}\n"],"mappings":";;;;AA0BA,IAAa,mBAAb,cAAgE,SAAe;CAC7E,eAAyB,CAAC,WAAY;CAEtC;CAEA;CAEA;CAEA,QAAQ;CAER,UAAU;CAEV;CAEA,YAAYA,QAAoC;EAC9C,OAAO;EACP,KAAK,OAAO,OAAO,QAAQ,OAAO,KAAK;EACvC,KAAK,OAAO,OAAO;EACnB,KAAK,SAAS,OAAO,OAAO,EAAE,MAAM,OAAO,KAAM,IAAG;EACpD,KAAK,UAAU,OAAO,WAAW,KAAK;CACvC;CAED,WAAuB;AACrB,SAAO,KAAKC;CACb;;;;;CAMD,SAASC,OAAmB;EAC1B,KAAKD,SAAS;GACZ,GAAG,KAAKA;GACR,GAAG;EACJ;CACF;CAED,MAAM,OACJE,OACAC,SACY;EACZ,MAAM,eAAe,aAAa,KAAK,QAAQ,QAAQ;EAEvD,MAAM,cAAc,MAAM,mCAAmC,cAC3D,cACA,YAAY,KAAK,KAAK,OAAO,aAAkB,CAChD;AAED,MAAI,SAAS,WAAW,YAAY,IAAI,KAAK,QAC3C,QAAO,MAAM,mCAAmC,cAC9C,cACA,YAAa,YAA+B,OAAO,OAAO,aAAa,CACxE;EAGH,KAAKH,SAAS;AACd,SAAO;CACR;AACF"}

package/dist/agents/annotation.cjs

@@ -13,6 +13,17 @@ const PreHookAnnotation = __langchain_langgraph.Annotation.Root({
 		default: () => []
 	})
 });
+const createAgentBaseAnnotation = () => __langchain_langgraph.Annotation.Root({ messages: (0, __langchain_langgraph.Annotation)({
+	reducer: __langchain_langgraph.messagesStateReducer,
+	default: () => []
+}) });
+const createAgentAnnotation = () => __langchain_langgraph.Annotation.Root({
+	messages: (0, __langchain_langgraph.Annotation)({
+		reducer: __langchain_langgraph.messagesStateReducer,
+		default: () => []
+	}),
+	structuredResponse: (0, __langchain_langgraph.Annotation)()
+});
 function createAgentAnnotationConditional(hasStructuredResponse = true) {
 	const baseAnnotation = { messages: (0, __langchain_langgraph.Annotation)({
 		reducer: __langchain_langgraph.messagesStateReducer,
@@ -144,6 +155,9 @@ function createAnnotationForZodType(zodType) {
 }
 
 //#endregion
+exports.PreHookAnnotation = PreHookAnnotation;
+exports.createAgentAnnotation = createAgentAnnotation;
 exports.createAgentAnnotationConditional = createAgentAnnotationConditional;
+exports.createAgentBaseAnnotation = createAgentBaseAnnotation;
 exports.enhanceStateSchemaWithMessageReducer = enhanceStateSchemaWithMessageReducer;
 //# sourceMappingURL=annotation.cjs.map

package/dist/agents/annotation.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"annotation.cjs","names":["PreHookAnnotation: AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>","Annotation","messagesStateReducer","stateSchema: AnyAnnotationRoot | InteropZodObject","annotationFields: Record<string, any>","_: unknown","update: unknown","current: unknown[]","update: unknown[]","config: (typeof ZOD_TYPE_CONFIGS)[keyof typeof ZOD_TYPE_CONFIGS]","defaultValueFn: () => unknown","zodType: any","config"],"sources":["../../src/agents/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 { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport {\n  isInteropZodSchema,\n  getInteropZodObjectShape,\n  getInteropZodDefaultGetter,\n  type InteropZodObject,\n} from \"@langchain/core/utils/types\";\n\n/**\n * Special type to indicate that no response format is provided.\n * When this type is used, the structuredResponse property should not be present in the result.\n */\nexport type ResponseFormatUndefined = {\n  __responseFormatUndefined: true;\n};\n\nexport const PreHookAnnotation: AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> = Annotation.Root({\n  llmInputMessages: Annotation<BaseMessage[], Messages>({\n    reducer: (_, update) => messagesStateReducer([], update),\n    default: () => [],\n  }),\n  messages: Annotation<BaseMessage[], Messages>({\n    reducer: messagesStateReducer,\n    default: () => [],\n  }),\n});\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n\n// Base annotation without structuredResponse (for ResponseFormatUndefined)\nexport const createAgentBaseAnnotation = () =>\n  Annotation.Root({\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  });\n\n// Full annotation with structuredResponse (for regular cases)\nexport const createAgentAnnotation = <\n  T extends Record<string, any> = Record<string, any>\n>(): AnnotationRoot<{\n  structuredResponse: LastValue<T>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> =>\n  Annotation.Root({\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n    structuredResponse: Annotation<T>(),\n  });\n\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\nexport function createAgentAnnotationConditional<\n  T extends Record<string, any> | ResponseFormatUndefined\n>(\n  hasStructuredResponse = true\n): AnnotationRoot<{\n  structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> {\n  const baseAnnotation = {\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  };\n\n  if (!hasStructuredResponse) {\n    return Annotation.Root(baseAnnotation) as AnnotationRoot<{\n      structuredResponse: LastValue<\n        T extends ResponseFormatUndefined ? never : T\n      >;\n      messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n    }>;\n  }\n\n  return Annotation.Root({\n    ...baseAnnotation,\n    structuredResponse:\n      Annotation<T extends ResponseFormatUndefined ? never : T>(),\n  });\n}\n\n// Helper type to select the right annotation based on the response format type\nexport type ReactAgentAnnotation<\n  T extends Record<string, any> | ResponseFormatUndefined\n> = T extends ResponseFormatUndefined\n  ? ReturnType<typeof createAgentBaseAnnotation>\n  : T extends Record<string, any>\n  ? ReturnType<typeof createAgentAnnotation<T>>\n  : never;\n\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> =\n  A extends AnyAnnotationRoot\n    ? A\n    : A extends InteropZodObject\n    ? AnnotationRoot<InteropZodToStateDefinition<A>>\n    : never;\n\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n\n/**\n * Enhances a state schema to ensure proper message handling.\n * If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer\n * and infers appropriate reducers for all other fields.\n *\n * @param stateSchema - The state schema to enhance.\n * @returns The enhanced state schema.\n */\nexport function enhanceStateSchemaWithMessageReducer(\n  stateSchema: AnyAnnotationRoot | InteropZodObject\n) {\n  /**\n   * If it's already an annotation, return as-is\n   */\n  if (typeof stateSchema === \"object\" && \"State\" in stateSchema) {\n    return stateSchema;\n  }\n\n  /**\n   * If it's a Zod schema, create annotations for all fields\n   */\n  if (isInteropZodSchema(stateSchema)) {\n    const shape = getInteropZodObjectShape(stateSchema);\n    const annotationFields: Record<string, any> = {};\n\n    /**\n     * Process each field in the Zod schema\n     */\n    for (const [key, zodType] of Object.entries(shape)) {\n      annotationFields[key] =\n        key === \"messages\"\n          ? /**\n             * Special handling for messages field - always use messagesStateReducer\n             */\n            Annotation<BaseMessage[]>({\n              reducer: messagesStateReducer,\n              default: () => [],\n            })\n          : /**\n             * For other fields, create appropriate annotations based on type\n             */\n            createAnnotationForZodType(zodType);\n    }\n\n    /**\n     * Ensure messages field exists even if not in the Zod schema\n     */\n    if (!annotationFields.messages) {\n      annotationFields.messages = Annotation<BaseMessage[]>({\n        reducer: messagesStateReducer,\n        default: () => [],\n      });\n    }\n\n    return Annotation.Root(annotationFields);\n  }\n\n  /**\n   * Fallback: create a base annotation with message reducer only\n   */\n  return Annotation.Root({\n    messages: Annotation<BaseMessage[]>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  });\n}\n\n/**\n * Type configuration for different Zod types\n */\nconst ZOD_TYPE_CONFIGS = {\n  ZodString: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: \"\",\n  },\n  ZodNumber: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: 0,\n  },\n  ZodBoolean: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: false,\n  },\n  ZodArray: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: [] as unknown[],\n  },\n  ZodRecord: {\n    reducer: (current: unknown[], update: unknown[]) => ({\n      ...current,\n      ...update,\n    }),\n    fallbackDefault: {} as Record<string, unknown>,\n  },\n  ZodObject: {\n    reducer: (current: unknown[], update: unknown[]) => ({\n      ...current,\n      ...update,\n    }),\n    fallbackDefault: {} as Record<string, unknown>,\n  },\n} as const;\n\n/**\n * Creates an annotation based on type configuration and default value\n */\nfunction createAnnotationFromConfig(\n  config: (typeof ZOD_TYPE_CONFIGS)[keyof typeof ZOD_TYPE_CONFIGS],\n  defaultValueFn: () => unknown\n) {\n  return Annotation<any>({\n    reducer: config.reducer,\n    default: defaultValueFn,\n  });\n}\n\n/**\n * Creates an appropriate annotation for a given Zod type.\n */\nfunction createAnnotationForZodType(zodType: any): any {\n  const typeName = zodType._def?.typeName;\n  const defaultGetter = getInteropZodDefaultGetter(zodType);\n  const isOptional = typeName === \"ZodOptional\";\n\n  /**\n   * Handle Zod wrapper first\n   */\n  if (\n    typeName === \"ZodDefault\" ||\n    typeName === \"ZodOptional\" ||\n    typeName === \"ZodNullable\"\n  ) {\n    const innerTypeName = zodType._def.innerType._def?.typeName;\n    const config =\n      ZOD_TYPE_CONFIGS[innerTypeName as keyof typeof ZOD_TYPE_CONFIGS];\n    return config\n      ? createAnnotationFromConfig(\n          config,\n          () =>\n            defaultGetter?.() ||\n            (isOptional ? undefined : config?.fallbackDefault)\n        )\n      : Annotation<unknown>({\n          reducer: (_: unknown, update: unknown) => update,\n          default: () => defaultGetter?.(),\n        });\n  }\n\n  /**\n   * Handle regular types\n   */\n  const config = ZOD_TYPE_CONFIGS[typeName as keyof typeof ZOD_TYPE_CONFIGS];\n  if (config) {\n    return createAnnotationFromConfig(\n      config,\n      () => defaultGetter?.() || config?.fallbackDefault\n    );\n  }\n\n  /**\n   * Fallback for unknown types\n   */\n  return Annotation<unknown>({\n    reducer: (_: unknown, update: unknown) => update,\n    default: () => defaultGetter?.(),\n  });\n}\n"],"mappings":";;;;;AA0BA,MAAaA,oBAGRC,iCAAW,KAAK;CACnB,wDAAsD;EACpD,SAAS,CAAC,GAAG,2DAAgC,CAAE,GAAE,OAAO;EACxD,SAAS,MAAM,CAAE;CAClB,EAAC;CACF,gDAA8C;EAC5C,SAASC;EACT,SAAS,MAAM,CAAE;CAClB,EAAC;AACH,EAAC;AA4BF,SAAgB,iCAGd,wBAAwB,MAIvB;CACD,MAAM,iBAAiB,EACrB,gDAA8C;EAC5C,SAASA;EACT,SAAS,MAAM,CAAE;CAClB,EAAC,CACH;AAED,KAAI,CAAC,sBACH,QAAOD,iCAAW,KAAK,eAAe;AAQxC,QAAOA,iCAAW,KAAK;EACrB,GAAG;EACH,2DAC6D;CAC9D,EAAC;AACH;;;;;;;;;AA4BD,SAAgB,qCACdE,aACA;;;;AAIA,KAAI,OAAO,gBAAgB,YAAY,WAAW,YAChD,QAAO;;;;AAMT,0DAAuB,YAAY,EAAE;EACnC,MAAM,mEAAiC,YAAY;EACnD,MAAMC,mBAAwC,CAAE;;;;AAKhD,OAAK,MAAM,CAAC,KAAK,QAAQ,IAAI,OAAO,QAAQ,MAAM,EAChD,iBAAiB,OACf,QAAQ,mDAIsB;GACxB,SAASF;GACT,SAAS,MAAM,CAAE;EAClB,EAAC,GAIF,2BAA2B,QAAQ;;;;AAM3C,MAAI,CAAC,iBAAiB,UACpB,iBAAiB,iDAAqC;GACpD,SAASA;GACT,SAAS,MAAM,CAAE;EAClB,EAAC;AAGJ,SAAOD,iCAAW,KAAK,iBAAiB;CACzC;;;;AAKD,QAAOA,iCAAW,KAAK,EACrB,gDAAoC;EAClC,SAASC;EACT,SAAS,MAAM,CAAE;CAClB,EAAC,CACH,EAAC;AACH;;;;AAKD,MAAM,mBAAmB;CACvB,WAAW;EACT,SAAS,CAACG,GAAYC,WAAoB;EAC1C,iBAAiB;CAClB;CACD,WAAW;EACT,SAAS,CAACD,GAAYC,WAAoB;EAC1C,iBAAiB;CAClB;CACD,YAAY;EACV,SAAS,CAACD,GAAYC,WAAoB;EAC1C,iBAAiB;CAClB;CACD,UAAU;EACR,SAAS,CAACD,GAAYC,WAAoB;EAC1C,iBAAiB,CAAE;CACpB;CACD,WAAW;EACT,SAAS,CAACC,SAAoBC,YAAuB;GACnD,GAAG;GACH,GAAG;EACJ;EACD,iBAAiB,CAAE;CACpB;CACD,WAAW;EACT,SAAS,CAACD,SAAoBC,YAAuB;GACnD,GAAG;GACH,GAAG;EACJ;EACD,iBAAiB,CAAE;CACpB;AACF;;;;AAKD,SAAS,2BACPC,QACAC,gBACA;AACA,8CAAuB;EACrB,SAAS,OAAO;EAChB,SAAS;CACV,EAAC;AACH;;;;AAKD,SAAS,2BAA2BC,SAAmB;CACrD,MAAM,WAAW,QAAQ,MAAM;CAC/B,MAAM,6EAA2C,QAAQ;CACzD,MAAM,aAAa,aAAa;;;;AAKhC,KACE,aAAa,gBACb,aAAa,iBACb,aAAa,eACb;EACA,MAAM,gBAAgB,QAAQ,KAAK,UAAU,MAAM;EACnD,MAAMC,WACJ,iBAAiB;AACnB,SAAOA,WACH,2BACEA,UACA,MACE,iBAAiB,KAChB,aAAa,SAAYA,UAAQ,iBACrC,yCACmB;GAClB,SAAS,CAACP,GAAYC,WAAoB;GAC1C,SAAS,MAAM,iBAAiB;EACjC,EAAC;CACP;;;;CAKD,MAAM,SAAS,iBAAiB;AAChC,KAAI,OACF,QAAO,2BACL,QACA,MAAM,iBAAiB,IAAI,QAAQ,gBACpC;;;;AAMH,8CAA2B;EACzB,SAAS,CAACD,GAAYC,WAAoB;EAC1C,SAAS,MAAM,iBAAiB;CACjC,EAAC;AACH"}
+{"version":3,"file":"annotation.cjs","names":["PreHookAnnotation: AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>","Annotation","messagesStateReducer","stateSchema: AnyAnnotationRoot | InteropZodObject","annotationFields: Record<string, any>","_: unknown","update: unknown","current: unknown[]","update: unknown[]","config: (typeof ZOD_TYPE_CONFIGS)[keyof typeof ZOD_TYPE_CONFIGS]","defaultValueFn: () => unknown","zodType: any","config"],"sources":["../../src/agents/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 { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport {\n  isInteropZodSchema,\n  getInteropZodObjectShape,\n  getInteropZodDefaultGetter,\n  type InteropZodObject,\n} from \"@langchain/core/utils/types\";\n\n/**\n * Special type to indicate that no response format is provided.\n * When this type is used, the structuredResponse property should not be present in the result.\n */\nexport type ResponseFormatUndefined = {\n  __responseFormatUndefined: true;\n};\n\nexport const PreHookAnnotation: AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> = Annotation.Root({\n  llmInputMessages: Annotation<BaseMessage[], Messages>({\n    reducer: (_, update) => messagesStateReducer([], update),\n    default: () => [],\n  }),\n  messages: Annotation<BaseMessage[], Messages>({\n    reducer: messagesStateReducer,\n    default: () => [],\n  }),\n});\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n\n// Base annotation without structuredResponse (for ResponseFormatUndefined)\nexport const createAgentBaseAnnotation = () =>\n  Annotation.Root({\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  });\n\n// Full annotation with structuredResponse (for regular cases)\nexport const createAgentAnnotation = <\n  T extends Record<string, any> = Record<string, any>\n>(): AnnotationRoot<{\n  structuredResponse: LastValue<T>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> =>\n  Annotation.Root({\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n    structuredResponse: Annotation<T>(),\n  });\n\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\nexport function createAgentAnnotationConditional<\n  T extends Record<string, any> | ResponseFormatUndefined\n>(\n  hasStructuredResponse = true\n): AnnotationRoot<{\n  structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> {\n  const baseAnnotation = {\n    messages: Annotation<BaseMessage[], Messages>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  };\n\n  if (!hasStructuredResponse) {\n    return Annotation.Root(baseAnnotation) as AnnotationRoot<{\n      structuredResponse: LastValue<\n        T extends ResponseFormatUndefined ? never : T\n      >;\n      messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n    }>;\n  }\n\n  return Annotation.Root({\n    ...baseAnnotation,\n    structuredResponse:\n      Annotation<T extends ResponseFormatUndefined ? never : T>(),\n  });\n}\n\n// Helper type to select the right annotation based on the response format type\nexport type ReactAgentAnnotation<\n  T extends Record<string, any> | ResponseFormatUndefined\n> = T extends ResponseFormatUndefined\n  ? ReturnType<typeof createAgentBaseAnnotation>\n  : T extends Record<string, any>\n  ? ReturnType<typeof createAgentAnnotation<T>>\n  : never;\n\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> =\n  A extends AnyAnnotationRoot\n    ? A\n    : A extends InteropZodObject\n    ? AnnotationRoot<InteropZodToStateDefinition<A>>\n    : never;\n\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n\n/**\n * Enhances a state schema to ensure proper message handling.\n * If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer\n * and infers appropriate reducers for all other fields.\n *\n * @param stateSchema - The state schema to enhance.\n * @returns The enhanced state schema.\n */\nexport function enhanceStateSchemaWithMessageReducer(\n  stateSchema: AnyAnnotationRoot | InteropZodObject\n) {\n  /**\n   * If it's already an annotation, return as-is\n   */\n  if (typeof stateSchema === \"object\" && \"State\" in stateSchema) {\n    return stateSchema;\n  }\n\n  /**\n   * If it's a Zod schema, create annotations for all fields\n   */\n  if (isInteropZodSchema(stateSchema)) {\n    const shape = getInteropZodObjectShape(stateSchema);\n    const annotationFields: Record<string, any> = {};\n\n    /**\n     * Process each field in the Zod schema\n     */\n    for (const [key, zodType] of Object.entries(shape)) {\n      annotationFields[key] =\n        key === \"messages\"\n          ? /**\n             * Special handling for messages field - always use messagesStateReducer\n             */\n            Annotation<BaseMessage[]>({\n              reducer: messagesStateReducer,\n              default: () => [],\n            })\n          : /**\n             * For other fields, create appropriate annotations based on type\n             */\n            createAnnotationForZodType(zodType);\n    }\n\n    /**\n     * Ensure messages field exists even if not in the Zod schema\n     */\n    if (!annotationFields.messages) {\n      annotationFields.messages = Annotation<BaseMessage[]>({\n        reducer: messagesStateReducer,\n        default: () => [],\n      });\n    }\n\n    return Annotation.Root(annotationFields);\n  }\n\n  /**\n   * Fallback: create a base annotation with message reducer only\n   */\n  return Annotation.Root({\n    messages: Annotation<BaseMessage[]>({\n      reducer: messagesStateReducer,\n      default: () => [],\n    }),\n  });\n}\n\n/**\n * Type configuration for different Zod types\n */\nconst ZOD_TYPE_CONFIGS = {\n  ZodString: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: \"\",\n  },\n  ZodNumber: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: 0,\n  },\n  ZodBoolean: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: false,\n  },\n  ZodArray: {\n    reducer: (_: unknown, update: unknown) => update,\n    fallbackDefault: [] as unknown[],\n  },\n  ZodRecord: {\n    reducer: (current: unknown[], update: unknown[]) => ({\n      ...current,\n      ...update,\n    }),\n    fallbackDefault: {} as Record<string, unknown>,\n  },\n  ZodObject: {\n    reducer: (current: unknown[], update: unknown[]) => ({\n      ...current,\n      ...update,\n    }),\n    fallbackDefault: {} as Record<string, unknown>,\n  },\n} as const;\n\n/**\n * Creates an annotation based on type configuration and default value\n */\nfunction createAnnotationFromConfig(\n  config: (typeof ZOD_TYPE_CONFIGS)[keyof typeof ZOD_TYPE_CONFIGS],\n  defaultValueFn: () => unknown\n) {\n  return Annotation<any>({\n    reducer: config.reducer,\n    default: defaultValueFn,\n  });\n}\n\n/**\n * Creates an appropriate annotation for a given Zod type.\n */\nfunction createAnnotationForZodType(zodType: any): any {\n  const typeName = zodType._def?.typeName;\n  const defaultGetter = getInteropZodDefaultGetter(zodType);\n  const isOptional = typeName === \"ZodOptional\";\n\n  /**\n   * Handle Zod wrapper first\n   */\n  if (\n    typeName === \"ZodDefault\" ||\n    typeName === \"ZodOptional\" ||\n    typeName === \"ZodNullable\"\n  ) {\n    const innerTypeName = zodType._def.innerType._def?.typeName;\n    const config =\n      ZOD_TYPE_CONFIGS[innerTypeName as keyof typeof ZOD_TYPE_CONFIGS];\n    return config\n      ? createAnnotationFromConfig(\n          config,\n          () =>\n            defaultGetter?.() ||\n            (isOptional ? undefined : config?.fallbackDefault)\n        )\n      : Annotation<unknown>({\n          reducer: (_: unknown, update: unknown) => update,\n          default: () => defaultGetter?.(),\n        });\n  }\n\n  /**\n   * Handle regular types\n   */\n  const config = ZOD_TYPE_CONFIGS[typeName as keyof typeof ZOD_TYPE_CONFIGS];\n  if (config) {\n    return createAnnotationFromConfig(\n      config,\n      () => defaultGetter?.() || config?.fallbackDefault\n    );\n  }\n\n  /**\n   * Fallback for unknown types\n   */\n  return Annotation<unknown>({\n    reducer: (_: unknown, update: unknown) => update,\n    default: () => defaultGetter?.(),\n  });\n}\n"],"mappings":";;;;;AA0BA,MAAaA,oBAGRC,iCAAW,KAAK;CACnB,wDAAsD;EACpD,SAAS,CAAC,GAAG,2DAAgC,CAAE,GAAE,OAAO;EACxD,SAAS,MAAM,CAAE;CAClB,EAAC;CACF,gDAA8C;EAC5C,SAASC;EACT,SAAS,MAAM,CAAE;CAClB,EAAC;AACH,EAAC;AAIF,MAAa,4BAA4B,MACvCD,iCAAW,KAAK,EACd,gDAA8C;CAC5C,SAASC;CACT,SAAS,MAAM,CAAE;AAClB,EAAC,CACH,EAAC;AAGJ,MAAa,wBAAwB,MAMnCD,iCAAW,KAAK;CACd,gDAA8C;EAC5C,SAASC;EACT,SAAS,MAAM,CAAE;CAClB,EAAC;CACF,2DAAmC;AACpC,EAAC;AAGJ,SAAgB,iCAGd,wBAAwB,MAIvB;CACD,MAAM,iBAAiB,EACrB,gDAA8C;EAC5C,SAASA;EACT,SAAS,MAAM,CAAE;CAClB,EAAC,CACH;AAED,KAAI,CAAC,sBACH,QAAOD,iCAAW,KAAK,eAAe;AAQxC,QAAOA,iCAAW,KAAK;EACrB,GAAG;EACH,2DAC6D;CAC9D,EAAC;AACH;;;;;;;;;AA4BD,SAAgB,qCACdE,aACA;;;;AAIA,KAAI,OAAO,gBAAgB,YAAY,WAAW,YAChD,QAAO;;;;AAMT,0DAAuB,YAAY,EAAE;EACnC,MAAM,mEAAiC,YAAY;EACnD,MAAMC,mBAAwC,CAAE;;;;AAKhD,OAAK,MAAM,CAAC,KAAK,QAAQ,IAAI,OAAO,QAAQ,MAAM,EAChD,iBAAiB,OACf,QAAQ,mDAIsB;GACxB,SAASF;GACT,SAAS,MAAM,CAAE;EAClB,EAAC,GAIF,2BAA2B,QAAQ;;;;AAM3C,MAAI,CAAC,iBAAiB,UACpB,iBAAiB,iDAAqC;GACpD,SAASA;GACT,SAAS,MAAM,CAAE;EAClB,EAAC;AAGJ,SAAOD,iCAAW,KAAK,iBAAiB;CACzC;;;;AAKD,QAAOA,iCAAW,KAAK,EACrB,gDAAoC;EAClC,SAASC;EACT,SAAS,MAAM,CAAE;CAClB,EAAC,CACH,EAAC;AACH;;;;AAKD,MAAM,mBAAmB;CACvB,WAAW;EACT,SAAS,CAACG,GAAYC,WAAoB;EAC1C,iBAAiB;CAClB;CACD,WAAW;EACT,SAAS,CAACD,GAAYC,WAAoB;EAC1C,iBAAiB;CAClB;CACD,YAAY;EACV,SAAS,CAACD,GAAYC,WAAoB;EAC1C,iBAAiB;CAClB;CACD,UAAU;EACR,SAAS,CAACD,GAAYC,WAAoB;EAC1C,iBAAiB,CAAE;CACpB;CACD,WAAW;EACT,SAAS,CAACC,SAAoBC,YAAuB;GACnD,GAAG;GACH,GAAG;EACJ;EACD,iBAAiB,CAAE;CACpB;CACD,WAAW;EACT,SAAS,CAACD,SAAoBC,YAAuB;GACnD,GAAG;GACH,GAAG;EACJ;EACD,iBAAiB,CAAE;CACpB;AACF;;;;AAKD,SAAS,2BACPC,QACAC,gBACA;AACA,8CAAuB;EACrB,SAAS,OAAO;EAChB,SAAS;CACV,EAAC;AACH;;;;AAKD,SAAS,2BAA2BC,SAAmB;CACrD,MAAM,WAAW,QAAQ,MAAM;CAC/B,MAAM,6EAA2C,QAAQ;CACzD,MAAM,aAAa,aAAa;;;;AAKhC,KACE,aAAa,gBACb,aAAa,iBACb,aAAa,eACb;EACA,MAAM,gBAAgB,QAAQ,KAAK,UAAU,MAAM;EACnD,MAAMC,WACJ,iBAAiB;AACnB,SAAOA,WACH,2BACEA,UACA,MACE,iBAAiB,KAChB,aAAa,SAAYA,UAAQ,iBACrC,yCACmB;GAClB,SAAS,CAACP,GAAYC,WAAoB;GAC1C,SAAS,MAAM,iBAAiB;EACjC,EAAC;CACP;;;;CAKD,MAAM,SAAS,iBAAiB;AAChC,KAAI,OACF,QAAO,2BACL,QACA,MAAM,iBAAiB,IAAI,QAAQ,gBACpC;;;;AAMH,8CAA2B;EACzB,SAAS,CAACD,GAAYC,WAAoB;EAC1C,SAAS,MAAM,iBAAiB;CACjC,EAAC;AACH"}

package/dist/agents/annotation.d.cts

@@ -1,7 +1,7 @@
 import { InteropZodObject } from "@langchain/core/utils/types";
-import { AnnotationRoot, BinaryOperatorAggregate, LastValue, Messages } from "@langchain/langgraph";
-import * as _langchain_core_messages23 from "@langchain/core/messages";
+import * as _langchain_core_messages0 from "@langchain/core/messages";
 import { BaseMessage } from "@langchain/core/messages";
+import { AnnotationRoot, BinaryOperatorAggregate, LastValue, Messages } from "@langchain/langgraph";
 import { InteropZodToStateDefinition } from "@langchain/langgraph/zod";
 
 //#region src/agents/annotation.d.ts
@@ -20,15 +20,18 @@ declare const PreHookAnnotation: AnnotationRoot<{
 type PreHookAnnotation = typeof PreHookAnnotation;
 // Base annotation without structuredResponse (for ResponseFormatUndefined)
 declare const createAgentBaseAnnotation: () => AnnotationRoot<{
-  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages23.MessageStructure, _langchain_core_messages23.MessageType>[], Messages>;
+  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[], Messages>;
 }>;
 // Full annotation with structuredResponse (for regular cases)
 declare const createAgentAnnotation: <T extends Record<string, any> = Record<string, any>>() => AnnotationRoot<{
   structuredResponse: LastValue<T>;
-  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages23.MessageStructure, _langchain_core_messages23.MessageType>[], Messages>;
+  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[], Messages>;
 }>;
 // Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse
-
+declare function createAgentAnnotationConditional<T extends Record<string, any> | ResponseFormatUndefined>(hasStructuredResponse?: boolean): AnnotationRoot<{
+  structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;
+  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;
+}>;
 // Helper type to select the right annotation based on the response format type
 type ReactAgentAnnotation<T extends Record<string, any> | ResponseFormatUndefined> = T extends ResponseFormatUndefined ? ReturnType<typeof createAgentBaseAnnotation> : T extends Record<string, any> ? ReturnType<typeof createAgentAnnotation<T>> : never;
 type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;
@@ -41,7 +44,9 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  * @param stateSchema - The state schema to enhance.
  * @returns The enhanced state schema.
  */
-
+declare function enhanceStateSchemaWithMessageReducer(stateSchema: AnyAnnotationRoot | InteropZodObject): AnnotationRoot<{
+  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[], BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[]>;
+}> | AnyAnnotationRoot;
 //#endregion
-export { AnyAnnotationRoot, PreHookAnnotation, ReactAgentAnnotation, ResponseFormatUndefined, ToAnnotationRoot };
+export { AnyAnnotationRoot, PreHookAnnotation, ReactAgentAnnotation, ResponseFormatUndefined, ToAnnotationRoot, createAgentAnnotation, createAgentAnnotationConditional, createAgentBaseAnnotation, enhanceStateSchemaWithMessageReducer };
 //# sourceMappingURL=annotation.d.cts.map

package/dist/agents/annotation.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"annotation.d.cts","names":["BaseMessage","Messages","AnnotationRoot","BinaryOperatorAggregate","LastValue","InteropZodToStateDefinition","InteropZodObject","ResponseFormatUndefined","PreHookAnnotation","createAgentBaseAnnotation","_langchain_core_messages23","MessageStructure","MessageType","createAgentAnnotation","Record","T","createAgentAnnotationConditional","ReactAgentAnnotation","ReturnType","ToAnnotationRoot","AnyAnnotationRoot","A","enhanceStateSchemaWithMessageReducer"],"sources":["../../src/agents/annotation.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { Messages, AnnotationRoot, type BinaryOperatorAggregate, type LastValue } from \"@langchain/langgraph\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport { type InteropZodObject } from \"@langchain/core/utils/types\";\n/**\n * Special type to indicate that no response format is provided.\n * When this type is used, the structuredResponse property should not be present in the result.\n */\nexport type ResponseFormatUndefined = {\n    __responseFormatUndefined: true;\n};\nexport declare const PreHookAnnotation: AnnotationRoot<{\n    llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n// Base annotation without structuredResponse (for ResponseFormatUndefined)\nexport declare const createAgentBaseAnnotation: () => AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Full annotation with structuredResponse (for regular cases)\nexport declare const createAgentAnnotation: <T extends Record<string, any> = Record<string, any>>() => AnnotationRoot<{\n    structuredResponse: LastValue<T>;\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\nexport declare function createAgentAnnotationConditional<T extends Record<string, any> | ResponseFormatUndefined>(hasStructuredResponse?: boolean): AnnotationRoot<{\n    structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\n// Helper type to select the right annotation based on the response format type\nexport type ReactAgentAnnotation<T extends Record<string, any> | ResponseFormatUndefined> = T extends ResponseFormatUndefined ? ReturnType<typeof createAgentBaseAnnotation> : T extends Record<string, any> ? ReturnType<typeof createAgentAnnotation<T>> : never;\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n/**\n * Enhances a state schema to ensure proper message handling.\n * If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer\n * and infers appropriate reducers for all other fields.\n *\n * @param stateSchema - The state schema to enhance.\n * @returns The enhanced state schema.\n */\nexport declare function enhanceStateSchemaWithMessageReducer(stateSchema: AnyAnnotationRoot | InteropZodObject): AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[]>;\n}> | AnyAnnotationRoot;\n"],"mappings":";;;;;;;;;;AASA;AAGA;AAGE,KANUO,uBAAAA,GAMV;EAAA,yBAF4CP,EAAAA,IAAAA;CAAW;AAAnCG,cADDK,iBACCL,EADkBD,cAClBC,CAAAA;EAAuB,gBACPH,EADhBG,uBACgBH,CADQA,WACRA,EAAAA,EADuBC,QACvBD,CAAAA;EAAW,QAAIC,EAAvCE,uBAAuCF,CAAfD,WAAeC,EAAAA,EAAAA,QAAAA,CAAAA;CAAQ,CAAA;AAFrBC,KAI5BM,iBAAAA,GAJ4BN,OAIDM,iBAJCN;AAAc;AAI1CM,cAESC,yBAFkBD,EAAAA,GAAAA,GAEeN,cAFE,CAAA;EAEnCO,QAAAA,EACPN,uBACZ,CADoCH,WACpC,CAAAU,0BAAAA,CADmFC,gBAAAA,EAAgBD,0BAAAA,CAAqCE,WAAAA,CACxI,EAAA,EADwJX,QACxJ,CAAA;CAAA,CAAA;;AADmGS,cAGhFG,qBAHqHD,EAAAA,CAAAA,UAGnFE,MAHmFF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAG7DE,MAH6DF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAAAA,GAGnCV,cAHmCU,CAAAA;EAAW,kBAA/GZ,EAIdI,SAJcJ,CAIJe,CAJIf,CAAAA;EAAW,QAAyGC,EAK5IE,uBAL4IF,CAKpHD,WALoHC,CAIzHS,0BAAAA,CACoDC,gBAAAA,EAAgBD,0BAAAA,CAAqCE,WAAAA,CALgBX,EAAAA,EAKAA,QALAA,CAAAA;CAAQ,CAAA;;;;AAIhIc,KAStBE,oBATsBF,CAAAA,UASSD,MATTC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS+BR,uBAT/BQ,CAAAA,GAS0DA,CAT1DA,SASoER,uBATpEQ,GAS8FG,UAT9FH,CAAAA,OASgHN,yBAThHM,CAAAA,GAS6IA,CAT7IA,SASuJD,MATvJC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS6KG,UAT7KH,CAAAA,OAS+LF,qBAT/LE,CASqNA,CATrNA,CAAAA,CAAAA,GAAAA,KAAAA;AAAVX,KAUZe,gBAVYf,CAAAA,UAUegB,iBAVfhB,GAUmCE,gBAVnCF,CAAAA,GAUuDiB,CAVvDjB,SAUiEgB,iBAVjEhB,GAUqFiB,CAVrFjB,GAUyFiB,CAVzFjB,SAUmGE,gBAVnGF,GAUsHF,cAVtHE,CAUqIC,2BAVrID,CAUiKiB,CAVjKjB,CAAAA,CAAAA,GAAAA,KAAAA;AAASM,KAWrBU,iBAAAA,GAAoBlB,cAVqDS,CAAAA,GAAAA,CAAAA;;;;;;AAFgC;AAUrH"}
+{"version":3,"file":"annotation.d.cts","names":["BaseMessage","Messages","AnnotationRoot","BinaryOperatorAggregate","LastValue","InteropZodToStateDefinition","InteropZodObject","ResponseFormatUndefined","PreHookAnnotation","createAgentBaseAnnotation","_langchain_core_messages0","MessageStructure","MessageType","createAgentAnnotation","Record","T","createAgentAnnotationConditional","ReactAgentAnnotation","ReturnType","ToAnnotationRoot","AnyAnnotationRoot","A","enhanceStateSchemaWithMessageReducer"],"sources":["../../src/agents/annotation.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { Messages, AnnotationRoot, type BinaryOperatorAggregate, type LastValue } from \"@langchain/langgraph\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport { type InteropZodObject } from \"@langchain/core/utils/types\";\n/**\n * Special type to indicate that no response format is provided.\n * When this type is used, the structuredResponse property should not be present in the result.\n */\nexport type ResponseFormatUndefined = {\n    __responseFormatUndefined: true;\n};\nexport declare const PreHookAnnotation: AnnotationRoot<{\n    llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n// Base annotation without structuredResponse (for ResponseFormatUndefined)\nexport declare const createAgentBaseAnnotation: () => AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Full annotation with structuredResponse (for regular cases)\nexport declare const createAgentAnnotation: <T extends Record<string, any> = Record<string, any>>() => AnnotationRoot<{\n    structuredResponse: LastValue<T>;\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\nexport declare function createAgentAnnotationConditional<T extends Record<string, any> | ResponseFormatUndefined>(hasStructuredResponse?: boolean): AnnotationRoot<{\n    structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\n// Helper type to select the right annotation based on the response format type\nexport type ReactAgentAnnotation<T extends Record<string, any> | ResponseFormatUndefined> = T extends ResponseFormatUndefined ? ReturnType<typeof createAgentBaseAnnotation> : T extends Record<string, any> ? ReturnType<typeof createAgentAnnotation<T>> : never;\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n/**\n * Enhances a state schema to ensure proper message handling.\n * If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer\n * and infers appropriate reducers for all other fields.\n *\n * @param stateSchema - The state schema to enhance.\n * @returns The enhanced state schema.\n */\nexport declare function enhanceStateSchemaWithMessageReducer(stateSchema: AnyAnnotationRoot | InteropZodObject): AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[]>;\n}> | AnyAnnotationRoot;\n"],"mappings":";;;;;;;;;;AASA;AAGA;AAGE,KANUO,uBAAAA,GAMV;EAAA,yBAF4CP,EAAAA,IAAAA;CAAW;AAAnCG,cADDK,iBACCL,EADkBD,cAClBC,CAAAA;EAAuB,gBACPH,EADhBG,uBACgBH,CADQA,WACRA,EAAAA,EADuBC,QACvBD,CAAAA;EAAW,QAAIC,EAAvCE,uBAAuCF,CAAfD,WAAeC,EAAAA,EAAAA,QAAAA,CAAAA;CAAQ,CAAA;AAFrBC,KAI5BM,iBAAAA,GAJ4BN,OAIDM,iBAJCN;AAAc;AAI1CM,cAESC,yBAFkBD,EAAAA,GAAAA,GAEeN,cAFE,CAAA;EAEnCO,QAAAA,EACPN,uBACZ,CADoCH,WACpC,CAAAU,yBAAAA,CADmFC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CACxI,EAAA,EADwJX,QACxJ,CAAA;CAAA,CAAA;;AADmGS,cAGhFG,qBAHqHD,EAAAA,CAAAA,UAGnFE,MAHmFF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAG7DE,MAH6DF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAAAA,GAGnCV,cAHmCU,CAAAA;EAAW,kBAA/GZ,EAIdI,SAJcJ,CAIJe,CAJIf,CAAAA;EAAW,QAAyGC,EAK5IE,uBAL4IF,CAKpHD,WALoHC,CAIzHS,yBAAAA,CACoDC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CALgBX,EAAAA,EAKAA,QALAA,CAAAA;CAAQ,CAAA;;AAD9F,iBAS5Ce,gCAT4C,CAAA,UASDF,MATC,CAAA,MAAA,EAAA,GAAA,CAAA,GASqBP,uBATrB,CAAA,CAAA,qBAAA,CAAA,EAAA,OAAA,CAAA,EASgFL,cAThF,CAAA;EAI/CW,kBAAAA,EAMGT,SAHtB,CAGgCW,CAHhC,SAG0CR,uBAH1C,GAAA,KAAA,GAG4EQ,CAH5E,CAAA;EAAA,QAAA,EAIYZ,uBAJZ,CAIoCH,WAJpC,EAAA,EAImDC,QAJnD,CAAA;CAAA,CAAA;;AAFgCc,KAStBE,oBATsBF,CAAAA,UASSD,MATTC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS+BR,uBAT/BQ,CAAAA,GAS0DA,CAT1DA,SASoER,uBATpEQ,GAS8FG,UAT9FH,CAAAA,OASgHN,yBAThHM,CAAAA,GAS6IA,CAT7IA,SASuJD,MATvJC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS6KG,UAT7KH,CAAAA,OAS+LF,qBAT/LE,CASqNA,CATrNA,CAAAA,CAAAA,GAAAA,KAAAA;AAAVX,KAUZe,gBAVYf,CAAAA,UAUegB,iBAVfhB,GAUmCE,gBAVnCF,CAAAA,GAUuDiB,CAVvDjB,SAUiEgB,iBAVjEhB,GAUqFiB,CAVrFjB,GAUyFiB,CAVzFjB,SAUmGE,gBAVnGF,GAUsHF,cAVtHE,CAUqIC,2BAVrID,CAUiKiB,CAVjKjB,CAAAA,CAAAA,GAAAA,KAAAA;AAASM,KAWrBU,iBAAAA,GAAoBlB,cAVqDS,CAAAA,GAAAA,CAAAA;;;;;;AAFgC;AAKrH;;AAAmEG,iBAgB3CQ,oCAAAA,CAhB2CR,WAAAA,EAgBOM,iBAhBPN,GAgB2BR,gBAhB3BQ,CAAAA,EAgB8CZ,cAhB9CY,CAAAA;EAAM,QAAgBP,EAiB3EJ,uBAjB2EI,CAiBnDP,WAjBmDO,CAgBqBG,yBAAAA,CACzBC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CAjBjDL,EAAAA,EAiBiEP,WAjBjEO,CAiBxCG,yBAAAA,CAAwJC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CAjBrKL,EAAAA,CAAAA;CAAuB,CAAA,GAkB3Ga,iBAjB6BL"}

package/dist/agents/annotation.d.ts

@@ -1,4 +1,4 @@
-import * as _langchain_core_messages23 from "@langchain/core/messages";
+import * as _langchain_core_messages0 from "@langchain/core/messages";
 import { BaseMessage } from "@langchain/core/messages";
 import { AnnotationRoot, BinaryOperatorAggregate, LastValue, Messages } from "@langchain/langgraph";
 import { InteropZodObject } from "@langchain/core/utils/types";
@@ -20,15 +20,18 @@ declare const PreHookAnnotation: AnnotationRoot<{
 type PreHookAnnotation = typeof PreHookAnnotation;
 // Base annotation without structuredResponse (for ResponseFormatUndefined)
 declare const createAgentBaseAnnotation: () => AnnotationRoot<{
-  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages23.MessageStructure, _langchain_core_messages23.MessageType>[], Messages>;
+  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[], Messages>;
 }>;
 // Full annotation with structuredResponse (for regular cases)
 declare const createAgentAnnotation: <T extends Record<string, any> = Record<string, any>>() => AnnotationRoot<{
   structuredResponse: LastValue<T>;
-  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages23.MessageStructure, _langchain_core_messages23.MessageType>[], Messages>;
+  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[], Messages>;
 }>;
 // Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse
-
+declare function createAgentAnnotationConditional<T extends Record<string, any> | ResponseFormatUndefined>(hasStructuredResponse?: boolean): AnnotationRoot<{
+  structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;
+  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;
+}>;
 // Helper type to select the right annotation based on the response format type
 type ReactAgentAnnotation<T extends Record<string, any> | ResponseFormatUndefined> = T extends ResponseFormatUndefined ? ReturnType<typeof createAgentBaseAnnotation> : T extends Record<string, any> ? ReturnType<typeof createAgentAnnotation<T>> : never;
 type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;
@@ -41,7 +44,9 @@ type AnyAnnotationRoot = AnnotationRoot<any>;
  * @param stateSchema - The state schema to enhance.
  * @returns The enhanced state schema.
  */
-
+declare function enhanceStateSchemaWithMessageReducer(stateSchema: AnyAnnotationRoot | InteropZodObject): AnnotationRoot<{
+  messages: BinaryOperatorAggregate<BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[], BaseMessage<_langchain_core_messages0.MessageStructure, _langchain_core_messages0.MessageType>[]>;
+}> | AnyAnnotationRoot;
 //#endregion
-export { AnyAnnotationRoot, PreHookAnnotation, ReactAgentAnnotation, ResponseFormatUndefined, ToAnnotationRoot };
+export { AnyAnnotationRoot, PreHookAnnotation, ReactAgentAnnotation, ResponseFormatUndefined, ToAnnotationRoot, createAgentAnnotation, createAgentAnnotationConditional, createAgentBaseAnnotation, enhanceStateSchemaWithMessageReducer };
 //# sourceMappingURL=annotation.d.ts.map

package/dist/agents/annotation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"annotation.d.ts","names":["BaseMessage","Messages","AnnotationRoot","BinaryOperatorAggregate","LastValue","InteropZodToStateDefinition","InteropZodObject","ResponseFormatUndefined","PreHookAnnotation","createAgentBaseAnnotation","_langchain_core_messages23","MessageStructure","MessageType","createAgentAnnotation","Record","T","createAgentAnnotationConditional","ReactAgentAnnotation","ReturnType","ToAnnotationRoot","AnyAnnotationRoot","A","enhanceStateSchemaWithMessageReducer"],"sources":["../../src/agents/annotation.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { Messages, AnnotationRoot, type BinaryOperatorAggregate, type LastValue } from \"@langchain/langgraph\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport { type InteropZodObject } from \"@langchain/core/utils/types\";\n/**\n * Special type to indicate that no response format is provided.\n * When this type is used, the structuredResponse property should not be present in the result.\n */\nexport type ResponseFormatUndefined = {\n    __responseFormatUndefined: true;\n};\nexport declare const PreHookAnnotation: AnnotationRoot<{\n    llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n// Base annotation without structuredResponse (for ResponseFormatUndefined)\nexport declare const createAgentBaseAnnotation: () => AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Full annotation with structuredResponse (for regular cases)\nexport declare const createAgentAnnotation: <T extends Record<string, any> = Record<string, any>>() => AnnotationRoot<{\n    structuredResponse: LastValue<T>;\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\nexport declare function createAgentAnnotationConditional<T extends Record<string, any> | ResponseFormatUndefined>(hasStructuredResponse?: boolean): AnnotationRoot<{\n    structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\n// Helper type to select the right annotation based on the response format type\nexport type ReactAgentAnnotation<T extends Record<string, any> | ResponseFormatUndefined> = T extends ResponseFormatUndefined ? ReturnType<typeof createAgentBaseAnnotation> : T extends Record<string, any> ? ReturnType<typeof createAgentAnnotation<T>> : never;\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n/**\n * Enhances a state schema to ensure proper message handling.\n * If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer\n * and infers appropriate reducers for all other fields.\n *\n * @param stateSchema - The state schema to enhance.\n * @returns The enhanced state schema.\n */\nexport declare function enhanceStateSchemaWithMessageReducer(stateSchema: AnyAnnotationRoot | InteropZodObject): AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[]>;\n}> | AnyAnnotationRoot;\n"],"mappings":";;;;;;;;;;AASA;AAGA;AAGE,KANUO,uBAAAA,GAMV;EAAA,yBAF4CP,EAAAA,IAAAA;CAAW;AAAnCG,cADDK,iBACCL,EADkBD,cAClBC,CAAAA;EAAuB,gBACPH,EADhBG,uBACgBH,CADQA,WACRA,EAAAA,EADuBC,QACvBD,CAAAA;EAAW,QAAIC,EAAvCE,uBAAuCF,CAAfD,WAAeC,EAAAA,EAAAA,QAAAA,CAAAA;CAAQ,CAAA;AAFrBC,KAI5BM,iBAAAA,GAJ4BN,OAIDM,iBAJCN;AAAc;AAI1CM,cAESC,yBAFkBD,EAAAA,GAAAA,GAEeN,cAFE,CAAA;EAEnCO,QAAAA,EACPN,uBACZ,CADoCH,WACpC,CAAAU,0BAAAA,CADmFC,gBAAAA,EAAgBD,0BAAAA,CAAqCE,WAAAA,CACxI,EAAA,EADwJX,QACxJ,CAAA;CAAA,CAAA;;AADmGS,cAGhFG,qBAHqHD,EAAAA,CAAAA,UAGnFE,MAHmFF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAG7DE,MAH6DF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAAAA,GAGnCV,cAHmCU,CAAAA;EAAW,kBAA/GZ,EAIdI,SAJcJ,CAIJe,CAJIf,CAAAA;EAAW,QAAyGC,EAK5IE,uBAL4IF,CAKpHD,WALoHC,CAIzHS,0BAAAA,CACoDC,gBAAAA,EAAgBD,0BAAAA,CAAqCE,WAAAA,CALgBX,EAAAA,EAKAA,QALAA,CAAAA;CAAQ,CAAA;;;;AAIhIc,KAStBE,oBATsBF,CAAAA,UASSD,MATTC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS+BR,uBAT/BQ,CAAAA,GAS0DA,CAT1DA,SASoER,uBATpEQ,GAS8FG,UAT9FH,CAAAA,OASgHN,yBAThHM,CAAAA,GAS6IA,CAT7IA,SASuJD,MATvJC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS6KG,UAT7KH,CAAAA,OAS+LF,qBAT/LE,CASqNA,CATrNA,CAAAA,CAAAA,GAAAA,KAAAA;AAAVX,KAUZe,gBAVYf,CAAAA,UAUegB,iBAVfhB,GAUmCE,gBAVnCF,CAAAA,GAUuDiB,CAVvDjB,SAUiEgB,iBAVjEhB,GAUqFiB,CAVrFjB,GAUyFiB,CAVzFjB,SAUmGE,gBAVnGF,GAUsHF,cAVtHE,CAUqIC,2BAVrID,CAUiKiB,CAVjKjB,CAAAA,CAAAA,GAAAA,KAAAA;AAASM,KAWrBU,iBAAAA,GAAoBlB,cAVqDS,CAAAA,GAAAA,CAAAA;;;;;;AAFgC;AAUrH"}
+{"version":3,"file":"annotation.d.ts","names":["BaseMessage","Messages","AnnotationRoot","BinaryOperatorAggregate","LastValue","InteropZodToStateDefinition","InteropZodObject","ResponseFormatUndefined","PreHookAnnotation","createAgentBaseAnnotation","_langchain_core_messages0","MessageStructure","MessageType","createAgentAnnotation","Record","T","createAgentAnnotationConditional","ReactAgentAnnotation","ReturnType","ToAnnotationRoot","AnyAnnotationRoot","A","enhanceStateSchemaWithMessageReducer"],"sources":["../../src/agents/annotation.d.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { Messages, AnnotationRoot, type BinaryOperatorAggregate, type LastValue } from \"@langchain/langgraph\";\nimport type { InteropZodToStateDefinition } from \"@langchain/langgraph/zod\";\nimport { type InteropZodObject } from \"@langchain/core/utils/types\";\n/**\n * Special type to indicate that no response format is provided.\n * When this type is used, the structuredResponse property should not be present in the result.\n */\nexport type ResponseFormatUndefined = {\n    __responseFormatUndefined: true;\n};\nexport declare const PreHookAnnotation: AnnotationRoot<{\n    llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n// Base annotation without structuredResponse (for ResponseFormatUndefined)\nexport declare const createAgentBaseAnnotation: () => AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Full annotation with structuredResponse (for regular cases)\nexport declare const createAgentAnnotation: <T extends Record<string, any> = Record<string, any>>() => AnnotationRoot<{\n    structuredResponse: LastValue<T>;\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], Messages>;\n}>;\n// Create annotation conditionally - for ResponseFormatUndefined, don't include structuredResponse\nexport declare function createAgentAnnotationConditional<T extends Record<string, any> | ResponseFormatUndefined>(hasStructuredResponse?: boolean): AnnotationRoot<{\n    structuredResponse: LastValue<T extends ResponseFormatUndefined ? never : T>;\n    messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\n// Helper type to select the right annotation based on the response format type\nexport type ReactAgentAnnotation<T extends Record<string, any> | ResponseFormatUndefined> = T extends ResponseFormatUndefined ? ReturnType<typeof createAgentBaseAnnotation> : T extends Record<string, any> ? ReturnType<typeof createAgentAnnotation<T>> : never;\nexport type ToAnnotationRoot<A extends AnyAnnotationRoot | InteropZodObject> = A extends AnyAnnotationRoot ? A : A extends InteropZodObject ? AnnotationRoot<InteropZodToStateDefinition<A>> : never;\nexport type AnyAnnotationRoot = AnnotationRoot<any>;\n/**\n * Enhances a state schema to ensure proper message handling.\n * If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer\n * and infers appropriate reducers for all other fields.\n *\n * @param stateSchema - The state schema to enhance.\n * @returns The enhanced state schema.\n */\nexport declare function enhanceStateSchemaWithMessageReducer(stateSchema: AnyAnnotationRoot | InteropZodObject): AnnotationRoot<{\n    messages: BinaryOperatorAggregate<BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[], BaseMessage<import(\"@langchain/core/messages\").MessageStructure, import(\"@langchain/core/messages\").MessageType>[]>;\n}> | AnyAnnotationRoot;\n"],"mappings":";;;;;;;;;;AASA;AAGA;AAGE,KANUO,uBAAAA,GAMV;EAAA,yBAF4CP,EAAAA,IAAAA;CAAW;AAAnCG,cADDK,iBACCL,EADkBD,cAClBC,CAAAA;EAAuB,gBACPH,EADhBG,uBACgBH,CADQA,WACRA,EAAAA,EADuBC,QACvBD,CAAAA;EAAW,QAAIC,EAAvCE,uBAAuCF,CAAfD,WAAeC,EAAAA,EAAAA,QAAAA,CAAAA;CAAQ,CAAA;AAFrBC,KAI5BM,iBAAAA,GAJ4BN,OAIDM,iBAJCN;AAAc;AAI1CM,cAESC,yBAFkBD,EAAAA,GAAAA,GAEeN,cAFE,CAAA;EAEnCO,QAAAA,EACPN,uBACZ,CADoCH,WACpC,CAAAU,yBAAAA,CADmFC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CACxI,EAAA,EADwJX,QACxJ,CAAA;CAAA,CAAA;;AADmGS,cAGhFG,qBAHqHD,EAAAA,CAAAA,UAGnFE,MAHmFF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAG7DE,MAH6DF,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,CAAAA,GAAAA,GAGnCV,cAHmCU,CAAAA;EAAW,kBAA/GZ,EAIdI,SAJcJ,CAIJe,CAJIf,CAAAA;EAAW,QAAyGC,EAK5IE,uBAL4IF,CAKpHD,WALoHC,CAIzHS,yBAAAA,CACoDC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CALgBX,EAAAA,EAKAA,QALAA,CAAAA;CAAQ,CAAA;;AAD9F,iBAS5Ce,gCAT4C,CAAA,UASDF,MATC,CAAA,MAAA,EAAA,GAAA,CAAA,GASqBP,uBATrB,CAAA,CAAA,qBAAA,CAAA,EAAA,OAAA,CAAA,EASgFL,cAThF,CAAA;EAI/CW,kBAAAA,EAMGT,SAHtB,CAGgCW,CAHhC,SAG0CR,uBAH1C,GAAA,KAAA,GAG4EQ,CAH5E,CAAA;EAAA,QAAA,EAIYZ,uBAJZ,CAIoCH,WAJpC,EAAA,EAImDC,QAJnD,CAAA;CAAA,CAAA;;AAFgCc,KAStBE,oBATsBF,CAAAA,UASSD,MATTC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS+BR,uBAT/BQ,CAAAA,GAS0DA,CAT1DA,SASoER,uBATpEQ,GAS8FG,UAT9FH,CAAAA,OASgHN,yBAThHM,CAAAA,GAS6IA,CAT7IA,SASuJD,MATvJC,CAAAA,MAAAA,EAAAA,GAAAA,CAAAA,GAS6KG,UAT7KH,CAAAA,OAS+LF,qBAT/LE,CASqNA,CATrNA,CAAAA,CAAAA,GAAAA,KAAAA;AAAVX,KAUZe,gBAVYf,CAAAA,UAUegB,iBAVfhB,GAUmCE,gBAVnCF,CAAAA,GAUuDiB,CAVvDjB,SAUiEgB,iBAVjEhB,GAUqFiB,CAVrFjB,GAUyFiB,CAVzFjB,SAUmGE,gBAVnGF,GAUsHF,cAVtHE,CAUqIC,2BAVrID,CAUiKiB,CAVjKjB,CAAAA,CAAAA,GAAAA,KAAAA;AAASM,KAWrBU,iBAAAA,GAAoBlB,cAVqDS,CAAAA,GAAAA,CAAAA;;;;;;AAFgC;AAKrH;;AAAmEG,iBAgB3CQ,oCAAAA,CAhB2CR,WAAAA,EAgBOM,iBAhBPN,GAgB2BR,gBAhB3BQ,CAAAA,EAgB8CZ,cAhB9CY,CAAAA;EAAM,QAAgBP,EAiB3EJ,uBAjB2EI,CAiBnDP,WAjBmDO,CAgBqBG,yBAAAA,CACzBC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CAjBjDL,EAAAA,EAiBiEP,WAjBjEO,CAiBxCG,yBAAAA,CAAwJC,gBAAAA,EAAgBD,yBAAAA,CAAqCE,WAAAA,CAjBrKL,EAAAA,CAAAA;CAAuB,CAAA,GAkB3Ga,iBAjB6BL"}