@latitude-data/telemetry 1.0.2 → 1.0.3

package/dist/index.js

@@ -1584,44 +1584,6 @@ class LatitudeInstrumentation {
         // Note: this is an estimation to not bundle a tokenizer
         return Math.ceil(length / 4);
     }
-    withTraceContext(ctx, fn) {
-        return context.with(this.telemetry.resume(ctx), fn);
-    }
-    async wrapToolHandler(fn, ...args) {
-        const toolArguments = args[0];
-        const { id, name } = args[1];
-        const $tool = this.telemetry.tool(context.active(), {
-            name,
-            call: {
-                id,
-                arguments: toolArguments,
-            },
-        });
-        let result;
-        try {
-            result = await context.with($tool.context, async () => await fn(...args));
-        }
-        catch (error) {
-            if (error.name === 'ToolExecutionPausedError') {
-                $tool.fail(error);
-                throw error;
-            }
-            $tool.end({
-                result: {
-                    value: error.message,
-                    isError: true,
-                },
-            });
-            throw error;
-        }
-        $tool.end({
-            result: {
-                value: result,
-                isError: false,
-            },
-        });
-        return result;
-    }
     async wrapRenderChain(fn, ...args) {
         const { prompt, parameters } = args[0];
         const $prompt = this.telemetry.prompt(context.active(), {
@@ -1711,8 +1673,8 @@ class LatitudeInstrumentation {
         }
         $tool.end({
             result: {
-                value: result,
-                isError: false, // Note: currently unknown
+                value: result.result,
+                isError: result.isError,
             },
         });
         return result;