@databricks/sdk-forecasting 0.0.0-dev → 0.1.0-dev.1

package/dist/v1/transport.js

@@ -0,0 +1,57 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+import { defaultCredentials } from '@databricks/sdk-auth/credentials';
+import { newFetchHttpClient } from '@databricks/sdk-core/http';
+/** Creates a new HTTP client with the given options. */
+export function newHttpClient(options) {
+    const opts = options ?? {};
+    // If an HTTP client is provided, use it as-is. Throw if other options are
+    // also set, since they would be silently ignored.
+    if (opts.httpClient !== undefined) {
+        if (opts.credentials !== undefined || opts.timeout !== undefined) {
+            throw new Error('httpClient cannot be combined with credentials or timeout');
+        }
+        return opts.httpClient;
+    }
+    const credentials = opts.credentials ?? defaultCredentials();
+    const base = newFetchHttpClient();
+    let client = new AuthHttpClient(base, credentials);
+    if (opts.timeout !== undefined) {
+        client = new TimeoutHttpClient(client, opts.timeout);
+    }
+    return client;
+}
+/** Wraps an HttpClient and adds authentication headers to requests. */
+class AuthHttpClient {
+    base;
+    credentials;
+    constructor(base, credentials) {
+        this.base = base;
+        this.credentials = credentials;
+    }
+    async send(request) {
+        const authHeaders = await this.credentials.authHeaders();
+        // Do not modify the original request.
+        const headers = new Headers(request.headers);
+        for (const h of authHeaders) {
+            headers.set(h.key, h.value);
+        }
+        return this.base.send({ ...request, headers });
+    }
+}
+/** Wraps an HttpClient and applies a default timeout to requests. */
+class TimeoutHttpClient {
+    base;
+    timeout;
+    constructor(base, timeout) {
+        this.base = base;
+        this.timeout = timeout;
+    }
+    async send(request) {
+        const timeoutSignal = AbortSignal.timeout(this.timeout);
+        const signal = request.signal !== undefined
+            ? AbortSignal.any([request.signal, timeoutSignal])
+            : timeoutSignal;
+        return this.base.send({ ...request, signal });
+    }
+}
+//# sourceMappingURL=transport.js.map

package/dist/v1/transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../src/v1/transport.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,kBAAkB,EAAC,MAAM,kCAAkC,CAAC;AAMpE,OAAO,EAAC,kBAAkB,EAAC,MAAM,2BAA2B,CAAC;AAG7D,wDAAwD;AACxD,MAAM,UAAU,aAAa,CAAC,OAAuB;IACnD,MAAM,IAAI,GAAG,OAAO,IAAI,EAAE,CAAC;IAE3B,0EAA0E;IAC1E,kDAAkD;IAClD,IAAI,IAAI,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACjE,MAAM,IAAI,KAAK,CACb,2DAA2D,CAC5D,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,kBAAkB,EAAE,CAAC;IAE7D,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAC;IAClC,IAAI,MAAM,GAAe,IAAI,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;IAE/D,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAC/B,MAAM,GAAG,IAAI,iBAAiB,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,uEAAuE;AACvE,MAAM,cAAc;IAEC;IACA;IAFnB,YACmB,IAAgB,EAChB,WAAwB;QADxB,SAAI,GAAJ,IAAI,CAAY;QAChB,gBAAW,GAAX,WAAW,CAAa;IACxC,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,EAAE,CAAC;QACzD,sCAAsC;QACtC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC7C,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;YAC5B,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC,GAAG,OAAO,EAAE,OAAO,EAAC,CAAC,CAAC;IAC/C,CAAC;CACF;AAED,qEAAqE;AACrE,MAAM,iBAAiB;IAEF;IACA;IAFnB,YACmB,IAAgB,EAChB,OAAe;QADf,SAAI,GAAJ,IAAI,CAAY;QAChB,YAAO,GAAP,OAAO,CAAQ;IAC/B,CAAC;IAEJ,KAAK,CAAC,IAAI,CAAC,OAAoB;QAC7B,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,MAAM,MAAM,GACV,OAAO,CAAC,MAAM,KAAK,SAAS;YAC1B,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAClD,CAAC,CAAC,aAAa,CAAC;QACpB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAC,GAAG,OAAO,EAAE,MAAM,EAAC,CAAC,CAAC;IAC9C,CAAC;CACF"}

