@broberg/ai-sdk 0.23.0 → 0.24.0

package/dist/index.d.ts

@@ -2017,8 +2017,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.23.0";
-declare const SDK_TAG: "@broberg/ai-sdk@0.23.0";
+declare const VERSION: "0.24.0";
+declare const SDK_TAG: "@broberg/ai-sdk@0.24.0";
 
 /** Built-in defaults. Every entry is overridable via AiConfig.defaults or a
  *  per-call override.

package/dist/index.js

@@ -2571,6 +2571,78 @@ var aiConfigSchema = z.object({
   availability: availabilitySchema.optional()
 });
 
+// src/version.ts
+var VERSION = "0.24.0";
+var SDK_TAG = "@broberg/ai-sdk@0.24.0";
+
+// src/cost/sinks/upmetrics.ts
+function upmetricsSink(config) {
+  const doFetch = config.fetch ?? fetch;
+  const url = `${config.baseUrl.replace(/\/$/, "")}/api/agent`;
+  return {
+    async record(usage) {
+      try {
+        const startedAt = usage.ts || (/* @__PURE__ */ new Date()).toISOString();
+        const endedAt = new Date(
+          new Date(startedAt).getTime() + (usage.latencyMs || 0)
+        ).toISOString();
+        const agentKind = config.agentKind ?? (usage.capability === "embedding" ? "embedding" : "chatbot");
+        const body = {
+          mode: "record",
+          agent_kind: agentKind,
+          agent_name: config.agentName,
+          provider: usage.provider,
+          model: usage.model,
+          status: "success",
+          input_tokens: usage.inputTokens,
+          output_tokens: usage.outputTokens,
+          cache_read_tokens: usage.cacheReadTokens,
+          cache_creation_tokens: usage.cacheCreationTokens,
+          cost_usd: usage.costUsd,
+          duration_ms: usage.latencyMs,
+          started_at: startedAt,
+          ended_at: endedAt,
+          tags: {
+            // Consumer attribution labels (e.g. tenantId) ride in tags so no new
+            // top-level field risks the strict-shape ingest schema (F011). The
+            // SDK-owned keys win — a label can never clobber capability/transport/sdk.
+            ...usage.labels,
+            capability: usage.capability,
+            transport: usage.transport,
+            sdk: SDK_TAG
+          }
+        };
+        if (usage.tier !== void 0) body.tier = usage.tier;
+        if (usage.purpose !== void 0) body.purpose = usage.purpose;
+        if (usage.toolCalls) {
+          body.tool_calls = usage.toolCalls.map((t) => ({
+            name: t.name,
+            count: t.count,
+            error_count: t.errorCount ?? 0
+          }));
+        }
+        void config.complianceMode;
+        const res = await doFetch(url, {
+          method: "POST",
+          headers: {
+            "content-type": "application/json",
+            "X-Upmetrics-Key": config.apiKey
+          },
+          body: JSON.stringify(body)
+        });
+        if (!res.ok) {
+          const text = await res.text().catch(() => "");
+          config.onError?.(
+            new Error(`upmetricsSink: ingest returned ${res.status}: ${text.slice(0, 200)}`)
+          );
+        }
+      } catch (err) {
+        config.onError?.(err);
+      }
+    }
+  };
+}
+
 // src/client.ts
 var DEFAULT_IMAGE_SPEC = {
   provider: "fal",
@@ -2608,9 +2680,20 @@ var DEFAULT_MODERATION_SPEC = { provider: "mistral", model: "mistral-moderation-
 var DEFAULT_PODCAST_SPEC = { provider: "elevenlabs", model: "eleven_v3", transport: "http" };
 var DEFAULT_TTS_SPEC = { provider: "elevenlabs", model: "eleven_multilingual_v2", transport: "http" };
 var DEFAULT_BATCH_SPEC = { provider: "mistral", model: "mistral-small-latest", transport: "http" };
+function defaultCostSink() {
+  const apiKey = process.env.UPMETRICS_API_KEY;
+  if (!apiKey) return void 0;
+  return upmetricsSink({
+    baseUrl: process.env.UPMETRICS_BASE_URL ?? "https://upmetrics.org",
+    apiKey,
+    agentName: process.env.UPMETRICS_AGENT_NAME ?? process.env.npm_package_name ?? "unknown",
+    complianceMode: process.env.UPMETRICS_COMPLIANCE === "1"
+  });
+}
 function createAI(config = {}) {
   const cfg = aiConfigSchema.parse(config);
   const providers = cfg.providers ?? defaultProviders;
+  const costSink = cfg.costSink ?? defaultCostSink();
   const budget = cfg.budget ? new BudgetGuard(cfg.budget) : void 0;
   const estTokens = (s) => Math.ceil(s.length / 4);
   async function preflight(spec, estInTokens, estOutTokens) {
@@ -2639,9 +2722,9 @@ function createAI(config = {}) {
     return usage;
   }
   async function report(usage) {
-    if (!cfg.costSink) return;
+    if (!costSink) return;
     try {
-      await cfg.costSink.record(usage);
+      await costSink.record(usage);
     } catch {
     }
   }
@@ -3170,10 +3253,6 @@ var stubProviders = {
   fal: falStubAdapter
 };
 
-// src/version.ts
-var VERSION = "0.23.0";
-var SDK_TAG = "@broberg/ai-sdk@0.23.0";
-
 // src/availability/refresh.ts
 var NOT_REFRESHED = { refreshed: false, checked: 0, markedUnavailable: [] };
 var DEFAULT_TTL_MS = 60 * 60 * 1e3;
@@ -3263,74 +3342,6 @@ function multiSink(sinks) {
   };
 }
 
-// src/cost/sinks/upmetrics.ts
-function upmetricsSink(config) {
-  const doFetch = config.fetch ?? fetch;
-  const url = `${config.baseUrl.replace(/\/$/, "")}/api/agent`;
-  return {
-    async record(usage) {
-      try {
-        const startedAt = usage.ts || (/* @__PURE__ */ new Date()).toISOString();
-        const endedAt = new Date(
-          new Date(startedAt).getTime() + (usage.latencyMs || 0)
-        ).toISOString();
-        const agentKind = config.agentKind ?? (usage.capability === "embedding" ? "embedding" : "chatbot");
-        const body = {
-          mode: "record",
-          agent_kind: agentKind,
-          agent_name: config.agentName,
-          provider: usage.provider,
-          model: usage.model,
-          status: "success",
-          input_tokens: usage.inputTokens,
-          output_tokens: usage.outputTokens,
-          cache_read_tokens: usage.cacheReadTokens,
-          cache_creation_tokens: usage.cacheCreationTokens,
-          cost_usd: usage.costUsd,
-          duration_ms: usage.latencyMs,
-          started_at: startedAt,
-          ended_at: endedAt,
-          tags: {
-            // Consumer attribution labels (e.g. tenantId) ride in tags so no new
-            // top-level field risks the strict-shape ingest schema (F011). The
-            // SDK-owned keys win — a label can never clobber capability/transport/sdk.
-            ...usage.labels,
-            capability: usage.capability,
-            transport: usage.transport,
-            sdk: SDK_TAG
-          }
-        };
-        if (usage.tier !== void 0) body.tier = usage.tier;
-        if (usage.purpose !== void 0) body.purpose = usage.purpose;
-        if (usage.toolCalls) {
-          body.tool_calls = usage.toolCalls.map((t) => ({
-            name: t.name,
-            count: t.count,
-            error_count: t.errorCount ?? 0
-          }));
-        }
-        void config.complianceMode;
-        const res = await doFetch(url, {
-          method: "POST",
-          headers: {
-            "content-type": "application/json",
-            "X-Upmetrics-Key": config.apiKey
-          },
-          body: JSON.stringify(body)
-        });
-        if (!res.ok) {
-          const text = await res.text().catch(() => "");
-          config.onError?.(
-            new Error(`upmetricsSink: ingest returned ${res.status}: ${text.slice(0, 200)}`)
-          );
-        }
-      } catch (err) {
-        config.onError?.(err);
-      }
-    }
-  };
-}
-
 // src/cost/sinks/discord.ts
 function discordSink(config) {
   const doFetch = config.fetch ?? fetch;