@agentforge/core 0.15.7 → 0.15.8

package/dist/index.cjs

@@ -2211,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';
 
 /**
@@ -1901,11 +1901,11 @@ declare function getToolDescription<TInput, TOutput>(tool: Tool<TInput, TOutput>
 /**
  * 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
      */
@@ -1919,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
  *
@@ -1949,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
  *
@@ -1971,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
  *
@@ -1996,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';
 
 /**
@@ -1901,11 +1901,11 @@ declare function getToolDescription<TInput, TOutput>(tool: Tool<TInput, TOutput>
 /**
  * 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
      */
@@ -1919,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
  *
@@ -1949,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
  *
@@ -1971,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
  *
@@ -1996,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

@@ -2036,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.7",
+  "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",