@flutchai/flutch-sdk 0.1.20 → 0.1.23

package/dist/index.cjs

@@ -177,8 +177,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");
   }
 });
 exports.EndpointRegistry = void 0;
@@ -450,8 +450,8 @@ exports.ENDPOINT_METADATA_KEY = void 0; var UI_ENDPOINT_CLASS_METADATA_KEY, UI_E
 var init_endpoint_decorators = __esm({
   "src/agent-ui/endpoint.decorators.ts"() {
     exports.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");
   }
 });
 exports.UIEndpointsDiscoveryService = void 0;
@@ -4448,8 +4448,14 @@ exports.EventProcessor = class EventProcessor {
   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: {},
@@ -4513,6 +4519,7 @@ exports.EventProcessor = class EventProcessor {
           input: block.input || "",
           output: ""
         };
+        state.pendingToolBlocks.push(state.currentBlock);
         this.sendDelta(
           channel,
           {
@@ -4631,24 +4638,36 @@ exports.EventProcessor = class EventProcessor {
     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;
     }