@ai-sdk/gateway 3.0.179 → 3.0.180

package/dist/index.js

@@ -36,7 +36,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/gateway-provider.ts
-var import_provider_utils19 = require("@ai-sdk/provider-utils");
+var import_provider_utils20 = require("@ai-sdk/provider-utils");
 
 // src/errors/as-gateway-error.ts
 var import_provider = require("@ai-sdk/provider");
@@ -2145,6 +2145,210 @@ var perplexitySearchToolFactory = (0, import_provider_utils18.createProviderTool
 });
 var perplexitySearch = (config = {}) => perplexitySearchToolFactory(config);
 
+// src/tool/tako-search.ts
+var import_provider_utils19 = require("@ai-sdk/provider-utils");
+var import_zod4 = require("zod");
+var takoDataSourceInputSchema = import_zod4.z.object({
+  count: import_zod4.z.number().optional().describe("Maximum number of data cards to return (1-20)."),
+  include_contents: import_zod4.z.boolean().optional().describe("Inline underlying card data. This can add a data charge."),
+  mode: import_zod4.z.enum(["inline", "url"]).optional().describe(
+    "Requested data delivery mode. Search card data is always inline."
+  ),
+  content_format: import_zod4.z.enum(["card_json", "csv", "json_compact", "json_records"]).optional().describe("Serialization for inlined card data."),
+  max_rows: import_zod4.z.number().optional().describe("Maximum rows for inlined card data."),
+  node_ids: import_zod4.z.array(import_zod4.z.string()).optional().describe("Data Graph node IDs to prioritize. Maximum 20."),
+  strict: import_zod4.z.boolean().optional().describe(
+    "Only return cards matching node_ids. Requires a non-empty node_ids."
+  )
+});
+var takoWebSourceInputSchema = import_zod4.z.object({
+  count: import_zod4.z.number().optional().describe("Maximum number of web results to return (1-20)."),
+  include_contents: import_zod4.z.boolean().optional().describe("Inline extracted web page text. This can add a data charge."),
+  category: import_zod4.z.enum(["finance", "news", "sports"]).optional().describe("Optional web-result category filter."),
+  include_domains: import_zod4.z.array(import_zod4.z.string()).optional().describe("Only return results from these bare domains."),
+  exclude_domains: import_zod4.z.array(import_zod4.z.string()).optional().describe("Exclude results from these bare domains."),
+  snippet_max_chars: import_zod4.z.number().optional().describe("Maximum characters in each web-result snippet."),
+  highlights: import_zod4.z.boolean().optional().describe(
+    "Include highlighted passages in web results. Defaults to true in AI Gateway."
+  ),
+  article_content_max_chars: import_zod4.z.number().optional().describe(
+    "Maximum extracted characters per web page when including contents."
+  ),
+  published_after: import_zod4.z.string().optional().describe("Keep results published on or after this ISO date (YYYY-MM-DD)."),
+  published_before: import_zod4.z.string().optional().describe(
+    "Keep results published on or before this ISO date (YYYY-MM-DD)."
+  )
+});
+var takoSearchInputSchema = (0, import_provider_utils19.lazySchema)(
+  () => (0, import_provider_utils19.zodSchema)(
+    import_zod4.z.object({
+      query: import_zod4.z.string().describe(
+        'Natural-language search query. Include the entity, metric, and time period. Quote a phrase to force it to one entity, for example "Tesla":PRODUCT price.'
+      ),
+      effort: import_zod4.z.enum(["deep", "fast", "instant"]).optional().describe(
+        "Search effort. fast is the balanced default, instant favors cached results and low latency, and deep broadens retrieval with reranking at higher cost and latency."
+      ),
+      sources: import_zod4.z.object({
+        data: takoDataSourceInputSchema.optional(),
+        web: takoWebSourceInputSchema.optional()
+      }).optional().describe(
+        "Sources to search. Omit to search both curated data and the web. When provided, only keys present are searched."
+      ),
+      location: import_zod4.z.object({
+        latitude: import_zod4.z.number().describe("Latitude between -90 and 90."),
+        longitude: import_zod4.z.number().describe("Longitude between -180 and 180.")
+      }).optional().describe("End-user coordinates for localized results."),
+      country_code: import_zod4.z.string().optional().describe("Two-letter ISO 3166-1 country code, such as 'US'."),
+      locale: import_zod4.z.string().optional().describe("BCP-47 locale, such as 'en-US'."),
+      timezone: import_zod4.z.string().optional().describe("IANA timezone, such as 'America/New_York'."),
+      output_settings: import_zod4.z.object({
+        image_dark_mode: import_zod4.z.boolean().optional().describe("Render card preview images in dark mode."),
+        force_refresh: import_zod4.z.boolean().optional().describe(
+          "Instant-effort only. Request a refreshed instant result."
+        )
+      }).optional().describe("Controls card rendering in the search response."),
+      include_related: import_zod4.z.number().optional().describe("Maximum related search suggestions to include (1-20).")
+    })
+  )
+);
+var takoDatasetCellSchema = import_zod4.z.union([import_zod4.z.boolean(), import_zod4.z.number(), import_zod4.z.string()]).nullable();
+var takoResultContentSchema = import_zod4.z.object({
+  content_format: import_zod4.z.enum(["card_json", "csv", "json_compact", "json_records"]).nullish(),
+  cost: import_zod4.z.number().optional(),
+  data: import_zod4.z.string().nullish(),
+  records: import_zod4.z.array(import_zod4.z.record(import_zod4.z.string(), takoDatasetCellSchema)).nullish(),
+  dataset: import_zod4.z.object({
+    columns: import_zod4.z.array(
+      import_zod4.z.object({
+        name: import_zod4.z.string(),
+        type: import_zod4.z.enum(["boolean", "date", "datetime", "number", "string"]),
+        unit: import_zod4.z.string().nullish()
+      })
+    ),
+    rows: import_zod4.z.array(import_zod4.z.array(takoDatasetCellSchema)),
+    total_rows: import_zod4.z.number(),
+    truncated: import_zod4.z.boolean(),
+    ref: import_zod4.z.string(),
+    sources: import_zod4.z.array(
+      import_zod4.z.object({
+        name: import_zod4.z.string(),
+        index: import_zod4.z.enum(["data", "web"]).optional()
+      })
+    ),
+    provenance: import_zod4.z.enum(["query", "web_extraction"]).optional()
+  }).nullish(),
+  card_data: import_zod4.z.object({}).passthrough().nullish(),
+  card_data_schema: import_zod4.z.object({}).passthrough().nullish(),
+  url: import_zod4.z.string().nullish(),
+  expires_at: import_zod4.z.string().nullish(),
+  total_rows: import_zod4.z.number().nullish(),
+  truncated: import_zod4.z.boolean().optional(),
+  export_pricing: import_zod4.z.object({
+    baseline_usd: import_zod4.z.number(),
+    free_rows: import_zod4.z.number(),
+    max_rows_ceiling: import_zod4.z.number(),
+    row_cpm_usd: import_zod4.z.number()
+  }).nullish(),
+  manifest: import_zod4.z.array(
+    import_zod4.z.object({
+      dtype: import_zod4.z.enum(["boolean", "date", "datetime", "number", "string"]).nullish(),
+      entity: import_zod4.z.string().nullish(),
+      metric: import_zod4.z.string().nullish(),
+      name: import_zod4.z.string().nullish(),
+      unit: import_zod4.z.string().nullish()
+    })
+  ).nullish()
+}).passthrough();
+var takoCardSchema = import_zod4.z.object({
+  card_id: import_zod4.z.string().nullish(),
+  title: import_zod4.z.string().nullish(),
+  description: import_zod4.z.string().nullish(),
+  semantic_description: import_zod4.z.string().nullish(),
+  webpage_url: import_zod4.z.string().nullish(),
+  image_url: import_zod4.z.string().nullish(),
+  embed_url: import_zod4.z.string().nullish(),
+  sources: import_zod4.z.array(
+    import_zod4.z.object({
+      source_name: import_zod4.z.string().nullish(),
+      source_description: import_zod4.z.string().nullish(),
+      source_index: import_zod4.z.enum(["data", "web"]),
+      source_text: import_zod4.z.string().nullish(),
+      url: import_zod4.z.string().nullish()
+    })
+  ).nullish(),
+  methodologies: import_zod4.z.array(
+    import_zod4.z.object({
+      methodology_name: import_zod4.z.string().nullable(),
+      methodology_description: import_zod4.z.string().nullable()
+    })
+  ).nullish(),
+  source_indexes: import_zod4.z.array(import_zod4.z.enum(["data", "web"])).nullish(),
+  card_type: import_zod4.z.string().nullish(),
+  relevance: import_zod4.z.enum(["High", "Low", "Medium"]).nullish(),
+  content: takoResultContentSchema.nullish(),
+  exportable: import_zod4.z.boolean().optional(),
+  relevance_score: import_zod4.z.number().nullish(),
+  nodes: import_zod4.z.array(
+    import_zod4.z.object({
+      id: import_zod4.z.string(),
+      type: import_zod4.z.enum(["entity", "metric"]),
+      name: import_zod4.z.string(),
+      description: import_zod4.z.string().nullish()
+    })
+  ).nullish(),
+  metric_definitions: import_zod4.z.array(import_zod4.z.object({ name: import_zod4.z.string(), definition: import_zod4.z.string() })).nullish(),
+  data_freshness: import_zod4.z.object({
+    coverage_end: import_zod4.z.string().nullish(),
+    data_as_of: import_zod4.z.string().nullish(),
+    last_updated: import_zod4.z.string().nullish()
+  }).nullish()
+}).passthrough();
+var takoWebResultSchema = import_zod4.z.object({
+  title: import_zod4.z.string(),
+  url: import_zod4.z.string(),
+  snippet: import_zod4.z.string().nullish(),
+  source_name: import_zod4.z.string().nullish(),
+  publish_date: import_zod4.z.string().nullish(),
+  content: takoResultContentSchema.nullish(),
+  citation_number: import_zod4.z.number().nullish()
+}).passthrough();
+var takoSearchOutputSchema = (0, import_provider_utils19.lazySchema)(
+  () => (0, import_provider_utils19.zodSchema)(
+    import_zod4.z.union([
+      import_zod4.z.object({
+        request_id: import_zod4.z.string(),
+        cards: import_zod4.z.array(takoCardSchema).optional(),
+        web_results: import_zod4.z.array(takoWebResultSchema).optional(),
+        usage: import_zod4.z.object({
+          total_cost_usd: import_zod4.z.number(),
+          compute: import_zod4.z.object({ cost_usd: import_zod4.z.number() }).nullish(),
+          data: import_zod4.z.object({ cost_usd: import_zod4.z.number(), datasets: import_zod4.z.number() }).nullish()
+        }).nullish(),
+        related: import_zod4.z.array(import_zod4.z.object({}).passthrough()).nullish()
+      }).passthrough(),
+      import_zod4.z.object({
+        error: import_zod4.z.enum([
+          "api_error",
+          "configuration_error",
+          "execution_error",
+          "invalid_input",
+          "rate_limit",
+          "timeout",
+          "unknown_tool"
+        ]),
+        statusCode: import_zod4.z.number().optional(),
+        message: import_zod4.z.string()
+      })
+    ])
+  )
+);
+var takoSearchToolFactory = (0, import_provider_utils19.createProviderToolFactoryWithOutputSchema)({
+  id: "gateway.tako_search",
+  inputSchema: takoSearchInputSchema,
+  outputSchema: takoSearchOutputSchema
+});
+var takoSearch = (config = {}) => takoSearchToolFactory(config);
+
 // src/gateway-tools.ts
 var gatewayTools = {
   /**
@@ -2171,7 +2375,16 @@ var gatewayTools = {
    * Provides ranked search results with advanced filtering options including
    * domain, language, date range, and recency filters.
    */
-  perplexitySearch
+  perplexitySearch,
+  /**
+   * Search the web and Tako's curated knowledge graph in one call for
+   * token-efficient web excerpts and structured data results grounded in
+   * premium sources, each with an embed-ready visualization.
+   *
+   * Supports effort, per-source web and data controls, localization, and inline
+   * contents for agents that need to reason over underlying data.
+   */
+  takoSearch
 };
 
 // src/vercel-environment.ts
