feat(appkit): migrate analytics to the modular @databricks/sdk-*

.oxlintrc.json

@@ -26,11 +26,8 @@
       {
         "patterns": [
           {
-            "group": [
-              "@databricks/sdk-experimental",
-              "@databricks/sdk-experimental/**"
-            ],
-            "message": "Import the Databricks SDK only through the wrapper in packages/shared/src/workspace-client. Add a re-export there if you need a new symbol."
+            "group": ["@databricks/sdk-*", "@databricks/sdk-*/**"],
+            "message": "Import the Databricks SDK only through the wrapper in packages/shared/src/workspace-client (legacy.ts for @databricks/sdk-experimental, modular.ts for the modular @databricks/sdk-* packages). Add a re-export there if you need a new symbol."
           }
         ]
       }

knip.json

@@ -10,7 +10,12 @@
     "packages/appkit": {
       "ignoreDependencies": [
         "vitest",
+        "@databricks/sdk-auth",
+        "@databricks/sdk-core",
         "@databricks/sdk-experimental",
+        "@databricks/sdk-options",
+        "@databricks/sdk-statementexecution",
+        "@databricks/sdk-warehouses",
         "@mlflow/core",
         "autoevals"
       ]

packages/appkit/package.json

@@ -71,7 +71,12 @@
   "dependencies": {
     "@ast-grep/napi": "0.37.0",
     "@databricks/lakebase": "workspace:*",
+    "@databricks/sdk-auth": "0.51.0",
+    "@databricks/sdk-core": "0.51.0",
     "@databricks/sdk-experimental": "0.17.0",
+    "@databricks/sdk-options": "0.51.0",
+    "@databricks/sdk-statementexecution": "0.52.0",
+    "@databricks/sdk-warehouses": "0.53.0",
     "@opentelemetry/api": "1.9.0",
     "@opentelemetry/api-logs": "0.219.0",
     "@opentelemetry/auto-instrumentations-node": "0.77.0",

packages/appkit/src/cache/index.ts

@@ -6,6 +6,7 @@ import { createLakebasePool } from "../connectors/lakebase";
 import { getClientOptions } from "../context/client-options";
 import { getCurrentPrincipalKey } from "../context/execution-context";
 import { AppKitError, ExecutionError, InitializationError } from "../errors";
+import { httpStatusOf } from "../errors/http-status";
 import { createLogger } from "../logging/logger";
 import type { Counter, TelemetryProvider } from "../telemetry";
 import { SpanStatusCode, TelemetryManager } from "../telemetry";
@@ -339,7 +340,13 @@ export class CacheManager {
               if (sharedController.signal.aborted) {
                 throw error;
               }
-              if (error instanceof AppKitError || error instanceof ApiError) {
+              // Keep any error that carries an HTTP status: the modular SDK's
+              // ApiError is a different class from the legacy one.
+              if (
+                error instanceof AppKitError ||
+                error instanceof ApiError ||
+                httpStatusOf(error) !== undefined
+              ) {
                 throw error;
               }
               throw ExecutionError.statementFailed(

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

@@ -997,6 +997,25 @@ describe("CacheManager", () => {
       );
     });
 
+    test("should re-throw a modular SDK ApiError without wrapping (keeps its status)", async () => {
+      const { ModularApiError } = await import("shared/workspace-client");
+      const cache = await CacheManager.getInstance({
+        storage: createMockStorage(),
+      });
+      const body = new TextEncoder().encode(
+        JSON.stringify({
+          error_code: "INVALID_PARAMETER_VALUE",
+          message: "Run 1 does not exist.",
+        }),
+      );
+      const apiError = ModularApiError.fromHttpError(400, undefined, body);
+      const fn = vi.fn().mockRejectedValue(apiError);
+
+      await expect(cache.getOrExecute(["key"], fn, "user1")).rejects.toBe(
+        apiError,
+      );
+    });
+
     test("should allow retry after error", async () => {
       const cache = await CacheManager.getInstance({
         storage: createMockStorage(),

packages/appkit/src/connectors/sql-warehouse/arrow-schema.ts

@@ -54,12 +54,12 @@ export function parseDatabricksType(typeText: string): DataType {
 export function buildEmptyArrowIPCBase64(
   columns: Array<{
     name?: string;
-    type_text?: string;
-    type_name?: string;
+    typeText?: string;
+    typeName?: string;
   }>,
 ): string {
   const fields = columns.map((col, index) => {
-    const typeText = col.type_text ?? col.type_name ?? "STRING";
+    const typeText = col.typeText ?? col.typeName ?? "STRING";
     let dataType: DataType;
     try {
       dataType = parseDatabricksType(typeText);