Springboot 整合 Spring AI 实现项目接入ChatGPT

在当今科技快速发展的时代，人工智能（AI）已经成为了许多应用中的核心部分。尤其是自然语言处理领域的模型如ChatGPT，正在帮助开发者提升用户体验和交互能力。本文将带你深入了解如何在Spring Boot项目中整合Spring AI，以实现与ChatGPT的接入。

1. 环境准备

首先，你需要确保已经安装了以下环境： - JDK 8或以上版本 - Maven - Spring Boot

2. 创建Spring Boot项目

使用Spring Initializr（https://start.spring.io/）创建一个新的Spring Boot项目。选择以下依赖： - Spring Web - Spring Boot DevTools - Spring AI（如果存在的话）

下载项目后，将其导入你喜欢的IDE中。

3. 配置依赖

在你的pom.xml中添加相关依赖。由于Spring AI可能会与ChatGPT的API进行交互，这里我们假定需要使用HTTP客户端来发送请求。

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-json</artifactId>
    </dependency>
    <dependency>
        <groupId>org.springframework.cloud</groupId>
        <artifactId>spring-cloud-starter-openfeign</artifactId>
    </dependency>
</dependencies>

4. 配置API密钥

要接入ChatGPT，你需要获取OpenAI的API密钥。将其添加到application.properties或application.yml文件中：

openai.api.key=你的_api_key

5. 创建ChatGPT服务

接下来，我们创建一个服务类来处理与ChatGPT的交互。我们将使用Spring的RestTemplate或Feign客户端来发送HTTP请求。

import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.client.RestTemplate;
import org.springframework.http.*;
import com.fasterxml.jackson.databind.JsonNode;

@Service
public class ChatGPTService {

    @Value("${openai.api.key}")
    private String apiKey;

    private final RestTemplate restTemplate = new RestTemplate();

    public String chat(String userMessage) {
        String url = "https://api.openai.com/v1/chat/completions";

        // 设置请求头
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.set("Authorization", "Bearer " + apiKey);

        // 设置请求体
        String requestBody = "{ \"model\": \"gpt-3.5-turbo\", \"messages\": [{\"role\": \"user\", \"content\": \"" + userMessage + "\"}] }";

        HttpEntity<String> requestEntity = new HttpEntity<>(requestBody, headers);

        // 发送POST请求并接收响应
        ResponseEntity<JsonNode> responseEntity = restTemplate.exchange(url, HttpMethod.POST, requestEntity, JsonNode.class);

        // 解析响应并返回回复内容
        JsonNode responseBody = responseEntity.getBody();
        return responseBody.get("choices").get(0).get("message").get("content").asText();
    }
}

6. 创建控制器

接下来，我们创建一个控制器来接收用户的请求，并通过ChatGPT服务返回响应。

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("/api/chat")
public class ChatController {

    @Autowired
    private ChatGPTService chatGPTService;

    @PostMapping
    public String chat(@RequestBody String userMessage) {
        return chatGPTService.chat(userMessage);
    }
}

7. 测试API

启动你的Spring Boot应用程序后，你可以使用工具（如Postman或curl）测试API：

curl -X POST http://localhost:8080/api/chat -d "你好，ChatGPT！"

结论

通过以上步骤，我们成功地将ChatGPT接入到Spring Boot项目中。这个过程展示了如何使用RestTemplate发送HTTP请求到OpenAI的API，并获取与用户输入相关的智能回复。你可以根据实际需求进一步扩展功能，比如增加异常处理、改进请求参数等。希望这篇文章能帮助你在Spring Boot项目中顺利实现与ChatGPT的整合！