langchain 1.0.0-alpha.8 → 1.0.0-alpha.9

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.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"}
+{"version":3,"file":"ReactAgent.js","names":["options: CreateAgentParams<StructuredResponseFormat, ContextSchema>","#toolBehaviorVersion","beforeAgentNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[]","beforeModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[]","afterModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[]","afterAgentNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[]","wrapModelCallHookMiddleware: [\n      AgentMiddleware,\n      /**\n       * ToDo: better type to get the state of middleware\n       */\n      () => any\n    ][]","#agentNode","beforeAgentNode: BeforeAgentNode | undefined","beforeModelNode: BeforeModelNode | undefined","afterModelNode: AfterModelNode | undefined","afterAgentNode: AfterAgentNode | undefined","entryNode: string","#createBeforeAgentRouter","#createBeforeModelRouter","#getModelPaths","#createModelRouter","#createAfterModelSequenceRouter","#createAfterModelRouter","#createToolsRouter","#graph","toolClasses: (ClientTool | ServerTool)[]","includeModelRequest: boolean","paths: (\"tools\" | \"model_request\" | typeof END)[]","shouldReturnDirect: Set<string>","exitNode: string | typeof END","state: any","state: BuiltInState","allowJump: boolean","state: Omit<BuiltInState, \"jumpTo\"> & { jumpTo?: JumpTo }","allowed: string[]","nextDefault: string","#initializeMiddlewareStates","state: InvokeStateParameter<TMiddleware>","#populatePrivateState","config?: RunnableConfig","config?: InvokeConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >","config?: StreamConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >","params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }","config?: StreamConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    > & { version?: \"v1\" | \"v2\" }","streamOptions?: Parameters<Runnable[\"streamEvents\"]>[2]","config: RunnableConfig","options?: GetStateOptions","options?: CheckpointListOptions","namespace?: string","recurse?: boolean","inputConfig: LangGraphRunnableConfig","values: Record<string, unknown> | unknown","asNode?: string"],"sources":["../../src/agents/ReactAgent.ts"],"sourcesContent":["/* eslint-disable no-instanceof/no-instanceof */\n/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { InteropZodObject } from \"@langchain/core/utils/types\";\n\nimport {\n  AnnotationRoot,\n  StateGraph,\n  END,\n  START,\n  Send,\n  Command,\n  CompiledStateGraph,\n  type GetStateOptions,\n  type LangGraphRunnableConfig,\n} from \"@langchain/langgraph\";\nimport type { CheckpointListOptions } from \"@langchain/langgraph-checkpoint\";\nimport { ToolMessage, AIMessage } from \"@langchain/core/messages\";\nimport { IterableReadableStream } from \"@langchain/core/utils/stream\";\nimport type { Runnable, RunnableConfig } from \"@langchain/core/runnables\";\nimport type { StreamEvent } from \"@langchain/core/tracers/log_stream\";\n\nimport { createAgentAnnotationConditional } from \"./annotation.js\";\nimport {\n  isClientTool,\n  validateLLMHasNoBoundTools,\n  wrapToolCall,\n} from \"./utils.js\";\n\nimport { AgentNode } from \"./nodes/AgentNode.js\";\nimport { ToolNode } from \"./nodes/ToolNode.js\";\nimport { BeforeAgentNode } from \"./nodes/BeforeAgentNode.js\";\nimport { BeforeModelNode } from \"./nodes/BeforeModelNode.js\";\nimport { AfterModelNode } from \"./nodes/AfterModelNode.js\";\nimport { AfterAgentNode } from \"./nodes/AfterAgentNode.js\";\nimport {\n  initializeMiddlewareStates,\n  parseJumpToTarget,\n} from \"./nodes/utils.js\";\n\nimport type { WithStateGraphNodes } from \"./types.js\";\nimport type { ClientTool, ServerTool } from \"./tools.js\";\n\nimport type {\n  CreateAgentParams,\n  BuiltInState,\n  JumpTo,\n  UserInput,\n} from \"./types.js\";\nimport type {\n  PrivateState,\n  InvokeConfiguration,\n  StreamConfiguration,\n} from \"./runtime.js\";\nimport type {\n  AgentMiddleware,\n  InferMiddlewareContextInputs,\n  InferMiddlewareStates,\n  InferMiddlewareInputStates,\n  InferContextInput,\n  ToAnnotationRoot,\n  AnyAnnotationRoot,\n} from \"./middleware/types.js\";\nimport { type ResponseFormatUndefined } from \"./responses.js\";\n\n// Helper type to get the state definition with middleware states\ntype MergedAgentState<\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined,\n  TMiddleware extends readonly AgentMiddleware[]\n> = (StructuredResponseFormat extends ResponseFormatUndefined\n  ? Omit<BuiltInState, \"jumpTo\">\n  : Omit<BuiltInState, \"jumpTo\"> & {\n      structuredResponse: StructuredResponseFormat;\n    }) &\n  InferMiddlewareStates<TMiddleware>;\n\ntype InvokeStateParameter<TMiddleware extends readonly AgentMiddleware[]> =\n  | (UserInput & InferMiddlewareInputStates<TMiddleware>)\n  | Command<any, any, any>\n  | null;\n\ntype AgentGraph<\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddleware extends readonly AgentMiddleware[] = []\n> = CompiledStateGraph<\n  any,\n  any,\n  any,\n  any,\n  MergedAgentState<StructuredResponseFormat, TMiddleware>,\n  ToAnnotationRoot<ContextSchema>[\"spec\"],\n  unknown\n>;\n\nexport class ReactAgent<\n  StructuredResponseFormat extends\n    | Record<string, any>\n    | ResponseFormatUndefined = Record<string, any>,\n  ContextSchema extends\n    | AnyAnnotationRoot\n    | InteropZodObject = AnyAnnotationRoot,\n  TMiddleware extends readonly AgentMiddleware[] = readonly AgentMiddleware[]\n> {\n  #graph: AgentGraph<StructuredResponseFormat, ContextSchema, TMiddleware>;\n\n  #toolBehaviorVersion: \"v1\" | \"v2\" = \"v2\";\n\n  #agentNode: AgentNode<any, AnyAnnotationRoot>;\n\n  constructor(\n    public options: CreateAgentParams<StructuredResponseFormat, ContextSchema>\n  ) {\n    this.#toolBehaviorVersion = options.version ?? this.#toolBehaviorVersion;\n\n    /**\n     * validate that model option is provided\n     */\n    if (!options.model) {\n      throw new Error(\"`model` option is required to create an agent.\");\n    }\n\n    /**\n     * Check if the LLM already has bound tools and throw if it does.\n     */\n    if (typeof options.model !== \"string\") {\n      validateLLMHasNoBoundTools(options.model);\n    }\n\n    /**\n     * define complete list of tools based on options and middleware\n     */\n    const middlewareTools = (this.options.middleware\n      ?.filter((m) => m.tools)\n      .flatMap((m) => m.tools) ?? []) as (ClientTool | ServerTool)[];\n    const toolClasses = [...(options.tools ?? []), ...middlewareTools];\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    /**\n     * Create a schema that merges agent base schema with middleware state schemas\n     * Using Zod with withLangGraph ensures LangGraph Studio gets proper metadata\n     */\n    const schema = createAgentAnnotationConditional<TMiddleware>(\n      this.options.responseFormat !== undefined,\n      this.options.middleware as TMiddleware\n    );\n\n    const workflow = new StateGraph(\n      schema as unknown as AnnotationRoot<any>,\n      this.options.contextSchema\n    );\n\n    const allNodeWorkflows = workflow as WithStateGraphNodes<\n      \"tools\" | \"model_request\" | string,\n      typeof workflow\n    >;\n\n    // Generate node names for middleware nodes that have hooks\n    const beforeAgentNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[] = [];\n    const beforeModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[] = [];\n    const afterModelNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[] = [];\n    const afterAgentNodes: {\n      index: number;\n      name: string;\n      allowed?: string[];\n    }[] = [];\n    const wrapModelCallHookMiddleware: [\n      AgentMiddleware,\n      /**\n       * ToDo: better type to get the state of middleware\n       */\n      () => any\n    ][] = [];\n\n    this.#agentNode = new AgentNode({\n      model: this.options.model,\n      systemPrompt: this.options.systemPrompt,\n      includeAgentName: this.options.includeAgentName,\n      name: this.options.name,\n      responseFormat: this.options.responseFormat,\n      middleware: this.options.middleware,\n      toolClasses,\n      shouldReturnDirect,\n      signal: this.options.signal,\n      wrapModelCallHookMiddleware,\n    });\n\n    const middlewareNames = new Set<string>();\n    const middleware = this.options.middleware ?? [];\n    for (let i = 0; i < middleware.length; i++) {\n      let beforeAgentNode: BeforeAgentNode | undefined;\n      let beforeModelNode: BeforeModelNode | undefined;\n      let afterModelNode: AfterModelNode | undefined;\n      let afterAgentNode: AfterAgentNode | undefined;\n      const m = middleware[i];\n      if (middlewareNames.has(m.name)) {\n        throw new Error(`Middleware ${m.name} is defined multiple times`);\n      }\n\n      middlewareNames.add(m.name);\n      if (m.beforeAgent) {\n        beforeAgentNode = new BeforeAgentNode(m, {\n          getPrivateState: () => this.#agentNode.getState()._privateState,\n        });\n        const name = `${m.name}.before_agent`;\n        beforeAgentNodes.push({\n          index: i,\n          name,\n          allowed: m.beforeAgentJumpTo,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          beforeAgentNode,\n          beforeAgentNode.nodeOptions\n        );\n      }\n      if (m.beforeModel) {\n        beforeModelNode = new BeforeModelNode(m, {\n          getPrivateState: () => this.#agentNode.getState()._privateState,\n        });\n        const name = `${m.name}.before_model`;\n        beforeModelNodes.push({\n          index: i,\n          name,\n          allowed: m.beforeModelJumpTo,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          beforeModelNode,\n          beforeModelNode.nodeOptions\n        );\n      }\n      if (m.afterModel) {\n        afterModelNode = new AfterModelNode(m, {\n          getPrivateState: () => this.#agentNode.getState()._privateState,\n        });\n        const name = `${m.name}.after_model`;\n        afterModelNodes.push({\n          index: i,\n          name,\n          allowed: m.afterModelJumpTo,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          afterModelNode,\n          afterModelNode.nodeOptions\n        );\n      }\n      if (m.afterAgent) {\n        afterAgentNode = new AfterAgentNode(m, {\n          getPrivateState: () => this.#agentNode.getState()._privateState,\n        });\n        const name = `${m.name}.after_agent`;\n        afterAgentNodes.push({\n          index: i,\n          name,\n          allowed: m.afterAgentJumpTo,\n        });\n        allNodeWorkflows.addNode(\n          name,\n          afterAgentNode,\n          afterAgentNode.nodeOptions\n        );\n      }\n\n      if (m.wrapModelCall) {\n        wrapModelCallHookMiddleware.push([\n          m,\n          () => ({\n            ...beforeAgentNode?.getState(),\n            ...beforeModelNode?.getState(),\n            ...afterModelNode?.getState(),\n            ...afterAgentNode?.getState(),\n          }),\n        ]);\n      }\n    }\n\n    /**\n     * Add Nodes\n     */\n    allNodeWorkflows.addNode(\n      \"model_request\",\n      this.#agentNode,\n      AgentNode.nodeOptions\n    );\n\n    /**\n     * Collect and compose wrapToolCall handlers from middleware\n     * Wrap each handler with error handling and validation\n     */\n    const wrapToolCallHandler = wrapToolCall(middleware);\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        wrapToolCall: wrapToolCallHandler,\n        getPrivateState: () => this.#agentNode.getState()._privateState,\n      });\n      allNodeWorkflows.addNode(\"tools\", toolNode);\n    }\n\n    /**\n     * Add Edges\n     */\n    // Determine the entry node (runs once at start): before_agent -> before_model -> model_request\n    let entryNode: string;\n    if (beforeAgentNodes.length > 0) {\n      entryNode = beforeAgentNodes[0].name;\n    } else if (beforeModelNodes.length > 0) {\n      entryNode = beforeModelNodes[0].name;\n    } else {\n      entryNode = \"model_request\";\n    }\n\n    // Determine the loop entry node (beginning of agent loop, excludes before_agent)\n    // This is where tools will loop back to for the next iteration\n    const loopEntryNode =\n      beforeModelNodes.length > 0 ? beforeModelNodes[0].name : \"model_request\";\n\n    // Determine the exit node (runs once at end): after_agent or END\n    const exitNode =\n      afterAgentNodes.length > 0\n        ? afterAgentNodes[afterAgentNodes.length - 1].name\n        : END;\n\n    allNodeWorkflows.addEdge(START, entryNode);\n\n    // Connect beforeAgent nodes (run once at start)\n    for (let i = 0; i < beforeAgentNodes.length; i++) {\n      const node = beforeAgentNodes[i];\n      const current = node.name;\n      const isLast = i === beforeAgentNodes.length - 1;\n      const nextDefault = isLast ? loopEntryNode : beforeAgentNodes[i + 1].name;\n\n      if (node.allowed && node.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = node.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n        // Replace END with exitNode (which could be an afterAgent node)\n        const destinations = Array.from(\n          new Set([\n            nextDefault,\n            ...allowedMapped.map((dest) => (dest === END ? exitNode : dest)),\n          ])\n        ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n        allNodeWorkflows.addConditionalEdges(\n          current,\n          this.#createBeforeAgentRouter(\n            toolClasses.filter(isClientTool),\n            nextDefault,\n            exitNode\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(current, nextDefault);\n      }\n    }\n\n    // Connect beforeModel nodes; add conditional routing ONLY if allowed jumps are specified\n    for (let i = 0; i < beforeModelNodes.length; i++) {\n      const node = beforeModelNodes[i];\n      const current = node.name;\n      const isLast = i === beforeModelNodes.length - 1;\n      const nextDefault = isLast\n        ? \"model_request\"\n        : beforeModelNodes[i + 1].name;\n\n      if (node.allowed && node.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = node.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n        const destinations = Array.from(\n          new Set([nextDefault, ...allowedMapped])\n        ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n        allNodeWorkflows.addConditionalEdges(\n          current,\n          this.#createBeforeModelRouter(\n            toolClasses.filter(isClientTool),\n            nextDefault\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(current, nextDefault);\n      }\n    }\n\n    // Connect agent to last afterModel node (for reverse order execution)\n    const lastAfterModelNode = afterModelNodes.at(-1);\n    if (afterModelNodes.length > 0 && lastAfterModelNode) {\n      allNodeWorkflows.addEdge(\"model_request\", lastAfterModelNode.name);\n    } else {\n      // If no afterModel nodes, connect model_request directly to model paths\n      const modelPaths = this.#getModelPaths(toolClasses.filter(isClientTool));\n      // Replace END with exitNode in destinations, since exitNode might be an afterAgent node\n      const destinations = modelPaths.map((p) =>\n        p === END ? exitNode : p\n      ) as (\"tools\" | \"model_request\" | typeof END)[];\n      if (destinations.length === 1) {\n        allNodeWorkflows.addEdge(\"model_request\", destinations[0]);\n      } else {\n        allNodeWorkflows.addConditionalEdges(\n          \"model_request\",\n          this.#createModelRouter(exitNode),\n          destinations\n        );\n      }\n    }\n\n    // Connect afterModel nodes in reverse sequence; add conditional routing ONLY if allowed jumps are specified per node\n    for (let i = afterModelNodes.length - 1; i > 0; i--) {\n      const node = afterModelNodes[i];\n      const current = node.name;\n      const nextDefault = afterModelNodes[i - 1].name;\n\n      if (node.allowed && node.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = node.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n        const destinations = Array.from(\n          new Set([nextDefault, ...allowedMapped])\n        ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n        allNodeWorkflows.addConditionalEdges(\n          current,\n          this.#createAfterModelSequenceRouter(\n            toolClasses.filter(isClientTool),\n            node.allowed,\n            nextDefault\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(current, nextDefault);\n      }\n    }\n\n    // Connect first afterModel node (last to execute) to model paths with jumpTo support\n    if (afterModelNodes.length > 0) {\n      const firstAfterModel = afterModelNodes[0];\n      const firstAfterModelNode = firstAfterModel.name;\n\n      // Include exitNode in the paths since afterModel should be able to route to after_agent or END\n      const modelPaths = this.#getModelPaths(\n        toolClasses.filter(isClientTool),\n        true\n      ).filter(\n        (p) => p !== \"tools\" || toolClasses.filter(isClientTool).length > 0\n      );\n\n      const allowJump = Boolean(\n        firstAfterModel.allowed && firstAfterModel.allowed.length > 0\n      );\n\n      // Replace END with exitNode in destinations, since exitNode might be an afterAgent node\n      const destinations = modelPaths.map((p) =>\n        p === END ? exitNode : p\n      ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n      allNodeWorkflows.addConditionalEdges(\n        firstAfterModelNode,\n        this.#createAfterModelRouter(\n          toolClasses.filter(isClientTool),\n          allowJump,\n          exitNode\n        ),\n        destinations\n      );\n    }\n\n    // Connect afterAgent nodes (run once at end, in reverse order like afterModel)\n    for (let i = afterAgentNodes.length - 1; i > 0; i--) {\n      const node = afterAgentNodes[i];\n      const current = node.name;\n      const nextDefault = afterAgentNodes[i - 1].name;\n\n      if (node.allowed && node.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = node.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n        const destinations = Array.from(\n          new Set([nextDefault, ...allowedMapped])\n        ) as (\"tools\" | \"model_request\" | typeof END)[];\n\n        allNodeWorkflows.addConditionalEdges(\n          current,\n          this.#createAfterModelSequenceRouter(\n            toolClasses.filter(isClientTool),\n            node.allowed,\n            nextDefault\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(current, nextDefault);\n      }\n    }\n\n    // Connect the first afterAgent node (last to execute) to END\n    if (afterAgentNodes.length > 0) {\n      const firstAfterAgent = afterAgentNodes[0];\n      const firstAfterAgentNode = firstAfterAgent.name;\n\n      if (firstAfterAgent.allowed && firstAfterAgent.allowed.length > 0) {\n        const hasTools = toolClasses.filter(isClientTool).length > 0;\n        const allowedMapped = firstAfterAgent.allowed\n          .map((t) => parseJumpToTarget(t))\n          .filter((dest) => dest !== \"tools\" || hasTools);\n\n        /**\n         * For after_agent, only use explicitly allowed destinations (don't add loopEntryNode)\n         * The default destination (when no jump occurs) should be END\n         */\n        const destinations = Array.from(new Set([END, ...allowedMapped])) as (\n          | \"tools\"\n          | \"model_request\"\n          | typeof END\n        )[];\n\n        allNodeWorkflows.addConditionalEdges(\n          firstAfterAgentNode,\n          this.#createAfterModelSequenceRouter(\n            toolClasses.filter(isClientTool),\n            firstAfterAgent.allowed,\n            END as string\n          ),\n          destinations\n        );\n      } else {\n        allNodeWorkflows.addEdge(firstAfterAgentNode, END);\n      }\n    }\n\n    /**\n     * add edges for tools node\n     */\n    if (toolClasses.filter(isClientTool).length > 0) {\n      // Tools should return to loop entry node (not including before_agent)\n      const toolReturnTarget = loopEntryNode;\n\n      if (shouldReturnDirect.size > 0) {\n        allNodeWorkflows.addConditionalEdges(\n          \"tools\",\n          this.#createToolsRouter(shouldReturnDirect, exitNode),\n          [toolReturnTarget, exitNode as string]\n        );\n      } else {\n        allNodeWorkflows.addEdge(\"tools\", toolReturnTarget);\n      }\n    }\n\n    /**\n     * compile the graph\n     */\n    this.#graph = allNodeWorkflows.compile({\n      checkpointer: this.options.checkpointer,\n      store: this.options.store,\n      name: this.options.name,\n      description: this.options.description,\n    }) as AgentGraph<StructuredResponseFormat, ContextSchema, TMiddleware>;\n  }\n\n  /**\n   * Get the compiled {@link https://docs.langchain.com/oss/javascript/langgraph/use-graph-api | StateGraph}.\n   */\n  get graph(): AgentGraph<\n    StructuredResponseFormat,\n    ContextSchema,\n    TMiddleware\n  > {\n    return this.#graph;\n  }\n\n  /**\n   * Get possible edge destinations from model node.\n   * @param toolClasses names of tools to call\n   * @param includeModelRequest whether to include \"model_request\" as a valid path (for jumpTo routing)\n   * @returns list of possible edge destinations\n   */\n  #getModelPaths(\n    toolClasses: (ClientTool | ServerTool)[],\n    includeModelRequest: boolean = false\n  ): (\"tools\" | \"model_request\" | typeof END)[] {\n    const paths: (\"tools\" | \"model_request\" | typeof END)[] = [];\n    if (toolClasses.length > 0) {\n      paths.push(\"tools\");\n    }\n\n    if (includeModelRequest) {\n      paths.push(\"model_request\");\n    }\n\n    paths.push(END);\n\n    return paths;\n  }\n\n  /**\n   * Create routing function for tools node conditional edges.\n   */\n  #createToolsRouter(\n    shouldReturnDirect: Set<string>,\n    exitNode: string | typeof END\n  ) {\n    /**\n     * ToDo: fix type\n     */\n    return (state: any) => {\n      const messages = state.messages;\n      const lastMessage = messages[messages.length - 1];\n\n      // Check if we just executed a returnDirect tool\n      if (\n        ToolMessage.isInstance(lastMessage) &&\n        lastMessage.name &&\n        shouldReturnDirect.has(lastMessage.name)\n      ) {\n        // If we have a response format, route to agent to generate structured response\n        // Otherwise, return directly to exit node (could be after_agent or END)\n        return this.options.responseFormat ? \"model_request\" : exitNode;\n      }\n\n      // For non-returnDirect tools, always route back to agent\n      return \"model_request\";\n    };\n  }\n\n  /**\n   * Create routing function for model node conditional edges.\n   * @param exitNode - The exit node to route to (could be after_agent or END)\n   */\n  #createModelRouter(exitNode: string | typeof END = END) {\n    /**\n     * determine if the agent should continue or not\n     */\n    return (state: BuiltInState) => {\n      const messages = state.messages;\n      const lastMessage = messages.at(-1);\n\n      if (\n        !AIMessage.isInstance(lastMessage) ||\n        !lastMessage.tool_calls ||\n        lastMessage.tool_calls.length === 0\n      ) {\n        return exitNode;\n      }\n\n      // Check if all tool calls are for structured response extraction\n      const hasOnlyStructuredResponseCalls = lastMessage.tool_calls.every(\n        (toolCall) => toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (hasOnlyStructuredResponseCalls) {\n        // If all tool calls are for structured response extraction, go to exit node\n        // The AgentNode will handle these internally and return the structured response\n        return exitNode;\n      }\n\n      /**\n       * The tool node processes a single message.\n       */\n      if (this.#toolBehaviorVersion === \"v1\") {\n        return \"tools\";\n      }\n\n      /**\n       * Route to tools node (filter out any structured response tool calls)\n       */\n      const regularToolCalls = lastMessage.tool_calls.filter(\n        (toolCall) => !toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (regularToolCalls.length === 0) {\n        return exitNode;\n      }\n\n      return regularToolCalls.map(\n        (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n      );\n    };\n  }\n\n  /**\n   * Create routing function for jumpTo functionality after afterModel hooks.\n   *\n   * This router checks if the `jumpTo` property is set in the state after afterModel middleware\n   * execution. If set, it routes to the specified target (\"model_request\" or \"tools\").\n   * If not set, it falls back to the normal model routing logic for afterModel context.\n   *\n   * The jumpTo property is automatically cleared after use to prevent infinite loops.\n   *\n   * @param toolClasses - Available tool classes for validation\n   * @param allowJump - Whether jumping is allowed\n   * @param exitNode - The exit node to route to (could be after_agent or END)\n   * @returns Router function that handles jumpTo logic and normal routing\n   */\n  #createAfterModelRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    allowJump: boolean,\n    exitNode: string | typeof END\n  ) {\n    const hasStructuredResponse = Boolean(this.options.responseFormat);\n\n    return (state: Omit<BuiltInState, \"jumpTo\"> & { jumpTo?: JumpTo }) => {\n      // First, check if we just processed a structured response\n      // If so, ignore any existing jumpTo and go to exitNode\n      const messages = state.messages;\n      const lastMessage = messages.at(-1);\n      if (\n        AIMessage.isInstance(lastMessage) &&\n        (!lastMessage.tool_calls || lastMessage.tool_calls.length === 0)\n      ) {\n        return exitNode;\n      }\n\n      // Check if jumpTo is set in the state and allowed\n      if (allowJump && state.jumpTo) {\n        if (state.jumpTo === END) {\n          return exitNode;\n        }\n        if (state.jumpTo === \"tools\") {\n          // If trying to jump to tools but no tools are available, go to exitNode\n          if (toolClasses.length === 0) {\n            return exitNode;\n          }\n          return new Send(\"tools\", { ...state, jumpTo: undefined });\n        }\n        // destination === \"model_request\"\n        return new Send(\"model_request\", { ...state, jumpTo: undefined });\n      }\n\n      // check if there are pending tool calls\n      const toolMessages = messages.filter(ToolMessage.isInstance);\n      const lastAiMessage = messages.filter(AIMessage.isInstance).at(-1);\n      const pendingToolCalls = lastAiMessage?.tool_calls?.filter(\n        (call) => !toolMessages.some((m) => m.tool_call_id === call.id)\n      );\n      if (pendingToolCalls && pendingToolCalls.length > 0) {\n        return pendingToolCalls.map(\n          (toolCall) => new Send(\"tools\", { ...state, lg_tool_call: toolCall })\n        );\n      }\n\n      // if we exhausted all tool calls, but still have no structured response tool calls,\n      // go back to model_request\n      const hasStructuredResponseCalls = lastAiMessage?.tool_calls?.some(\n        (toolCall) => toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (\n        pendingToolCalls &&\n        pendingToolCalls.length === 0 &&\n        !hasStructuredResponseCalls &&\n        hasStructuredResponse\n      ) {\n        return \"model_request\";\n      }\n\n      if (\n        !AIMessage.isInstance(lastMessage) ||\n        !lastMessage.tool_calls ||\n        lastMessage.tool_calls.length === 0\n      ) {\n        return exitNode;\n      }\n\n      // Check if all tool calls are for structured response extraction\n      const hasOnlyStructuredResponseCalls = lastMessage.tool_calls.every(\n        (toolCall) => toolCall.name.startsWith(\"extract-\")\n      );\n\n      // Check if there are any regular tool calls (non-structured response)\n      const hasRegularToolCalls = lastMessage.tool_calls.some(\n        (toolCall) => !toolCall.name.startsWith(\"extract-\")\n      );\n\n      if (hasOnlyStructuredResponseCalls || !hasRegularToolCalls) {\n        return exitNode;\n      }\n\n      /**\n       * For routing from afterModel nodes, always use simple string paths\n       * The Send API is handled at the model_request node level\n       */\n      return \"tools\";\n    };\n  }\n\n  /**\n   * Router for afterModel sequence nodes (connecting later middlewares to earlier ones),\n   * honoring allowed jump targets and defaulting to the next node.\n   */\n  #createAfterModelSequenceRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    allowed: string[],\n    nextDefault: string\n  ) {\n    const allowedSet = new Set(allowed.map((t) => parseJumpToTarget(t)));\n    return (state: BuiltInState) => {\n      if (state.jumpTo) {\n        const dest = parseJumpToTarget(state.jumpTo);\n        if (dest === END && allowedSet.has(END)) {\n          return END;\n        }\n        if (dest === \"tools\" && allowedSet.has(\"tools\")) {\n          if (toolClasses.length === 0) return END;\n          return new Send(\"tools\", { ...state, jumpTo: undefined });\n        }\n        if (dest === \"model_request\" && allowedSet.has(\"model_request\")) {\n          return new Send(\"model_request\", { ...state, jumpTo: undefined });\n        }\n      }\n      return nextDefault as any;\n    };\n  }\n\n  /**\n   * Create routing function for jumpTo functionality after beforeAgent hooks.\n   * Falls back to the default next node if no jumpTo is present.\n   * When jumping to END, routes to exitNode (which could be an afterAgent node).\n   */\n  #createBeforeAgentRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    nextDefault: string,\n    exitNode: string | typeof END\n  ) {\n    return (state: BuiltInState) => {\n      if (!state.jumpTo) {\n        return nextDefault;\n      }\n      const destination = parseJumpToTarget(state.jumpTo);\n      if (destination === END) {\n        // When beforeAgent jumps to END, route to exitNode (first afterAgent node)\n        return exitNode;\n      }\n      if (destination === \"tools\") {\n        if (toolClasses.length === 0) {\n          return exitNode;\n        }\n        return new Send(\"tools\", { ...state, jumpTo: undefined });\n      }\n      // destination === \"model_request\"\n      return new Send(\"model_request\", { ...state, jumpTo: undefined });\n    };\n  }\n\n  /**\n   * Create routing function for jumpTo functionality after beforeModel hooks.\n   * Falls back to the default next node if no jumpTo is present.\n   */\n  #createBeforeModelRouter(\n    toolClasses: (ClientTool | ServerTool)[],\n    nextDefault: string\n  ) {\n    return (state: BuiltInState) => {\n      if (!state.jumpTo) {\n        return nextDefault;\n      }\n      const destination = parseJumpToTarget(state.jumpTo);\n      if (destination === END) {\n        return END;\n      }\n      if (destination === \"tools\") {\n        if (toolClasses.length === 0) {\n          return END;\n        }\n        return new Send(\"tools\", { ...state, jumpTo: undefined });\n      }\n      // destination === \"model_request\"\n      return new Send(\"model_request\", { ...state, jumpTo: undefined });\n    };\n  }\n\n  /**\n   * Initialize middleware states if not already present in the input state.\n   */\n  async #initializeMiddlewareStates(\n    state: InvokeStateParameter<TMiddleware>\n  ): Promise<InvokeStateParameter<TMiddleware>> {\n    if (\n      !this.options.middleware ||\n      this.options.middleware.length === 0 ||\n      state instanceof Command ||\n      !state\n    ) {\n      return state;\n    }\n\n    const defaultStates = await initializeMiddlewareStates(\n      this.options.middleware,\n      state\n    );\n    const updatedState = { ...state } as InvokeStateParameter<TMiddleware>;\n    if (!updatedState) {\n      return updatedState;\n    }\n\n    // Only add defaults for keys that don't exist in current state\n    for (const [key, value] of Object.entries(defaultStates)) {\n      if (!(key in updatedState)) {\n        updatedState[key as keyof typeof updatedState] = value;\n      }\n    }\n\n    return updatedState;\n  }\n\n  /**\n   * Populate the private state of the agent node from the previous state.\n   */\n  async #populatePrivateState(config?: RunnableConfig) {\n    /**\n     * not needed if thread_id is not provided\n     */\n    if (!config?.configurable?.thread_id) {\n      return;\n    }\n    const prevState = (await this.#graph.getState(config as any)) as {\n      values: {\n        _privateState: PrivateState;\n      };\n    };\n\n    /**\n     * not need if state is empty\n     */\n    if (!prevState.values._privateState) {\n      return;\n    }\n\n    this.#agentNode.setState({\n      structuredResponse: undefined,\n      _privateState: prevState.values._privateState,\n    });\n  }\n\n  /**\n   * Executes the agent with the given state and returns the final state after all processing.\n   *\n   * This method runs the agent's entire workflow synchronously, including:\n   * - Processing the input messages through any configured middleware\n   * - Calling the language model to generate responses\n   * - Executing any tool calls made by the model\n   * - Running all middleware hooks (beforeModel, afterModel, etc.)\n   *\n   * @param state - The initial state for the agent execution. Can be:\n   *   - An object containing `messages` array and any middleware-specific state properties\n   *   - A Command object for more advanced control flow\n   *\n   * @param config - Optional runtime configuration including:\n   * @param config.context - The context for the agent execution.\n   * @param config.configurable - LangGraph configuration options like `thread_id`, `run_id`, etc.\n   * @param config.store - The store for the agent execution for persisting state, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Memory storage}.\n   * @param config.signal - An optional {@link https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal | `AbortSignal`} for the agent execution.\n   * @param config.recursionLimit - The recursion limit for the agent execution.\n   *\n   * @returns A Promise that resolves to the final agent state after execution completes.\n   *          The returned state includes:\n   *          - a `messages` property containing an array with all messages (input, AI responses, tool calls/results)\n   *          - a `structuredResponse` property containing the structured response (if configured)\n   *          - all state values defined in the middleware\n   *\n   * @example\n   * ```typescript\n   * const agent = new ReactAgent({\n   *   llm: myModel,\n   *   tools: [calculator, webSearch],\n   *   responseFormat: z.object({\n   *     weather: z.string(),\n   *   }),\n   * });\n   *\n   * const result = await agent.invoke({\n   *   messages: [{ role: \"human\", content: \"What's the weather in Paris?\" }]\n   * });\n   *\n   * console.log(result.structuredResponse.weather); // outputs: \"It's sunny and 75°F.\"\n   * ```\n   */\n  async invoke(\n    state: InvokeStateParameter<TMiddleware>,\n    config?: InvokeConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >\n  ) {\n    type FullState = MergedAgentState<StructuredResponseFormat, TMiddleware>;\n    const initializedState = await this.#initializeMiddlewareStates(state);\n    await this.#populatePrivateState(config);\n\n    return this.#graph.invoke(\n      initializedState,\n      config as unknown as InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    ) as Promise<FullState>;\n  }\n\n  /**\n   * Executes the agent with streaming, returning an async iterable of state updates as they occur.\n   *\n   * This method runs the agent's workflow similar to `invoke`, but instead of waiting for\n   * completion, it streams high-level state updates in real-time. This allows you to:\n   * - Display intermediate results to users as they're generated\n   * - Monitor the agent's progress through each step\n   * - React to state changes as nodes complete\n   *\n   * For more granular event-level streaming (like individual LLM tokens), use `streamEvents` instead.\n   *\n   * @param state - The initial state for the agent execution. Can be:\n   *   - An object containing `messages` array and any middleware-specific state properties\n   *   - A Command object for more advanced control flow\n   *\n   * @param config - Optional runtime configuration including:\n   * @param config.context - The context for the agent execution.\n   * @param config.configurable - LangGraph configuration options like `thread_id`, `run_id`, etc.\n   * @param config.store - The store for the agent execution for persisting state, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Memory storage}.\n   * @param config.signal - An optional {@link https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal | `AbortSignal`} for the agent execution.\n   * @param config.streamMode - The streaming mode for the agent execution, see more in {@link https://docs.langchain.com/oss/javascript/langgraph/streaming#supported-stream-modes | Supported stream modes}.\n   * @param config.recursionLimit - The recursion limit for the agent execution.\n   *\n   * @returns A Promise that resolves to an IterableReadableStream of state updates.\n   *          Each update contains the current state after a node completes.\n   *\n   * @example\n   * ```typescript\n   * const agent = new ReactAgent({\n   *   llm: myModel,\n   *   tools: [calculator, webSearch]\n   * });\n   *\n   * const stream = await agent.stream({\n   *   messages: [{ role: \"human\", content: \"What's 2+2 and the weather in NYC?\" }]\n   * });\n   *\n   * for await (const chunk of stream) {\n   *   console.log(chunk); // State update from each node\n   * }\n   * ```\n   */\n  async stream(\n    state: InvokeStateParameter<TMiddleware>,\n    config?: StreamConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    >\n  ): Promise<IterableReadableStream<any>> {\n    const initializedState = await this.#initializeMiddlewareStates(state);\n    return this.#graph.stream(initializedState, config as Record<string, any>);\n  }\n\n  /**\n   * Visualize the graph as a PNG image.\n   * @param params - Parameters for the drawMermaidPng method.\n   * @param params.withStyles - Whether to include styles in the graph.\n   * @param params.curveStyle - The style of the graph's curves.\n   * @param params.nodeColors - The colors of the graph's nodes.\n   * @param params.wrapLabelNWords - The maximum number of words to wrap in a node's label.\n   * @param params.backgroundColor - The background color of the graph.\n   * @returns PNG image as a buffer\n   */\n  async drawMermaidPng(params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }) {\n    const representation = await this.#graph.getGraphAsync();\n    const image = await representation.drawMermaidPng(params);\n    const arrayBuffer = await image.arrayBuffer();\n    const buffer = new Uint8Array(arrayBuffer);\n    return buffer;\n  }\n\n  /**\n   * Draw the graph as a Mermaid string.\n   * @param params - Parameters for the drawMermaid method.\n   * @param params.withStyles - Whether to include styles in the graph.\n   * @param params.curveStyle - The style of the graph's curves.\n   * @param params.nodeColors - The colors of the graph's nodes.\n   * @param params.wrapLabelNWords - The maximum number of words to wrap in a node's label.\n   * @param params.backgroundColor - The background color of the graph.\n   * @returns Mermaid string\n   */\n  async drawMermaid(params?: {\n    withStyles?: boolean;\n    curveStyle?: string;\n    nodeColors?: Record<string, string>;\n    wrapLabelNWords?: number;\n    backgroundColor?: string;\n  }) {\n    const representation = await this.#graph.getGraphAsync();\n    return representation.drawMermaid(params);\n  }\n\n  /**\n   * The following are internal methods to enable support for LangGraph Platform.\n   * They are not part of the createAgent public API.\n   *\n   * Note: we intentionally return as `never` to avoid type errors due to type inference.\n   */\n\n  /**\n   * @internal\n   */\n  streamEvents(\n    state: InvokeStateParameter<TMiddleware>,\n    config?: StreamConfiguration<\n      InferContextInput<ContextSchema> &\n        InferMiddlewareContextInputs<TMiddleware>\n    > & { version?: \"v1\" | \"v2\" },\n    streamOptions?: Parameters<Runnable[\"streamEvents\"]>[2]\n  ): IterableReadableStream<StreamEvent> {\n    return this.#graph.streamEvents(\n      state,\n      {\n        ...(config as any),\n        version: config?.version ?? \"v2\",\n      },\n      streamOptions\n    );\n  }\n  /**\n   * @internal\n   */\n  getGraphAsync(config?: RunnableConfig) {\n    return this.#graph.getGraphAsync(config) as never;\n  }\n  /**\n   * @internal\n   */\n  getState(config: RunnableConfig, options?: GetStateOptions) {\n    return this.#graph.getState(config, options) as never;\n  }\n  /**\n   * @internal\n   */\n  getStateHistory(config: RunnableConfig, options?: CheckpointListOptions) {\n    return this.#graph.getStateHistory(config, options) as never;\n  }\n  /**\n   * @internal\n   */\n  getSubgraphs(namespace?: string, recurse?: boolean) {\n    return this.#graph.getSubgraphs(namespace, recurse) as never;\n  }\n  /**\n   * @internal\n   */\n  getSubgraphAsync(namespace?: string, recurse?: boolean) {\n    return this.#graph.getSubgraphsAsync(namespace, recurse) as never;\n  }\n  /**\n   * @internal\n   */\n  updateState(\n    inputConfig: LangGraphRunnableConfig,\n    values: Record<string, unknown> | unknown,\n    asNode?: string\n  ) {\n    return this.#graph.updateState(inputConfig, values, asNode) as never;\n  }\n}\n"],"mappings":";;;;;;;;;;;;;AAoGA,IAAa,aAAb,MAQE;CACA;CAEA,uBAAoC;CAEpC;CAEA,YACSA,SACP;EADO;EAEP,KAAKC,uBAAuB,QAAQ,WAAW,KAAKA;;;;AAKpD,MAAI,CAAC,QAAQ,MACX,OAAM,IAAI,MAAM;;;;AAMlB,MAAI,OAAO,QAAQ,UAAU,UAC3B,2BAA2B,QAAQ,MAAM;;;;EAM3C,MAAM,kBAAmB,KAAK,QAAQ,YAClC,OAAO,CAAC,MAAM,EAAE,MAAM,CACvB,QAAQ,CAAC,MAAM,EAAE,MAAM,IAAI,CAAE;EAChC,MAAM,cAAc,CAAC,GAAI,QAAQ,SAAS,CAAE,GAAG,GAAG,eAAgB;;;;;EAMlE,MAAM,qBAAqB,IAAI,IAC7B,YACG,OAAO,aAAa,CACpB,OAAO,CAAC,SAAS,kBAAkB,QAAQ,KAAK,aAAa,CAC7D,IAAI,CAAC,SAAS,KAAK,KAAK;;;;;EAO7B,MAAM,SAAS,iCACb,KAAK,QAAQ,mBAAmB,QAChC,KAAK,QAAQ,WACd;EAED,MAAM,WAAW,IAAI,WACnB,QACA,KAAK,QAAQ;EAGf,MAAM,mBAAmB;EAMzB,MAAMC,mBAIA,CAAE;EACR,MAAMC,mBAIA,CAAE;EACR,MAAMC,kBAIA,CAAE;EACR,MAAMC,kBAIA,CAAE;EACR,MAAMC,8BAMA,CAAE;EAER,KAAKC,aAAa,IAAI,UAAU;GAC9B,OAAO,KAAK,QAAQ;GACpB,cAAc,KAAK,QAAQ;GAC3B,kBAAkB,KAAK,QAAQ;GAC/B,MAAM,KAAK,QAAQ;GACnB,gBAAgB,KAAK,QAAQ;GAC7B,YAAY,KAAK,QAAQ;GACzB;GACA;GACA,QAAQ,KAAK,QAAQ;GACrB;EACD;EAED,MAAM,kCAAkB,IAAI;EAC5B,MAAM,aAAa,KAAK,QAAQ,cAAc,CAAE;AAChD,OAAK,IAAI,IAAI,GAAG,IAAI,WAAW,QAAQ,KAAK;GAC1C,IAAIC;GACJ,IAAIC;GACJ,IAAIC;GACJ,IAAIC;GACJ,MAAM,IAAI,WAAW;AACrB,OAAI,gBAAgB,IAAI,EAAE,KAAK,CAC7B,OAAM,IAAI,MAAM,CAAC,WAAW,EAAE,EAAE,KAAK,0BAA0B,CAAC;GAGlE,gBAAgB,IAAI,EAAE,KAAK;AAC3B,OAAI,EAAE,aAAa;IACjB,kBAAkB,IAAI,gBAAgB,GAAG,EACvC,iBAAiB,MAAM,KAAKJ,WAAW,UAAU,CAAC,cACnD;IACD,MAAM,OAAO,GAAG,EAAE,KAAK,aAAa,CAAC;IACrC,iBAAiB,KAAK;KACpB,OAAO;KACP;KACA,SAAS,EAAE;IACZ,EAAC;IACF,iBAAiB,QACf,MACA,iBACA,gBAAgB,YACjB;GACF;AACD,OAAI,EAAE,aAAa;IACjB,kBAAkB,IAAI,gBAAgB,GAAG,EACvC,iBAAiB,MAAM,KAAKA,WAAW,UAAU,CAAC,cACnD;IACD,MAAM,OAAO,GAAG,EAAE,KAAK,aAAa,CAAC;IACrC,iBAAiB,KAAK;KACpB,OAAO;KACP;KACA,SAAS,EAAE;IACZ,EAAC;IACF,iBAAiB,QACf,MACA,iBACA,gBAAgB,YACjB;GACF;AACD,OAAI,EAAE,YAAY;IAChB,iBAAiB,IAAI,eAAe,GAAG,EACrC,iBAAiB,MAAM,KAAKA,WAAW,UAAU,CAAC,cACnD;IACD,MAAM,OAAO,GAAG,EAAE,KAAK,YAAY,CAAC;IACpC,gBAAgB,KAAK;KACnB,OAAO;KACP;KACA,SAAS,EAAE;IACZ,EAAC;IACF,iBAAiB,QACf,MACA,gBACA,eAAe,YAChB;GACF;AACD,OAAI,EAAE,YAAY;IAChB,iBAAiB,IAAI,eAAe,GAAG,EACrC,iBAAiB,MAAM,KAAKA,WAAW,UAAU,CAAC,cACnD;IACD,MAAM,OAAO,GAAG,EAAE,KAAK,YAAY,CAAC;IACpC,gBAAgB,KAAK;KACnB,OAAO;KACP;KACA,SAAS,EAAE;IACZ,EAAC;IACF,iBAAiB,QACf,MACA,gBACA,eAAe,YAChB;GACF;AAED,OAAI,EAAE,eACJ,4BAA4B,KAAK,CAC/B,GACA,OAAO;IACL,GAAG,iBAAiB,UAAU;IAC9B,GAAG,iBAAiB,UAAU;IAC9B,GAAG,gBAAgB,UAAU;IAC7B,GAAG,gBAAgB,UAAU;GAC9B,EACF,EAAC;EAEL;;;;EAKD,iBAAiB,QACf,iBACA,KAAKA,YACL,UAAU,YACX;;;;;EAMD,MAAM,sBAAsB,aAAa,WAAW;;;;AAKpD,MAAI,YAAY,OAAO,aAAa,CAAC,SAAS,GAAG;GAC/C,MAAM,WAAW,IAAI,SAAS,YAAY,OAAO,aAAa,EAAE;IAC9D,QAAQ,KAAK,QAAQ;IACrB,cAAc;IACd,iBAAiB,MAAM,KAAKA,WAAW,UAAU,CAAC;GACnD;GACD,iBAAiB,QAAQ,SAAS,SAAS;EAC5C;;;;EAMD,IAAIK;AACJ,MAAI,iBAAiB,SAAS,GAC5B,YAAY,iBAAiB,GAAG;WACvB,iBAAiB,SAAS,GACnC,YAAY,iBAAiB,GAAG;OAEhC,YAAY;EAKd,MAAM,gBACJ,iBAAiB,SAAS,IAAI,iBAAiB,GAAG,OAAO;EAG3D,MAAM,WACJ,gBAAgB,SAAS,IACrB,gBAAgB,gBAAgB,SAAS,GAAG,OAC5C;EAEN,iBAAiB,QAAQ,OAAO,UAAU;AAG1C,OAAK,IAAI,IAAI,GAAG,IAAI,iBAAiB,QAAQ,KAAK;GAChD,MAAM,OAAO,iBAAiB;GAC9B,MAAM,UAAU,KAAK;GACrB,MAAM,SAAS,MAAM,iBAAiB,SAAS;GAC/C,MAAM,cAAc,SAAS,gBAAgB,iBAAiB,IAAI,GAAG;AAErE,OAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;IAC3C,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,KAAK,QACxB,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;IAEjD,MAAM,eAAe,MAAM,KACzB,IAAI,IAAI,CACN,aACA,GAAG,cAAc,IAAI,CAAC,SAAU,SAAS,MAAM,WAAW,KAAM,AACjE,GACF;IAED,iBAAiB,oBACf,SACA,KAAKC,yBACH,YAAY,OAAO,aAAa,EAChC,aACA,SACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,SAAS,YAAY;EAEjD;AAGD,OAAK,IAAI,IAAI,GAAG,IAAI,iBAAiB,QAAQ,KAAK;GAChD,MAAM,OAAO,iBAAiB;GAC9B,MAAM,UAAU,KAAK;GACrB,MAAM,SAAS,MAAM,iBAAiB,SAAS;GAC/C,MAAM,cAAc,SAChB,kBACA,iBAAiB,IAAI,GAAG;AAE5B,OAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;IAC3C,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,KAAK,QACxB,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;IACjD,MAAM,eAAe,MAAM,KACzB,IAAI,IAAI,CAAC,aAAa,GAAG,aAAc,GACxC;IAED,iBAAiB,oBACf,SACA,KAAKC,yBACH,YAAY,OAAO,aAAa,EAChC,YACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,SAAS,YAAY;EAEjD;EAGD,MAAM,qBAAqB,gBAAgB,GAAG,GAAG;AACjD,MAAI,gBAAgB,SAAS,KAAK,oBAChC,iBAAiB,QAAQ,iBAAiB,mBAAmB,KAAK;OAC7D;GAEL,MAAM,aAAa,KAAKC,eAAe,YAAY,OAAO,aAAa,CAAC;GAExE,MAAM,eAAe,WAAW,IAAI,CAAC,MACnC,MAAM,MAAM,WAAW,EACxB;AACD,OAAI,aAAa,WAAW,GAC1B,iBAAiB,QAAQ,iBAAiB,aAAa,GAAG;QAE1D,iBAAiB,oBACf,iBACA,KAAKC,mBAAmB,SAAS,EACjC,aACD;EAEJ;AAGD,OAAK,IAAI,IAAI,gBAAgB,SAAS,GAAG,IAAI,GAAG,KAAK;GACnD,MAAM,OAAO,gBAAgB;GAC7B,MAAM,UAAU,KAAK;GACrB,MAAM,cAAc,gBAAgB,IAAI,GAAG;AAE3C,OAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;IAC3C,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,KAAK,QACxB,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;IACjD,MAAM,eAAe,MAAM,KACzB,IAAI,IAAI,CAAC,aAAa,GAAG,aAAc,GACxC;IAED,iBAAiB,oBACf,SACA,KAAKC,gCACH,YAAY,OAAO,aAAa,EAChC,KAAK,SACL,YACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,SAAS,YAAY;EAEjD;AAGD,MAAI,gBAAgB,SAAS,GAAG;GAC9B,MAAM,kBAAkB,gBAAgB;GACxC,MAAM,sBAAsB,gBAAgB;GAG5C,MAAM,aAAa,KAAKF,eACtB,YAAY,OAAO,aAAa,EAChC,KACD,CAAC,OACA,CAAC,MAAM,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS,EACnE;GAED,MAAM,YAAY,QAChB,gBAAgB,WAAW,gBAAgB,QAAQ,SAAS,EAC7D;GAGD,MAAM,eAAe,WAAW,IAAI,CAAC,MACnC,MAAM,MAAM,WAAW,EACxB;GAED,iBAAiB,oBACf,qBACA,KAAKG,wBACH,YAAY,OAAO,aAAa,EAChC,WACA,SACD,EACD,aACD;EACF;AAGD,OAAK,IAAI,IAAI,gBAAgB,SAAS,GAAG,IAAI,GAAG,KAAK;GACnD,MAAM,OAAO,gBAAgB;GAC7B,MAAM,UAAU,KAAK;GACrB,MAAM,cAAc,gBAAgB,IAAI,GAAG;AAE3C,OAAI,KAAK,WAAW,KAAK,QAAQ,SAAS,GAAG;IAC3C,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,KAAK,QACxB,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;IACjD,MAAM,eAAe,MAAM,KACzB,IAAI,IAAI,CAAC,aAAa,GAAG,aAAc,GACxC;IAED,iBAAiB,oBACf,SACA,KAAKD,gCACH,YAAY,OAAO,aAAa,EAChC,KAAK,SACL,YACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,SAAS,YAAY;EAEjD;AAGD,MAAI,gBAAgB,SAAS,GAAG;GAC9B,MAAM,kBAAkB,gBAAgB;GACxC,MAAM,sBAAsB,gBAAgB;AAE5C,OAAI,gBAAgB,WAAW,gBAAgB,QAAQ,SAAS,GAAG;IACjE,MAAM,WAAW,YAAY,OAAO,aAAa,CAAC,SAAS;IAC3D,MAAM,gBAAgB,gBAAgB,QACnC,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC,CAChC,OAAO,CAAC,SAAS,SAAS,WAAW,SAAS;;;;;IAMjD,MAAM,eAAe,MAAM,KAAK,IAAI,IAAI,CAAC,KAAK,GAAG,aAAc,GAAE;IAMjE,iBAAiB,oBACf,qBACA,KAAKA,gCACH,YAAY,OAAO,aAAa,EAChC,gBAAgB,SAChB,IACD,EACD,aACD;GACF,OACC,iBAAiB,QAAQ,qBAAqB,IAAI;EAErD;;;;AAKD,MAAI,YAAY,OAAO,aAAa,CAAC,SAAS,GAAG;GAE/C,MAAM,mBAAmB;AAEzB,OAAI,mBAAmB,OAAO,GAC5B,iBAAiB,oBACf,SACA,KAAKE,mBAAmB,oBAAoB,SAAS,EACrD,CAAC,kBAAkB,QAAmB,EACvC;QAED,iBAAiB,QAAQ,SAAS,iBAAiB;EAEtD;;;;EAKD,KAAKC,SAAS,iBAAiB,QAAQ;GACrC,cAAc,KAAK,QAAQ;GAC3B,OAAO,KAAK,QAAQ;GACpB,MAAM,KAAK,QAAQ;GACnB,aAAa,KAAK,QAAQ;EAC3B,EAAC;CACH;;;;CAKD,IAAI,QAIF;AACA,SAAO,KAAKA;CACb;;;;;;;CAQD,eACEC,aACAC,sBAA+B,OACa;EAC5C,MAAMC,QAAoD,CAAE;AAC5D,MAAI,YAAY,SAAS,GACvB,MAAM,KAAK,QAAQ;AAGrB,MAAI,qBACF,MAAM,KAAK,gBAAgB;EAG7B,MAAM,KAAK,IAAI;AAEf,SAAO;CACR;;;;CAKD,mBACEC,oBACAC,UACA;;;;AAIA,SAAO,CAACC,UAAe;GACrB,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,SAAS,SAAS;AAG/C,OACE,YAAY,WAAW,YAAY,IACnC,YAAY,QACZ,mBAAmB,IAAI,YAAY,KAAK,CAIxC,QAAO,KAAK,QAAQ,iBAAiB,kBAAkB;AAIzD,UAAO;EACR;CACF;;;;;CAMD,mBAAmBD,WAAgC,KAAK;;;;AAItD,SAAO,CAACE,UAAwB;GAC9B,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AAEnC,OACE,CAAC,UAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,EAElC,QAAO;GAIT,MAAM,iCAAiC,YAAY,WAAW,MAC5D,CAAC,aAAa,SAAS,KAAK,WAAW,WAAW,CACnD;AAED,OAAI,+BAGF,QAAO;;;;AAMT,OAAI,KAAK1B,yBAAyB,KAChC,QAAO;;;;GAMT,MAAM,mBAAmB,YAAY,WAAW,OAC9C,CAAC,aAAa,CAAC,SAAS,KAAK,WAAW,WAAW,CACpD;AAED,OAAI,iBAAiB,WAAW,EAC9B,QAAO;AAGT,UAAO,iBAAiB,IACtB,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;EACF;CACF;;;;;;;;;;;;;;;CAgBD,wBACEoB,aACAO,WACAH,UACA;EACA,MAAM,wBAAwB,QAAQ,KAAK,QAAQ,eAAe;AAElE,SAAO,CAACI,UAA8D;GAGpE,MAAM,WAAW,MAAM;GACvB,MAAM,cAAc,SAAS,GAAG,GAAG;AACnC,OACE,UAAU,WAAW,YAAY,KAChC,CAAC,YAAY,cAAc,YAAY,WAAW,WAAW,GAE9D,QAAO;AAIT,OAAI,aAAa,MAAM,QAAQ;AAC7B,QAAI,MAAM,WAAW,IACnB,QAAO;AAET,QAAI,MAAM,WAAW,SAAS;AAE5B,SAAI,YAAY,WAAW,EACzB,QAAO;AAET,YAAO,IAAI,KAAK,SAAS;MAAE,GAAG;MAAO,QAAQ;KAAW;IACzD;AAED,WAAO,IAAI,KAAK,iBAAiB;KAAE,GAAG;KAAO,QAAQ;IAAW;GACjE;GAGD,MAAM,eAAe,SAAS,OAAO,YAAY,WAAW;GAC5D,MAAM,gBAAgB,SAAS,OAAO,UAAU,WAAW,CAAC,GAAG,GAAG;GAClE,MAAM,mBAAmB,eAAe,YAAY,OAClD,CAAC,SAAS,CAAC,aAAa,KAAK,CAAC,MAAM,EAAE,iBAAiB,KAAK,GAAG,CAChE;AACD,OAAI,oBAAoB,iBAAiB,SAAS,EAChD,QAAO,iBAAiB,IACtB,CAAC,aAAa,IAAI,KAAK,SAAS;IAAE,GAAG;IAAO,cAAc;GAAU,GACrE;GAKH,MAAM,6BAA6B,eAAe,YAAY,KAC5D,CAAC,aAAa,SAAS,KAAK,WAAW,WAAW,CACnD;AAED,OACE,oBACA,iBAAiB,WAAW,KAC5B,CAAC,8BACD,sBAEA,QAAO;AAGT,OACE,CAAC,UAAU,WAAW,YAAY,IAClC,CAAC,YAAY,cACb,YAAY,WAAW,WAAW,EAElC,QAAO;GAIT,MAAM,iCAAiC,YAAY,WAAW,MAC5D,CAAC,aAAa,SAAS,KAAK,WAAW,WAAW,CACnD;GAGD,MAAM,sBAAsB,YAAY,WAAW,KACjD,CAAC,aAAa,CAAC,SAAS,KAAK,WAAW,WAAW,CACpD;AAED,OAAI,kCAAkC,CAAC,oBACrC,QAAO;;;;;AAOT,UAAO;EACR;CACF;;;;;CAMD,gCACER,aACAS,SACAC,aACA;EACA,MAAM,aAAa,IAAI,IAAI,QAAQ,IAAI,CAAC,MAAM,kBAAkB,EAAE,CAAC;AACnE,SAAO,CAACJ,UAAwB;AAC9B,OAAI,MAAM,QAAQ;IAChB,MAAM,OAAO,kBAAkB,MAAM,OAAO;AAC5C,QAAI,SAAS,OAAO,WAAW,IAAI,IAAI,CACrC,QAAO;AAET,QAAI,SAAS,WAAW,WAAW,IAAI,QAAQ,EAAE;AAC/C,SAAI,YAAY,WAAW,EAAG,QAAO;AACrC,YAAO,IAAI,KAAK,SAAS;MAAE,GAAG;MAAO,QAAQ;KAAW;IACzD;AACD,QAAI,SAAS,mBAAmB,WAAW,IAAI,gBAAgB,CAC7D,QAAO,IAAI,KAAK,iBAAiB;KAAE,GAAG;KAAO,QAAQ;IAAW;GAEnE;AACD,UAAO;EACR;CACF;;;;;;CAOD,yBACEN,aACAU,aACAN,UACA;AACA,SAAO,CAACE,UAAwB;AAC9B,OAAI,CAAC,MAAM,OACT,QAAO;GAET,MAAM,cAAc,kBAAkB,MAAM,OAAO;AACnD,OAAI,gBAAgB,IAElB,QAAO;AAET,OAAI,gBAAgB,SAAS;AAC3B,QAAI,YAAY,WAAW,EACzB,QAAO;AAET,WAAO,IAAI,KAAK,SAAS;KAAE,GAAG;KAAO,QAAQ;IAAW;GACzD;AAED,UAAO,IAAI,KAAK,iBAAiB;IAAE,GAAG;IAAO,QAAQ;GAAW;EACjE;CACF;;;;;CAMD,yBACEN,aACAU,aACA;AACA,SAAO,CAACJ,UAAwB;AAC9B,OAAI,CAAC,MAAM,OACT,QAAO;GAET,MAAM,cAAc,kBAAkB,MAAM,OAAO;AACnD,OAAI,gBAAgB,IAClB,QAAO;AAET,OAAI,gBAAgB,SAAS;AAC3B,QAAI,YAAY,WAAW,EACzB,QAAO;AAET,WAAO,IAAI,KAAK,SAAS;KAAE,GAAG;KAAO,QAAQ;IAAW;GACzD;AAED,UAAO,IAAI,KAAK,iBAAiB;IAAE,GAAG;IAAO,QAAQ;GAAW;EACjE;CACF;;;;CAKD,MAAMK,4BACJC,OAC4C;AAC5C,MACE,CAAC,KAAK,QAAQ,cACd,KAAK,QAAQ,WAAW,WAAW,KACnC,iBAAiB,WACjB,CAAC,MAED,QAAO;EAGT,MAAM,gBAAgB,MAAM,2BAC1B,KAAK,QAAQ,YACb,MACD;EACD,MAAM,eAAe,EAAE,GAAG,MAAO;AACjC,MAAI,CAAC,aACH,QAAO;AAIT,OAAK,MAAM,CAAC,KAAK,MAAM,IAAI,OAAO,QAAQ,cAAc,CACtD,KAAI,EAAE,OAAO,eACX,aAAa,OAAoC;AAIrD,SAAO;CACR;;;;CAKD,MAAMC,sBAAsBC,QAAyB;;;;AAInD,MAAI,CAAC,QAAQ,cAAc,UACzB;EAEF,MAAM,YAAa,MAAM,KAAKf,OAAO,SAAS,OAAc;;;;AAS5D,MAAI,CAAC,UAAU,OAAO,cACpB;EAGF,KAAKb,WAAW,SAAS;GACvB,oBAAoB;GACpB,eAAe,UAAU,OAAO;EACjC,EAAC;CACH;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA6CD,MAAM,OACJ0B,OACAG,QAIA;EAEA,MAAM,mBAAmB,MAAM,KAAKJ,4BAA4B,MAAM;EACtE,MAAM,KAAKE,sBAAsB,OAAO;AAExC,SAAO,KAAKd,OAAO,OACjB,kBACA,OAED;CACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CA4CD,MAAM,OACJa,OACAI,QAIsC;EACtC,MAAM,mBAAmB,MAAM,KAAKL,4BAA4B,MAAM;AACtE,SAAO,KAAKZ,OAAO,OAAO,kBAAkB,OAA8B;CAC3E;;;;;;;;;;;CAYD,MAAM,eAAekB,QAMlB;EACD,MAAM,iBAAiB,MAAM,KAAKlB,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,YAAYkB,QAMf;EACD,MAAM,iBAAiB,MAAM,KAAKlB,OAAO,eAAe;AACxD,SAAO,eAAe,YAAY,OAAO;CAC1C;;;;;;;;;;CAYD,aACEa,OACAM,QAIAC,eACqC;AACrC,SAAO,KAAKpB,OAAO,aACjB,OACA;GACE,GAAI;GACJ,SAAS,QAAQ,WAAW;EAC7B,GACD,cACD;CACF;;;;CAID,cAAce,QAAyB;AACrC,SAAO,KAAKf,OAAO,cAAc,OAAO;CACzC;;;;CAID,SAASqB,QAAwBC,SAA2B;AAC1D,SAAO,KAAKtB,OAAO,SAAS,QAAQ,QAAQ;CAC7C;;;;CAID,gBAAgBqB,QAAwBE,SAAiC;AACvE,SAAO,KAAKvB,OAAO,gBAAgB,QAAQ,QAAQ;CACpD;;;;CAID,aAAawB,WAAoBC,SAAmB;AAClD,SAAO,KAAKzB,OAAO,aAAa,WAAW,QAAQ;CACpD;;;;CAID,iBAAiBwB,WAAoBC,SAAmB;AACtD,SAAO,KAAKzB,OAAO,kBAAkB,WAAW,QAAQ;CACzD;;;;CAID,YACE0B,aACAC,QACAC,QACA;AACA,SAAO,KAAK5B,OAAO,YAAY,aAAa,QAAQ,OAAO;CAC5D;AACF"}

