@latitude-data/telemetry 1.0.2 → 1.0.4

package/dist/index.js

@@ -3,7 +3,6 @@ import * as otel from '@opentelemetry/api';
 import { propagation, trace, context } from '@opentelemetry/api';
 import { ATTR_HTTP_REQUEST_METHOD, ATTR_HTTP_RESPONSE_STATUS_CODE, ATTR_SERVICE_NAME } from '@opentelemetry/semantic-conventions';
 import { ATTR_GEN_AI_OPERATION_NAME, ATTR_GEN_AI_TOOL_CALL_ID, ATTR_GEN_AI_TOOL_TYPE, ATTR_GEN_AI_TOOL_NAME, ATTR_GEN_AI_SYSTEM, ATTR_GEN_AI_RESPONSE_FINISH_REASONS, ATTR_GEN_AI_RESPONSE_MODEL, ATTR_GEN_AI_USAGE_OUTPUT_TOKENS, ATTR_GEN_AI_USAGE_INPUT_TOKENS } from '@opentelemetry/semantic-conventions/incubating';
-import { v4 } from 'uuid';
 import { BaggageSpanProcessor, ALLOW_ALL_BAGGAGE_KEYS } from '@opentelemetry/baggage-span-processor';
 import { AsyncLocalStorageContextManager } from '@opentelemetry/context-async-hooks';
 import { CompositePropagator, W3CTraceContextPropagator, W3CBaggagePropagator } from '@opentelemetry/core';
@@ -12,14 +11,13 @@ import { registerInstrumentations } from '@opentelemetry/instrumentation';
 import { Resource } from '@opentelemetry/resources';
 import { NodeTracerProvider, SimpleSpanProcessor, BatchSpanProcessor } from '@opentelemetry/sdk-trace-node';
 import { AnthropicInstrumentation } from '@traceloop/instrumentation-anthropic';
-import { AzureOpenAIInstrumentation } from '@traceloop/instrumentation-azure';
 import { BedrockInstrumentation } from '@traceloop/instrumentation-bedrock';
 import { CohereInstrumentation } from '@traceloop/instrumentation-cohere';
 import { LangChainInstrumentation } from '@traceloop/instrumentation-langchain';
 import { LlamaIndexInstrumentation } from '@traceloop/instrumentation-llamaindex';
 import { OpenAIInstrumentation } from '@traceloop/instrumentation-openai';
 import { TogetherInstrumentation } from '@traceloop/instrumentation-together';
