@flutchai/flutch-sdk 0.2.4 → 0.2.6

package/dist/index.cjs

@@ -4422,6 +4422,48 @@ function generateTextSummary(data, toolCallId) {
 
 // src/graph/attachment-tool-node.ts
 var DEFAULT_ATTACHMENT_THRESHOLD = Number(process.env.ATTACHMENT_THRESHOLD) || 4e3;
+var attachmentDataStore = /* @__PURE__ */ new Map();
+var cleanupTimers = /* @__PURE__ */ new Map();
+var AUTO_CLEANUP_MS = 10 * 60 * 1e3;
+function storeAttachmentData(key, data, threadId) {
+  const scope = threadId || "__global__";
+  let threadStore = attachmentDataStore.get(scope);
+  if (!threadStore) {
+    threadStore = /* @__PURE__ */ new Map();
+    attachmentDataStore.set(scope, threadStore);
+  }
+  threadStore.set(key, data);
+  const existingTimer = cleanupTimers.get(scope);
+  if (existingTimer) {
+    clearTimeout(existingTimer);
+  }
+  const timer = setTimeout(() => {
+    attachmentDataStore.delete(scope);
+    cleanupTimers.delete(scope);
+  }, AUTO_CLEANUP_MS);
+  if (timer.unref) timer.unref();
+  cleanupTimers.set(scope, timer);
+}
+function getAttachmentData(key, threadId) {
+  const scope = threadId || "__global__";
+  return attachmentDataStore.get(scope)?.get(key);
+}
+function clearAttachmentDataStore(threadId) {
+  if (threadId) {
+    attachmentDataStore.delete(threadId);
+    const timer = cleanupTimers.get(threadId);
+    if (timer) {
+      clearTimeout(timer);
+      cleanupTimers.delete(threadId);
+    }
+  } else {
+    attachmentDataStore.clear();
+    for (const timer of cleanupTimers.values()) {
+      clearTimeout(timer);
+    }
+    cleanupTimers.clear();
+  }
+}
 async function executeToolWithAttachments(params) {
   const {
     toolCall,
@@ -4433,17 +4475,23 @@ async function executeToolWithAttachments(params) {
     logger: logger2,
     threshold = DEFAULT_ATTACHMENT_THRESHOLD,
     injectIntoArg = "data",
-    sourceAttachmentId
+    sourceAttachmentId,
+    threadId
   } = params;
   const argsWithInjection = { ...enrichedArgs };
   try {
     if (shouldInjectData(argsWithInjection, attachments, injectIntoArg)) {
       const attachment = sourceAttachmentId ? attachments[sourceAttachmentId] : getLatestAttachment(attachments);
       if (attachment) {
-        argsWithInjection[injectIntoArg] = typeof attachment.data === "string" ? attachment.data : JSON.stringify(attachment.data);
-        logger2?.debug(
-          `[Attachment] Auto-injected data from attachment "${attachment.toolCallId}" into ${toolCall.name}.${injectIntoArg}`
-        );
+        const attachmentKey = sourceAttachmentId || attachment.toolCallId;
+        const storedData = getAttachmentData(attachmentKey, threadId);
+        const data = storedData !== void 0 ? storedData : attachment.data;
+        if (data != null) {
+          argsWithInjection[injectIntoArg] = typeof data === "string" ? data : JSON.stringify(data);
+          logger2?.debug(
+            `[Attachment] Auto-injected data from attachment "${attachment.toolCallId}" into ${toolCall.name}.${injectIntoArg} (source: ${storedData !== void 0 ? "memory" : "state"})`
+          );
+        }
       }
     }
   } catch (e) {
@@ -4463,8 +4511,14 @@ async function executeToolWithAttachments(params) {
         toolCall.name,
         toolCall.id
       );
+      storeAttachmentData(toolCall.id, attachment.data, threadId);
+      const stateAttachment = {
+        ...attachment,
+        data: null
+        // Data stored in memory, not in graph state
+      };
       logger2?.debug(
-        `[Attachment] Stored large result (${content.length} chars) as attachment "${toolCall.id}"`
+        `[Attachment] Stored large result (${content.length} chars) as attachment "${toolCall.id}" (data in memory, metadata in state)`
       );
       const toolMessage2 = new messages.ToolMessage({
         content: attachment.summary,
@@ -4473,7 +4527,7 @@ async function executeToolWithAttachments(params) {
       });
       return {
         toolMessage: toolMessage2,
-        attachment: { key: toolCall.id, value: attachment }
+        attachment: { key: toolCall.id, value: stateAttachment }
       };
     }
   } catch (e) {
@@ -4505,7 +4559,8 @@ function getLatestAttachment(attachments) {
 }
 var logger = new common.Logger("ApiCallTracer");
 var DEFAULT_TRACER_OPTIONS = {
-  maxStringLength: 5e3,
+  maxStringLength: 1e5,
+  // 100KB - enough for most tool outputs, prevents overflow
   maxDepth: 15
 };
 var cachedDispatch;
@@ -4571,6 +4626,9 @@ function sanitizeTraceData(value, depth = 0, seen = /* @__PURE__ */ new WeakSet(
     return null;
   }
   if (typeof value === "string") {
+    if (value.length > opts.maxStringLength) {
+      return `${value.slice(0, opts.maxStringLength)}\u2026 [truncated: ${value.length - opts.maxStringLength} chars]`;
+    }
     return value;
   }
   if (typeof value === "number" || typeof value === "boolean") {
@@ -4735,6 +4793,19 @@ exports.EventProcessor = class EventProcessor {
     }
     return [];
   }
+  /**
+   * Extract attachments from various input formats
+   * Handles both array format (IAttachment[]) and object format (Record<string, IGraphAttachment>)
+   */
+  extractAttachments(attachments) {
+    if (!attachments) {
+      return [];
+    }
+    const items = Array.isArray(attachments) ? attachments : typeof attachments === "object" ? Object.values(attachments) : [];
+    return items.filter(
+      (item) => item != null && "type" in item && "value" in item
+    );
+  }
   /**
    * Send delta to UI (unified format)
    */
@@ -4872,6 +4943,14 @@ exports.EventProcessor = class EventProcessor {
         const blocks = this.normalizeContentBlocks(event.data.content);
         this.processContentStream(acc, channel, blocks, onPartial);
       }
+      if (event.name === "send_attachments" && event.data.attachments) {
+        const attachments = event.data.attachments || [];
+        acc.attachments = [...acc.attachments, ...attachments];
+        this.logger.debug("[ATTACHMENTS] Extracted from send_attachments event", {
+          extractedCount: attachments.length,
+          totalAccCount: acc.attachments.length
+        });
+      }
       return;
     }
     if (event.event === "on_chat_model_stream" && event.data?.chunk?.content) {
@@ -4911,7 +4990,21 @@ exports.EventProcessor = class EventProcessor {
       }
       if (toolBlock && toolBlock.type === "tool_use") {
         const output = event.data?.output;
-        const outputString = typeof output === "string" ? output : JSON.stringify(output, null, 2);
+        let outputString;
+        try {
+          outputString = typeof output === "string" ? output : JSON.stringify(output, null, 2);
+        } catch {
+          outputString = typeof output?.content === "string" ? output.content : "[Output too large to display]";
+        }
+        const MAX_TOOL_OUTPUT_LENGTH = 5e4;
+        if (outputString.length > MAX_TOOL_OUTPUT_LENGTH) {
+          let cutAt = outputString.lastIndexOf("\n", MAX_TOOL_OUTPUT_LENGTH);
+          if (cutAt < MAX_TOOL_OUTPUT_LENGTH * 0.8) {
+            cutAt = MAX_TOOL_OUTPUT_LENGTH;
+          }
+          outputString = outputString.slice(0, cutAt) + `
+... [truncated: ${outputString.length - cutAt} chars]`;
+        }
         toolBlock.output = outputString;
         this.sendDelta(
           channel,
@@ -4955,32 +5048,6 @@ exports.EventProcessor = class EventProcessor {
       });
       return;
     }
-    if (event.event === "on_chain_end") {
-      const channel = event.metadata?.stream_channel ?? "text" /* TEXT */;
-      if (channel === "text" /* TEXT */) {
-        const output = event.data.output;
-        if (output?.answer) {
-          acc.attachments = [
-            ...acc.attachments,
-            ...output.answer.attachments || []
-          ];
-          acc.metadata = { ...acc.metadata, ...output.answer.metadata || {} };
-        } else if (output?.generation) {
-          acc.attachments = [
-            ...acc.attachments,
-            ...output.generation.attachments || []
-          ];
-          acc.metadata = {
-            ...acc.metadata,
-            ...output.generation.metadata || {}
-          };
-        } else if (output?.attachments || output?.metadata) {
-          acc.attachments = [...acc.attachments, ...output.attachments || []];
-          acc.metadata = { ...acc.metadata, ...output.metadata || {} };
-        }
-      }
-      return;
-    }
   }
   /**
    * Build final result from accumulator
@@ -5024,7 +5091,9 @@ exports.EventProcessor = class EventProcessor {
       textChains: allChains.filter((c) => c.channel === "text").length,
       processingChains: allChains.filter((c) => c.channel === "processing").length,
       totalSteps: allChains.reduce((sum, c) => sum + c.steps.length, 0),
-      textLength: text.length
+      textLength: text.length,
+      attachmentsCount: acc.attachments?.length || 0,
+      attachments: acc.attachments
     });
     return {
       content: {
@@ -5165,11 +5234,21 @@ exports.LangGraphEngine = class LangGraphEngine {
       this.logger.debug("[ENGINE] Signal assigned to preparedPayload.signal");
     }
     const input = await this.deserializeInput(preparedPayload.input || {});
-    const result = await graph.invoke(input, {
-      ...preparedPayload.config,
-      signal: preparedPayload.signal
-    });
-    return this.processGraphResult(result);
+    try {
+      const result = await graph.invoke(input, {
+        ...preparedPayload.config,
+        signal: preparedPayload.signal
+      });
+      return this.processGraphResult(result);
+    } finally {
+      const threadId = this.extractThreadId(preparedPayload);
+      if (threadId) {
+        clearAttachmentDataStore(threadId);
+        this.logger.debug(
+          `[ENGINE] Cleared attachment data store for thread: ${threadId}`
+        );
+      }
+    }
   }
   async streamGraph(graph, preparedPayload, onPartial, signal) {
     const acc = this.eventProcessor.createAccumulator();
@@ -5210,6 +5289,13 @@ exports.LangGraphEngine = class LangGraphEngine {
       this.logger.error(`[STREAM-ERROR] Stack trace: ${streamError.stack}`);
     } finally {
       await this.sendTraceFromAccumulator(acc, preparedPayload, streamError);
+      const threadId = this.extractThreadId(preparedPayload);
+      if (threadId) {
+        clearAttachmentDataStore(threadId);
+        this.logger.debug(
+          `[ENGINE] Cleared attachment data store for thread: ${threadId}`
+        );
+      }
     }
     const { content, trace } = this.eventProcessor.getResult(acc);
     this.logger.debug("[STREAM-RESULT] Got result from EventProcessor", {
@@ -5388,6 +5474,12 @@ exports.LangGraphEngine = class LangGraphEngine {
       });
     }
   }
+  /**
+   * Extract threadId from payload (used for attachment store cleanup)
+   */
+  extractThreadId(preparedPayload) {
+    return preparedPayload.configurable?.thread_id || preparedPayload.configurable?.context?.threadId || preparedPayload.config?.configurable?.thread_id || preparedPayload.config?.configurable?.context?.threadId || void 0;
+  }
   /**
    * Process graph execution result
    */
@@ -5413,6 +5505,9 @@ async function createStaticMessage(content, config) {
   await dispatch.dispatchCustomEvent("send_static_message", { content }, config);
   return message;
 }
+async function dispatchAttachments(attachments, config) {
+  await dispatch.dispatchCustomEvent("send_attachments", { attachments }, config);
+}
 
 // src/core/universal-graph.module.ts
 init_builder_registry_service();
@@ -7550,14 +7645,17 @@ exports.WithEndpoints = WithEndpoints;
 exports.WithUIEndpoints = WithUIEndpoints;
 exports._internals = _internals;
 exports.bootstrap = bootstrap;
+exports.clearAttachmentDataStore = clearAttachmentDataStore;
 exports.createEndpointDescriptors = createEndpointDescriptors;
 exports.createGraphAttachment = createGraphAttachment;
 exports.createMongoClientAdapter = createMongoClientAdapter;
 exports.createStaticMessage = createStaticMessage;
+exports.dispatchAttachments = dispatchAttachments;
 exports.executeToolWithAttachments = executeToolWithAttachments;
 exports.findCallbackMethod = findCallbackMethod;
 exports.findEndpointMethod = findEndpointMethod;
 exports.generateAttachmentSummary = generateAttachmentSummary;
+exports.getAttachmentData = getAttachmentData;
 exports.getCallbackMetadata = getCallbackMetadata;
 exports.getEndpointMetadata = getEndpointMetadata;
 exports.getUIEndpointClassMetadata = getUIEndpointClassMetadata;
@@ -7567,6 +7665,7 @@ exports.hasUIEndpoints = hasUIEndpoints;
 exports.registerFinanceExampleCallback = registerFinanceExampleCallback;
 exports.registerUIEndpointsFromClass = registerUIEndpointsFromClass;
 exports.sanitizeTraceData = sanitizeTraceData;
+exports.storeAttachmentData = storeAttachmentData;
 exports.traceApiCall = traceApiCall;
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map