@flutchai/flutch-sdk 0.7.6 → 0.7.8

package/dist/index.d.cts

@@ -366,8 +366,9 @@ declare class GraphController {
         timestamp: string;
     }>;
     getSupportedGraphTypes(): Promise<string[]>;
-    generateAnswer(payload: IGraphRequestPayload): Promise<IGraphResponsePayload>;
+    generateAnswer(payload: IGraphRequestPayload, res: Response): Promise<IGraphResponsePayload>;
     streamAnswer(payload: IGraphRequestPayload, res: Response): Promise<void>;
+    private cancelOnDisconnect;
     cancelGeneration(requestId: string): Promise<{
         message: string;
     }>;
@@ -1428,7 +1429,7 @@ declare class McpRuntimeHttpClient implements McpRuntimeClient {
     private readonly baseUrl;
     constructor(mcpRuntimeUrl?: string);
     getTools(): Promise<McpTool[]>;
-    executeTool(name: string, args: any, context?: any, mcpServers?: Record<string, any>[]): Promise<ToolExecutionResult>;
+    executeTool(name: string, args: any, context?: any, mcpServers?: Record<string, any>[], signal?: AbortSignal): Promise<ToolExecutionResult>;
     getToolStats(): Promise<any>;
     isHealthy(): Promise<boolean>;
     executeToolWithEvents(toolCallId: string, toolName: string, enrichedArgs: Record<string, any>, executionContext: Record<string, any>, config?: RunnableConfig, mcpServers?: Record<string, any>[]): Promise<{
@@ -1635,6 +1636,7 @@ declare class UniversalGraphService implements IGraphService {
 declare function bootstrap(AppModule: any, options?: {
     port?: number;
     globalPrefix?: string;
+    bodyLimit?: string;
 }): Promise<_nestjs_common.INestApplication<any>>;
 
 interface ApiCallTracerOptions {

package/dist/index.d.ts

@@ -366,8 +366,9 @@ declare class GraphController {
         timestamp: string;
     }>;
     getSupportedGraphTypes(): Promise<string[]>;
-    generateAnswer(payload: IGraphRequestPayload): Promise<IGraphResponsePayload>;
+    generateAnswer(payload: IGraphRequestPayload, res: Response): Promise<IGraphResponsePayload>;
     streamAnswer(payload: IGraphRequestPayload, res: Response): Promise<void>;
+    private cancelOnDisconnect;
     cancelGeneration(requestId: string): Promise<{
         message: string;
     }>;
@@ -1428,7 +1429,7 @@ declare class McpRuntimeHttpClient implements McpRuntimeClient {
     private readonly baseUrl;
     constructor(mcpRuntimeUrl?: string);
     getTools(): Promise<McpTool[]>;
-    executeTool(name: string, args: any, context?: any, mcpServers?: Record<string, any>[]): Promise<ToolExecutionResult>;
+    executeTool(name: string, args: any, context?: any, mcpServers?: Record<string, any>[], signal?: AbortSignal): Promise<ToolExecutionResult>;
     getToolStats(): Promise<any>;
     isHealthy(): Promise<boolean>;
     executeToolWithEvents(toolCallId: string, toolName: string, enrichedArgs: Record<string, any>, executionContext: Record<string, any>, config?: RunnableConfig, mcpServers?: Record<string, any>[]): Promise<{
@@ -1635,6 +1636,7 @@ declare class UniversalGraphService implements IGraphService {
 declare function bootstrap(AppModule: any, options?: {
     port?: number;
     globalPrefix?: string;
+    bodyLimit?: string;
 }): Promise<_nestjs_common.INestApplication<any>>;
 
 interface ApiCallTracerOptions {

package/dist/index.js

@@ -6,10 +6,10 @@ import * as fs from 'fs';
 import * as path2 from 'path';
 import * as os from 'os';
 import { ConfigService, ConfigModule } from '@nestjs/config';
+import { Response, json, urlencoded } from 'express';
 import * as net from 'net';
 import mongoose from 'mongoose';
 import { MongoDBSaver } from '@langchain/langgraph-checkpoint-mongodb';
-import { Response } from 'express';
 import * as crypto from 'crypto';
 import { randomUUID, createHash, randomBytes } from 'crypto';
 import { Registry, collectDefaultMetrics, Counter, Histogram, Gauge } from 'prom-client';
@@ -1960,6 +1960,14 @@ async function bootstrap(AppModule, options = {}) {
   setupRedisMock();
   const app = await NestFactory.create(AppModule);
   const logger3 = new Logger("Bootstrap");
+  const bodyLimit = options.bodyLimit || process.env.GRAPH_BODY_LIMIT || "10mb";
+  app.use(json({
+    limit: bodyLimit
+  }));
+  app.use(urlencoded({
+    limit: bodyLimit,
+    extended: true
+  }));
   const requestedPort = options.port || parseInt(process.env.PORT || "3100", 10);
   const port = await findAvailablePort(requestedPort);
   const globalPrefix = options.globalPrefix;
@@ -2093,7 +2101,8 @@ var GraphController = class _GraphController {
   async getSupportedGraphTypes() {
     return this.graphService.getSupportedGraphTypes();
   }
-  async generateAnswer(payload) {
+  async generateAnswer(payload, res) {
+    this.cancelOnDisconnect(res, payload.requestId);
     try {
       return await this.graphService.generateAnswer(payload);
     } catch (error) {
@@ -2106,6 +2115,7 @@ var GraphController = class _GraphController {
     res.setHeader("Transfer-Encoding", "chunked");
     res.setHeader("Cache-Control", "no-cache");
     res.setHeader("Connection", "keep-alive");
+    this.cancelOnDisconnect(res, payload.requestId);
     try {
       this.logger.debug(`[CONTROLLER] Starting streamAnswer for requestId: ${payload.requestId}`);
       const result = await this.graphService.streamAnswer(payload, (chunk) => {
@@ -2143,6 +2153,19 @@ var GraphController = class _GraphController {
       res.end();
     }
   }
+  /**
+  * A caller that hangs up before the answer is done — its own timeout, a
+  * restart, a cancelled parent run — no longer waits for it, so the run is
+  * cancelled: left going, it keeps calling models and tools nobody reads.
+  */
+  cancelOnDisconnect(res, requestId) {
+    if (!requestId) return;
+    res.on("close", () => {
+      if (res.writableFinished) return;
+      this.logger.warn(`[CONTROLLER] Caller disconnected before the answer: cancelling ${requestId}`);
+      this.graphService.cancelGeneration(requestId).catch((error) => this.logger.warn(`[CONTROLLER] Cancel after disconnect failed for ${requestId}: ${error.message}`));
+    });
+  }
   async cancelGeneration(requestId) {
     try {
       await this.graphService.cancelGeneration(requestId);
@@ -2216,9 +2239,13 @@ _ts_decorate8([
     description: "Answer generated"
   }),
   _ts_param4(0, Body()),
+  _ts_param4(1, Res({
+    passthrough: true
+  })),
   _ts_metadata6("design:type", Function),
   _ts_metadata6("design:paramtypes", [
-    typeof IGraphRequestPayload === "undefined" ? Object : IGraphRequestPayload
+    typeof IGraphRequestPayload === "undefined" ? Object : IGraphRequestPayload,
+    typeof Response === "undefined" ? Object : Response
   ]),
   _ts_metadata6("design:returntype", Promise)
 ], GraphController.prototype, "generateAnswer", null);
@@ -6999,7 +7026,7 @@ var McpRuntimeHttpClient = class _McpRuntimeHttpClient {
   /**
   * Execute a tool by name with given arguments
   */
-  async executeTool(name, args, context, mcpServers) {
+  async executeTool(name, args, context, mcpServers, signal) {
     try {
       this.logger.debug(`Executing tool: ${name} with args:`, args);
       const payload = {
@@ -7012,7 +7039,9 @@ var McpRuntimeHttpClient = class _McpRuntimeHttpClient {
       if (mcpServers && mcpServers.length > 0) {
         payload.mcpServers = mcpServers;
       }
-      const response = await this.httpClient.post("/tools/execute", payload);
+      const response = signal ? await this.httpClient.post("/tools/execute", payload, {
+        signal
+      }) : await this.httpClient.post("/tools/execute", payload);
       this.logger.log(`Tool ${name} executed successfully`);
       return response.data;
     } catch (error) {
@@ -7080,7 +7109,7 @@ var McpRuntimeHttpClient = class _McpRuntimeHttpClient {
           toolName
         ]
       }, JSON.stringify(enrichedArgs), parsedConfig.runId, void 0, parsedConfig.tags, parsedConfig.metadata, toolName);
-      const result = await this.executeTool(toolName, enrichedArgs, executionContext, mcpServers);
+      const result = await this.executeTool(toolName, enrichedArgs, executionContext, mcpServers, config?.signal);
       const content = result.success ? JSON.stringify(result) : result.error || JSON.stringify(result);
       await runManager?.handleToolEnd(content);
       return {
@@ -7122,8 +7151,8 @@ function toolSearchTerms(text) {
 }
 __name(toolSearchTerms, "toolSearchTerms");
 function schemaParams(schema) {
-  const json = schema;
-  const properties = json?.properties ?? json?.shape ?? {};
+  const json2 = schema;
+  const properties = json2?.properties ?? json2?.shape ?? {};
   return Object.entries(properties).flatMap(([name, property]) => property?.description ? [
     name,
     property.description
@@ -8838,8 +8867,8 @@ var IV_LENGTH = 16;
 function encryptTokens(tokens, key) {
   const iv = crypto.randomBytes(IV_LENGTH);
   const cipher = crypto.createCipheriv(ALGORITHM, Buffer.from(key), iv);
-  const json = JSON.stringify(tokens);
-  let encrypted = cipher.update(json, "utf8", "hex");
+  const json2 = JSON.stringify(tokens);
+  let encrypted = cipher.update(json2, "utf8", "hex");
   encrypted += cipher.final("hex");
   return iv.toString("hex") + ":" + encrypted;
 }