修改对话逻辑提交

src/main/java/com/rf/AIquantum/dao/dto/SseResultDataDto.java

@@ -0,0 +1,18 @@
+package com.rf.AIquantum.dao.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
+
+@JpaDto
+@Data
+@AllArgsConstructor
+@NoArgsConstructor
+public class SseResultDataDto {
+
+    private String dialogueId;
+
+    private String type;
+
+    private String content;
+}

src/main/java/com/rf/AIquantum/dialogue/dao/repository/DialogueRepository.java

@@ -28,4 +28,10 @@ public interface DialogueRepository extends BaseRepository<DialogueEntity, Strin
     @Query(value = "select * from t_dialogue_info where phone = ?1 " +
             " AND status = ?2 AND if( ?3 is not null and  ?3 != '', (dialogue_name like CONCAT('%',  ?3, '%')), 1 = 1) order by create_time desc",nativeQuery = true)
     Page<DialogueEntity> findByPhoneAndStatusAndSearch(String phone, int status, String search, PageRequest of);
+
+    @Modifying
+    @Query(value = " insert into t_dialogue_info(id,create_time,update_time,dialogue_name,phone,status) values " +
+            "(:#{#dialogueEntity.id},:#{#dialogueEntity.createTime},:#{#dialogueEntity.updateTime},:#{#dialogueEntity.dialogueName}," +
+            ":#{#dialogueEntity.phone},:#{#dialogueEntity.status}) ", nativeQuery = true)
+    void insert(@Param("dialogueEntity") DialogueEntity dialogueEntity);
 }

src/main/java/com/rf/AIquantum/dialogue/rest/ChatHistoryController.java

@@ -11,6 +11,7 @@ import com.rf.AIquantum.dialogue.service.ChatHistoryService;
 import com.rf.AIquantum.user.dao.model.UserEntity;
 import com.rf.AIquantum.user.service.UserService;
 import com.rf.AIquantum.utils.Constant;
+import com.rf.AIquantum.utils.FileUtil;
 import com.rf.AIquantum.utils.JWTUtil;
 import com.rf.AIquantum.utils.Result;
 import io.swagger.annotations.Api;
@@ -22,6 +23,8 @@ import org.springframework.util.DigestUtils;
 import org.springframework.web.bind.annotation.*;
 
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
 import java.io.UnsupportedEncodingException;
 import java.nio.charset.StandardCharsets;
 import java.util.Date;
@@ -117,4 +120,47 @@ public class ChatHistoryController extends BaseController {
         return success();
     }
 
+    @PostMapping("/exportPdf")
+    @ApiOperation(value = "导出PDF")
+    public String exportPdf(@RequestBody String json, HttpServletResponse response) {
+        List<ChatHistoryEntity> chatHistoryEntityList = JSONArray.parseArray(json,ChatHistoryEntity.class);
+        if (chatHistoryEntityList.size() > 0) {
+            return "导出内容为空";
+            /*UserRecordEntity userRecordEntity = this.userRecordService.getUserRecordById(id);
+            if (userRecordEntity == null) {
+                return "该记录异常，下载失败";
+            }
+            String PDFPath = generatePDFRecord(userRecordEntity, "pdf");
+            if (PDFPath.equals("false")) {
+                return "报告版本较早，暂无法下载";
+            }
+//            List<String> htmlFilePathList = new ArrayList<>();
+//            htmlFilePathList.add(PDFPath);
+//
+//            //生成PDF测试记录
+//            try {
+//                html2pdf(htmlFilePathList);
+//            } catch (Exception e) {
+//                log.error("生成测试文件失败" + e.getMessage());
+//                e.printStackTrace();
+//                return "下载失败";
+//            }
+//            String[] split = userRecordEntity.getFileName().split("/");
+//            String fileName = split[split.length - 1];
+//            String PDFName = fileName.replace(".xlsx", ".pdf");
+            //下载PDF文件
+            File z = new File("./h2p/export_pdf/PDFReport/" + PDFPath);
+            if (!z.exists()) return "下载失败";
+            try {
+                FileUtil.downloadFile(response, z, PDFPath, true);
+                return "下载成功";
+            } catch (UnsupportedEncodingException e) {
+                e.printStackTrace();
+                return "下载失败";
+            }*/
+        }else {
+            return "导出内容为空";
+        }
+    }
+
 }

