devbox
正经夜光杯
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

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

作者：正经夜光杯 | 2024-08-06 13:19:16

赞

踩

springboot调用chatgpt

目录

一、说明

二、代码

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<Gpt35TurboVO> objects = new ArrayList<>();
  12.         objects.add(gpt35TurboVO);
  13.         Map<Object, Object> 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<Gpt35TurboVO> chatContext = this.getChatContext(chatBotSingleQuestionVO);
  21.         String requestUrl = iTbKeyManagerService.getproxyUrl();
  22.         Map<Object, Object> 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<Gpt35TurboVO> getChatContext(ChatBotSingleQuestionVO chatBotSingleQuestionVO) {
  7.  
  8.         List<Gpt35TurboVO> 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<TbKeyManager> 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<TbDialogueMain> 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<Gpt35TurboVO> chatContext = this.getChatDigContext(streamParametersVO, tbDialogueMain);
  42.  
  43.         String requestUrl = iTbKeyManagerService.getproxyUrl();
  44.         Map<Object, Object> 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<Gpt35TurboVO> getChatDigContext(StreamParametersVO streamParametersVO, TbDialogueMain tbDialogueMain) {
  8.  
  9.         List<Gpt35TurboVO> 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<TbDialogueProcess> tbDialogueProcessesDesc = iTbDialogueProcessService
  49.                 .selectTbDialogueProcessListByLimitDesc(tbDialogueProcess);
  50.  
  51.         if (CollectionUtil.isNotEmpty(tbDialogueProcessesDesc)) {
  52.             //获取到倒数10条数据后将数据正序配好
  53.             List<TbDialogueProcess> 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<Gpt35TurboVO> imageContext = this.getImageContext(imageParametersVO);
  18.         String requestImageUrl = iTbKeyManagerService.getImageProxyUrl();
  19.         Map<Object, Object> 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.     }

 

 

 

本文内容由网友自发贡献，转载请注明出处：【wpsshop博客】
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号