@databricks/appkit 0.54.0 → 0.55.1

package/dist/plugins/ai-search/ai-search.js

@@ -0,0 +1,296 @@
+import { createLogger } from "../../logging/logger.js";
+import { getWorkspaceClient } from "../../context/execution-context.js";
+import { formatWarningBanner } from "../../utils/banner.js";
+import "../../context/index.js";
+import { Plugin } from "../../plugin/plugin.js";
+import { toPlugin } from "../../plugin/to-plugin.js";
+import "../../plugin/index.js";
+import { AiSearchConnector } from "../../connectors/ai-search/client.js";
+import { aiSearchDefaults } from "./defaults.js";
+import manifest_default from "./manifest.js";
+
+//#region src/plugins/ai-search/ai-search.ts
+const logger = createLogger("ai-search");
+const querySettings = { default: aiSearchDefaults };
+var AiSearchPlugin = class extends Plugin {
+	static manifest = manifest_default;
+	static description = "Query Databricks Vector Search indexes with hybrid search, reranking, and pagination";
+	connector;
+	constructor(config) {
+		super(config);
+		this.config = {
+			...config,
+			indexes: config.indexes ?? this._defaultIndexes()
+		};
+		this.connector = new AiSearchConnector({
+			timeout: config.timeout,
+			telemetry: config.telemetry
+		});
+	}
+	/**
+	* Seeds a `default` index from `DATABRICKS_VS_INDEX_NAME` when no `indexes`
+	* are configured, so `aiSearch()` works with just the env var.
+	*/
+	_defaultIndexes() {
+		const indexName = process.env.DATABRICKS_VS_INDEX_NAME;
+		return indexName ? { default: { indexName } } : {};
+	}
+	async setup() {
+		for (const [alias, idx] of Object.entries(this.config.indexes ?? {})) if (idx.pagination && !idx.endpointName) throw new Error(`Index "${alias}" has pagination enabled but is missing "endpointName"`);
+		if (process.env.NODE_ENV === "development") await this._autoDiscoverColumns();
+		else for (const [alias, idx] of Object.entries(this.config.indexes ?? {})) if (!idx.columns || idx.columns.length === 0) throw new Error(`Index "${alias}" has no columns configured. Vector Search queries require "columns"; set them explicitly (auto-discovered only in development).`);
+	}
+	/**
+	* For each configured index missing `columns`, fill them from its Delta-Sync
+	* source table (all source columns minus embedding vectors). Best-effort:
+	* failures are logged and skipped, never thrown. A partial `columns_to_sync`
+	* isn't honored, so the discovered list is a starting point to trim.
+	*/
+	async _autoDiscoverColumns() {
+		const discovered = {};
+		for (const [alias, idx] of Object.entries(this.config.indexes ?? {})) {
+			if (idx.columns && idx.columns.length > 0) continue;
+			const indexName = idx.indexName ?? process.env.DATABRICKS_VS_INDEX_NAME;
+			if (!indexName) continue;
+			try {
+				const client = getWorkspaceClient();
+				const info = await this.connector.getIndex(client, indexName);
+				const sourceTable = info.delta_sync_index_spec?.source_table;
+				if (!sourceTable) continue;
+				const excluded = new Set((info.delta_sync_index_spec?.embedding_vector_columns ?? []).map((c) => c.name));
+				const columns = (await this.connector.getSourceColumns(client, sourceTable)).filter((c) => !excluded.has(c));
+				if (columns.length > 0) {
+					idx.columns = columns;
+					discovered[alias] = columns;
+				}
+			} catch (error) {
+				logger.warn("Could not auto-discover columns for index \"%s\": %s", alias, error instanceof Error ? error.message : String(error));
+			}
+		}
+		if (Object.keys(discovered).length > 0) logger.warn("\n%s", this._formatColumnDiscoveryBanner(discovered));
+	}
+	_formatColumnDiscoveryBanner(discovered) {
+		const lines = ["AI SEARCH: columns auto-discovered (dev mode — would fail in production)", ""];
+		for (const [alias, columns] of Object.entries(discovered)) lines.push(`  ${alias}: ${columns.join(", ")}`);
+		lines.push("");
+		lines.push("Set `columns` explicitly in the plugin config before deploying.");
+		return formatWarningBanner(lines);
+	}
+	injectRoutes(router) {
+		this.route(router, {
+			name: "query",
+			method: "post",
+			path: "/:alias/query",
+			handler: async (req, res) => {
+				const indexConfig = this._resolveOr404(req, res);
+				if (!indexConfig) return;
+				const body = req.body;
+				if (!body.queryText && !body.queryVector) {
+					res.status(400).json({
+						error: "queryText or queryVector is required",
+						plugin: this.name
+					});
+					return;
+				}
+				const { columns: _clientColumns, ...safeBody } = body;
+				const plugin = indexConfig.auth === "on-behalf-of-user" ? this.asUser(req) : this;
+				const queryType = safeBody.queryType ?? indexConfig.queryType ?? "hybrid";
+				try {
+					const result = await plugin.execute(async (signal) => {
+						const prepared = await this._prepareQuery(safeBody, indexConfig);
+						return this.connector.query(getWorkspaceClient(), {
+							indexName: indexConfig.indexName,
+							...prepared
+						}, signal);
+					}, querySettings);
+					this._sendResult(res, result, queryType);
+				} catch (error) {
+					this._handleError(res, error, "Query failed");
+				}
+			}
+		});
+		this.route(router, {
+			name: "queryNextPage",
+			method: "post",
+			path: "/:alias/next-page",
+			handler: async (req, res) => {
+				const indexConfig = this._resolveOr404(req, res);
+				if (!indexConfig) return;
+				if (!indexConfig.pagination) {
+					res.status(400).json({
+						error: `Pagination is not enabled for index "${req.params.alias}"`,
+						plugin: this.name
+					});
+					return;
+				}
+				if (!indexConfig.endpointName) {
+					res.status(400).json({
+						error: `Index "${req.params.alias}" is missing endpointName required for pagination`,
+						plugin: this.name
+					});
+					return;
+				}
+				const { pageToken, queryType } = req.body;
+				if (!pageToken) {
+					res.status(400).json({
+						error: "pageToken is required",
+						plugin: this.name
+					});
+					return;
+				}
+				const pageQueryType = queryType ?? indexConfig.queryType ?? "hybrid";
+				try {
+					const result = await (indexConfig.auth === "on-behalf-of-user" ? this.asUser(req) : this).execute(async (signal) => this.connector.queryNextPage(getWorkspaceClient(), {
+						indexName: indexConfig.indexName,
+						endpointName: indexConfig.endpointName,
+						pageToken
+					}, signal), querySettings);
+					this._sendResult(res, result, pageQueryType);
+				} catch (error) {
+					this._handleError(res, error, "Next-page query failed");
+				}
+			}
+		});
+		this.route(router, {
+			name: "getConfig",
+			method: "get",
+			path: "/:alias/config",
+			handler: async (req, res) => {
+				const indexConfig = this._resolveOr404(req, res);
+				if (!indexConfig) return;
+				res.json({
+					alias: req.params.alias,
+					columns: indexConfig.columns,
+					queryType: indexConfig.queryType ?? "hybrid",
+					numResults: indexConfig.numResults ?? 20,
+					reranker: !!indexConfig.reranker,
+					pagination: !!indexConfig.pagination
+				});
+			}
+		});
+	}
+	/**
+	* Index aliases + non-sensitive query metadata, serialized to the client so
+	* the UI can discover available indexes instead of hardcoding an alias.
+	*/
+	clientConfig() {
+		return { indexes: Object.entries(this.config.indexes ?? {}).map(([alias, idx]) => ({
+			alias,
+			queryType: idx.queryType ?? "hybrid",
+			pagination: !!idx.pagination
+		})) };
+	}
+	/**
+	* Programmatic query API — available as `appkit.aiSearch.query()`.
+	* When called through `asUser(req)`, executes with the user's credentials.
+	*
+	* @remarks `T` types each result's `data` but is an unchecked cast — the row
+	* shape isn't validated at runtime.
+	*/
+	async query(alias, request) {
+		const indexConfig = this._resolveIndex(alias);
+		if (!indexConfig) throw new Error(`No index configured with alias "${alias}"`);
+		const prepared = await this._prepareQuery(request, indexConfig);
+		const result = await this.execute(async (signal) => this.connector.query(getWorkspaceClient(), {
+			indexName: indexConfig.indexName,
+			...prepared
+		}, signal), querySettings);
+		if (!result.ok) throw new Error(`Vector search query failed for index "${alias}": ${result.message}`);
+		return this._parseResponse(result.data, prepared.queryType);
+	}
+	async shutdown() {}
+	exports() {
+		return { query: this.query.bind(this) };
+	}
+	_resolveIndex(alias) {
+		const idx = this.config.indexes?.[alias];
+		if (!idx) return void 0;
+		const indexName = idx.indexName ?? process.env.DATABRICKS_VS_INDEX_NAME;
+		if (!indexName) return void 0;
+		return {
+			...idx,
+			indexName
+		};
+	}
+	/** Resolve an index by route alias, or send a 404 and return null. */
+	_resolveOr404(req, res) {
+		const indexConfig = this._resolveIndex(req.params.alias);
+		if (!indexConfig) {
+			res.status(404).json({
+				error: `No index configured with alias "${req.params.alias}"`,
+				plugin: this.name
+			});
+			return null;
+		}
+		return indexConfig;
+	}
+	/** Send an execution result as JSON, or its error status/message. */
+	_sendResult(res, result, queryType) {
+		if (!result.ok) {
+			res.status(result.status).json({
+				error: result.message,
+				plugin: this.name
+			});
+			return;
+		}
+		res.json(this._parseResponse(result.data, queryType));
+	}
+	async _prepareQuery(request, indexConfig) {
+		const queryType = request.queryType ?? indexConfig.queryType ?? "hybrid";
+		let queryText = request.queryText;
+		let queryVector = request.queryVector;
+		if (indexConfig.embeddingFn && queryText && !queryVector && queryType !== "full_text") try {
+			queryVector = await indexConfig.embeddingFn(queryText);
+			if (queryType === "ann") queryText = void 0;
+		} catch (error) {
+			throw new Error(`Embedding generation failed: ${error instanceof Error ? error.message : String(error)}`);
+		}
+		const columns = request.columns ?? indexConfig.columns ?? [];
+		return {
+			queryText,
+			queryVector,
+			queryType,
+			columns,
+			numResults: request.numResults ?? indexConfig.numResults ?? 20,
+			filters: request.filters,
+			reranker: this._resolveReranker(request.reranker, indexConfig, columns)
+		};
+	}
+	_resolveReranker(requestReranker, indexConfig, columns) {
+		if (!(requestReranker ?? indexConfig.reranker)) return void 0;
+		if (typeof indexConfig.reranker === "object") return indexConfig.reranker;
+		const columnsToRerank = columns.filter((c) => c !== "id");
+		return columnsToRerank.length > 0 ? { columnsToRerank } : void 0;
+	}
+	_parseResponse(raw, queryType) {
+		const columnNames = raw.manifest.columns.map((c) => c.name);
+		const scoreIndex = columnNames.indexOf("score");
+		return {
+			results: raw.result.data_array.map((row) => {
+				const data = {};
+				for (let i = 0; i < columnNames.length; i++) if (i !== scoreIndex) data[columnNames[i]] = row[i];
+				return {
+					score: scoreIndex >= 0 ? row[scoreIndex] : 0,
+					data
+				};
+			}),
+			totalCount: raw.result.row_count,
+			queryTimeMs: raw.debug_info?.response_time ?? raw.debug_info?.latency_ms ?? 0,
+			queryType,
+			nextPageToken: raw.next_page_token ?? null
+		};
+	}
+	_handleError(res, error, fallbackMessage) {
+		logger.error("%s: %O", fallbackMessage, error);
+		const message = process.env.NODE_ENV !== "production" && error instanceof Error ? error.message : fallbackMessage;
+		res.status(500).json({
+			error: message,
+			plugin: this.name
+		});
+	}
+};
+const aiSearch = toPlugin(AiSearchPlugin);
+
+//#endregion
+export { AiSearchPlugin, aiSearch };
+//# sourceMappingURL=ai-search.js.map

