> Discover all available pages from the documentation index: https://mastra.zisheng.pro/zh-TW/llms.txt

# ![Databricks 標誌](https://models.dev/logos/databricks.svg)Databricks

透過 Mastra 的模型路由器存取 30 個 Databricks 模型。系統會自動使用 `DATABRICKS_TOKEN` 環境變數完成身分驗證。 Configure `DATABRICKS_HOST` as well.

如需進一步了解，請參閱[Databricks 文件](https://docs.databricks.com/aws/en/machine-learning/foundation-models/)。

```bash
DATABRICKS_HOST=your-value
DATABRICKS_TOKEN=your-api-token
```

```typescript
import { Agent } from '@mastra/core/agent'

const agent = new Agent({
  id: 'my-agent',
  name: 'My Agent',
  instructions: 'You are a helpful assistant',
  model: 'databricks/databricks-claude-haiku-4-5',
})

// Generate a response
const response = await agent.generate('Hello!')

// Stream a response
const stream = await agent.stream('Tell me a story')
for await (const chunk of stream) {
  console.log(chunk)
}
```

> **資訊:** Mastra 使用與 OpenAI 相容的 `/chat/completions` endpoint。部分 Provider 特有功能可能無法使用，詳情請參閱[Databricks 文件](https://docs.databricks.com/aws/en/machine-learning/foundation-models/)。

## 模型

| Model                                         | Context | Tools | Reasoning | Image | Audio | Video | Input $/1M | Output $/1M |
| --------------------------------------------- | ------- | ----- | --------- | ----- | ----- | ----- | ---------- | ----------- |
| `databricks/databricks-claude-haiku-4-5`      | 200K    |       |           |       |       |       | $1         | $5          |
| `databricks/databricks-claude-opus-4-1`       | 200K    |       |           |       |       |       | $15        | $75         |
| `databricks/databricks-claude-opus-4-5`       | 200K    |       |           |       |       |       | $5         | $25         |
| `databricks/databricks-claude-opus-4-6`       | 1.0M    |       |           |       |       |       | $5         | $25         |
| `databricks/databricks-claude-opus-4-7`       | 1.0M    |       |           |       |       |       | $5         | $25         |
| `databricks/databricks-claude-sonnet-4`       | 200K    |       |           |       |       |       | $3         | $15         |
| `databricks/databricks-claude-sonnet-4-5`     | 200K    |       |           |       |       |       | $3         | $15         |
| `databricks/databricks-claude-sonnet-4-6`     | 1.0M    |       |           |       |       |       | $3         | $15         |
| `databricks/databricks-gemini-2-5-flash`      | 1.0M    |       |           |       |       |       | $0.30      | $3          |
| `databricks/databricks-gemini-2-5-pro`        | 1.0M    |       |           |       |       |       | $1         | $10         |
| `databricks/databricks-gemini-3-1-flash-lite` | 1.0M    |       |           |       |       |       | $0.25      | $2          |
| `databricks/databricks-gemini-3-1-pro`        | 1.0M    |       |           |       |       |       | $2         | $12         |
| `databricks/databricks-gemini-3-flash`        | 1.0M    |       |           |       |       |       | $0.50      | $3          |
| `databricks/databricks-gemini-3-pro`          | 1.0M    |       |           |       |       |       | $2         | $12         |
| `databricks/databricks-glm-5-2`               | 1.0M    |       |           |       |       |       | $1         | $4          |
| `databricks/databricks-gpt-5`                 | 400K    |       |           |       |       |       | $1         | $10         |
| `databricks/databricks-gpt-5-1`               | 400K    |       |           |       |       |       | $1         | $10         |
| `databricks/databricks-gpt-5-2`               | 400K    |       |           |       |       |       | $2         | $14         |
| `databricks/databricks-gpt-5-4`               | 1.1M    |       |           |       |       |       | $3         | $15         |
| `databricks/databricks-gpt-5-4-mini`          | 400K    |       |           |       |       |       | $0.75      | $5          |
| `databricks/databricks-gpt-5-4-nano`          | 400K    |       |           |       |       |       | $0.20      | $1          |
| `databricks/databricks-gpt-5-5`               | 1.1M    |       |           |       |       |       | $5         | $30         |
| `databricks/databricks-gpt-5-6-luna`          | 400K    |       |           |       |       |       | $1         | $6          |
| `databricks/databricks-gpt-5-6-sol`           | 1.1M    |       |           |       |       |       | $5         | $30         |
| `databricks/databricks-gpt-5-6-terra`         | 1.1M    |       |           |       |       |       | $3         | $15         |
| `databricks/databricks-gpt-5-mini`            | 400K    |       |           |       |       |       | $0.25      | $2          |
| `databricks/databricks-gpt-5-nano`            | 400K    |       |           |       |       |       | $0.05      | $0.40       |
| `databricks/databricks-gpt-oss-120b`          | 131K    |       |           |       |       |       | $0.07      | $0.28       |
| `databricks/databricks-gpt-oss-20b`           | 131K    |       |           |       |       |       | $0.05      | $0.20       |
| `databricks/databricks-kimi-k2-7-code`        | 262K    |       |           |       |       |       | $0.95      | $4          |

## 進階設定

### 自訂標頭

```typescript
const agent = new Agent({
  id: 'custom-agent',
  name: 'custom-agent',
  model: {
    url: 'https://${DATABRICKS_HOST}/ai-gateway/mlflow/v1',
    id: 'databricks/databricks-claude-haiku-4-5',
    apiKey: process.env.DATABRICKS_TOKEN,
    headers: {
      'X-Custom-Header': 'value',
    },
  },
})
```

### 動態選擇模型

```typescript
const agent = new Agent({
  id: 'dynamic-agent',
  name: 'Dynamic Agent',
  model: ({ requestContext }) => {
    const useAdvanced = requestContext.task === 'complex'
    return useAdvanced
      ? 'databricks/databricks-kimi-k2-7-code'
      : 'databricks/databricks-claude-haiku-4-5'
  },
})
```