@probelabs/probe 0.6.0-rc299 → 0.6.0-rc301

package/cjs/agent/simpleTelemetry.cjs

@@ -22,7 +22,8 @@ var simpleTelemetry_exports = {};
 __export(simpleTelemetry_exports, {
   SimpleAppTracer: () => SimpleAppTracer,
   SimpleTelemetry: () => SimpleTelemetry,
-  initializeSimpleTelemetryFromOptions: () => initializeSimpleTelemetryFromOptions
+  initializeSimpleTelemetryFromOptions: () => initializeSimpleTelemetryFromOptions,
+  truncateForSpan: () => truncateForSpan
 });
 module.exports = __toCommonJS(simpleTelemetry_exports);
 var import_fs = require("fs");
@@ -138,6 +139,14 @@ function patchConsole() {
 }
 
 // src/agent/simpleTelemetry.js
+function truncateForSpan(text, maxLen = 4096) {
+  if (!text || text.length <= maxLen) return text || "";
+  const half = Math.floor((maxLen - 40) / 2);
+  const omitted = text.length - half * 2;
+  return text.substring(0, half) + `
+... [${omitted} chars omitted] ...
+` + text.substring(text.length - half);
+}
 var SimpleTelemetry = class {
   constructor(options = {}) {
     this.serviceName = options.serviceName || "probe-agent";
@@ -530,7 +539,7 @@ var SimpleAppTracer = class {
       "tokens.context_remaining": tokenData.maxContextTokens ? tokenData.maxContextTokens - (tokenData.contextTokens || 0) : null
     });
   }
-  async withSpan(spanName, fn, attributes = {}) {
+  async withSpan(spanName, fn, attributes = {}, onResult = null) {
     if (!this.isEnabled()) {
       return fn();
     }
@@ -541,6 +550,12 @@ var SimpleAppTracer = class {
     try {
       const result = await fn();
       span.setStatus("OK");
+      if (onResult) {
+        try {
+          onResult(span, result);
+        } catch (_) {
+        }
+      }
       return result;
     } catch (error) {
       span.setStatus("ERROR");
@@ -578,5 +593,6 @@ function initializeSimpleTelemetryFromOptions(options) {
 0 && (module.exports = {
   SimpleAppTracer,
   SimpleTelemetry,
-  initializeSimpleTelemetryFromOptions
+  initializeSimpleTelemetryFromOptions,
+  truncateForSpan
 });