Databricks

Accédez à 30 modèles Databricks via le routeur de modèles de Mastra. L’authentification est gérée automatiquement à l’aide de la variable d’environnement DATABRICKS_TOKEN. Configurez également DATABRICKS_HOST.

Pour en savoir plus, consultez la documentation 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);
}

info

Mastra utilise le point de terminaison /chat/completions compatible avec OpenAI. Certaines fonctionnalités propres au Provider peuvent ne pas être disponibles. Consultez la documentation Databricks pour plus de détails.

Modèles

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

Configuration avancée

En-têtes personnalisés

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"
    }
  }
});

Sélection dynamique du modèle

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";
  }
});