package/dist/plugins/ai-search/ai-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-search.js","names":["manifest"],"sources":["../../../src/plugins/ai-search/ai-search.ts"],"sourcesContent":["import type express from \"express\";\nimport type { IAppRouter, PluginExecutionSettings } from \"shared\";\nimport { AiSearchConnector } from \"../../connectors/ai-search/client\";\nimport type {\n  VsQueryParams,\n  VsRawResponse,\n} from \"../../connectors/ai-search/types\";\nimport { getWorkspaceClient } from \"../../context\";\nimport { createLogger } from \"../../logging/logger\";\nimport { Plugin, toPlugin } from \"../../plugin\";\nimport type { PluginManifest } from \"../../registry\";\nimport { formatWarningBanner } from \"../../utils/banner\";\nimport { aiSearchDefaults } from \"./defaults\";\nimport manifest from \"./manifest.json\";\nimport type {\n  IAiSearchConfig,\n  IndexConfig,\n  IndexSummary,\n  SearchQueryType,\n  SearchRequest,\n  SearchResponse,\n  SearchResult,\n} from \"./types\";\n\nconst logger = createLogger(\"ai-search\");\n\nconst querySettings: PluginExecutionSettings = {\n  default: aiSearchDefaults,\n};\n\nexport class AiSearchPlugin extends Plugin<IAiSearchConfig> {\n  static manifest = manifest as PluginManifest<\"aiSearch\">;\n\n  protected static description =\n    \"Query Databricks Vector Search indexes with hybrid search, reranking, and pagination\";\n  protected declare config: IAiSearchConfig;\n\n  private connector: AiSearchConnector;\n\n  constructor(config: IAiSearchConfig) {\n    super(config);\n    this.config = {\n      ...config,\n      indexes: config.indexes ?? this._defaultIndexes(),\n    };\n    this.connector = new AiSearchConnector({\n      timeout: config.timeout,\n      telemetry: config.telemetry,\n    });\n  }\n\n  /**\n   * Seeds a `default` index from `DATABRICKS_VS_INDEX_NAME` when no `indexes`\n   * are configured, so `aiSearch()` works with just the env var.\n   */\n  private _defaultIndexes(): Record<string, IndexConfig> {\n    const indexName = process.env.DATABRICKS_VS_INDEX_NAME;\n    return indexName ? { default: { indexName } } : {};\n  }\n\n  async setup(): Promise<void> {\n    // pagination needs an endpointName the framework's resource validation\n    // can't see, so check it here.\n    for (const [alias, idx] of Object.entries(this.config.indexes ?? {})) {\n      if (idx.pagination && !idx.endpointName) {\n        throw new Error(\n          `Index \"${alias}\" has pagination enabled but is missing \"endpointName\"`,\n        );\n      }\n    }\n\n    // Dev fills in missing `columns` from the source table; prod can't query\n    // without them (VS requires `columns`), so fail fast at boot.\n    if (process.env.NODE_ENV === \"development\") {\n      await this._autoDiscoverColumns();\n    } else {\n      for (const [alias, idx] of Object.entries(this.config.indexes ?? {})) {\n        if (!idx.columns || idx.columns.length === 0) {\n          throw new Error(\n            `Index \"${alias}\" has no columns configured. Vector Search queries require \"columns\"; set them explicitly (auto-discovered only in development).`,\n          );\n        }\n      }\n    }\n  }\n\n  /**\n   * For each configured index missing `columns`, fill them from its Delta-Sync\n   * source table (all source columns minus embedding vectors). Best-effort:\n   * failures are logged and skipped, never thrown. A partial `columns_to_sync`\n   * isn't honored, so the discovered list is a starting point to trim.\n   */\n  private async _autoDiscoverColumns(): Promise<void> {\n    const discovered: Record<string, string[]> = {};\n    for (const [alias, idx] of Object.entries(this.config.indexes ?? {})) {\n      if (idx.columns && idx.columns.length > 0) continue;\n      const indexName = idx.indexName ?? process.env.DATABRICKS_VS_INDEX_NAME;\n      if (!indexName) continue;\n      try {\n        const client = getWorkspaceClient();\n        const info = await this.connector.getIndex(client, indexName);\n        const sourceTable = info.delta_sync_index_spec?.source_table;\n        if (!sourceTable) continue;\n        const excluded = new Set(\n          (info.delta_sync_index_spec?.embedding_vector_columns ?? []).map(\n            (c) => c.name,\n          ),\n        );\n        const columns = (\n          await this.connector.getSourceColumns(client, sourceTable)\n        ).filter((c) => !excluded.has(c));\n        if (columns.length > 0) {\n          idx.columns = columns;\n          discovered[alias] = columns;\n        }\n      } catch (error) {\n        logger.warn(\n          'Could not auto-discover columns for index \"%s\": %s',\n          alias,\n          error instanceof Error ? error.message : String(error),\n        );\n      }\n    }\n    if (Object.keys(discovered).length > 0) {\n      logger.warn(\"\\n%s\", this._formatColumnDiscoveryBanner(discovered));\n    }\n  }\n\n  private _formatColumnDiscoveryBanner(\n    discovered: Record<string, string[]>,\n  ): string {\n    const lines = [\n      \"AI SEARCH: columns auto-discovered (dev mode — would fail in production)\",\n      \"\",\n    ];\n    for (const [alias, columns] of Object.entries(discovered)) {\n      lines.push(`  ${alias}: ${columns.join(\", \")}`);\n    }\n    lines.push(\"\");\n    lines.push(\n      \"Set `columns` explicitly in the plugin config before deploying.\",\n    );\n\n    return formatWarningBanner(lines);\n  }\n\n  injectRoutes(router: IAppRouter) {\n    this.route(router, {\n      name: \"query\",\n      method: \"post\",\n      path: \"/:alias/query\",\n      handler: async (req: express.Request, res: express.Response) => {\n        const indexConfig = this._resolveOr404(req, res);\n        if (!indexConfig) return;\n\n        const body: SearchRequest = req.body;\n        if (!body.queryText && !body.queryVector) {\n          res.status(400).json({\n            error: \"queryText or queryVector is required\",\n            plugin: this.name,\n          });\n          return;\n        }\n\n        // Drop client-supplied `columns` so an HTTP caller can't widen the\n        // projection past what the app configured. (query() callers are\n        // trusted and keep the override.)\n        const { columns: _clientColumns, ...safeBody } = body;\n        const plugin =\n          indexConfig.auth === \"on-behalf-of-user\" ? this.asUser(req) : this;\n        const queryType =\n          safeBody.queryType ?? indexConfig.queryType ?? \"hybrid\";\n\n        try {\n          // Prepare inside execute so a self-managed embeddingFn runs in the\n          // same OBO context as the query, not as the service principal.\n          const result = await plugin.execute(async (signal) => {\n            const prepared = await this._prepareQuery(safeBody, indexConfig);\n            return this.connector.query(\n              getWorkspaceClient(),\n              { indexName: indexConfig.indexName, ...prepared },\n              signal,\n            );\n          }, querySettings);\n\n          this._sendResult(res, result, queryType);\n        } catch (error) {\n          this._handleError(res, error, \"Query failed\");\n        }\n      },\n    });\n\n    this.route(router, {\n      name: \"queryNextPage\",\n      method: \"post\",\n      path: \"/:alias/next-page\",\n      handler: async (req: express.Request, res: express.Response) => {\n        const indexConfig = this._resolveOr404(req, res);\n        if (!indexConfig) return;\n\n        if (!indexConfig.pagination) {\n          res.status(400).json({\n            error: `Pagination is not enabled for index \"${req.params.alias}\"`,\n            plugin: this.name,\n          });\n          return;\n        }\n\n        if (!indexConfig.endpointName) {\n          res.status(400).json({\n            error: `Index \"${req.params.alias}\" is missing endpointName required for pagination`,\n            plugin: this.name,\n          });\n          return;\n        }\n\n        const { pageToken, queryType } = req.body;\n        if (!pageToken) {\n          res.status(400).json({\n            error: \"pageToken is required\",\n            plugin: this.name,\n          });\n          return;\n        }\n        // Echo the original query's queryType so paged responses stay\n        // consistent with page 1; fall back to the index default.\n        const pageQueryType = queryType ?? indexConfig.queryType ?? \"hybrid\";\n\n        try {\n          const plugin =\n            indexConfig.auth === \"on-behalf-of-user\" ? this.asUser(req) : this;\n\n          const result = await plugin.execute(\n            async (signal) =>\n              this.connector.queryNextPage(\n                getWorkspaceClient(),\n                {\n                  indexName: indexConfig.indexName,\n                  endpointName: indexConfig.endpointName as string,\n                  pageToken,\n                },\n                signal,\n              ),\n            querySettings,\n          );\n\n          this._sendResult(res, result, pageQueryType);\n        } catch (error) {\n          this._handleError(res, error, \"Next-page query failed\");\n        }\n      },\n    });\n\n    this.route(router, {\n      name: \"getConfig\",\n      method: \"get\",\n      path: \"/:alias/config\",\n      handler: async (req: express.Request, res: express.Response) => {\n        const indexConfig = this._resolveOr404(req, res);\n        if (!indexConfig) return;\n        res.json({\n          alias: req.params.alias,\n          columns: indexConfig.columns,\n          queryType: indexConfig.queryType ?? \"hybrid\",\n          numResults: indexConfig.numResults ?? 20,\n          reranker: !!indexConfig.reranker,\n          pagination: !!indexConfig.pagination,\n        });\n      },\n    });\n  }\n\n  /**\n   * Index aliases + non-sensitive query metadata, serialized to the client so\n   * the UI can discover available indexes instead of hardcoding an alias.\n   */\n  clientConfig(): { indexes: IndexSummary[] } {\n    const indexes = Object.entries(this.config.indexes ?? {}).map(\n      ([alias, idx]) => ({\n        alias,\n        queryType: idx.queryType ?? \"hybrid\",\n        pagination: !!idx.pagination,\n      }),\n    );\n    return { indexes };\n  }\n\n  /**\n   * Programmatic query API — available as `appkit.aiSearch.query()`.\n   * When called through `asUser(req)`, executes with the user's credentials.\n   *\n   * @remarks `T` types each result's `data` but is an unchecked cast — the row\n   * shape isn't validated at runtime.\n   */\n  async query<T extends Record<string, unknown> = Record<string, unknown>>(\n    alias: string,\n    request: SearchRequest,\n  ): Promise<SearchResponse<T>> {\n    const indexConfig = this._resolveIndex(alias);\n    if (!indexConfig) {\n      throw new Error(`No index configured with alias \"${alias}\"`);\n    }\n\n    const prepared = await this._prepareQuery(request, indexConfig);\n\n    const result = await this.execute(\n      async (signal) =>\n        this.connector.query(\n          getWorkspaceClient(),\n          { indexName: indexConfig.indexName, ...prepared },\n          signal,\n        ),\n      querySettings,\n    );\n\n    if (!result.ok) {\n      throw new Error(\n        `Vector search query failed for index \"${alias}\": ${result.message}`,\n      );\n    }\n\n    return this._parseResponse(result.data, prepared.queryType);\n  }\n\n  async shutdown(): Promise<void> {\n    // No streams or persistent connections to clean up\n  }\n\n  exports() {\n    return {\n      query: this.query.bind(this),\n    };\n  }\n\n  private _resolveIndex(\n    alias: string,\n  ): (IndexConfig & { indexName: string }) | undefined {\n    const idx = this.config.indexes?.[alias];\n    if (!idx) return undefined;\n    const indexName = idx.indexName ?? process.env.DATABRICKS_VS_INDEX_NAME;\n    if (!indexName) return undefined;\n    return { ...idx, indexName };\n  }\n\n  /** Resolve an index by route alias, or send a 404 and return null. */\n  private _resolveOr404(\n    req: express.Request,\n    res: express.Response,\n  ): (IndexConfig & { indexName: string }) | null {\n    const indexConfig = this._resolveIndex(req.params.alias);\n    if (!indexConfig) {\n      res.status(404).json({\n        error: `No index configured with alias \"${req.params.alias}\"`,\n        plugin: this.name,\n      });\n      return null;\n    }\n    return indexConfig;\n  }\n\n  /** Send an execution result as JSON, or its error status/message. */\n  private _sendResult(\n    res: express.Response,\n    result: Awaited<ReturnType<typeof this.execute<VsRawResponse>>>,\n    queryType: SearchQueryType,\n  ): void {\n    if (!result.ok) {\n      res\n        .status(result.status)\n        .json({ error: result.message, plugin: this.name });\n      return;\n    }\n    res.json(this._parseResponse(result.data, queryType));\n  }\n\n  private async _prepareQuery(\n    request: SearchRequest,\n    indexConfig: IndexConfig,\n  ): Promise<Omit<VsQueryParams, \"indexName\">> {\n    const queryType = request.queryType ?? indexConfig.queryType ?? \"hybrid\";\n    let queryText = request.queryText;\n    let queryVector = request.queryVector;\n\n    // full_text uses no vector; hybrid keeps the text for its keyword half.\n    if (\n      indexConfig.embeddingFn &&\n      queryText &&\n      !queryVector &&\n      queryType !== \"full_text\"\n    ) {\n      try {\n        queryVector = await indexConfig.embeddingFn(queryText);\n        if (queryType === \"ann\") queryText = undefined;\n      } catch (error) {\n        throw new Error(\n          `Embedding generation failed: ${error instanceof Error ? error.message : String(error)}`,\n        );\n      }\n    }\n\n    const columns = request.columns ?? indexConfig.columns ?? [];\n    return {\n      queryText,\n      queryVector,\n      queryType,\n      columns,\n      numResults: request.numResults ?? indexConfig.numResults ?? 20,\n      filters: request.filters,\n      reranker: this._resolveReranker(request.reranker, indexConfig, columns),\n    };\n  }\n\n  private _resolveReranker(\n    requestReranker: boolean | undefined,\n    indexConfig: IndexConfig,\n    columns: string[],\n  ): { columnsToRerank: string[] } | undefined {\n    const shouldRerank = requestReranker ?? indexConfig.reranker;\n    if (!shouldRerank) return undefined;\n\n    if (typeof indexConfig.reranker === \"object\") {\n      return indexConfig.reranker;\n    }\n    // Auto-derive from returnable columns (excluding the id). With no columns\n    // resolved there's nothing to rerank on, so skip it.\n    const columnsToRerank = columns.filter((c) => c !== \"id\");\n    return columnsToRerank.length > 0 ? { columnsToRerank } : undefined;\n  }\n\n  private _parseResponse<\n    T extends Record<string, unknown> = Record<string, unknown>,\n  >(raw: VsRawResponse, queryType: SearchQueryType): SearchResponse<T> {\n    const columnNames = raw.manifest.columns.map((c) => c.name);\n    const scoreIndex = columnNames.indexOf(\"score\");\n\n    const results: SearchResult<T>[] = raw.result.data_array.map((row) => {\n      const data: Record<string, unknown> = {};\n      for (let i = 0; i < columnNames.length; i++) {\n        if (i !== scoreIndex) data[columnNames[i]] = row[i];\n      }\n      return {\n        score: scoreIndex >= 0 ? (row[scoreIndex] as number) : 0,\n        data: data as T,\n      };\n    });\n\n    return {\n      results,\n      totalCount: raw.result.row_count,\n      queryTimeMs:\n        raw.debug_info?.response_time ?? raw.debug_info?.latency_ms ?? 0,\n      queryType,\n      nextPageToken: raw.next_page_token ?? null,\n    };\n  }\n\n  private _handleError(\n    res: express.Response,\n    error: unknown,\n    fallbackMessage: string,\n  ): void {\n    logger.error(\"%s: %O\", fallbackMessage, error);\n    // Match Plugin.execute(): the raw message is only exposed outside production.\n    const isDev = process.env.NODE_ENV !== \"production\";\n    const message =\n      isDev && error instanceof Error ? error.message : fallbackMessage;\n    res.status(500).json({ error: message, plugin: this.name });\n  }\n}\n\nexport const aiSearch = toPlugin(AiSearchPlugin);\n"],"mappings":";;;;;;;;;;;;AAwBA,MAAM,SAAS,aAAa,YAAY;AAExC,MAAM,gBAAyC,EAC7C,SAAS,kBACV;AAED,IAAa,iBAAb,cAAoC,OAAwB;CAC1D,OAAO,WAAWA;CAElB,OAAiB,cACf;CAGF,AAAQ;CAER,YAAY,QAAyB;AACnC,QAAM,OAAO;AACb,OAAK,SAAS;GACZ,GAAG;GACH,SAAS,OAAO,WAAW,KAAK,iBAAiB;GAClD;AACD,OAAK,YAAY,IAAI,kBAAkB;GACrC,SAAS,OAAO;GAChB,WAAW,OAAO;GACnB,CAAC;;;;;;CAOJ,AAAQ,kBAA+C;EACrD,MAAM,YAAY,QAAQ,IAAI;AAC9B,SAAO,YAAY,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,EAAE;;CAGpD,MAAM,QAAuB;AAG3B,OAAK,MAAM,CAAC,OAAO,QAAQ,OAAO,QAAQ,KAAK,OAAO,WAAW,EAAE,CAAC,CAClE,KAAI,IAAI,cAAc,CAAC,IAAI,aACzB,OAAM,IAAI,MACR,UAAU,MAAM,wDACjB;AAML,MAAI,QAAQ,IAAI,aAAa,cAC3B,OAAM,KAAK,sBAAsB;MAEjC,MAAK,MAAM,CAAC,OAAO,QAAQ,OAAO,QAAQ,KAAK,OAAO,WAAW,EAAE,CAAC,CAClE,KAAI,CAAC,IAAI,WAAW,IAAI,QAAQ,WAAW,EACzC,OAAM,IAAI,MACR,UAAU,MAAM,kIACjB;;;;;;;;CAYT,MAAc,uBAAsC;EAClD,MAAM,aAAuC,EAAE;AAC/C,OAAK,MAAM,CAAC,OAAO,QAAQ,OAAO,QAAQ,KAAK,OAAO,WAAW,EAAE,CAAC,EAAE;AACpE,OAAI,IAAI,WAAW,IAAI,QAAQ,SAAS,EAAG;GAC3C,MAAM,YAAY,IAAI,aAAa,QAAQ,IAAI;AAC/C,OAAI,CAAC,UAAW;AAChB,OAAI;IACF,MAAM,SAAS,oBAAoB;IACnC,MAAM,OAAO,MAAM,KAAK,UAAU,SAAS,QAAQ,UAAU;IAC7D,MAAM,cAAc,KAAK,uBAAuB;AAChD,QAAI,CAAC,YAAa;IAClB,MAAM,WAAW,IAAI,KAClB,KAAK,uBAAuB,4BAA4B,EAAE,EAAE,KAC1D,MAAM,EAAE,KACV,CACF;IACD,MAAM,WACJ,MAAM,KAAK,UAAU,iBAAiB,QAAQ,YAAY,EAC1D,QAAQ,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;AACjC,QAAI,QAAQ,SAAS,GAAG;AACtB,SAAI,UAAU;AACd,gBAAW,SAAS;;YAEf,OAAO;AACd,WAAO,KACL,wDACA,OACA,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,CACvD;;;AAGL,MAAI,OAAO,KAAK,WAAW,CAAC,SAAS,EACnC,QAAO,KAAK,QAAQ,KAAK,6BAA6B,WAAW,CAAC;;CAItE,AAAQ,6BACN,YACQ;EACR,MAAM,QAAQ,CACZ,4EACA,GACD;AACD,OAAK,MAAM,CAAC,OAAO,YAAY,OAAO,QAAQ,WAAW,CACvD,OAAM,KAAK,KAAK,MAAM,IAAI,QAAQ,KAAK,KAAK,GAAG;AAEjD,QAAM,KAAK,GAAG;AACd,QAAM,KACJ,kEACD;AAED,SAAO,oBAAoB,MAAM;;CAGnC,aAAa,QAAoB;AAC/B,OAAK,MAAM,QAAQ;GACjB,MAAM;GACN,QAAQ;GACR,MAAM;GACN,SAAS,OAAO,KAAsB,QAA0B;IAC9D,MAAM,cAAc,KAAK,cAAc,KAAK,IAAI;AAChD,QAAI,CAAC,YAAa;IAElB,MAAM,OAAsB,IAAI;AAChC,QAAI,CAAC,KAAK,aAAa,CAAC,KAAK,aAAa;AACxC,SAAI,OAAO,IAAI,CAAC,KAAK;MACnB,OAAO;MACP,QAAQ,KAAK;MACd,CAAC;AACF;;IAMF,MAAM,EAAE,SAAS,gBAAgB,GAAG,aAAa;IACjD,MAAM,SACJ,YAAY,SAAS,sBAAsB,KAAK,OAAO,IAAI,GAAG;IAChE,MAAM,YACJ,SAAS,aAAa,YAAY,aAAa;AAEjD,QAAI;KAGF,MAAM,SAAS,MAAM,OAAO,QAAQ,OAAO,WAAW;MACpD,MAAM,WAAW,MAAM,KAAK,cAAc,UAAU,YAAY;AAChE,aAAO,KAAK,UAAU,MACpB,oBAAoB,EACpB;OAAE,WAAW,YAAY;OAAW,GAAG;OAAU,EACjD,OACD;QACA,cAAc;AAEjB,UAAK,YAAY,KAAK,QAAQ,UAAU;aACjC,OAAO;AACd,UAAK,aAAa,KAAK,OAAO,eAAe;;;GAGlD,CAAC;AAEF,OAAK,MAAM,QAAQ;GACjB,MAAM;GACN,QAAQ;GACR,MAAM;GACN,SAAS,OAAO,KAAsB,QAA0B;IAC9D,MAAM,cAAc,KAAK,cAAc,KAAK,IAAI;AAChD,QAAI,CAAC,YAAa;AAElB,QAAI,CAAC,YAAY,YAAY;AAC3B,SAAI,OAAO,IAAI,CAAC,KAAK;MACnB,OAAO,wCAAwC,IAAI,OAAO,MAAM;MAChE,QAAQ,KAAK;MACd,CAAC;AACF;;AAGF,QAAI,CAAC,YAAY,cAAc;AAC7B,SAAI,OAAO,IAAI,CAAC,KAAK;MACnB,OAAO,UAAU,IAAI,OAAO,MAAM;MAClC,QAAQ,KAAK;MACd,CAAC;AACF;;IAGF,MAAM,EAAE,WAAW,cAAc,IAAI;AACrC,QAAI,CAAC,WAAW;AACd,SAAI,OAAO,IAAI,CAAC,KAAK;MACnB,OAAO;MACP,QAAQ,KAAK;MACd,CAAC;AACF;;IAIF,MAAM,gBAAgB,aAAa,YAAY,aAAa;AAE5D,QAAI;KAIF,MAAM,SAAS,OAFb,YAAY,SAAS,sBAAsB,KAAK,OAAO,IAAI,GAAG,MAEpC,QAC1B,OAAO,WACL,KAAK,UAAU,cACb,oBAAoB,EACpB;MACE,WAAW,YAAY;MACvB,cAAc,YAAY;MAC1B;MACD,EACD,OACD,EACH,cACD;AAED,UAAK,YAAY,KAAK,QAAQ,cAAc;aACrC,OAAO;AACd,UAAK,aAAa,KAAK,OAAO,yBAAyB;;;GAG5D,CAAC;AAEF,OAAK,MAAM,QAAQ;GACjB,MAAM;GACN,QAAQ;GACR,MAAM;GACN,SAAS,OAAO,KAAsB,QAA0B;IAC9D,MAAM,cAAc,KAAK,cAAc,KAAK,IAAI;AAChD,QAAI,CAAC,YAAa;AAClB,QAAI,KAAK;KACP,OAAO,IAAI,OAAO;KAClB,SAAS,YAAY;KACrB,WAAW,YAAY,aAAa;KACpC,YAAY,YAAY,cAAc;KACtC,UAAU,CAAC,CAAC,YAAY;KACxB,YAAY,CAAC,CAAC,YAAY;KAC3B,CAAC;;GAEL,CAAC;;;;;;CAOJ,eAA4C;AAQ1C,SAAO,EAAE,SAPO,OAAO,QAAQ,KAAK,OAAO,WAAW,EAAE,CAAC,CAAC,KACvD,CAAC,OAAO,UAAU;GACjB;GACA,WAAW,IAAI,aAAa;GAC5B,YAAY,CAAC,CAAC,IAAI;GACnB,EACF,EACiB;;;;;;;;;CAUpB,MAAM,MACJ,OACA,SAC4B;EAC5B,MAAM,cAAc,KAAK,cAAc,MAAM;AAC7C,MAAI,CAAC,YACH,OAAM,IAAI,MAAM,mCAAmC,MAAM,GAAG;EAG9D,MAAM,WAAW,MAAM,KAAK,cAAc,SAAS,YAAY;EAE/D,MAAM,SAAS,MAAM,KAAK,QACxB,OAAO,WACL,KAAK,UAAU,MACb,oBAAoB,EACpB;GAAE,WAAW,YAAY;GAAW,GAAG;GAAU,EACjD,OACD,EACH,cACD;AAED,MAAI,CAAC,OAAO,GACV,OAAM,IAAI,MACR,yCAAyC,MAAM,KAAK,OAAO,UAC5D;AAGH,SAAO,KAAK,eAAe,OAAO,MAAM,SAAS,UAAU;;CAG7D,MAAM,WAA0B;CAIhC,UAAU;AACR,SAAO,EACL,OAAO,KAAK,MAAM,KAAK,KAAK,EAC7B;;CAGH,AAAQ,cACN,OACmD;EACnD,MAAM,MAAM,KAAK,OAAO,UAAU;AAClC,MAAI,CAAC,IAAK,QAAO;EACjB,MAAM,YAAY,IAAI,aAAa,QAAQ,IAAI;AAC/C,MAAI,CAAC,UAAW,QAAO;AACvB,SAAO;GAAE,GAAG;GAAK;GAAW;;;CAI9B,AAAQ,cACN,KACA,KAC8C;EAC9C,MAAM,cAAc,KAAK,cAAc,IAAI,OAAO,MAAM;AACxD,MAAI,CAAC,aAAa;AAChB,OAAI,OAAO,IAAI,CAAC,KAAK;IACnB,OAAO,mCAAmC,IAAI,OAAO,MAAM;IAC3D,QAAQ,KAAK;IACd,CAAC;AACF,UAAO;;AAET,SAAO;;;CAIT,AAAQ,YACN,KACA,QACA,WACM;AACN,MAAI,CAAC,OAAO,IAAI;AACd,OACG,OAAO,OAAO,OAAO,CACrB,KAAK;IAAE,OAAO,OAAO;IAAS,QAAQ,KAAK;IAAM,CAAC;AACrD;;AAEF,MAAI,KAAK,KAAK,eAAe,OAAO,MAAM,UAAU,CAAC;;CAGvD,MAAc,cACZ,SACA,aAC2C;EAC3C,MAAM,YAAY,QAAQ,aAAa,YAAY,aAAa;EAChE,IAAI,YAAY,QAAQ;EACxB,IAAI,cAAc,QAAQ;AAG1B,MACE,YAAY,eACZ,aACA,CAAC,eACD,cAAc,YAEd,KAAI;AACF,iBAAc,MAAM,YAAY,YAAY,UAAU;AACtD,OAAI,cAAc,MAAO,aAAY;WAC9B,OAAO;AACd,SAAM,IAAI,MACR,gCAAgC,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,GACvF;;EAIL,MAAM,UAAU,QAAQ,WAAW,YAAY,WAAW,EAAE;AAC5D,SAAO;GACL;GACA;GACA;GACA;GACA,YAAY,QAAQ,cAAc,YAAY,cAAc;GAC5D,SAAS,QAAQ;GACjB,UAAU,KAAK,iBAAiB,QAAQ,UAAU,aAAa,QAAQ;GACxE;;CAGH,AAAQ,iBACN,iBACA,aACA,SAC2C;AAE3C,MAAI,EADiB,mBAAmB,YAAY,UACjC,QAAO;AAE1B,MAAI,OAAO,YAAY,aAAa,SAClC,QAAO,YAAY;EAIrB,MAAM,kBAAkB,QAAQ,QAAQ,MAAM,MAAM,KAAK;AACzD,SAAO,gBAAgB,SAAS,IAAI,EAAE,iBAAiB,GAAG;;CAG5D,AAAQ,eAEN,KAAoB,WAA+C;EACnE,MAAM,cAAc,IAAI,SAAS,QAAQ,KAAK,MAAM,EAAE,KAAK;EAC3D,MAAM,aAAa,YAAY,QAAQ,QAAQ;AAa/C,SAAO;GACL,SAZiC,IAAI,OAAO,WAAW,KAAK,QAAQ;IACpE,MAAM,OAAgC,EAAE;AACxC,SAAK,IAAI,IAAI,GAAG,IAAI,YAAY,QAAQ,IACtC,KAAI,MAAM,WAAY,MAAK,YAAY,MAAM,IAAI;AAEnD,WAAO;KACL,OAAO,cAAc,IAAK,IAAI,cAAyB;KACjD;KACP;KACD;GAIA,YAAY,IAAI,OAAO;GACvB,aACE,IAAI,YAAY,iBAAiB,IAAI,YAAY,cAAc;GACjE;GACA,eAAe,IAAI,mBAAmB;GACvC;;CAGH,AAAQ,aACN,KACA,OACA,iBACM;AACN,SAAO,MAAM,UAAU,iBAAiB,MAAM;EAG9C,MAAM,UADQ,QAAQ,IAAI,aAAa,gBAE5B,iBAAiB,QAAQ,MAAM,UAAU;AACpD,MAAI,OAAO,IAAI,CAAC,KAAK;GAAE,OAAO;GAAS,QAAQ,KAAK;GAAM,CAAC;;;AAI/D,MAAa,WAAW,SAAS,eAAe"}

