@uselemma/tracing 2.10.0 → 2.12.0

package/dist/debug-mode.js

@@ -18,10 +18,10 @@ function lemmaDebug(prefix, msg, data) {
     if (!isDebugModeEnabled())
         return;
     if (data !== undefined) {
-        console.log(`[lemma:${prefix}] ${msg}`, data);
+        console.log(`[LEMMA:${prefix}] ${msg}`, data);
     }
     else {
-        console.log(`[lemma:${prefix}] ${msg}`);
+        console.log(`[LEMMA:${prefix}] ${msg}`);
     }
 }
 //# sourceMappingURL=debug-mode.js.map

package/dist/trace-wrapper.d.ts

@@ -5,23 +5,17 @@ export type TraceContext = {
     /** Unique identifier for this agent run. */
     runId: string;
     /**
-     * Signal the run is complete. When `autoEndRoot` is enabled (the default),
-     * this is a no-op and returns `false` — the span is ended automatically.
-     * When `autoEndRoot` is disabled, this ends the top-level span and returns `true`.
+     * Record the run output. Sets `ai.agent.output` on the span — call this
+     * before returning to store a specific value. If omitted, the function's
+     * return value is captured automatically.
      */
-    onComplete: (result: unknown) => boolean;
+    onComplete: (result: unknown) => void;
     /** Record an error on the span. Marks the span as errored. */
     recordError: (error: unknown) => void;
 };
 export type WrapAgentOptions = {
     /** Mark this run as an experiment in Lemma. */
     isExperiment?: boolean;
-    /**
-     * If `true` (the default), the top-level span is automatically ended when
-     * the wrapped function returns or throws. Set to `false` to manage span
-     * lifetime manually via `ctx.onComplete`.
-     */
-    autoEndRoot?: boolean;
 };
 /**
  * Wraps an agent function with OpenTelemetry tracing, automatically creating
@@ -30,6 +24,10 @@ export type WrapAgentOptions = {
  * The returned function creates a new root span on every invocation, attaches
  * agent metadata (name, run ID, experiment flag), and handles error recording.
  *
+ * `ai.agent.input` and `ai.agent.output` are set as JSON strings for Lemma
+ * ingestion and UI. The span ends automatically when the wrapped function
+ * returns or throws.
+ *
  * @example
  * const myAgent = wrapAgent<{ topic: string }>(
  *   'my-agent',
@@ -44,7 +42,6 @@ export type WrapAgentOptions = {
  * @param fn - The agent function to wrap. Receives a {@link TraceContext} as its first argument and the call-time input as its second.
  * @param options - Configuration for the agent trace.
  * @param options.isExperiment - Mark this run as an experiment in Lemma.
- * @param options.autoEndRoot - Automatically end the top-level span when the wrapped function returns or throws (default: `true`). Set to `false` to end manually via `ctx.onComplete`.
  * @returns An async function that accepts an `input`, executes `fn` inside a traced context, and returns `{ result, runId, span }`.
  */
 export declare function wrapAgent<Input = unknown>(agentName: string, fn: (traceContext: TraceContext, input: Input) => any, options?: WrapAgentOptions): (this: any, input: Input) => Promise<{

package/dist/trace-wrapper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"trace-wrapper.d.ts","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgC,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAKxE,MAAM,MAAM,YAAY,GAAG;IACzB,wDAAwD;IACxD,IAAI,EAAE,IAAI,CAAC;IACX,4CAA4C;IAC5C,KAAK,EAAE,MAAM,CAAC;IACd;;;;OAIG;IACH,UAAU,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,OAAO,CAAC;IACzC,8DAA8D;IAC9D,WAAW,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACvC,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,+CAA+C;IAC/C,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB;;;;OAIG;IACH,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAgB,SAAS,CAAC,KAAK,GAAG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,KAAK,GAAG,EAAE,OAAO,CAAC,EAAE,gBAAgB,UAC/F,GAAG,SAAS,KAAK;;;;GAkEhE"}
+{"version":3,"file":"trace-wrapper.d.ts","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgC,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAKxE,MAAM,MAAM,YAAY,GAAG;IACzB,wDAAwD;IACxD,IAAI,EAAE,IAAI,CAAC;IACX,4CAA4C;IAC5C,KAAK,EAAE,MAAM,CAAC;IACd;;;;OAIG;IACH,UAAU,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;IACtC,8DAA8D;IAC9D,WAAW,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACvC,CAAC;AAEF,MAAM,MAAM,gBAAgB,GAAG;IAC7B,+CAA+C;IAC/C,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,SAAS,CAAC,KAAK,GAAG,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,KAAK,GAAG,EAAE,OAAO,CAAC,EAAE,gBAAgB,UAC/F,GAAG,SAAS,KAAK;;;;GAqDhE"}

package/dist/trace-wrapper.js

@@ -12,6 +12,10 @@ const experiment_mode_1 = require("./experiment-mode");
  * The returned function creates a new root span on every invocation, attaches
  * agent metadata (name, run ID, experiment flag), and handles error recording.
  *
