@latitude-data/telemetry 1.0.2 → 1.0.4

package/dist/index.d.ts

@@ -4,20 +4,6 @@ import { SpanProcessor, ReadableSpan, SpanExporter } from '@opentelemetry/sdk-tr
 import * as latitude from '@latitude-data/sdk';
 import { z } from 'zod';
 import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
-import * as anthropic from '@anthropic-ai/sdk';
-import * as bedrock from '@aws-sdk/client-bedrock-runtime';
-import * as azure from '@azure/openai';
-import * as aiplatform from '@google-cloud/aiplatform';
-import * as vertexai from '@google-cloud/vertexai';
-import * as langchain_runnables from '@langchain/core/runnables';
-import * as langchain_vectorstores from '@langchain/core/vectorstores';
-import * as cohere from 'cohere-ai';
-import * as langchain_agents from 'langchain/agents';
-import * as langchain_chains from 'langchain/chains';
-import * as langchain_tools from 'langchain/tools';
-import * as llamaindex from 'llamaindex';
-import * as openai from 'openai';
-import * as togetherai from 'together-ai';
 
 interface RedactSpanProcessorOptions {
     attributes: (string | RegExp)[];
@@ -35,121 +21,33 @@ declare class RedactSpanProcessor implements SpanProcessor {
 }
 declare const DEFAULT_REDACT_SPAN_PROCESSOR: () => RedactSpanProcessor;
 
-declare enum SegmentSource {
+type LatitudeInstrumentationOptions = {
+    module: typeof latitude.Latitude;
+    completions?: boolean;
+};
+
+declare enum LogSources {
     API = "api",
-    AgentAsTool = "agent_as_tool",// TODO(tracing): deprecated, use SegmentType.Document with DocumentType.Agent instead
+    AgentAsTool = "agent_as_tool",
     Copilot = "copilot",
     EmailTrigger = "email_trigger",
-    Evaluation = "evaluation",// Note: from prompts of llm evaluations
+    Evaluation = "evaluation",
     Experiment = "experiment",
+    IntegrationTrigger = "integration_trigger",
     Playground = "playground",
     ScheduledTrigger = "scheduled_trigger",
-    IntegrationTrigger = "integration_trigger",
     SharedPrompt = "shared_prompt",
+    ShadowTest = "shadow_test",
+    ABTestChallenger = "ab_test_challenger",
     User = "user"
 }
-declare enum SegmentType {
-    Document = "document",
-    Step = "step"
-}
-declare const segmentBaggageSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<z.objectUtil.extendShape<{
-    id: z.ZodString;
-    parentId: z.ZodOptional<z.ZodString>;
-    source: z.ZodNativeEnum<typeof SegmentSource>;
-}, {
-    type: z.ZodLiteral<SegmentType.Document>;
-    data: z.ZodObject<{
-        logUuid: z.ZodOptional<z.ZodString>;
-        commitUuid: z.ZodString;
-        documentUuid: z.ZodString;
-        experimentUuid: z.ZodOptional<z.ZodString>;
-        externalId: z.ZodOptional<z.ZodString>;
-    }, "strip", z.ZodTypeAny, {
-        commitUuid: string;
-        documentUuid: string;
-        logUuid?: string | undefined;
-        experimentUuid?: string | undefined;
-        externalId?: string | undefined;
-    }, {
-        commitUuid: string;
-        documentUuid: string;
-        logUuid?: string | undefined;
-        experimentUuid?: string | undefined;
-        externalId?: string | undefined;
-    }>;
-}>, "strip", z.ZodTypeAny, {
-    id: string;
-    type: SegmentType.Document;
-    data: {
-        commitUuid: string;
-        documentUuid: string;
-        logUuid?: string | undefined;
-        experimentUuid?: string | undefined;
-        externalId?: string | undefined;
-    };
-    source: SegmentSource;
-    parentId?: string | undefined;
-}, {
-    id: string;
-    type: SegmentType.Document;
-    data: {
-        commitUuid: string;
-        documentUuid: string;
-        logUuid?: string | undefined;
-        experimentUuid?: string | undefined;
-        externalId?: string | undefined;
-    };
-    source: SegmentSource;
-    parentId?: string | undefined;
-}>, z.ZodObject<z.objectUtil.extendShape<{
-    id: z.ZodString;
-    parentId: z.ZodOptional<z.ZodString>;
-    source: z.ZodNativeEnum<typeof SegmentSource>;
-}, {
-    type: z.ZodLiteral<SegmentType.Step>;
-    data: z.ZodOptional<z.ZodUndefined>;
-}>, "strip", z.ZodTypeAny, {
-    id: string;
-    type: SegmentType.Step;
-    source: SegmentSource;
-    data?: undefined;
-    parentId?: string | undefined;
-}, {
-    id: string;
-    type: SegmentType.Step;
-    source: SegmentSource;
-    data?: undefined;
-    parentId?: string | undefined;
-}>]>;
-type SegmentBaggage<T extends SegmentType = SegmentType> = Extract<z.infer<typeof segmentBaggageSchema>, {
-    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;
-}>;
+}, z.core.$strip>;
 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;
