@databricks/sdk-aifunctions 0.6.0 → 0.6.1

package/dist/package.js

@@ -1,6 +1,6 @@
 //#region package.json
 var name = "@databricks/sdk-aifunctions";
-var version = "0.6.0";
+var version = "0.6.1";
 
 //#endregion
 export { name, version };

package/dist/v1/client.d.ts

@@ -1,4 +1,4 @@
-import { AiClassifyRequest, AiClassifyResponse, AiDecideRequest, AiDecideResponse, AiExtractRequest, AiExtractResponse, AiParseDocumentRequest, AiParseDocumentResponse } from "./model.js";
+import { AiClassifyRequest, AiClassifyResponse, AiDecideRequest, AiDecideResponse, AiExtractRequest, AiExtractResponse, AiParseDocumentRequest, AiParseDocumentResponse, Batch, CancelBatchRequest, CancelBatchResponse, CreateBatchRequest, GetBatchRequest, ListBatchesRequest, ListBatchesResponse } from "./model.js";
 import { CallOptions } from "@databricks/sdk-options/call";
 import { ClientOptions } from "@databricks/sdk-options/client";
 
@@ -10,6 +10,15 @@ declare class AiFunctionsClient {
   private config;
   constructor(options: ClientOptions);
   private resolveConfig;
+  /** Cancel a batch. Cancellation is asynchronous: requests in flight drain first, so read the batch to observe the `CANCELLING` and `CANCELLED` statuses. */
+  cancelBatch(req: CancelBatchRequest, options?: CallOptions): Promise<CancelBatchResponse>;
+  /** Run an AI function over every request in a file on a Unity Catalog volume. The batch runs asynchronously; read it to follow its progress. */
+  createBatch(req: CreateBatchRequest, options?: CallOptions): Promise<Batch>;
+  /** Read one batch, with its status and progress. */
+  getBatch(req: GetBatchRequest, options?: CallOptions): Promise<Batch>;
+  /** List the batches the caller created in this workspace. */
+  listBatches(req: ListBatchesRequest, options?: CallOptions): Promise<ListBatchesResponse>;
+  listBatchesIter(req: ListBatchesRequest, options?: CallOptions): AsyncGenerator<Batch>;
   /** Classifies content according to a set of provided labels. For REST API requests, the default rate limit is 1,200 requests per minute per workspace. Contact your <Databricks> account team to request a higher limit. */
   aiClassify(req: AiClassifyRequest, options?: CallOptions): Promise<AiClassifyResponse>;
   /** Turn text and structured data into decisions your application can use. Define questions and criteria to choose an option, estimate a probability, or assign a score given a provided state. */

package/dist/v1/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","names":[],"sources":["../../src/v1/client.ts"],"mappings":";;;;;cA6Ca,iBAAA;EAAA,iBACM,OAAA;EAAA,iBACA,MAAA;EAAA,iBAIA,SAAA;EAAA,QAIT,MAAA;cAEI,OAAA,EAAS,aAAA;EAAA,QAUb,aAAA;EAQI;EAFN,UAAA,CACJ,GAAA,EAAK,iBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,kBAAA;EAAR;EA2BG,QAAA,CACJ,GAAA,EAAK,eAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,gBAAA;EADC;EA4BN,SAAA,CACJ,GAAA,EAAK,gBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,iBAAA;EA9BR;EAyDG,eAAA,CACJ,GAAA,EAAK,sBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,uBAAA;AAAA"}
+{"version":3,"file":"client.d.ts","names":[],"sources":["../../src/v1/client.ts"],"mappings":";;;;;cAyDa,iBAAA;EAAA,iBACM,OAAA;EAAA,iBACA,MAAA;EAAA,iBAIA,SAAA;EAAA,QAIT,MAAA;cAEI,OAAA,EAAS,aAAA;EAAA,QAUb,aAAA;EAQI;EAFN,WAAA,CACJ,GAAA,EAAK,kBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,mBAAA;EAAR;EA2BG,WAAA,CACJ,GAAA,EAAK,kBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,KAAA;EADC;EA4BN,QAAA,CAAS,GAAA,EAAK,eAAA,EAAiB,OAAA,GAAU,WAAA,GAAc,OAAA,CAAQ,KAAA;EA3BlE;EAqDG,WAAA,CACJ,GAAA,EAAK,kBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,mBAAA;EAkCJ,eAAA,CACL,GAAA,EAAK,kBAAA,EACL,OAAA,GAAU,WAAA,GACT,cAAA,CAAe,KAAA;EAlEmD;EAiF/D,UAAA,CACJ,GAAA,EAAK,iBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,kBAAA;EAzDJ;EAoFD,QAAA,CACJ,GAAA,EAAK,eAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,gBAAA;EArFA;EAgHL,SAAA,CACJ,GAAA,EAAK,gBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,iBAAA;EAhFJ;EA2GD,eAAA,CACJ,GAAA,EAAK,sBAAA,EACL,OAAA,GAAU,WAAA,GACT,OAAA,CAAQ,uBAAA;AAAA"}

package/dist/v1/client.js

@@ -1,7 +1,7 @@
 import { resolveClientConfig } from "./transport.js";
 import { buildHttpRequest, executeCall, executeHttpCall, marshalRequest, parseResponse } from "./utils.js";
 import { name, version } from "../package.js";
-import { marshalAiClassifyRequestSchema, marshalAiDecideRequestSchema, marshalAiExtractRequestSchema, marshalAiParseDocumentRequestSchema, unmarshalAiClassifyResponseSchema, unmarshalAiDecideResponseSchema, unmarshalAiExtractResponseSchema, unmarshalAiParseDocumentResponseSchema } from "./model.js";
+import { marshalAiClassifyRequestSchema, marshalAiDecideRequestSchema, marshalAiExtractRequestSchema, marshalAiParseDocumentRequestSchema, marshalBatchSchema, marshalCancelBatchRequestSchema, unmarshalAiClassifyResponseSchema, unmarshalAiDecideResponseSchema, unmarshalAiExtractResponseSchema, unmarshalAiParseDocumentResponseSchema, unmarshalBatchSchema, unmarshalCancelBatchResponseSchema, unmarshalListBatchesResponseSchema } from "./model.js";
 import { VERSION } from "@databricks/sdk-auth";
 import { createDefault } from "@databricks/sdk-core/clientinfo";
 import { NoOpLogger } from "@databricks/sdk-core/logger";
@@ -31,6 +31,98 @@ var AiFunctionsClient = class {
 		this.config ??= resolveClientConfig(this.options);
 		return this.config;
 	}
+	/** Cancel a batch. Cancellation is asynchronous: requests in flight drain first, so read the batch to observe the `CANCELLING` and `CANCELLED` statuses. */
+	async cancelBatch(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/ai-functions/batches/${req.name ?? ""}/cancel`;
+		const body = marshalRequest(req, marshalCancelBatchRequestSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalCancelBatchResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Run an AI function over every request in a file on a Unity Catalog volume. The batch runs asynchronously; read it to follow its progress. */
+	async createBatch(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/ai-functions/batches`;
+		const body = marshalRequest(req.batch, marshalBatchSchema);
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers({ "Content-Type": "application/json" });
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("POST", url, headers, callSignal, body),
+				httpClient,
+				logger: this.logger
+			}), unmarshalBatchSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** Read one batch, with its status and progress. */
+	async getBatch(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/ai-functions/batches/${req.name ?? ""}`;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", url, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalBatchSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	/** List the batches the caller created in this workspace. */
+	async listBatches(req, options) {
+		const { host, workspaceId, httpClient } = await this.resolveConfig();
+		const url = `${host}/api/2.0/ai-functions/batches`;
+		const params = new URLSearchParams();
+		if (req.pageSize !== void 0) params.append("page_size", String(req.pageSize));
+		if (req.pageToken !== void 0) params.append("page_token", req.pageToken);
+		const query = params.toString();
+		const fullUrl = query !== "" ? `${url}?${query}` : url;
+		let resp;
+		const call = async (callSignal) => {
+			const headers = new Headers();
+			if (workspaceId !== void 0) headers.set("X-Databricks-Workspace-Id", workspaceId);
+			headers.set("User-Agent", this.userAgent);
+			resp = parseResponse(await executeHttpCall({
+				request: buildHttpRequest("GET", fullUrl, headers, callSignal),
+				httpClient,
+				logger: this.logger
+			}), unmarshalListBatchesResponseSchema);
+		};
+		await executeCall(call, options);
+		if (resp === void 0) throw new Error("operation completed without a result.");
+		return resp;
+	}
+	async *listBatchesIter(req, options) {
+		const pageReq = { ...req };
+		for (;;) {
+			const resp = await this.listBatches(pageReq, options);
+			for (const item of resp.batches ?? []) yield item;
+			if (resp.nextPageToken === void 0 || resp.nextPageToken === "") return;
+			pageReq.pageToken = resp.nextPageToken;
+		}
+	}
 	/** Classifies content according to a set of provided labels. For REST API requests, the default rate limit is 1,200 requests per minute per workspace. Contact your <Databricks> account team to request a higher limit. */
 	async aiClassify(req, options) {
 		const { host, workspaceId, httpClient } = await this.resolveConfig();

package/dist/v1/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","names":["pkgJson.version","AUTH_VERSION"],"sources":["../../src/v1/client.ts"],"sourcesContent":["// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.\n\nimport {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';\nimport {createDefault} from '@databricks/sdk-core/clientinfo';\nimport type {Logger} from '@databricks/sdk-core/logger';\nimport {NoOpLogger} from '@databricks/sdk-core/logger';\nimport type {CallOptions} from '@databricks/sdk-options/call';\nimport type {ClientOptions} from '@databricks/sdk-options/client';\nimport type {ResolvedClientConfig} from './transport';\nimport {resolveClientConfig} from './transport';\nimport {\n  buildHttpRequest,\n  executeCall,\n  executeHttpCall,\n  marshalRequest,\n  parseResponse,\n} from './utils';\nimport pkgJson from '../../package.json' with {type: 'json'};\nimport type {\n  AiClassifyRequest,\n  AiClassifyResponse,\n  AiDecideRequest,\n  AiDecideResponse,\n  AiExtractRequest,\n  AiExtractResponse,\n  AiParseDocumentRequest,\n  AiParseDocumentResponse,\n} from './model';\nimport {\n  marshalAiClassifyRequestSchema,\n  marshalAiDecideRequestSchema,\n  marshalAiExtractRequestSchema,\n  marshalAiParseDocumentRequestSchema,\n  unmarshalAiClassifyResponseSchema,\n  unmarshalAiDecideResponseSchema,\n  unmarshalAiExtractResponseSchema,\n  unmarshalAiParseDocumentResponseSchema,\n} from './model';\n\n// Package identity segment for this client to be used in the User-Agent header.\nconst PACKAGE_SEGMENT = {\n  key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\\/sdk-/, ''),\n  value: pkgJson.version,\n};\n\nexport class AiFunctionsClient {\n  private readonly options: ClientOptions;\n  private readonly logger: Logger;\n  // User-Agent header value. Composed once at construction from\n  // createDefault() merged with this package's identity and the active\n  // credential's name.\n  private readonly userAgent: string;\n  // Memoized configuration. The profile is resolved once, lazily, on the first\n  // request, then reused; host, workspaceId/accountId, and credentials are\n  // filled from it when not set explicitly on the options.\n  private config: Promise<ResolvedClientConfig> | undefined;\n\n  constructor(options: ClientOptions) {\n    this.options = options;\n    this.logger = options.logger ?? new NoOpLogger();\n    const info = createDefault()\n      .with(PACKAGE_SEGMENT)\n      .with({key: 'sdk-js-auth', value: AUTH_VERSION})\n      .with({key: 'auth', value: options.credentials?.name() ?? 'default'});\n    this.userAgent = info.toString();\n  }\n\n  private resolveConfig(): Promise<ResolvedClientConfig> {\n    this.config ??= resolveClientConfig(this.options);\n    return this.config;\n  }\n\n  /** Classifies content according to a set of provided labels. For REST API requests, the default rate limit is 1,200 requests per minute per workspace. Contact your <Databricks> account team to request a higher limit. */\n  async aiClassify(\n    req: AiClassifyRequest,\n    options?: CallOptions\n  ): Promise<AiClassifyResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-classify`;\n    const body = marshalRequest(req, marshalAiClassifyRequestSchema);\n    let resp: AiClassifyResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiClassifyResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Turn text and structured data into decisions your application can use. Define questions and criteria to choose an option, estimate a probability, or assign a score given a provided state. */\n  async aiDecide(\n    req: AiDecideRequest,\n    options?: CallOptions\n  ): Promise<AiDecideResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-decide`;\n    const body = marshalRequest(req, marshalAiDecideRequestSchema);\n    let resp: AiDecideResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiDecideResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Extracts structured data from text and documents according to a provided schema. For REST API requests, the default rate limit is 120 requests per minute per workspace. Contact your <Databricks> account team to request a higher limit. */\n  async aiExtract(\n    req: AiExtractRequest,\n    options?: CallOptions\n  ): Promise<AiExtractResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-extract`;\n    const body = marshalRequest(req, marshalAiExtractRequestSchema);\n    let resp: AiExtractResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiExtractResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Parse structured content from unstructured documents. */\n  async aiParseDocument(\n    req: AiParseDocumentRequest,\n    options?: CallOptions\n  ): Promise<AiParseDocumentResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-parse-document`;\n    const body = marshalRequest(req, marshalAiParseDocumentRequestSchema);\n    let resp: AiParseDocumentResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiParseDocumentResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n}\n"],"mappings":";;;;;;;;;AAwCA,MAAM,kBAAkB;CACtB,KAAK,iBAAyB,QAAQ,iBAAiB,GAAG;CAC1D,OAAOA;CACR;AAED,IAAa,oBAAb,MAA+B;CAC7B,AAAiB;CACjB,AAAiB;CAIjB,AAAiB;CAIjB,AAAQ;CAER,YAAY,SAAwB;AAClC,OAAK,UAAU;AACf,OAAK,SAAS,QAAQ,UAAU,IAAI,YAAY;AAKhD,OAAK,YAJQ,eAAe,CACzB,KAAK,gBAAgB,CACrB,KAAK;GAAC,KAAK;GAAe,OAAOC;GAAa,CAAC,CAC/C,KAAK;GAAC,KAAK;GAAQ,OAAO,QAAQ,aAAa,MAAM,IAAI;GAAU,CAAC,CACjD,UAAU;;CAGlC,AAAQ,gBAA+C;AACrD,OAAK,WAAW,oBAAoB,KAAK,QAAQ;AACjD,SAAO,KAAK;;;CAId,MAAM,WACJ,KACA,SAC6B;EAC7B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,+BAA+B;EAChE,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,kCAAkC;;AAEnE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,SACJ,KACA,SAC2B;EAC3B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,6BAA6B;EAC9D,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,gCAAgC;;AAEjE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,UACJ,KACA,SAC4B;EAC5B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,8BAA8B;EAC/D,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,iCAAiC;;AAElE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,gBACJ,KACA,SACkC;EAClC,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,oCAAoC;EACrE,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,uCAAuC;;AAExE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO"}
+{"version":3,"file":"client.js","names":["pkgJson.version","AUTH_VERSION"],"sources":["../../src/v1/client.ts"],"sourcesContent":["// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.\n\nimport {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';\nimport {createDefault} from '@databricks/sdk-core/clientinfo';\nimport type {Logger} from '@databricks/sdk-core/logger';\nimport {NoOpLogger} from '@databricks/sdk-core/logger';\nimport type {CallOptions} from '@databricks/sdk-options/call';\nimport type {ClientOptions} from '@databricks/sdk-options/client';\nimport type {ResolvedClientConfig} from './transport';\nimport {resolveClientConfig} from './transport';\nimport {\n  buildHttpRequest,\n  executeCall,\n  executeHttpCall,\n  marshalRequest,\n  parseResponse,\n} from './utils';\nimport pkgJson from '../../package.json' with {type: 'json'};\nimport type {\n  AiClassifyRequest,\n  AiClassifyResponse,\n  AiDecideRequest,\n  AiDecideResponse,\n  AiExtractRequest,\n  AiExtractResponse,\n  AiParseDocumentRequest,\n  AiParseDocumentResponse,\n  Batch,\n  CancelBatchRequest,\n  CancelBatchResponse,\n  CreateBatchRequest,\n  GetBatchRequest,\n  ListBatchesRequest,\n  ListBatchesResponse,\n} from './model';\nimport {\n  marshalAiClassifyRequestSchema,\n  marshalAiDecideRequestSchema,\n  marshalAiExtractRequestSchema,\n  marshalAiParseDocumentRequestSchema,\n  marshalBatchSchema,\n  marshalCancelBatchRequestSchema,\n  unmarshalAiClassifyResponseSchema,\n  unmarshalAiDecideResponseSchema,\n  unmarshalAiExtractResponseSchema,\n  unmarshalAiParseDocumentResponseSchema,\n  unmarshalBatchSchema,\n  unmarshalCancelBatchResponseSchema,\n  unmarshalListBatchesResponseSchema,\n} from './model';\n\n// Package identity segment for this client to be used in the User-Agent header.\nconst PACKAGE_SEGMENT = {\n  key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\\/sdk-/, ''),\n  value: pkgJson.version,\n};\n\nexport class AiFunctionsClient {\n  private readonly options: ClientOptions;\n  private readonly logger: Logger;\n  // User-Agent header value. Composed once at construction from\n  // createDefault() merged with this package's identity and the active\n  // credential's name.\n  private readonly userAgent: string;\n  // Memoized configuration. The profile is resolved once, lazily, on the first\n  // request, then reused; host, workspaceId/accountId, and credentials are\n  // filled from it when not set explicitly on the options.\n  private config: Promise<ResolvedClientConfig> | undefined;\n\n  constructor(options: ClientOptions) {\n    this.options = options;\n    this.logger = options.logger ?? new NoOpLogger();\n    const info = createDefault()\n      .with(PACKAGE_SEGMENT)\n      .with({key: 'sdk-js-auth', value: AUTH_VERSION})\n      .with({key: 'auth', value: options.credentials?.name() ?? 'default'});\n    this.userAgent = info.toString();\n  }\n\n  private resolveConfig(): Promise<ResolvedClientConfig> {\n    this.config ??= resolveClientConfig(this.options);\n    return this.config;\n  }\n\n  /** Cancel a batch. Cancellation is asynchronous: requests in flight drain first, so read the batch to observe the `CANCELLING` and `CANCELLED` statuses. */\n  async cancelBatch(\n    req: CancelBatchRequest,\n    options?: CallOptions\n  ): Promise<CancelBatchResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/batches/${req.name ?? ''}/cancel`;\n    const body = marshalRequest(req, marshalCancelBatchRequestSchema);\n    let resp: CancelBatchResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalCancelBatchResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Run an AI function over every request in a file on a Unity Catalog volume. The batch runs asynchronously; read it to follow its progress. */\n  async createBatch(\n    req: CreateBatchRequest,\n    options?: CallOptions\n  ): Promise<Batch> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/batches`;\n    const body = marshalRequest(req.batch, marshalBatchSchema);\n    let resp: Batch | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalBatchSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Read one batch, with its status and progress. */\n  async getBatch(req: GetBatchRequest, options?: CallOptions): Promise<Batch> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/batches/${req.name ?? ''}`;\n    let resp: Batch | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers();\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('GET', url, headers, callSignal);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalBatchSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** List the batches the caller created in this workspace. */\n  async listBatches(\n    req: ListBatchesRequest,\n    options?: CallOptions\n  ): Promise<ListBatchesResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/batches`;\n    const params = new URLSearchParams();\n    if (req.pageSize !== undefined) {\n      params.append('page_size', String(req.pageSize));\n    }\n    if (req.pageToken !== undefined) {\n      params.append('page_token', req.pageToken);\n    }\n    const query = params.toString();\n    const fullUrl = query !== '' ? `${url}?${query}` : url;\n    let resp: ListBatchesResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers();\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('GET', fullUrl, headers, callSignal);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalListBatchesResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  async *listBatchesIter(\n    req: ListBatchesRequest,\n    options?: CallOptions\n  ): AsyncGenerator<Batch> {\n    const pageReq: ListBatchesRequest = {...req};\n    for (;;) {\n      const resp = await this.listBatches(pageReq, options);\n      for (const item of resp.batches ?? []) {\n        yield item;\n      }\n      if (resp.nextPageToken === undefined || resp.nextPageToken === '') {\n        return;\n      }\n      pageReq.pageToken = resp.nextPageToken;\n    }\n  }\n\n  /** Classifies content according to a set of provided labels. For REST API requests, the default rate limit is 1,200 requests per minute per workspace. Contact your <Databricks> account team to request a higher limit. */\n  async aiClassify(\n    req: AiClassifyRequest,\n    options?: CallOptions\n  ): Promise<AiClassifyResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-classify`;\n    const body = marshalRequest(req, marshalAiClassifyRequestSchema);\n    let resp: AiClassifyResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiClassifyResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Turn text and structured data into decisions your application can use. Define questions and criteria to choose an option, estimate a probability, or assign a score given a provided state. */\n  async aiDecide(\n    req: AiDecideRequest,\n    options?: CallOptions\n  ): Promise<AiDecideResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-decide`;\n    const body = marshalRequest(req, marshalAiDecideRequestSchema);\n    let resp: AiDecideResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiDecideResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Extracts structured data from text and documents according to a provided schema. For REST API requests, the default rate limit is 120 requests per minute per workspace. Contact your <Databricks> account team to request a higher limit. */\n  async aiExtract(\n    req: AiExtractRequest,\n    options?: CallOptions\n  ): Promise<AiExtractResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-extract`;\n    const body = marshalRequest(req, marshalAiExtractRequestSchema);\n    let resp: AiExtractResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiExtractResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n\n  /** Parse structured content from unstructured documents. */\n  async aiParseDocument(\n    req: AiParseDocumentRequest,\n    options?: CallOptions\n  ): Promise<AiParseDocumentResponse> {\n    const {host, workspaceId, httpClient} = await this.resolveConfig();\n    const url = `${host}/api/2.0/ai-functions/ai-parse-document`;\n    const body = marshalRequest(req, marshalAiParseDocumentRequestSchema);\n    let resp: AiParseDocumentResponse | undefined;\n    const call = async (callSignal?: AbortSignal): Promise<void> => {\n      const headers = new Headers({'Content-Type': 'application/json'});\n      if (workspaceId !== undefined) {\n        headers.set('X-Databricks-Workspace-Id', workspaceId);\n      }\n      headers.set('User-Agent', this.userAgent);\n      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);\n      const respBody = await executeHttpCall({\n        request: httpReq,\n        httpClient,\n        logger: this.logger,\n      });\n      resp = parseResponse(respBody, unmarshalAiParseDocumentResponseSchema);\n    };\n    await executeCall(call, options);\n    if (resp === undefined) {\n      throw new Error('operation completed without a result.');\n    }\n    return resp;\n  }\n}\n"],"mappings":";;;;;;;;;AAoDA,MAAM,kBAAkB;CACtB,KAAK,iBAAyB,QAAQ,iBAAiB,GAAG;CAC1D,OAAOA;CACR;AAED,IAAa,oBAAb,MAA+B;CAC7B,AAAiB;CACjB,AAAiB;CAIjB,AAAiB;CAIjB,AAAQ;CAER,YAAY,SAAwB;AAClC,OAAK,UAAU;AACf,OAAK,SAAS,QAAQ,UAAU,IAAI,YAAY;AAKhD,OAAK,YAJQ,eAAe,CACzB,KAAK,gBAAgB,CACrB,KAAK;GAAC,KAAK;GAAe,OAAOC;GAAa,CAAC,CAC/C,KAAK;GAAC,KAAK;GAAQ,OAAO,QAAQ,aAAa,MAAM,IAAI;GAAU,CAAC,CACjD,UAAU;;CAGlC,AAAQ,gBAA+C;AACrD,OAAK,WAAW,oBAAoB,KAAK,QAAQ;AACjD,SAAO,KAAK;;;CAId,MAAM,YACJ,KACA,SAC8B;EAC9B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK,gCAAgC,IAAI,QAAQ,GAAG;EACnE,MAAM,OAAO,eAAe,KAAK,gCAAgC;EACjE,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,mCAAmC;;AAEpE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,YACJ,KACA,SACgB;EAChB,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,IAAI,OAAO,mBAAmB;EAC1D,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,qBAAqB;;AAEtD,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,SAAS,KAAsB,SAAuC;EAC1E,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK,gCAAgC,IAAI,QAAQ;EAChE,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,SAAS;AAC7B,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,OAAO,KAAK,SAAS,WAAW;IAG/D;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,qBAAqB;;AAEtD,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,YACJ,KACA,SAC8B;EAC9B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,SAAS,IAAI,iBAAiB;AACpC,MAAI,IAAI,aAAa,OACnB,QAAO,OAAO,aAAa,OAAO,IAAI,SAAS,CAAC;AAElD,MAAI,IAAI,cAAc,OACpB,QAAO,OAAO,cAAc,IAAI,UAAU;EAE5C,MAAM,QAAQ,OAAO,UAAU;EAC/B,MAAM,UAAU,UAAU,KAAK,GAAG,IAAI,GAAG,UAAU;EACnD,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,SAAS;AAC7B,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,OAAO,SAAS,SAAS,WAAW;IAGnE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,mCAAmC;;AAEpE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;CAGT,OAAO,gBACL,KACA,SACuB;EACvB,MAAM,UAA8B,EAAC,GAAG,KAAI;AAC5C,WAAS;GACP,MAAM,OAAO,MAAM,KAAK,YAAY,SAAS,QAAQ;AACrD,QAAK,MAAM,QAAQ,KAAK,WAAW,EAAE,CACnC,OAAM;AAER,OAAI,KAAK,kBAAkB,UAAa,KAAK,kBAAkB,GAC7D;AAEF,WAAQ,YAAY,KAAK;;;;CAK7B,MAAM,WACJ,KACA,SAC6B;EAC7B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,+BAA+B;EAChE,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,kCAAkC;;AAEnE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,SACJ,KACA,SAC2B;EAC3B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,6BAA6B;EAC9D,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,gCAAgC;;AAEjE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,UACJ,KACA,SAC4B;EAC5B,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,8BAA8B;EAC/D,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,iCAAiC;;AAElE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO;;;CAIT,MAAM,gBACJ,KACA,SACkC;EAClC,MAAM,EAAC,MAAM,aAAa,eAAc,MAAM,KAAK,eAAe;EAClE,MAAM,MAAM,GAAG,KAAK;EACpB,MAAM,OAAO,eAAe,KAAK,oCAAoC;EACrE,IAAI;EACJ,MAAM,OAAO,OAAO,eAA4C;GAC9D,MAAM,UAAU,IAAI,QAAQ,EAAC,gBAAgB,oBAAmB,CAAC;AACjE,OAAI,gBAAgB,OAClB,SAAQ,IAAI,6BAA6B,YAAY;AAEvD,WAAQ,IAAI,cAAc,KAAK,UAAU;AAOzC,UAAO,cALU,MAAM,gBAAgB;IACrC,SAFc,iBAAiB,QAAQ,KAAK,SAAS,YAAY,KAAK;IAGtE;IACA,QAAQ,KAAK;IACd,CAAC,EAC6B,uCAAuC;;AAExE,QAAM,YAAY,MAAM,QAAQ;AAChC,MAAI,SAAS,OACX,OAAM,IAAI,MAAM,wCAAwC;AAE1D,SAAO"}

