tracia 0.3.9 → 0.3.12

package/dist/index.d.mts

@@ -8,9 +8,17 @@ interface RunVariables {
 }
 interface RunOptions {
     model?: string;
+    /** Run a specific prompt version instead of the latest */
+    version?: number;
     tags?: string[];
     userId?: string;
     sessionId?: string;
+    /** Trace ID to group related spans together */
+    traceId?: string;
+    /** Parent span ID for chaining spans in a sequence */
+    parentSpanId?: string;
+    /** Full conversation messages for multi-turn (skips template rendering) */
+    messages?: LocalPromptMessage[];
 }
 interface TokenUsage {
     inputTokens: number;
@@ -33,6 +41,8 @@ interface RunResult {
     toolCalls?: ToolCall[];
     /** Parsed JSON when the prompt has an output schema configured */
     structuredOutput?: Record<string, unknown>;
+    /** Full conversation messages for multi-turn continuation */
+    messages?: LocalPromptMessage[];
 }
 declare enum TraciaErrorCode {
     UNAUTHORIZED = "UNAUTHORIZED",
@@ -53,7 +63,8 @@ declare enum TraciaErrorCode {
 declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
-    GOOGLE = "google"
+    GOOGLE = "google",
+    AMAZON_BEDROCK = "amazon_bedrock"
 }
 interface ApiSuccessResponse {
     text: string;
@@ -66,6 +77,7 @@ interface ApiSuccessResponse {
     finishReason?: FinishReason;
     toolCalls?: ToolCall[];
     structuredOutput?: Record<string, unknown>;
+    messages?: LocalPromptMessage[];
 }
 type MessageRole = 'system' | 'developer' | 'user' | 'assistant' | 'tool';
 interface PromptMessage {
@@ -112,7 +124,7 @@ type SpanStatus = 'SUCCESS' | 'ERROR';
 interface SpanListItem {
     id: string;
     spanId: string;
-    traceId: string;
+    traceId: string | null;
     parentSpanId: string | null;
     promptSlug: string;
     model: string;
@@ -127,7 +139,7 @@ interface SpanListItem {
 interface Span {
     id: string;
     spanId: string;
-    traceId: string;
+    traceId: string | null;
     parentSpanId: string | null;
     promptSlug: string;
     promptVersion: number;

package/dist/index.d.ts

@@ -8,9 +8,17 @@ interface RunVariables {
 }
 interface RunOptions {
     model?: string;
+    /** Run a specific prompt version instead of the latest */
+    version?: number;
     tags?: string[];
     userId?: string;
     sessionId?: string;
+    /** Trace ID to group related spans together */
+    traceId?: string;
+    /** Parent span ID for chaining spans in a sequence */
+    parentSpanId?: string;
+    /** Full conversation messages for multi-turn (skips template rendering) */
+    messages?: LocalPromptMessage[];
 }
 interface TokenUsage {
     inputTokens: number;
@@ -33,6 +41,8 @@ interface RunResult {
     toolCalls?: ToolCall[];
     /** Parsed JSON when the prompt has an output schema configured */
     structuredOutput?: Record<string, unknown>;
+    /** Full conversation messages for multi-turn continuation */
+    messages?: LocalPromptMessage[];
 }
 declare enum TraciaErrorCode {
     UNAUTHORIZED = "UNAUTHORIZED",
@@ -53,7 +63,8 @@ declare enum TraciaErrorCode {
 declare enum LLMProvider {
     OPENAI = "openai",
     ANTHROPIC = "anthropic",
-    GOOGLE = "google"
+    GOOGLE = "google",
+    AMAZON_BEDROCK = "amazon_bedrock"
 }
 interface ApiSuccessResponse {
     text: string;
@@ -66,6 +77,7 @@ interface ApiSuccessResponse {
     finishReason?: FinishReason;
     toolCalls?: ToolCall[];
     structuredOutput?: Record<string, unknown>;
+    messages?: LocalPromptMessage[];
 }
 type MessageRole = 'system' | 'developer' | 'user' | 'assistant' | 'tool';
 interface PromptMessage {
@@ -112,7 +124,7 @@ type SpanStatus = 'SUCCESS' | 'ERROR';
 interface SpanListItem {
     id: string;
     spanId: string;
-    traceId: string;
+    traceId: string | null;
     parentSpanId: string | null;
     promptSlug: string;
     model: string;
@@ -127,7 +139,7 @@ interface SpanListItem {
 interface Span {
     id: string;
     spanId: string;
-    traceId: string;
+    traceId: string | null;
     parentSpanId: string | null;
     promptSlug: string;
     promptVersion: number;

package/dist/index.js

@@ -72,11 +72,12 @@ var LLMProvider = /* @__PURE__ */ ((LLMProvider2) => {
   LLMProvider2["OPENAI"] = "openai";
   LLMProvider2["ANTHROPIC"] = "anthropic";
   LLMProvider2["GOOGLE"] = "google";
+  LLMProvider2["AMAZON_BEDROCK"] = "amazon_bedrock";
   return LLMProvider2;
 })(LLMProvider || {});
 
 // src/client.ts
-var SDK_VERSION = "0.3.9";
+var SDK_VERSION = "0.3.12";
 var DEFAULT_TIMEOUT_MS = 12e4;
 function mapApiErrorCodeToTraciaErrorCode(apiCode) {
   const codeMap = {
@@ -201,6 +202,9 @@ var Prompts = class {
     if (options?.model) {
       requestBody.model = options.model;
     }
+    if (options?.version != null) {
+      requestBody.version = options.version;
+    }
     if (options?.tags && options.tags.length > 0) {
       requestBody.tags = options.tags;
     }
@@ -210,6 +214,15 @@ var Prompts = class {
     if (options?.sessionId) {
       requestBody.sessionId = options.sessionId;
     }
+    if (options?.traceId) {
+      requestBody.traceId = options.traceId;
+    }
+    if (options?.parentSpanId) {
+      requestBody.parentSpanId = options.parentSpanId;
+    }
+    if (options?.messages && options.messages.length > 0) {
+      requestBody.messages = options.messages;
+    }
     const response = await this.client.post(
       `/api/v1/prompts/${encodeURIComponent(slug)}/run`,
       requestBody
@@ -224,7 +237,8 @@ var Prompts = class {
       cost: response.cost,
       finishReason: response.finishReason,
       toolCalls: response.toolCalls,
-      structuredOutput: response.structuredOutput
+      structuredOutput: response.structuredOutput,
+      messages: response.messages
     };
   }
 };
@@ -318,7 +332,17 @@ var MODEL_TO_PROVIDER = {
   "gemini-2.5-flash-preview-09-2025": "google" /* GOOGLE */,
   "gemini-2.5-pro": "google" /* GOOGLE */,
   "gemini-3-pro-preview": "google" /* GOOGLE */,
-  "gemini-3-flash-preview": "google" /* GOOGLE */
+  "gemini-3-flash-preview": "google" /* GOOGLE */,
+  // Amazon Bedrock
+  "anthropic.claude-haiku-4-5-20251001-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  "anthropic.claude-sonnet-4-20250514-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  "anthropic.claude-sonnet-4-5-20250929-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  "anthropic.claude-opus-4-5-20251101-v1:0": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  "anthropic.claude-opus-4-6-v1": "amazon_bedrock" /* AMAZON_BEDROCK */,
+  "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 */
 };
 function getProviderForModel(modelId) {
   return MODEL_TO_PROVIDER[modelId];
@@ -329,6 +353,7 @@ var aiSdk = null;
 var openaiProvider = null;
 var anthropicProvider = null;
 var googleProvider = null;
+var bedrockProvider = null;
 async function loadAISdk() {
   if (aiSdk) return aiSdk;
   try {
@@ -377,6 +402,18 @@ async function loadGoogleProvider() {
     );
   }
 }
+async function loadBedrockProvider() {
+  if (bedrockProvider) return bedrockProvider;
+  try {
+    bedrockProvider = await import("@ai-sdk/amazon-bedrock");
+    return bedrockProvider;
+  } catch {
+    throw new TraciaError(
+      "MISSING_PROVIDER_SDK" /* MISSING_PROVIDER_SDK */,
+      "Amazon Bedrock provider not installed. Install it with: npm install @ai-sdk/amazon-bedrock"
+    );
+  }
+}
 function combineAbortSignals(userSignal, timeoutMs) {
   if (!timeoutMs && !userSignal) return void 0;
   if (timeoutMs && !userSignal) return AbortSignal.timeout(timeoutMs);
@@ -398,10 +435,39 @@ function combineAbortSignals(userSignal, timeoutMs) {
 function sanitizeErrorMessage(message) {
   return message.replace(/\b(sk-|tr_|key-|api[_-]?key[=:\s]+)[a-zA-Z0-9_-]{10,}\b/gi, "[REDACTED]").replace(/Bearer\s+[a-zA-Z0-9_.-]+/gi, "Bearer [REDACTED]").replace(/Basic\s+[a-zA-Z0-9+/=]{20,}/gi, "Basic [REDACTED]").replace(/(authorization[=:\s]+)[^\s,}]+/gi, "$1[REDACTED]");
 }
+var BEDROCK_VENDOR_PREFIXES = ["anthropic.", "amazon.", "meta.", "mistral.", "cohere.", "deepseek."];
+var KNOWN_REGION_PREFIXES = ["us", "eu", "ap", "sa", "ca", "me", "af"];
+function getBedrockRegionPrefix(region) {
+  return region.split("-")[0];
+}
+function applyBedrockRegionPrefix(model, region) {
+  const prefix = getBedrockRegionPrefix(region);
+  const firstDot = model.indexOf(".");
+  if (firstDot === -1) return `${prefix}.${model}`;
+  const beforeDot = model.substring(0, firstDot);
+  if (KNOWN_REGION_PREFIXES.includes(beforeDot)) {
+    return `${prefix}.${model.substring(firstDot + 1)}`;
+  }
+  return `${prefix}.${model}`;
+}
+function isBedrockModel(model) {
+  if (BEDROCK_VENDOR_PREFIXES.some((prefix) => model.startsWith(prefix))) {
+    return true;
+  }
+  const firstDot = model.indexOf(".");
+  if (firstDot > 0 && KNOWN_REGION_PREFIXES.includes(model.substring(0, firstDot))) {
+    const afterRegion = model.substring(firstDot + 1);
+    return BEDROCK_VENDOR_PREFIXES.some((prefix) => afterRegion.startsWith(prefix));
+  }
+  return false;
+}
 function resolveProvider(model, explicitProvider) {
   if (explicitProvider) return explicitProvider;
   const fromRegistry = getProviderForModel(model);
   if (fromRegistry) return fromRegistry;
+  if (isBedrockModel(model)) {
+    return "amazon_bedrock" /* AMAZON_BEDROCK */;
+  }
   if (model.startsWith("gpt-") || model.startsWith("o1") || model.startsWith("o3") || model.startsWith("o4")) {
     return "openai" /* OPENAI */;
   }
@@ -433,6 +499,12 @@ async function getLanguageModel(provider, model, apiKey) {
       const google = createGoogleGenerativeAI({ apiKey });
       return google(model);
     }
+    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 });
+      return bedrock(applyBedrockRegionPrefix(model, region));
+    }
     default:
       throw new TraciaError(
         "UNSUPPORTED_MODEL" /* UNSUPPORTED_MODEL */,
@@ -1001,7 +1073,8 @@ var SPAN_STATUS_ERROR = "ERROR";
 var ENV_VAR_MAP = {
   ["openai" /* OPENAI */]: "OPENAI_API_KEY",
   ["anthropic" /* ANTHROPIC */]: "ANTHROPIC_API_KEY",
-  ["google" /* GOOGLE */]: "GOOGLE_API_KEY"
+  ["google" /* GOOGLE */]: "GOOGLE_API_KEY",
+  ["amazon_bedrock" /* AMAZON_BEDROCK */]: "BEDROCK_API_KEY"
 };
 function convertResponsesItemToMessage(item) {
   if ("role" in item && (item.role === "developer" || item.role === "user")) {
@@ -1676,6 +1749,7 @@ var Tracia = class {
     const envVar = ENV_VAR_MAP[provider];
     const key = process.env[envVar];
     if (!key) {
+      if (provider === "amazon_bedrock" /* AMAZON_BEDROCK */) return "";
       throw new TraciaError(
         "MISSING_PROVIDER_API_KEY" /* MISSING_PROVIDER_API_KEY */,
         `Missing API key for ${provider}. Set the ${envVar} environment variable or provide providerApiKey in options.`