package/dist/plugins/ai-search/defaults.js

@@ -0,0 +1,14 @@
+//#region src/plugins/ai-search/defaults.ts
+const aiSearchDefaults = {
+	cache: { enabled: false },
+	retry: {
+		enabled: true,
+		initialDelay: 1e3,
+		attempts: 3
+	},
+	timeout: 3e4
+};
+
+//#endregion
+export { aiSearchDefaults };
+//# sourceMappingURL=defaults.js.map

package/dist/plugins/ai-search/defaults.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"defaults.js","names":[],"sources":["../../../src/plugins/ai-search/defaults.ts"],"sourcesContent":["import type { PluginExecuteConfig } from \"shared\";\n\nexport const aiSearchDefaults: PluginExecuteConfig = {\n  cache: { enabled: false },\n  retry: { enabled: true, initialDelay: 1000, attempts: 3 },\n  timeout: 30_000,\n};\n"],"mappings":";AAEA,MAAa,mBAAwC;CACnD,OAAO,EAAE,SAAS,OAAO;CACzB,OAAO;EAAE,SAAS;EAAM,cAAc;EAAM,UAAU;EAAG;CACzD,SAAS;CACV"}

package/dist/plugins/ai-search/index.d.ts

@@ -0,0 +1,2 @@
+import { IAiSearchConfig, IndexConfig, IndexSummary, RerankerConfig, SearchFilters, SearchQueryType, SearchRequest, SearchResponse, SearchResult } from "./types.js";
+import { AiSearchPlugin, aiSearch } from "./ai-search.js";

