BJ_ZTEITS-Developers / parking-cloud-yudao

Commit b091498040b989ebc5a2afb72539ee47d2430220

Authored by cherishsince
1 parent 3e91032e

【优化】优化AI 角色列表返回信息 和 修改模型id类型

Inline Side-by-side
Showing 4 changed files with 34 additions and 26 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/model/vo/role/AiChatRoleAddReqVO.java
  View file @b091498
... ... @@ -42,7 +42,7 @@ public class AiChatRoleAddReqVO {
42 42 private Long modelId;
43 43  
44 44 @NotNull
45   - @Schema(description = "开启状态 open、close")
  45 + @Schema(description = "开启状态 0、open 1、close")
46 46 private Boolean publicStatus;
47 47  
48 48 @NotNull
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/model/vo/role/AiChatRoleListRespVO.java
  View file @b091498
... ... @@ -14,24 +14,39 @@ import lombok.experimental.Accessors;
14 14 @Accessors(chain = true)
15 15 public class AiChatRoleListRespVO {
16 16  
17   - @Schema(description = "id")
  17 + @Schema(description = "编号", example = "1")
18 18 private Long id;
19 19  
20   - @Schema(description = "用户id")
21   - private Long userId;
22   -
23   - @Schema(description = "角色名字")
  20 + @Schema(description = "角色名称", example = "小红书写作")
24 21 private String name;
25 22  
26   - @Schema(description = "角色介绍，详细描述角色的功能或用途")
27   - private String introduce;
  23 + @Schema(description = "角色头像", example = "http://...")
  24 + private String avatar;
  25 +
  26 + @Schema(description = "角色分类", example = "writing")
  27 + private String category;
  28 +
  29 + @Schema(description = "角色描述", example = "角色描述")
  30 + private String description;
  31 +
  32 + @Schema(description = "角色欢迎语", example = "欢迎...")
  33 + private String welcomeMessage;
  34 +
  35 + @Schema(description = "角色设定（消息）", example = "你是拥有丰富的小红书写作经验作者xxxx")
  36 + private String systemMessage;
  37 +
  38 + @Schema(description = "用户编号", example = "1")
  39 + private Long userId;
  40 +
  41 + @Schema(description = "模型编号", example = "1")
  42 + private Long modelId;
28 43  
29   - @Schema(description = "分类，角色所属的类别，如娱乐、创作等")
30   - private String classify;
  44 + @Schema(description = "是否公开 true - 公开；false - 私有", example = "true")
  45 + private Boolean publicStatus;
31 46  
32   - @Schema(description = "状态 open、close")
33   - private String enable;
  47 + @Schema(description = "排序值 asc", example = "1")
  48 + private Integer sort;
34 49  
35   - @Schema(description = "角色的使用次数统计")
36   - private Integer useCount;
  50 + @Schema(description = "状态 0、开启 1、关闭", example = "1")
  51 + private Integer status;
37 52 }
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiChatRoleDO.java
  View file @b091498
1 1 package cn.iocoder.yudao.module.ai.dal.dataobject.model;
2 2  
3   -import cn.iocoder.yudao.framework.ai.chat.messages.MessageType;
4 3 import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
5   -import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
6 4 import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
7 5 import com.baomidou.mybatisplus.annotation.*;
8   -import com.baomidou.mybatisplus.extension.handlers.AbstractJsonTypeHandler;
9 6 import lombok.*;
10   -import lombok.experimental.Accessors;
11   -
12   -import java.io.Serializable;
13   -import java.util.List;
14 7  
15 8 /**
16 9 * AI 聊天角色 DO
... ... @@ -67,12 +60,12 @@ public class AiChatRoleDO extends BaseDO {
67 60 /**
68 61 * 模型编号
69 62 *
70   - * 关联 {@link AiChatModalDO#getId()} 字段
  63 + * 关联 {@link AiChatModelDO#getId()} 字段
71 64 */
72   - private String modelId;
  65 + private Long modelId;
73 66  
74 67 /**
75   - * 是否公开
  68 + * 是否公开 true - 公开；false - 私有
76 69 *
77 70 * true - 公开；false - 私有
78 71 */
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/AiChatRoleServiceImpl.java
  View file @b091498
... ... @@ -12,7 +12,7 @@ import cn.iocoder.yudao.module.ai.convert.AiChatRoleConvert;
12 12 import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatRoleDO;
13 13 import cn.iocoder.yudao.module.ai.dal.mysql.AiChatRoleMapper;
14 14 import cn.iocoder.yudao.module.ai.enums.AiChatRoleCategoryEnum;
15   -import cn.iocoder.yudao.module.ai.service.AiChatModalService;
  15 +import cn.iocoder.yudao.module.ai.service.AiChatModelService;
16 16 import cn.iocoder.yudao.module.ai.service.AiChatRoleService;
17 17 import lombok.AllArgsConstructor;
18 18 import lombok.extern.slf4j.Slf4j;
... ... @@ -32,7 +32,7 @@ import java.util.List;
32 32 public class AiChatRoleServiceImpl implements AiChatRoleService {
33 33  
34 34 private final AiChatRoleMapper aiChatRoleMapper;
35   - private final AiChatModalService aiChatModalService;
  35 + private final AiChatModelService aiChatModalService;
36 36  
37 37 @Override
38 38 public PageResult<AiChatRoleListRespVO> list(AiChatRoleListReqVO req) {
... ...