package/dist/agents/annotation.cjs

@@ -1,163 +1,46 @@
 const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
 const __langchain_langgraph = require_rolldown_runtime.__toESM(require("@langchain/langgraph"));
-const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));
+const zod_v3 = require_rolldown_runtime.__toESM(require("zod/v3"));
+const __langchain_langgraph_zod = require_rolldown_runtime.__toESM(require("@langchain/langgraph/zod"));
 
 //#region src/agents/annotation.ts
-const PreHookAnnotation = __langchain_langgraph.Annotation.Root({
-	llmInputMessages: (0, __langchain_langgraph.Annotation)({
-		reducer: (_, update) => (0, __langchain_langgraph.messagesStateReducer)([], update),
-		default: () => []
-	}),
-	messages: (0, __langchain_langgraph.Annotation)({
-		reducer: __langchain_langgraph.messagesStateReducer,
-		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,
-		default: () => []
-	}) };
-	if (!hasStructuredResponse) return __langchain_langgraph.Annotation.Root(baseAnnotation);
-	return __langchain_langgraph.Annotation.Root({
-		...baseAnnotation,
-		structuredResponse: (0, __langchain_langgraph.Annotation)()
-	});
-}
-/**
-* Enhances a state schema to ensure proper message handling.
-* If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer
-* and infers appropriate reducers for all other fields.
-*
-* @param stateSchema - The state schema to enhance.
-* @returns The enhanced state schema.
-*/
-function enhanceStateSchemaWithMessageReducer(stateSchema) {
+function createAgentAnnotationConditional(hasStructuredResponse = true, middlewareList = []) {
 	/**
-	* If it's already an annotation, return as-is
+	* Create Zod schema object to preserve jsonSchemaExtra
+	* metadata for LangGraph Studio using v3-compatible withLangGraph
 	*/
-	if (typeof stateSchema === "object" && "State" in stateSchema) return stateSchema;
+	const zodSchema = {
+		messages: (0, __langchain_langgraph_zod.withLangGraph)(zod_v3.z.custom(), __langchain_langgraph.MessagesZodMeta),
+		jumpTo: zod_v3.z.union([
+			zod_v3.z.literal("model_request"),
+			zod_v3.z.literal("tools"),
+			zod_v3.z.undefined()
+		]).optional()
+	};
 	/**
-	* If it's a Zod schema, create annotations for all fields
+	* Add middleware state properties to the Zod schema
 	*/
-	if ((0, __langchain_core_utils_types.isInteropZodSchema)(stateSchema)) {
-		const shape = (0, __langchain_core_utils_types.getInteropZodObjectShape)(stateSchema);
-		const annotationFields = {};
-		/**
-		* Process each field in the Zod schema
-		*/
-		for (const [key, zodType] of Object.entries(shape)) annotationFields[key] = key === "messages" ? (0, __langchain_langgraph.Annotation)({
-			reducer: __langchain_langgraph.messagesStateReducer,
-			default: () => []
-		}) : createAnnotationForZodType(zodType);
-		/**
-		* Ensure messages field exists even if not in the Zod schema
-		*/
-		if (!annotationFields.messages) annotationFields.messages = (0, __langchain_langgraph.Annotation)({
-			reducer: __langchain_langgraph.messagesStateReducer,
-			default: () => []
-		});
-		return __langchain_langgraph.Annotation.Root(annotationFields);
+	for (const middleware of middlewareList) if (middleware.stateSchema) {
+		const { shape } = middleware.stateSchema;
+		for (const [key, schema] of Object.entries(shape)) {
+			/**
+			* Skip private state properties
+			*/
+			if (key.startsWith("_")) continue;
+			if (!(key in zodSchema)) zodSchema[key] = schema;
+		}
 	}
-	/**
-	* Fallback: create a base annotation with message reducer only
-	*/
-	return __langchain_langgraph.Annotation.Root({ messages: (0, __langchain_langgraph.Annotation)({
-		reducer: __langchain_langgraph.messagesStateReducer,
-		default: () => []
-	}) });
-}
-/**
-* Type configuration for different Zod types
-*/
-const ZOD_TYPE_CONFIGS = {
-	ZodString: {
-		reducer: (_, update) => update,
-		fallbackDefault: ""
-	},
-	ZodNumber: {
-		reducer: (_, update) => update,
-		fallbackDefault: 0
-	},
-	ZodBoolean: {
-		reducer: (_, update) => update,
-		fallbackDefault: false
-	},
-	ZodArray: {
-		reducer: (_, update) => update,
-		fallbackDefault: []
-	},
-	ZodRecord: {
-		reducer: (current, update) => ({
-			...current,
-			...update
-		}),
-		fallbackDefault: {}
-	},
-	ZodObject: {
-		reducer: (current, update) => ({
-			...current,
-			...update
-		}),
-		fallbackDefault: {}
-	}
-};
-/**
-* Creates an annotation based on type configuration and default value
-*/
-function createAnnotationFromConfig(config, defaultValueFn) {
-	return (0, __langchain_langgraph.Annotation)({
-		reducer: config.reducer,
-		default: defaultValueFn
-	});
-}
-/**
-* Creates an appropriate annotation for a given Zod type.
-*/
-function createAnnotationForZodType(zodType) {
-	const typeName = zodType._def?.typeName;
-	const defaultGetter = (0, __langchain_core_utils_types.getInteropZodDefaultGetter)(zodType);
-	const isOptional = typeName === "ZodOptional";
-	/**
-	* Handle Zod wrapper first
-	*/
-	if (typeName === "ZodDefault" || typeName === "ZodOptional" || typeName === "ZodNullable") {
-		const innerTypeName = zodType._def.innerType._def?.typeName;
-		const config$1 = ZOD_TYPE_CONFIGS[innerTypeName];
-		return config$1 ? createAnnotationFromConfig(config$1, () => defaultGetter?.() || (isOptional ? void 0 : config$1?.fallbackDefault)) : (0, __langchain_langgraph.Annotation)({
-			reducer: (_, update) => update,
-			default: () => defaultGetter?.()
-		});
-	}
-	/**
-	* Handle regular types
-	*/
-	const config = ZOD_TYPE_CONFIGS[typeName];
-	if (config) return createAnnotationFromConfig(config, () => defaultGetter?.() || config?.fallbackDefault);
-	/**
-	* Fallback for unknown types
-	*/
-	return (0, __langchain_langgraph.Annotation)({
-		reducer: (_, update) => update,
-		default: () => defaultGetter?.()
-	});
+	if (hasStructuredResponse) zodSchema.structuredResponse = zod_v3.z.any().optional();
+	return zod_v3.z.object(zodSchema);
 }
+const PreHookAnnotation = zod_v3.z.object({
+	llmInputMessages: (0, __langchain_langgraph_zod.withLangGraph)(zod_v3.z.custom(), {
+		reducer: { fn: (_x, update) => __langchain_langgraph.MessagesZodMeta.reducer.fn([], update) },
+		default: () => []
+	}),
+	messages: (0, __langchain_langgraph_zod.withLangGraph)(zod_v3.z.custom(), __langchain_langgraph.MessagesZodMeta)
+});
 
 //#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;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"}
+{"version":3,"file":"annotation.cjs","names":["middlewareList: TMiddleware","zodSchema: Record<string, any>","z","MessagesZodMeta","PreHookAnnotation: AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>","_x: Messages","update: Messages"],"sources":["../../src/agents/annotation.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { BaseMessage } from \"@langchain/core/messages\";\nimport { z } from \"zod/v3\";\nimport {\n  Messages,\n  AnnotationRoot,\n  MessagesZodMeta,\n  type BinaryOperatorAggregate,\n} from \"@langchain/langgraph\";\nimport { withLangGraph } from \"@langchain/langgraph/zod\";\n\nimport type { AgentMiddleware } from \"./middleware/types.js\";\n\nexport function createAgentAnnotationConditional<\n  TMiddleware extends readonly AgentMiddleware<any, any, any>[] = []\n>(\n  hasStructuredResponse = true,\n  middlewareList: TMiddleware = [] as unknown as TMiddleware\n) {\n  /**\n   * Create Zod schema object to preserve jsonSchemaExtra\n   * metadata for LangGraph Studio using v3-compatible withLangGraph\n   */\n  const zodSchema: Record<string, any> = {\n    messages: withLangGraph(z.custom<BaseMessage[]>(), MessagesZodMeta),\n    jumpTo: z\n      .union([z.literal(\"model_request\"), z.literal(\"tools\"), z.undefined()])\n      .optional(),\n  };\n\n  /**\n   * Add middleware state properties to the Zod schema\n   */\n  for (const middleware of middlewareList) {\n    if (middleware.stateSchema) {\n      const { shape } = middleware.stateSchema;\n      for (const [key, schema] of Object.entries(shape)) {\n        /**\n         * Skip private state properties\n         */\n        if (key.startsWith(\"_\")) {\n          continue;\n        }\n\n        if (!(key in zodSchema)) {\n          zodSchema[key] = schema;\n        }\n      }\n    }\n  }\n\n  // Only include structuredResponse when responseFormat is defined\n  if (hasStructuredResponse) {\n    zodSchema.structuredResponse = z.any().optional();\n  }\n\n  return z.object(zodSchema);\n}\n\nexport const PreHookAnnotation: AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}> = z.object({\n  llmInputMessages: withLangGraph(z.custom<BaseMessage[]>(), {\n    reducer: {\n      fn: (_x: Messages, update: Messages) =>\n        MessagesZodMeta.reducer!.fn([], update),\n    },\n    default: () => [],\n  }),\n  /**\n   * Use MessagesZodMeta to preserve jsonSchemaExtra metadata\n   * for LangGraph Studio UI to render proper messages input field\n   */\n  messages: withLangGraph(z.custom<BaseMessage[]>(), MessagesZodMeta),\n}) as unknown as AnnotationRoot<{\n  llmInputMessages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n  messages: BinaryOperatorAggregate<BaseMessage[], Messages>;\n}>;\nexport type PreHookAnnotation = typeof PreHookAnnotation;\n"],"mappings":";;;;;;AAaA,SAAgB,iCAGd,wBAAwB,MACxBA,iBAA8B,CAAE,GAChC;;;;;CAKA,MAAMC,YAAiC;EACrC,uDAAwBC,SAAE,QAAuB,EAAEC,sCAAgB;EACnE,QAAQD,SACL,MAAM;GAACA,SAAE,QAAQ,gBAAgB;GAAEA,SAAE,QAAQ,QAAQ;GAAEA,SAAE,WAAW;EAAC,EAAC,CACtE,UAAU;CACd;;;;AAKD,MAAK,MAAM,cAAc,eACvB,KAAI,WAAW,aAAa;EAC1B,MAAM,EAAE,OAAO,GAAG,WAAW;AAC7B,OAAK,MAAM,CAAC,KAAK,OAAO,IAAI,OAAO,QAAQ,MAAM,EAAE;;;;AAIjD,OAAI,IAAI,WAAW,IAAI,CACrB;AAGF,OAAI,EAAE,OAAO,YACX,UAAU,OAAO;EAEpB;CACF;AAIH,KAAI,uBACF,UAAU,qBAAqBA,SAAE,KAAK,CAAC,UAAU;AAGnD,QAAOA,SAAE,OAAO,UAAU;AAC3B;AAED,MAAaE,oBAGRF,SAAE,OAAO;CACZ,+DAAgCA,SAAE,QAAuB,EAAE;EACzD,SAAS,EACP,IAAI,CAACG,IAAcC,WACjBH,sCAAgB,QAAS,GAAG,CAAE,GAAE,OAAO,CAC1C;EACD,SAAS,MAAM,CAAE;CAClB,EAAC;CAKF,uDAAwBD,SAAE,QAAuB,EAAEC,sCAAgB;AACpE,EAAC"}

package/dist/agents/annotation.js

@@ -1,158 +1,45 @@
-import { Annotation, messagesStateReducer } from "@langchain/langgraph";
-import { getInteropZodDefaultGetter, getInteropZodObjectShape, isInteropZodSchema } from "@langchain/core/utils/types";
+import { MessagesZodMeta } from "@langchain/langgraph";
+import { z } from "zod/v3";
+import { withLangGraph } from "@langchain/langgraph/zod";
 
 //#region src/agents/annotation.ts
-const PreHookAnnotation = Annotation.Root({
-	llmInputMessages: Annotation({
-		reducer: (_, update) => messagesStateReducer([], update),
-		default: () => []
-	}),
-	messages: Annotation({
-		reducer: messagesStateReducer,
-		default: () => []
-	})
-});
-const createAgentBaseAnnotation = () => Annotation.Root({ messages: Annotation({
-	reducer: messagesStateReducer,
-	default: () => []
-}) });
-const createAgentAnnotation = () => Annotation.Root({
-	messages: Annotation({
-		reducer: messagesStateReducer,
-		default: () => []
-	}),
-	structuredResponse: Annotation()
-});
-function createAgentAnnotationConditional(hasStructuredResponse = true) {
-	const baseAnnotation = { messages: Annotation({
-		reducer: messagesStateReducer,
-		default: () => []
-	}) };
-	if (!hasStructuredResponse) return Annotation.Root(baseAnnotation);
-	return Annotation.Root({
-		...baseAnnotation,
-		structuredResponse: Annotation()
-	});
-}
-/**
-* Enhances a state schema to ensure proper message handling.
-* If a Zod schema is provided, it creates an annotation that includes the messagesStateReducer
-* and infers appropriate reducers for all other fields.
-*
-* @param stateSchema - The state schema to enhance.
-* @returns The enhanced state schema.
-*/
-function enhanceStateSchemaWithMessageReducer(stateSchema) {
+function createAgentAnnotationConditional(hasStructuredResponse = true, middlewareList = []) {
 	/**
-	* If it's already an annotation, return as-is
+	* Create Zod schema object to preserve jsonSchemaExtra
+	* metadata for LangGraph Studio using v3-compatible withLangGraph
 	*/
-	if (typeof stateSchema === "object" && "State" in stateSchema) return stateSchema;
+	const zodSchema = {
+		messages: withLangGraph(z.custom(), MessagesZodMeta),
+		jumpTo: z.union([
+			z.literal("model_request"),
+			z.literal("tools"),
+			z.undefined()
+		]).optional()
+	};
 	/**
-	* If it's a Zod schema, create annotations for all fields
+	* Add middleware state properties to the Zod schema
 	*/
-	if (isInteropZodSchema(stateSchema)) {
-		const shape = getInteropZodObjectShape(stateSchema);
-		const annotationFields = {};
-		/**
-		* Process each field in the Zod schema
-		*/
-		for (const [key, zodType] of Object.entries(shape)) annotationFields[key] = key === "messages" ? Annotation({
-			reducer: messagesStateReducer,
-			default: () => []
-		}) : createAnnotationForZodType(zodType);
-		/**
-		* Ensure messages field exists even if not in the Zod schema
-		*/
-		if (!annotationFields.messages) annotationFields.messages = Annotation({
-			reducer: messagesStateReducer,
-			default: () => []
-		});
-		return Annotation.Root(annotationFields);
+	for (const middleware of middlewareList) if (middleware.stateSchema) {
+		const { shape } = middleware.stateSchema;
+		for (const [key, schema] of Object.entries(shape)) {
+			/**
+			* Skip private state properties
+			*/
+			if (key.startsWith("_")) continue;
+			if (!(key in zodSchema)) zodSchema[key] = schema;
+		}
 	}
-	/**
-	* Fallback: create a base annotation with message reducer only
-	*/
-	return Annotation.Root({ messages: Annotation({
-		reducer: messagesStateReducer,
-		default: () => []
-	}) });
-}
-/**
-* Type configuration for different Zod types
-*/
-const ZOD_TYPE_CONFIGS = {
-	ZodString: {
-		reducer: (_, update) => update,
-		fallbackDefault: ""
-	},
-	ZodNumber: {
-		reducer: (_, update) => update,
-		fallbackDefault: 0
-	},
-	ZodBoolean: {
-		reducer: (_, update) => update,
-		fallbackDefault: false
-	},
-	ZodArray: {
-		reducer: (_, update) => update,
-		fallbackDefault: []
-	},
-	ZodRecord: {
-		reducer: (current, update) => ({
-			...current,
-			...update
-		}),
-		fallbackDefault: {}
-	},
-	ZodObject: {
-		reducer: (current, update) => ({
-			...current,
-			...update
-		}),
-		fallbackDefault: {}
-	}
-};
-/**
-* Creates an annotation based on type configuration and default value
-*/
-function createAnnotationFromConfig(config, defaultValueFn) {
-	return Annotation({
-		reducer: config.reducer,
-		default: defaultValueFn
-	});
-}
-/**
-* Creates an appropriate annotation for a given Zod type.
-*/
-function createAnnotationForZodType(zodType) {
-	const typeName = zodType._def?.typeName;
-	const defaultGetter = getInteropZodDefaultGetter(zodType);
-	const isOptional = typeName === "ZodOptional";
-	/**
-	* Handle Zod wrapper first
-	*/
-	if (typeName === "ZodDefault" || typeName === "ZodOptional" || typeName === "ZodNullable") {
-		const innerTypeName = zodType._def.innerType._def?.typeName;
-		const config$1 = ZOD_TYPE_CONFIGS[innerTypeName];
-		return config$1 ? createAnnotationFromConfig(config$1, () => defaultGetter?.() || (isOptional ? void 0 : config$1?.fallbackDefault)) : Annotation({
-			reducer: (_, update) => update,
-			default: () => defaultGetter?.()
-		});
-	}
-	/**
-	* Handle regular types
-	*/
-	const config = ZOD_TYPE_CONFIGS[typeName];
-	if (config) return createAnnotationFromConfig(config, () => defaultGetter?.() || config?.fallbackDefault);
-	/**
-	* Fallback for unknown types
-	*/
-	return Annotation({
-		reducer: (_, update) => update,
-		default: () => defaultGetter?.()
-	});
+	if (hasStructuredResponse) zodSchema.structuredResponse = z.any().optional();
+	return z.object(zodSchema);
 }
+const PreHookAnnotation = z.object({
+	llmInputMessages: withLangGraph(z.custom(), {
+		reducer: { fn: (_x, update) => MessagesZodMeta.reducer.fn([], update) },
+		default: () => []
+	}),
+	messages: withLangGraph(z.custom(), MessagesZodMeta)
+});
 
 //#endregion
-export { PreHookAnnotation, createAgentAnnotation, createAgentAnnotationConditional, createAgentBaseAnnotation, enhanceStateSchemaWithMessageReducer };
+export { createAgentAnnotationConditional };
 //# sourceMappingURL=annotation.js.map