Migrate chatbots to Vercel AI Gateway

.env.example

@@ -1,5 +1,5 @@
-OPENROUTER_API_KEY=
-OPENROUTER_MODEL=
+AI_GATEWAY_API_KEY=
+AI_GATEWAY_MODEL=
 
 UPSTASH_REDIS_REST_URL=
 UPSTASH_REDIS_REST_TOKEN=

app/api/demo/chat/route.ts

@@ -1,5 +1,6 @@
 import {
   convertToModelMessages,
+  gateway,
   jsonSchema,
   stepCountIs,
   streamText,
@@ -16,7 +17,6 @@ import {
   chatRateLimit,
   getClientIp,
 } from "@/lib/demo/rate-limit";
-import { openrouter } from "@/lib/ai/openrouter";
 
 const SYSTEM_PROMPT = `You are an AI assistant demonstrating the Agent Auth Protocol — a standard for authorizing AI agents to act on behalf of users.
 
@@ -375,9 +375,7 @@ export async function POST(req: Request) {
   }
 
   const result = streamText({
-    model: openrouter.chat(
-      process.env.OPENROUTER_MODEL ?? "anthropic/claude-opus-4.6",
-    ),
+    model: gateway(process.env.AI_GATEWAY_MODEL ?? "anthropic/claude-opus-4.6"),
     system: systemPrompt,
     messages: await convertToModelMessages(messages),
     tools: buildTools(session),

lib/ai/server.ts

@@ -1,5 +1,6 @@
 import {
 	convertToModelMessages,
+	gateway,
 	stepCountIs,
 	streamText,
 	wrapLanguageModel,
@@ -15,7 +16,6 @@ import {
 	getSpecContent,
 	getSpecVersions,
 } from "@/lib/spec";
-import { openrouter } from "@/lib/ai/openrouter";
 
 // Build unified Orama search index: docs + spec sections
 const searchServer = initAdvancedSearch({
@@ -157,9 +157,7 @@ export async function POST(req: Request) {
 	}
 
 	const model = wrapLanguageModel({
-		model: openrouter.chat(
-			process.env.OPENROUTER_MODEL ?? "moonshotai/kimi-k2.5",
-		),
+		model: gateway(process.env.AI_GATEWAY_MODEL ?? "moonshotai/kimi-k2.5"),
 		middleware: middlewares,
 	});
 

package.json

@@ -11,7 +11,6 @@
   "dependencies": {
     "@ai-sdk/react": "^3.0.118",
     "@auth/agent": "latest",
-    "@openrouter/ai-sdk-provider": "^2.3.3",
     "@radix-ui/react-avatar": "^1.1.10",
     "@radix-ui/react-dialog": "^1.1.14",
     "@radix-ui/react-dropdown-menu": "^2.1.15",