package/dist/v1/index.d.ts

@@ -1,3 +1,3 @@
-import { AiClassifyOptions, AiClassifyRequest, AiClassifyResponse, AiClassifyResponseMetadata, AiDecideOptions, AiDecideRequest, AiDecideResponse, AiDecideResponseMetadata, AiExtractBbox, AiExtractCitation, AiExtractOptions, AiExtractRequest, AiExtractResponse, AiExtractResponseMetadata, AiParseDocumentFileMetadata, AiParseDocumentOptions, AiParseDocumentPageError, AiParseDocumentRequest, AiParseDocumentResponse, AiParseDocumentResponseMetadata } from "./model.js";
+import { AiClassifyOptions, AiClassifyRequest, AiClassifyResponse, AiClassifyResponseMetadata, AiDecideOptions, AiDecideRequest, AiDecideResponse, AiDecideResponseMetadata, AiExtractBbox, AiExtractCitation, AiExtractOptions, AiExtractRequest, AiExtractResponse, AiExtractResponseMetadata, AiParseDocumentFileMetadata, AiParseDocumentOptions, AiParseDocumentPageError, AiParseDocumentRequest, AiParseDocumentResponse, AiParseDocumentResponseMetadata, Batch, BatchRequestCounts, BatchStatus, CancelBatchRequest, CancelBatchResponse, CreateBatchRequest, GetBatchRequest, ListBatchesRequest, ListBatchesResponse } from "./model.js";
 import { AiFunctionsClient } from "./client.js";
-export { type AiClassifyOptions, type AiClassifyRequest, type AiClassifyResponse, type AiClassifyResponseMetadata, type AiDecideOptions, type AiDecideRequest, type AiDecideResponse, type AiDecideResponseMetadata, type AiExtractBbox, type AiExtractCitation, type AiExtractOptions, type AiExtractRequest, type AiExtractResponse, type AiExtractResponseMetadata, AiFunctionsClient, type AiParseDocumentFileMetadata, type AiParseDocumentOptions, type AiParseDocumentPageError, type AiParseDocumentRequest, type AiParseDocumentResponse, type AiParseDocumentResponseMetadata };
+export { type AiClassifyOptions, type AiClassifyRequest, type AiClassifyResponse, type AiClassifyResponseMetadata, type AiDecideOptions, type AiDecideRequest, type AiDecideResponse, type AiDecideResponseMetadata, type AiExtractBbox, type AiExtractCitation, type AiExtractOptions, type AiExtractRequest, type AiExtractResponse, type AiExtractResponseMetadata, AiFunctionsClient, type AiParseDocumentFileMetadata, type AiParseDocumentOptions, type AiParseDocumentPageError, type AiParseDocumentRequest, type AiParseDocumentResponse, type AiParseDocumentResponseMetadata, type Batch, type BatchRequestCounts, BatchStatus, type CancelBatchRequest, type CancelBatchResponse, type CreateBatchRequest, type GetBatchRequest, type ListBatchesRequest, type ListBatchesResponse };

package/dist/v1/index.js

@@ -1,3 +1,4 @@
+import { BatchStatus } from "./model.js";
 import { AiFunctionsClient } from "./client.js";
 
-export { AiFunctionsClient };
+export { AiFunctionsClient, BatchStatus };

package/dist/v1/model.d.ts

@@ -1,7 +1,21 @@
+import { Temporal } from "@js-temporal/polyfill";
 import { z } from "zod";
-import { JsonValue } from "@databricks/sdk-core/wkt";
+import { JsonObject, JsonValue } from "@databricks/sdk-core/wkt";
 
 //#region src/v1/model.d.ts
