import type { ToolDef } from "@fabric-harness/sdk";
import { type DatabricksStatementClient } from "./sql-exec.js";
export interface DatabricksAiQueryToolOptions {
    warehouseId: string;
    name?: string;
    description?: string;
    pollIntervalMs?: number;
    timeoutMs?: number;
}
export type DatabricksAiQueryPolicy = {
    allowedEndpoints: readonly string[];
    allowAnyEndpoint?: never;
} | {
    allowedEndpoints?: never;
    allowAnyEndpoint: true;
};
/**
 * `ai_query(endpoint, request)` — run model inference inside the SQL warehouse against a serving
 * endpoint. Drives both serving and SQL-warehouse consumption. Uses named SQL parameters, so the
 * endpoint and request are never string-interpolated into SQL.
 */
export declare function databricksAiQueryTool(client: DatabricksStatementClient, endpointPolicy: DatabricksAiQueryPolicy, options: DatabricksAiQueryToolOptions): ToolDef<{
    endpoint: string;
    request: string;
}, unknown>;
//# sourceMappingURL=ai.d.ts.map