+ * `ai.agent.input` and `ai.agent.output` are set as JSON strings for Lemma
+ * ingestion and UI. The span ends automatically when the wrapped function
+ * returns or throws.
+ *
  * @example
  * const myAgent = wrapAgent<{ topic: string }>(
  *   'my-agent',
@@ -26,63 +30,48 @@ const experiment_mode_1 = require("./experiment-mode");
  * @param fn - The agent function to wrap. Receives a {@link TraceContext} as its first argument and the call-time input as its second.
  * @param options - Configuration for the agent trace.
  * @param options.isExperiment - Mark this run as an experiment in Lemma.
- * @param options.autoEndRoot - Automatically end the top-level span when the wrapped function returns or throws (default: `true`). Set to `false` to end manually via `ctx.onComplete`.
  * @returns An async function that accepts an `input`, executes `fn` inside a traced context, and returns `{ result, runId, span }`.
  */
 function wrapAgent(agentName, fn, options) {
     const wrappedFunction = async function (input) {
-        // Obtain the Lemma tracer from the global OTel provider
         const tracer = api_1.trace.getTracer("lemma");
-        // Generate a unique run ID and open a new span for this agent invocation
         const runId = (0, uuid_1.v4)();
-        const autoEndRoot = options?.autoEndRoot !== false; // default true
         const span = tracer.startSpan("ai.agent.run", {
             attributes: {
                 "ai.agent.name": agentName,
                 "lemma.run_id": runId,
                 "lemma.is_experiment": (0, experiment_mode_1.isExperimentModeEnabled)() || options?.isExperiment === true,
-                "lemma.auto_end_root": autoEndRoot,
             },
         }, api_1.ROOT_CONTEXT);
-        (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span started", { agentName, runId, autoEndRoot });
-        // Propagate the span as the active context so child spans are nested correctly
+        span.setAttribute("ai.agent.input", JSON.stringify(input) ?? "null");
+        (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span started", { agentName, runId });
         const ctx = api_1.trace.setSpan(api_1.ROOT_CONTEXT, span);
-        let rootEnded = false;
+        let outputSet = false;
         try {
             return await api_1.context.with(ctx, async () => {
-                const onComplete = (_result) => {
-                    if (!autoEndRoot && !rootEnded) {
-                        rootEnded = true;
-                        span.end();
-                        (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span ended via onComplete", { runId });
-                        return true;
-                    }
-                    (0, debug_mode_1.lemmaDebug)("trace-wrapper", "onComplete called but span not ended (autoEndRoot active or already ended)", { runId });
-                    return false;
+                const onComplete = (result) => {
+                    span.setAttribute("ai.agent.output", JSON.stringify(result) ?? "null");
+                    outputSet = true;
+                    (0, debug_mode_1.lemmaDebug)("trace-wrapper", "onComplete called", { runId });
                 };
                 const recordError = (error) => {
                     span.recordException(error instanceof Error ? error : new Error(String(error)));
                     span.setStatus({ code: 2 }); // SpanStatusCode.ERROR
                 };
                 const result = await fn.call(this, { span, runId, onComplete, recordError }, input);
-                // Auto-end the span if autoEndRoot is enabled and onComplete hasn't ended it yet
-                if (autoEndRoot && !rootEnded) {
-                    rootEnded = true;
-                    span.end();
-                    (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span auto-ended after fn returned", { runId });
+                if (!outputSet) {
+                    span.setAttribute("ai.agent.output", JSON.stringify(result) ?? "null");
                 }
+                span.end();
+                (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span ended after fn returned", { runId });
                 return { result, runId, span };
             });
         }
         catch (err) {
-            // Record the exception on the span, mark it as errored, and end it
             span.recordException(err);
             span.setStatus({ code: 2 }); // SpanStatusCode.ERROR
-            if (!rootEnded) {
-                rootEnded = true;
-                span.end();
-                (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span ended on error", { runId, error: String(err) });
-            }
+            span.end();
+            (0, debug_mode_1.lemmaDebug)("trace-wrapper", "span ended on error", { runId, error: String(err) });
             throw err;
         }
     };

package/dist/trace-wrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"trace-wrapper.js","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":";;AAuDA,8BAmEC;AA1HD,4CAAwE;AACxE,+BAAoC;AACpC,6CAA0C;AAC1C,uDAA4D;AA4B5D;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,SAAgB,SAAS,CAAkB,SAAiB,EAAE,EAAqD,EAAE,OAA0B;IAC7I,MAAM,eAAe,GAAG,KAAK,WAAsB,KAAY;QAC7D,wDAAwD;QACxD,MAAM,MAAM,GAAG,WAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAExC,yEAAyE;QACzE,MAAM,KAAK,GAAG,IAAA,SAAM,GAAE,CAAC;QACvB,MAAM,WAAW,GAAG,OAAO,EAAE,WAAW,KAAK,KAAK,CAAC,CAAC,eAAe;QACnE,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE;YAC5C,UAAU,EAAE;gBACV,eAAe,EAAE,SAAS;gBAC1B,cAAc,EAAE,KAAK;gBACrB,qBAAqB,EAAE,IAAA,yCAAuB,GAAE,IAAI,OAAO,EAAE,YAAY,KAAK,IAAI;gBAClF,qBAAqB,EAAE,WAAW;aACnC;SACF,EAAE,kBAAY,CAAC,CAAC;QAEjB,IAAA,uBAAU,EAAC,eAAe,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;QAE/E,+EAA+E;QAC/E,MAAM,GAAG,GAAG,WAAK,CAAC,OAAO,CAAC,kBAAY,EAAE,IAAI,CAAC,CAAC;QAC9C,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,IAAI,CAAC;YACH,OAAO,MAAM,aAAO,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;gBACxC,MAAM,UAAU,GAAG,CAAC,OAAgB,EAAW,EAAE;oBAC/C,IAAI,CAAC,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;wBAC/B,SAAS,GAAG,IAAI,CAAC;wBACjB,IAAI,CAAC,GAAG,EAAE,CAAC;wBACX,IAAA,uBAAU,EAAC,eAAe,EAAE,2BAA2B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;wBACpE,OAAO,IAAI,CAAC;oBACd,CAAC;oBACD,IAAA,uBAAU,EAAC,eAAe,EAAE,4EAA4E,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;oBACrH,OAAO,KAAK,CAAC;gBACf,CAAC,CAAC;gBAEF,MAAM,WAAW,GAAG,CAAC,KAAc,EAAE,EAAE;oBACrC,IAAI,CAAC,eAAe,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAChF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;gBACtD,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,KAAK,CAAC,CAAC;gBAEpF,iFAAiF;gBACjF,IAAI,WAAW,IAAI,CAAC,SAAS,EAAE,CAAC;oBAC9B,SAAS,GAAG,IAAI,CAAC;oBACjB,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,IAAA,uBAAU,EAAC,eAAe,EAAE,mCAAmC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC9E,CAAC;gBAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,mEAAmE;YACnE,IAAI,CAAC,eAAe,CAAC,GAAY,CAAC,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;YACpD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS,GAAG,IAAI,CAAC;gBACjB,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,IAAA,uBAAU,EAAC,eAAe,EAAE,qBAAqB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,eAAe,CAAC;AACzB,CAAC"}
+{"version":3,"file":"trace-wrapper.js","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":";;AAoDA,8BAsDC;AA1GD,4CAAwE;AACxE,+BAAoC;AACpC,6CAA0C;AAC1C,uDAA4D;AAsB5D;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,SAAgB,SAAS,CAAkB,SAAiB,EAAE,EAAqD,EAAE,OAA0B;IAC7I,MAAM,eAAe,GAAG,KAAK,WAAsB,KAAY;QAC7D,MAAM,MAAM,GAAG,WAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,KAAK,GAAG,IAAA,SAAM,GAAE,CAAC;QAEvB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE;YAC5C,UAAU,EAAE;gBACV,eAAe,EAAE,SAAS;gBAC1B,cAAc,EAAE,KAAK;gBACrB,qBAAqB,EAAE,IAAA,yCAAuB,GAAE,IAAI,OAAO,EAAE,YAAY,KAAK,IAAI;aACnF;SACF,EAAE,kBAAY,CAAC,CAAC;QAEjB,IAAI,CAAC,YAAY,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,CAAC;QAErE,IAAA,uBAAU,EAAC,eAAe,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QAElE,MAAM,GAAG,GAAG,WAAK,CAAC,OAAO,CAAC,kBAAY,EAAE,IAAI,CAAC,CAAC;QAC9C,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,IAAI,CAAC;YACH,OAAO,MAAM,aAAO,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;gBACxC,MAAM,UAAU,GAAG,CAAC,MAAe,EAAQ,EAAE;oBAC3C,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,CAAC;oBACvE,SAAS,GAAG,IAAI,CAAC;oBACjB,IAAA,uBAAU,EAAC,eAAe,EAAE,mBAAmB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC9D,CAAC,CAAC;gBAEF,MAAM,WAAW,GAAG,CAAC,KAAc,EAAE,EAAE;oBACrC,IAAI,CAAC,eAAe,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBAChF,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;gBACtD,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,KAAK,CAAC,CAAC;gBAEpF,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,IAAI,CAAC,YAAY,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,CAAC;gBACzE,CAAC;gBAED,IAAI,CAAC,GAAG,EAAE,CAAC;gBACX,IAAA,uBAAU,EAAC,eAAe,EAAE,8BAA8B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAEvE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;YACjC,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,eAAe,CAAC,GAAY,CAAC,CAAC;YACnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,uBAAuB;YACpD,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,IAAA,uBAAU,EAAC,eAAe,EAAE,qBAAqB,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAClF,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,eAAe,CAAC;AACzB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uselemma/tracing",
-  "version": "2.10.0",
+  "version": "2.12.0",
   "description": "OpenTelemetry-based tracing module for Lemma",
   "license": "MIT",
   "author": "Lemma",