@databricks/sdk-vectorsearch 0.1.0-dev.3 → 0.1.0-dev.5

package/src/v1/transport.ts

@@ -1,73 +0,0 @@
-// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
-
-import type {Credentials} from '@databricks/sdk-auth';
-import {defaultCredentials} from '@databricks/sdk-auth/credentials';
-import type {
-  HttpClient,
-  HttpRequest,
-  HttpResponse,
-} from '@databricks/sdk-core/http';
-import {newFetchHttpClient} from '@databricks/sdk-core/http';
-import type {ClientOptions} from '@databricks/sdk-options/client';
-
-/** Creates a new HTTP client with the given options. */
-export function newHttpClient(options?: ClientOptions): HttpClient {
-  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: HttpClient = 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 implements HttpClient {
-  constructor(
-    private readonly base: HttpClient,
-    private readonly credentials: Credentials
-  ) {}
-
-  async send(request: HttpRequest): Promise<HttpResponse> {
-    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 implements HttpClient {
-  constructor(
-    private readonly base: HttpClient,
-    private readonly timeout: number
-  ) {}
-
-  async send(request: HttpRequest): Promise<HttpResponse> {
-    const timeoutSignal = AbortSignal.timeout(this.timeout);
-    const signal =
-      request.signal !== undefined
-        ? AbortSignal.any([request.signal, timeoutSignal])
-        : timeoutSignal;
-    return this.base.send({...request, signal});
-  }
-}

package/src/v1/utils.ts

@@ -1,180 +0,0 @@
-// Code generated from API definition by Databricks SDK Generator. DO NOT EDIT.
-
-import type {Options} from '@databricks/sdk-core/ops';
-import {execute, retryOn} from '@databricks/sdk-core/ops';
-import {ApiError} from '@databricks/sdk-core/apierror';
-import type {
-  HttpClient,
-  HttpRequest,
-  HttpResponse,
-} from '@databricks/sdk-core/http';
-import type {Logger} from '@databricks/sdk-core/logger';
-import type {CallOptions} from '@databricks/sdk-options/call';
-import type {LroOptions} from '@databricks/sdk-options/lro';
-import JSONBig from 'json-bigint';
-import type {z} from 'zod';
-
-// 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});
-
-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 async function executeCall(
-  call: (signal?: AbortSignal) => Promise<void>,
-  options?: CallOptions
-): Promise<void> {
-  const opts: Options = {
-    ...(options?.retrier !== undefined && {retrier: options.retrier}),
-    ...(options?.rateLimiter !== undefined && {
-      rateLimiter: options.rateLimiter,
-    }),
-    ...(options?.timeout !== undefined && {timeout: options.timeout}),
-  };
-  return execute(options?.signal, call, opts);
-}
-
-/**
- * Sentinel thrown by a polling call to signal that the operation has not
- * yet reached a terminal state. {@link executeWait} treats this error as
- * retriable; any other error aborts the wait.
- */
-export class StillRunningError extends Error {}
-
-/**
- * Polls until the call returns without throwing {@link StillRunningError}.
- * Abort and overall-deadline behavior come from the supplied LroOptions.
- */
-export async function executeWait(
-  call: (signal?: AbortSignal) => Promise<void>,
-  options?: LroOptions
-): Promise<void> {
-  const opts: Options = {
-    ...(options?.timeout !== undefined && {timeout: options.timeout}),
-    retrier: () =>
-      retryOn({}, (err: Error) => err instanceof StillRunningError),
-  };
-  return execute(options?.signal, call, opts);
-}
-
-async function readAll(
-  body: ReadableStream<Uint8Array> | null
-): Promise<Uint8Array> {
-  if (body === null) {
-    return new Uint8Array(0);
-  }
-  const reader = body.getReader();
-  const chunks: Uint8Array[] = [];
-  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: HttpCallOptions
-): Promise<Uint8Array> {
-  opts.logger.debug('HTTP request', {
-    method: opts.request.method,
-    url: opts.request.url,
-  });
-
-  let resp: HttpResponse;
-  try {
-    resp = await opts.httpClient.send(opts.request);
-  } catch (e: unknown) {
-    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: string,
-  url: string,
-  headers: Headers,
-  signal?: AbortSignal,
-  body?: string | ReadableStream<Uint8Array>
-): HttpRequest {
-  const req: HttpRequest = {url, method, headers};
-  if (body !== undefined) {
-    req.body = body;
-  }
-  if (signal !== undefined) {
-    req.signal = signal;
-  }
-  return req;
-}
-
-export function parseResponse<T>(body: Uint8Array, schema: z.ZodType<T>): T {
-  const text = new TextDecoder().decode(body);
-  const parsed: unknown = jsonBigint.parse(text);
-  return schema.parse(parsed);
-}
-
-export function marshalRequest(data: unknown, schema: z.ZodType): string {
-  return jsonBigint.stringify(schema.parse(data));
-}
-
-export function flattenQueryParams(
-  prefix: string,
-  value: unknown,
-  params: URLSearchParams
-): void {
-  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 as Record<string, unknown>)) {
-      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}`);
-  }
-}