package/dist/plugins/ai-search/index.js

@@ -0,0 +1,3 @@
+import { AiSearchPlugin, aiSearch } from "./ai-search.js";
+
+export {  };

package/dist/plugins/ai-search/manifest.js

@@ -0,0 +1,73 @@
+//#region src/plugins/ai-search/manifest.json
+var manifest_default = {
+	$schema: "https://databricks.github.io/appkit/schemas/plugin-manifest.schema.json",
+	name: "aiSearch",
+	displayName: "AI Search Plugin",
+	stability: "beta",
+	description: "Query Databricks Vector Search indexes with built-in hybrid search, reranking, and pagination",
+	resources: {
+		"required": [{
+			"type": "vector_search_index",
+			"alias": "Vector Search Index",
+			"resourceKey": "vector-search-index",
+			"description": "A Databricks Vector Search index to query. Index names configured via plugin config.",
+			"permission": "SELECT",
+			"fields": { "id": {
+				"env": "DATABRICKS_VS_INDEX_NAME",
+				"description": "Three-level UC name of the default index (catalog.schema.index_name)"
+			} }
+		}],
+		"optional": []
+	},
+	scaffolding: { "rules": { "must": [
+		"List the vector search index's source-table columns (unity-catalog tables API) so the user can pick which to return",
+		"Ask the user which of those columns the search should return",
+		"Write aiSearch({ indexes: { default: { columns: [...] } } }) in the server file; the client queries the 'default' alias"
+	] } },
+	config: { "schema": {
+		"type": "object",
+		"properties": {
+			"indexes": {
+				"type": "object",
+				"description": "Map of alias names to index configurations",
+				"additionalProperties": {
+					"type": "object",
+					"properties": {
+						"indexName": {
+							"type": "string",
+							"description": "Three-level UC name: catalog.schema.index_name"
+						},
+						"columns": {
+							"type": "array",
+							"items": { "type": "string" },
+							"description": "Columns to return in results"
+						},
+						"queryType": {
+							"type": "string",
+							"enum": [
+								"ann",
+								"hybrid",
+								"full_text"
+							],
+							"default": "hybrid"
+						},
+						"numResults": {
+							"type": "number",
+							"default": 20
+						}
+					}
+				}
+			},
+			"timeout": {
+				"type": "number",
+				"default": 3e4,
+				"description": "Query execution timeout in milliseconds"
+			}
+		},
+		"required": ["indexes"]
+	} }
+};
+
+//#endregion
+export { manifest_default as default };
+//# sourceMappingURL=manifest.js.map

