• SpringBoot调用ChatGPT-API实现智能对话

    目录

    一、说明

    二、代码

    2.1、对话测试

    2.2、单次对话

    2.3、连续对话

    2.4、AI绘画

    一、说明

    我们在登录chatgpt官网进行对话是不收费的，但需要魔法。在调用官网的API时，在代码层面上使用，通过API KEY进行对话是收费的，不过刚注册的小伙伴有免费5美金的体验额度，在不调用绘画模型，只是做为简单的问答，个人使用是没问题的。

    ChatGPT官网

    API使用情况

    二、代码

    2.1、对话测试

    Gpt35TurboVO

    1. import lombok.Data;
    2.  
    3. @Data
    4. public class Gpt35TurboVO {
    5.     private String role;     // 角色一般为 user
    6.     private String content;  // 询问内容
    7. }

    Controller

    1.     @GetMapping(value = "/test", produces = "text/event-stream;charset=UTF-8")
    2.     public String test(@RequestParam String message) {
    3.         //回复用户
    4.         String apikey = "sk-****";
    5.         //请求ChatGPT的URL
    6.         String url = "https://api.openai.com/v1/chat/completions";
    7.  
    8.         Gpt35TurboVO gpt35TurboVO = new Gpt35TurboVO();
    9.         gpt35TurboVO.setRole("user");
    10.         gpt35TurboVO.setContent(message);
    11.         List objects = new ArrayList<>();
    12.         objects.add(gpt35TurboVO);
    13.         Map objectObjectHashMap = new HashMap<>();
    14.         objectObjectHashMap.put("model", "gpt-3.5-turbo");  //使用的模型
    15.         objectObjectHashMap.put("messages", objects);       //提问信息
    16.         objectObjectHashMap.put("stream", false);            //流
    17.         objectObjectHashMap.put("temperature", 0);          //GPT回答温度（随机因子）
    18.         objectObjectHashMap.put("frequency_penalty", 0);    //重复度惩罚因子
    19.         objectObjectHashMap.put("presence_penalty", 0.6);   //控制主题的重复度
    20.         String postData = JSONUtil.toJsonStr(objectObjectHashMap);
    21.  
    22.         String result2 = HttpRequest.post(url)
    23.                 .header("Authorization", "Bearer " + apikey)//头信息，多个头信息多次调用此方法即可
    24.                 .header("Content-Type", "application/json")
    25.                 .body(postData)//表单内容
    26.                 .timeout(200000)//超时，毫秒
    27.                 .execute().body();
    28.  
    29.         System.out.println(result2);
    30.  
    31.         return result2;
    32.  
    33.     }

    返回结果

    2.2、单次对话

    ChatBotSingleQuestionVO

    1. import lombok.Data;
    2.  
    3. /**
    4.  * 应用管理-单次提问-VO
    5.  * @author lf
    6.  * @date 2023/8/18
    7.  */
    8. @Data
    9. public class ChatBotSingleQuestionVO {
    10.  
    11.     /**
    12.      * 用户输入的询问内容
    13.      */
    14.     private String prompt;
    15.  
    16.     /**
    17.      * 角色扮演ID
    18.      */
    19.     private Integer rolePlayId;
    20.  
    21. }

    Redis锁工具类

    1. import org.redisson.api.RLock;
    2. import org.redisson.api.RedissonClient;
    3. import org.springframework.beans.factory.annotation.Autowired;
    4. import org.springframework.stereotype.Component;
    5.  
    6. import java.util.concurrent.TimeUnit;
    7.  
    8. /**
    9.  * redis锁工具类
    10.  *
    11.  * @author ruoyi
    12.  */
    13. @Component
    14. public class RedisLock {
    15.     @Autowired
    16.     private RedissonClient redissonClient;
    17.  
    18.     /**
    19.      * 获取锁
    20.      *
    21.      * @param lockKey 锁实例key
    22.      * @return 锁信息
    23.      */
    24.     public RLock getRLock(String lockKey) {
    25.         return redissonClient.getLock(lockKey);
    26.     }
    27.  
    28.     /**
    29.      * 加锁
    30.      *
    31.      * @param lockKey 锁实例key
    32.      * @return 锁信息
    33.      */
    34.     public RLock lock(String lockKey) {
    35.         RLock lock = getRLock(lockKey);
    36.         lock.lock();
    37.         return lock;
    38.     }
    39.  
    40.     /**
    41.      * 加锁
    42.      *
    43.      * @param lockKey   锁实例key
    44.      * @param leaseTime 上锁后自动释放锁时间
    45.      * @return true=成功；false=失败
    46.      */
    47.     public Boolean tryLock(String lockKey, long leaseTime) {
    48.         return tryLock(lockKey, 0, leaseTime, TimeUnit.SECONDS);
    49.     }
    50.  
    51.     /**
    52.      * 加锁
    53.      *
    54.      * @param lockKey   锁实例key
    55.      * @param leaseTime 上锁后自动释放锁时间
    56.      * @param unit      时间颗粒度
    57.      * @return true=加锁成功；false=加锁失败
    58.      */
    59.     public Boolean tryLock(String lockKey, long leaseTime, TimeUnit unit) {
    60.         return tryLock(lockKey, 0, leaseTime, unit);
    61.     }
    62.  
    63.     /**
    64.      * 加锁
    65.      *
    66.      * @param lockKey   锁实例key
    67.      * @param waitTime  最多等待时间
    68.      * @param leaseTime 上锁后自动释放锁时间
    69.      * @param unit      时间颗粒度
    70.      * @return true=加锁成功；false=加锁失败
    71.      */
    72.     public Boolean tryLock(String lockKey, long waitTime, long leaseTime, TimeUnit unit) {
    73.         RLock rLock = getRLock(lockKey);
    74.         boolean tryLock = false;
    75.         try {
    76.             tryLock = rLock.tryLock(waitTime, leaseTime, unit);
    77.         } catch (InterruptedException e) {
    78.             return false;
    79.         }
    80.         return tryLock;
    81.     }
    82.  
    83.     /**
    84.      * 释放锁
    85.      *
    86.      * @param lockKey 锁实例key
    87.      */
    88.     public void unlock(String lockKey) {
    89.         RLock lock = getRLock(lockKey);
    90.         lock.unlock();
    91.     }
    92.  
    93.     /**
    94.      * 释放锁
    95.      *
    96.      * @param lock 锁信息
    97.      */
    98.     public void unlock(RLock lock) {
    99.         lock.unlock();
    100.     }
    101. }

    Controller

    1.     @PostMapping("/chatBotSingleQuestion/api")
    2.     public AjaxResult chatBotSingleQuestion(@RequestBody ChatBotSingleQuestionVO chatBotSingleQuestionVO) {
    3.         String answerContent = iChatBotSingleQuestionService.chatBotSingleQuestion(chatBotSingleQuestionVO);
    4.         return success("success", answerContent);
    5.     }

    Impl

    1.     /**
    2.      * 应用管理-用户单次提问-不支持续问对话
    3.      * @param chatBotSingleQuestionVO
    4.      * @return
    5.      */
    6.     @Override
    7.     @Transactional
    8.     public String chatBotSingleQuestion(ChatBotSingleQuestionVO chatBotSingleQuestionVO) {
    9.  
    10.         if (Objects.isNull(chatBotSingleQuestionVO.getRolePlayId())){
    11.             throw new RuntimeException("参数不可为空");
    12.         }
    13.  
    14.  
    15.         String lockName = "QA_" + SecurityUtils.getUserId();
    16.  
    17.         //回答的内容
    18.         String answerContent = "";
    19.  
    20.         try{
    21.             RLock rLock = redisLock.getRLock(lockName);
    22.             boolean locked = rLock.isLocked();
    23.  
    24.             if (locked) {
    25.                 throw new RuntimeException("正在回复中...");
    26.             }
    27.  
    28.             //对同一用户访问加锁
    29.             redisLock.lock(lockName);
    30.             this.chatBefore(chatBotSingleQuestionVO);
    31.             InputStream is = this.sendRequestBeforeChat(chatBotSingleQuestionVO);
    32.             String line = "";
    33.             BufferedReader reader = new BufferedReader(new InputStreamReader(is));
    34.             while ((line = reader.readLine()) != null) {
    35.                 //首先对行数据进行处理
    36.                 if (StrUtil.isNotBlank(line) &&
    37.                         !StrUtil.equals(line, "event: answer") &&
    38.                         !StrUtil.equals(line, "event: chatResponse") &&
    39.                         !StrUtil.contains(line, "data: {\"quoteLen\"")) {
    40.                     line = CollectionUtil.removeEmpty(StrUtil.split(line, "data: ")).get(0);
    41.                     if (!StrUtil.contains(line, "[DONE]")) {
    42.                         String oneWord = catchTextGpt(line);
    43.                         if (StrUtil.isNotBlank(oneWord)) {
    44.                             answerContent = answerContent + oneWord;
    45.                         }
    46.                     }
    47.                     WebSocketService.sendInfo(line, SecurityUtils.getUserId() + "");
    48.                     TimeUnit.MILLISECONDS.sleep(50);
    49.                 }
    50.             }
    51.             //处理完了后将次条聊天记录进行记录
    52.             if (StrUtil.isNotBlank(answerContent)) {
    53.                 //保存聊天记录
    54.                 this.saveDialogueProcess(chatBotSingleQuestionVO, answerContent);
    55.                 //更新提问次数
    56.                 this.upddateAppModel(chatBotSingleQuestionVO);
    57.             }
    58.             is.close();
    59.             reader.close();
    60.  
    61.  
    62.         } catch (Exception e) {
    63.             throw new RuntimeException(e.getMessage());
    64.         }finally {
    65.             redisLock.unlock(lockName);
    66.         }
    67.         return answerContent;
    68.     }

    sendRequestBeforeChat方法

    1.     /**
    2.      * 这块为问询,不包含对话模式
    3.      *
    4.      * @param chatBotSingleQuestionVO
    5.      * @return
    6.      * @throws Exception
    7.      */
    8.     @Transactional
    9.     public InputStream sendRequestBeforeChat(ChatBotSingleQuestionVO chatBotSingleQuestionVO) throws Exception {
    10.         InputStream in = null;
    11.         // 通知内容添加文本铭感词汇过滤
    12.         //其余错误见返回码说明
    13.         //正常返回0
    14.         //违禁词检测
    15.         this.disableWordCheck(chatBotSingleQuestionVO.getPrompt());
    16.         String apikeyRefresh = getOpenAiKey();
    17.         if (StrUtil.isBlank(apikeyRefresh)) {
    18.             throw new RuntimeException("无可用key");
    19.         }
    20.         List chatContext = this.getChatContext(chatBotSingleQuestionVO);
    21.         String requestUrl = iTbKeyManagerService.getproxyUrl();
    22.         Map objectObjectHashMap = new HashMap<>();
    23.         objectObjectHashMap.put("model", "gpt-3.5-turbo");
    24.         objectObjectHashMap.put("messages", chatContext);
    25.         objectObjectHashMap.put("stream", true);
    26.         objectObjectHashMap.put("temperature", 0);
    27.         objectObjectHashMap.put("frequency_penalty", 0);
    28.         objectObjectHashMap.put("presence_penalty", 0.6);
    29.         String bodyJson = JSONUtil.toJsonStr(objectObjectHashMap);
    30.         URL url = new URL(requestUrl); // 接口地址
    31.         HttpURLConnection urlConnection = (HttpURLConnection) url.openConnection();
    32.         urlConnection.setRequestMethod("POST");
    33.         urlConnection.setDoOutput(true);
    34.         urlConnection.setDoInput(true);
    35.         urlConnection.setUseCaches(false);
    36.         urlConnection.setRequestProperty("Connection", "Keep-Alive");
    37.         urlConnection.setRequestProperty("Charset", "UTF-8");
    38.         urlConnection.setRequestProperty("Authorization", "Bearer " + apikeyRefresh);
    39.         urlConnection.setRequestProperty("Content-Type", "application/json; charset=UTF-8");
    40.         byte[] dataBytes = bodyJson.getBytes();
    41.         urlConnection.setRequestProperty("Content-Length", String.valueOf(dataBytes.length));
    42.         OutputStream os = urlConnection.getOutputStream();
    43.         os.write(dataBytes);
    44.         in = new BufferedInputStream(urlConnection.getInputStream());
    45.         os.flush();
    46.         os.close();
    47.         return in;
    48.  
    49.     }

    catchTextGpt方法

    1.     /**
    2.      * 处理单独打印的文字
    3.      *
    4.      * @param str
    5.      * @return
    6.      */
    7.     public String catchTextGpt(String str) {
    8.         String choices = JsonUtil.parseMiddleData(str, "choices");
    9.         JSONArray jsonArray = JSONUtil.parseArray(choices);
    10.         String string = jsonArray.getByPath("[0].delta").toString();
    11.         String content = JsonUtil.parseMiddleData(string, "content");
    12.         return content;
    13.     }

    getChatContext方法

    1.     /**
    2.      * 角色扮演
    3.      * @param chatBotSingleQuestionVO
    4.      * @return 结果
    5.      */
    6.     public List getChatContext(ChatBotSingleQuestionVO chatBotSingleQuestionVO) {
    7.  
    8.         List messages = new ArrayList<>();
    9.  
    10.         //首先获取角色，设置默认角色
    11.         String default_role_content = iconfigService.selectConfigByKey("default_role");
    12.         if (StrUtil.isBlank(default_role_content)) {
    13.             default_role_content = "我是一个全能机器人,可以回答你任何问题";
    14.         }
    15.  
    16.         TbAppModel tbAppModel = iTbAppModelService.selectTbAppModelById(Long.valueOf(chatBotSingleQuestionVO.getRolePlayId()));
    17.         if (!Objects.isNull(tbAppModel)){
    18.             default_role_content = tbAppModel.getAppModelContent();
    19.         }
    20.         //小程序专用
    21.         //是否通过微信小程序审核的设定语
    22.         String is_open_pass_wx_promt = iconfigService.selectConfigByKey("is_open_pass_wx_promt");
    23.         if (StrUtil.isNotBlank(is_open_pass_wx_promt) && !StrUtil.equals(is_open_pass_wx_promt, "0")) {
    24.             String pass_wx_promt = iconfigService.selectConfigByKey("pass_wx_promt");
    25.             default_role_content = default_role_content + "。并且你必须遵循:" + pass_wx_promt;
    26.         }
    27.  
    28.         //设定系统所扮演的角色
    29.         Gpt35TurboVO gpt35TurboVOSys = new Gpt35TurboVO();
    30.         gpt35TurboVOSys.setRole("system");
    31.         gpt35TurboVOSys.setContent(default_role_content);
    32.         messages.add(gpt35TurboVOSys);
    33.  
    34.  
    35.         //最后查询用户最新询问的问题
    36.         Gpt35TurboVO gpt35TurboUser = new Gpt35TurboVO();
    37.         gpt35TurboUser.setRole("user");
    38.         gpt35TurboUser.setContent(chatBotSingleQuestionVO.getPrompt());
    39.         messages.add(gpt35TurboUser);
    40.         return messages;
    41.     }

    getOpenAiKey方法

    1.     /**
    2.      * 查询key是否可用
    3.      * @return 结果
    4.      */
    5.     public String getOpenAiKey() {
    6.         //模仿查到的key集合
    7.         TbKeyManager tbKeyManager = new TbKeyManager();
    8.         tbKeyManager.setIsUse(1);
    9.         //可用的key
    10.         List tbKeyManagers = iTbKeyManagerService.selectTbKeyManagerList(tbKeyManager);
    11.         //判断是否key额度用完
    12.         if (CollectionUtil.isEmpty(tbKeyManagers) || tbKeyManagers.size() <= 0) {
    13.             throw new RuntimeException("key额度耗尽");
    14.         }
    15.         //获取第一个key,然后将第一个key存入缓存
    16.         String key = tbKeyManagers.get(0).getSecretKey();
    17.         redisTemplate.opsForValue().set("apikey", key);
    18.         //检查key
    19.         changeKey(tbKeyManagers.get(0));
    20.         return key;
    21.     }

    2.3、连续对话

    Controller

    1.     @PostMapping(value = "/chatBotNoId/api")
    2.     public AjaxResult continuousDialogue(@RequestBody StreamParametersVO streamParametersVO) {
    3.         String answerContent = iChatGtpService.continuousDialogueSocketStream(streamParametersVO);
    4.         return success("success", answerContent);
    5.     }

    Impl

    1.     /**
    2.      * 用户直接发起连续对话，系统同意创建对话主题，用户不用手动新建主题
    3.      *
    4.      * @param streamParametersVO
    5.      */
    6.     @Override
    7.     public String continuousDialogueSocketStream(StreamParametersVO streamParametersVO) {
    8.  
    9.         //判断是否isNewOpen填写参数，表示是否先开对话
    10.         if (Objects.isNull(streamParametersVO.getIsNewOpen())) {
    11.             throw new RuntimeException("isNewOpen参数未填");
    12.         }
    13.  
    14.         if (streamParametersVO.getIsNewOpen()) {
    15.             //新开对话，创建新的对话主题
    16.             tbModelTable = new TbModelTable();
    17.             //主题名称
    18.             tbModelTable.setModelName("Dialogue_" + SecurityUtils.getUserId() + "_" + DateTime.now());
    19.             //设置模板角色
    20.             if (Objects.nonNull(streamParametersVO.getDialogueRoleId())) {
    21.                 tbModelTable.setId(Long.valueOf(streamParametersVO.getDialogueRoleId()));
    22.             } else {
    23.                 tbModelTable.setId(1L);
    24.             }
    25.             tbDialogueMain = tbDialogueMainService.creatNewDig(tbModelTable);
    26.         } else {
    27.             //非新开对话，查询本次的对话主题
    28.             TbDialogueMain tbDialogueMainParam = new TbDialogueMain();
    29.             //设置模板角色
    30.             if (Objects.nonNull(streamParametersVO.getDialogueRoleId())) {
    31.                 tbDialogueMainParam.setDialogueRoleId(Long.valueOf(streamParametersVO.getDialogueRoleId()));
    32.             } else {
    33.                 tbDialogueMainParam.setDialogueRoleId(1L);
    34.             }
    35.  
    36.             tbDialogueMainParam.setUserId(SecurityUtils.getUserId());
    37.             List tbDialogueMains = iTbDialogueMainService.selectTbDialogueMainList(tbDialogueMainParam);
    38.  
    39.             if (CollectionUtil.isEmpty(tbDialogueMains)) {
    40.                 //创建新的对话主题
    41.                 tbModelTable = new TbModelTable();
    42.                 //主题名称
    43.                 tbModelTable.setModelName("Dialogue_" + SecurityUtils.getUserId() + "_" + DateTime.now());
    44.                 //设置模板角色
    45.                 tbModelTable.setId(Long.valueOf(streamParametersVO.getDialogueRoleId()));
    46.                 tbDialogueMain = tbDialogueMainService.creatNewDig(tbModelTable);
    47.             } else {
    48.                 tbDialogueMain = tbDialogueMains.get(0);
    49.             }
    50.  
    51.         }
    52.  
    53.         //设置对话ID
    54.         streamParametersVO.setDialogueId(tbDialogueMain.getId());
    55.  
    56.         String lockName = "chat_" + SecurityUtils.getUserId();
    57.         //回答的内容
    58.         String answerContent = "";
    59.         try {
    60.  
    61.             RLock rLock = redisLock.getRLock(lockName);
    62.             boolean locked = rLock.isLocked();
    63.  
    64.             if (locked) {
    65.                 throw new RuntimeException("正在回复中...");
    66.             }
    67.  
    68.             //对同一用户访问加锁
    69.             redisLock.lock(lockName);
    70.  
    71.             //进来做校验
    72.             TbDialogueMain tbDialogueMain = this.paramVerify(streamParametersVO);
    73.  
    74.             String userId = SecurityUtils.getUserId() + "";
    75.  
    76.             //将提问数据封装为流,并请求OpenAI的接口
    77.             InputStream inputStream = this.sendRequestBefore(streamParametersVO, tbDialogueMain);
    78.  
    79.             String line = null;
    80.  
    81.             BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream));
    82.  
    83.             while ((line = reader.readLine()) != null) {
    84.  
    85.                 //首先对行数据进行处理
    86.                 if (StrUtil.isNotBlank(line) &&
    87.                         !StrUtil.equals(line, "event: answer") &&
    88.                         !StrUtil.equals(line, "event: chatResponse") &&
    89.                         !StrUtil.contains(line, "data: {\"quoteLen\"")) {
    90.  
    91.                     line = CollectionUtil.removeEmpty(StrUtil.split(line, "data: ")).get(0);
    92.  
    93.                     if (StrUtil.contains(line, "[DONE]")) {
    94.                     } else {
    95.  
    96.                         String oneWord = catchTextGpt(line);
    97.                         if (StrUtil.isNotBlank(oneWord)) {
    98.                             answerContent = answerContent + oneWord;
    99.                         }
    100.                     }
    101.  
    102.                     WebSocketService.sendInfo(line, userId);
    103.                     TimeUnit.MILLISECONDS.sleep(50);
    104.  
    105.                 }
    106.             }
    107.  
    108.             //处理完了后,将此条聊天记录进行保存
    109.             if (StrUtil.isNotBlank(answerContent)) {
    110.                 //保存聊天记录
    111.                 this.saveDig(streamParametersVO, answerContent);
    112.             }
    113.             inputStream.close();
    114.             reader.close();
    115.  
    116.         } catch (Exception e) {
    117.             throw new RuntimeException(e.getMessage());
    118.         } finally {
    119.             //解锁
    120.             redisLock.unlock(lockName);
    121.             //清除正在问话的标识
    122.             redisTemplate.delete(SecurityUtils.getUserId() + "");
    123.         }
    124.  
    125.         return answerContent;
    126.     }

    sendRequestBefore方法

    1.     /**
    2.      * 这块为 - 流对话模式的封装
    3.      *
    4.      * @param streamParametersVO
    5.      * @param tbDialogueMain
    6.      * @return
    7.      * @throws Exception
    8.      */
    9.     @Transactional
    10.     public InputStream sendRequestBefore(StreamParametersVO streamParametersVO, TbDialogueMain tbDialogueMain) throws Exception {
    11.  
    12.         InputStream in = null;
    13.         //提问内容
    14.         String prompt = streamParametersVO.getPrompt();
    15.  
    16.         // 获取当前的用户
    17.         String userId = SecurityUtils.getUserId() + "";
    18.         Object o = redisTemplate.opsForValue().get(userId);
    19.  
    20.         if (!Objects.isNull(o)) {
    21.             throw new RuntimeException("正在回复");
    22.         }
    23.  
    24.         redisTemplate.opsForValue().set(userId, true, 30, TimeUnit.SECONDS);
    25.  
    26.         if (StrUtil.isBlank(prompt)) {
    27.             throw new RuntimeException("输入内容为空");
    28.         }
    29.  
    30.         // 通知内容添加文本铭感词汇过滤
    31.         // 其余错误见返回码说明
    32.         // 违禁词检测 正常返回0
    33.         this.disableWordCheck(prompt);
    34.  
    35.         String apikeyRefresh = getOpenAiKey();
    36.         if (StrUtil.isBlank(apikeyRefresh)) {
    37.             throw new RuntimeException("无可用key");
    38.         }
    39.  
    40.         //处理提问内容（指定系统角色+对话上下文+最新的提问内容）
    41.         List chatContext = this.getChatDigContext(streamParametersVO, tbDialogueMain);
    42.  
    43.         String requestUrl = iTbKeyManagerService.getproxyUrl();
    44.         Map objectObjectHashMap = new HashMap<>();
    45.         objectObjectHashMap.put("model", "gpt-3.5-turbo");
    46.         objectObjectHashMap.put("messages", chatContext);
    47.         objectObjectHashMap.put("stream", true);
    48.         objectObjectHashMap.put("temperature", 0);
    49.         objectObjectHashMap.put("frequency_penalty", 0);
    50.         objectObjectHashMap.put("presence_penalty", 0.6);
    51.         String bodyJson = JSONUtil.toJsonStr(objectObjectHashMap);
    52.         URL url = new URL(requestUrl); // 接口地址
    53.         HttpURLConnection urlConnection = (HttpURLConnection) url.openConnection();
    54.         urlConnection.setRequestMethod("POST");
    55.         urlConnection.setDoOutput(true);
    56.         urlConnection.setDoInput(true);
    57.         urlConnection.setUseCaches(false);
    58.         urlConnection.setRequestProperty("Connection", "Keep-Alive");
    59.         urlConnection.setRequestProperty("Charset", "UTF-8");
    60.         urlConnection.setRequestProperty("Authorization", "Bearer " + apikeyRefresh);
    61.         urlConnection.setRequestProperty("Content-Type", "application/json; charset=UTF-8");
    62.         byte[] dataBytes = bodyJson.getBytes();
    63.         urlConnection.setRequestProperty("Content-Length", String.valueOf(dataBytes.length));
    64.         OutputStream os = urlConnection.getOutputStream();
    65.         os.write(dataBytes);
    66.         in = new BufferedInputStream(urlConnection.getInputStream());
    67.         os.flush();
    68.         os.close();
    69.         return in;
    70.  
    71.     }

    getChatDigContext方法

    1.     /**
    2.      * 获取对话上下文
    3.      *
    4.      * @param streamParametersVO
    5.      * @return
    6.      */
    7.     public List getChatDigContext(StreamParametersVO streamParametersVO, TbDialogueMain tbDialogueMain) {
    8.  
    9.         List messages = new ArrayList<>();
    10.  
    11.         //首先获取角色,默认角色
    12.         String default_role_content = iconfigService.selectConfigByKey("default_role");
    13.         if (StrUtil.isBlank(default_role_content)) {
    14.             default_role_content = "我是一个全能机器人,可以回答你任何问题";
    15.         }
    16.  
    17.         //根据用户传递过来的Id查询角色模型数据
    18.         TbModelTable tbModelTable = iTbModelTableService.selectTbModelTableById(tbDialogueMain.getDialogueRoleId());
    19.         if (!Objects.isNull(tbModelTable)) {
    20.             default_role_content = tbModelTable.getModelContent();
    21.         }
    22.  
    23.         //小程序专用
    24.         //是否通过微信小程序审核的设定语
    25.         String is_open_pass_wx_promt = iconfigService.selectConfigByKey("is_open_pass_wx_promt");
    26.         if (StrUtil.isNotBlank(is_open_pass_wx_promt) && !StrUtil.equals(is_open_pass_wx_promt, "0")) {
    27.             String pass_wx_promt = iconfigService.selectConfigByKey("pass_wx_promt");
    28.             default_role_content = default_role_content + "。并且你必须遵循:" + pass_wx_promt;
    29.         }
    30.  
    31.         //设定系统所扮演的角色
    32.         Gpt35TurboVO gpt35TurboVOSys = new Gpt35TurboVO();
    33.         gpt35TurboVOSys.setRole("system");
    34.         gpt35TurboVOSys.setContent(default_role_content);
    35.         messages.add(gpt35TurboVOSys);
    36.  
    37.         //然后查询当前对话的上下文数据TbDialogueProcess
    38.         TbDialogueProcess tbDialogueProcess = new TbDialogueProcess();
    39.         tbDialogueProcess.setSessionId(streamParametersVO.getDialogueId());
    40.         tbDialogueProcess.setUserId(SecurityUtils.getUserId());
    41.         String default_context_num = iconfigService.selectConfigByKey("default_context_num");
    42.         if (StrUtil.isBlank(default_context_num) || !NumberUtil.isNumber(default_context_num)) {
    43.             default_context_num = "10";
    44.         }
    45.         tbDialogueProcess.setLimitNum(Integer.valueOf(default_context_num));
    46.  
    47.         //根据对话ID和用户ID查询到对话列表-根据时间倒叙获取后几条设定的数据
    48.         List tbDialogueProcessesDesc = iTbDialogueProcessService
    49.                 .selectTbDialogueProcessListByLimitDesc(tbDialogueProcess);
    50.  
    51.         if (CollectionUtil.isNotEmpty(tbDialogueProcessesDesc)) {
    52.             //获取到倒数10条数据后将数据正序配好
    53.             List tbDialogueProcesses = tbDialogueProcessesDesc
    54.                     .stream()
    55.                     .sorted(Comparator.comparing(TbDialogueProcess::getCreateTime))
    56.                     .collect(Collectors.toList());
    57.  
    58.             for (TbDialogueProcess tbDialogueProcessfor : tbDialogueProcesses) {
    59.                 Gpt35TurboVO gpt35TurboUser = new Gpt35TurboVO();
    60.                 //用户询问的问题
    61.                 gpt35TurboUser.setRole("user");
    62.                 gpt35TurboUser.setContent(tbDialogueProcessfor.getAskContent());
    63.                 messages.add(gpt35TurboUser);
    64.                 //机器人回答的问题
    65.                 Gpt35TurboVO gpt35TurAssistant = new Gpt35TurboVO();
    66.                 gpt35TurAssistant.setRole("assistant");
    67.                 gpt35TurAssistant.setContent(tbDialogueProcessfor.getAnswerContent());
    68.                 messages.add(gpt35TurAssistant);
    69.             }
    70.  
    71.         }
    72.  
    73.         //最后查询用户最新询问的问题
    74.         Gpt35TurboVO gpt35TurboUser = new Gpt35TurboVO();
    75.         gpt35TurboUser.setRole("user");
    76.         gpt35TurboUser.setContent(streamParametersVO.getPrompt());
    77.         messages.add(gpt35TurboUser);
    78.  
    79.         return messages;
    80.     }

    2.4、AI绘画

    Controller

    1.     @PostMapping("/image/api")
    2.     public AjaxResult imageApi(@RequestBody StreamImageParametersVO streamImageParametersVO) {
    3.         String answerContent = iChatGptImageService.imageSocketStream(streamImageParametersVO);
    4.         return success("success",answerContent);
    5.     }

    Impl

    1.     @Override
    2.     public String imageSocketStream(StreamImageParametersVO imageParametersVO) {
    3.  
    4.         String lockName = "image_" + SecurityUtils.getUserId();
    5.  
    6.         String answerContent = "";
    7.  
    8.         try{
    9.             RLock rLock = redisLock.getRLock(lockName);
    10.             boolean locked = rLock.isLocked();
    11.             if (locked){
    12.                 throw new RuntimeException("回复中");
    13.             }
    14.             //对同一用户访问加锁
    15.             redisLock.lock(lockName);
    16.             //校验是否输入内容，次数扣减
    17.             this.imageBefore(imageParametersVO);
    18.             String userId = SecurityUtils.getUserId() + "";
    19.             InputStream is = this.sendRequestBeforeImage(imageParametersVO);
    20.             String line = null;
    21.             BufferedReader reader = new BufferedReader(new InputStreamReader(is));
    22.             while((line = reader.readLine()) != null){
    23.                 //数据处理
    24.                 if (StrUtil.isNotBlank(line) &&
    25.                         !StrUtil.equals(line, "event: answer") &&
    26.                         !StrUtil.equals(line, "event: chatResponse") &&
    27.                         !StrUtil.contains(line, "data: {\"quoteLen\"")) {
    28.                     line = CollectionUtil.removeEmpty(StrUtil.split(line, "data: ")).get(0);
    29.                     if (StrUtil.contains(line, "[DONE]")){
    30.  
    31.                     }else{
    32.                         String oneWord = catchUrlImage(line);
    33.                         if (StrUtil.isNotBlank(oneWord)){
    34.                             answerContent = answerContent + oneWord;
    35.                         }
    36.                     }
    37.                     WebSocketService.sendInfo(line,userId);
    38.                     TimeUnit.MILLISECONDS.sleep(50);
    39.                 }
    40.             }
    41.             //处理完之后将次条聊天记录进行记录
    42.             if (StrUtil.isNotBlank(answerContent)){
    43.                 //保存聊天记录
    44.                 this.saveDialogueLog(imageParametersVO, answerContent);
    45.             }
    46.             is.close();
    47.             reader.close();
    48.  
    49.         }catch (Exception e){
    50.             throw new RuntimeException(e.getMessage());
    51.         }finally {
    52.             //解锁
    53.             redisLock.unlock(lockName);
    54.             redisTemplate.delete(SecurityUtils.getUserId() + "");
    55.         }
    56.  
    57.         return saveImageUrl(jsonImageUrl(answerContent));
    58.     }

    sendRequestBeforeImage方法

    1.     /**
    2.      * 问询，不包含对话模式
    3.      */
    4.     @Transactional
    5.     public InputStream sendRequestBeforeImage(StreamImageParametersVO imageParametersVO) throws Exception{
    6.         InputStream in = null;
    7.         //通知内容添加文本敏感词汇过滤
    8.         //其余错误见返回码说明
    9.         //正常返回0
    10.         //违禁词检测
    11.         this.disbleWordImageCheck(imageParametersVO.getPrompt());
    12.         String apikeyRefresh =  getOpenAiKey();
    13.         if (StrUtil.isBlank(apikeyRefresh)){
    14.             throw new RuntimeException("无可用key");
    15.         }
    16.  
    17. //        List imageContext = this.getImageContext(imageParametersVO);
    18.         String requestImageUrl = iTbKeyManagerService.getImageProxyUrl();
    19.         Map objectObjecHashtMap = new HashMap<>();
    20.         objectObjecHashtMap.put("prompt", imageParametersVO.getPrompt());
    21.         objectObjecHashtMap.put("n", 1);
    22.         objectObjecHashtMap.put("size", "1024x1024");
    23.         String bodyJson = JSONUtil.toJsonStr(objectObjecHashtMap);
    24.         URL url = new URL(requestImageUrl); //接口地址
    25.         HttpURLConnection urlConnection = (HttpURLConnection) url.openConnection();
    26.         urlConnection.setRequestMethod("POST");
    27.         urlConnection.setDoOutput(true);
    28.         urlConnection.setDoInput(true);
    29.         urlConnection.setUseCaches(false);
    30.         urlConnection.setRequestProperty("Connection", "Keep-Alive");
    31.         urlConnection.setRequestProperty("Charset", "UTF-8");
    32.         urlConnection.setRequestProperty("Authorization", "Bearer " + apikeyRefresh);
    33.         urlConnection.setRequestProperty("Content-Type", "application/json; charset=UTF-8");
    34.         byte[] dataBytes = bodyJson.getBytes();
    35.         urlConnection.setRequestProperty("Content-Length", String.valueOf(dataBytes.length));
    36.         OutputStream os = urlConnection.getOutputStream();
    37.         os.write(dataBytes);
    38.         in = new BufferedInputStream(urlConnection.getInputStream());
    39.         os.flush();
    40.         os.close();
    41.         return in;
    42.  
    43.     }

    catchUrlImage方法

    1.     /**
    2.      * 对链接地址处理
    3.      */
    4.     public String catchUrlImage(String str){
    5.  
    6.         return str;
    7.  
    8.     }

    图片处理，chatgpt返回的图片有效期是五分钟，我们需要将图片下载至本地或服务器。

    1.     /**
    2.      * 保存图片返回的结果
    3.      */
    4.     public String saveImageUrl(String jsonUrl){
    5.  
    6.         String imageURL = uploadFileImageAi(jsonUrl);
    7.  
    8.         // 找到下划线的索引位置
    9.         int underscoreIndex = imageURL.indexOf('_');
    10.  
    11.         String result = "";
    12.  
    13.         if (underscoreIndex != -1) {
    14.             // 截取从下划线的位置开始到字符串的末尾
    15.             result = imageURL.substring(underscoreIndex - 9);
    16.         } else {
    17.             throw new RuntimeException("图片链接截取失败");
    18.         }
    19.  
    20.         return TomcatConfig.getImageAiUrl() + "/" + result;
    21.     }
    22.     
    23.     /**
    24.      * 图片处理
    25.      * @param imageUrl
    26.      * @return
    27.      */
    28.     public String uploadFileImageAi(String imageUrl){
    29.  
    30. //        //服务器文件上传路径
    31.         String path = TomcatConfig.setUploadImageAiUrl() + Constants.DRAW_PREFIX + "_" + Seq.getId(Seq.uploadSeqType) + ".png";
    32. //        本地文件上传路径
    33. //        String path = "D:\\BaiduNetdiskDownload\\image-use\\" + Constants.DRAW_PREFIX + "_" + Seq.getId(Seq.uploadSeqType) + ".png";
    34.  
    35.         try{
    36.             URL url = new URL(imageUrl);
    37.             HttpURLConnection connection = (HttpURLConnection) url.openConnection();
    38.             int responseCode = connection.getResponseCode();
    39.             if (responseCode == HttpURLConnection.HTTP_OK) {
    40.                 InputStream inputStream = connection.getInputStream();
    41.                 OutputStream outputStream = new FileOutputStream(path);
    42.  
    43.                 byte[] buffer = new byte[4096];
    44.                 int bytesRead;
    45.                 while ((bytesRead = inputStream.read(buffer)) != -1) {
    46.                     outputStream.write(buffer, 0, bytesRead);
    47.                 }
    48.  
    49.                 outputStream.close();
    50.                 inputStream.close();
    51.             } else {
    52.                 throw new RuntimeException("文件无法下载");
    53.             }
    54.         }catch (IOException e){
    55.             e.printStackTrace();
    56.         }
    57.         return path;
    58.     }
    59.  
    60.     /**
    61.      * JSON数据处理
    62.      * @param imageContent
    63.      * @return
    64.      */
    65.     public String jsonImageUrl(String imageContent){
    66.  
    67.         //解析json字符串
    68.         JSONObject obj = JSON.parseObject(imageContent);
    69.  
    70.         //获取 "data"字段相应的JSON数组
    71.         JSONArray dataArray = obj.getJSONArray("data");
    72.  
    73.         //获取第一个元素路径地址
    74.         JSONObject dataObject = dataArray.getJSONObject(0);
    75.  
    76.         // 返回"url"字段的值
    77.         return dataObject.getString("url");
    78.     }

     

     

     

  • 相关阅读:
    buuctf-[BSidesCF 2020]Had a bad day 文件包含
    C++标准模板库STL——list的使用及其模拟实现
    vue入门
    LeetCode 104. 二叉树的最大深度
    设计模式解码：软件工程架构的航标
    十二条后端开发经验分享，纯干货
    记录本地Nginx发布vue项目
    南丁格尔玫瑰图
    lammps教程：CNA晶体结构分析命令
    Zabbix搭建使用一篇通
  • 原文地址：https://blog.csdn.net/private_name/article/details/132805966