fix: sdk import

apps/dev-playground/client/src/routeTree.gen.ts

@@ -13,11 +13,8 @@ import { Route as TypeSafetyRouteRouteImport } from './routes/type-safety.route'
 import { Route as TelemetryRouteRouteImport } from './routes/telemetry.route'
 import { Route as SqlHelpersRouteRouteImport } from './routes/sql-helpers.route'
 import { Route as ReconnectRouteRouteImport } from './routes/reconnect.route'
-
-import { Route as GenieRouteRouteImport } from './routes/genie.route'
-
 import { Route as LakebaseRouteRouteImport } from './routes/lakebase.route'
-
+import { Route as GenieRouteRouteImport } from './routes/genie.route'
 import { Route as DataVisualizationRouteRouteImport } from './routes/data-visualization.route'
 import { Route as ArrowAnalyticsRouteRouteImport } from './routes/arrow-analytics.route'
 import { Route as AnalyticsRouteRouteImport } from './routes/analytics.route'
@@ -43,17 +40,16 @@ const ReconnectRouteRoute = ReconnectRouteRouteImport.update({
   path: '/reconnect',
   getParentRoute: () => rootRouteImport,
 } as any)
-
-const GenieRouteRoute = GenieRouteRouteImport.update({
-  id: '/genie',
-  path: '/genie',
-  getParentRoute: () => rootRouteImport,
-} as any)
 const LakebaseRouteRoute = LakebaseRouteRouteImport.update({
   id: '/lakebase',
   path: '/lakebase',
   getParentRoute: () => rootRouteImport,
 } as any)