package/dist/v1/utils.d.ts

@@ -0,0 +1,22 @@
+import type { Call } from '@databricks/sdk-core/api';
+import type { HttpClient, HttpRequest } from '@databricks/sdk-core/http';
+import type { Logger } from '@databricks/sdk-core/logger';
+import type { CallOptions } from '@databricks/sdk-options/call';
+import type { z } from 'zod';
+export interface HttpCallOptions {
+    readonly request: HttpRequest;
+    readonly httpClient: HttpClient;
+    readonly logger: Logger;
+}
+/**
+ * Translates public CallOptions to the internal Options shape accepted by
+ * execute(). Even though the shapes match today, this isolates the public
+ * API from the executor's internal type so they can diverge.
+ */
+export declare function executeCall(call: Call, options?: CallOptions): Promise<void>;
+export declare function executeHttpCall(opts: HttpCallOptions): Promise<Uint8Array>;
+export declare function buildHttpRequest(method: string, url: string, headers: Headers, signal?: AbortSignal, body?: string | ReadableStream<Uint8Array>): HttpRequest;
+export declare function parseResponse<T>(body: Uint8Array, schema: z.ZodType<T>): T;
+export declare function marshalRequest(data: unknown, schema: z.ZodType): string;
+export declare function flattenQueryParams(prefix: string, value: unknown, params: URLSearchParams): void;
+//# sourceMappingURL=utils.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAC,IAAI,EAAU,MAAM,0BAA0B,CAAC;AAG5D,OAAO,KAAK,EACV,UAAU,EACV,WAAW,EAEZ,MAAM,2BAA2B,CAAC;AACnC,OAAO,KAAK,EAAC,MAAM,EAAC,MAAM,6BAA6B,CAAC;AACxD,OAAO,KAAK,EAAC,WAAW,EAAC,MAAM,8BAA8B,CAAC;AAE9D,OAAO,KAAK,EAAC,CAAC,EAAC,MAAM,KAAK,CAAC;AAO3B,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAChC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAC/B,IAAI,EAAE,IAAI,EACV,OAAO,CAAC,EAAE,WAAW,GACpB,OAAO,CAAC,IAAI,CAAC,CASf;AA2BD,wBAAsB,eAAe,CACnC,IAAI,EAAE,eAAe,GACpB,OAAO,CAAC,UAAU,CAAC,CA2BrB;AAED,wBAAgB,gBAAgB,CAC9B,MAAM,EAAE,MAAM,EACd,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,OAAO,EAChB,MAAM,CAAC,EAAE,WAAW,EACpB,IAAI,CAAC,EAAE,MAAM,GAAG,cAAc,CAAC,UAAU,CAAC,GACzC,WAAW,CASb;AAED,wBAAgB,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAI1E;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC,OAAO,GAAG,MAAM,CAEvE;AAED,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,OAAO,EACd,MAAM,EAAE,eAAe,GACtB,IAAI,CAuBN"}

package/dist/v1/utils.js

