deepagents 1.4.0 → 1.4.1

package/dist/index.d.ts

@@ -1,17 +1,18 @@
-import * as langchain0 from "langchain";
-import { AgentMiddleware, AgentMiddleware as AgentMiddleware$1, AgentTypeConfig, InterruptOnConfig, ReactAgent, StructuredTool, ToolMessage } from "langchain";
+import * as langchain9 from "langchain";
+import { AgentMiddleware, AgentMiddleware as AgentMiddleware$1, AgentTypeConfig, InferMiddlewareStates, InterruptOnConfig, ReactAgent, ResponseFormat, StructuredTool, SystemMessage, ToolMessage } from "langchain";
 import * as _Command from "@langchain/langgraph";
 import { AnnotationRoot, Command } from "@langchain/langgraph";
 import { z } from "zod/v4";
 import * as zod_v30 from "zod/v3";
+import { z as z$1 } from "zod/v3";
 import * as _messages from "@langchain/core/messages";
 import * as zod0 from "zod";
 import * as zod_v4_core0 from "zod/v4/core";
 import { BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
 import * as _langchain_core_language_models_base0 from "@langchain/core/language_models/base";
 import { BaseLanguageModel, LanguageModelLike } from "@langchain/core/language_models/base";
-import * as _langchain_core_tools1 from "@langchain/core/tools";
-import { StructuredTool as StructuredTool$1 } from "@langchain/core/tools";
+import * as _langchain_core_tools3 from "@langchain/core/tools";
+import { ClientTool, ServerTool, StructuredTool as StructuredTool$1 } from "@langchain/core/tools";
 import { Runnable } from "@langchain/core/runnables";
 import { InteropZodObject } from "@langchain/core/utils/types";
 
@@ -306,13 +307,13 @@ declare function createFilesystemMiddleware(options?: FilesystemMiddlewareOption
     created_at: z.ZodString;
     modified_at: z.ZodString;
   }, z.core.$strip>>>;
-}, z.core.$strip>, undefined, unknown, (langchain0.DynamicStructuredTool<z.ZodObject<{
+}, z.core.$strip>, undefined, unknown, (langchain9.DynamicStructuredTool<z.ZodObject<{
   path: z.ZodDefault<z.ZodOptional<z.ZodString>>;
 }, z.core.$strip>, {
   path: string;
 }, {
   path?: string | undefined;
-}, string, "ls"> | langchain0.DynamicStructuredTool<z.ZodObject<{
+}, string, "ls"> | langchain9.DynamicStructuredTool<z.ZodObject<{
   file_path: z.ZodString;
   offset: z.ZodDefault<z.ZodOptional<z.ZodCoercedNumber<unknown>>>;
   limit: z.ZodDefault<z.ZodOptional<z.ZodCoercedNumber<unknown>>>;
@@ -324,7 +325,7 @@ declare function createFilesystemMiddleware(options?: FilesystemMiddlewareOption
   file_path: string;
   offset?: unknown;
   limit?: unknown;
-}, string, "read_file"> | langchain0.DynamicStructuredTool<z.ZodObject<{
+}, string, "read_file"> | langchain9.DynamicStructuredTool<z.ZodObject<{
   file_path: z.ZodString;
   content: z.ZodString;
 }, z.core.$strip>, {
@@ -336,7 +337,7 @@ declare function createFilesystemMiddleware(options?: FilesystemMiddlewareOption
 }, string | ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>> | Command<unknown, {
   files: Record<string, FileData>;
   messages: ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>>[];
-}, string>, "write_file"> | langchain0.DynamicStructuredTool<z.ZodObject<{
+}, string>, "write_file"> | langchain9.DynamicStructuredTool<z.ZodObject<{
   file_path: z.ZodString;
   old_string: z.ZodString;
   new_string: z.ZodString;
@@ -354,7 +355,7 @@ declare function createFilesystemMiddleware(options?: FilesystemMiddlewareOption
 }, string | ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>> | Command<unknown, {
   files: Record<string, FileData>;
   messages: ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>>[];
-}, string>, "edit_file"> | langchain0.DynamicStructuredTool<z.ZodObject<{
+}, string>, "edit_file"> | langchain9.DynamicStructuredTool<z.ZodObject<{
   pattern: z.ZodString;
   path: z.ZodDefault<z.ZodOptional<z.ZodString>>;
 }, z.core.$strip>, {
@@ -363,7 +364,7 @@ declare function createFilesystemMiddleware(options?: FilesystemMiddlewareOption
 }, {
   pattern: string;
   path?: string | undefined;
-}, string, "glob"> | langchain0.DynamicStructuredTool<z.ZodObject<{
+}, string, "glob"> | langchain9.DynamicStructuredTool<z.ZodObject<{
   pattern: z.ZodString;
   path: z.ZodDefault<z.ZodOptional<z.ZodString>>;
   glob: z.ZodNullable<z.ZodOptional<z.ZodString>>;
@@ -375,7 +376,7 @@ declare function createFilesystemMiddleware(options?: FilesystemMiddlewareOption
   pattern: string;
   path?: string | undefined;
   glob?: string | null | undefined;
-}, string, "grep"> | langchain0.DynamicStructuredTool<z.ZodObject<{
+}, string, "grep"> | langchain9.DynamicStructuredTool<z.ZodObject<{
   command: z.ZodString;
 }, z.core.$strip>, {
   command: string;
@@ -410,7 +411,7 @@ interface SubAgent {
   /** The model for the agent. Defaults to default_model */
   model?: LanguageModelLike | string;
   /** Additional middleware to append after default_middleware */
-  middleware?: AgentMiddleware$1[];
+  middleware?: readonly AgentMiddleware$1[];
   /** The tool configs to use for the agent */
   interruptOn?: Record<string, boolean | InterruptOnConfig>;
 }
@@ -438,7 +439,22 @@ interface SubAgentMiddlewareOptions {
 /**
  * Create subagent middleware with task tool
  */
-declare function createSubAgentMiddleware(options: SubAgentMiddlewareOptions): AgentMiddleware$1;
+declare function createSubAgentMiddleware(options: SubAgentMiddlewareOptions): AgentMiddleware$1<undefined, undefined, unknown, readonly [langchain9.DynamicStructuredTool<z$1.ZodObject<{
+  description: z$1.ZodString;
+  subagent_type: z$1.ZodString;
+}, "strip", z$1.ZodTypeAny, {
+  description: string;
+  subagent_type: string;
+}, {
+  description: string;
+  subagent_type: string;
+}>, {
+  description: string;
+  subagent_type: string;
+}, {
+  description: string;
+  subagent_type: string;
+}, string | Command<unknown, Record<string, unknown>, string>, "task">]>;
 //#endregion
 //#region src/middleware/patch_tool_calls.d.ts
 /**
@@ -461,7 +477,7 @@ declare function createSubAgentMiddleware(options: SubAgentMiddlewareOptions): A
  * });
  * ```
  */
-declare function createPatchToolCallsMiddleware(): AgentMiddleware<undefined, undefined, unknown, readonly (_langchain_core_tools1.ClientTool | _langchain_core_tools1.ServerTool)[]>;
+declare function createPatchToolCallsMiddleware(): AgentMiddleware<undefined, undefined, unknown, readonly (_langchain_core_tools3.ClientTool | _langchain_core_tools3.ServerTool)[]>;
 //#endregion
 //#region src/backends/state.d.ts
 /**
@@ -918,24 +934,49 @@ declare abstract class BaseSandbox implements SandboxBackendProtocol {
   edit(filePath: string, oldString: string, newString: string, replaceAll?: boolean): Promise<EditResult>;
 }
 //#endregion
-//#region src/agent.d.ts
+//#region src/types.d.ts
+/**
+ * Helper type to extract middleware from a SubAgent definition
+ * Handles both mutable and readonly middleware arrays
+ */
+type ExtractSubAgentMiddleware<T> = T extends {
+  middleware?: infer M;
+} ? M extends readonly AgentMiddleware[] ? M : M extends AgentMiddleware[] ? M : readonly [] : readonly [];
+/**
+ * Helper type to flatten and merge middleware from all subagents
+ */
+type FlattenSubAgentMiddleware<T extends readonly (SubAgent | CompiledSubAgent)[]> = T extends readonly [] ? readonly [] : T extends readonly [infer First, ...infer Rest] ? Rest extends readonly (SubAgent | CompiledSubAgent)[] ? readonly [...ExtractSubAgentMiddleware<First>, ...FlattenSubAgentMiddleware<Rest>] : ExtractSubAgentMiddleware<First> : readonly [];
+/**
+ * Helper type to merge states from subagent middleware
+ */
+type InferSubAgentMiddlewareStates<T extends readonly (SubAgent | CompiledSubAgent)[]> = InferMiddlewareStates<FlattenSubAgentMiddleware<T>>;
+/**
+ * Combined state type including custom middleware and subagent middleware states
+ */
+type MergedDeepAgentState<TMiddleware extends readonly AgentMiddleware[], TSubagents extends readonly (SubAgent | CompiledSubAgent)[]> = InferMiddlewareStates<TMiddleware> & InferSubAgentMiddlewareStates<TSubagents>;
 /**
  * Configuration parameters for creating a Deep Agent
  * Matches Python's create_deep_agent parameters
+ *
+ * @typeParam TResponse - The structured response type when using responseFormat
+ * @typeParam ContextSchema - The context schema type
+ * @typeParam TMiddleware - The middleware array type for proper type inference
+ * @typeParam TSubagents - The subagents array type for extracting subagent middleware states
+ * @typeParam TTools - The tools array type
  */
-interface CreateDeepAgentParams<ContextSchema extends AnnotationRoot<any> | InteropZodObject = AnnotationRoot<any>> {
+interface CreateDeepAgentParams<TResponse extends ResponseFormat = ResponseFormat, ContextSchema extends AnnotationRoot<any> | InteropZodObject = AnnotationRoot<any>, TMiddleware extends readonly AgentMiddleware[] = readonly AgentMiddleware[], TSubagents extends readonly (SubAgent | CompiledSubAgent)[] = readonly (SubAgent | CompiledSubAgent)[], TTools extends readonly (ClientTool | ServerTool)[] = readonly (ClientTool | ServerTool)[]> {
   /** The model to use (model name string or LanguageModelLike instance). Defaults to claude-sonnet-4-5-20250929 */
   model?: BaseLanguageModel | string;
   /** Tools the agent should have access to */
-  tools?: StructuredTool$1[];
+  tools?: TTools | StructuredTool$1[];
   /** Custom system prompt for the agent. This will be combined with the base agent prompt */
-  systemPrompt?: string;
+  systemPrompt?: string | SystemMessage;
   /** Custom middleware to apply after standard middleware */
-  middleware?: AgentMiddleware[];
+  middleware?: TMiddleware;
   /** List of subagent specifications for task delegation */
-  subagents?: (SubAgent | CompiledSubAgent)[];
-  /** Structured output response format for the agent */
-  responseFormat?: any;
+  subagents?: TSubagents;
+  /** Structured output response format for the agent (Zod schema or other format) */
+  responseFormat?: TResponse;
   /** Optional schema for context (not persisted between invocations) */
   contextSchema?: ContextSchema;
   /** Optional checkpointer for persisting agent state between runs */
@@ -956,6 +997,8 @@ interface CreateDeepAgentParams<ContextSchema extends AnnotationRoot<any> | Inte
   /** The name of the agent */
   name?: string;
 }
+//#endregion
+//#region src/agent.d.ts
 /**
  * Create a Deep Agent with middleware-based architecture.
  *
@@ -969,9 +1012,25 @@ interface CreateDeepAgentParams<ContextSchema extends AnnotationRoot<any> | Inte
  * - Human-in-the-loop (humanInTheLoopMiddleware) - optional
  *
  * @param params Configuration parameters for the agent
- * @returns ReactAgent instance ready for invocation
+ * @returns ReactAgent instance ready for invocation with properly inferred state types
+ *
+ * @example
+ * ```typescript
+ * // Middleware with custom state
+ * const ResearchMiddleware = createMiddleware({
+ *   name: "ResearchMiddleware",
+ *   stateSchema: z.object({ research: z.string().default("") }),
+ * });
+ *
+ * const agent = createDeepAgent({
+ *   middleware: [ResearchMiddleware],
+ * });
+ *
+ * const result = await agent.invoke({ messages: [...] });
+ * // result.research is properly typed as string
+ * ```
  */
-declare function createDeepAgent<ContextSchema extends InteropZodObject = InteropZodObject>(params?: CreateDeepAgentParams<ContextSchema>): ReactAgent<AgentTypeConfig<Record<string, any>, undefined, ContextSchema, (AgentMiddleware<any, any, any, readonly (_langchain_core_tools1.ClientTool | _langchain_core_tools1.ServerTool)[]> | AgentMiddleware<zod_v30.ZodObject<{
+declare function createDeepAgent<TResponse extends ResponseFormat = ResponseFormat, ContextSchema extends InteropZodObject = InteropZodObject, const TMiddleware extends readonly AgentMiddleware[] = readonly [], const TSubagents extends readonly (SubAgent | CompiledSubAgent)[] = readonly [], const TTools extends readonly (ClientTool | ServerTool)[] = readonly []>(params?: CreateDeepAgentParams<TResponse, ContextSchema, TMiddleware, TSubagents, TTools>): ReactAgent<AgentTypeConfig<TResponse, undefined, ContextSchema, readonly [AgentMiddleware<zod_v30.ZodObject<{
   todos: zod_v30.ZodDefault<zod_v30.ZodArray<zod_v30.ZodObject<{
     content: zod_v30.ZodString;
     status: zod_v30.ZodEnum<["pending", "in_progress", "completed"]>;
@@ -992,7 +1051,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
     content: string;
     status: "completed" | "in_progress" | "pending";
   }[] | undefined;
-}>, undefined, unknown, readonly [_langchain_core_tools1.DynamicStructuredTool<zod_v30.ZodObject<{
+}>, undefined, unknown, readonly [langchain9.DynamicStructuredTool<zod_v30.ZodObject<{
   todos: zod_v30.ZodArray<zod_v30.ZodObject<{
     content: zod_v30.ZodString;
     status: zod_v30.ZodEnum<["pending", "in_progress", "completed"]>;
@@ -1029,19 +1088,19 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
     status: "completed" | "in_progress" | "pending";
   }[];
   messages: _messages.ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>>[];
-}, string>, "write_todos">]> | AgentMiddleware<zod0.ZodObject<{
+}, string>, "write_todos">]>, AgentMiddleware<zod0.ZodObject<{
   files: zod0.ZodDefault<zod0.ZodRecord<zod0.ZodString, zod0.ZodObject<{
     content: zod0.ZodArray<zod0.ZodString>;
     created_at: zod0.ZodString;
     modified_at: zod0.ZodString;
   }, zod_v4_core0.$strip>>>;
-}, zod_v4_core0.$strip>, undefined, unknown, (_langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, zod_v4_core0.$strip>, undefined, unknown, (langchain9.DynamicStructuredTool<zod0.ZodObject<{
   path: zod0.ZodDefault<zod0.ZodOptional<zod0.ZodString>>;
 }, zod_v4_core0.$strip>, {
   path: string;
 }, {
   path?: string | undefined;
-}, string, "ls"> | _langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, string, "ls"> | langchain9.DynamicStructuredTool<zod0.ZodObject<{
   file_path: zod0.ZodString;
   offset: zod0.ZodDefault<zod0.ZodOptional<zod0.ZodCoercedNumber<unknown>>>;
   limit: zod0.ZodDefault<zod0.ZodOptional<zod0.ZodCoercedNumber<unknown>>>;
@@ -1053,7 +1112,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   file_path: string;
   offset?: unknown;
   limit?: unknown;
-}, string, "read_file"> | _langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, string, "read_file"> | langchain9.DynamicStructuredTool<zod0.ZodObject<{
   file_path: zod0.ZodString;
   content: zod0.ZodString;
 }, zod_v4_core0.$strip>, {
@@ -1065,7 +1124,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
 }, string | _messages.ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>> | _Command.Command<unknown, {
   files: Record<string, FileData>;
   messages: _messages.ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>>[];
-}, string>, "write_file"> | _langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, string>, "write_file"> | langchain9.DynamicStructuredTool<zod0.ZodObject<{
   file_path: zod0.ZodString;
   old_string: zod0.ZodString;
   new_string: zod0.ZodString;
@@ -1083,7 +1142,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
 }, string | _messages.ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>> | _Command.Command<unknown, {
   files: Record<string, FileData>;
   messages: _messages.ToolMessage<_messages.MessageStructure<_messages.MessageToolSet>>[];
-}, string>, "edit_file"> | _langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, string>, "edit_file"> | langchain9.DynamicStructuredTool<zod0.ZodObject<{
   pattern: zod0.ZodString;
   path: zod0.ZodDefault<zod0.ZodOptional<zod0.ZodString>>;
 }, zod_v4_core0.$strip>, {
@@ -1092,7 +1151,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
 }, {
   pattern: string;
   path?: string | undefined;
-}, string, "glob"> | _langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, string, "glob"> | langchain9.DynamicStructuredTool<zod0.ZodObject<{
   pattern: zod0.ZodString;
   path: zod0.ZodDefault<zod0.ZodOptional<zod0.ZodString>>;
   glob: zod0.ZodNullable<zod0.ZodOptional<zod0.ZodString>>;
@@ -1104,13 +1163,28 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   pattern: string;
   path?: string | undefined;
   glob?: string | null | undefined;
-}, string, "grep"> | _langchain_core_tools1.DynamicStructuredTool<zod0.ZodObject<{
+}, string, "grep"> | langchain9.DynamicStructuredTool<zod0.ZodObject<{
   command: zod0.ZodString;
 }, zod_v4_core0.$strip>, {
   command: string;
 }, {
   command: string;
-}, string, "execute">)[]> | AgentMiddleware<undefined, zod_v30.ZodObject<{
+}, string, "execute">)[]>, AgentMiddleware<undefined, undefined, unknown, readonly [langchain9.DynamicStructuredTool<zod_v30.ZodObject<{
+  description: zod_v30.ZodString;
+  subagent_type: zod_v30.ZodString;
+}, "strip", zod_v30.ZodTypeAny, {
+  description: string;
+  subagent_type: string;
+}, {
+  description: string;
+  subagent_type: string;
+}>, {
+  description: string;
+  subagent_type: string;
+}, {
+  description: string;
+  subagent_type: string;
+}, string | _Command.Command<unknown, Record<string, unknown>, string>, "task">]>, AgentMiddleware<undefined, zod_v30.ZodObject<{
   trigger: zod_v30.ZodOptional<zod_v30.ZodUnion<[zod_v30.ZodEffects<zod_v30.ZodObject<{
     fraction: zod_v30.ZodOptional<zod_v30.ZodNumber>;
     tokens: zod_v30.ZodOptional<zod_v30.ZodNumber>;
@@ -1180,7 +1254,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   maxTokensBeforeSummary: zod_v30.ZodOptional<zod_v30.ZodNumber>;
   messagesToKeep: zod_v30.ZodOptional<zod_v30.ZodNumber>;
 } & {
-  model: zod_v30.ZodOptional<zod_v30.ZodType<BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions>, zod_v30.ZodTypeDef, BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions>>>;
+  model: zod_v30.ZodOptional<zod_v30.ZodType<_langchain_core_language_models_base0.BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions>, zod_v30.ZodTypeDef, _langchain_core_language_models_base0.BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions>>>;
 }, "strip", zod_v30.ZodTypeAny, {
   trigger?: {
     fraction?: number | undefined;
@@ -1202,7 +1276,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   summaryPrefix?: string | undefined;
   maxTokensBeforeSummary?: number | undefined;
   messagesToKeep?: number | undefined;
-  model?: BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions> | undefined;
+  model?: _langchain_core_language_models_base0.BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions> | undefined;
 }, {
   trigger?: {
     fraction?: number | undefined;
@@ -1224,7 +1298,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   summaryPrefix?: string | undefined;
   maxTokensBeforeSummary?: number | undefined;
   messagesToKeep?: number | undefined;
-  model?: BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions> | undefined;
+  model?: _langchain_core_language_models_base0.BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions> | undefined;
 }>, {
   trigger?: {
     fraction?: number | undefined;
@@ -1246,8 +1320,8 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   summaryPrefix?: string | undefined;
   maxTokensBeforeSummary?: number | undefined;
   messagesToKeep?: number | undefined;
-  model?: BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions> | undefined;
-}, readonly (_langchain_core_tools1.ClientTool | _langchain_core_tools1.ServerTool)[]> | AgentMiddleware<undefined, zod_v30.ZodObject<{
+  model?: _langchain_core_language_models_base0.BaseLanguageModel<any, _langchain_core_language_models_base0.BaseLanguageModelCallOptions> | undefined;
+}, readonly (ClientTool | ServerTool)[]>, AgentMiddleware<undefined, zod_v30.ZodObject<{
   enableCaching: zod_v30.ZodOptional<zod_v30.ZodBoolean>;
   ttl: zod_v30.ZodOptional<zod_v30.ZodEnum<["5m", "1h"]>>;
   minMessagesToCache: zod_v30.ZodOptional<zod_v30.ZodNumber>;
@@ -1267,7 +1341,7 @@ declare function createDeepAgent<ContextSchema extends InteropZodObject = Intero
   ttl?: "1h" | "5m" | undefined;
   minMessagesToCache?: number | undefined;
   unsupportedModelBehavior?: "ignore" | "raise" | "warn" | undefined;
-}, readonly (_langchain_core_tools1.ClientTool | _langchain_core_tools1.ServerTool)[]> | AgentMiddleware<undefined, undefined, unknown, readonly (_langchain_core_tools1.ClientTool | _langchain_core_tools1.ServerTool)[]>)[], readonly StructuredTool$1<_langchain_core_tools1.ToolSchemaBase, any, any, any>[]>>;
+}, readonly (ClientTool | ServerTool)[]>, AgentMiddleware<undefined, undefined, unknown, readonly (ClientTool | ServerTool)[]>, ...TMiddleware, ...FlattenSubAgentMiddleware<TSubagents>], TTools>>;
 //#endregion
 //#region src/config.d.ts
 /**
@@ -1398,7 +1472,7 @@ interface SkillsMiddlewareOptions {
  * @param options - Configuration options
  * @returns AgentMiddleware for skills loading and injection
  */
-declare function createSkillsMiddleware(options: SkillsMiddlewareOptions): AgentMiddleware;
+declare function createSkillsMiddleware(options: SkillsMiddlewareOptions): AgentMiddleware<any, undefined, unknown, readonly (_langchain_core_tools3.ClientTool | _langchain_core_tools3.ServerTool)[]>;
 //#endregion
 //#region src/middleware/agent-memory.d.ts
 /**
@@ -1422,7 +1496,7 @@ interface AgentMemoryMiddlewareOptions {
  * @param options - Configuration options
  * @returns AgentMiddleware for memory loading and injection
  */
-declare function createAgentMemoryMiddleware(options: AgentMemoryMiddlewareOptions): AgentMiddleware<any, undefined, unknown, readonly (_langchain_core_tools1.ClientTool | _langchain_core_tools1.ServerTool)[]>;
+declare function createAgentMemoryMiddleware(options: AgentMemoryMiddlewareOptions): AgentMiddleware<any, undefined, unknown, readonly (_langchain_core_tools3.ClientTool | _langchain_core_tools3.ServerTool)[]>;
 //#endregion
 //#region src/skills/loader.d.ts
 /**
@@ -1506,5 +1580,5 @@ declare function parseSkillMetadata(skillMdPath: string, source: "user" | "proje
  */
 declare function listSkills(options: ListSkillsOptions): SkillMetadata[];
 //#endregion
-export { type AgentMemoryMiddlewareOptions, type BackendFactory, type BackendProtocol, BaseSandbox, type CompiledSubAgent, CompositeBackend, type CreateDeepAgentParams, type EditResult, type ExecuteResponse, type FileData, type FileDownloadResponse, type FileInfo, type FileOperationError, type FileUploadResponse, FilesystemBackend, type FilesystemMiddlewareOptions, type GrepMatch, type ListSkillsOptions, MAX_SKILL_DESCRIPTION_LENGTH, MAX_SKILL_FILE_SIZE, MAX_SKILL_NAME_LENGTH, type MaybePromise, type SandboxBackendProtocol, type Settings, type SettingsOptions, type SkillMetadata, type SkillsMiddlewareOptions, StateBackend, StoreBackend, type SubAgent, type SubAgentMiddlewareOptions, type WriteResult, createAgentMemoryMiddleware, createDeepAgent, createFilesystemMiddleware, createPatchToolCallsMiddleware, createSettings, createSkillsMiddleware, createSubAgentMiddleware, findProjectRoot, isSandboxBackend, listSkills, parseSkillMetadata };
+export { type AgentMemoryMiddlewareOptions, type BackendFactory, type BackendProtocol, BaseSandbox, type CompiledSubAgent, CompositeBackend, type CreateDeepAgentParams, type EditResult, type ExecuteResponse, type FileData, type FileDownloadResponse, type FileInfo, type FileOperationError, type FileUploadResponse, FilesystemBackend, type FilesystemMiddlewareOptions, type GrepMatch, type ListSkillsOptions, MAX_SKILL_DESCRIPTION_LENGTH, MAX_SKILL_FILE_SIZE, MAX_SKILL_NAME_LENGTH, type MaybePromise, type MergedDeepAgentState, type SandboxBackendProtocol, type Settings, type SettingsOptions, type SkillMetadata, type SkillsMiddlewareOptions, StateBackend, StoreBackend, type SubAgent, type SubAgentMiddlewareOptions, type WriteResult, createAgentMemoryMiddleware, createDeepAgent, createFilesystemMiddleware, createPatchToolCallsMiddleware, createSettings, createSkillsMiddleware, createSubAgentMiddleware, findProjectRoot, isSandboxBackend, listSkills, parseSkillMetadata };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.js

@@ -1,4 +1,4 @@
-import { AIMessage, ToolMessage, anthropicPromptCachingMiddleware, createAgent, createMiddleware, humanInTheLoopMiddleware, summarizationMiddleware, todoListMiddleware, tool } from "langchain";
+import { AIMessage, SystemMessage, ToolMessage, anthropicPromptCachingMiddleware, createAgent, createMiddleware, humanInTheLoopMiddleware, summarizationMiddleware, todoListMiddleware, tool } from "langchain";
 import { Command, REMOVE_ALL_MESSAGES, getCurrentTaskInput, isCommand } from "@langchain/langgraph";
 import { z } from "zod/v4";
 import micromatch from "micromatch";
@@ -810,7 +810,7 @@ function createFilesystemMiddleware(options = {}) {
 					filesUpdate: null
 				};
 			}
-			if (result instanceof ToolMessage) {
+			if (ToolMessage.isInstance(result)) {
 				const processed = await processToolMessage(result);
 				if (processed.filesUpdate) return new Command({ update: {
 					files: processed.filesUpdate,
@@ -824,7 +824,7 @@ function createFilesystemMiddleware(options = {}) {
 				let hasLargeResults = false;
 				const accumulatedFiles = { ...update.files || {} };
 				const processedMessages = [];
-				for (const msg of update.messages) if (msg instanceof ToolMessage) {
+				for (const msg of update.messages) if (ToolMessage.isInstance(msg)) {
 					const processed = await processToolMessage(msg);
 					processedMessages.push(processed.message);
 					if (processed.filesUpdate) {
@@ -2638,13 +2638,35 @@ const BASE_PROMPT = `In order to complete the objective that the user asks of yo
 * - Human-in-the-loop (humanInTheLoopMiddleware) - optional
 *
 * @param params Configuration parameters for the agent
-* @returns ReactAgent instance ready for invocation
+* @returns ReactAgent instance ready for invocation with properly inferred state types
+*
+* @example
+* ```typescript
+* // Middleware with custom state
+* const ResearchMiddleware = createMiddleware({
+*   name: "ResearchMiddleware",
+*   stateSchema: z.object({ research: z.string().default("") }),
+* });
+*
+* const agent = createDeepAgent({
+*   middleware: [ResearchMiddleware],
+* });
+*
+* const result = await agent.invoke({ messages: [...] });
+* // result.research is properly typed as string
+* ```
 */
 function createDeepAgent(params = {}) {
 	const { model = "claude-sonnet-4-5-20250929", tools = [], systemPrompt, middleware: customMiddleware = [], subagents = [], responseFormat, contextSchema, checkpointer, store, backend, interruptOn, name } = params;
-	const finalSystemPrompt = systemPrompt ? `${systemPrompt}\n\n${BASE_PROMPT}` : BASE_PROMPT;
+	const finalSystemPrompt = systemPrompt ? typeof systemPrompt === "string" ? `${systemPrompt}\n\n${BASE_PROMPT}` : new SystemMessage({ content: [{
+		type: "text",
+		text: BASE_PROMPT
+	}, ...typeof systemPrompt.content === "string" ? [{
+		type: "text",
+		text: systemPrompt.content
+	}] : systemPrompt.content] }) : BASE_PROMPT;
 	const filesystemBackend = backend ? backend : (config) => new StateBackend(config);
-	const middleware = [
+	const builtInMiddleware = [
 		todoListMiddleware(),
 		createFilesystemMiddleware({ backend: filesystemBackend }),
 		createSubAgentMiddleware({
@@ -2673,13 +2695,12 @@ function createDeepAgent(params = {}) {
 		anthropicPromptCachingMiddleware({ unsupportedModelBehavior: "ignore" }),
 		createPatchToolCallsMiddleware()
 	];
-	if (interruptOn) middleware.push(humanInTheLoopMiddleware({ interruptOn }));
-	middleware.push(...customMiddleware);
+	if (interruptOn) builtInMiddleware.push(humanInTheLoopMiddleware({ interruptOn }));
 	return createAgent({
 		model,
 		systemPrompt: finalSystemPrompt,
 		tools,
-		middleware,
+		middleware: [...builtInMiddleware, ...customMiddleware],
 		responseFormat,
 		contextSchema,
 		checkpointer,