Sarvam AI

Accédez à 2 modèles Sarvam AI via le routeur de modèles de Mastra. L’authentification est gérée automatiquement à l’aide de la variable d’environnement SARVAM_API_KEY.

Pour en savoir plus, consultez la documentation Sarvam AI.

.env

SARVAM_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: "sarvam/sarvam-105b"
});

// 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 Sarvam AI pour plus de détails.

Modèles

2 available models

Model	Context	Tools	Reasoning	Image	Audio	Video	Input $/1M	Output $/1M
sarvam/sarvam-105b	131K						—	—
sarvam/sarvam-30b	66K						—	—

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://api.sarvam.ai/v1",
    id: "sarvam/sarvam-105b",
    apiKey: process.env.SARVAM_API_KEY,
    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
      ? "sarvam/sarvam-30b"
      : "sarvam/sarvam-105b";
  }
});