@uselemma/tracing 0.12.0 → 0.14.0

package/dist/index.d.ts

@@ -1,3 +1,3 @@
 export { CandidatePromptManager } from "./candidate-prompt-manager";
-export { wrapAgent } from "./trace-wrapper";
+export { wrapAgent, type TraceContext } from "./trace-wrapper";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,KAAK,YAAY,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,uEAAoE;AAA3D,kIAAA,sBAAsB,OAAA;AAC/B,iDAA4C;AAAnC,0GAAA,SAAS,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,uEAAoE;AAA3D,kIAAA,sBAAsB,OAAA;AAC/B,iDAA+D;AAAtD,0GAAA,SAAS,OAAA"}

package/dist/trace-wrapper.d.ts

@@ -1,8 +1,17 @@
-export declare function wrapAgent<A extends unknown[], F extends (...args: A) => ReturnType<F>>(agentName: string, options: {
+import { Span } from "@opentelemetry/api";
+export type TraceContext = {
+    span: Span;
+    runId: string;
+    onFinish: (result: unknown) => void;
+    onError: (error: unknown) => void;
+};
+export declare function wrapAgent<A extends unknown[]>(agentName: string, options: {
     isExperiment?: boolean;
     initialState?: any;
-}, fn: F, ...args: A): Promise<{
-    result: ReturnType<F>;
+    endOnExit?: boolean;
+}, fn: (traceContext: TraceContext, ...args: A) => any): (this: any, ...args: A) => Promise<{
+    result: any;
     runId: string;
+    span: Span;
 }>;
 //# sourceMappingURL=trace-wrapper.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"trace-wrapper.d.ts","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":"AAGA,wBAAsB,SAAS,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,CAAC,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE;IAAE,YAAY,CAAC,EAAE,OAAO,CAAC;IAAC,YAAY,CAAC,EAAE,GAAG,CAAA;CAAE,EAAE,EAAE,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC;;;GA0B1L"}
+{"version":3,"file":"trace-wrapper.d.ts","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":"AAAA,OAAO,EAAkB,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAG1D,MAAM,MAAM,YAAY,GAAG;IACzB,IAAI,EAAE,IAAI,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,CAAC;IACpC,OAAO,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,IAAI,CAAC;CACnC,CAAC;AAEF,wBAAgB,SAAS,CAAC,CAAC,SAAS,OAAO,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE;IAAE,YAAY,CAAC,EAAE,OAAO,CAAC;IAAC,YAAY,CAAC,EAAE,GAAG,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,CAAA;CAAE,EAAE,EAAE,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,GAAG,IAAI,EAAE,CAAC,KAAK,GAAG,UACnJ,GAAG,WAAW,CAAC;;;;GAgD9D"}

package/dist/trace-wrapper.js

@@ -3,29 +3,45 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.wrapAgent = wrapAgent;
 const api_1 = require("@opentelemetry/api");
 const uuid_1 = require("uuid");
-async function wrapAgent(agentName, options, fn, ...args) {
-    const tracer = api_1.trace.getTracer("lemma");
-    const runId = (0, uuid_1.v4)();
-    const span = tracer.startSpan(agentName, {
-        attributes: {
-            "lemma.agent.run_id": runId,
-            "lemma.agent.input": JSON.stringify(options.initialState),
-            "lemma.agent.is_experiment": options.isExperiment,
-        },
-    });
-    const ctx = api_1.trace.setSpan(api_1.context.active(), span);
-    try {
-        return await api_1.context.with(ctx, async () => {
-            const result = await fn(...args);
-            span.end();
-            return { result, runId };
+function wrapAgent(agentName, options, fn) {
+    const wrappedFunction = async function (...args) {
+        const tracer = api_1.trace.getTracer("lemma");
+        const runId = (0, uuid_1.v4)();
+        const span = tracer.startSpan(agentName, {
+            attributes: {
+                "lemma.agent.run_id": runId,
+                "lemma.agent.input": JSON.stringify(options.initialState),
+                "lemma.agent.is_experiment": options.isExperiment,
+            },
         });
-    }
-    catch (err) {
-        span.recordException(err);
-        span.setStatus({ code: 2 });
-        span.end();
-        throw err;
-    }
+        const ctx = api_1.trace.setSpan(api_1.context.active(), span);
+        try {
+            return await api_1.context.with(ctx, async () => {
+                const onFinish = (result) => {
+                    span.setAttribute("lemma.agent.output", JSON.stringify(result));
+                    span.end();
+                };
+                const onError = (error) => {
+                    span.recordException(error instanceof Error ? error : new Error(String(error)));
+                    span.setStatus({ code: 2 });
+                    span.end();
+                };
+                const result = await fn.call(this, { span, runId, onFinish, onError }, ...args);
+                if (options?.endOnExit !== false) {
+                    span.end();
+                }
+                return { result, runId, span };
+            });
+        }
+        catch (err) {
+            span.recordException(err);
+            span.setStatus({ code: 2 });
+            if (options?.endOnExit !== false) {
+                span.end();
+            }
+            throw err;
+        }
+    };
+    return wrappedFunction;
 }
 //# sourceMappingURL=trace-wrapper.js.map

package/dist/trace-wrapper.js.map

@@ -1 +1 @@
-{"version":3,"file":"trace-wrapper.js","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":";;AAGA,8BA0BC;AA7BD,4CAAoD;AACpD,+BAAoC;AAE7B,KAAK,UAAU,SAAS,CAA+D,SAAiB,EAAE,OAAuD,EAAE,EAAK,EAAE,GAAG,IAAO;IACzL,MAAM,MAAM,GAAG,WAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAExC,MAAM,KAAK,GAAG,IAAA,SAAM,GAAE,CAAC;IACvB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE;QACvC,UAAU,EAAE;YACV,oBAAoB,EAAE,KAAK;YAC3B,mBAAmB,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC;YACzD,2BAA2B,EAAE,OAAO,CAAC,YAAY;SAClD;KACF,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;IAElD,IAAI,CAAC;QACH,OAAO,MAAM,aAAO,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,GAAG,EAAE,CAAC;YACX,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,eAAe,CAAC,GAAY,CAAC,CAAC;QACnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC5B,IAAI,CAAC,GAAG,EAAE,CAAC;QACX,MAAM,GAAG,CAAC;IACZ,CAAC;AACH,CAAC"}
+{"version":3,"file":"trace-wrapper.js","sourceRoot":"","sources":["../src/trace-wrapper.ts"],"names":[],"mappings":";;AAUA,8BAiDC;AA3DD,4CAA0D;AAC1D,+BAAoC;AASpC,SAAgB,SAAS,CAAsB,SAAiB,EAAE,OAA4E,EAAE,EAAmD;IACjM,MAAM,eAAe,GAAG,KAAK,WAAsB,GAAG,IAAO;QAC3D,MAAM,MAAM,GAAG,WAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAExC,MAAM,KAAK,GAAG,IAAA,SAAM,GAAE,CAAC;QACvB,MAAM,IAAI,GAAG,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE;YACvC,UAAU,EAAE;gBACV,oBAAoB,EAAE,KAAK;gBAC3B,mBAAmB,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,CAAC;gBACzD,2BAA2B,EAAE,OAAO,CAAC,YAAY;aAClD;SACF,CAAC,CAAC;QAEH,MAAM,GAAG,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;QAElD,IAAI,CAAC;YACH,OAAO,MAAM,aAAO,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;gBACxC,MAAM,QAAQ,GAAG,CAAC,MAAe,EAAE,EAAE;oBACnC,IAAI,CAAC,YAAY,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;oBAChE,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,MAAM,OAAO,GAAG,CAAC,KAAc,EAAE,EAAE;oBACjC,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;oBAC5B,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC,CAAC;gBAEF,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,GAAG,IAAI,CAAC,CAAC;gBAEhF,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,EAAE,CAAC;oBACjC,IAAI,CAAC,GAAG,EAAE,CAAC;gBACb,CAAC;gBAED,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;YAE5B,IAAI,OAAO,EAAE,SAAS,KAAK,KAAK,EAAE,CAAC;gBACjC,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,CAAC;YAED,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": "0.12.0",
+  "version": "0.14.0",
   "description": "OpenTelemetry-based tracing module for Lemma",
   "license": "MIT",
   "author": "Lemma",