@broberg/ai-sdk 0.61.0 → 0.61.1

package/dist/index.d.ts

@@ -48,6 +48,8 @@ interface SearchCredentials {
     braveApiKey?: string;
     cloudflareAccountId?: string;
     cloudflareApiToken?: string;
+    /** AI Gateway the search runs through. Default "default" (env CLOUDFLARE_GATEWAY_ID). */
+    cloudflareGatewayId?: string;
     tavilyApiKey?: string;
 }
 interface SearchCache {
@@ -2332,8 +2334,8 @@ declare const falStubAdapter: ProviderAdapter;
  *  wires the live adapters. */
 declare const stubProviders: Record<string, ProviderAdapter>;
 
-declare const VERSION: "0.61.0";
-declare const SDK_TAG: "@broberg/ai-sdk@0.61.0";
+declare const VERSION: "0.61.1";
+declare const SDK_TAG: "@broberg/ai-sdk@0.61.1";
 
 /** Built-in defaults. Every entry is overridable via AiConfig.defaults or a
  *  per-call override.

package/dist/index.js

@@ -58,7 +58,7 @@ async function cloudflareSearch(req, creds, f) {
   const res = await f(`https://api.cloudflare.com/client/v4/accounts/${encodeURIComponent(creds.accountId)}/ai/websearch/`, {
     method: "POST",
     headers: { authorization: `Bearer ${creds.apiToken}`, "content-type": "application/json" },
-    body: JSON.stringify({ query: req.query, provider, limit: Math.min(req.limit ?? CLOUDFLARE_MAX_LIMIT, CLOUDFLARE_MAX_LIMIT) })
+    body: JSON.stringify({ query: req.query, provider, limit: Math.min(req.limit ?? CLOUDFLARE_MAX_LIMIT, CLOUDFLARE_MAX_LIMIT), options: { gateway: { id: creds.gatewayId } } })
   });
   if (!res.ok) throw Object.assign(new Error(`cloudflare search (${provider}): HTTP ${res.status} ${(await res.text()).slice(0, 200)}`), { status: res.status });
   const body = await res.json();
@@ -290,7 +290,8 @@ async function callProvider(provider, req, opts) {
     const apiToken = c.cloudflareApiToken ?? env.CLOUDFLARE_API_TOKEN;
     if (!accountId) throw new SearchKeyMissingError(provider, "CLOUDFLARE_ACCOUNT_ID");
     if (!apiToken) throw new SearchKeyMissingError(provider, "CLOUDFLARE_API_TOKEN");
-    return cloudflareSearch({ ...req, provider }, { accountId, apiToken }, f);
+    const gatewayId = c.cloudflareGatewayId ?? env.CLOUDFLARE_GATEWAY_ID ?? "default";
+    return cloudflareSearch({ ...req, provider }, { accountId, apiToken, gatewayId }, f);
   }
   throw new Error(`search: unknown provider "${String(provider)}"`);
 }
@@ -3898,8 +3899,8 @@ var aiConfigSchema = z.object({
 import { randomUUID } from "crypto";
 
 // src/version.ts
-var VERSION = "0.61.0";
-var SDK_TAG = "@broberg/ai-sdk@0.61.0";
+var VERSION = "0.61.1";
+var SDK_TAG = "@broberg/ai-sdk@0.61.1";
 
 // src/cost/sinks/upmetrics.ts
 function isRetryableStatus(status) {