The Grid AI

透過 Mastra 的模型路由器存取 9 個 The Grid AI 模型。系統會自動使用 THEGRID_API_KEY 環境變數完成身分驗證。

如需進一步了解，請參閱The Grid AI 文件。

.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)
}

資訊

Mastra 使用與 OpenAI 相容的 /chat/completions endpoint。部分 Provider 特有功能可能無法使用，詳情請參閱The Grid AI 文件。

模型

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						—	—

進階設定

自訂標頭

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

動態選擇模型

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