AI-ROUTER

透過 Mastra 的模型路由器存取 5 個 AI-ROUTER 模型。系統會自動使用 AI_ROUTER_API_KEY 環境變數進行驗證。

詳情請參閱AI-ROUTER 文件。

.env

AI_ROUTER_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: 'ai-router/gpt-5.4',
})

// 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 端點。部分 Provider 專屬功能可能無法使用，詳情請參閱AI-ROUTER 文件。

模型

5 available models

Model	Context	Tools	Reasoning	Image	Audio	Video	Input $/1M	Output $/1M
ai-router/gpt-5.4	1.1M						$3	$15
ai-router/gpt-5.5	1.1M						$5	$30
ai-router/gpt-5.6-luna	1.1M						$1	$6
ai-router/gpt-5.6-sol	1.1M						$5	$30
ai-router/gpt-5.6-terra	1.1M						$3	$15

進階設定

自訂標頭

src/mastra/agents/my-agent.ts

const agent = new Agent({
  id: 'custom-agent',
  name: 'custom-agent',
  model: {
    url: 'https://api.ai-router.dev/v1',
    id: 'ai-router/gpt-5.4',
    apiKey: process.env.AI_ROUTER_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 ? 'ai-router/gpt-5.6-terra' : 'ai-router/gpt-5.4'
  },
})