style(Ai对话): 优化代码

src/main/java/cn/com/poc/ai_dialogues/aggregate/AiDialoguesService.java

@@ -17,6 +17,7 @@ public interface AiDialoguesService {
      *
      * @param type   对话类型
      * @param userId 用户ID
+     * @return 对话ID
      */
     String create(AiDialoguesTypeEnum type, Long userId) throws Exception;
 
@@ -26,6 +27,7 @@ public interface AiDialoguesService {
      * @param fileUrl 文件地址
      * @param type    对话类型
      * @param userId  用户ID
+     * @return 对话ID
      */
     String create(AiDialoguesTypeEnum type, String fileUrl, Long userId) throws Exception;
 
@@ -60,9 +62,7 @@ public interface AiDialoguesService {
     /**
      * 获取对话上下文
      *
-     * @param dialogueId
-     * @return
-     * @throws Exception
+     * @param dialogueId 对话ID
      */
     List<DialoguesContextDto> getDialogueContext(String dialogueId);
 
@@ -70,9 +70,10 @@ public interface AiDialoguesService {
     /**
      * 问题联想
      *
-     * @param question
-     * @return
+     * @param dialogueId 对话ID - 非必填
+     * @param question   问题文本
+     * @return 联想结果列表
      */
-    List<String> questionAssociation(String question);
+    List<String> questionAssociation(String dialogueId, String question);
 
 }

src/main/java/cn/com/poc/ai_dialogues/aggregate/impl/AiDialoguesServiceImpl.java

@@ -152,11 +152,11 @@ public class AiDialoguesServiceImpl implements AiDialoguesService {
                     }
             ).collect(Collectors.toList());
         }
-        return null;
+        return Collections.emptyList();
     }
 
     @Override
-    public List<String> questionAssociation(String question) {
+    public List<String> questionAssociation(String dialogueId, String question) {
         if (StringUtils.isBlank(question)) {
             logger.warn("question is blank");
             return Collections.emptyList();

src/main/java/cn/com/poc/ai_dialogues/constant/AiDialoguesTypeEnum.java

@@ -16,9 +16,9 @@ public enum AiDialoguesTypeEnum {
 
     ;
 
-    private String type;
+    private final String type;
 
-    private String desc;
+    private final String desc;
 
     AiDialoguesTypeEnum(String type, String desc) {
         this.type = type;

src/main/java/cn/com/poc/ai_dialogues/rest/AiDialoguesRest.java

@@ -26,6 +26,8 @@ public interface AiDialoguesRest extends BaseRest {
 
     /**
      * 获取对话列表
+     *
+     * @param type 对话类型  - AiDialoguesTypeEnum
      */
     List<AiDialoguesRecordDto> getList(@RequestParam String type) throws Exception;
 
@@ -45,13 +47,18 @@ public interface AiDialoguesRest extends BaseRest {
 
     /**
      * 获取对话上下文
+     *
+     * @param dialogueId 对话ID
      */
     List<DialoguesContextDto> getDialogueContext(@RequestParam String dialogueId) throws Exception;
 
 
     /**
      * 问题联想
+     *
+     * @param dialogueId 对话ID
+     * @param question   问题
      */
-    List<String> questionAssociation( @RequestParam String question);
+    List<String> questionAssociation(@RequestParam(required = false) String dialogueId, @RequestParam String question);
 
 }

src/main/java/cn/com/poc/ai_dialogues/rest/impl/AiDialoguesRestImpl.java

@@ -15,7 +15,6 @@ import cn.hutool.core.util.ObjectUtil;
 import org.springframework.stereotype.Component;
 
 import javax.annotation.Resource;
-import java.util.Collections;
 import java.util.List;
 
 @Component
@@ -75,7 +74,7 @@ public class AiDialoguesRestImpl implements AiDialoguesRest {
     }
 
     @Override
-    public List<String> questionAssociation(String question) {
-        return aiDialoguesService.questionAssociation(question);
+    public List<String> questionAssociation(String dialogueId, String question) {
+        return aiDialoguesService.questionAssociation(dialogueId,question);
     }
 }
\ No newline at end of file