+/** The status of a batch. */
+declare const BatchStatus: {
+  readonly BATCH_STATUS_UNSPECIFIED: "BATCH_STATUS_UNSPECIFIED"; /** Accepted, not started yet. */
+  readonly PENDING: "PENDING"; /** Requests are being processed. */
+  readonly IN_PROGRESS: "IN_PROGRESS"; /** Every request is processed and the output file is being written. */
+  readonly FINALIZING: "FINALIZING"; /** The output is written and available. */
+  readonly COMPLETED: "COMPLETED"; /** The completion window elapsed. The output holds the requests that finished in time. */
+  readonly EXPIRED: "EXPIRED"; /** Cancellation was accepted and requests in flight are draining. */
+  readonly CANCELLING: "CANCELLING"; /** Cancellation finished. No output file is written. */
+  readonly CANCELLED: "CANCELLED"; /** The batch failed. */
+  readonly FAILED: "FAILED";
+};
+type BatchStatus = (typeof BatchStatus)[keyof typeof BatchStatus] | (string & {});
 /** ai_classify */
 interface AiClassifyOptions {
   /** Natural-language guidance that steers how the text is classified (up to 20,000 characters). */
@@ -292,6 +306,68 @@ interface AiParseDocumentResponseMetadata {
   /** Describes the source file; present only for file-path input. */
   fileMetadata?: AiParseDocumentFileMetadata | undefined;
 }
+/** An asynchronous run of one AI function over a file of requests. */
+interface Batch {
+  /** ID of the batch, in the form `batch_<id>`. The service assigns it. */
+  name?: string | undefined;
+  /** The AI function to run for every request, for example `ai_extract`. */
+  endpoint?: string | undefined;
+  /** Path of the JSONL input file on a Unity Catalog volume. Each line is one request. */
+  inputRef?: string | undefined;
+  /** Function options applied to every request in the batch, for example the extraction schema. */
+  params?: JsonObject | undefined;
+  /**
+   * Path on a Unity Catalog volume to write the JSONL output to. When it is omitted, the output
+   * stays on Databricks-managed storage and a finished batch returns a short-lived download URL
+   * in this field instead.
+   */
+  outputRef?: string | undefined;
+  /** Output only. Status of the batch. */
+  status?: BatchStatus | undefined;
+  /** Output only. Time the batch was created. */
+  createTime?: Temporal.Instant | undefined;
+  /** Output only. Request counts. Returned when one batch is read, not when batches are listed. */
+  requestCounts?: BatchRequestCounts | undefined;
+  /**
+   * Output only. Percentage of the requests that finished, from 0 to 100. Returned when one batch
+   * is read, not when batches are listed.
+   */
+  percentComplete?: number | undefined;
+}
+/** Counts of the requests in a batch. */
+interface BatchRequestCounts {
+  /** Output only. Number of requests in the input file. */
+  total?: bigint | undefined;
+  /** Output only. Number of requests that finished successfully. */
+  succeeded?: bigint | undefined;
+  /** Output only. Number of requests that failed and will not be retried. */
+  failed?: bigint | undefined;
+}
+interface CancelBatchRequest {
+  /** ID of the batch, in the form `batch_<id>`. */
+  name?: string | undefined;
+}
+interface CancelBatchResponse {}
+interface CreateBatchRequest {
+  /** The batch to create. */
+  batch?: Batch | undefined;
+}
+interface GetBatchRequest {
+  /** ID of the batch, in the form `batch_<id>`. */
+  name?: string | undefined;
+}
+interface ListBatchesRequest {
+  /** Maximum number of batches to return in one page. */
+  pageSize?: number | undefined;
+  /** Page token from an earlier response. Omit it to read the first page. */
+  pageToken?: string | undefined;
+}
+interface ListBatchesResponse {
+  /** The batches in this page. */
+  batches?: Batch[] | undefined;
+  /** Token that reads the next page. It is empty on the last page. */
+  nextPageToken?: string | undefined;
+}
 declare const unmarshalAiClassifyResponseSchema: z.ZodType<AiClassifyResponse>;
 declare const unmarshalAiClassifyResponseMetadataSchema: z.ZodType<AiClassifyResponseMetadata>;
 declare const unmarshalAiDecideResponseSchema: z.ZodType<AiDecideResponse>;
@@ -304,6 +380,10 @@ declare const unmarshalAiParseDocumentFileMetadataSchema: z.ZodType<AiParseDocum
 declare const unmarshalAiParseDocumentPageErrorSchema: z.ZodType<AiParseDocumentPageError>;
 declare const unmarshalAiParseDocumentResponseSchema: z.ZodType<AiParseDocumentResponse>;
 declare const unmarshalAiParseDocumentResponseMetadataSchema: z.ZodType<AiParseDocumentResponseMetadata>;
+declare const unmarshalBatchSchema: z.ZodType<Batch>;
+declare const unmarshalBatchRequestCountsSchema: z.ZodType<BatchRequestCounts>;
+declare const unmarshalCancelBatchResponseSchema: z.ZodType<CancelBatchResponse>;
+declare const unmarshalListBatchesResponseSchema: z.ZodType<ListBatchesResponse>;
 declare const marshalAiClassifyOptionsSchema: z.ZodType;
 declare const marshalAiClassifyRequestSchema: z.ZodType;
 declare const marshalAiDecideOptionsSchema: z.ZodType;
@@ -312,6 +392,9 @@ declare const marshalAiExtractOptionsSchema: z.ZodType;
 declare const marshalAiExtractRequestSchema: z.ZodType;
 declare const marshalAiParseDocumentOptionsSchema: z.ZodType;
 declare const marshalAiParseDocumentRequestSchema: z.ZodType;
+declare const marshalBatchSchema: z.ZodType;
+declare const marshalBatchRequestCountsSchema: z.ZodType;
+declare const marshalCancelBatchRequestSchema: z.ZodType;
 //#endregion
-export { AiClassifyOptions, AiClassifyRequest, AiClassifyResponse, AiClassifyResponseMetadata, AiDecideOptions, AiDecideRequest, AiDecideResponse, AiDecideResponseMetadata, AiExtractBbox, AiExtractCitation, AiExtractOptions, AiExtractRequest, AiExtractResponse, AiExtractResponseMetadata, AiParseDocumentFileMetadata, AiParseDocumentOptions, AiParseDocumentPageError, AiParseDocumentRequest, AiParseDocumentResponse, AiParseDocumentResponseMetadata, marshalAiClassifyOptionsSchema, marshalAiClassifyRequestSchema, marshalAiDecideOptionsSchema, marshalAiDecideRequestSchema, marshalAiExtractOptionsSchema, marshalAiExtractRequestSchema, marshalAiParseDocumentOptionsSchema, marshalAiParseDocumentRequestSchema, unmarshalAiClassifyResponseMetadataSchema, unmarshalAiClassifyResponseSchema, unmarshalAiDecideResponseMetadataSchema, unmarshalAiDecideResponseSchema, unmarshalAiExtractBboxSchema, unmarshalAiExtractCitationSchema, unmarshalAiExtractResponseMetadataSchema, unmarshalAiExtractResponseSchema, unmarshalAiParseDocumentFileMetadataSchema, unmarshalAiParseDocumentPageErrorSchema, unmarshalAiParseDocumentResponseMetadataSchema, unmarshalAiParseDocumentResponseSchema };
+export { AiClassifyOptions, AiClassifyRequest, AiClassifyResponse, AiClassifyResponseMetadata, AiDecideOptions, AiDecideRequest, AiDecideResponse, AiDecideResponseMetadata, AiExtractBbox, AiExtractCitation, AiExtractOptions, AiExtractRequest, AiExtractResponse, AiExtractResponseMetadata, AiParseDocumentFileMetadata, AiParseDocumentOptions, AiParseDocumentPageError, AiParseDocumentRequest, AiParseDocumentResponse, AiParseDocumentResponseMetadata, Batch, BatchRequestCounts, BatchStatus, CancelBatchRequest, CancelBatchResponse, CreateBatchRequest, GetBatchRequest, ListBatchesRequest, ListBatchesResponse, marshalAiClassifyOptionsSchema, marshalAiClassifyRequestSchema, marshalAiDecideOptionsSchema, marshalAiDecideRequestSchema, marshalAiExtractOptionsSchema, marshalAiExtractRequestSchema, marshalAiParseDocumentOptionsSchema, marshalAiParseDocumentRequestSchema, marshalBatchRequestCountsSchema, marshalBatchSchema, marshalCancelBatchRequestSchema, unmarshalAiClassifyResponseMetadataSchema, unmarshalAiClassifyResponseSchema, unmarshalAiDecideResponseMetadataSchema, unmarshalAiDecideResponseSchema, unmarshalAiExtractBboxSchema, unmarshalAiExtractCitationSchema, unmarshalAiExtractResponseMetadataSchema, unmarshalAiExtractResponseSchema, unmarshalAiParseDocumentFileMetadataSchema, unmarshalAiParseDocumentPageErrorSchema, unmarshalAiParseDocumentResponseMetadataSchema, unmarshalAiParseDocumentResponseSchema, unmarshalBatchRequestCountsSchema, unmarshalBatchSchema, unmarshalCancelBatchResponseSchema, unmarshalListBatchesResponseSchema };
 //# sourceMappingURL=model.d.ts.map

package/dist/v1/model.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"model.d.ts","names":[],"sources":["../../src/v1/model.ts"],"mappings":";;;;;UAiBiB,iBAAA;EAAA;EAEf,YAAA;;EAEA,OAAA;EAFA;EAIA,UAAA;EAAA;EAEA,sBAAA;EAEA;EAAA,gBAAA;AAAA;AAAA,UAGe,iBAAA;EAAiB;EAEhC,OAAA,GAAU,SAAA;EAAA;EAEV,MAAA,GAAS,SAAA;EAEC;EAAV,OAAA,GAAU,iBAAA;AAAA;AAAA,UAGK,kBAAA;EAPL;EASV,QAAA,GAAW,SAAA;EAPF;EAST,QAAA,GAAW,0BAAA;AAAA;AAAA,UAGI,0BAAA;EAVY;EAY3B,OAAA;AAAA;;UAIe,eAAA;EAXf;EAaA,OAAA;AAAA;AAAA,UAGe,eAAA;EAdsB;EAgBrC,KAAA,GAAQ,SAAA;EAbO;;;;;AAMjB;;;;;AAKA;;;;;;;;;;;;;;;;AA0EA;;;;;;;;;;AAkEA;;;;;AAMA;;;;;AAQA;;;;;;;;;;;AAYA;;;;;;;;;EAjGE,SAAA,GAAY,SAAA;EAiHU;EA/GtB,OAAA,GAAU,eAAA;AAAA;AAAA,UAGK,gBAAA;EAiHL;;;;;;;;;;;;;AAOZ;;;;;;;;;;AAOA;;;;;;;;;;;AAYA;;;;;;;;;;AAYA;;;;;;;;;;AAYA;;;;;AAOA;EA7GE,QAAA,GAAW,SAAA;;EAEX,QAAA,GAAW,wBAAA;AAAA;AAAA,UAGI,wBAAA;EA4GL;EA1GV,OAAA;AAAA;AA6GF;AAAA,UAzGiB,aAAA;;EAEf,KAAA;EA2Gc;EAzGd,MAAA;AAAA;;UAIe,iBAAA;EAmGJ;EAjGX,EAAA;EAmGc;EAjGd,KAAA;EAmGW;EAjGX,IAAA;EAiG0C;EA/F1C,IAAA,GAAO,aAAA;AAAA;;UAIQ,gBAAA;EAgGf;EA9FA,YAAA;EAkGA;EAhGA,OAAA;EAgG0C;EA9F1C,IAAA;EAiGW;;;;;;;EAzFX,eAAA;EAyF0E;EAvF1E,sBAAA;AAAA;AAAA,UAGe,gBAAA;EAiGgD;EA/F/D,OAAA,GAAU,SAAA;EA+F8C;EA7FxD,MAAA,GAAS,SAAA;EA6FiF;EA3F1F,OAAA,GAAU,gBAAA;AAAA;AAAA,UAGK,iBAAA;;EAEf,QAAA,GAAW,SAAA;EA+FiC;EA7F5C,QAAA,GAAW,yBAAA;AAAA;AAAA,UAGI,yBAAA;EA0FuD;EAxFtE,OAAA;EAyGK;EAvGL,IAAA;EAgG6D;EA9F7D,SAAA;EA8FsD;EA5FtD,SAAA,GAAY,iBAAA;AAAA;;UAIG,2BAAA;EAgHZ;EA9GH,QAAA;EA+FkD;EA7FlD,QAAA;EA6F2C;EA3F3C,QAAA;EA2FgE;EAzFhE,oBAAA;AAAA;;UAIe,sBAAA;EAsGuC;EApGtD,OAAA;EAoG+C;EAlG/C,uBAAA;EAkGwE;EAhGxE,eAAA;EAuHW;EArHX,SAAA;AAAA;;UAIe,wBAAA;EAiHgC;EA/G/C,YAAA;EA+GwE;EA7GxE,MAAA;AAAA;AAAA,UAGe,sBAAA;;EAEf,OAAA;EAkHqD;EAhHrD,OAAA,GAAU,sBAAA;AAAA;AAAA,UAGK,uBAAA;EA6GyE;EA3GxF,QAAA,GAAW,SAAA;EA4IN;EA1IL,WAAA,GAAc,wBAAA;EA0HkD;EAxHhE,QAAA,GAAW,+BAAA;AAAA;AAAA,UAGI,+BAAA;EAqH6E;EAnH5F,EAAA;EAqIW;EAnIX,OAAA;;EAEA,YAAA,GAAe,2BAAA;AAAA;AAAA,cAGJ,iCAAA,EAAmC,CAAA,CAAE,OAAA,CAAQ,kBAAA;AAAA,cAa7C,yCAAA,EAA2C,CAAA,CAAE,OAAA,CAAQ,0BAAA;AAAA,cASrD,+BAAA,EAAiC,CAAA,CAAE,OAAA,CAAQ,gBAAA;AAAA,cAU3C,uCAAA,EAAyC,CAAA,CAAE,OAAA,CAAQ,wBAAA;AAAA,cASnD,4BAAA,EAA8B,CAAA,CAAE,OAAA,CAAQ,aAAA;AAAA,cAiBxC,gCAAA,EAAkC,CAAA,CAAE,OAAA,CAAQ,iBAAA;AAAA,cAuB5C,gCAAA,EAAkC,CAAA,CAAE,OAAA,CAAQ,iBAAA;AAAA,cAU5C,wCAAA,EAA0C,CAAA,CAAE,OAAA,CAAQ,yBAAA;AAAA,cAiBpD,0CAAA,EAA4C,CAAA,CAAE,OAAA,CAAQ,2BAAA;AAAA,cAkBtD,uCAAA,EAAyC,CAAA,CAAE,OAAA,CAAQ,wBAAA;AAAA,cAcnD,sCAAA,EAAwC,CAAA,CAAE,OAAA,CAAQ,uBAAA;AAAA,cAiBlD,8CAAA,EAAgD,CAAA,CAAE,OAAA,CAAQ,+BAAA;AAAA,cAe1D,8BAAA,EAAgC,CAAA,CAAE,OAAA;AAAA,cAgBlC,8BAAA,EAAgC,CAAA,CAAE,OAAA;AAAA,cAYlC,4BAAA,EAA8B,CAAA,CAAE,OAAA;AAAA,cAQhC,4BAAA,EAA8B,CAAA,CAAE,OAAA;AAAA,cAYhC,6BAAA,EAA+B,CAAA,CAAE,OAAA;AAAA,cAgBjC,6BAAA,EAA+B,CAAA,CAAE,OAAA;AAAA,cAYjC,mCAAA,EAAqC,CAAA,CAAE,OAAA;AAAA,cAcvC,mCAAA,EAAqC,CAAA,CAAE,OAAA"}
+{"version":3,"file":"model.d.ts","names":[],"sources":["../../src/v1/model.ts"],"mappings":";;;;;;cAuBa,WAAA;EAAA,+DAkBH;EAAA,6BAAA;EAAA;;;;;mCACE;EAAA;;KAAA,WAAA,WACA,WAAA,eAA0B,WAAA;;UAIrB,iBAAA;EAAiB;EAEhC,YAAA;EAFgC;EAIhC,OAAA;EAAA;EAEA,UAAA;EAEA;EAAA,sBAAA;EAEgB;EAAhB,gBAAA;AAAA;AAAA,UAGe,iBAAA;;EAEf,OAAA,GAAU,SAAA;EAED;EAAT,MAAA,GAAS,SAAA;EAEkB;EAA3B,OAAA,GAAU,iBAAA;AAAA;AAAA,UAGK,kBAAA;EALf;EAOA,QAAA,GAAW,SAAA;EALX;EAOA,QAAA,GAAW,0BAAA;AAAA;AAAA,UAGI,0BAAA;EAPA;EASf,OAAA;AAAA;;UAIe,eAAA;EAXJ;EAaX,OAAA;AAAA;AAAA,UAGe,eAAA;EAdsB;EAgBrC,KAAA,GAAQ,SAAA;EAbiC;;;;AAM3C;;;;;AAKA;;;;;;;;;;;;;;;;AA0EA;;;;;;;;;;AAkEA;;;;;AAMA;;;;;AAQA;;;;;;;;;;;AAYA;;;;;;;;;;EAjGE,SAAA,GAAY,SAAA;EAoHG;EAlHf,OAAA,GAAU,eAAA;AAAA;AAAA,UAGK,gBAAA;EAmHN;;;;;;;;;;;;AAKX;;;;;;;;;;AAOA;;;;;;;;;;;AAYA;;;;;;;;;;AAYA;;;;;;;;;;AAYA;;;;;AAOA;;EA7GE,QAAA,GAAW,SAAA;EAiHqB;EA/GhC,QAAA,GAAW,wBAAA;AAAA;AAAA,UAGI,wBAAA;EA4GiB;EA1GhC,OAAA;AAAA;;UAIe,aAAA;EA2GJ;EAzGX,KAAA;EA6GW;EA3GX,MAAA;AAAA;;UAIe,iBAAA;EAqGf;EAnGA,EAAA;EAqGA;EAnGA,KAAA;EAmG0C;EAjG1C,IAAA;EAoGe;EAlGf,IAAA,GAAO,aAAA;AAAA;;UAIQ,gBAAA;EAkGf;EAhGA,YAAA;EAkGe;EAhGf,OAAA;EAgG0C;EA9F1C,IAAA;EAkGoB;;;;;;;EA1FpB,eAAA;EA4FA;EA1FA,sBAAA;AAAA;AAAA,UAGe,gBAAA;EA6FN;EA3FT,OAAA,GAAU,SAAA;EAmGV;EAjGA,MAAA,GAAS,SAAA;EAmGT;EAjGA,OAAA,GAAU,gBAAA;AAAA;AAAA,UAGK,iBAAA;EAgGC;EA9FhB,QAAA,GAAW,SAAA;EAmGI;EAjGf,QAAA,GAAW,yBAAA;AAAA;AAAA,UAGI,yBAAA;;EAEf,OAAA;EAkGA;EAhGA,IAAA;EAoGA;EAlGA,SAAA;EAkGM;EAhGN,SAAA,GAAY,iBAAA;AAAA;;UAIG,2BAAA;EAiGX;EA/FJ,QAAA;EAmGkC;EAjGlC,QAAA;EAiGkC;EA/FlC,QAAA;EAiGe;EA/Ff,oBAAA;AAAA;;UAIe,sBAAA;EAgGA;EA9Ff,OAAA;;EAEA,uBAAA;EA8FI;EA5FJ,eAAA;EA+FiC;EA7FjC,SAAA;AAAA;;UAIe,wBAAA;EAgGmB;EA9FlC,YAAA;EAgGe;EA9Ff,MAAA;AAAA;AAAA,UAGe,sBAAA;EA6FF;EA3Fb,OAAA;EA8FW;EA5FX,OAAA,GAAU,sBAAA;AAAA;AAAA,UAGK,uBAAA;EAyF+B;EAvF9C,QAAA,GAAW,SAAA;EAuF6C;EArFxD,WAAA,GAAc,wBAAA;EAqF4D;EAnF1E,QAAA,GAAW,+BAAA;AAAA;AAAA,UAGI,+BAAA;EA6FgD;EA3F/D,EAAA;EA2FwD;EAzFxD,OAAA;EAyF0F;EAvF1F,YAAA,GAAe,2BAAA;AAAA;;UAIA,KAAA;EA4FsC;EA1FrD,IAAA;EA0F8C;EAxF9C,QAAA;EAwFsE;EAtFtE,QAAA;EAgGW;EA9FX,MAAA,GAAS,UAAA;;;;;;EAMT,SAAA;EAwFsF;EAtFtF,MAAA,GAAS,WAAA;EA8GN;EA5GH,UAAA,GAAa,QAAA,CAAS,OAAA;EA6F4B;EA3FlD,aAAA,GAAgB,kBAAA;EA2F2B;;;;EAtF3C,eAAA;AAAA;;UAIe,kBAAA;EAmG8B;EAjG7C,KAAA;EAiGuD;EA/FvD,SAAA;EA+FwE;EA7FxE,MAAA;AAAA;AAAA,UAGe,kBAAA;EAiHuC;EA/GtD,IAAA;AAAA;AAAA,UAIe,mBAAA;AAAA,UAEA,kBAAA;EAmHJ;EAjHX,KAAA,GAAQ,KAAA;AAAA;AAAA,UAGO,eAAA;EA8GsC;EA5GrD,IAAA;AAAA;AAAA,UAGe,kBAAA;EAyGyE;EAvGxF,QAAA;EAwIK;EAtIL,SAAA;AAAA;AAAA,UAGe,mBAAA;EAmH0C;EAjHzD,OAAA,GAAU,KAAA;EAiHkF;EA/G5F,aAAA;AAAA;AAAA,cAGW,iCAAA,EAAmC,CAAA,CAAE,OAAA,CAAQ,kBAAA;AAAA,cAa7C,yCAAA,EAA2C,CAAA,CAAE,OAAA,CAAQ,0BAAA;AAAA,cASrD,+BAAA,EAAiC,CAAA,CAAE,OAAA,CAAQ,gBAAA;AAAA,cAU3C,uCAAA,EAAyC,CAAA,CAAE,OAAA,CAAQ,wBAAA;AAAA,cASnD,4BAAA,EAA8B,CAAA,CAAE,OAAA,CAAQ,aAAA;AAAA,cAiBxC,gCAAA,EAAkC,CAAA,CAAE,OAAA,CAAQ,iBAAA;AAAA,cAuB5C,gCAAA,EAAkC,CAAA,CAAE,OAAA,CAAQ,iBAAA;AAAA,cAU5C,wCAAA,EAA0C,CAAA,CAAE,OAAA,CAAQ,yBAAA;AAAA,cAiBpD,0CAAA,EAA4C,CAAA,CAAE,OAAA,CAAQ,2BAAA;AAAA,cAkBtD,uCAAA,EAAyC,CAAA,CAAE,OAAA,CAAQ,wBAAA;AAAA,cAcnD,sCAAA,EAAwC,CAAA,CAAE,OAAA,CAAQ,uBAAA;AAAA,cAiBlD,8CAAA,EAAgD,CAAA,CAAE,OAAA,CAAQ,+BAAA;AAAA,cAe1D,oBAAA,EAAsB,CAAA,CAAE,OAAA,CAAQ,KAAA;AAAA,cA2BhC,iCAAA,EAAmC,CAAA,CAAE,OAAA,CAAQ,kBAAA;AAAA,cAsB7C,kCAAA,EAAoC,CAAA,CAAE,OAAA,CAAQ,mBAAA;AAAA,cAG9C,kCAAA,EAAoC,CAAA,CAAE,OAAA,CAAQ,mBAAA;AAAA,cAW9C,8BAAA,EAAgC,CAAA,CAAE,OAAA;AAAA,cAgBlC,8BAAA,EAAgC,CAAA,CAAE,OAAA;AAAA,cAYlC,4BAAA,EAA8B,CAAA,CAAE,OAAA;AAAA,cAQhC,4BAAA,EAA8B,CAAA,CAAE,OAAA;AAAA,cAYhC,6BAAA,EAA+B,CAAA,CAAE,OAAA;AAAA,cAgBjC,6BAAA,EAA+B,CAAA,CAAE,OAAA;AAAA,cAYjC,mCAAA,EAAqC,CAAA,CAAE,OAAA;AAAA,cAcvC,mCAAA,EAAqC,CAAA,CAAE,OAAA;AAAA,cAUvC,kBAAA,EAAoB,CAAA,CAAE,OAAA;AAAA,cA2BtB,+BAAA,EAAiC,CAAA,CAAE,OAAA;AAAA,cAYnC,+BAAA,EAAiC,CAAA,CAAE,OAAA"}

