@dbx-tools/search 0.6.9

package/lib/src/lakebase.js

@@ -0,0 +1,218 @@
+/**
+ * A Lakebase (Postgres) full-text search backend - the FALLBACK used when no
+ * Databricks Vector Search endpoint/index is configured but a Lakebase pool is
+ * available. It provisions a single table per index, indexes a generated
+ * `tsvector`, and answers queries with `websearch_to_tsquery` + `ts_rank`.
+ *
+ * The whole point is parity: this backend returns the EXACT same
+ * `@dbx-tools/shared-search` shapes (`SearchResult` / `SearchHit` /
+ * `UpsertResult`) as the Vector Search backend, so the client, the Mastra
+ * tools, the routes, and the React search box cannot tell which one answered.
+ * A hit's `id` is the primary key, its `score` is the text-rank, and `fields`
+ * is the stored document minus the internal columns.
+ *
+ * The Postgres pool is built the same way `@dbx-tools/appkit-mastra` builds its
+ * memory pool: the AppKit `lakebase` plugin resolves a service-principal
+ * `PoolConfig` (connection target + OAuth token-refresh `password` callback),
+ * and this backend constructs a `pg.Pool` from it. It never re-implements auth.
+ *
+ * @module
+ */
+import { ExecutionError } from "@databricks/appkit";
+import { log, object, string } from "@dbx-tools/shared-core";
+import { Pool } from "pg";
+const logger = log.logger("search/lakebase");
+/** The internal columns every search table carries, excluded from a hit's `fields`. */
+const RESERVED_COLUMNS = new Set(["id", "search_text", "document", "search_vector"]);
+/**
+ * A Postgres full-text backend. One instance is shared across indexes; each
+ * index maps to a table whose name is derived from the index reference.
+ */
+export class LakebaseSearchBackend {
+    pgConfigFactory;
+    schema;
+    poolFactory;
+    pool;
+    poolPromise;
+    provisioned = new Set();
+    constructor(pgConfigFactory, schema = "public", 
+    /** How a pool is built from the resolved config. Overridable for tests. */
+    poolFactory = (config) => new Pool(config)) {
+        this.pgConfigFactory = pgConfigFactory;
+        this.schema = schema;
+        this.poolFactory = poolFactory;
+    }
+    /** Lazily build (and cache) the pg pool from the resolved Lakebase config. */
+    async getPool() {
+        if (this.pool)
+            return this.pool;
+        this.poolPromise ??= (async () => {
+            const config = await this.pgConfigFactory();
+            const pool = this.poolFactory(config);
+            this.pool = pool;
+            return pool;
+        })();
+        return this.poolPromise;
+    }
+    /** Close the pool so a restarted app rebuilds it. */
+    async close() {
+        const pool = this.pool;
+        this.pool = undefined;
+        this.poolPromise = undefined;
+        this.provisioned.clear();
+        if (pool)
+            await pool.end();
+    }
+    /**
+     * Search a Lakebase-backed index. Returns hits sorted most-relevant-first,
+     * shaped identically to the Vector Search backend.
+     */
+    async search(index, query, options = {}) {
+        const text = string.trimToEmpty(query);
+        const table = this.tableFor(index);
+        const limit = options.limit ?? 10;
+        const pool = await this.getPool();
+        // `websearch_to_tsquery` accepts a bare search box string (quoted phrases,
+        // `or`, `-term`); an empty query returns the most recent rows so an empty
+        // search box still shows content, matching a keyword index's behavior.
+        const sql = text
+            ? `SELECT id, document, ts_rank(search_vector, websearch_to_tsquery('english', $1)) AS score
+           FROM ${table}
+          WHERE search_vector @@ websearch_to_tsquery('english', $1)
+          ORDER BY score DESC
+          LIMIT $2`
+            : `SELECT id, document, 0::float4 AS score FROM ${table} ORDER BY id LIMIT $2`;
+        const params = text ? [text, limit] : [limit];
+        const { rows } = await this.query(pool, sql, params, options.signal);
+        const hits = rows
+            .map((row) => ({
+            id: String(row.id),
+            score: Number(row.score) || 0,
+            fields: this.toFields(row.document),
+        }))
+            .filter((hit) => options.scoreThreshold === undefined || hit.score >= options.scoreThreshold);
+        return { query: text, index, hits, count: hits.length };
+    }
+    /**
+     * Ensure the table + full-text index exist and seed documents when empty.
+     * Idempotent, so it is safe to call on every boot.
+     */
+    async provision(index, options = {}) {
+        const table = this.tableFor(index);
+        const pool = await this.getPool();
+        await this.ensureTable(pool, table, options.signal);
+        const { rows } = await this.query(pool, `SELECT count(*)::text AS count FROM ${table}`, [], options.signal);
+        const existing = Number(rows[0]?.count ?? "0");
+        const seed = options.seed ?? [];
+        if (existing === 0 && seed.length > 0) {
+            await this.upsert(pool, table, seed, options.textColumn ?? "text", options.signal);
+            logger.info("index-seeded", { index, table, count: seed.length });
+            return seed.length;
+        }
+        return existing;
+    }
+    /** Add or update documents by primary key. */
+    async addDocuments(index, documents, textColumn = "text", signal) {
+        const table = this.tableFor(index);
+        const pool = await this.getPool();
+        await this.ensureTable(pool, table, signal);
+        await this.upsert(pool, table, documents, textColumn, signal);
+        return { index, count: documents.length };
+    }
+    /** Delete documents by primary key. */
+    async deleteDocuments(index, ids, signal) {
+        const table = this.tableFor(index);
+        const pool = await this.getPool();
+        await this.query(pool, `DELETE FROM ${table} WHERE id = ANY($1)`, [ids.map(String)], signal);
+        return { index, count: ids.length };
+    }
+    /** Create the table + GIN index once per table (memoized across calls). */
+    async ensureTable(pool, table, signal) {
+        if (this.provisioned.has(table))
+            return;
+        // `document` holds the whole row; `search_text` is the indexed text; the
+        // generated `search_vector` keeps the tsvector in lockstep with it so a
+        // write never has to compute the vector by hand.
+        await this.query(pool, `CREATE SCHEMA IF NOT EXISTS ${this.ident(this.schema)}`, [], signal);
+        await this.query(pool, `CREATE TABLE IF NOT EXISTS ${table} (
+         id text PRIMARY KEY,
+         search_text text NOT NULL DEFAULT '',
+         document jsonb NOT NULL DEFAULT '{}'::jsonb,
+         search_vector tsvector GENERATED ALWAYS AS (to_tsvector('english', search_text)) STORED
+       )`, [], signal);
+        await this.query(pool, `CREATE INDEX IF NOT EXISTS ${this.ident(`${this.bareName(table)}_fts`)}
+         ON ${table} USING gin (search_vector)`, [], signal);
+        this.provisioned.add(table);
+        logger.info("index-created", { table });
+    }
+    /** Upsert rows: the whole document as jsonb + a flattened text blob to index. */
+    async upsert(pool, table, documents, textColumn, signal) {
+        for (const doc of documents) {
+            const id = string.trimToNull(String(doc.id ?? doc.ID ?? "")) ?? undefined;
+            if (id === undefined) {
+                throw new ExecutionError("search (lakebase): a document is missing an `id`", {
+                    context: { operation: "addDocuments" },
+                });
+            }
+            const searchText = this.searchText(doc, textColumn);
+            await this.query(pool, `INSERT INTO ${table} (id, search_text, document)
+           VALUES ($1, $2, $3::jsonb)
+         ON CONFLICT (id) DO UPDATE
+           SET search_text = EXCLUDED.search_text, document = EXCLUDED.document`, [id, searchText, JSON.stringify(doc)], signal);
+        }
+    }
+    /** The text a row is indexed by: the text column first, then any other string field. */
+    searchText(doc, textColumn) {
+        const primary = string.trimToEmpty(String(doc[textColumn] ?? ""));
+        const rest = Object.entries(doc)
+            .filter(([key, value]) => key !== textColumn && key !== "id" && typeof value === "string")
+            .map(([, value]) => value);
+        return [primary, ...rest].filter(Boolean).join("\n");
+    }
+    /** A hit's `fields`: the stored document minus the reserved/internal keys. */
+    toFields(document) {
+        if (!object.isRecord(document))
+            return {};
+        const fields = {};
+        for (const [key, value] of Object.entries(document)) {
+            if (RESERVED_COLUMNS.has(key))
+                continue;
+            fields[key] = value;
+        }
+        return fields;
+    }
+    /** The fully-qualified table name for an index reference. */
+    tableFor(index) {
+        return `${this.ident(this.schema)}.${this.ident(this.bareName(index))}`;
+    }
+    /** A safe bare table name derived from an index reference. */
+    bareName(reference) {
+        const last = reference.split(".").filter(Boolean).pop() ?? reference;
+        const slug = last
+            .toLowerCase()
+            .replace(/[^a-z0-9_]+/g, "_")
+            .replace(/^_+|_+$/g, "");
+        return slug.length > 0 ? slug : "documents";
+    }
+    /** Quote a Postgres identifier. */
+    ident(name) {
+        return `"${name.replace(/"/g, '""')}"`;
+    }
+    /** Run one query under external cancellation. */
+    async query(pool, sql, params, signal) {
+        if (signal?.aborted)
+            throw ExecutionError.canceled();
+        const client = await pool.connect();
+        const onAbort = () => void client.release(true);
+        signal?.addEventListener("abort", onAbort, { once: true });
+        try {
+            const result = await client.query(sql, params);
+            return { rows: result.rows };
+        }
+        finally {
+            signal?.removeEventListener("abort", onAbort);
+            client.release();
+        }
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibGFrZWJhc2UuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvbGFrZWJhc2UudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUE7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7R0FtQkc7QUFFSCxPQUFPLEVBQUUsY0FBYyxFQUFFLE1BQU0sb0JBQW9CLENBQUM7QUFDcEQsT0FBTyxFQUFFLEdBQUcsRUFBRSxNQUFNLEVBQUUsTUFBTSxFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFDN0QsT0FBTyxFQUFFLElBQUksRUFBbUIsTUFBTSxJQUFJLENBQUM7QUFRM0MsTUFBTSxNQUFNLEdBQUcsR0FBRyxDQUFDLE1BQU0sQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDO0FBRTdDLHVGQUF1RjtBQUN2RixNQUFNLGdCQUFnQixHQUFHLElBQUksR0FBRyxDQUFDLENBQUMsSUFBSSxFQUFFLGFBQWEsRUFBRSxVQUFVLEVBQUUsZUFBZSxDQUFDLENBQUMsQ0FBQztBQWtCckY7OztHQUdHO0FBQ0gsTUFBTSxPQUFPLHFCQUFxQjtJQU1iO0lBQ0E7SUFFQTtJQVJYLElBQUksQ0FBbUI7SUFDdkIsV0FBVyxDQUE0QjtJQUM5QixXQUFXLEdBQUcsSUFBSSxHQUFHLEVBQVUsQ0FBQztJQUVqRCxZQUNtQixlQUF1RCxFQUN2RCxTQUFTLFFBQVE7SUFDbEMsMkVBQTJFO0lBQzFELGNBQTRDLENBQUMsTUFBTSxFQUFFLEVBQUUsQ0FBQyxJQUFJLElBQUksQ0FBQyxNQUFNLENBQUM7UUFIeEUsb0JBQWUsR0FBZixlQUFlLENBQXdDO1FBQ3ZELFdBQU0sR0FBTixNQUFNLENBQVc7UUFFakIsZ0JBQVcsR0FBWCxXQUFXLENBQTZEO0lBQ3hGLENBQUM7SUFFSiw4RUFBOEU7SUFDdEUsS0FBSyxDQUFDLE9BQU87UUFDbkIsSUFBSSxJQUFJLENBQUMsSUFBSTtZQUFFLE9BQU8sSUFBSSxDQUFDLElBQUksQ0FBQztRQUNoQyxJQUFJLENBQUMsV0FBVyxLQUFLLENBQUMsS0FBSyxJQUFJLEVBQUU7WUFDL0IsTUFBTSxNQUFNLEdBQUcsTUFBTSxJQUFJLENBQUMsZUFBZSxFQUFFLENBQUM7WUFDNUMsTUFBTSxJQUFJLEdBQUcsSUFBSSxDQUFDLFdBQVcsQ0FBQyxNQUFNLENBQUMsQ0FBQztZQUN0QyxJQUFJLENBQUMsSUFBSSxHQUFHLElBQUksQ0FBQztZQUNqQixPQUFPLElBQUksQ0FBQztRQUNkLENBQUMsQ0FBQyxFQUFFLENBQUM7UUFDTCxPQUFPLElBQUksQ0FBQyxXQUFXLENBQUM7SUFDMUIsQ0FBQztJQUVELHFEQUFxRDtJQUNyRCxLQUFLLENBQUMsS0FBSztRQUNULE1BQU0sSUFBSSxHQUFHLElBQUksQ0FBQyxJQUFJLENBQUM7UUFDdkIsSUFBSSxDQUFDLElBQUksR0FBRyxTQUFTLENBQUM7UUFDdEIsSUFBSSxDQUFDLFdBQVcsR0FBRyxTQUFTLENBQUM7UUFDN0IsSUFBSSxDQUFDLFdBQVcsQ0FBQyxLQUFLLEVBQUUsQ0FBQztRQUN6QixJQUFJLElBQUk7WUFBRSxNQUFNLElBQUksQ0FBQyxHQUFHLEVBQUUsQ0FBQztJQUM3QixDQUFDO0lBRUQ7OztPQUdHO0lBQ0gsS0FBSyxDQUFDLE1BQU0sQ0FDVixLQUFhLEVBQ2IsS0FBYSxFQUNiLFVBQWlDLEVBQUU7UUFFbkMsTUFBTSxJQUFJLEdBQUcsTUFBTSxDQUFDLFdBQVcsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUN2QyxNQUFNLEtBQUssR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ25DLE1BQU0sS0FBSyxHQUFHLE9BQU8sQ0FBQyxLQUFLLElBQUksRUFBRSxDQUFDO1FBQ2xDLE1BQU0sSUFBSSxHQUFHLE1BQU0sSUFBSSxDQUFDLE9BQU8sRUFBRSxDQUFDO1FBRWxDLDJFQUEyRTtRQUMzRSwwRUFBMEU7UUFDMUUsdUVBQXVFO1FBQ3ZFLE1BQU0sR0FBRyxHQUFHLElBQUk7WUFDZCxDQUFDLENBQUM7a0JBQ1UsS0FBSzs7O21CQUdKO1lBQ2IsQ0FBQyxDQUFDLGdEQUFnRCxLQUFLLHVCQUF1QixDQUFDO1FBQ2pGLE1BQU0sTUFBTSxHQUFHLElBQUksQ0FBQyxDQUFDLENBQUMsQ0FBQyxJQUFJLEVBQUUsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQUMsS0FBSyxDQUFDLENBQUM7UUFFOUMsTUFBTSxFQUFFLElBQUksRUFBRSxHQUFHLE1BQU0sSUFBSSxDQUFDLEtBQUssQ0FDL0IsSUFBSSxFQUNKLEdBQUcsRUFDSCxNQUFNLEVBQ04sT0FBTyxDQUFDLE1BQU0sQ0FDZixDQUFDO1FBQ0YsTUFBTSxJQUFJLEdBQWdCLElBQUk7YUFDM0IsR0FBRyxDQUFDLENBQUMsR0FBRyxFQUFFLEVBQUUsQ0FBQyxDQUFDO1lBQ2IsRUFBRSxFQUFFLE1BQU0sQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDO1lBQ2xCLEtBQUssRUFBRSxNQUFNLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUM7WUFDN0IsTUFBTSxFQUFFLElBQUksQ0FBQyxRQUFRLENBQUMsR0FBRyxDQUFDLFFBQVEsQ0FBQztTQUNwQyxDQUFDLENBQUM7YUFDRixNQUFNLENBQUMsQ0FBQyxHQUFHLEVBQUUsRUFBRSxDQUFDLE9BQU8sQ0FBQyxjQUFjLEtBQUssU0FBUyxJQUFJLEdBQUcsQ0FBQyxLQUFLLElBQUksT0FBTyxDQUFDLGNBQWMsQ0FBQyxDQUFDO1FBQ2hHLE9BQU8sRUFBRSxLQUFLLEVBQUUsSUFBSSxFQUFFLEtBQUssRUFBRSxJQUFJLEVBQUUsS0FBSyxFQUFFLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FBQztJQUMxRCxDQUFDO0lBRUQ7OztPQUdHO0lBQ0gsS0FBSyxDQUFDLFNBQVMsQ0FBQyxLQUFhLEVBQUUsVUFBb0MsRUFBRTtRQUNuRSxNQUFNLEtBQUssR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ25DLE1BQU0sSUFBSSxHQUFHLE1BQU0sSUFBSSxDQUFDLE9BQU8sRUFBRSxDQUFDO1FBQ2xDLE1BQU0sSUFBSSxDQUFDLFdBQVcsQ0FBQyxJQUFJLEVBQUUsS0FBSyxFQUFFLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQztRQUVwRCxNQUFNLEVBQUUsSUFBSSxFQUFFLEdBQUcsTUFBTSxJQUFJLENBQUMsS0FBSyxDQUMvQixJQUFJLEVBQ0osdUNBQXVDLEtBQUssRUFBRSxFQUM5QyxFQUFFLEVBQ0YsT0FBTyxDQUFDLE1BQU0sQ0FDZixDQUFDO1FBQ0YsTUFBTSxRQUFRLEdBQUcsTUFBTSxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsRUFBRSxLQUFLLElBQUksR0FBRyxDQUFDLENBQUM7UUFDL0MsTUFBTSxJQUFJLEdBQUcsT0FBTyxDQUFDLElBQUksSUFBSSxFQUFFLENBQUM7UUFDaEMsSUFBSSxRQUFRLEtBQUssQ0FBQyxJQUFJLElBQUksQ0FBQyxNQUFNLEdBQUcsQ0FBQyxFQUFFLENBQUM7WUFDdEMsTUFBTSxJQUFJLENBQUMsTUFBTSxDQUFDLElBQUksRUFBRSxLQUFLLEVBQUUsSUFBSSxFQUFFLE9BQU8sQ0FBQyxVQUFVLElBQUksTUFBTSxFQUFFLE9BQU8sQ0FBQyxNQUFNLENBQUMsQ0FBQztZQUNuRixNQUFNLENBQUMsSUFBSSxDQUFDLGNBQWMsRUFBRSxFQUFFLEtBQUssRUFBRSxLQUFLLEVBQUUsS0FBSyxFQUFFLElBQUksQ0FBQyxNQUFNLEVBQUUsQ0FBQyxDQUFDO1lBQ2xFLE9BQU8sSUFBSSxDQUFDLE1BQU0sQ0FBQztRQUNyQixDQUFDO1FBQ0QsT0FBTyxRQUFRLENBQUM7SUFDbEIsQ0FBQztJQUVELDhDQUE4QztJQUM5QyxLQUFLLENBQUMsWUFBWSxDQUNoQixLQUFhLEVBQ2IsU0FBMkIsRUFDM0IsVUFBVSxHQUFHLE1BQU0sRUFDbkIsTUFBb0I7UUFFcEIsTUFBTSxLQUFLLEdBQUcsSUFBSSxDQUFDLFFBQVEsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUNuQyxNQUFNLElBQUksR0FBRyxNQUFNLElBQUksQ0FBQyxPQUFPLEVBQUUsQ0FBQztRQUNsQyxNQUFNLElBQUksQ0FBQyxXQUFXLENBQUMsSUFBSSxFQUFFLEtBQUssRUFBRSxNQUFNLENBQUMsQ0FBQztRQUM1QyxNQUFNLElBQUksQ0FBQyxNQUFNLENBQUMsSUFBSSxFQUFFLEtBQUssRUFBRSxTQUFTLEVBQUUsVUFBVSxFQUFFLE1BQU0sQ0FBQyxDQUFDO1FBQzlELE9BQU8sRUFBRSxLQUFLLEVBQUUsS0FBSyxFQUFFLFNBQVMsQ0FBQyxNQUFNLEVBQUUsQ0FBQztJQUM1QyxDQUFDO0lBRUQsdUNBQXVDO0lBQ3ZDLEtBQUssQ0FBQyxlQUFlLENBQ25CLEtBQWEsRUFDYixHQUEyQixFQUMzQixNQUFvQjtRQUVwQixNQUFNLEtBQUssR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ25DLE1BQU0sSUFBSSxHQUFHLE1BQU0sSUFBSSxDQUFDLE9BQU8sRUFBRSxDQUFDO1FBQ2xDLE1BQU0sSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLEVBQUUsZUFBZSxLQUFLLHFCQUFxQixFQUFFLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxNQUFNLENBQUMsQ0FBQyxFQUFFLE1BQU0sQ0FBQyxDQUFDO1FBQzdGLE9BQU8sRUFBRSxLQUFLLEVBQUUsS0FBSyxFQUFFLEdBQUcsQ0FBQyxNQUFNLEVBQUUsQ0FBQztJQUN0QyxDQUFDO0lBRUQsMkVBQTJFO0lBQ25FLEtBQUssQ0FBQyxXQUFXLENBQUMsSUFBVSxFQUFFLEtBQWEsRUFBRSxNQUFvQjtRQUN2RSxJQUFJLElBQUksQ0FBQyxXQUFXLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQztZQUFFLE9BQU87UUFDeEMseUVBQXlFO1FBQ3pFLHdFQUF3RTtRQUN4RSxpREFBaUQ7UUFDakQsTUFBTSxJQUFJLENBQUMsS0FBSyxDQUFDLElBQUksRUFBRSwrQkFBK0IsSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLEVBQUUsRUFBRSxFQUFFLEVBQUUsTUFBTSxDQUFDLENBQUM7UUFDN0YsTUFBTSxJQUFJLENBQUMsS0FBSyxDQUNkLElBQUksRUFDSiw4QkFBOEIsS0FBSzs7Ozs7U0FLaEMsRUFDSCxFQUFFLEVBQ0YsTUFBTSxDQUNQLENBQUM7UUFDRixNQUFNLElBQUksQ0FBQyxLQUFLLENBQ2QsSUFBSSxFQUNKLDhCQUE4QixJQUFJLENBQUMsS0FBSyxDQUFDLEdBQUcsSUFBSSxDQUFDLFFBQVEsQ0FBQyxLQUFLLENBQUMsTUFBTSxDQUFDO2NBQy9ELEtBQUssNEJBQTRCLEVBQ3pDLEVBQUUsRUFDRixNQUFNLENBQ1AsQ0FBQztRQUNGLElBQUksQ0FBQyxXQUFXLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQzVCLE1BQU0sQ0FBQyxJQUFJLENBQUMsZUFBZSxFQUFFLEVBQUUsS0FBSyxFQUFFLENBQUMsQ0FBQztJQUMxQyxDQUFDO0lBRUQsaUZBQWlGO0lBQ3pFLEtBQUssQ0FBQyxNQUFNLENBQ2xCLElBQVUsRUFDVixLQUFhLEVBQ2IsU0FBMkIsRUFDM0IsVUFBa0IsRUFDbEIsTUFBb0I7UUFFcEIsS0FBSyxNQUFNLEdBQUcsSUFBSSxTQUFTLEVBQUUsQ0FBQztZQUM1QixNQUFNLEVBQUUsR0FBRyxNQUFNLENBQUMsVUFBVSxDQUFDLE1BQU0sQ0FBQyxHQUFHLENBQUMsRUFBRSxJQUFJLEdBQUcsQ0FBQyxFQUFFLElBQUksRUFBRSxDQUFDLENBQUMsSUFBSSxTQUFTLENBQUM7WUFDMUUsSUFBSSxFQUFFLEtBQUssU0FBUyxFQUFFLENBQUM7Z0JBQ3JCLE1BQU0sSUFBSSxjQUFjLENBQUMsa0RBQWtELEVBQUU7b0JBQzNFLE9BQU8sRUFBRSxFQUFFLFNBQVMsRUFBRSxjQUFjLEVBQUU7aUJBQ3ZDLENBQUMsQ0FBQztZQUNMLENBQUM7WUFDRCxNQUFNLFVBQVUsR0FBRyxJQUFJLENBQUMsVUFBVSxDQUFDLEdBQUcsRUFBRSxVQUFVLENBQUMsQ0FBQztZQUNwRCxNQUFNLElBQUksQ0FBQyxLQUFLLENBQ2QsSUFBSSxFQUNKLGVBQWUsS0FBSzs7O2dGQUdvRCxFQUN4RSxDQUFDLEVBQUUsRUFBRSxVQUFVLEVBQUUsSUFBSSxDQUFDLFNBQVMsQ0FBQyxHQUFHLENBQUMsQ0FBQyxFQUNyQyxNQUFNLENBQ1AsQ0FBQztRQUNKLENBQUM7SUFDSCxDQUFDO0lBRUQsd0ZBQXdGO0lBQ2hGLFVBQVUsQ0FBQyxHQUFtQixFQUFFLFVBQWtCO1FBQ3hELE1BQU0sT0FBTyxHQUFHLE1BQU0sQ0FBQyxXQUFXLENBQUMsTUFBTSxDQUFDLEdBQUcsQ0FBQyxVQUFVLENBQUMsSUFBSSxFQUFFLENBQUMsQ0FBQyxDQUFDO1FBQ2xFLE1BQU0sSUFBSSxHQUFHLE1BQU0sQ0FBQyxPQUFPLENBQUMsR0FBRyxDQUFDO2FBQzdCLE1BQU0sQ0FBQyxDQUFDLENBQUMsR0FBRyxFQUFFLEtBQUssQ0FBQyxFQUFFLEVBQUUsQ0FBQyxHQUFHLEtBQUssVUFBVSxJQUFJLEdBQUcsS0FBSyxJQUFJLElBQUksT0FBTyxLQUFLLEtBQUssUUFBUSxDQUFDO2FBQ3pGLEdBQUcsQ0FBQyxDQUFDLENBQUMsRUFBRSxLQUFLLENBQUMsRUFBRSxFQUFFLENBQUMsS0FBZSxDQUFDLENBQUM7UUFDdkMsT0FBTyxDQUFDLE9BQU8sRUFBRSxHQUFHLElBQUksQ0FBQyxDQUFDLE1BQU0sQ0FBQyxPQUFPLENBQUMsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLENBQUM7SUFDdkQsQ0FBQztJQUVELDhFQUE4RTtJQUN0RSxRQUFRLENBQUMsUUFBaUI7UUFDaEMsSUFBSSxDQUFDLE1BQU0sQ0FBQyxRQUFRLENBQUMsUUFBUSxDQUFDO1lBQUUsT0FBTyxFQUFFLENBQUM7UUFDMUMsTUFBTSxNQUFNLEdBQTRCLEVBQUUsQ0FBQztRQUMzQyxLQUFLLE1BQU0sQ0FBQyxHQUFHLEVBQUUsS0FBSyxDQUFDLElBQUksTUFBTSxDQUFDLE9BQU8sQ0FBQyxRQUFRLENBQUMsRUFBRSxDQUFDO1lBQ3BELElBQUksZ0JBQWdCLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQztnQkFBRSxTQUFTO1lBQ3hDLE1BQU0sQ0FBQyxHQUFHLENBQUMsR0FBRyxLQUFLLENBQUM7UUFDdEIsQ0FBQztRQUNELE9BQU8sTUFBTSxDQUFDO0lBQ2hCLENBQUM7SUFFRCw2REFBNkQ7SUFDckQsUUFBUSxDQUFDLEtBQWE7UUFDNUIsT0FBTyxHQUFHLElBQUksQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLE1BQU0sQ0FBQyxJQUFJLElBQUksQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLFFBQVEsQ0FBQyxLQUFLLENBQUMsQ0FBQyxFQUFFLENBQUM7SUFDMUUsQ0FBQztJQUVELDhEQUE4RDtJQUN0RCxRQUFRLENBQUMsU0FBaUI7UUFDaEMsTUFBTSxJQUFJLEdBQUcsU0FBUyxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsQ0FBQyxNQUFNLENBQUMsT0FBTyxDQUFDLENBQUMsR0FBRyxFQUFFLElBQUksU0FBUyxDQUFDO1FBQ3JFLE1BQU0sSUFBSSxHQUFHLElBQUk7YUFDZCxXQUFXLEVBQUU7YUFDYixPQUFPLENBQUMsY0FBYyxFQUFFLEdBQUcsQ0FBQzthQUM1QixPQUFPLENBQUMsVUFBVSxFQUFFLEVBQUUsQ0FBQyxDQUFDO1FBQzNCLE9BQU8sSUFBSSxDQUFDLE1BQU0sR0FBRyxDQUFDLENBQUMsQ0FBQyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsV0FBVyxDQUFDO0lBQzlDLENBQUM7SUFFRCxtQ0FBbUM7SUFDM0IsS0FBSyxDQUFDLElBQVk7UUFDeEIsT0FBTyxJQUFJLElBQUksQ0FBQyxPQUFPLENBQUMsSUFBSSxFQUFFLElBQUksQ0FBQyxHQUFHLENBQUM7SUFDekMsQ0FBQztJQUVELGlEQUFpRDtJQUN6QyxLQUFLLENBQUMsS0FBSyxDQUNqQixJQUFVLEVBQ1YsR0FBVyxFQUNYLE1BQWlCLEVBQ2pCLE1BQW9CO1FBRXBCLElBQUksTUFBTSxFQUFFLE9BQU87WUFBRSxNQUFNLGNBQWMsQ0FBQyxRQUFRLEVBQUUsQ0FBQztRQUNyRCxNQUFNLE1BQU0sR0FBRyxNQUFNLElBQUksQ0FBQyxPQUFPLEVBQUUsQ0FBQztRQUNwQyxNQUFNLE9BQU8sR0FBRyxHQUFHLEVBQUUsQ0FBQyxLQUFLLE1BQU0sQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUM7UUFDaEQsTUFBTSxFQUFFLGdCQUFnQixDQUFDLE9BQU8sRUFBRSxPQUFPLEVBQUUsRUFBRSxJQUFJLEVBQUUsSUFBSSxFQUFFLENBQUMsQ0FBQztRQUMzRCxJQUFJLENBQUM7WUFDSCxNQUFNLE1BQU0sR0FBRyxNQUFNLE1BQU0sQ0FBQyxLQUFLLENBQUMsR0FBRyxFQUFFLE1BQU0sQ0FBQyxDQUFDO1lBQy9DLE9BQU8sRUFBRSxJQUFJLEVBQUUsTUFBTSxDQUFDLElBQVcsRUFBRSxDQUFDO1FBQ3RDLENBQUM7Z0JBQVMsQ0FBQztZQUNULE1BQU0sRUFBRSxtQkFBbUIsQ0FBQyxPQUFPLEVBQUUsT0FBTyxDQUFDLENBQUM7WUFDOUMsTUFBTSxDQUFDLE9BQU8sRUFBRSxDQUFDO1FBQ25CLENBQUM7SUFDSCxDQUFDO0NBQ0YiLCJzb3VyY2VzQ29udGVudCI6WyIvKipcbiAqIEEgTGFrZWJhc2UgKFBvc3RncmVzKSBmdWxsLXRleHQgc2VhcmNoIGJhY2tlbmQgLSB0aGUgRkFMTEJBQ0sgdXNlZCB3aGVuIG5vXG4gKiBEYXRhYnJpY2tzIFZlY3RvciBTZWFyY2ggZW5kcG9pbnQvaW5kZXggaXMgY29uZmlndXJlZCBidXQgYSBMYWtlYmFzZSBwb29sIGlzXG4gKiBhdmFpbGFibGUuIEl0IHByb3Zpc2lvbnMgYSBzaW5nbGUgdGFibGUgcGVyIGluZGV4LCBpbmRleGVzIGEgZ2VuZXJhdGVkXG4gKiBgdHN2ZWN0b3JgLCBhbmQgYW5zd2VycyBxdWVyaWVzIHdpdGggYHdlYnNlYXJjaF90b190c3F1ZXJ5YCArIGB0c19yYW5rYC5cbiAqXG4gKiBUaGUgd2hvbGUgcG9pbnQgaXMgcGFyaXR5OiB0aGlzIGJhY2tlbmQgcmV0dXJucyB0aGUgRVhBQ1Qgc2FtZVxuICogYEBkYngtdG9vbHMvc2hhcmVkLXNlYXJjaGAgc2hhcGVzIChgU2VhcmNoUmVzdWx0YCAvIGBTZWFyY2hIaXRgIC9cbiAqIGBVcHNlcnRSZXN1bHRgKSBhcyB0aGUgVmVjdG9yIFNlYXJjaCBiYWNrZW5kLCBzbyB0aGUgY2xpZW50LCB0aGUgTWFzdHJhXG4gKiB0b29scywgdGhlIHJvdXRlcywgYW5kIHRoZSBSZWFjdCBzZWFyY2ggYm94IGNhbm5vdCB0ZWxsIHdoaWNoIG9uZSBhbnN3ZXJlZC5cbiAqIEEgaGl0J3MgYGlkYCBpcyB0aGUgcHJpbWFyeSBrZXksIGl0cyBgc2NvcmVgIGlzIHRoZSB0ZXh0LXJhbmssIGFuZCBgZmllbGRzYFxuICogaXMgdGhlIHN0b3JlZCBkb2N1bWVudCBtaW51cyB0aGUgaW50ZXJuYWwgY29sdW1ucy5cbiAqXG4gKiBUaGUgUG9zdGdyZXMgcG9vbCBpcyBidWlsdCB0aGUgc2FtZSB3YXkgYEBkYngtdG9vbHMvYXBwa2l0LW1hc3RyYWAgYnVpbGRzIGl0c1xuICogbWVtb3J5IHBvb2w6IHRoZSBBcHBLaXQgYGxha2ViYXNlYCBwbHVnaW4gcmVzb2x2ZXMgYSBzZXJ2aWNlLXByaW5jaXBhbFxuICogYFBvb2xDb25maWdgIChjb25uZWN0aW9uIHRhcmdldCArIE9BdXRoIHRva2VuLXJlZnJlc2ggYHBhc3N3b3JkYCBjYWxsYmFjayksXG4gKiBhbmQgdGhpcyBiYWNrZW5kIGNvbnN0cnVjdHMgYSBgcGcuUG9vbGAgZnJvbSBpdC4gSXQgbmV2ZXIgcmUtaW1wbGVtZW50cyBhdXRoLlxuICpcbiAqIEBtb2R1bGVcbiAqL1xuXG5pbXBvcnQgeyBFeGVjdXRpb25FcnJvciB9IGZyb20gXCJAZGF0YWJyaWNrcy9hcHBraXRcIjtcbmltcG9ydCB7IGxvZywgb2JqZWN0LCBzdHJpbmcgfSBmcm9tIFwiQGRieC10b29scy9zaGFyZWQtY29yZVwiO1xuaW1wb3J0IHsgUG9vbCwgdHlwZSBQb29sQ29uZmlnIH0gZnJvbSBcInBnXCI7XG5pbXBvcnQgdHlwZSB7XG4gIFNlYXJjaERvY3VtZW50LFxuICBTZWFyY2hIaXQsXG4gIFNlYXJjaFJlc3VsdCxcbiAgVXBzZXJ0UmVzdWx0LFxufSBmcm9tIFwiQGRieC10b29scy9zaGFyZWQtc2VhcmNoXCI7XG5cbmNvbnN0IGxvZ2dlciA9IGxvZy5sb2dnZXIoXCJzZWFyY2gvbGFrZWJhc2VcIik7XG5cbi8qKiBUaGUgaW50ZXJuYWwgY29sdW1ucyBldmVyeSBzZWFyY2ggdGFibGUgY2FycmllcywgZXhjbHVkZWQgZnJvbSBhIGhpdCdzIGBmaWVsZHNgLiAqL1xuY29uc3QgUkVTRVJWRURfQ09MVU1OUyA9IG5ldyBTZXQoW1wiaWRcIiwgXCJzZWFyY2hfdGV4dFwiLCBcImRvY3VtZW50XCIsIFwic2VhcmNoX3ZlY3RvclwiXSk7XG5cbi8qKiBPcHRpb25zIGZvciBhIHNpbmdsZS1pbmRleCBMYWtlYmFzZSBzZWFyY2ggKG1pcnJvcnMgdGhlIGNsaWVudCdzIGBTZWFyY2hPcHRpb25zYCkuICovXG5leHBvcnQgaW50ZXJmYWNlIExha2ViYXNlU2VhcmNoT3B0aW9ucyB7XG4gIGxpbWl0PzogbnVtYmVyO1xuICBzY29yZVRocmVzaG9sZD86IG51bWJlcjtcbiAgc2lnbmFsPzogQWJvcnRTaWduYWw7XG59XG5cbi8qKiBPcHRpb25zIGZvciBwcm92aXNpb25pbmcgYSBMYWtlYmFzZS1iYWNrZWQgaW5kZXguICovXG5leHBvcnQgaW50ZXJmYWNlIExha2ViYXNlUHJvdmlzaW9uT3B0aW9ucyB7XG4gIC8qKiBUZXh0IGNvbHVtbiBlbWJlZGRlZCBpbnRvIHRoZSBzZWFyY2ggdmVjdG9yLiBEZWZhdWx0cyB0byBgdGV4dGAuICovXG4gIHRleHRDb2x1bW4/OiBzdHJpbmc7XG4gIC8qKiBEb2N1bWVudHMgdG8gc2VlZCB3aGVuIHRoZSB0YWJsZSBpcyBlbXB0eS4gKi9cbiAgc2VlZD86IFNlYXJjaERvY3VtZW50W107XG4gIHNpZ25hbD86IEFib3J0U2lnbmFsO1xufVxuXG4vKipcbiAqIEEgUG9zdGdyZXMgZnVsbC10ZXh0IGJhY2tlbmQuIE9uZSBpbnN0YW5jZSBpcyBzaGFyZWQgYWNyb3NzIGluZGV4ZXM7IGVhY2hcbiAqIGluZGV4IG1hcHMgdG8gYSB0YWJsZSB3aG9zZSBuYW1lIGlzIGRlcml2ZWQgZnJvbSB0aGUgaW5kZXggcmVmZXJlbmNlLlxuICovXG5leHBvcnQgY2xhc3MgTGFrZWJhc2VTZWFyY2hCYWNrZW5kIHtcbiAgcHJpdmF0ZSBwb29sOiBQb29sIHwgdW5kZWZpbmVkO1xuICBwcml2YXRlIHBvb2xQcm9taXNlOiBQcm9taXNlPFBvb2w+IHwgdW5kZWZpbmVkO1xuICBwcml2YXRlIHJlYWRvbmx5IHByb3Zpc2lvbmVkID0gbmV3IFNldDxzdHJpbmc+KCk7XG5cbiAgY29uc3RydWN0b3IoXG4gICAgcHJpdmF0ZSByZWFkb25seSBwZ0NvbmZpZ0ZhY3Rvcnk6ICgpID0+IFByb21pc2U8UG9vbENvbmZpZz4gfCBQb29sQ29uZmlnLFxuICAgIHByaXZhdGUgcmVhZG9ubHkgc2NoZW1hID0gXCJwdWJsaWNcIixcbiAgICAvKiogSG93IGEgcG9vbCBpcyBidWlsdCBmcm9tIHRoZSByZXNvbHZlZCBjb25maWcuIE92ZXJyaWRhYmxlIGZvciB0ZXN0cy4gKi9cbiAgICBwcml2YXRlIHJlYWRvbmx5IHBvb2xGYWN0b3J5OiAoY29uZmlnOiBQb29sQ29uZmlnKSA9PiBQb29sID0gKGNvbmZpZykgPT4gbmV3IFBvb2woY29uZmlnKSxcbiAgKSB7fVxuXG4gIC8qKiBMYXppbHkgYnVpbGQgKGFuZCBjYWNoZSkgdGhlIHBnIHBvb2wgZnJvbSB0aGUgcmVzb2x2ZWQgTGFrZWJhc2UgY29uZmlnLiAqL1xuICBwcml2YXRlIGFzeW5jIGdldFBvb2woKTogUHJvbWlzZTxQb29sPiB7XG4gICAgaWYgKHRoaXMucG9vbCkgcmV0dXJuIHRoaXMucG9vbDtcbiAgICB0aGlzLnBvb2xQcm9taXNlID8/PSAoYXN5bmMgKCkgPT4ge1xuICAgICAgY29uc3QgY29uZmlnID0gYXdhaXQgdGhpcy5wZ0NvbmZpZ0ZhY3RvcnkoKTtcbiAgICAgIGNvbnN0IHBvb2wgPSB0aGlzLnBvb2xGYWN0b3J5KGNvbmZpZyk7XG4gICAgICB0aGlzLnBvb2wgPSBwb29sO1xuICAgICAgcmV0dXJuIHBvb2w7XG4gICAgfSkoKTtcbiAgICByZXR1cm4gdGhpcy5wb29sUHJvbWlzZTtcbiAgfVxuXG4gIC8qKiBDbG9zZSB0aGUgcG9vbCBzbyBhIHJlc3RhcnRlZCBhcHAgcmVidWlsZHMgaXQuICovXG4gIGFzeW5jIGNsb3NlKCk6IFByb21pc2U8dm9pZD4ge1xuICAgIGNvbnN0IHBvb2wgPSB0aGlzLnBvb2w7XG4gICAgdGhpcy5wb29sID0gdW5kZWZpbmVkO1xuICAgIHRoaXMucG9vbFByb21pc2UgPSB1bmRlZmluZWQ7XG4gICAgdGhpcy5wcm92aXNpb25lZC5jbGVhcigpO1xuICAgIGlmIChwb29sKSBhd2FpdCBwb29sLmVuZCgpO1xuICB9XG5cbiAgLyoqXG4gICAqIFNlYXJjaCBhIExha2ViYXNlLWJhY2tlZCBpbmRleC4gUmV0dXJucyBoaXRzIHNvcnRlZCBtb3N0LXJlbGV2YW50LWZpcnN0LFxuICAgKiBzaGFwZWQgaWRlbnRpY2FsbHkgdG8gdGhlIFZlY3RvciBTZWFyY2ggYmFja2VuZC5cbiAgICovXG4gIGFzeW5jIHNlYXJjaChcbiAgICBpbmRleDogc3RyaW5nLFxuICAgIHF1ZXJ5OiBzdHJpbmcsXG4gICAgb3B0aW9uczogTGFrZWJhc2VTZWFyY2hPcHRpb25zID0ge30sXG4gICk6IFByb21pc2U8U2VhcmNoUmVzdWx0PiB7XG4gICAgY29uc3QgdGV4dCA9IHN0cmluZy50cmltVG9FbXB0eShxdWVyeSk7XG4gICAgY29uc3QgdGFibGUgPSB0aGlzLnRhYmxlRm9yKGluZGV4KTtcbiAgICBjb25zdCBsaW1pdCA9IG9wdGlvbnMubGltaXQgPz8gMTA7XG4gICAgY29uc3QgcG9vbCA9IGF3YWl0IHRoaXMuZ2V0UG9vbCgpO1xuXG4gICAgLy8gYHdlYnNlYXJjaF90b190c3F1ZXJ5YCBhY2NlcHRzIGEgYmFyZSBzZWFyY2ggYm94IHN0cmluZyAocXVvdGVkIHBocmFzZXMsXG4gICAgLy8gYG9yYCwgYC10ZXJtYCk7IGFuIGVtcHR5IHF1ZXJ5IHJldHVybnMgdGhlIG1vc3QgcmVjZW50IHJvd3Mgc28gYW4gZW1wdHlcbiAgICAvLyBzZWFyY2ggYm94IHN0aWxsIHNob3dzIGNvbnRlbnQsIG1hdGNoaW5nIGEga2V5d29yZCBpbmRleCdzIGJlaGF2aW9yLlxuICAgIGNvbnN0IHNxbCA9IHRleHRcbiAgICAgID8gYFNFTEVDVCBpZCwgZG9jdW1lbnQsIHRzX3Jhbmsoc2VhcmNoX3ZlY3Rvciwgd2Vic2VhcmNoX3RvX3RzcXVlcnkoJ2VuZ2xpc2gnLCAkMSkpIEFTIHNjb3JlXG4gICAgICAgICAgIEZST00gJHt0YWJsZX1cbiAgICAgICAgICBXSEVSRSBzZWFyY2hfdmVjdG9yIEBAIHdlYnNlYXJjaF90b190c3F1ZXJ5KCdlbmdsaXNoJywgJDEpXG4gICAgICAgICAgT1JERVIgQlkgc2NvcmUgREVTQ1xuICAgICAgICAgIExJTUlUICQyYFxuICAgICAgOiBgU0VMRUNUIGlkLCBkb2N1bWVudCwgMDo6ZmxvYXQ0IEFTIHNjb3JlIEZST00gJHt0YWJsZX0gT1JERVIgQlkgaWQgTElNSVQgJDJgO1xuICAgIGNvbnN0IHBhcmFtcyA9IHRleHQgPyBbdGV4dCwgbGltaXRdIDogW2xpbWl0XTtcblxuICAgIGNvbnN0IHsgcm93cyB9ID0gYXdhaXQgdGhpcy5xdWVyeTx7IGlkOiBzdHJpbmc7IGRvY3VtZW50OiB1bmtub3duOyBzY29yZTogbnVtYmVyIH0+KFxuICAgICAgcG9vbCxcbiAgICAgIHNxbCxcbiAgICAgIHBhcmFtcyxcbiAgICAgIG9wdGlvbnMuc2lnbmFsLFxuICAgICk7XG4gICAgY29uc3QgaGl0czogU2VhcmNoSGl0W10gPSByb3dzXG4gICAgICAubWFwKChyb3cpID0+ICh7XG4gICAgICAgIGlkOiBTdHJpbmcocm93LmlkKSxcbiAgICAgICAgc2NvcmU6IE51bWJlcihyb3cuc2NvcmUpIHx8IDAsXG4gICAgICAgIGZpZWxkczogdGhpcy50b0ZpZWxkcyhyb3cuZG9jdW1lbnQpLFxuICAgICAgfSkpXG4gICAgICAuZmlsdGVyKChoaXQpID0+IG9wdGlvbnMuc2NvcmVUaHJlc2hvbGQgPT09IHVuZGVmaW5lZCB8fCBoaXQuc2NvcmUgPj0gb3B0aW9ucy5zY29yZVRocmVzaG9sZCk7XG4gICAgcmV0dXJuIHsgcXVlcnk6IHRleHQsIGluZGV4LCBoaXRzLCBjb3VudDogaGl0cy5sZW5ndGggfTtcbiAgfVxuXG4gIC8qKlxuICAgKiBFbnN1cmUgdGhlIHRhYmxlICsgZnVsbC10ZXh0IGluZGV4IGV4aXN0IGFuZCBzZWVkIGRvY3VtZW50cyB3aGVuIGVtcHR5LlxuICAgKiBJZGVtcG90ZW50LCBzbyBpdCBpcyBzYWZlIHRvIGNhbGwgb24gZXZlcnkgYm9vdC5cbiAgICovXG4gIGFzeW5jIHByb3Zpc2lvbihpbmRleDogc3RyaW5nLCBvcHRpb25zOiBMYWtlYmFzZVByb3Zpc2lvbk9wdGlvbnMgPSB7fSk6IFByb21pc2U8bnVtYmVyPiB7XG4gICAgY29uc3QgdGFibGUgPSB0aGlzLnRhYmxlRm9yKGluZGV4KTtcbiAgICBjb25zdCBwb29sID0gYXdhaXQgdGhpcy5nZXRQb29sKCk7XG4gICAgYXdhaXQgdGhpcy5lbnN1cmVUYWJsZShwb29sLCB0YWJsZSwgb3B0aW9ucy5zaWduYWwpO1xuXG4gICAgY29uc3QgeyByb3dzIH0gPSBhd2FpdCB0aGlzLnF1ZXJ5PHsgY291bnQ6IHN0cmluZyB9PihcbiAgICAgIHBvb2wsXG4gICAgICBgU0VMRUNUIGNvdW50KCopOjp0ZXh0IEFTIGNvdW50IEZST00gJHt0YWJsZX1gLFxuICAgICAgW10sXG4gICAgICBvcHRpb25zLnNpZ25hbCxcbiAgICApO1xuICAgIGNvbnN0IGV4aXN0aW5nID0gTnVtYmVyKHJvd3NbMF0/LmNvdW50ID8/IFwiMFwiKTtcbiAgICBjb25zdCBzZWVkID0gb3B0aW9ucy5zZWVkID8/IFtdO1xuICAgIGlmIChleGlzdGluZyA9PT0gMCAmJiBzZWVkLmxlbmd0aCA+IDApIHtcbiAgICAgIGF3YWl0IHRoaXMudXBzZXJ0KHBvb2wsIHRhYmxlLCBzZWVkLCBvcHRpb25zLnRleHRDb2x1bW4gPz8gXCJ0ZXh0XCIsIG9wdGlvbnMuc2lnbmFsKTtcbiAgICAgIGxvZ2dlci5pbmZvKFwiaW5kZXgtc2VlZGVkXCIsIHsgaW5kZXgsIHRhYmxlLCBjb3VudDogc2VlZC5sZW5ndGggfSk7XG4gICAgICByZXR1cm4gc2VlZC5sZW5ndGg7XG4gICAgfVxuICAgIHJldHVybiBleGlzdGluZztcbiAgfVxuXG4gIC8qKiBBZGQgb3IgdXBkYXRlIGRvY3VtZW50cyBieSBwcmltYXJ5IGtleS4gKi9cbiAgYXN5bmMgYWRkRG9jdW1lbnRzKFxuICAgIGluZGV4OiBzdHJpbmcsXG4gICAgZG9jdW1lbnRzOiBTZWFyY2hEb2N1bWVudFtdLFxuICAgIHRleHRDb2x1bW4gPSBcInRleHRcIixcbiAgICBzaWduYWw/OiBBYm9ydFNpZ25hbCxcbiAgKTogUHJvbWlzZTxVcHNlcnRSZXN1bHQ+IHtcbiAgICBjb25zdCB0YWJsZSA9IHRoaXMudGFibGVGb3IoaW5kZXgpO1xuICAgIGNvbnN0IHBvb2wgPSBhd2FpdCB0aGlzLmdldFBvb2woKTtcbiAgICBhd2FpdCB0aGlzLmVuc3VyZVRhYmxlKHBvb2wsIHRhYmxlLCBzaWduYWwpO1xuICAgIGF3YWl0IHRoaXMudXBzZXJ0KHBvb2wsIHRhYmxlLCBkb2N1bWVudHMsIHRleHRDb2x1bW4sIHNpZ25hbCk7XG4gICAgcmV0dXJuIHsgaW5kZXgsIGNvdW50OiBkb2N1bWVudHMubGVuZ3RoIH07XG4gIH1cblxuICAvKiogRGVsZXRlIGRvY3VtZW50cyBieSBwcmltYXJ5IGtleS4gKi9cbiAgYXN5bmMgZGVsZXRlRG9jdW1lbnRzKFxuICAgIGluZGV4OiBzdHJpbmcsXG4gICAgaWRzOiBBcnJheTxzdHJpbmcgfCBudW1iZXI+LFxuICAgIHNpZ25hbD86IEFib3J0U2lnbmFsLFxuICApOiBQcm9taXNlPFVwc2VydFJlc3VsdD4ge1xuICAgIGNvbnN0IHRhYmxlID0gdGhpcy50YWJsZUZvcihpbmRleCk7XG4gICAgY29uc3QgcG9vbCA9IGF3YWl0IHRoaXMuZ2V0UG9vbCgpO1xuICAgIGF3YWl0IHRoaXMucXVlcnkocG9vbCwgYERFTEVURSBGUk9NICR7dGFibGV9IFdIRVJFIGlkID0gQU5ZKCQxKWAsIFtpZHMubWFwKFN0cmluZyldLCBzaWduYWwpO1xuICAgIHJldHVybiB7IGluZGV4LCBjb3VudDogaWRzLmxlbmd0aCB9O1xuICB9XG5cbiAgLyoqIENyZWF0ZSB0aGUgdGFibGUgKyBHSU4gaW5kZXggb25jZSBwZXIgdGFibGUgKG1lbW9pemVkIGFjcm9zcyBjYWxscykuICovXG4gIHByaXZhdGUgYXN5bmMgZW5zdXJlVGFibGUocG9vbDogUG9vbCwgdGFibGU6IHN0cmluZywgc2lnbmFsPzogQWJvcnRTaWduYWwpOiBQcm9taXNlPHZvaWQ+IHtcbiAgICBpZiAodGhpcy5wcm92aXNpb25lZC5oYXModGFibGUpKSByZXR1cm47XG4gICAgLy8gYGRvY3VtZW50YCBob2xkcyB0aGUgd2hvbGUgcm93OyBgc2VhcmNoX3RleHRgIGlzIHRoZSBpbmRleGVkIHRleHQ7IHRoZVxuICAgIC8vIGdlbmVyYXRlZCBgc2VhcmNoX3ZlY3RvcmAga2VlcHMgdGhlIHRzdmVjdG9yIGluIGxvY2tzdGVwIHdpdGggaXQgc28gYVxuICAgIC8vIHdyaXRlIG5ldmVyIGhhcyB0byBjb21wdXRlIHRoZSB2ZWN0b3IgYnkgaGFuZC5cbiAgICBhd2FpdCB0aGlzLnF1ZXJ5KHBvb2wsIGBDUkVBVEUgU0NIRU1BIElGIE5PVCBFWElTVFMgJHt0aGlzLmlkZW50KHRoaXMuc2NoZW1hKX1gLCBbXSwgc2lnbmFsKTtcbiAgICBhd2FpdCB0aGlzLnF1ZXJ5KFxuICAgICAgcG9vbCxcbiAgICAgIGBDUkVBVEUgVEFCTEUgSUYgTk9UIEVYSVNUUyAke3RhYmxlfSAoXG4gICAgICAgICBpZCB0ZXh0IFBSSU1BUlkgS0VZLFxuICAgICAgICAgc2VhcmNoX3RleHQgdGV4dCBOT1QgTlVMTCBERUZBVUxUICcnLFxuICAgICAgICAgZG9jdW1lbnQganNvbmIgTk9UIE5VTEwgREVGQVVMVCAne30nOjpqc29uYixcbiAgICAgICAgIHNlYXJjaF92ZWN0b3IgdHN2ZWN0b3IgR0VORVJBVEVEIEFMV0FZUyBBUyAodG9fdHN2ZWN0b3IoJ2VuZ2xpc2gnLCBzZWFyY2hfdGV4dCkpIFNUT1JFRFxuICAgICAgIClgLFxuICAgICAgW10sXG4gICAgICBzaWduYWwsXG4gICAgKTtcbiAgICBhd2FpdCB0aGlzLnF1ZXJ5KFxuICAgICAgcG9vbCxcbiAgICAgIGBDUkVBVEUgSU5ERVggSUYgTk9UIEVYSVNUUyAke3RoaXMuaWRlbnQoYCR7dGhpcy5iYXJlTmFtZSh0YWJsZSl9X2Z0c2ApfVxuICAgICAgICAgT04gJHt0YWJsZX0gVVNJTkcgZ2luIChzZWFyY2hfdmVjdG9yKWAsXG4gICAgICBbXSxcbiAgICAgIHNpZ25hbCxcbiAgICApO1xuICAgIHRoaXMucHJvdmlzaW9uZWQuYWRkKHRhYmxlKTtcbiAgICBsb2dnZXIuaW5mbyhcImluZGV4LWNyZWF0ZWRcIiwgeyB0YWJsZSB9KTtcbiAgfVxuXG4gIC8qKiBVcHNlcnQgcm93czogdGhlIHdob2xlIGRvY3VtZW50IGFzIGpzb25iICsgYSBmbGF0dGVuZWQgdGV4dCBibG9iIHRvIGluZGV4LiAqL1xuICBwcml2YXRlIGFzeW5jIHVwc2VydChcbiAgICBwb29sOiBQb29sLFxuICAgIHRhYmxlOiBzdHJpbmcsXG4gICAgZG9jdW1lbnRzOiBTZWFyY2hEb2N1bWVudFtdLFxuICAgIHRleHRDb2x1bW46IHN0cmluZyxcbiAgICBzaWduYWw/OiBBYm9ydFNpZ25hbCxcbiAgKTogUHJvbWlzZTx2b2lkPiB7XG4gICAgZm9yIChjb25zdCBkb2Mgb2YgZG9jdW1lbnRzKSB7XG4gICAgICBjb25zdCBpZCA9IHN0cmluZy50cmltVG9OdWxsKFN0cmluZyhkb2MuaWQgPz8gZG9jLklEID8/IFwiXCIpKSA/PyB1bmRlZmluZWQ7XG4gICAgICBpZiAoaWQgPT09IHVuZGVmaW5lZCkge1xuICAgICAgICB0aHJvdyBuZXcgRXhlY3V0aW9uRXJyb3IoXCJzZWFyY2ggKGxha2ViYXNlKTogYSBkb2N1bWVudCBpcyBtaXNzaW5nIGFuIGBpZGBcIiwge1xuICAgICAgICAgIGNvbnRleHQ6IHsgb3BlcmF0aW9uOiBcImFkZERvY3VtZW50c1wiIH0sXG4gICAgICAgIH0pO1xuICAgICAgfVxuICAgICAgY29uc3Qgc2VhcmNoVGV4dCA9IHRoaXMuc2VhcmNoVGV4dChkb2MsIHRleHRDb2x1bW4pO1xuICAgICAgYXdhaXQgdGhpcy5xdWVyeShcbiAgICAgICAgcG9vbCxcbiAgICAgICAgYElOU0VSVCBJTlRPICR7dGFibGV9IChpZCwgc2VhcmNoX3RleHQsIGRvY3VtZW50KVxuICAgICAgICAgICBWQUxVRVMgKCQxLCAkMiwgJDM6Ompzb25iKVxuICAgICAgICAgT04gQ09ORkxJQ1QgKGlkKSBETyBVUERBVEVcbiAgICAgICAgICAgU0VUIHNlYXJjaF90ZXh0ID0gRVhDTFVERUQuc2VhcmNoX3RleHQsIGRvY3VtZW50ID0gRVhDTFVERUQuZG9jdW1lbnRgLFxuICAgICAgICBbaWQsIHNlYXJjaFRleHQsIEpTT04uc3RyaW5naWZ5KGRvYyldLFxuICAgICAgICBzaWduYWwsXG4gICAgICApO1xuICAgIH1cbiAgfVxuXG4gIC8qKiBUaGUgdGV4dCBhIHJvdyBpcyBpbmRleGVkIGJ5OiB0aGUgdGV4dCBjb2x1bW4gZmlyc3QsIHRoZW4gYW55IG90aGVyIHN0cmluZyBmaWVsZC4gKi9cbiAgcHJpdmF0ZSBzZWFyY2hUZXh0KGRvYzogU2VhcmNoRG9jdW1lbnQsIHRleHRDb2x1bW46IHN0cmluZyk6IHN0cmluZyB7XG4gICAgY29uc3QgcHJpbWFyeSA9IHN0cmluZy50cmltVG9FbXB0eShTdHJpbmcoZG9jW3RleHRDb2x1bW5dID8/IFwiXCIpKTtcbiAgICBjb25zdCByZXN0ID0gT2JqZWN0LmVudHJpZXMoZG9jKVxuICAgICAgLmZpbHRlcigoW2tleSwgdmFsdWVdKSA9PiBrZXkgIT09IHRleHRDb2x1bW4gJiYga2V5ICE9PSBcImlkXCIgJiYgdHlwZW9mIHZhbHVlID09PSBcInN0cmluZ1wiKVxuICAgICAgLm1hcCgoWywgdmFsdWVdKSA9PiB2YWx1ZSBhcyBzdHJpbmcpO1xuICAgIHJldHVybiBbcHJpbWFyeSwgLi4ucmVzdF0uZmlsdGVyKEJvb2xlYW4pLmpvaW4oXCJcXG5cIik7XG4gIH1cblxuICAvKiogQSBoaXQncyBgZmllbGRzYDogdGhlIHN0b3JlZCBkb2N1bWVudCBtaW51cyB0aGUgcmVzZXJ2ZWQvaW50ZXJuYWwga2V5cy4gKi9cbiAgcHJpdmF0ZSB0b0ZpZWxkcyhkb2N1bWVudDogdW5rbm93bik6IFJlY29yZDxzdHJpbmcsIHVua25vd24+IHtcbiAgICBpZiAoIW9iamVjdC5pc1JlY29yZChkb2N1bWVudCkpIHJldHVybiB7fTtcbiAgICBjb25zdCBmaWVsZHM6IFJlY29yZDxzdHJpbmcsIHVua25vd24+ID0ge307XG4gICAgZm9yIChjb25zdCBba2V5LCB2YWx1ZV0gb2YgT2JqZWN0LmVudHJpZXMoZG9jdW1lbnQpKSB7XG4gICAgICBpZiAoUkVTRVJWRURfQ09MVU1OUy5oYXMoa2V5KSkgY29udGludWU7XG4gICAgICBmaWVsZHNba2V5XSA9IHZhbHVlO1xuICAgIH1cbiAgICByZXR1cm4gZmllbGRzO1xuICB9XG5cbiAgLyoqIFRoZSBmdWxseS1xdWFsaWZpZWQgdGFibGUgbmFtZSBmb3IgYW4gaW5kZXggcmVmZXJlbmNlLiAqL1xuICBwcml2YXRlIHRhYmxlRm9yKGluZGV4OiBzdHJpbmcpOiBzdHJpbmcge1xuICAgIHJldHVybiBgJHt0aGlzLmlkZW50KHRoaXMuc2NoZW1hKX0uJHt0aGlzLmlkZW50KHRoaXMuYmFyZU5hbWUoaW5kZXgpKX1gO1xuICB9XG5cbiAgLyoqIEEgc2FmZSBiYXJlIHRhYmxlIG5hbWUgZGVyaXZlZCBmcm9tIGFuIGluZGV4IHJlZmVyZW5jZS4gKi9cbiAgcHJpdmF0ZSBiYXJlTmFtZShyZWZlcmVuY2U6IHN0cmluZyk6IHN0cmluZyB7XG4gICAgY29uc3QgbGFzdCA9IHJlZmVyZW5jZS5zcGxpdChcIi5cIikuZmlsdGVyKEJvb2xlYW4pLnBvcCgpID8/IHJlZmVyZW5jZTtcbiAgICBjb25zdCBzbHVnID0gbGFzdFxuICAgICAgLnRvTG93ZXJDYXNlKClcbiAgICAgIC5yZXBsYWNlKC9bXmEtejAtOV9dKy9nLCBcIl9cIilcbiAgICAgIC5yZXBsYWNlKC9eXyt8XyskL2csIFwiXCIpO1xuICAgIHJldHVybiBzbHVnLmxlbmd0aCA+IDAgPyBzbHVnIDogXCJkb2N1bWVudHNcIjtcbiAgfVxuXG4gIC8qKiBRdW90ZSBhIFBvc3RncmVzIGlkZW50aWZpZXIuICovXG4gIHByaXZhdGUgaWRlbnQobmFtZTogc3RyaW5nKTogc3RyaW5nIHtcbiAgICByZXR1cm4gYFwiJHtuYW1lLnJlcGxhY2UoL1wiL2csICdcIlwiJyl9XCJgO1xuICB9XG5cbiAgLyoqIFJ1biBvbmUgcXVlcnkgdW5kZXIgZXh0ZXJuYWwgY2FuY2VsbGF0aW9uLiAqL1xuICBwcml2YXRlIGFzeW5jIHF1ZXJ5PFQ+KFxuICAgIHBvb2w6IFBvb2wsXG4gICAgc3FsOiBzdHJpbmcsXG4gICAgcGFyYW1zOiB1bmtub3duW10sXG4gICAgc2lnbmFsPzogQWJvcnRTaWduYWwsXG4gICk6IFByb21pc2U8eyByb3dzOiBUW10gfT4ge1xuICAgIGlmIChzaWduYWw/LmFib3J0ZWQpIHRocm93IEV4ZWN1dGlvbkVycm9yLmNhbmNlbGVkKCk7XG4gICAgY29uc3QgY2xpZW50ID0gYXdhaXQgcG9vbC5jb25uZWN0KCk7XG4gICAgY29uc3Qgb25BYm9ydCA9ICgpID0+IHZvaWQgY2xpZW50LnJlbGVhc2UodHJ1ZSk7XG4gICAgc2lnbmFsPy5hZGRFdmVudExpc3RlbmVyKFwiYWJvcnRcIiwgb25BYm9ydCwgeyBvbmNlOiB0cnVlIH0pO1xuICAgIHRyeSB7XG4gICAgICBjb25zdCByZXN1bHQgPSBhd2FpdCBjbGllbnQucXVlcnkoc3FsLCBwYXJhbXMpO1xuICAgICAgcmV0dXJuIHsgcm93czogcmVzdWx0LnJvd3MgYXMgVFtdIH07XG4gICAgfSBmaW5hbGx5IHtcbiAgICAgIHNpZ25hbD8ucmVtb3ZlRXZlbnRMaXN0ZW5lcihcImFib3J0XCIsIG9uQWJvcnQpO1xuICAgICAgY2xpZW50LnJlbGVhc2UoKTtcbiAgICB9XG4gIH1cbn1cbiJdfQ==

package/lib/src/plugin.d.ts

@@ -0,0 +1,216 @@
+/**
+ * AppKit plugin (registered name: `search`) that turns a Databricks AI
+ * Search (Vector Search) index into a batteries-included search surface. It is
+ * the "shortcut" half of this package: register it with nothing but an index
+ * name and you get, all at once -
+ *
+ *   - a `search` / `universal_search` / (opt-in) `add_documents` tool set for
+ *     both Mastra and AppKit agents (autocomplete is a small-`limit` search);
+ *   - HTTP routes under `/api/search` a browser search box calls directly
+ *     (`POST /` search, `POST /universal` federated, `GET /indexes` catalogue,
+ *     and `POST /documents` when writes are enabled);
+ *   - a `clientConfig()` payload so a UI knows the indexes, default, and page
+ *     size at boot with no round-trip;
+ *   - `exports()` so app code can `appkit.search.search(...)` directly.
+ *
+ * Everything runs under the caller's OBO identity (routes wrap in `asUser`),
+ * so Unity Catalog ACLs on the index apply. Registering the plugin resolves
+ * and logs the effective config (default index, known indexes, page size,
+ * mode) so a misconfiguration shows up in the boot log rather than on the
+ * first search.
+ *
+ * @module
+ */
+import { Plugin, ResourceType, type IAppRouter, type ResourceRequirement } from "@databricks/appkit";
+import { type AgentToolDefinition, type ToolProvider } from "@databricks/appkit/beta";
+import { search as searchContract } from "@dbx-tools/shared-search";
+import { type SearchPluginConfig } from "./config.ts";
+/**
+ * AppKit plugin exposing AI Search as a search box, an agent tool set, and a
+ * direct API.
+ *
+ * @example
+ * ```ts
+ * import { createApp, server } from "@databricks/appkit";
+ * import { plugin as searchPlugin } from "@dbx-tools/search";
+ *
+ * await createApp({
+ *   plugins: [
+ *     server(),
+ *     // zero-config: reads DATABRICKS_VECTOR_SEARCH_INDEX
+ *     searchPlugin.search(),
+ *     // or go deeper:
+ *     // searchPlugin.search({
+ *     //   index: "main.support.docs",
+ *     //   indexes: ["main.support.docs", "main.catalog.products"],
+ *     //   columns: ["title", "url", "body"],
+ *     //   mode: "hybrid",
+ *     // }),
+ *   ],
+ * });
+ * ```
+ */
+export declare class SearchPlugin extends Plugin<SearchPluginConfig> implements ToolProvider {
+    static manifest: {
+        name: "search";
+        displayName: string;
+        description: string;
+        stability: "beta";
+        resources: {
+            required: never[];
+            optional: {
+                type: ResourceType.VECTOR_SEARCH_INDEX;
+                alias: string;
+                resourceKey: string;
+                description: string;
+                permission: "SELECT";
+                fields: {
+                    name: {
+                        env: string;
+                        description: string;
+                        discovery: {
+                            type: "cli";
+                            cliCommand: string;
+                            selectField: string;
+                        };
+                    };
+                };
+            }[];
+        };
+        config: {
+            schema: import("json-schema").JSONSchema7;
+        };
+    };
+    /** The base path AppKit mounts this plugin's routes under. */
+    private get basePath();
+    /**
+     * Promote the index to a required resource once a deployment pins a default,
+     * through plugin config or either environment name.
+     */
+    static getResourceRequirements(config: SearchPluginConfig): ResourceRequirement[];
+    /**
+     * The tools this plugin offers to an AppKit agent. `search` /
+     * `universal_search` are reads; `add_documents` / `create_index` /
+     * `sync_index` are only offered when the write surface is enabled. None is
+     * autoInheritable: every tool runs under the caller's identity, so it must be
+     * granted explicitly.
+     */
+    private get tools();
+    /** Prime the shared runtime from config and log the effective policy at boot. */
+    setup(): Promise<void>;
+    /**
+     * Build the Lakebase FALLBACK backend, or return `undefined` when Vector
+     * Search should be used. Falls back only when there is NO Vector Search
+     * endpoint configured (plugin config or `SEARCH_ENDPOINT` /
+     * `DATABRICKS_VECTOR_SEARCH_ENDPOINT`) AND the sibling AppKit `lakebase`
+     * plugin is registered. The pg pool is built from that plugin's
+     * service-principal config exactly like `@dbx-tools/appkit-mastra`'s memory
+     * pool - no auth is re-implemented here.
+     */
+    private resolveLakebaseBackend;
+    /**
+     * Ensure the endpoint + index exist and seed them, honoring `ensureOnSetup`.
+     * Uses boot-time SDK auth (env / config profile) via the client's
+     * out-of-request fallback. Failures are logged, never thrown - a search app
+     * should still start even if provisioning is slow or a permission is missing.
+     */
+    private runEnsureOnSetup;
+    /** Build a Vector Search `schema_json` map from a seed document's value types. */
+    private inferSchema;
+    /** Drop the shared runtime so a restarted app re-resolves config. */
+    shutdown(): Promise<void>;
+    abortActiveOperations(): void;
+    /**
+     * Mount the search routes under `/api/search`. Each is wrapped in
+     * `asUser(req)` so the query runs as the requesting user and the index's
+     * Unity Catalog ACLs apply. `GET /indexes` needs no user scope - it just
+     * echoes the configured catalogue.
+     */
+    injectRoutes(router: IAppRouter): void;
+    /**
+     * Run a route body, turning a failure into a JSON error response.
+     *
+     * AppKit's `route()` registers the handler as-is, so a rejection escapes as
+     * an unhandled promise rejection and takes the whole process down rather than
+     * failing the one request. Every route here goes through this, so a bad index
+     * name or a Vector Search API error returns 500 with a message instead of
+     * killing the server.
+     */
+    private respond;
+    /** Surface the index catalogue + defaults so a search box needs no round-trip. */
+    clientConfig(): Record<string, unknown>;
+    /** AppKit `ToolProvider`: the tool definitions offered to an agent. */
+    getAgentTools(): AgentToolDefinition[];
+    /** AppKit `ToolProvider`: run one tool call, validating input against its schema. */
+    executeAgentTool(name: string, args: unknown, signal?: AbortSignal): Promise<unknown>;
+    exports(): {
+        /** Search one index (default when omitted). Runs as the current context user. */
+        search: (request: searchContract.SearchRequest, signal?: AbortSignal) => Promise<{
+            query: string;
+            hits: {
+                id: string;
+                score: number;
+                fields: Record<string, unknown>;
+                index?: string | undefined;
+            }[];
+            count: number;
+            index?: string | undefined;
+        }>;
+        /** Search across every configured index and merge the hits. */
+        universalSearch: (request: searchContract.UniversalSearchRequest, signal?: AbortSignal) => Promise<{
+            query: string;
+            hits: {
+                id: string;
+                score: number;
+                fields: Record<string, unknown>;
+                index?: string | undefined;
+            }[];
+            count: number;
+            index?: string | undefined;
+        }>;
+        /** Add or update documents in a direct-access index (throws when writes are disabled). */
+        addDocuments: (request: {
+            index?: string;
+            documents: unknown;
+        }, signal?: AbortSignal) => Promise<{
+            index: string;
+            count: number;
+        }>;
+        /** Create a Vector Search index (throws when writes are disabled). */
+        createIndex: (request: searchContract.CreateIndexRequest, signal?: AbortSignal) => Promise<import("./client.ts").IndexInfo>;
+        /** Sync a Delta Sync index from its source table (throws when writes are disabled). */
+        syncIndex: (request: searchContract.SyncIndexRequest, signal?: AbortSignal) => Promise<{
+            index: string;
+            synced: boolean;
+        }>;
+    };
+    private runSearch;
+    private runUniversalSearch;
+    private runAddDocuments;
+    private runCreateIndex;
+    private runSyncIndex;
+}
+/**
+ * Register the AI Search plugin with AppKit.
+ *
+ * @example
+ * ```ts
+ * import { createApp, server } from "@databricks/appkit";
+ * import { plugin as searchPlugin, tool as searchToolModule } from "@dbx-tools/search";
+ * import { agents, plugin as mastraPlugin } from "@dbx-tools/appkit-mastra";
+ *
+ * const support = agents.createAgent({
+ *   instructions: "Answer from the docs; use `search` to find them.",
+ *   tools: () => ({ search: searchToolModule.searchTool() }),
+ * });
+ *
+ * await createApp({
+ *   plugins: [
+ *     server(),
+ *     searchPlugin.search({ index: "main.support.docs" }),
+ *     mastraPlugin.mastra({ agents: support }),
+ *   ],
+ * });
+ * ```
+ */
+export declare const search: import("@databricks/appkit").ToPlugin<typeof SearchPlugin, SearchPluginConfig, "search">;