tracia 0.3.12 → 0.4.1

package/dist/index.d.mts

@@ -64,7 +64,13 @@ declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
     GOOGLE = "google",
-    AMAZON_BEDROCK = "amazon_bedrock"
+    AMAZON_BEDROCK = "amazon_bedrock",
+    VOYAGE = "voyage"
+}
+declare enum SpanKind {
+    ROOT = "ROOT",
+    LLM = "LLM",
+    EMBEDDING = "EMBEDDING"
 }
 interface ApiSuccessResponse {
     text: string;
@@ -341,7 +347,8 @@ interface CreateSpanPayload {
     model: string;
     provider: LLMProvider;
     input: {
-        messages: LocalPromptMessage[];
+        messages?: LocalPromptMessage[];
+        text?: string | string[];
     };
     variables: Record<string, string> | null;
     output: string | null;
@@ -363,6 +370,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;
@@ -561,6 +570,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 */
@@ -612,6 +680,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.
  *
@@ -691,6 +764,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;
 }
 
@@ -814,6 +892,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.
@@ -859,4 +960,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

@@ -64,7 +64,13 @@ declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
     GOOGLE = "google",
-    AMAZON_BEDROCK = "amazon_bedrock"
+    AMAZON_BEDROCK = "amazon_bedrock",
+    VOYAGE = "voyage"
+}
+declare enum SpanKind {
+    ROOT = "ROOT",
+    LLM = "LLM",
+    EMBEDDING = "EMBEDDING"
 }
 interface ApiSuccessResponse {
     text: string;
@@ -341,7 +347,8 @@ interface CreateSpanPayload {
     model: string;
     provider: LLMProvider;
     input: {
-        messages: LocalPromptMessage[];
+        messages?: LocalPromptMessage[];
+        text?: string | string[];
     };
     variables: Record<string, string> | null;
     output: string | null;
@@ -363,6 +370,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;
@@ -561,6 +570,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 */
@@ -612,6 +680,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.
  *
@@ -691,6 +764,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;
 }
 
@@ -814,6 +892,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.
@@ -859,4 +960,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.js

@@ -32,6 +32,7 @@ var index_exports = {};
 __export(index_exports, {
   Eval: () => Eval,
   LLMProvider: () => LLMProvider,
+  SpanKind: () => SpanKind,
   Tracia: () => Tracia,
   TraciaError: () => TraciaError,
   TraciaErrorCode: () => TraciaErrorCode,
@@ -73,11 +74,18 @@ var LLMProvider = /* @__PURE__ */ ((LLMProvider2) => {
   LLMProvider2["ANTHROPIC"] = "anthropic";
   LLMProvider2["GOOGLE"] = "google";
   LLMProvider2["AMAZON_BEDROCK"] = "amazon_bedrock";
+  LLMProvider2["VOYAGE"] = "voyage";
   return LLMProvider2;
 })(LLMProvider || {});
+var SpanKind = /* @__PURE__ */ ((SpanKind2) => {
+  SpanKind2["ROOT"] = "ROOT";
+  SpanKind2["LLM"] = "LLM";
+  SpanKind2["EMBEDDING"] = "EMBEDDING";
+  return SpanKind2;
+})(SpanKind || {});
 
 // src/client.ts
-var SDK_VERSION = "0.3.12";
+var SDK_VERSION = "0.4.1";
 var DEFAULT_TIMEOUT_MS = 12e4;
 function mapApiErrorCodeToTraciaErrorCode(apiCode) {
   const codeMap = {
@@ -342,7 +350,23 @@ var MODEL_TO_PROVIDER = {
   "amazon.nova-micro-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
   "amazon.nova-lite-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
   "amazon.nova-pro-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
-  "mistral.pixtral-large-2502-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */
+  "mistral.pixtral-large-2502-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  // OpenAI - Embedding models
+  "text-embedding-3-small": "openai" /* OPENAI */,
+  "text-embedding-3-large": "openai" /* OPENAI */,
+  "text-embedding-ada-002": "openai" /* OPENAI */,
+  // Google - Embedding models
+  "text-embedding-004": "google" /* GOOGLE */,
+  // Amazon Bedrock - Embedding models
+  "amazon.titan-embed-text-v2:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  "cohere.embed-english-v3": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  // Voyage - Embedding models
+  "voyage-3": "voyage" /* VOYAGE */,
+  "voyage-3-large": "voyage" /* VOYAGE */,
+  "voyage-3-lite": "voyage" /* VOYAGE */,
+  "voyage-code-3": "voyage" /* VOYAGE */,
+  "voyage-finance-2": "voyage" /* VOYAGE */,
+  "voyage-law-2": "voyage" /* VOYAGE */
 };
 function getProviderForModel(modelId) {
   return MODEL_TO_PROVIDER[modelId];
@@ -477,6 +501,9 @@ function resolveProvider(model, explicitProvider) {
   if (model.startsWith("gemini-")) {
     return "google" /* GOOGLE */;
   }
+  if (model.startsWith("voyage-")) {
+    return "voyage" /* VOYAGE */;
+  }
   throw new TraciaError(
     "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
     `Cannot determine provider for model: ${model}. Specify provider explicitly.`
@@ -502,9 +529,14 @@ async function getLanguageModel(provider, model, apiKey) {
     case "amazon_bedrock" /* AMAZON_BEDROCK */: {
       const { createAmazonBedrock } = await loadBedrockProvider();
       const region = process.env.AWS_REGION ?? "eu-central-1";
-      const bedrock = apiKey ? createAmazonBedrock({ apiKey, region }) : createAmazonBedrock({ region });
+      const bedrock = createAmazonBedrock({ region });
       return bedrock(applyBedrockRegionPrefix(model, region));
     }
+    case "voyage" /* VOYAGE */:
+      throw new TraciaError(
+        "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
+        "Voyage is an embedding-only provider. Use runEmbedding() instead of runLocal()."
+      );
     default:
       throw new TraciaError(
         "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
@@ -512,6 +544,108 @@ async function getLanguageModel(provider, model, apiKey) {
       );
   }
 }
+async function getEmbeddingModel(provider, model, apiKey) {
+  switch (provider) {
+    case "openai" /* OPENAI */: {
+      const { createOpenAI } = await loadOpenAIProvider();
+      const openai = createOpenAI({ apiKey });
+      return openai.textEmbeddingModel(model);
+    }
+    case "google" /* GOOGLE */: {
+      const { createGoogleGenerativeAI } = await loadGoogleProvider();
+      const google = createGoogleGenerativeAI({ apiKey });
+      return google.textEmbeddingModel(model);
+    }
+    case "amazon_bedrock" /* AMAZON_BEDROCK */: {
+      const { createAmazonBedrock } = await loadBedrockProvider();
+      const region = process.env.AWS_REGION ?? "eu-central-1";
+      const bedrock = createAmazonBedrock({ region });
+      return bedrock.textEmbeddingModel(applyBedrockRegionPrefix(model, region));
+    }
+    case "anthropic" /* ANTHROPIC */:
+      throw new TraciaError(
+        "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
+        "Anthropic does not offer embedding models. Use OpenAI, Google, or Amazon Bedrock instead."
+      );
+    default:
+      throw new TraciaError(
+        "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
+        `Unsupported provider for embeddings: ${provider}`
+      );
+  }
+}
+async function callVoyageEmbedding(input, model, apiKey, timeoutMs) {
+  const response = await fetch("https://api.voyageai.com/v1/embeddings", {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${apiKey}`
+    },
+    body: JSON.stringify({ input, model }),
+    signal: timeoutMs ? AbortSignal.timeout(timeoutMs) : void 0
+  });
+  if (!response.ok) {
+    const body = await response.text();
+    throw new TraciaError(
+      "PROVIDER_ERROR" /* PROVIDER_ERROR */,
+      `Voyage AI API error ${response.status}: ${sanitizeErrorMessage(body)}`
+    );
+  }
+  const data = await response.json();
+  const embeddings = data.data.sort((a, b) => a.index - b.index).map((item) => ({ values: item.embedding, index: item.index }));
+  return {
+    embeddings,
+    totalTokens: data.usage?.total_tokens ?? 0,
+    provider: "voyage" /* VOYAGE */
+  };
+}
+async function embedText(options) {
+  const provider = resolveProvider(options.model, options.provider);
+  const inputs = Array.isArray(options.input) ? options.input : [options.input];
+  if (provider === "voyage" /* VOYAGE */) {
+    if (options.dimensions) {
+      throw new TraciaError(
+        "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
+        "Voyage AI does not support custom embedding dimensions"
+      );
+    }
+    return callVoyageEmbedding(inputs, options.model, options.apiKey, options.timeoutMs);
+  }
+  const embeddingModel = await getEmbeddingModel(provider, options.model, options.apiKey);
+  try {
+    const { embedMany } = await loadAISdk();
+    const providerOptions = {};
+    if (options.dimensions) {
+      if (provider === "openai" /* OPENAI */) {
+        providerOptions.openai = { dimensions: options.dimensions };
+      } else if (provider === "google" /* GOOGLE */) {
+        providerOptions.google = { outputDimensionality: options.dimensions };
+      }
+    }
+    const result = await embedMany({
+      model: embeddingModel,
+      values: inputs,
+      abortSignal: options.timeoutMs ? AbortSignal.timeout(options.timeoutMs) : void 0,
+      ...Object.keys(providerOptions).length > 0 && { providerOptions }
+    });
+    const embeddings = result.embeddings.map((values, index) => ({
+      values,
+      index
+    }));
+    return {
+      embeddings,
+      totalTokens: result.usage?.tokens ?? 0,
+      provider
+    };
+  } catch (error) {
+    if (error instanceof TraciaError) throw error;
+    const rawMessage = error instanceof Error ? error.message : String(error);
+    throw new TraciaError(
+      "PROVIDER_ERROR" /* PROVIDER_ERROR */,
+      `${provider} embedding error: ${sanitizeErrorMessage(rawMessage)}`
+    );
+  }
+}
 function convertMessages(messages) {
   return messages.map((msg) => {
     if (msg.role === "tool") {
@@ -548,9 +682,12 @@ function convertMessages(messages) {
       };
     }
     const role = msg.role === "developer" ? "system" : msg.role;
+    if (typeof msg.content === "string") {
+      return { role, content: msg.content };
+    }
     return {
       role,
-      content: typeof msg.content === "string" ? msg.content : msg.content.map((b) => b.type === "text" ? b.text : "").join("")
+      content: msg.content.map((b) => b.type === "text" ? b.text : "").join("")
     };
   });
 }
@@ -957,6 +1094,20 @@ var TraciaSession = class {
       abort: () => responsesStream2.abort()
     };
   }
+  /**
+   * Generate embeddings, automatically linking the span to this session.
+   * See Tracia.runEmbedding() for full documentation.
+   */
+  async runEmbedding(input) {
+    const inputWithSession = {
+      ...input,
+      traceId: this.traceId ?? void 0,
+      parentSpanId: this.lastSpanId ?? void 0
+    };
+    const result = await this.tracia.runEmbedding(inputWithSession);
+    this.updateSessionState(result.spanId, result.traceId);
+    return result;
+  }
   updateSessionState(spanId, traceId) {
     if (!spanId) return;
     if (!this.traceId && traceId) {
@@ -1074,7 +1225,8 @@ var ENV_VAR_MAP = {
   ["openai" /* OPENAI */]: "OPENAI_API_KEY",
   ["anthropic" /* ANTHROPIC */]: "ANTHROPIC_API_KEY",
   ["google" /* GOOGLE */]: "GOOGLE_API_KEY",
-  ["amazon_bedrock" /* AMAZON_BEDROCK */]: "BEDROCK_API_KEY"
+  ["amazon_bedrock" /* AMAZON_BEDROCK */]: "BEDROCK_API_KEY",
+  ["voyage" /* VOYAGE */]: "VOYAGE_API_KEY"
 };
 function convertResponsesItemToMessage(item) {
   if ("role" in item && (item.role === "developer" || item.role === "user")) {
@@ -1596,6 +1748,128 @@ var Tracia = class {
     signal2.addEventListener("abort", onAbort, { once: true });
     return controller.signal;
   }
+  /**
+   * 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
+   * ```
+   */
+  async runEmbedding(input) {
+    this.validateRunEmbeddingInput(input);
+    let spanId = "";
+    let traceId = "";
+    if (input.sendTrace !== false) {
+      if (input.spanId && !isValidSpanIdFormat(input.spanId)) {
+        throw new TraciaError(
+          "INVALID_REQUEST" /* INVALID_REQUEST */,
+          `Invalid span ID format. Must match: sp_ + 16 hex characters (e.g., sp_1234567890abcdef)`
+        );
+      }
+      spanId = input.spanId || generateSpanId();
+      traceId = input.traceId || generateTraceId();
+    }
+    const provider = resolveProvider(input.model, input.provider);
+    const apiKey = this.getProviderApiKey(provider, input.providerApiKey);
+    const startTime = Date.now();
+    let embeddingResult = null;
+    let errorMessage = null;
+    try {
+      embeddingResult = await embedText({
+        model: input.model,
+        input: input.input,
+        apiKey,
+        provider: input.provider,
+        dimensions: input.dimensions,
+        timeoutMs: input.timeoutMs
+      });
+    } catch (error) {
+      if (error instanceof TraciaError) {
+        errorMessage = error.message;
+      } else {
+        errorMessage = error instanceof Error ? error.message : String(error);
+      }
+    }
+    const latencyMs = Date.now() - startTime;
+    const inputTexts = Array.isArray(input.input) ? input.input : [input.input];
+    if (spanId) {
+      const embeddingCount = embeddingResult?.embeddings.length ?? 0;
+      const embeddingDimensions = embeddingResult?.embeddings[0]?.values.length ?? 0;
+      this.scheduleSpanCreation(spanId, {
+        spanId,
+        model: input.model,
+        provider: embeddingResult?.provider ?? provider,
+        input: { text: inputTexts },
+        variables: null,
+        output: embeddingResult ? JSON.stringify({ dimensions: embeddingDimensions, count: embeddingCount }) : null,
+        status: errorMessage ? SPAN_STATUS_ERROR : SPAN_STATUS_SUCCESS,
+        error: errorMessage,
+        latencyMs,
+        inputTokens: embeddingResult?.totalTokens ?? 0,
+        outputTokens: 0,
+        totalTokens: embeddingResult?.totalTokens ?? 0,
+        tags: input.tags,
+        userId: input.userId,
+        sessionId: input.sessionId,
+        traceId,
+        parentSpanId: input.parentSpanId,
+        spanKind: "EMBEDDING" /* EMBEDDING */
+      });
+    }
+    if (errorMessage || !embeddingResult) {
+      throw new TraciaError("PROVIDER_ERROR" /* PROVIDER_ERROR */, errorMessage ?? "Embedding call returned no result");
+    }
+    return {
+      embeddings: embeddingResult.embeddings,
+      spanId,
+      traceId,
+      latencyMs,
+      usage: { totalTokens: embeddingResult.totalTokens },
+      cost: null,
+      provider: embeddingResult.provider,
+      model: input.model
+    };
+  }
+  validateRunEmbeddingInput(input) {
+    if (!input.model || input.model.trim() === "") {
+      throw new TraciaError(
+        "INVALID_REQUEST" /* INVALID_REQUEST */,
+        "model is required and cannot be empty"
+      );
+    }
+    if (!input.input || Array.isArray(input.input) && input.input.length === 0) {
+      throw new TraciaError(
+        "INVALID_REQUEST" /* INVALID_REQUEST */,
+        "input is required and cannot be empty"
+      );
+    }
+    if (typeof input.input === "string" && input.input.trim() === "") {
+      throw new TraciaError(
+        "INVALID_REQUEST" /* INVALID_REQUEST */,
+        "input text cannot be empty"
+      );
+    }
+    if (Array.isArray(input.input) && input.input.some((text) => text.trim() === "")) {
+      throw new TraciaError(
+        "INVALID_REQUEST" /* INVALID_REQUEST */,
+        "input array cannot contain empty strings"
+      );
+    }
+  }
   async flush() {
     await Promise.all(this.pendingSpans.values());
   }
@@ -1762,6 +2036,7 @@ var Tracia = class {
 0 && (module.exports = {
   Eval,
   LLMProvider,
+  SpanKind,
   Tracia,
   TraciaError,
   TraciaErrorCode,