Vercel AI SDK

npm install ai @ai-sdk/openai zod

// lib/ai.ts
import { createOpenAI } from "@ai-sdk/openai";

export const cokeapi = createOpenAI({
  baseURL: "https://api.cokeapi.com/v1",
  apiKey: process.env.COKEAPI_KEY!,
});

import { generateText } from "ai";
import { cokeapi } from "@/lib/ai";

const { text, usage } = await generateText({
  model: cokeapi("gpt-4o-mini"),
  prompt: "用一句话介绍 CokeAPI",
});

console.log(text);
console.log(`tokens: ${usage.totalTokens}`);

// app/api/chat/route.ts
import { streamText, convertToModelMessages, type UIMessage } from "ai";
import { cokeapi } from "@/lib/ai";

export const runtime = "nodejs";

export async function POST(req: Request) {
  const { messages }: { messages: UIMessage[] } = await req.json();

  const result = streamText({
    model: cokeapi("grok-4.20-fast"),
    messages: convertToModelMessages(messages),
  });

  return result.toUIMessageStreamResponse();
}

"use client";
import { useChat } from "@ai-sdk/react";

export default function Chat() {
  const { messages, input, handleInputChange, handleSubmit } = useChat({
    api: "/api/chat",
  });

  return (
    <div>
      {messages.map((m) => (
        <div key={m.id}>
          <b>{m.role}:</b> {m.parts.map((p) => p.type === "text" ? p.text : "").join("")}
        </div>
      ))}
      <form onSubmit={handleSubmit}>
        <input value={input} onChange={handleInputChange} />
        <button type="submit">Send</button>
      </form>
    </div>
  );
}

import { generateObject } from "ai";
import { z } from "zod";

const { object } = await generateObject({
  model: cokeapi("gpt-4o-mini"),
  schema: z.object({
    name: z.string(),
    tags: z.array(z.string()),
  }),
  prompt: "给 CokeAPI 起 3 个标签",
});

console.log(object); // { name: "CokeAPI", tags: [...] }

import { generateText, tool } from "ai";
import { z } from "zod";

const { text } = await generateText({
  model: cokeapi("grok-4.20-expert"),
  prompt: "北京天气?",
  tools: {
    getWeather: tool({
      description: "查询某城市天气",
      parameters: z.object({ city: z.string() }),
      execute: async ({ city }) => `${city} 晴天 25°C`,
    }),
  },
});

import { experimental_generateImage as generateImage } from "ai";

const { image } = await generateImage({
  model: cokeapi.image("gpt-image-2"),
  prompt: "一只在京都樱花树下的赛博狐狸",
  size: "1024x1024",
});

console.log(image.base64); // base64 PNG