tracia 0.4.0 → 0.4.1

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",

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",

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.1";
 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 = {};
@@ -1175,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")) {