package/dist/plugins/ai-search/manifest.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manifest.js","names":[],"sources":["../../../src/plugins/ai-search/manifest.json"],"sourcesContent":[""],"mappings":""}

package/dist/plugins/ai-search/types.d.ts

@@ -0,0 +1,80 @@
+import { BasePluginConfig } from "../../shared/src/plugin.js";
+import "../../shared/src/index.js";
+
+//#region src/plugins/ai-search/types.d.ts
+/** Vector Search query mode: semantic (`ann`), keyword+semantic (`hybrid`), or keyword-only (`full_text`). */
+type SearchQueryType = "ann" | "hybrid" | "full_text";
+interface IAiSearchConfig extends BasePluginConfig {
+  timeout?: number;
+  indexes?: Record<string, IndexConfig>;
+}
+interface IndexConfig {
+  /**
+   * Three-level UC name: catalog.schema.index_name. Defaults to the
+   * `DATABRICKS_VS_INDEX_NAME` env var when omitted — so multiple aliases that
+   * omit it all resolve to that same physical index. Set it explicitly per
+   * alias when they should point at distinct indexes.
+   */
+  indexName?: string;
+  /**
+   * Columns to return in results. Optional: in development the plugin
+   * auto-discovers them from the index's source table when omitted (and warns
+   * that they should be set explicitly for production).
+   */
+  columns?: string[];
+  /** Default search mode */
+  queryType?: SearchQueryType;
+  /** Max results per query */
+  numResults?: number;
+  /** Enable built-in reranker. Pass true to rerank all non-id columns, or an object for fine control. */
+  reranker?: boolean | RerankerConfig;
+  /**
+   * Auth mode for the built-in HTTP routes — "service-principal" (default)
+   * uses the app's SP, "on-behalf-of-user" proxies the logged-in user's token.
+   * Programmatic callers select per call via `appkit.aiSearch.asUser(req)`.
+   */
+  auth?: "service-principal" | "on-behalf-of-user";
+  /** Enable cursor pagination */
+  pagination?: boolean;
+  /** VS endpoint name (required when pagination is true) */
+  endpointName?: string;
+  /**
+   * For self-managed embedding indexes: converts query text to an embedding vector.
+   * When provided, the plugin calls this function and sends query_vector to VS.
+   * When omitted, query_text is sent and VS computes embeddings server-side (managed mode).
+   */
+  embeddingFn?: (text: string) => Promise<number[]>;
+}
+interface RerankerConfig {
+  columnsToRerank: string[];
+}
+/** Public summary of a configured index, exposed to the client via `clientConfig()`. */
+interface IndexSummary {
+  alias: string;
+  queryType: SearchQueryType;
+  pagination: boolean;
+}
+type SearchFilters = Record<string, string | number | boolean | (string | number)[]>;
+interface SearchRequest {
+  queryText?: string;
+  queryVector?: number[];
+  columns?: string[];
+  numResults?: number;
+  queryType?: SearchQueryType;
+  filters?: SearchFilters;
+  reranker?: boolean;
+}
+interface SearchResponse<T extends Record<string, unknown> = Record<string, unknown>> {
+  results: SearchResult<T>[];
+  totalCount: number;
+  queryTimeMs: number;
+  queryType: SearchQueryType;
+  nextPageToken: string | null;
+}
+interface SearchResult<T extends Record<string, unknown> = Record<string, unknown>> {
+  score: number;
+  data: T;
+}
+//#endregion
+export { IAiSearchConfig, IndexConfig, IndexSummary, RerankerConfig, SearchFilters, SearchQueryType, SearchRequest, SearchResponse, SearchResult };
+//# sourceMappingURL=types.d.ts.map