+const GenieRouteRoute = GenieRouteRouteImport.update({
+  id: '/genie',
+  path: '/genie',
+  getParentRoute: () => rootRouteImport,
+} as any)
 const DataVisualizationRouteRoute = DataVisualizationRouteRouteImport.update({
   id: '/data-visualization',
   path: '/data-visualization',
@@ -80,11 +76,8 @@ export interface FileRoutesByFullPath {
   '/analytics': typeof AnalyticsRouteRoute
   '/arrow-analytics': typeof ArrowAnalyticsRouteRoute
   '/data-visualization': typeof DataVisualizationRouteRoute
-
   '/genie': typeof GenieRouteRoute
-
   '/lakebase': typeof LakebaseRouteRoute
-
   '/reconnect': typeof ReconnectRouteRoute
   '/sql-helpers': typeof SqlHelpersRouteRoute
   '/telemetry': typeof TelemetryRouteRoute
@@ -95,11 +88,8 @@ export interface FileRoutesByTo {
   '/analytics': typeof AnalyticsRouteRoute
   '/arrow-analytics': typeof ArrowAnalyticsRouteRoute
   '/data-visualization': typeof DataVisualizationRouteRoute
-
   '/genie': typeof GenieRouteRoute
-
   '/lakebase': typeof LakebaseRouteRoute
-
   '/reconnect': typeof ReconnectRouteRoute
   '/sql-helpers': typeof SqlHelpersRouteRoute
   '/telemetry': typeof TelemetryRouteRoute
@@ -111,11 +101,8 @@ export interface FileRoutesById {
   '/analytics': typeof AnalyticsRouteRoute
   '/arrow-analytics': typeof ArrowAnalyticsRouteRoute
   '/data-visualization': typeof DataVisualizationRouteRoute
-
   '/genie': typeof GenieRouteRoute
-
   '/lakebase': typeof LakebaseRouteRoute
-
   '/reconnect': typeof ReconnectRouteRoute
   '/sql-helpers': typeof SqlHelpersRouteRoute
   '/telemetry': typeof TelemetryRouteRoute
@@ -128,11 +115,8 @@ export interface FileRouteTypes {
     | '/analytics'
     | '/arrow-analytics'
     | '/data-visualization'
-
     | '/genie'
-
     | '/lakebase'
-
     | '/reconnect'
     | '/sql-helpers'
     | '/telemetry'
@@ -143,11 +127,8 @@ export interface FileRouteTypes {
     | '/analytics'
     | '/arrow-analytics'
     | '/data-visualization'
-
     | '/genie'
-
     | '/lakebase'
-
     | '/reconnect'
     | '/sql-helpers'
     | '/telemetry'
@@ -158,11 +139,8 @@ export interface FileRouteTypes {
     | '/analytics'
     | '/arrow-analytics'
     | '/data-visualization'
-
     | '/genie'
-
     | '/lakebase'
-
     | '/reconnect'
     | '/sql-helpers'
     | '/telemetry'
@@ -174,11 +152,8 @@ export interface RootRouteChildren {
   AnalyticsRouteRoute: typeof AnalyticsRouteRoute
   ArrowAnalyticsRouteRoute: typeof ArrowAnalyticsRouteRoute
   DataVisualizationRouteRoute: typeof DataVisualizationRouteRoute
-
   GenieRouteRoute: typeof GenieRouteRoute
-
   LakebaseRouteRoute: typeof LakebaseRouteRoute
-
   ReconnectRouteRoute: typeof ReconnectRouteRoute
   SqlHelpersRouteRoute: typeof SqlHelpersRouteRoute
   TelemetryRouteRoute: typeof TelemetryRouteRoute
@@ -215,21 +190,20 @@ declare module '@tanstack/react-router' {
       preLoaderRoute: typeof ReconnectRouteRouteImport
       parentRoute: typeof rootRouteImport
     }
-
-    '/genie': {
-      id: '/genie'
-      path: '/genie'
-      fullPath: '/genie'
-      preLoaderRoute: typeof GenieRouteRouteImport
-      parentRoute: typeof rootRouteImport
-    }
     '/lakebase': {
       id: '/lakebase'
       path: '/lakebase'
       fullPath: '/lakebase'
       preLoaderRoute: typeof LakebaseRouteRouteImport
       parentRoute: typeof rootRouteImport
     }
+    '/genie': {
+      id: '/genie'
+      path: '/genie'
+      fullPath: '/genie'
+      preLoaderRoute: typeof GenieRouteRouteImport
+      parentRoute: typeof rootRouteImport
+    }
     '/data-visualization': {
       id: '/data-visualization'
       path: '/data-visualization'
@@ -266,11 +240,8 @@ const rootRouteChildren: RootRouteChildren = {
   AnalyticsRouteRoute: AnalyticsRouteRoute,
   ArrowAnalyticsRouteRoute: ArrowAnalyticsRouteRoute,
   DataVisualizationRouteRoute: DataVisualizationRouteRoute,
-
   GenieRouteRoute: GenieRouteRoute,
-
   LakebaseRouteRoute: LakebaseRouteRoute,
-
   ReconnectRouteRoute: ReconnectRouteRoute,
   SqlHelpersRouteRoute: SqlHelpersRouteRoute,
   TelemetryRouteRoute: TelemetryRouteRoute,

packages/appkit/src/cache/tests/cache-manager.test.ts

@@ -48,10 +48,15 @@ vi.mock("../storage/persistent", () => ({
   }),
 }));
 
-// Mock WorkspaceClient
-vi.mock("@databricks/sdk-experimental", () => ({
-  WorkspaceClient: vi.fn().mockImplementation(() => ({})),
-}));
+// Mock WorkspaceClient (preserve Time, TimeUnits, and other SDK exports)
+vi.mock("@databricks/sdk-experimental", async (importOriginal) => {
+  const actual =
+    await importOriginal<typeof import("@databricks/sdk-experimental")>();
+  return {
+    ...actual,
+    WorkspaceClient: vi.fn().mockImplementation(() => ({})),
+  };
+});
 
 /** Create a mock storage for testing */
 function createMockStorage(persistent = false): CacheStorage {

packages/appkit/src/connectors/genie/client.ts

@@ -1,5 +1,5 @@
 import type { WorkspaceClient } from "@databricks/sdk-experimental";
-import { Time, TimeUnits } from "@databricks/sdk-experimental";
+import * as SDK from "@databricks/sdk-experimental";
 import type { GenieMessage } from "@databricks/sdk-experimental/dist/apis/dashboards";
 import type { Waiter } from "@databricks/sdk-experimental/dist/wait";
 import { createLogger } from "../../logging";
@@ -12,6 +12,7 @@ import type {
   GenieStreamEvent,
 } from "./types";
 
+const { Time, TimeUnits } = SDK;
 const logger = createLogger("connectors:genie");
 
 type CreateMessageWaiter = Waiter<GenieMessage, GenieMessage>;