@nimbus-dev/connectors 0.2.1 → 0.2.3

package/connectors/dagster/src/tools.ts

@@ -0,0 +1,187 @@
+import { z } from "zod";
+import { requiredBaseUrl, requiredEnv } from "../../../shared/env-json-api.ts";
+import { postGraphql } from "../../../shared/graphql-json.ts";
+import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
+import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
+import type { ZodToolRegistrar } from "../../../shared/run-read-only-mcp-connector.ts";
+import { asRecord } from "../../../shared/search-filter.ts";
+import { filterDagsterJobs } from "./search-filter.ts";
+
+function apiToken(): string {
+  // Required for Dagster Cloud; self-hosted OSS may use a placeholder. The
+  // gateway injects it regardless to keep spawn wiring uniform.
+  return requiredEnv("DAGSTER_API_TOKEN");
+}
+
+const JOBS_QUERY = `
+query NimbusJobs {
+  repositoriesOrError {
+    __typename
+    ... on RepositoryConnection {
+      nodes {
+        name
+        location { name }
+        pipelines { id name description isJob tags { key value } }
+      }
+    }
+    ... on PythonError { message }
+  }
+}
+`;
+
+async function dagsterGraphql<T>(query: string): Promise<T> {
+  return postGraphql<T>({
+    url: `${requiredBaseUrl("DAGSTER_BASE_URL")}/graphql`,
+    label: "Dagster",
+    headers: { "Dagster-Cloud-Api-Token": apiToken() },
+    query,
+  });
+}
+
+interface FlatJob {
+  name: string;
+  repository: string;
+  location: string | null;
+  id: string | null;
+  description: string | null;
+  isJob: boolean;
+  tags: Array<{ key: string; value: string }>;
+}
+
+function str(row: Record<string, unknown>, key: string): string | null {
+  const v = row[key];
+  return typeof v === "string" && v !== "" ? v : null;
+}
+
+function tagPairs(raw: unknown): Array<{ key: string; value: string }> {
+  if (!Array.isArray(raw)) {
+    return [];
+  }
+  const out: Array<{ key: string; value: string }> = [];
+  for (const t of raw) {
+    const tr = asRecord(t);
+    if (tr === undefined) {
+      continue;
+    }
+    const key = str(tr, "key");
+    if (key === null) {
+      continue;
+    }
+    out.push({ key, value: str(tr, "value") ?? "" });
+  }
+  return out;
+}
+
+function pipelinesOf(node: unknown): {
+  repository: string;
+  location: string | null;
+  rows: unknown[];
+} {
+  const repo = asRecord(node);
+  if (repo === undefined) {
+    return { repository: "", location: null, rows: [] };
+  }
+  const locationRow = asRecord(repo["location"]);
+  const pipelines = repo["pipelines"];
+  return {
+    repository: str(repo, "name") ?? "",
+    location: locationRow === undefined ? null : str(locationRow, "name"),
+    rows: Array.isArray(pipelines) ? pipelines : [],
+  };
+}
+
+function pipelineToJob(p: unknown, repository: string, location: string | null): FlatJob | null {
+  const pr = asRecord(p);
+  if (pr === undefined) {
+    return null;
+  }
+  const name = str(pr, "name");
+  if (name === null) {
+    return null;
+  }
+  return {
+    name,
+    repository,
+    location,
+    id: str(pr, "id"),
+    description: str(pr, "description"),
+    isJob: pr["isJob"] === true,
+    tags: tagPairs(pr["tags"]),
+  };
+}
+
+function flattenJobs(data: unknown): FlatJob[] {
+  const root = asRecord(data) ?? {};
+  const repos = asRecord(root["repositoriesOrError"]) ?? {};
+  if (repos["__typename"] === "PythonError") {
+    const message = str(repos, "message") ?? "unknown error";
+    throw new Error(`Dagster repositoriesOrError PythonError: ${message}`);
+  }
+  const nodes = repos["nodes"];
+  if (!Array.isArray(nodes)) {
+    return [];
+  }
+  const out: FlatJob[] = [];
+  for (const node of nodes) {
+    const { repository, location, rows } = pipelinesOf(node);
+    for (const p of rows) {
+      const job = pipelineToJob(p, repository, location);
+      if (job !== null) {
+        out.push(job);
+      }
+    }
+  }
+  return out;
+}
+
+async function listJobs(): Promise<FlatJob[]> {
+  return flattenJobs(await dagsterGraphql<unknown>(JOBS_QUERY));
+}
+
+/** Tool names exposed by this connector — for contract/introspection tests. */
+export const DAGSTER_TOOL_NAMES = ["dagster_get", "dagster_list", "dagster_search"] as const;
+
+export function registerDagsterTools(reg: ZodToolRegistrar): void {
+  reg(
+    "dagster_list",
+    "List Dagster jobs. Walks the repositories catalog via a single GraphQL query (repositoriesOrError → nodes[].pipelines[]) against POST <base_url>/graphql and flattens to jobs, each carrying its repository and code location. Returns an { items: [...] } envelope.",
+    z.object({
+      limit: z.number().int().min(1).max(2000).optional(),
+    }),
+    async (p) => {
+      const jobs = await listJobs();
+      const limited = p.limit === undefined ? jobs : jobs.slice(0, p.limit);
+      return jsonResult({ items: limited });
+    },
+  );
+
+  reg(
+    "dagster_get",
+    "Fetch one Dagster job by its stable id. The id is either the full location:repository:jobName triple or just the bare jobName (the first match wins when only a name is given). Throws when no job matches.",
+    z.object({
+      id: z.string().min(1),
+    }),
+    async (p) => {
+      const jobs = await listJobs();
+      const match = jobs.find((j) => {
+        const triple = `${j.location ?? "_"}:${j.repository}:${j.name}`;
+        return triple === p.id || j.name === p.id;
+      });
+      if (match === undefined) {
+        throw new Error(`Dagster: job ${p.id} not found`);
+      }
+      return jsonResult(match);
+    },
+  );
+
+  reg(
+    "dagster_search",
+    "Substring search across Dagster jobs. Matches the query (case-insensitive) against the job name, repository, code location, description, and tag keys/values. Returns a { matches: [...] } envelope.",
+    searchToolInputSchema(2000),
+    async (p) => {
+      const jobs = await listJobs();
+      const matches = filterDagsterJobs(jobs, { query: p.query, limit: p.limit });
+      return jsonResult({ matches });
+    },
+  );
+}

