The Grid AI

Access 9 The Grid AI models through Mastra's model router. Authentication is handled automatically using the THEGRID_API_KEY environment variable.

Learn more in the The Grid AI documentation.

.env

THEGRID_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: "the-grid-ai/agent-max"
});

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

note

Mastra uses the OpenAI-compatible /chat/completions endpoint. Some provider-specific features may not be available. Check the The Grid AI documentation for details.

Models

9 available models

Model	Context	Tools	Reasoning	Image	Audio	Video	Input $/1M	Output $/1M
the-grid-ai/agent-max	1.0M						—	—
the-grid-ai/agent-prime	197K						—	—
the-grid-ai/agent-standard	128K						—	—
the-grid-ai/code-max	1.0M						—	—
the-grid-ai/code-prime	197K						—	—
the-grid-ai/code-standard	128K						—	—
the-grid-ai/text-max	1.0M						—	—
the-grid-ai/text-prime	197K						—	—
the-grid-ai/text-standard	128K						—	—

Model availability, capabilities, context windows, and pricing are sourced from models.dev and may change.

Advanced configuration

Custom headers

src/mastra/agents/my-agent.ts

const agent = new Agent({
  id: "custom-agent",
  name: "custom-agent",
  model: {
    url: "https://api.thegrid.ai/v1",
    id: "the-grid-ai/agent-max",
    apiKey: process.env.THEGRID_API_KEY,
    headers: {
      "X-Custom-Header": "value"
    }
  }
});

Dynamic model selection

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
      ? "the-grid-ai/text-standard"
      : "the-grid-ai/agent-max";
  }
});