langchain 1.0.0-alpha.6 → 1.0.0-alpha.8

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

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

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

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

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

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

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

@@ -1,54 +1,53 @@
-import { JsonSchemaFormat, ProviderStrategy, ResponseFormat, ResponseFormatUndefined, ToolStrategy, TypedToolStrategy } from "../responses.cjs";
-import { AnyAnnotationRoot, PreHookAnnotation, ToAnnotationRoot } from "../annotation.cjs";
+import { JUMP_TO_TARGETS } from "./constants.cjs";
+import { AnyAnnotationRoot, ResponseFormatUndefined, ToAnnotationRoot } from "../annotation.cjs";
+import { JsonSchemaFormat, ProviderStrategy, ResponseFormat, ToolStrategy, TypedToolStrategy } from "../responses.cjs";
 import { ToolNode } from "../nodes/ToolNode.cjs";
-import { AgentRuntime, ClientTool, ServerTool } from "../types.cjs";
+import { ClientTool, ServerTool } from "../types.cjs";
 import { LanguageModelLike } from "@langchain/core/language_models/base";
-import { z } from "zod/v3";
-import { InferInteropZodInput, InteropZodObject, InteropZodType } from "@langchain/core/utils/types";
-import { LangGraphRunnableConfig, START } from "@langchain/langgraph";
-import { BaseMessage, BaseMessageLike, SystemMessage } from "@langchain/core/messages";
-import { All, BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
-import { Runnable } from "@langchain/core/runnables";
+import { InferInteropZodInput, InferInteropZodOutput, InteropZodDefault, InteropZodObject, InteropZodOptional, InteropZodType } from "@langchain/core/utils/types";
+import { BaseMessage } from "@langchain/core/messages";
+import { LangGraphRunnableConfig, PregelOptions, Runtime, START } from "@langchain/langgraph";
+import { BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
+import { Messages as Messages$1 } from "@langchain/langgraph/";
 
 //#region src/agents/middlewareAgent/types.d.ts
-type N = typeof START | "model_request" | "tools";
-interface BuiltInState {
-  messages: BaseMessage[];
-}
+
 /**
- * Information about a tool call that has been executed.
+ * Represents information about an interrupt.
  */
-interface ToolCall {
+interface Interrupt<TValue = unknown> {
   /**
-   * The ID of the tool call.
+   * The ID of the interrupt.
    */
   id: string;
   /**
-   * The name of the tool that was called.
+   * The requests for human input.
    */
-  name: string;
+  value: TValue;
+}
+interface BuiltInState {
+  messages: BaseMessage[];
+  __interrupt__?: Interrupt[];
   /**
-   * The arguments that were passed to the tool.
+   * Optional property to control routing after afterModel middleware execution.
+   * When set by middleware, the agent will jump to the specified node instead of
+   * following normal routing logic. The property is automatically cleared after use.
+   *
+   * - "model_request": Jump back to the model for another LLM call
+   * - "tools": Jump to tool execution (requires tools to be available)
    */
-  args: Record<string, any>;
+  jumpTo?: JumpToTarget;
 }
 /**
- * Information about a tool result from a tool execution.
+ * Base input type for `.invoke` and `.stream` methods.
  */
-interface ToolResult {
-  /**
-   * The ID of the tool call.
-   */
-  id: string;
-  /**
-   * The result of the tool call.
-   */
-  result: any;
-  /**
-   * An optional error message if the tool call failed.
-   */
-  error?: string;
-}
+type UserInput = {
+  messages: Messages$1;
+};
+/**
+ * Information about a tool call that has been executed.
+ */
+
 /**
  * Configuration for modifying a model call at runtime.
  * All fields are optional and only provided fields will override defaults.
@@ -65,7 +64,7 @@ interface ModelRequest {
   /**
    * The system message for this step.
    */
-  systemMessage?: BaseMessage;
+  systemPrompt?: string;
   /**
    * Tool choice configuration (model-specific format).
    * Can be one of:
@@ -82,40 +81,27 @@ interface ModelRequest {
   };
   /**
    * The tools to make available for this step.
-   * Can be tool names (strings) or tool instances.
    */
-  tools: (ClientTool | ServerTool)[];
+  tools: (ServerTool | ClientTool)[];
 }
 /**
- * Runtime information available to middleware (readonly).
+ * Type helper to check if TContext is an optional Zod schema
  */
-interface Runtime<TContext = unknown> {
-  readonly toolCalls: ToolCall[];
-  readonly toolResults: ToolResult[];
-  // readonly tokenUsage: {
-  //   readonly inputTokens: number;
-  //   readonly outputTokens: number;
-  //   readonly totalTokens: number;
-  // };
+type IsOptionalZodObject<T> = T extends InteropZodOptional<any> ? true : false;
+type IsDefaultZodObject<T> = T extends InteropZodDefault<any> ? true : false;
+type WithMaybeContext<TContext> = undefined extends TContext ? {
+  readonly context?: TContext;
+} : IsOptionalZodObject<TContext> extends true ? {
+  readonly context?: TContext;
+} : IsDefaultZodObject<TContext> extends true ? {
+  readonly context?: TContext;
+} : {
   readonly context: TContext;
-}
-/**
- * Control flow interface for middleware.
- */
-interface Controls<TState = unknown> {
-  jumpTo(target: "model" | "tools", stateUpdate?: Partial<TState>): ControlAction<TState>;
-  terminate(result?: Partial<TState> | Error): ControlAction<TState>;
-}
+};
 /**
- * Control action type returned by control methods.
+ * Runtime information available to middleware (readonly).
  */
-type ControlAction<TStateSchema> = {
-  type: "jump" | "terminate" | "retry";
-  target?: string;
-  stateUpdate?: Partial<TStateSchema>;
-  result?: any;
-  error?: Error;
-};
+type Runtime$1<TContext = unknown> = Partial<Omit<Runtime<TContext>, "context" | "configurable">> & WithMaybeContext<TContext> & PrivateState;
 /**
  * Result type for middleware functions.
  */
@@ -134,12 +120,12 @@ type FilterPrivateProps<T> = { [K in keyof T as K extends `_${string}` ? never :
  * Helper type to infer the state schema type from a middleware
  * This filters out private properties (those starting with underscore)
  */
-type InferMiddlewareState<T extends AgentMiddleware<any, any, any>> = T extends AgentMiddleware<infer S, any, any> ? S extends z.ZodObject<any> ? FilterPrivateProps<z.infer<S>> : {} : {};
+type InferMiddlewareState<T extends AgentMiddleware<any, any, any>> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodOutput<S>> : {} : {};
 /**
  * Helper type to infer the input state schema type from a middleware (all properties optional)
  * This filters out private properties (those starting with underscore)
  */
-type InferMiddlewareInputState<T extends AgentMiddleware<any, any, any>> = T extends AgentMiddleware<infer S, any, any> ? S extends z.ZodObject<any> ? FilterPrivateProps<z.input<S>> : {} : {};
+type InferMiddlewareInputState<T extends AgentMiddleware<any, any, any>> = T extends AgentMiddleware<infer S, any, any> ? S extends InteropZodObject ? FilterPrivateProps<InferInteropZodInput<S>> : {} : {};
 /**
  * Helper type to infer merged state from an array of middleware (just the middleware states)
  */
@@ -155,22 +141,37 @@ type InferMiddlewareInputStates<T extends readonly AgentMiddleware<any, any, any
 /**
  * Helper type to infer the input context schema type from a middleware (with optional defaults)
  */
-type InferMiddlewareContextInput<T extends AgentMiddleware<any, any, any>> = T extends AgentMiddleware<any, infer C, any> ? C extends z.ZodObject<any> ? z.input<C> : {} : {};
+type InferMiddlewareContextInput<T extends AgentMiddleware<any, any, any>> = T extends AgentMiddleware<any, infer C, any> ? C extends InteropZodOptional<infer Inner> ? InferInteropZodInput<Inner> | undefined : C extends InteropZodObject ? InferInteropZodInput<C> : {} : {};
 /**
  * Helper type to infer merged context from an array of middleware
  */
 
+/**
+ * Helper to merge two context types, preserving undefined unions
+ */
+type MergeContextTypes<A, B> = [A] extends [undefined] ? [B] extends [undefined] ? undefined : B | undefined : [B] extends [undefined] ? A | undefined : [A] extends [B] ? A : [B] extends [A] ? B : A & B;
 /**
  * Helper type to infer merged input context from an array of middleware (with optional defaults)
  */
-type InferMiddlewareContextInputs<T extends readonly AgentMiddleware<any, any, any>[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware<any, any, any> ? Rest extends readonly AgentMiddleware<any, any, any>[] ? InferMiddlewareContextInput<First> & InferMiddlewareContextInputs<Rest> : InferMiddlewareContextInput<First> : {} : {};
+type InferMiddlewareContextInputs<T extends readonly AgentMiddleware<any, any, any>[]> = T extends readonly [] ? {} : T extends readonly [infer First, ...infer Rest] ? First extends AgentMiddleware<any, any, any> ? Rest extends readonly AgentMiddleware<any, any, any>[] ? MergeContextTypes<InferMiddlewareContextInput<First>, InferMiddlewareContextInputs<Rest>> : InferMiddlewareContextInput<First> : {} : {};
+/**
+ * jump targets (user facing)
+ */
+type JumpToTarget = (typeof JUMP_TO_TARGETS)[number];
+/**
+ * jump targets (internal)
+ */
+
 /**
  * Base middleware interface.
  */
-interface AgentMiddleware<TSchema extends z.ZodObject<z.ZodRawShape> | undefined = undefined, TContextSchema extends z.ZodObject<z.ZodRawShape> | undefined = undefined, TFullContext = any> {
+interface AgentMiddleware<TSchema extends InteropZodObject | undefined = undefined, TContextSchema extends InteropZodObject | InteropZodDefault<InteropZodObject> | InteropZodOptional<InteropZodObject> | undefined = undefined, TFullContext = any> {
   stateSchema?: TSchema;
   contextSchema?: TContextSchema;
   name: string;
+  beforeModelJumpTo?: JumpToTarget[];
+  afterModelJumpTo?: JumpToTarget[];
+  tools?: (ClientTool | ServerTool)[];
   /**
    * Runs before each LLM call, can modify call parameters, changes are not persistent
    * e.g. if you change `model`, it will only be changed for the next model call
@@ -180,24 +181,29 @@ interface AgentMiddleware<TSchema extends z.ZodObject<z.ZodRawShape> | undefined
    * @param runtime - Runtime context and metadata
    * @returns Modified options or undefined to pass through
    */
-  prepareModelRequest?(request: ModelRequest, state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TFullContext>): Promise<Partial<ModelRequest> | void> | Partial<ModelRequest> | void;
-  beforeModel?(state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TFullContext>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>): Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>>;
-  afterModel?(state: (TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState, runtime: Runtime<TFullContext>, controls: Controls<(TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}) & AgentBuiltInState>): Promise<MiddlewareResult<Partial<TSchema extends z.ZodObject<any> ? z.infer<TSchema> : {}>>>;
+  modifyModelRequest?(request: ModelRequest, state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime$1<TFullContext>): Promise<Partial<ModelRequest> | void> | Partial<ModelRequest> | void;
+  /**
+   * Logic to handle model invocation errors and optionally retry.
+   *
+   * @param error - The exception that occurred during model invocation.
+   * @param request - The original model request that failed.
+   * @param state - The current agent state.
+   * @param runtime - The runtime context.
+   * @param attempt - The current attempt number (1-indexed).
+   * @returns Modified request to retry with, or undefined/null to propagate the error (re-raise).
+   */
+  retryModelRequest?(error: Error, request: ModelRequest, state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime$1<TFullContext>, attempt: number): Promise<ModelRequest | void> | ModelRequest | void;
+  beforeModel?(state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime$1<TFullContext>): Promise<MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>>;
+  afterModel?(state: (TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}) & AgentBuiltInState, runtime: Runtime$1<TFullContext>): Promise<MiddlewareResult<Partial<TSchema extends InteropZodObject ? InferInteropZodInput<TSchema> : {}>>>;
 }
 /**
  * Information about a tool call that has been executed.
  */
 
-/**
- * Duplicate of the Prompt type from ../types.ts
- */
-type Prompt<ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot> = SystemMessage | string | ((state: AgentBuiltInState, config: LangGraphRunnableConfig<ToAnnotationRoot<ContextSchema>["State"]>) => BaseMessageLike[] | Promise<BaseMessageLike[]>) | Runnable;
-type DynamicLLMFunction<ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot> = (state: AgentBuiltInState & PreHookAnnotation["State"], runtime: AgentRuntime<ToAnnotationRoot<ContextSchema>["State"]>) => Promise<LanguageModelLike> | LanguageModelLike;
 type CreateAgentParams<StructuredResponseType extends Record<string, any> = Record<string, any>, ContextSchema extends AnyAnnotationRoot | InteropZodObject = AnyAnnotationRoot, ResponseFormatType = InteropZodType<StructuredResponseType> | InteropZodType<unknown>[] | JsonSchemaFormat | JsonSchemaFormat[] | ResponseFormat | TypedToolStrategy<StructuredResponseType> | ToolStrategy<StructuredResponseType> | ProviderStrategy<StructuredResponseType> | ResponseFormatUndefined> = {
-  /** The chat model that can utilize OpenAI-style tool calling. */
-  llm?: LanguageModelLike | DynamicLLMFunction<ContextSchema>;
   /**
-   * Initializes a ChatModel based on the provided model name and provider.
+   * Defines a model to use for the agent. You can either pass in an instance of a LangChain chat model
+   * or a string. If a string is provided the agent initializes a ChatModel based on the provided model name and provider.
    * It supports various model providers and allows for runtime configuration of model parameters.
    *
    * @uses {@link initChatModel}
@@ -208,9 +214,38 @@ type CreateAgentParams<StructuredResponseType extends Record<string, any> = Reco
    *   // ...
    * });
    * ```
+   *
+   * @example
+   * ```ts
+   * import { ChatOpenAI } from "@langchain/openai";
+   * const agent = createAgent({
+   *   model: new ChatOpenAI({ model: "gpt-4o" }),
+   *   // ...
+   * });
+   * ```
+   */
+  model: string | LanguageModelLike;
+  /**
+   * A list of tools or a ToolNode.
+   *
+   * @example
+   * ```ts
+   * import { tool } from "langchain";
+   *
+   * const weatherTool = tool(() => "Sunny!", {
+   *   name: "get_weather",
+   *   description: "Get the weather for a location",
+   *   schema: z.object({
+   *     location: z.string().describe("The location to get weather for"),
+   *   }),
+   * });
+   *
+   * const agent = createAgent({
+   *   tools: [weatherTool],
+   *   // ...
+   * });
+   * ```
    */
-  model?: string;
-  /** A list of tools or a ToolNode. */
   tools?: ToolNode | (ServerTool | ClientTool)[];
   /**
    * An optional prompt for the LLM. This takes full graph state BEFORE the LLM is called and prepares the input to LLM.
@@ -226,9 +261,9 @@ type CreateAgentParams<StructuredResponseType extends Record<string, any> = Reco
    * Prior to `v0.2.46`, the prompt was set using `stateModifier` / `messagesModifier` parameters.
    * This is now deprecated and will be removed in a future release.
    *
-   * Cannot be used together with `prepareModelRequest`.
+   * Cannot be used together with `modifyModelRequest`.
    */
-  prompt?: Prompt<ContextSchema>;
+  systemPrompt?: string;
   /**
    * An optional schema for the context. It allows to pass in a typed context object into the agent
    * invocation and allows to access it in hooks such as `prompt` and middleware.
@@ -267,10 +302,10 @@ type CreateAgentParams<StructuredResponseType extends Record<string, any> = Reco
   checkpointSaver?: BaseCheckpointSaver | boolean;
   /** An optional checkpoint saver to persist the agent's state. Alias of "checkpointSaver". */
   checkpointer?: BaseCheckpointSaver | boolean;
-  /** An optional list of node names to interrupt before running. */
-  interruptBefore?: N[] | All;
-  /** An optional list of node names to interrupt after running. */
-  interruptAfter?: N[] | All;
+  /**
+   * An optional store to persist the agent's state.
+   * @see {@link https://docs.langchain.com/oss/javascript/langgraph/memory#memory-storage | Long-term memory}
+   */
   store?: BaseStore;
   /**
    * An optional schema for the final agent output.
@@ -364,17 +399,74 @@ type CreateAgentParams<StructuredResponseType extends Record<string, any> = Reco
    */
   version?: "v1" | "v2";
 };
+/**
+ * Helper type to check if a type is optional (includes undefined)
+ */
+type IsOptionalType<T> = undefined extends T ? true : false;
+/**
+ * Extract non-undefined part of a union that includes undefined
+ */
+type ExtractNonUndefined<T> = T extends undefined ? never : T;
 /**
  * Helper type to check if all properties of a type are optional
  */
-type IsAllOptional<T> = T extends Record<string, any> ? {} extends T ? true : false : true;
+type IsAllOptional<T> =
+// If T includes undefined, then it's optional (can be omitted entirely)
+undefined extends T ? true : IsOptionalType<T> extends true ? true : ExtractNonUndefined<T> extends Record<string, any> ? {} extends ExtractNonUndefined<T> ? true : false : IsOptionalType<T>;
 /**
  * Helper type to extract input type from context schema (with optional defaults)
  */
 type InferContextInput<ContextSchema extends AnyAnnotationRoot | InteropZodObject> = ContextSchema extends InteropZodObject ? InferInteropZodInput<ContextSchema> : ContextSchema extends AnyAnnotationRoot ? ToAnnotationRoot<ContextSchema>["State"] : {};
 /**
- * Helper type to get the required config type based on context schema
+ * Helper to check if ContextSchema is the default (AnyAnnotationRoot)
+ */
+
+interface RunLevelPrivateState {
+  /**
+   * The number of times the model has been called at the run level.
+   * This includes multiple agent invocations.
+   */
+  runModelCallCount: number;
+}
+interface ThreadLevelPrivateState {
+  /**
+   * The number of times the model has been called at the thread level.
+   * This includes multiple agent invocations within different environments
+   * using the same thread.
+   */
+  threadLevelCallCount: number;
+}
+/**
+ * As private state we consider all information we want to track within
+ * the lifecycle of the agent, without exposing it to the user. These informations
+ * are propagated to the user as _readonly_ runtime properties.
+ */
+interface PrivateState extends ThreadLevelPrivateState, RunLevelPrivateState {}
+/**
+ * Pregel options that are propagated to the agent
+ */
+type CreateAgentPregelOptions = "configurable" | "durability" | "store" | "cache" | "signal" | "recursionLimit" | "maxConcurrency" | "timeout";
+/**
+ * Decide whether provided configuration requires a context
+ */
+type InvokeConfiguration<ContextSchema extends Record<string, any>> =
+/**
+ * If the context schema is a default object, `context` can be optional
+ */
+ContextSchema extends InteropZodDefault<any> ? Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> & {
+  context?: Partial<ContextSchema>;
+} : IsAllOptional<ContextSchema> extends true ? Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> & {
+  context?: Partial<ContextSchema>;
+} : Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> & WithMaybeContext<ContextSchema>;
+type StreamConfiguration<ContextSchema extends Record<string, any>> =
+/**
+ * If the context schema is a default object, `context` can be optional
  */
+ContextSchema extends InteropZodDefault<any> ? Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions>> & {
+  context?: Partial<ContextSchema>;
+} : IsAllOptional<ContextSchema> extends true ? Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions | "streamMode">> & {
+  context?: Partial<ContextSchema>;
+} : Partial<Pick<PregelOptions<any, any, any>, CreateAgentPregelOptions | "streamMode">> & WithMaybeContext<ContextSchema>;
 //#endregion
-export { AgentBuiltInState, AgentMiddleware, BuiltInState, Controls, CreateAgentParams, InferContextInput, InferMiddlewareContextInputs, InferMiddlewareInputStates, InferMiddlewareStates, IsAllOptional, MiddlewareResult, ModelRequest, Runtime };
+export { AgentBuiltInState, AgentMiddleware, BuiltInState, CreateAgentParams, InferContextInput, InferMiddlewareContextInputs, InferMiddlewareInputStates, InferMiddlewareStates, InvokeConfiguration, JumpToTarget, MiddlewareResult, ModelRequest, Runtime$1 as Runtime, StreamConfiguration, UserInput };
 //# sourceMappingURL=types.d.cts.map