Databricks

通过 Mastra 的模型路由器访问 30 个 Databricks 模型。系统会自动使用 DATABRICKS_TOKEN 环境变量完成身份验证。 Configure DATABRICKS_HOST as well.

更多信息请参阅Databricks 文档。

.env

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

src/mastra/agents/my-agent.ts

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 文档。

模型

30 available 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

高级配置

自定义请求头

src/mastra/agents/my-agent.ts

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',
    },
  },
})

动态选择模型

src/mastra/agents/my-agent.ts

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'
  },
})