@flutchai/flutch-sdk 0.7.2 → 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;
@@ -5686,7 +5691,7 @@ exports.EventProcessor = class _EventProcessor {
     const metadata = sanitizeTraceData(event?.metadata, 0, /* @__PURE__ */ new WeakSet(), {
       maxStringLength: TRACE_MAX_STRING
     });
-    const data = sanitizeTraceData(compactTraceMessages(event?.data), 0, /* @__PURE__ */ new WeakSet(), {
+    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();
@@ -5708,39 +5713,39 @@ exports.EventProcessor = class _EventProcessor {
 exports.EventProcessor = _ts_decorate21([
   common.Injectable()
 ], exports.EventProcessor);
-var TRACE_MAX_STRING = 2e4;
-var TRACE_KEEP_LAST_MESSAGES = 6;
-function compactTraceMessages(value, depth = 0) {
+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)) {
-    const items = isHistory(value) ? compactHistory(value) : value;
-    return items.map((item) => compactTraceMessages(item, depth + 1));
+    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] = compactTraceMessages(item, depth + 1);
+    out[key] = refTraceMessages(item, store, depth + 1);
   }
   return out;
 }
-__name(compactTraceMessages, "compactTraceMessages");
+__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 > TRACE_KEEP_LAST_MESSAGES + 2 && items.every((item) => item !== null && typeof item === "object" && !Array.isArray(item) && ("lc_kwargs" in item || "content" in item && ("type" in item || "role" in item)));
+  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");
-function compactHistory(messages) {
-  const keep = TRACE_KEEP_LAST_MESSAGES + 1;
-  if (messages.length <= keep + 1) return messages;
-  const omitted = messages.length - keep;
-  return [
-    messages[0],
-    {
-      type: "trace_omitted",
-      omittedMessages: omitted
-    },
-    ...messages.slice(-TRACE_KEEP_LAST_MESSAGES)
-  ];
-}
-__name(compactHistory, "compactHistory");
 var als = new async_hooks.AsyncLocalStorage();
 function withFlutchContext(ctx, fn) {
   return als.run(ctx, fn);
@@ -6014,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(),
@@ -6100,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: {
@@ -9133,7 +9140,6 @@ exports.buildEffortRequestFields = buildEffortRequestFields;
 exports.buildOAuthAuthorizationUrl = buildOAuthAuthorizationUrl;
 exports.buildOpenAIModelConfig = buildOpenAIModelConfig;
 exports.clearAttachmentDataStore = clearAttachmentDataStore;
-exports.compactTraceMessages = compactTraceMessages;
 exports.createEndpointDescriptors = createEndpointDescriptors;
 exports.createGraphAttachment = createGraphAttachment;
 exports.createMongoClientAdapter = createMongoClientAdapter;
@@ -9175,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;