@flutchai/flutch-sdk 0.7.6 → 0.7.8

package/dist/index.cjs

@@ -8,10 +8,10 @@ var fs = require('fs');
 var path2 = require('path');
 var os = require('os');
 var config = require('@nestjs/config');
+var express = require('express');
 var net = require('net');
 var mongoose = require('mongoose');
 var langgraphCheckpointMongodb = require('@langchain/langgraph-checkpoint-mongodb');
-var express = require('express');
 var crypto = require('crypto');
 var promClient = require('prom-client');
 var messages = require('@langchain/core/messages');
@@ -1990,6 +1990,14 @@ async function bootstrap(AppModule, options = {}) {
   setupRedisMock();
   const app = await core.NestFactory.create(AppModule);
   const logger3 = new common.Logger("Bootstrap");
+  const bodyLimit = options.bodyLimit || process.env.GRAPH_BODY_LIMIT || "10mb";
+  app.use(express.json({
+    limit: bodyLimit
+  }));
+  app.use(express.urlencoded({
+    limit: bodyLimit,
+    extended: true
+  }));
   const requestedPort = options.port || parseInt(process.env.PORT || "3100", 10);
   const port = await findAvailablePort(requestedPort);
   const globalPrefix = options.globalPrefix;
@@ -2123,7 +2131,8 @@ exports.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) {
@@ -2136,6 +2145,7 @@ exports.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) => {
@@ -2173,6 +2183,19 @@ exports.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);
@@ -2246,9 +2269,13 @@ _ts_decorate8([
     description: "Answer generated"
   }),
   _ts_param4(0, common.Body()),
+  _ts_param4(1, common.Res({
+    passthrough: true
+  })),
   _ts_metadata6("design:type", Function),
   _ts_metadata6("design:paramtypes", [
-    typeof IGraphRequestPayload === "undefined" ? Object : IGraphRequestPayload
+    typeof IGraphRequestPayload === "undefined" ? Object : IGraphRequestPayload,
+    typeof express.Response === "undefined" ? Object : express.Response
   ]),
   _ts_metadata6("design:returntype", Promise)
 ], exports.GraphController.prototype, "generateAnswer", null);
@@ -7029,7 +7056,7 @@ exports.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 = {
@@ -7042,7 +7069,9 @@ exports.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) {
@@ -7110,7 +7139,7 @@ exports.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 {
@@ -7152,8 +7181,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
@@ -8868,8 +8897,8 @@ var IV_LENGTH = 16;
 function encryptTokens(tokens, key) {
   const iv = crypto__namespace.randomBytes(IV_LENGTH);
   const cipher = crypto__namespace.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;
 }