@tool-bridge/server 0.1.0 → 0.3.0

package/dist/tavily-B2YHCNRD.js

@@ -0,0 +1,349 @@
+import {
+  createProviderPlugin,
+  guardedFetch,
+  requireApiKey,
+  upstreamError
+} from "./chunk-CSJEZHQO.js";
+import "./chunk-K7FVSFUX.js";
+import {
+  TBError,
+  external_exports
+} from "./chunk-JLTYA2B3.js";
+import "./chunk-MLKGABMK.js";
+
+// ../plugins/src/tavily/api.ts
+var SERVICE = "tavily";
+var API_BASE = "https://api.tavily.com";
+var REQUEST_TIMEOUT_MS = 3e4;
+function record(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value) ? value : void 0;
+}
+function text(value) {
+  if (typeof value !== "string") return void 0;
+  const trimmed = value.trim();
+  return trimmed === "" ? void 0 : trimmed;
+}
+function errorMessage(payload, status) {
+  if (typeof payload === "string" && payload.trim() !== "") return payload;
+  const body = record(payload);
+  return text(body?.detail) ?? text(body?.error) ?? text(body?.message) ?? `Tavily \u8FD4\u56DE HTTP ${status}`;
+}
+async function request(ctx, options) {
+  const headers = {
+    accept: "application/json",
+    authorization: `Bearer ${requireApiKey(ctx, SERVICE)}`
+  };
+  if (options.body !== void 0) headers["content-type"] = "application/json";
+  let response;
+  try {
+    response = await guardedFetch(`${API_BASE}${options.path}`, {
+      method: options.method,
+      headers,
+      signal: AbortSignal.timeout(REQUEST_TIMEOUT_MS),
+      ...options.body === void 0 ? {} : { body: JSON.stringify(options.body) }
+    });
+  } catch (error) {
+    if (error instanceof TBError) throw error;
+    if (error instanceof Error && error.name === "TimeoutError") {
+      throw upstreamError(504, `Tavily ${REQUEST_TIMEOUT_MS / 1e3}s \u5185\u6CA1\u6709\u8FD4\u56DE`);
+    }
+    throw upstreamError(502, error instanceof Error ? `Tavily \u8BF7\u6C42\u5931\u8D25:${error.message}` : "Tavily \u8BF7\u6C42\u5931\u8D25");
+  }
+  const raw = await response.text().catch(() => "");
+  let payload = null;
+  if (raw !== "") {
+    try {
+      payload = JSON.parse(raw);
+    } catch {
+      if (response.ok) throw upstreamError(502, "Tavily \u8FD4\u56DE\u4E86\u975E JSON \u54CD\u5E94");
+      payload = raw;
+    }
+  }
+  if (!response.ok) throw upstreamError(response.status, errorMessage(payload, response.status));
+  if (raw === "") return {};
+  const result = record(payload);
+  if (result === void 0) throw upstreamError(502, "Tavily \u54CD\u5E94\u4E0D\u662F\u5BF9\u8C61");
+  return result;
+}
+function search(input, ctx) {
+  return request(ctx, { method: "POST", path: "/search", body: input });
+}
+function extract(input, ctx) {
+  return request(ctx, { method: "POST", path: "/extract", body: input });
+}
+function map(input, ctx) {
+  return request(ctx, { method: "POST", path: "/map", body: input });
+}
+function crawl(input, ctx) {
+  return request(ctx, { method: "POST", path: "/crawl", body: input });
+}
+function createResearch(input, ctx) {
+  return request(ctx, { method: "POST", path: "/research", body: input });
+}
+function getResearch(input, ctx) {
+  return request(ctx, { method: "GET", path: `/research/${encodeURIComponent(input.request_id)}` });
+}
+function getUsage(_input, ctx) {
+  return request(ctx, { method: "GET", path: "/usage" });
+}
+
+// ../plugins/src/tavily/schema.ts
+var searchInput = external_exports.strictObject({
+  query: external_exports.string().min(1).describe("The search query to execute with Tavily."),
+  search_depth: external_exports.enum(["advanced", "basic", "fast", "ultra-fast"]).describe("Controls the latency-versus-relevance tradeoff for Tavily Search.").optional(),
+  chunks_per_source: external_exports.int().min(1).max(3).describe("The maximum number of chunks to return per source.").optional(),
+  max_results: external_exports.int().min(0).max(20).describe("The maximum number of search results to return.").optional(),
+  topic: external_exports.enum(["general", "news", "finance"]).describe("The search category used by Tavily.").optional(),
+  time_range: external_exports.enum(["day", "week", "month", "year", "d", "w", "m", "y"]).describe("The date range shortcut used to filter results.").optional(),
+  start_date: external_exports.iso.date().describe("Only return results after this YYYY-MM-DD date.").optional(),
+  end_date: external_exports.iso.date().describe("Only return results before this YYYY-MM-DD date.").optional(),
+  include_answer: external_exports.union([external_exports.boolean().describe("Whether to include an answer."), external_exports.enum(["basic", "advanced"]).describe("The answer detail level.")]).optional(),
+  include_raw_content: external_exports.union([external_exports.boolean().describe("Whether to include raw content."), external_exports.enum(["markdown", "text"]).describe("The raw content format.")]).optional(),
+  include_images: external_exports.boolean().describe("Whether to include top-level and per-result images.").optional(),
+  include_image_descriptions: external_exports.boolean().describe("Whether to include descriptions for returned images.").optional(),
+  include_favicon: external_exports.boolean().describe("Whether to include favicons for returned results.").optional(),
+  include_domains: external_exports.array(external_exports.string().min(1).describe("A domain name to include.")).min(1).describe("Domains that Tavily should include in the search results.").optional(),
+  exclude_domains: external_exports.array(external_exports.string().min(1).describe("A domain name to exclude.")).min(1).describe("Domains that Tavily should exclude from the search results.").optional(),
+  country: external_exports.string().describe("A country name used to boost results for general searches.").optional(),
+  auto_parameters: external_exports.boolean().describe("Whether Tavily should auto-configure search parameters.").optional(),
+  exact_match: external_exports.boolean().describe("Whether Tavily should require quoted exact phrases to match exactly.").optional(),
+  include_usage: external_exports.boolean().describe("Whether to include credit usage details in the response.").optional()
+}).describe("The input payload for a Tavily Search request.");
+var searchOutput = external_exports.looseObject({
+  query: external_exports.string().describe("The search query that was executed."),
+  results: external_exports.array(external_exports.looseObject({
+    title: external_exports.string().describe("The source title.").optional(),
+    url: external_exports.string().describe("The source URL.").optional(),
+    content: external_exports.string().describe("The extracted snippet or summary for the source.").optional(),
+    score: external_exports.number().describe("The relevance score of the source.").optional(),
+    raw_content: external_exports.string().describe("The cleaned page content when include_raw_content is enabled.").nullable().optional(),
+    favicon: external_exports.string().describe("The favicon URL for the result when include_favicon is enabled.").nullable().optional(),
+    images: external_exports.array(external_exports.looseObject({
+      url: external_exports.string().describe("The image URL.").optional(),
+      description: external_exports.string().describe("A short description for the image.").optional()
+    }).describe("An image item returned by Tavily.")).describe("Images extracted from this result.").optional(),
+    published_date: external_exports.string().describe("The published date for the result when Tavily can determine it.").optional()
+  }).describe("A single Tavily search result.")).describe("The ranked search results returned by Tavily."),
+  response_time: external_exports.union([external_exports.number().describe("The total response time reported by Tavily."), external_exports.string().describe("The total response time reported by Tavily.")]),
+  request_id: external_exports.string().describe("A unique request identifier for Tavily support and debugging."),
+  answer: external_exports.string().describe("A Tavily-generated answer for the query."),
+  images: external_exports.array(external_exports.looseObject({
+    url: external_exports.string().describe("The image URL.").optional(),
+    description: external_exports.string().describe("A short description for the image.").optional()
+  }).describe("An image item returned by Tavily.")).describe("Query-related images returned by Tavily."),
+  auto_parameters: external_exports.looseObject({}).describe("Auto-selected parameters returned by Tavily when enabled."),
+  usage: external_exports.looseObject({
+    credits: external_exports.number().describe("The number of API credits consumed by the request.").optional()
+  }).describe("Credit usage details returned by Tavily.")
+}).describe("The Tavily Search response payload.");
+var extractInput = external_exports.strictObject({
+  urls: external_exports.array(external_exports.url().describe("A source URL.")).min(1).describe("The URLs that Tavily should extract content from."),
+  query: external_exports.string().describe("An optional query used to rerank extracted chunks.").optional(),
+  chunks_per_source: external_exports.int().min(1).max(5).describe("The maximum number of chunks to return per source when query is provided.").optional(),
+  extract_depth: external_exports.enum(["basic", "advanced"]).describe("Controls whether Tavily uses basic or advanced extraction.").optional(),
+  include_images: external_exports.boolean().describe("Whether to include images found on each page.").optional(),
+  include_favicon: external_exports.boolean().describe("Whether to include the favicon URL for each page.").optional(),
+  format: external_exports.enum(["markdown", "text"]).describe("The format of the extracted page content.").optional(),
+  timeout: external_exports.number().min(1).max(60).describe("The extraction timeout in seconds.").optional(),
+  include_usage: external_exports.boolean().describe("Whether to include credit usage details in the response.").optional()
+}).describe("The input payload for a Tavily Extract request.");
+var extractOutput = external_exports.looseObject({
+  results: external_exports.array(external_exports.looseObject({
+    url: external_exports.string().describe("The processed source URL.").optional(),
+    raw_content: external_exports.string().describe("The extracted page content in the selected format.").optional(),
+    images: external_exports.array(external_exports.looseObject({
+      url: external_exports.string().describe("The image URL.").optional(),
+      description: external_exports.string().describe("A short description for the image.").optional()
+    }).describe("An image item returned by Tavily.")).describe("Images extracted from the source.").optional(),
+    favicon: external_exports.string().describe("The favicon URL for the source when include_favicon is enabled.").optional()
+  }).describe("An extracted Tavily result item.")).describe("The successful extraction results returned by Tavily."),
+  failed_results: external_exports.array(external_exports.looseObject({
+    url: external_exports.string().describe("The source URL that failed.").optional(),
+    error: external_exports.string().describe("The failure reason returned by Tavily.").optional()
+  }).describe("A failed Tavily extraction result.")).describe("URLs that Tavily could not extract successfully."),
+  response_time: external_exports.number().describe("The total response time reported by Tavily."),
+  request_id: external_exports.string().describe("A unique request identifier for Tavily support and debugging."),
+  usage: external_exports.looseObject({
+    credits: external_exports.number().describe("The number of API credits consumed by the request.").optional()
+  }).describe("Credit usage details returned by Tavily.")
+}).describe("The Tavily Extract response payload.");
+var mapInput = external_exports.strictObject({
+  url: external_exports.url().describe("The root URL that Tavily should map."),
+  instructions: external_exports.string().describe("Natural-language instructions that guide the mapping.").optional(),
+  max_depth: external_exports.int().min(1).max(5).describe("The maximum mapping depth.").optional(),
+  max_breadth: external_exports.int().min(1).max(500).describe("The maximum number of links to follow per level.").optional(),
+  limit: external_exports.int().min(1).describe("The maximum number of links Tavily should process.").optional(),
+  select_paths: external_exports.array(external_exports.string().min(1).describe("A path selection pattern.")).min(1).describe("Regex patterns used to include only matching URL paths.").optional(),
+  select_domains: external_exports.array(external_exports.string().min(1).describe("A domain selection pattern.")).min(1).describe("Regex patterns used to include only matching domains.").optional(),
+  exclude_paths: external_exports.array(external_exports.string().min(1).describe("A path exclusion pattern.")).min(1).describe("Regex patterns used to exclude matching URL paths.").optional(),
+  exclude_domains: external_exports.array(external_exports.string().min(1).describe("A domain exclusion pattern.")).min(1).describe("Regex patterns used to exclude matching domains.").optional(),
+  allow_external: external_exports.boolean().describe("Whether external domain links can appear in the results.").optional(),
+  timeout: external_exports.number().min(10).max(150).describe("The mapping timeout in seconds.").optional(),
+  include_usage: external_exports.boolean().describe("Whether to include credit usage details in the response.").optional()
+}).describe("The input payload for a Tavily Map request.");
+var mapOutput = external_exports.looseObject({
+  base_url: external_exports.string().describe("The base URL that Tavily mapped."),
+  results: external_exports.array(external_exports.string().min(1)).describe("The URLs discovered during the mapping operation."),
+  response_time: external_exports.number().describe("The total response time reported by Tavily."),
+  request_id: external_exports.string().describe("A unique request identifier for Tavily support and debugging."),
+  usage: external_exports.looseObject({
+    credits: external_exports.number().describe("The number of API credits consumed by the request.").optional()
+  }).describe("Credit usage details returned by Tavily.")
+}).describe("The Tavily Map response payload.");
+var crawlInput = external_exports.strictObject({
+  url: external_exports.url().describe("The root URL that Tavily should crawl."),
+  instructions: external_exports.string().describe("Natural-language instructions that guide the crawl.").optional(),
+  max_depth: external_exports.int().min(1).max(5).describe("The maximum crawl depth.").optional(),
+  max_breadth: external_exports.int().min(1).max(500).describe("The maximum number of links to follow per crawl level.").optional(),
+  limit: external_exports.int().min(1).describe("The maximum number of links Tavily should process.").optional(),
+  select_paths: external_exports.array(external_exports.string().min(1).describe("A path selection pattern.")).min(1).describe("Regex patterns used to include only matching URL paths.").optional(),
+  select_domains: external_exports.array(external_exports.string().min(1).describe("A domain selection pattern.")).min(1).describe("Regex patterns used to include only matching domains.").optional(),
+  exclude_paths: external_exports.array(external_exports.string().min(1).describe("A path exclusion pattern.")).min(1).describe("Regex patterns used to exclude matching URL paths.").optional(),
+  exclude_domains: external_exports.array(external_exports.string().min(1).describe("A domain exclusion pattern.")).min(1).describe("Regex patterns used to exclude matching domains.").optional(),
+  allow_external: external_exports.boolean().describe("Whether external domain links can appear in the results.").optional(),
+  include_images: external_exports.boolean().describe("Whether to include images in crawled results.").optional(),
+  extract_depth: external_exports.enum(["basic", "advanced"]).describe("Controls whether Tavily uses basic or advanced extraction.").optional(),
+  format: external_exports.enum(["markdown", "text"]).describe("The format of the extracted page content.").optional(),
+  include_favicon: external_exports.boolean().describe("Whether to include a favicon URL for each crawled result.").optional(),
+  timeout: external_exports.number().min(10).max(150).describe("The crawl timeout in seconds.").optional(),
+  include_usage: external_exports.boolean().describe("Whether to include credit usage details in the response.").optional()
+}).describe("The input payload for a Tavily Crawl request.");
+var crawlOutput = external_exports.looseObject({
+  base_url: external_exports.string().describe("The base URL that Tavily crawled."),
+  results: external_exports.array(external_exports.looseObject({
+    url: external_exports.string().describe("The processed source URL.").optional(),
+    raw_content: external_exports.string().describe("The extracted page content in the selected format.").optional(),
+    images: external_exports.array(external_exports.looseObject({
+      url: external_exports.string().describe("The image URL.").optional(),
+      description: external_exports.string().describe("A short description for the image.").optional()
+    }).describe("An image item returned by Tavily.")).describe("Images extracted from the source.").optional(),
+    favicon: external_exports.string().describe("The favicon URL for the source when include_favicon is enabled.").optional()
+  }).describe("An extracted Tavily result item.")).describe("The extracted results returned by Tavily Crawl."),
+  response_time: external_exports.number().describe("The total response time reported by Tavily."),
+  request_id: external_exports.string().describe("A unique request identifier for Tavily support and debugging."),
+  usage: external_exports.looseObject({
+    credits: external_exports.number().describe("The number of API credits consumed by the request.").optional()
+  }).describe("Credit usage details returned by Tavily.")
+}).describe("The Tavily Crawl response payload.");
+var createResearchInput = external_exports.strictObject({
+  input: external_exports.string().min(1).describe("The research task or question to investigate."),
+  model: external_exports.enum(["mini", "pro", "auto"]).describe("Research model to use.").optional(),
+  stream: external_exports.literal(false).describe("Must be false or omitted. Tavily SSE streaming is not supported.").optional(),
+  output_schema: external_exports.looseObject({}).describe("JSON Schema for structured research output.").optional(),
+  citation_format: external_exports.enum(["numbered", "mla", "apa", "chicago"]).describe("The format for citations in the research report.").optional(),
+  include_domains: external_exports.array(external_exports.string().min(1).describe("A domain to prioritize.")).max(20).describe("Soft source preference domains.").optional(),
+  exclude_domains: external_exports.array(external_exports.string().min(1).describe("A domain to exclude.")).max(20).describe("Hard source blocklist domains.").optional(),
+  output_length: external_exports.enum(["short", "standard", "long"]).describe("The target research response length.").optional(),
+  files: external_exports.array(external_exports.strictObject({
+    name: external_exports.string().min(1).describe("The file name, including extension."),
+    data: external_exports.string().min(1).describe("The base64-encoded file contents."),
+    type: external_exports.literal("base64").describe("The encoded file content type.")
+  }).describe("A base64-encoded .txt, .md, or .json file attached to a Tavily Research request.")).min(1).max(5).describe("Up to 5 .txt, .md, or .json files to use as additional research sources.").optional()
+}).describe("The input payload for creating a Tavily Research task.");
+var createResearchOutput = external_exports.looseObject({
+  request_id: external_exports.string().min(1).describe("A unique identifier for the research task.").optional(),
+  status: external_exports.enum(["pending", "in_progress", "completed", "failed"]).describe("The current research task status.").optional(),
+  input: external_exports.string().describe("The research task or question investigated.").optional(),
+  model: external_exports.enum(["mini", "pro", "auto"]).describe("The model used by the research agent.").optional(),
+  content: external_exports.unknown().describe("The final research report content.").optional(),
+  sources: external_exports.array(external_exports.looseObject({}).describe("A source used in a Tavily Research result.")).describe("Sources used in the research.").optional(),
+  response_time: external_exports.number().describe("The response time reported by Tavily.").optional()
+}).describe("A Tavily Research task response.");
+var getResearchInput = external_exports.strictObject({
+  request_id: external_exports.string().min(1).describe("The unique identifier of the research task.")
+}).describe("The input payload for retrieving a Tavily Research task.");
+var getResearchOutput = external_exports.looseObject({
+  request_id: external_exports.string().min(1).describe("A unique identifier for the research task.").optional(),
+  status: external_exports.enum(["pending", "in_progress", "completed", "failed"]).describe("The current research task status.").optional(),
+  input: external_exports.string().describe("The research task or question investigated.").optional(),
+  model: external_exports.enum(["mini", "pro", "auto"]).describe("The model used by the research agent.").optional(),
+  content: external_exports.unknown().describe("The final research report content.").optional(),
+  sources: external_exports.array(external_exports.looseObject({}).describe("A source used in a Tavily Research result.")).describe("Sources used in the research.").optional(),
+  response_time: external_exports.number().describe("The response time reported by Tavily.").optional()
+}).describe("A Tavily Research task response.");
+var getUsageInput = external_exports.strictObject({}).describe("The input payload for a Tavily usage request.");
+var getUsageOutput = external_exports.strictObject({
+  key: external_exports.looseObject({
+    usage: external_exports.number().describe("The total usage for the API key.").optional(),
+    limit: external_exports.number().describe("The usage limit for the API key.").optional(),
+    search_usage: external_exports.number().describe("Search credits consumed by the API key.").optional(),
+    extract_usage: external_exports.number().describe("Extract credits consumed by the API key.").optional(),
+    crawl_usage: external_exports.number().describe("Crawl credits consumed by the API key.").optional(),
+    map_usage: external_exports.number().describe("Map credits consumed by the API key.").optional(),
+    research_usage: external_exports.number().describe("Research credits consumed by the API key.").optional()
+  }).describe("Usage details for the specific API key."),
+  account: external_exports.looseObject({
+    current_plan: external_exports.string().describe("The current account plan name.").optional(),
+    plan_usage: external_exports.number().describe("The total plan usage for the account.").optional(),
+    plan_limit: external_exports.number().describe("The plan limit for the account.").optional(),
+    paygo_usage: external_exports.number().describe("The pay-as-you-go usage for the account.").optional(),
+    paygo_limit: external_exports.number().describe("The pay-as-you-go limit for the account.").optional()
+  }).describe("Plan and usage information for the account.")
+}).describe("The Tavily usage response payload.");
+var tavilyActions = {
+  search: {
+    description: "Execute a Tavily Search query and return ranked source results.",
+    effect: "write",
+    inputSchema: searchInput,
+    outputSchema: external_exports.toJSONSchema(searchOutput, { io: "output", unrepresentable: "any" })
+  },
+  extract: {
+    description: "Extract structured page content from one or more URLs with Tavily.",
+    effect: "write",
+    inputSchema: extractInput,
+    outputSchema: external_exports.toJSONSchema(extractOutput, { io: "output", unrepresentable: "any" })
+  },
+  map: {
+    description: "Discover URLs from a website with Tavily Map.",
+    effect: "write",
+    inputSchema: mapInput,
+    outputSchema: external_exports.toJSONSchema(mapOutput, { io: "output", unrepresentable: "any" })
+  },
+  crawl: {
+    description: "Crawl a website and extract content from discovered pages with Tavily.",
+    effect: "write",
+    inputSchema: crawlInput,
+    outputSchema: external_exports.toJSONSchema(crawlOutput, { io: "output", unrepresentable: "any" })
+  },
+  create_research: {
+    description: "Start an asynchronous Tavily Research task and return a request ID for polling.",
+    effect: "write",
+    inputSchema: createResearchInput,
+    outputSchema: external_exports.toJSONSchema(createResearchOutput, { io: "output", unrepresentable: "any" })
+  },
+  get_research: {
+    description: "Get the current status and result for a Tavily Research task.",
+    effect: "read",
+    inputSchema: getResearchInput,
+    outputSchema: external_exports.toJSONSchema(getResearchOutput, { io: "output", unrepresentable: "any" })
+  },
+  get_usage: {
+    description: "Get API key and account usage details from Tavily.",
+    effect: "read",
+    inputSchema: getUsageInput,
+    outputSchema: external_exports.toJSONSchema(getUsageOutput, { io: "output", unrepresentable: "any" })
+  }
+};
+
+// ../plugins/src/tavily/index.ts
+function createTavilyPlugin() {
+  return createProviderPlugin({
+    description: "Tavily",
+    actions: tavilyActions,
+    credentialProbe: "get_usage",
+    handlers: {
+      search,
+      extract,
+      map,
+      crawl,
+      create_research: createResearch,
+      get_research: getResearch,
+      get_usage: getUsage
+    }
+  });
+}
+var tavily_default = createTavilyPlugin();
+export {
+  createTavilyPlugin,
+  tavily_default as default
+};