@@ -0,0 +1,113 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+import { execute } from '@databricks/sdk-core/api';
+import { ApiError } from '@databricks/sdk-core/apierror';
+import JSONBig from 'json-bigint';
+// JSON codec that preserves int64 precision. On the way in, large integer
+// literals come back as bigint instead of being rounded to JS Number. On the
+// way out, bigint values are emitted as raw JSON number digits.
+const jsonBigint = JSONBig({ useNativeBigInt: true });
+/**
+ * Translates public CallOptions to the internal Options shape accepted by
+ * execute(). Even though the shapes match today, this isolates the public
+ * API from the executor's internal type so they can diverge.
+ */
+export async function executeCall(call, options) {
+    const opts = {
+        ...(options?.retrier !== undefined && { retrier: options.retrier }),
+        ...(options?.rateLimiter !== undefined && {
+            rateLimiter: options.rateLimiter,
+        }),
+        ...(options?.timeout !== undefined && { timeout: options.timeout }),
+    };
+    return execute(options?.signal, call, opts);
+}
+async function readAll(body) {
+    if (body === null) {
+        return new Uint8Array(0);
+    }
+    const reader = body.getReader();
+    const chunks = [];
+    for (;;) {
+        const { done, value } = await reader.read();
+        if (done) {
+            break;
+        }
+        chunks.push(value);
+    }
+    const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
+    const result = new Uint8Array(totalLength);
+    let offset = 0;
+    for (const chunk of chunks) {
+        result.set(chunk, offset);
+        offset += chunk.length;
+    }
+    return result;
+}
+export async function executeHttpCall(opts) {
+    opts.logger.debug('HTTP request', {
+        method: opts.request.method,
+        url: opts.request.url,
+    });
+    let resp;
+    try {
+        resp = await opts.httpClient.send(opts.request);
+    }
+    catch (e) {
+        opts.logger.debug('HTTP request failed');
+        throw e;
+    }
+    const body = await readAll(resp.body);
+    opts.logger.debug('HTTP response', {
+        statusCode: resp.statusCode,
+        body: new TextDecoder().decode(body),
+    });
+    const apiErr = ApiError.fromHttpError(resp.statusCode, resp.headers, body);
+    if (apiErr !== undefined) {
+        throw apiErr;
+    }
+    return body;
+}
+export function buildHttpRequest(method, url, headers, signal, body) {
+    const req = { url, method, headers };
+    if (body !== undefined) {
+        req.body = body;
+    }
+    if (signal !== undefined) {
+        req.signal = signal;
+    }
+    return req;
+}
+export function parseResponse(body, schema) {
+    const text = new TextDecoder().decode(body);
+    const parsed = jsonBigint.parse(text);
+    return schema.parse(parsed);
+}
+export function marshalRequest(data, schema) {
+    return jsonBigint.stringify(schema.parse(data));
+}
+export function flattenQueryParams(prefix, value, params) {
+    if (value === null || value === undefined) {
+        return;
+    }
+    if (Array.isArray(value)) {
+        // arrays of objects are not yet supported
+        for (const item of value) {
+            params.append(prefix, String(item));
+        }
+    }
+    else if (typeof value === 'object') {
+        for (const [key, val] of Object.entries(value)) {
+            flattenQueryParams(`${prefix}.${key}`, val, params);
+        }
+    }
+    else if (typeof value === 'string' ||
+        typeof value === 'number' ||
+        typeof value === 'boolean' ||
+        typeof value === 'bigint') {
+        params.append(prefix, String(value));
+    }
+    else {
+        throw new Error(`Unsupported query parameter type: ${typeof value}`);
+    }
+}
+//# sourceMappingURL=utils.js.map

package/dist/v1/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/v1/utils.ts"],"names":[],"mappings":"AAAA,+EAA+E;AAG/E,OAAO,EAAC,OAAO,EAAC,MAAM,0BAA0B,CAAC;AACjD,OAAO,EAAC,QAAQ,EAAC,MAAM,+BAA+B,CAAC;AAQvD,OAAO,OAAO,MAAM,aAAa,CAAC;AAGlC,0EAA0E;AAC1E,6EAA6E;AAC7E,gEAAgE;AAChE,MAAM,UAAU,GAAG,OAAO,CAAC,EAAC,eAAe,EAAE,IAAI,EAAC,CAAC,CAAC;AAQpD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAU,EACV,OAAqB;IAErB,MAAM,IAAI,GAAY;QACpB,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;QACjE,GAAG,CAAC,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI;YACxC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;QACF,GAAG,CAAC,OAAO,EAAE,OAAO,KAAK,SAAS,IAAI,EAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAC,CAAC;KAClE,CAAC;IACF,OAAO,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,OAAO,CACpB,IAAuC;IAEvC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;QAClB,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;IAC3B,CAAC;IACD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;IAChC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,SAAS,CAAC;QACR,MAAM,EAAC,IAAI,EAAE,KAAK,EAAC,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,IAAI,EAAE,CAAC;YACT,MAAM;QACR,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAqB;IAErB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,cAAc,EAAE;QAChC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;QAC3B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;KACtB,CAAC,CAAC;IAEH,IAAI,IAAkB,CAAC;IACvB,IAAI,CAAC;QACH,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClD,CAAC;IAAC,OAAO,CAAU,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,MAAM,CAAC,CAAC;IACV,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE;QACjC,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,IAAI,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;KACrC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,MAAM,MAAM,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,MAAc,EACd,GAAW,EACX,OAAgB,EAChB,MAAoB,EACpB,IAA0C;IAE1C,MAAM,GAAG,GAAgB,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAC,CAAC;IAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACvB,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,aAAa,CAAI,IAAgB,EAAE,MAAoB;IACrE,MAAM,IAAI,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,MAAM,GAAY,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC/C,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa,EAAE,MAAiB;IAC7D,OAAO,UAAU,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,MAAc,EACd,KAAc,EACd,MAAuB;IAEvB,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO;IACT,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,0CAA0C;QAC1C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACrC,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAgC,CAAC,EAAE,CAAC;YAC1E,kBAAkB,CAAC,GAAG,MAAM,IAAI,GAAG,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IACL,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,qCAAqC,OAAO,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;AACH,CAAC"}

