@agentforge/core 0.15.6 → 0.15.8

package/dist/index.cjs

@@ -794,21 +794,42 @@ function toolBuilder() {
 // src/langchain/converter.ts
 var import_tools = require("@langchain/core/tools");
 var import_zod_to_json_schema = require("zod-to-json-schema");
+function serializeToolResult(result) {
+  if (typeof result === "string") {
+    return result;
+  }
+  if (typeof result === "object" && result !== null) {
+    return JSON.stringify(result, null, 2);
+  }
+  return String(result);
+}
+function isJsonSchemaObject(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function isJsonSchemaDefinitionMap(value) {
+  if (!isJsonSchemaObject(value)) {
+    return false;
+  }
+  return Object.values(value).every(isJsonSchemaObject);
+}
+function extractToolSchema(jsonSchema) {
+  if (!isJsonSchemaObject(jsonSchema)) {
+    return {};
+  }
+  const ref = jsonSchema.$ref;
+  const definitions = jsonSchema.definitions;
+  if (typeof ref !== "string" || !isJsonSchemaDefinitionMap(definitions)) {
+    return jsonSchema;
+  }
+  const refName = ref.replace("#/definitions/", "");
+  return definitions[refName] ?? jsonSchema;
+}
 function toLangChainTool(tool) {
   return new import_tools.DynamicStructuredTool({
     name: tool.metadata.name,
     description: tool.metadata.description,
     schema: tool.schema,
-    func: async (input) => {
-      const result = await tool.invoke(input);
-      if (typeof result === "string") {
-        return result;
-      }
-      if (typeof result === "object" && result !== null) {
-        return JSON.stringify(result, null, 2);
-      }
-      return String(result);
-    }
+    func: async (input) => serializeToolResult(await tool.invoke(input))
   });
 }
 function toLangChainTools(tools) {
@@ -820,11 +841,7 @@ function getToolJsonSchema(tool) {
     $refStrategy: "none"
     // Don't use $ref for nested schemas
   });
-  if (jsonSchema.$ref && jsonSchema.definitions) {
-    const refName = jsonSchema.$ref.replace("#/definitions/", "");
-    return jsonSchema.definitions[refName] || jsonSchema;
-  }
-  return jsonSchema;
+  return extractToolSchema(jsonSchema);
 }
 function getToolDescription(tool) {
   const { metadata } = tool;
@@ -2194,46 +2211,84 @@ function createToolSimulator(config) {
 
 // src/langgraph/state.ts
 var import_langgraph = require("@langchain/langgraph");
+function entriesOf(value) {
+  return Object.entries(value);
+}
+function keysOf(value) {
+  return Object.keys(value);
+}
+function hasOwnProperty(value, key) {
+  return Object.prototype.hasOwnProperty.call(value, key);
+}
+function useLatestValue(_left, right) {
+  return right;
+}
+function setStateValue(target, key, value) {
+  target[key] = value;
+}
+function setStateDefinitionValue(target, key, value) {
+  target[key] = value;
+}
+function createChannel(channelConfig) {
+  if (channelConfig.reducer) {
+    return (0, import_langgraph.Annotation)({
+      reducer: channelConfig.reducer,
+      default: channelConfig.default
+    });
+  }
+  if (channelConfig.default) {
+    return (0, import_langgraph.Annotation)({
+      reducer: useLatestValue,
+      default: channelConfig.default
+    });
+  }
+  return (0, import_langgraph.Annotation)();
+}
 function createStateAnnotation(config) {
   const stateDefinition = {};
-  for (const [key, channelConfig] of Object.entries(config)) {
-    if (channelConfig.reducer) {
-      stateDefinition[key] = (0, import_langgraph.Annotation)({
-        reducer: channelConfig.reducer,
-        default: channelConfig.default
-      });
-    } else if (channelConfig.default) {
-      stateDefinition[key] = (0, import_langgraph.Annotation)({
-        reducer: (_left, right) => right,
-        default: channelConfig.default
-      });
-    } else {
-      stateDefinition[key] = (0, import_langgraph.Annotation)();
-    }
+  for (const [key, channelConfig] of entriesOf(config)) {
+    setStateDefinitionValue(
+      stateDefinition,
+      key,
+      createChannel(channelConfig)
+    );
   }
   return import_langgraph.Annotation.Root(stateDefinition);
 }
 function validateState(state, config) {
   const validated = {};
-  for (const [key, channelConfig] of Object.entries(config)) {
-    if (channelConfig.schema && key in state) {
-      validated[key] = channelConfig.schema.parse(state[key]);
-    } else if (key in state) {
-      validated[key] = state[key];
+  for (const [key, channelConfig] of entriesOf(config)) {
+    if (channelConfig.schema && hasOwnProperty(state, key)) {
+      setStateValue(
+        validated,
+        key,
+        channelConfig.schema.parse(state[key])
+      );
+    } else if (hasOwnProperty(state, key)) {
+      setStateValue(validated, key, state[key]);
     } else if (channelConfig.default) {
-      validated[key] = channelConfig.default();
+      setStateValue(validated, key, channelConfig.default());
     }
   }
   return validated;
 }
 function mergeState(currentState, update, config) {
   const merged = { ...currentState };
-  for (const [key, value] of Object.entries(update)) {
+  for (const key of keysOf(update)) {
+    const value = update[key];
     const channelConfig = config[key];
-    if (channelConfig?.reducer && key in merged) {
-      merged[key] = channelConfig.reducer(merged[key], value);
+    const reducer = channelConfig?.reducer;
+    if (reducer && hasOwnProperty(merged, key)) {
+      setStateValue(
+        merged,
+        key,
+        reducer(
+          merged[key],
+          value
+        )
+      );
     } else {
-      merged[key] = value;
+      setStateValue(merged, key, value);
     }
   }
   return merged;

package/dist/index.d.cts

@@ -1,6 +1,6 @@
-import { z, ZodType, ZodTypeDef } from 'zod';
+import { z, ZodTypeAny, output, ZodType, ZodTypeDef } from 'zod';
 import { DynamicStructuredTool } from '@langchain/core/tools';
-import { AnnotationRoot, StateDefinition, StateGraph, END, MemorySaver, BaseCheckpointSaver, CheckpointTuple } from '@langchain/langgraph';
+import { AnnotationRoot, BaseChannel, StateGraph, END, MemorySaver, BaseCheckpointSaver, CheckpointTuple } from '@langchain/langgraph';
 import { RunnableConfig } from '@langchain/core/runnables';
 
 /**
@@ -1794,6 +1794,13 @@ declare function createToolSimulator(config: ToolSimulatorConfig): {
  * ```
  */
 
+type RuntimeSchema<TInput = unknown> = z.ZodSchema<TInput>;
+type JsonSchemaObject = Record<string, unknown>;
+interface LangChainConvertibleTool<TInput = unknown, TOutput = unknown> {
+    metadata: ToolMetadata;
+    schema: RuntimeSchema<TInput>;
+    invoke(input: TInput): Promise<TOutput>;
+}
 /**
  * Convert an AgentForge tool to a LangChain DynamicStructuredTool
  *
@@ -1826,7 +1833,7 @@ declare function createToolSimulator(config: ToolSimulatorConfig): {
  * });
  * ```
  */
-declare function toLangChainTool(tool: Tool<any, any>): DynamicStructuredTool<any>;
+declare function toLangChainTool<TInput, TOutput>(tool: Tool<TInput, TOutput>): DynamicStructuredTool<RuntimeSchema<TInput>, TInput, TInput, string>;
 /**
  * Convert multiple AgentForge tools to LangChain tools
  *
@@ -1844,7 +1851,7 @@ declare function toLangChainTool(tool: Tool<any, any>): DynamicStructuredTool<an
  * });
  * ```
  */
-declare function toLangChainTools(tools: Tool<any, any>[]): DynamicStructuredTool<any>[];
+declare function toLangChainTools(tools: readonly LangChainConvertibleTool[]): DynamicStructuredTool[];
 /**
  * Get the JSON Schema representation of a tool's input schema
  *
@@ -1859,7 +1866,7 @@ declare function toLangChainTools(tools: Tool<any, any>[]): DynamicStructuredToo
  * console.log(JSON.stringify(schema, null, 2));
  * ```
  */
-declare function getToolJsonSchema(tool: Tool<any, any>): Record<string, any>;
+declare function getToolJsonSchema<TInput, TOutput>(tool: Tool<TInput, TOutput>): JsonSchemaObject;
 /**
  * Get tool metadata in a format suitable for LLM prompts
  *
@@ -1880,7 +1887,7 @@ declare function getToolJsonSchema(tool: Tool<any, any>): Record<string, any>;
  * // ...
  * ```
  */
-declare function getToolDescription(tool: Tool<any, any>): string;
+declare function getToolDescription<TInput, TOutput>(tool: Tool<TInput, TOutput>): string;
 
 /**
  * LangGraph State Utilities
@@ -1894,11 +1901,11 @@ declare function getToolDescription(tool: Tool<any, any>): string;
 /**
  * State channel configuration with optional Zod schema validation
  */
-interface StateChannelConfig<T = any, U = T> {
+interface StateChannelConfig<T = unknown, U = T> {
     /**
      * Optional Zod schema for runtime validation
      */
-    schema?: ZodType<T, ZodTypeDef, any>;
+    schema?: ZodType<T, ZodTypeDef, unknown>;
     /**
      * Optional reducer function for aggregating updates
      */
@@ -1912,6 +1919,52 @@ interface StateChannelConfig<T = any, U = T> {
      */
     description?: string;
 }
+type StateChannelConfigLike = {
+    schema?: ZodTypeAny;
+    reducer?: (left: never, right: never) => unknown;
+    default?: () => unknown;
+    description?: string;
+};
+type StateConfigMap = Record<string, StateChannelConfigLike>;
+type IsExact<TLeft, TRight> = [
+    TLeft
+] extends [TRight] ? ([TRight] extends [TLeft] ? true : false) : false;
+type HasReducer<TChannel extends StateChannelConfigLike> = TChannel extends {
+    reducer: (left: unknown, right: unknown) => unknown;
+} ? true : false;
+type SchemaValue<TChannel extends StateChannelConfigLike> = TChannel extends {
+    schema: infer TSchema extends ZodTypeAny;
+} ? output<TSchema> : never;
+type DefaultValue<TChannel extends StateChannelConfigLike> = TChannel extends {
+    default: () => infer TValue;
+} ? TValue : never;
+type ReducerValue<TChannel extends StateChannelConfigLike> = TChannel extends {
+    reducer: (left: infer TValue, right: unknown) => infer TResult;
+} ? IsExact<TValue, TResult> extends true ? TValue : never : never;
+type ReducerUpdate<TChannel extends StateChannelConfigLike> = TChannel extends {
+    reducer: (left: unknown, right: infer TUpdate) => unknown;
+} ? TUpdate : never;
+type ChannelValue<TChannel extends StateChannelConfigLike> = HasReducer<TChannel> extends true ? ReducerValue<TChannel> : [SchemaValue<TChannel>] extends [never] ? [DefaultValue<TChannel>] extends [never] ? unknown : DefaultValue<TChannel> : SchemaValue<TChannel>;
+type ChannelUpdate<TChannel extends StateChannelConfigLike> = HasReducer<TChannel> extends true ? ReducerUpdate<TChannel> : ChannelValue<TChannel>;
+type StateShape<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]: ChannelValue<TConfig[K]>;
+};
+type StateUpdateShape<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]?: ChannelUpdate<TConfig[K]>;
+};
+type StateChannelDefinition<TChannel extends StateChannelConfigLike> = BaseChannel<ChannelValue<TChannel>, ChannelUpdate<TChannel>>;
+type StateAnnotationDefinition<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]: StateChannelDefinition<TConfig[K]>;
+};
+type DefaultedKeys<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]-?: TConfig[K] extends {
+        default: () => ChannelValue<TConfig[K]>;
+    } ? K : never;
+}[keyof TConfig];
+type InputStateKeys<TConfig extends StateConfigMap, TState> = Extract<keyof TConfig, keyof TState>;
+type ValidatedState<TConfig extends StateConfigMap, TState extends Partial<Record<keyof TConfig, unknown>>> = {
+    [K in InputStateKeys<TConfig, TState> | DefaultedKeys<TConfig>]: ChannelValue<TConfig[K]>;
+};
 /**
  * Create a type-safe state annotation with optional Zod validation
  *
@@ -1942,7 +1995,7 @@ interface StateChannelConfig<T = any, U = T> {
  * type State = typeof AgentState.State;
  * ```
  */
-declare function createStateAnnotation<T extends Record<string, StateChannelConfig>>(config: T): AnnotationRoot<StateDefinition>;
+declare function createStateAnnotation<TConfig extends StateConfigMap>(config: TConfig): AnnotationRoot<StateAnnotationDefinition<TConfig>>;
 /**
  * Validate state against Zod schemas
  *
@@ -1964,7 +2017,7 @@ declare function createStateAnnotation<T extends Record<string, StateChannelConf
  * );
  * ```
  */
-declare function validateState<T extends Record<string, StateChannelConfig>>(state: Record<string, any>, config: T): Record<string, any>;
+declare function validateState<TConfig extends StateConfigMap, TState extends Partial<Record<keyof TConfig, unknown>>>(state: TState, config: TConfig): ValidatedState<TConfig, TState>;
 /**
  * Merge state updates using configured reducers
  *
@@ -1989,7 +2042,7 @@ declare function validateState<T extends Record<string, StateChannelConfig>>(sta
  * // Result: { messages: ['a', 'b', 'c'] }
  * ```
  */
-declare function mergeState<T extends Record<string, StateChannelConfig>>(currentState: Record<string, any>, update: Record<string, any>, config: T): Record<string, any>;
+declare function mergeState<TConfig extends StateConfigMap>(currentState: Partial<StateShape<TConfig>>, update: StateUpdateShape<TConfig>, config: TConfig): Partial<StateShape<TConfig>>;
 
 /**
  * Sequential Workflow Builder

package/dist/index.d.ts

@@ -1,6 +1,6 @@
-import { z, ZodType, ZodTypeDef } from 'zod';
+import { z, ZodTypeAny, output, ZodType, ZodTypeDef } from 'zod';
 import { DynamicStructuredTool } from '@langchain/core/tools';
-import { AnnotationRoot, StateDefinition, StateGraph, END, MemorySaver, BaseCheckpointSaver, CheckpointTuple } from '@langchain/langgraph';
+import { AnnotationRoot, BaseChannel, StateGraph, END, MemorySaver, BaseCheckpointSaver, CheckpointTuple } from '@langchain/langgraph';
 import { RunnableConfig } from '@langchain/core/runnables';
 
 /**
@@ -1794,6 +1794,13 @@ declare function createToolSimulator(config: ToolSimulatorConfig): {
  * ```
  */
 
+type RuntimeSchema<TInput = unknown> = z.ZodSchema<TInput>;
+type JsonSchemaObject = Record<string, unknown>;
+interface LangChainConvertibleTool<TInput = unknown, TOutput = unknown> {
+    metadata: ToolMetadata;
+    schema: RuntimeSchema<TInput>;
+    invoke(input: TInput): Promise<TOutput>;
+}
 /**
  * Convert an AgentForge tool to a LangChain DynamicStructuredTool
  *
@@ -1826,7 +1833,7 @@ declare function createToolSimulator(config: ToolSimulatorConfig): {
  * });
  * ```
  */
-declare function toLangChainTool(tool: Tool<any, any>): DynamicStructuredTool<any>;
+declare function toLangChainTool<TInput, TOutput>(tool: Tool<TInput, TOutput>): DynamicStructuredTool<RuntimeSchema<TInput>, TInput, TInput, string>;
 /**
  * Convert multiple AgentForge tools to LangChain tools
  *
@@ -1844,7 +1851,7 @@ declare function toLangChainTool(tool: Tool<any, any>): DynamicStructuredTool<an
  * });
  * ```
  */
-declare function toLangChainTools(tools: Tool<any, any>[]): DynamicStructuredTool<any>[];
+declare function toLangChainTools(tools: readonly LangChainConvertibleTool[]): DynamicStructuredTool[];
 /**
  * Get the JSON Schema representation of a tool's input schema
  *
@@ -1859,7 +1866,7 @@ declare function toLangChainTools(tools: Tool<any, any>[]): DynamicStructuredToo
  * console.log(JSON.stringify(schema, null, 2));
  * ```
  */
-declare function getToolJsonSchema(tool: Tool<any, any>): Record<string, any>;
+declare function getToolJsonSchema<TInput, TOutput>(tool: Tool<TInput, TOutput>): JsonSchemaObject;
 /**
  * Get tool metadata in a format suitable for LLM prompts
  *
@@ -1880,7 +1887,7 @@ declare function getToolJsonSchema(tool: Tool<any, any>): Record<string, any>;
  * // ...
  * ```
  */
-declare function getToolDescription(tool: Tool<any, any>): string;
+declare function getToolDescription<TInput, TOutput>(tool: Tool<TInput, TOutput>): string;
 
 /**
  * LangGraph State Utilities
@@ -1894,11 +1901,11 @@ declare function getToolDescription(tool: Tool<any, any>): string;
 /**
  * State channel configuration with optional Zod schema validation
  */
-interface StateChannelConfig<T = any, U = T> {
+interface StateChannelConfig<T = unknown, U = T> {
     /**
      * Optional Zod schema for runtime validation
      */
-    schema?: ZodType<T, ZodTypeDef, any>;
+    schema?: ZodType<T, ZodTypeDef, unknown>;
     /**
      * Optional reducer function for aggregating updates
      */
@@ -1912,6 +1919,52 @@ interface StateChannelConfig<T = any, U = T> {
      */
     description?: string;
 }
+type StateChannelConfigLike = {
+    schema?: ZodTypeAny;
+    reducer?: (left: never, right: never) => unknown;
+    default?: () => unknown;
+    description?: string;
+};
+type StateConfigMap = Record<string, StateChannelConfigLike>;
+type IsExact<TLeft, TRight> = [
+    TLeft
+] extends [TRight] ? ([TRight] extends [TLeft] ? true : false) : false;
+type HasReducer<TChannel extends StateChannelConfigLike> = TChannel extends {
+    reducer: (left: unknown, right: unknown) => unknown;
+} ? true : false;
+type SchemaValue<TChannel extends StateChannelConfigLike> = TChannel extends {
+    schema: infer TSchema extends ZodTypeAny;
+} ? output<TSchema> : never;
+type DefaultValue<TChannel extends StateChannelConfigLike> = TChannel extends {
+    default: () => infer TValue;
+} ? TValue : never;
+type ReducerValue<TChannel extends StateChannelConfigLike> = TChannel extends {
+    reducer: (left: infer TValue, right: unknown) => infer TResult;
+} ? IsExact<TValue, TResult> extends true ? TValue : never : never;
+type ReducerUpdate<TChannel extends StateChannelConfigLike> = TChannel extends {
+    reducer: (left: unknown, right: infer TUpdate) => unknown;
+} ? TUpdate : never;
+type ChannelValue<TChannel extends StateChannelConfigLike> = HasReducer<TChannel> extends true ? ReducerValue<TChannel> : [SchemaValue<TChannel>] extends [never] ? [DefaultValue<TChannel>] extends [never] ? unknown : DefaultValue<TChannel> : SchemaValue<TChannel>;
+type ChannelUpdate<TChannel extends StateChannelConfigLike> = HasReducer<TChannel> extends true ? ReducerUpdate<TChannel> : ChannelValue<TChannel>;
+type StateShape<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]: ChannelValue<TConfig[K]>;
+};
+type StateUpdateShape<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]?: ChannelUpdate<TConfig[K]>;
+};
+type StateChannelDefinition<TChannel extends StateChannelConfigLike> = BaseChannel<ChannelValue<TChannel>, ChannelUpdate<TChannel>>;
+type StateAnnotationDefinition<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]: StateChannelDefinition<TConfig[K]>;
+};
+type DefaultedKeys<TConfig extends StateConfigMap> = {
+    [K in keyof TConfig]-?: TConfig[K] extends {
+        default: () => ChannelValue<TConfig[K]>;
+    } ? K : never;
+}[keyof TConfig];
+type InputStateKeys<TConfig extends StateConfigMap, TState> = Extract<keyof TConfig, keyof TState>;
+type ValidatedState<TConfig extends StateConfigMap, TState extends Partial<Record<keyof TConfig, unknown>>> = {
+    [K in InputStateKeys<TConfig, TState> | DefaultedKeys<TConfig>]: ChannelValue<TConfig[K]>;
+};
 /**
  * Create a type-safe state annotation with optional Zod validation
  *
@@ -1942,7 +1995,7 @@ interface StateChannelConfig<T = any, U = T> {
  * type State = typeof AgentState.State;
  * ```
  */
