@broberg/ai-sdk 0.62.0 → 0.62.2

package/dist/index.d.ts

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

package/dist/index.js

@@ -185,6 +185,9 @@ function classifyFailure(err) {
   return { errorCode: "unknown", errorKind: "other" };
 }
 
+// src/search/index.ts
+import { createHash } from "crypto";
+
 // src/search/types.ts
 var SearchKeyMissingError = class extends Error {
   constructor(provider, envVar) {
@@ -270,6 +273,15 @@ var sharedCapStore = /* @__PURE__ */ (() => {
     getSpent: (k) => m.get(k) ?? 0
   };
 })();
+var COOLDOWN_MS = 10 * 6e4;
+var COOLDOWN_STATUS = /* @__PURE__ */ new Set(["401", "402", "403"]);
+var coolingUntil = /* @__PURE__ */ new Map();
+function cooldownKey(provider, opts) {
+  const c = opts.credentials ?? {};
+  const env = opts.byok ? {} : process.env;
+  const secret = provider === "brave" ? c.braveApiKey ?? env.BRAVE_API_KEY : provider === "tavily" ? c.tavilyApiKey ?? env.TAVILY_API_KEY : `${c.cloudflareAccountId ?? env.CLOUDFLARE_ACCOUNT_ID}:${c.cloudflareApiToken ?? env.CLOUDFLARE_API_TOKEN}`;
+  return `${provider}:${createHash("sha256").update(String(secret)).digest("hex").slice(0, 16)}`;
+}
 var copenhagenDay = (d) => new Intl.DateTimeFormat("en-CA", { timeZone: "Europe/Copenhagen" }).format(d);
 async function callProvider(provider, req, opts) {
   const f = opts.fetch ?? fetch;
@@ -325,11 +337,16 @@ async function search(req, opts = {}) {
   const tenant = req.labels?.tenantId ?? "_";
   const day = copenhagenDay((opts.now ?? (() => /* @__PURE__ */ new Date()))());
   const capKey = JSON.stringify([tenant, day]);
+  const nowMs = () => (opts.now ?? (() => /* @__PURE__ */ new Date()))().getTime();
   let lastErr;
-  for (const provider of chain) {
+  for (const [i, provider] of chain.entries()) {
     const key = searchCacheKey(provider, req);
     const hit = cache?.get(key);
     if (hit) return { items: hit.items, meta: { ...hit.meta, cached: true, costUsd: 0, latencyMs: 0 } };
+    const coolKey = cooldownKey(provider, opts);
+    const until = coolingUntil.get(coolKey);
+    if (until !== void 0 && until <= nowMs()) coolingUntil.delete(coolKey);
+    else if (until !== void 0 && i < chain.length - 1) continue;
     const price = SEARCH_PRICE_USD[provider];
     const cap = opts.dailyCapUsd ?? Infinity;
     if (!await capStore.reserve(capKey, price, cap)) {
@@ -343,6 +360,7 @@ async function search(req, opts = {}) {
         items: dedupe(out.items),
         meta: { provider, latencyMs, cached: false, costUsd: SEARCH_PRICE_USD[provider], ...out.requestId ? { requestId: out.requestId } : {} }
       };
+      coolingUntil.delete(coolKey);
       cache?.set(key, result, ttl);
       try {
         await sink?.record(usageRow(provider, req, result.meta.costUsd, latencyMs));
@@ -352,6 +370,7 @@ async function search(req, opts = {}) {
     } catch (e) {
       lastErr = e;
       await capStore.release(capKey, price);
+      if (COOLDOWN_STATUS.has(classifyFailure(e).errorCode)) coolingUntil.set(coolKey, nowMs() + COOLDOWN_MS);
       if (sink?.recordFailure) {
         try {
           await sink.recordFailure({
@@ -3899,8 +3918,8 @@ var aiConfigSchema = z.object({
 import { randomUUID } from "crypto";
 
 // src/version.ts
-var VERSION = "0.62.0";
-var SDK_TAG = "@broberg/ai-sdk@0.62.0";
+var VERSION = "0.62.2";
+var SDK_TAG = "@broberg/ai-sdk@0.62.2";
 
 // src/cost/sinks/upmetrics.ts
 function isRetryableStatus(status) {