@@ -2183,7 +2396,7 @@ async function getVercelRequestId() {
 }
 
 // src/version.ts
-var VERSION = true ? "3.0.179" : "0.0.0-test";
+var VERSION = true ? "3.0.180" : "0.0.0-test";
 
 // src/gateway-provider.ts
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
@@ -2193,11 +2406,11 @@ function createGatewayProvider(options = {}) {
   let metadataCache = null;
   const cacheRefreshMillis = (_a11 = options.metadataCacheRefreshMillis) != null ? _a11 : 1e3 * 60 * 5;
   let lastFetchTime = 0;
-  const baseURL = (_b11 = (0, import_provider_utils19.withoutTrailingSlash)(options.baseURL)) != null ? _b11 : "https://ai-gateway.vercel.sh/v3/ai";
+  const baseURL = (_b11 = (0, import_provider_utils20.withoutTrailingSlash)(options.baseURL)) != null ? _b11 : "https://ai-gateway.vercel.sh/v3/ai";
   const getHeaders = async () => {
     try {
       const auth = await getGatewayAuthToken(options);
-      return (0, import_provider_utils19.withUserAgentSuffix)(
+      return (0, import_provider_utils20.withUserAgentSuffix)(
         {
           Authorization: `Bearer ${auth.token}`,
           "ai-gateway-protocol-version": AI_GATEWAY_PROTOCOL_VERSION,
@@ -2216,19 +2429,19 @@ function createGatewayProvider(options = {}) {
     }
   };
   const createO11yHeaders = () => {
-    const deploymentId = (0, import_provider_utils19.loadOptionalSetting)({
+    const deploymentId = (0, import_provider_utils20.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_DEPLOYMENT_ID"
     });
-    const environment = (0, import_provider_utils19.loadOptionalSetting)({
+    const environment = (0, import_provider_utils20.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_ENV"
     });
-    const region = (0, import_provider_utils19.loadOptionalSetting)({
+    const region = (0, import_provider_utils20.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_REGION"
     });
-    const projectId = (0, import_provider_utils19.loadOptionalSetting)({
+    const projectId = (0, import_provider_utils20.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_PROJECT_ID"
     });
@@ -2394,7 +2607,7 @@ function createGatewayProvider(options = {}) {
 }
 var gateway = createGatewayProvider();
 async function getGatewayAuthToken(options) {
-  const apiKey = (0, import_provider_utils19.loadOptionalSetting)({
+  const apiKey = (0, import_provider_utils20.loadOptionalSetting)({
     settingValue: options.apiKey,
     environmentVariableName: "AI_GATEWAY_API_KEY"
   });