package/dist/v1/model.js

@@ -1,3 +1,4 @@
+import { Temporal } from "@js-temporal/polyfill";
 import { z } from "zod";
 
 //#region src/v1/model.ts
@@ -9,6 +10,19 @@ const jsonValueSchema = z.lazy(() => z.union([
 	z.record(z.string(), jsonValueSchema),
 	z.array(jsonValueSchema)
 ]));
+const jsonObjectSchema = z.record(z.string(), jsonValueSchema);
+/** The status of a batch. */
+const BatchStatus = {
+	BATCH_STATUS_UNSPECIFIED: "BATCH_STATUS_UNSPECIFIED",
+	PENDING: "PENDING",
+	IN_PROGRESS: "IN_PROGRESS",
+	FINALIZING: "FINALIZING",
+	COMPLETED: "COMPLETED",
+	EXPIRED: "EXPIRED",
+	CANCELLING: "CANCELLING",
+	CANCELLED: "CANCELLED",
+	FAILED: "FAILED"
+};
 const unmarshalAiClassifyResponseSchema = z.object({
 	response: jsonValueSchema.optional(),
 	metadata: z.lazy(() => unmarshalAiClassifyResponseMetadataSchema).optional()
@@ -125,6 +139,56 @@ const unmarshalAiParseDocumentResponseMetadataSchema = z.object({
 	version: d.version,
 	fileMetadata: d.file_metadata
 }));
+const unmarshalBatchSchema = z.object({
+	name: z.string().optional(),
+	endpoint: z.string().optional(),
+	input_ref: z.string().optional(),
+	params: jsonObjectSchema.optional(),
+	output_ref: z.string().optional(),
+	status: z.string().optional(),
+	create_time: z.string().transform((s) => Temporal.Instant.from(s)).optional(),
+	request_counts: z.lazy(() => unmarshalBatchRequestCountsSchema).optional(),
+	percent_complete: z.number().optional()
+}).transform((d) => ({
+	name: d.name,
+	endpoint: d.endpoint,
+	inputRef: d.input_ref,
+	params: d.params,
+	outputRef: d.output_ref,
+	status: d.status,
+	createTime: d.create_time,
+	requestCounts: d.request_counts,
+	percentComplete: d.percent_complete
+}));
+const unmarshalBatchRequestCountsSchema = z.object({
+	total: z.union([
+		z.number(),
+		z.bigint(),
+		z.string()
+	]).transform((v) => BigInt(v)).optional(),
+	succeeded: z.union([
+		z.number(),
+		z.bigint(),
+		z.string()
+	]).transform((v) => BigInt(v)).optional(),
+	failed: z.union([
+		z.number(),
+		z.bigint(),
+		z.string()
+	]).transform((v) => BigInt(v)).optional()
+}).transform((d) => ({
+	total: d.total,
+	succeeded: d.succeeded,
+	failed: d.failed
+}));
+const unmarshalCancelBatchResponseSchema = z.object({});
+const unmarshalListBatchesResponseSchema = z.object({
+	batches: z.array(z.lazy(() => unmarshalBatchSchema)).optional(),
+	next_page_token: z.string().optional()
+}).transform((d) => ({
+	batches: d.batches,
+	nextPageToken: d.next_page_token
+}));
 const marshalAiClassifyOptionsSchema = z.object({
 	instructions: z.string().optional(),
 	version: z.string().optional(),
@@ -197,7 +261,38 @@ const marshalAiParseDocumentRequestSchema = z.object({
 	content: d.content,
 	options: d.options
 }));
+const marshalBatchSchema = z.object({
+	name: z.string().optional(),
+	endpoint: z.string().optional(),
+	inputRef: z.string().optional(),
+	params: jsonObjectSchema.optional(),
+	outputRef: z.string().optional(),
+	status: z.string().optional(),
+	createTime: z.any().transform((d) => d.toString()).optional(),
+	requestCounts: z.lazy(() => marshalBatchRequestCountsSchema).optional(),
+	percentComplete: z.number().optional()
+}).transform((d) => ({
+	name: d.name,
+	endpoint: d.endpoint,
+	input_ref: d.inputRef,
+	params: d.params,
+	output_ref: d.outputRef,
+	status: d.status,
+	create_time: d.createTime,
+	request_counts: d.requestCounts,
+	percent_complete: d.percentComplete
+}));
+const marshalBatchRequestCountsSchema = z.object({
+	total: z.bigint().optional(),
+	succeeded: z.bigint().optional(),
+	failed: z.bigint().optional()
+}).transform((d) => ({
+	total: d.total,
+	succeeded: d.succeeded,
+	failed: d.failed
+}));
+const marshalCancelBatchRequestSchema = z.object({ name: z.string().optional() }).transform((d) => ({ name: d.name }));
 
 //#endregion
-export { marshalAiClassifyOptionsSchema, marshalAiClassifyRequestSchema, marshalAiDecideOptionsSchema, marshalAiDecideRequestSchema, marshalAiExtractOptionsSchema, marshalAiExtractRequestSchema, marshalAiParseDocumentOptionsSchema, marshalAiParseDocumentRequestSchema, unmarshalAiClassifyResponseMetadataSchema, unmarshalAiClassifyResponseSchema, unmarshalAiDecideResponseMetadataSchema, unmarshalAiDecideResponseSchema, unmarshalAiExtractBboxSchema, unmarshalAiExtractCitationSchema, unmarshalAiExtractResponseMetadataSchema, unmarshalAiExtractResponseSchema, unmarshalAiParseDocumentFileMetadataSchema, unmarshalAiParseDocumentPageErrorSchema, unmarshalAiParseDocumentResponseMetadataSchema, unmarshalAiParseDocumentResponseSchema };
+export { BatchStatus, marshalAiClassifyOptionsSchema, marshalAiClassifyRequestSchema, marshalAiDecideOptionsSchema, marshalAiDecideRequestSchema, marshalAiExtractOptionsSchema, marshalAiExtractRequestSchema, marshalAiParseDocumentOptionsSchema, marshalAiParseDocumentRequestSchema, marshalBatchRequestCountsSchema, marshalBatchSchema, marshalCancelBatchRequestSchema, unmarshalAiClassifyResponseMetadataSchema, unmarshalAiClassifyResponseSchema, unmarshalAiDecideResponseMetadataSchema, unmarshalAiDecideResponseSchema, unmarshalAiExtractBboxSchema, unmarshalAiExtractCitationSchema, unmarshalAiExtractResponseMetadataSchema, unmarshalAiExtractResponseSchema, unmarshalAiParseDocumentFileMetadataSchema, unmarshalAiParseDocumentPageErrorSchema, unmarshalAiParseDocumentResponseMetadataSchema, unmarshalAiParseDocumentResponseSchema, unmarshalBatchRequestCountsSchema, unmarshalBatchSchema, unmarshalCancelBatchResponseSchema, unmarshalListBatchesResponseSchema };
 //# sourceMappingURL=model.js.map

package/dist/v1/model.js.map

@@ -1 +1 @@
-{"version":3,"file":"model.js","names":[],"sources":["../../src/v1/model.ts"],"sourcesContent":["// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.\n\nimport type {JsonValue} from '@databricks/sdk-core/wkt';\nimport {z} from 'zod';\n\nconst jsonValueSchema: z.ZodType<JsonValue> = z.lazy(() =>\n  z.union([\n    z.null(),\n    z.number(),\n    z.string(),\n    z.boolean(),\n    z.record(z.string(), jsonValueSchema),\n    z.array(jsonValueSchema),\n  ])\n);\n\n/** ai_classify */\nexport interface AiClassifyOptions {\n  /** Natural-language guidance that steers how the text is classified (up to 20,000 characters). */\n  instructions?: string | undefined;\n  /** The function version to invoke. Defaults to the latest version. Supported versions: [\"2.1\"]. */\n  version?: string | undefined;\n  /** When true, allows more than one label to be returned per input. */\n  multilabel?: boolean | undefined;\n  /** When true, includes a per-label confidence score in the response. */\n  enableConfidenceScores?: boolean | undefined;\n  /** When true, includes a rationale explaining each classification in the response. */\n  enableRationales?: boolean | undefined;\n}\n\nexport interface AiClassifyRequest {\n  /** The content to classify. It accepts a plain string or the response object of [ai_parse_document](:method:AiFunctions/AiParseDocument). */\n  content?: JsonValue | undefined;\n  /** The label set to classify as. Either a JSON array of label strings (e.g. [\"spam\", \"not_spam\"]), or a JSON object mapping each label to a description (e.g. {\"spam\": \"unsolicited bulk message\", \"not_spam\": \"a legitimate message\"}). Accepts 2 to 500 labels, each 1 to 100 characters. */\n  labels?: JsonValue | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiClassifyOptions | undefined;\n}\n\nexport interface AiClassifyResponse {\n  /** The function result as a JSON value. An array of per-label objects: one element in single-label mode (the default), or multiple elements when `multilabel` is true. When `enable_confidence_scores` and `enable_rationales` are true, `confidence_score` and `rationale` are included in each response value, respectively. */\n  response?: JsonValue | undefined;\n  /** Additional metadata returned by AI Classify. */\n  metadata?: AiClassifyResponseMetadata | undefined;\n}\n\nexport interface AiClassifyResponseMetadata {\n  /** The resolved function version. */\n  version?: string | undefined;\n}\n\n/** ai_decide */\nexport interface AiDecideOptions {\n  /** The function API version to invoke. Defaults to \"1.0\". Supported versions: [\"1.0\"]. */\n  version?: string | undefined;\n}\n\nexport interface AiDecideRequest {\n  /** A string, JSON object, or array containing the content, related context, and examples needed to answer the provided questions. For example, provide a support message, a conversation, or records describing the current state of an application. All questions receive this same state. */\n  state?: JsonValue | undefined;\n  /**\n   * A JSON object mapping question IDs to their definitions. Each ID must contain non-whitespace text; its answer is returned with the same ID in `response.answers`.\n   *\n   * Each definition is an object with the required fields `type` and `instructions`. The `criteria` field is optional for the type `noul` but is required for the types `choice` and `score`.\n   *\n   * The `instructions` field describes the judgment to make and can be a string, object, or array. Use an object or array to include supporting context alongside the instructions.\n   *\n   * The `type` can be one of:\n   *\n   * - `choice`: Selects one option from a defined set. Requires `criteria` to be an object mapping 1 to 255 nonempty option names to descriptions. The criteria description can be a string, object, array, or null when the name needs no additional detail. For example:\n   *\n   * ```json\n   * {\n   * \"team\": {\n   * \"type\": \"choice\",\n   * \"instructions\": \"Which team should handle this ticket?\",\n   * \"criteria\": {\n   * \"billing\": \"Payments, charges, and refunds\",\n   * \"technical_support\": null\n   * }\n   * }\n   * }\n   * ```\n   *\n   * - `noul`: Estimates the probability that the answer to a true-or-false question is true. `criteria` can take the fields `true` or `false`, or both, with descriptions that are strings, objects, or arrays. Omit `criteria` to use the question alone. The instructions or at least one criteria description must contain non-whitespace text, a nonempty object, or a nonempty array. For example, both of the following are valid:\n   *\n   * ```json\n   * {\n   * \"escalate\": {\n   * \"type\": \"noul\",\n   * \"instructions\": \"Does this ticket need escalation?\",\n   * \"criteria\": {\n   * \"true\": \"Suspected fraud or an exception to standard policy\",\n   * \"false\": \"A routine issue frontline support can resolve\"\n   * }\n   * }\n   * }\n   * ```\n   *\n   * or\n   *\n   * ```json\n   * {\n   * \"escalate\": {\n   * \"type\": \"noul\",\n   * \"instructions\": \"Does this ticket need escalation?\"\n   * }\n   * }\n   * ```\n   *\n   * - `score`: Rates the state on an ordered scale. Requires `criteria` to be an array of 2 to 10 level descriptions, ordered from low to high. Descriptions must contain non-whitespace text, a nonempty object, or a nonempty array. Array positions define levels starting at 0. For example:\n   *\n   * ```json\n   * {\n   * \"urgency\": {\n   * \"type\": \"score\",\n   * \"instructions\": \"How urgent is this ticket?\",\n   * \"criteria\": [\n   * \"Routine: can wait a few days\",\n   * \"Time-sensitive: needs attention today\",\n   * \"Critical: needs immediate action\"\n   * ]\n   * }\n   * }\n   * ```\n   */\n  questions?: JsonValue | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiDecideOptions | undefined;\n}\n\nexport interface AiDecideResponse {\n  /**\n   * A JSON object containing an `answers` map with one answer for each question, keyed by the same IDs as `questions`.\n   *\n   * Each answer has a `type` matching its question and the following fields:\n   *\n   * - `choice`: choice is the highest-probability option from `criteria`. `probabilities` maps every option name to its probability. `confidence` is a number from 0 to 1 indicating how well the state supports the assessment. The `probabilities` values sum to 1. For example:\n   *\n   * ```json\n   * {\n   * \"answers\": {\n   * \"team\": {\n   * \"type\": \"choice\",\n   * \"choice\": \"billing\",\n   * \"probabilities\": {\n   * \"billing\": 0.85,\n   * \"technical_support\": 0.15\n   * },\n   * \"confidence\": 0.9\n   * }\n   * }\n   * }\n   * ```\n   *\n   * - `noul`: `probability` is a number from 0 to 1 estimating the probability that the answer is true. For example:\n   *\n   * ```json\n   * {\n   * \"answers\": {\n   * \"escalate\": {\n   * \"type\": \"noul\",\n   * \"probability\": 0.8\n   * }\n   * }\n   * }\n   * ```\n   *\n   * - `score`: score is the probability-weighted mean of the zero-based level indices. It can fall between levels, from 0 to the number of levels minus 1. `probabilities` maps each level index to its probability, and `legend` maps each index to its original description. Both maps use string keys such as \"0\", \"1\", and \"2\". `confidence` is a number from 0 to 1 indicating how well the state supports the assessment. The `probabilities` values sum to 1. For example:\n   *\n   * ```json\n   * {\n   * \"answers\": {\n   * \"urgency\": {\n   * \"type\": \"score\",\n   * \"score\": 1.6,\n   * \"probabilities\": {\n   * \"0\": 0.1,\n   * \"1\": 0.2,\n   * \"2\": 0.7\n   * },\n   * \"legend\": {\n   * \"0\": \"Routine: can wait a few days\",\n   * \"1\": \"Time-sensitive: needs attention today\",\n   * \"2\": \"Critical: needs immediate action\"\n   * },\n   * \"confidence\": 0.85\n   * }\n   * }\n   * }\n   * ```\n   */\n  response?: JsonValue | undefined;\n  /** Metadata identifying the function API version used for this request. */\n  metadata?: AiDecideResponseMetadata | undefined;\n}\n\nexport interface AiDecideResponseMetadata {\n  /** The function API version used to evaluate the request. */\n  version?: string | undefined;\n}\n\n/** A bounding box on a source page; used by bbox-input citations. */\nexport interface AiExtractBbox {\n  /** Pixel coordinates on the page image as [x0, y0, x1, y1]. */\n  coord?: bigint[] | undefined;\n  /** 0-based page index the box is on. */\n  pageId?: bigint | undefined;\n}\n\n/** A citation locating an extracted value in the source. start/stop are set for span (STRING input) citations, bbox for bbox (parsed-document input) citations. */\nexport interface AiExtractCitation {\n  /** Integer matching a citation_ids entry on an extracted field. */\n  id?: bigint | undefined;\n  /** Inclusive 0-based character offset into the input string; set for span citations. */\n  start?: bigint | undefined;\n  /** Exclusive 0-based character offset into the input string; set for span citations. */\n  stop?: bigint | undefined;\n  /** Bounding boxes locating the citation on the source pages; set for bbox citations. */\n  bbox?: AiExtractBbox[] | undefined;\n}\n\n/** ai_extract */\nexport interface AiExtractOptions {\n  /** Natural-language guidance that steers how data is extracted (up to 20,000 characters). */\n  instructions?: string | undefined;\n  /** The function version to invoke. Defaults to the latest version. Supported versions: [\"2.1\"]. */\n  version?: string | undefined;\n  /** Extraction mode. Supported modes: \"precision\" — more powerful extraction for complex schemas, long documents, and reasoning-heavy extractions. Defaults to none (standard extraction). */\n  mode?: string | undefined;\n  /**\n   * When true, includes citation metadata locating each extracted value in the source. Depending on the type of input, citations can be one of two types:\n   *\n   * For raw text (STRING) inputs, a citation is a span of text in the original input. Each object in `metadata.citations` has an `id` (integer matching a `citation_ids` entry on a field), a `start` (inclusive 0-based character offset into the input string), and a `stop` (exclusive 0-based character offset into the input string).\n   *\n   * For PDF documents and images (when using ai_extract downstream of ai_parse_document), a citation is a bounding box in the original input. Each object in `metadata.citations` has an `id` (integer matching a `citation_ids` entry on a field) and a `bbox` (array of {coord, page_id} objects, identical in shape to element.bbox in ai_parse_document output; coord is pixel coordinates on the page image as [x0, y0, x1, y1], and page_id is a 0-based page index).\n   */\n  enableCitations?: boolean | undefined;\n  /** When true, includes a per-field confidence score in the response. */\n  enableConfidenceScores?: boolean | undefined;\n}\n\nexport interface AiExtractRequest {\n  /** The text to extract from. It accepts a plain string or the response object of [ai_parse_document](:method:AiFunctions/AiParseDocument). */\n  content?: JsonValue | undefined;\n  /** The extraction schema defining the fields to extract. Either a JSON array of field names, assumed to be strings (e.g. [\"company\", \"valuation\"]), or a JSON object mapping each field to its type/description/nullability (e.g. {\"company\": {\"type\": \"string\", \"description\": \"the company name\"}}). Accepts up to 256 fields, 12 levels of nesting, and 500 enum values. Supported field types are string, integer, number, boolean, and enum. */\n  schema?: JsonValue | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiExtractOptions | undefined;\n}\n\nexport interface AiExtractResponse {\n  /** The function result as a JSON value. When `enable_confidence_scores` and `enable_citations` are true, `confidence` and `citation_ids` are included in each response field, respectively. */\n  response?: JsonValue | undefined;\n  /** Additional metadata returned by AI Extract. */\n  metadata?: AiExtractResponseMetadata | undefined;\n}\n\nexport interface AiExtractResponseMetadata {\n  /** The resolved function version. */\n  version?: string | undefined;\n  /** The resolved extraction mode; present when a non-default mode was used. */\n  mode?: string | undefined;\n  /** How the source was chunked for citation offsets (span for text input, bbox for parsed-document input); present when citations are enabled. */\n  chunkType?: string | undefined;\n  /** Citation objects locating each result in the source; present when citations are enabled. */\n  citations?: AiExtractCitation[] | undefined;\n}\n\n/** Metadata about the source file; present only for file-path input. */\nexport interface AiParseDocumentFileMetadata {\n  /** Unity Catalog volume path of the source file. */\n  filePath?: string | undefined;\n  /** Base name of the source file. */\n  fileName?: string | undefined;\n  /** Size of the source file in bytes. */\n  fileSize?: bigint | undefined;\n  /** Last-modified timestamp of the source file, as an HTTP date string. */\n  fileModificationTime?: string | undefined;\n}\n\n/** ai_parse_document */\nexport interface AiParseDocumentOptions {\n  /** The ai_parse_document output schema version. Supported value: \"2.0\". */\n  version?: string | undefined;\n  /** Element types for which an AI-generated description is produced. Use \"*\" (default) to generate descriptions for all supported element types, \"figure\" to generate them for figures only, or \"\" (empty string) to generate none. Only figure descriptions are supported for version \"2.0\", so \"*\" and \"figure\" produce the same behavior. */\n  descriptionElementTypes?: string | undefined;\n  /** Unity Catalog volume path where rendered page and element images are written. */\n  imageOutputPath?: string | undefined;\n  /** Pages to parse (1-indexed), as a comma-separated list of page numbers or ranges (e.g. \"1,3,5-10\"). */\n  pageRange?: string | undefined;\n}\n\n/** A single page that failed to parse while the overall request succeeded. */\nexport interface AiParseDocumentPageError {\n  /** Message describing why the page failed. */\n  errorMessage?: string | undefined;\n  /** 0-based index of the page that failed. */\n  pageId?: bigint | undefined;\n}\n\nexport interface AiParseDocumentRequest {\n  /** The document to parse. The REST API accepts a Unity Catalog volume path to the source file or a base64 data URL containing the file bytes (for example, data:application/pdf;base64,<encoded-file-bytes> or data:image/jpeg;base64,<encoded-file-bytes>). Supported formats: PDF, DOCX, DOC, PPTX, PPT, JPG, JPEG, PNG, TIFF. Accepts up to 100 pages and 100 MB per document. */\n  content?: string | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiParseDocumentOptions | undefined;\n}\n\nexport interface AiParseDocumentResponse {\n  /** The parsed document as a JSON value, containing the extracted pages and elements. */\n  document?: JsonValue | undefined;\n  /** Per-page partial-failure details; present when the request succeeds (2xx) but individual pages fail. */\n  errorStatus?: AiParseDocumentPageError[] | undefined;\n  /** Additional metadata returned by AI Parse Document. */\n  metadata?: AiParseDocumentResponseMetadata | undefined;\n}\n\nexport interface AiParseDocumentResponseMetadata {\n  /** Unique identifier for the parse request. */\n  id?: string | undefined;\n  /** The resolved function version. */\n  version?: string | undefined;\n  /** Describes the source file; present only for file-path input. */\n  fileMetadata?: AiParseDocumentFileMetadata | undefined;\n}\n\nexport const unmarshalAiClassifyResponseSchema: z.ZodType<AiClassifyResponse> =\n  z\n    .object({\n      response: jsonValueSchema.optional(),\n      metadata: z\n        .lazy(() => unmarshalAiClassifyResponseMetadataSchema)\n        .optional(),\n    })\n    .transform(d => ({\n      response: d.response,\n      metadata: d.metadata,\n    }));\n\nexport const unmarshalAiClassifyResponseMetadataSchema: z.ZodType<AiClassifyResponseMetadata> =\n  z\n    .object({\n      version: z.string().optional(),\n    })\n    .transform(d => ({\n      version: d.version,\n    }));\n\nexport const unmarshalAiDecideResponseSchema: z.ZodType<AiDecideResponse> = z\n  .object({\n    response: jsonValueSchema.optional(),\n    metadata: z.lazy(() => unmarshalAiDecideResponseMetadataSchema).optional(),\n  })\n  .transform(d => ({\n    response: d.response,\n    metadata: d.metadata,\n  }));\n\nexport const unmarshalAiDecideResponseMetadataSchema: z.ZodType<AiDecideResponseMetadata> =\n  z\n    .object({\n      version: z.string().optional(),\n    })\n    .transform(d => ({\n      version: d.version,\n    }));\n\nexport const unmarshalAiExtractBboxSchema: z.ZodType<AiExtractBbox> = z\n  .object({\n    coord: z\n      .array(\n        z.union([z.number(), z.bigint(), z.string()]).transform(v => BigInt(v))\n      )\n      .optional(),\n    page_id: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n  })\n  .transform(d => ({\n    coord: d.coord,\n    pageId: d.page_id,\n  }));\n\nexport const unmarshalAiExtractCitationSchema: z.ZodType<AiExtractCitation> = z\n  .object({\n    id: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n    start: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n    stop: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n    bbox: z.array(z.lazy(() => unmarshalAiExtractBboxSchema)).optional(),\n  })\n  .transform(d => ({\n    id: d.id,\n    start: d.start,\n    stop: d.stop,\n    bbox: d.bbox,\n  }));\n\nexport const unmarshalAiExtractResponseSchema: z.ZodType<AiExtractResponse> = z\n  .object({\n    response: jsonValueSchema.optional(),\n    metadata: z.lazy(() => unmarshalAiExtractResponseMetadataSchema).optional(),\n  })\n  .transform(d => ({\n    response: d.response,\n    metadata: d.metadata,\n  }));\n\nexport const unmarshalAiExtractResponseMetadataSchema: z.ZodType<AiExtractResponseMetadata> =\n  z\n    .object({\n      version: z.string().optional(),\n      mode: z.string().optional(),\n      chunk_type: z.string().optional(),\n      citations: z\n        .array(z.lazy(() => unmarshalAiExtractCitationSchema))\n        .optional(),\n    })\n    .transform(d => ({\n      version: d.version,\n      mode: d.mode,\n      chunkType: d.chunk_type,\n      citations: d.citations,\n    }));\n\nexport const unmarshalAiParseDocumentFileMetadataSchema: z.ZodType<AiParseDocumentFileMetadata> =\n  z\n    .object({\n      file_path: z.string().optional(),\n      file_name: z.string().optional(),\n      file_size: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n      file_modification_time: z.string().optional(),\n    })\n    .transform(d => ({\n      filePath: d.file_path,\n      fileName: d.file_name,\n      fileSize: d.file_size,\n      fileModificationTime: d.file_modification_time,\n    }));\n\nexport const unmarshalAiParseDocumentPageErrorSchema: z.ZodType<AiParseDocumentPageError> =\n  z\n    .object({\n      error_message: z.string().optional(),\n      page_id: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n    })\n    .transform(d => ({\n      errorMessage: d.error_message,\n      pageId: d.page_id,\n    }));\n\nexport const unmarshalAiParseDocumentResponseSchema: z.ZodType<AiParseDocumentResponse> =\n  z\n    .object({\n      document: jsonValueSchema.optional(),\n      error_status: z\n        .array(z.lazy(() => unmarshalAiParseDocumentPageErrorSchema))\n        .optional(),\n      metadata: z\n        .lazy(() => unmarshalAiParseDocumentResponseMetadataSchema)\n        .optional(),\n    })\n    .transform(d => ({\n      document: d.document,\n      errorStatus: d.error_status,\n      metadata: d.metadata,\n    }));\n\nexport const unmarshalAiParseDocumentResponseMetadataSchema: z.ZodType<AiParseDocumentResponseMetadata> =\n  z\n    .object({\n      id: z.string().optional(),\n      version: z.string().optional(),\n      file_metadata: z\n        .lazy(() => unmarshalAiParseDocumentFileMetadataSchema)\n        .optional(),\n    })\n    .transform(d => ({\n      id: d.id,\n      version: d.version,\n      fileMetadata: d.file_metadata,\n    }));\n\nexport const marshalAiClassifyOptionsSchema: z.ZodType = z\n  .object({\n    instructions: z.string().optional(),\n    version: z.string().optional(),\n    multilabel: z.boolean().optional(),\n    enableConfidenceScores: z.boolean().optional(),\n    enableRationales: z.boolean().optional(),\n  })\n  .transform(d => ({\n    instructions: d.instructions,\n    version: d.version,\n    multilabel: d.multilabel,\n    enable_confidence_scores: d.enableConfidenceScores,\n    enable_rationales: d.enableRationales,\n  }));\n\nexport const marshalAiClassifyRequestSchema: z.ZodType = z\n  .object({\n    content: jsonValueSchema.optional(),\n    labels: jsonValueSchema.optional(),\n    options: z.lazy(() => marshalAiClassifyOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    content: d.content,\n    labels: d.labels,\n    options: d.options,\n  }));\n\nexport const marshalAiDecideOptionsSchema: z.ZodType = z\n  .object({\n    version: z.string().optional(),\n  })\n  .transform(d => ({\n    version: d.version,\n  }));\n\nexport const marshalAiDecideRequestSchema: z.ZodType = z\n  .object({\n    state: jsonValueSchema.optional(),\n    questions: jsonValueSchema.optional(),\n    options: z.lazy(() => marshalAiDecideOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    state: d.state,\n    questions: d.questions,\n    options: d.options,\n  }));\n\nexport const marshalAiExtractOptionsSchema: z.ZodType = z\n  .object({\n    instructions: z.string().optional(),\n    version: z.string().optional(),\n    mode: z.string().optional(),\n    enableCitations: z.boolean().optional(),\n    enableConfidenceScores: z.boolean().optional(),\n  })\n  .transform(d => ({\n    instructions: d.instructions,\n    version: d.version,\n    mode: d.mode,\n    enable_citations: d.enableCitations,\n    enable_confidence_scores: d.enableConfidenceScores,\n  }));\n\nexport const marshalAiExtractRequestSchema: z.ZodType = z\n  .object({\n    content: jsonValueSchema.optional(),\n    schema: jsonValueSchema.optional(),\n    options: z.lazy(() => marshalAiExtractOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    content: d.content,\n    schema: d.schema,\n    options: d.options,\n  }));\n\nexport const marshalAiParseDocumentOptionsSchema: z.ZodType = z\n  .object({\n    version: z.string().optional(),\n    descriptionElementTypes: z.string().optional(),\n    imageOutputPath: z.string().optional(),\n    pageRange: z.string().optional(),\n  })\n  .transform(d => ({\n    version: d.version,\n    description_element_types: d.descriptionElementTypes,\n    image_output_path: d.imageOutputPath,\n    page_range: d.pageRange,\n  }));\n\nexport const marshalAiParseDocumentRequestSchema: z.ZodType = z\n  .object({\n    content: z.string().optional(),\n    options: z.lazy(() => marshalAiParseDocumentOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    content: d.content,\n    options: d.options,\n  }));\n"],"mappings":";;;AAKA,MAAM,kBAAwC,EAAE,WAC9C,EAAE,MAAM;CACN,EAAE,MAAM;CACR,EAAE,QAAQ;CACV,EAAE,QAAQ;CACV,EAAE,SAAS;CACX,EAAE,OAAO,EAAE,QAAQ,EAAE,gBAAgB;CACrC,EAAE,MAAM,gBAAgB;CACzB,CAAC,CACH;AAwTD,MAAa,oCACX,EACG,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,UAAU,EACP,WAAW,0CAA0C,CACrD,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACb,EAAE;AAEP,MAAa,4CACX,EACG,OAAO,EACN,SAAS,EAAE,QAAQ,CAAC,UAAU,EAC/B,CAAC,CACD,WAAU,OAAM,EACf,SAAS,EAAE,SACZ,EAAE;AAEP,MAAa,kCAA+D,EACzE,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,UAAU,EAAE,WAAW,wCAAwC,CAAC,UAAU;CAC3E,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACb,EAAE;AAEL,MAAa,0CACX,EACG,OAAO,EACN,SAAS,EAAE,QAAQ,CAAC,UAAU,EAC/B,CAAC,CACD,WAAU,OAAM,EACf,SAAS,EAAE,SACZ,EAAE;AAEP,MAAa,+BAAyD,EACnE,OAAO;CACN,OAAO,EACJ,MACC,EAAE,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAAC,WAAU,MAAK,OAAO,EAAE,CAAC,CACxE,CACA,UAAU;CACb,SAAS,EACN,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,OAAO,EAAE;CACT,QAAQ,EAAE;CACX,EAAE;AAEL,MAAa,mCAAiE,EAC3E,OAAO;CACN,IAAI,EACD,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,OAAO,EACJ,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,MAAM,EACH,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,MAAM,EAAE,MAAM,EAAE,WAAW,6BAA6B,CAAC,CAAC,UAAU;CACrE,CAAC,CACD,WAAU,OAAM;CACf,IAAI,EAAE;CACN,OAAO,EAAE;CACT,MAAM,EAAE;CACR,MAAM,EAAE;CACT,EAAE;AAEL,MAAa,mCAAiE,EAC3E,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,UAAU,EAAE,WAAW,yCAAyC,CAAC,UAAU;CAC5E,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACb,EAAE;AAEL,MAAa,2CACX,EACG,OAAO;CACN,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,YAAY,EAAE,QAAQ,CAAC,UAAU;CACjC,WAAW,EACR,MAAM,EAAE,WAAW,iCAAiC,CAAC,CACrD,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,MAAM,EAAE;CACR,WAAW,EAAE;CACb,WAAW,EAAE;CACd,EAAE;AAEP,MAAa,6CACX,EACG,OAAO;CACN,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,WAAW,EACR,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,wBAAwB,EAAE,QAAQ,CAAC,UAAU;CAC9C,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACZ,UAAU,EAAE;CACZ,sBAAsB,EAAE;CACzB,EAAE;AAEP,MAAa,0CACX,EACG,OAAO;CACN,eAAe,EAAE,QAAQ,CAAC,UAAU;CACpC,SAAS,EACN,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,cAAc,EAAE;CAChB,QAAQ,EAAE;CACX,EAAE;AAEP,MAAa,yCACX,EACG,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,cAAc,EACX,MAAM,EAAE,WAAW,wCAAwC,CAAC,CAC5D,UAAU;CACb,UAAU,EACP,WAAW,+CAA+C,CAC1D,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,aAAa,EAAE;CACf,UAAU,EAAE;CACb,EAAE;AAEP,MAAa,iDACX,EACG,OAAO;CACN,IAAI,EAAE,QAAQ,CAAC,UAAU;CACzB,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,eAAe,EACZ,WAAW,2CAA2C,CACtD,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,IAAI,EAAE;CACN,SAAS,EAAE;CACX,cAAc,EAAE;CACjB,EAAE;AAEP,MAAa,iCAA4C,EACtD,OAAO;CACN,cAAc,EAAE,QAAQ,CAAC,UAAU;CACnC,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,YAAY,EAAE,SAAS,CAAC,UAAU;CAClC,wBAAwB,EAAE,SAAS,CAAC,UAAU;CAC9C,kBAAkB,EAAE,SAAS,CAAC,UAAU;CACzC,CAAC,CACD,WAAU,OAAM;CACf,cAAc,EAAE;CAChB,SAAS,EAAE;CACX,YAAY,EAAE;CACd,0BAA0B,EAAE;CAC5B,mBAAmB,EAAE;CACtB,EAAE;AAEL,MAAa,iCAA4C,EACtD,OAAO;CACN,SAAS,gBAAgB,UAAU;CACnC,QAAQ,gBAAgB,UAAU;CAClC,SAAS,EAAE,WAAW,+BAA+B,CAAC,UAAU;CACjE,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,QAAQ,EAAE;CACV,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,+BAA0C,EACpD,OAAO,EACN,SAAS,EAAE,QAAQ,CAAC,UAAU,EAC/B,CAAC,CACD,WAAU,OAAM,EACf,SAAS,EAAE,SACZ,EAAE;AAEL,MAAa,+BAA0C,EACpD,OAAO;CACN,OAAO,gBAAgB,UAAU;CACjC,WAAW,gBAAgB,UAAU;CACrC,SAAS,EAAE,WAAW,6BAA6B,CAAC,UAAU;CAC/D,CAAC,CACD,WAAU,OAAM;CACf,OAAO,EAAE;CACT,WAAW,EAAE;CACb,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,gCAA2C,EACrD,OAAO;CACN,cAAc,EAAE,QAAQ,CAAC,UAAU;CACnC,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,iBAAiB,EAAE,SAAS,CAAC,UAAU;CACvC,wBAAwB,EAAE,SAAS,CAAC,UAAU;CAC/C,CAAC,CACD,WAAU,OAAM;CACf,cAAc,EAAE;CAChB,SAAS,EAAE;CACX,MAAM,EAAE;CACR,kBAAkB,EAAE;CACpB,0BAA0B,EAAE;CAC7B,EAAE;AAEL,MAAa,gCAA2C,EACrD,OAAO;CACN,SAAS,gBAAgB,UAAU;CACnC,QAAQ,gBAAgB,UAAU;CAClC,SAAS,EAAE,WAAW,8BAA8B,CAAC,UAAU;CAChE,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,QAAQ,EAAE;CACV,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,sCAAiD,EAC3D,OAAO;CACN,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,yBAAyB,EAAE,QAAQ,CAAC,UAAU;CAC9C,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACtC,WAAW,EAAE,QAAQ,CAAC,UAAU;CACjC,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,2BAA2B,EAAE;CAC7B,mBAAmB,EAAE;CACrB,YAAY,EAAE;CACf,EAAE;AAEL,MAAa,sCAAiD,EAC3D,OAAO;CACN,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,SAAS,EAAE,WAAW,oCAAoC,CAAC,UAAU;CACtE,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,SAAS,EAAE;CACZ,EAAE"}
+{"version":3,"file":"model.js","names":[],"sources":["../../src/v1/model.ts"],"sourcesContent":["// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.\n\nimport {Temporal} from '@js-temporal/polyfill';\nimport type {JsonValue, JsonObject} from '@databricks/sdk-core/wkt';\nimport {z} from 'zod';\n\nconst jsonValueSchema: z.ZodType<JsonValue> = z.lazy(() =>\n  z.union([\n    z.null(),\n    z.number(),\n    z.string(),\n    z.boolean(),\n    z.record(z.string(), jsonValueSchema),\n    z.array(jsonValueSchema),\n  ])\n);\nconst jsonObjectSchema: z.ZodType<JsonObject> = z.record(\n  z.string(),\n  jsonValueSchema\n);\n\n/** The status of a batch. */\n// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.\nexport const BatchStatus = {\n  BATCH_STATUS_UNSPECIFIED: 'BATCH_STATUS_UNSPECIFIED',\n  /** Accepted, not started yet. */\n  PENDING: 'PENDING',\n  /** Requests are being processed. */\n  IN_PROGRESS: 'IN_PROGRESS',\n  /** Every request is processed and the output file is being written. */\n  FINALIZING: 'FINALIZING',\n  /** The output is written and available. */\n  COMPLETED: 'COMPLETED',\n  /** The completion window elapsed. The output holds the requests that finished in time. */\n  EXPIRED: 'EXPIRED',\n  /** Cancellation was accepted and requests in flight are draining. */\n  CANCELLING: 'CANCELLING',\n  /** Cancellation finished. No output file is written. */\n  CANCELLED: 'CANCELLED',\n  /** The batch failed. */\n  FAILED: 'FAILED',\n} as const;\nexport type BatchStatus =\n  | (typeof BatchStatus)[keyof typeof BatchStatus]\n  | (string & {});\n\n/** ai_classify */\nexport interface AiClassifyOptions {\n  /** Natural-language guidance that steers how the text is classified (up to 20,000 characters). */\n  instructions?: string | undefined;\n  /** The function version to invoke. Defaults to the latest version. Supported versions: [\"2.1\"]. */\n  version?: string | undefined;\n  /** When true, allows more than one label to be returned per input. */\n  multilabel?: boolean | undefined;\n  /** When true, includes a per-label confidence score in the response. */\n  enableConfidenceScores?: boolean | undefined;\n  /** When true, includes a rationale explaining each classification in the response. */\n  enableRationales?: boolean | undefined;\n}\n\nexport interface AiClassifyRequest {\n  /** The content to classify. It accepts a plain string or the response object of [ai_parse_document](:method:AiFunctions/AiParseDocument). */\n  content?: JsonValue | undefined;\n  /** The label set to classify as. Either a JSON array of label strings (e.g. [\"spam\", \"not_spam\"]), or a JSON object mapping each label to a description (e.g. {\"spam\": \"unsolicited bulk message\", \"not_spam\": \"a legitimate message\"}). Accepts 2 to 500 labels, each 1 to 100 characters. */\n  labels?: JsonValue | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiClassifyOptions | undefined;\n}\n\nexport interface AiClassifyResponse {\n  /** The function result as a JSON value. An array of per-label objects: one element in single-label mode (the default), or multiple elements when `multilabel` is true. When `enable_confidence_scores` and `enable_rationales` are true, `confidence_score` and `rationale` are included in each response value, respectively. */\n  response?: JsonValue | undefined;\n  /** Additional metadata returned by AI Classify. */\n  metadata?: AiClassifyResponseMetadata | undefined;\n}\n\nexport interface AiClassifyResponseMetadata {\n  /** The resolved function version. */\n  version?: string | undefined;\n}\n\n/** ai_decide */\nexport interface AiDecideOptions {\n  /** The function API version to invoke. Defaults to \"1.0\". Supported versions: [\"1.0\"]. */\n  version?: string | undefined;\n}\n\nexport interface AiDecideRequest {\n  /** A string, JSON object, or array containing the content, related context, and examples needed to answer the provided questions. For example, provide a support message, a conversation, or records describing the current state of an application. All questions receive this same state. */\n  state?: JsonValue | undefined;\n  /**\n   * A JSON object mapping question IDs to their definitions. Each ID must contain non-whitespace text; its answer is returned with the same ID in `response.answers`.\n   *\n   * Each definition is an object with the required fields `type` and `instructions`. The `criteria` field is optional for the type `noul` but is required for the types `choice` and `score`.\n   *\n   * The `instructions` field describes the judgment to make and can be a string, object, or array. Use an object or array to include supporting context alongside the instructions.\n   *\n   * The `type` can be one of:\n   *\n   * - `choice`: Selects one option from a defined set. Requires `criteria` to be an object mapping 1 to 255 nonempty option names to descriptions. The criteria description can be a string, object, array, or null when the name needs no additional detail. For example:\n   *\n   * ```json\n   * {\n   * \"team\": {\n   * \"type\": \"choice\",\n   * \"instructions\": \"Which team should handle this ticket?\",\n   * \"criteria\": {\n   * \"billing\": \"Payments, charges, and refunds\",\n   * \"technical_support\": null\n   * }\n   * }\n   * }\n   * ```\n   *\n   * - `noul`: Estimates the probability that the answer to a true-or-false question is true. `criteria` can take the fields `true` or `false`, or both, with descriptions that are strings, objects, or arrays. Omit `criteria` to use the question alone. The instructions or at least one criteria description must contain non-whitespace text, a nonempty object, or a nonempty array. For example, both of the following are valid:\n   *\n   * ```json\n   * {\n   * \"escalate\": {\n   * \"type\": \"noul\",\n   * \"instructions\": \"Does this ticket need escalation?\",\n   * \"criteria\": {\n   * \"true\": \"Suspected fraud or an exception to standard policy\",\n   * \"false\": \"A routine issue frontline support can resolve\"\n   * }\n   * }\n   * }\n   * ```\n   *\n   * or\n   *\n   * ```json\n   * {\n   * \"escalate\": {\n   * \"type\": \"noul\",\n   * \"instructions\": \"Does this ticket need escalation?\"\n   * }\n   * }\n   * ```\n   *\n   * - `score`: Rates the state on an ordered scale. Requires `criteria` to be an array of 2 to 10 level descriptions, ordered from low to high. Descriptions must contain non-whitespace text, a nonempty object, or a nonempty array. Array positions define levels starting at 0. For example:\n   *\n   * ```json\n   * {\n   * \"urgency\": {\n   * \"type\": \"score\",\n   * \"instructions\": \"How urgent is this ticket?\",\n   * \"criteria\": [\n   * \"Routine: can wait a few days\",\n   * \"Time-sensitive: needs attention today\",\n   * \"Critical: needs immediate action\"\n   * ]\n   * }\n   * }\n   * ```\n   */\n  questions?: JsonValue | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiDecideOptions | undefined;\n}\n\nexport interface AiDecideResponse {\n  /**\n   * A JSON object containing an `answers` map with one answer for each question, keyed by the same IDs as `questions`.\n   *\n   * Each answer has a `type` matching its question and the following fields:\n   *\n   * - `choice`: choice is the highest-probability option from `criteria`. `probabilities` maps every option name to its probability. `confidence` is a number from 0 to 1 indicating how well the state supports the assessment. The `probabilities` values sum to 1. For example:\n   *\n   * ```json\n   * {\n   * \"answers\": {\n   * \"team\": {\n   * \"type\": \"choice\",\n   * \"choice\": \"billing\",\n   * \"probabilities\": {\n   * \"billing\": 0.85,\n   * \"technical_support\": 0.15\n   * },\n   * \"confidence\": 0.9\n   * }\n   * }\n   * }\n   * ```\n   *\n   * - `noul`: `probability` is a number from 0 to 1 estimating the probability that the answer is true. For example:\n   *\n   * ```json\n   * {\n   * \"answers\": {\n   * \"escalate\": {\n   * \"type\": \"noul\",\n   * \"probability\": 0.8\n   * }\n   * }\n   * }\n   * ```\n   *\n   * - `score`: score is the probability-weighted mean of the zero-based level indices. It can fall between levels, from 0 to the number of levels minus 1. `probabilities` maps each level index to its probability, and `legend` maps each index to its original description. Both maps use string keys such as \"0\", \"1\", and \"2\". `confidence` is a number from 0 to 1 indicating how well the state supports the assessment. The `probabilities` values sum to 1. For example:\n   *\n   * ```json\n   * {\n   * \"answers\": {\n   * \"urgency\": {\n   * \"type\": \"score\",\n   * \"score\": 1.6,\n   * \"probabilities\": {\n   * \"0\": 0.1,\n   * \"1\": 0.2,\n   * \"2\": 0.7\n   * },\n   * \"legend\": {\n   * \"0\": \"Routine: can wait a few days\",\n   * \"1\": \"Time-sensitive: needs attention today\",\n   * \"2\": \"Critical: needs immediate action\"\n   * },\n   * \"confidence\": 0.85\n   * }\n   * }\n   * }\n   * ```\n   */\n  response?: JsonValue | undefined;\n  /** Metadata identifying the function API version used for this request. */\n  metadata?: AiDecideResponseMetadata | undefined;\n}\n\nexport interface AiDecideResponseMetadata {\n  /** The function API version used to evaluate the request. */\n  version?: string | undefined;\n}\n\n/** A bounding box on a source page; used by bbox-input citations. */\nexport interface AiExtractBbox {\n  /** Pixel coordinates on the page image as [x0, y0, x1, y1]. */\n  coord?: bigint[] | undefined;\n  /** 0-based page index the box is on. */\n  pageId?: bigint | undefined;\n}\n\n/** A citation locating an extracted value in the source. start/stop are set for span (STRING input) citations, bbox for bbox (parsed-document input) citations. */\nexport interface AiExtractCitation {\n  /** Integer matching a citation_ids entry on an extracted field. */\n  id?: bigint | undefined;\n  /** Inclusive 0-based character offset into the input string; set for span citations. */\n  start?: bigint | undefined;\n  /** Exclusive 0-based character offset into the input string; set for span citations. */\n  stop?: bigint | undefined;\n  /** Bounding boxes locating the citation on the source pages; set for bbox citations. */\n  bbox?: AiExtractBbox[] | undefined;\n}\n\n/** ai_extract */\nexport interface AiExtractOptions {\n  /** Natural-language guidance that steers how data is extracted (up to 20,000 characters). */\n  instructions?: string | undefined;\n  /** The function version to invoke. Defaults to the latest version. Supported versions: [\"2.1\"]. */\n  version?: string | undefined;\n  /** Extraction mode. Supported modes: \"precision\" — more powerful extraction for complex schemas, long documents, and reasoning-heavy extractions. Defaults to none (standard extraction). */\n  mode?: string | undefined;\n  /**\n   * When true, includes citation metadata locating each extracted value in the source. Depending on the type of input, citations can be one of two types:\n   *\n   * For raw text (STRING) inputs, a citation is a span of text in the original input. Each object in `metadata.citations` has an `id` (integer matching a `citation_ids` entry on a field), a `start` (inclusive 0-based character offset into the input string), and a `stop` (exclusive 0-based character offset into the input string).\n   *\n   * For PDF documents and images (when using ai_extract downstream of ai_parse_document), a citation is a bounding box in the original input. Each object in `metadata.citations` has an `id` (integer matching a `citation_ids` entry on a field) and a `bbox` (array of {coord, page_id} objects, identical in shape to element.bbox in ai_parse_document output; coord is pixel coordinates on the page image as [x0, y0, x1, y1], and page_id is a 0-based page index).\n   */\n  enableCitations?: boolean | undefined;\n  /** When true, includes a per-field confidence score in the response. */\n  enableConfidenceScores?: boolean | undefined;\n}\n\nexport interface AiExtractRequest {\n  /** The text to extract from. It accepts a plain string or the response object of [ai_parse_document](:method:AiFunctions/AiParseDocument). */\n  content?: JsonValue | undefined;\n  /** The extraction schema defining the fields to extract. Either a JSON array of field names, assumed to be strings (e.g. [\"company\", \"valuation\"]), or a JSON object mapping each field to its type/description/nullability (e.g. {\"company\": {\"type\": \"string\", \"description\": \"the company name\"}}). Accepts up to 256 fields, 12 levels of nesting, and 500 enum values. Supported field types are string, integer, number, boolean, and enum. */\n  schema?: JsonValue | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiExtractOptions | undefined;\n}\n\nexport interface AiExtractResponse {\n  /** The function result as a JSON value. When `enable_confidence_scores` and `enable_citations` are true, `confidence` and `citation_ids` are included in each response field, respectively. */\n  response?: JsonValue | undefined;\n  /** Additional metadata returned by AI Extract. */\n  metadata?: AiExtractResponseMetadata | undefined;\n}\n\nexport interface AiExtractResponseMetadata {\n  /** The resolved function version. */\n  version?: string | undefined;\n  /** The resolved extraction mode; present when a non-default mode was used. */\n  mode?: string | undefined;\n  /** How the source was chunked for citation offsets (span for text input, bbox for parsed-document input); present when citations are enabled. */\n  chunkType?: string | undefined;\n  /** Citation objects locating each result in the source; present when citations are enabled. */\n  citations?: AiExtractCitation[] | undefined;\n}\n\n/** Metadata about the source file; present only for file-path input. */\nexport interface AiParseDocumentFileMetadata {\n  /** Unity Catalog volume path of the source file. */\n  filePath?: string | undefined;\n  /** Base name of the source file. */\n  fileName?: string | undefined;\n  /** Size of the source file in bytes. */\n  fileSize?: bigint | undefined;\n  /** Last-modified timestamp of the source file, as an HTTP date string. */\n  fileModificationTime?: string | undefined;\n}\n\n/** ai_parse_document */\nexport interface AiParseDocumentOptions {\n  /** The ai_parse_document output schema version. Supported value: \"2.0\". */\n  version?: string | undefined;\n  /** Element types for which an AI-generated description is produced. Use \"*\" (default) to generate descriptions for all supported element types, \"figure\" to generate them for figures only, or \"\" (empty string) to generate none. Only figure descriptions are supported for version \"2.0\", so \"*\" and \"figure\" produce the same behavior. */\n  descriptionElementTypes?: string | undefined;\n  /** Unity Catalog volume path where rendered page and element images are written. */\n  imageOutputPath?: string | undefined;\n  /** Pages to parse (1-indexed), as a comma-separated list of page numbers or ranges (e.g. \"1,3,5-10\"). */\n  pageRange?: string | undefined;\n}\n\n/** A single page that failed to parse while the overall request succeeded. */\nexport interface AiParseDocumentPageError {\n  /** Message describing why the page failed. */\n  errorMessage?: string | undefined;\n  /** 0-based index of the page that failed. */\n  pageId?: bigint | undefined;\n}\n\nexport interface AiParseDocumentRequest {\n  /** The document to parse. The REST API accepts a Unity Catalog volume path to the source file or a base64 data URL containing the file bytes (for example, data:application/pdf;base64,<encoded-file-bytes> or data:image/jpeg;base64,<encoded-file-bytes>). Supported formats: PDF, DOCX, DOC, PPTX, PPT, JPG, JPEG, PNG, TIFF. Accepts up to 100 pages and 100 MB per document. */\n  content?: string | undefined;\n  /** Function options. Omitted fields fall back to their documented defaults. */\n  options?: AiParseDocumentOptions | undefined;\n}\n\nexport interface AiParseDocumentResponse {\n  /** The parsed document as a JSON value, containing the extracted pages and elements. */\n  document?: JsonValue | undefined;\n  /** Per-page partial-failure details; present when the request succeeds (2xx) but individual pages fail. */\n  errorStatus?: AiParseDocumentPageError[] | undefined;\n  /** Additional metadata returned by AI Parse Document. */\n  metadata?: AiParseDocumentResponseMetadata | undefined;\n}\n\nexport interface AiParseDocumentResponseMetadata {\n  /** Unique identifier for the parse request. */\n  id?: string | undefined;\n  /** The resolved function version. */\n  version?: string | undefined;\n  /** Describes the source file; present only for file-path input. */\n  fileMetadata?: AiParseDocumentFileMetadata | undefined;\n}\n\n/** An asynchronous run of one AI function over a file of requests. */\nexport interface Batch {\n  /** ID of the batch, in the form `batch_<id>`. The service assigns it. */\n  name?: string | undefined;\n  /** The AI function to run for every request, for example `ai_extract`. */\n  endpoint?: string | undefined;\n  /** Path of the JSONL input file on a Unity Catalog volume. Each line is one request. */\n  inputRef?: string | undefined;\n  /** Function options applied to every request in the batch, for example the extraction schema. */\n  params?: JsonObject | undefined;\n  /**\n   * Path on a Unity Catalog volume to write the JSONL output to. When it is omitted, the output\n   * stays on Databricks-managed storage and a finished batch returns a short-lived download URL\n   * in this field instead.\n   */\n  outputRef?: string | undefined;\n  /** Output only. Status of the batch. */\n  status?: BatchStatus | undefined;\n  /** Output only. Time the batch was created. */\n  createTime?: Temporal.Instant | undefined;\n  /** Output only. Request counts. Returned when one batch is read, not when batches are listed. */\n  requestCounts?: BatchRequestCounts | undefined;\n  /**\n   * Output only. Percentage of the requests that finished, from 0 to 100. Returned when one batch\n   * is read, not when batches are listed.\n   */\n  percentComplete?: number | undefined;\n}\n\n/** Counts of the requests in a batch. */\nexport interface BatchRequestCounts {\n  /** Output only. Number of requests in the input file. */\n  total?: bigint | undefined;\n  /** Output only. Number of requests that finished successfully. */\n  succeeded?: bigint | undefined;\n  /** Output only. Number of requests that failed and will not be retried. */\n  failed?: bigint | undefined;\n}\n\nexport interface CancelBatchRequest {\n  /** ID of the batch, in the form `batch_<id>`. */\n  name?: string | undefined;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-empty-object-type\nexport interface CancelBatchResponse {}\n\nexport interface CreateBatchRequest {\n  /** The batch to create. */\n  batch?: Batch | undefined;\n}\n\nexport interface GetBatchRequest {\n  /** ID of the batch, in the form `batch_<id>`. */\n  name?: string | undefined;\n}\n\nexport interface ListBatchesRequest {\n  /** Maximum number of batches to return in one page. */\n  pageSize?: number | undefined;\n  /** Page token from an earlier response. Omit it to read the first page. */\n  pageToken?: string | undefined;\n}\n\nexport interface ListBatchesResponse {\n  /** The batches in this page. */\n  batches?: Batch[] | undefined;\n  /** Token that reads the next page. It is empty on the last page. */\n  nextPageToken?: string | undefined;\n}\n\nexport const unmarshalAiClassifyResponseSchema: z.ZodType<AiClassifyResponse> =\n  z\n    .object({\n      response: jsonValueSchema.optional(),\n      metadata: z\n        .lazy(() => unmarshalAiClassifyResponseMetadataSchema)\n        .optional(),\n    })\n    .transform(d => ({\n      response: d.response,\n      metadata: d.metadata,\n    }));\n\nexport const unmarshalAiClassifyResponseMetadataSchema: z.ZodType<AiClassifyResponseMetadata> =\n  z\n    .object({\n      version: z.string().optional(),\n    })\n    .transform(d => ({\n      version: d.version,\n    }));\n\nexport const unmarshalAiDecideResponseSchema: z.ZodType<AiDecideResponse> = z\n  .object({\n    response: jsonValueSchema.optional(),\n    metadata: z.lazy(() => unmarshalAiDecideResponseMetadataSchema).optional(),\n  })\n  .transform(d => ({\n    response: d.response,\n    metadata: d.metadata,\n  }));\n\nexport const unmarshalAiDecideResponseMetadataSchema: z.ZodType<AiDecideResponseMetadata> =\n  z\n    .object({\n      version: z.string().optional(),\n    })\n    .transform(d => ({\n      version: d.version,\n    }));\n\nexport const unmarshalAiExtractBboxSchema: z.ZodType<AiExtractBbox> = z\n  .object({\n    coord: z\n      .array(\n        z.union([z.number(), z.bigint(), z.string()]).transform(v => BigInt(v))\n      )\n      .optional(),\n    page_id: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n  })\n  .transform(d => ({\n    coord: d.coord,\n    pageId: d.page_id,\n  }));\n\nexport const unmarshalAiExtractCitationSchema: z.ZodType<AiExtractCitation> = z\n  .object({\n    id: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n    start: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n    stop: z\n      .union([z.number(), z.bigint(), z.string()])\n      .transform(v => BigInt(v))\n      .optional(),\n    bbox: z.array(z.lazy(() => unmarshalAiExtractBboxSchema)).optional(),\n  })\n  .transform(d => ({\n    id: d.id,\n    start: d.start,\n    stop: d.stop,\n    bbox: d.bbox,\n  }));\n\nexport const unmarshalAiExtractResponseSchema: z.ZodType<AiExtractResponse> = z\n  .object({\n    response: jsonValueSchema.optional(),\n    metadata: z.lazy(() => unmarshalAiExtractResponseMetadataSchema).optional(),\n  })\n  .transform(d => ({\n    response: d.response,\n    metadata: d.metadata,\n  }));\n\nexport const unmarshalAiExtractResponseMetadataSchema: z.ZodType<AiExtractResponseMetadata> =\n  z\n    .object({\n      version: z.string().optional(),\n      mode: z.string().optional(),\n      chunk_type: z.string().optional(),\n      citations: z\n        .array(z.lazy(() => unmarshalAiExtractCitationSchema))\n        .optional(),\n    })\n    .transform(d => ({\n      version: d.version,\n      mode: d.mode,\n      chunkType: d.chunk_type,\n      citations: d.citations,\n    }));\n\nexport const unmarshalAiParseDocumentFileMetadataSchema: z.ZodType<AiParseDocumentFileMetadata> =\n  z\n    .object({\n      file_path: z.string().optional(),\n      file_name: z.string().optional(),\n      file_size: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n      file_modification_time: z.string().optional(),\n    })\n    .transform(d => ({\n      filePath: d.file_path,\n      fileName: d.file_name,\n      fileSize: d.file_size,\n      fileModificationTime: d.file_modification_time,\n    }));\n\nexport const unmarshalAiParseDocumentPageErrorSchema: z.ZodType<AiParseDocumentPageError> =\n  z\n    .object({\n      error_message: z.string().optional(),\n      page_id: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n    })\n    .transform(d => ({\n      errorMessage: d.error_message,\n      pageId: d.page_id,\n    }));\n\nexport const unmarshalAiParseDocumentResponseSchema: z.ZodType<AiParseDocumentResponse> =\n  z\n    .object({\n      document: jsonValueSchema.optional(),\n      error_status: z\n        .array(z.lazy(() => unmarshalAiParseDocumentPageErrorSchema))\n        .optional(),\n      metadata: z\n        .lazy(() => unmarshalAiParseDocumentResponseMetadataSchema)\n        .optional(),\n    })\n    .transform(d => ({\n      document: d.document,\n      errorStatus: d.error_status,\n      metadata: d.metadata,\n    }));\n\nexport const unmarshalAiParseDocumentResponseMetadataSchema: z.ZodType<AiParseDocumentResponseMetadata> =\n  z\n    .object({\n      id: z.string().optional(),\n      version: z.string().optional(),\n      file_metadata: z\n        .lazy(() => unmarshalAiParseDocumentFileMetadataSchema)\n        .optional(),\n    })\n    .transform(d => ({\n      id: d.id,\n      version: d.version,\n      fileMetadata: d.file_metadata,\n    }));\n\nexport const unmarshalBatchSchema: z.ZodType<Batch> = z\n  .object({\n    name: z.string().optional(),\n    endpoint: z.string().optional(),\n    input_ref: z.string().optional(),\n    params: jsonObjectSchema.optional(),\n    output_ref: z.string().optional(),\n    status: z.string().optional(),\n    create_time: z\n      .string()\n      .transform(s => Temporal.Instant.from(s))\n      .optional(),\n    request_counts: z.lazy(() => unmarshalBatchRequestCountsSchema).optional(),\n    percent_complete: z.number().optional(),\n  })\n  .transform(d => ({\n    name: d.name,\n    endpoint: d.endpoint,\n    inputRef: d.input_ref,\n    params: d.params,\n    outputRef: d.output_ref,\n    status: d.status,\n    createTime: d.create_time,\n    requestCounts: d.request_counts,\n    percentComplete: d.percent_complete,\n  }));\n\nexport const unmarshalBatchRequestCountsSchema: z.ZodType<BatchRequestCounts> =\n  z\n    .object({\n      total: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n      succeeded: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n      failed: z\n        .union([z.number(), z.bigint(), z.string()])\n        .transform(v => BigInt(v))\n        .optional(),\n    })\n    .transform(d => ({\n      total: d.total,\n      succeeded: d.succeeded,\n      failed: d.failed,\n    }));\n\nexport const unmarshalCancelBatchResponseSchema: z.ZodType<CancelBatchResponse> =\n  z.object({});\n\nexport const unmarshalListBatchesResponseSchema: z.ZodType<ListBatchesResponse> =\n  z\n    .object({\n      batches: z.array(z.lazy(() => unmarshalBatchSchema)).optional(),\n      next_page_token: z.string().optional(),\n    })\n    .transform(d => ({\n      batches: d.batches,\n      nextPageToken: d.next_page_token,\n    }));\n\nexport const marshalAiClassifyOptionsSchema: z.ZodType = z\n  .object({\n    instructions: z.string().optional(),\n    version: z.string().optional(),\n    multilabel: z.boolean().optional(),\n    enableConfidenceScores: z.boolean().optional(),\n    enableRationales: z.boolean().optional(),\n  })\n  .transform(d => ({\n    instructions: d.instructions,\n    version: d.version,\n    multilabel: d.multilabel,\n    enable_confidence_scores: d.enableConfidenceScores,\n    enable_rationales: d.enableRationales,\n  }));\n\nexport const marshalAiClassifyRequestSchema: z.ZodType = z\n  .object({\n    content: jsonValueSchema.optional(),\n    labels: jsonValueSchema.optional(),\n    options: z.lazy(() => marshalAiClassifyOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    content: d.content,\n    labels: d.labels,\n    options: d.options,\n  }));\n\nexport const marshalAiDecideOptionsSchema: z.ZodType = z\n  .object({\n    version: z.string().optional(),\n  })\n  .transform(d => ({\n    version: d.version,\n  }));\n\nexport const marshalAiDecideRequestSchema: z.ZodType = z\n  .object({\n    state: jsonValueSchema.optional(),\n    questions: jsonValueSchema.optional(),\n    options: z.lazy(() => marshalAiDecideOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    state: d.state,\n    questions: d.questions,\n    options: d.options,\n  }));\n\nexport const marshalAiExtractOptionsSchema: z.ZodType = z\n  .object({\n    instructions: z.string().optional(),\n    version: z.string().optional(),\n    mode: z.string().optional(),\n    enableCitations: z.boolean().optional(),\n    enableConfidenceScores: z.boolean().optional(),\n  })\n  .transform(d => ({\n    instructions: d.instructions,\n    version: d.version,\n    mode: d.mode,\n    enable_citations: d.enableCitations,\n    enable_confidence_scores: d.enableConfidenceScores,\n  }));\n\nexport const marshalAiExtractRequestSchema: z.ZodType = z\n  .object({\n    content: jsonValueSchema.optional(),\n    schema: jsonValueSchema.optional(),\n    options: z.lazy(() => marshalAiExtractOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    content: d.content,\n    schema: d.schema,\n    options: d.options,\n  }));\n\nexport const marshalAiParseDocumentOptionsSchema: z.ZodType = z\n  .object({\n    version: z.string().optional(),\n    descriptionElementTypes: z.string().optional(),\n    imageOutputPath: z.string().optional(),\n    pageRange: z.string().optional(),\n  })\n  .transform(d => ({\n    version: d.version,\n    description_element_types: d.descriptionElementTypes,\n    image_output_path: d.imageOutputPath,\n    page_range: d.pageRange,\n  }));\n\nexport const marshalAiParseDocumentRequestSchema: z.ZodType = z\n  .object({\n    content: z.string().optional(),\n    options: z.lazy(() => marshalAiParseDocumentOptionsSchema).optional(),\n  })\n  .transform(d => ({\n    content: d.content,\n    options: d.options,\n  }));\n\nexport const marshalBatchSchema: z.ZodType = z\n  .object({\n    name: z.string().optional(),\n    endpoint: z.string().optional(),\n    inputRef: z.string().optional(),\n    params: jsonObjectSchema.optional(),\n    outputRef: z.string().optional(),\n    status: z.string().optional(),\n    createTime: z\n      .any()\n      .transform((d: Temporal.Instant) => d.toString())\n      .optional(),\n    requestCounts: z.lazy(() => marshalBatchRequestCountsSchema).optional(),\n    percentComplete: z.number().optional(),\n  })\n  .transform(d => ({\n    name: d.name,\n    endpoint: d.endpoint,\n    input_ref: d.inputRef,\n    params: d.params,\n    output_ref: d.outputRef,\n    status: d.status,\n    create_time: d.createTime,\n    request_counts: d.requestCounts,\n    percent_complete: d.percentComplete,\n  }));\n\nexport const marshalBatchRequestCountsSchema: z.ZodType = z\n  .object({\n    total: z.bigint().optional(),\n    succeeded: z.bigint().optional(),\n    failed: z.bigint().optional(),\n  })\n  .transform(d => ({\n    total: d.total,\n    succeeded: d.succeeded,\n    failed: d.failed,\n  }));\n\nexport const marshalCancelBatchRequestSchema: z.ZodType = z\n  .object({\n    name: z.string().optional(),\n  })\n  .transform(d => ({\n    name: d.name,\n  }));\n"],"mappings":";;;;AAMA,MAAM,kBAAwC,EAAE,WAC9C,EAAE,MAAM;CACN,EAAE,MAAM;CACR,EAAE,QAAQ;CACV,EAAE,QAAQ;CACV,EAAE,SAAS;CACX,EAAE,OAAO,EAAE,QAAQ,EAAE,gBAAgB;CACrC,EAAE,MAAM,gBAAgB;CACzB,CAAC,CACH;AACD,MAAM,mBAA0C,EAAE,OAChD,EAAE,QAAQ,EACV,gBACD;;AAID,MAAa,cAAc;CACzB,0BAA0B;CAE1B,SAAS;CAET,aAAa;CAEb,YAAY;CAEZ,WAAW;CAEX,SAAS;CAET,YAAY;CAEZ,WAAW;CAEX,QAAQ;CACT;AAkYD,MAAa,oCACX,EACG,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,UAAU,EACP,WAAW,0CAA0C,CACrD,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACb,EAAE;AAEP,MAAa,4CACX,EACG,OAAO,EACN,SAAS,EAAE,QAAQ,CAAC,UAAU,EAC/B,CAAC,CACD,WAAU,OAAM,EACf,SAAS,EAAE,SACZ,EAAE;AAEP,MAAa,kCAA+D,EACzE,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,UAAU,EAAE,WAAW,wCAAwC,CAAC,UAAU;CAC3E,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACb,EAAE;AAEL,MAAa,0CACX,EACG,OAAO,EACN,SAAS,EAAE,QAAQ,CAAC,UAAU,EAC/B,CAAC,CACD,WAAU,OAAM,EACf,SAAS,EAAE,SACZ,EAAE;AAEP,MAAa,+BAAyD,EACnE,OAAO;CACN,OAAO,EACJ,MACC,EAAE,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAAC,WAAU,MAAK,OAAO,EAAE,CAAC,CACxE,CACA,UAAU;CACb,SAAS,EACN,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,OAAO,EAAE;CACT,QAAQ,EAAE;CACX,EAAE;AAEL,MAAa,mCAAiE,EAC3E,OAAO;CACN,IAAI,EACD,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,OAAO,EACJ,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,MAAM,EACH,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,MAAM,EAAE,MAAM,EAAE,WAAW,6BAA6B,CAAC,CAAC,UAAU;CACrE,CAAC,CACD,WAAU,OAAM;CACf,IAAI,EAAE;CACN,OAAO,EAAE;CACT,MAAM,EAAE;CACR,MAAM,EAAE;CACT,EAAE;AAEL,MAAa,mCAAiE,EAC3E,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,UAAU,EAAE,WAAW,yCAAyC,CAAC,UAAU;CAC5E,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACb,EAAE;AAEL,MAAa,2CACX,EACG,OAAO;CACN,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,YAAY,EAAE,QAAQ,CAAC,UAAU;CACjC,WAAW,EACR,MAAM,EAAE,WAAW,iCAAiC,CAAC,CACrD,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,MAAM,EAAE;CACR,WAAW,EAAE;CACb,WAAW,EAAE;CACd,EAAE;AAEP,MAAa,6CACX,EACG,OAAO;CACN,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,WAAW,EACR,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,wBAAwB,EAAE,QAAQ,CAAC,UAAU;CAC9C,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,UAAU,EAAE;CACZ,UAAU,EAAE;CACZ,sBAAsB,EAAE;CACzB,EAAE;AAEP,MAAa,0CACX,EACG,OAAO;CACN,eAAe,EAAE,QAAQ,CAAC,UAAU;CACpC,SAAS,EACN,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,cAAc,EAAE;CAChB,QAAQ,EAAE;CACX,EAAE;AAEP,MAAa,yCACX,EACG,OAAO;CACN,UAAU,gBAAgB,UAAU;CACpC,cAAc,EACX,MAAM,EAAE,WAAW,wCAAwC,CAAC,CAC5D,UAAU;CACb,UAAU,EACP,WAAW,+CAA+C,CAC1D,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,UAAU,EAAE;CACZ,aAAa,EAAE;CACf,UAAU,EAAE;CACb,EAAE;AAEP,MAAa,iDACX,EACG,OAAO;CACN,IAAI,EAAE,QAAQ,CAAC,UAAU;CACzB,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,eAAe,EACZ,WAAW,2CAA2C,CACtD,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,IAAI,EAAE;CACN,SAAS,EAAE;CACX,cAAc,EAAE;CACjB,EAAE;AAEP,MAAa,uBAAyC,EACnD,OAAO;CACN,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,UAAU,EAAE,QAAQ,CAAC,UAAU;CAC/B,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,QAAQ,iBAAiB,UAAU;CACnC,YAAY,EAAE,QAAQ,CAAC,UAAU;CACjC,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC7B,aAAa,EACV,QAAQ,CACR,WAAU,MAAK,SAAS,QAAQ,KAAK,EAAE,CAAC,CACxC,UAAU;CACb,gBAAgB,EAAE,WAAW,kCAAkC,CAAC,UAAU;CAC1E,kBAAkB,EAAE,QAAQ,CAAC,UAAU;CACxC,CAAC,CACD,WAAU,OAAM;CACf,MAAM,EAAE;CACR,UAAU,EAAE;CACZ,UAAU,EAAE;CACZ,QAAQ,EAAE;CACV,WAAW,EAAE;CACb,QAAQ,EAAE;CACV,YAAY,EAAE;CACd,eAAe,EAAE;CACjB,iBAAiB,EAAE;CACpB,EAAE;AAEL,MAAa,oCACX,EACG,OAAO;CACN,OAAO,EACJ,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,WAAW,EACR,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACb,QAAQ,EACL,MAAM;EAAC,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAE,EAAE,QAAQ;EAAC,CAAC,CAC3C,WAAU,MAAK,OAAO,EAAE,CAAC,CACzB,UAAU;CACd,CAAC,CACD,WAAU,OAAM;CACf,OAAO,EAAE;CACT,WAAW,EAAE;CACb,QAAQ,EAAE;CACX,EAAE;AAEP,MAAa,qCACX,EAAE,OAAO,EAAE,CAAC;AAEd,MAAa,qCACX,EACG,OAAO;CACN,SAAS,EAAE,MAAM,EAAE,WAAW,qBAAqB,CAAC,CAAC,UAAU;CAC/D,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACvC,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,eAAe,EAAE;CAClB,EAAE;AAEP,MAAa,iCAA4C,EACtD,OAAO;CACN,cAAc,EAAE,QAAQ,CAAC,UAAU;CACnC,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,YAAY,EAAE,SAAS,CAAC,UAAU;CAClC,wBAAwB,EAAE,SAAS,CAAC,UAAU;CAC9C,kBAAkB,EAAE,SAAS,CAAC,UAAU;CACzC,CAAC,CACD,WAAU,OAAM;CACf,cAAc,EAAE;CAChB,SAAS,EAAE;CACX,YAAY,EAAE;CACd,0BAA0B,EAAE;CAC5B,mBAAmB,EAAE;CACtB,EAAE;AAEL,MAAa,iCAA4C,EACtD,OAAO;CACN,SAAS,gBAAgB,UAAU;CACnC,QAAQ,gBAAgB,UAAU;CAClC,SAAS,EAAE,WAAW,+BAA+B,CAAC,UAAU;CACjE,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,QAAQ,EAAE;CACV,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,+BAA0C,EACpD,OAAO,EACN,SAAS,EAAE,QAAQ,CAAC,UAAU,EAC/B,CAAC,CACD,WAAU,OAAM,EACf,SAAS,EAAE,SACZ,EAAE;AAEL,MAAa,+BAA0C,EACpD,OAAO;CACN,OAAO,gBAAgB,UAAU;CACjC,WAAW,gBAAgB,UAAU;CACrC,SAAS,EAAE,WAAW,6BAA6B,CAAC,UAAU;CAC/D,CAAC,CACD,WAAU,OAAM;CACf,OAAO,EAAE;CACT,WAAW,EAAE;CACb,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,gCAA2C,EACrD,OAAO;CACN,cAAc,EAAE,QAAQ,CAAC,UAAU;CACnC,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,iBAAiB,EAAE,SAAS,CAAC,UAAU;CACvC,wBAAwB,EAAE,SAAS,CAAC,UAAU;CAC/C,CAAC,CACD,WAAU,OAAM;CACf,cAAc,EAAE;CAChB,SAAS,EAAE;CACX,MAAM,EAAE;CACR,kBAAkB,EAAE;CACpB,0BAA0B,EAAE;CAC7B,EAAE;AAEL,MAAa,gCAA2C,EACrD,OAAO;CACN,SAAS,gBAAgB,UAAU;CACnC,QAAQ,gBAAgB,UAAU;CAClC,SAAS,EAAE,WAAW,8BAA8B,CAAC,UAAU;CAChE,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,QAAQ,EAAE;CACV,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,sCAAiD,EAC3D,OAAO;CACN,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,yBAAyB,EAAE,QAAQ,CAAC,UAAU;CAC9C,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACtC,WAAW,EAAE,QAAQ,CAAC,UAAU;CACjC,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,2BAA2B,EAAE;CAC7B,mBAAmB,EAAE;CACrB,YAAY,EAAE;CACf,EAAE;AAEL,MAAa,sCAAiD,EAC3D,OAAO;CACN,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,SAAS,EAAE,WAAW,oCAAoC,CAAC,UAAU;CACtE,CAAC,CACD,WAAU,OAAM;CACf,SAAS,EAAE;CACX,SAAS,EAAE;CACZ,EAAE;AAEL,MAAa,qBAAgC,EAC1C,OAAO;CACN,MAAM,EAAE,QAAQ,CAAC,UAAU;CAC3B,UAAU,EAAE,QAAQ,CAAC,UAAU;CAC/B,UAAU,EAAE,QAAQ,CAAC,UAAU;CAC/B,QAAQ,iBAAiB,UAAU;CACnC,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC7B,YAAY,EACT,KAAK,CACL,WAAW,MAAwB,EAAE,UAAU,CAAC,CAChD,UAAU;CACb,eAAe,EAAE,WAAW,gCAAgC,CAAC,UAAU;CACvE,iBAAiB,EAAE,QAAQ,CAAC,UAAU;CACvC,CAAC,CACD,WAAU,OAAM;CACf,MAAM,EAAE;CACR,UAAU,EAAE;CACZ,WAAW,EAAE;CACb,QAAQ,EAAE;CACV,YAAY,EAAE;CACd,QAAQ,EAAE;CACV,aAAa,EAAE;CACf,gBAAgB,EAAE;CAClB,kBAAkB,EAAE;CACrB,EAAE;AAEL,MAAa,kCAA6C,EACvD,OAAO;CACN,OAAO,EAAE,QAAQ,CAAC,UAAU;CAC5B,WAAW,EAAE,QAAQ,CAAC,UAAU;CAChC,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC9B,CAAC,CACD,WAAU,OAAM;CACf,OAAO,EAAE;CACT,WAAW,EAAE;CACb,QAAQ,EAAE;CACX,EAAE;AAEL,MAAa,kCAA6C,EACvD,OAAO,EACN,MAAM,EAAE,QAAQ,CAAC,UAAU,EAC5B,CAAC,CACD,WAAU,OAAM,EACf,MAAM,EAAE,MACT,EAAE"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@databricks/sdk-aifunctions",
-  "version": "0.6.0",
+  "version": "0.6.1",
   "description": "",
   "type": "module",
   "exports": {