fix(databricks): address PR review feedback

- Move Databricks scripts from packages/core/script/ to providers/databricks/scripts/
  with their own package.json; remove @databricks/sdk-experimental from packages/core
- Add Zod runtime validation to fetchFilteredGatewayRoutes (replaces unsafe cast)
- Replace isOpenAIcompatible boolean with openAICompatiblePackages Set in schema.ts
- Clarify provider.toml env comment with PAT vs OAuth M2M auth modes
- Shorten dense root README.md entry to a one-liner linking to provider README
- Add bun.lock to .gitignore
- Add databricks-claude-opus-4-7 model TOML

Author: prasadkona

README.md

@@ -186,7 +186,7 @@ See existing providers in the `providers/` directory for reference:
 - `providers/anthropic/` - Anthropic Claude models
 - `providers/openai/` - OpenAI GPT models
 - `providers/google/` - Google Gemini models
-- `providers/databricks/` - Databricks Foundation Model APIs on **AI Gateway**: default **`mlflow/v1`** for chat and embeddings via **`@databricks/ai-sdk-provider`**; per-model **`[provider]`** for **Claude** (Anthropic Messages), **Gemini** (native API), and **OpenAI Responses**. See [providers/databricks/README.md](providers/databricks/README.md) for discovery, authentication, and validation scripts.
+- `providers/databricks/` - Databricks Foundation Model APIs via AI Gateway. See [providers/databricks/README.md](providers/databricks/README.md).
 
 ### Working on frontend
 

package.json