-};
 
 type StartSpanOptions = {
     name?: string;
@@ -179,6 +77,9 @@ type StartCompletionSpanOptions = StartSpanOptions & {
     model: string;
     configuration: Record<string, unknown>;
     input: Record<string, unknown>[];
+    versionUuid?: string;
+    promptUuid?: string;
+    experimentUuid?: string;
 };
 type EndCompletionSpanOptions = EndSpanOptions & {
     output: Record<string, unknown>[];
@@ -205,22 +106,17 @@ type EndHttpSpanOptions = EndSpanOptions & {
         body: string | Record<string, unknown>;
     };
 };
-type SegmentOptions = {
-    attributes?: otel.Attributes;
-    baggage?: Record<string, otel.BaggageEntry>;
-    _internal?: {
-        id?: string;
-        source?: SegmentSource;
-    };
-};
-type PromptSegmentOptions = SegmentOptions & {
-    logUuid?: string;
+type PromptSpanOptions = StartSpanOptions & {
+    documentLogUuid?: string;
     versionUuid?: string;
     promptUuid: string;
+    projectId?: string;
     experimentUuid?: string;
+    testDeploymentId?: number;
     externalId?: string;
     template: string;
     parameters?: Record<string, unknown>;
+    source?: LogSources;
 };
 
 type TelemetryContext = otel.Context;
@@ -234,44 +130,36 @@ declare class ScopedTracerProvider implements otel.TracerProvider {
 }
 declare const DEFAULT_SPAN_EXPORTER: (apiKey: string) => OTLPTraceExporter;
 declare enum Instrumentation {
-    Latitude = "latitude",
-    OpenAI = "openai",
     Anthropic = "anthropic",
-    AzureOpenAI = "azure",
-    VercelAI = "vercelai",
-    VertexAI = "vertexai",
     AIPlatform = "aiplatform",
     Bedrock = "bedrock",
-    TogetherAI = "togetherai",
     Cohere = "cohere",
     Langchain = "langchain",
-    LlamaIndex = "llamaindex"
+    Latitude = "latitude",
+    LlamaIndex = "llamaindex",
+    OpenAI = "openai",
+    TogetherAI = "togetherai",
+    VertexAI = "vertexai"
 }
 type TelemetryOptions = {
+    disableBatch?: boolean;
+    exporter?: SpanExporter;
+    processors?: SpanProcessor[];
+    propagators?: TextMapPropagator[];
     instrumentations?: {
         [Instrumentation.Latitude]?: typeof latitude.Latitude | LatitudeInstrumentationOptions;
-        [Instrumentation.OpenAI]?: typeof openai.OpenAI;
-        [Instrumentation.Anthropic]?: typeof anthropic;
-        [Instrumentation.AzureOpenAI]?: typeof azure;
-        [Instrumentation.VercelAI]?: 'manual';
-        [Instrumentation.VertexAI]?: typeof vertexai;
-        [Instrumentation.AIPlatform]?: typeof aiplatform;
-        [Instrumentation.Bedrock]?: typeof bedrock;
-        [Instrumentation.TogetherAI]?: typeof togetherai.Together;
-        [Instrumentation.Cohere]?: typeof cohere;
+        [Instrumentation.AIPlatform]?: any;
+        [Instrumentation.Anthropic]?: any;
+        [Instrumentation.Bedrock]?: any;
+        [Instrumentation.Cohere]?: any;
+        [Instrumentation.OpenAI]?: any;
+        [Instrumentation.LlamaIndex]?: any;
+        [Instrumentation.TogetherAI]?: any;
+        [Instrumentation.VertexAI]?: any;
         [Instrumentation.Langchain]?: {
-            chainsModule: typeof langchain_chains;
-            agentsModule: typeof langchain_agents;
-            toolsModule: typeof langchain_tools;
-            vectorStoreModule: typeof langchain_vectorstores;
-            runnablesModule: typeof langchain_runnables;
+            callbackManagerModule?: any;
         };
-        [Instrumentation.LlamaIndex]?: typeof llamaindex;
     };
-    disableBatch?: boolean;
-    exporter?: SpanExporter;
-    processors?: SpanProcessor[];
-    propagators?: TextMapPropagator[];
 };
 declare class LatitudeTelemetry {
     private options;
@@ -286,15 +174,7 @@ declare class LatitudeTelemetry {
     private initInstrumentations;
     instrument(): void;
     uninstrument(): void;
-    baggage(ctx: otel.Context | TraceContext): TraceBaggage | undefined;
-    pause(ctx: otel.Context): {
-        traceparent: string;
-        tracestate?: string | undefined;
-        baggage?: string | undefined;
-    };
     resume(ctx: TraceContext): otel.Context;
-    restored(ctx: otel.Context): boolean;
-    restore(ctx: otel.Context): otel.Context;
     tool(ctx: otel.Context, options: StartToolSpanOptions): {
         context: otel.Context;
         end: (options: EndToolSpanOptions) => void;
@@ -325,12 +205,12 @@ declare class LatitudeTelemetry {
         end: (options: EndHttpSpanOptions) => void;
         fail: (_error: Error, _options?: ErrorOptions) => void;
     };
-    prompt(ctx: otel.Context, options: PromptSegmentOptions): {
+    prompt(ctx: otel.Context, options: PromptSpanOptions): {
         context: otel.Context;
         end: (_options?: EndSpanOptions) => void;
         fail: (_error: Error, _options?: ErrorOptions) => void;
     };
-    step(ctx: otel.Context, options?: SegmentOptions): {
+    step(ctx: otel.Context, options?: StartSpanOptions): {
         context: otel.Context;
         end: (_options?: EndSpanOptions) => void;
         fail: (_error: Error, _options?: ErrorOptions) => void;
@@ -338,4 +218,4 @@ declare class LatitudeTelemetry {
 }
 
 export { BACKGROUND, DEFAULT_REDACT_SPAN_PROCESSOR, DEFAULT_SPAN_EXPORTER, Instrumentation, LatitudeTelemetry, RedactSpanProcessor };
-export type { EndCompletionSpanOptions, EndHttpSpanOptions, EndSpanOptions, EndToolSpanOptions, ErrorOptions, PromptSegmentOptions, RedactSpanProcessorOptions, SegmentOptions, StartCompletionSpanOptions, StartHttpSpanOptions, StartSpanOptions, StartToolSpanOptions, TelemetryContext, TelemetryOptions };
+export type { EndCompletionSpanOptions, EndHttpSpanOptions, EndSpanOptions, EndToolSpanOptions, ErrorOptions, PromptSpanOptions as PromptSegmentOptions, RedactSpanProcessorOptions, StartCompletionSpanOptions, StartHttpSpanOptions, StartSpanOptions, StartToolSpanOptions, TelemetryContext, TelemetryOptions };