OpenAi Java 流式返回接入最简单指南：使用 Spring Boot 构建 ChatGPT

2023-09-16 20:37:51

利用 Java 库构建 ChatGPT：开启对话式人工智能之旅

在人工智能领域，OpenAI 的 ChatGPT 堪称一颗闪耀的新星。作为一款基于 Transformer 神经网络的大型语言模型，ChatGPT 以其生成流畅文本、回答问题和创作内容的能力而闻名。然而，ChatGPT 的官方 API 使用费用较高，对于个人开发者而言可能难以负担。本文将介绍如何利用 OpenAI 提供的 Java 库免费构建 ChatGPT，让你开启与人工智能对话的精彩旅程。

第一步：配置你的 Java 环境

要开始构建 ChatGPT，你需要创建一个新的 Spring Boot 项目。可以使用 Spring Initializr 来快速生成项目。选择 "Spring Boot" 和 "Java"，然后点击 "Generate" 按钮。

接下来，在项目中添加 OpenAI Java 库。在 Maven 的 pom.xml 文件中添加以下依赖：

<dependency>
  <groupId>com.openai</groupId>
  <artifactId>openai-java</artifactId>
  <version>3.10.0</version>
</dependency>

第二步：获取 OpenAI API 密钥

在你的 Spring Boot 项目中，你需要配置 OpenAI API 密钥。可以在 application.properties 文件中添加以下配置：

openai.api-key=<YOUR_API_KEY>

其中 <YOUR_API_KEY> 是你的 OpenAI API 密钥。

第三步：创建 ChatGPT 控制器

下一步是创建一个 ChatGPT Controller 来处理 ChatGPT 的请求。在 src/main/java/com/example/demo/ChatGPTController.java 文件中创建以下代码：

import com.openai.OpenAI;
import com.openai.request.CompletionRequest;
import com.openai.response.CompletionResponse;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

@RestController
public class ChatGPTController {

    @PostMapping("/chat")
    public CompletionResponse chat(@RequestBody String prompt) {
        OpenAI openAI = new OpenAI(System.getenv("OPENAI_API_KEY"));
        CompletionRequest request = new CompletionRequest();
        request.setModel("text-davinci-003");
        request.setPrompt(prompt);
        return openAI.completions(request);
    }
}