@ai.ntellect/core 0.7.10 → 0.7.12

package/dist/types/index.d.ts

@@ -68,7 +68,7 @@ export type GraphContext<T extends ZodSchema> = {
  * @template I - Input schema type
  * @template O - Output schema type
  */
-export interface Node<T extends ZodSchema, P = any> {
+export type GraphNodeConfig<T extends ZodSchema, P = any> = {
     /** Name of the node */
     name: string;
     /** Description of the node */
@@ -79,10 +79,13 @@ export interface Node<T extends ZodSchema, P = any> {
     execute: (context: GraphContext<T>, params?: P, tools?: {
         eventEmitter: IEventEmitter;
     }) => Promise<void>;
-    /** Optional condition for node execution */
+    /** Optional condition for node start execution */
     condition?: (context: GraphContext<T>, params?: P) => boolean;
-    /** Array of next node names */
-    next?: string[] | ((context: GraphContext<T>) => string[]);
+    /** Array of next node names or objects with conditions for the next node */
+    next?: Array<string | {
+        node: string;
+        condition: (context: GraphContext<T>) => boolean;
+    }> | string | ((context: GraphContext<T>) => string[]);
     /** Array of event names that trigger this node */
     events?: string[];
     /** Wait for a single event before continuing */
@@ -108,17 +111,17 @@ export interface Node<T extends ZodSchema, P = any> {
     };
     /** Error handler function */
     onError?: (error: Error) => void;
-}
+};
 /**
  * Interface for graph definition
  * @interface
  * @template T - Schema type
  */
-export interface GraphDefinition<T extends ZodSchema> {
+export type GraphConfig<T extends ZodSchema> = {
     /** Name of the graph */
     name: string;
     /** Array of nodes in the graph */
-    nodes: Node<T, any>[];
+    nodes: GraphNodeConfig<T, any>[];
     /** Initial context */
     context: SchemaType<T>;
     /** Schema for validation */
@@ -131,7 +134,7 @@ export interface GraphDefinition<T extends ZodSchema> {
     eventEmitter?: IEventEmitter | EventEmitter;
     /** Array of events */
     events?: string[];
-}
+};
 /**
  * Configuration type for Meilisearch
  * @typedef {Object} MeilisearchConfig

package/dist/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../types/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAI9C;;;;;;;GAOG;AACH,MAAM,WAAW,iBAAiB;IAChC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,IAAI,CAAC;CACjB;AAID;;;GAGG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,kDAAkD;IAClD,EAAE,EAAE,MAAM,CAAC;IACX,kCAAkC;IAClC,eAAe,EAAE,MAAM,CAAC;IACxB,yCAAyC;IACzC,cAAc,EAAE,MAAM,CAAC;IACvB,uCAAuC;IACvC,WAAW,EAAE,OAAO,CAAC;IACrB,wBAAwB;IACxB,SAAS,EAAE,IAAI,CAAC;CACjB,CAAC;AAIF;;;GAGG;AACH,MAAM,MAAM,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;AAErE;;;GAGG;AACH,MAAM,MAAM,YAAY,CAAC,CAAC,SAAS,SAAS,IAAI;IAC9C,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,WAAW,IAAI,CAAC,CAAC,SAAS,SAAS,EAAE,CAAC,GAAG,GAAG;IAChD,uBAAuB;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,8BAA8B;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,6BAA6B;IAC7B,MAAM,CAAC,EAAE,CAAC,SAAS,IAAI,GAAG,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;IAC/C,oCAAoC;IACpC,OAAO,EAAE,CACP,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,EACxB,MAAM,CAAC,EAAE,CAAC,EACV,KAAK,CAAC,EAAE;QAAE,YAAY,EAAE,aAAa,CAAA;KAAE,KACpC,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,4CAA4C;IAC5C,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK,OAAO,CAAC;IAC9D,+BAA+B;IAC/B,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC;IAC3D,kDAAkD;IAClD,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,gDAAgD;IAChD,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,6CAA6C;IAC7C,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,sCAAsC;IACtC,eAAe,CAAC,EAAE;QAChB,MAAM,EAAE,MAAM,EAAE,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,OAAO,CAAC;KACzC,CAAC;IACF,0BAA0B;IAC1B,KAAK,CAAC,EAAE;QACN,uCAAuC;QACvC,WAAW,EAAE,MAAM,CAAC;QACpB,4CAA4C;QAC5C,KAAK,EAAE,MAAM,CAAC;QACd,6BAA6B;QAC7B,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAC1E,yCAAyC;QACzC,gBAAgB,CAAC,EAAE,OAAO,CAAC;KAC5B,CAAC;IACF,6BAA6B;IAC7B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC;AAED;;;;GAIG;AACH,MAAM,WAAW,eAAe,CAAC,CAAC,SAAS,SAAS;IAClD,wBAAwB;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,kCAAkC;IAClC,KAAK,EAAE,IAAI,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;IACtB,sBAAsB;IACtB,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACvB,4BAA4B;IAC5B,MAAM,EAAE,CAAC,CAAC;IACV,2BAA2B;IAC3B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;IAC3D,sBAAsB;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,6BAA6B;IAC7B,YAAY,CAAC,EAAE,aAAa,GAAG,YAAY,CAAC;IAC5C,sBAAsB;IACtB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB;AAID;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,2BAA2B;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,iCAAiC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,qCAAqC;IACrC,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAChC,mCAAmC;IACnC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,qCAAqC;IACrC,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAChC,mCAAmC;IACnC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,CAAC;AAEF,MAAM,WAAW,UAAU,CAAC,CAAC,SAAS,SAAS;IAC7C,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,GAAG,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC;IACjC,SAAS,CAAC,EAAE,CAAC,CAAC,SAAS,SAAS,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9E"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../types/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,CAAC;AAChC,OAAO,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAI9C;;;;;;;GAOG;AACH,MAAM,WAAW,iBAAiB;IAChC,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,GAAG,CAAC,EAAE,MAAM,CAAC;CACd;AAED;;;;;;;;;GASG;AACH,MAAM,WAAW,cAAc;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,IAAI,CAAC;CACjB;AAID;;;GAGG;AACH,MAAM,MAAM,gBAAgB,GAAG;IAC7B,kDAAkD;IAClD,EAAE,EAAE,MAAM,CAAC;IACX,kCAAkC;IAClC,eAAe,EAAE,MAAM,CAAC;IACxB,yCAAyC;IACzC,cAAc,EAAE,MAAM,CAAC;IACvB,uCAAuC;IACvC,WAAW,EAAE,OAAO,CAAC;IACrB,wBAAwB;IACxB,SAAS,EAAE,IAAI,CAAC;CACjB,CAAC;AAIF;;;GAGG;AACH,MAAM,MAAM,UAAU,CAAC,CAAC,IAAI,CAAC,SAAS,SAAS,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;AAErE;;;GAGG;AACH,MAAM,MAAM,YAAY,CAAC,CAAC,SAAS,SAAS,IAAI;IAC9C,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,MAAM,eAAe,CAAC,CAAC,SAAS,SAAS,EAAE,CAAC,GAAG,GAAG,IAAI;IAC1D,uBAAuB;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,8BAA8B;IAC9B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,6BAA6B;IAC7B,MAAM,CAAC,EAAE,CAAC,SAAS,IAAI,GAAG,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;IAC/C,oCAAoC;IACpC,OAAO,EAAE,CACP,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,EACxB,MAAM,CAAC,EAAE,CAAC,EACV,KAAK,CAAC,EAAE;QAAE,YAAY,EAAE,aAAa,CAAA;KAAE,KACpC,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,kDAAkD;IAClD,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK,OAAO,CAAC;IAE9D,4EAA4E;IAC5E,IAAI,CAAC,EACD,KAAK,CACD,MAAM,GACN;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,OAAO,CAAA;KAAE,CACrE,GACD,MAAM,GACN,CAAC,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC;IAC7C,kDAAkD;IAClD,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;IAClB,gDAAgD;IAChD,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,6CAA6C;IAC7C,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,sCAAsC;IACtC,eAAe,CAAC,EAAE;QAChB,MAAM,EAAE,MAAM,EAAE,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,WAAW,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,KAAK,OAAO,CAAC;KACzC,CAAC;IACF,0BAA0B;IAC1B,KAAK,CAAC,EAAE;QACN,uCAAuC;QACvC,WAAW,EAAE,MAAM,CAAC;QACpB,4CAA4C;QAC5C,KAAK,EAAE,MAAM,CAAC;QACd,6BAA6B;QAC7B,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;QAC1E,yCAAyC;QACzC,gBAAgB,CAAC,EAAE,OAAO,CAAC;KAC5B,CAAC;IACF,6BAA6B;IAC7B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;CAClC,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,WAAW,CAAC,CAAC,SAAS,SAAS,IAAI;IAC7C,wBAAwB;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,kCAAkC;IAClC,KAAK,EAAE,eAAe,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;IACjC,sBAAsB;IACtB,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC;IACvB,4BAA4B;IAC5B,MAAM,EAAE,CAAC,CAAC;IACV,2BAA2B;IAC3B,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;IAC3D,sBAAsB;IACtB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,6BAA6B;IAC7B,YAAY,CAAC,EAAE,aAAa,GAAG,YAAY,CAAC;IAC5C,sBAAsB;IACtB,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB,CAAC;AAIF;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG;IAC9B,2BAA2B;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,iCAAiC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,qCAAqC;IACrC,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAChC,mCAAmC;IACnC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,mBAAmB,GAAG;IAChC,qCAAqC;IACrC,oBAAoB,CAAC,EAAE,MAAM,EAAE,CAAC;IAChC,mCAAmC;IACnC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B,CAAC;AAEF,MAAM,WAAW,UAAU,CAAC,CAAC,SAAS,SAAS;IAC7C,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,GAAG,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,KAAK,GAAG,KAAK,GAAG,MAAM,CAAC;IACjC,SAAS,CAAC,EAAE,CAAC,CAAC,SAAS,SAAS,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9E"}

package/graph/event-manager.ts

@@ -1,7 +1,7 @@
 import { Observable, Subject, filter } from "rxjs";
 import { ZodSchema } from "zod";
 import { IEventEmitter } from "../interfaces";
-import { GraphContext, GraphEvent, Node } from "../types";
+import { GraphContext, GraphEvent, GraphNodeConfig } from "../types";
 import { GraphNode } from "./node";
 
 /**
@@ -31,7 +31,7 @@ export class GraphEventManager<T extends ZodSchema> {
    */
   constructor(
     private eventEmitter: IEventEmitter,
-    private nodes: Map<string, Node<T, any>>,
+    private nodes: Map<string, GraphNodeConfig<T, any>>,
     name: string,
     context: GraphContext<T>,
     graphEvents?: string[],

package/graph/index.ts

@@ -2,11 +2,17 @@ import { EventEmitter } from "events";
 import { BehaviorSubject, Subject } from "rxjs";
 import { ZodSchema } from "zod";
 import { GraphObservable, IEventEmitter } from "../interfaces";
-import { GraphContext, GraphDefinition, GraphEvent, Node } from "../types";
+import {
+  GraphConfig,
+  GraphContext,
+  GraphEvent,
+  GraphNodeConfig,
+} from "../types";
 import { GraphEventManager } from "./event-manager";
 import { GraphLogger } from "./logger";
 import { GraphNode, NodeParams } from "./node";
 import { GraphObserver } from "./observer";
+import { GraphVisualizer } from "./visualizer";
 
 /**
  * @module GraphFlow
@@ -29,7 +35,7 @@ export class GraphFlow<T extends ZodSchema> {
   private graphEvents?: string[];
   private entryNode?: string;
   private verbose: boolean = false;
-  public nodes: Map<string, Node<T, any>>;
+  public nodes: Map<string, GraphNodeConfig<T, any>>;
 
   private eventSubject: Subject<GraphEvent<T>> = new Subject();
   private stateSubject: BehaviorSubject<GraphContext<T>>;
@@ -43,16 +49,16 @@ export class GraphFlow<T extends ZodSchema> {
   /**
    * Creates a new instance of GraphFlow
    * @param {string} name - The name of the graph flow
-   * @param {GraphDefinition<T>} config - Configuration object containing nodes, schema, context, and error handlers
+   * @param {GraphConfig<T>} config - Configuration object containing nodes, schema, context, and error handlers
    * @param {Object} options - Optional options for the graph flow
    */
   constructor(
     public name: string,
-    config: GraphDefinition<T>,
+    config: GraphConfig<T>,
     options: { verbose?: boolean } = {}
   ) {
     this.nodes = new Map(
-      config.nodes.map((node: Node<T, any>) => [node.name, node])
+      config.nodes.map((node: GraphNodeConfig<T, any>) => [node.name, node])
     );
     this.validator = config.schema;
     this.context = config.schema.parse(config.context) as GraphContext<T>;
@@ -75,7 +81,7 @@ export class GraphFlow<T extends ZodSchema> {
       config.onError
     );
     this.nodeExecutor = new GraphNode(
-      this.nodes as Map<string, Node<T, any>>,
+      this.nodes as Map<string, GraphNodeConfig<T, any>>,
       this.logger,
       this.eventManager,
       this.eventSubject,
@@ -288,9 +294,9 @@ export class GraphFlow<T extends ZodSchema> {
 
   /**
    * Updates the graph definition with new configuration
-   * @param {GraphDefinition<T>} definition - New graph definition
+   * @param {GraphConfig<T>} definition - New graph definition
    */
-  public load(definition: GraphDefinition<T>): void {
+  public load(definition: GraphConfig<T>): void {
     // Clear all existing nodes
     this.nodes.clear();
     // Wipe out old node-based event listeners
@@ -358,10 +364,10 @@ export class GraphFlow<T extends ZodSchema> {
 
   /**
    * Adds a new node to the graph
-   * @param {Node<T>} node - Node to add
+   * @param {GraphNodeConfig<T, any>} node - Node to add
    * @throws {Error} If node with same name already exists
    */
-  public addNode(node: Node<T, any>): void {
+  public addNode(node: GraphNodeConfig<T, any>): void {
     this.nodes.set(node.name, node);
     this.eventManager.setupEventListeners();
   }
@@ -379,7 +385,7 @@ export class GraphFlow<T extends ZodSchema> {
    * Returns all nodes in the graph
    * @returns {Node<T>[]} Array of all nodes
    */
-  public getNodes(): Node<T, any>[] {
+  public getNodes(): GraphNodeConfig<T, any>[] {
     return Array.from(this.nodes.values());
   }
 
@@ -392,4 +398,12 @@ export class GraphFlow<T extends ZodSchema> {
     this.eventSubject.complete();
     this.stateSubject.complete();
   }
+
+  /**
+   * Creates a visualizer instance for the current graph
+   * @returns {GraphVisualizer<T>} A visualizer instance
+   */
+  public createVisualizer(): GraphVisualizer<T> {
+    return new GraphVisualizer(this.nodes);
+  }
 }

package/graph/node.ts

@@ -1,7 +1,7 @@
 import { IEventEmitter } from "interfaces";
 import { BehaviorSubject, Subject } from "rxjs";
 import { ZodSchema } from "zod";
-import { GraphContext, GraphEvent, Node } from "../types";
+import { GraphContext, GraphEvent, GraphNodeConfig } from "../types";
 import { GraphEventManager } from "./event-manager";
 
 /**
@@ -29,7 +29,7 @@ export class GraphNode<T extends ZodSchema> {
    * @param stateSubject - Subject for managing graph state
    */
   constructor(
-    private nodes: Map<string, Node<T, any>>,
+    private nodes: Map<string, GraphNodeConfig<T, any>>,
     private logger: GraphLogger,
     private eventManager: GraphEventManager<T>,
     private eventSubject: Subject<GraphEvent<T>>,
@@ -144,7 +144,9 @@ export class GraphNode<T extends ZodSchema> {
       if (!triggeredByEvent && node.next) {
         const nextNodes =
           typeof node.next === "function" ? node.next(contextProxy) : node.next;
-        for (const nextNodeName of nextNodes) {
+        for (const nextNode of nextNodes) {
+          const nextNodeName =
+            typeof nextNode === "string" ? nextNode : nextNode.node;
           await this.executeNode(nextNodeName, context, undefined, false);
         }
       }
@@ -167,7 +169,7 @@ export class GraphNode<T extends ZodSchema> {
    * @private
    */
   private async validateParams(
-    node: Node<T, any>,
+    node: GraphNodeConfig<T, any>,
     params: any,
     nodeName: string
   ): Promise<void> {
@@ -194,7 +196,7 @@ export class GraphNode<T extends ZodSchema> {
    * @private
    */
   private async handleEvents(
-    node: Node<T, any>,
+    node: GraphNodeConfig<T, any>,
     nodeName: string,
     context: GraphContext<T>
   ): Promise<void> {
@@ -218,7 +220,7 @@ export class GraphNode<T extends ZodSchema> {
    * @private
    */
   private async executeWithRetry(
-    node: Node<T, any>,
+    node: GraphNodeConfig<T, any>,
     contextProxy: GraphContext<T>,
     nodeName: string,
     params?: NodeParams
@@ -266,7 +268,7 @@ export class GraphNode<T extends ZodSchema> {
    * @private
    */
   private async handleCorrelatedEvents(
-    node: Node<T, any>,
+    node: GraphNodeConfig<T, any>,
     nodeName: string
   ): Promise<void> {
     if (node.correlateEvents) {
@@ -312,7 +314,7 @@ export class GraphNode<T extends ZodSchema> {
    * @private
    */
   private async handleWaitForEvents(
-    node: Node<T, any>,
+    node: GraphNodeConfig<T, any>,
     nodeName: string
   ): Promise<void> {
     if (node.waitForEvents) {

package/graph/visualizer.ts

@@ -0,0 +1,243 @@
+import { z, ZodSchema } from "zod";
+import { GraphNodeConfig } from "../types";
+import { GraphFlow } from "./index";
+
+interface NodeVisualData {
+  id: string;
+  label: string;
+  type?: string;
+  events?: string[];
+  contextChanges?: string[];
+}
+
+interface EdgeVisualData {
+  from: string;
+  to: string;
+  condition?: string;
+}
+
+interface NextNode {
+  node: string;
+  condition: (context: any) => boolean;
+}
+
+export class GraphVisualizer<T extends ZodSchema> {
+  private nodes: NodeVisualData[] = [];
+  private edges: EdgeVisualData[] = [];
+
+  constructor(private graphNodes: Map<string, GraphNodeConfig<T, any>>) {
+    this.processNodes();
+  }
+
+  private processNodes(): void {
+    this.graphNodes.forEach((node, nodeName) => {
+      // Extraire les modifications de contexte depuis la fonction execute
+      const contextChanges: string[] = [];
+      const executeStr = node.execute.toString();
+      const contextAssignments = executeStr.match(/context\.\w+\s*=/g) || [];
+      contextAssignments.forEach((assignment) => {
+        const variable = assignment
+          .replace("context.", "")
+          .replace("=", "")
+          .trim();
+        contextChanges.push(variable);
+      });
+
+      this.nodes.push({
+        id: nodeName,
+        label: nodeName,
+        events: node.events,
+        contextChanges,
+      });
+
+      if (node.next) {
+        if (Array.isArray(node.next)) {
+          node.next.forEach((nextNode: string | NextNode) => {
+            if (typeof nextNode === "string") {
+              this.edges.push({ from: nodeName, to: nextNode });
+            } else {
+              const conditionStr = nextNode.condition
+                .toString()
+                .replace(/.*=> /, "")
+                .replace(/ctx\./g, "")
+                .replace(/[{}]/g, "")
+                .replace(/\|\|/g, " or ")
+                .replace(/&&/g, " and ")
+                .replace(/!/g, "not ");
+
+              this.edges.push({
+                from: nodeName,
+                to: nextNode.node,
+                condition: conditionStr,
+              });
+            }
+          });
+        } else if (typeof node.next === "string") {
+          this.edges.push({ from: nodeName, to: node.next });
+        }
+      }
+    });
+  }
+
+  /**
+   * Generates a Mermaid flowchart representation of the graph
+   */
+  public toMermaid(): string {
+    let mmd = "flowchart TD\n";
+
+    // Add nodes with context changes
+    this.nodes.forEach((node) => {
+      let nodeLabel = node.label;
+      if (node.contextChanges?.length) {
+        nodeLabel += `\nSet: ${node.contextChanges.join(", ")}`;
+      }
+      if (node.events?.length) {
+        // Créer des nœuds d'événements externes
+        node.events.forEach((event) => {
+          mmd += `    ${event}((${event}))\n`;
+          mmd += `    ${event} -.->|event| ${node.id}\n`;
+        });
+      }
+      mmd += `    ${node.id}["${nodeLabel}"]\n`;
+    });
+
+    // Add edges with escaped conditions
+    this.edges.forEach((edge) => {
+      const escapedCondition = edge.condition?.replace(/[|]/g, "\\|");
+      const arrow = escapedCondition ? `-->|${escapedCondition}|` : "-->";
+      mmd += `    ${edge.from} ${arrow} ${edge.to}\n`;
+    });
+
+    return mmd;
+  }
+
+  /**
+   * Returns the nodes and edges data
+   */
+  public getVisualizationData() {
+    return {
+      nodes: this.nodes,
+      edges: this.edges,
+    };
+  }
+}
+
+// Example usage
+const graph = new GraphFlow("CryptoTradeAssistant", {
+  name: "CryptoTradeAssistant",
+  entryNode: "Initialize",
+  nodes: [
+    {
+      name: "Initialize",
+      execute: async (context) => {
+        context.walletBalance = 1000;
+        context.targetPrice = 2000;
+        context.gasFees = 50;
+        context.isWalletConnected = false;
+        console.log("Initializing crypto assistant");
+      },
+      next: ["ConnectWallet"],
+    },
+    {
+      name: "ConnectWallet",
+      execute: async (context) => {
+        context.isWalletConnected = true;
+        console.log("Connecting wallet");
+      },
+      next: ["CheckMarketConditions"],
+      events: ["walletConnected"],
+    },
+    {
+      name: "CheckMarketConditions",
+      execute: async (context) => {
+        context.currentPrice = 1950;
+        context.marketVolatility = "low";
+        context.hasLiquidity = true;
+        console.log("Checking market conditions");
+      },
+      next: [
+        {
+          node: "PrepareTransaction",
+          condition: (ctx) =>
+            ctx.isWalletConnected &&
+            ctx.currentPrice < ctx.targetPrice &&
+            ctx.hasLiquidity,
+        },
+        {
+          node: "WaitForBetterConditions",
+          condition: (ctx) =>
+            !ctx.hasLiquidity || ctx.currentPrice >= ctx.targetPrice,
+        },
+      ],
+      events: ["priceUpdate", "liquidityUpdate"],
+    },
+    {
+      name: "WaitForBetterConditions",
+      execute: async () => console.log("Waiting for better conditions"),
+      next: ["CheckMarketConditions"],
+      events: ["marketUpdate"],
+    },
+    {
+      name: "PrepareTransaction",
+      execute: async (context) => {
+        context.txHash = "0x123...";
+        context.estimatedGas = 45;
+        console.log("Preparing transaction");
+      },
+      next: [
+        {
+          node: "ExecuteTransaction",
+          condition: (ctx) => ctx.estimatedGas < ctx.gasFees,
+        },
+        {
+          node: "WaitForBetterConditions",
+          condition: (ctx) => ctx.estimatedGas >= ctx.gasFees,
+        },
+      ],
+    },
+    {
+      name: "ExecuteTransaction",
+      execute: async (context) => {
+        context.txStatus = "pending";
+        console.log("Executing transaction");
+      },
+      next: ["MonitorTransaction"],
+      events: ["txSubmitted"],
+    },
+    {
+      name: "MonitorTransaction",
+      execute: async (context) => {
+        context.txStatus = "confirmed";
+        context.walletBalance = context.walletBalance - context.currentPrice;
+        console.log("Monitoring transaction");
+      },
+      next: ["NotifyUser"],
+      events: ["txConfirmed", "txFailed"],
+    },
+    {
+      name: "NotifyUser",
+      execute: async (context) => {
+        context.notificationSent = true;
+        console.log("Notifying user");
+      },
+      next: ["CheckMarketConditions"],
+    },
+  ],
+  context: {},
+  schema: z.object({
+    walletBalance: z.number().optional(),
+    targetPrice: z.number().optional(),
+    currentPrice: z.number().optional(),
+    gasFees: z.number().optional(),
+    estimatedGas: z.number().optional(),
+    isWalletConnected: z.boolean().optional(),
+    marketVolatility: z.string().optional(),
+    hasLiquidity: z.boolean().optional(),
+    txHash: z.string().optional(),
+    txStatus: z.string().optional(),
+    notificationSent: z.boolean().optional(),
+  }),
+});
+
+const visualizer = graph.createVisualizer();
+console.log(visualizer.toMermaid());

package/index.ts

@@ -16,6 +16,7 @@ export * from "./graph/event-manager";
 export * from "./graph/index";
 export * from "./graph/node";
 export * from "./graph/observer";
+export * from "./graph/visualizer";
 
 export * from "./modules/memory";
 export * from "./modules/memory/adapters/in-memory";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai.ntellect/core",
-  "version": "0.7.10",
+  "version": "0.7.12",
   "description": "",
   "main": "dist/index.js",
   "scripts": {

package/test/graph/controller.test.ts

@@ -2,7 +2,7 @@ import { expect } from "chai";
 import { z } from "zod";
 import { GraphController } from "../../graph/controller";
 import { GraphFlow } from "../../graph/index";
-import { Node } from "../../types";
+import { GraphNodeConfig } from "../../types";
 
 describe("GraphController", () => {
   const TestSchema = z.object({
@@ -11,7 +11,7 @@ describe("GraphController", () => {
   });
 
   const createTestGraph = (name: string): GraphFlow<typeof TestSchema> => {
-    const nodes: Node<typeof TestSchema>[] = [
+    const nodes: GraphNodeConfig<typeof TestSchema>[] = [
       {
         name: "start",
         execute: async (context, params) => {