@latitude-data/telemetry 1.0.1 → 1.0.3

package/dist/index.d.ts

@@ -35,37 +35,18 @@ declare class RedactSpanProcessor implements SpanProcessor {
 }
 declare const DEFAULT_REDACT_SPAN_PROCESSOR: () => RedactSpanProcessor;
 
-declare const traceContextSchema: z.ZodObject<{
-    traceparent: z.ZodString;
-    tracestate: z.ZodOptional<z.ZodString>;
-    baggage: z.ZodOptional<z.ZodString>;
-}, "strip", z.ZodTypeAny, {
-    traceparent: string;
-    tracestate?: string | undefined;
-    baggage?: string | undefined;
-}, {
-    traceparent: string;
-    tracestate?: string | undefined;
-    baggage?: string | undefined;
-}>;
-type TraceContext = z.infer<typeof traceContextSchema>;
-type TraceBaggage = {
-    segment: Pick<SegmentBaggage, 'id' | 'parentId'>;
-    segments: (SegmentBaggage & Pick<TraceContext, 'traceparent' | 'tracestate'> & {
-        paused?: boolean;
-    })[];
-};
-
 declare enum SegmentSource {
     API = "api",
-    Playground = "playground",
+    AgentAsTool = "agent_as_tool",// TODO(tracing): deprecated, use SegmentType.Document with DocumentType.Agent instead
+    Copilot = "copilot",
+    EmailTrigger = "email_trigger",
     Evaluation = "evaluation",// Note: from prompts of llm evaluations
     Experiment = "experiment",
-    User = "user",
+    Playground = "playground",
+    ScheduledTrigger = "scheduled_trigger",
+    IntegrationTrigger = "integration_trigger",
     SharedPrompt = "shared_prompt",
-    AgentAsTool = "agent_as_tool",// TODO(tracing): deprecated, use SegmentType.Document with DocumentType.Agent instead
-    EmailTrigger = "email_trigger",
-    ScheduledTrigger = "scheduled_trigger"
+    User = "user"
 }
 declare enum SegmentType {
     Document = "document",
@@ -144,6 +125,27 @@ type SegmentBaggage<T extends SegmentType = SegmentType> = Extract<z.infer<typeo
     type: T;
 }>;
 
+declare const traceContextSchema: z.ZodObject<{
+    traceparent: z.ZodString;
+    tracestate: z.ZodOptional<z.ZodString>;
+    baggage: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    traceparent: string;
+    tracestate?: string | undefined;
+    baggage?: string | undefined;
+}, {
+    traceparent: string;
+    tracestate?: string | undefined;
+    baggage?: string | undefined;
+}>;
+type TraceContext = z.infer<typeof traceContextSchema>;
+type TraceBaggage = {
+    segment: Pick<SegmentBaggage, 'id' | 'parentId'>;
+    segments: (SegmentBaggage & Pick<TraceContext, 'traceparent' | 'tracestate'> & {
+        paused?: boolean;
+    })[];
+};
+
 type LatitudeInstrumentationOptions = {
     module: typeof latitude.Latitude;
     completions?: boolean;

package/dist/index.js

@@ -121,14 +121,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) {
@@ -228,7 +230,7 @@ const SPAN_SPECIFICATIONS = {
         name: 'Segment',
         description: 'A (partial) segment of a trace',
         isGenAI: false,
-        isHidden: true,
+        isHidden: false,
     },
     [SpanType.Unknown]: {
         name: 'Unknown',
@@ -411,16 +413,6 @@ z.object({
     isError: z.boolean().optional(),
     text: 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) {
@@ -758,17 +750,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;
@@ -881,6 +874,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";
@@ -916,6 +917,7 @@ var DocumentTriggerType;
 (function (DocumentTriggerType) {
     DocumentTriggerType["Email"] = "email";
     DocumentTriggerType["Scheduled"] = "scheduled";
+    DocumentTriggerType["Integration"] = "integration";
 })(DocumentTriggerType || (DocumentTriggerType = {}));
 var DocumentTriggerParameters;
 (function (DocumentTriggerParameters) {
@@ -1582,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 { toolId, toolName } = args[1];
-        const $tool = this.telemetry.tool(context.active(), {
-            name: toolName,
-            call: {
-                id: toolId,
-                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(), {
@@ -1639,25 +1603,6 @@ class LatitudeInstrumentation {
         $prompt.end();
         return result;
     }
-    async wrapRenderAgent(fn, ...args) {
-        const { prompt, parameters } = args[0];
-        const $prompt = this.telemetry.prompt(context.active(), {
-            versionUuid: prompt.versionUuid,
-            promptUuid: prompt.uuid,
-            template: prompt.content,
-            parameters: parameters,
-        });
-        let result;
-        try {
-            result = await 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(context.active());
         let result;
@@ -1728,8 +1673,8 @@ class LatitudeInstrumentation {
         }
         $tool.end({
             result: {
-                value: result,
-                isError: false, // Note: currently unknown
+                value: result.result,
+                isError: result.isError,
             },
         });
         return result;