-declare function createStateAnnotation<T extends Record<string, StateChannelConfig>>(config: T): AnnotationRoot<StateDefinition>;
+declare function createStateAnnotation<TConfig extends StateConfigMap>(config: TConfig): AnnotationRoot<StateAnnotationDefinition<TConfig>>;
 /**
  * Validate state against Zod schemas
  *
@@ -1964,7 +2017,7 @@ declare function createStateAnnotation<T extends Record<string, StateChannelConf
  * );
  * ```
  */
-declare function validateState<T extends Record<string, StateChannelConfig>>(state: Record<string, any>, config: T): Record<string, any>;
+declare function validateState<TConfig extends StateConfigMap, TState extends Partial<Record<keyof TConfig, unknown>>>(state: TState, config: TConfig): ValidatedState<TConfig, TState>;
 /**
  * Merge state updates using configured reducers
  *
@@ -1989,7 +2042,7 @@ declare function validateState<T extends Record<string, StateChannelConfig>>(sta
  * // Result: { messages: ['a', 'b', 'c'] }
  * ```
  */
-declare function mergeState<T extends Record<string, StateChannelConfig>>(currentState: Record<string, any>, update: Record<string, any>, config: T): Record<string, any>;
+declare function mergeState<TConfig extends StateConfigMap>(currentState: Partial<StateShape<TConfig>>, update: StateUpdateShape<TConfig>, config: TConfig): Partial<StateShape<TConfig>>;
 
 /**
  * Sequential Workflow Builder

package/dist/index.js

@@ -619,21 +619,42 @@ function toolBuilder() {
 // src/langchain/converter.ts
 import { DynamicStructuredTool } from "@langchain/core/tools";
 import { zodToJsonSchema } from "zod-to-json-schema";
+function serializeToolResult(result) {
+  if (typeof result === "string") {
+    return result;
+  }
+  if (typeof result === "object" && result !== null) {
+    return JSON.stringify(result, null, 2);
+  }
+  return String(result);
+}
+function isJsonSchemaObject(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function isJsonSchemaDefinitionMap(value) {
+  if (!isJsonSchemaObject(value)) {
+    return false;
+  }
+  return Object.values(value).every(isJsonSchemaObject);
+}
+function extractToolSchema(jsonSchema) {
+  if (!isJsonSchemaObject(jsonSchema)) {
+    return {};
+  }
+  const ref = jsonSchema.$ref;
+  const definitions = jsonSchema.definitions;
+  if (typeof ref !== "string" || !isJsonSchemaDefinitionMap(definitions)) {
+    return jsonSchema;
+  }
+  const refName = ref.replace("#/definitions/", "");
+  return definitions[refName] ?? jsonSchema;
+}
 function toLangChainTool(tool) {
   return new DynamicStructuredTool({
     name: tool.metadata.name,
     description: tool.metadata.description,
     schema: tool.schema,
-    func: async (input) => {
-      const result = await tool.invoke(input);
-      if (typeof result === "string") {
-        return result;
-      }
-      if (typeof result === "object" && result !== null) {
-        return JSON.stringify(result, null, 2);
-      }
-      return String(result);
-    }
+    func: async (input) => serializeToolResult(await tool.invoke(input))
   });
 }
 function toLangChainTools(tools) {
@@ -645,11 +666,7 @@ function getToolJsonSchema(tool) {
     $refStrategy: "none"
     // Don't use $ref for nested schemas
   });
-  if (jsonSchema.$ref && jsonSchema.definitions) {
-    const refName = jsonSchema.$ref.replace("#/definitions/", "");
-    return jsonSchema.definitions[refName] || jsonSchema;
-  }
-  return jsonSchema;
+  return extractToolSchema(jsonSchema);
 }
 function getToolDescription(tool) {
   const { metadata } = tool;
@@ -2019,46 +2036,84 @@ function createToolSimulator(config) {
 
 // src/langgraph/state.ts
 import { Annotation } from "@langchain/langgraph";
+function entriesOf(value) {
+  return Object.entries(value);
+}
+function keysOf(value) {
+  return Object.keys(value);
+}
+function hasOwnProperty(value, key) {
+  return Object.prototype.hasOwnProperty.call(value, key);
+}
+function useLatestValue(_left, right) {
+  return right;
+}
+function setStateValue(target, key, value) {
+  target[key] = value;
+}
+function setStateDefinitionValue(target, key, value) {
+  target[key] = value;
+}
+function createChannel(channelConfig) {
+  if (channelConfig.reducer) {
+    return Annotation({
+      reducer: channelConfig.reducer,
+      default: channelConfig.default
+    });
+  }
+  if (channelConfig.default) {
+    return Annotation({
+      reducer: useLatestValue,
+      default: channelConfig.default
+    });
+  }
+  return Annotation();
+}
 function createStateAnnotation(config) {
   const stateDefinition = {};
-  for (const [key, channelConfig] of Object.entries(config)) {
-    if (channelConfig.reducer) {
-      stateDefinition[key] = Annotation({
-        reducer: channelConfig.reducer,
-        default: channelConfig.default
-      });
-    } else if (channelConfig.default) {
-      stateDefinition[key] = Annotation({
-        reducer: (_left, right) => right,
-        default: channelConfig.default
-      });
-    } else {
-      stateDefinition[key] = Annotation();
-    }
+  for (const [key, channelConfig] of entriesOf(config)) {
+    setStateDefinitionValue(
+      stateDefinition,
+      key,
+      createChannel(channelConfig)
+    );
   }
   return Annotation.Root(stateDefinition);
 }
 function validateState(state, config) {
   const validated = {};
-  for (const [key, channelConfig] of Object.entries(config)) {
-    if (channelConfig.schema && key in state) {
-      validated[key] = channelConfig.schema.parse(state[key]);
-    } else if (key in state) {
-      validated[key] = state[key];
+  for (const [key, channelConfig] of entriesOf(config)) {
+    if (channelConfig.schema && hasOwnProperty(state, key)) {
+      setStateValue(
+        validated,
+        key,
+        channelConfig.schema.parse(state[key])
+      );
+    } else if (hasOwnProperty(state, key)) {
+      setStateValue(validated, key, state[key]);
     } else if (channelConfig.default) {
-      validated[key] = channelConfig.default();
+      setStateValue(validated, key, channelConfig.default());
     }
   }
   return validated;
 }
 function mergeState(currentState, update, config) {
   const merged = { ...currentState };
-  for (const [key, value] of Object.entries(update)) {
+  for (const key of keysOf(update)) {
+    const value = update[key];
     const channelConfig = config[key];
-    if (channelConfig?.reducer && key in merged) {
-      merged[key] = channelConfig.reducer(merged[key], value);
+    const reducer = channelConfig?.reducer;
+    if (reducer && hasOwnProperty(merged, key)) {
+      setStateValue(
+        merged,
+        key,
+        reducer(
+          merged[key],
+          value
+        )
+      );
     } else {
-      merged[key] = value;
+      setStateValue(merged, key, value);
     }
   }
   return merged;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentforge/core",
-  "version": "0.15.6",
+  "version": "0.15.8",
   "description": "Production-ready TypeScript agent framework built on LangGraph with orchestration, middleware, and typed abstractions.",
   "type": "module",
   "main": "./dist/index.cjs",