@flutchai/flutch-sdk 0.1.25 → 0.1.26

package/dist/index.d.cts

@@ -2,8 +2,9 @@ import { JSONSchema7 } from 'json-schema';
 import * as _nestjs_common from '@nestjs/common';
 import { OnModuleInit, Logger, Type, CanActivate, ExecutionContext, DynamicModule } from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
-import { HumanMessage, AIMessageChunk } from '@langchain/core/messages';
+import { HumanMessage, AIMessage, AIMessageChunk } from '@langchain/core/messages';
 import { BaseChannel, CompiledStateGraph, LangGraphRunnableConfig } from '@langchain/langgraph';
+import { MongoClient } from 'mongodb';
 import Redis from 'ioredis';
 import { Registry } from 'prom-client';
 import { Response, Request } from 'express';
@@ -956,6 +957,28 @@ declare class CallbackController {
     handleCallback(req: CallbackRequest): Promise<CallbackResult>;
 }
 
+interface BaseGraphContext {
+    messageId?: string;
+    threadId: string;
+    userId: string;
+    agentId: string;
+    companyId?: string;
+}
+interface BaseGraphConfig<TSettings = any, TContext extends BaseGraphContext = BaseGraphContext> {
+    thread_id: string;
+    checkpoint_ns: string;
+    checkpoint_id: string;
+    context: TContext;
+    metadata: {
+        userId: string;
+        agentId: string;
+        requestId: string;
+        graphType: string;
+        version: string;
+        workflowType: string;
+    };
+    graphSettings?: TSettings;
+}
 interface IGraphManifest {
     companySlug: string;
     name: string;
@@ -1000,14 +1023,9 @@ declare abstract class AbstractGraphBuilder<V extends string = string> {
     protected manifest?: IGraphManifest;
     constructor();
     abstract buildGraph(config: any): Promise<any>;
-    protected createGraphContext(payload: IGraphRequestPayload): {
-        messageId?: string;
-        threadId: string;
-        userId: string;
-        agentId: string;
-        companyId?: string;
-    };
+    protected createGraphContext(payload: IGraphRequestPayload): BaseGraphContext;
     prepareConfig(payload: IGraphRequestPayload): Promise<any>;
+    protected customizeConfig(config: any, payload: IGraphRequestPayload): Promise<any>;
     protected loadManifest(): Promise<IGraphManifest | null>;
     protected loadManifestSync(): IGraphManifest | null;
     protected validateManifest(manifest: IGraphManifest): void;
@@ -1180,14 +1198,25 @@ declare class GraphEngineFactory {
     getEngine(engineType: GraphEngineType): IGraphEngine;
 }
 
+declare function createStaticMessage(content: string, config: LangGraphRunnableConfig): Promise<AIMessage>;
+
+interface MongoDBConfig {
+    uri?: string;
+    dbName?: string;
+    checkpointCollectionName?: string;
+    checkpointWritesCollectionName?: string;
+}
 interface UniversalGraphModuleOptions {
     engineType?: GraphEngineType;
     versioning?: VersioningConfig[];
+    mongodb?: MongoDBConfig;
 }
 declare class UniversalGraphModule {
     static forRoot(options: UniversalGraphModuleOptions): DynamicModule;
 }
 
+declare function createMongoClientAdapter(mongooseClient: any): MongoClient;
+
 declare enum ModelProvider {
     FLUTCH = "flutch",
     FLUTCH_MISTRAL = "flutch-mistral",
@@ -1568,4 +1597,4 @@ declare class StaticDiscovery implements ServiceDiscoveryProvider {
     }>>;
 }
 
-export { type ACLValidationResult, AbstractGraphBuilder, type ApiCallTracerOptions, AttachmentType, type AuditEntry, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, type BaseGraphState, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, type CallbackContext, CallbackController, type CallbackEntry, type CallbackHandler, type CallbackMetadata, CallbackMetrics, type CallbackPatch, type CallbackPatchHandler, CallbackPatchService, CallbackRateLimiter, type CallbackRecord, CallbackRegistry, type CallbackRequest, type CallbackResult, CallbackStore, CallbackTokenGuard, type CallbackUser, type ChannelState, type ChartType, ChatFeature, type ChatModelCreator, type ChatModelOrRunnable, type ChatModelWithTools, type CitationValue, type CompiledGraphFor, type ConcreteModels, type CustomDocument, DEFAULT_TRACER_OPTIONS, type DataEnvelope, ENDPOINT_METADATA_KEY, type EmbeddingModelCreator, Endpoint, type EndpointDescriptor, type EndpointHandler, type EndpointMetadata, type EndpointOptions, EndpointRegistry, EventProcessor, type ExtendedCallbackContext, type ExtendedCallbackHandler, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, type IAgentToolConfig, type IAgentToolConfiguration, type IAttachment, type IBaseEntity, type IChartDataPoint, type IChartDataset, type IChartValue, type IContentBlock, type IContentChain, type IDeletionInfo, type IGenericGraphType, type IGraphBuilder, type IGraphCompiler, type IGraphConfigurable, type IGraphEngine, type IGraphManifest, type IGraphMetadata, type IGraphMetadataRegistry, type IGraphRequestPayload, type IGraphResponsePayload, type IGraphRunnableConfig, type IGraphService, type IGraphServiceRegistry, type IGraphSettingsRepository, type IGraphTraceEvent, type IGraphTracer, type IGraphTypeMetadata, type IGraphTypeRegistry, type IModelInitializer, type IReasoningChain, type IReasoningStep, type IStoredMessageContent, type ISystemToolCredentials, type IToolCall, type IToolCatalog, type IToolConfigOption, type IToolExecutionContext, type ITracingEvent, type IUsageMetrics, type IUsageRecorder, type IdempotencyConfig, type IdempotencyEntry, IdempotencyManager, IdempotencyStatus, type KnowledgeBaseInfo, type LLMCallRecord, type LLMConfig, type LLModel, LangGraphEngine, type MappedChannels, McpConverter, type McpRuntimeClient, McpRuntimeHttpClient, type McpTool, McpToolFilter, type Model, type ModelByIdConfig, type ModelByIdWithTypeConfig, type ModelConfig, type ModelConfigFetcher, type ModelConfigWithToken, type ModelConfigWithTokenAndType, type ModelCreator, type ModelCreators, ModelInitializer, ModelProvider, ModelType, type RateLimitConfig, type RateLimitResult, type RequestContext, type RerankModelCreator, type RetrieveQueryOptions, type RetrieverConfig, RetrieverSearchType, RetrieverService, type RouterConfig, type ServiceDiscoveryProvider, SmartCallbackRouter, StaticDiscovery, type StreamAccumulator, StreamChannel, type StrictGraphRunnableConfig, TelegramPatchHandler, type ToolExecutionResult, type TraceApiCallResult, type TracingLevel, UIDispatchController, type UIDispatchDto, UIEndpoint, type UIEndpointClassMetadata, type UIEndpointMethodMetadata, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, type UniversalGraphModuleOptions, UniversalGraphService, type VersionResolution, type VersionResolutionOptions, type VersionRoute, VersionedGraphService, type VersioningConfig, VoyageAIRerank, type VoyageAIRerankConfig, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };
+export { type ACLValidationResult, AbstractGraphBuilder, type ApiCallTracerOptions, AttachmentType, type AuditEntry, type BaseGraphConfig, type BaseGraphContext, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, type BaseGraphState, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, type CallbackContext, CallbackController, type CallbackEntry, type CallbackHandler, type CallbackMetadata, CallbackMetrics, type CallbackPatch, type CallbackPatchHandler, CallbackPatchService, CallbackRateLimiter, type CallbackRecord, CallbackRegistry, type CallbackRequest, type CallbackResult, CallbackStore, CallbackTokenGuard, type CallbackUser, type ChannelState, type ChartType, ChatFeature, type ChatModelCreator, type ChatModelOrRunnable, type ChatModelWithTools, type CitationValue, type CompiledGraphFor, type ConcreteModels, type CustomDocument, DEFAULT_TRACER_OPTIONS, type DataEnvelope, ENDPOINT_METADATA_KEY, type EmbeddingModelCreator, Endpoint, type EndpointDescriptor, type EndpointHandler, type EndpointMetadata, type EndpointOptions, EndpointRegistry, EventProcessor, type ExtendedCallbackContext, type ExtendedCallbackHandler, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, type IAgentToolConfig, type IAgentToolConfiguration, type IAttachment, type IBaseEntity, type IChartDataPoint, type IChartDataset, type IChartValue, type IContentBlock, type IContentChain, type IDeletionInfo, type IGenericGraphType, type IGraphBuilder, type IGraphCompiler, type IGraphConfigurable, type IGraphEngine, type IGraphManifest, type IGraphMetadata, type IGraphMetadataRegistry, type IGraphRequestPayload, type IGraphResponsePayload, type IGraphRunnableConfig, type IGraphService, type IGraphServiceRegistry, type IGraphSettingsRepository, type IGraphTraceEvent, type IGraphTracer, type IGraphTypeMetadata, type IGraphTypeRegistry, type IModelInitializer, type IReasoningChain, type IReasoningStep, type IStoredMessageContent, type ISystemToolCredentials, type IToolCall, type IToolCatalog, type IToolConfigOption, type IToolExecutionContext, type ITracingEvent, type IUsageMetrics, type IUsageRecorder, type IdempotencyConfig, type IdempotencyEntry, IdempotencyManager, IdempotencyStatus, type KnowledgeBaseInfo, type LLMCallRecord, type LLMConfig, type LLModel, LangGraphEngine, type MappedChannels, McpConverter, type McpRuntimeClient, McpRuntimeHttpClient, type McpTool, McpToolFilter, type Model, type ModelByIdConfig, type ModelByIdWithTypeConfig, type ModelConfig, type ModelConfigFetcher, type ModelConfigWithToken, type ModelConfigWithTokenAndType, type ModelCreator, type ModelCreators, ModelInitializer, ModelProvider, ModelType, type MongoDBConfig, type RateLimitConfig, type RateLimitResult, type RequestContext, type RerankModelCreator, type RetrieveQueryOptions, type RetrieverConfig, RetrieverSearchType, RetrieverService, type RouterConfig, type ServiceDiscoveryProvider, SmartCallbackRouter, StaticDiscovery, type StreamAccumulator, StreamChannel, type StrictGraphRunnableConfig, TelegramPatchHandler, type ToolExecutionResult, type TraceApiCallResult, type TracingLevel, UIDispatchController, type UIDispatchDto, UIEndpoint, type UIEndpointClassMetadata, type UIEndpointMethodMetadata, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, type UniversalGraphModuleOptions, UniversalGraphService, type VersionResolution, type VersionResolutionOptions, type VersionRoute, VersionedGraphService, type VersioningConfig, VoyageAIRerank, type VoyageAIRerankConfig, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, createMongoClientAdapter, createStaticMessage, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };

package/dist/index.d.ts

@@ -2,8 +2,9 @@ import { JSONSchema7 } from 'json-schema';
 import * as _nestjs_common from '@nestjs/common';
 import { OnModuleInit, Logger, Type, CanActivate, ExecutionContext, DynamicModule } from '@nestjs/common';
 import { ConfigService } from '@nestjs/config';
-import { HumanMessage, AIMessageChunk } from '@langchain/core/messages';
+import { HumanMessage, AIMessage, AIMessageChunk } from '@langchain/core/messages';
 import { BaseChannel, CompiledStateGraph, LangGraphRunnableConfig } from '@langchain/langgraph';
+import { MongoClient } from 'mongodb';
 import Redis from 'ioredis';
 import { Registry } from 'prom-client';
 import { Response, Request } from 'express';
@@ -956,6 +957,28 @@ declare class CallbackController {
     handleCallback(req: CallbackRequest): Promise<CallbackResult>;
 }
 
+interface BaseGraphContext {
+    messageId?: string;
+    threadId: string;
+    userId: string;
+    agentId: string;
+    companyId?: string;
+}
+interface BaseGraphConfig<TSettings = any, TContext extends BaseGraphContext = BaseGraphContext> {
+    thread_id: string;
+    checkpoint_ns: string;
+    checkpoint_id: string;
+    context: TContext;
+    metadata: {
+        userId: string;
+        agentId: string;
+        requestId: string;
+        graphType: string;
+        version: string;
+        workflowType: string;
+    };
+    graphSettings?: TSettings;
+}
 interface IGraphManifest {
     companySlug: string;
     name: string;
@@ -1000,14 +1023,9 @@ declare abstract class AbstractGraphBuilder<V extends string = string> {
     protected manifest?: IGraphManifest;
     constructor();
     abstract buildGraph(config: any): Promise<any>;
-    protected createGraphContext(payload: IGraphRequestPayload): {
-        messageId?: string;
-        threadId: string;
-        userId: string;
-        agentId: string;
-        companyId?: string;
-    };
+    protected createGraphContext(payload: IGraphRequestPayload): BaseGraphContext;
     prepareConfig(payload: IGraphRequestPayload): Promise<any>;
+    protected customizeConfig(config: any, payload: IGraphRequestPayload): Promise<any>;
     protected loadManifest(): Promise<IGraphManifest | null>;
     protected loadManifestSync(): IGraphManifest | null;
     protected validateManifest(manifest: IGraphManifest): void;
@@ -1180,14 +1198,25 @@ declare class GraphEngineFactory {
     getEngine(engineType: GraphEngineType): IGraphEngine;
 }
 
+declare function createStaticMessage(content: string, config: LangGraphRunnableConfig): Promise<AIMessage>;
+
+interface MongoDBConfig {
+    uri?: string;
+    dbName?: string;
+    checkpointCollectionName?: string;
+    checkpointWritesCollectionName?: string;
+}
 interface UniversalGraphModuleOptions {
     engineType?: GraphEngineType;
     versioning?: VersioningConfig[];
+    mongodb?: MongoDBConfig;
 }
 declare class UniversalGraphModule {
     static forRoot(options: UniversalGraphModuleOptions): DynamicModule;
 }
 
+declare function createMongoClientAdapter(mongooseClient: any): MongoClient;
+
 declare enum ModelProvider {
     FLUTCH = "flutch",
     FLUTCH_MISTRAL = "flutch-mistral",
@@ -1568,4 +1597,4 @@ declare class StaticDiscovery implements ServiceDiscoveryProvider {
     }>>;
 }
 
-export { type ACLValidationResult, AbstractGraphBuilder, type ApiCallTracerOptions, AttachmentType, type AuditEntry, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, type BaseGraphState, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, type CallbackContext, CallbackController, type CallbackEntry, type CallbackHandler, type CallbackMetadata, CallbackMetrics, type CallbackPatch, type CallbackPatchHandler, CallbackPatchService, CallbackRateLimiter, type CallbackRecord, CallbackRegistry, type CallbackRequest, type CallbackResult, CallbackStore, CallbackTokenGuard, type CallbackUser, type ChannelState, type ChartType, ChatFeature, type ChatModelCreator, type ChatModelOrRunnable, type ChatModelWithTools, type CitationValue, type CompiledGraphFor, type ConcreteModels, type CustomDocument, DEFAULT_TRACER_OPTIONS, type DataEnvelope, ENDPOINT_METADATA_KEY, type EmbeddingModelCreator, Endpoint, type EndpointDescriptor, type EndpointHandler, type EndpointMetadata, type EndpointOptions, EndpointRegistry, EventProcessor, type ExtendedCallbackContext, type ExtendedCallbackHandler, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, type IAgentToolConfig, type IAgentToolConfiguration, type IAttachment, type IBaseEntity, type IChartDataPoint, type IChartDataset, type IChartValue, type IContentBlock, type IContentChain, type IDeletionInfo, type IGenericGraphType, type IGraphBuilder, type IGraphCompiler, type IGraphConfigurable, type IGraphEngine, type IGraphManifest, type IGraphMetadata, type IGraphMetadataRegistry, type IGraphRequestPayload, type IGraphResponsePayload, type IGraphRunnableConfig, type IGraphService, type IGraphServiceRegistry, type IGraphSettingsRepository, type IGraphTraceEvent, type IGraphTracer, type IGraphTypeMetadata, type IGraphTypeRegistry, type IModelInitializer, type IReasoningChain, type IReasoningStep, type IStoredMessageContent, type ISystemToolCredentials, type IToolCall, type IToolCatalog, type IToolConfigOption, type IToolExecutionContext, type ITracingEvent, type IUsageMetrics, type IUsageRecorder, type IdempotencyConfig, type IdempotencyEntry, IdempotencyManager, IdempotencyStatus, type KnowledgeBaseInfo, type LLMCallRecord, type LLMConfig, type LLModel, LangGraphEngine, type MappedChannels, McpConverter, type McpRuntimeClient, McpRuntimeHttpClient, type McpTool, McpToolFilter, type Model, type ModelByIdConfig, type ModelByIdWithTypeConfig, type ModelConfig, type ModelConfigFetcher, type ModelConfigWithToken, type ModelConfigWithTokenAndType, type ModelCreator, type ModelCreators, ModelInitializer, ModelProvider, ModelType, type RateLimitConfig, type RateLimitResult, type RequestContext, type RerankModelCreator, type RetrieveQueryOptions, type RetrieverConfig, RetrieverSearchType, RetrieverService, type RouterConfig, type ServiceDiscoveryProvider, SmartCallbackRouter, StaticDiscovery, type StreamAccumulator, StreamChannel, type StrictGraphRunnableConfig, TelegramPatchHandler, type ToolExecutionResult, type TraceApiCallResult, type TracingLevel, UIDispatchController, type UIDispatchDto, UIEndpoint, type UIEndpointClassMetadata, type UIEndpointMethodMetadata, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, type UniversalGraphModuleOptions, UniversalGraphService, type VersionResolution, type VersionResolutionOptions, type VersionRoute, VersionedGraphService, type VersioningConfig, VoyageAIRerank, type VoyageAIRerankConfig, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };
+export { type ACLValidationResult, AbstractGraphBuilder, type ApiCallTracerOptions, AttachmentType, type AuditEntry, type BaseGraphConfig, type BaseGraphContext, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, type BaseGraphState, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, type CallbackContext, CallbackController, type CallbackEntry, type CallbackHandler, type CallbackMetadata, CallbackMetrics, type CallbackPatch, type CallbackPatchHandler, CallbackPatchService, CallbackRateLimiter, type CallbackRecord, CallbackRegistry, type CallbackRequest, type CallbackResult, CallbackStore, CallbackTokenGuard, type CallbackUser, type ChannelState, type ChartType, ChatFeature, type ChatModelCreator, type ChatModelOrRunnable, type ChatModelWithTools, type CitationValue, type CompiledGraphFor, type ConcreteModels, type CustomDocument, DEFAULT_TRACER_OPTIONS, type DataEnvelope, ENDPOINT_METADATA_KEY, type EmbeddingModelCreator, Endpoint, type EndpointDescriptor, type EndpointHandler, type EndpointMetadata, type EndpointOptions, EndpointRegistry, EventProcessor, type ExtendedCallbackContext, type ExtendedCallbackHandler, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, type IAgentToolConfig, type IAgentToolConfiguration, type IAttachment, type IBaseEntity, type IChartDataPoint, type IChartDataset, type IChartValue, type IContentBlock, type IContentChain, type IDeletionInfo, type IGenericGraphType, type IGraphBuilder, type IGraphCompiler, type IGraphConfigurable, type IGraphEngine, type IGraphManifest, type IGraphMetadata, type IGraphMetadataRegistry, type IGraphRequestPayload, type IGraphResponsePayload, type IGraphRunnableConfig, type IGraphService, type IGraphServiceRegistry, type IGraphSettingsRepository, type IGraphTraceEvent, type IGraphTracer, type IGraphTypeMetadata, type IGraphTypeRegistry, type IModelInitializer, type IReasoningChain, type IReasoningStep, type IStoredMessageContent, type ISystemToolCredentials, type IToolCall, type IToolCatalog, type IToolConfigOption, type IToolExecutionContext, type ITracingEvent, type IUsageMetrics, type IUsageRecorder, type IdempotencyConfig, type IdempotencyEntry, IdempotencyManager, IdempotencyStatus, type KnowledgeBaseInfo, type LLMCallRecord, type LLMConfig, type LLModel, LangGraphEngine, type MappedChannels, McpConverter, type McpRuntimeClient, McpRuntimeHttpClient, type McpTool, McpToolFilter, type Model, type ModelByIdConfig, type ModelByIdWithTypeConfig, type ModelConfig, type ModelConfigFetcher, type ModelConfigWithToken, type ModelConfigWithTokenAndType, type ModelCreator, type ModelCreators, ModelInitializer, ModelProvider, ModelType, type MongoDBConfig, type RateLimitConfig, type RateLimitResult, type RequestContext, type RerankModelCreator, type RetrieveQueryOptions, type RetrieverConfig, RetrieverSearchType, RetrieverService, type RouterConfig, type ServiceDiscoveryProvider, SmartCallbackRouter, StaticDiscovery, type StreamAccumulator, StreamChannel, type StrictGraphRunnableConfig, TelegramPatchHandler, type ToolExecutionResult, type TraceApiCallResult, type TracingLevel, UIDispatchController, type UIDispatchDto, UIEndpoint, type UIEndpointClassMetadata, type UIEndpointMethodMetadata, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, type UniversalGraphModuleOptions, UniversalGraphService, type VersionResolution, type VersionResolutionOptions, type VersionRoute, VersionedGraphService, type VersioningConfig, VoyageAIRerank, type VoyageAIRerankConfig, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, createMongoClientAdapter, createStaticMessage, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };

package/dist/index.js

@@ -8,8 +8,12 @@ import { createHash, randomUUID, randomBytes } from 'crypto';
 import { Registry, collectDefaultMetrics, Counter, Histogram, Gauge } from 'prom-client';
 import { NestFactory, MetadataScanner, ModuleRef, DiscoveryModule } from '@nestjs/core';
 import * as net from 'net';
-import { ConfigModule } from '@nestjs/config';
+import { ConfigService, ConfigModule } from '@nestjs/config';
+import mongoose from 'mongoose';
+import { MongoDBSaver } from '@langchain/langgraph-checkpoint-mongodb';
 import * as LangGraph from '@langchain/langgraph';
+import { AIMessage } from '@langchain/core/messages';
+import { dispatchCustomEvent } from '@langchain/core/callbacks/dispatch';
 import { DynamicStructuredTool } from '@langchain/core/tools';
 import { z } from 'zod';
 import axios2 from 'axios';
@@ -3164,13 +3168,37 @@ var _AbstractGraphBuilder = class _AbstractGraphBuilder {
   /**
    * Basic configuration preparation for graph execution
    * Automatically creates context with tracer and usageRecorder
-   * Can be overridden in child classes to add specific logic
+   * Handles message deserialization and LangGraph-compatible structure
+   * FINAL method - cannot be overridden. Use customizeConfig() hook instead.
    */
   async prepareConfig(payload) {
     const context = this.createGraphContext(payload);
-    return {
+    let message = payload.message;
+    if (payload.message && typeof payload.message === "object" && "lc" in payload.message) {
+      try {
+        const { load } = await import('@langchain/core/load');
+        message = await load(JSON.stringify(payload.message));
+        this.logger.debug({
+          message: "Deserialized BaseMessage using load()",
+          type: message.constructor?.name
+        });
+      } catch (error) {
+        this.logger.warn({
+          message: "Failed to deserialize message",
+          error: error.message
+        });
+      }
+    }
+    const baseConfig = {
+      // Input for LangGraph (messages array)
+      input: message ? {
+        messages: [message]
+      } : void 0,
+      // Configurable settings for LangGraph checkpointing and context
       configurable: {
         thread_id: payload.threadId,
+        checkpoint_ns: this.graphType,
+        checkpoint_id: `${payload.threadId}-${Date.now()}`,
         context,
         // Add metadata for compatibility
         metadata: {
@@ -3178,10 +3206,39 @@ var _AbstractGraphBuilder = class _AbstractGraphBuilder {
           agentId: payload.agentId,
           requestId: payload.requestId,
           graphType: this.graphType,
-          version: this.version
-        }
+          version: this.version,
+          workflowType: this.graphType
+        },
+        // Graph-specific settings from payload
+        graphSettings: payload.graphSettings || {}
       }
     };
+    const finalConfig = await this.customizeConfig(baseConfig, payload);
+    return finalConfig;
+  }
+  /**
+   * Hook for customizing config after base preparation
+   * Override this method in child classes to add/modify config fields
+   *
+   * @param config - Base config prepared by SDK
+   * @param payload - Original request payload
+   * @returns Modified config
+   *
+   * @example
+   * ```typescript
+   * protected async customizeConfig(config: any, payload: IGraphRequestPayload): Promise<any> {
+   *   // Add custom fields
+   *   config.configurable.myCustomField = "value";
+   *
+   *   // Modify existing fields
+   *   config.configurable.context.customData = await this.loadCustomData(payload);
+   *
+   *   return config;
+   * }
+   * ```
+   */
+  async customizeConfig(config, payload) {
+    return config;
   }
   /**
    * Load graph manifest (if using manifest-based approach)
@@ -3894,6 +3951,24 @@ async function bootstrap(AppModule, options = {}) {
 // src/core/index.ts
 init_builder_registry_service();
 
+// src/core/mongodb/mongodb-client.adapter.ts
+function createMongoClientAdapter(mongooseClient) {
+  if (!mongooseClient || typeof mongooseClient.db !== "function") {
+    throw new Error(
+      "Invalid MongoDB client: missing required methods. Expected MongoClient from mongoose.Connection.getClient()"
+    );
+  }
+  const requiredMethods = ["db", "close", "connect"];
+  for (const method of requiredMethods) {
+    if (typeof mongooseClient[method] !== "function") {
+      throw new Error(
+        `Invalid MongoDB client: missing required method '${method}'`
+      );
+    }
+  }
+  return mongooseClient;
+}
+
 // src/messages/attachments.ts
 var AttachmentType = /* @__PURE__ */ ((AttachmentType2) => {
   AttachmentType2["IMAGE"] = "image";
@@ -4334,8 +4409,8 @@ function getLangGraphDispatch() {
   if (cachedDispatch !== void 0) {
     return cachedDispatch;
   }
-  const { dispatchCustomEvent } = LangGraph;
-  cachedDispatch = typeof dispatchCustomEvent === "function" ? dispatchCustomEvent : null;
+  const { dispatchCustomEvent: dispatchCustomEvent2 } = LangGraph;
+  cachedDispatch = typeof dispatchCustomEvent2 === "function" ? dispatchCustomEvent2 : null;
   return cachedDispatch;
 }
 function sanitizeTraceData(value, depth = 0, seen = /* @__PURE__ */ new WeakSet(), options) {
@@ -4642,6 +4717,14 @@ var EventProcessor = class {
    */
   processEvent(acc, event, onPartial) {
     this.captureTraceEvent(acc, event);
+    if (event.event === "on_custom_event" && event.data) {
+      const channel = event.metadata?.stream_channel ?? "text" /* TEXT */;
+      if (event.name === "send_static_message" && event.data.content) {
+        const blocks = this.normalizeContentBlocks(event.data.content);
+        this.processContentStream(acc, channel, blocks, onPartial);
+      }
+      return;
+    }
     if (event.event === "on_chat_model_stream" && event.data?.chunk?.content) {
       const channel = event.metadata?.stream_channel ?? "text" /* TEXT */;
       const blocks = this.normalizeContentBlocks(event.data.chunk.content);
@@ -4766,6 +4849,7 @@ var EventProcessor = class {
       }
       if (state.currentBlock) {
         state.contentChain.push(state.currentBlock);
+        state.currentBlock = null;
       }
       if (state.contentChain.length > 0) {
         allChains.push({
@@ -5128,6 +5212,13 @@ LangGraphEngine = __decorateClass([
   Injectable(),
   __decorateParam(1, Optional())
 ], LangGraphEngine);
+async function createStaticMessage(content, config) {
+  const message = new AIMessage({
+    content
+  });
+  await dispatchCustomEvent("send_static_message", { content }, config);
+  return message;
+}
 
 // src/core/universal-graph.module.ts
 init_builder_registry_service();
@@ -5315,6 +5406,58 @@ var UniversalGraphModule = class {
         },
         inject: [CallbackRegistry]
       },
+      // MongoDB connection (optional - only if mongodb config provided)
+      ...options.mongodb ? [
+        {
+          provide: "MONGO_CONNECTION",
+          useFactory: async (configService) => {
+            const logger2 = new Logger("UniversalGraphModule");
+            const mongoUri = options.mongodb?.uri || configService.get("MONGODB_URI") || process.env.MONGODB_URI;
+            const dbName = options.mongodb?.dbName || configService.get("MONGO_DB_NAME") || process.env.MONGO_DB_NAME;
+            if (!mongoUri) {
+              throw new Error(
+                "MONGODB_URI is not defined in options, config, or environment"
+              );
+            }
+            logger2.log(
+              `Connecting to MongoDB: ${mongoUri?.substring(0, 50) + "..."}`
+            );
+            try {
+              await mongoose.connect(mongoUri, { dbName });
+              logger2.log(
+                `Successfully connected to MongoDB (db: ${dbName})`
+              );
+              return mongoose.connection;
+            } catch (error) {
+              logger2.error("Failed to connect to MongoDB", error);
+              throw error;
+            }
+          },
+          inject: [ConfigService]
+        },
+        // MongoDB checkpointer
+        {
+          provide: "CHECKPOINTER",
+          useFactory: async (connection, configService) => {
+            const logger2 = new Logger("UniversalGraphModule");
+            const dbName = options.mongodb?.dbName || configService.get("MONGO_DB_NAME") || process.env.MONGO_DB_NAME;
+            const checkpointCollectionName = options.mongodb?.checkpointCollectionName || "checkpoints";
+            const checkpointWritesCollectionName = options.mongodb?.checkpointWritesCollectionName || "checkpoint_writes";
+            logger2.log(
+              `Creating CHECKPOINTER with collections: ${checkpointCollectionName}, ${checkpointWritesCollectionName}`
+            );
+            const mongooseClient = connection.getClient();
+            const mongoClient = createMongoClientAdapter(mongooseClient);
+            return new MongoDBSaver({
+              client: mongoClient,
+              dbName,
+              checkpointCollectionName,
+              checkpointWritesCollectionName
+            });
+          },
+          inject: ["MONGO_CONNECTION", ConfigService]
+        }
+      ] : [],
       {
         provide: "GRAPH_ENGINE",
         useFactory: (langGraphEngine) => langGraphEngine,
@@ -7186,6 +7329,6 @@ StaticDiscovery = __decorateClass([
   Injectable()
 ], StaticDiscovery);
 
-export { AbstractGraphBuilder, AttachmentType, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, CallbackController, CallbackMetrics, CallbackPatchService, CallbackRateLimiter, CallbackRegistry, CallbackStore, CallbackTokenGuard, ChatFeature, DEFAULT_TRACER_OPTIONS, ENDPOINT_METADATA_KEY, Endpoint, EndpointRegistry, EventProcessor, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, IdempotencyManager, IdempotencyStatus, LangGraphEngine, McpConverter, McpRuntimeHttpClient, McpToolFilter, ModelInitializer, ModelProvider, ModelType, RetrieverSearchType, RetrieverService, SmartCallbackRouter, StaticDiscovery, StreamChannel, TelegramPatchHandler, UIDispatchController, UIEndpoint, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, UniversalGraphService, VersionedGraphService, VoyageAIRerank, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };
+export { AbstractGraphBuilder, AttachmentType, GraphController as BaseGraphServiceController, UniversalGraphModule as BaseGraphServiceModule, BuilderRegistryService, Callback, CallbackACL, CallbackAuditAction, CallbackAuditor, CallbackController, CallbackMetrics, CallbackPatchService, CallbackRateLimiter, CallbackRegistry, CallbackStore, CallbackTokenGuard, ChatFeature, DEFAULT_TRACER_OPTIONS, ENDPOINT_METADATA_KEY, Endpoint, EndpointRegistry, EventProcessor, FileBasedDiscovery, GraphController, GraphEngineFactory, GraphEngineType, GraphManifestSchema, GraphManifestValidator, GraphServiceTokens, GraphTypeUtils, IdempotencyManager, IdempotencyStatus, LangGraphEngine, McpConverter, McpRuntimeHttpClient, McpToolFilter, ModelInitializer, ModelProvider, ModelType, RetrieverSearchType, RetrieverService, SmartCallbackRouter, StaticDiscovery, StreamChannel, TelegramPatchHandler, UIDispatchController, UIEndpoint, UIEndpointsDiscoveryService, UniversalCallbackService, UniversalGraphModule, UniversalGraphService, VersionedGraphService, VoyageAIRerank, WebPatchHandler, WithCallbacks, WithEndpoints, WithUIEndpoints, bootstrap, createEndpointDescriptors, createMongoClientAdapter, createStaticMessage, findCallbackMethod, findEndpointMethod, getCallbackMetadata, getEndpointMetadata, getUIEndpointClassMetadata, getUIEndpointMethodsMetadata, hasCallbacks, hasUIEndpoints, registerFinanceExampleCallback, registerUIEndpointsFromClass, sanitizeTraceData, traceApiCall };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map