@flutchai/flutch-sdk 0.1.19 → 0.1.22

package/dist/index.cjs

@@ -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: {},
@@ -4488,6 +4494,11 @@ exports.EventProcessor = class EventProcessor {
    */
   sendDelta(channel, delta, onPartial) {
     if (!onPartial) return;
+    if (delta.type === "step_started" || delta.type === "tool_output_chunk" || delta.type === "tool_input_chunk") {
+      this.logger.debug(
+        `[DELTA] type=${delta.type} channel=${channel} stepId=${delta.stepId || delta.step?.id} name=${delta.step?.name || "N/A"}`
+      );
+    }
     onPartial(
       JSON.stringify({
         channel,
@@ -4513,6 +4524,7 @@ exports.EventProcessor = class EventProcessor {
           input: block.input || "",
           output: ""
         };
+        state.pendingToolBlocks.push(state.currentBlock);
         this.sendDelta(
           channel,
           {
@@ -4631,24 +4643,39 @@ 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;
+      this.logger.debug(
+        `[on_tool_end] channel=${channel} pendingCount=${state.pendingToolBlocks.length} pendingIds=${state.pendingToolBlocks.map((b) => b.id).join(",")} currentBlock=${state.currentBlock?.id}`
+      );
+      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;
     }
@@ -5513,12 +5540,22 @@ ${paramDescriptions.join("\n")}`;
       name: mcpTool.name,
       description: enhancedDescription,
       schema,
-      func: async (input) => {
+      func: async (input, _runManager, config) => {
         logger2.log(`\u{1F527} [${mcpTool.name}] LLM INPUT: ${JSON.stringify(input)}`);
+        const configurable = config?.configurable;
+        const context = {
+          agentId: configurable?.agentId,
+          userId: configurable?.userId,
+          threadId: configurable?.thread_id
+        };
+        logger2.debug(
+          `\u{1F527} [${mcpTool.name}] Execution context: ${JSON.stringify(context)}`
+        );
         try {
           const request = {
             name: mcpTool.name,
-            arguments: input ?? {}
+            arguments: input ?? {},
+            context
           };
           logger2.log(`\u{1F527} [${mcpTool.name}] Calling MCP Runtime...`);
           const response = await axios2__default.default.post(
@@ -5639,13 +5676,14 @@ var McpToolFilter = class _McpToolFilter {
       );
       const mcpClient = {
         getTools: async () => dynamicTools,
-        executeTool: async (name, args) => {
+        executeTool: async (name, args, context) => {
           this.logger.debug(`[DEBUG] Executing tool ${name} with args:`, args);
           const response2 = await axios2__default.default.post(
             `${this.mcpRuntimeUrl}/tools/execute`,
             {
               name,
-              arguments: args || {}
+              arguments: args || {},
+              context
             }
           );
           return response2.data;
@@ -5688,12 +5726,13 @@ var McpToolFilter = class _McpToolFilter {
       this.logger.debug(`Retrieved ${allTools.length} total MCP tools`);
       const mcpClient = {
         getTools: async () => allTools,
-        executeTool: async (name, args) => {
+        executeTool: async (name, args, context) => {
           const response2 = await axios2__default.default.post(
             `${this.mcpRuntimeUrl}/tools/execute`,
             {
               name,
-              arguments: args || {}
+              arguments: args || {},
+              context
             }
           );
           return response2.data;