buttons;
+
+ /**
+ * 任务编号
+ *
+ * 1. midjourney proxy：关联的 task id
+ */
+ private String taskId;
+
+}
+
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeDO.java
new file mode 100644
index 0000000000..e1327a50ef
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeDO.java
@@ -0,0 +1,64 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.knowledge;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiModelDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+/**
+ * AI 知识库 DO
+ *
+ * @author xiaoxin
+ */
+@TableName(value = "ai_knowledge", autoResultMap = true)
+@KeySequence("ai_knowledge_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiKnowledgeDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 知识库名称
+ */
+ private String name;
+ /**
+ * 知识库描述
+ */
+ private String description;
+
+ /**
+ * 向量模型编号
+ *
+ * 关联 {@link AiModelDO#getId()}
+ */
+ private Long embeddingModelId;
+ /**
+ * 模型标识
+ *
+ * 冗余 {@link AiModelDO#getModel()}
+ */
+ private String embeddingModel;
+
+ /**
+ * topK
+ */
+ private Integer topK;
+ /**
+ * 相似度阈值
+ */
+ private Double similarityThreshold;
+
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeDocumentDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeDocumentDO.java
new file mode 100644
index 0000000000..ac014e926b
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeDocumentDO.java
@@ -0,0 +1,69 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.knowledge;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+/**
+ * AI 知识库-文档 DO
+ *
+ * @author xiaoxin
+ */
+@TableName(value = "ai_knowledge_document")
+@KeySequence("ai_knowledge_document_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiKnowledgeDocumentDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 知识库编号
+ *
+ * 关联 {@link AiKnowledgeDO#getId()}
+ */
+ private Long knowledgeId;
+ /**
+ * 文档名称
+ */
+ private String name;
+ /**
+ * 文件 URL
+ */
+ private String url;
+ /**
+ * 内容
+ */
+ private String content;
+ /**
+ * 文档长度
+ */
+ private Integer contentLength;
+
+ /**
+ * 文档 token 数量
+ */
+ private Integer tokens;
+ /**
+ * 分片最大 Token 数
+ */
+ private Integer segmentMaxTokens;
+
+ /**
+ * 召回次数
+ */
+ private Integer retrievalCount;
+
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeSegmentDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeSegmentDO.java
new file mode 100644
index 0000000000..cccbd6846b
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/knowledge/AiKnowledgeSegmentDO.java
@@ -0,0 +1,72 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.knowledge;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+/**
+ * AI 知识库-文档分段 DO
+ *
+ * @author xiaoxin
+ */
+@TableName(value = "ai_knowledge_segment")
+@KeySequence("ai_knowledge_segment_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiKnowledgeSegmentDO extends BaseDO {
+
+ /**
+ * 向量库的编号 - 空值
+ */
+ public static final String VECTOR_ID_EMPTY = "";
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 知识库编号
+ *
+ * 关联 {@link AiKnowledgeDO#getId()}
+ */
+ private Long knowledgeId;
+ /**
+ * 文档编号
+ *
+ * 关联 {@link AiKnowledgeDocumentDO#getId()}
+ */
+ private Long documentId;
+ /**
+ * 切片内容
+ */
+ private String content;
+ /**
+ * 切片内容长度
+ */
+ private Integer contentLength;
+
+ /**
+ * 向量库的编号
+ */
+ private String vectorId;
+ /**
+ * token 数量
+ */
+ private Integer tokens;
+
+ /**
+ * 召回次数
+ */
+ private Integer retrievalCount;
+
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/mindmap/AiMindMapDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/mindmap/AiMindMapDO.java
new file mode 100644
index 0000000000..db788b7e83
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/mindmap/AiMindMapDO.java
@@ -0,0 +1,66 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.mindmap;
+
+import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiModelDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+/**
+ * AI 思维导图 DO
+ *
+ * @author xiaoxin
+ */
+@TableName(value = "ai_mind_map")
+@KeySequence("ai_mind_map_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiMindMapDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+
+ /**
+ * 用户编号
+ *
+ * 关联 AdminUserDO 的 userId 字段
+ */
+ private Long userId;
+
+ /**
+ * 平台
+ *
+ * 枚举 {@link AiPlatformEnum}
+ */
+ private String platform;
+ /**
+ * 模型编号
+ *
+ * 关联 {@link AiModelDO#getId()}
+ */
+ private Long modelId;
+ /**
+ * 模型
+ */
+ private String model;
+
+ /**
+ * 生成内容提示
+ */
+ private String prompt;
+
+ /**
+ * 生成的内容
+ */
+ private String generatedContent;
+
+ /**
+ * 错误信息
+ */
+ private String errorMessage;
+
+}
\ No newline at end of file
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiApiKeyDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiApiKeyDO.java
new file mode 100644
index 0000000000..f2c683a503
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiApiKeyDO.java
@@ -0,0 +1,54 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.model;
+
+import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.*;
+
+/**
+ * AI API 秘钥 DO
+ *
+ * @author 芋道源码
+ */
+@TableName("ai_api_key")
+@KeySequence("ai_chat_conversation_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class AiApiKeyDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 名称
+ */
+ private String name;
+ /**
+ * 密钥
+ */
+ private String apiKey;
+ /**
+ * 平台
+ *
+ * 枚举 {@link AiPlatformEnum}
+ */
+ private String platform;
+ /**
+ * API 地址
+ */
+ private String url;
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiChatRoleDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiChatRoleDO.java
new file mode 100644
index 0000000000..bb6a3ca48d
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiChatRoleDO.java
@@ -0,0 +1,103 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.model;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.framework.mybatis.core.type.LongListTypeHandler;
+import cn.iocoder.yudao.module.ai.dal.dataobject.knowledge.AiKnowledgeDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.*;
+
+import java.util.List;
+
+/**
+ * AI 聊天角色 DO
+ *
+ * @author fansili
+ * @since 2024/4/24 19:39
+ */
+@TableName(value = "ai_chat_role", autoResultMap = true)
+@KeySequence("ai_chat_role_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class AiChatRoleDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 角色名称
+ */
+ private String name;
+ /**
+ * 角色头像
+ */
+ private String avatar;
+ /**
+ * 角色分类
+ */
+ private String category;
+ /**
+ * 角色描述
+ */
+ private String description;
+ /**
+ * 角色设定
+ */
+ private String systemMessage;
+
+ /**
+ * 用户编号
+ *
+ * 关联 AdminUserDO 的 userId 字段
+ */
+ private Long userId;
+
+ /**
+ * 模型编号
+ *
+ * 关联 {@link AiModelDO#getId()} 字段
+ */
+ private Long modelId;
+
+ /**
+ * 引用的知识库编号列表
+ *
+ * 关联 {@link AiKnowledgeDO#getId()} 字段
+ */
+ @TableField(typeHandler = LongListTypeHandler.class)
+ private List knowledgeIds;
+ /**
+ * 引用的工具编号列表
+ *
+ * 关联 {@link AiToolDO#getId()} 字段
+ */
+ @TableField(typeHandler = LongListTypeHandler.class)
+ private List toolIds;
+
+ /**
+ * 是否公开
+ *
+ * 1. true - 公开；由管理员在【角色管理】所创建
+ * 2. false - 私有；由个人在【我的角色】所创建
+ */
+ private Boolean publicStatus;
+
+ /**
+ * 排序值
+ */
+ private Integer sort;
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiModelDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiModelDO.java
new file mode 100644
index 0000000000..9e54f94c5a
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiModelDO.java
@@ -0,0 +1,88 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.model;
+
+import cn.iocoder.yudao.module.ai.enums.model.AiModelTypeEnum;
+import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.*;
+
+/**
+ * AI 模型 DO
+ *
+ * 默认模型：{@link #status} 为开启，并且 {@link #sort} 排序第一
+ *
+ * @author fansili
+ * @since 2024/4/24 19:39
+ */
+@TableName("ai_model")
+@KeySequence("ai_model_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class AiModelDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * API 秘钥编号
+ *
+ * 关联 {@link AiApiKeyDO#getId()}
+ */
+ private Long keyId;
+ /**
+ * 模型名称
+ */
+ private String name;
+ /**
+ * 模型标志
+ */
+ private String model;
+ /**
+ * 平台
+ *
+ * 枚举 {@link AiPlatformEnum}
+ */
+ private String platform;
+ /**
+ * 类型
+ *
+ * 枚举 {@link AiModelTypeEnum}
+ */
+ private Integer type;
+
+ /**
+ * 排序值
+ */
+ private Integer sort;
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+ // ========== 对话配置 ==========
+
+ /**
+ * 温度参数
+ *
+ * 用于调整生成回复的随机性和多样性程度：较低的温度值会使输出更收敛于高频词汇，较高的则增加多样性
+ */
+ private Double temperature;
+ /**
+ * 单条回复的最大 Token 数量
+ */
+ private Integer maxTokens;
+ /**
+ * 上下文的最大 Message 数量
+ */
+ private Integer maxContexts;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiToolDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiToolDO.java
new file mode 100644
index 0000000000..7773e978cc
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/model/AiToolDO.java
@@ -0,0 +1,48 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.model;
+
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.module.ai.service.model.tool.DirectoryListToolFunction;
+import cn.iocoder.yudao.module.ai.service.model.tool.WeatherQueryToolFunction;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.*;
+
+/**
+ * AI 工具 DO
+ *
+ * @author 芋道源码
+ */
+@TableName("ai_tool")
+@KeySequence("ai_tool_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+@Builder
+@NoArgsConstructor
+@AllArgsConstructor
+public class AiToolDO extends BaseDO {
+
+ /**
+ * 工具编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 工具名称
+ *
+ * 对应 Bean 的名字，例如说：
+ * 1. {@link DirectoryListToolFunction} 的 Bean 名字是 directory_list
+ * 2. {@link WeatherQueryToolFunction} 的 Bean 名字是 weather_query
+ */
+ private String name;
+ /**
+ * 工具描述
+ */
+ private String description;
+ /**
+ * 状态
+ *
+ * 枚举 {@link cn.iocoder.yudao.framework.common.enums.CommonStatusEnum}
+ */
+ private Integer status;
+
+}
\ No newline at end of file
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/music/AiMusicDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/music/AiMusicDO.java
new file mode 100644
index 0000000000..cc16133a70
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/music/AiMusicDO.java
@@ -0,0 +1,119 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.music;
+
+import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.module.ai.enums.music.AiMusicGenerateModeEnum;
+import cn.iocoder.yudao.module.ai.enums.music.AiMusicStatusEnum;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * AI 音乐 DO
+ *
+ * @author xiaoxin
+ */
+@TableName(value = "ai_music", autoResultMap = true)
+@KeySequence("ai_music_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiMusicDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+
+ /**
+ * 用户编号
+ *
+ * 关联 AdminUserDO 的 userId 字段
+ */
+ private Long userId;
+
+ /**
+ * 音乐名称
+ */
+ private String title;
+
+ /**
+ * 歌词
+ */
+ private String lyric;
+
+ /**
+ * 图片地址
+ */
+ private String imageUrl;
+ /**
+ * 音频地址
+ */
+ private String audioUrl;
+ /**
+ * 视频地址
+ */
+ private String videoUrl;
+
+ /**
+ * 音乐状态
+ *
+ * 枚举 {@link AiMusicStatusEnum}
+ */
+ private Integer status;
+
+ /**
+ * 生成模式
+ *
+ * 枚举 {@link AiMusicGenerateModeEnum}
+ */
+ private Integer generateMode;
+
+ /**
+ * 描述词
+ */
+ private String description;
+
+ /**
+ * 平台
+ *
+ * 枚举 {@link AiPlatformEnum}
+ */
+ private String platform;
+ // TODO @芋艿：modelId？
+ /**
+ * 模型
+ */
+ private String model;
+
+ /**
+ * 音乐风格标签
+ */
+ @TableField(typeHandler = JacksonTypeHandler.class)
+ private List tags;
+
+ /**
+ * 音乐时长
+ */
+ private Double duration;
+
+ /**
+ * 是否公开
+ */
+ private Boolean publicStatus;
+
+ /**
+ * 任务编号
+ */
+ private String taskId;
+
+ /**
+ * 错误信息
+ */
+ private String errorMessage;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/workflow/AiWorkflowDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/workflow/AiWorkflowDO.java
new file mode 100644
index 0000000000..d844f7da2e
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/workflow/AiWorkflowDO.java
@@ -0,0 +1,51 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.workflow;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+/**
+ * AI 工作流 DO
+ *
+ * @author lesan
+ */
+@TableName(value = "ai_workflow", autoResultMap = true)
+@KeySequence("ai_workflow") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiWorkflowDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+ /**
+ * 工作流名称
+ */
+ private String name;
+ /**
+ * 工作流标识
+ */
+ private String code;
+
+ /**
+ * 工作流模型 JSON 数据
+ */
+ private String graph;
+
+ /**
+ * 备注
+ */
+ private String remark;
+
+ /**
+ * 状态
+ *
+ * 枚举 {@link CommonStatusEnum}
+ */
+ private Integer status;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/write/AiWriteDO.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/write/AiWriteDO.java
new file mode 100644
index 0000000000..75e1e29cef
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/write/AiWriteDO.java
@@ -0,0 +1,104 @@
+package cn.iocoder.yudao.module.ai.dal.dataobject.write;
+
+import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiModelDO;
+import cn.iocoder.yudao.module.ai.enums.DictTypeConstants;
+import cn.iocoder.yudao.module.ai.enums.write.AiWriteTypeEnum;
+import com.baomidou.mybatisplus.annotation.KeySequence;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+
+/**
+ * AI 写作 DO
+ *
+ * @author xiaoxin
+ */
+@TableName("ai_write")
+@KeySequence("ai_write_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
+@Data
+public class AiWriteDO extends BaseDO {
+
+ /**
+ * 编号
+ */
+ @TableId
+ private Long id;
+
+ /**
+ * 用户编号
+ *
+ * 关联 AdminUserDO 的 userId 字段
+ */
+ private Long userId;
+
+ /**
+ * 写作类型
+ *
+ * 枚举 {@link AiWriteTypeEnum}
+ */
+ private Integer type;
+
+ /**
+ * 平台
+ *
+ * 枚举 {@link AiPlatformEnum}
+ */
+ private String platform;
+ /**
+ * 模型编号
+ *
+ * 关联 {@link AiModelDO#getId()}
+ */
+ private Long modelId;
+ /**
+ * 模型
+ */
+ private String model;
+
+ /**
+ * 生成内容提示
+ */
+ private String prompt;
+
+ /**
+ * 生成的内容
+ */
+ private String generatedContent;
+ /**
+ * 原文
+ */
+ private String originalContent;
+
+ /**
+ * 长度提示词
+ *
+ * 字典：{@link DictTypeConstants#AI_WRITE_LENGTH}
+ */
+ private Integer length;
+ /**
+ * 格式提示词
+ *
+ * 字典：{@link DictTypeConstants#AI_WRITE_FORMAT}
+ */
+ private Integer format;
+ /**
+ * 语气提示词
+ *
+ * 字典：{@link DictTypeConstants#AI_WRITE_TONE}
+ */
+ private Integer tone;
+ /**
+ * 语言提示词
+ *
+ * 字典：{@link DictTypeConstants#AI_WRITE_LANGUAGE}
+ */
+ private Integer language;
+
+ /**
+ * 错误信息
+ */
+ private String errorMessage;
+
+}
\ No newline at end of file
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/chat/AiChatConversationMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/chat/AiChatConversationMapper.java
new file mode 100644
index 0000000000..ce9da2f24d
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/chat/AiChatConversationMapper.java
@@ -0,0 +1,38 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.chat;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.chat.vo.conversation.AiChatConversationPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.chat.AiChatConversationDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+/**
+ * AI 聊天对话 Mapper
+ *
+ * @author 芋道源码
+ */
+@Mapper
+public interface AiChatConversationMapper extends BaseMapperX {
+
+ default List selectListByUserId(Long userId) {
+ return selectList(AiChatConversationDO::getUserId, userId);
+ }
+
+ default List selectListByUserIdAndPinned(Long userId, boolean pinned) {
+ return selectList(new LambdaQueryWrapperX()
+ .eq(AiChatConversationDO::getUserId, userId)
+ .eq(AiChatConversationDO::getPinned, pinned));
+ }
+
+ default PageResult selectChatConversationPage(AiChatConversationPageReqVO pageReqVO) {
+ return selectPage(pageReqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiChatConversationDO::getUserId, pageReqVO.getUserId())
+ .likeIfPresent(AiChatConversationDO::getTitle, pageReqVO.getTitle())
+ .betweenIfPresent(AiChatConversationDO::getCreateTime, pageReqVO.getCreateTime())
+ .orderByDesc(AiChatConversationDO::getId));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/chat/AiChatMessageMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/chat/AiChatMessageMapper.java
new file mode 100644
index 0000000000..5020f3944e
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/chat/AiChatMessageMapper.java
@@ -0,0 +1,59 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.chat;
+
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.core.map.MapUtil;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.common.util.collection.CollectionUtils;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.chat.vo.conversation.AiChatConversationPageReqVO;
+import cn.iocoder.yudao.module.ai.controller.admin.chat.vo.message.AiChatMessagePageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.chat.AiChatConversationDO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.chat.AiChatMessageDO;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * AI 聊天对话 Mapper
+ *
+ * @author fansili
+ */
+@Mapper
+public interface AiChatMessageMapper extends BaseMapperX {
+
+ default List selectListByConversationId(Long conversationId) {
+ return selectList(new LambdaQueryWrapperX()
+ .eq(AiChatMessageDO::getConversationId, conversationId)
+ .orderByAsc(AiChatMessageDO::getId));
+ }
+
+ default Map selectCountMapByConversationId(Collection conversationIds) {
+ // SQL count 查询
+ List> result = selectMaps(new QueryWrapper()
+ .select("COUNT(id) AS count, conversation_id AS conversationId")
+ .in("conversation_id", conversationIds)
+ .groupBy("conversation_id"));
+ if (CollUtil.isEmpty(result)) {
+ return Collections.emptyMap();
+ }
+ // 转换数据
+ return CollectionUtils.convertMap(result,
+ record -> MapUtil.getLong(record, "conversationId"),
+ record -> MapUtil.getInt(record, "count" ));
+ }
+
+ default PageResult selectPage(AiChatMessagePageReqVO pageReqVO) {
+ return selectPage(pageReqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiChatMessageDO::getConversationId, pageReqVO.getConversationId())
+ .eqIfPresent(AiChatMessageDO::getUserId, pageReqVO.getUserId())
+ .likeIfPresent(AiChatMessageDO::getContent, pageReqVO.getContent())
+ .betweenIfPresent(AiChatMessageDO::getCreateTime, pageReqVO.getCreateTime())
+ .orderByDesc(AiChatMessageDO::getId));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/image/AiImageMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/image/AiImageMapper.java
new file mode 100644
index 0000000000..f87c9472bc
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/image/AiImageMapper.java
@@ -0,0 +1,57 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.image;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImagePageReqVO;
+import cn.iocoder.yudao.module.ai.controller.admin.image.vo.AiImagePublicPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.image.AiImageDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+/**
+ * AI 绘图 Mapper
+ *
+ * @author fansili
+ */
+@Mapper
+public interface AiImageMapper extends BaseMapperX {
+
+ default AiImageDO selectByTaskId(String taskId) {
+ return selectOne(AiImageDO::getTaskId, taskId);
+ }
+
+ default PageResult selectPage(AiImagePageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiImageDO::getUserId, reqVO.getUserId())
+ .eqIfPresent(AiImageDO::getPlatform, reqVO.getPlatform())
+ .eqIfPresent(AiImageDO::getStatus, reqVO.getStatus())
+ .eqIfPresent(AiImageDO::getPublicStatus, reqVO.getPublicStatus())
+ .betweenIfPresent(AiImageDO::getCreateTime, reqVO.getCreateTime())
+ .orderByDesc(AiImageDO::getId));
+ }
+
+ default PageResult selectPageMy(Long userId, AiImagePageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiImageDO::getPrompt, reqVO.getPrompt())
+ // 情况一：公开
+ .eq(Boolean.TRUE.equals(reqVO.getPublicStatus()), AiImageDO::getPublicStatus, reqVO.getPublicStatus())
+ // 情况二：私有
+ .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiImageDO::getUserId, userId)
+ .orderByDesc(AiImageDO::getId));
+ }
+
+ default PageResult selectPage(AiImagePublicPageReqVO pageReqVO) {
+ return selectPage(pageReqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiImageDO::getPublicStatus, Boolean.TRUE)
+ .likeIfPresent(AiImageDO::getPrompt, pageReqVO.getPrompt())
+ .orderByDesc(AiImageDO::getId));
+ }
+
+ default List selectListByStatusAndPlatform(Integer status, String platform) {
+ return selectList(AiImageDO::getStatus, status,
+ AiImageDO::getPlatform, platform);
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeDocumentMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeDocumentMapper.java
new file mode 100644
index 0000000000..55f04bb328
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeDocumentMapper.java
@@ -0,0 +1,43 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.knowledge;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.knowledge.vo.document.AiKnowledgeDocumentPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.knowledge.AiKnowledgeDocumentDO;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * AI 知识库文档 Mapper
+ *
+ * @author xiaoxin
+ */
+@Mapper
+public interface AiKnowledgeDocumentMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiKnowledgeDocumentPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiKnowledgeDocumentDO::getKnowledgeId, reqVO.getKnowledgeId())
+ .likeIfPresent(AiKnowledgeDocumentDO::getName, reqVO.getName())
+ .orderByDesc(AiKnowledgeDocumentDO::getId));
+ }
+
+ default void updateRetrievalCountIncr(Collection ids) {
+ update(new LambdaUpdateWrapper()
+ .setSql(" retrieval_count = retrieval_count + 1")
+ .in(AiKnowledgeDocumentDO::getId, ids));
+ }
+
+ default List selectListByStatus(Integer status) {
+ return selectList(AiKnowledgeDocumentDO::getStatus, status);
+ }
+
+ default List selectListByKnowledgeId(Long knowledgeId) {
+ return selectList(AiKnowledgeDocumentDO::getKnowledgeId, knowledgeId);
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeMapper.java
new file mode 100644
index 0000000000..3433c0b973
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeMapper.java
@@ -0,0 +1,32 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.knowledge;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.knowledge.vo.knowledge.AiKnowledgePageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.knowledge.AiKnowledgeDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+/**
+ * AI 知识库 Mapper
+ *
+ * @author xiaoxin
+ */
+@Mapper
+public interface AiKnowledgeMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiKnowledgePageReqVO pageReqVO) {
+ return selectPage(pageReqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiKnowledgeDO::getName, pageReqVO.getName())
+ .eqIfPresent(AiKnowledgeDO::getStatus, pageReqVO.getStatus())
+ .betweenIfPresent(AiKnowledgeDO::getCreateTime, pageReqVO.getCreateTime())
+ .orderByDesc(AiKnowledgeDO::getId));
+ }
+
+ default List selectListByStatus(Integer status) {
+ return selectList(AiKnowledgeDO::getStatus, status);
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeSegmentMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeSegmentMapper.java
new file mode 100644
index 0000000000..1b9ca867f5
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/knowledge/AiKnowledgeSegmentMapper.java
@@ -0,0 +1,67 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.knowledge;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.MPJLambdaWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.knowledge.vo.segment.AiKnowledgeSegmentPageReqVO;
+import cn.iocoder.yudao.module.ai.controller.admin.knowledge.vo.segment.AiKnowledgeSegmentProcessRespVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.knowledge.AiKnowledgeSegmentDO;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
+import com.github.yulichang.wrapper.MPJLambdaWrapper;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * AI 知识库分片 Mapper
+ *
+ * @author xiaoxin
+ */
+@Mapper
+public interface AiKnowledgeSegmentMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiKnowledgeSegmentPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .eq(AiKnowledgeSegmentDO::getDocumentId, reqVO.getDocumentId())
+ .likeIfPresent(AiKnowledgeSegmentDO::getContent, reqVO.getContent())
+ .eqIfPresent(AiKnowledgeSegmentDO::getStatus, reqVO.getStatus())
+ .orderByDesc(AiKnowledgeSegmentDO::getId));
+ }
+
+ default List selectListByVectorIds(List vectorIds) {
+ return selectList(new LambdaQueryWrapperX()
+ .in(AiKnowledgeSegmentDO::getVectorId, vectorIds)
+ .orderByDesc(AiKnowledgeSegmentDO::getId));
+ }
+
+ default List selectListByDocumentId(Long documentId) {
+ return selectList(new LambdaQueryWrapperX()
+ .eq(AiKnowledgeSegmentDO::getDocumentId, documentId)
+ .orderByDesc(AiKnowledgeSegmentDO::getId));
+ }
+
+ default List selectListByKnowledgeIdAndStatus(Long knowledgeId, Integer status) {
+ return selectList(AiKnowledgeSegmentDO::getKnowledgeId, knowledgeId,
+ AiKnowledgeSegmentDO::getStatus, status);
+ }
+
+ default List selectProcessList(Collection documentIds) {
+ MPJLambdaWrapper wrapper = new MPJLambdaWrapperX()
+ .selectAs(AiKnowledgeSegmentDO::getDocumentId, AiKnowledgeSegmentProcessRespVO::getDocumentId)
+ .selectCount(AiKnowledgeSegmentDO::getId, "count")
+ .select("COUNT(CASE WHEN vector_id > '" + AiKnowledgeSegmentDO.VECTOR_ID_EMPTY
+ + "' THEN 1 ELSE NULL END) AS embeddingCount")
+ .in(AiKnowledgeSegmentDO::getDocumentId, documentIds)
+ .groupBy(AiKnowledgeSegmentDO::getDocumentId);
+ return selectJoinList(AiKnowledgeSegmentProcessRespVO.class, wrapper);
+ }
+
+ default void updateRetrievalCountIncrByIds(List ids) {
+ update(new LambdaUpdateWrapper()
+ .setSql(" retrieval_count = retrieval_count + 1")
+ .in(AiKnowledgeSegmentDO::getId, ids));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/mindmap/AiMindMapMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/mindmap/AiMindMapMapper.java
new file mode 100644
index 0000000000..0292ef473d
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/mindmap/AiMindMapMapper.java
@@ -0,0 +1,26 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.mindmap;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.mindmap.vo.AiMindMapPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.mindmap.AiMindMapDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * AI 思维导图 Mapper
+ *
+ * @author xiaoxin
+ */
+@Mapper
+public interface AiMindMapMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiMindMapPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiMindMapDO::getUserId, reqVO.getUserId())
+ .eqIfPresent(AiMindMapDO::getPrompt, reqVO.getPrompt())
+ .betweenIfPresent(AiMindMapDO::getCreateTime, reqVO.getCreateTime())
+ .orderByDesc(AiMindMapDO::getId));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiApiKeyMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiApiKeyMapper.java
new file mode 100644
index 0000000000..0a2efe36f8
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiApiKeyMapper.java
@@ -0,0 +1,35 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.model;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.QueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeyPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiApiKeyDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * AI API 密钥 Mapper
+ *
+ * @author 芋道源码
+ */
+@Mapper
+public interface AiApiKeyMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiApiKeyPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiApiKeyDO::getName, reqVO.getName())
+ .eqIfPresent(AiApiKeyDO::getPlatform, reqVO.getPlatform())
+ .eqIfPresent(AiApiKeyDO::getStatus, reqVO.getStatus())
+ .orderByDesc(AiApiKeyDO::getId));
+ }
+
+ default AiApiKeyDO selectFirstByPlatformAndStatus(String platform, Integer status) {
+ return selectOne(new QueryWrapperX()
+ .eq("platform", platform)
+ .eq("status", status)
+ .limitN(1)
+ .orderByAsc("id"));
+ }
+
+}
\ No newline at end of file
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiChatMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiChatMapper.java
new file mode 100644
index 0000000000..bfe2caf52a
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiChatMapper.java
@@ -0,0 +1,47 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.model;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.QueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.model.vo.model.AiModelPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiModelDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import javax.annotation.Nullable;
+import java.util.List;
+
+/**
+ * API 模型 Mapper
+ *
+ * @author fansili
+ */
+@Mapper
+public interface AiChatMapper extends BaseMapperX {
+
+ default AiModelDO selectFirstByStatus(Integer type, Integer status) {
+ return selectOne(new QueryWrapperX()
+ .eq("type", type)
+ .eq("status", status)
+ .limitN(1)
+ .orderByAsc("sort"));
+ }
+
+ default PageResult selectPage(AiModelPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiModelDO::getName, reqVO.getName())
+ .eqIfPresent(AiModelDO::getModel, reqVO.getModel())
+ .eqIfPresent(AiModelDO::getPlatform, reqVO.getPlatform())
+ .orderByAsc(AiModelDO::getSort));
+ }
+
+ default List selectListByStatusAndType(Integer status, Integer type,
+ @Nullable String platform) {
+ return selectList(new LambdaQueryWrapperX()
+ .eq(AiModelDO::getStatus, status)
+ .eq(AiModelDO::getType, type)
+ .eqIfPresent(AiModelDO::getPlatform, platform)
+ .orderByAsc(AiModelDO::getSort));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiChatRoleMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiChatRoleMapper.java
new file mode 100644
index 0000000000..ed91edf3fa
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiChatRoleMapper.java
@@ -0,0 +1,54 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.model;
+
+import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.model.vo.chatRole.AiChatRolePageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiChatRoleDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+/**
+ * AI 聊天角色 Mapper
+ *
+ * @author fansili
+ */
+@Mapper
+public interface AiChatRoleMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiChatRolePageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiChatRoleDO::getName, reqVO.getName())
+ .eqIfPresent(AiChatRoleDO::getCategory, reqVO.getCategory())
+ .eqIfPresent(AiChatRoleDO::getPublicStatus, reqVO.getPublicStatus())
+ .orderByAsc(AiChatRoleDO::getSort));
+ }
+
+ default PageResult selectPageByMy(AiChatRolePageReqVO reqVO, Long userId) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiChatRoleDO::getName, reqVO.getName())
+ .eqIfPresent(AiChatRoleDO::getCategory, reqVO.getCategory())
+ // 情况一：公开
+ .eq(Boolean.TRUE.equals(reqVO.getPublicStatus()), AiChatRoleDO::getPublicStatus, reqVO.getPublicStatus())
+ // 情况二：私有
+ .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiChatRoleDO::getUserId, userId)
+ .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiChatRoleDO::getStatus, CommonStatusEnum.ENABLE.getStatus())
+ .orderByAsc(AiChatRoleDO::getSort));
+ }
+
+ default List selectListGroupByCategory(Integer status) {
+ return selectList(new LambdaQueryWrapperX()
+ .select(AiChatRoleDO::getCategory)
+ .eq(AiChatRoleDO::getStatus, status)
+ .groupBy(AiChatRoleDO::getCategory));
+ }
+
+ default List selectListByName(String name) {
+ return selectList(new LambdaQueryWrapperX()
+ .likeIfPresent(AiChatRoleDO::getName, name)
+ .orderByAsc(AiChatRoleDO::getSort));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiToolMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiToolMapper.java
new file mode 100644
index 0000000000..d5d296692a
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/model/AiToolMapper.java
@@ -0,0 +1,35 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.model;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.model.vo.tool.AiToolPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.model.AiToolDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+/**
+ * AI 工具 Mapper
+ *
+ * @author 芋道源码
+ */
+@Mapper
+public interface AiToolMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiToolPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .likeIfPresent(AiToolDO::getName, reqVO.getName())
+ .eqIfPresent(AiToolDO::getDescription, reqVO.getDescription())
+ .eqIfPresent(AiToolDO::getStatus, reqVO.getStatus())
+ .betweenIfPresent(AiToolDO::getCreateTime, reqVO.getCreateTime())
+ .orderByDesc(AiToolDO::getId));
+ }
+
+ default List selectListByStatus(Integer status) {
+ return selectList(new LambdaQueryWrapperX()
+ .eq(AiToolDO::getStatus, status)
+ .orderByDesc(AiToolDO::getId));
+ }
+
+}
\ No newline at end of file
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/music/AiMusicMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/music/AiMusicMapper.java
new file mode 100644
index 0000000000..025f5e0185
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/music/AiMusicMapper.java
@@ -0,0 +1,44 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.music;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.music.vo.AiMusicPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.music.AiMusicDO;
+import org.apache.ibatis.annotations.Mapper;
+
+import java.util.List;
+
+/**
+ * AI 音乐 Mapper
+ *
+ * @author xiaoxin
+ */
+@Mapper
+public interface AiMusicMapper extends BaseMapperX {
+
+ default List selectListByStatus(Integer status) {
+ return selectList(AiMusicDO::getStatus, status);
+ }
+
+ default PageResult selectPage(AiMusicPageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiMusicDO::getUserId, reqVO.getUserId())
+ .eqIfPresent(AiMusicDO::getTitle, reqVO.getTitle())
+ .eqIfPresent(AiMusicDO::getStatus, reqVO.getStatus())
+ .eqIfPresent(AiMusicDO::getGenerateMode, reqVO.getGenerateMode())
+ .betweenIfPresent(AiMusicDO::getCreateTime, reqVO.getCreateTime())
+ .eqIfPresent(AiMusicDO::getPublicStatus, reqVO.getPublicStatus())
+ .orderByDesc(AiMusicDO::getId));
+ }
+
+ default PageResult selectPageByMy(AiMusicPageReqVO reqVO, Long userId) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ // 情况一：公开
+ .eq(Boolean.TRUE.equals(reqVO.getPublicStatus()), AiMusicDO::getPublicStatus, reqVO.getPublicStatus())
+ // 情况二：私有
+ .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiMusicDO::getUserId, userId)
+ .orderByAsc(AiMusicDO::getId));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/workflow/AiWorkflowMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/workflow/AiWorkflowMapper.java
new file mode 100644
index 0000000000..3770dbf0b5
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/workflow/AiWorkflowMapper.java
@@ -0,0 +1,30 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.workflow;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.workflow.vo.AiWorkflowPageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.workflow.AiWorkflowDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * AI 工作流 Mapper
+ *
+ * @author lesan
+ */
+@Mapper
+public interface AiWorkflowMapper extends BaseMapperX {
+
+ default AiWorkflowDO selectByCode(String code) {
+ return selectOne(AiWorkflowDO::getCode, code);
+ }
+
+ default PageResult selectPage(AiWorkflowPageReqVO pageReqVO) {
+ return selectPage(pageReqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiWorkflowDO::getStatus, pageReqVO.getStatus())
+ .likeIfPresent(AiWorkflowDO::getName, pageReqVO.getName())
+ .likeIfPresent(AiWorkflowDO::getCode, pageReqVO.getCode())
+ .betweenIfPresent(AiWorkflowDO::getCreateTime, pageReqVO.getCreateTime()));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/write/AiWriteMapper.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/write/AiWriteMapper.java
new file mode 100644
index 0000000000..c4983eb444
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/dal/mysql/write/AiWriteMapper.java
@@ -0,0 +1,27 @@
+package cn.iocoder.yudao.module.ai.dal.mysql.write;
+
+import cn.iocoder.yudao.framework.common.pojo.PageResult;
+import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
+import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
+import cn.iocoder.yudao.module.ai.controller.admin.write.vo.AiWritePageReqVO;
+import cn.iocoder.yudao.module.ai.dal.dataobject.write.AiWriteDO;
+import org.apache.ibatis.annotations.Mapper;
+
+/**
+ * AI 写作 Mapper
+ *
+ * @author xiaoxin
+ */
+@Mapper
+public interface AiWriteMapper extends BaseMapperX {
+
+ default PageResult selectPage(AiWritePageReqVO reqVO) {
+ return selectPage(reqVO, new LambdaQueryWrapperX()
+ .eqIfPresent(AiWriteDO::getUserId, reqVO.getUserId())
+ .eqIfPresent(AiWriteDO::getType, reqVO.getType())
+ .eqIfPresent(AiWriteDO::getPlatform, reqVO.getPlatform())
+ .betweenIfPresent(AiWriteDO::getCreateTime, reqVO.getCreateTime())
+ .orderByDesc(AiWriteDO::getId));
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/AiChatRoleEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/AiChatRoleEnum.java
new file mode 100644
index 0000000000..1479274959
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/AiChatRoleEnum.java
@@ -0,0 +1,50 @@
+package cn.iocoder.yudao.module.ai.enums;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+/**
+ * AI 内置聊天角色的枚举
+ *
+ * @author xiaoxin
+ */
+@AllArgsConstructor
+@Getter
+public enum AiChatRoleEnum {
+
+ AI_WRITE_ROLE("写作助手", """
+ 你是一位出色的写作助手，能够帮助用户生成创意和灵感，并在用户提供场景和提示词时生成对应的回复。你的任务包括：
+ 1. 撰写建议：根据用户提供的主题或问题，提供详细的写作建议、情节发展方向、角色设定以及背景描写，确保内容结构清晰、有逻辑。
+ 2. 回复生成：根据用户提供的场景和提示词，生成合适的对话或文字回复，确保语气和风格符合场景需求。
+ 除此之外不需要除了正文内容外的其他回复，如标题、开头、任何解释性语句或道歉。
+ """),
+
+ AI_MIND_MAP_ROLE("导图助手", """
+ 你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。markdown 只需要输出一级标题，二级标题，三级标题，四级标题，最多输出四级，除此之外不要输出任何其他 markdown 标记。下面是一个合格的例子：
+ # Geek-AI 助手
+ ## 完整的开源系统
+ ### 前端开源
+ ### 后端开源
+ ## 支持各种大模型
+ ### OpenAI
+ ### Azure
+ ### 文心一言
+ ### 通义千问
+ ## 集成多种收费方式
+ ### 支付宝
+ ### 微信
+ 除此之外不要任何解释性语句。
+ """),
+ ;
+
+ /**
+ * 角色名
+ */
+ private final String name;
+
+ /**
+ * 角色设定
+ */
+ private final String systemMessage;
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/DictTypeConstants.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/DictTypeConstants.java
new file mode 100644
index 0000000000..73782a2cbb
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/DictTypeConstants.java
@@ -0,0 +1,16 @@
+package cn.iocoder.yudao.module.ai.enums;
+
+/**
+ * AI 字典类型的枚举类
+ *
+ * @author xiaoxin
+ */
+public interface DictTypeConstants {
+
+ // ========== AI Write ==========
+ String AI_WRITE_FORMAT = "ai_write_format"; // 写作格式
+ String AI_WRITE_LENGTH = "ai_write_length"; // 写作长度
+ String AI_WRITE_LANGUAGE = "ai_write_language"; // 写作语言
+ String AI_WRITE_TONE = "ai_write_tone"; // 写作语气
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/ErrorCodeConstants.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/ErrorCodeConstants.java
new file mode 100644
index 0000000000..8a8a388324
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/ErrorCodeConstants.java
@@ -0,0 +1,68 @@
+package cn.iocoder.yudao.module.ai.enums;
+
+import cn.iocoder.yudao.framework.common.exception.ErrorCode;
+
+/**
+ * AI 错误码枚举类
+ *
+ * ai 系统，使用 1-040-000-000 段
+ */
+public interface ErrorCodeConstants {
+
+ // ========== API 密钥 1-040-000-000 ==========
+ ErrorCode API_KEY_NOT_EXISTS = new ErrorCode(1_040_000_000, "API 密钥不存在");
+ ErrorCode API_KEY_DISABLE = new ErrorCode(1_040_000_001, "API 密钥已禁用！");
+
+ // ========== API 模型 1-040-001-000 ==========
+ ErrorCode MODEL_NOT_EXISTS = new ErrorCode(1_040_001_000, "模型不存在!");
+ ErrorCode MODEL_DISABLE = new ErrorCode(1_040_001_001, "模型({})已禁用!");
+ ErrorCode MODEL_DEFAULT_NOT_EXISTS = new ErrorCode(1_040_001_002, "操作失败，找不到默认模型");
+ ErrorCode MODEL_USE_TYPE_ERROR = new ErrorCode(1_040_001_003, "操作失败，该模型的模型类型不正确");
+
+ // ========== API 聊天角色 1-040-002-000 ==========
+ ErrorCode CHAT_ROLE_NOT_EXISTS = new ErrorCode(1_040_002_000, "聊天角色不存在");
+ ErrorCode CHAT_ROLE_DISABLE = new ErrorCode(1_040_001_001, "聊天角色({})已禁用!");
+
+ // ========== API 聊天会话 1-040-003-000 ==========
+ ErrorCode CHAT_CONVERSATION_NOT_EXISTS = new ErrorCode(1_040_003_000, "对话不存在!");
+ ErrorCode CHAT_CONVERSATION_MODEL_ERROR = new ErrorCode(1_040_003_001, "操作失败，该聊天模型的配置不完整");
+
+ // ========== API 聊天消息 1-040-004-000 ==========
+ ErrorCode CHAT_MESSAGE_NOT_EXIST = new ErrorCode(1_040_004_000, "消息不存在!");
+ ErrorCode CHAT_STREAM_ERROR = new ErrorCode(1_040_004_001, "对话生成异常!");
+
+ // ========== API 绘画 1-040-005-000 ==========
+ ErrorCode IMAGE_NOT_EXISTS = new ErrorCode(1_040_005_000, "图片不存在!");
+ ErrorCode IMAGE_MIDJOURNEY_SUBMIT_FAIL = new ErrorCode(1_040_005_001, "Midjourney 提交失败!原因：{}");
+ ErrorCode IMAGE_CUSTOM_ID_NOT_EXISTS = new ErrorCode(1_040_005_002, "Midjourney 按钮 customId 不存在! {}");
+
+ // ========== API 音乐 1-040-006-000 ==========
+ ErrorCode MUSIC_NOT_EXISTS = new ErrorCode(1_040_006_000, "音乐不存在!");
+
+ // ========== API 写作 1-040-007-000 ==========
+ ErrorCode WRITE_NOT_EXISTS = new ErrorCode(1_040_007_000, "作文不存在!");
+ ErrorCode WRITE_STREAM_ERROR = new ErrorCode(1_040_07_001, "写作生成异常!");
+
+ // ========== API 思维导图 1-040-008-000 ==========
+ ErrorCode MIND_MAP_NOT_EXISTS = new ErrorCode(1_040_008_000, "思维导图不存在!");
+
+ // ========== API 知识库 1-040-009-000 ==========
+ ErrorCode KNOWLEDGE_NOT_EXISTS = new ErrorCode(1_040_009_000, "知识库不存在!");
+
+ ErrorCode KNOWLEDGE_DOCUMENT_NOT_EXISTS = new ErrorCode(1_040_009_101, "文档不存在!");
+ ErrorCode KNOWLEDGE_DOCUMENT_FILE_EMPTY = new ErrorCode(1_040_009_102, "文档内容为空!");
+ ErrorCode KNOWLEDGE_DOCUMENT_FILE_DOWNLOAD_FAIL = new ErrorCode(1_040_009_102, "文件下载失败!");
+ ErrorCode KNOWLEDGE_DOCUMENT_FILE_READ_FAIL = new ErrorCode(1_040_009_102, "文档加载失败!");
+
+ ErrorCode KNOWLEDGE_SEGMENT_NOT_EXISTS = new ErrorCode(1_040_009_202, "段落不存在!");
+ ErrorCode KNOWLEDGE_SEGMENT_CONTENT_TOO_LONG = new ErrorCode(1_040_009_203, "内容 Token 数为 {}，超过最大限制 {}");
+
+ // ========== AI 工具 1-040-010-000 ==========
+ ErrorCode TOOL_NOT_EXISTS = new ErrorCode(1_040_010_000, "工具不存在");
+ ErrorCode TOOL_NAME_NOT_EXISTS = new ErrorCode(1_040_010_001, "工具({})找不到 Bean");
+
+ // ========== AI 工作流 1-040-011-000 ==========
+ ErrorCode WORKFLOW_NOT_EXISTS = new ErrorCode(1_040_011_000, "工作流不存在");
+ ErrorCode WORKFLOW_CODE_EXISTS = new ErrorCode(1_040_011_001, "工作流标识已存在");
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/image/AiImageStatusEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/image/AiImageStatusEnum.java
new file mode 100644
index 0000000000..cf80761503
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/image/AiImageStatusEnum.java
@@ -0,0 +1,37 @@
+package cn.iocoder.yudao.module.ai.enums.image;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+/**
+ * AI 绘画状态的枚举
+ *
+ * @author fansili
+ */
+@AllArgsConstructor
+@Getter
+public enum AiImageStatusEnum {
+
+ IN_PROGRESS(10, "进行中"),
+ SUCCESS(20, "已完成"),
+ FAIL(30, "已失败");
+
+ /**
+ * 状态
+ */
+ private final Integer status;
+ /**
+ * 状态名
+ */
+ private final String name;
+
+ public static AiImageStatusEnum valueOfStatus(Integer status) {
+ for (AiImageStatusEnum statusEnum : AiImageStatusEnum.values()) {
+ if (statusEnum.getStatus().equals(status)) {
+ return statusEnum;
+ }
+ }
+ throw new IllegalArgumentException("未知会话状态： " + status);
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java
new file mode 100644
index 0000000000..bdba3e8915
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiModelTypeEnum.java
@@ -0,0 +1,41 @@
+package cn.iocoder.yudao.module.ai.enums.model;
+
+import cn.iocoder.yudao.framework.common.core.ArrayValuable;
+import lombok.Getter;
+import lombok.RequiredArgsConstructor;
+
+import java.util.Arrays;
+
+/**
+ * AI 模型类型的枚举
+ *
+ * @author 芋道源码
+ */
+@Getter
+@RequiredArgsConstructor
+public enum AiModelTypeEnum implements ArrayValuable {
+
+ CHAT(1, "对话"),
+ IMAGE(2, "图片"),
+ VOICE(3, "语音"),
+ VIDEO(4, "视频"),
+ EMBEDDING(5, "向量"),
+ RERANK(6, "重排序");
+
+ /**
+ * 类型
+ */
+ private final Integer type;
+ /**
+ * 类型名
+ */
+ private final String name;
+
+ public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiModelTypeEnum::getType).toArray(Integer[]::new);
+
+ @Override
+ public Integer[] array() {
+ return ARRAYS;
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiPlatformEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiPlatformEnum.java
new file mode 100644
index 0000000000..cebe0b9568
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/model/AiPlatformEnum.java
@@ -0,0 +1,69 @@
+package cn.iocoder.yudao.module.ai.enums.model;
+
+import cn.iocoder.yudao.framework.common.core.ArrayValuable;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+
+/**
+ * AI 模型平台
+ *
+ * @author fansili
+ */
+@Getter
+@AllArgsConstructor
+public enum AiPlatformEnum implements ArrayValuable {
+
+ // ========== 国内平台 ==========
+
+ TONG_YI("TongYi", "通义千问"), // 阿里
+ YI_YAN("YiYan", "文心一言"), // 百度
+ DEEP_SEEK("DeepSeek", "DeepSeek"), // DeepSeek
+ ZHI_PU("ZhiPu", "智谱"), // 智谱 AI
+ XING_HUO("XingHuo", "星火"), // 讯飞
+ DOU_BAO("DouBao", "豆包"), // 字节
+ HUN_YUAN("HunYuan", "混元"), // 腾讯
+ SILICON_FLOW("SiliconFlow", "硅基流动"), // 硅基流动
+ MINI_MAX("MiniMax", "MiniMax"), // 稀宇科技
+ MOONSHOT("Moonshot", "月之暗灭"), // KIMI
+ BAI_CHUAN("BaiChuan", "百川智能"), // 百川智能
+
+ // ========== 国外平台 ==========
+
+ OPENAI("OpenAI", "OpenAI"), // OpenAI 官方
+ AZURE_OPENAI("AzureOpenAI", "AzureOpenAI"), // OpenAI 微软
+ OLLAMA("Ollama", "Ollama"),
+
+ STABLE_DIFFUSION("StableDiffusion", "StableDiffusion"), // Stability AI
+ MIDJOURNEY("Midjourney", "Midjourney"), // Midjourney
+ SUNO("Suno", "Suno"), // Suno AI
+
+ ;
+
+ /**
+ * 平台
+ */
+ private final String platform;
+ /**
+ * 平台名
+ */
+ private final String name;
+
+ public static final String[] ARRAYS = Arrays.stream(values()).map(AiPlatformEnum::getPlatform).toArray(String[]::new);
+
+ public static AiPlatformEnum validatePlatform(String platform) {
+ for (AiPlatformEnum platformEnum : AiPlatformEnum.values()) {
+ if (platformEnum.getPlatform().equals(platform)) {
+ return platformEnum;
+ }
+ }
+ throw new IllegalArgumentException("非法平台： " + platform);
+ }
+
+ @Override
+ public String[] array() {
+ return ARRAYS;
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/music/AiMusicGenerateModeEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/music/AiMusicGenerateModeEnum.java
new file mode 100644
index 0000000000..82052943bc
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/music/AiMusicGenerateModeEnum.java
@@ -0,0 +1,37 @@
+package cn.iocoder.yudao.module.ai.enums.music;
+
+import cn.iocoder.yudao.framework.common.core.ArrayValuable;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+
+/**
+ * AI 音乐生成模式的枚举
+ *
+ * @author xiaoxin
+ */
+@AllArgsConstructor
+@Getter
+public enum AiMusicGenerateModeEnum implements ArrayValuable {
+
+ DESCRIPTION(1, "描述模式"),
+ LYRIC(2, "歌词模式");
+
+ /**
+ * 模式
+ */
+ private final Integer mode;
+ /**
+ * 模式名
+ */
+ private final String name;
+
+ public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiMusicGenerateModeEnum::getMode).toArray(Integer[]::new);
+
+ @Override
+ public Integer[] array() {
+ return ARRAYS;
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/music/AiMusicStatusEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/music/AiMusicStatusEnum.java
new file mode 100644
index 0000000000..9340a59762
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/music/AiMusicStatusEnum.java
@@ -0,0 +1,39 @@
+package cn.iocoder.yudao.module.ai.enums.music;
+
+import cn.iocoder.yudao.framework.common.core.ArrayValuable;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+
+/**
+ * AI 音乐状态的枚举
+ *
+ * @author xiaoxin
+ */
+@AllArgsConstructor
+@Getter
+public enum AiMusicStatusEnum implements ArrayValuable {
+
+ IN_PROGRESS(10, "进行中"),
+ SUCCESS(20, "已完成"),
+ FAIL(30, "已失败");
+
+ /**
+ * 状态
+ */
+ private final Integer status;
+
+ /**
+ * 状态名
+ */
+ private final String name;
+
+ public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiMusicStatusEnum::getStatus).toArray(Integer[]::new);
+
+ @Override
+ public Integer[] array() {
+ return ARRAYS;
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/write/AiWriteTypeEnum.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/write/AiWriteTypeEnum.java
new file mode 100644
index 0000000000..6d91433b7b
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/enums/write/AiWriteTypeEnum.java
@@ -0,0 +1,42 @@
+package cn.iocoder.yudao.module.ai.enums.write;
+
+import cn.iocoder.yudao.framework.common.core.ArrayValuable;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+
+/**
+ * AI 写作类型的枚举
+ *
+ * @author xiaoxin
+ */
+@AllArgsConstructor
+@Getter
+public enum AiWriteTypeEnum implements ArrayValuable {
+
+ WRITING(1, "撰写", "请撰写一篇关于 [{}] 的文章。文章的内容格式：{}，语气：{}，语言：{}，长度：{}。请确保涵盖主要内容，不需要除了正文内容外的其他回复，如标题、额外的解释或道歉。"),
+ REPLY(2, "回复", "请针对如下内容：[{}] 做个回复。回复内容参考：[{}], 回复格式：{}，语气：{}，语言：{}，长度：{}。不需要除了正文内容外的其他回复，如标题、开头、额外的解释或道歉。");
+
+ /**
+ * 类型
+ */
+ private final Integer type;
+ /**
+ * 类型名
+ */
+ private final String name;
+
+ /**
+ * 模版
+ */
+ private final String prompt;
+
+ public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiWriteTypeEnum::getType).toArray(Integer[]::new);
+
+ @Override
+ public Integer[] array() {
+ return ARRAYS;
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java
new file mode 100644
index 0000000000..a28d726b90
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/AiAutoConfiguration.java
@@ -0,0 +1,253 @@
+package cn.iocoder.yudao.module.ai.framework.ai.config;
+
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.extra.spring.SpringUtil;
+import cn.iocoder.yudao.module.ai.framework.ai.core.AiModelFactory;
+import cn.iocoder.yudao.module.ai.framework.ai.core.AiModelFactoryImpl;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.baichuan.BaiChuanChatModel;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.deepseek.DeepSeekChatModel;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.doubao.DouBaoChatModel;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.hunyuan.HunYuanChatModel;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.siliconflow.SiliconFlowApiConstants;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.siliconflow.SiliconFlowChatModel;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.suno.api.SunoApi;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.xinghuo.XingHuoChatModel;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.ai.autoconfigure.vectorstore.milvus.MilvusServiceClientProperties;
+import org.springframework.ai.autoconfigure.vectorstore.milvus.MilvusVectorStoreProperties;
+import org.springframework.ai.autoconfigure.vectorstore.qdrant.QdrantVectorStoreProperties;
+import org.springframework.ai.autoconfigure.vectorstore.redis.RedisVectorStoreProperties;
+import org.springframework.ai.embedding.BatchingStrategy;
+import org.springframework.ai.embedding.TokenCountBatchingStrategy;
+import org.springframework.ai.model.tool.ToolCallingManager;
+import org.springframework.ai.openai.OpenAiChatModel;
+import org.springframework.ai.openai.OpenAiChatOptions;
+import org.springframework.ai.openai.api.OpenAiApi;
+import org.springframework.ai.tokenizer.JTokkitTokenCountEstimator;
+import org.springframework.ai.tokenizer.TokenCountEstimator;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+/**
+ * 芋道 AI 自动配置
+ *
+ * @author fansili
+ */
+@Configuration
+@EnableConfigurationProperties({ YudaoAiProperties.class,
+ QdrantVectorStoreProperties.class, // 解析 Qdrant 配置
+ RedisVectorStoreProperties.class, // 解析 Redis 配置
+ MilvusVectorStoreProperties.class, MilvusServiceClientProperties.class // 解析 Milvus 配置
+})
+@Slf4j
+public class AiAutoConfiguration {
+
+ @Bean
+ public AiModelFactory aiModelFactory() {
+ return new AiModelFactoryImpl();
+ }
+
+ // ========== 各种 AI Client 创建 ==========
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.deepseek.enable", havingValue = "true")
+ public DeepSeekChatModel deepSeekChatModel(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.DeepSeekProperties properties = yudaoAiProperties.getDeepseek();
+ return buildDeepSeekChatModel(properties);
+ }
+
+ public DeepSeekChatModel buildDeepSeekChatModel(YudaoAiProperties.DeepSeekProperties properties) {
+ if (StrUtil.isEmpty(properties.getModel())) {
+ properties.setModel(DeepSeekChatModel.MODEL_DEFAULT);
+ }
+ OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
+ .openAiApi(OpenAiApi.builder()
+ .baseUrl(DeepSeekChatModel.BASE_URL)
+ .apiKey(properties.getApiKey())
+ .build())
+ .defaultOptions(OpenAiChatOptions.builder()
+ .model(properties.getModel())
+ .temperature(properties.getTemperature())
+ .maxTokens(properties.getMaxTokens())
+ .topP(properties.getTopP())
+ .build())
+ .toolCallingManager(getToolCallingManager())
+ .build();
+ return new DeepSeekChatModel(openAiChatModel);
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.doubao.enable", havingValue = "true")
+ public DouBaoChatModel douBaoChatClient(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.DouBaoProperties properties = yudaoAiProperties.getDoubao();
+ return buildDouBaoChatClient(properties);
+ }
+
+ public DouBaoChatModel buildDouBaoChatClient(YudaoAiProperties.DouBaoProperties properties) {
+ if (StrUtil.isEmpty(properties.getModel())) {
+ properties.setModel(DouBaoChatModel.MODEL_DEFAULT);
+ }
+ OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
+ .openAiApi(OpenAiApi.builder()
+ .baseUrl(DouBaoChatModel.BASE_URL)
+ .apiKey(properties.getApiKey())
+ .build())
+ .defaultOptions(OpenAiChatOptions.builder()
+ .model(properties.getModel())
+ .temperature(properties.getTemperature())
+ .maxTokens(properties.getMaxTokens())
+ .topP(properties.getTopP())
+ .build())
+ .toolCallingManager(getToolCallingManager())
+ .build();
+ return new DouBaoChatModel(openAiChatModel);
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.siliconflow.enable", havingValue = "true")
+ public SiliconFlowChatModel siliconFlowChatClient(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.SiliconFlowProperties properties = yudaoAiProperties.getSiliconflow();
+ return buildSiliconFlowChatClient(properties);
+ }
+
+ public SiliconFlowChatModel buildSiliconFlowChatClient(YudaoAiProperties.SiliconFlowProperties properties) {
+ if (StrUtil.isEmpty(properties.getModel())) {
+ properties.setModel(SiliconFlowApiConstants.MODEL_DEFAULT);
+ }
+ OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
+ .openAiApi(OpenAiApi.builder()
+ .baseUrl(SiliconFlowApiConstants.DEFAULT_BASE_URL)
+ .apiKey(properties.getApiKey())
+ .build())
+ .defaultOptions(OpenAiChatOptions.builder()
+ .model(properties.getModel())
+ .temperature(properties.getTemperature())
+ .maxTokens(properties.getMaxTokens())
+ .topP(properties.getTopP())
+ .build())
+ .toolCallingManager(getToolCallingManager())
+ .build();
+ return new SiliconFlowChatModel(openAiChatModel);
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.hunyuan.enable", havingValue = "true")
+ public HunYuanChatModel hunYuanChatClient(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.HunYuanProperties properties = yudaoAiProperties.getHunyuan();
+ return buildHunYuanChatClient(properties);
+ }
+
+ public HunYuanChatModel buildHunYuanChatClient(YudaoAiProperties.HunYuanProperties properties) {
+ if (StrUtil.isEmpty(properties.getModel())) {
+ properties.setModel(HunYuanChatModel.MODEL_DEFAULT);
+ }
+ // 特殊：由于混元大模型不提供 deepseek，而是通过知识引擎，所以需要区分下 URL
+ if (StrUtil.isEmpty(properties.getBaseUrl())) {
+ properties.setBaseUrl(
+ StrUtil.startWithIgnoreCase(properties.getModel(), "deepseek") ? HunYuanChatModel.DEEP_SEEK_BASE_URL
+ : HunYuanChatModel.BASE_URL);
+ }
+ // 创建 OpenAiChatModel、HunYuanChatModel 对象
+ OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
+ .openAiApi(OpenAiApi.builder()
+ .baseUrl(properties.getBaseUrl())
+ .apiKey(properties.getApiKey())
+ .build())
+ .defaultOptions(OpenAiChatOptions.builder()
+ .model(properties.getModel())
+ .temperature(properties.getTemperature())
+ .maxTokens(properties.getMaxTokens())
+ .topP(properties.getTopP())
+ .build())
+ .toolCallingManager(getToolCallingManager())
+ .build();
+ return new HunYuanChatModel(openAiChatModel);
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.xinghuo.enable", havingValue = "true")
+ public XingHuoChatModel xingHuoChatClient(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.XingHuoProperties properties = yudaoAiProperties.getXinghuo();
+ return buildXingHuoChatClient(properties);
+ }
+
+ public XingHuoChatModel buildXingHuoChatClient(YudaoAiProperties.XingHuoProperties properties) {
+ if (StrUtil.isEmpty(properties.getModel())) {
+ properties.setModel(XingHuoChatModel.MODEL_DEFAULT);
+ }
+ OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
+ .openAiApi(OpenAiApi.builder()
+ .baseUrl(XingHuoChatModel.BASE_URL)
+ .apiKey(properties.getAppKey() + ":" + properties.getSecretKey())
+ .build())
+ .defaultOptions(OpenAiChatOptions.builder()
+ .model(properties.getModel())
+ .temperature(properties.getTemperature())
+ .maxTokens(properties.getMaxTokens())
+ .topP(properties.getTopP())
+ .build())
+ .toolCallingManager(getToolCallingManager())
+ .build();
+ return new XingHuoChatModel(openAiChatModel);
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.baichuan.enable", havingValue = "true")
+ public BaiChuanChatModel baiChuanChatClient(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.BaiChuanProperties properties = yudaoAiProperties.getBaichuan();
+ return buildBaiChuanChatClient(properties);
+ }
+
+ public BaiChuanChatModel buildBaiChuanChatClient(YudaoAiProperties.BaiChuanProperties properties) {
+ if (StrUtil.isEmpty(properties.getModel())) {
+ properties.setModel(BaiChuanChatModel.MODEL_DEFAULT);
+ }
+ OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
+ .openAiApi(OpenAiApi.builder()
+ .baseUrl(BaiChuanChatModel.BASE_URL)
+ .apiKey(properties.getApiKey())
+ .build())
+ .defaultOptions(OpenAiChatOptions.builder()
+ .model(properties.getModel())
+ .temperature(properties.getTemperature())
+ .maxTokens(properties.getMaxTokens())
+ .topP(properties.getTopP())
+ .build())
+ .toolCallingManager(getToolCallingManager())
+ .build();
+ return new BaiChuanChatModel(openAiChatModel);
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.midjourney.enable", havingValue = "true")
+ public MidjourneyApi midjourneyApi(YudaoAiProperties yudaoAiProperties) {
+ YudaoAiProperties.MidjourneyProperties config = yudaoAiProperties.getMidjourney();
+ return new MidjourneyApi(config.getBaseUrl(), config.getApiKey(), config.getNotifyUrl());
+ }
+
+ @Bean
+ @ConditionalOnProperty(value = "yudao.ai.suno.enable", havingValue = "true")
+ public SunoApi sunoApi(YudaoAiProperties yudaoAiProperties) {
+ return new SunoApi(yudaoAiProperties.getSuno().getBaseUrl());
+ }
+
+ // ========== RAG 相关 ==========
+
+ @Bean
+ public TokenCountEstimator tokenCountEstimator() {
+ return new JTokkitTokenCountEstimator();
+ }
+
+ @Bean
+ public BatchingStrategy batchingStrategy() {
+ return new TokenCountBatchingStrategy();
+ }
+
+ private static ToolCallingManager getToolCallingManager() {
+ return SpringUtil.getBean(ToolCallingManager.class);
+ }
+
+}
\ No newline at end of file
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/YudaoAiProperties.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/YudaoAiProperties.java
new file mode 100644
index 0000000000..7f8046768a
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/config/YudaoAiProperties.java
@@ -0,0 +1,164 @@
+package cn.iocoder.yudao.module.ai.framework.ai.config;
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+
+/**
+ * 芋道 AI 配置类
+ *
+ * @author fansili
+ * @since 1.0
+ */
+@ConfigurationProperties(prefix = "yudao.ai")
+@Data
+public class YudaoAiProperties {
+
+ /**
+ * DeepSeek
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private DeepSeekProperties deepseek;
+
+ /**
+ * 字节豆包
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private DouBaoProperties doubao;
+
+ /**
+ * 腾讯混元
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private HunYuanProperties hunyuan;
+
+ /**
+ * 硅基流动
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private SiliconFlowProperties siliconflow;
+
+ /**
+ * 讯飞星火
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private XingHuoProperties xinghuo;
+
+ /**
+ * 百川
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private BaiChuanProperties baichuan;
+
+ /**
+ * Midjourney 绘图
+ */
+ private MidjourneyProperties midjourney;
+
+ /**
+ * Suno 音乐
+ */
+ @SuppressWarnings("SpellCheckingInspection")
+ private SunoProperties suno;
+
+ @Data
+ public static class DeepSeekProperties {
+
+ private String enable;
+ private String apiKey;
+
+ private String model;
+ private Double temperature;
+ private Integer maxTokens;
+ private Double topP;
+
+ }
+
+ @Data
+ public static class DouBaoProperties {
+
+ private String enable;
+ private String apiKey;
+
+ private String model;
+ private Double temperature;
+ private Integer maxTokens;
+ private Double topP;
+
+ }
+
+ @Data
+ public static class HunYuanProperties {
+
+ private String enable;
+ private String baseUrl;
+ private String apiKey;
+
+ private String model;
+ private Double temperature;
+ private Integer maxTokens;
+ private Double topP;
+
+ }
+
+ @Data
+ public static class SiliconFlowProperties {
+
+ private String enable;
+ private String apiKey;
+
+ private String model;
+ private Double temperature;
+ private Integer maxTokens;
+ private Double topP;
+
+ }
+
+ @Data
+ public static class XingHuoProperties {
+
+ private String enable;
+ private String appId;
+ private String appKey;
+ private String secretKey;
+
+ private String model;
+ private Double temperature;
+ private Integer maxTokens;
+ private Double topP;
+
+ }
+
+ @Data
+ public static class BaiChuanProperties {
+
+ private String enable;
+ private String apiKey;
+
+ private String model;
+ private Double temperature;
+ private Integer maxTokens;
+ private Double topP;
+
+ }
+
+ @Data
+ public static class MidjourneyProperties {
+
+ private String enable;
+ private String baseUrl;
+
+ private String apiKey;
+ private String notifyUrl;
+
+ }
+
+ @Data
+ public static class SunoProperties {
+
+ private boolean enable = false;
+
+ private String baseUrl;
+
+ }
+
+}
diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/core/AiModelFactory.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/core/AiModelFactory.java
new file mode 100644
index 0000000000..659fa1f92b
--- /dev/null
+++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/framework/ai/core/AiModelFactory.java
@@ -0,0 +1,113 @@
+package cn.iocoder.yudao.module.ai.framework.ai.core;
+
+import cn.iocoder.yudao.module.ai.enums.model.AiPlatformEnum;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi;
+import cn.iocoder.yudao.module.ai.framework.ai.core.model.suno.api.SunoApi;
+import org.springframework.ai.chat.model.ChatModel;
+import org.springframework.ai.embedding.EmbeddingModel;
+import org.springframework.ai.image.ImageModel;
+import org.springframework.ai.vectorstore.VectorStore;
+
+import java.util.Map;
+
+/**
+ * AI Model 模型工厂的接口类
+ *
+ * @author fansili
+ */
+public interface AiModelFactory {
+
+ /**
+ * 基于指定配置，获得 ChatModel 对象
+ *
+ * 如果不存在，则进行创建
+ *
+ * @param platform 平台
+ * @param apiKey API KEY
+ * @param url API URL
+ * @return ChatModel 对象
+ */
+ ChatModel getOrCreateChatModel(AiPlatformEnum platform, String apiKey, String url);
+
+ /**
+ * 基于默认配置，获得 ChatModel 对象
+ *
+ * 默认配置，指的是在 application.yaml 配置文件中的 spring.ai 相关的配置
+ *
+ * @param platform 平台
+ * @return ChatModel 对象
+ */
+ ChatModel getDefaultChatModel(AiPlatformEnum platform);
+
+ /**
+ * 基于默认配置，获得 ImageModel 对象
+ *
+ * 默认配置，指的是在 application.yaml 配置文件中的 spring.ai 相关的配置
+ *
+ * @param platform 平台
+ * @return ImageModel 对象
+ */
+ ImageModel getDefaultImageModel(AiPlatformEnum platform);
+
+ /**
+ * 基于指定配置，获得 ImageModel 对象
+ *
+ * 如果不存在，则进行创建
+ *
+ * @param platform 平台
+ * @param apiKey API KEY
+ * @param url API URL
+ * @return ImageModel 对象
+ */
+ ImageModel getOrCreateImageModel(AiPlatformEnum platform, String apiKey, String url);
+
+ /**
+ * 基于指定配置，获得 MidjourneyApi 对象
+ *
+ * 如果不存在，则进行创建
+ *
+ * @param apiKey API KEY
+ * @param url API URL
+ * @return MidjourneyApi 对象
+ */
+ MidjourneyApi getOrCreateMidjourneyApi(String apiKey, String url);
+
+ /**
+ * 基于指定配置，获得 SunoApi 对象
+ *
+ * 如果不存在，则进行创建
+ *
+ * @param apiKey API KEY
+ * @param url API URL
+ * @return SunoApi 对象
+ */
+ SunoApi getOrCreateSunoApi(String apiKey, String url);
+
+ /**
+ * 基于指定配置，获得 EmbeddingModel 对象
+ *
+ * 如果不存在，则进行创建
+ *
+ * @param platform 平台
+ * @param apiKey API KEY
+ * @param url API URL
+ * @param model 模型
+ * @return ChatModel 对象
+ */
+ EmbeddingModel getOrCreateEmbeddingModel(AiPlatformEnum platform, String apiKey, String url, String model);
+
+ /**
+ * 基于指定配置，获得 VectorStore 对象
+ *
+ * 如果不存在，则进行创建
+ *
+ * @param type 向量存储类型
+ * @param embeddingModel 向量模型
+ * @param metadataFields 元数据字段
+ * @return VectorStore 对象
+ */
+ VectorStore getOrCreateVectorStore(Class type,
+ EmbeddingModel embeddingModel,
+ Map