Cerebras
Accédez à 3 modèles Cerebras via le routeur de modèles de Mastra. L’authentification est gérée automatiquement à l’aide de la variable d’environnement CEREBRAS_API_KEY.
Pour en savoir plus, consultez la documentation Cerebras.
.env
CEREBRAS_API_KEY=your-api-key
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: "cerebras/gemma-4-31b"
});
// 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);
}
ModèlesLien direct vers Modèles
| Model | Context | Tools | Reasoning | Image | Audio | Video | Input $/1M | Output $/1M |
|---|---|---|---|---|---|---|---|---|
cerebras/gemma-4-31b | 131K | $0.99 | $1 | |||||
cerebras/gpt-oss-120b | 131K | $0.35 | $0.75 | |||||
cerebras/zai-glm-4.7 | 131K | $2 | $3 |
Configuration avancéeLien direct vers Configuration avancée
En-têtes personnalisésLien direct vers En-têtes personnalisés
src/mastra/agents/my-agent.ts
const agent = new Agent({
id: "custom-agent",
name: "custom-agent",
model: {
id: "cerebras/gemma-4-31b",
apiKey: process.env.CEREBRAS_API_KEY,
headers: {
"X-Custom-Header": "value"
}
}
});
Sélection dynamique du modèleLien direct vers 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
? "cerebras/zai-glm-4.7"
: "cerebras/gemma-4-31b";
}
});
Installation directe du ProviderLien direct vers Installation directe du Provider
Ce Provider peut aussi être installé directement comme package autonome, utilisable à la place de la chaîne du routeur de modèles Mastra. Consultez la documentation du package pour plus de détails.
- npm
- pnpm
- Yarn
- Bun
npm install @ai-sdk/cerebras
pnpm add @ai-sdk/cerebras
yarn add @ai-sdk/cerebras
bun add @ai-sdk/cerebras
Pour une documentation détaillée propre au Provider, consultez la documentation du Provider AI SDK Cerebras provider.