package/dist/plugins/ai-search/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","names":[],"sources":["../../../src/plugins/ai-search/types.ts"],"mappings":";;;;;KAGY,eAAA;AAAA,UAEK,eAAA,SAAwB,gBAAA;EACvC,OAAA;EACA,OAAA,GAAU,MAAA,SAAe,WAAA;AAAA;AAAA,UAGV,WAAA;EAPU;AAE3B;;;;;EAYE,SAAA;EAZuD;;;;;EAkBvD,OAAA;EAhByB;EAkBzB,SAAA,GAAY,eAAA;EAlBwB;EAoBpC,UAAA;EAjB0B;EAmB1B,QAAA,aAAqB,cAAA;EAJT;;;;;EAUZ,IAAA;EAZA;EAcA,UAAA;EAZY;EAcZ,YAAA;EAVA;;;;;EAgBA,WAAA,IAAe,IAAA,aAAiB,OAAA;AAAA;AAAA,UAGjB,cAAA;EACf,eAAA;AAAA;AADF;AAAA,UAKiB,YAAA;EACf,KAAA;EACA,SAAA,EAAW,eAAA;EACX,UAAA;AAAA;AAAA,KAGU,aAAA,GAAgB,MAAA;AAAA,UAKX,aAAA;EACf,SAAA;EACA,WAAA;EACA,OAAA;EACA,UAAA;EACA,SAAA,GAAY,eAAA;EACZ,OAAA,GAAU,aAAA;EACV,QAAA;AAAA;AAAA,UAGe,cAAA,WACL,MAAA,oBAA0B,MAAA;EAEpC,OAAA,EAAS,YAAA,CAAa,CAAA;EACtB,UAAA;EACA,WAAA;EACA,SAAA,EAAW,eAAA;EACX,aAAA;AAAA;AAAA,UAGe,YAAA,WACL,MAAA,oBAA0B,MAAA;EAEpC,KAAA;EACA,IAAA,EAAM,CAAA;AAAA"}