@@ -3,7 +3,8 @@
   "private": true,
   "workspaces": {
     "packages": [
-      "packages/*"
+      "packages/*",
+      "providers/databricks"
     ],
     "catalog": {
       "typescript": "5.8.2",
@@ -20,9 +21,9 @@
     "venice:generate": "bun ./packages/core/script/generate-venice.ts",
     "vercel:generate": "bun ./packages/core/script/generate-vercel.ts",
     "wandb:generate": "bun ./packages/core/script/generate-wandb.ts",
-    "databricks:list-gateway": "bun ./packages/core/script/list-databricks-ai-gateway.ts",
-    "databricks:test-inference": "bun ./packages/core/script/test-databricks.ts",
-    "databricks:probe-capabilities": "bun ./packages/core/script/probe-databricks-capabilities.ts"
+    "databricks:list-gateway": "bun ./providers/databricks/scripts/list-databricks-ai-gateway.ts",
+    "databricks:test-inference": "bun ./providers/databricks/scripts/test-databricks.ts",
+    "databricks:probe-capabilities": "bun ./providers/databricks/scripts/probe-databricks-capabilities.ts"
   },
   "dependencies": {
     "@cloudflare/workers-types": "^4.20250801.0",

packages/core/package.json

@@ -4,7 +4,6 @@
   "$schema": "https://json.schemastore.org/package.json",
   "type": "module",
   "dependencies": {
-    "@databricks/sdk-experimental": "^0.16.0",
     "zod": "catalog:"
   },
   "main": "./src/index.ts",

packages/core/src/schema.ts

@@ -147,9 +147,11 @@ export const Provider = z
   .refine(
     (data) => {
       const isOpenAI = data.npm === "@ai-sdk/openai";
-      const isOpenAIcompatible =
-        data.npm === "@ai-sdk/openai-compatible" ||
-        data.npm === "@databricks/ai-sdk-provider";
+      const openAICompatiblePackages = new Set([
+        "@ai-sdk/openai-compatible",
+        "@databricks/ai-sdk-provider",
+      ]);
+      const isOpenAIcompatible = openAICompatiblePackages.has(data.npm);
       const isOpenrouter = data.npm === "@openrouter/ai-sdk-provider";
       const isAnthropic = data.npm === "@ai-sdk/anthropic";
       const hasApi = data.api !== undefined;
@@ -173,7 +175,7 @@ export const Provider = z
     },
     {
       message:
-        "'api' is required for @ai-sdk/openai-compatible, @databricks/ai-sdk-provider, and openrouter; optional for anthropic and openai; forbidden otherwise",
+        "'api' is required for openai-compatible packages and openrouter; optional for anthropic and openai; forbidden otherwise",
       path: ["api"],
     },
   );

providers/databricks/README.md

@@ -203,7 +203,7 @@ cd packages/web && bun run build
 | `bun run databricks:test-inference -- --profile PROFILE` | Exercises each catalog route against AI Gateway (`--only`, `--delay-ms`, `--json` supported). |
 
 
-Scripts: `[list-databricks-ai-gateway.ts](../../packages/core/script/list-databricks-ai-gateway.ts)`, `[test-databricks.ts](../../packages/core/script/test-databricks.ts)`.
+Scripts: `[list-databricks-ai-gateway.ts](./scripts/list-databricks-ai-gateway.ts)`, `[test-databricks.ts](./scripts/test-databricks.ts)`.
 
 ---
 

providers/databricks/models/databricks-claude-opus-4-7.toml

@@ -0,0 +1,21 @@
+name = "Databricks Claude Opus 4.7"
+family = "claude-opus"
+release_date = "2025-06-01"
+last_updated = "2026-04-20"
+attachment = true
+reasoning = true
+tool_call = true
+temperature = true
+open_weights = false
+
+[limit]
+context = 200000
+output = 64000
+
+[modalities]
+input = ["text", "image", "pdf"]
+output = ["text"]
+
+[provider]
+npm = "@ai-sdk/anthropic"
+api = "https://<workspace-numeric-id>.ai-gateway.cloud.databricks.com/anthropic"

providers/databricks/package.json

@@ -0,0 +1,12 @@
+{
+  "name": "models.dev-provider-databricks",
+  "private": true,
+  "type": "module",
+  "devDependencies": {
+    "@databricks/sdk-experimental": "^0.16.0",
+    "@tsconfig/bun": "catalog:",
+    "@types/bun": "catalog:",
+    "@types/node": "catalog:",
+    "zod": "catalog:"
+  }
+}

providers/databricks/provider.toml

@@ -1,12 +1,13 @@
 name = "Databricks"
-# Environment hints. Use the subset that matches each auth mode. Integrators substitute
-# <workspace-numeric-id> in `api` (or set baseURL after discovery—see README).
+# PAT auth:   DATABRICKS_HOST + DATABRICKS_TOKEN
+# OAuth M2M:  DATABRICKS_HOST + DATABRICKS_CLIENT_ID + DATABRICKS_CLIENT_SECRET
+# DATABRICKS_WORKSPACE_ID: only needed when constructing the AI Gateway URL manually (see README)
 env = [
   "DATABRICKS_HOST",
-  "DATABRICKS_WORKSPACE_ID",
   "DATABRICKS_TOKEN",
   "DATABRICKS_CLIENT_ID",
   "DATABRICKS_CLIENT_SECRET",
+  "DATABRICKS_WORKSPACE_ID",
 ]
 npm = "@databricks/ai-sdk-provider"
 api = "https://<workspace-numeric-id>.ai-gateway.cloud.databricks.com/mlflow/v1"

…e/script/databricks-ai-gateway-shared.ts …/scripts/databricks-ai-gateway-shared.tspackages/core/script/databricks-ai-gateway-shared.ts renamed to providers/databricks/scripts/databricks-ai-gateway-shared.ts packages/core/script/databricks-ai-gateway-shared.ts renamed to providers/databricks/scripts/databricks-ai-gateway-shared.ts

@@ -3,21 +3,30 @@
  */
 
 import { WorkspaceClient } from "@databricks/sdk-experimental";
+import { z } from "zod";
 
-export interface Destination {
-  name?: string;
-  type?: string;
-}
+const DestinationSchema = z.object({
+  name: z.string().optional(),
+  type: z.string().optional(),
+});
 
-export interface Endpoint {
-  name?: string;
-  ai_gateway_url?: string;
-  config?: { destinations?: Destination[] };
-}
+const EndpointSchema = z.object({
+  name: z.string().optional(),
+  ai_gateway_url: z.string().optional(),
+  config: z
+    .object({
+      destinations: z.array(DestinationSchema).optional(),
+    })
+    .optional(),
+});
 
-export interface EndpointsResponse {
-  endpoints?: Endpoint[];
-}
+const EndpointsResponseSchema = z.object({
+  endpoints: z.array(EndpointSchema).optional(),
+});
+
+export type Destination = z.infer<typeof DestinationSchema>;
+export type Endpoint = z.infer<typeof EndpointSchema>;
+export type EndpointsResponse = z.infer<typeof EndpointsResponseSchema>;
 
 export interface FilteredGatewayRoute {
   gateway_name: string;
@@ -57,17 +66,24 @@ export function filterEndpoints(endpoints: Endpoint[]): FilteredGatewayRoute[] {
 export async function fetchFilteredGatewayRoutes(
   client: WorkspaceClient,
 ): Promise<FilteredGatewayRoute[]> {
-  const raw = (await client.apiClient.request(
+  const raw = await client.apiClient.request(
     {
       path: "/api/ai-gateway/v2/endpoints",
       method: "GET",
       headers: new Headers(),
       raw: false,
     },
     undefined,
-  )) as EndpointsResponse;
+  );
+
+  const parsed = EndpointsResponseSchema.safeParse(raw);
+  if (!parsed.success) {
+    throw new Error(
+      `Unexpected response shape from /api/ai-gateway/v2/endpoints: ${parsed.error.message}`,
+    );
+  }
 
-  return filterEndpoints(raw.endpoints ?? []);
+  return filterEndpoints(parsed.data.endpoints ?? []);
 }
 
 /** OpenAI-compatible base URL for chat/embeddings (no trailing slash). */