@latitude-data/telemetry 1.0.1 → 1.0.3

package/dist/index.cjs

@@ -141,14 +141,16 @@ const env = { GATEWAY_BASE_URL: GET_GATEWAY_BASE_URL() };
 var SegmentSource;
 (function (SegmentSource) {
     SegmentSource["API"] = "api";
-    SegmentSource["Playground"] = "playground";
-    SegmentSource["Evaluation"] = "evaluation";
-    SegmentSource["Experiment"] = "experiment";
-    SegmentSource["User"] = "user";
-    SegmentSource["SharedPrompt"] = "shared_prompt";
     SegmentSource["AgentAsTool"] = "agent_as_tool";
+    SegmentSource["Copilot"] = "copilot";
     SegmentSource["EmailTrigger"] = "email_trigger";
+    SegmentSource["Evaluation"] = "evaluation";
+    SegmentSource["Experiment"] = "experiment";
+    SegmentSource["Playground"] = "playground";
     SegmentSource["ScheduledTrigger"] = "scheduled_trigger";
+    SegmentSource["IntegrationTrigger"] = "integration_trigger";
+    SegmentSource["SharedPrompt"] = "shared_prompt";
+    SegmentSource["User"] = "user";
 })(SegmentSource || (SegmentSource = {}));
 var SegmentType;
 (function (SegmentType) {
@@ -248,7 +250,7 @@ const SPAN_SPECIFICATIONS = {
         name: 'Segment',
         description: 'A (partial) segment of a trace',
         isGenAI: false,
-        isHidden: true,
+        isHidden: false,
     },
     [SpanType.Unknown]: {
         name: 'Unknown',
@@ -431,16 +433,6 @@ zod.z.object({
     isError: zod.z.boolean().optional(),
     text: zod.z.string().optional(),
 });
-var FinishReason;
-(function (FinishReason) {
-    FinishReason["Stop"] = "stop";
-    FinishReason["Length"] = "length";
-    FinishReason["ContentFilter"] = "content-filter";
-    FinishReason["ToolCalls"] = "tool-calls";
-    FinishReason["Error"] = "error";
-    FinishReason["Other"] = "other";
-    FinishReason["Unknown"] = "unknown";
-})(FinishReason || (FinishReason = {}));
 
 var ParameterType;
 (function (ParameterType) {
@@ -778,17 +770,18 @@ var LegacyChainEventTypes;
 
 var ChainEventTypes;
 (function (ChainEventTypes) {
+    ChainEventTypes["ChainCompleted"] = "chain-completed";
+    ChainEventTypes["ChainError"] = "chain-error";
     ChainEventTypes["ChainStarted"] = "chain-started";
-    ChainEventTypes["StepStarted"] = "step-started";
-    ChainEventTypes["ProviderStarted"] = "provider-started";
+    ChainEventTypes["IntegrationWakingUp"] = "integration-waking-up";
     ChainEventTypes["ProviderCompleted"] = "provider-completed";
-    ChainEventTypes["ToolsStarted"] = "tools-started";
-    ChainEventTypes["ToolCompleted"] = "tool-completed";
+    ChainEventTypes["ProviderStarted"] = "provider-started";
     ChainEventTypes["StepCompleted"] = "step-completed";
-    ChainEventTypes["ChainCompleted"] = "chain-completed";
-    ChainEventTypes["ChainError"] = "chain-error";
+    ChainEventTypes["StepStarted"] = "step-started";
+    ChainEventTypes["ToolCompleted"] = "tool-completed";
     ChainEventTypes["ToolsRequested"] = "tools-requested";
-    ChainEventTypes["IntegrationWakingUp"] = "integration-waking-up";
+    ChainEventTypes["ToolResult"] = "tool-result";
+    ChainEventTypes["ToolsStarted"] = "tools-started";
 })(ChainEventTypes || (ChainEventTypes = {}));
 
 var IntegrationType;
@@ -901,6 +894,14 @@ var EvaluationResultableType;
     EvaluationResultableType["Number"] = "evaluation_resultable_numbers";
 })(EvaluationResultableType || (EvaluationResultableType = {}));
 
+var MessageRole;
+(function (MessageRole) {
+    MessageRole["system"] = "system";
+    MessageRole["user"] = "user";
+    MessageRole["assistant"] = "assistant";
+    MessageRole["tool"] = "tool";
+})(MessageRole || (MessageRole = {}));
+
 var ModifiedDocumentType;
 (function (ModifiedDocumentType) {
     ModifiedDocumentType["Created"] = "created";
@@ -936,6 +937,7 @@ var DocumentTriggerType;
 (function (DocumentTriggerType) {
     DocumentTriggerType["Email"] = "email";
     DocumentTriggerType["Scheduled"] = "scheduled";
+    DocumentTriggerType["Integration"] = "integration";
 })(DocumentTriggerType || (DocumentTriggerType = {}));
 var DocumentTriggerParameters;
 (function (DocumentTriggerParameters) {
@@ -1602,44 +1604,6 @@ class LatitudeInstrumentation {
         // Note: this is an estimation to not bundle a tokenizer
         return Math.ceil(length / 4);
     }
-    withTraceContext(ctx, fn) {
-        return otel.context.with(this.telemetry.resume(ctx), fn);
-    }
-    async wrapToolHandler(fn, ...args) {
-        const toolArguments = args[0];
-        const { toolId, toolName } = args[1];
-        const $tool = this.telemetry.tool(otel.context.active(), {
-            name: toolName,
-            call: {
-                id: toolId,
-                arguments: toolArguments,
-            },
-        });
-        let result;
-        try {
-            result = await otel.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(otel.context.active(), {
@@ -1659,25 +1623,6 @@ class LatitudeInstrumentation {
         $prompt.end();
         return result;
     }
-    async wrapRenderAgent(fn, ...args) {
-        const { prompt, parameters } = args[0];
-        const $prompt = this.telemetry.prompt(otel.context.active(), {
-            versionUuid: prompt.versionUuid,
-            promptUuid: prompt.uuid,
-            template: prompt.content,
-            parameters: parameters,
-        });
-        let result;
-        try {
-            result = await otel.context.with($prompt.context, async () => await fn(...args));
-        }
-        catch (error) {
-            $prompt.fail(error);
-            throw error;
-        }
-        $prompt.end();
-        return result;
-    }
     async wrapRenderStep(fn, ...args) {
         const $step = this.telemetry.step(otel.context.active());
         let result;
@@ -1748,8 +1693,8 @@ class LatitudeInstrumentation {
         }
         $tool.end({
             result: {
-                value: result,
-                isError: false, // Note: currently unknown
+                value: result.result,
+                isError: result.isError,
             },
         });
         return result;