package/connectors/databricks/src/search-filter.ts

@@ -1,16 +1,13 @@
 import {
   asRecord,
   makeQueryFilter,
+  nestedString,
   type SearchMatchOptions,
+  stringField,
 } from "../../../shared/search-filter.ts";
 
 export type DatabricksSearchMatchOptions = SearchMatchOptions;
 
-function stringAt(row: Record<string, unknown>, key: string): string {
-  const v = row[key];
-  return typeof v === "string" ? v : "";
-}
-
 function numberAt(row: Record<string, unknown>, key: string): string {
   const v = row[key];
   return typeof v === "number" && Number.isFinite(v) ? String(v) : "";
@@ -21,8 +18,11 @@ function fieldsOf(item: unknown): readonly string[] | null {
   if (row === undefined) {
     return null;
   }
-  const settings = asRecord(row["settings"]) ?? {};
-  return [stringAt(settings, "name"), stringAt(row, "creator_user_name"), numberAt(row, "job_id")];
+  return [
+    nestedString(row, ["settings", "name"]),
+    stringField(row, "creator_user_name"),
+    numberAt(row, "job_id"),
+  ];
 }
 
 export const filterDatabricksJobs = makeQueryFilter(fieldsOf);

package/connectors/databricks/src/server.ts

@@ -1,81 +1,6 @@
-import { z } from "zod";
-import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
-import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
 import { runReadOnlyMcpConnector } from "../../../shared/run-read-only-mcp-connector.ts";
-import { filterDatabricksJobs } from "./search-filter.ts";
-
-function trimTrailingSlash(s: string): string {
-  return s.endsWith("/") ? s.slice(0, -1) : s;
-}
-
-function apiBase(): string {
-  const v = process.env["DATABRICKS_HOST"]?.trim();
-  if (v === undefined || v === "") {
-    throw new Error("DATABRICKS_HOST is not set");
-  }
-  return trimTrailingSlash(v);
-}
-
-function authHeader(): Record<string, string> {
-  const k = process.env["DATABRICKS_TOKEN"]?.trim();
-  if (k === undefined || k === "") {
-    throw new Error("DATABRICKS_TOKEN is not set");
-  }
-  return { Authorization: `Bearer ${k}`, Accept: "application/json" };
-}
-
-async function dbGet(path: string): Promise<unknown> {
-  const res = await fetch(`${apiBase()}${path}`, { headers: authHeader() });
-  const text = await res.text();
-  if (!res.ok) {
-    throw new Error(`Databricks ${String(res.status)}: ${text.slice(0, 400)}`);
-  }
-  return JSON.parse(text) as unknown;
-}
-
-function jobsFrom(root: unknown): unknown[] {
-  const jobs = (root as { jobs?: unknown } | null)?.jobs;
-  return Array.isArray(jobs) ? jobs : [];
-}
+import { registerDatabricksTools } from "./tools.ts";
 
 await runReadOnlyMcpConnector("nimbus-databricks", (reg) => {
-  reg(
-    "databricks_list",
-    "List Databricks jobs (`GET /api/2.1/jobs/list`). Returns a single page; `limit` (1..100, default 25) caps the page size. Returns the raw `{ jobs, has_more, next_page_token }` envelope.",
-    z.object({
-      limit: z.number().int().min(1).max(100).optional(),
-    }),
-    async (p) => {
-      const limit = p.limit ?? 25;
-      const params = new URLSearchParams({ limit: String(limit) });
-      return jsonResult(await dbGet(`/api/2.1/jobs/list?${params.toString()}`));
-    },
-  );
-
-  reg(
-    "databricks_get",
-    "Fetch one Databricks job by numeric `jobId` (`GET /api/2.1/jobs/get?job_id={jobId}`). Throws when no job with that id exists.",
-    z.object({
-      jobId: z.number().int(),
-    }),
-    async (p) => {
-      const params = new URLSearchParams({ job_id: String(p.jobId) });
-      return jsonResult(await dbGet(`/api/2.1/jobs/get?${params.toString()}`));
-    },
-  );
-
-  reg(
-    "databricks_search",
-    "Substring search across Databricks jobs. Matches the query (case-insensitive) against the job's `settings.name`, `creator_user_name`, and `job_id`. Returns a `{ matches: [...] }` envelope.",
-    searchToolInputSchema(100),
-    async (p) => {
-      const params = new URLSearchParams({ limit: "100" });
-      const root = await dbGet(`/api/2.1/jobs/list?${params.toString()}`);
-      const matches = filterDatabricksJobs(jobsFrom(root), {
-        query: p.query,
-        limit: p.limit,
-      });
-      return jsonResult({ matches });
-    },
-  );
+  registerDatabricksTools(reg);
 });

package/connectors/databricks/src/tools.ts

@@ -0,0 +1,66 @@
+import { z } from "zod";
+import { createJsonGetter, envAuthHeaders, requiredBaseUrl } from "../../../shared/env-json-api.ts";
+import { searchToolInputSchema } from "../../../shared/mcp-search-tool.ts";
+import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
+import type { ZodToolRegistrar } from "../../../shared/run-read-only-mcp-connector.ts";
+import { filterDatabricksJobs } from "./search-filter.ts";
+
+const dbGet = createJsonGetter({
+  base: () => requiredBaseUrl("DATABRICKS_HOST"),
+  label: "Databricks",
+  headers: envAuthHeaders({ env: "DATABRICKS_TOKEN" }),
+});
+
+function jobsFrom(root: unknown): unknown[] {
+  const jobs = (root as { jobs?: unknown } | null)?.jobs;
+  return Array.isArray(jobs) ? jobs : [];
+}
+
+/** Tool names exposed by this connector — for contract/introspection tests. */
+export const DATABRICKS_TOOL_NAMES = [
+  "databricks_list",
+  "databricks_get",
+  "databricks_search",
+] as const;
+
+export function registerDatabricksTools(reg: ZodToolRegistrar): void {
+  reg(
+    "databricks_list",
+    "List Databricks jobs (`GET /api/2.1/jobs/list`). Returns a single page; `limit` (1..100, default 25) caps the page size. Returns the raw `{ jobs, has_more, next_page_token }` envelope.",
+    z.object({
+      limit: z.number().int().min(1).max(100).optional(),
+    }),
+    async (p) => {
+      const limit = p.limit ?? 25;
+      const params = new URLSearchParams({ limit: String(limit) });
+      return jsonResult(await dbGet(`/api/2.1/jobs/list?${params.toString()}`));
+    },
+  );
+
+  reg(
+    "databricks_get",
+    "Fetch one Databricks job by numeric `jobId` (`GET /api/2.1/jobs/get?job_id={jobId}`). Throws when no job with that id exists.",
+    z.object({
+      jobId: z.number().int(),
+    }),
+    async (p) => {
+      const params = new URLSearchParams({ job_id: String(p.jobId) });
+      return jsonResult(await dbGet(`/api/2.1/jobs/get?${params.toString()}`));
+    },
+  );
+
+  reg(
+    "databricks_search",
+    "Substring search across Databricks jobs. Matches the query (case-insensitive) against the job's `settings.name`, `creator_user_name`, and `job_id`. Returns a `{ matches: [...] }` envelope.",
+    searchToolInputSchema(100),
+    async (p) => {
+      const params = new URLSearchParams({ limit: "100" });
+      const root = await dbGet(`/api/2.1/jobs/list?${params.toString()}`);
+      const matches = filterDatabricksJobs(jobsFrom(root), {
+        query: p.query,
+        limit: p.limit,
+      });
+      return jsonResult({ matches });
+    },
+  );
+}

package/connectors/datadog/src/server.ts

@@ -1,56 +1,10 @@
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { z } from "zod";
-
-import {
-  createRegisterSimpleTool,
-  createZodToolRegistrar,
-  mcpJsonResult as jsonResult,
-} from "../../../shared/mcp-tool-kit.ts";
-
-function siteHost(): string {
-  const s = process.env["DD_SITE"]?.trim() || "datadoghq.com";
-  return `api.${s}`;
-}
-
-function headers(): Record<string, string> {
-  const ak = process.env["DD_API_KEY"]?.trim();
-  const app = process.env["DD_APP_KEY"]?.trim();
-  if (ak === undefined || ak === "" || app === undefined || app === "") {
-    throw new Error("DD_API_KEY and DD_APP_KEY must be set");
-  }
-  return {
-    "DD-API-KEY": ak,
-    "DD-APPLICATION-KEY": app,
-    Accept: "application/json",
-  };
-}
-
-async function ddGet(path: string): Promise<unknown> {
-  const res = await fetch(`https://${siteHost()}${path}`, { headers: headers() });
-  const text = await res.text();
-  if (!res.ok) {
-    throw new Error(`Datadog ${String(res.status)}: ${text.slice(0, 400)}`);
-  }
-  return JSON.parse(text) as unknown;
-}
+import { registerDatadogTools } from "./tools.ts";
 
 const mcp = new McpServer({ name: "nimbus-datadog", version: "0.1.0" });
-const reg = createZodToolRegistrar(createRegisterSimpleTool(mcp));
-
-reg("datadog_monitor_list", "List monitors.", z.object({}), async () =>
-  jsonResult(await ddGet("/api/v1/monitor")),
-);
 
-reg(
-  "datadog_incident_list",
-  "List incidents (v2).",
-  z.object({ limit: z.number().int().min(1).max(50).optional() }),
-  async (p) => {
-    const lim = p.limit ?? 10;
-    return jsonResult(await ddGet(`/api/v2/incidents?page[size]=${String(lim)}`));
-  },
-);
+registerDatadogTools(mcp);
 
 const transport = new StdioServerTransport();
 await mcp.connect(transport);

package/connectors/datadog/src/tools.ts

@@ -0,0 +1,48 @@
+import { z } from "zod";
+import { createJsonGetter, optionalEnv } from "../../../shared/env-json-api.ts";
+import {
+  createRegisterSimpleTool,
+  createZodToolRegistrar,
+  mcpJsonResult as jsonResult,
+} from "../../../shared/mcp-tool-kit.ts";
+
+/** Both keys or neither: one alone authenticates nothing, so the refusal names the pair. */
+function headers(): Record<string, string> {
+  const ak = process.env["DD_API_KEY"]?.trim();
+  const app = process.env["DD_APP_KEY"]?.trim();
+  if (ak === undefined || ak === "" || app === undefined || app === "") {
+    throw new Error("DD_API_KEY and DD_APP_KEY must be set");
+  }
+  return {
+    "DD-API-KEY": ak,
+    "DD-APPLICATION-KEY": app,
+    Accept: "application/json",
+  };
+}
+
+const ddGet = createJsonGetter({
+  base: () => `https://api.${optionalEnv("DD_SITE", "datadoghq.com")}`,
+  label: "Datadog",
+  headers,
+});
+
+/** Tool names exposed by this connector — for contract/introspection tests. */
+export const DATADOG_TOOL_NAMES = ["datadog_incident_list", "datadog_monitor_list"] as const;
+
+export function registerDatadogTools(server: { tool: (...args: never) => unknown }): void {
+  const reg = createZodToolRegistrar(createRegisterSimpleTool(server));
+
+  reg("datadog_monitor_list", "List monitors.", z.object({}), async () =>
+    jsonResult(await ddGet("/api/v1/monitor")),
+  );
+
+  reg(
+    "datadog_incident_list",
+    "List incidents (v2).",
+    z.object({ limit: z.number().int().min(1).max(50).optional() }),
+    async (p) => {
+      const lim = p.limit ?? 10;
+      return jsonResult(await ddGet(`/api/v2/incidents?page[size]=${String(lim)}`));
+    },
+  );
+}

package/connectors/dataprofile/src/profile.ts

@@ -1,6 +1,6 @@
 import type { FileHandle } from "node:fs/promises";
-import { open, readdir } from "node:fs/promises";
-import { isAbsolute, join, relative, resolve, sep } from "node:path";
+import { open } from "node:fs/promises";
+import { relative, resolve } from "node:path";
 import {
   type DataColumn,
   firstLineAndRows,
@@ -10,6 +10,8 @@ import {
   parseJsonColumns,
   parseJsonlColumns,
 } from "@nimbus-dev/sdk";
+import { requiredEnv } from "../../../shared/env-json-api.ts";
+import { assertWithinDir, walkFiles } from "../../../shared/walk-files.ts";
 
 /**
  * Local data profiling reader (Tier-5, no-row-data). The MCP-server-side
@@ -53,21 +55,11 @@ export interface DataModel {
 }
 
 export function dataDir(): string {
-  const dir = process.env["DATAPROFILE_DIR"]?.trim();
-  if (dir === undefined || dir === "") {
-    throw new Error("DATAPROFILE_DIR is not set");
-  }
-  return resolve(dir);
+  return resolve(requiredEnv("DATAPROFILE_DIR"));
 }
 
 export function assertWithinDataDir(candidate: string, root: string): void {
-  const rel = relative(root, candidate);
-  if (rel === "") {
-    return;
-  }
-  if (rel.startsWith("..") || rel.startsWith(`..${sep}`) || isAbsolute(rel)) {
-    throw new Error("path escapes the configured data-profile dir");
-  }
+  assertWithinDir(candidate, root, "the configured data-profile dir");
 }
 
 function extOf(name: string): string {
@@ -218,37 +210,15 @@ async function profileFile(
   };
 }
 
-async function collectDataFiles(
-  root: string,
-): Promise<Array<{ path: string; format: DataFileFormat }>> {
-  const found: Array<{ path: string; format: DataFileFormat }> = [];
-  async function walk(dir: string, depth: number): Promise<void> {
-    if (depth > MAX_WALK_DEPTH || found.length >= MAX_FILES) {
-      return;
-    }
-    let entries: import("node:fs").Dirent[];
-    try {
-      entries = await readdir(dir, { withFileTypes: true });
-    } catch {
-      return;
-    }
-    for (const entry of entries) {
-      if (found.length >= MAX_FILES) {
-        return;
-      }
-      const full = join(dir, entry.name);
-      if (entry.isDirectory()) {
-        await walk(full, depth + 1);
-      } else if (entry.isFile()) {
-        const format = EXT_FORMAT[extOf(entry.name)];
-        if (format !== undefined) {
-          found.push({ path: full, format });
-        }
-      }
-    }
-  }
-  await walk(root, 0);
-  return found;
+function collectDataFiles(root: string): Promise<Array<{ path: string; format: DataFileFormat }>> {
+  return walkFiles(root, {
+    maxFiles: MAX_FILES,
+    maxDepth: MAX_WALK_DEPTH,
+    select: (entry, full) => {
+      const format = EXT_FORMAT[extOf(entry.name)];
+      return format === undefined ? undefined : { path: full, format };
+    },
+  });
 }
 
 /** Profile all data files under the configured dir into schema-only `DataModel` views. */
@@ -260,7 +230,7 @@ export async function listDataModels(
   const out: DataModel[] = [];
   for (const { path, format } of files) {
     assertWithinDataDir(path, root);
-    const model = await profileFile(path, root, format, readParquet);
+    const model = await profileFile(path, root, format, readParquet); // NOSONAR S9382: one file at a time on purpose — up to MAX_FILES (2000) files, a text file read whole up to 64 MiB, so a fan-out could hold gigabytes and thousands of open handles at once.
     if (model !== null) {
       out.push(model);
     }

package/connectors/dbt/src/server.ts

@@ -1,96 +1,6 @@
-import { z } from "zod";
-import { mcpJsonResult as jsonResult } from "../../../shared/mcp-tool-kit.ts";
 import { runReadOnlyMcpConnector } from "../../../shared/run-read-only-mcp-connector.ts";
-import { filterDbtJobs } from "./search-filter.ts";
-
-const DEFAULT_API_BASE = "https://cloud.getdbt.com";
-
-function trimTrailingSlash(s: string): string {
-  return s.endsWith("/") ? s.slice(0, -1) : s;
-}
-
-function apiBase(): string {
-  const v = process.env["DBT_API_BASE"]?.trim();
-  return v === undefined || v === "" ? DEFAULT_API_BASE : trimTrailingSlash(v);
-}
-
-function authHeader(): Record<string, string> {
-  const t = process.env["DBT_TOKEN"]?.trim();
-  if (t === undefined || t === "") {
-    throw new Error("DBT_TOKEN is not set");
-  }
-  return { Authorization: `Token ${t}`, Accept: "application/json" };
-}
-
-async function dbtGet(path: string): Promise<unknown> {
-  const res = await fetch(`${apiBase()}/api/v2${path}`, { headers: authHeader() });
-  const text = await res.text();
-  if (!res.ok) {
-    throw new Error(`dbt Cloud ${String(res.status)}: ${text.slice(0, 400)}`);
-  }
-  return JSON.parse(text) as unknown;
-}
-
-function dataFrom(root: unknown): unknown[] {
-  if (Array.isArray(root)) {
-    return root;
-  }
-  const data = (root as { data?: unknown } | null)?.data;
-  return Array.isArray(data) ? data : [];
-}
+import { registerDbtTools } from "./tools.ts";
 
 await runReadOnlyMcpConnector("nimbus-dbt", (reg) => {
-  reg(
-    "dbt_list",
-    "List dbt Cloud accounts, or jobs for an account. Without `accountId`, returns the account's accounts (`/accounts/`). With `accountId`, returns that account's jobs (`/accounts/{accountId}/jobs/`), capped at `limit` (default 100). Returns the raw dbt Cloud envelope (`{ data: [...] }`).",
-    z.object({
-      accountId: z.number().int().optional(),
-      limit: z.number().int().min(1).max(500).optional(),
-    }),
-    async (p) => {
-      if (p.accountId === undefined) {
-        return jsonResult(await dbtGet("/accounts/"));
-      }
-      const search = new URLSearchParams({ limit: String(p.limit ?? 100) });
-      return jsonResult(
-        await dbtGet(
-          `/accounts/${encodeURIComponent(String(p.accountId))}/jobs/?${search.toString()}`,
-        ),
-      );
-    },
-  );
-
-  reg(
-    "dbt_get",
-    "Fetch one dbt Cloud job by `accountId` + `jobId` (`/accounts/{accountId}/jobs/{jobId}/`). Returns the raw dbt Cloud envelope. Throws when no such job exists.",
-    z.object({
-      accountId: z.number().int(),
-      jobId: z.number().int(),
-    }),
-    async (p) => {
-      return jsonResult(
-        await dbtGet(
-          `/accounts/${encodeURIComponent(String(p.accountId))}/jobs/${encodeURIComponent(String(p.jobId))}/`,
-        ),
-      );
-    },
-  );
-
-  reg(
-    "dbt_search",
-    "Substring search across an account's dbt Cloud jobs. Matches the query against job name, dbt version, and id (case-insensitive). Returns a `{ matches: [...] }` envelope.",
-    z.object({
-      accountId: z.number().int(),
-      query: z.string().min(1),
-      limit: z.number().int().min(1).max(200).optional(),
-    }),
-    async (p) => {
-      const search = new URLSearchParams({ limit: "500" });
-      const root = await dbtGet(
-        `/accounts/${encodeURIComponent(String(p.accountId))}/jobs/?${search.toString()}`,
-      );
-      const matches = filterDbtJobs(dataFrom(root), { query: p.query, limit: p.limit });
-      return jsonResult({ matches });
-    },
-  );
+  registerDbtTools(reg);
 });