@fallom/trace 0.2.16 → 0.2.18

package/dist/index.mjs

@@ -41,7 +41,7 @@ import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
 // node_modules/@opentelemetry/resources/build/esm/Resource.js
 import { diag } from "@opentelemetry/api";
 
-// node_modules/@opentelemetry/semantic-conventions/build/esm/resource/SemanticResourceAttributes.js
+// node_modules/@opentelemetry/resources/node_modules/@opentelemetry/semantic-conventions/build/esm/resource/SemanticResourceAttributes.js
 var SemanticResourceAttributes = {
   /**
    * Name of the cloud provider.
@@ -1056,6 +1056,54 @@ function clearPromptContext() {
   promptContext = null;
 }
 
+// src/trace/wrappers/shared-utils.ts
+function sanitizeMetadataOnly(key, value) {
+  const contentKeys = [
+    "text",
+    "content",
+    "message",
+    "messages",
+    "object",
+    "prompt",
+    "system",
+    "input",
+    "output",
+    "response",
+    "toolCalls",
+    "toolResults",
+    "steps",
+    "reasoning",
+    "rawResponse",
+    "rawCall",
+    "body",
+    "candidates",
+    "parts"
+  ];
+  if (contentKeys.includes(key)) {
+    if (typeof value === "string") {
+      return `[content omitted: ${value.length} chars]`;
+    }
+    if (Array.isArray(value)) {
+      return `[content omitted: ${value.length} items]`;
+    }
+    if (typeof value === "object" && value !== null) {
+      return "[content omitted]";
+    }
+  }
+  if (typeof value === "string") {
+    if (value.startsWith("data:image/")) {
+      return "[base64 image omitted]";
+    }
+    if (value.length > 1e3) {
+      return `[large string omitted: ${value.length} chars]`;
+    }
+  }
+  if (value instanceof Uint8Array || value && value.type === "Buffer") {
+    return "[binary data omitted]";
+  }
+  return value;
+}
+
 // src/trace/wrappers/openai.ts
 function wrapOpenAI(client, sessionCtx) {
   const originalCreate = client.chat.completions.create.bind(
@@ -1103,6 +1151,13 @@ function wrapOpenAI(client, sessionCtx) {
       if (response?.usage) {
         attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
       }
+      try {
+        attributes["fallom.raw.metadata"] = JSON.stringify(
+          response,
+          sanitizeMetadataOnly
+        );
+      } catch {
+      }
       const waterfallTimings = {
         requestStart: 0,
         requestEnd: endTime - startTime,
@@ -1229,6 +1284,13 @@ function wrapAnthropic(client, sessionCtx) {
       if (response?.usage) {
         attributes["fallom.raw.usage"] = JSON.stringify(response.usage);
       }
+      try {
+        attributes["fallom.raw.metadata"] = JSON.stringify(
+          response,
+          sanitizeMetadataOnly
+        );
+      } catch {
+      }
       const waterfallTimings = {
         requestStart: 0,
         requestEnd: endTime - startTime,
@@ -1349,6 +1411,13 @@ function wrapGoogleAI(model, sessionCtx) {
       if (result?.usageMetadata) {
         attributes["fallom.raw.usage"] = JSON.stringify(result.usageMetadata);
       }
+      try {
+        attributes["fallom.raw.metadata"] = JSON.stringify(
+          result,
+          sanitizeMetadataOnly
+        );
+      } catch {
+      }
       const waterfallTimings = {
         requestStart: 0,
         requestEnd: endTime - startTime,
@@ -1498,20 +1567,36 @@ function createGenerateTextWrapper(aiModule, sessionCtx, debug = false) {
           tools: params?.tools ? Object.keys(params.tools) : void 0,
           maxSteps: params?.maxSteps
         });
-        const mapToolCall = (tc) => ({
-          toolCallId: tc?.toolCallId,
-          toolName: tc?.toolName,
-          args: tc?.args ?? tc?.input,
-          // v4: args, v5: input
-          type: tc?.type
-        });
-        const mapToolResult = (tr) => ({
-          toolCallId: tr?.toolCallId,
-          toolName: tr?.toolName,
-          result: tr?.result ?? tr?.output,
-          // v4: result, v5: output
-          type: tr?.type
-        });
+        const mapToolCall = (tc) => {
+          let args2 = tc?.args ?? tc?.input;
+          if (args2 === void 0 && tc) {
+            const { type, toolCallId, toolName, providerExecuted, dynamic, invalid, error, providerMetadata, ...rest } = tc;
+            if (Object.keys(rest).length > 0) {
+              args2 = rest;
+            }
+          }
+          return {
+            toolCallId: tc?.toolCallId,
+            toolName: tc?.toolName,
+            args: args2,
+            type: tc?.type
+          };
+        };
+        const mapToolResult = (tr) => {
+          let result2 = tr?.result ?? tr?.output;
+          if (result2 === void 0 && tr) {
+            const { type, toolCallId, toolName, ...rest } = tr;
+            if (Object.keys(rest).length > 0) {
+              result2 = rest;
+            }
+          }
+          return {
+            toolCallId: tr?.toolCallId,
+            toolName: tr?.toolName,
+            result: result2,
+            type: tr?.type
+          };
+        };
         attributes["fallom.raw.response"] = JSON.stringify({
           text: result?.text,
           finishReason: result?.finishReason,
@@ -1545,6 +1630,13 @@ function createGenerateTextWrapper(aiModule, sessionCtx, debug = false) {
           result.experimental_providerMetadata
         );
       }
+      try {
+        attributes["fallom.raw.metadata"] = JSON.stringify(
+          result,
+          sanitizeMetadataOnly
+        );
+      } catch {
+      }
       const totalDurationMs = endTime - startTime;
       const sortedToolTimings = Array.from(toolTimings.values()).sort(
         (a, b) => a.startTime - b.startTime
@@ -1717,7 +1809,9 @@ function createStreamTextWrapper(aiModule, sessionCtx, debug = false) {
     let wrappedParams = params;
     if (params.tools && typeof params.tools === "object") {
       const wrappedTools = {};
-      for (const [toolName, tool] of Object.entries(params.tools)) {
+      for (const [toolName, tool] of Object.entries(
+        params.tools
+      )) {
         if (tool && typeof tool.execute === "function") {
           const originalExecute = tool.execute;
           wrappedTools[toolName] = {
@@ -1800,10 +1894,54 @@ function createStreamTextWrapper(aiModule, sessionCtx, debug = false) {
               "\u{1F50D} [Fallom Debug] streamText toolCalls:",
               JSON.stringify(toolCalls, null, 2)
             );
+            if (toolCalls?.[0]) {
+              console.log(
+                "\u{1F50D} [Fallom Debug] streamText toolCalls[0] keys:",
+                Object.keys(toolCalls[0])
+              );
+              console.log(
+                "\u{1F50D} [Fallom Debug] streamText toolCalls[0] full:",
+                JSON.stringify(
+                  toolCalls[0],
+                  Object.getOwnPropertyNames(toolCalls[0]),
+                  2
+                )
+              );
+            }
             console.log(
               "\u{1F50D} [Fallom Debug] streamText steps count:",
               steps?.length
             );
+            if (steps?.[0]?.toolCalls?.[0]) {
+              const tc = steps[0].toolCalls[0];
+              console.log(
+                "\u{1F50D} [Fallom Debug] steps[0].toolCalls[0] keys:",
+                Object.keys(tc)
+              );
+              console.log(
+                "\u{1F50D} [Fallom Debug] steps[0].toolCalls[0].args (v4):",
+                tc.args
+              );
+              console.log(
+                "\u{1F50D} [Fallom Debug] steps[0].toolCalls[0].input (v5):",
+                tc.input
+              );
+            }
+            if (steps?.[0]?.toolResults?.[0]) {
+              const tr = steps[0].toolResults[0];
+              console.log(
+                "\u{1F50D} [Fallom Debug] steps[0].toolResults[0] keys:",
+                Object.keys(tr)
+              );
+              console.log(
+                "\u{1F50D} [Fallom Debug] steps[0].toolResults[0].result (v4):",
+                typeof tr.result === "string" ? tr.result.slice(0, 200) : tr.result
+              );
+              console.log(
+                "\u{1F50D} [Fallom Debug] steps[0].toolResults[0].output (v5):",
+                typeof tr.output === "string" ? tr.output.slice(0, 200) : tr.output
+              );
+            }
           }
           let providerMetadata = result?.experimental_providerMetadata;
           if (providerMetadata && typeof providerMetadata.then === "function") {
@@ -1819,20 +1957,46 @@ function createStreamTextWrapper(aiModule, sessionCtx, debug = false) {
             "fallom.is_streaming": true
           };
           if (captureContent2) {
-            const mapToolCall = (tc) => ({
-              toolCallId: tc?.toolCallId,
-              toolName: tc?.toolName,
-              args: tc?.args ?? tc?.input,
-              // v4: args, v5: input
-              type: tc?.type
-            });
-            const mapToolResult = (tr) => ({
-              toolCallId: tr?.toolCallId,
-              toolName: tr?.toolName,
-              result: tr?.result ?? tr?.output,
-              // v4: result, v5: output
-              type: tr?.type
-            });
+            const mapToolCall = (tc) => {
+              let args2 = tc?.args ?? tc?.input;
+              if (args2 === void 0 && tc) {
+                const {
+                  type,
+                  toolCallId,
+                  toolName,
+                  providerExecuted,
+                  dynamic,
+                  invalid,
+                  error,
+                  providerMetadata: providerMetadata2,
+                  ...rest
+                } = tc;
+                if (Object.keys(rest).length > 0) {
+                  args2 = rest;
+                }
+              }
+              return {
+                toolCallId: tc?.toolCallId,
+                toolName: tc?.toolName,
+                args: args2,
+                type: tc?.type
+              };
+            };
+            const mapToolResult = (tr) => {
+              let result2 = tr?.result ?? tr?.output;
+              if (result2 === void 0 && tr) {
+                const { type, toolCallId, toolName, ...rest } = tr;
+                if (Object.keys(rest).length > 0) {
+                  result2 = rest;
+                }
+              }
+              return {
+                toolCallId: tr?.toolCallId,
+                toolName: tr?.toolName,
+                result: result2,
+                type: tr?.type
+              };
+            };
             attributes["fallom.raw.request"] = JSON.stringify({
               prompt: params?.prompt,
               messages: params?.messages,
@@ -1873,6 +2037,13 @@ function createStreamTextWrapper(aiModule, sessionCtx, debug = false) {
           if (firstTokenTime) {
             attributes["fallom.time_to_first_token_ms"] = firstTokenTime - startTime;
           }
+          try {
+            attributes["fallom.raw.metadata"] = JSON.stringify(
+              result,
+              sanitizeMetadataOnly
+            );
+          } catch {
+          }
           const totalDurationMs = endTime - startTime;
           const sortedToolTimings = Array.from(toolTimings.values()).sort(
             (a, b) => a.startTime - b.startTime
@@ -1897,8 +2068,12 @@ function createStreamTextWrapper(aiModule, sessionCtx, debug = false) {
             });
           }
           if (sortedToolTimings.length > 0) {
-            const firstToolStart = Math.min(...sortedToolTimings.map((t) => t.startTime));
-            const lastToolEnd = Math.max(...sortedToolTimings.map((t) => t.endTime));
+            const firstToolStart = Math.min(
+              ...sortedToolTimings.map((t) => t.startTime)
+            );
+            const lastToolEnd = Math.max(
+              ...sortedToolTimings.map((t) => t.endTime)
+            );
             if (firstToolStart > 10) {
               waterfallTimings.phases.push({
                 type: "llm",
@@ -2078,6 +2253,10 @@ function createGenerateObjectWrapper(aiModule, sessionCtx, debug = false) {
           result.experimental_providerMetadata
         );
       }
+      try {
+        attributes["fallom.raw.metadata"] = JSON.stringify(result, sanitizeMetadataOnly);
+      } catch {
+      }
       const promptCtx = getPromptContext();
       sendTrace({
         config_key: ctx.configKey,
@@ -2197,6 +2376,10 @@ function createStreamObjectWrapper(aiModule, sessionCtx, debug = false) {
         if (providerMetadata) {
           attributes["fallom.raw.providerMetadata"] = JSON.stringify(providerMetadata);
         }
+        try {
+          attributes["fallom.raw.metadata"] = JSON.stringify(result, sanitizeMetadataOnly);
+        } catch {
+        }
         const promptCtx = getPromptContext();
         sendTrace({
           config_key: ctx.configKey,
@@ -2293,6 +2476,13 @@ function wrapMastraAgent(agent, sessionCtx) {
         attributes["fallom.raw.request"] = JSON.stringify(input);
         attributes["fallom.raw.response"] = JSON.stringify(result);
       }
+      try {
+        attributes["fallom.raw.metadata"] = JSON.stringify(
+          result,
+          sanitizeMetadataOnly
+        );
+      } catch {
+      }
       sendTrace({
         config_key: ctx.configKey,
         session_id: ctx.sessionId,

package/dist/models-SEFDGZU2.mjs

@@ -0,0 +1,8 @@
+import {
+  get,
+  init
+} from "./chunk-KFD5AQ7V.mjs";
+export {
+  get,
+  init
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fallom/trace",
-  "version": "0.2.16",
+  "version": "0.2.18",
   "description": "Model A/B testing and tracing for LLM applications. Zero latency, production-ready.",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",