tracia 0.4.0 → 0.4.2

package/dist/index.d.mts

@@ -64,7 +64,8 @@ declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
     GOOGLE = "google",
-    AMAZON_BEDROCK = "amazon_bedrock"
+    AMAZON_BEDROCK = "amazon_bedrock",
+    VOYAGE = "voyage"
 }
 declare enum SpanKind {
     ROOT = "ROOT",
@@ -242,7 +243,15 @@ interface ToolCallPart {
     name: string;
     arguments: Record<string, unknown>;
 }
-type ContentPart = TextPart | ToolCallPart;
+/**
+ * Image part in user messages.
+ */
+interface ImagePart {
+    type: 'image';
+    image: string | Buffer | ArrayBuffer | Uint8Array;
+    mimeType?: string;
+}
+type ContentPart = TextPart | ToolCallPart | ImagePart;
 /**
  * Message format for LLM conversations.
  *
@@ -959,4 +968,4 @@ declare class Tracia {
     private getProviderApiKey;
 }
 
-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 };
+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 ImagePart, 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,8 @@ declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
     GOOGLE = "google",
-    AMAZON_BEDROCK = "amazon_bedrock"
+    AMAZON_BEDROCK = "amazon_bedrock",
+    VOYAGE = "voyage"
 }
 declare enum SpanKind {
     ROOT = "ROOT",
@@ -242,7 +243,15 @@ interface ToolCallPart {
     name: string;
     arguments: Record<string, unknown>;
 }
-type ContentPart = TextPart | ToolCallPart;
+/**
+ * Image part in user messages.
+ */
+interface ImagePart {
+    type: 'image';
+    image: string | Buffer | ArrayBuffer | Uint8Array;
+    mimeType?: string;
+}
+type ContentPart = TextPart | ToolCallPart | ImagePart;
 /**
  * Message format for LLM conversations.
  *
@@ -959,4 +968,4 @@ declare class Tracia {
     private getProviderApiKey;
 }
 
-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 };
+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 ImagePart, 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

@@ -74,6 +74,7 @@ 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) => {
@@ -84,7 +85,7 @@ var SpanKind = /* @__PURE__ */ ((SpanKind2) => {
 })(SpanKind || {});
 
 // src/client.ts
-var SDK_VERSION = "0.4.0";
+var SDK_VERSION = "0.4.2";
 var DEFAULT_TIMEOUT_MS = 12e4;
 function mapApiErrorCodeToTraciaErrorCode(apiCode) {
   const codeMap = {
@@ -358,7 +359,14 @@ var MODEL_TO_PROVIDER = {
   "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 */
+  "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];
@@ -493,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.`
@@ -521,6 +532,11 @@ async function getLanguageModel(provider, model, apiKey) {
       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 */,
@@ -558,10 +574,44 @@ async function getEmbeddingModel(provider, model, apiKey) {
       );
   }
 }
+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 embeddingModel = await getEmbeddingModel(provider, options.model, options.apiKey);
   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 = {};
@@ -635,6 +685,22 @@ function convertMessages(messages) {
     if (typeof msg.content === "string") {
       return { role, content: msg.content };
     }
+    if (role === "user") {
+      return {
+        role,
+        content: msg.content.map((part) => {
+          if (part.type === "image") {
+            const imagePart = part;
+            return {
+              type: "image",
+              image: imagePart.image,
+              ...imagePart.mimeType && { mimeType: imagePart.mimeType }
+            };
+          }
+          return part;
+        })
+      };
+    }
     return {
       role,
       content: msg.content.map((b) => b.type === "text" ? b.text : "").join("")
@@ -1175,7 +1241,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")) {