Java 调用OpenAI完成聊天

灰太狼 2023-10-13 21:51 23阅读 0赞

这段时间比较火的chatGPT，准确来说应该只是openAI的一个小部分，这里对openAI的功能接口进行一个java实现，以文本补全、聊天（也就是chatGPT）和图像生成作为演示，体会一下AI的强大力量。

## 一、openAI账号创建以及测试 ##

### 1.账号 ###

访问openai.com注册，按照提示输入注册就行，这里注意的就是需要外国手机号，还浪费了我几十块钱，可恶。。

### 2.创建API KEY ###

在个人中心里面创建一下api密钥，这里要注意key只有创建时可以看到，创建完之后就不可见了，因此在创建时要注意复制保存，当然如果实在忘了，删除重新创建一个就行  
![在这里插入图片描述][0eca7ac20c37452ea6daa042345dd184.png]

## 二、配置Maven ##

<dependency>
                <groupId>com.netflix.feign</groupId>
                <artifactId>feign-core</artifactId>
                <version>8.18.0</version>
            </dependency>
            <dependency>
                <groupId>com.netflix.feign</groupId>
                <artifactId>feign-jackson</artifactId>
                <version>8.18.0</version>
            </dependency>
    
            <dependency>
                <groupId>com.netflix.feign</groupId>
                <artifactId>feign-okhttp</artifactId>
                <version>8.18.0</version>
            </dependency>

## 三、配置yml文件 ##

key就是OpenAI官网申请的API KEY

open-ai:
       url: https://api.openai.com/
       key: xxxxxxxxx

## 三、Controller（以流的方式返回） ##

