tracia 0.3.10 → 0.4.0

package/dist/index.d.mts

@@ -8,6 +8,8 @@ interface RunVariables {
 }
 interface RunOptions {
     model?: string;
+    /** Run a specific prompt version instead of the latest */
+    version?: number;
     tags?: string[];
     userId?: string;
     sessionId?: string;
@@ -61,7 +63,13 @@ declare enum TraciaErrorCode {
 declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
-    GOOGLE = "google"
+    GOOGLE = "google",
+    AMAZON_BEDROCK = "amazon_bedrock"
+}
+declare enum SpanKind {
+    ROOT = "ROOT",
+    LLM = "LLM",
+    EMBEDDING = "EMBEDDING"
 }
 interface ApiSuccessResponse {
     text: string;
@@ -338,7 +346,8 @@ interface CreateSpanPayload {
     model: string;
     provider: LLMProvider;
     input: {
-        messages: LocalPromptMessage[];
+        messages?: LocalPromptMessage[];
+        text?: string | string[];
     };
     variables: Record<string, string> | null;
     output: string | null;
@@ -360,6 +369,8 @@ interface CreateSpanPayload {
     traceId?: string;
     /** Parent span ID for chaining spans in a sequence */
     parentSpanId?: string;
+    /** Span kind override (for embedding spans) */
+    spanKind?: SpanKind;
 }
 interface CreateSpanResult {
     spanId: string;
@@ -558,6 +569,65 @@ interface ResponsesStream {
     /** Abort the stream */
     abort(): void;
 }
+interface RunEmbeddingInput {
+    /** Text(s) to embed */
+    input: string | string[];
+    /** Embedding model (e.g., 'text-embedding-3-small') */
+    model: string;
+    /** Explicitly specify the provider */
+    provider?: LLMProvider;
+    /** Provider API key override */
+    providerApiKey?: string;
+    /** Optional dimension override */
+    dimensions?: number;
+    /** Whether to send span to Tracia (default: true) */
+    sendTrace?: boolean;
+    /** Custom span ID */
+    spanId?: string;
+    /** Tags for the span */
+    tags?: string[];
+    /** User ID for the span */
+    userId?: string;
+    /** Session ID for the span */
+    sessionId?: string;
+    /** Trace ID to group related spans together */
+    traceId?: string;
+    /** Parent span ID for chaining spans in a sequence */
+    parentSpanId?: string;
+    /** Timeout in milliseconds */
+    timeoutMs?: number;
+}
+interface EmbeddingVector {
+    /** The embedding values */
+    values: number[];
+    /** Index of this embedding in the input array */
+    index: number;
+}
+interface EmbeddingUsage {
+    /** Total tokens used for the embedding request */
+    totalTokens: number;
+}
+interface RunEmbeddingResult {
+    /** The generated embeddings */
+    embeddings: EmbeddingVector[];
+    /** Span ID for this request */
+    spanId: string;
+    /** Trace ID grouping related spans */
+    traceId: string;
+    /** Latency in milliseconds */
+    latencyMs: number;
+    /** Token usage */
+    usage: EmbeddingUsage;
+    /**
+     * Estimated cost in USD.
+     * Currently always `null` — cost is calculated server-side when the span is ingested.
+     */
+    cost: number | null;
+    /** The provider used */
+    provider: LLMProvider;
+    /** The model used */
+    model: string;
+}
 /** @deprecated Use SpanStatus instead */
 type TraceStatus = SpanStatus;
 /** @deprecated Use SpanListItem instead */
@@ -609,6 +679,11 @@ type SessionRunLocalInput = Omit<RunLocalInput, 'traceId' | 'parentSpanId'>;
  * as those are managed by the session.
  */
 type SessionRunResponsesInput = Omit<RunResponsesInput, 'traceId' | 'parentSpanId'>;
+/**
+ * Input for session.runEmbedding() - same as RunEmbeddingInput but without traceId/parentSpanId
+ * as those are managed by the session.
+ */
+type SessionRunEmbeddingInput = Omit<RunEmbeddingInput, 'traceId' | 'parentSpanId'>;
 /**
  * A session for grouping related spans together under a single trace.
  *
@@ -688,6 +763,11 @@ declare class TraciaSession {
     }): Promise<RunResponsesResult>;
     private runResponsesNonStreaming;
     private runResponsesStreaming;
+    /**
+     * Generate embeddings, automatically linking the span to this session.
+     * See Tracia.runEmbedding() for full documentation.
+     */
+    runEmbedding(input: SessionRunEmbeddingInput): Promise<RunEmbeddingResult>;
     private updateSessionState;
 }
 
@@ -811,6 +891,29 @@ declare class Tracia {
     private createResponsesStream;
     private createLocalStream;
     private combineAbortSignals;
+    /**
+     * Generate embeddings for text input(s) using an embedding model.
+     *
+     * @example Single text
+     * ```typescript
+     * const result = await tracia.runEmbedding({
+     *   model: 'text-embedding-3-small',
+     *   input: 'Hello world',
+     * })
+     * console.log(result.embeddings[0].values.length) // 1536
+     * ```
+     *
+     * @example Batch
+     * ```typescript
+     * const result = await tracia.runEmbedding({
+     *   model: 'text-embedding-3-small',
+     *   input: ['Hello', 'World'],
+     * })
+     * console.log(result.embeddings.length) // 2
+     * ```
+     */
+    runEmbedding(input: RunEmbeddingInput): Promise<RunEmbeddingResult>;
+    private validateRunEmbeddingInput;
     flush(): Promise<void>;
     /**
      * Create a new session for grouping related spans together under a single trace.
@@ -856,4 +959,4 @@ declare class Tracia {
     private getProviderApiKey;
 }
 
-export { type ContentPart, type CreatePromptOptions, type CreateSpanPayload, type CreateSpanResult, type CreateTracePayload, type CreateTraceResult, Eval, type EvaluateOptions, type EvaluateResult, type FinishReason, type JsonSchemaProperty, LLMProvider, type ListSpansOptions, type ListSpansResult, type ListTracesOptions, type ListTracesResult, type LocalPromptMessage, type LocalStream, type MessageRole, type Prompt, type PromptListItem, type PromptMessage, type ResponsesEvent, type ResponsesInputItem, type ResponsesOutputItem, type ResponsesStream, type RunLocalInput, type RunLocalResult, type RunOptions, type RunResponsesInput, type RunResponsesResult, type RunResult, type RunVariables, type SessionRunLocalInput, type SessionRunResponsesInput, type Span, type SpanListItem, type SpanStatus, type StreamResult, type TextPart, type TokenUsage, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolParameters, type Trace, type TraceListItem, type TraceStatus, Tracia, TraciaError, TraciaErrorCode, type TraciaOptions, TraciaSession, type UpdatePromptOptions };
+export { type ContentPart, type CreatePromptOptions, type CreateSpanPayload, type CreateSpanResult, type CreateTracePayload, type CreateTraceResult, type EmbeddingUsage, type EmbeddingVector, Eval, type EvaluateOptions, type EvaluateResult, type FinishReason, type JsonSchemaProperty, LLMProvider, type ListSpansOptions, type ListSpansResult, type ListTracesOptions, type ListTracesResult, type LocalPromptMessage, type LocalStream, type MessageRole, type Prompt, type PromptListItem, type PromptMessage, type ResponsesEvent, type ResponsesInputItem, type ResponsesOutputItem, type ResponsesStream, type RunEmbeddingInput, type RunEmbeddingResult, type RunLocalInput, type RunLocalResult, type RunOptions, type RunResponsesInput, type RunResponsesResult, type RunResult, type RunVariables, type SessionRunEmbeddingInput, type SessionRunLocalInput, type SessionRunResponsesInput, type Span, SpanKind, type SpanListItem, type SpanStatus, type StreamResult, type TextPart, type TokenUsage, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolParameters, type Trace, type TraceListItem, type TraceStatus, Tracia, TraciaError, TraciaErrorCode, type TraciaOptions, TraciaSession, type UpdatePromptOptions };

package/dist/index.d.ts

@@ -8,6 +8,8 @@ interface RunVariables {
 }
 interface RunOptions {
     model?: string;
+    /** Run a specific prompt version instead of the latest */
+    version?: number;
     tags?: string[];
     userId?: string;
     sessionId?: string;
@@ -61,7 +63,13 @@ declare enum TraciaErrorCode {
 declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
-    GOOGLE = "google"
+    GOOGLE = "google",
+    AMAZON_BEDROCK = "amazon_bedrock"
+}
+declare enum SpanKind {
+    ROOT = "ROOT",
+    LLM = "LLM",
+    EMBEDDING = "EMBEDDING"
 }
 interface ApiSuccessResponse {
     text: string;
@@ -338,7 +346,8 @@ interface CreateSpanPayload {
     model: string;
     provider: LLMProvider;
     input: {
-        messages: LocalPromptMessage[];
+        messages?: LocalPromptMessage[];
+        text?: string | string[];
     };
     variables: Record<string, string> | null;
     output: string | null;
@@ -360,6 +369,8 @@ interface CreateSpanPayload {
     traceId?: string;
     /** Parent span ID for chaining spans in a sequence */
     parentSpanId?: string;
+    /** Span kind override (for embedding spans) */
+    spanKind?: SpanKind;
 }
 interface CreateSpanResult {
     spanId: string;
@@ -558,6 +569,65 @@ interface ResponsesStream {
     /** Abort the stream */
     abort(): void;
 }
+interface RunEmbeddingInput {
+    /** Text(s) to embed */
+    input: string | string[];
+    /** Embedding model (e.g., 'text-embedding-3-small') */
+    model: string;
+    /** Explicitly specify the provider */
+    provider?: LLMProvider;
+    /** Provider API key override */
+    providerApiKey?: string;
+    /** Optional dimension override */
+    dimensions?: number;
+    /** Whether to send span to Tracia (default: true) */
+    sendTrace?: boolean;
+    /** Custom span ID */
+    spanId?: string;
+    /** Tags for the span */
+    tags?: string[];
+    /** User ID for the span */
+    userId?: string;
+    /** Session ID for the span */
+    sessionId?: string;
+    /** Trace ID to group related spans together */
+    traceId?: string;
+    /** Parent span ID for chaining spans in a sequence */
+    parentSpanId?: string;
+    /** Timeout in milliseconds */
+    timeoutMs?: number;
+}
+interface EmbeddingVector {
+    /** The embedding values */
+    values: number[];
+    /** Index of this embedding in the input array */
+    index: number;
+}
+interface EmbeddingUsage {
+    /** Total tokens used for the embedding request */
+    totalTokens: number;
+}
+interface RunEmbeddingResult {
+    /** The generated embeddings */
+    embeddings: EmbeddingVector[];
+    /** Span ID for this request */
+    spanId: string;
+    /** Trace ID grouping related spans */
+    traceId: string;
+    /** Latency in milliseconds */
+    latencyMs: number;
+    /** Token usage */
+    usage: EmbeddingUsage;
+    /**
+     * Estimated cost in USD.
+     * Currently always `null` — cost is calculated server-side when the span is ingested.
+     */
+    cost: number | null;
+    /** The provider used */
+    provider: LLMProvider;
+    /** The model used */
+    model: string;
+}
 /** @deprecated Use SpanStatus instead */
 type TraceStatus = SpanStatus;
 /** @deprecated Use SpanListItem instead */
@@ -609,6 +679,11 @@ type SessionRunLocalInput = Omit<RunLocalInput, 'traceId' | 'parentSpanId'>;
  * as those are managed by the session.
  */
 type SessionRunResponsesInput = Omit<RunResponsesInput, 'traceId' | 'parentSpanId'>;
+/**
+ * Input for session.runEmbedding() - same as RunEmbeddingInput but without traceId/parentSpanId
+ * as those are managed by the session.
+ */
+type SessionRunEmbeddingInput = Omit<RunEmbeddingInput, 'traceId' | 'parentSpanId'>;
 /**
  * A session for grouping related spans together under a single trace.
  *
@@ -688,6 +763,11 @@ declare class TraciaSession {
     }): Promise<RunResponsesResult>;
     private runResponsesNonStreaming;
     private runResponsesStreaming;
+    /**
+     * Generate embeddings, automatically linking the span to this session.
+     * See Tracia.runEmbedding() for full documentation.
+     */
+    runEmbedding(input: SessionRunEmbeddingInput): Promise<RunEmbeddingResult>;
     private updateSessionState;
 }
 
@@ -811,6 +891,29 @@ declare class Tracia {
     private createResponsesStream;
     private createLocalStream;
     private combineAbortSignals;
+    /**
+     * Generate embeddings for text input(s) using an embedding model.
+     *
+     * @example Single text
+     * ```typescript
+     * const result = await tracia.runEmbedding({
+     *   model: 'text-embedding-3-small',
+     *   input: 'Hello world',
+     * })
+     * console.log(result.embeddings[0].values.length) // 1536
+     * ```
+     *
+     * @example Batch
+     * ```typescript
+     * const result = await tracia.runEmbedding({
+     *   model: 'text-embedding-3-small',
+     *   input: ['Hello', 'World'],
+     * })
+     * console.log(result.embeddings.length) // 2
+     * ```
+     */
+    runEmbedding(input: RunEmbeddingInput): Promise<RunEmbeddingResult>;
+    private validateRunEmbeddingInput;
     flush(): Promise<void>;
     /**
      * Create a new session for grouping related spans together under a single trace.
@@ -856,4 +959,4 @@ declare class Tracia {
     private getProviderApiKey;
 }
 
-export { type ContentPart, type CreatePromptOptions, type CreateSpanPayload, type CreateSpanResult, type CreateTracePayload, type CreateTraceResult, Eval, type EvaluateOptions, type EvaluateResult, type FinishReason, type JsonSchemaProperty, LLMProvider, type ListSpansOptions, type ListSpansResult, type ListTracesOptions, type ListTracesResult, type LocalPromptMessage, type LocalStream, type MessageRole, type Prompt, type PromptListItem, type PromptMessage, type ResponsesEvent, type ResponsesInputItem, type ResponsesOutputItem, type ResponsesStream, type RunLocalInput, type RunLocalResult, type RunOptions, type RunResponsesInput, type RunResponsesResult, type RunResult, type RunVariables, type SessionRunLocalInput, type SessionRunResponsesInput, type Span, type SpanListItem, type SpanStatus, type StreamResult, type TextPart, type TokenUsage, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolParameters, type Trace, type TraceListItem, type TraceStatus, Tracia, TraciaError, TraciaErrorCode, type TraciaOptions, TraciaSession, type UpdatePromptOptions };
+export { type ContentPart, type CreatePromptOptions, type CreateSpanPayload, type CreateSpanResult, type CreateTracePayload, type CreateTraceResult, type EmbeddingUsage, type EmbeddingVector, Eval, type EvaluateOptions, type EvaluateResult, type FinishReason, type JsonSchemaProperty, LLMProvider, type ListSpansOptions, type ListSpansResult, type ListTracesOptions, type ListTracesResult, type LocalPromptMessage, type LocalStream, type MessageRole, type Prompt, type PromptListItem, type PromptMessage, type ResponsesEvent, type ResponsesInputItem, type ResponsesOutputItem, type ResponsesStream, type RunEmbeddingInput, type RunEmbeddingResult, type RunLocalInput, type RunLocalResult, type RunOptions, type RunResponsesInput, type RunResponsesResult, type RunResult, type RunVariables, type SessionRunEmbeddingInput, type SessionRunLocalInput, type SessionRunResponsesInput, type Span, SpanKind, type SpanListItem, type SpanStatus, type StreamResult, type TextPart, type TokenUsage, type ToolCall, type ToolCallPart, type ToolChoice, type ToolDefinition, type ToolParameters, type Trace, type TraceListItem, type TraceStatus, Tracia, TraciaError, TraciaErrorCode, type TraciaOptions, TraciaSession, type UpdatePromptOptions };