@flutchai/flutch-sdk 0.1.20 → 0.1.23

package/dist/index.d.cts

@@ -3,7 +3,7 @@ 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 { CompiledStateGraph, LangGraphRunnableConfig, BaseChannel } from '@langchain/langgraph';
+import { BaseChannel, CompiledStateGraph, LangGraphRunnableConfig } from '@langchain/langgraph';
 import Redis from 'ioredis';
 import { Registry } from 'prom-client';
 import { Response, Request } from 'express';
@@ -1124,6 +1124,7 @@ interface LLMCallRecord {
 interface ChannelState {
     contentChain: IContentBlock[];
     currentBlock: IContentBlock | null;
+    pendingToolBlocks: IContentBlock[];
 }
 interface StreamAccumulator {
     channels: Map<StreamChannel, ChannelState>;

package/dist/index.d.ts

@@ -3,7 +3,7 @@ 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 { CompiledStateGraph, LangGraphRunnableConfig, BaseChannel } from '@langchain/langgraph';
+import { BaseChannel, CompiledStateGraph, LangGraphRunnableConfig } from '@langchain/langgraph';
 import Redis from 'ioredis';
 import { Registry } from 'prom-client';
 import { Response, Request } from 'express';
@@ -1124,6 +1124,7 @@ interface LLMCallRecord {
 interface ChannelState {
     contentChain: IContentBlock[];
     currentBlock: IContentBlock | null;
+    pendingToolBlocks: IContentBlock[];
 }
 interface StreamAccumulator {
     channels: Map<StreamChannel, ChannelState>;

package/dist/index.js

@@ -1,5 +1,5 @@
 import 'reflect-metadata';
-import { Injectable, Post, UseGuards, Req, Controller, Inject, Get, Body, Res, Param, Logger, Optional, Module, UnauthorizedException, ForbiddenException, ValidationPipe, HttpException, HttpStatus, NotFoundException, InternalServerErrorException, Headers } from '@nestjs/common';
+import { Injectable, Post, UseGuards, Req, Controller, Inject, Get, Body, Res, Param, Logger, Optional, Module, UnauthorizedException, Headers, NotFoundException, InternalServerErrorException, ForbiddenException, ValidationPipe, HttpException, HttpStatus } from '@nestjs/common';
 import { ApiOperation, ApiResponse, ApiTags, ApiParam, DocumentBuilder, SwaggerModule } from '@nestjs/swagger';
 import * as fs from 'fs';
 import * as path2 from 'path';
@@ -148,8 +148,8 @@ function findCallbackMethod(target, handler) {
 var CALLBACK_METADATA_KEY, CALLBACK_CLASS_METADATA_KEY;
 var init_callback_decorators = __esm({
   "src/callbacks/callback.decorators.ts"() {
-    CALLBACK_METADATA_KEY = /* @__PURE__ */ Symbol("callbacks");
-    CALLBACK_CLASS_METADATA_KEY = /* @__PURE__ */ Symbol("callback_class");
+    CALLBACK_METADATA_KEY = Symbol("callbacks");
+    CALLBACK_CLASS_METADATA_KEY = Symbol("callback_class");
   }
 });
 var EndpointRegistry;
@@ -421,8 +421,8 @@ var ENDPOINT_METADATA_KEY, UI_ENDPOINT_CLASS_METADATA_KEY, UI_ENDPOINT_METHOD_ME
 var init_endpoint_decorators = __esm({
   "src/agent-ui/endpoint.decorators.ts"() {
     ENDPOINT_METADATA_KEY = "graph:endpoints";
-    UI_ENDPOINT_CLASS_METADATA_KEY = /* @__PURE__ */ Symbol("ui_endpoint_class");
-    UI_ENDPOINT_METHOD_METADATA_KEY = /* @__PURE__ */ Symbol("ui_endpoint_methods");
+    UI_ENDPOINT_CLASS_METADATA_KEY = Symbol("ui_endpoint_class");
+    UI_ENDPOINT_METHOD_METADATA_KEY = Symbol("ui_endpoint_methods");
   }
 });
 var UIEndpointsDiscoveryService;
@@ -4419,8 +4419,14 @@ var EventProcessor = class {
   createAccumulator() {
     return {
       channels: /* @__PURE__ */ new Map([
-        ["text" /* TEXT */, { contentChain: [], currentBlock: null }],
-        ["processing" /* PROCESSING */, { contentChain: [], currentBlock: null }]
+        [
+          "text" /* TEXT */,
+          { contentChain: [], currentBlock: null, pendingToolBlocks: [] }
+        ],
+        [
+          "processing" /* PROCESSING */,
+          { contentChain: [], currentBlock: null, pendingToolBlocks: [] }
+        ]
       ]),
       attachments: [],
       metadata: {},
@@ -4484,6 +4490,7 @@ var EventProcessor = class {
           input: block.input || "",
           output: ""
         };
+        state.pendingToolBlocks.push(state.currentBlock);
         this.sendDelta(
           channel,
           {
@@ -4602,24 +4609,36 @@ var EventProcessor = class {
     if (event.event === "on_tool_end") {
       const channel = event.metadata?.stream_channel ?? "text" /* TEXT */;
       const state = acc.channels.get(channel);
-      if (state?.currentBlock && state.currentBlock.type === "tool_use") {
+      if (!state) return;
+      const toolBlock = state.pendingToolBlocks.shift();
+      if (toolBlock && toolBlock.type === "tool_use") {
         const output = event.data?.output;
         const outputString = typeof output === "string" ? output : JSON.stringify(output, null, 2);
-        state.currentBlock.output = outputString;
+        toolBlock.output = outputString;
         this.sendDelta(
           channel,
           {
             type: "tool_output_chunk",
-            stepId: state.currentBlock.id,
+            stepId: toolBlock.id,
             chunk: outputString
           },
           onPartial
         );
         this.logger.log("\u2705 Tool execution completed", {
           toolName: event.name,
+          toolBlockId: toolBlock.id,
           outputPreview: outputString.substring(0, 200) + (outputString.length > 200 ? "..." : ""),
           runId: event.run_id
         });
+      } else {
+        this.logger.warn(
+          "\u26A0\uFE0F on_tool_end received but no pending tool block found",
+          {
+            toolName: event.name,
+            runId: event.run_id,
+            pendingCount: state.pendingToolBlocks.length
+          }
+        );
       }
       return;
     }