src/main/java/com/rf/AIquantum/dialogue/rest/DialogueController.java

@@ -6,6 +6,7 @@ import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import com.auth0.jwt.interfaces.DecodedJWT;
 import com.rf.AIquantum.base.rest.BaseController;
+import com.rf.AIquantum.dao.dto.SseResultDataDto;
 import com.rf.AIquantum.dialogue.dao.model.ChatHistoryEntity;
 import com.rf.AIquantum.dialogue.dao.model.DialogueEntity;
 import com.rf.AIquantum.dialogue.service.ChatHistoryService;
@@ -89,19 +90,20 @@ public class DialogueController extends BaseController {
                 }
             }
         }
-        if (dialogueId == null || dialogueId.equals("")){
+        List<ChatHistoryEntity> chatHistoryEntities = chatHistoryService.findChatHistoryByDialogueIdAndStatus(dialogueId);
+        if (chatHistoryEntities.size() < 1 ){
             if (content.length() > 50){
                 content = content.substring(0,50);
             }
             //新建对话
             DialogueEntity dialogueEntity = new DialogueEntity();
+            dialogueEntity.setId(dialogueId);
             dialogueEntity.setDialogueName(content);
             dialogueEntity.setPhone(phone);
             dialogueEntity.setStatus(1);
             dialogueEntity.setCreateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss.SSS"));
             dialogueEntity.setUpdateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss.SSS"));
-            dialogueEntity = this.dialogueService.save(dialogueEntity);
-            dialogueId = dialogueEntity.getId();
+            this.dialogueService.insert(dialogueEntity);
         }
         ChatHistoryEntity chatHistoryEntity = new ChatHistoryEntity();
         chatHistoryEntity.setDialogueId(dialogueId);
@@ -114,7 +116,7 @@ public class DialogueController extends BaseController {
         chatHistoryEntity.setUpdateTime(DateUtil.format(new Date(), "yyyy-MM-dd HH:mm:ss.SSS"));
         this.chatHistoryService.save(chatHistoryEntity);
         //调用模型相关操作
-        /*List<ChatHistoryEntity> chatHistoryEntities = this.chatHistoryService.findChatHistoryByDialogueIdAndStatus(dialogueId);
+        /*chatHistoryEntities = this.chatHistoryService.findChatHistoryByDialogueIdAndStatus(dialogueId);
         JSONArray messages = new JSONArray();
         for (ChatHistoryEntity chatHistory : chatHistoryEntities) {
             JSONArray contents = new JSONArray();
@@ -147,12 +149,30 @@ public class DialogueController extends BaseController {
         }
         content = jsonSystem.getString("response");*/
         String sseContent = "";
-        content = "";
-        for (int i = 0; i < 10; i++) {
-            sseContent = "消息"+i;
+        content = "<think>";
+        SseResultDataDto sseResultDataDto = new SseResultDataDto();
+        sseResultDataDto.setDialogueId(dialogueId);
+        String i0 = " 我是深度思考的内容，我是深度思考的内容，深度思考的内容  # 占位回复\n\n当前服务尚未部署实际的语言模型。\n\n## 您的选择\n- **模型**: `default`\n- **上传的图片数量**: 0\n\n## 示例 Markdown 样式\n- **加粗**: `**加粗文本**`\n- *斜体*: `*斜体文本*`\n- ~~删除线~~: `~~删除线文本~~`\n- [超链接](https://www.bing.com): `[超链接](URL)`\n- 图片:\n\n![示例图片](https://vuejs.org/images/logo.png)\n\n## 列表示例\n1. 第一项\n2. 第二项\n   - 子项 1\n   - 子项 2\n\n> 引用示例：这是一个引用块。\n\n```python\n# 代码块示例\ndef hello():\n    print('Hello, World!')\n```\n\n- 行内公式: `$E = mc^2$` 显示为 $E = mc^2$\n- 块级公式:\n\n$$\n\\int_a^b f(x) \\ dx = F(b) - F(a)\n$$\n\n上述公式显示为：\n\n$$\n\\int_a^b f(x) \\ dx = F(b) - F(a)\n$$\n`";
+        String[] i1= i0.split("");
+        boolean think = true;
+        for (int i = 0; i < i1.length; i++) {
+            sseContent = i1[i];
+            if (think) {
+                if (i == 15) {
+                    content = content + "</think>";
+                    think = false;
+                    sseResultDataDto.setType("text");
+                }else {
+                    sseResultDataDto.setType("think");
+                }
+            }else {
+                sseResultDataDto.setType("text");
+            }
+            sseResultDataDto.setContent(sseContent);
+
             content = content + sseContent;
-            sseEmitterService.sendMessage(dialogueId,sseContent);
-            Thread.sleep(1000);
+            sseEmitterService.sendMessage(dialogueId, sseResultDataDto);
+            Thread.sleep(100);
         }
         chatHistoryEntity = new ChatHistoryEntity();
         chatHistoryEntity.setDialogueId(dialogueId);
@@ -245,6 +265,7 @@ public class DialogueController extends BaseController {
             response = httpclient.execute(httpPost);
             if (response.getStatusLine().getStatusCode() == 200){
                 responseData =  EntityUtils.toString(response.getEntity());
+                System.out.println("22:"+responseData);
             }
         } catch (IOException e) {
             e.printStackTrace();

src/main/java/com/rf/AIquantum/dialogue/service/DialogueService.java

@@ -16,4 +16,6 @@ public interface DialogueService {
     DialogueEntity save(DialogueEntity dialogueEntity);
 
     Page<DialogueEntity> findByPhoneAndStatusAndSearch(int pageNum, int pageSize, String phone, int status, String search);
+
+    void insert(DialogueEntity dialogueEntity);
 }

src/main/java/com/rf/AIquantum/dialogue/service/impl/DialogueServiceImpl.java

@@ -10,8 +10,11 @@ import com.rf.AIquantum.utils.Constant;
 import com.rf.AIquantum.utils.PageRequestUtil;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.domain.Page;
+import org.springframework.data.jpa.repository.Modifying;
+import org.springframework.data.jpa.repository.Query;
 import org.springframework.stereotype.Service;
 
+import javax.transaction.Transactional;
 import java.util.Date;
 import java.util.List;
 
@@ -36,4 +39,12 @@ public class DialogueServiceImpl implements DialogueService {
         Page<DialogueEntity> page = this.dialogueRepository.findByPhoneAndStatusAndSearch(phone,status,search, PageRequestUtil.of(pageNum, pageSize));
         return page;
     }
+
+    @Transactional
+    @Modifying
+    @Query
+    @Override
+    public void insert(DialogueEntity dialogueEntity) {
+        this.dialogueRepository.insert(dialogueEntity);
+    }
 }

src/main/java/com/rf/AIquantum/utils/SseEmitterService.java

@@ -5,6 +5,8 @@ package com.rf.AIquantum.utils;
  * @Date:2025/3/10:18:25
  * @Description:
  */
+import com.alibaba.fastjson.JSONObject;
+import com.rf.AIquantum.dao.dto.SseResultDataDto;
 import org.springframework.stereotype.Component;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 
@@ -24,7 +26,7 @@ public class SseEmitterService {
         emitters.remove(clientId);
     }
 
-    public void sendMessage(String clientId, Object message) {
+    public void sendMessage(String clientId, SseResultDataDto message) {
         SseEmitter emitter = emitters.get(clientId);
         if (emitter != null) {
             try {