BJ_ZTEITS-Developers / parking-cloud-yudao

Authored by cherishsince 2024-04-14 18:38:06 +0800

Showing 3 changed files with 102 additions and 0 deletions

yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/ChatConversationTypeEnum.java 0 → 100644

	1	+package cn.iocoder.yudao.module.ai.enums;
	2	+
	3	+import lombok.AllArgsConstructor;
	4	+import lombok.Getter;
	5	+
	6	+/**
	7	+ * 对话类型
	8	+ * 创建对话、继续对话
	9	+ *
	10	+ * @author fansili
	11	+ * @time 2024/4/14 18:15
	12	+ * @since 1.0
	13	+ */
	14	+@AllArgsConstructor
	15	+@Getter
	16	+public enum ChatConversationTypeEnum {
	17	+
	18	+ NEW("new", "新建对话"),
	19	+ CONTINUE("continue", "继续对话"),
	20	+
	21	+ ;
	22	+
	23	+ private String type;
	24	+
	25	+ private String name;
	26	+
	27	+ public static ChatConversationTypeEnum valueOfType(String type) {
	28	+ for (ChatConversationTypeEnum itemEnum : ChatConversationTypeEnum.values()) {
	29	+ if (itemEnum.getType().equals(type)) {
	30	+ return itemEnum;
	31	+ }
	32	+ }
	33	+ throw new IllegalArgumentException("Invalid MessageType value: " + type);
	34	+ }
	35	+
	36	+}
...	...

yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dataobject/AiChatConversationDO.java 0 → 100644

	1	+package cn.iocoder.yudao.module.ai.dataobject;
	2	+
	3	+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
	4	+import com.baomidou.mybatisplus.annotation.TableName;
	5	+import io.swagger.v3.oas.annotations.media.Schema;
	6	+import lombok.Data;
	7	+import lombok.experimental.Accessors;
	8	+
	9	+/**
	10	+ * chat 交谈
	11	+ *
	12	+ * @author fansili
	13	+ * @time 2024/4/14 17:35
	14	+ * @since 1.0
	15	+ */
	16	+@Data
	17	+@Accessors(chain = true)
	18	+@TableName("ai_chat_conversation")
	19	+public class AiChatConversationDO extends BaseDO {
	20	+
	21	+ @Schema(description = "id")
	22	+ private Long id;
	23	+
	24	+ @Schema(description = "用户id")
	25	+ private Long userId;
	26	+
	27	+ @Schema(description = "chat角色Id")
	28	+ private Long chatRoleId;
	29	+
	30	+ @Schema(description = "chat角色名称")
	31	+ private String chatRoleName;
	32	+
	33	+ @Schema(description = "对话标题(有程序自动生成)")
	34	+ private String chatTitle;
	35	+
	36	+ @Schema(description = "对话标题(有程序自动生成)")
	37	+ private Integer chatCount;
	38	+
	39	+ @Schema(description = "对话类型(roleChat、userChat)")
	40	+ private String chatType;
	41	+
	42	+}
...	...

yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/mapper/AiChatConversationMapper.java 0 → 100644

	1	+package cn.iocoder.yudao.module.ai.mapper;
	2	+
	3	+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
	4	+import cn.iocoder.yudao.module.ai.dataobject.AiChatConversationDO;
	5	+import org.apache.ibatis.annotations.Mapper;
	6	+import org.apache.ibatis.annotations.Param;
	7	+import org.apache.ibatis.annotations.Update;
	8	+import org.springframework.stereotype.Repository;
	9	+
	10	+/**
	11	+ * message mapper
	12	+ *
	13	+ * @fansili
	14	+ * @since v1.0
	15	+ */
	16	+@Repository
	17	+@Mapper
	18	+public interface AiChatConversationMapper extends BaseMapperX<AiChatConversationDO> {
	19	+
	20	+
	21	+ @Update("update ai_chat_conversation set chat_count = chat_count + 1 where id = #{id}")
	22	+ void updateIncrChatCount(@Param("id") Long id);
	23	+
	24	+}
...	...