@flutchai/flutch-sdk 0.7.1 → 0.7.3

package/dist/index.cjs

@@ -5212,6 +5212,10 @@ exports.EventProcessor = class _EventProcessor {
       attachments: [],
       metadata: {},
       traceEvents: [],
+      traceMessages: {
+        messages: {},
+        keys: /* @__PURE__ */ new WeakMap()
+      },
       traceStartedAt: null,
       traceCompletedAt: null
     };
@@ -5613,6 +5617,7 @@ exports.EventProcessor = class _EventProcessor {
     const completedAt = acc.traceCompletedAt ?? startedAt;
     const trace = acc.traceEvents.length > 0 ? {
       events: acc.traceEvents,
+      messages: acc.traceMessages.messages,
       startedAt,
       completedAt,
       durationMs: Math.max(0, completedAt - startedAt),
@@ -5640,14 +5645,14 @@ exports.EventProcessor = class _EventProcessor {
     };
   }
   captureTraceEvent(acc, event) {
-    const normalized = this.normalizeTraceEvent(event);
+    const normalized = this.normalizeTraceEvent(event, acc.traceMessages);
     if (!normalized) {
       return;
     }
     acc.traceEvents.push(normalized);
     this.logger.debug(`[TRACE] Captured trace event ${normalized.type} for node ${normalized.nodeName} (${acc.traceEvents.length} total)`);
   }
-  normalizeTraceEvent(event) {
+  normalizeTraceEvent(event, traceMessages) {
     const type = event?.event ?? event?.type;
     if (!type) {
       return null;
@@ -5683,8 +5688,12 @@ exports.EventProcessor = class _EventProcessor {
     if (!hasLangGraphNode && type.startsWith("on_chain")) {
       return null;
     }
-    const metadata = sanitizeTraceData(event?.metadata);
-    const data = sanitizeTraceData(event?.data);
+    const metadata = sanitizeTraceData(event?.metadata, 0, /* @__PURE__ */ new WeakSet(), {
+      maxStringLength: TRACE_MAX_STRING
+    });
+    const data = sanitizeTraceData(refTraceMessages(event?.data, traceMessages), 0, /* @__PURE__ */ new WeakSet(), {
+      maxStringLength: TRACE_MAX_STRING
+    });
     const timestampSource = event?.timestamp ?? event?.time ?? event?.data?.timestamp ?? event?.data?.ts ?? Date.now();
     const timestamp = Number(timestampSource) || Date.now();
     const streamChannel = typeof event?.metadata?.stream_channel === "string" ? event.metadata.stream_channel : typeof metadata?.stream_channel === "string" ? metadata.stream_channel : void 0;
@@ -5704,6 +5713,39 @@ exports.EventProcessor = class _EventProcessor {
 exports.EventProcessor = _ts_decorate21([
   common.Injectable()
 ], exports.EventProcessor);
+var TRACE_MAX_STRING = 1e5;
+function refTraceMessages(value, store, depth = 0) {
+  if (depth > 6 || value === null || typeof value !== "object") return value;
+  if (Array.isArray(value)) {
+    if (isHistory(value)) {
+      return value.map((message) => ({
+        _traceMsg: messageKey(message, store)
+      }));
+    }
+    return value.map((item) => refTraceMessages(item, store, depth + 1));
+  }
+  const out = {};
+  for (const [key, item] of Object.entries(value)) {
+    out[key] = refTraceMessages(item, store, depth + 1);
+  }
+  return out;
+}
+__name(refTraceMessages, "refTraceMessages");
+function messageKey(message, store) {
+  const known = store.keys.get(message);
+  if (known) return known;
+  const key = `m${Object.keys(store.messages).length + 1}`;
+  store.keys.set(message, key);
+  store.messages[key] = sanitizeTraceData(message, 0, /* @__PURE__ */ new WeakSet(), {
+    maxStringLength: TRACE_MAX_STRING
+  });
+  return key;
+}
+__name(messageKey, "messageKey");
+function isHistory(items) {
+  return items.length > 0 && items.every((item) => item !== null && typeof item === "object" && !Array.isArray(item) && ("lc_kwargs" in item || "content" in item && ("type" in item || "role" in item)));
+}
+__name(isHistory, "isHistory");
 var als = new async_hooks.AsyncLocalStorage();
 function withFlutchContext(ctx, fn) {
   return als.run(ctx, fn);
@@ -5977,6 +6019,7 @@ exports.LangGraphEngine = class _LangGraphEngine {
           agentId: context.agentId || "unknown",
           companyId: context.companyId || "unknown",
           events: trace.events,
+          messages: trace.messages,
           totalEvents: trace.totalEvents,
           startedAt: trace.startedAt,
           completedAt: trace.completedAt || Date.now(),
@@ -6063,6 +6106,7 @@ exports.LangGraphEngine = class _LangGraphEngine {
         durationMs: payload.durationMs,
         status: payload.status || "success",
         error: payload.error,
+        messages: payload.messages,
         events: payload.events.map((event) => ({
           timestamp: event.timestamp ? new Date(event.timestamp).toISOString() : (/* @__PURE__ */ new Date()).toISOString(),
           meta: {
@@ -9137,6 +9181,7 @@ exports.modelSupportsPromptCache = modelSupportsPromptCache;
 exports.normalizeToolConfigs = normalizeToolConfigs;
 exports.parseToolSearchResult = parseToolSearchResult;
 exports.redactConnectionString = redactConnectionString;
+exports.refTraceMessages = refTraceMessages;
 exports.registerFinanceExampleCallback = registerFinanceExampleCallback;
 exports.registerUIEndpointsFromClass = registerUIEndpointsFromClass;
 exports.resolveOAuthProviderConfig = resolveOAuthProviderConfig;