-import { VertexAIInstrumentation, AIPlatformInstrumentation } from '@traceloop/instrumentation-vertexai';
+import { AIPlatformInstrumentation, VertexAIInstrumentation } from '@traceloop/instrumentation-vertexai';
 
 class RedactSpanProcessor {
     options;
@@ -76,6 +74,7 @@ const DEFAULT_REDACT_SPAN_PROCESSOR = () => new RedactSpanProcessor({
     attributes: [
         /^.*auth.*$/i,
         /^.*authorization.*$/i,
+        /^(?!ai\.).*usage.*$/i,
         /^(?!gen_ai\.).*token.*$/i,
         /^.*secret.*$/i,
         /^.*key.*$/i,
@@ -118,289 +117,6 @@ function GET_GATEWAY_BASE_URL() {
 }
 const env = { GATEWAY_BASE_URL: GET_GATEWAY_BASE_URL() };
 
-var SegmentSource;
-(function (SegmentSource) {
-    SegmentSource["API"] = "api";
-    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) {
-    SegmentType["Document"] = "document";
-    SegmentType["Step"] = "step";
-})(SegmentType || (SegmentType = {}));
-const SEGMENT_SPECIFICATIONS = {
-    [SegmentType.Document]: {
-        name: 'Prompt',
-        description: 'A prompt',
-    },
-    [SegmentType.Step]: {
-        name: 'Step',
-        description: 'A step in a prompt',
-    },
-};
-const baseSegmentBaggageSchema = z.object({
-    id: z.string(),
-    parentId: z.string().optional(),
-    source: z.nativeEnum(SegmentSource),
-});
-z.discriminatedUnion('type', [
-    baseSegmentBaggageSchema.extend({
-        type: z.literal(SegmentType.Document),
-        data: z.object({
-            logUuid: z.string().optional(), // TODO(tracing): temporal related log, remove when observability is ready
-            commitUuid: z.string(),
-            documentUuid: z.string(),
-            experimentUuid: z.string().optional(),
-            externalId: z.string().optional(),
-        }),
-    }),
-    baseSegmentBaggageSchema.extend({
-        type: z.literal(SegmentType.Step),
-        data: z.undefined().optional(),
-    }),
-]);
-
-var SpanKind;
-(function (SpanKind) {
-    SpanKind["Internal"] = "internal";
-    SpanKind["Server"] = "server";
-    SpanKind["Client"] = "client";
-    SpanKind["Producer"] = "producer";
-    SpanKind["Consumer"] = "consumer";
-})(SpanKind || (SpanKind = {}));
-// Note: loosely based on OpenTelemetry GenAI semantic conventions
-var SpanType;
-(function (SpanType) {
-    SpanType["Tool"] = "tool";
-    SpanType["Completion"] = "completion";
-    SpanType["Embedding"] = "embedding";
-    SpanType["Retrieval"] = "retrieval";
-    SpanType["Reranking"] = "reranking";
-    SpanType["Http"] = "http";
-    SpanType["Segment"] = "segment";
-    SpanType["Unknown"] = "unknown";
-})(SpanType || (SpanType = {}));
-const SPAN_SPECIFICATIONS = {
-    [SpanType.Tool]: {
-        name: 'Tool',
-        description: 'A tool call',
-        isGenAI: true,
-        isHidden: false,
-    },
-    [SpanType.Completion]: {
-        name: 'Completion',
-        description: 'A completion call',
-        isGenAI: true,
-        isHidden: false,
-    },
-    [SpanType.Embedding]: {
-        name: 'Embedding',
-        description: 'An embedding call',
-        isGenAI: true,
-        isHidden: false,
-    },
-    [SpanType.Retrieval]: {
-        name: 'Retrieval',
-        description: 'A retrieval call',
-        isGenAI: true,
-        isHidden: false,
-    },
-    [SpanType.Reranking]: {
-        name: 'Reranking',
-        description: 'A reranking call',
-        isGenAI: true,
-        isHidden: false,
-    },
-    [SpanType.Http]: {
-        name: 'HTTP',
-        description: 'An HTTP request',
-        isGenAI: false,
-        isHidden: true,
-    },
-    [SpanType.Segment]: {
-        name: 'Segment',
-        description: 'A (partial) segment of a trace',
-        isGenAI: false,
-        isHidden: false,
-    },
-    [SpanType.Unknown]: {
-        name: 'Unknown',
-        description: 'An unknown span',
-        isGenAI: false,
-        isHidden: true,
-    },
-};
-var SpanStatus;
-(function (SpanStatus) {
-    SpanStatus["Unset"] = "unset";
-    SpanStatus["Ok"] = "ok";
-    SpanStatus["Error"] = "error";
-})(SpanStatus || (SpanStatus = {}));
-
-// Note: Traces are unmaterialized but this context is used to propagate the trace
-// See www.w3.org/TR/trace-context and w3c.github.io/baggage
-z.object({
-    traceparent: z.string(), // <version>-<trace-id>-<span-id>-<trace-flags>
-    tracestate: z.string().optional(), // <key>=urlencoded(<value>)[,<key>=urlencoded(<value>)]*
-    baggage: z.string().optional(), // <key>=urlencoded(<value>)[,<key>=urlencoded(<value>)]*
-});
-
-/* Note: Instrumentation scopes from all language SDKs */
-const SCOPE_LATITUDE = 'so.latitude.instrumentation';
-var InstrumentationScope;
-(function (InstrumentationScope) {
-    InstrumentationScope["Manual"] = "manual";
-    InstrumentationScope["Latitude"] = "latitude";
-    InstrumentationScope["OpenAI"] = "openai";
-    InstrumentationScope["Anthropic"] = "anthropic";
-    InstrumentationScope["AzureOpenAI"] = "azure";
-    InstrumentationScope["VercelAI"] = "vercelai";
-    InstrumentationScope["VertexAI"] = "vertexai";
-    InstrumentationScope["AIPlatform"] = "aiplatform";
-    InstrumentationScope["MistralAI"] = "mistralai";
-    InstrumentationScope["Bedrock"] = "bedrock";
-    InstrumentationScope["Sagemaker"] = "sagemaker";
-    InstrumentationScope["TogetherAI"] = "togetherai";
-    InstrumentationScope["Replicate"] = "replicate";
-    InstrumentationScope["Groq"] = "groq";
-    InstrumentationScope["Cohere"] = "cohere";
-    InstrumentationScope["LiteLLM"] = "litellm";
-    InstrumentationScope["Langchain"] = "langchain";
-    InstrumentationScope["LlamaIndex"] = "llamaindex";
-    InstrumentationScope["DSPy"] = "dspy";
-    InstrumentationScope["Haystack"] = "haystack";
-    InstrumentationScope["Ollama"] = "ollama";
-    InstrumentationScope["Transformers"] = "transformers";
-    InstrumentationScope["AlephAlpha"] = "alephalpha";
-})(InstrumentationScope || (InstrumentationScope = {}));
-/* Note: non-standard OpenTelemetry semantic conventions used in Latitude */
-const ATTR_LATITUDE = 'latitude';
-const ATTR_LATITUDE_TYPE = `${ATTR_LATITUDE}.type`;
-const ATTR_LATITUDE_SEGMENT_ID = `${ATTR_LATITUDE}.segment.id`;
-const ATTR_LATITUDE_SEGMENT_PARENT_ID = `${ATTR_LATITUDE}.segment.parent_id`;
-const ATTR_LATITUDE_SEGMENTS = `${ATTR_LATITUDE}.segments`;
-const GEN_AI_TOOL_TYPE_VALUE_FUNCTION = 'function';
-const ATTR_GEN_AI_TOOL_CALL_ARGUMENTS = 'gen_ai.tool.call.arguments';
-const ATTR_GEN_AI_TOOL_RESULT_VALUE = 'gen_ai.tool.result.value';
-const ATTR_GEN_AI_TOOL_RESULT_IS_ERROR = 'gen_ai.tool.result.is_error';
-const ATTR_GEN_AI_REQUEST = 'gen_ai.request';
-const ATTR_GEN_AI_REQUEST_CONFIGURATION = 'gen_ai.request.configuration';
-const ATTR_GEN_AI_REQUEST_TEMPLATE = 'gen_ai.request.template';
-const ATTR_GEN_AI_REQUEST_PARAMETERS = 'gen_ai.request.parameters';
-const ATTR_GEN_AI_REQUEST_MESSAGES = 'gen_ai.request.messages';
-const ATTR_GEN_AI_RESPONSE = 'gen_ai.response';
-const ATTR_GEN_AI_RESPONSE_MESSAGES = 'gen_ai.response.messages';
-const ATTR_GEN_AI_USAGE_PROMPT_TOKENS = 'gen_ai.usage.prompt_tokens';
-const ATTR_GEN_AI_USAGE_CACHED_TOKENS = 'gen_ai.usage.cached_tokens';
-const ATTR_GEN_AI_USAGE_REASONING_TOKENS = 'gen_ai.usage.reasoning_tokens'; // prettier-ignore
-const ATTR_GEN_AI_USAGE_COMPLETION_TOKENS = 'gen_ai.usage.completion_tokens'; // prettier-ignore
-const ATTR_GEN_AI_PROMPTS = 'gen_ai.prompt'; // gen_ai.prompt.{index}.{role/content/...}
-const ATTR_GEN_AI_COMPLETIONS = 'gen_ai.completion'; // gen_ai.completion.{index}.{role/content/...}
-const ATTR_GEN_AI_MESSAGE_ROLE = 'role';
-const ATTR_GEN_AI_MESSAGE_CONTENT = 'content'; // string or object
-const ATTR_GEN_AI_MESSAGE_TOOL_NAME = 'tool_name';
-const ATTR_GEN_AI_MESSAGE_TOOL_CALL_ID = 'tool_call_id';
-const ATTR_GEN_AI_MESSAGE_TOOL_RESULT_IS_ERROR = 'is_error';
-const ATTR_GEN_AI_MESSAGE_TOOL_CALLS = 'tool_calls'; // gen_ai.completion.{index}.tool_calls.{index}.{id/name/arguments}
-const ATTR_GEN_AI_MESSAGE_TOOL_CALLS_ID = 'id';
-const ATTR_GEN_AI_MESSAGE_TOOL_CALLS_NAME = 'name';
-const ATTR_GEN_AI_MESSAGE_TOOL_CALLS_ARGUMENTS = 'arguments';
-const GEN_AI_RESPONSE_FINISH_REASON_VALUE_STOP = 'stop';
-const GEN_AI_RESPONSE_FINISH_REASON_VALUE_TOOL_CALLS = 'tool_calls';
-const ATTR_HTTP_REQUEST_URL = 'http.request.url';
-const ATTR_HTTP_REQUEST_BODY = 'http.request.body';
-const ATTR_HTTP_REQUEST_HEADER = 'http.request.header';
-const ATTR_HTTP_RESPONSE_BODY = 'http.response.body';
-const ATTR_HTTP_RESPONSE_HEADER = 'http.response.header';
-/* Note: Schemas for span ingestion following OpenTelemetry service request specification */
-var Otlp;
-(function (Otlp) {
-    Otlp.attributeValueSchema = z.object({
-        stringValue: z.string().optional(),
-        intValue: z.number().optional(),
-        boolValue: z.boolean().optional(),
-        arrayValue: z
-            .object({
-            values: z.array(z.object({
-                stringValue: z.string().optional(),
-                intValue: z.number().optional(),
-                boolValue: z.boolean().optional(),
-            })),
-        })
-            .optional(),
-    });
-    Otlp.attributeSchema = z.object({
-        key: z.string(),
-        value: Otlp.attributeValueSchema,
-    });
-    Otlp.eventSchema = z.object({
-        name: z.string(),
-        timeUnixNano: z.string(),
-        attributes: z.array(Otlp.attributeSchema).optional(),
-    });
-    Otlp.linkSchema = z.object({
-        traceId: z.string(),
-        spanId: z.string(),
-        attributes: z.array(Otlp.attributeSchema).optional(),
-    });
-    (function (StatusCode) {
-        StatusCode[StatusCode["Unset"] = 0] = "Unset";
-        StatusCode[StatusCode["Ok"] = 1] = "Ok";
-        StatusCode[StatusCode["Error"] = 2] = "Error";
-    })(Otlp.StatusCode || (Otlp.StatusCode = {}));
-    Otlp.statusSchema = z.object({
-        code: z.number(),
-        message: z.string().optional(),
-    });
-    (function (SpanKind) {
-        SpanKind[SpanKind["Internal"] = 0] = "Internal";
-        SpanKind[SpanKind["Server"] = 1] = "Server";
-        SpanKind[SpanKind["Client"] = 2] = "Client";
-        SpanKind[SpanKind["Producer"] = 3] = "Producer";
-        SpanKind[SpanKind["Consumer"] = 4] = "Consumer";
-    })(Otlp.SpanKind || (Otlp.SpanKind = {}));
-    Otlp.spanSchema = z.object({
-        traceId: z.string(),
-        spanId: z.string(),
-        parentSpanId: z.string().optional(),
-        name: z.string(),
-        kind: z.number(),
-        startTimeUnixNano: z.string(),
-        endTimeUnixNano: z.string(),
-        status: Otlp.statusSchema.optional(),
-        events: z.array(Otlp.eventSchema).optional(),
-        links: z.array(Otlp.linkSchema).optional(),
-        attributes: z.array(Otlp.attributeSchema).optional(),
-    });
-    Otlp.scopeSchema = z.object({
-        name: z.string(),
-        version: z.string().optional(),
-    });
-    Otlp.scopeSpanSchema = z.object({
-        scope: Otlp.scopeSchema,
-        spans: z.array(Otlp.spanSchema),
-    });
-    Otlp.resourceSchema = z.object({
-        attributes: z.array(Otlp.attributeSchema),
-    });
-    Otlp.resourceSpanSchema = z.object({
-        resource: Otlp.resourceSchema,
-        scopeSpans: z.array(Otlp.scopeSpanSchema),
-    });
-    Otlp.serviceRequestSchema = z.object({
-        resourceSpans: z.array(Otlp.resourceSpanSchema),
-    });
-})(Otlp || (Otlp = {}));
-
 var StreamEventTypes;
 (function (StreamEventTypes) {
     StreamEventTypes["Latitude"] = "latitude-event";
@@ -425,17 +141,27 @@ var LatitudeTool;
     LatitudeTool["RunCode"] = "code";
     LatitudeTool["WebSearch"] = "search";
     LatitudeTool["WebExtract"] = "extract";
+    LatitudeTool["Think"] = "think";
+    LatitudeTool["TODO"] = "todo";
 })(LatitudeTool || (LatitudeTool = {}));
 var LatitudeToolInternalName;
 (function (LatitudeToolInternalName) {
     LatitudeToolInternalName["RunCode"] = "lat_tool_run_code";
     LatitudeToolInternalName["WebSearch"] = "lat_tool_web_search";
     LatitudeToolInternalName["WebExtract"] = "lat_tool_web_extract";
+    LatitudeToolInternalName["Think"] = "think";
+    LatitudeToolInternalName["TODO"] = "todo_write";
 })(LatitudeToolInternalName || (LatitudeToolInternalName = {}));
+[
+    LatitudeTool.Think,
+    LatitudeTool.TODO,
+];
 
 const actualOutputConfiguration = z.object({
     messageSelection: z.enum(['last', 'all']), // Which assistant messages to select
-    contentFilter: z.enum(['text', 'image', 'file', 'tool_call']).optional(),
+    contentFilter: z
+        .enum(['text', 'reasoning', 'image', 'file', 'tool_call'])
+        .optional(),
     parsingFormat: z.enum(['string', 'json']),
     fieldAccessor: z.string().optional(), // Field accessor to get the output from if it's a key-value format
 });
@@ -445,11 +171,11 @@ const expectedOutputConfiguration = z.object({
 });
 const baseEvaluationConfiguration = z.object({
     reverseScale: z.boolean(), // If true, lower is better, otherwise, higher is better
-    actualOutput: actualOutputConfiguration.optional(), // Optional for backwards compatibility
-    expectedOutput: expectedOutputConfiguration.optional(), // Optional for backwards compatibility
+    actualOutput: actualOutputConfiguration,
+    expectedOutput: expectedOutputConfiguration.optional(),
 });
 const baseEvaluationResultMetadata = z.object({
-    // Configuration snapshot is defined in every metric specification
+    // configuration: Configuration snapshot is defined in every metric specification
     actualOutput: z.string(),
     expectedOutput: z.string().optional(),
     datasetLabel: z.string().optional(),
@@ -458,7 +184,78 @@ const baseEvaluationResultError = z.object({
     message: z.string(),
 });
 
+const compositeEvaluationConfiguration = baseEvaluationConfiguration.extend({
+    evaluationUuids: z.array(z.string()),
+    minThreshold: z.number().optional(), // Threshold percentage
+    maxThreshold: z.number().optional(), // Threshold percentage
+});
+const compositeEvaluationResultMetadata = baseEvaluationResultMetadata.extend({
+    results: z.record(z.string(), // Evaluation uuid
+    z.object({
+        uuid: z.string(), // Result uuid (for side effects)
+        name: z.string(), // Evaluation name
+        score: z.number(), // Normalized score
+        reason: z.string(),
+        passed: z.boolean(),
+    })),
+});
+const compositeEvaluationResultError = baseEvaluationResultError.extend({
+    errors: z
+        .record(z.string(), // Evaluation uuid
+    z.object({
+        uuid: z.string(), // Result uuid (for side effects)
+        name: z.string(), // Evaluation name
+        message: z.string(),
+    }))
+        .optional(),
+});
+// AVERAGE
+const compositeEvaluationAverageConfiguration = compositeEvaluationConfiguration.extend({});
+compositeEvaluationResultMetadata.extend({
+    configuration: compositeEvaluationAverageConfiguration,
+});
+compositeEvaluationResultError.extend({});
+const CompositeEvaluationAverageSpecification = {
+    };
+// WEIGHTED
+const compositeEvaluationWeightedConfiguration = compositeEvaluationConfiguration.extend({
+    weights: z.record(z.string(), // Evaluation uuid
+    z.number()),
+});
+compositeEvaluationResultMetadata.extend({
+    configuration: compositeEvaluationWeightedConfiguration,
+});
+compositeEvaluationResultError.extend({});
+const CompositeEvaluationWeightedSpecification = {
+    };
+// CUSTOM
+const compositeEvaluationCustomConfiguration = compositeEvaluationConfiguration.extend({
+    formula: z.string(),
+});
+compositeEvaluationResultMetadata.extend({
+    configuration: compositeEvaluationCustomConfiguration,
+});
+compositeEvaluationResultError.extend({});
+const CompositeEvaluationCustomSpecification = {
+    };
+/* ------------------------------------------------------------------------- */
+var CompositeEvaluationMetric;
+(function (CompositeEvaluationMetric) {
+    CompositeEvaluationMetric["Average"] = "average";
+    CompositeEvaluationMetric["Weighted"] = "weighted";
+    CompositeEvaluationMetric["Custom"] = "custom";
+})(CompositeEvaluationMetric || (CompositeEvaluationMetric = {}));
+const CompositeEvaluationSpecification = {
+    // prettier-ignore
+    metrics: {
+        [CompositeEvaluationMetric.Average]: CompositeEvaluationAverageSpecification,
+        [CompositeEvaluationMetric.Weighted]: CompositeEvaluationWeightedSpecification,
+        [CompositeEvaluationMetric.Custom]: CompositeEvaluationCustomSpecification,
+    },
+};
+
 const humanEvaluationConfiguration = baseEvaluationConfiguration.extend({
+    enableControls: z.boolean().optional(),
     criteria: z.string().optional(),
 });
 const humanEvaluationResultMetadata = baseEvaluationResultMetadata.extend({
@@ -599,7 +396,9 @@ const LlmEvaluationSpecification = {
 };
 
 const ruleEvaluationConfiguration = baseEvaluationConfiguration.extend({});
-const ruleEvaluationResultMetadata = baseEvaluationResultMetadata.extend({});
+const ruleEvaluationResultMetadata = baseEvaluationResultMetadata.extend({
+    reason: z.string().optional(),
+});
 const ruleEvaluationResultError = baseEvaluationResultError.extend({});
 // EXACT MATCH
 const ruleEvaluationExactMatchConfiguration = ruleEvaluationConfiguration.extend({
@@ -709,12 +508,14 @@ var EvaluationType;
     EvaluationType["Rule"] = "rule";
     EvaluationType["Llm"] = "llm";
     EvaluationType["Human"] = "human";
+    EvaluationType["Composite"] = "composite";
 })(EvaluationType || (EvaluationType = {}));
-const EvaluationTypeSchema = z.nativeEnum(EvaluationType);
+const EvaluationTypeSchema = z.enum(EvaluationType);
 const EvaluationMetricSchema = z.union([
-    z.nativeEnum(RuleEvaluationMetric),
-    z.nativeEnum(LlmEvaluationMetric),
-    z.nativeEnum(HumanEvaluationMetric),
+    z.enum(RuleEvaluationMetric),
+    z.enum(LlmEvaluationMetric),
+    z.enum(HumanEvaluationMetric),
+    z.enum(CompositeEvaluationMetric),
 ]);
 const EvaluationConfigurationSchema = z.custom();
 // prettier-ignore
@@ -725,6 +526,7 @@ z.custom();
     [EvaluationType.Rule]: RuleEvaluationSpecification,
     [EvaluationType.Llm]: LlmEvaluationSpecification,
     [EvaluationType.Human]: HumanEvaluationSpecification,
+    [EvaluationType.Composite]: CompositeEvaluationSpecification,
 });
 z.object({
     name: z.string(),
@@ -738,7 +540,6 @@ z.object({
     enableSuggestions: z.boolean().nullable().optional(),
     autoApplySuggestions: z.boolean().nullable().optional(),
 });
-Object.values(SegmentSource).filter((source) => source !== SegmentSource.Evaluation && source !== SegmentSource.Experiment);
 
 var LegacyChainEventTypes;
 (function (LegacyChainEventTypes) {
@@ -759,17 +560,71 @@ var ChainEventTypes;
     ChainEventTypes["StepCompleted"] = "step-completed";
     ChainEventTypes["StepStarted"] = "step-started";
     ChainEventTypes["ToolCompleted"] = "tool-completed";
-    ChainEventTypes["ToolsRequested"] = "tools-requested";
     ChainEventTypes["ToolResult"] = "tool-result";
     ChainEventTypes["ToolsStarted"] = "tools-started";
 })(ChainEventTypes || (ChainEventTypes = {}));
 
+z.object({
+    name: z.string(),
+    provider: z.string(),
+    model: z.string(),
+    temperature: z.number(),
+});
+// Experiment ran from a dataset
+const experimentDatasetSourceSchema = z.object({
+    source: z.literal('dataset'),
+    datasetId: z.number(),
+    fromRow: z.number(),
+    toRow: z.number(),
+    datasetLabels: z.record(z.string(), z.string()),
+    parametersMap: z.record(z.string(), z.number()),
+});
+// Experiment ran from last logs (from commit and creation time of experiment)
+const experimentLogsSourceSchema = z.object({
+    source: z.literal('logs'),
+    count: z.number(),
+});
+// Experiment ran with manual parameters (currently only used for prompts with no parameters)
+const experimentManualSourceSchema = z.object({
+    source: z.literal('manual'),
+    count: z.number(),
+    parametersMap: z.record(z.string(), z.number()),
+});
+z.discriminatedUnion('source', [
+    experimentDatasetSourceSchema,
+    experimentLogsSourceSchema,
+    experimentManualSourceSchema,
+]);
+
+var QuotaType;
+(function (QuotaType) {
+    QuotaType["Seats"] = "seats";
+    QuotaType["Runs"] = "runs";
+    QuotaType["Credits"] = "credits";
+})(QuotaType || (QuotaType = {}));
+var GrantSource;
+(function (GrantSource) {
+    GrantSource["System"] = "system";
+    GrantSource["Subscription"] = "subscription";
+    GrantSource["Purchase"] = "purchase";
+    GrantSource["Reward"] = "reward";
+    GrantSource["Promocode"] = "promocode";
+})(GrantSource || (GrantSource = {}));
+
+var ModifiedDocumentType;
+(function (ModifiedDocumentType) {
+    ModifiedDocumentType["Created"] = "created";
+    ModifiedDocumentType["Updated"] = "updated";
+    ModifiedDocumentType["UpdatedPath"] = "updated_path";
+    ModifiedDocumentType["Deleted"] = "deleted";
+})(ModifiedDocumentType || (ModifiedDocumentType = {}));
+
 var IntegrationType;
 (function (IntegrationType) {
     IntegrationType["Latitude"] = "latitude";
     IntegrationType["ExternalMCP"] = "custom_mcp";
-    IntegrationType["HostedMCP"] = "mcp_server";
     IntegrationType["Pipedream"] = "pipedream";
+    IntegrationType["HostedMCP"] = "mcp_server";
 })(IntegrationType || (IntegrationType = {}));
 var HostedIntegrationType;
 (function (HostedIntegrationType) {
@@ -866,13 +721,40 @@ var HostedIntegrationType;
     // Loops = 'loops', // Does not exist
 })(HostedIntegrationType || (HostedIntegrationType = {}));
 
-// TODO(evalsv2): Remove
-var EvaluationResultableType;
-(function (EvaluationResultableType) {
-    EvaluationResultableType["Boolean"] = "evaluation_resultable_booleans";
-    EvaluationResultableType["Text"] = "evaluation_resultable_texts";
-    EvaluationResultableType["Number"] = "evaluation_resultable_numbers";
-})(EvaluationResultableType || (EvaluationResultableType = {}));
+var LogSources;
+(function (LogSources) {
+    LogSources["API"] = "api";
+    LogSources["AgentAsTool"] = "agent_as_tool";
+    LogSources["Copilot"] = "copilot";
+    LogSources["EmailTrigger"] = "email_trigger";
+    LogSources["Evaluation"] = "evaluation";
+    LogSources["Experiment"] = "experiment";
+    LogSources["IntegrationTrigger"] = "integration_trigger";
+    LogSources["Playground"] = "playground";
+    LogSources["ScheduledTrigger"] = "scheduled_trigger";
+    LogSources["SharedPrompt"] = "shared_prompt";
+    LogSources["ShadowTest"] = "shadow_test";
+    LogSources["ABTestChallenger"] = "ab_test_challenger";
+    LogSources["User"] = "user";
+})(LogSources || (LogSources = {}));
+
+var RunSourceGroup;
+(function (RunSourceGroup) {
+    RunSourceGroup["Production"] = "production";
+    RunSourceGroup["Playground"] = "playground";
+})(RunSourceGroup || (RunSourceGroup = {}));
+({
+    [RunSourceGroup.Production]: [
+        LogSources.API,
+        LogSources.Copilot,
+        LogSources.EmailTrigger,
+        LogSources.IntegrationTrigger,
+        LogSources.ScheduledTrigger,
+        LogSources.SharedPrompt,
+        LogSources.User,
+    ],
+    [RunSourceGroup.Playground]: [LogSources.Playground, LogSources.Experiment],
+});
 
 var MessageRole;
 (function (MessageRole) {
@@ -882,13 +764,242 @@ var MessageRole;
     MessageRole["tool"] = "tool";
 })(MessageRole || (MessageRole = {}));
 
-var ModifiedDocumentType;
-(function (ModifiedDocumentType) {
-    ModifiedDocumentType["Created"] = "created";
-    ModifiedDocumentType["Updated"] = "updated";
-    ModifiedDocumentType["UpdatedPath"] = "updated_path";
-    ModifiedDocumentType["Deleted"] = "deleted";
-})(ModifiedDocumentType || (ModifiedDocumentType = {}));
+var SpanKind;
+(function (SpanKind) {
+    SpanKind["Internal"] = "internal";
+    SpanKind["Server"] = "server";
+    SpanKind["Client"] = "client";
+    SpanKind["Producer"] = "producer";
+    SpanKind["Consumer"] = "consumer";
+})(SpanKind || (SpanKind = {}));
+// Note: loosely based on OpenTelemetry GenAI semantic conventions
+var SpanType;
+(function (SpanType) {
+    SpanType["Tool"] = "tool";
+    SpanType["Completion"] = "completion";
+    SpanType["Embedding"] = "embedding";
+    SpanType["Retrieval"] = "retrieval";
+    SpanType["Reranking"] = "reranking";
+    SpanType["Http"] = "http";
+    SpanType["Unknown"] = "unknown";
+    SpanType["Prompt"] = "prompt";
+    SpanType["Step"] = "step";
+})(SpanType || (SpanType = {}));
+const SPAN_SPECIFICATIONS = {
+    [SpanType.Tool]: {
+        name: 'Tool',
+        description: 'A tool call',
+        isGenAI: true,
+        isHidden: false,
+    },
+    [SpanType.Completion]: {
+        name: 'Completion',
+        description: 'A completion call',
+        isGenAI: true,
+        isHidden: false,
+    },
+    [SpanType.Embedding]: {
+        name: 'Embedding',
+        description: 'An embedding call',
+        isGenAI: true,
+        isHidden: false,
+    },
+    [SpanType.Retrieval]: {
+        name: 'Retrieval',
+        description: 'A retrieval call',
+        isGenAI: true,
+        isHidden: false,
+    },
+    [SpanType.Reranking]: {
+        name: 'Reranking',
+        description: 'A reranking call',
+        isGenAI: true,
+        isHidden: false,
+    },
+    [SpanType.Http]: {
+        name: 'HTTP',
+        description: 'An HTTP request',
+        isGenAI: false,
+        isHidden: true,
+    },
+    [SpanType.Unknown]: {
+        name: 'Unknown',
+        description: 'An unknown span',
+        isGenAI: false,
+        isHidden: true,
+    },
+    [SpanType.Prompt]: {
+        name: 'Prompt',
+        description: 'A prompt span',
+        isGenAI: false,
+        isHidden: false,
+    },
+    [SpanType.Step]: {
+        name: 'Step',
+        description: 'A step span',
+        isGenAI: false,
+        isHidden: false,
+    },
+};
+var SpanStatus;
+(function (SpanStatus) {
+    SpanStatus["Unset"] = "unset";
+    SpanStatus["Ok"] = "ok";
+    SpanStatus["Error"] = "error";
+})(SpanStatus || (SpanStatus = {}));
+
+// Note: Traces are unmaterialized but this context is used to propagate the trace
+// See www.w3.org/TR/trace-context and w3c.github.io/baggage
+z.object({
+    traceparent: z.string(), // <version>-<trace-id>-<span-id>-<trace-flags>
+    tracestate: z.string().optional(), // <key>=urlencoded(<value>)[,<key>=urlencoded(<value>)]*
+    baggage: z.string().optional(), // <key>=urlencoded(<value>)[,<key>=urlencoded(<value>)]*
+});
+
+/* Note: Instrumentation scopes from all language SDKs */
+const SCOPE_LATITUDE = 'so.latitude.instrumentation';
+var InstrumentationScope;
+(function (InstrumentationScope) {
+    InstrumentationScope["Manual"] = "manual";
+    InstrumentationScope["Latitude"] = "latitude";
+    InstrumentationScope["OpenAI"] = "openai";
+    InstrumentationScope["Anthropic"] = "anthropic";
+    InstrumentationScope["AzureOpenAI"] = "azure";
+    InstrumentationScope["VercelAI"] = "vercelai";
+    InstrumentationScope["VertexAI"] = "vertexai";
+    InstrumentationScope["AIPlatform"] = "aiplatform";
+    InstrumentationScope["MistralAI"] = "mistralai";
+    InstrumentationScope["Bedrock"] = "bedrock";
+    InstrumentationScope["Sagemaker"] = "sagemaker";
+    InstrumentationScope["TogetherAI"] = "togetherai";
+    InstrumentationScope["Replicate"] = "replicate";
+    InstrumentationScope["Groq"] = "groq";
+    InstrumentationScope["Cohere"] = "cohere";
+    InstrumentationScope["LiteLLM"] = "litellm";
+    InstrumentationScope["Langchain"] = "langchain";
+    InstrumentationScope["LlamaIndex"] = "llamaindex";
+    InstrumentationScope["DSPy"] = "dspy";
+    InstrumentationScope["Haystack"] = "haystack";
+    InstrumentationScope["Ollama"] = "ollama";
+    InstrumentationScope["Transformers"] = "transformers";
+    InstrumentationScope["AlephAlpha"] = "alephalpha";
+})(InstrumentationScope || (InstrumentationScope = {}));
+/* Note: non-standard OpenTelemetry semantic conventions used in Latitude */
+const ATTR_LATITUDE = 'latitude';
+const ATTR_LATITUDE_TYPE = `${ATTR_LATITUDE}.type`;
+const ATTR_LATITUDE_TEST_DEPLOYMENT_ID = `${ATTR_LATITUDE}.test_deployment_id`;
+const GEN_AI_TOOL_TYPE_VALUE_FUNCTION = 'function';
+const ATTR_GEN_AI_TOOL_CALL_ARGUMENTS = 'gen_ai.tool.call.arguments';
+const ATTR_GEN_AI_TOOL_RESULT_VALUE = 'gen_ai.tool.result.value';
+const ATTR_GEN_AI_TOOL_RESULT_IS_ERROR = 'gen_ai.tool.result.is_error';
+const ATTR_GEN_AI_REQUEST = 'gen_ai.request';
+const ATTR_GEN_AI_REQUEST_CONFIGURATION = 'gen_ai.request.configuration';
+const ATTR_GEN_AI_REQUEST_TEMPLATE = 'gen_ai.request.template';
+const ATTR_GEN_AI_REQUEST_PARAMETERS = 'gen_ai.request.parameters';
+const ATTR_GEN_AI_REQUEST_MESSAGES = 'gen_ai.request.messages';
+const ATTR_GEN_AI_RESPONSE = 'gen_ai.response';
+const ATTR_GEN_AI_RESPONSE_MESSAGES = 'gen_ai.response.messages';
+const ATTR_GEN_AI_USAGE_PROMPT_TOKENS = 'gen_ai.usage.prompt_tokens';
+const ATTR_GEN_AI_USAGE_CACHED_TOKENS = 'gen_ai.usage.cached_tokens';
+const ATTR_GEN_AI_USAGE_REASONING_TOKENS = 'gen_ai.usage.reasoning_tokens'; // prettier-ignore
+const ATTR_GEN_AI_USAGE_COMPLETION_TOKENS = 'gen_ai.usage.completion_tokens'; // prettier-ignore
+const ATTR_GEN_AI_PROMPTS = 'gen_ai.prompt'; // gen_ai.prompt.{index}.{role/content/...}
+const ATTR_GEN_AI_COMPLETIONS = 'gen_ai.completion'; // gen_ai.completion.{index}.{role/content/...}
+const ATTR_GEN_AI_MESSAGE_ROLE = 'role';
+const ATTR_GEN_AI_MESSAGE_CONTENT = 'content'; // string or object
+const ATTR_GEN_AI_MESSAGE_TOOL_NAME = 'tool_name';
+const ATTR_GEN_AI_MESSAGE_TOOL_CALL_ID = 'tool_call_id';
+const ATTR_GEN_AI_MESSAGE_TOOL_RESULT_IS_ERROR = 'is_error';
+const ATTR_GEN_AI_MESSAGE_TOOL_CALLS = 'tool_calls'; // gen_ai.completion.{index}.tool_calls.{index}.{id/name/arguments}
+const ATTR_GEN_AI_MESSAGE_TOOL_CALLS_ID = 'id';
+const ATTR_GEN_AI_MESSAGE_TOOL_CALLS_NAME = 'name';
+const ATTR_GEN_AI_MESSAGE_TOOL_CALLS_ARGUMENTS = 'arguments';
+const GEN_AI_RESPONSE_FINISH_REASON_VALUE_STOP = 'stop';
+const GEN_AI_RESPONSE_FINISH_REASON_VALUE_TOOL_CALLS = 'tool_calls';
+const ATTR_HTTP_REQUEST_URL = 'http.request.url';
+const ATTR_HTTP_REQUEST_BODY = 'http.request.body';
+const ATTR_HTTP_REQUEST_HEADER = 'http.request.header';
+const ATTR_HTTP_RESPONSE_BODY = 'http.response.body';
+const ATTR_HTTP_RESPONSE_HEADER = 'http.response.header';
+/* Note: Schemas for span ingestion following OpenTelemetry service request specification */
+var Otlp;
+(function (Otlp) {
+    Otlp.attributeValueSchema = z.object({
+        stringValue: z.string().optional(),
+        intValue: z.number().optional(),
+        boolValue: z.boolean().optional(),
+        arrayValue: z
+            .object({
+            values: z.array(z.object({
+                stringValue: z.string().optional(),
+                intValue: z.number().optional(),
+                boolValue: z.boolean().optional(),
+            })),
+        })
+            .optional(),
+    });
+    Otlp.attributeSchema = z.object({
+        key: z.string(),
+        value: Otlp.attributeValueSchema,
+    });
+    Otlp.eventSchema = z.object({
+        name: z.string(),
+        timeUnixNano: z.string(),
+        attributes: z.array(Otlp.attributeSchema).optional(),
+    });
+    Otlp.linkSchema = z.object({
+        traceId: z.string(),
+        spanId: z.string(),
+        attributes: z.array(Otlp.attributeSchema).optional(),
+    });
+    (function (StatusCode) {
+        StatusCode[StatusCode["Unset"] = 0] = "Unset";
+        StatusCode[StatusCode["Ok"] = 1] = "Ok";
+        StatusCode[StatusCode["Error"] = 2] = "Error";
+    })(Otlp.StatusCode || (Otlp.StatusCode = {}));
+    Otlp.statusSchema = z.object({
+        code: z.number(),
+        message: z.string().optional(),
+    });
+    (function (SpanKind) {
+        SpanKind[SpanKind["Internal"] = 0] = "Internal";
+        SpanKind[SpanKind["Server"] = 1] = "Server";
+        SpanKind[SpanKind["Client"] = 2] = "Client";
+        SpanKind[SpanKind["Producer"] = 3] = "Producer";
+        SpanKind[SpanKind["Consumer"] = 4] = "Consumer";
+    })(Otlp.SpanKind || (Otlp.SpanKind = {}));
+    Otlp.spanSchema = z.object({
+        traceId: z.string(),
+        spanId: z.string(),
+        parentSpanId: z.string().optional(),
+        name: z.string(),
+        kind: z.number(),
+        startTimeUnixNano: z.string(),
+        endTimeUnixNano: z.string(),
+        status: Otlp.statusSchema.optional(),
+        events: z.array(Otlp.eventSchema).optional(),
+        links: z.array(Otlp.linkSchema).optional(),
+        attributes: z.array(Otlp.attributeSchema).optional(),
+    });
+    Otlp.scopeSchema = z.object({
+        name: z.string(),
+        version: z.string().optional(),
+    });
+    Otlp.scopeSpanSchema = z.object({
+        scope: Otlp.scopeSchema,
+        spans: z.array(Otlp.spanSchema),
+    });
+    Otlp.resourceSchema = z.object({
+        attributes: z.array(Otlp.attributeSchema),
+    });
+    Otlp.resourceSpanSchema = z.object({
+        resource: Otlp.resourceSchema,
+        scopeSpans: z.array(Otlp.scopeSpanSchema),
+    });
+    Otlp.serviceRequestSchema = z.object({
+        resourceSpans: z.array(Otlp.resourceSpanSchema),
+    });
+})(Otlp || (Otlp = {}));
 
 // TODO(tracing): deprecated
 const HEAD_COMMIT = 'live';
@@ -919,6 +1030,12 @@ var DocumentTriggerType;
     DocumentTriggerType["Scheduled"] = "scheduled";
     DocumentTriggerType["Integration"] = "integration";
 })(DocumentTriggerType || (DocumentTriggerType = {}));
+var DocumentTriggerStatus;
+(function (DocumentTriggerStatus) {
+    DocumentTriggerStatus["Pending"] = "pending";
+    DocumentTriggerStatus["Deployed"] = "deployed";
+    DocumentTriggerStatus["Deprecated"] = "deprecated";
+})(DocumentTriggerStatus || (DocumentTriggerStatus = {}));
 var DocumentTriggerParameters;
 (function (DocumentTriggerParameters) {
     DocumentTriggerParameters["SenderEmail"] = "senderEmail";
@@ -930,11 +1047,9 @@ var DocumentTriggerParameters;
 
 class ManualInstrumentation {
     enabled;
-    source;
     tracer;
-    constructor(source, tracer) {
+    constructor(tracer) {
         this.enabled = false;
-        this.source = source;
         this.tracer = tracer;
     }
     isEnabled() {
@@ -946,96 +1061,9 @@ class ManualInstrumentation {
     disable() {
         this.enabled = false;
     }
-    baggage(ctx) {
-        if ('traceparent' in ctx) {
-            ctx = propagation.extract(otel.ROOT_CONTEXT, ctx);
-        }
-        const baggage = Object.fromEntries(propagation.getBaggage(ctx)?.getAllEntries() || []);
-        if (!(ATTR_LATITUDE_SEGMENT_ID in baggage) ||
-            !(ATTR_LATITUDE_SEGMENTS in baggage)) {
-            return undefined;
-        }
-        const segment = {
-            id: baggage[ATTR_LATITUDE_SEGMENT_ID].value,
-            parentId: baggage[ATTR_LATITUDE_SEGMENT_PARENT_ID]?.value,
-        };
-        let segments = [];
-        try {
-            segments = JSON.parse(baggage[ATTR_LATITUDE_SEGMENTS].value);
-        }
-        catch (error) {
-            return undefined;
-        }
-        if (segments.length < 1) {
-            return undefined;
-        }
-        return { segment, segments };
-    }
-    setBaggage(ctx, baggage, extra) {
-        let parent = Object.fromEntries(propagation.getBaggage(ctx)?.getAllEntries() || []);
-        parent = Object.fromEntries(Object.entries(parent).filter(([attribute]) => attribute !== ATTR_LATITUDE_SEGMENT_ID &&
-            attribute !== ATTR_LATITUDE_SEGMENT_PARENT_ID &&
-            attribute !== ATTR_LATITUDE_SEGMENTS));
-        if (!baggage) {
-            const payload = propagation.createBaggage({ ...parent, ...(extra || {}) });
-            return propagation.setBaggage(ctx, payload);
-        }
-        let jsonSegments = '';
-        try {
-            jsonSegments = JSON.stringify(baggage.segments);
-        }
-        catch (error) {
-            jsonSegments = '[]';
-        }
-        const payload = propagation.createBaggage({
-            ...parent,
-            [ATTR_LATITUDE_SEGMENT_ID]: { value: baggage.segment.id },
-            ...(baggage.segment.parentId && {
-                [ATTR_LATITUDE_SEGMENT_PARENT_ID]: { value: baggage.segment.parentId },
-            }),
-            [ATTR_LATITUDE_SEGMENTS]: { value: jsonSegments },
-            ...(extra || {}),
-        });
-        return propagation.setBaggage(ctx, payload);
-    }
-    pause(ctx) {
-        const baggage = this.baggage(ctx);
-        if (baggage) {
-            baggage.segments.at(-1).paused = true;
-        }
-        ctx = this.setBaggage(ctx, baggage);
-        let carrier = {};
-        propagation.inject(ctx, carrier);
-        return carrier;
-    }
     resume(ctx) {
         return propagation.extract(otel.ROOT_CONTEXT, ctx);
     }
-    restored(ctx) {
-        const baggage = this.baggage(ctx);
-        return !baggage?.segments.some((segment) => segment.paused);
-    }
-    restore(ctx) {
-        let baggage = this.baggage(ctx);
-        if (!baggage)
-            return ctx;
-        const segments = baggage.segments;
-        while (segments.at(-1)?.paused)
-            segments.pop();
-        const segment = segments.at(-1);
-        if (!segment)
-            return otel.ROOT_CONTEXT;
-        baggage = {
-            segment: { id: segment.id, parentId: segment.parentId },
-            segments: segments,
-        };
-        ctx = this.setBaggage(ctx, baggage);
-        let carrier = {};
-        propagation.inject(ctx, carrier);
-        carrier.traceparent = segment.traceparent;
-        carrier.tracestate = segment.tracestate;
-        return this.resume(carrier);
-    }
     capitalize(str) {
         if (str.length === 0)
             return str;
@@ -1368,6 +1396,9 @@ class ManualInstrumentation {
                 [ATTR_GEN_AI_REQUEST_MESSAGES]: jsonInput,
                 ...attrInput,
                 ...(start.attributes || {}),
+                ['latitude.commitUuid']: start.versionUuid,
+                ['latitude.documentUuid']: start.promptUuid,
+                ['latitude.experimentUuid']: start.experimentUuid,
             },
         });
         return {
@@ -1482,49 +1513,7 @@ class ManualInstrumentation {
             fail: span.fail,
         };
     }
-    segment(ctx, type, data, options) {
-        options = options || {};
-        let baggage = this.baggage(ctx);
-        const parent = baggage?.segments.at(-1);
-        const segments = baggage?.segments || [];
-        segments.push({
-            ...{
-                id: options._internal?.id || v4(),
-                ...(parent?.id && { parentId: parent.id }),
-                source: options._internal?.source || parent?.source || this.source,
-                type: type,
-                data: data,
-            },
-            traceparent: 'undefined',
-            tracestate: undefined,
-        });
-        const segment = segments.at(-1);
-        baggage = {
-            segment: { id: segment.id, parentId: segment.parentId },
-            segments: segments,
-        };
-        ctx = this.setBaggage(ctx, baggage, options.baggage);
-        // Dummy wrapper to force the same trace and carry on some segment attributes
-        const span = this.span(ctx, SEGMENT_SPECIFICATIONS[type].name, SpanType.Segment, { attributes: options.attributes });
-        let carrier = {};
-        propagation.inject(span.context, carrier);
-        baggage.segments.at(-1).traceparent = carrier.traceparent;
-        baggage.segments.at(-1).tracestate = carrier.tracestate;
-        // Fix current segment span segments attribute now that we know the trace
-        trace
-            .getSpan(span.context)
-            .setAttribute(ATTR_LATITUDE_SEGMENTS, JSON.stringify(baggage.segments));
-        ctx = this.setBaggage(span.context, baggage, options.baggage);
-        return { context: ctx, end: span.end, fail: span.fail };
-    }
-    prompt(ctx, { logUuid, versionUuid, promptUuid, experimentUuid, externalId, template, parameters, ...rest }) {
-        const baggage = {
-            ...(logUuid && { logUuid }), // TODO(tracing): temporal related log, remove when observability is ready
-            commitUuid: versionUuid || HEAD_COMMIT,
-            documentUuid: promptUuid,
-            ...(experimentUuid && { experimentUuid }),
-            ...(externalId && { externalId }),
-        };
+    prompt(ctx, { documentLogUuid, versionUuid, promptUuid, projectId, experimentUuid, testDeploymentId, externalId, template, parameters, name, source, ...rest }) {
         let jsonParameters = '';
         try {
             jsonParameters = JSON.stringify(parameters || {});
@@ -1535,23 +1524,32 @@ class ManualInstrumentation {
         const attributes = {
             [ATTR_GEN_AI_REQUEST_TEMPLATE]: template,
             [ATTR_GEN_AI_REQUEST_PARAMETERS]: jsonParameters,
+            ['latitude.commitUuid']: versionUuid || HEAD_COMMIT,
+            ['latitude.documentUuid']: promptUuid,
+            ['latitude.projectId']: projectId,
+            ...(documentLogUuid && { ['latitude.documentLogUuid']: documentLogUuid }),
+            ...(experimentUuid && { ['latitude.experimentUuid']: experimentUuid }),
+            ...(testDeploymentId && {
+                [ATTR_LATITUDE_TEST_DEPLOYMENT_ID]: testDeploymentId,
+            }),
+            ...(externalId && { ['latitude.externalId']: externalId }),
+            ...(source && { ['latitude.source']: source }),
             ...(rest.attributes || {}),
         };
-        return this.segment(ctx, SegmentType.Document, baggage, {
-            ...rest,
+        return this.span(ctx, name || `prompt-${promptUuid}`, SpanType.Prompt, {
             attributes,
         });
     }
     step(ctx, options) {
-        return this.segment(ctx, SegmentType.Step, undefined, options);
+        return this.span(ctx, 'step', SpanType.Step, options);
     }
 }
 
 class LatitudeInstrumentation {
     options;
     telemetry;
-    constructor(source, tracer, options) {
-        this.telemetry = new ManualInstrumentation(source, tracer);
+    constructor(tracer, options) {
+        this.telemetry = new ManualInstrumentation(tracer);
         this.options = options;
     }
     isEnabled() {
@@ -1584,44 +1582,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 +1671,8 @@ class LatitudeInstrumentation {
         }
         $tool.end({
             result: {
-                value: result,
-                isError: false, // Note: currently unknown
+                value: result.result,
+                isError: result.isError,
             },
         });
         return result;
@@ -1747,18 +1707,16 @@ const DEFAULT_SPAN_EXPORTER = (apiKey) => new OTLPTraceExporter({
 // Note: Only exporting typescript instrumentations
 var Instrumentation;
 (function (Instrumentation) {
-    Instrumentation["Latitude"] = "latitude";
-    Instrumentation["OpenAI"] = "openai";
     Instrumentation["Anthropic"] = "anthropic";
-    Instrumentation["AzureOpenAI"] = "azure";
-    Instrumentation["VercelAI"] = "vercelai";
-    Instrumentation["VertexAI"] = "vertexai";
     Instrumentation["AIPlatform"] = "aiplatform";
     Instrumentation["Bedrock"] = "bedrock";
-    Instrumentation["TogetherAI"] = "togetherai";
     Instrumentation["Cohere"] = "cohere";
     Instrumentation["Langchain"] = "langchain";
+    Instrumentation["Latitude"] = "latitude";
     Instrumentation["LlamaIndex"] = "llamaindex";
+    Instrumentation["OpenAI"] = "openai";
+    Instrumentation["TogetherAI"] = "togetherai";
+    Instrumentation["VertexAI"] = "vertexai";
 })(Instrumentation || (Instrumentation = {}));
 class LatitudeTelemetry {
     options;
@@ -1824,136 +1782,37 @@ class LatitudeTelemetry {
     initInstrumentations() {
         this.instrumentations = [];
         const tracer = this.tracer(InstrumentationScope.Manual);
-        this.telemetry = new ManualInstrumentation(SegmentSource.API, tracer);
+        this.telemetry = new ManualInstrumentation(tracer);
         this.instrumentations.push(this.telemetry);
         const latitude = this.options.instrumentations?.latitude;
         if (latitude) {
             const tracer = this.tracer(Instrumentation.Latitude);
-            const instrumentation = new LatitudeInstrumentation(SegmentSource.API, tracer, typeof latitude === 'object' ? latitude : { module: latitude });
-            this.instrumentations.push(instrumentation);
-        }
-        const openai = this.options.instrumentations?.openai;
-        if (openai) {
-            const provider = this.tracerProvider(Instrumentation.OpenAI);
-            const instrumentation = new OpenAIInstrumentation({ enrichTokens: true });
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(openai);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
+            const instrumentation = new LatitudeInstrumentation(tracer, typeof latitude === 'object' ? latitude : { module: latitude });
             this.instrumentations.push(instrumentation);
         }
-        const anthropic = this.options.instrumentations?.anthropic;
-        if (anthropic) {
-            const provider = this.tracerProvider(Instrumentation.Anthropic);
-            const instrumentation = new AnthropicInstrumentation();
+        const configureInstrumentation = (instrumentationType, InstrumentationConstructor, instrumentationOptions) => {
+            const providerPkg = this.options.instrumentations?.[instrumentationType];
+            const provider = this.tracerProvider(instrumentationType);
+            const instrumentation = new InstrumentationConstructor(instrumentationOptions); // prettier-ignore
             instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(anthropic);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const azure = this.options.instrumentations?.azure;
-        if (azure) {
-            const provider = this.tracerProvider(Instrumentation.AzureOpenAI);
-            const instrumentation = new AzureOpenAIInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(azure);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const vertexai = this.options.instrumentations?.vertexai;
-        if (vertexai) {
-            const provider = this.tracerProvider(Instrumentation.VertexAI);
-            const instrumentation = new VertexAIInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(vertexai);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const aiplatform = this.options.instrumentations?.aiplatform;
-        if (aiplatform) {
-            const provider = this.tracerProvider(Instrumentation.AIPlatform);
-            const instrumentation = new AIPlatformInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(aiplatform);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const bedrock = this.options.instrumentations?.bedrock;
-        if (bedrock) {
-            const provider = this.tracerProvider(Instrumentation.Bedrock);
-            const instrumentation = new BedrockInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(bedrock);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const togetherai = this.options.instrumentations?.togetherai;
-        if (togetherai) {
-            const provider = this.tracerProvider(Instrumentation.TogetherAI);
-            const instrumentation = new TogetherInstrumentation({
-                enrichTokens: true,
-            });
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(togetherai);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const cohere = this.options.instrumentations?.cohere;
-        if (cohere) {
-            const provider = this.tracerProvider(Instrumentation.Cohere);
-            const instrumentation = new CohereInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(cohere);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const langchain = this.options.instrumentations?.langchain;
-        if (langchain) {
-            const provider = this.tracerProvider(Instrumentation.Langchain);
-            const instrumentation = new LangChainInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(langchain);
-            registerInstrumentations({
-                instrumentations: [instrumentation],
-                tracerProvider: provider,
-            });
-            this.instrumentations.push(instrumentation);
-        }
-        const llamaindex = this.options.instrumentations?.llamaindex;
-        if (llamaindex) {
-            const provider = this.tracerProvider(Instrumentation.LlamaIndex);
-            const instrumentation = new LlamaIndexInstrumentation();
-            instrumentation.setTracerProvider(provider);
-            instrumentation.manuallyInstrument(llamaindex);
+            if (providerPkg) {
+                instrumentation.manuallyInstrument(providerPkg);
+            }
             registerInstrumentations({
                 instrumentations: [instrumentation],
                 tracerProvider: provider,
             });
             this.instrumentations.push(instrumentation);
-        }
+        };
+        configureInstrumentation(Instrumentation.Anthropic, AnthropicInstrumentation); // prettier-ignore
+        configureInstrumentation(Instrumentation.AIPlatform, AIPlatformInstrumentation); // prettier-ignore
+        configureInstrumentation(Instrumentation.Bedrock, BedrockInstrumentation); // prettier-ignore
+        configureInstrumentation(Instrumentation.Cohere, CohereInstrumentation); // prettier-ignore
+        configureInstrumentation(Instrumentation.Langchain, LangChainInstrumentation); // prettier-ignore
+        configureInstrumentation(Instrumentation.LlamaIndex, LlamaIndexInstrumentation); // prettier-ignore
+        configureInstrumentation(Instrumentation.OpenAI, OpenAIInstrumentation, { enrichTokens: true }); // prettier-ignore
+        configureInstrumentation(Instrumentation.TogetherAI, TogetherInstrumentation, { enrichTokens: true }); // prettier-ignore
+        configureInstrumentation(Instrumentation.VertexAI, VertexAIInstrumentation); // prettier-ignore
     }
     instrument() {
         this.instrumentations.forEach((instrumentation) => {
@@ -1967,21 +1826,9 @@ class LatitudeTelemetry {
                 instrumentation.disable();
         });
     }
-    baggage(ctx) {
-        return this.telemetry.baggage(ctx);
-    }
-    pause(ctx) {
-        return this.telemetry.pause(ctx);
-    }
     resume(ctx) {
         return this.telemetry.resume(ctx);
     }
-    restored(ctx) {
-        return this.telemetry.restored(ctx);
-    }
-    restore(ctx) {
-        return this.telemetry.restore(ctx);
-    }
     tool(ctx, options) {
         return this.telemetry.tool(ctx, options);
     }