大家好,我是Yuan,这篇文章主要介绍如何在springboot项目中集成调用chatgpt的api功能。
1. 概述
在本教程中,我们将学习如何在 Spring Boot 中调用 OpenAI ChatGPT API。我们将创建一个 Spring Boot 应用程序,该应用程序将通过调用 OpenAI ChatGPT API 生成对提示的响应。
2.OpenAI ChatGPT API
在开始本教程之前,让我们先探讨一下我们将在本教程中使用的 OpenAI ChatGPT API。我们将调用创建聊天完成 API 来生成对提示的响应。
2.1. API参数与认证
我们看一下API的强制请求参数:
- model – 这是我们将向其发送请求的模型的版本。该模型有 。我们将使用 gpt-3.5-turbo模型,这是该模型公开的最新版本
- 消息——消息是对模型的提示。每条消息都需要两个字段: role 和 content。角色 字段指定消息的发送者。请求中它将是“用户” , 响应中将是“助手” 。内容 字段是实际的消息
为了使用 API 进行身份验证,我们将生成一个OpenAI API 密钥(地址:https://platform.openai.com/account/api-keys)(访问需要魔法)。我们将在调用 API 时在Authorization标头中设置此密钥。
cURL 格式的示例请求如下所示:
$ curl https://api.openai.com/v1/chat/completions \-H "Content-Type: application/json" \-H "Authorization: Bearer $OPENAI_API_KEY" \-d '{"model": "gpt-3.5-turbo","messages": [{"role": "user", "content": "Hello!"}]}'
此外,API 接受许多可选参数来修改响应。
在以下部分中,我们将重点关注一个简单的请求,但让我们看一下一些有助于调整响应的可选参数:
n – 如果我们想增加 生成的响应数量,则可以指定。默认值为 1。
温度——控制 响应的随机性。默认值为 1(最随机)。
max_tokens – 用于 限制响应中令牌的最大数量。默认值是无穷大,这意味着响应将与模型可以生成的一样长。一般来说,最好将此值设置为合理的数字,以避免生成很长的响应并产生很高的成本。
2.2. API响应
API 响应将是一个带有一些元数据和 选择 字段的 JSON 对象。选择 字段 将是一个对象数组。每个对象都有一个 文本 字段,其中包含对提示的响应。
选择数组中的对象数量 将等于 请求中的可选n参数。如果 未指定 n参数,则选项 数组将包含单个对象。
这是一个示例响应:
{"id": "chatcmpl-123","object": "chat.completion","created": 1677652288,"choices": [{"index": 0,"message": {"role": "assistant","content": "\n\nHello there, how may I assist you today?"},"finish_reason": "stop"}],"usage": {"prompt_tokens": 9,"completion_tokens": 12,"total_tokens": 21}
}
响应中的使用字段 将包含提示和响应中使用的令牌数量。这用于计算 API 调用的成本。
3. 代码示例
我们将创建一个使用 OpenAI ChatGPT API 的 Spring Boot 应用程序。为此,我们将创建一个 Spring Boot Rest API,该 API 接受提示作为请求参数,将其传递给 OpenAI ChatGPT API,并将响应作为响应正文返回。
3.1. 依赖关系
首先,我们创建一个 Spring Boot 项目。我们需要该项目的Spring Boot Starter Web依赖项:
<dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId>
</dependency>
3.2. DTO
接下来,我们创建一个与 OpenAI ChatGPT API 的请求参数对应的DTO :
public class ChatRequest {private String model;private List<Message> messages;private int n;private double temperature;public ChatRequest(String model, String prompt) {this.model = model;this.messages = new ArrayList<>();this.messages.add(new Message("user", prompt));}// getters and setters
}
我们还定义Message类:
public class Message {private String role;private String content;// constructor, getters and setters
}
同样,让我们为响应创建一个 DTO:
public class ChatResponse {private List<Choice> choices;// constructors, getters and setterspublic static class Choice {private int index;private Message message;// constructors, getters and setters}
}
3.3. Controller
接下来,让我们创建一个Controller,它将接受提示作为请求参数并返回响应作为响应正文:
@RestController
public class ChatController {@Qualifier("openaiRestTemplate")@Autowiredprivate RestTemplate restTemplate;@Value("${openai.model}")private String model;@Value("${openai.api.url}")private String apiUrl;@GetMapping("/chat")public String chat(@RequestParam String prompt) {// create a requestChatRequest request = new ChatRequest(model, prompt);// call the APIChatResponse response = restTemplate.postForObject(apiUrl, request, ChatResponse.class);if (response == null || response.getChoices() == null || response.getChoices().isEmpty()) {return "No response";}// return the first responsereturn response.getChoices().get(0).getMessage().getContent();}
}
让我们看一下代码的一些重要部分:
我们使用@Qualifier 注释来注入 我们将在下一节中创建的RestTemplate bean
使用 RestTemplate bean,我们使用 postForObject () 方法调用 OpenAI ChatGPT API。postForObject ()方法将 URL、请求对象和响应类作为参数
最后,我们读取回复的选择列表并返回第一个回复
3.4. RestTemplate
接下来,我们定义一个自定义RestTemplate bean,它将使用 OpenAI API 密钥进行身份验证:
@Configuration
public class OpenAIRestTemplateConfig {@Value("${openai.api.key}")private String openaiApiKey;@Bean@Qualifier("openaiRestTemplate")public RestTemplate openaiRestTemplate() {RestTemplate restTemplate = new RestTemplate();restTemplate.getInterceptors().add((request, body, execution) -> {request.getHeaders().add("Authorization", "Bearer " + openaiApiKey);return execution.execute(request, body);});return restTemplate;}
}
在这里,我们向基本 RestTemplate 添加了一个拦截器,并添加了 Authorization标头。
3.5. 属性
最后,让我们在application.properties文件中提供 API 的属性:
openai.model=gpt-3.5-turbo
openai.api.url=https://api.openai.com/v1/chat/completions
openai.api.key=your-api-key
4. 运行应用程序
我们现在可以运行该应用程序并在浏览器中测试它:
在浏览器中调用 API 时聊天 GPT 响应
正如我们所看到的,应用程序生成了对提示的响应。请注意,响应可能会有所不同,因为它是由模型生成的。
5. 结论
在本教程中,我们探索了 OpenAI ChatGPT API 以生成对提示的响应。我们创建了一个 Spring Boot 应用程序,它调用 API 来生成对提示的响应。