@Api(tags = "客户端-客户管理")
    @RestController
    @RequestMapping("/client/chatUser")
    public class ChatUserClientController {
        
    
        @Resource
        private ChatUserService chatUserService;
    
        /**
         * 返回数据流 优化用户体验
         * @return
         */
    
        @ApiOperation(value = "发送聊天",produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
        @PostMapping("/sendMessage")
        public ResponseBodyEmitter sendMessageByStream(@RequestBody OpenAIChatSendMessage openAIChatSendMessage) {
        
            return chatUserService.sendMessage(openAIChatSendMessage);
        }
    
    }

## 四、接口层 ##

public interface ChatUserService extends IService<ChatUser>{
        
        ResponseBodyEmitter sendMessage(OpenAIChatSendMessage openAIChatSendMessage);
    }

## 五、实现层 ##

可以根据自己的需求在此处添加判断逻辑

@Service
    public class ChatUserServiceImpl extends ServiceImpl<ChatUserMapper, ChatUser> implements ChatUserService {
        
    
        @Resource
        private OpenAIApiDao openAIApiDao;
    
        @Override
        public ResponseBodyEmitter sendMessage(OpenAIChatSendMessage openAIChatSendMessage) {
        
            List<OpenAIChatMessage> messages = openAIChatSendMessage.getMessages();
            if (CollectionUtils.isEmpty(messages) || messages.size() == 1) {
        
                throw new NingException(ServiceCode.FAILED);
            }
    		// 可在次数加判断逻辑
    
            ResponseBodyEmitter emitter = new ResponseBodyEmitter();
    
            CompletableFuture.runAsync(() -> {
        
                	 // 调用 OpenAI 返回
                    openAIChatSendMessage.setStream(true);
    
                    Response response = openAIApiDao.sendChatByOpenAI(openAIChatSendMessage);
    
                    try {
        
                        try (InputStream inputStream = response.body().asInputStream()) {
        
                            byte[] buffer = new byte[1024];
                            int bytesRead = -1;
                            while ((bytesRead = inputStream.read(buffer)) != -1) {
        
                                emitter.send(Arrays.copyOf(buffer, bytesRead));
                            }
                        } finally {
        
                            emitter.complete();
                            response.close();
                        }
                    } catch (IOException e) {
        
                        emitter.completeWithError(e);
                    }
            });
    
    
            return emitter;
        }
     }

## 六、OpenAIApiDao ##

public interface OpenAIApiDao {
        
    
        Response sendChatByOpenAI(OpenAIChatSendMessage openAIChatSendMessage);
    
    }

## 七、feign初始化、调用 ##

包含跳过https校验

@Slf4j
    @Service
    public class OpenAIApiDaoImpl implements OpenAIApiDao {
        
    
        private static OpenAIChatApi OPEN_AI_CHAT_API;
    
        @Value("${open-ai.url}")
        protected String url;
    
        @Value("${open-ai.key}")
        private String key;
    
    
        private static SSLSocketFactory createTrustAllSslSocketFactory(TrustManager[] trustManagers) throws NoSuchAlgorithmException, KeyManagementException {
        
            SSLContext sslContext = SSLContext.getInstance("TLS");
            sslContext.init(null, trustManagers, new SecureRandom());
            return sslContext.getSocketFactory();
        }
    
    
        @PostConstruct
        @SneakyThrows
        protected void init() {
        
    
    		// 跳过https校验
    
            // 1. 创建一个不进行任何证书校验的 TrustManager
            TrustManager[] trustAllCerts = new TrustManager[] {
        
                    new X509TrustManager() {
        
                        public void checkClientTrusted(X509Certificate[] chain, String authType) {
        }
                        public void checkServerTrusted(X509Certificate[] chain, String authType) {
        }
                        public X509Certificate[] getAcceptedIssuers() {
         return new X509Certificate[0]; }
                    }
            };
    
            // 2. 创建一个 OkHttpClient.Builder 对象，并将上面创建的 TrustManager 设置到 OkHttpClient.Builder 中
            OkHttpClient.Builder builder = new OkHttpClient.Builder()
                    .hostnameVerifier((hostname, session) -> true)
                    .sslSocketFactory(createTrustAllSslSocketFactory(trustAllCerts), new X509TrustManager() {
        
                        @Override
                        public void checkClientTrusted(X509Certificate[] chain, String authType) throws CertificateException {
        }
    
                        @Override
                        public void checkServerTrusted(X509Certificate[] chain, String authType) throws CertificateException {
        }
    
                        @Override
                        public X509Certificate[] getAcceptedIssuers() {
        
                            return new X509Certificate[0];
                        }
                    });
    
            OPEN_AI_CHAT_API = Feign.builder()
                    .client(new feign.okhttp.OkHttpClient(builder.build()))
                    .decoder(new Decoder.Default())
                    .encoder(new JacksonEncoder())
                    .errorDecoder(new ErrorDecoder.Default())
                    .target(OpenAIChatApi.class, url);
    
        }
    
        @Override
        public Response sendChatByOpenAI(OpenAIChatSendMessage openAIChatSendMessage) {
        
    
            return OPEN_AI_CHAT_API.sendChatMessage(key, openAIChatSendMessage);
        }
    }

## 八、调用层 ##

public interface OpenAIChatApi {
        
    
        /**
         * 聊天消息发送
         */
    
        @RequestLine("POST v1/chat/completions")
        @Headers({
        "Content-Type: application/json","Authorization: {key}","Accept: application/octet-stream"})
        Response sendChatMessage(@Param("key") String key, OpenAIChatSendMessage messages);
    
    }

## 九、VO类 ##

### 1.OpenAIChatMessage ###

@Data
    public class OpenAIChatMessage {
        
    
        private String role;
    
        private String content;
    }

### 2.OpenAIChatSendMessage ###

@Data
    public class OpenAIChatSendMessage {
        
    
        @ApiModelProperty("消息记录")
        private List<OpenAIChatMessage> messages;
    
    
        @ApiModelProperty("是否使用流式传输")
        private Boolean stream;
    
    
        @ApiModelProperty("温度（准确率，温度越高消费时间约长）")
        private BigDecimal temperature;
    
        @ApiModelProperty("指定生成文本的概率阈值。这个参数可以控制生成文本的多样性，以避免生成过于相似的文本。")
        private BigDecimal top_p;
    }

ok，齐活~

[0eca7ac20c37452ea6daa042345dd184.png]: https://img-blog.csdnimg.cn/0eca7ac20c37452ea6daa042345dd184.png