feat(opencode): Add Databricks provider support

packages/opencode/src/auth/index.ts

@@ -20,6 +20,7 @@ export namespace Auth {
     .object({
       type: z.literal("api"),
       key: z.string(),
+      host: z.string().optional(), // For providers like Databricks that need a host URL
     })
     .meta({ ref: "ApiAuth" })
 

packages/opencode/src/cli/cmd/auth.ts

@@ -276,6 +276,7 @@ export const AuthLoginCommand = cmd({
           google: 4,
           openrouter: 5,
           vercel: 6,
+          databricks: 7,
         }
         let provider = await prompts.autocomplete({
           message: "Select provider",
@@ -344,6 +345,19 @@ export const AuthLoginCommand = cmd({
           )
         }
 
+        if (provider === "databricks") {
+          prompts.log.info(
+            "Databricks Foundation Model APIs authentication:\n" +
+              "  Enter your workspace URL and Personal Access Token\n" +
+              "  Create token at: Workspace > Settings > Developer > Access tokens\n\n" +
+              "Authentication options (in priority order):\n" +
+              "  1. PAT: Enter your Personal Access Token below, or set DATABRICKS_TOKEN\n" +
+              "  2. OAuth M2M: Set DATABRICKS_CLIENT_ID + DATABRICKS_CLIENT_SECRET\n" +
+              "  3. Azure AD Service Principal: Set ARM_CLIENT_ID + ARM_CLIENT_SECRET + ARM_TENANT_ID\n" +
+              "  4. Azure CLI (auto): For Azure Databricks, will use 'az account get-access-token' if logged in",
+          )
+        }
+
         if (provider === "opencode") {
           prompts.log.info("Create an api key at https://opencode.ai/auth")
         }
@@ -358,6 +372,22 @@ export const AuthLoginCommand = cmd({
           )
         }
 
+        // For Databricks, prompt for host first
+        let host: string | undefined
+        if (provider === "databricks") {
+          const hostInput = await prompts.text({
+            message: "Enter your Databricks workspace URL",
+            placeholder: "https://your-workspace.cloud.databricks.com",
+            validate: (x) => {
+              if (!x || x.length === 0) return "Required"
+              if (!x.startsWith("https://")) return "Must start with https://"
+              return undefined
+            },
+          })
+          if (prompts.isCancel(hostInput)) throw new UI.CancelledError()
+          host = hostInput.replace(/\/$/, "") // Remove trailing slash
+        }
+
         const key = await prompts.password({
           message: "Enter your API key",
           validate: (x) => (x && x.length > 0 ? undefined : "Required"),
@@ -366,6 +396,7 @@ export const AuthLoginCommand = cmd({
         await Auth.set(provider, {
           type: "api",
           key,
+          host,
         })
 
         prompts.outro("Done")

packages/opencode/src/cli/cmd/tui/component/dialog-provider.tsx

@@ -1,4 +1,4 @@
-import { createMemo, createSignal, onMount, Show } from "solid-js"
+import { createMemo, createSignal, onMount, Show, createEffect } from "solid-js"
 import { useSync } from "@tui/context/sync"
 import { map, pipe, sortBy } from "remeda"
 import { DialogSelect } from "@tui/ui/dialog-select"
@@ -97,6 +97,10 @@ export function createDialogProviderOptions() {
               }
             }
             if (method.type === "api") {
+              // Databricks requires both host and API key
+              if (provider.id === "databricks") {
+                return dialog.replace(() => <DatabricksApiMethod providerID={provider.id} title={method.label} />)
+              }
               return dialog.replace(() => <ApiMethod providerID={provider.id} title={method.label} />)
             }
           },
@@ -212,6 +216,119 @@ function CodeMethod(props: CodeMethodProps) {
   )
 }
 
+interface DatabricksApiMethodProps {
+  providerID: string
+  title: string
+}
+function DatabricksApiMethod(props: DatabricksApiMethodProps) {
+  const { theme } = useTheme()
+  const dialog = useDialog()
+  const sdk = useSDK()
+  const sync = useSync()
+  // Get host from environment variable
+  const envHost = typeof process !== "undefined" ? process.env["DATABRICKS_HOST"] : undefined
+
+  // Check if we have a valid token in the CLI cache for this host
+  onMount(async () => {
+    if (!envHost) return
+
+    const normalizedHost = envHost.replace(/\/$/, "")
+    const homedir = typeof process !== "undefined" ? (process.env["HOME"] ?? process.env["USERPROFILE"]) : undefined
+    if (!homedir) return
+
+    try {
+      const tokenCachePath = `${homedir}/.databricks/token-cache.json`
+      const file = Bun.file(tokenCachePath)
+      if (!(await file.exists())) return
+
+      const cache = (await file.json()) as {
+        tokens: Record<string, { access_token: string; expiry: string; refresh_token?: string }>
+      }
+
+      const tokenEntry = cache.tokens[normalizedHost]
+      if (!tokenEntry) return
+
+      // Check if token is valid or can be refreshed
+      const expiry = new Date(tokenEntry.expiry)
+      const hasValidToken = expiry.getTime() - 5 * 60 * 1000 > Date.now()
+      const canRefresh = Boolean(tokenEntry.refresh_token)
+
+      if (hasValidToken || canRefresh) {
+        // We have CLI auth available, skip prompts and go straight to model selection
+        // Dispose and bootstrap to pick up the CLI token
+        await sdk.client.instance.dispose()
+        await sync.bootstrap()
+        dialog.replace(() => <DialogModel providerID={props.providerID} />)
+      }
+    } catch {
+      // Token cache not available or invalid, continue with normal flow
+    }
+  })
+
+  return (
+    <DialogPrompt
+      title="Databricks Host URL"
+      placeholder="https://your-workspace.cloud.databricks.com"
+      value={envHost ? envHost.replace(/\/$/, "") : undefined}
+      description={() => (
+        <box gap={1}>
+          <text fg={theme.textMuted}>Enter your Databricks workspace URL</text>
+          <text fg={theme.textMuted}>Examples:</text>
+          <text fg={theme.textMuted}>  • https://dbc-xxx.cloud.databricks.com (AWS/GCP)</text>
+          <text fg={theme.textMuted}>  • https://adb-xxx.azuredatabricks.net (Azure)</text>
+        </box>
+      )}
+      onConfirm={(value) => {
+        if (!value) return
+        // Remove trailing slash if present
+        const cleanHost = value.replace(/\/$/, "")
+        dialog.replace(() => (
+          <DatabricksApiKeyMethod providerID={props.providerID} title={props.title} host={cleanHost} />
+        ))
+      }}
+    />
+  )
+}
+
+interface DatabricksApiKeyMethodProps {
+  providerID: string
+  title: string
+  host: string
+}
+function DatabricksApiKeyMethod(props: DatabricksApiKeyMethodProps) {
+  const dialog = useDialog()
+  const sdk = useSDK()
+  const sync = useSync()
+  const { theme } = useTheme()
+
+  return (
+    <DialogPrompt
+      title={props.title}
+      placeholder="API key (Personal Access Token)"
+      description={
+        <box gap={1}>
+          <text fg={theme.textMuted}>Enter your Databricks Personal Access Token</text>
+          <text fg={theme.textMuted}>Create at: Workspace → Settings → Developer → Access tokens</text>
+        </box>
+      }
+      onConfirm={async (value) => {
+        if (!value) return
+        sdk.client.auth.set({
+          providerID: props.providerID,
+          auth: {
+            type: "api",
+            key: value,
+            host: props.host,
+          },
+        })
+        await sdk.client.instance.dispose()
+        await sync.bootstrap()
+        dialog.replace(() => <DialogModel providerID={props.providerID} />)
+      }}
+    />
+  )
+}
+
 interface ApiMethodProps {
   providerID: string
   title: string

packages/opencode/src/cli/cmd/tui/ui/dialog-prompt.tsx

@@ -20,8 +20,12 @@ export function DialogPrompt(props: DialogPromptProps) {
 
   useKeyboard((evt) => {
     if (evt.name === "return") {
+      evt.preventDefault()
       props.onConfirm?.(textarea.plainText)
     }
+    if (evt.name === "escape") {
+      props.onCancel?.()
+    }
   })
 
   onMount(() => {
@@ -47,6 +51,12 @@ export function DialogPrompt(props: DialogPromptProps) {
           onSubmit={() => {
             props.onConfirm?.(textarea.plainText)
           }}
+          onKeyDown={(e) => {
+            if (e.name === "return") {
+              e.preventDefault()
+              props.onConfirm?.(textarea.plainText)
+            }
+          }}
           height={3}
           keyBindings={[{ name: "return", action: "submit" }]}
           ref={(val: TextareaRenderable) => (textarea = val)}

packages/opencode/src/provider/auth.ts

@@ -121,11 +121,13 @@ export namespace ProviderAuth {
     z.object({
       providerID: z.string(),
       key: z.string(),
+      host: z.string().optional(),
     }),
     async (input) => {
       await Auth.set(input.providerID, {
         type: "api",
         key: input.key,
+        host: input.host,
       })
     },
   )