@dbx-tools/search 0.6.108 → 0.6.112

package/lib/src/native.js

@@ -0,0 +1,79 @@
+/**
+ * Adapter from AppKit's native AI Search provider to the dbx-tools extension
+ * contract used by federated search and agent tools.
+ *
+ * @module
+ */
+import { ValidationError } from "@databricks/appkit";
+import { search as sharedSearch, } from "@dbx-tools/shared-search";
+import { string } from "@dbx-tools/shared-core";
+import { indexConfigFor } from "./config.js";
+function filters(value) {
+    if (!value)
+        return undefined;
+    const result = {};
+    for (const [key, item] of Object.entries(value)) {
+        if (typeof item === "string" ||
+            typeof item === "number" ||
+            typeof item === "boolean" ||
+            (Array.isArray(item) &&
+                item.every((entry) => typeof entry === "string" || typeof entry === "number"))) {
+            result[key] = item;
+            continue;
+        }
+        throw new ValidationError(`AI Search filter "${key}" must be a string, number, boolean, or string/number array`);
+    }
+    return result;
+}
+function hitId(data, primaryKey, index) {
+    const value = data[primaryKey ?? "id"] ?? Object.values(data)[0] ?? index;
+    return String(value);
+}
+function aliasFor(index, config) {
+    const alias = indexConfigFor(config, index)?.alias ?? string.trimToNull(index);
+    if (!alias)
+        throw new ValidationError("AI Search requires a configured index alias");
+    return alias;
+}
+/** Build a read backend that delegates every Vector Search query to AppKit. */
+export function nativeAiSearchBackend(provider, config) {
+    const backend = {
+        supportsLifecycle: provider.providerKind !== "lakebase",
+        async search(index, query, options = {}) {
+            options.signal?.throwIfAborted();
+            const known = indexConfigFor(config, index);
+            const alias = aliasFor(index, config);
+            const resolvedQueryType = sharedSearch.toAiSearchQueryType(options.mode);
+            const resolvedFilters = filters(options.filter);
+            const response = await provider.query(alias, {
+                queryText: query,
+                numResults: options.limit ?? config.pageSize,
+                ...(options.columns ? { columns: [...options.columns] } : {}),
+                ...(resolvedQueryType ? { queryType: resolvedQueryType } : {}),
+                ...(resolvedFilters ? { filters: resolvedFilters } : {}),
+            });
+            options.signal?.throwIfAborted();
+            const hits = response.results
+                .map((result, resultIndex) => ({
+                id: hitId(result.data, known?.primaryKey, resultIndex),
+                score: result.score,
+                fields: result.data,
+            }))
+                .filter((hit) => options.scoreThreshold === undefined || hit.score >= options.scoreThreshold);
+            return { query, index, hits, count: hits.length };
+        },
+    };
+    const addDocuments = provider.addDocuments?.bind(provider);
+    if (!addDocuments)
+        return backend;
+    return {
+        ...backend,
+        async addDocuments(index, documents, signal) {
+            signal?.throwIfAborted();
+            const result = await addDocuments(aliasFor(index, config), documents);
+            signal?.throwIfAborted();
+            return result;
+        },
+    };
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoibmF0aXZlLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL25hdGl2ZS50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQTs7Ozs7R0FLRztBQUVILE9BQU8sRUFBRSxlQUFlLEVBQUUsTUFBTSxvQkFBb0IsQ0FBQztBQU1yRCxPQUFPLEVBQ0wsTUFBTSxJQUFJLFlBQVksR0FNdkIsTUFBTSwwQkFBMEIsQ0FBQztBQUNsQyxPQUFPLEVBQUUsTUFBTSxFQUFFLE1BQU0sd0JBQXdCLENBQUM7QUFFaEQsT0FBTyxFQUFFLGNBQWMsRUFBNkIsTUFBTSxhQUFhLENBQUM7QUFTeEUsU0FBUyxPQUFPLENBQUMsS0FBdUM7SUFDdEQsSUFBSSxDQUFDLEtBQUs7UUFBRSxPQUFPLFNBQVMsQ0FBQztJQUM3QixNQUFNLE1BQU0sR0FBd0IsRUFBRSxDQUFDO0lBQ3ZDLEtBQUssTUFBTSxDQUFDLEdBQUcsRUFBRSxJQUFJLENBQUMsSUFBSSxNQUFNLENBQUMsT0FBTyxDQUFDLEtBQUssQ0FBQyxFQUFFLENBQUM7UUFDaEQsSUFDRSxPQUFPLElBQUksS0FBSyxRQUFRO1lBQ3hCLE9BQU8sSUFBSSxLQUFLLFFBQVE7WUFDeEIsT0FBTyxJQUFJLEtBQUssU0FBUztZQUN6QixDQUFDLEtBQUssQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDO2dCQUNsQixJQUFJLENBQUMsS0FBSyxDQUFDLENBQUMsS0FBSyxFQUFFLEVBQUUsQ0FBQyxPQUFPLEtBQUssS0FBSyxRQUFRLElBQUksT0FBTyxLQUFLLEtBQUssUUFBUSxDQUFDLENBQUMsRUFDaEYsQ0FBQztZQUNELE1BQU0sQ0FBQyxHQUFHLENBQUMsR0FBRyxJQUFJLENBQUM7WUFDbkIsU0FBUztRQUNYLENBQUM7UUFDRCxNQUFNLElBQUksZUFBZSxDQUN2QixxQkFBcUIsR0FBRyw2REFBNkQsQ0FDdEYsQ0FBQztJQUNKLENBQUM7SUFDRCxPQUFPLE1BQU0sQ0FBQztBQUNoQixDQUFDO0FBRUQsU0FBUyxLQUFLLENBQ1osSUFBNkIsRUFDN0IsVUFBOEIsRUFDOUIsS0FBYTtJQUViLE1BQU0sS0FBSyxHQUFHLElBQUksQ0FBQyxVQUFVLElBQUksSUFBSSxDQUFDLElBQUksTUFBTSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsQ0FBQyxDQUFDLENBQUMsSUFBSSxLQUFLLENBQUM7SUFDMUUsT0FBTyxNQUFNLENBQUMsS0FBSyxDQUFDLENBQUM7QUFDdkIsQ0FBQztBQUVELFNBQVMsUUFBUSxDQUFDLEtBQWEsRUFBRSxNQUE0QjtJQUMzRCxNQUFNLEtBQUssR0FBRyxjQUFjLENBQUMsTUFBTSxFQUFFLEtBQUssQ0FBQyxFQUFFLEtBQUssSUFBSSxNQUFNLENBQUMsVUFBVSxDQUFDLEtBQUssQ0FBQyxDQUFDO0lBQy9FLElBQUksQ0FBQyxLQUFLO1FBQUUsTUFBTSxJQUFJLGVBQWUsQ0FBQyw2Q0FBNkMsQ0FBQyxDQUFDO0lBQ3JGLE9BQU8sS0FBSyxDQUFDO0FBQ2YsQ0FBQztBQUVELCtFQUErRTtBQUMvRSxNQUFNLFVBQVUscUJBQXFCLENBQ25DLFFBQTBCLEVBQzFCLE1BQTRCO0lBRTVCLE1BQU0sT0FBTyxHQUFzQjtRQUNqQyxpQkFBaUIsRUFBRSxRQUFRLENBQUMsWUFBWSxLQUFLLFVBQVU7UUFDdkQsS0FBSyxDQUFDLE1BQU0sQ0FBQyxLQUFhLEVBQUUsS0FBYSxFQUFFLFVBQXlCLEVBQUU7WUFDcEUsT0FBTyxDQUFDLE1BQU0sRUFBRSxjQUFjLEVBQUUsQ0FBQztZQUNqQyxNQUFNLEtBQUssR0FBRyxjQUFjLENBQUMsTUFBTSxFQUFFLEtBQUssQ0FBQyxDQUFDO1lBQzVDLE1BQU0sS0FBSyxHQUFHLFFBQVEsQ0FBQyxLQUFLLEVBQUUsTUFBTSxDQUFDLENBQUM7WUFDdEMsTUFBTSxpQkFBaUIsR0FBRyxZQUFZLENBQUMsbUJBQW1CLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFDO1lBQ3pFLE1BQU0sZUFBZSxHQUFHLE9BQU8sQ0FBQyxPQUFPLENBQUMsTUFBTSxDQUFDLENBQUM7WUFDaEQsTUFBTSxRQUFRLEdBQUcsTUFBTSxRQUFRLENBQUMsS0FBSyxDQUFDLEtBQUssRUFBRTtnQkFDM0MsU0FBUyxFQUFFLEtBQUs7Z0JBQ2hCLFVBQVUsRUFBRSxPQUFPLENBQUMsS0FBSyxJQUFJLE1BQU0sQ0FBQyxRQUFRO2dCQUM1QyxHQUFHLENBQUMsT0FBTyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsRUFBRSxPQUFPLEVBQUUsQ0FBQyxHQUFHLE9BQU8sQ0FBQyxPQUFPLENBQUMsRUFBRSxDQUFDLENBQUMsQ0FBQyxFQUFFLENBQUM7Z0JBQzdELEdBQUcsQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDLENBQUMsRUFBRSxTQUFTLEVBQUUsaUJBQWlCLEVBQUUsQ0FBQyxDQUFDLENBQUMsRUFBRSxDQUFDO2dCQUM5RCxHQUFHLENBQUMsZUFBZSxDQUFDLENBQUMsQ0FBQyxFQUFFLE9BQU8sRUFBRSxlQUFlLEVBQUUsQ0FBQyxDQUFDLENBQUMsRUFBRSxDQUFDO2FBQ3pELENBQUMsQ0FBQztZQUNILE9BQU8sQ0FBQyxNQUFNLEVBQUUsY0FBYyxFQUFFLENBQUM7WUFDakMsTUFBTSxJQUFJLEdBQWdCLFFBQVEsQ0FBQyxPQUFPO2lCQUN2QyxHQUFHLENBQUMsQ0FBQyxNQUFNLEVBQUUsV0FBVyxFQUFFLEVBQUUsQ0FBQyxDQUFDO2dCQUM3QixFQUFFLEVBQUUsS0FBSyxDQUFDLE1BQU0sQ0FBQyxJQUFJLEVBQUUsS0FBSyxFQUFFLFVBQVUsRUFBRSxXQUFXLENBQUM7Z0JBQ3RELEtBQUssRUFBRSxNQUFNLENBQUMsS0FBSztnQkFDbkIsTUFBTSxFQUFFLE1BQU0sQ0FBQyxJQUFJO2FBQ3BCLENBQUMsQ0FBQztpQkFDRixNQUFNLENBQ0wsQ0FBQyxHQUFHLEVBQUUsRUFBRSxDQUFDLE9BQU8sQ0FBQyxjQUFjLEtBQUssU0FBUyxJQUFJLEdBQUcsQ0FBQyxLQUFLLElBQUksT0FBTyxDQUFDLGNBQWMsQ0FDckYsQ0FBQztZQUNKLE9BQU8sRUFBRSxLQUFLLEVBQUUsS0FBSyxFQUFFLElBQUksRUFBRSxLQUFLLEVBQUUsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDO1FBQ3BELENBQUM7S0FDRixDQUFDO0lBQ0YsTUFBTSxZQUFZLEdBQUcsUUFBUSxDQUFDLFlBQVksRUFBRSxJQUFJLENBQUMsUUFBUSxDQUFDLENBQUM7SUFDM0QsSUFBSSxDQUFDLFlBQVk7UUFBRSxPQUFPLE9BQU8sQ0FBQztJQUNsQyxPQUFPO1FBQ0wsR0FBRyxPQUFPO1FBQ1YsS0FBSyxDQUFDLFlBQVksQ0FDaEIsS0FBYSxFQUNiLFNBQTJCLEVBQzNCLE1BQW9CO1lBRXBCLE1BQU0sRUFBRSxjQUFjLEVBQUUsQ0FBQztZQUN6QixNQUFNLE1BQU0sR0FBRyxNQUFNLFlBQVksQ0FBQyxRQUFRLENBQUMsS0FBSyxFQUFFLE1BQU0sQ0FBQyxFQUFFLFNBQVMsQ0FBQyxDQUFDO1lBQ3RFLE1BQU0sRUFBRSxjQUFjLEVBQUUsQ0FBQztZQUN6QixPQUFPLE1BQU0sQ0FBQztRQUNoQixDQUFDO0tBQ0YsQ0FBQztBQUNKLENBQUMiLCJzb3VyY2VzQ29udGVudCI6WyIvKipcbiAqIEFkYXB0ZXIgZnJvbSBBcHBLaXQncyBuYXRpdmUgQUkgU2VhcmNoIHByb3ZpZGVyIHRvIHRoZSBkYngtdG9vbHMgZXh0ZW5zaW9uXG4gKiBjb250cmFjdCB1c2VkIGJ5IGZlZGVyYXRlZCBzZWFyY2ggYW5kIGFnZW50IHRvb2xzLlxuICpcbiAqIEBtb2R1bGVcbiAqL1xuXG5pbXBvcnQgeyBWYWxpZGF0aW9uRXJyb3IgfSBmcm9tIFwiQGRhdGFicmlja3MvYXBwa2l0XCI7XG5pbXBvcnQgdHlwZSB7XG4gIFNlYXJjaEZpbHRlcnMgYXMgQXBwS2l0U2VhcmNoRmlsdGVycyxcbiAgU2VhcmNoUmVxdWVzdCBhcyBBcHBLaXRTZWFyY2hSZXF1ZXN0LFxuICBTZWFyY2hSZXNwb25zZSBhcyBBcHBLaXRTZWFyY2hSZXNwb25zZSxcbn0gZnJvbSBcIkBkYXRhYnJpY2tzL2FwcGtpdC9iZXRhXCI7XG5pbXBvcnQge1xuICBzZWFyY2ggYXMgc2hhcmVkU2VhcmNoLFxuICB0eXBlIFNlYXJjaERvY3VtZW50LFxuICB0eXBlIFNlYXJjaEhpdCxcbiAgdHlwZSBTZWFyY2hSZXF1ZXN0IGFzIEV4dGVuc2lvblNlYXJjaFJlcXVlc3QsXG4gIHR5cGUgU2VhcmNoUmVzdWx0LFxuICB0eXBlIFVwc2VydFJlc3VsdCxcbn0gZnJvbSBcIkBkYngtdG9vbHMvc2hhcmVkLXNlYXJjaFwiO1xuaW1wb3J0IHsgc3RyaW5nIH0gZnJvbSBcIkBkYngtdG9vbHMvc2hhcmVkLWNvcmVcIjtcbmltcG9ydCB0eXBlIHsgU2VhcmNoT3B0aW9ucywgU2VhcmNoUmVhZEJhY2tlbmQgfSBmcm9tIFwiLi9jbGllbnQudHNcIjtcbmltcG9ydCB7IGluZGV4Q29uZmlnRm9yLCB0eXBlIFJlc29sdmVkU2VhcmNoQ29uZmlnIH0gZnJvbSBcIi4vY29uZmlnLnRzXCI7XG5cbi8qKiBNaW5pbWFsIHF1ZXJ5IHN1cmZhY2UgZXhwb3NlZCBieSBBcHBLaXQncyBuYXRpdmUgYGFpU2VhcmNoYCBwbHVnaW4uICovXG5leHBvcnQgaW50ZXJmYWNlIEFpU2VhcmNoUHJvdmlkZXIge1xuICBwcm92aWRlcktpbmQ/OiBcImxha2ViYXNlXCI7XG4gIHF1ZXJ5KGFsaWFzOiBzdHJpbmcsIHJlcXVlc3Q6IEFwcEtpdFNlYXJjaFJlcXVlc3QpOiBQcm9taXNlPEFwcEtpdFNlYXJjaFJlc3BvbnNlPjtcbiAgYWRkRG9jdW1lbnRzPyhhbGlhczogc3RyaW5nLCBkb2N1bWVudHM6IFNlYXJjaERvY3VtZW50W10pOiBQcm9taXNlPFVwc2VydFJlc3VsdD47XG59XG5cbmZ1bmN0aW9uIGZpbHRlcnModmFsdWU6IEV4dGVuc2lvblNlYXJjaFJlcXVlc3RbXCJmaWx0ZXJcIl0pOiBBcHBLaXRTZWFyY2hGaWx0ZXJzIHwgdW5kZWZpbmVkIHtcbiAgaWYgKCF2YWx1ZSkgcmV0dXJuIHVuZGVmaW5lZDtcbiAgY29uc3QgcmVzdWx0OiBBcHBLaXRTZWFyY2hGaWx0ZXJzID0ge307XG4gIGZvciAoY29uc3QgW2tleSwgaXRlbV0gb2YgT2JqZWN0LmVudHJpZXModmFsdWUpKSB7XG4gICAgaWYgKFxuICAgICAgdHlwZW9mIGl0ZW0gPT09IFwic3RyaW5nXCIgfHxcbiAgICAgIHR5cGVvZiBpdGVtID09PSBcIm51bWJlclwiIHx8XG4gICAgICB0eXBlb2YgaXRlbSA9PT0gXCJib29sZWFuXCIgfHxcbiAgICAgIChBcnJheS5pc0FycmF5KGl0ZW0pICYmXG4gICAgICAgIGl0ZW0uZXZlcnkoKGVudHJ5KSA9PiB0eXBlb2YgZW50cnkgPT09IFwic3RyaW5nXCIgfHwgdHlwZW9mIGVudHJ5ID09PSBcIm51bWJlclwiKSlcbiAgICApIHtcbiAgICAgIHJlc3VsdFtrZXldID0gaXRlbTtcbiAgICAgIGNvbnRpbnVlO1xuICAgIH1cbiAgICB0aHJvdyBuZXcgVmFsaWRhdGlvbkVycm9yKFxuICAgICAgYEFJIFNlYXJjaCBmaWx0ZXIgXCIke2tleX1cIiBtdXN0IGJlIGEgc3RyaW5nLCBudW1iZXIsIGJvb2xlYW4sIG9yIHN0cmluZy9udW1iZXIgYXJyYXlgLFxuICAgICk7XG4gIH1cbiAgcmV0dXJuIHJlc3VsdDtcbn1cblxuZnVuY3Rpb24gaGl0SWQoXG4gIGRhdGE6IFJlY29yZDxzdHJpbmcsIHVua25vd24+LFxuICBwcmltYXJ5S2V5OiBzdHJpbmcgfCB1bmRlZmluZWQsXG4gIGluZGV4OiBudW1iZXIsXG4pOiBzdHJpbmcge1xuICBjb25zdCB2YWx1ZSA9IGRhdGFbcHJpbWFyeUtleSA/PyBcImlkXCJdID8/IE9iamVjdC52YWx1ZXMoZGF0YSlbMF0gPz8gaW5kZXg7XG4gIHJldHVybiBTdHJpbmcodmFsdWUpO1xufVxuXG5mdW5jdGlvbiBhbGlhc0ZvcihpbmRleDogc3RyaW5nLCBjb25maWc6IFJlc29sdmVkU2VhcmNoQ29uZmlnKTogc3RyaW5nIHtcbiAgY29uc3QgYWxpYXMgPSBpbmRleENvbmZpZ0Zvcihjb25maWcsIGluZGV4KT8uYWxpYXMgPz8gc3RyaW5nLnRyaW1Ub051bGwoaW5kZXgpO1xuICBpZiAoIWFsaWFzKSB0aHJvdyBuZXcgVmFsaWRhdGlvbkVycm9yKFwiQUkgU2VhcmNoIHJlcXVpcmVzIGEgY29uZmlndXJlZCBpbmRleCBhbGlhc1wiKTtcbiAgcmV0dXJuIGFsaWFzO1xufVxuXG4vKiogQnVpbGQgYSByZWFkIGJhY2tlbmQgdGhhdCBkZWxlZ2F0ZXMgZXZlcnkgVmVjdG9yIFNlYXJjaCBxdWVyeSB0byBBcHBLaXQuICovXG5leHBvcnQgZnVuY3Rpb24gbmF0aXZlQWlTZWFyY2hCYWNrZW5kKFxuICBwcm92aWRlcjogQWlTZWFyY2hQcm92aWRlcixcbiAgY29uZmlnOiBSZXNvbHZlZFNlYXJjaENvbmZpZyxcbik6IFNlYXJjaFJlYWRCYWNrZW5kIHtcbiAgY29uc3QgYmFja2VuZDogU2VhcmNoUmVhZEJhY2tlbmQgPSB7XG4gICAgc3VwcG9ydHNMaWZlY3ljbGU6IHByb3ZpZGVyLnByb3ZpZGVyS2luZCAhPT0gXCJsYWtlYmFzZVwiLFxuICAgIGFzeW5jIHNlYXJjaChpbmRleDogc3RyaW5nLCBxdWVyeTogc3RyaW5nLCBvcHRpb25zOiBTZWFyY2hPcHRpb25zID0ge30pOiBQcm9taXNlPFNlYXJjaFJlc3VsdD4ge1xuICAgICAgb3B0aW9ucy5zaWduYWw/LnRocm93SWZBYm9ydGVkKCk7XG4gICAgICBjb25zdCBrbm93biA9IGluZGV4Q29uZmlnRm9yKGNvbmZpZywgaW5kZXgpO1xuICAgICAgY29uc3QgYWxpYXMgPSBhbGlhc0ZvcihpbmRleCwgY29uZmlnKTtcbiAgICAgIGNvbnN0IHJlc29sdmVkUXVlcnlUeXBlID0gc2hhcmVkU2VhcmNoLnRvQWlTZWFyY2hRdWVyeVR5cGUob3B0aW9ucy5tb2RlKTtcbiAgICAgIGNvbnN0IHJlc29sdmVkRmlsdGVycyA9IGZpbHRlcnMob3B0aW9ucy5maWx0ZXIpO1xuICAgICAgY29uc3QgcmVzcG9uc2UgPSBhd2FpdCBwcm92aWRlci5xdWVyeShhbGlhcywge1xuICAgICAgICBxdWVyeVRleHQ6IHF1ZXJ5LFxuICAgICAgICBudW1SZXN1bHRzOiBvcHRpb25zLmxpbWl0ID8/IGNvbmZpZy5wYWdlU2l6ZSxcbiAgICAgICAgLi4uKG9wdGlvbnMuY29sdW1ucyA/IHsgY29sdW1uczogWy4uLm9wdGlvbnMuY29sdW1uc10gfSA6IHt9KSxcbiAgICAgICAgLi4uKHJlc29sdmVkUXVlcnlUeXBlID8geyBxdWVyeVR5cGU6IHJlc29sdmVkUXVlcnlUeXBlIH0gOiB7fSksXG4gICAgICAgIC4uLihyZXNvbHZlZEZpbHRlcnMgPyB7IGZpbHRlcnM6IHJlc29sdmVkRmlsdGVycyB9IDoge30pLFxuICAgICAgfSk7XG4gICAgICBvcHRpb25zLnNpZ25hbD8udGhyb3dJZkFib3J0ZWQoKTtcbiAgICAgIGNvbnN0IGhpdHM6IFNlYXJjaEhpdFtdID0gcmVzcG9uc2UucmVzdWx0c1xuICAgICAgICAubWFwKChyZXN1bHQsIHJlc3VsdEluZGV4KSA9PiAoe1xuICAgICAgICAgIGlkOiBoaXRJZChyZXN1bHQuZGF0YSwga25vd24/LnByaW1hcnlLZXksIHJlc3VsdEluZGV4KSxcbiAgICAgICAgICBzY29yZTogcmVzdWx0LnNjb3JlLFxuICAgICAgICAgIGZpZWxkczogcmVzdWx0LmRhdGEsXG4gICAgICAgIH0pKVxuICAgICAgICAuZmlsdGVyKFxuICAgICAgICAgIChoaXQpID0+IG9wdGlvbnMuc2NvcmVUaHJlc2hvbGQgPT09IHVuZGVmaW5lZCB8fCBoaXQuc2NvcmUgPj0gb3B0aW9ucy5zY29yZVRocmVzaG9sZCxcbiAgICAgICAgKTtcbiAgICAgIHJldHVybiB7IHF1ZXJ5LCBpbmRleCwgaGl0cywgY291bnQ6IGhpdHMubGVuZ3RoIH07XG4gICAgfSxcbiAgfTtcbiAgY29uc3QgYWRkRG9jdW1lbnRzID0gcHJvdmlkZXIuYWRkRG9jdW1lbnRzPy5iaW5kKHByb3ZpZGVyKTtcbiAgaWYgKCFhZGREb2N1bWVudHMpIHJldHVybiBiYWNrZW5kO1xuICByZXR1cm4ge1xuICAgIC4uLmJhY2tlbmQsXG4gICAgYXN5bmMgYWRkRG9jdW1lbnRzKFxuICAgICAgaW5kZXg6IHN0cmluZyxcbiAgICAgIGRvY3VtZW50czogU2VhcmNoRG9jdW1lbnRbXSxcbiAgICAgIHNpZ25hbD86IEFib3J0U2lnbmFsLFxuICAgICk6IFByb21pc2U8VXBzZXJ0UmVzdWx0PiB7XG4gICAgICBzaWduYWw/LnRocm93SWZBYm9ydGVkKCk7XG4gICAgICBjb25zdCByZXN1bHQgPSBhd2FpdCBhZGREb2N1bWVudHMoYWxpYXNGb3IoaW5kZXgsIGNvbmZpZyksIGRvY3VtZW50cyk7XG4gICAgICBzaWduYWw/LnRocm93SWZBYm9ydGVkKCk7XG4gICAgICByZXR1cm4gcmVzdWx0O1xuICAgIH0sXG4gIH07XG59XG4iXX0=

package/lib/src/plugin.d.ts

@@ -1,51 +1,55 @@
 /**
- * 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 -
+ * AppKit extension plugin (registered name: `search`) around a sibling
+ * `aiSearch` provider. Native AppKit owns Vector Search query execution; this
+ * plugin adds the surfaces that remain useful across native and Lakebase
+ * providers:
  *
  *   - 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
+ *   - compatibility and extension routes under `/api/search`
  *     (`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.
+ * Vector Search reads delegate to the registered native AppKit `aiSearch`
+ * plugin, preserving its OBO, caching, reranking, and response handling. The
+ * alternative `lakebaseAiSearch` provider supplies the same query contract
+ * over PostgreSQL full-text search.
  *
  * @module
  */
-import { Plugin, ResourceType, type IAppRouter, type ResourceRequirement } from "@databricks/appkit";
+import { Plugin, type IAppRouter } from "@databricks/appkit";
 import { type AgentToolDefinition, type ToolProvider } from "@databricks/appkit/beta";
 import { search as sharedSearch } 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.
+ * AppKit extension plugin for agent tools, federated search, and optional
+ * Vector Search lifecycle operations. Register `aiSearch()` or
+ * `lakebaseAiSearch()` before this plugin.
  *
  * @example
  * ```ts
  * import { createApp, server } from "@databricks/appkit";
+ * import { aiSearch } from "@databricks/appkit/beta";
  * 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",
- *     // }),
+ *     aiSearch({
+ *       indexes: {
+ *         docs: {
+ *           indexName: "main.support.docs",
+ *           columns: ["id", "title", "body"],
+ *         },
+ *       },
+ *     }),
+ *     searchPlugin.search({
+ *       index: "main.support.docs",
+ *       indexes: [{ name: "main.support.docs", alias: "docs" }],
+ *     }),
  *   ],
  * });
  * ```
@@ -58,24 +62,7 @@ export declare class SearchPlugin extends Plugin<SearchPluginConfig> implements
         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;
-                        };
-                    };
-                };
-            }[];
+            optional: never[];
         };
         config: {
             schema: import("json-schema").JSONSchema7;
@@ -83,11 +70,6 @@ export declare class SearchPlugin extends Plugin<SearchPluginConfig> implements
     };
     /** 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` /
@@ -98,16 +80,8 @@ export declare class SearchPlugin extends Plugin<SearchPluginConfig> implements
     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;
+    /** Resolve the registered AppKit-compatible AI Search provider. */
+    private resolveProviderBackend;
     /**
      * Ensure the endpoint + index exist and seed them, honoring `ensureOnSetup`.
      * Uses boot-time SDK auth (env / config profile) via the client's
@@ -121,10 +95,8 @@ export declare class SearchPlugin extends Plugin<SearchPluginConfig> implements
     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.
+     * Mount extension routes under `/api/search`. Single-index queries stay on
+     * the provider's native `/api/ai-search/:alias/query` surface.
      */
     injectRoutes(router: IAppRouter): void;
     /**