package/package.json

@@ -1,7 +1,41 @@
 {
   "name": "@databricks/sdk-forecasting",
-  "version": "0.0.0-dev",
-  "description": "Bootstrap placeholder; real contents in a later release.",
-  "main": "index.js",
-  "license": "Apache-2.0"
+  "version": "0.1.0-dev.1",
+  "description": "",
+  "type": "module",
+  "exports": {
+    "./v1": {
+      "types": "./dist/v1/index.d.ts",
+      "import": "./dist/v1/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "src",
+    "LICENSE"
+  ],
+  "scripts": {
+    "build": "tsc -b",
+    "lint": "eslint src --ext .ts",
+    "lint:fix": "eslint src --ext .ts --fix",
+    "format": "prettier --write \"src/**/*.ts\"",
+    "format:check": "prettier --check \"src/**/*.ts\"",
+    "typecheck": "tsc --noEmit",
+    "clean": "rm -rf dist tsconfig.tsbuildinfo",
+    "test": "echo 'no tests'",
+    "test:browser": "echo 'no tests'"
+  },
+  "author": "Databricks",
+  "license": "Apache-2.0",
+  "dependencies": {
+    "@databricks/sdk-auth": ">=0.1.0-dev.2 <1.0.0",
+    "@databricks/sdk-core": ">=0.1.0-dev.3 <1.0.0",
+    "@databricks/sdk-options": ">=0.1.0-dev.2 <1.0.0",
+    "@js-temporal/polyfill": "^0.5.0",
+    "json-bigint": "^1.0.0",
+    "zod": "^4.3.6"
+  },
+  "engines": {
+    "node": ">=22.0.0"
+  }
 }

package/src/v1/client.ts

@@ -0,0 +1,225 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+
+import {VERSION as AUTH_VERSION} from '@databricks/sdk-auth';
+import type {Call} from '@databricks/sdk-core/api';
+import {retryOn} from '@databricks/sdk-core/api';
+import {createDefault} from '@databricks/sdk-core/clientinfo';
+import type {Logger} from '@databricks/sdk-core/logger';
+import {NoOpLogger} from '@databricks/sdk-core/logger';
+import type {CallOptions} from '@databricks/sdk-options/call';
+import type {ClientOptions} from '@databricks/sdk-options/client';
+import type {HttpClient} from '@databricks/sdk-core/http';
+import {newHttpClient} from './transport';
+import {
+  buildHttpRequest,
+  executeCall,
+  executeHttpCall,
+  marshalRequest,
+  parseResponse,
+} from './utils';
+import pkgJson from '../../package.json' with {type: 'json'};
+import type {
+  CreateForecastingExperimentRequest,
+  CreateForecastingExperimentResponse,
+  ForecastingExperiment,
+  GetForecastingExperimentRequest,
+} from './model';
+import {
+  ForecastingExperiment_State,
+  marshalCreateForecastingExperimentRequestSchema,
+  unmarshalCreateForecastingExperimentResponseSchema,
+  unmarshalForecastingExperimentSchema,
+} from './model';
+
+// Package identity segment for this client to be used in the User-Agent header.
+const PACKAGE_SEGMENT = {
+  key: 'sdk-js-' + pkgJson.name.replace(/^@[^/]+\/sdk-/, ''),
+  value: pkgJson.version,
+};
+
+class StillRunningError extends Error {}
+
+export class ForecastingClient {
+  private readonly host: string;
+  // Workspace ID used to route workspace-level calls on unified hosts (SPOG).
+  // When set, workspace-level methods send X-Databricks-Org-Id on every
+  // request.
+  private readonly workspaceId: string | undefined;
+  private readonly httpClient: HttpClient;
+  private readonly logger: Logger;
+  // User-Agent header value. Composed once at construction from
+  // createDefault() merged with this package's identity and the active
+  // credential's name.
+  private readonly userAgent: string;
+
+  constructor(options: ClientOptions) {
+    if (options.host === undefined) {
+      throw new Error('Host is required.');
+    }
+    this.host = options.host.replace(/\/$/, '');
+    this.workspaceId = options.workspaceId;
+    this.logger = options.logger ?? new NoOpLogger();
+    const info = createDefault()
+      .with(PACKAGE_SEGMENT)
+      .with({key: 'sdk-js-auth', value: AUTH_VERSION})
+      .with({key: 'auth', value: options.credentials?.name() ?? 'default'});
+    this.userAgent = info.toString();
+    this.httpClient = newHttpClient(options);
+  }
+
+  /** Creates a serverless forecasting experiment. Returns the experiment ID. */
+  async createForecastingExperiment(
+    req: CreateForecastingExperimentRequest,
+    options?: CallOptions
+  ): Promise<CreateForecastingExperimentResponse> {
+    const url = `${this.host}/api/2.0/automl/create-forecasting-experiment`;
+    const body = marshalRequest(
+      req,
+      marshalCreateForecastingExperimentRequestSchema
+    );
+    let resp: CreateForecastingExperimentResponse | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers({'Content-Type': 'application/json'});
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('POST', url, headers, callSignal, body);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(
+        respBody,
+        unmarshalCreateForecastingExperimentResponseSchema
+      );
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+
+  async createForecastingExperimentWaiter(
+    req: CreateForecastingExperimentRequest,
+    options?: CallOptions
+  ): Promise<CreateForecastingExperimentWaiter> {
+    const resp = await this.createForecastingExperiment(req, options);
+    if (resp.experimentId === undefined) {
+      throw new Error(
+        'response field experimentId required for polling is missing'
+      );
+    }
+    return new CreateForecastingExperimentWaiter(this, resp.experimentId);
+  }
+
+  /** Public RPC to get forecasting experiment */
+  async getForecastingExperiment(
+    req: GetForecastingExperimentRequest,
+    options?: CallOptions
+  ): Promise<ForecastingExperiment> {
+    const url = `${this.host}/api/2.0/automl/get-forecasting-experiment/${req.experimentId ?? ''}`;
+    let resp: ForecastingExperiment | undefined;
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const headers = new Headers();
+      if (this.workspaceId !== undefined) {
+        headers.set('X-Databricks-Org-Id', this.workspaceId);
+      }
+      headers.set('User-Agent', this.userAgent);
+      const httpReq = buildHttpRequest('GET', url, headers, callSignal);
+      const respBody = await executeHttpCall({
+        request: httpReq,
+        httpClient: this.httpClient,
+        logger: this.logger,
+      });
+      resp = parseResponse(respBody, unmarshalForecastingExperimentSchema);
+    };
+    await executeCall(call, options);
+    if (resp === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return resp;
+  }
+}
+
+export class CreateForecastingExperimentWaiter {
+  constructor(
+    private readonly client: ForecastingClient,
+    readonly experimentId: string
+  ) {}
+
+  /**
+   * Polls until the operation reaches a terminal state.
+   *
+   * Throws if a failure state is reached.
+   */
+  async wait(options?: CallOptions): Promise<ForecastingExperiment> {
+    let result: ForecastingExperiment | undefined;
+
+    const call: Call = async (callSignal?: AbortSignal): Promise<void> => {
+      const pollResp = await this.client.getForecastingExperiment(
+        {
+          experimentId: this.experimentId,
+        },
+        {...options, ...(callSignal !== undefined && {signal: callSignal})}
+      );
+
+      const status = pollResp.state;
+      if (status === undefined) {
+        throw new Error('response missing required status field');
+      }
+
+      switch (status) {
+        case ForecastingExperiment_State.SUCCEEDED:
+          result = pollResp;
+          return;
+        case ForecastingExperiment_State.FAILED:
+        case ForecastingExperiment_State.CANCELLED: {
+          const msg = '(no message)';
+          throw new Error(`terminal state ${status}: ${msg}`);
+        }
+        default:
+          throw new StillRunningError();
+      }
+    };
+
+    const retryOptions: CallOptions = {
+      ...(options?.signal !== undefined && {signal: options.signal}),
+      retrier: () =>
+        retryOn({}, (err: Error) => {
+          return err instanceof StillRunningError;
+        }),
+    };
+    await executeCall(call, retryOptions);
+    if (result === undefined) {
+      throw new Error('API call completed without a result.');
+    }
+    return result;
+  }
+
+  /** Checks whether the operation has reached a terminal state. */
+  async done(options?: CallOptions): Promise<boolean> {
+    const pollResp = await this.client.getForecastingExperiment(
+      {
+        experimentId: this.experimentId,
+      },
+      options
+    );
+
+    const status = pollResp.state;
+    if (status === undefined) {
+      throw new Error('response missing required status field');
+    }
+
+    switch (status) {
+      case ForecastingExperiment_State.SUCCEEDED:
+      case ForecastingExperiment_State.FAILED:
+      case ForecastingExperiment_State.CANCELLED:
+        return true;
+      default:
+        return false;
+    }
+  }
+}

package/src/v1/index.ts

@@ -0,0 +1,12 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+
+export {ForecastingClient, CreateForecastingExperimentWaiter} from './client';
+
+export {ForecastingExperiment_State} from './model';
+
+export type {
+  CreateForecastingExperimentRequest,
+  CreateForecastingExperimentResponse,
+  ForecastingExperiment,
+  GetForecastingExperimentRequest,
+} from './model';

package/src/v1/model.ts

@@ -0,0 +1,146 @@
+// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
+
+import {z} from 'zod';
+
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested enum name.
+export enum ForecastingExperiment_State {
+  /** The forecasting experiment is pending and has not started yet. */
+  PENDING = 'PENDING',
+  /** The forecasting experiment is currently running. */
+  RUNNING = 'RUNNING',
+  /** The forecasting experiment has completed successfully. */
+  SUCCEEDED = 'SUCCEEDED',
+  /** The forecasting experiment has failed. */
+  FAILED = 'FAILED',
+  /** The forecasting experiment has been cancelled. */
+  CANCELLED = 'CANCELLED',
+}
+
+export interface CreateForecastingExperimentRequest {
+  /** The fully qualified path of a Unity Catalog table, formatted as catalog_name.schema_name.table_name, used as training data for the forecasting model. */
+  trainDataPath?: string | undefined;
+  /** The column in the input training table used as the prediction target for model training. The values in this column are used as the ground truth for model training. */
+  targetColumn?: string | undefined;
+  /** The column in the input training table that represents each row's timestamp. */
+  timeColumn?: string | undefined;
+  /**
+   * The time interval between consecutive rows in the time series data.
+   * Possible values include: '1 second', '1 minute', '5 minutes', '10 minutes', '15 minutes', '30 minutes', 'Hourly', 'Daily', 'Weekly', 'Monthly', 'Quarterly', 'Yearly'.
+   */
+  forecastGranularity?: string | undefined;
+  /** The number of time steps into the future to make predictions, calculated as a multiple of forecast_granularity. This value represents how far ahead the model should forecast. */
+  forecastHorizon?: bigint | undefined;
+  /** The evaluation metric used to optimize the forecasting model. */
+  primaryMetric?: string | undefined;
+  /** List of frameworks to include for model tuning. Possible values are 'Prophet', 'ARIMA', 'DeepAR'. An empty list includes all supported frameworks. */
+  trainingFrameworks?: string[] | undefined;
+  /** The path in the workspace to store the created experiment. */
+  experimentPath?: string | undefined;
+  /** The maximum duration for the experiment in minutes. The experiment stops automatically if it exceeds this limit. */
+  maxRuntime?: bigint | undefined;
+  /** // The column in the training table used for custom data splits. Values must be 'train', 'validate', or 'test'. */
+  splitColumn?: string | undefined;
+  /** The column in the training table used to customize weights for each time series. */
+  customWeightsColumn?: string | undefined;
+  /** The fully qualified path of a Unity Catalog model, formatted as catalog_name.schema_name.model_name, used to store the best model. */
+  registerTo?: string | undefined;
+  /** The region code(s) to automatically add holiday features. Currently supports only one region. */
+  holidayRegions?: string[] | undefined;
+  /** The column in the training table used to group the dataset for predicting individual time series. */
+  timeseriesIdentifierColumns?: string[] | undefined;
+  /** The fully qualified path of a Unity Catalog table, formatted as catalog_name.schema_name.table_name, used to store predictions. */
+  predictionDataPath?: string | undefined;
+  /**
+   * Specifies the list of feature columns to include in model training. These columns must exist
+   * in the training data and be of type string, numerical, or boolean. If not specified, no additional
+   * features will be included.
+   * Note: Certain columns are automatically handled:
+   * - Automatically excluded: split_column, target_column, custom_weights_column.
+   * - Automatically included: time_column.
+   */
+  includeFeatures?: string[] | undefined;
+  /** The fully qualified path of a Unity Catalog table, formatted as catalog_name.schema_name.table_name, used to store future feature data for predictions. */
+  futureFeatureDataPath?: string | undefined;
+}
+
+export interface CreateForecastingExperimentResponse {
+  /** The unique ID of the created forecasting experiment */
+  experimentId?: string | undefined;
+}
+
+/** Represents a forecasting experiment with its unique identifier, URL, and state. */
+export interface ForecastingExperiment {
+  /** The unique ID for the forecasting experiment. */
+  experimentId?: string | undefined;
+  /** The URL to the forecasting experiment page. */
+  experimentPageUrl?: string | undefined;
+  /** The current state of the forecasting experiment. */
+  state?: ForecastingExperiment_State | undefined;
+}
+
+export interface GetForecastingExperimentRequest {
+  /** The unique ID of a forecasting experiment */
+  experimentId?: string | undefined;
+}
+
+export const unmarshalCreateForecastingExperimentResponseSchema: z.ZodType<CreateForecastingExperimentResponse> =
+  z
+    .object({
+      experiment_id: z.string().optional(),
+    })
+    .transform(d => ({
+      experimentId: d.experiment_id,
+    }));
+
+export const unmarshalForecastingExperimentSchema: z.ZodType<ForecastingExperiment> =
+  z
+    .object({
+      experiment_id: z.string().optional(),
+      experiment_page_url: z.string().optional(),
+      state: z.enum(ForecastingExperiment_State).optional(),
+    })
+    .transform(d => ({
+      experimentId: d.experiment_id,
+      experimentPageUrl: d.experiment_page_url,
+      state: d.state,
+    }));
+
+export const marshalCreateForecastingExperimentRequestSchema: z.ZodType = z
+  .object({
+    trainDataPath: z.string().optional(),
+    targetColumn: z.string().optional(),
+    timeColumn: z.string().optional(),
+    forecastGranularity: z.string().optional(),
+    forecastHorizon: z.bigint().optional(),
+    primaryMetric: z.string().optional(),
+    trainingFrameworks: z.array(z.string()).optional(),
+    experimentPath: z.string().optional(),
+    maxRuntime: z.bigint().optional(),
+    splitColumn: z.string().optional(),
+    customWeightsColumn: z.string().optional(),
+    registerTo: z.string().optional(),
+    holidayRegions: z.array(z.string()).optional(),
+    timeseriesIdentifierColumns: z.array(z.string()).optional(),
+    predictionDataPath: z.string().optional(),
+    includeFeatures: z.array(z.string()).optional(),
+    futureFeatureDataPath: z.string().optional(),
+  })
+  .transform(d => ({
+    train_data_path: d.trainDataPath,
+    target_column: d.targetColumn,
+    time_column: d.timeColumn,
+    forecast_granularity: d.forecastGranularity,
+    forecast_horizon: d.forecastHorizon,
+    primary_metric: d.primaryMetric,
+    training_frameworks: d.trainingFrameworks,
+    experiment_path: d.experimentPath,
+    max_runtime: d.maxRuntime,
+    split_column: d.splitColumn,
+    custom_weights_column: d.customWeightsColumn,
+    register_to: d.registerTo,
+    holiday_regions: d.holidayRegions,
+    timeseries_identifier_columns: d.timeseriesIdentifierColumns,
+    prediction_data_path: d.predictionDataPath,
+    include_features: d.includeFeatures,
+    future_feature_data_path: d.futureFeatureDataPath,
+  }));