From 97f61cffb8f843a3830bcd0a5f6ab00b07d6348c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=A7=A6=E8=8B=B1=E6=9D=B0?= <327782001@qq.com> Date: Wed, 22 Mar 2023 21:10:25 +0800 Subject: [PATCH] =?UTF-8?q?fix:=E4=BD=BF=E7=94=A8chatgpt=E6=8E=A5=E5=8F=A3?= =?UTF-8?q?=E8=B0=83=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pom.xml | 17 ++++ .../com/kwan/spring5/pojo/ChoiceMessage.java | 22 +++++ .../java/com/kwan/spring5/pojo/Choices.java | 25 +++++ .../spring5/pojo/CompletionChatRequest.java | 41 ++++++++ .../spring5/pojo/CompletionChatResponse.java | 41 ++++++++ .../com/kwan/spring5/utils/OpenAIChatGpt.java | 99 +++++++++++++++++++ target/classes/spring1.xml | 4 - target/classes/spring23.xml | 19 ---- 8 files changed, 245 insertions(+), 23 deletions(-) create mode 100644 src/main/java/com/kwan/spring5/pojo/ChoiceMessage.java create mode 100644 src/main/java/com/kwan/spring5/pojo/Choices.java create mode 100644 src/main/java/com/kwan/spring5/pojo/CompletionChatRequest.java create mode 100644 src/main/java/com/kwan/spring5/pojo/CompletionChatResponse.java create mode 100644 src/main/java/com/kwan/spring5/utils/OpenAIChatGpt.java delete mode 100644 target/classes/spring23.xml diff --git a/pom.xml b/pom.xml index 72ac854..d8fa1a6 100644 --- a/pom.xml +++ b/pom.xml @@ -88,5 +88,22 @@ guava 31.1-jre + + org.projectlombok + lombok + 1.18.26 + + + + com.theokanning.openai-gpt3-java + service + 0.11.0 + + + + com.theokanning.openai-gpt3-java + client + 0.11.1 + \ No newline at end of file diff --git a/src/main/java/com/kwan/spring5/pojo/ChoiceMessage.java b/src/main/java/com/kwan/spring5/pojo/ChoiceMessage.java new file mode 100644 index 0000000..0717291 --- /dev/null +++ b/src/main/java/com/kwan/spring5/pojo/ChoiceMessage.java @@ -0,0 +1,22 @@ +package com.kwan.spring5.pojo; + +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; + +@Builder +@NoArgsConstructor +@AllArgsConstructor +@Data +public class ChoiceMessage { + /** + * 角色 + */ + String role; + /** + * 具体内容 + */ + String content; +} + diff --git a/src/main/java/com/kwan/spring5/pojo/Choices.java b/src/main/java/com/kwan/spring5/pojo/Choices.java new file mode 100644 index 0000000..b157fb6 --- /dev/null +++ b/src/main/java/com/kwan/spring5/pojo/Choices.java @@ -0,0 +1,25 @@ +package com.kwan.spring5.pojo; + +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; + +@Builder +@NoArgsConstructor +@AllArgsConstructor +@Data +public class Choices { + /** + * 内容对象 + */ + private ChoiceMessage message; + /** + * 结束原因 + */ + private String finish_reason; + /** + * 索引位置 + */ + private Integer index; +} diff --git a/src/main/java/com/kwan/spring5/pojo/CompletionChatRequest.java b/src/main/java/com/kwan/spring5/pojo/CompletionChatRequest.java new file mode 100644 index 0000000..f56ccb7 --- /dev/null +++ b/src/main/java/com/kwan/spring5/pojo/CompletionChatRequest.java @@ -0,0 +1,41 @@ +package com.kwan.spring5.pojo; + + +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; + +import java.util.List; +import java.util.Map; + + +/** + * 请求参数 + * + * @author : qinyingjie + * @version : 2.2.0 + * @date : 2023/3/22 21:00 + */ +@Builder +@NoArgsConstructor +@AllArgsConstructor +@Data +public class CompletionChatRequest { + String model; + List messages; + Integer maxTokens; + Double temperature; + Double topP; + Integer n; + Boolean stream; + Integer logprobs; + Boolean echo; + List stop; + Double presencePenalty; + Double frequencyPenalty; + Integer bestOf; + Map logitBias; + String user; +} + diff --git a/src/main/java/com/kwan/spring5/pojo/CompletionChatResponse.java b/src/main/java/com/kwan/spring5/pojo/CompletionChatResponse.java new file mode 100644 index 0000000..06b281c --- /dev/null +++ b/src/main/java/com/kwan/spring5/pojo/CompletionChatResponse.java @@ -0,0 +1,41 @@ +package com.kwan.spring5.pojo; + +import com.theokanning.openai.Usage; +import lombok.AllArgsConstructor; +import lombok.Builder; +import lombok.Data; +import lombok.NoArgsConstructor; + +import java.util.List; + +@Builder +@NoArgsConstructor +@AllArgsConstructor +@Data +public class CompletionChatResponse { + /** + * id + */ + String id; + /** + * chat.completion + */ + String object; + /** + * 创建时间 + */ + long created; + /** + * 模型 + */ + String model; + /** + * 结果集 + */ + List choices; + /** + * 使用情况 + */ + Usage usage; +} + diff --git a/src/main/java/com/kwan/spring5/utils/OpenAIChatGpt.java b/src/main/java/com/kwan/spring5/utils/OpenAIChatGpt.java new file mode 100644 index 0000000..4f1a144 --- /dev/null +++ b/src/main/java/com/kwan/spring5/utils/OpenAIChatGpt.java @@ -0,0 +1,99 @@ +package com.kwan.spring5.utils; + +import cn.hutool.http.HttpRequest; +import cn.hutool.http.HttpUtil; +import com.alibaba.fastjson2.JSON; +import com.kwan.spring5.pojo.ChoiceMessage; +import com.kwan.spring5.pojo.Choices; +import com.kwan.spring5.pojo.CompletionChatRequest; +import com.kwan.spring5.pojo.CompletionChatResponse; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +/** + * chatgpt-api + * + * @author : qinyingjie + * @version : 2.2.0 + * @date : 2023/3/22 20:40 + */ +public class OpenAIChatGpt { + /** + * 请求地址 + */ + final static private String chatCompletionUrl = "https://api.openai.com/v1/chat/completions"; + /** + * 模型 + */ + final static private String model = "gpt-3.5-turbo"; + /** + * 这里使用你自己的认证信息 + */ + final static private String Authorization = "Bearer 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"; + + + public static void main(String[] args) { + String question = "默写静夜思"; + //这里是我根据模型请求封装的模型实体 + CompletionChatRequest completionChatRequest = new CompletionChatRequest(); + //封装http请求 + HttpRequest post = HttpUtil.createPost(chatCompletionUrl).timeout(300000); + Map headers = new HashMap<>(); + headers.put("Authorization", Authorization); + post.addHeaders(headers); + post.contentType("application/json;charset=UTF-8"); + completionChatRequest.setModel(model); + ChoiceMessage message = new ChoiceMessage("user", question); + List messages = new ArrayList<>(); + messages.add(message); + completionChatRequest.setMessages(messages); + post.body(JSON.toJSONString(completionChatRequest)); + String body = post.execute().body(); + CompletionChatResponse completionChatResponse = JSON.parseObject(body, CompletionChatResponse.class); + final List choices = completionChatResponse.getChoices(); + for (Choices choice : choices) { + final ChoiceMessage message1 = choice.getMessage(); + final String content = message1.getContent(); + System.out.println(content); + } + System.out.println("成功"); + } + +// public static CompletionChatResponse chatCompletion(String prompt) {//prompt即要将要发送至gtp的内容 +// //这里我把用户发送和机器人返回的20条内容内容存在了redis中，用于关联对话上下文场景，如果不需要，注释掉即可，直接new一个ArrayList +// List messages = getUserCacheMessages(); +// int size = messages.size(); +// +// if (size > 19) { +// messages.remove(0); +// } +// +// messages.add(message); +// setUserCacheMessages(messages);//把新的放到redis +// completionChatRequest.setMessages(messages); +// post.body(JSON.toJSONString(completionChatRequest)); +// String body = post.execute().body(); +// CompletionChatResponse completionChatResponse = JSON.parseObject(body, CompletionChatResponse.class); +// return completionChatResponse; +// } +// +// public static List getUserCacheMessages(){ +// RedisCache redisCache = SpringUtils.getBean(RedisCache.class); +// List cacheMessages = redisCache.getCacheObject(JwtTokenUtil.getUserId + ":messages"); +// if (StringUtils.isEmpty(cacheMessages)){ +// cacheMessages= new ArrayList<>(); +// } +// return cacheMessages; +// } +// +// public static void setUserCacheMessages(List messages){ +// while(messages.size()>20){ +// messages.remove(0); +// } +// RedisCache redisCache = SpringUtils.getBean(RedisCache.class); +// redisCache.setCacheObject(JwtTokenUtil.getUserId + ":messages", messages, 30, TimeUnit.MINUTES); +// } +} diff --git a/target/classes/spring1.xml b/target/classes/spring1.xml index 2967571..5814835 100644 --- a/target/classes/spring1.xml +++ b/target/classes/spring1.xml @@ -9,7 +9,6 @@ - - - \ No newline at end of file diff --git a/target/classes/spring23.xml b/target/classes/spring23.xml deleted file mode 100644 index f4b9c04..0000000 --- a/target/classes/spring23.xml +++ /dev/null @@ -1,19 +0,0 @@ - - - - - - - - - - \ No newline at end of file -- GitLab