跨模型的 API 调用与集成：使用 Spring AI 构建可移植的 API 代码，支持多个 AI 模型（如 OpenAI、Anthropic 等）的调用

在现代 AI 系统中，集成多个 AI 模型（如 OpenAI、Anthropic 等）往往是为了充分利用不同模型在特定任务中的优势。然而，不同模型的 API 调用和配置方式各不相同，为了避免重复造轮子并提升系统的灵活性，我们可以设计一个统一的接口来调用不同的 AI 模型。

本文将介绍如何使用 Spring AI 框架来实现这一目标。Spring AI 提供了方便的 API 客户端和集成工具，使得跨多个 AI 模型的调用变得更加简单。通过构建一个统一的接口，我们可以在 Spring Boot 项目中灵活切换并调用不同的 AI 模型。

1. Spring AI 简介

Spring AI 是 Spring 生态系统中的一部分，旨在提供一种简单、可扩展的方式来集成人工智能模型。Spring AI 使得开发者可以使用现有的 Spring 体系结构方便地访问多种 AI 模型和服务，例如 OpenAI、Anthropic、HuggingFace 等。

通过 Spring AI，开发者可以：

• 简化 AI 模型的集成：通过 Spring 的依赖注入和配置管理功能，轻松集成第三方 AI 服务。
• 跨模型调用支持：可以统一管理多个 AI 模型，并根据需要动态切换和调用。
• 自动化与优化：Spring AI 提供了模型缓存、负载均衡、API 调用优化等功能。

2. 设计可移植的 API 调用层

与前文中提到的纯 Python 实现类似，我们首先需要定义一个通用的接口类，来统一不同模型的 API 调用。这可以通过 Spring 的 @Component 注解和依赖注入来实现。首先，我们定义一个抽象的 AI 模型接口。

2.1 统一的 API 接口设计

public interface AIModel {
   
    void initialize(String apiKey, String endpoint);
    String generateText(String prompt);
}

2.2 实现具体模型的接口

然后，我们为不同的模型实现这个接口。以下是 OpenAI 和 Anthropic 模型的实现。

OpenAIModel 实现

import org.springframework.stereotype.Component;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;

@Component
public class OpenAIModel implements AIModel {
   

    private String apiKey;
    private String endpoint = "https://api.openai.com/v1/completions";
    private RestTemplate restTemplate;

    public OpenAIModel(RestTemplate restTemplate) {
   
        this.restTemplate = restTemplate;
    }

    @Override
    public void initialize(String apiKey, String endpoint) {
   
        this.apiKey = apiKey;
        if (endpoint !=