@ai.ntellect/core 0.6.21 → 0.7.1

package/graph/logger.ts

@@ -0,0 +1,70 @@
+/**
+ * Handles logging operations for a graph instance
+ * Provides methods for adding, retrieving, and managing logs with optional verbose output
+ */
+export class GraphLogger {
+  private logs: string[] = [];
+  private verbose: boolean = false;
+
+  /**
+   * Creates a new GraphLogger instance
+   * @param graphName - The name of the graph this logger is associated with
+   * @param verbose - Whether to output logs to console in real-time
+   */
+  constructor(private graphName: string, verbose: boolean = false) {
+    this.verbose = verbose;
+  }
+
+  /**
+   * Adds a new log entry with timestamp
+   * @param message - The message to log
+   */
+  public addLog(message: string): void {
+    const timestamp = new Date().toISOString();
+    const logMessage = `[${timestamp}] ${message}`;
+    this.logs.push(logMessage);
+    if (this.verbose) {
+      console.log(`${this.graphName} - ${message}`);
+    }
+  }
+
+  /**
+   * Returns a copy of all stored logs
+   * @returns Array of log messages
+   */
+  public getLogs(): string[] {
+    return [...this.logs];
+  }
+
+  /**
+   * Clears all stored logs
+   */
+  public clearLogs(): void {
+    this.logs = [];
+  }
+
+  /**
+   * Sets the verbose mode
+   * @param enabled - Whether to enable verbose mode
+   */
+  public setVerbose(enabled: boolean): void {
+    this.verbose = enabled;
+  }
+
+  /**
+   * Gets the current verbose mode status
+   * @returns Current verbose mode state
+   */
+  public isVerbose(): boolean {
+    return this.verbose;
+  }
+
+  /**
+   * Logs a message to console with graph name prefix
+   * @param message - The message to log
+   * @param data - Optional data to log
+   */
+  log(message: string, data?: any): void {
+    console.log(`[Graph ${this.graphName}] ${message}`, data);
+  }
+}

package/graph/node.ts

