From f5b1791b062185fa180ccfbfc67a3de12b465780 Mon Sep 17 00:00:00 2001 From: Pawel Kosiec Date: Tue, 10 Feb 2026 17:15:25 +0100 Subject: [PATCH] chore: deprecate the Lakebase V1 connector --- docs/docs/architecture.md | 2 +- docs/docs/plugins.md | 2 +- packages/appkit/src/cache/index.ts | 4 +- .../appkit/src/cache/storage/persistent.ts | 6 +- .../src/cache/tests/cache-manager.test.ts | 4 +- .../appkit/src/cache/tests/persistent.test.ts | 2 +- packages/appkit/src/connectors/index.ts | 2 +- .../{lakebase => lakebase-v1}/client.ts | 62 ++++++++++++------- .../src/connectors/lakebase-v1/defaults.ts | 15 +++++ .../src/connectors/lakebase-v1/index.ts | 6 ++ .../{lakebase => lakebase-v1}/types.ts | 37 +++++++++-- .../{lakebase => lakebase-v1}/utils.ts | 10 +-- .../src/connectors/lakebase/defaults.ts | 10 --- .../appkit/src/connectors/lakebase/index.ts | 2 - .../{lakebase.test.ts => lakebase-v1.test.ts} | 46 +++++++------- 15 files changed, 129 insertions(+), 81 deletions(-) rename packages/appkit/src/connectors/{lakebase => lakebase-v1}/client.ts (90%) create mode 100644 packages/appkit/src/connectors/lakebase-v1/defaults.ts create mode 100644 packages/appkit/src/connectors/lakebase-v1/index.ts rename packages/appkit/src/connectors/{lakebase => lakebase-v1}/types.ts (52%) rename packages/appkit/src/connectors/{lakebase => lakebase-v1}/utils.ts (88%) delete mode 100644 packages/appkit/src/connectors/lakebase/defaults.ts delete mode 100644 packages/appkit/src/connectors/lakebase/index.ts rename packages/appkit/src/connectors/tests/{lakebase.test.ts => lakebase-v1.test.ts} (92%) diff --git a/docs/docs/architecture.md b/docs/docs/architecture.md index b29fbbc1..d9d26c4a 100644 --- a/docs/docs/architecture.md +++ b/docs/docs/architecture.md @@ -83,7 +83,7 @@ The Node.js backend layer built with `@databricks/appkit`: Integration with Databricks services: - **SQL Warehouses**: Execute analytical queries with Arrow or JSON format -- **Lakebase**: Access data from Lakebase +- **Lakebase V1 (Provisioned)**: Access data from Lakebase Provisioned. Support for Lakebase Autoscaling coming soon. ## See also diff --git a/docs/docs/plugins.md b/docs/docs/plugins.md index 1c8fee2a..a1b35119 100644 --- a/docs/docs/plugins.md +++ b/docs/docs/plugins.md @@ -292,7 +292,7 @@ await createApp({ }); ``` -Storage auto-selects **Lakebase persistent cache when healthy**, otherwise falls back to in-memory. +Storage auto-selects **Lakebase V1 (Provisioned) persistent cache when healthy**, otherwise falls back to in-memory. Support for Lakebase Autoscaling coming soon. ### Plugin-level caching diff --git a/packages/appkit/src/cache/index.ts b/packages/appkit/src/cache/index.ts index 140f36b8..952a7987 100644 --- a/packages/appkit/src/cache/index.ts +++ b/packages/appkit/src/cache/index.ts @@ -1,7 +1,7 @@ import { createHash } from "node:crypto"; import { WorkspaceClient } from "@databricks/sdk-experimental"; import type { CacheConfig, CacheStorage } from "shared"; -import { LakebaseConnector } from "@/connectors"; +import { LakebaseV1Connector } from "@/connectors"; import { AppKitError, ExecutionError, InitializationError } from "../errors"; import { createLogger } from "../logging/logger"; import type { Counter, TelemetryProvider } from "../telemetry"; @@ -147,7 +147,7 @@ export class CacheManager { // try to use lakebase storage try { const workspaceClient = new WorkspaceClient({}); - const connector = new LakebaseConnector({ workspaceClient }); + const connector = new LakebaseV1Connector({ workspaceClient }); const isHealthy = await connector.healthCheck(); if (isHealthy) { diff --git a/packages/appkit/src/cache/storage/persistent.ts b/packages/appkit/src/cache/storage/persistent.ts index 2ae9ff99..393385cd 100644 --- a/packages/appkit/src/cache/storage/persistent.ts +++ b/packages/appkit/src/cache/storage/persistent.ts @@ -1,6 +1,6 @@ import { createHash } from "node:crypto"; import type { CacheConfig, CacheEntry, CacheStorage } from "shared"; -import type { LakebaseConnector } from "../../connectors"; +import type { LakebaseV1Connector } from "../../connectors"; import { InitializationError, ValidationError } from "../../errors"; import { createLogger } from "../../logging/logger"; import { lakebaseStorageDefaults } from "./defaults"; @@ -22,7 +22,7 @@ const logger = createLogger("cache:persistent"); * */ export class PersistentStorage implements CacheStorage { - private readonly connector: LakebaseConnector; + private readonly connector: LakebaseV1Connector; private readonly tableName: string; private readonly maxBytes: number; private readonly maxEntryBytes: number; @@ -30,7 +30,7 @@ export class PersistentStorage implements CacheStorage { private readonly evictionCheckProbability: number; private initialized: boolean; - constructor(config: CacheConfig, connector: LakebaseConnector) { + constructor(config: CacheConfig, connector: LakebaseV1Connector) { this.connector = connector; this.maxBytes = config.maxBytes ?? lakebaseStorageDefaults.maxBytes; this.maxEntryBytes = diff --git a/packages/appkit/src/cache/tests/cache-manager.test.ts b/packages/appkit/src/cache/tests/cache-manager.test.ts index 65d6c0fd..a4e54438 100644 --- a/packages/appkit/src/cache/tests/cache-manager.test.ts +++ b/packages/appkit/src/cache/tests/cache-manager.test.ts @@ -2,10 +2,10 @@ import type { CacheStorage } from "shared"; import { afterEach, beforeEach, describe, expect, test, vi } from "vitest"; import { CacheManager } from "../../index"; -// Mock LakebaseConnector +// Mock LakebaseV1Connector const mockLakebaseHealthCheck = vi.fn(); vi.mock("@/connectors", () => ({ - LakebaseConnector: vi.fn().mockImplementation(() => ({ + LakebaseV1Connector: vi.fn().mockImplementation(() => ({ healthCheck: mockLakebaseHealthCheck, close: vi.fn().mockResolvedValue(undefined), })), diff --git a/packages/appkit/src/cache/tests/persistent.test.ts b/packages/appkit/src/cache/tests/persistent.test.ts index 6e658f4b..fc3ead27 100644 --- a/packages/appkit/src/cache/tests/persistent.test.ts +++ b/packages/appkit/src/cache/tests/persistent.test.ts @@ -1,7 +1,7 @@ import { beforeEach, describe, expect, test, vi } from "vitest"; import { PersistentStorage } from "../storage"; -/** Mock LakebaseConnector for testing */ +/** Mock LakebaseV1Connector for testing */ const createMockConnector = () => ({ query: vi.fn(), healthCheck: vi.fn().mockResolvedValue(true), diff --git a/packages/appkit/src/connectors/index.ts b/packages/appkit/src/connectors/index.ts index 70702b4b..01d23cf2 100644 --- a/packages/appkit/src/connectors/index.ts +++ b/packages/appkit/src/connectors/index.ts @@ -1,2 +1,2 @@ -export * from "./lakebase"; +export * from "./lakebase-v1"; export * from "./sql-warehouse"; diff --git a/packages/appkit/src/connectors/lakebase/client.ts b/packages/appkit/src/connectors/lakebase-v1/client.ts similarity index 90% rename from packages/appkit/src/connectors/lakebase/client.ts rename to packages/appkit/src/connectors/lakebase-v1/client.ts index 89420eff..ed3e2fea 100644 --- a/packages/appkit/src/connectors/lakebase/client.ts +++ b/packages/appkit/src/connectors/lakebase-v1/client.ts @@ -18,37 +18,51 @@ import { } from "../../errors"; import { createLogger } from "../../logging/logger"; import { deepMerge } from "../../utils"; -import { lakebaseDefaults } from "./defaults"; +import { lakebaseV1Defaults } from "./defaults"; import type { - LakebaseConfig, - LakebaseConnectionConfig, - LakebaseCredentials, + LakebaseV1Config, + LakebaseV1ConnectionConfig, + LakebaseV1Credentials, } from "./types"; -const logger = createLogger("connectors:lakebase"); +const logger = createLogger("connectors:lakebase-v1"); /** - * Enterprise-grade connector for Databricks Lakebase + * Enterprise-grade connector for Databricks Lakebase Provisioned + * + * @deprecated This connector is for Lakebase Provisioned only. + * For new projects, use Lakebase Autoscaling instead: https://docs.databricks.com/aws/en/oltp/projects/ + * + * This connector is compatible with Lakebase Provisioned: https://docs.databricks.com/aws/en/oltp/instances/ + * + * Lakebase Autoscaling offers: + * - Automatic compute scaling + * - Scale-to-zero for cost optimization + * - Database branching for development + * - Instant restore capabilities + * + * Use the new LakebaseConnector (coming in a future release) for Lakebase Autoscaling support. + * * @example Simplest - everything from env/context * ```typescript - * const connector = new LakebaseConnector(); + * const connector = new LakebaseV1Connector(); * await connector.query('SELECT * FROM users'); * ``` * * @example With explicit connection string * ```typescript - * const connector = new LakebaseConnector({ + * const connector = new LakebaseV1Connector({ * connectionString: 'postgresql://...' * }); * ``` */ -export class LakebaseConnector { - private readonly name: string = "lakebase"; +export class LakebaseV1Connector { + private readonly name: string = "lakebase-v1"; private readonly CACHE_BUFFER_MS = 2 * 60 * 1000; - private readonly config: LakebaseConfig; - private readonly connectionConfig: LakebaseConnectionConfig; + private readonly config: LakebaseV1Config; + private readonly connectionConfig: LakebaseV1ConnectionConfig; private pool: pg.Pool | null = null; - private credentials: LakebaseCredentials | null = null; + private credentials: LakebaseV1Credentials | null = null; // telemetry private readonly telemetry: TelemetryProvider; @@ -57,8 +71,8 @@ export class LakebaseConnector { queryDuration: Histogram; }; - constructor(userConfig?: Partial) { - this.config = deepMerge(lakebaseDefaults, userConfig); + constructor(userConfig?: Partial) { + this.config = deepMerge(lakebaseV1Defaults, userConfig); this.connectionConfig = this.parseConnectionConfig(); this.telemetry = TelemetryManager.getProvider( @@ -68,13 +82,13 @@ export class LakebaseConnector { this.telemetryMetrics = { queryCount: this.telemetry .getMeter() - .createCounter("lakebase.query.count", { + .createCounter("lakebase.v1.query.count", { description: "Total number of queries executed", unit: "1", }), queryDuration: this.telemetry .getMeter() - .createHistogram("lakebase.query.duration", { + .createHistogram("lakebase.v1.query.duration", { description: "Duration of queries executed", unit: "ms", }), @@ -107,10 +121,10 @@ export class LakebaseConnector { const startTime = Date.now(); return this.telemetry.startActiveSpan( - "lakebase.query", + "lakebase.v1.query", { attributes: { - "db.system": "lakebase", + "db.system": "lakebase-v1", "db.statement": sql.substring(0, 500), "db.retry_count": retryCount, }, @@ -178,10 +192,10 @@ export class LakebaseConnector { ): Promise { const startTime = Date.now(); return this.telemetry.startActiveSpan( - "lakebase.transaction", + "lakebase.v1.transaction", { attributes: { - "db.system": "lakebase", + "db.system": "lakebase-v1", "db.retry_count": retryCount, }, }, @@ -249,7 +263,7 @@ export class LakebaseConnector { /** Check if database connection is healthy */ async healthCheck(): Promise { return this.telemetry.startActiveSpan( - "lakebase.healthCheck", + "lakebase.v1.healthCheck", {}, async (span) => { try { @@ -491,7 +505,7 @@ export class LakebaseConnector { } /** Parse connection configuration from config or environment */ - private parseConnectionConfig(): LakebaseConnectionConfig { + private parseConnectionConfig(): LakebaseV1ConnectionConfig { if (this.config.connectionString) { return this.parseConnectionString(this.config.connectionString); } @@ -539,7 +553,7 @@ export class LakebaseConnector { private parseConnectionString( connectionString: string, - ): LakebaseConnectionConfig { + ): LakebaseV1ConnectionConfig { const url = new URL(connectionString); const appName = url.searchParams.get("appName"); if (!appName) { diff --git a/packages/appkit/src/connectors/lakebase-v1/defaults.ts b/packages/appkit/src/connectors/lakebase-v1/defaults.ts new file mode 100644 index 00000000..1a8abdf5 --- /dev/null +++ b/packages/appkit/src/connectors/lakebase-v1/defaults.ts @@ -0,0 +1,15 @@ +import type { LakebaseV1Config } from "./types"; + +/** + * Default configuration for Lakebase V1 connector + * + * @deprecated This connector is for Lakebase Provisioned only. + * For new projects, use Lakebase Autoscaling: https://docs.databricks.com/aws/en/oltp/projects/ + */ +export const lakebaseV1Defaults: LakebaseV1Config = { + port: 5432, + sslMode: "require", + maxPoolSize: 10, + idleTimeoutMs: 30_000, + connectionTimeoutMs: 10_000, +}; diff --git a/packages/appkit/src/connectors/lakebase-v1/index.ts b/packages/appkit/src/connectors/lakebase-v1/index.ts new file mode 100644 index 00000000..a173693f --- /dev/null +++ b/packages/appkit/src/connectors/lakebase-v1/index.ts @@ -0,0 +1,6 @@ +export { LakebaseV1Connector } from "./client"; +export type { + LakebaseV1Config, + LakebaseV1ConnectionConfig, + LakebaseV1Credentials, +} from "./types"; diff --git a/packages/appkit/src/connectors/lakebase/types.ts b/packages/appkit/src/connectors/lakebase-v1/types.ts similarity index 52% rename from packages/appkit/src/connectors/lakebase/types.ts rename to packages/appkit/src/connectors/lakebase-v1/types.ts index a8591b50..fcaddaad 100644 --- a/packages/appkit/src/connectors/lakebase/types.ts +++ b/packages/appkit/src/connectors/lakebase-v1/types.ts @@ -1,8 +1,23 @@ import type { WorkspaceClient } from "@databricks/sdk-experimental"; import type { TelemetryOptions } from "shared"; -/** Configuration for LakebaseConnector */ -export interface LakebaseConfig { +/** + * Configuration for LakebaseV1Connector + * + * @deprecated This connector is for Lakebase Provisioned only. + * For new projects, use Lakebase Autoscaling instead: https://docs.databricks.com/aws/en/oltp/projects/ + * + * This connector is compatible with Lakebase Provisioned: https://docs.databricks.com/aws/en/oltp/instances/ + * + * Lakebase Autoscaling offers: + * - Automatic compute scaling + * - Scale-to-zero for cost optimization + * - Database branching for development + * - Instant restore capabilities + * + * Use the new LakebaseConnector (coming in a future release) for Lakebase Autoscaling support. + */ +export interface LakebaseV1Config { /** Databricks workspace client */ workspaceClient?: WorkspaceClient; @@ -40,8 +55,13 @@ export interface LakebaseConfig { [key: string]: unknown; } -/** Lakebase credentials for authentication */ -export interface LakebaseCredentials { +/** + * Lakebase V1 credentials for authentication + * + * @deprecated This type is for Lakebase Provisioned only. + * For new projects, use Lakebase Autoscaling: https://docs.databricks.com/aws/en/oltp/projects/ + */ +export interface LakebaseV1Credentials { /** Username */ username: string; /** Password */ @@ -50,8 +70,13 @@ export interface LakebaseCredentials { expiresAt: number; } -/** Internal connection configuration */ -export interface LakebaseConnectionConfig { +/** + * Internal connection configuration for Lakebase V1 + * + * @deprecated This type is for Lakebase Provisioned only. + * For new projects, use Lakebase Autoscaling: https://docs.databricks.com/aws/en/oltp/projects/ + */ +export interface LakebaseV1ConnectionConfig { /** Database host */ readonly host: string; /** Database name */ diff --git a/packages/appkit/src/connectors/lakebase/utils.ts b/packages/appkit/src/connectors/lakebase-v1/utils.ts similarity index 88% rename from packages/appkit/src/connectors/lakebase/utils.ts rename to packages/appkit/src/connectors/lakebase-v1/utils.ts index 5bf726d4..be362962 100644 --- a/packages/appkit/src/connectors/lakebase/utils.ts +++ b/packages/appkit/src/connectors/lakebase-v1/utils.ts @@ -1,8 +1,8 @@ import { ConfigurationError, ValidationError } from "../../errors"; -import type { LakebaseConnectionConfig } from "./types"; +import type { LakebaseV1ConnectionConfig } from "./types"; export interface ParsedConnectionString { - connectionParams: LakebaseConnectionConfig; + connectionParams: LakebaseV1ConnectionConfig; originalConnectionString: string; } @@ -40,7 +40,7 @@ export function parseConnectionString( (url.searchParams.get("sslmode") as "require" | "disable" | "prefer") || "require"; - const connectionParams: LakebaseConnectionConfig = { + const connectionParams: LakebaseV1ConnectionConfig = { host: url.hostname, database: dbName, port: url.port ? parseInt(url.port, 10) : 5432, @@ -57,7 +57,7 @@ export function parseConnectionString( throw ValidationError.missingField("database"); } - const connectionParams: LakebaseConnectionConfig = { + const connectionParams: LakebaseV1ConnectionConfig = { host: connectionStringOrHost, database, port: port || 5432, @@ -71,7 +71,7 @@ export function parseConnectionString( } /** Parse connection configuration from environment variables */ -export function parseFromEnv(): LakebaseConnectionConfig { +export function parseFromEnv(): LakebaseV1ConnectionConfig { const host = process.env.PGHOST; const database = process.env.PGDATABASE; const port = process.env.PGPORT ? parseInt(process.env.PGPORT, 10) : 5432; diff --git a/packages/appkit/src/connectors/lakebase/defaults.ts b/packages/appkit/src/connectors/lakebase/defaults.ts deleted file mode 100644 index c0253347..00000000 --- a/packages/appkit/src/connectors/lakebase/defaults.ts +++ /dev/null @@ -1,10 +0,0 @@ -import type { LakebaseConfig } from "./types"; - -/** Default configuration for Lakebase connector */ -export const lakebaseDefaults: LakebaseConfig = { - port: 5432, - sslMode: "require", - maxPoolSize: 10, - idleTimeoutMs: 30_000, - connectionTimeoutMs: 10_000, -}; diff --git a/packages/appkit/src/connectors/lakebase/index.ts b/packages/appkit/src/connectors/lakebase/index.ts deleted file mode 100644 index fe028c40..00000000 --- a/packages/appkit/src/connectors/lakebase/index.ts +++ /dev/null @@ -1,2 +0,0 @@ -export { LakebaseConnector } from "./client"; -export type { LakebaseConfig } from "./types"; diff --git a/packages/appkit/src/connectors/tests/lakebase.test.ts b/packages/appkit/src/connectors/tests/lakebase-v1.test.ts similarity index 92% rename from packages/appkit/src/connectors/tests/lakebase.test.ts rename to packages/appkit/src/connectors/tests/lakebase-v1.test.ts index 2b8b30b1..b1b5992c 100644 --- a/packages/appkit/src/connectors/tests/lakebase.test.ts +++ b/packages/appkit/src/connectors/tests/lakebase-v1.test.ts @@ -1,5 +1,5 @@ import { beforeEach, describe, expect, test, vi } from "vitest"; -import { LakebaseConnector } from "../lakebase"; +import { LakebaseV1Connector } from "../lakebase-v1"; // Mock pg module vi.mock("pg", () => { @@ -55,7 +55,7 @@ vi.mock("@databricks/sdk-experimental", () => { }; }); -describe("LakebaseConnector", () => { +describe("LakebaseV1Connector", () => { beforeEach(() => { vi.clearAllMocks(); // Set required env vars @@ -68,7 +68,7 @@ describe("LakebaseConnector", () => { test("should throw error when maxPoolSize is less than 1", () => { expect( () => - new LakebaseConnector({ + new LakebaseV1Connector({ maxPoolSize: 0, workspaceClient: {} as any, }), @@ -76,11 +76,11 @@ describe("LakebaseConnector", () => { }); test("should create connector with valid config", () => { - const connector = new LakebaseConnector({ + const connector = new LakebaseV1Connector({ workspaceClient: {} as any, }); - expect(connector).toBeInstanceOf(LakebaseConnector); + expect(connector).toBeInstanceOf(LakebaseV1Connector); }); test("should throw when env vars are missing", () => { @@ -88,7 +88,7 @@ describe("LakebaseConnector", () => { delete process.env.PGDATABASE; delete process.env.PGAPPNAME; - expect(() => new LakebaseConnector()).toThrow( + expect(() => new LakebaseV1Connector()).toThrow( "Lakebase connection not configured", ); }); @@ -96,20 +96,20 @@ describe("LakebaseConnector", () => { test("should throw when PGPORT is invalid", () => { process.env.PGPORT = "invalid"; - expect(() => new LakebaseConnector()).toThrow("Invalid value for port"); + expect(() => new LakebaseV1Connector()).toThrow("Invalid value for port"); }); test("should parse env vars correctly", () => { process.env.PGPORT = "5433"; process.env.PGSSLMODE = "disable"; - const connector = new LakebaseConnector(); + const connector = new LakebaseV1Connector(); - expect(connector).toBeInstanceOf(LakebaseConnector); + expect(connector).toBeInstanceOf(LakebaseV1Connector); }); test("should use explicit config over env vars", () => { - const connector = new LakebaseConnector({ + const connector = new LakebaseV1Connector({ host: "explicit-host.databricks.com", database: "explicit-db", appName: "explicit-app", @@ -118,12 +118,12 @@ describe("LakebaseConnector", () => { workspaceClient: {} as any, }); - expect(connector).toBeInstanceOf(LakebaseConnector); + expect(connector).toBeInstanceOf(LakebaseV1Connector); }); }); describe("query", () => { - let connector: LakebaseConnector; + let connector: LakebaseV1Connector; let mockQuery: ReturnType; let mockMe: ReturnType; let mockRequest: ReturnType; @@ -144,7 +144,7 @@ describe("LakebaseConnector", () => { }); mockQuery.mockResolvedValue({ rows: [{ result: 1 }] }); - connector = new LakebaseConnector({ + connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" }, @@ -224,7 +224,7 @@ describe("LakebaseConnector", () => { }); describe("transaction", () => { - let connector: LakebaseConnector; + let connector: LakebaseV1Connector; let mockConnect: ReturnType; let mockMe: ReturnType; let mockRequest: ReturnType; @@ -249,7 +249,7 @@ describe("LakebaseConnector", () => { }; mockConnect.mockResolvedValue(mockClient); - connector = new LakebaseConnector({ + connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" }, @@ -285,7 +285,7 @@ describe("LakebaseConnector", () => { }); describe("healthCheck", () => { - let connector: LakebaseConnector; + let connector: LakebaseV1Connector; let mockQuery: ReturnType; let mockMe: ReturnType; let mockRequest: ReturnType; @@ -304,7 +304,7 @@ describe("LakebaseConnector", () => { expiration_time: new Date(Date.now() + 3600000).toISOString(), }); - connector = new LakebaseConnector({ + connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" }, @@ -338,7 +338,7 @@ describe("LakebaseConnector", () => { }); describe("close", () => { - let connector: LakebaseConnector; + let connector: LakebaseV1Connector; let mockEnd: ReturnType; let mockQuery: ReturnType; let mockMe: ReturnType; @@ -361,7 +361,7 @@ describe("LakebaseConnector", () => { mockQuery.mockResolvedValue({ rows: [{ result: 1 }] }); mockEnd.mockResolvedValue(undefined); - connector = new LakebaseConnector({ + connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" }, @@ -404,7 +404,7 @@ describe("LakebaseConnector", () => { mockMe.mockResolvedValue({ userName: null }); mockRequest.mockResolvedValue({ token: "test-token" }); - const connector = new LakebaseConnector({ + const connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" }, @@ -420,7 +420,7 @@ describe("LakebaseConnector", () => { mockMe.mockResolvedValue({ userName: "test-user@example.com" }); mockRequest.mockResolvedValue({ error: "unauthorized" }); // missing token and expiration_time - const connector = new LakebaseConnector({ + const connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" }, @@ -434,7 +434,7 @@ describe("LakebaseConnector", () => { }); describe("transient error codes", () => { - let connector: LakebaseConnector; + let connector: LakebaseV1Connector; let mockQuery: ReturnType; let mockMe: ReturnType; let mockRequest: ReturnType; @@ -453,7 +453,7 @@ describe("LakebaseConnector", () => { expiration_time: new Date(Date.now() + 3600000).toISOString(), }); - connector = new LakebaseConnector({ + connector = new LakebaseV1Connector({ workspaceClient: { currentUser: { me: mockMe }, config: { host: "https://test.databricks.com" },