package/dist/plugins/beta-exports.generated.d.ts

@@ -1,2 +1,4 @@
 import { agents } from "./agents/agents.js";
-import "./agents/index.js";
+import "./agents/index.js";
+import { aiSearch } from "./ai-search/ai-search.js";
+import "./ai-search/index.js";

package/dist/plugins/beta-exports.generated.js

@@ -1,4 +1,6 @@
 import { agents } from "./agents/agents.js";
 import "./agents/index.js";
+import { aiSearch } from "./ai-search/ai-search.js";
+import "./ai-search/index.js";
 
 export {  };

package/dist/plugins/jobs/index.d.ts

@@ -1,2 +1,2 @@
-import { IJobsConfig, JobAPI, JobConfig, JobHandle, JobsExport } from "./types.js";
+import { IJobsConfig, JobAPI, JobConfig, JobsExport } from "./types.js";
 import { jobs } from "./plugin.js";

package/dist/plugins/jobs/plugin.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"plugin.d.ts","names":[],"sources":["../../../src/plugins/jobs/plugin.ts"],"mappings":";;;;;;;;;cAsFM,UAAA,SAAmB,MAAA;EAAA,OAChB,QAAA,EAAuB,cAAA;EAAA,UAEZ,MAAA,EAAQ,WAAA;EAAA,QAClB,SAAA;EAAA,QACA,MAAA;EAAA,QACA,UAAA;EAAA,QACA,OAAA;;;;;SAMD,YAAA,CAAa,MAAA,EAAQ,WAAA,GAAc,MAAA,SAAe,SAAA;EAAf;;;;EAAA,OAgCnC,uBAAA,CAAwB,MAAA,EAAQ,WAAA,GAAc,mBAAA;cAqBzC,MAAA,EAAQ,WAAA;EA0Bd,KAAA,CAAA,GAAK,OAAA;EAAA,YAMC,MAAA,CAAA;EAAA,QAIJ,QAAA;EAAA,QAcA,aAAA;EAAA,QAYA,cAAA;EAhIqB;;;;EAAA,QA6IrB,eAAA;EA1IkB;;;;EAAA,UAoKhB,YAAA,CAAa,MAAA,WAAiB,MAAA;EA1JjC;;;;EAAA,QAgXC,WAAA;EAAA,QA0BA,gBAAA;EA1W+B;;;;;EAAA,QAsX/B,eAAA;EAAA,QAuCM,UAAA;EA4Cd,YAAA,CAAa,MAAA,EAAQ,UAAA;EA6IrB,OAAA,CAAA,GAAW,UAAA;EAsBX,YAAA,CAAA,GAAgB,MAAA;AAAA;;;;cAiBL,IAAA,EAAI,QAAA,QAAA,UAAA,EAAA,WAAA"}
+{"version":3,"file":"plugin.d.ts","names":[],"sources":["../../../src/plugins/jobs/plugin.ts"],"mappings":";;;;;;;;;cAoFM,UAAA,SAAmB,MAAA;EAAA,OAChB,QAAA,EAAuB,cAAA;EAAA,UAEZ,MAAA,EAAQ,WAAA;EAAA,QAClB,SAAA;EAAA,QACA,MAAA;EAAA,QACA,UAAA;EAAA,QACA,OAAA;;;;;SAMD,YAAA,CAAa,MAAA,EAAQ,WAAA,GAAc,MAAA,SAAe,SAAA;EAAf;;;;EAAA,OAgCnC,uBAAA,CAAwB,MAAA,EAAQ,WAAA,GAAc,mBAAA;cAqBzC,MAAA,EAAQ,WAAA;EA0Bd,KAAA,CAAA,GAAK,OAAA;EAAA,YAMC,MAAA,CAAA;EAAA,QAIJ,QAAA;EAAA,QAcA,aAAA;EAAA,QAYA,cAAA;EAhIqB;;;;EAAA,QA6IrB,eAAA;EA1IkB;;;;EAAA,UAoKhB,YAAA,CAAa,MAAA,WAAiB,MAAA;EA1JjC;;;;EAAA,QA6WC,WAAA;EAAA,QA0BA,gBAAA;EAvW+B;;;;;EAAA,QAmX/B,eAAA;EAAA,QAuCM,UAAA;EA4Cd,YAAA,CAAa,MAAA,EAAQ,UAAA;EA6IrB,OAAA,CAAA,GAAW,UAAA;EAcX,YAAA,CAAA,GAAgB,MAAA;AAAA;;;;cAiBL,IAAA,EAAI,QAAA,QAAA,UAAA,EAAA,WAAA"}

package/dist/plugins/jobs/plugin.js

@@ -498,12 +498,7 @@ var JobsPlugin = class JobsPlugin extends Plugin {
 	exports() {
 		const resolveJob = (jobKey) => {
 			if (!this.jobKeys.includes(jobKey)) throw new Error(`Unknown job "${jobKey}". Available jobs: ${this.jobKeys.join(", ")}`);
-			return {
-				...this.createJobAPI(jobKey),
-				asUser: (req) => {
-					return this.asUser(req).createJobAPI(jobKey);
-				}
-			};
+			return this.createJobAPI(jobKey);
 		};
 		return resolveJob;
 	}