@@ -0,0 +1,398 @@
+import { BehaviorSubject, Subject } from "rxjs";
+import { ZodSchema } from "zod";
+import { GraphContext, GraphEvent, Node } from "../types";
+import { GraphEventManager } from "./event-manager";
+import { GraphLogger } from "./logger";
+
+/**
+ * Represents a node in the graph that can execute operations and manage state
+ * @template T - The Zod schema type for validation
+ */
+export class GraphNode<T extends ZodSchema> {
+  /**
+   * Creates a new GraphNode instance
+   * @param nodes - Map of all nodes in the graph
+   * @param logger - Logger instance for tracking node operations
+   * @param eventManager - Manager for handling graph events
+   * @param eventSubject - Subject for emitting events
+   * @param stateSubject - Subject for managing graph state
+   */
+  constructor(
+    private nodes: Map<string, Node<T, any>>,
+    private logger: GraphLogger,
+    private eventManager: GraphEventManager<T>,
+    private eventSubject: Subject<GraphEvent<T>>,
+    private stateSubject: BehaviorSubject<GraphContext<T>>
+  ) {}
+
+  /**
+   * Emits an event with the specified type and payload
+   * @param type - The type of event to emit
+   * @param payload - The data associated with the event
+   * @private
+   */
+  private emitEvent(type: string, payload: any) {
+    this.logger.addLog(`📢 Event: ${type}`);
+    const event = {
+      type,
+      payload: {
+        ...payload,
+        name:
+          type === "nodeStateChanged"
+            ? payload.name || payload.nodeName
+            : payload.name,
+        context: { ...payload.context },
+      },
+      timestamp: Date.now(),
+    };
+
+    this.eventSubject.next(event);
+    this.eventManager.emitEvent(type, event);
+
+    // Update state subject only for state changes
+    if (type === "nodeStateChanged") {
+      this.stateSubject.next({ ...payload.context });
+    }
+  }
+
+  /**
+   * Executes a node with the given name and context
+   * @param nodeName - The name of the node to execute
+   * @param context - The current graph context
+   * @param inputs - Input data for the node
+   * @param triggeredByEvent - Whether the execution was triggered by an event
+   * @throws Error if the node is not found or execution fails
+   */
+  async executeNode(
+    nodeName: string,
+    context: GraphContext<T>,
+    inputs: any,
+    triggeredByEvent: boolean = false
+  ): Promise<void> {
+    const node = this.nodes.get(nodeName);
+    if (!node) throw new Error(`Node "${nodeName}" not found.`);
+
+    this.logger.addLog(`🚀 Starting node "${nodeName}"`);
+    this.emitEvent("nodeStarted", { name: nodeName, context: { ...context } });
+
+    try {
+      const contextProxy = new Proxy(context, {
+        set: (target, prop, value) => {
+          const oldValue = target[prop];
+          target[prop] = value;
+
+          this.emitEvent("nodeStateChanged", {
+            nodeName,
+            name: nodeName,
+            property: prop.toString(),
+            oldValue,
+            newValue: value,
+            context: { ...target },
+          });
+
+          return true;
+        },
+        get: (target, prop) => {
+          return target[prop];
+        },
+      });
+
+      if (node.condition && !node.condition(contextProxy)) {
+        this.logger.addLog(
+          `⏭️ Skipping node "${nodeName}" - condition not met`
+        );
+        return;
+      }
+
+      if (node.inputs) {
+        await this.validateInputs(node, inputs, nodeName);
+      }
+
+      if (node.retry && node.retry.maxAttempts > 0) {
+        await this.executeWithRetry(node, contextProxy, inputs, nodeName);
+      } else {
+        await node.execute(contextProxy, inputs);
+      }
+
+      if (node.outputs) {
+        await this.validateOutputs(node, contextProxy, nodeName);
+      }
+
+      if (!triggeredByEvent) {
+        const nextNodes =
+          typeof node.next === "function"
+            ? node.next(contextProxy)
+            : node.next || [];
+
+        for (const nextNodeName of nextNodes) {
+          const nextNode = this.nodes.get(nextNodeName);
+          if (
+            nextNode &&
+            (!nextNode.condition || nextNode.condition(contextProxy))
+          ) {
+            await this.executeNode(
+              nextNodeName,
+              contextProxy,
+              undefined,
+              false
+            );
+          }
+        }
+      }
+
+      if (!triggeredByEvent) {
+        await this.handleEvents(node, nodeName, contextProxy);
+      }
+
+      this.logger.addLog(`✅ Node "${nodeName}" executed successfully`);
+      this.emitEvent("nodeCompleted", {
+        name: nodeName,
+        context: { ...contextProxy },
+      });
+    } catch (error: any) {
+      const errorToThrow =
+        error instanceof Error
+          ? error
+          : new Error(error.message || "Unknown error");
+
+      this.logger.addLog(
+        `❌ Error in node "${nodeName}": ${errorToThrow.message}`
+      );
+
+      this.emitEvent("nodeError", {
+        name: nodeName,
+        error: errorToThrow,
+        context,
+      });
+
+      throw errorToThrow;
+    }
+  }
+
+  /**
+   * Validates the inputs for a node using its schema
+   * @param node - The node whose inputs need validation
+   * @param inputs - The input data to validate
+   * @param nodeName - The name of the node (for error messages)
+   * @throws Error if validation fails
+   * @private
+   */
+  private async validateInputs(
+    node: Node<T, any>,
+    inputs: any,
+    nodeName: string
+  ): Promise<void> {
+    if (!inputs) {
+      throw new Error(`Inputs required for node "${nodeName}"`);
+    }
+
+    try {
+      return node.inputs!.parse(inputs);
+    } catch (error: any) {
+      throw new Error(
+        error.errors?.[0]?.message || error.message || "Input validation failed"
+      );
+    }
+  }
+
+  /**
+   * Validates the outputs of a node against its schema
+   * @param node - The node whose outputs need validation
+   * @param context - The current graph context
+   * @param nodeName - The name of the node (for error messages)
+   * @throws Error if validation fails
+   * @private
+   */
+  private async validateOutputs(
+    node: Node<T, any>,
+    context: GraphContext<T>,
+    nodeName: string
+  ): Promise<void> {
+    try {
+      node.outputs!.parse(context);
+    } catch (error: any) {
+      throw new Error(
+        error.errors?.[0]?.message ||
+          error.message ||
+          "Output validation failed"
+      );
+    }
+  }
+
+  /**
+   * Handles event-related operations for a node
+   * @param node - The node whose events need handling
+   * @param nodeName - The name of the node
+   * @param context - The current graph context
+   * @private
+   */
+  private async handleEvents(
+    node: Node<T, any>,
+    nodeName: string,
+    context: GraphContext<T>
+  ): Promise<void> {
+    if (node.correlateEvents) {
+      await this.handleCorrelatedEvents(node, nodeName);
+    }
+
+    if (node.waitForEvents) {
+      await this.handleWaitForEvents(node, nodeName);
+    }
+  }
+
+  /**
+   * Executes a node with retry logic
+   * @param node - The node to execute
+   * @param contextProxy - The proxied graph context
+   * @param inputs - Input data for the node
+   * @param nodeName - The name of the node
+   * @throws Error if all retry attempts fail
+   * @private
+   */
+  private async executeWithRetry(
+    node: Node<T, any>,
+    contextProxy: GraphContext<T>,
+    inputs: any,
+    nodeName: string
+  ): Promise<void> {
+    let attempts = 0;
+    let lastError: Error | null = null;
+
+    while (attempts < node.retry!.maxAttempts) {
+      try {
+        this.logger.addLog(
+          `🔄 Attempt ${attempts + 1}/${node.retry!.maxAttempts}`
+        );
+        await node.execute(contextProxy, inputs);
+        return;
+      } catch (error: any) {
+        lastError = error instanceof Error ? error : new Error(error.message);
+        attempts++;
+        this.logger.addLog(
+          `❌ Attempt ${attempts} failed: ${lastError.message}`
+        );
+
+        if (attempts === node.retry!.maxAttempts) {
+          if (node.retry!.onRetryFailed && lastError) {
+            await this.handleRetryFailure(
+              node,
+              lastError,
+              contextProxy,
+              nodeName
+            );
+          }
+          throw lastError;
+        }
+
+        await new Promise((resolve) =>
+          setTimeout(resolve, node.retry?.delay || 0)
+        );
+      }
+    }
+  }
+
+  /**
+   * Handles the failure of retry attempts
+   * @param node - The node that failed
+   * @param error - The error that caused the failure
+   * @param context - The current graph context
+   * @param nodeName - The name of the node
+   * @private
+   */
+  private async handleRetryFailure(
+    node: Node<T, any>,
+    error: Error,
+    context: GraphContext<T>,
+    nodeName: string
+  ): Promise<void> {
+    this.logger.addLog(
+      `🔄 Executing retry failure handler for node "${nodeName}"`
+    );
+    try {
+      if (node.retry?.onRetryFailed) {
+        await node.retry.onRetryFailed(error, context);
+        if (node.retry.continueOnFailed) {
+          this.logger.addLog(
+            `✅ Retry failure handler succeeded - continuing execution`
+          );
+          return;
+        }
+        this.logger.addLog(
+          `⚠️ Retry failure handler executed but node will still fail`
+        );
+      }
+    } catch (handlerError: any) {
+      this.logger.addLog(
+        `❌ Retry failure handler failed: ${handlerError.message}`
+      );
+      throw handlerError;
+    }
+  }
+
+  /**
+   * Handles correlated events for a node
+   * @param node - The node with correlated events
+   * @param nodeName - The name of the node
+   * @throws Error if correlation fails or timeout occurs
+   * @private
+   */
+  private async handleCorrelatedEvents(
+    node: Node<T, any>,
+    nodeName: string
+  ): Promise<void> {
+    if (node.correlateEvents) {
+      const { events, timeout, correlation } = node.correlateEvents;
+      this.logger.addLog(
+        `⏳ Node "${nodeName}" waiting for correlated events: ${events.join(
+          ", "
+        )}`
+      );
+
+      try {
+        // Attendre les événements
+        const receivedEvents = await this.eventManager.waitForEvents(
+          events,
+          timeout
+        );
+
+        // Vérifier la corrélation
+        if (!correlation(receivedEvents)) {
+          this.logger.addLog(
+            `❌ Event correlation failed for node "${nodeName}"`
+          );
+          throw new Error(`Event correlation failed for node "${nodeName}"`);
+        }
+
+        this.logger.addLog(
+          `✅ Event correlation succeeded for node "${nodeName}"`
+        );
+      } catch (error) {
+        this.logger.addLog(
+          `❌ Error waiting for events: ${(error as Error).message}`
+        );
+        throw error;
+      }
+    }
+  }
+
+  /**
+   * Handles waiting for events
+   * @param node - The node waiting for events
+   * @param nodeName - The name of the node
+   * @throws Error if timeout occurs
+   * @private
+   */
+  private async handleWaitForEvents(
+    node: Node<T, any>,
+    nodeName: string
+  ): Promise<void> {
+    if (node.waitForEvents) {
+      const { events, timeout } = node.waitForEvents;
+      this.logger.addLog(
+        `⏳ Node "${nodeName}" waiting for events: ${events.join(", ")}`
+      );
+      await this.eventManager.waitForEvents(events, timeout);
+      this.logger.addLog(`✅ All events received for node "${nodeName}"`);
+    }
+  }
+}