使用 Vercel AI SDK 热切换 AI 模型

我想展示 Vercel AI SDK 在模型选择方面究竟有多灵活。

滚动式代码讲解

这里有一个 ask 函数，它接收一个提示词和一个类型为 LanguageModel.

可以使用 AI SDK 提供的任意模型调用这个函数。

例如，可以使用 Anthropic 模型调用：

也可以使用 OpenAI：

import { generateText, type LanguageModel } from "ai";

export const ask = async (
  prompt: string,
  model: LanguageModel,
) => {
  const { text } = await generateText({
    model,
    prompt,
  });

  return text;
};

这为应用的构建方式提供了极大灵活性。

这个 ask 函数与它使用的模型完全解耦。

这个 LanguageModel 类型由 AI SDK 暴露，让你可以使用依赖注入。换句话说，可以把任意模型注入系统。

在我看来，这是 AI SDK 的核心卖点之一。

下一个示例会介绍聊天历史，以及如何长期保留一段对话的历史。