Springboot集成OpenAi chatgpt3.5官方开源项目opanAI-Java

前言

本文主要介绍Springboot集成openai-java完成openai官方接口的调用，官方有多种语言的demo示例

OPENAI开源openai-java项目地址：https://github.com/TheoKanning/openai-java

准备工作

必要的前提，要使用chatgpt必须要魔法

魔法
openai帐号（需要apiKey）
springboot+maven的项目

开始

1、maven中引入openai-java

目前用的版本是0.12.0

	<!-- openai --><dependency><groupId>com.theokanning.openai-gpt3-java</groupId><artifactId>api</artifactId><version>${openai.version}</version></dependency><dependency><groupId>com.theokanning.openai-gpt3-java</groupId><artifactId>client</artifactId><version>${openai.version}</version></dependency><dependency><groupId>com.theokanning.openai-gpt3-java</groupId><artifactId>service</artifactId><version>${openai.version}</version></dependency>

2、配置

测试我把配置信息放在了yml中

openai:proxyHost: 127.0.0.1proxyPort: 7890keys:- sk-xxxxxxxxxxxxxxxxxxxxxxxx

配置类

@Configuration
@ConfigurationProperties(prefix = "openai")
public class OpenAiModel {/*** 代理地址*/private static String proxyHost;/*** 代理端口*/private static Integer proxyPort;/*** openai apikey*/private static List<String> keys;// 省略 get set	
}

3、使用

调用api的核心类是OpenAiService，不清楚是不是魔法的问题，我直接调用会ping不通，请求超时，必须设置代理。

/*** openAiService 工厂* @author WuHao* @since 2023/5/24 10:00*/
public class AiServiceFactory {private static final Duration DEFAULT_TIMEOUT = Duration.ofSeconds(10L);public static OpenAiService createService() {String token = Optional.ofNullable(OpenAiModel.getKeys()).orElseThrow(() -> new RuntimeException("ApiKey不能为空，请检查参数配置")).stream().findFirst().orElse(null);Assert.notEmpty(token,() -> new RuntimeException("ApiKey不能为空，请检查参数配置"));ObjectMapper mapper = OpenAiService.defaultObjectMapper();// 设置代理Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(OpenAiModel.getProxyHost(), OpenAiModel.getProxyPort()));OkHttpClient client = OpenAiService.defaultClient(token, DEFAULT_TIMEOUT).newBuilder().proxy(proxy).build();Retrofit retrofit = OpenAiService.defaultRetrofit(client, mapper);return new OpenAiService(retrofit.create(OpenAiApi.class), client.dispatcher().executorService());}}

测试类

@GetMapping("/testChat")public String testChat() throws UnsupportedEncodingException {OpenAiService service = AiServiceFactory.createService();final List<ChatMessage> messages = new ArrayList<>();final ChatMessage systemMessage = new ChatMessage(ChatMessageRole.SYSTEM.value(), URLDecoder.decode("取一个3个字的中文名字，要求姓氏为吴", "UTF-8"));messages.add(systemMessage);ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder().model("gpt-3.5-turbo").messages(messages).n(1).maxTokens(50).logitBias(new HashMap<>()).build();service.streamChatCompletion(chatCompletionRequest).doOnError(Throwable::printStackTrace).blockingForEach(System.err::println);service.shutdownExecutor();return null;