@flutchai/flutch-sdk 0.7.7 → 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<{

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<{

package/dist/index.js

@@ -2101,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) {
@@ -2114,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) => {
@@ -2151,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);
@@ -2224,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);
@@ -7007,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 = {
@@ -7020,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) {
@@ -7088,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 {