tracia 0.4.0 → 0.4.2

package/dist/index.mjs

@@ -32,6 +32,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) => {
@@ -42,7 +43,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 = {
@@ -316,7 +317,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];
@@ -451,6 +459,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.`
@@ -479,6 +490,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 */,
@@ -516,10 +532,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 = {};
@@ -593,6 +643,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("")
@@ -1133,7 +1199,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")) {