diff --git a/README.md b/README.md
index f8ce58b7..78ea6497 100644
--- a/README.md
+++ b/README.md
@@ -150,6 +150,13 @@
QQ技术交流群
技术讨论
+
+
+
+微信技术交流群
+技术讨论
+ |
+
diff --git a/docs/工作流模块说明.md b/docs/工作流模块说明.md
new file mode 100644
index 00000000..8d2f4396
--- /dev/null
+++ b/docs/工作流模块说明.md
@@ -0,0 +1,432 @@
+# Ruoyi-AI 工作流模块详细说明文档
+
+## 概述
+
+Ruoyi-AI 工作流模块是一个基于 LangGraph4j 的智能工作流引擎，支持可视化工作流设计、AI 模型集成、条件分支、人机交互等高级功能。该模块采用微服务架构，提供完整的 RESTful API 和流式响应支持。
+
+## 模块架构
+
+### 1. 模块结构
+
+```
+ruoyi-ai/
+├── ruoyi-modules/
+│ └── ruoyi-workflow/ # 工作流核心模块
+│ ├── pom.xml
+│ └── src/main/java/org/ruoyi/workflow/
+│ └── controller/ # 控制器层
+│ ├── WorkflowController.java
+│ ├── WorkflowRuntimeController.java
+│ └── admin/ # 管理端控制器
+│ ├── AdminWorkflowController.java
+│ └── AdminWorkflowComponentController.java
+└── ruoyi-modules-api/
+ └── ruoyi-workflow-api/ # 工作流API模块
+ ├── pom.xml
+ └── src/main/java/org/ruoyi/workflow/
+ ├── entity/ # 实体类
+ ├── dto/ # 数据传输对象
+ ├── service/ # 服务接口
+ ├── mapper/ # 数据访问层
+ ├── workflow/ # 工作流核心逻辑
+ ├── enums/ # 枚举类
+ ├── util/ # 工具类
+ └── exception/ # 异常处理
+```
+
+### 2. 核心依赖
+
+- **LangGraph4j**: 1.5.3 - 工作流图执行引擎
+- **LangChain4j**: 1.2.0 - AI 模型集成框架
+- **Spring Boot**: 3.x - 应用框架
+- **MyBatis Plus**: 数据访问层
+- **Redis**: 缓存和状态管理
+- **Swagger/OpenAPI**: API 文档
+
+## 核心功能
+
+### 1. 工作流管理
+
+#### 1.1 工作流定义
+- **创建工作流**: 支持自定义标题、描述、公开性设置
+- **编辑工作流**: 可视化节点编辑、连接线配置
+- **版本控制**: 支持工作流的版本管理和回滚
+- **权限管理**: 支持公开/私有工作流设置
+
+#### 1.2 工作流执行
+- **流式执行**: 基于 SSE 的实时流式响应
+- **状态管理**: 完整的执行状态跟踪
+- **错误处理**: 详细的错误信息和异常处理
+- **中断恢复**: 支持工作流中断和恢复执行
+
+### 2. 节点类型
+
+#### 2.1 基础节点
+- **Start**: 开始节点，定义工作流入口
+- **End**: 结束节点，定义工作流出口
+
+#### 2.2 AI 模型节点
+- **Answer**: 大语言模型问答节点
+- **Dalle3**: DALL-E 3 图像生成
+- **Tongyiwanx**: 通义万相图像生成
+- **Classifier**: 内容分类节点
+
+#### 2.3 数据处理节点
+- **DocumentExtractor**: 文档信息提取
+- **KeywordExtractor**: 关键词提取
+- **FaqExtractor**: 常见问题提取
+- **KnowledgeRetrieval**: 知识库检索
+
+#### 2.4 控制流节点
+- **Switcher**: 条件分支节点
+- **HumanFeedback**: 人机交互节点
+
+#### 2.5 外部集成节点
+- **Google**: Google 搜索集成
+- **MailSend**: 邮件发送
+- **HttpRequest**: HTTP 请求
+- **Template**: 模板转换
+
+### 3. 数据流管理
+
+#### 3.1 输入输出定义
+```java
+// 节点输入输出数据结构
+public class NodeIOData {
+ private String name; // 参数名称
+ private NodeIODataContent content; // 参数内容
+}
+
+// 支持的数据类型
+public enum WfIODataTypeEnum {
+ TEXT, // 文本
+ NUMBER, // 数字
+ BOOLEAN, // 布尔值
+ FILES, // 文件
+ OPTIONS // 选项
+}
+```
+
+#### 3.2 参数引用
+- **节点间引用**: 支持上游节点输出作为下游节点输入
+- **参数映射**: 自动处理参数名称映射
+- **类型转换**: 自动进行数据类型转换
+
+## 数据库设计
+
+### 1. 核心表结构
+
+#### 1.1 工作流定义表 (t_workflow)
+```sql
+CREATE TABLE t_workflow (
+ id BIGINT AUTO_INCREMENT PRIMARY KEY,
+ uuid VARCHAR(32) NOT NULL DEFAULT '',
+ title VARCHAR(100) NOT NULL DEFAULT '',
+ remark TEXT NOT NULL DEFAULT '',
+ user_id BIGINT NOT NULL DEFAULT 0,
+ is_public TINYINT(1) NOT NULL DEFAULT 0,
+ is_enable TINYINT(1) NOT NULL DEFAULT 1,
+ create_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+ update_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+ is_deleted TINYINT(1) NOT NULL DEFAULT 0
+);
+```
+
+#### 1.2 工作流节点表 (t_workflow_node)
+```sql
+CREATE TABLE t_workflow_node (
+ id BIGINT AUTO_INCREMENT PRIMARY KEY,
+ uuid VARCHAR(32) NOT NULL DEFAULT '',
+ workflow_id BIGINT NOT NULL DEFAULT 0,
+ workflow_component_id BIGINT NOT NULL DEFAULT 0,
+ user_id BIGINT NOT NULL DEFAULT 0,
+ title VARCHAR(100) NOT NULL DEFAULT '',
+ remark VARCHAR(500) NOT NULL DEFAULT '',
+ input_config JSON NOT NULL DEFAULT ('{}'),
+ node_config JSON NOT NULL DEFAULT ('{}'),
+ position_x DOUBLE NOT NULL DEFAULT 0,
+ position_y DOUBLE NOT NULL DEFAULT 0,
+ create_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+ update_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+ is_deleted TINYINT(1) NOT NULL DEFAULT 0
+);
+```
+
+#### 1.3 工作流边表 (t_workflow_edge)
+```sql
+CREATE TABLE t_workflow_edge (
+ id BIGINT AUTO_INCREMENT PRIMARY KEY,
+ uuid VARCHAR(32) NOT NULL DEFAULT '',
+ workflow_id BIGINT NOT NULL DEFAULT 0,
+ source_node_uuid VARCHAR(32) NOT NULL DEFAULT '',
+ source_handle VARCHAR(32) NOT NULL DEFAULT '',
+ target_node_uuid VARCHAR(32) NOT NULL DEFAULT '',
+ create_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+ update_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+ is_deleted TINYINT(1) NOT NULL DEFAULT 0
+);
+```
+
+#### 1.4 工作流运行时表 (t_workflow_runtime)
+```sql
+CREATE TABLE t_workflow_runtime (
+ id BIGINT AUTO_INCREMENT PRIMARY KEY,
+ uuid VARCHAR(32) NOT NULL DEFAULT '',
+ user_id BIGINT NOT NULL DEFAULT 0,
+ workflow_id BIGINT NOT NULL DEFAULT 0,
+ input JSON NOT NULL DEFAULT ('{}'),
+ output JSON NOT NULL DEFAULT ('{}'),
+ status SMALLINT NOT NULL DEFAULT 1,
+ status_remark VARCHAR(250) NOT NULL DEFAULT '',
+ create_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
+ update_time DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
+ is_deleted TINYINT(1) NOT NULL DEFAULT 0
+);
+```
+
+#### 1.5 工作流组件表 (t_workflow_component)
+```sql
+CREATE TABLE t_workflow_component (
+ id BIGINT AUTO_INCREMENT PRIMARY KEY,
+ uuid VARCHAR(32) DEFAULT '' NOT NULL,
+ name VARCHAR(32) DEFAULT '' NOT NULL,
+ title VARCHAR(100) DEFAULT '' NOT NULL,
+ remark TEXT NOT NULL,
+ display_order INT DEFAULT 0 NOT NULL,
+ is_enable TINYINT(1) DEFAULT 0 NOT NULL,
+ create_time DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL,
+ update_time DATETIME DEFAULT CURRENT_TIMESTAMP NOT NULL,
+ is_deleted TINYINT(1) DEFAULT 0 NOT NULL
+);
+```
+
+## API 接口
+
+### 1. 工作流管理接口
+
+#### 1.1 基础操作
+```http
+# 创建工作流
+POST /workflow/add
+Content-Type: application/json
+{
+ "title": "工作流标题",
+ "remark": "工作流描述",
+ "isPublic": false
+}
+
+# 更新工作流
+POST /workflow/update
+Content-Type: application/json
+{
+ "uuid": "工作流UUID",
+ "title": "新标题",
+ "remark": "新描述"
+}
+
+# 删除工作流
+POST /workflow/del/{uuid}
+
+# 启用/禁用工作流
+POST /workflow/enable/{uuid}?enable=true
+```
+
+#### 1.2 搜索和查询
+```http
+# 搜索我的工作流
+GET /workflow/mine/search?keyword=关键词&isPublic=true¤tPage=1&pageSize=10
+
+# 搜索公开工作流
+GET /workflow/public/search?keyword=关键词¤tPage=1&pageSize=10
+
+# 获取工作流组件列表
+GET /workflow/public/component/list
+```
+
+### 2. 工作流执行接口
+
+#### 2.1 流式执行
+```http
+# 流式执行工作流
+POST /workflow/run
+Content-Type: application/json
+Accept: text/event-stream
+{
+ "uuid": "工作流UUID",
+ "inputs": [
+ {
+ "name": "input",
+ "content": {
+ "type": 1,
+ "textContent": "用户输入内容"
+ }
+ }
+ ]
+}
+```
+
+#### 2.2 运行时管理
+```http
+# 恢复中断的工作流
+POST /workflow/runtime/resume/{runtimeUuid}
+Content-Type: application/json
+{
+ "feedbackContent": "用户反馈内容"
+}
+
+# 查询工作流执行历史
+GET /workflow/runtime/page?wfUuid=工作流UUID¤tPage=1&pageSize=10
+
+# 查询运行时节点详情
+GET /workflow/runtime/nodes/{runtimeUuid}
+
+# 清理运行时数据
+POST /workflow/runtime/clear?wfUuid=工作流UUID
+```
+
+### 3. 管理端接口
+
+#### 3.1 工作流管理
+```http
+# 搜索所有工作流
+POST /admin/workflow/search
+Content-Type: application/json
+{
+ "title": "搜索关键词",
+ "isPublic": true,
+ "isEnable": true
+}
+
+# 启用/禁用工作流
+POST /admin/workflow/enable?uuid=工作流UUID&isEnable=true
+```
+
+## 核心实现
+
+### 1. 工作流引擎 (WorkflowEngine)
+
+工作流引擎是整个模块的核心，负责：
+- 工作流图的构建和编译
+- 节点执行调度
+- 状态管理和持久化
+- 流式输出处理
+
+```java
+public class WorkflowEngine {
+ // 核心执行方法
+ public void run(User user, List userInputs, SseEmitter sseEmitter) {
+ // 1. 验证工作流状态
+ // 2. 创建运行时实例
+ // 3. 构建状态图
+ // 4. 执行工作流
+ // 5. 处理流式输出
+ }
+
+ // 恢复执行方法
+ public void resume(String userInput) {
+ // 1. 更新状态
+ // 2. 继续执行
+ }
+}
+```
+
+### 2. 节点工厂 (WfNodeFactory)
+
+节点工厂负责根据组件类型创建对应的节点实例：
+
+```java
+public class WfNodeFactory {
+ public static AbstractWfNode create(WorkflowComponent component,
+ WorkflowNode node,
+ WfState wfState,
+ WfNodeState nodeState) {
+ // 根据组件类型创建对应的节点实例
+ switch (component.getName()) {
+ case "Answer":
+ return new LLMAnswerNode(component, node, wfState, nodeState);
+ case "Switcher":
+ return new SwitcherNode(component, node, wfState, nodeState);
+ // ... 其他节点类型
+ }
+ }
+}
+```
+
+### 3. 图构建器 (WorkflowGraphBuilder)
+
+图构建器负责将工作流定义转换为可执行的状态图：
+
+```java
+public class WorkflowGraphBuilder {
+ public StateGraph build(WorkflowNode startNode) {
+ // 1. 构建编译节点树
+ // 2. 转换为状态图
+ // 3. 添加节点和边
+ // 4. 处理条件分支
+ // 5. 处理并行执行
+ }
+}
+```
+
+## 流式响应机制
+
+### 1. SSE 事件类型
+
+工作流执行过程中会发送多种类型的 SSE 事件：
+
+```javascript
+// 节点开始执行
+[NODE_RUN_节点UUID] - 节点执行开始事件
+
+// 节点输入数据
+[NODE_INPUT_节点UUID] - 节点输入数据事件
+
+// 节点输出数据
+[NODE_OUTPUT_节点UUID] - 节点输出数据事件
+
+// 流式内容块
+[NODE_CHUNK_节点UUID] - 流式内容块事件
+
+// 等待用户输入
+[NODE_WAIT_FEEDBACK_BY_节点UUID] - 等待用户输入事件
+```
+
+### 2. 流式处理流程
+
+1. **初始化**: 创建工作流运行时实例
+2. **节点执行**: 逐个执行工作流节点
+3. **实时输出**: 通过 SSE 实时推送执行结果
+4. **状态更新**: 实时更新节点和工作流状态
+5. **错误处理**: 捕获并处理执行过程中的错误
+
+
+## 扩展开发
+
+### 1. 自定义节点开发
+
+要开发自定义工作流节点，需要：
+
+1. **创建节点类**：继承 `AbstractWfNode`
+2. **实现处理逻辑**：重写 `onProcess()` 方法
+3. **定义配置类**：创建节点配置类
+4. **注册组件**：在组件表中注册新组件
+
+```java
+public class CustomNode extends AbstractWfNode {
+ @Override
+ protected NodeProcessResult onProcess() {
+ // 实现自定义处理逻辑
+ List outputs = new ArrayList<>();
+ // ... 处理逻辑
+ return NodeProcessResult.success(outputs);
+ }
+}
+```
+
+### 2. 自定义组件注册
+
+```sql
+-- 在 t_workflow_component 表中添加新组件
+INSERT INTO t_workflow_component (uuid, name, title, remark, is_enable)
+VALUES (REPLACE(UUID(), '-', ''), 'CustomNode', '自定义节点', '自定义节点描述', true);
+```
diff --git a/pom.xml b/pom.xml
index b164fbfb..6f282a46 100644
--- a/pom.xml
+++ b/pom.xml
@@ -270,13 +270,6 @@
${lock4j.version}
-
-
- com.xuxueli
- xxl-job-core
- ${xxl-job.version}
-
-
com.alibaba
transmittable-thread-local
@@ -373,6 +366,23 @@
langchain4j-community-neo4j
${langchain4j-neo4j.version}
+ ruoyi-aihuman
+ ${revision}
+
+
+
+
+ org.ruoyi
+ ruoyi-workflow
+ ${revision}
+
+
+
+ org.ruoyi
+ ruoyi-workflow-api
+ ${revision}
+
+
diff --git a/ruoyi-admin/pom.xml b/ruoyi-admin/pom.xml
index 3cf17cc7..5aa1fba8 100644
--- a/ruoyi-admin/pom.xml
+++ b/ruoyi-admin/pom.xml
@@ -61,6 +61,14 @@
org.ruoyi
ruoyi-graph
+
+ org.ruoyi
+ ruoyi-workflow
+
+
+
+ org.ruoyi
+ ruoyi-aihuman
diff --git a/ruoyi-admin/src/main/java/org/ruoyi/RuoYiAIApplication.java b/ruoyi-admin/src/main/java/org/ruoyi/RuoYiAIApplication.java
index cc65b7b4..ea5d664f 100644
--- a/ruoyi-admin/src/main/java/org/ruoyi/RuoYiAIApplication.java
+++ b/ruoyi-admin/src/main/java/org/ruoyi/RuoYiAIApplication.java
@@ -11,7 +11,7 @@ import org.springframework.scheduling.annotation.EnableScheduling;
*
* @author Lion Li
*/
-@SpringBootApplication
+@SpringBootApplication(scanBasePackages = {"org.ruoyi", "org.ruoyi.aihuman"})
@EnableScheduling
@EnableAsync
public class RuoYiAIApplication {
@@ -22,4 +22,4 @@ public class RuoYiAIApplication {
application.run(args);
System.out.println("(♥◠‿◠)ﾉﾞ RuoYiAI启动成功 ლ(´ڡ`ლ)ﾞ");
}
-}
+}
\ No newline at end of file
diff --git a/ruoyi-admin/src/main/resources/application-dev.yml b/ruoyi-admin/src/main/resources/application-dev.yml
index a722f88b..cb082f50 100644
--- a/ruoyi-admin/src/main/resources/application-dev.yml
+++ b/ruoyi-admin/src/main/resources/application-dev.yml
@@ -37,6 +37,8 @@ spring:
connectionTestQuery: SELECT 1
# 多久检查一次连接的活性
keepaliveTime: 30000
+ mail:
+ username: xx
--- # redis 单机配置(单机与集群只能开启一个另一个需要注释掉)
spring.data:
@@ -102,7 +104,15 @@ pdf:
#百炼模型配置
dashscope:
key: sk-xxxx
- model: qvq-max
+
+local:
+ images: xx
+
+
+
+ files: xx
+
+
--- # Neo4j 知识图谱配置
neo4j:
diff --git a/ruoyi-admin/src/main/resources/application.yml b/ruoyi-admin/src/main/resources/application.yml
index 6d5e6d2f..4c6d38a0 100644
--- a/ruoyi-admin/src/main/resources/application.yml
+++ b/ruoyi-admin/src/main/resources/application.yml
@@ -156,6 +156,8 @@ security:
# actuator 监控配置
- /actuator
- /actuator/**
+ - /workflow/**
+ - /admin/workflow/**
# 多租户配置
tenant:
# 是否开启
@@ -328,3 +330,19 @@ spring:
servers-configuration: classpath:mcp-server.json
request-timeout: 300s
+# 向量库配置
+vector-store:
+ # 向量存储类型 可选(weaviate/milvus)
+ # 如需修改向量库类型，请修改此配置值!
+ type: weaviate
+
+ # Weaviate配置
+ weaviate:
+ protocol: http
+ host: 127.0.0.1:6038
+ classname: LocalKnowledge
+ # Milvus配置
+ milvus:
+ url: http://localhost:19530
+ collectionname: LocalKnowledge
+
diff --git a/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/request/ChatRequest.java b/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/request/ChatRequest.java
index f0eacdfc..ba351cf7 100644
--- a/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/request/ChatRequest.java
+++ b/ruoyi-common/ruoyi-common-chat/src/main/java/org/ruoyi/common/chat/request/ChatRequest.java
@@ -26,11 +26,18 @@ public class ChatRequest {
*/
private String prompt;
+
/**
* 系统提示词
*/
private String sysPrompt;
+
+ /**
+ * 消息id
+ */
+ private Long messageId;
+
/**
* 是否开启流式对话
*/
@@ -72,6 +79,11 @@ public class ChatRequest {
*/
private Boolean hasAttachment;
+ /**
+ * 是否启用深度思考
+ */
+ private Boolean enableThinking;
+
/**
* 是否自动切换模型
*/
@@ -82,9 +94,4 @@ public class ChatRequest {
*/
private String token;
- /**
- * 消息ID（保存消息成功后设置，用于后续扣费更新）
- */
- private Long messageId;
-
}
diff --git a/ruoyi-common/ruoyi-common-core/src/main/java/org/ruoyi/common/core/config/VectorStoreProperties.java b/ruoyi-common/ruoyi-common-core/src/main/java/org/ruoyi/common/core/config/VectorStoreProperties.java
new file mode 100644
index 00000000..b4bb135d
--- /dev/null
+++ b/ruoyi-common/ruoyi-common-core/src/main/java/org/ruoyi/common/core/config/VectorStoreProperties.java
@@ -0,0 +1,62 @@
+package org.ruoyi.common.core.config;
+
+import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+/**
+ * 向量库配置属性
+ *
+ * @author ageer
+ */
+@Data
+@Component
+@ConfigurationProperties(prefix = "vector-store")
+public class VectorStoreProperties {
+
+ /**
+ * 向量库类型
+ */
+ private String type;
+
+ /**
+ * Weaviate配置
+ */
+ private Weaviate weaviate = new Weaviate();
+
+ /**
+ * Milvus配置
+ */
+ private Milvus milvus = new Milvus();
+
+ @Data
+ public static class Weaviate {
+ /**
+ * 协议
+ */
+ private String protocol;
+
+ /**
+ * 主机地址
+ */
+ private String host;
+
+ /**
+ * 类名
+ */
+ private String classname;
+ }
+
+ @Data
+ public static class Milvus {
+ /**
+ * 连接URL
+ */
+ private String url;
+
+ /**
+ * 集合名称
+ */
+ private String collectionname;
+ }
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/pom.xml b/ruoyi-modules-api/pom.xml
index a3f0ef00..2cf63600 100644
--- a/ruoyi-modules-api/pom.xml
+++ b/ruoyi-modules-api/pom.xml
@@ -17,6 +17,7 @@
ruoyi-chat-api
ruoyi-knowledge-api
ruoyi-system-api
+ ruoyi-workflow-api
diff --git a/ruoyi-modules-api/ruoyi-chat-api/pom.xml b/ruoyi-modules-api/ruoyi-chat-api/pom.xml
index c1af20a5..7e08c9af 100644
--- a/ruoyi-modules-api/ruoyi-chat-api/pom.xml
+++ b/ruoyi-modules-api/ruoyi-chat-api/pom.xml
@@ -16,7 +16,7 @@
17
17
UTF-8
- 1.0.0
+ 1.0.0-M7
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/ChatModel.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/ChatModel.java
index e8f0e308..34465613 100644
--- a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/ChatModel.java
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/ChatModel.java
@@ -1,6 +1,7 @@
package org.ruoyi.domain;
+import com.alibaba.excel.annotation.ExcelProperty;
import com.baomidou.mybatisplus.annotation.TableId;
import com.baomidou.mybatisplus.annotation.TableName;
import lombok.Data;
@@ -81,6 +82,11 @@ public class ChatModel extends BaseEntity {
*/
private Integer priority;
+ /**
+ * 模型供应商
+ */
+ private String ProviderName;
+
/**
* 备注
*/
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/McpInfo.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/McpInfo.java
new file mode 100644
index 00000000..f8392ad6
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/McpInfo.java
@@ -0,0 +1,67 @@
+package org.ruoyi.domain;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
+import com.baomidou.mybatisplus.annotation.TableName;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import org.ruoyi.annotation.DataColumn;
+import org.ruoyi.core.domain.BaseEntity;
+
+import java.util.Date;
+
+/**
+ * MCP对象 mcp_info
+ *
+ * @author ageerle
+ * @date Sat Aug 09 16:50:58 CST 2025
+ */
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName("mcp_info")
+public class McpInfo extends BaseEntity {
+
+
+ /**
+ * id
+ */
+ @TableId(value = "mcp_id", type = IdType.AUTO)
+ private Integer mcpId;
+
+ /**
+ * 服务器名称
+ */
+ private String serverName;
+
+ /**
+ * 链接方式
+ */
+
+ private String transportType;
+
+ /**
+ * Command
+ */
+ private String command;
+
+ /**
+ * Args
+ */
+ private String arguments;
+
+ private String description;
+
+ /**
+ * Env
+ */
+ private String env;
+
+ /**
+ * 是否启用
+ */
+ private Boolean status;
+
+
+
+
+}
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/ChatModelBo.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/ChatModelBo.java
index b828515b..34ee975e 100644
--- a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/ChatModelBo.java
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/ChatModelBo.java
@@ -1,5 +1,6 @@
package org.ruoyi.domain.bo;
+import com.alibaba.excel.annotation.ExcelProperty;
import io.github.linpeilie.annotations.AutoMapper;
import jakarta.validation.constraints.NotBlank;
import jakarta.validation.constraints.NotNull;
@@ -85,6 +86,10 @@ public class ChatModelBo extends BaseEntity {
@NotBlank(message = "密钥不能为空", groups = { AddGroup.class, EditGroup.class })
private String apiKey;
+ /**
+ * 模型供应商
+ */
+ private String ProviderName;
/**
* 备注
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/McpInfoBo.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/McpInfoBo.java
new file mode 100644
index 00000000..2228fb99
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/bo/McpInfoBo.java
@@ -0,0 +1,59 @@
+package org.ruoyi.domain.bo;
+
+import io.github.linpeilie.annotations.AutoMapper;
+import jakarta.validation.constraints.NotNull;
+import lombok.Data;
+import org.ruoyi.domain.McpInfo;
+
+
+import java.io.Serializable;
+
+/**
+ * MCP业务对象 mcp_info
+ *
+ * @author ageerle
+ * @date Sat Aug 09 16:50:58 CST 2025
+ */
+@Data
+
+@AutoMapper(target = McpInfo.class, reverseConvertGenerate = false)
+public class McpInfoBo implements Serializable {
+
+ /**
+ * id
+ */
+ @NotNull(message = "id不能为空" )
+ private Integer mcpId;
+
+ /**
+ * 服务器名称
+ */
+ private String serverName;
+
+ /**
+ * 链接方式
+ */
+ private String transportType;
+
+ /**
+ * Command
+ */
+ private String command;
+
+ /**
+ * Args
+ */
+ private String arguments;
+ private String description;
+ /**
+ * Env
+ */
+ private String env;
+
+ /**
+ * 是否启用
+ */
+ private Boolean status;
+
+
+}
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/ChatModelVo.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/ChatModelVo.java
index 0638c13a..062a378a 100644
--- a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/ChatModelVo.java
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/ChatModelVo.java
@@ -70,6 +70,11 @@ public class ChatModelVo implements Serializable {
@ExcelProperty(value = "是否显示")
private String modelShow;
+ /**
+ * 模型维度
+ */
+ private Integer dimension;
+
/**
* 系统提示词
*/
@@ -95,6 +100,12 @@ public class ChatModelVo implements Serializable {
@ExcelProperty(value = "优先级")
private Integer priority;
+ /**
+ * 模型供应商
+ */
+ @ExcelProperty(value = "模型供应商")
+ private String ProviderName;
+
/**
* 备注
*/
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/McpInfoVo.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/McpInfoVo.java
new file mode 100644
index 00000000..b5cc27c1
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/domain/vo/McpInfoVo.java
@@ -0,0 +1,65 @@
+package org.ruoyi.domain.vo;
+
+import com.alibaba.excel.annotation.ExcelIgnoreUnannotated;
+import com.alibaba.excel.annotation.ExcelProperty;
+import io.github.linpeilie.annotations.AutoMapper;
+import lombok.Data;
+import org.ruoyi.common.excel.annotation.ExcelDictFormat;
+import org.ruoyi.common.excel.convert.ExcelDictConvert;
+import org.ruoyi.domain.McpInfo;
+
+import java.io.Serializable;
+
+
+/**
+ * MCP视图对象 mcp_info
+ *
+ * @author jiyi
+ * @date Sat Aug 09 16:50:58 CST 2025
+ */
+@Data
+@ExcelIgnoreUnannotated
+@AutoMapper(target = McpInfo.class)
+public class McpInfoVo implements Serializable {
+ private Integer mcpId;
+
+ /**
+ * 服务器名称
+ */
+ @ExcelProperty(value = "服务器名称")
+ private String serverName;
+
+ /**
+ * 链接方式
+ */
+ @ExcelProperty(value = "链接方式", converter = ExcelDictConvert.class)
+ @ExcelDictFormat(dictType = "mcp_transport_type")
+ private String transportType;
+
+ /**
+ * Command
+ */
+ @ExcelProperty(value = "Command")
+ private String command;
+
+ /**
+ * Args
+ */
+ @ExcelProperty(value = "Args")
+ private String arguments;
+ @ExcelProperty(value = "Description")
+ private String description;
+ /**
+ * Env
+ */
+ @ExcelProperty(value = "Env")
+ private String env;
+
+ /**
+ * 是否启用
+ */
+ @ExcelProperty(value = "是否启用")
+ private Boolean status;
+
+
+}
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/mapper/McpInfoMapper.java b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/mapper/McpInfoMapper.java
new file mode 100644
index 00000000..fb26aaac
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/java/org/ruoyi/mapper/McpInfoMapper.java
@@ -0,0 +1,33 @@
+package org.ruoyi.mapper;
+
+
+import org.apache.ibatis.annotations.*;
+import org.ruoyi.core.mapper.BaseMapperPlus;
+import org.ruoyi.domain.McpInfo;
+import org.ruoyi.domain.vo.McpInfoVo;
+
+import java.util.List;
+
+/**
+ * MCPMapper接口
+ *
+ * @author jiuyi
+ * @date Sat Aug 09 16:50:58 CST 2025
+ */
+@Mapper
+public interface McpInfoMapper extends BaseMapperPlus {
+ @Select("SELECT * FROM mcp_info WHERE server_name = #{serverName}")
+ McpInfo selectByServerName(@Param("serverName") String serverName);
+
+ @Select("SELECT * FROM mcp_info WHERE status = 1")
+ List selectActiveServers();
+
+ @Select("SELECT server_name FROM mcp_info WHERE status = 1")
+ List selectActiveServerNames();
+
+ @Update("UPDATE mcp_info SET status = #{status} WHERE server_name = #{serverName}")
+ int updateActiveStatus(@Param("serverName") String serverName, @Param("status") Boolean status);
+
+ @Delete("DELETE FROM mcp_info WHERE server_name = #{serverName}")
+ int deleteByServerName(@Param("serverName") String serverName);
+}
diff --git a/ruoyi-modules-api/ruoyi-chat-api/src/main/resources/mapper/McpInfoMapper.xml b/ruoyi-modules-api/ruoyi-chat-api/src/main/resources/mapper/McpInfoMapper.xml
new file mode 100644
index 00000000..f2a28e2a
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-chat-api/src/main/resources/mapper/McpInfoMapper.xml
@@ -0,0 +1,7 @@
+
+
+
+
+
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/pom.xml b/ruoyi-modules-api/ruoyi-knowledge-api/pom.xml
index f8082a67..fc7e01bc 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/pom.xml
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/pom.xml
@@ -74,6 +74,18 @@
1.19.6
+
+ io.milvus
+ milvus-sdk-java
+ 2.6.4
+
+
+
+
+ dev.langchain4j
+ langchain4j-milvus
+
+
dev.langchain4j
langchain4j-open-ai
@@ -101,11 +113,10 @@
commons-compress
-
- com.alibaba
- dashscope-sdk-java
- 2.19.0
-
+
+ org.ruoyi
+ ruoyi-chat-api
+
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeInfo.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeInfo.java
index 627cd3c6..f892b3c1 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeInfo.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/KnowledgeInfo.java
@@ -83,6 +83,11 @@ public class KnowledgeInfo extends BaseEntity {
*/
private String vectorModelName;
+ /**
+ * 向量化模型id
+ */
+ private Long embeddingModelId;
+
/**
* 向量化模型名称
*/
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeInfoBo.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeInfoBo.java
index 39083871..af453492 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeInfoBo.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/KnowledgeInfoBo.java
@@ -92,7 +92,11 @@ public class KnowledgeInfoBo extends BaseEntity {
/**
* 向量化模型名称
*/
- @NotBlank(message = "向量模型不能为空", groups = { AddGroup.class, EditGroup.class })
+ private Long embeddingModelId;
+
+ /**
+ * 向量化模型名称
+ */
private String embeddingModelName;
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/QueryVectorBo.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/QueryVectorBo.java
index ff3a26e5..0d5b2e6e 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/QueryVectorBo.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/QueryVectorBo.java
@@ -31,7 +31,12 @@ public class QueryVectorBo {
private String vectorModelName;
/**
- * 向量化模型名称
+ * 向量化模型ID
+ */
+ private Long embeddingModelId;
+
+ /**
+ * 向量化模型ID
*/
private String embeddingModelName;
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/StoreEmbeddingBo.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/StoreEmbeddingBo.java
index e4d8c381..eedfe4d6 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/StoreEmbeddingBo.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/bo/StoreEmbeddingBo.java
@@ -32,9 +32,14 @@ public class StoreEmbeddingBo {
private List fids;
/**
- * 向量库模型名称
+ * 向量库名称
*/
- private String vectorModelName;
+ private String vectorStoreName;
+
+ /**
+ * 向量化模型id
+ */
+ private Long embeddingModelId;
/**
* 向量化模型名称
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeInfoVo.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeInfoVo.java
index 2dc73d45..c907e940 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeInfoVo.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/domain/vo/KnowledgeInfoVo.java
@@ -101,6 +101,11 @@ public class KnowledgeInfoVo implements Serializable {
*/
private String vectorModelName;
+ /**
+ * 向量化模型id
+ */
+ private Long embeddingModelId;
+
/**
* 向量化模型名称
*/
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/BaseEmbedModelService.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/BaseEmbedModelService.java
new file mode 100644
index 00000000..9b3d0021
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/BaseEmbedModelService.java
@@ -0,0 +1,26 @@
+package org.ruoyi.embedding;
+
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import org.ruoyi.domain.vo.ChatModelVo;
+import org.ruoyi.embedding.model.ModalityType;
+
+import java.util.Set;
+
+/**
+ * BaseEmbedModelService 接口，扩展了 EmbeddingModel 接口
+ * 该接口定义了嵌入模型服务的基本配置和功能方法
+ */
+public interface BaseEmbedModelService extends EmbeddingModel {
+ /**
+ * 根据配置信息配置嵌入模型
+ * @param config 包含模型配置信息的 ChatModelVo 对象
+ */
+ void configure(ChatModelVo config);
+
+ /**
+ * 获取当前嵌入模型支持的所有模态类型
+ * @return 返回支持的模态类型集合
+ */
+ Set getSupportedModalities();
+
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/EmbeddingModelFactory.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/EmbeddingModelFactory.java
new file mode 100644
index 00000000..3acf8a91
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/EmbeddingModelFactory.java
@@ -0,0 +1,120 @@
+package org.ruoyi.embedding;
+
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.ruoyi.domain.vo.ChatModelVo;
+import org.ruoyi.service.IChatModelService;
+import org.springframework.beans.factory.NoSuchBeanDefinitionException;
+import org.springframework.context.ApplicationContext;
+import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+/**
+ * 嵌入模型工厂服务类
+ * 负责创建和管理各种嵌入模型实例
+ */
+@Service
+@RequiredArgsConstructor
+@Slf4j
+public class EmbeddingModelFactory {
+
+ private final ApplicationContext applicationContext;
+
+ private final IChatModelService chatModelService;
+
+ // 模型缓存，使用ConcurrentHashMap保证线程安全
+ private final Map modelCache = new ConcurrentHashMap<>();
+
+ /**
+ * 创建嵌入模型实例
+ * 如果模型已存在于缓存中，则直接返回；否则创建新的实例
+ *
+ * @param embeddingModelName 嵌入模型名称
+ * @param dimension 模型维度大小
+ */
+ public BaseEmbedModelService createModel(String embeddingModelName, Integer dimension) {
+ return modelCache.computeIfAbsent(embeddingModelName, name -> {
+ ChatModelVo modelConfig = chatModelService.selectModelByName(embeddingModelName);
+ if (modelConfig == null) {
+ throw new IllegalArgumentException("未找到模型配置，name=" + name);
+ }
+ if (modelConfig.getDimension() != null) {
+ modelConfig.setDimension(dimension);
+ }
+ return createModelInstance(modelConfig.getProviderName(), modelConfig);
+ });
+ }
+
+ /**
+ * 检查模型是否支持多模态
+ *
+ * @param embeddingModelName 嵌入模型名称
+ * @return boolean 如果模型支持多模态则返回true，否则返回false
+ */
+ public boolean isMultimodalModel(String embeddingModelName) {
+ return createModel(embeddingModelName, null) instanceof MultiModalEmbedModelService;
+ }
+
+ /**
+ * 创建多模态嵌入模型实例
+ *
+ * @param embeddingModelName 嵌入模型名称
+ * @return MultiModalEmbedModelService 多模态嵌入模型服务实例
+ * @throws IllegalArgumentException 当模型不支持多模态时抛出
+ */
+ public MultiModalEmbedModelService createMultimodalModel(String embeddingModelName) {
+ BaseEmbedModelService model = createModel(embeddingModelName, null);
+ if (model instanceof MultiModalEmbedModelService) {
+ return (MultiModalEmbedModelService) model;
+ }
+ throw new IllegalArgumentException("该模型不支持多模态");
+ }
+
+ /**
+ * 刷新模型缓存
+ * 根据给定的嵌入模型ID从缓存中移除对应的模型
+ *
+ * @param embeddingModelId 嵌入模型的唯一标识ID
+ */
+ public void refreshModel(Long embeddingModelId) {
+ // 从模型缓存中移除指定ID的模型
+ modelCache.remove(embeddingModelId);
+ }
+
+ /**
+ * 获取所有支持模型工厂的列表
+ *
+ * @return List 支持的模型工厂名称列表
+ */
+ public List getSupportedFactories() {
+ return new ArrayList<>(applicationContext.getBeansOfType(BaseEmbedModelService.class)
+ .keySet());
+ }
+
+ /**
+ * 创建具体的模型实例
+ * 根据提供的工厂名称和配置信息创建并配置模型实例
+ *
+ * @param factory 工厂名称，用于标识模型类型
+ * @param config 模型配置信息
+ * @return BaseEmbedModelService 配置好的模型实例
+ * @throws IllegalArgumentException 当无法获取指定的模型实例时抛出
+ */
+ private BaseEmbedModelService createModelInstance(String factory, ChatModelVo config) {
+ try {
+ // 从Spring上下文中获取模型实例
+ BaseEmbedModelService model = applicationContext.getBean(factory, BaseEmbedModelService.class);
+ // 配置模型参数
+ model.configure(config);
+ log.info("成功创建嵌入模型: factory={}, modelId={}", config.getProviderName(), config.getId());
+
+ return model;
+ } catch (NoSuchBeanDefinitionException e) {
+ throw new IllegalArgumentException("获取不到嵌入模型: " + factory, e);
+ }
+ }
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/MultiModalEmbedModelService.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/MultiModalEmbedModelService.java
new file mode 100644
index 00000000..062ff00f
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/MultiModalEmbedModelService.java
@@ -0,0 +1,35 @@
+package org.ruoyi.embedding;
+
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.model.output.Response;
+import org.ruoyi.embedding.model.MultiModalInput;
+
+
+/**
+ * 多模态嵌入模型服务接口，继承自基础嵌入模型服务
+ * 该接口提供了处理图像、视频以及多模态数据并转换为嵌入向量的功能
+ */
+public interface MultiModalEmbedModelService extends BaseEmbedModelService {
+ /**
+ * 将图像数据转换为嵌入向量
+ * @param imageDataUrl 图像的地址，必须是公开可访问的URL
+ * @return 包含嵌入向量的响应对象，可能包含状态信息和嵌入结果
+ */
+ Response embedImage(String imageDataUrl);
+
+ /**
+ * 将视频数据转换为嵌入向量
+ * @param videoDataUrl 视频的地址，必须是公开可访问的URL
+ * @return 包含嵌入向量的响应对象，可能包含状态信息和嵌入结果
+ */
+ Response embedVideo(String videoDataUrl);
+
+
+ /**
+ * 处理多模态输入并返回嵌入向量的方法
+ *
+ * @param input 包含多种模态信息（如图像、文本等）的输入对象
+ * @return Response 包含嵌入向量的响应对象，Embedding通常表示输入数据的向量表示
+ */
+ Response embedMultiModal(MultiModalInput input);
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/AliBaiLianBaseEmbedProvider.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/AliBaiLianBaseEmbedProvider.java
new file mode 100644
index 00000000..1511a0fe
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/AliBaiLianBaseEmbedProvider.java
@@ -0,0 +1,14 @@
+package org.ruoyi.embedding.impl;
+
+
+import org.springframework.stereotype.Component;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-09-30-下午3:00
+ * @Description: 阿里百炼基础嵌入模型（兼容openai）
+ */
+@Component("alibailian")
+public class AliBaiLianBaseEmbedProvider extends OpenAiEmbeddingProvider{
+
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/AliBaiLianMultiEmbeddingProvider.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/AliBaiLianMultiEmbeddingProvider.java
new file mode 100644
index 00000000..ad3e8374
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/AliBaiLianMultiEmbeddingProvider.java
@@ -0,0 +1,281 @@
+package org.ruoyi.embedding.impl;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.output.Response;
+import dev.langchain4j.model.output.TokenUsage;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.*;
+import org.ruoyi.domain.vo.ChatModelVo;
+import org.ruoyi.embedding.MultiModalEmbedModelService;
+import org.ruoyi.embedding.model.AliyunMultiModalEmbedRequest;
+import org.ruoyi.embedding.model.AliyunMultiModalEmbedResponse;
+import org.ruoyi.embedding.model.ModalityType;
+import org.ruoyi.embedding.model.MultiModalInput;
+import org.springframework.stereotype.Component;
+
+import java.io.IOException;
+import java.util.*;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * 阿里云百炼多模态嵌入模型服务实现类
+ * 实现了MultiModalEmbedModelService接口，提供文本、图像和视频的嵌入向量生成服务
+ */
+@Component("bailianMultiModel")
+@Slf4j
+public class AliBaiLianMultiEmbeddingProvider implements MultiModalEmbedModelService {
+ private ChatModelVo chatModelVo;
+
+ private final OkHttpClient okHttpClient;
+
+ /**
+ * 构造函数，初始化HTTP客户端
+ * 设置连接超时、读取超时和写入超时时间
+ */
+ public AliBaiLianMultiEmbeddingProvider() {
+ this.okHttpClient = new OkHttpClient.Builder()
+ .connectTimeout(30, TimeUnit.SECONDS)
+ .readTimeout(60, TimeUnit.SECONDS)
+ .writeTimeout(30, TimeUnit.SECONDS)
+ .build();
+ }
+
+ /**
+ * 图像嵌入向量生成
+ * @param imageDataUrl 图像数据的URL
+ * @return 包含图像嵌入向量的Response对象
+ */
+ @Override
+ public Response embedImage(String imageDataUrl) {
+ return embedSingleModality("image", imageDataUrl);
+ }
+
+ /**
+ * 视频嵌入向量生成
+ * @param videoDataUrl 视频数据的URL
+ * @return 包含视频嵌入向量的Response对象
+ */
+ @Override
+ public Response embedVideo(String videoDataUrl) {
+ return embedSingleModality("video", videoDataUrl);
+ }
+
+ /**
+ * 多模态嵌入向量生成
+ * 支持同时处理文本、图像和视频等多种模态的数据
+ * @param input 包含多种模态输入的对象
+ * @return 包含多模态嵌入向量的Response对象
+ */
+ @Override
+ public Response embedMultiModal(MultiModalInput input) {
+ try {
+ // 构建请求内容
+ List> contents = buildContentMap(input);
+ if (contents.isEmpty()) {
+ throw new IllegalArgumentException("至少提供一种模态的内容");
+ }
+
+ // 构建请求
+ AliyunMultiModalEmbedRequest request = buildRequest(contents, chatModelVo);
+ AliyunMultiModalEmbedResponse resp = executeRequest(request, chatModelVo);
+
+ // 转换为 embeddings
+ Response> response = toEmbeddings(resp);
+ List embeddings = response.content();
+
+ if (embeddings.isEmpty()) {
+ log.warn("阿里云混合模态嵌入返回为空");
+ return Response.from(Embedding.from(new float[0]), response.tokenUsage());
+ }
+
+ // 多模态通常取第一个向量作为代表，也可以根据业务场景返回多个
+ return Response.from(embeddings.get(0), response.tokenUsage());
+
+ } catch (Exception e) {
+ log.error("阿里云混合模态嵌入失败", e);
+ throw new IllegalArgumentException("阿里云混合模态嵌入失败", e);
+ }
+ }
+
+ /**
+ * 配置模型参数
+ * @param config 模型配置信息
+ */
+ @Override
+ public void configure(ChatModelVo config) {
+ this.chatModelVo = config;
+ }
+
+ /**
+ * 获取支持的模态类型
+ * @return 支持的模态类型集合
+ */
+ @Override
+ public Set getSupportedModalities() {
+ return Set.of(ModalityType.TEXT, ModalityType.VIDEO, ModalityType.IMAGE);
+ }
+
+ /**
+ * 批量文本嵌入向量生成
+ * @param textSegments 文本段列表
+ * @return 包含所有文本嵌入向量的Response对象
+ */
+ @Override
+ public Response> embedAll(List textSegments) {
+ if (textSegments.isEmpty()) return Response.from(Collections.emptyList());
+
+ try {
+ List> contents = new ArrayList<>();
+ for (TextSegment segment : textSegments) {
+ contents.add(Map.of("text", segment.text()));
+ }
+
+ AliyunMultiModalEmbedRequest request = buildRequest(contents, chatModelVo);
+ AliyunMultiModalEmbedResponse resp = executeRequest(request, chatModelVo);
+
+ return toEmbeddings(resp);
+ } catch (Exception e) {
+ log.error("阿里云文本嵌入失败", e);
+ throw new IllegalArgumentException("阿里云文本嵌入失败", e);
+ }
+ }
+
+ /**
+ * 单模态嵌入（图片/视频/单条文本）复用方法
+ * @param key 模态类型（image/video/text）
+ * @param dataUrl 数据URL
+ * @return 包含嵌入向量的Response对象
+ */
+
+ public Response embedSingleModality(String key, String dataUrl) {
+ try {
+ AliyunMultiModalEmbedRequest request = buildRequest(List.of(Map.of(key, dataUrl)), chatModelVo);
+ AliyunMultiModalEmbedResponse resp = executeRequest(request, chatModelVo);
+
+ Response> response = toEmbeddings(resp);
+ List embeddings = response.content();
+
+ if (embeddings.isEmpty()) {
+ log.warn("阿里云 {} 嵌入返回为空", key);
+ return Response.from(Embedding.from(new float[0]), response.tokenUsage());
+ }
+
+ return Response.from(embeddings.get(0), response.tokenUsage());
+ } catch (Exception e) {
+ log.error("阿里云 {} 嵌入失败", key, e);
+ throw new IllegalArgumentException("阿里云 " + key + " 嵌入失败", e);
+ }
+ }
+
+ /**
+ * 构建请求对象
+ * @param contents 请求内容列表
+ * @param chatModelVo 模型配置信息
+ * @return 构建好的请求对象
+ */
+ private AliyunMultiModalEmbedRequest buildRequest(List> contents, ChatModelVo chatModelVo) {
+ if (contents.isEmpty()) throw new IllegalArgumentException("请求内容不能为空");
+ return AliyunMultiModalEmbedRequest.create(chatModelVo.getModelName(), contents);
+ }
+
+ /**
+ * 执行 HTTP 请求并解析响应
+ * @param request 请求对象
+ * @param chatModelVo 模型配置信息
+ * @return API响应对象
+ * @throws IOException IO异常
+ */
+ private AliyunMultiModalEmbedResponse executeRequest(AliyunMultiModalEmbedRequest request, ChatModelVo chatModelVo) throws IOException {
+ String jsonBody = request.toJson();
+ RequestBody body = RequestBody.create(jsonBody, MediaType.get("application/json"));
+
+ Request httpRequest = new Request.Builder()
+ .url(chatModelVo.getApiHost())
+ .addHeader("Authorization", "Bearer " + chatModelVo.getApiKey())
+ .post(body)
+ .build();
+
+ try (okhttp3.Response response = okHttpClient.newCall(httpRequest).execute()) {
+ if (!response.isSuccessful()) {
+ String err = response.body() != null ? response.body().string() : "无错误信息";
+ throw new IllegalArgumentException("API调用失败: " + response.code() + " - " + err, null);
+ }
+
+ ResponseBody responseBody = response.body();
+ if (responseBody == null) throw new IllegalArgumentException("响应体为空", null);
+
+ return parseEmbeddingsFromResponse(responseBody.string());
+ }
+ }
+
+ /**
+ * 解析嵌入向量列表
+ * @param responseBody API响应的JSON字符串
+ * @return 嵌入向量响应对象
+ * @throws IOException IO异常
+ */
+ private AliyunMultiModalEmbedResponse parseEmbeddingsFromResponse(String responseBody) throws IOException {
+ ObjectMapper objectMapper1 = new ObjectMapper();
+ return objectMapper1.readValue(responseBody, AliyunMultiModalEmbedResponse.class);
+ }
+
+ /**
+ * 构建 API 请求内容 Map
+ * @param input 多模态输入对象
+ * @return 包含各种模态内容的Map列表
+ */
+ private List> buildContentMap(MultiModalInput input) {
+ List> contents = new ArrayList<>();
+
+ if (input.getText() != null && !input.getText().isBlank()) {
+ contents.add(Map.of("text", input.getText()));
+ }
+ if (input.getImageUrl() != null && !input.getImageUrl().isBlank()) {
+ contents.add(Map.of("image", input.getImageUrl()));
+ }
+ if (input.getVideoUrl() != null && !input.getVideoUrl().isBlank()) {
+ contents.add(Map.of("video", input.getVideoUrl()));
+ }
+ if (input.getMultiImageUrls() != null && input.getMultiImageUrls().length > 0) {
+ contents.add(Map.of("multi_images", Arrays.asList(input.getMultiImageUrls())));
+ }
+
+ return contents;
+ }
+
+ /**
+ * 将 API 原始响应解析为 LangChain4j 的 Response
+ * @param resp API原始响应对象
+ * @return 包含嵌入向量和token使用情况的Response对象
+ */
+ private Response> toEmbeddings(AliyunMultiModalEmbedResponse resp) {
+ if (resp == null || resp.output() == null || resp.output().embeddings() == null) {
+ return Response.from(Collections.emptyList());
+ }
+
+ // 转换 double -> float
+ List embeddings = resp.output().embeddings().stream()
+ .map(item -> {
+ float[] vector = new float[item.embedding().size()];
+ for (int i = 0; i < item.embedding().size(); i++) {
+ vector[i] = item.embedding().get(i).floatValue();
+ }
+ return Embedding.from(vector);
+ })
+ .toList();
+
+ // 构建 TokenUsage
+ TokenUsage tokenUsage = null;
+ if (resp.usage() != null) {
+ tokenUsage = new TokenUsage(
+ resp.usage().input_tokens(),
+ resp.usage().image_tokens(),
+ resp.usage().input_tokens() +resp.usage().image_tokens()
+ );
+ }
+
+ return Response.from(embeddings, tokenUsage);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/OllamaEmbeddingProvider.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/OllamaEmbeddingProvider.java
new file mode 100644
index 00000000..1a179be7
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/OllamaEmbeddingProvider.java
@@ -0,0 +1,42 @@
+package org.ruoyi.embedding.impl;
+
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.ollama.OllamaEmbeddingModel;
+import dev.langchain4j.model.output.Response;
+import org.ruoyi.domain.vo.ChatModelVo;
+import org.ruoyi.embedding.BaseEmbedModelService;
+import org.ruoyi.embedding.model.ModalityType;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-09-30-下午3:00
+ * @Description: Ollama嵌入模型
+ */
+@Component("ollama")
+public class OllamaEmbeddingProvider implements BaseEmbedModelService {
+ private ChatModelVo chatModelVo;
+
+ @Override
+ public void configure(ChatModelVo config) {
+ this.chatModelVo = config;
+ }
+ @Override
+ public Set getSupportedModalities() {
+ return Set.of(ModalityType.TEXT);
+ }
+
+ // ollama不能设置embedding维度，使用milvus时请注意！！创建向量表时需要先设定维度大小
+ @Override
+ public Response> embedAll(List textSegments) {
+ return OllamaEmbeddingModel.builder()
+ .baseUrl(chatModelVo.getApiHost())
+ .modelName(chatModelVo.getModelName())
+ .build()
+ .embedAll(textSegments);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/OpenAiEmbeddingProvider.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/OpenAiEmbeddingProvider.java
new file mode 100644
index 00000000..8a0c9f62
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/OpenAiEmbeddingProvider.java
@@ -0,0 +1,44 @@
+package org.ruoyi.embedding.impl;
+
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
+import dev.langchain4j.model.output.Response;
+import org.ruoyi.domain.vo.ChatModelVo;
+import org.ruoyi.embedding.BaseEmbedModelService;
+import org.ruoyi.embedding.model.ModalityType;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-09-30-下午3:59
+ * @Description: OpenAi嵌入模型
+ */
+@Component("openai")
+public class OpenAiEmbeddingProvider implements BaseEmbedModelService {
+ protected ChatModelVo chatModelVo;
+
+ @Override
+ public void configure(ChatModelVo config) {
+ this.chatModelVo = config;
+ }
+
+ @Override
+ public Set getSupportedModalities() {
+ return Set.of(ModalityType.TEXT);
+ }
+
+ @Override
+ public Response> embedAll(List textSegments) {
+ return OpenAiEmbeddingModel.builder()
+ .baseUrl(chatModelVo.getApiHost())
+ .apiKey(chatModelVo.getApiKey())
+ .modelName(chatModelVo.getModelName())
+ .dimensions(chatModelVo.getDimension())
+ .build()
+ .embedAll(textSegments);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/SiliconFlowEmbeddingProvider.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/SiliconFlowEmbeddingProvider.java
new file mode 100644
index 00000000..e0ccd2b9
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/SiliconFlowEmbeddingProvider.java
@@ -0,0 +1,18 @@
+package org.ruoyi.embedding.impl;
+
+
+import org.ruoyi.embedding.BaseEmbedModelService;
+import org.ruoyi.embedding.model.ModalityType;
+import org.springframework.stereotype.Component;
+
+import java.util.Set;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-09-30-下午3:59
+ * @Description: 硅基流动（兼容 OpenAi）
+ */
+@Component("siliconflow")
+public class SiliconFlowEmbeddingProvider extends OpenAiEmbeddingProvider {
+
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/ZhiPuAiEmbeddingProvider.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/ZhiPuAiEmbeddingProvider.java
new file mode 100644
index 00000000..621e47c6
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/impl/ZhiPuAiEmbeddingProvider.java
@@ -0,0 +1,44 @@
+package org.ruoyi.embedding.impl;
+
+import dev.langchain4j.community.model.zhipu.ZhipuAiEmbeddingModel;
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.output.Response;
+import org.ruoyi.domain.vo.ChatModelVo;
+import org.ruoyi.embedding.BaseEmbedModelService;
+import org.ruoyi.embedding.model.ModalityType;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-09-30-下午4:02
+ * @Description: 智谱AI
+ */
+@Component("zhipu")
+public class ZhiPuAiEmbeddingProvider implements BaseEmbedModelService {
+ private ChatModelVo chatModelVo;
+
+ @Override
+ public void configure(ChatModelVo config) {
+ this.chatModelVo = config;
+ }
+
+ @Override
+ public Set getSupportedModalities() {
+ return Set.of();
+ }
+
+ @Override
+ public Response> embedAll(List textSegments) {
+ return ZhipuAiEmbeddingModel.builder()
+ .baseUrl(chatModelVo.getApiHost())
+ .apiKey(chatModelVo.getApiKey())
+ .model(chatModelVo.getModelName())
+ .dimensions(chatModelVo.getDimension())
+ .build()
+ .embedAll(textSegments);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/AliyunMultiModalEmbedRequest.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/AliyunMultiModalEmbedRequest.java
new file mode 100644
index 00000000..eb880588
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/AliyunMultiModalEmbedRequest.java
@@ -0,0 +1,44 @@
+package org.ruoyi.embedding.model;
+
+import org.ruoyi.common.json.utils.JsonUtils;
+import lombok.Data;
+
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-10-1-上午10:00
+ * @Description: 阿里云多模态嵌入请求
+ */
+@Data
+public class AliyunMultiModalEmbedRequest {
+ private String model;
+ private Input input;
+
+ /**
+ * 表示输入数据的记录类(Record)
+ * 该类用于封装一个包含多个映射关系列表的输入数据结构
+ *
+ * @param contents 包含多个Map的列表，每个Map中存储String类型的键和Object类型的值
+ */
+ public record Input(List> contents) { }
+
+ /**
+ * 创建请求对象
+ */
+ public static AliyunMultiModalEmbedRequest create(String modelName, List> contents) {
+ AliyunMultiModalEmbedRequest request = new AliyunMultiModalEmbedRequest();
+ request.setModel(modelName);
+ Input input = new Input(contents);
+ request.setInput(input);
+ return request;
+ }
+
+ /**
+ * 转换为JSON字符串
+ */
+ public String toJson() {
+ return JsonUtils.toJsonString(this);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/AliyunMultiModalEmbedResponse.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/AliyunMultiModalEmbedResponse.java
new file mode 100644
index 00000000..03446d47
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/AliyunMultiModalEmbedResponse.java
@@ -0,0 +1,44 @@
+package org.ruoyi.embedding.model;
+
+import java.util.List;
+
+/**
+ * 阿里云多模态嵌入 API 响应数据模型
+ */
+public record AliyunMultiModalEmbedResponse(
+ Output output, // 输出结果对象
+ String request_id, // 请求唯一标识
+ String code, // 错误码
+ String message, // 错误消息
+ Usage usage // 用量信息
+) {
+
+ /**
+ * 输出对象，包含嵌入向量结果
+ */
+ public record Output(
+ List embeddings // 嵌入向量列表
+ ) {
+ }
+
+ /**
+ * 单个嵌入向量条目
+ */
+ public record EmbeddingItem(
+ int index, // 输入内容的索引
+ List embedding, // 生成的 1024 维向量
+ String type // 输入的类型 (text/image/video/multi_images)
+ ) {
+ }
+
+ /**
+ * 用量统计信息
+ */
+ public record Usage(
+ int input_tokens, // 本次请求输入的 Token 数量
+ int image_tokens, // 本次请求输入的图像 Token 数量
+ int image_count, // 本次请求输入的图像数量
+ int duration // 本次请求输入的视频时长（秒）
+ ) {
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/ModalityType.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/ModalityType.java
new file mode 100644
index 00000000..782aac62
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/ModalityType.java
@@ -0,0 +1,8 @@
+package org.ruoyi.embedding.model;
+
+/**
+ * 模态类型
+ */
+public enum ModalityType {
+ TEXT, IMAGE, AUDIO, VIDEO, MULTI
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/MultiModalInput.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/MultiModalInput.java
new file mode 100644
index 00000000..f2a31e98
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/embedding/model/MultiModalInput.java
@@ -0,0 +1,71 @@
+package org.ruoyi.embedding.model;
+
+import cn.hutool.core.util.ArrayUtil;
+import cn.hutool.core.util.StrUtil;
+import lombok.Builder;
+import lombok.Data;
+
+/**
+ * @Author: Robust_H
+ * @Date: 2025-09-30-下午2:13
+ * @Description: 多模态输入
+ */
+@Data
+@Builder
+public class MultiModalInput {
+ private String text;
+ private byte[] imageData;
+ private byte[] videoData;
+ private String imageMimeType;
+ private String videoMimeType;
+ private String[] multiImageUrls;
+ private String imageUrl;
+ private String videoUrl;
+
+ /**
+ * 检查是否有文本内容
+ */
+ public boolean hasText() {
+ return StrUtil.isNotBlank(text);
+ }
+
+ /**
+ * 检查是否有图片内容
+ */
+ public boolean hasImage() {
+ return ArrayUtil.isNotEmpty(imageData) || StrUtil.isNotBlank(imageUrl);
+ }
+
+ /**
+ * 检查是否有视频内容
+ */
+ public boolean hasVideo() {
+ return ArrayUtil.isNotEmpty(videoData) || StrUtil.isNotBlank(videoUrl);
+ }
+
+ /**
+ * 检查是否有多图片
+ */
+ public boolean hasMultiImages() {
+ return ArrayUtil.isNotEmpty(multiImageUrls);
+ }
+
+ /**
+ * 检查是否有任何内容
+ */
+ public boolean hasAnyContent() {
+ return hasText() || hasImage() || hasVideo() || hasMultiImages();
+ }
+
+ /**
+ * 获取内容的数量
+ */
+ public int getContentCount() {
+ int count = 0;
+ if (hasText()) count++;
+ if (hasImage()) count++;
+ if (hasVideo()) count++;
+ if (hasMultiImages()) count++;
+ return count;
+ }
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/VectorStoreService.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/VectorStoreService.java
index 29bfecc3..2937fc44 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/VectorStoreService.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/VectorStoreService.java
@@ -1,5 +1,6 @@
package org.ruoyi.service;
+import org.ruoyi.common.core.exception.ServiceException;
import org.ruoyi.domain.bo.QueryVectorBo;
import org.ruoyi.domain.bo.StoreEmbeddingBo;
@@ -11,15 +12,15 @@ import java.util.List;
*/
public interface VectorStoreService {
- void storeEmbeddings(StoreEmbeddingBo storeEmbeddingBo);
+ void storeEmbeddings(StoreEmbeddingBo storeEmbeddingBo) throws ServiceException;
List getQueryVector(QueryVectorBo queryVectorBo);
- void createSchema(String kid,String modelName);
+ void createSchema(String kid, String embeddingModelName);
- void removeById(String id,String modelName);
+ void removeById(String id,String modelName) throws ServiceException;
- void removeByDocId(String docId, String kid);
+ void removeByDocId(String docId, String kid) throws ServiceException;
- void removeByFid(String fid, String kid);
+ void removeByFid(String fid, String kid) throws ServiceException;
}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
index 799ce729..c9aeeab4 100644
--- a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
@@ -1,37 +1,14 @@
package org.ruoyi.service.impl;
-import cn.hutool.json.JSONObject;
-import com.google.protobuf.ServiceException;
-import dev.langchain4j.data.embedding.Embedding;
-import dev.langchain4j.data.segment.TextSegment;
-import dev.langchain4j.model.embedding.EmbeddingModel;
-import dev.langchain4j.model.ollama.OllamaEmbeddingModel;
-import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
-import dev.langchain4j.store.embedding.EmbeddingMatch;
-import dev.langchain4j.store.embedding.EmbeddingSearchRequest;
-import dev.langchain4j.store.embedding.EmbeddingStore;
-import dev.langchain4j.store.embedding.weaviate.WeaviateEmbeddingStore;
-import io.weaviate.client.Config;
-import io.weaviate.client.WeaviateClient;
-import io.weaviate.client.base.Result;
-import io.weaviate.client.v1.batch.api.ObjectsBatchDeleter;
-import io.weaviate.client.v1.batch.model.BatchDeleteResponse;
-import io.weaviate.client.v1.filters.Operator;
-import io.weaviate.client.v1.filters.WhereFilter;
-import io.weaviate.client.v1.graphql.model.GraphQLResponse;
-import io.weaviate.client.v1.schema.model.Property;
-import io.weaviate.client.v1.schema.model.Schema;
-import io.weaviate.client.v1.schema.model.WeaviateClass;
import lombok.RequiredArgsConstructor;
-import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
-import org.ruoyi.common.core.service.ConfigService;
import org.ruoyi.domain.bo.QueryVectorBo;
import org.ruoyi.domain.bo.StoreEmbeddingBo;
import org.ruoyi.service.VectorStoreService;
+import org.ruoyi.service.strategy.VectorStoreStrategyFactory;
+import org.springframework.context.annotation.Primary;
import org.springframework.stereotype.Service;
import java.util.*;
-import java.util.stream.Collectors;
/**
* 向量库管理
@@ -39,235 +16,61 @@ import java.util.stream.Collectors;
* @author ageer
*/
@Service
+@Primary
@Slf4j
@RequiredArgsConstructor
public class VectorStoreServiceImpl implements VectorStoreService {
- private final ConfigService configService;
+ private final VectorStoreStrategyFactory strategyFactory;
-// private EmbeddingStore embeddingStore;
- private WeaviateClient client;
+ /**
+ * 获取当前配置的向量库策略
+ */
+ private VectorStoreService getCurrentStrategy() {
+ return strategyFactory.getStrategy();
+ }
@Override
public void createSchema(String kid, String modelName) {
- String protocol = configService.getConfigValue("weaviate", "protocol");
- String host = configService.getConfigValue("weaviate", "host");
- String className = configService.getConfigValue("weaviate", "classname")+kid;
- // 创建 Weaviate 客户端
- client= new WeaviateClient(new Config(protocol, host));
- // 检查类是否存在，如果不存在就创建 schema
- Result schemaResult = client.schema().getter().run();
- Schema schema = schemaResult.getResult();
- boolean classExists = false;
- for (WeaviateClass weaviateClass : schema.getClasses()) {
- if (weaviateClass.getClassName().equals(className)) {
- classExists = true;
- break;
- }
- }
- if (!classExists) {
- // 类不存在，创建 schema
- WeaviateClass build = WeaviateClass.builder()
- .className(className)
- .vectorizer("none")
- .properties(
- List.of(Property.builder().name("text").dataType(Collections.singletonList("text")).build(),
- Property.builder().name("fid").dataType(Collections.singletonList("text")).build(),
- Property.builder().name("kid").dataType(Collections.singletonList("text")).build(),
- Property.builder().name("docId").dataType(Collections.singletonList("text")).build())
- )
- .build();
- Result createResult = client.schema().classCreator().withClass(build).run();
- if (createResult.hasErrors()) {
- log.error("Schema 创建失败: {}", createResult.getError());
- } else {
- log.info("Schema 创建成功: {}", className);
- }
- }
-// embeddingStore = WeaviateEmbeddingStore.builder()
-// .scheme(protocol)
-// .host(host)
-// .objectClass(className)
-// .scheme(protocol)
-// .avoidDups(true)
-// .consistencyLevel("ALL")
-// .build();
+ VectorStoreService strategy = getCurrentStrategy();
+ strategy.createSchema(kid, modelName);
}
@Override
public void storeEmbeddings(StoreEmbeddingBo storeEmbeddingBo) {
- createSchema(storeEmbeddingBo.getKid(), storeEmbeddingBo.getVectorModelName());
- EmbeddingModel embeddingModel = getEmbeddingModel(storeEmbeddingBo.getEmbeddingModelName(),
- storeEmbeddingBo.getApiKey(), storeEmbeddingBo.getBaseUrl());
- List chunkList = storeEmbeddingBo.getChunkList();
- List fidList = storeEmbeddingBo.getFids();
- String kid = storeEmbeddingBo.getKid();
- String docId = storeEmbeddingBo.getDocId();
- log.info("向量存储条数记录: " + chunkList.size());
- long startTime = System.currentTimeMillis();
- for (int i = 0; i < chunkList.size(); i++) {
- String text = chunkList.get(i);
- String fid = fidList.get(i);
- Embedding embedding = embeddingModel.embed(text).content();
- Map properties = Map.of(
- "text", text,
- "fid",fid,
- "kid", kid,
- "docId", docId
- );
- Float[] vector = toObjectArray(embedding.vector());
- client.data().creator()
- .withClassName("LocalKnowledge" + kid) // 注意替换成实际类名
- .withProperties(properties)
- .withVector(vector)
- .run();
- }
- long endTime = System.currentTimeMillis();
- log.info("向量存储完成消耗时间："+ (endTime-startTime)/1000+"秒");
+ log.info("存储向量数据: kid={}, docId={}, 数据条数={}",
+ storeEmbeddingBo.getKid(), storeEmbeddingBo.getDocId(), storeEmbeddingBo.getChunkList().size());
+ VectorStoreService strategy = getCurrentStrategy();
+ strategy.storeEmbeddings(storeEmbeddingBo);
}
- private static Float[] toObjectArray(float[] primitive) {
- Float[] result = new Float[primitive.length];
- for (int i = 0; i < primitive.length; i++) {
- result[i] = primitive[i]; // 自动装箱
- }
- return result;
- }
@Override
public List getQueryVector(QueryVectorBo queryVectorBo) {
- createSchema(queryVectorBo.getKid(), queryVectorBo.getVectorModelName());
- EmbeddingModel embeddingModel = getEmbeddingModel(queryVectorBo.getEmbeddingModelName(),
- queryVectorBo.getApiKey(), queryVectorBo.getBaseUrl());
- Embedding queryEmbedding = embeddingModel.embed(queryVectorBo.getQuery()).content();
- float[] vector = queryEmbedding.vector();
- List vectorStrings = new ArrayList<>();
- for (float v : vector) {
- vectorStrings.add(String.valueOf(v));
- }
- String vectorStr = String.join(",", vectorStrings);
- String className = configService.getConfigValue("weaviate", "classname") ;
- // 构建 GraphQL 查询
- String graphQLQuery = String.format(
- "{\n" +
- " Get {\n" +
- " %s(nearVector: {vector: [%s]} limit: %d) {\n" +
- " text\n" +
- " fid\n" +
- " kid\n" +
- " docId\n" +
- " _additional {\n" +
- " distance\n" +
- " id\n" +
- " }\n" +
- " }\n" +
- " }\n" +
- "}",
- className+ queryVectorBo.getKid(),
- vectorStr,
- queryVectorBo.getMaxResults()
- );
-
- Result result = client.graphQL().raw().withQuery(graphQLQuery).run();
- List resultList = new ArrayList<>();
- if (result != null && !result.hasErrors()) {
- Object data = result.getResult().getData();
- JSONObject entries = new JSONObject(data);
- Map entriesMap = entries.get("Get", Map.class);
- cn.hutool.json.JSONArray objects = entriesMap.get(className + queryVectorBo.getKid());
- if(objects.isEmpty()){
- return resultList;
- }
- for (Object object : objects) {
- Map map = (Map) object;
- String content = map.get("text");
- resultList.add( content);
- }
- return resultList;
- } else {
- log.error("GraphQL 查询失败: {}", result.getError());
- return resultList;
- }
+ log.info("查询向量数据: kid={}, query={}, maxResults={}",
+ queryVectorBo.getKid(), queryVectorBo.getQuery(), queryVectorBo.getMaxResults());
+ VectorStoreService strategy = getCurrentStrategy();
+ return strategy.getQueryVector(queryVectorBo);
}
@Override
- @SneakyThrows
public void removeById(String id, String modelName) {
- String protocol = configService.getConfigValue("weaviate", "protocol");
- String host = configService.getConfigValue("weaviate", "host");
- String className = configService.getConfigValue("weaviate", "classname");
- String finalClassName = className + id;
- WeaviateClient client = new WeaviateClient(new Config(protocol, host));
- Result result = client.schema().classDeleter().withClassName(finalClassName).run();
- if (result.hasErrors()) {
- log.error("失败删除向量: " + result.getError());
- throw new ServiceException("失败删除向量数据!");
- } else {
- log.info("成功删除向量数据: " + result.getResult());
- }
+ log.info("根据ID删除向量数据: id={}, modelName={}", id, modelName);
+ VectorStoreService strategy = getCurrentStrategy();
+ strategy.removeById(id, modelName);
}
@Override
public void removeByDocId(String docId, String kid) {
- String className = configService.getConfigValue("weaviate", "classname") + kid;
- // 构建 Where 条件
- WhereFilter whereFilter = WhereFilter.builder()
- .path("docId")
- .operator(Operator.Equal)
- .valueText(docId)
- .build();
- ObjectsBatchDeleter deleter = client.batch().objectsBatchDeleter();
- Result result = deleter.withClassName(className)
- .withWhere(whereFilter)
- .run();
- if (result != null && !result.hasErrors()) {
- log.info("成功删除 docId={} 的所有向量数据", docId);
- } else {
- log.error("删除失败: {}", result.getError());
- }
+ log.info("根据docId删除向量数据: docId={}, kid={}", docId, kid);
+ VectorStoreService strategy = getCurrentStrategy();
+ strategy.removeByDocId(docId, kid);
}
@Override
public void removeByFid(String fid, String kid) {
- String className = configService.getConfigValue("weaviate", "classname") + kid;
- // 构建 Where 条件
- WhereFilter whereFilter = WhereFilter.builder()
- .path("fid")
- .operator(Operator.Equal)
- .valueText(fid)
- .build();
- ObjectsBatchDeleter deleter = client.batch().objectsBatchDeleter();
- Result result = deleter.withClassName(className)
- .withWhere(whereFilter)
- .run();
- if (result != null && !result.hasErrors()) {
- log.info("成功删除 fid={} 的所有向量数据", fid);
- } else {
- log.error("删除失败: {}", result.getError());
- }
+ log.info("根据fid删除向量数据: fid={}, kid={}", fid, kid);
+ VectorStoreService strategy = getCurrentStrategy();
+ strategy.removeByFid(fid, kid);
}
-
- /**
- * 获取向量模型
- */
- @SneakyThrows
- public EmbeddingModel getEmbeddingModel(String modelName, String apiKey, String baseUrl) {
- EmbeddingModel embeddingModel;
- if ("quentinz/bge-large-zh-v1.5".equals(modelName)) {
- embeddingModel = OllamaEmbeddingModel.builder()
- .baseUrl(baseUrl)
- .modelName(modelName)
- .build();
- } else if ("baai/bge-m3".equals(modelName)) {
- embeddingModel = OpenAiEmbeddingModel.builder()
- .apiKey(apiKey)
- .baseUrl(baseUrl)
- .modelName(modelName)
- .build();
- } else {
- throw new ServiceException("未找到对应向量化模型!");
- }
- return embeddingModel;
- }
-
}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/AbstractVectorStoreStrategy.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/AbstractVectorStoreStrategy.java
new file mode 100644
index 00000000..d35d9739
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/AbstractVectorStoreStrategy.java
@@ -0,0 +1,52 @@
+package org.ruoyi.service.strategy;
+
+import org.ruoyi.common.core.exception.ServiceException;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import dev.langchain4j.model.ollama.OllamaEmbeddingModel;
+import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
+import lombok.RequiredArgsConstructor;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.ruoyi.common.core.config.VectorStoreProperties;
+import org.ruoyi.common.core.utils.StringUtils;
+import org.ruoyi.service.VectorStoreService;
+import org.ruoyi.embedding.EmbeddingModelFactory;
+
+/**
+ * 向量库策略抽象基类
+ * 提供公共的方法实现，如embedding模型获取等
+ *
+ * @author Yzm
+ */
+@Slf4j
+@RequiredArgsConstructor
+public abstract class AbstractVectorStoreStrategy implements VectorStoreService {
+
+ protected final VectorStoreProperties vectorStoreProperties;
+
+ private final EmbeddingModelFactory embeddingModelFactory;
+
+ /**
+ * 获取向量模型
+ */
+ @SneakyThrows
+ protected EmbeddingModel getEmbeddingModel(String modelName, Integer dimension) {
+ return embeddingModelFactory.createModel(modelName, dimension);
+ }
+
+ /**
+ * 将float数组转换为Float对象数组
+ */
+ protected static Float[] toObjectArray(float[] primitive) {
+ Float[] result = new Float[primitive.length];
+ for (int i = 0; i < primitive.length; i++) {
+ result[i] = primitive[i]; // 自动装箱
+ }
+ return result;
+ }
+
+ /**
+ * 获取向量库类型标识
+ */
+ public abstract String getVectorStoreType();
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/VectorStoreStrategyFactory.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/VectorStoreStrategyFactory.java
new file mode 100644
index 00000000..0bab68cc
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/VectorStoreStrategyFactory.java
@@ -0,0 +1,57 @@
+package org.ruoyi.service.strategy;
+
+import jakarta.annotation.PostConstruct;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.ruoyi.common.core.config.VectorStoreProperties;
+import org.ruoyi.service.strategy.impl.MilvusVectorStoreStrategy;
+import org.ruoyi.service.strategy.impl.WeaviateVectorStoreStrategy;
+import org.ruoyi.service.VectorStoreService;
+import org.springframework.stereotype.Component;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * 向量库策略工厂
+ * 根据配置动态选择向量库实现
+ *
+ * @author Yzm
+ */
+@Slf4j
+@Component
+@RequiredArgsConstructor
+public class VectorStoreStrategyFactory {
+
+ private final VectorStoreProperties vectorStoreProperties;
+ private final WeaviateVectorStoreStrategy weaviateStrategy;
+ private final MilvusVectorStoreStrategy milvusStrategy;
+
+ private Map strategies;
+
+ @PostConstruct
+ public void init() {
+ strategies = new HashMap<>();
+ strategies.put("weaviate", weaviateStrategy);
+ strategies.put("milvus", milvusStrategy);
+ log.info("向量库策略工厂初始化完成，支持的策略: {}", strategies.keySet());
+ }
+
+ /**
+ * 获取当前配置的向量库策略
+ */
+ public VectorStoreService getStrategy() {
+ String vectorStoreType = vectorStoreProperties.getType();
+ if (vectorStoreType == null || vectorStoreType.trim().isEmpty()) {
+ vectorStoreType = "weaviate"; // 默认使用weaviate
+ }
+ VectorStoreService strategy = strategies.get(vectorStoreType.toLowerCase());
+ if (strategy == null) {
+ log.warn("未找到向量库策略: {}, 使用默认策略: weaviate", vectorStoreType);
+ strategy = strategies.get("weaviate");
+ }
+ log.debug("使用向量库策略: {}", vectorStoreType);
+ return strategy;
+ }
+
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/impl/MilvusVectorStoreStrategy.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/impl/MilvusVectorStoreStrategy.java
new file mode 100644
index 00000000..8d3d50fa
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/impl/MilvusVectorStoreStrategy.java
@@ -0,0 +1,157 @@
+package org.ruoyi.service.strategy.impl;
+
+import dev.langchain4j.data.document.Metadata;
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import dev.langchain4j.store.embedding.EmbeddingMatch;
+import dev.langchain4j.store.embedding.EmbeddingSearchRequest;
+import dev.langchain4j.store.embedding.EmbeddingStore;
+import dev.langchain4j.store.embedding.filter.Filter;
+import dev.langchain4j.store.embedding.filter.MetadataFilterBuilder;
+import dev.langchain4j.store.embedding.milvus.MilvusEmbeddingStore;
+import io.milvus.param.IndexType;
+import io.milvus.param.MetricType;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.ruoyi.common.core.config.VectorStoreProperties;
+import org.ruoyi.domain.bo.QueryVectorBo;
+import org.ruoyi.domain.bo.StoreEmbeddingBo;
+import org.ruoyi.embedding.EmbeddingModelFactory;
+import org.ruoyi.service.strategy.AbstractVectorStoreStrategy;
+import org.springframework.stereotype.Component;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.stream.IntStream;
+
+@Slf4j
+@Component
+public class MilvusVectorStoreStrategy extends AbstractVectorStoreStrategy {
+
+
+ private final Integer DIMENSION = 2048;
+
+ public MilvusVectorStoreStrategy(VectorStoreProperties vectorStoreProperties, EmbeddingModelFactory embeddingModelFactory) {
+ super(vectorStoreProperties, embeddingModelFactory);
+ }
+
+ // 缓存不同集合与 autoFlush 配置的 Milvus 连接
+ private final Map> storeCache = new ConcurrentHashMap<>();
+
+ private EmbeddingStore getMilvusStore(String collectionName, boolean autoFlushOnInsert) {
+ String key = collectionName + "|" + autoFlushOnInsert;
+ return storeCache.computeIfAbsent(key, k ->
+ MilvusEmbeddingStore.builder()
+ .uri(vectorStoreProperties.getMilvus().getUrl())
+ .collectionName(collectionName)
+ .dimension(DIMENSION)
+ .indexType(IndexType.IVF_FLAT)
+ .metricType(MetricType.L2)
+ .autoFlushOnInsert(autoFlushOnInsert)
+ .idFieldName("id")
+ .textFieldName("text")
+ .metadataFieldName("metadata")
+ .vectorFieldName("vector")
+ .build()
+ );
+ }
+
+ @Override
+ public void createSchema(String kid, String modelName) {
+ String collectionName = vectorStoreProperties.getMilvus().getCollectionname() + kid;
+ // 使用缓存获取连接以确保只初始化一次
+ EmbeddingStore store = getMilvusStore(collectionName, true);
+ log.info("Milvus集合初始化完成: {}", collectionName);
+ }
+
+ @Override
+ public void storeEmbeddings(StoreEmbeddingBo storeEmbeddingBo) {
+ EmbeddingModel embeddingModel = getEmbeddingModel(storeEmbeddingBo.getEmbeddingModelName(), DIMENSION);
+
+ List chunkList = storeEmbeddingBo.getChunkList();
+ List fidList = storeEmbeddingBo.getFids();
+ String kid = storeEmbeddingBo.getKid();
+ String docId = storeEmbeddingBo.getDocId();
+ String collectionName = vectorStoreProperties.getMilvus().getCollectionname() + kid;
+
+ log.info("Milvus向量存储条数记录: {}", chunkList.size());
+ long startTime = System.currentTimeMillis();
+
+ // 复用连接，写入场景使用 autoFlush=false 以提升批量插入性能
+ EmbeddingStore embeddingStore = getMilvusStore(collectionName, false);
+
+ IntStream.range(0, chunkList.size()).forEach(i -> {
+ String text = chunkList.get(i);
+ String fid = fidList.get(i);
+ Metadata metadata = new Metadata();
+ metadata.put("fid", fid);
+ metadata.put("kid", kid);
+ metadata.put("docId", docId);
+
+ TextSegment textSegment = TextSegment.from(text, metadata);
+ Embedding embedding = embeddingModel.embed(text).content();
+ embeddingStore.add(embedding, textSegment);
+ });
+ long endTime = System.currentTimeMillis();
+ log.info("Milvus向量存储完成消耗时间：{}秒", (endTime - startTime) / 1000);
+ }
+
+ @Override
+ public List getQueryVector(QueryVectorBo queryVectorBo) {
+ EmbeddingModel embeddingModel = getEmbeddingModel(queryVectorBo.getEmbeddingModelName(), DIMENSION);
+
+ Embedding queryEmbedding = embeddingModel.embed(queryVectorBo.getQuery()).content();
+ String collectionName = vectorStoreProperties.getMilvus().getCollectionname() + queryVectorBo.getKid();
+
+ // 查询复用连接，autoFlush 对查询无影响，此处保持 true
+ EmbeddingStore embeddingStore = getMilvusStore(collectionName, true);
+
+ List resultList = new ArrayList<>();
+ EmbeddingSearchRequest request = EmbeddingSearchRequest.builder()
+ .queryEmbedding(queryEmbedding)
+ .maxResults(queryVectorBo.getMaxResults())
+ .build();
+ List> matches = embeddingStore.search(request).matches();
+ for (EmbeddingMatch match : matches) {
+ TextSegment segment = match.embedded();
+ if (segment != null) {
+ resultList.add(segment.text());
+ }
+ }
+ return resultList;
+ }
+
+ @Override
+ @SneakyThrows
+ public void removeById(String id, String modelName) {
+ // 注意：此处原逻辑使用 collectionname + id，保持现状
+ EmbeddingStore embeddingStore = getMilvusStore(vectorStoreProperties.getMilvus().getCollectionname() + id, false);
+ embeddingStore.remove(id);
+ }
+
+ @Override
+ public void removeByDocId(String docId, String kid) {
+ String collectionName = vectorStoreProperties.getMilvus().getCollectionname() + kid;
+ EmbeddingStore embeddingStore = getMilvusStore(collectionName, false);
+ Filter filter = MetadataFilterBuilder.metadataKey("docId").isEqualTo(docId);
+ embeddingStore.removeAll(filter);
+ log.info("Milvus成功删除 docId={} 的所有向量数据", docId);
+ }
+
+ @Override
+ public void removeByFid(String fid, String kid) {
+ String collectionName = vectorStoreProperties.getMilvus().getCollectionname() + kid;
+ EmbeddingStore embeddingStore = getMilvusStore(collectionName, false);
+ Filter filter = MetadataFilterBuilder.metadataKey("fid").isEqualTo(fid);
+ embeddingStore.removeAll(filter);
+ log.info("Milvus成功删除 fid={} 的所有向量数据", fid);
+ }
+
+ @Override
+ public String getVectorStoreType() {
+ return "milvus";
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/impl/WeaviateVectorStoreStrategy.java b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/impl/WeaviateVectorStoreStrategy.java
new file mode 100644
index 00000000..3d61f8ac
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/strategy/impl/WeaviateVectorStoreStrategy.java
@@ -0,0 +1,232 @@
+package org.ruoyi.service.strategy.impl;
+
+import cn.hutool.json.JSONObject;
+import org.ruoyi.common.core.exception.ServiceException;
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import io.weaviate.client.Config;
+import io.weaviate.client.WeaviateClient;
+import io.weaviate.client.base.Result;
+import io.weaviate.client.v1.batch.api.ObjectsBatchDeleter;
+import io.weaviate.client.v1.batch.model.BatchDeleteResponse;
+import io.weaviate.client.v1.filters.Operator;
+import io.weaviate.client.v1.filters.WhereFilter;
+import io.weaviate.client.v1.graphql.model.GraphQLResponse;
+import io.weaviate.client.v1.schema.model.Property;
+import io.weaviate.client.v1.schema.model.Schema;
+import io.weaviate.client.v1.schema.model.WeaviateClass;
+import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
+import org.ruoyi.common.core.config.VectorStoreProperties;
+import org.ruoyi.domain.bo.QueryVectorBo;
+import org.ruoyi.domain.bo.StoreEmbeddingBo;
+import org.ruoyi.embedding.EmbeddingModelFactory;
+import org.ruoyi.service.strategy.AbstractVectorStoreStrategy;
+import org.springframework.stereotype.Component;
+import java.util.*;
+
+/**
+ * Weaviate向量库策略实现
+ *
+ * @author Yzm
+ */
+@Slf4j
+@Component
+public class WeaviateVectorStoreStrategy extends AbstractVectorStoreStrategy {
+
+ private WeaviateClient client;
+
+ public WeaviateVectorStoreStrategy(VectorStoreProperties vectorStoreProperties, EmbeddingModelFactory embeddingModelFactory) {
+ super(vectorStoreProperties, embeddingModelFactory);
+ }
+
+ @Override
+ public String getVectorStoreType() {
+ return "weaviate";
+ }
+
+ @Override
+ public void createSchema(String kid, String embeddingModelName) {
+ String protocol = vectorStoreProperties.getWeaviate().getProtocol();
+ String host = vectorStoreProperties.getWeaviate().getHost();
+ String className = vectorStoreProperties.getWeaviate().getClassname() + kid;
+ // 创建 Weaviate 客户端
+ client = new WeaviateClient(new Config(protocol, host));
+ // 检查类是否存在，如果不存在就创建 schema
+ Result schemaResult = client.schema().getter().run();
+ Schema schema = schemaResult.getResult();
+ boolean classExists = false;
+ for (WeaviateClass weaviateClass : schema.getClasses()) {
+ if (weaviateClass.getClassName().equals(className)) {
+ classExists = true;
+ break;
+ }
+ }
+ if (!classExists) {
+ // 类不存在，创建 schema
+ WeaviateClass build = WeaviateClass.builder()
+ .className(className)
+ .vectorizer("none")
+ .properties(
+ List.of(Property.builder().name("text").dataType(Collections.singletonList("text")).build(),
+ Property.builder().name("fid").dataType(Collections.singletonList("text")).build(),
+ Property.builder().name("kid").dataType(Collections.singletonList("text")).build(),
+ Property.builder().name("docId").dataType(Collections.singletonList("text")).build())
+ )
+ .build();
+ Result createResult = client.schema().classCreator().withClass(build).run();
+ if (createResult.hasErrors()) {
+ log.error("Schema 创建失败: {}", createResult.getError());
+ } else {
+ log.info("Schema 创建成功: {}", className);
+ }
+ }
+ }
+
+ @Override
+ public void storeEmbeddings(StoreEmbeddingBo storeEmbeddingBo) {
+ createSchema(storeEmbeddingBo.getKid(),storeEmbeddingBo.getEmbeddingModelName());
+ EmbeddingModel embeddingModel = getEmbeddingModel(storeEmbeddingBo.getEmbeddingModelName(), null);
+ List chunkList = storeEmbeddingBo.getChunkList();
+ List fidList = storeEmbeddingBo.getFids();
+ String kid = storeEmbeddingBo.getKid();
+ String docId = storeEmbeddingBo.getDocId();
+ log.info("向量存储条数记录: " + chunkList.size());
+ long startTime = System.currentTimeMillis();
+ for (int i = 0; i < chunkList.size(); i++) {
+ String text = chunkList.get(i);
+ String fid = fidList.get(i);
+ Embedding embedding = embeddingModel.embed(text).content();
+ Map properties = Map.of(
+ "text", text,
+ "fid", fid,
+ "kid", kid,
+ "docId", docId
+ );
+ Float[] vector = toObjectArray(embedding.vector());
+ client.data().creator()
+ .withClassName("LocalKnowledge" + kid)
+ .withProperties(properties)
+ .withVector(vector)
+ .run();
+ }
+ long endTime = System.currentTimeMillis();
+ log.info("向量存储完成消耗时间：" + (endTime - startTime) / 1000 + "秒");
+ }
+
+
+
+ @Override
+ public List getQueryVector(QueryVectorBo queryVectorBo) {
+ createSchema(queryVectorBo.getKid(),queryVectorBo.getEmbeddingModelName());
+ EmbeddingModel embeddingModel = getEmbeddingModel(queryVectorBo.getEmbeddingModelName(),null);
+ Embedding queryEmbedding = embeddingModel.embed(queryVectorBo.getQuery()).content();
+ float[] vector = queryEmbedding.vector();
+ List vectorStrings = new ArrayList<>();
+ for (float v : vector) {
+ vectorStrings.add(String.valueOf(v));
+ }
+ String vectorStr = String.join(",", vectorStrings);
+ String className = vectorStoreProperties.getWeaviate().getClassname();
+
+ // 构建 GraphQL 查询
+ String graphQLQuery = String.format(
+ "{\n" +
+ " Get {\n" +
+ " %s(nearVector: {vector: [%s]} limit: %d) {\n" +
+ " text\n" +
+ " fid\n" +
+ " kid\n" +
+ " docId\n" +
+ " _additional {\n" +
+ " distance\n" +
+ " id\n" +
+ " }\n" +
+ " }\n" +
+ " }\n" +
+ "}",
+ className + queryVectorBo.getKid(),
+ vectorStr,
+ queryVectorBo.getMaxResults()
+ );
+
+ Result result = client.graphQL().raw().withQuery(graphQLQuery).run();
+ List resultList = new ArrayList<>();
+ if (result != null && !result.hasErrors()) {
+ Object data = result.getResult().getData();
+ JSONObject entries = new JSONObject(data);
+ Map entriesMap = entries.get("Get", Map.class);
+ cn.hutool.json.JSONArray objects = entriesMap.get(className + queryVectorBo.getKid());
+ if (objects.isEmpty()) {
+ return resultList;
+ }
+ for (Object object : objects) {
+ Map map = (Map) object;
+ String content = map.get("text");
+ resultList.add(content);
+ }
+ return resultList;
+ } else {
+ log.error("GraphQL 查询失败: {}", result.getError());
+ return resultList;
+ }
+ }
+
+ @Override
+ @SneakyThrows
+ public void removeById(String id, String modelName) {
+ String protocol = vectorStoreProperties.getWeaviate().getProtocol();
+ String host = vectorStoreProperties.getWeaviate().getHost();
+ String className = vectorStoreProperties.getWeaviate().getClassname();
+ String finalClassName = className + id;
+ WeaviateClient client = new WeaviateClient(new Config(protocol, host));
+ Result result = client.schema().classDeleter().withClassName(finalClassName).run();
+ if (result.hasErrors()) {
+ log.error("失败删除向量: " + result.getError());
+ throw new ServiceException("失败删除向量数据!");
+ } else {
+ log.info("成功删除向量数据: " + result.getResult());
+ }
+ }
+
+ @Override
+ public void removeByDocId(String docId, String kid) {
+ String className = vectorStoreProperties.getWeaviate().getClassname() + kid;
+ // 构建 Where 条件
+ WhereFilter whereFilter = WhereFilter.builder()
+ .path("docId")
+ .operator(Operator.Equal)
+ .valueText(docId)
+ .build();
+ ObjectsBatchDeleter deleter = client.batch().objectsBatchDeleter();
+ Result result = deleter.withClassName(className)
+ .withWhere(whereFilter)
+ .run();
+ if (result != null && !result.hasErrors()) {
+ log.info("成功删除 docId={} 的所有向量数据", docId);
+ } else {
+ log.error("删除失败: {}", result.getError());
+ }
+ }
+
+ @Override
+ public void removeByFid(String fid, String kid) {
+ String className = vectorStoreProperties.getWeaviate().getClassname() + kid;
+ // 构建 Where 条件
+ WhereFilter whereFilter = WhereFilter.builder()
+ .path("fid")
+ .operator(Operator.Equal)
+ .valueText(fid)
+ .build();
+ ObjectsBatchDeleter deleter = client.batch().objectsBatchDeleter();
+ Result result = deleter.withClassName(className)
+ .withWhere(whereFilter)
+ .run();
+ if (result != null && !result.hasErrors()) {
+ log.info("成功删除 fid={} 的所有向量数据", fid);
+ } else {
+ log.error("删除失败: {}", result.getError());
+ }
+ }
+
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/vo/ChatConfigVo.java b/ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/vo/ChatConfigVo.java
index 148d2dc8..410e4a2d 100644
--- a/ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/vo/ChatConfigVo.java
+++ b/ruoyi-modules-api/ruoyi-system-api/src/main/java/org/ruoyi/system/domain/vo/ChatConfigVo.java
@@ -18,7 +18,7 @@ import java.io.Serializable;
*/
@Data
@ExcelIgnoreUnannotated
-@AutoMapper(target = ChatConfig.class)
+ @AutoMapper(target = ChatConfig.class)
public class ChatConfigVo implements Serializable {
@Serial
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/pom.xml b/ruoyi-modules-api/ruoyi-workflow-api/pom.xml
new file mode 100644
index 00000000..9049508f
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/pom.xml
@@ -0,0 +1,133 @@
+
+
+ 4.0.0
+
+ org.ruoyi
+ ruoyi-modules-api
+ ${revision}
+ ../pom.xml
+
+
+ ruoyi-workflow-api
+
+
+ 工作流API模块
+
+
+
+ 17
+ 17
+ UTF-8
+
+
+
+
+ org.springframework.boot
+ spring-boot-starter
+
+
+
+ com.fasterxml.jackson.core
+ jackson-databind
+
+
+
+ org.springframework
+ spring-web
+
+
+
+ org.ruoyi
+ ruoyi-system-api
+
+
+
+ org.ruoyi
+ ruoyi-common-satoken
+
+
+
+ org.ruoyi
+ ruoyi-common-mail
+
+
+
+ org.ruoyi
+ ruoyi-chat
+
+
+
+ dev.langchain4j
+ langchain4j-core
+ 1.2.0
+
+
+
+ cn.hutool
+ hutool-all
+ 5.8.12
+ compile
+
+
+
+ org.bsc.langgraph4j
+ langgraph4j-core
+ 1.5.3
+
+
+
+ org.bsc.langgraph4j
+ langgraph4j-langchain4j
+ 1.5.3
+
+
+
+ io.swagger.core.v3
+ swagger-annotations
+ 2.2.8
+
+
+
+ dev.langchain4j
+ langchain4j-open-ai
+ 1.2.0
+ compile
+
+
+
+ dev.langchain4j
+ langchain4j-community-dashscope
+ 1.2.0-beta8
+
+
+
+ com.baomidou
+ mybatis-plus-generator
+ 3.5.3.1
+
+
+
+ dev.langchain4j
+ langchain4j-http-client-jdk
+ 1.2.0
+
+
+
+ dev.langchain4j
+ langchain4j-document-parser-apache-poi
+ 1.2.0-beta8
+
+
+
+
+ com.google.api-client
+ google-api-client
+ 2.6.0
+
+
+
+
+
+
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/CodeGenerator.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/CodeGenerator.java
new file mode 100644
index 00000000..6b77b030
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/CodeGenerator.java
@@ -0,0 +1,43 @@
+package org.ruoyi.workflow;
+
+import com.baomidou.mybatisplus.generator.FastAutoGenerator;
+import com.baomidou.mybatisplus.generator.config.OutputFile;
+import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
+
+import java.sql.Types;
+import java.util.Collections;
+
+public class CodeGenerator {
+ public static void main(String[] args) {
+ FastAutoGenerator.create("jdbc:postgres://172.17.30.40:5432/aideepin?useUnicode=true&characterEncoding=utf8&serverTimezone=GMT%2B8&tinyInt1isBit=false&allowMultiQueries=true", "postgres", "postgres")
+ .globalConfig(builder -> {
+ builder.author("moyz") // 设置作者
+ .enableSwagger() // 开启 swagger 模式
+ .fileOverride() // 覆盖已生成文件
+ .outputDir("D://"); // 指定输出目录
+ })
+ .dataSourceConfig(builder -> builder.typeConvertHandler((globalConfig, typeRegistry, metaInfo) -> {
+ int typeCode = metaInfo.getJdbcType().TYPE_CODE;
+ if (typeCode == Types.SMALLINT) {
+ // 自定义类型转换
+ return DbColumnType.INTEGER;
+ }
+ return typeRegistry.getColumnType(metaInfo);
+
+ }))
+ .packageConfig(builder -> {
+ builder.mapper("com.adi.common.mapper")
+ .parent("")
+ .moduleName("")
+ .entity("po")
+ .serviceImpl("service.impl")
+ .pathInfo(Collections.singletonMap(OutputFile.xml, "D://mybatisplus-generatorcode")); // 设置mapperXml生成路径
+ })
+ .strategyConfig(builder -> {
+ builder.addInclude("adi_knowledge_base_qa_record") // 设置需要生成的表名
+ .addTablePrefix("adi_");
+ builder.mapperBuilder().enableBaseResultMap().enableMapperAnnotation().build();
+ })
+ .execute();
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/BaseResponse.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/BaseResponse.java
new file mode 100644
index 00000000..d6220abd
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/BaseResponse.java
@@ -0,0 +1,51 @@
+package org.ruoyi.workflow.base;
+
+import lombok.Data;
+import org.ruoyi.workflow.enums.ErrorEnum;
+
+import java.io.Serializable;
+
+@Data
+public class BaseResponse implements Serializable {
+
+ private static final long serialVersionUID = 1L;
+ /**
+ * 是否成功
+ */
+ private boolean success;
+ /**
+ * 状态码
+ */
+ private String code;
+ /**
+ * 提示
+ */
+ private String message;
+ /**
+ * 数据
+ */
+ private T data;
+
+ public BaseResponse() {
+ }
+
+ public BaseResponse(boolean success) {
+ this.success = success;
+ }
+
+ public BaseResponse(boolean success, T data) {
+ this.data = data;
+ this.success = success;
+ }
+
+ public BaseResponse(String code, String message, T data) {
+ this.code = code;
+ this.success = false;
+ this.message = message;
+ this.data = data;
+ }
+
+ public static BaseResponse success(String message) {
+ return new BaseResponse(ErrorEnum.SUCCESS.getCode(), message, "");
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/NodeInputConfigTypeHandler.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/NodeInputConfigTypeHandler.java
new file mode 100644
index 00000000..4bd44fbb
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/NodeInputConfigTypeHandler.java
@@ -0,0 +1,118 @@
+package org.ruoyi.workflow.base;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.node.ArrayNode;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.CollectionUtils;
+import org.apache.ibatis.type.BaseTypeHandler;
+import org.apache.ibatis.type.JdbcType;
+import org.apache.ibatis.type.MappedJdbcTypes;
+import org.apache.ibatis.type.MappedTypes;
+import org.ruoyi.workflow.enums.WfIODataTypeEnum;
+import org.ruoyi.workflow.util.JsonUtil;
+import org.ruoyi.workflow.workflow.WfNodeInputConfig;
+import org.ruoyi.workflow.workflow.def.WfNodeIO;
+import org.ruoyi.workflow.workflow.def.WfNodeParamRef;
+
+import java.sql.CallableStatement;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.ruoyi.workflow.workflow.WfNodeIODataUtil.INPUT_TYPE_TO_NODE_IO_DEF;
+
+@Slf4j
+@MappedJdbcTypes({JdbcType.JAVA_OBJECT})
+@MappedTypes({WfNodeInputConfig.class})
+public class NodeInputConfigTypeHandler extends BaseTypeHandler {
+
+ public static WfNodeInputConfig fillNodeInputConfig(String jsonSource) {
+ ObjectNode jsonNode = (ObjectNode) JsonUtil.toJsonNode(jsonSource);
+ return createNodeInputConfig(jsonNode);
+ }
+
+ public static WfNodeInputConfig createNodeInputConfig(ObjectNode jsonNode) {
+ List userInputs = new ArrayList<>();
+ WfNodeInputConfig result = new WfNodeInputConfig();
+ result.setUserInputs(userInputs);
+ result.setRefInputs(new ArrayList<>());
+ if (null == jsonNode) {
+ return result;
+ }
+ ArrayNode userInputsJson = jsonNode.withArray("user_inputs");
+ ArrayNode refInputs = jsonNode.withArray("ref_inputs");
+ if (!userInputsJson.isEmpty()) {
+ for (JsonNode userInput : userInputsJson) {
+ if (userInput instanceof ObjectNode objectNode) {
+ int type = objectNode.get("type").asInt();
+ Class nodeIOClass = INPUT_TYPE_TO_NODE_IO_DEF.get(WfIODataTypeEnum.getByValue(type));
+ WfNodeIO wfNodeIO = JsonUtil.fromJson(objectNode, nodeIOClass);
+ if (null != wfNodeIO) {
+ userInputs.add(wfNodeIO);
+ } else {
+ log.warn("用户输入格式不正确:{}", userInput);
+ }
+ }
+ }
+ }
+ if (!refInputs.isEmpty()) {
+ List list = JsonUtil.fromArrayNode(refInputs, WfNodeParamRef.class);
+ if (CollectionUtils.isNotEmpty(list)) {
+ result.setRefInputs(list);
+ } else {
+ log.warn("引用输入格式不正确:{}", refInputs);
+ }
+ }
+ return result;
+ }
+
+ @Override
+ public void setNonNullParameter(PreparedStatement ps, int i, WfNodeInputConfig parameter, JdbcType jdbcType) {
+// PGobject jsonObject = new PGobject();
+// jsonObject.setType("jsonb");
+// try {
+// jsonObject.setValue(JsonUtil.toJson(parameter));
+// ps.setObject(i, jsonObject);
+// } catch (Exception e) {
+// throw new RuntimeException(e);
+// }
+ }
+
+ @Override
+ public WfNodeInputConfig getNullableResult(ResultSet rs, String columnName) throws SQLException {
+ String jsonSource = rs.getString(columnName);
+ if (jsonSource != null) {
+ try {
+ return fillNodeInputConfig(jsonSource);
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+ return null;
+ }
+
+ @Override
+ public WfNodeInputConfig getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
+ String jsonSource = rs.getString(columnIndex);
+ if (jsonSource != null) {
+ return fillNodeInputConfig(jsonSource);
+ }
+ return null;
+ }
+
+ @Override
+ public WfNodeInputConfig getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
+ String jsonSource = cs.getString(columnIndex);
+ if (jsonSource != null) {
+ try {
+ return fillNodeInputConfig(jsonSource);
+ } catch (Exception e) {
+ throw new RuntimeException(e);
+ }
+ }
+ return null;
+ }
+}
\ No newline at end of file
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/ThreadContext.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/ThreadContext.java
new file mode 100644
index 00000000..cb7e992a
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/base/ThreadContext.java
@@ -0,0 +1,122 @@
+package org.ruoyi.workflow.base;
+
+import cn.dev33.satoken.stp.StpUtil;
+import org.apache.commons.lang3.StringUtils;
+import org.ruoyi.common.core.domain.model.LoginUser;
+import org.ruoyi.common.core.exception.base.BaseException;
+import org.ruoyi.common.satoken.utils.LoginHelper;
+import org.ruoyi.workflow.entity.User;
+import org.ruoyi.workflow.enums.UserStatusEnum;
+
+import static org.ruoyi.workflow.enums.ErrorEnum.A_USER_NOT_FOUND;
+
+/**
+ * 线程上下文适配器，统一接入 Sa-Token 登录态。
+ */
+public class ThreadContext {
+
+ private static final ThreadLocal CURRENT_USER = new ThreadLocal<>();
+ private static final ThreadLocal CURRENT_TOKEN = new ThreadLocal<>();
+
+ private ThreadContext() {
+ }
+
+ /**
+ * 获取当前登录的工作流用户。
+ */
+ public static User getCurrentUser() {
+ User cached = CURRENT_USER.get();
+ if (cached != null) {
+ return cached;
+ }
+ LoginUser loginUser = LoginHelper.getLoginUser();
+ if (loginUser == null) {
+ throw new BaseException(A_USER_NOT_FOUND.getInfo());
+ }
+ User mapped = mapToWorkflowUser(loginUser);
+ CURRENT_USER.set(mapped);
+ return mapped;
+ }
+
+ /**
+ * 允许在测试或特殊场景下显式设置当前用户。
+ */
+ public static void setCurrentUser(User user) {
+ if (user == null) {
+ CURRENT_USER.remove();
+ } else {
+ CURRENT_USER.set(user);
+ }
+ }
+
+ /**
+ * 获取当前登录用户 ID。
+ */
+ public static Long getCurrentUserId() {
+ Long userId = LoginHelper.getUserId();
+ if (userId != null) {
+ return userId;
+ }
+ return getCurrentUser().getId();
+ }
+
+ /**
+ * 获取当前访问 token。
+ */
+ public static String getToken() {
+ String token = CURRENT_TOKEN.get();
+ if (StringUtils.isNotBlank(token)) {
+ return token;
+ }
+ try {
+ token = StpUtil.getTokenValue();
+ } catch (Exception ignore) {
+ token = null;
+ }
+ if (StringUtils.isNotBlank(token)) {
+ CURRENT_TOKEN.set(token);
+ }
+ return token;
+ }
+
+ public static void setToken(String token) {
+ if (StringUtils.isBlank(token)) {
+ CURRENT_TOKEN.remove();
+ } else {
+ CURRENT_TOKEN.set(token);
+ }
+ }
+
+ public static boolean isLogin() {
+ return LoginHelper.isLogin();
+ }
+
+ public static User getExistCurrentUser() {
+ return getCurrentUser();
+ }
+
+ public static void unload() {
+ CURRENT_USER.remove();
+ CURRENT_TOKEN.remove();
+ }
+
+ private static User mapToWorkflowUser(LoginUser loginUser) {
+ User user = new User();
+ user.setId(loginUser.getUserId());
+ String nickname = loginUser.getNickName();
+ user.setName(StringUtils.defaultIfBlank(nickname, loginUser.getUsername()));
+ user.setEmail(loginUser.getUsername());
+ user.setUuid(String.valueOf(loginUser.getUserId()));
+ user.setUserStatus(UserStatusEnum.NORMAL);
+ user.setIsAdmin(LoginHelper.isSuperAdmin(loginUser.getUserId()));
+ user.setUnderstandContextMsgPairNum(0);
+ user.setQuotaByTokenDaily(0);
+ user.setQuotaByTokenMonthly(0);
+ user.setQuotaByRequestDaily(0);
+ user.setQuotaByRequestMonthly(0);
+ user.setQuotaByImageDaily(0);
+ user.setQuotaByImageMonthly(0);
+ user.setIsDeleted(false);
+ return user;
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/config/BeanConfig.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/config/BeanConfig.java
new file mode 100644
index 00000000..5623208b
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/config/BeanConfig.java
@@ -0,0 +1,76 @@
+package org.ruoyi.workflow.config;
+
+import com.fasterxml.jackson.annotation.JsonInclude;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.datatype.jdk8.Jdk8Module;
+import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
+import lombok.extern.slf4j.Slf4j;
+import org.ruoyi.workflow.util.LocalDateTimeUtil;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
+import org.springframework.core.task.AsyncTaskExecutor;
+import org.springframework.http.client.BufferingClientHttpRequestFactory;
+import org.springframework.http.client.SimpleClientHttpRequestFactory;
+import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder;
+import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
+import org.springframework.validation.beanvalidation.LocalValidatorFactoryBean;
+import org.springframework.web.client.RestTemplate;
+
+@Slf4j
+@Configuration
+public class BeanConfig {
+
+ @Bean
+ public RestTemplate restTemplate() {
+ log.info("Configuration:create restTemplate");
+ SimpleClientHttpRequestFactory requestFactory = new SimpleClientHttpRequestFactory();
+ // 设置建立连接超时时间 毫秒
+ requestFactory.setConnectTimeout(60000);
+ // 设置读取数据超时时间 毫秒
+ requestFactory.setReadTimeout(60000);
+ RestTemplate restTemplate = new RestTemplate();
+ // 注册LOG拦截器
+// restTemplate.setInterceptors(Lists.newArrayList(new LogClientHttpRequestInterceptor()));
+ restTemplate.setRequestFactory(new BufferingClientHttpRequestFactory(requestFactory));
+
+ return restTemplate;
+ }
+
+ @Bean
+ @Primary
+ public ObjectMapper objectMapper() {
+ log.info("Configuration:create objectMapper");
+ ObjectMapper objectMapper = new Jackson2ObjectMapperBuilder().createXmlMapper(false).build();
+ objectMapper.registerModules(LocalDateTimeUtil.getSimpleModule(), new JavaTimeModule(), new Jdk8Module());
+ //设置null值不参与序列化(字段不被显示)
+ objectMapper.setSerializationInclusion(JsonInclude.Include.NON_NULL);
+ return objectMapper;
+ }
+
+ @Bean(name = "mainExecutor")
+ @Primary
+ public AsyncTaskExecutor mainExecutor() {
+ int processorsNum = Runtime.getRuntime().availableProcessors();
+ log.info("mainExecutor,processorsNum:{}", processorsNum);
+ ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
+ executor.setCorePoolSize(processorsNum * 2);
+ executor.setMaxPoolSize(100);
+ return executor;
+ }
+
+ @Bean(name = "imagesExecutor")
+ public AsyncTaskExecutor imagesExecutor() {
+ int processorsNum = Runtime.getRuntime().availableProcessors();
+ ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
+ log.info("imagesExecutor corePoolSize:{},maxPoolSize:{}", processorsNum, processorsNum * 2);
+ executor.setCorePoolSize(processorsNum);
+ executor.setMaxPoolSize(processorsNum * 2);
+ return executor;
+ }
+
+ @Bean(name = "beanValidator")
+ public LocalValidatorFactoryBean validator() {
+ return new LocalValidatorFactoryBean();
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/cosntant/AdiConstant.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/cosntant/AdiConstant.java
new file mode 100644
index 00000000..d9b4bac7
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/cosntant/AdiConstant.java
@@ -0,0 +1,407 @@
+package org.ruoyi.workflow.cosntant;
+
+import dev.langchain4j.model.input.PromptTemplate;
+import lombok.extern.slf4j.Slf4j;
+
+import java.lang.reflect.Field;
+import java.util.ArrayList;
+import java.util.List;
+
+@Slf4j
+public class AdiConstant {
+ public static final int DEFAULT_PAGE_SIZE = 10;
+ /**
+ * 验证码id过期时间：1小时
+ */
+ public static final int AUTH_CAPTCHA_ID_EXPIRE = 1;
+ /**
+ * 验证码过期时间，5分钟
+ */
+ public static final int AUTH_CAPTCHA_EXPIRE = 5;
+ /**
+ * 注册激活码有效时长，8小时
+ */
+ public static final int AUTH_ACTIVE_CODE_EXPIRE = 8;
+ /**
+ * token存活时间（8小时）
+ */
+ public static final int USER_TOKEN_EXPIRE = 8;
+ public static final String DEFAULT_PASSWORD = "123456";
+ public static final int LOGIN_MAX_FAIL_TIMES = 3;
+ public static final String[] WEB_RESOURCES = {
+ "/swagger-ui/index.html",
+ "/swagger-ui",
+ "/swagger-resources",
+ "/v3/api-docs",
+ "/favicon.ico",
+ ".css",
+ ".js",
+ "/doc.html"
+ };
+ public static final int SECRET_KEY_TYPE_SYSTEM = 1;
+ public static final int SECRET_KEY_TYPE_CUSTOM = 2;
+ public static final String OPENAI_MESSAGE_DONE_FLAG = "[DONE]";
+ public static final String DEFAULT_MODEL = "gpt-3.5-turbo";
+ public static final String CREATE_IMAGE_RESP_FORMATS_B64JSON = "b64_json";
+ public static final String OPENAI_CREATE_IMAGE_RESP_FORMATS_URL = "url";
+ public static final List DALLE2_CREATE_IMAGE_SIZES = List.of("256x256", "512x512", "1024x1024");
+ public static final List DALLE3_CREATE_IMAGE_SIZES = List.of("1024x1024", "1024x1792", "1792x1024");
+ public static final PromptTemplate PROMPT_EXTRA_TEMPLATE = PromptTemplate.from("""
+ ## 要求
+ 尽可能准确地回答用户的问题
+
+ ## 用户的问题
+ {{question}}
+
+ ## 注意
+ {{extraInfo}}
+ """);
+ public static final PromptTemplate PROMPT_INFO_TEMPLATE = PromptTemplate.from("""
+ ## 要求
+ 根据已知信息，尽可能准确地回答用户的问题
+
+ ## 用户的问题
+ {{question}}
+
+ ## 已知信息
+ {{information}}
+
+ ## 注意
+ 回答的内容不能让用户感知到已知信息的存在
+ """);
+ /**
+ * 可能的 extraInfo 如适用转音频的要求： 2. 回答的内容要尽量口语化，以方便将内容转成语音
+ */
+ public static final PromptTemplate PROMPT_INFO_EXTRA_TEMPLATE = PromptTemplate.from("""
+ ## 要求
+ 根据已知信息，尽可能准确地回答用户的问题
+
+ ## 用户的问题
+ {{question}}
+
+ ## 已知信息
+ {{information}}
+
+ ## 注意
+ 1. 回答的内容不能让用户感知到已知信息的存在
+ {{extraInfo}}
+ """);
+ public static final String PROMPT_EXTRA_AUDIO = "2. 回答的内容要尽量口语化，以方便将内容转成语音";
+ public static final Double LLM_TEMPERATURE_DEFAULT = 0.7D;
+ public static final Double RAG_RETRIEVE_MIN_SCORE_DEFAULT = 0.6D;
+ public static final int tts_ = 1;
+ public static final String[] POI_DOC_TYPES = {"doc", "docx", "ppt", "pptx", "xls", "xlsx"};
+ public static final long SSE_TIMEOUT = (2 * 60 + 30) * 1000L; // 2.5分钟
+ public static final int RAG_TYPE_KB = 1;
+ public static final int RAG_TYPE_SEARCH = 2;
+ /**
+ * 每块文档长度（按token算）
+ */
+ public static final int RAG_MAX_SEGMENT_SIZE_IN_TOKENS = 1000;
+ /**
+ * 文档召回默认数量
+ */
+ public static final int RAG_RETRIEVE_NUMBER_DEFAULT = 3;
+ /**
+ * 文档召回最大数量
+ */
+ public static final int RAG_RETRIEVE_NUMBER_MAX = 5;
+ /**
+ * 向量搜索时命中所需的最低分数
+ */
+ public static final double RAG_MIN_SCORE = 0.6;
+ /**
+ * 默认的最大输入token数
+ */
+ public static final int LLM_MAX_INPUT_TOKENS_DEFAULT = 4096;
+ public static final String LLM_INPUT_TYPE_TEXT = "text";
+ public static final String LLM_INPUT_TYPE_IMAGE = "image";
+ public static final String LLM_INPUT_TYPE_AUDIO = "audio";
+ public static final String LLM_INPUT_TYPE_VIDEO = "video";
+ public static final String[] GRAPH_ENTITY_EXTRACTION_ENTITY_TYPES = {"organization", "person", "geo", "event"};
+ public static final String GRAPH_TUPLE_DELIMITER = "<|>";
+ public static final String GRAPH_RECORD_DELIMITER = "##";
+ public static final String GRAPH_COMPLETION_DELIMITER = "<|COMPLETE|>";
+ public static final List GRAPH_STORE_MAIN_FIELDS = List.of("name", "label", "text_segment_id", "description");
+ /**
+ * 唯一标识字段，如果该字段有指定，则根据该配置判断Vertex或Edge是否唯一，如知识库中根据 name、metadata->>kb_uuid 来做判断
+ */
+ public static final String GRAPH_METADATA_IDENTIFY_COLUMNS = "graph_metadata_identify_columns";
+ /**
+ * 内容追加字段
+ * 更新数据时，如遇到该标识中的字段，追加内容而不是替换
+ */
+ public static final String GRAPH_METADATA_APPEND_COLUMNS = "graph_metadata_append_columns_if_exist";
+ public static final int AI_IMAGE_TYPE_REGULAR = 1;
+ public static final int AI_IMAGE_TYPE_THUMBNAIL = 2;
+ public static final int AI_IMAGE_TYPE_REGULAR_MARK = 3;
+ public static final int AI_IMAGE_TYPE_THUMBNAIL_MARK = 4;
+ public static final String DOC_INDEX_TYPE_EMBEDDING = "embedding";
+ public static final String DOC_INDEX_TYPE_GRAPHICAL = "graphical";
+ public static final String DRAW_TYPE_PUBLIC = "public";
+ public static final String DRAW_TYPE_STARRED = "starred";
+ public static final String DRAW_TYPE_MINE = "mine";
+ public static final String MP_LIMIT_1 = "limit 1";
+ /**
+ * 文件存储在本地
+ */
+ public static final int STORAGE_LOCATION_LOCAL = 1;
+ /**
+ * 文件存储到阿里云OSS
+ */
+ public static final int STORAGE_LOCATION_ALI_OSS = 2;
+ public static final String URL_PREFIX_FILE = "/file/";
+ public static final String URL_PREFIX_IMAGE = "/image/";
+ public static final String URL_PREFIX_MY_IMAGE = "/my-image/";
+ public static final String URL_PREFIX_MY_THUMBNAIL = "/my-thumbnail/";
+ public static final List IMAGE_EXTENSIONS = List.of("jpg", "jpeg", "png", "gif", "bmp", "webp");
+ public static final String W_FAILED = "FAILED";
+ public static final String COLUMN_NAME_IS_DELETE = "is_deleted";
+ public static final String COLUMN_NAME_USER_ID = "user_id";
+ public static final String COLUMN_NAME_ID = "id";
+ public static final String COLUMN_NAME_UUID = "uuid";
+ public static final String FORM_DATA_BOUNDARY_PRE = "----WebKitFormBoundary";
+
+ private AdiConstant() {
+ }
+
+ public static class ConversationConstant {
+ public static final String DEFAULT_NAME = "通用智能助手";
+ public static final int ANSWER_CONTENT_TYPE_AUTO = 1;
+ public static final int ANSWER_CONTENT_TYPE_TEXT = 2;
+ public static final int ANSWER_CONTENT_TYPE_AUDIO = 3;
+ public static final String AUDIO_CONFIG_FIELD_ANSWER_VOICE = "answer_voice";
+ public static final String AUDIO_CONFIG_FIELD_VOICE_PLATFORM = "platform";
+
+ private ConversationConstant() {
+ }
+ }
+
+ public static class GenerateImage {
+ public static final int INTERACTING_METHOD_GENERATE_IMAGE = 1;
+ public static final int INTERACTING_METHOD_EDIT_IMAGE = 2;
+ public static final int INTERACTING_METHOD_VARIATION = 3;
+ public static final int INTERACTING_METHOD_BACKGROUND_GENERATION = 4;
+ public static final int STATUS_DOING = 1;
+ public static final int STATUS_FAIL = 2;
+ public static final int STATUS_SUCCESS = 3;
+
+ private GenerateImage() {
+ }
+ }
+
+ public static class MetadataKey {
+ public static final String KB_UUID = "kb_uuid";
+ public static final String KB_ITEM_UUID = "kb_item_uuid";
+ public static final String ENGINE_NAME = "engine_name";
+ public static final String SEARCH_UUID = "search_uuid";
+
+ private MetadataKey() {
+ }
+ }
+
+ public static class SysConfigKey {
+ public static final String DEEPSEEK_SETTING = "deepseek_setting";
+ public static final String OPENAI_SETTING = "openai_setting";
+ public static final String DASHSCOPE_SETTING = "dashscope_setting";
+ public static final String QIANFAN_SETTING = "qianfan_setting";
+ public static final String OLLAMA_SETTING = "ollama_setting";
+ public static final String SILICONFLOW_SETTING = "siliconflow_setting";
+ public static final String GOOGLE_SETTING = "google_setting";
+ public static final String BING_SETTING = "bing_setting";
+ public static final String BAIDU_SETTING = "baidu_setting";
+ public static final String REQUEST_TEXT_RATE_LIMIT = "request_text_rate_limit";
+ public static final String REQUEST_IMAGE_RATE_LIMIT = "request_image_rate_limit";
+ public static final String CONVERSATION_MAX_NUM = "conversation_max_num";
+ public static final String QUOTA_BY_TOKEN_DAILY = "quota_by_token_daily";
+ public static final String QUOTA_BY_TOKEN_MONTHLY = "quota_by_token_monthly";
+ public static final String QUOTA_BY_REQUEST_DAILY = "quota_by_request_daily";
+ public static final String QUOTA_BY_REQUEST_MONTHLY = "quota_by_request_monthly";
+ public static final String QUOTA_BY_IMAGE_DAILY = "quota_by_image_daily";
+ public static final String QUOTA_BY_IMAGE_MONTHLY = "quota_by_image_monthly";
+ public static final String QUOTA_BY_QA_ASK_DAILY = "quota_by_qa_ask_daily";
+ public static final String STORAGE_LOCATION = "storage_location";
+ public static final String STORAGE_LOCATION_ALI_OSS = "storage_location_ali_oss";
+ public static final String ASR_SETTING = "asr_setting";
+ public static final String TTS_SETTING = "tts_setting";
+
+ private SysConfigKey() {
+ }
+ }
+
+ public static class ModelPlatform {
+ public static final String DEEPSEEK = "deepseek";
+ public static final String OPENAI = "openai";
+ public static final String DASHSCOPE = "dashscope";
+ public static final String QIANFAN = "qianfan";
+ public static final String OLLAMA = "ollama";
+ public static final String SILICONFLOW = "siliconflow";
+
+ private ModelPlatform() {
+ }
+
+ // 获取所有公共静态常量（String类型的值）的列表
+ public static List getModelConstants() {
+ List list = new ArrayList<>();
+ Class clazz = ModelPlatform.class;
+ for (Field field : clazz.getDeclaredFields()) {
+ try {
+ String value = (String) field.get(null);
+ list.add(value);
+ } catch (ReflectiveOperationException e) {
+ log.error("error", e);
+ }
+
+ }
+ return list;
+ }
+ }
+
+ public static class ModelType {
+ public static final String TEXT = "text";
+ public static final String IMAGE = "image";
+ public static final String EMBEDDING = "embedding";
+ public static final String RERANK = "rerank";
+ public static final String ASR = "asr";
+ public static final String TTS = "tts";
+
+ private ModelType() {
+ }
+
+ public static List getModelType() {
+ List list = new ArrayList<>();
+ Class clazz = ModelType.class;
+ for (Field field : clazz.getDeclaredFields()) {
+ try {
+ String value = (String) field.get(null);
+ list.add(value);
+ } catch (ReflectiveOperationException e) {
+ log.error("error", e);
+ }
+
+ }
+ return list;
+ }
+ }
+
+ public static class SearchEngineName {
+ public static final String GOOGLE = "google";
+ public static final String BING = "bing";
+ public static final String BAIDU = "baidu";
+ public static final String[] GOOGLE_COUNTRIES = {"cn", "af", "al", "dz", "as", "ad", "ao", "ai", "aq", "ag", "ar", "am", "aw", "au", "at", "az", "bs", "bh", "bd", "bb", "by", "be", "bz", "bj", "bm", "bt", "bo", "ba", "bw", "bv", "br", "io", "bn", "bg", "bf", "bi", "kh", "cm", "ca", "cv", "ky", "cf", "td", "cl", "cx", "cc", "co", "km", "cg", "cd", "ck", "cr", "ci", "hr", "cu", "cy", "cz", "dk", "dj", "dm", "do", "ec", "eg", "sv", "gq", "er", "ee", "et", "fk", "fo", "fj", "fi", "fr", "gf", "pf", "tf", "ga", "gm", "ge", "de", "gh", "gi", "gr", "gl", "gd", "gp", "gu", "gt", "gn", "gw", "gy", "ht", "hm", "va", "hn", "hk", "hu", "is", "in", "id", "ir", "iq", "ie", "il", "it", "jm", "jp", "jo", "kz", "ke", "ki", "kp", "kr", "kw", "kg", "la", "lv", "lb", "ls", "lr", "ly", "li", "lt", "lu", "mo", "mk", "mg", "mw", "my", "mv", "ml", "mt", "mh", "mq", "mr", "mu", "yt", "mx", "fm", "md", "mc", "mn", "ms", "ma", "mz", "mm", "na", "nr", "np", "nl", "an", "nc", "nz", "ni", "ne", "ng", "nu", "nf", "mp", "no", "om", "pk", "pw", "ps", "pa", "pg", "py", "pe", "ph", "pn", "pl", "pt", "pr", "qa", "re", "ro", "ru", "rw", "sh", "kn", "lc", "pm", "vc", "ws", "sm", "st", "sa", "sn", "rs", "sc", "sl", "sg", "sk", "si", "sb", "so", "za", "gs", "es", "lk", "sd", "sr", "sj", "sz", "se", "ch", "sy", "tw", "tj", "tz", "th", "tl", "tg", "tk", "to", "tt", "tn", "tr", "tm", "tc", "tv", "ug", "ua", "ae", "uk", "gb", "us", "um", "uy", "uz", "vu", "ve", "vn", "vg", "vi", "wf", "eh", "ye", "zm", "zw"};
+ public static final String[] GOOGLE_LANGUAGES = {"zh-cn", "zh-tw", "af", "ak", "sq", "ws", "am", "ar", "hy", "az", "eu", "be", "bem", "bn", "bh", "xx-bork", "bs", "br", "bg", "bt", "km", "ca", "chr", "ny", "co", "hr", "cs", "da", "nl", "xx-elmer", "en", "eo", "et", "ee", "fo", "tl", "fi", "fr", "fy", "gaa", "gl", "ka", "de", "el", "kl", "gn", "gu", "xx-hacker", "ht", "ha", "haw", "iw", "hi", "hu", "is", "ig", "id", "ia", "ga", "it", "ja", "jw", "kn", "kk", "rw", "rn", "xx-klingon", "kg", "ko", "kri", "ku", "ckb", "ky", "lo", "la", "lv", "ln", "lt", "loz", "lg", "ach", "mk", "mg", "ms", "ml", "mt", "mv", "mi", "mr", "mfe", "mo", "mn", "sr-me", "my", "ne", "pcm", "nso", "no", "nn", "oc", "or", "om", "ps", "fa", "xx-pirate", "pl", "pt", "pt-br", "pt-pt", "pa", "qu", "ro", "rm", "nyn", "ru", "gd", "sr", "sh", "st", "tn", "crs", "sn", "sd", "si", "sk", "sl", "so", "es", "es-419", "su", "sw", "sv", "tg", "ta", "tt", "te", "th", "ti", "to", "lua", "tum", "tr", "tk", "tw", "ug", "uk", "ur", "uz", "vu", "vi", "cy", "wo", "xh", "yi", "yo", "zu"};
+
+ private SearchEngineName() {
+ }
+ }
+
+ public static class SSEEventName {
+ public static final String START = "[START]";
+ public static final String DONE = "[DONE]";
+ public static final String ERROR = "[ERROR]";
+ public static final String META = "[META]";
+ public static final String AUDIO = "[AUDIO]";
+ public static final String THINKING = "[THINKING]";
+ public static final String AI_SEARCH_SOURCE_LINKS = "[SOURCE_LINKS]";
+ public static final String WF_NODE_CHUNK = "[WF_NODE_CHUNK]";
+ public static final String WF_NODE_OUTPUT = "[WF_NODE_OUTPUT]";
+ public static final String STATE_CHANGED = "[STATE_CHANGED]";
+
+ private SSEEventName() {
+ }
+ }
+
+ public static class SSEEventData {
+
+ /**
+ * 状态：问题分析中
+ * 如敏感词校验等
+ */
+ public static final String STATE_QUESTION_ANALYSING = """
+ {"state":"question_analysing","remark":"问题分析中"}
+ """;
+
+ public static final String STATE_KNOWLEDGE_SEARCHING = """
+ {"state":"knowledge_searching","remark":"知识库搜索中"}
+ """;
+ //使用 THINKING 事件代替
+ public static final String STATE_THINKING = """
+ {"state":"thinking","remark":"推理中"}
+ """;
+ public static final String STATE_RESPONDING = """
+ {"state":"responding","remark":"回答中"}
+ """;
+ }
+
+ public static class WorkflowConstant {
+ public static final String DEFAULT_INPUT_PARAM_NAME = "input";
+ public static final String DEFAULT_OUTPUT_PARAM_NAME = "output";
+ public static final String DEFAULT_ERROR_OUTPUT_PARAM_NAME = "error_msg";
+ public static final String HUMAN_FEEDBACK_KEY = "human_feedback";
+ public static final int NODE_PROCESS_STATUS_READY = 1;
+ public static final int NODE_PROCESS_STATUS_DOING = 2;
+ public static final int NODE_PROCESS_STATUS_SUCCESS = 3;
+ public static final int NODE_PROCESS_STATUS_FAIL = 4;
+
+ public static final int WORKFLOW_PROCESS_STATUS_READY = 1;
+ public static final int WORKFLOW_PROCESS_STATUS_DOING = 2;
+ public static final int WORKFLOW_PROCESS_STATUS_SUCCESS = 3;
+ public static final int WORKFLOW_PROCESS_STATUS_FAIL = 4;
+ public static final int WORKFLOW_PROCESS_STATUS_WAITING_INPUT = 5;
+
+ public static final int WORKFLOW_NODE_PROCESS_TYPE_NORMAL = 1;
+ public static final int WORKFLOW_NODE_PROCESS_TYPE_CONDITIONAL = 2;
+ public static final int WORKFLOW_NODE_PROCESS_TYPE_PARALLEL = 3;
+
+ public static final int MAIL_SENDER_TYPE_SYS = 1;
+ public static final int MAIL_SENDER_TYPE_CUSTOM = 2;
+ }
+
+ public static class TokenEstimator {
+ public static String OPENAI = "openai";
+ public static String HUGGING_FACE = "huggingface";
+ public static String QWEN = "qwen";
+
+ public static List ALL = List.of(OPENAI, HUGGING_FACE, QWEN);
+ }
+
+ public static class EmbeddingModel {
+ public static String ALL_MINILM_L6 = "local:all-minilm-l6-v2";
+ }
+
+ public static class McpConstant {
+ public static final String TRANSPORT_TYPE_SSE = "sse";
+ public static final String TRANSPORT_TYPE_STDIO = "stdio";
+ public static final String INSTALL_TYPE_REMOTE = "remote";
+ public static final String INSTALL_TYPE_WASM = "wasm";
+ public static final String INSTALL_TYPE_LOCAL = "local";
+ public static final String INSTALL_TYPE_DOCKER = "docker";
+ }
+
+ public static class TtsConstant {
+
+ /**
+ * 语音合成器位置-客户端
+ */
+ public static final String SYNTHESIZER_CLIENT = "client";
+ /**
+ * 语音合成器位置-服务端
+ */
+ public static final String SYNTHESIZER_SERVER = "server";
+
+ /**
+ * 通义默认语音音色-龙应严(义正严辞女声)
+ */
+ public static final String DASHSCOPE_DEFAULT_VOICE = "longyingyan";
+ }
+
+ public static class CustomChatRequestParameterKeys {
+ /**
+ * 是否开启思考模式，默认不开启
+ */
+ public static final String ENABLE_THINKING = "enable_thinking";
+
+ private CustomChatRequestParameterKeys() {
+ }
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/cosntant/RedisKeyConstant.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/cosntant/RedisKeyConstant.java
new file mode 100644
index 00000000..77e86997
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/cosntant/RedisKeyConstant.java
@@ -0,0 +1,116 @@
+package org.ruoyi.workflow.cosntant;
+
+public class RedisKeyConstant {
+
+ /**
+ * 账号激活码的key
+ */
+ public static final String AUTH_ACTIVE_CODE = "auth:activeCode:{0}";
+ /**
+ * 注册时使用的验证码
+ * 参数：验证码id
+ * 值：验证码
+ */
+ public static final String AUTH_REGISTER_CAPTCHA_ID = "auth:register:captcha:{0}";
+ /**
+ * 登录时使用的验证码id缓存
+ * 参数：验证码id
+ * 值：验证码
+ */
+ public static final String AUTH_LOGIN_CAPTCHA_ID = "auth:login:captcha:{0}";
+ /**
+ * 注册验证码缓存
+ * 参数：验证码
+ * 值：1
+ */
+ public static final String AUTH_CAPTCHA = "auth:register:captcha:{0}";
+ /**
+ * 登录token
+ * {0}:用户token
+ * 值：json.format(user)
+ */
+ public static final String USER_TOKEN = "user:token:{0}";
+ /**
+ * 参数：游客的uuid
+ * 值：json.format(guest)
+ */
+ public static final String GUEST_UUID = "guest:uuid:{0}";
+ /**
+ * 登录失败次数
+ * 参数：用户邮箱
+ * 值: 失效次数
+ */
+ public static final String LOGIN_FAIL_COUNT = "user:login:fail:{0}";
+ /**
+ * 用户是否请求ai中
+ * 参数：用户id
+ * 值: 1或者0
+ */
+ public static final String USER_ASKING = "user:asking:{0}";
+ /**
+ * 用户是否画画中
+ * 参数：用户id
+ * 值: 1或者0
+ */
+ public static final String USER_DRAWING = "user:drawing:{0}";
+ /**
+ * 用户提问限流计数
+ * 参数：用户id
+ * 值: 当前时间窗口访问量
+ */
+ public static final String USER_REQUEST_TEXT_TIMES = "user:request-text:times:{0}";
+ public static final String USER_REQUEST_IMAGE_TIMES = "user:request-image:times:{0}";
+ /**
+ * 用户信息缓存
+ * 参数：用户id
+ * 值: user object
+ */
+ public static final String USER_INFO = "user:info:";
+ /**
+ * 找回密码的请求绑在
+ * 参数：随机数
+ * 值: 用户id，用于校验后续流程中的重置密码使用
+ */
+ public static final String FIND_MY_PASSWORD = "user:find:password:{0}";
+ /**
+ * qa提问次数（每天）
+ * 参数：用户id:日期yyyyMMdd
+ * 值：提问数量
+ */
+ public static final String AQ_ASK_TIMES = "qa:ask:limit:{0}:{1}";
+ /**
+ * 知识库知识点生成数量
+ * 值: 用户id
+ */
+ public static final String QA_ITEM_CREATE_LIMIT = "aq:item:create:{0}";
+ /**
+ * 信号(重新生成知识库统计数据)
+ * 值:知识库uuid
+ */
+ public static final String KB_STATISTIC_RECALCULATE_SIGNAL = "kb:statistic:recalculate:signal";
+ public static final String STATISTIC = "statistic";
+ public static final String STATISTIC_USER = "user";
+ public static final String STATISTIC_KNOWLEDGE_BASE = "kb";
+ public static final String STATISTIC_TOKEN_COST = "token-cost";
+ public static final String STATISTIC_CONVERSATION = "conversation";
+ public static final String STATISTIC_IMAGE_COST = "image-cost";
+ public static final String TOKEN_USAGE_KEY = "token:usage:{0}";
+ /**
+ * 用户正在对文档进行索引
+ * 值：用户id
+ */
+ public static final String USER_INDEXING = "user:indexing:{0}";
+ /**
+ * 用户评论并发限制
+ * 值：用户id
+ */
+ public static final String DRAW_COMMENT_LIMIT_KEY = "user:draw:comment-submitting:{0}";
+ public static final String WORKFLOW_KEY = "workflow";
+ public static final String WORKFLOW_COMPONENTS = "workflow:components";
+ public static final String WORKFLOW_COMPONENT_START_KEY = "workflow:component:start";
+ public static final String WORKFLOW_COMPONENT_KEY = "workflow:component";
+ public static final String WORKFLOW_COPY_DOING = "workflow:copy:doing:{0}";
+
+ private RedisKeyConstant() {
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfAddReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfAddReq.java
new file mode 100644
index 00000000..f201919d
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfAddReq.java
@@ -0,0 +1,17 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import jakarta.validation.constraints.NotBlank;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+@Data
+@Validated
+public class WfAddReq {
+
+ @NotBlank
+ private String title;
+
+ private String remark;
+
+ private Boolean isPublic;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfBaseInfoUpdateReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfBaseInfoUpdateReq.java
new file mode 100644
index 00000000..7b12335b
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfBaseInfoUpdateReq.java
@@ -0,0 +1,15 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import jakarta.validation.constraints.NotBlank;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+@Validated
+@Data
+public class WfBaseInfoUpdateReq {
+ @NotBlank
+ private String uuid;
+ private String title;
+ private String remark;
+ private Boolean isPublic;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfComponentReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfComponentReq.java
new file mode 100644
index 00000000..84da556f
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfComponentReq.java
@@ -0,0 +1,18 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import jakarta.validation.constraints.NotBlank;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+@Data
+@Validated
+public class WfComponentReq {
+ private String uuid;
+ @NotBlank(message = "标题不能为空")
+ private String name;
+ @NotBlank(message = "标题不能为空")
+ private String title;
+ private String remark;
+ private Boolean isEnable;
+ private Integer displayOrder;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfComponentSearchReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfComponentSearchReq.java
new file mode 100644
index 00000000..5ebc9dff
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfComponentSearchReq.java
@@ -0,0 +1,9 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import lombok.Data;
+
+@Data
+public class WfComponentSearchReq {
+ private String title;
+ private Boolean isEnable;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfEdgeReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfEdgeReq.java
new file mode 100644
index 00000000..40eb9a19
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfEdgeReq.java
@@ -0,0 +1,25 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import jakarta.validation.constraints.Min;
+import jakarta.validation.constraints.NotBlank;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+@Validated
+@Data
+public class WfEdgeReq {
+ private Long id;
+ @NotBlank
+ private String uuid;
+ @Min(1)
+ private Long workflowId;
+ @NotBlank
+ private String sourceNodeUuid;
+ private String sourceHandle;
+ @NotBlank
+ private String targetNodeUuid;
+ /**
+ * 是否新增
+ */
+ private Boolean isNew;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfNodeDto.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfNodeDto.java
new file mode 100644
index 00000000..b5c5baa5
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfNodeDto.java
@@ -0,0 +1,32 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import jakarta.validation.constraints.Min;
+import jakarta.validation.constraints.NotBlank;
+import jakarta.validation.constraints.NotNull;
+import jakarta.validation.constraints.Size;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+@Validated
+@Data
+public class WfNodeDto {
+ private Long id;
+ @NotBlank
+ @Size(min = 32, max = 32)
+ private String uuid;
+ private Long workflowId;
+ @Min(1)
+ private Long workflowComponentId;
+ @NotBlank
+ private String title;
+ private String remark;
+ @NotNull
+ private ObjectNode inputConfig;
+ @NotNull
+ private ObjectNode nodeConfig;
+ @NotNull
+ private Double positionX;
+ @NotNull
+ private Double positionY;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfRuntimeNodeDto.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfRuntimeNodeDto.java
new file mode 100644
index 00000000..4ca93525
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfRuntimeNodeDto.java
@@ -0,0 +1,17 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+@Validated
+@Data
+public class WfRuntimeNodeDto {
+ private Long id;
+ private String uuid;
+ private Long workflowRuntimeId;
+ private Long nodeId;
+ private ObjectNode input;
+ private ObjectNode output;
+ private Integer status;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfRuntimeResp.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfRuntimeResp.java
new file mode 100644
index 00000000..b5f5fd8a
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfRuntimeResp.java
@@ -0,0 +1,22 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import com.fasterxml.jackson.annotation.JsonFormat;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import lombok.Data;
+
+import java.time.LocalDateTime;
+
+@Data
+public class WfRuntimeResp {
+ private Long id;
+ private String uuid;
+ private Long workflowId;
+ private ObjectNode input;
+ private ObjectNode output;
+ private Integer status;
+ private String statusRemark;
+
+ private String workflowUuid;
+ @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss")
+ private LocalDateTime createTime;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfSearchReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfSearchReq.java
new file mode 100644
index 00000000..939c3970
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WfSearchReq.java
@@ -0,0 +1,10 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import lombok.Data;
+
+@Data
+public class WfSearchReq {
+ private String title;
+ private Boolean isEnable;
+ private Boolean isPublic;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowResp.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowResp.java
new file mode 100644
index 00000000..1e14b116
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowResp.java
@@ -0,0 +1,22 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import lombok.Data;
+
+import java.time.LocalDateTime;
+import java.util.List;
+
+@Data
+public class WorkflowResp {
+ private Long id;
+ private String uuid;
+ private String title;
+ private String remark;
+ private Boolean isPublic;
+ private Long userId;
+ private String userUuid;
+ private String userName;
+ private List nodes;
+ private List edges;
+ private LocalDateTime createTime;
+ private LocalDateTime updateTime;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowResumeReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowResumeReq.java
new file mode 100644
index 00000000..27680932
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowResumeReq.java
@@ -0,0 +1,8 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import lombok.Data;
+
+@Data
+public class WorkflowResumeReq {
+ private String feedbackContent;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowRunReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowRunReq.java
new file mode 100644
index 00000000..54a32a03
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowRunReq.java
@@ -0,0 +1,14 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import lombok.Data;
+
+import java.util.List;
+
+@Data
+public class WorkflowRunReq {
+ private List inputs;
+ private String uuid;
+
+
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowUpdateReq.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowUpdateReq.java
new file mode 100644
index 00000000..3e26c490
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/dto/workflow/WorkflowUpdateReq.java
@@ -0,0 +1,24 @@
+package org.ruoyi.workflow.dto.workflow;
+
+import jakarta.validation.constraints.NotBlank;
+import jakarta.validation.constraints.NotNull;
+import jakarta.validation.constraints.Size;
+import lombok.Data;
+import org.springframework.validation.annotation.Validated;
+
+import java.util.List;
+
+@Validated
+@Data
+public class WorkflowUpdateReq {
+ @NotBlank
+ private String uuid;
+ @Size(min = 1)
+ private List nodes;
+ @NotNull
+ private List edges;
+
+ private List deleteNodes;
+
+ private List deleteEdges;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/BaseEntity.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/BaseEntity.java
new file mode 100644
index 00000000..b657eb16
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/BaseEntity.java
@@ -0,0 +1,29 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableId;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+
+import java.io.Serializable;
+import java.time.LocalDateTime;
+
+@Data
+public class BaseEntity implements Serializable {
+
+ private static final long serialVersionUID = 1L;
+
+ @TableId(type = IdType.AUTO)
+ private Long id;
+
+ @TableField(value = "create_time")
+ private LocalDateTime createTime;
+
+ @TableField(value = "update_time")
+ private LocalDateTime updateTime;
+
+ @Schema(title = "是否删除（0：未删除，1：已删除）")
+ @TableField(value = "is_deleted")
+ private Boolean isDeleted;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/User.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/User.java
new file mode 100644
index 00000000..20ac4dc0
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/User.java
@@ -0,0 +1,66 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+import org.ruoyi.workflow.enums.UserStatusEnum;
+
+import java.time.LocalDateTime;
+
+@EqualsAndHashCode(callSuper = true)
+@Data
+@TableName("adi_user")
+@Schema(title = "User对象")
+public class User extends BaseEntity {
+
+ @Schema(name = "用户名称")
+ @TableField("name")
+ private String name;
+
+ @TableField("email")
+ private String email;
+
+ @TableField("password")
+ private String password;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @Schema(name = "上下文理解中需要携带的消息对数量（提示词及回复）")
+ @TableField("understand_context_msg_pair_num")
+ private Integer understandContextMsgPairNum;
+
+ @Schema(name = "token quota in one day")
+ @TableField("quota_by_token_daily")
+ private Integer quotaByTokenDaily;
+
+ @Schema(name = "token quota in one month")
+ @TableField("quota_by_token_monthly")
+ private Integer quotaByTokenMonthly;
+
+ @Schema(name = "request quota in one day")
+ @TableField("quota_by_request_daily")
+ private Integer quotaByRequestDaily;
+
+ @Schema(name = "request quota in one month")
+ @TableField("quota_by_request_monthly")
+ private Integer quotaByRequestMonthly;
+
+ @TableField("quota_by_image_daily")
+ private Integer quotaByImageDaily;
+
+ @TableField("quota_by_image_monthly")
+ private Integer quotaByImageMonthly;
+
+ @TableField("user_status")
+ private UserStatusEnum userStatus;
+
+ @TableField("active_time")
+ private LocalDateTime activeTime;
+
+ @Schema(title = "是否管理员（0：否，1：是）")
+ @TableField(value = "is_admin")
+ private Boolean isAdmin;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/Workflow.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/Workflow.java
new file mode 100644
index 00000000..4334b069
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/Workflow.java
@@ -0,0 +1,37 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.io.Serial;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName("t_workflow")
+@Schema(title = "工作流定义 | workflow definition")
+public class Workflow extends BaseEntity {
+
+ @Serial
+ private static final long serialVersionUID = 1L;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @TableField("title")
+ private String title;
+
+ @TableField("remark")
+ private String remark;
+
+ @TableField("user_id")
+ private Long userId;
+
+ @TableField("is_public")
+ private Boolean isPublic;
+
+ @TableField("is_enable")
+ private Boolean isEnable;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowComponent.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowComponent.java
new file mode 100644
index 00000000..d75e92c8
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowComponent.java
@@ -0,0 +1,37 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.io.Serial;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName(value = "t_workflow_component", autoResultMap = true)
+@Schema(title = "工作流组件")
+public class WorkflowComponent extends BaseEntity {
+
+ @Serial
+ private static final long serialVersionUID = 1L;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @TableField("name")
+ private String name;
+
+ @TableField("title")
+ private String title;
+
+ @TableField("remark")
+ private String remark;
+
+ @TableField("display_order")
+ private Integer displayOrder;
+
+ @TableField("is_enable")
+ private Boolean isEnable;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowEdge.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowEdge.java
new file mode 100644
index 00000000..0a9bffda
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowEdge.java
@@ -0,0 +1,34 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.io.Serial;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName("t_workflow_edge")
+@Schema(title = "工作流定义-边 | workflow definition edge")
+public class WorkflowEdge extends BaseEntity {
+
+ @Serial
+ private static final long serialVersionUID = 1L;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @TableField("workflow_id")
+ private Long workflowId;
+
+ @TableField("source_node_uuid")
+ private String sourceNodeUuid;
+
+ @TableField("source_handle")
+ private String sourceHandle;
+
+ @TableField("target_node_uuid")
+ private String targetNodeUuid;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowNode.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowNode.java
new file mode 100644
index 00000000..cc15504a
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowNode.java
@@ -0,0 +1,46 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.io.Serial;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName(value = "t_workflow_node", autoResultMap = true)
+@Schema(title = "工作流定义-节点 | workflow definition node")
+public class WorkflowNode extends BaseEntity {
+
+ @Serial
+ private static final long serialVersionUID = 1L;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @TableField("workflow_id")
+ private Long workflowId;
+
+ @TableField("workflow_component_id")
+ private Long workflowComponentId;
+
+ @TableField("title")
+ private String title;
+
+ @TableField("remark")
+ private String remark;
+
+ @TableField(value = "input_config")
+ private String inputConfig;
+
+ @TableField(value = "node_config")
+ private String nodeConfig;
+
+ @TableField("position_x")
+ private Double positionX;
+
+ @TableField("position_y")
+ private Double positionY;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowRuntime.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowRuntime.java
new file mode 100644
index 00000000..557de3f4
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowRuntime.java
@@ -0,0 +1,40 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.io.Serial;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName(value = "t_workflow_runtime", autoResultMap = true)
+@Schema(title = "工作流运行时 | Workflow runtime")
+public class WorkflowRuntime extends BaseEntity {
+
+ @Serial
+ private static final long serialVersionUID = 1L;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @TableField("user_id")
+ private Long userId;
+
+ @TableField("workflow_id")
+ private Long workflowId;
+
+ @TableField(value = "input")
+ private String input;
+
+ @TableField(value = "output")
+ private String output;
+
+ @TableField("status")
+ private Integer status;
+
+ @TableField("status_remark")
+ private String statusRemark;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowRuntimeNode.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowRuntimeNode.java
new file mode 100644
index 00000000..271d9eb7
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/entity/WorkflowRuntimeNode.java
@@ -0,0 +1,42 @@
+package org.ruoyi.workflow.entity;
+
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.baomidou.mybatisplus.annotation.TableName;
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.Data;
+import lombok.EqualsAndHashCode;
+
+import java.io.Serial;
+
+@Data
+@EqualsAndHashCode(callSuper = true)
+@TableName(value = "t_workflow_runtime_node", autoResultMap = true)
+@Schema(title = "工作流实例-节点 | Workflow runtime - node")
+public class WorkflowRuntimeNode extends BaseEntity {
+ @Serial
+ private static final long serialVersionUID = 1L;
+
+ @TableField("uuid")
+ private String uuid;
+
+ @TableField("user_id")
+ private Long userId;
+
+ @TableField("workflow_runtime_id")
+ private Long workflowRuntimeId;
+
+ @TableField("node_id")
+ private Long nodeId;
+
+ @TableField(value = "input")
+ private String input;
+
+ @TableField(value = "output")
+ private String output;
+
+ @TableField("status")
+ private Integer status;
+
+ @TableField("status_remark")
+ private String statusRemark;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/AiModelStatus.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/AiModelStatus.java
new file mode 100644
index 00000000..0dec4a94
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/AiModelStatus.java
@@ -0,0 +1,14 @@
+package org.ruoyi.workflow.enums;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+@Getter
+@AllArgsConstructor
+public enum AiModelStatus implements BaseEnum {
+ ACTIVE(1, "启用"),
+ INACTIVE(2, "停用");
+
+ private final Integer value;
+ private final String desc;
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/BaseEnum.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/BaseEnum.java
new file mode 100644
index 00000000..66a2fb63
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/BaseEnum.java
@@ -0,0 +1,12 @@
+package org.ruoyi.workflow.enums;
+
+import com.baomidou.mybatisplus.annotation.IEnum;
+
+public interface BaseEnum extends IEnum {
+ /**
+ * 获取对应名称
+ *
+ * @return String
+ */
+ String getDesc();
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/ErrorEnum.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/ErrorEnum.java
new file mode 100644
index 00000000..f9de801e
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/ErrorEnum.java
@@ -0,0 +1,127 @@
+package org.ruoyi.workflow.enums;
+
+import lombok.Getter;
+
+@Getter
+public enum ErrorEnum {
+ SUCCESS("00000", "成功"),
+ A_URL_NOT_FOUND("A0001", "地址不存在"),
+ A_PARAMS_ERROR("A0002", "参数校验不通过"),
+ A_REQUEST_TOO_MUCH("A0003", "访问次数太多"),
+ A_LOGIN_ERROR("A0004", "登陆失败，账号或密码错误"),
+ A_LOGIN_ERROR_MAX("A0005", "失败次数太多，请输入验证码重试"),
+ A_LOGIN_CAPTCHA_ERROR("A0006", "验证码不正确"),
+ A_USER_NOT_EXIST("A0007", "用户不存在"),
+ A_CONVERSATION_NOT_EXIST("A0008", "对话不存在"),
+ A_IMAGE_NUMBER_ERROR("A0009", "图片数量不对"),
+ A_IMAGE_SIZE_ERROR("A0010", "图片尺寸不对"),
+ A_FILE_NOT_EXIST("A0011", "文件不存在"),
+ A_DRAWING("A0012", "作图还未完成"),
+ A_USER_EXIST("A0013", "账号已经存在，请使用账号密码登录"),
+ A_FIND_PASSWORD_CODE_ERROR("A0014", "重置码已过期或不存在"),
+ A_USER_WAIT_CONFIRM("A0015", "用户未激活"),
+ A_USER_NOT_AUTH("A0016", "用户无权限"),
+ A_DATA_NOT_FOUND("A0017", "数据不存在"),
+ A_UPLOAD_FAIL("A0018", "上传失败"),
+ A_QA_ASK_LIMIT("A0019", "请求次数太多"),
+ A_QA_ITEM_LIMIT("A0020", "知识点生成已超额度"),
+ A_CONVERSATION_EXIST("A0021", "会话(角色)已存在"),
+ A_MODEL_NOT_FOUND("A0022", "模型不存在"),
+ A_MODEL_ALREADY_EXIST("A0023", "模型已存在"),
+ A_CONVERSATION_NOT_FOUND("A0024", "会话(角色)找不到"),
+ A_AI_IMAGE_NOT_FOUND("A0024", "图片找不到"),
+ A_ENABLE_MODEL_NOT_FOUND("A0025", "没有可用的模型"),
+ A_DOC_INDEX_DOING("A0026", "文档索引正在进行中，请稍后重试"),
+ A_PRESET_CONVERSATION_NOT_EXIST("A0027", "预设会话或角色不存在"),
+ A_CONVERSATION_TITLE_EXIST("A0028", "会话(角色)标题已存在"),
+ A_AI_IMAGE_NO_AUTH("A0029", "无权限查看该图片"),
+ A_USER_NOT_FOUND("A0030", "用户不存在"),
+ A_ACTIVE_CODE_INVALID("A0031", "激活码已失效"),
+ A_OLD_PASSWORD_INVALID("A0032", "原密码不正确"),
+ A_OPT_TOO_FREQUENTLY("A0032", "操作太频繁"),
+ A_DRAW_NOT_FOUND("A00033", "绘图记录找不到"),
+ A_WF_NOT_FOUND("A00034", "工作流找不到"),
+ A_WF_DISABLED("A0035", "工作流已停用"),
+ A_WF_NODE_NOT_FOUND("A0036", "工作流节点找不到"),
+ A_WF_NODE_CONFIG_NOT_FOUND("A0037", "工作流节点配置找不到"),
+ A_WF_NODE_CONFIG_ERROR("A0038", "工作流节点配置异常"),
+ A_WF_INPUT_INVALID("A0039", "工作流输入参数错误"),
+ A_WF_INPUT_MISSING("A0040", "工作流输入缺少参数"),
+ A_WF_MULTIPLE_START_NODE("A0041", "多个开始节点"),
+ A_WF_START_NODE_NOT_FOUND("A0042", "没有开始节点"),
+ A_WF_END_NODE_NOT_FOUND("A0043", "没有结束节点"),
+ A_WF_EDGE_NOT_FOUND("A0044", "工作流的边找不到"),
+ A_WF_RUNTIME_NOT_FOUND("A00045", "工作流运行时数据找不到"),
+ A_SEARCH_QUERY_IS_EMPTY("A00046", "搜索内容不能为空"),
+ A_WF_COMPONENT_NOT_FOUND("A00047", "工作流基础组件找不到"),
+ A_WF_RESUME_FAIL("A00048", "工作流恢复执行时失败"),
+ A_MAIL_SENDER_EMPTY("A00049", "邮件发送人不能为空"),
+ A_MAIL_SENDER_CONFIG_ERROR("A00050", "邮件发送人配置错误"),
+ A_MAIL_RECEIVER_EMPTY("A00051", "邮件接收人不能为空"),
+ A_MCP_SERVER_NOT_FOUND("A00052", "MCP服务找不到"),
+ A_USER_MCP_SERVER_NOT_FOUND("A00053", "用户的MCP服务找不到"),
+ A_PARAMS_INVALID_BY_("A00054", "参数校验异常:{0}"),
+ A_AI_MESSAGE_NOT_FOUND("A00055", "找不到AI的消息"),
+ A_USER_QUESTION_NOT_FOUND("A00056", "用户问题不存在"),
+ A_PLATFORM_NOT_MATCH("A0057", "平台不匹配"),
+ B_UNCAUGHT_ERROR("B0001", "未捕捉异常"),
+ B_COMMON_ERROR("B0002", "业务出错"),
+ B_GLOBAL_ERROR("B0003", "全局异常"),
+ B_SAVE_IMAGE_ERROR("B0004", "保存图片异常"),
+ B_FIND_IMAGE_404("B0005", "无法找到图片"),
+ B_DAILY_QUOTA_USED("B0006", "今天额度已经用完"),
+ B_MONTHLY_QUOTA_USED("B0007", "当月额度已经用完"),
+ B_LLM_NOT_SUPPORT("B0008", "LLM不支持该功能"),
+ B_LLM_SECRET_KEY_NOT_SET("B0009", "LLM的secret key没设置"),
+ B_MESSAGE_NOT_FOUND("B0008", "消息不存在"),
+ B_LLM_SERVICE_DISABLED("B0009", "LLM服务不可用"),
+ B_KNOWLEDGE_BASE_IS_EMPTY("B0010", "知识库内容为空"),
+ B_NO_ANSWER("B0011", "[无答案]"),
+ B_SAVE_FILE_ERROR("B0012", "保存文件异常"),
+ B_BREAK_SEARCH("B0013", "中断搜索"),
+ B_GRAPH_FILTER_NOT_FOUND("B0014", "图过滤器未定义"),
+ B_DB_ERROR("B0015", "数据库查询异常"),
+ B_ACTIVE_USER_ERROR("B0016", "激活用户失败"),
+ B_RESET_PASSWORD_ERROR("B0017", "重置密码失败"),
+ B_IMAGE_LOAD_ERROR("B0018", "加载图片失败"),
+ B_IO_EXCEPTION("B0019", "IO异常"),
+ B_SERVER_EXCEPTION("B0020", "服务端异常"),
+ B_DELETE_FILE_ERROR("B0021", "删除文件异常"),
+ B_WF_RUN_ERROR("B0022", "工作流运行异常"),
+ B_WF_NODE_DEFINITION_NOT_FOUND("B0023", "工作流节点定义找不到"),
+ B_DIR_CREATE_FAIL("B0024", "创建目录失败"),
+ B_LLM_TEMPERATURE_ERROR("B0025", "采样温度应该在 0.1-1之间"),
+ B_ASR_SETTING_NOT_FOUND("B0026", "语音识别设置未找到"),
+ B_URL_INVALID("B0027", "不是有效的网络地址"),
+ B_ASR_MODEL_NOT_FOUND("B0028", "语音识别模型未找到"),
+ B_TTS_SETTING_NOT_FOUND("B0029", "语音合成设置未找到"),
+ B_TTS_MODEL_NOT_FOUND("B0030", "语音合成模型未找到"),
+ B_VOICE_NOT_FOUND("B0031", "声音不存在"),
+ C_DRAW_FAIL("C0001", "大模型生成图片失败,原因:{0}"),
+ C_ALI_OSS_CONFIG_ERROR("C0002", "阿里云OSS初始化失败,原因:{0}"),
+ C_LLM_RESPONSE_INVALID("C0003", "大模型生成结果内容无效"),
+ C_WF_COMPONENT_DELETED_FAIL_BY_USED("C0004", "工作流组件已经被使用，无法被删除，可先停用");
+
+ private final String code;
+ private final String info;
+
+ ErrorEnum(String code, String info) {
+ this.code = code;
+ this.info = info;
+ }
+
+ public static ErrorEnum getErrorEnum(String code) {
+ ErrorEnum result = null;
+ for (ErrorEnum c : ErrorEnum.values()) {
+ if (c.getCode().equals(code)) {
+ result = c;
+ break;
+ }
+ }
+ if (null == result) {
+ result = B_COMMON_ERROR;
+ }
+ return result;
+ }
+
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/UserStatusEnum.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/UserStatusEnum.java
new file mode 100644
index 00000000..2be9a950
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/UserStatusEnum.java
@@ -0,0 +1,23 @@
+package org.ruoyi.workflow.enums;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+
+@Getter
+@AllArgsConstructor
+public enum UserStatusEnum implements BaseEnum {
+
+ WAIT_CONFIRM(1, "待验证"),
+ NORMAL(2, "正常"),
+ FREEZE(3, "冻结");
+
+ private final Integer value;
+ private final String desc;
+
+ public static UserStatusEnum getByValue(Integer val) {
+ return Arrays.stream(UserStatusEnum.values()).filter(item -> item.value.equals(val)).findFirst().orElse(null);
+ }
+
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/WfIODataTypeEnum.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/WfIODataTypeEnum.java
new file mode 100644
index 00000000..8f8989c6
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/enums/WfIODataTypeEnum.java
@@ -0,0 +1,25 @@
+package org.ruoyi.workflow.enums;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+import java.util.Arrays;
+
+@Getter
+@AllArgsConstructor
+public enum WfIODataTypeEnum implements BaseEnum {
+ TEXT(1, "文本"),
+ NUMBER(2, "数字"),
+ OPTIONS(3, "下拉选项"),
+ FILES(4, "文件列表"),
+ BOOL(5, "布尔值"),
+ REF_INPUT(6, "引用节点的输入参数"),
+ REF_OUTPUT(7, "引用节点的输出参数");
+
+ private final Integer value;
+ private final String desc;
+
+ public static WfIODataTypeEnum getByValue(Integer val) {
+ return Arrays.stream(WfIODataTypeEnum.values()).filter(item -> item.value.equals(val)).findFirst().orElse(null);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/helper/SSEEmitterHelper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/helper/SSEEmitterHelper.java
new file mode 100644
index 00000000..4d432c07
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/helper/SSEEmitterHelper.java
@@ -0,0 +1,138 @@
+package org.ruoyi.workflow.helper;
+
+import com.google.common.cache.Cache;
+import com.google.common.cache.CacheBuilder;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.ruoyi.workflow.cosntant.AdiConstant;
+import org.ruoyi.workflow.cosntant.RedisKeyConstant;
+import org.ruoyi.workflow.entity.User;
+import org.springframework.data.redis.core.StringRedisTemplate;
+import org.springframework.stereotype.Service;
+import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
+
+import java.io.IOException;
+import java.text.MessageFormat;
+import java.util.Objects;
+import java.util.concurrent.TimeUnit;
+
+@Slf4j
+@Service
+public class SSEEmitterHelper {
+
+ private static final Cache COMPLETED_SSE = CacheBuilder.newBuilder()
+ .expireAfterWrite(10, TimeUnit.MINUTES).build();
+
+ @Resource
+ private StringRedisTemplate stringRedisTemplate;
+
+ public static void parseAndSendPartialMsg(SseEmitter sseEmitter, String name, String content) {
+ if (Boolean.TRUE.equals(COMPLETED_SSE.getIfPresent(sseEmitter))) {
+ log.warn("sseEmitter already completed,name:{}", name);
+ return;
+ }
+ String[] lines = content.split("[\\r\\n]", -1);
+ if (lines.length > 1) {
+ sendPartial(sseEmitter, name, " " + lines[0]);
+ for (int i = 1; i < lines.length; i++) {
+ sendPartial(sseEmitter, name, "-_wrap_-");
+ sendPartial(sseEmitter, name, " " + lines[i]);
+ }
+ } else {
+ sendPartial(sseEmitter, name, " " + content);
+ }
+ }
+
+ public static void sendPartial(SseEmitter sseEmitter, String name, String msg) {
+ if (Boolean.TRUE.equals(COMPLETED_SSE.getIfPresent(sseEmitter))) {
+ log.warn("sseEmitter already completed,name:{}", name);
+ return;
+ }
+ try {
+ if (StringUtils.isNotBlank(name)) {
+ sseEmitter.send(SseEmitter.event().name(name).data(msg));
+ } else {
+ sseEmitter.send(msg);
+ }
+ } catch (IOException ioException) {
+ log.error("stream onNext error", ioException);
+ }
+ }
+
+
+ public boolean checkOrComplete(User user, SseEmitter sseEmitter) {
+ //Check: If still waiting response
+ String askingKey = MessageFormat.format(RedisKeyConstant.USER_ASKING, user.getId());
+ String askingVal = stringRedisTemplate.opsForValue().get(askingKey);
+ if (StringUtils.isNotBlank(askingVal)) {
+ sendErrorAndComplete(user.getId(), sseEmitter, "正在回复中...");
+ return false;
+ }
+ return true;
+ }
+
+
+ public void startSse(User user, SseEmitter sseEmitter, String data) {
+
+ String askingKey = MessageFormat.format(RedisKeyConstant.USER_ASKING, user.getId());
+ stringRedisTemplate.opsForValue().set(askingKey, "1", 15, TimeUnit.SECONDS);
+
+ try {
+ SseEmitter.SseEventBuilder builder = SseEmitter.event().name(AdiConstant.SSEEventName.START);
+ if (StringUtils.isNotBlank(data)) {
+ builder.data(data);
+ }
+ sseEmitter.send(builder);
+ } catch (IOException e) {
+ log.error("startSse error", e);
+ sseEmitter.completeWithError(e);
+ COMPLETED_SSE.put(sseEmitter, Boolean.TRUE);
+ stringRedisTemplate.delete(askingKey);
+ }
+ }
+
+ public void sendComplete(long userId, SseEmitter sseEmitter, String msg) {
+ if (Boolean.TRUE.equals(COMPLETED_SSE.getIfPresent(sseEmitter))) {
+ log.warn("sseEmitter already completed,userId:{}", userId);
+ delSseRequesting(userId);
+ return;
+ }
+ try {
+ sseEmitter.send(SseEmitter.event().name(AdiConstant.SSEEventName.DONE).data(msg));
+ } catch (IOException e) {
+ throw new RuntimeException(e);
+ } finally {
+ COMPLETED_SSE.put(sseEmitter, Boolean.TRUE);
+ delSseRequesting(userId);
+ sseEmitter.complete();
+ }
+ }
+
+
+ public void sendErrorAndComplete(long userId, SseEmitter sseEmitter, String errorMsg) {
+ if (Boolean.TRUE.equals(COMPLETED_SSE.getIfPresent(sseEmitter))) {
+ log.warn("sseEmitter already completed,ignore error:{}", errorMsg);
+ delSseRequesting(userId);
+ return;
+ }
+ try {
+ SseEmitter.SseEventBuilder event = SseEmitter.event();
+ event.name(AdiConstant.SSEEventName.ERROR);
+ event.data(Objects.toString(errorMsg, ""));
+ sseEmitter.send(event);
+ } catch (IOException e) {
+ log.warn("sendErrorAndComplete userId:{},errorMsg:{}", userId, errorMsg);
+ throw new RuntimeException(e);
+ } finally {
+ COMPLETED_SSE.put(sseEmitter, Boolean.TRUE);
+ delSseRequesting(userId);
+ sseEmitter.complete();
+ }
+ }
+
+ private void delSseRequesting(long userId) {
+ String askingKey = MessageFormat.format(RedisKeyConstant.USER_ASKING, userId);
+ stringRedisTemplate.delete(askingKey);
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowComponentMapper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowComponentMapper.java
new file mode 100644
index 00000000..0935eb58
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowComponentMapper.java
@@ -0,0 +1,11 @@
+package org.ruoyi.workflow.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+import org.ruoyi.workflow.entity.WorkflowComponent;
+
+@Mapper
+public interface WorkflowComponentMapper extends BaseMapper {
+ Integer countRefNodes(@Param("uuid") String uuid);
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowEdgeMapper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowEdgeMapper.java
new file mode 100644
index 00000000..28d21dec
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowEdgeMapper.java
@@ -0,0 +1,9 @@
+package org.ruoyi.workflow.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.ruoyi.workflow.entity.WorkflowEdge;
+
+@Mapper
+public interface WorkflowEdgeMapper extends BaseMapper {
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowMapper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowMapper.java
new file mode 100644
index 00000000..b8bb1988
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowMapper.java
@@ -0,0 +1,9 @@
+package org.ruoyi.workflow.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.ruoyi.workflow.entity.Workflow;
+
+@Mapper
+public interface WorkflowMapper extends BaseMapper {
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowNodeMapper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowNodeMapper.java
new file mode 100644
index 00000000..380eec9d
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowNodeMapper.java
@@ -0,0 +1,10 @@
+package org.ruoyi.workflow.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.ruoyi.workflow.entity.WorkflowNode;
+
+@Mapper
+public interface WorkflowNodeMapper extends BaseMapper {
+ WorkflowNode getStartNode(long workflowId);
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowRunMapper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowRunMapper.java
new file mode 100644
index 00000000..3e73dd6a
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowRunMapper.java
@@ -0,0 +1,13 @@
+package org.ruoyi.workflow.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import org.apache.ibatis.annotations.Mapper;
+import org.apache.ibatis.annotations.Param;
+import org.ruoyi.workflow.entity.WorkflowRuntime;
+
+@Mapper
+public interface WorkflowRunMapper extends BaseMapper {
+
+ Page pageByWfUuid(Page page, @Param("wfUuid") String wfUuid, @Param("userId") Long userId);
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowRuntimeNodeMapper.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowRuntimeNodeMapper.java
new file mode 100644
index 00000000..91345f1c
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/mapper/WorkflowRuntimeNodeMapper.java
@@ -0,0 +1,9 @@
+package org.ruoyi.workflow.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import org.apache.ibatis.annotations.Mapper;
+import org.ruoyi.workflow.entity.WorkflowRuntimeNode;
+
+@Mapper
+public interface WorkflowRuntimeNodeMapper extends BaseMapper {
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowComponentService.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowComponentService.java
new file mode 100644
index 00000000..04e3b03c
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowComponentService.java
@@ -0,0 +1,126 @@
+package org.ruoyi.workflow.service;
+
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.baomidou.mybatisplus.extension.toolkit.ChainWrappers;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.ruoyi.common.core.exception.base.BaseException;
+import org.ruoyi.workflow.dto.workflow.WfComponentReq;
+import org.ruoyi.workflow.dto.workflow.WfComponentSearchReq;
+import org.ruoyi.workflow.entity.WorkflowComponent;
+import org.ruoyi.workflow.enums.ErrorEnum;
+import org.ruoyi.workflow.mapper.WorkflowComponentMapper;
+import org.ruoyi.workflow.util.PrivilegeUtil;
+import org.ruoyi.workflow.util.UuidUtil;
+import org.ruoyi.workflow.workflow.WfComponentNameEnum;
+import org.springframework.beans.BeanUtils;
+import org.springframework.cache.annotation.CacheEvict;
+import org.springframework.cache.annotation.Cacheable;
+import org.springframework.context.annotation.Lazy;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+
+import static org.ruoyi.workflow.cosntant.RedisKeyConstant.WORKFLOW_COMPONENTS;
+import static org.ruoyi.workflow.cosntant.RedisKeyConstant.WORKFLOW_COMPONENT_START_KEY;
+import static org.ruoyi.workflow.enums.ErrorEnum.C_WF_COMPONENT_DELETED_FAIL_BY_USED;
+
+@Slf4j
+@Service
+public class WorkflowComponentService extends ServiceImpl {
+
+ @Lazy
+ @Resource
+ private WorkflowComponentService self;
+
+ @CacheEvict(cacheNames = {WORKFLOW_COMPONENTS, WORKFLOW_COMPONENT_START_KEY})
+ public WorkflowComponent addOrUpdate(WfComponentReq req) {
+ WorkflowComponent wfComponent;
+ if (StringUtils.isNotBlank(req.getUuid())) {
+ wfComponent = PrivilegeUtil.checkAndGetByUuid(req.getUuid(), this.query(), ErrorEnum.A_WF_COMPONENT_NOT_FOUND);
+
+ WorkflowComponent update = new WorkflowComponent();
+ BeanUtils.copyProperties(req, update, "id", "uuid");
+ update.setId(wfComponent.getId());
+ update.setName(req.getName());
+ update.setTitle(req.getTitle());
+ update.setRemark(req.getRemark());
+ update.setIsEnable(req.getIsEnable());
+ update.setDisplayOrder(req.getDisplayOrder());
+ this.baseMapper.updateById(update);
+
+ return update;
+ } else {
+ wfComponent = new WorkflowComponent();
+ BeanUtils.copyProperties(req, wfComponent, "id", "uuid");
+ wfComponent.setUuid(UuidUtil.createShort());
+ this.baseMapper.insert(wfComponent);
+
+ return wfComponent;
+ }
+ }
+
+ @CacheEvict(cacheNames = {WORKFLOW_COMPONENTS, WORKFLOW_COMPONENT_START_KEY})
+ public void enable(String uuid, Boolean isEnable) {
+ WorkflowComponent wfComponent = PrivilegeUtil.checkAndGetByUuid(uuid, this.query(), ErrorEnum.A_WF_COMPONENT_NOT_FOUND);
+ WorkflowComponent update = new WorkflowComponent();
+ update.setIsEnable(isEnable);
+ update.setId(wfComponent.getId());
+ this.baseMapper.updateById(update);
+ }
+
+ @CacheEvict(cacheNames = {WORKFLOW_COMPONENTS, WORKFLOW_COMPONENT_START_KEY})
+ public void deleteByUuid(String uuid) {
+ WorkflowComponent component = PrivilegeUtil.checkAndGetByUuid(uuid, this.query(), ErrorEnum.A_WF_COMPONENT_NOT_FOUND);
+ Integer refNodeCount = baseMapper.countRefNodes(uuid);
+ if (refNodeCount != null && refNodeCount > 0) {
+ throw new BaseException(C_WF_COMPONENT_DELETED_FAIL_BY_USED.getInfo());
+ }
+ boolean updated = ChainWrappers.lambdaUpdateChain(baseMapper)
+ .eq(WorkflowComponent::getId, component.getId())
+ .set(WorkflowComponent::getIsDeleted, true)
+ .set(WorkflowComponent::getIsEnable, false)
+ .update();
+ if (!updated) {
+ throw new BaseException(ErrorEnum.A_WF_COMPONENT_NOT_FOUND.getInfo());
+ }
+ }
+
+ public Page search(WfComponentSearchReq searchReq, Integer currentPage, Integer pageSize) {
+ LambdaQueryWrapper wrapper = new LambdaQueryWrapper<>();
+ wrapper.eq(WorkflowComponent::getIsDeleted, false);
+ wrapper.eq(null != searchReq.getIsEnable(), WorkflowComponent::getIsEnable, searchReq.getIsEnable());
+ wrapper.like(StringUtils.isNotBlank(searchReq.getTitle()), WorkflowComponent::getTitle, searchReq.getTitle());
+ wrapper.orderByAsc(List.of(WorkflowComponent::getDisplayOrder, WorkflowComponent::getId));
+ return baseMapper.selectPage(new Page<>(currentPage, pageSize), wrapper);
+ }
+
+ @Cacheable(cacheNames = WORKFLOW_COMPONENTS)
+ public List getAllEnable() {
+ return ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowComponent::getIsEnable, true)
+ .eq(WorkflowComponent::getIsDeleted, false)
+ .orderByAsc(List.of(WorkflowComponent::getDisplayOrder, WorkflowComponent::getId))
+ .list();
+ }
+
+ @Cacheable(cacheNames = WORKFLOW_COMPONENT_START_KEY)
+ public WorkflowComponent getStartComponent() {
+ List components = self.getAllEnable();
+ return components.stream()
+ .filter(component -> WfComponentNameEnum.START.getName().equals(component.getName()))
+ .findFirst()
+ .orElseThrow(() -> new BaseException(ErrorEnum.B_WF_NODE_DEFINITION_NOT_FOUND.getInfo()));
+ }
+
+ public WorkflowComponent getComponent(Long id) {
+ List components = self.getAllEnable();
+ return components.stream()
+ .filter(component -> component.getId().equals(id))
+ .findFirst()
+ .orElseThrow(() -> new BaseException(ErrorEnum.B_WF_NODE_DEFINITION_NOT_FOUND.getInfo()));
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowEdgeService.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowEdgeService.java
new file mode 100644
index 00000000..ba545e2d
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowEdgeService.java
@@ -0,0 +1,118 @@
+package org.ruoyi.workflow.service;
+
+import cn.hutool.core.collection.CollUtil;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.baomidou.mybatisplus.extension.toolkit.ChainWrappers;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.CollectionUtils;
+import org.ruoyi.common.core.exception.base.BaseException;
+import org.ruoyi.workflow.dto.workflow.WfEdgeReq;
+import org.ruoyi.workflow.entity.WorkflowEdge;
+import org.ruoyi.workflow.enums.ErrorEnum;
+import org.ruoyi.workflow.mapper.WorkflowEdgeMapper;
+import org.ruoyi.workflow.util.MPPageUtil;
+import org.ruoyi.workflow.util.UuidUtil;
+import org.springframework.beans.BeanUtils;
+import org.springframework.context.annotation.Lazy;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Slf4j
+@Service
+public class WorkflowEdgeService extends ServiceImpl {
+
+ @Lazy
+ @Resource
+ private WorkflowEdgeService self;
+
+ public List listDtoByWfId(long workflowId) {
+ List edges = ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowEdge::getWorkflowId, workflowId)
+ .eq(WorkflowEdge::getIsDeleted, false)
+ .list();
+ return MPPageUtil.convertToList(edges, WfEdgeReq.class);
+ }
+
+ @Transactional
+ public void createOrUpdateEdges(Long workflowId, List edges) {
+ List uuidList = new ArrayList<>();
+ for (WfEdgeReq edge : edges) {
+ WorkflowEdge newOne = new WorkflowEdge();
+ BeanUtils.copyProperties(edge, newOne);
+ newOne.setWorkflowId(workflowId);
+
+ WorkflowEdge old = self.getByUuid(edge.getUuid());
+ if (null != old) {
+ log.info("更新边,id:{},uuid:{},source:{},sourceHandle:{},target:{}",
+ edge.getId(), edge.getUuid(), edge.getSourceNodeUuid(), edge.getSourceHandle(), edge.getTargetNodeUuid());
+ newOne.setId(old.getId());
+ } else {
+ newOne.setId(null);
+ log.info("新增边,uuid:{},source:{},sourceHandle:{},target:{}",
+ edge.getUuid(), edge.getSourceNodeUuid(), edge.getSourceHandle(), edge.getTargetNodeUuid());
+ }
+ uuidList.add(edge.getUuid());
+ self.saveOrUpdate(newOne);
+ }
+ ChainWrappers.lambdaUpdateChain(baseMapper)
+ .eq(WorkflowEdge::getWorkflowId, workflowId)
+ .notIn(CollUtil.isNotEmpty(uuidList), WorkflowEdge::getUuid, uuidList)
+ .set(WorkflowEdge::getIsDeleted, true)
+ .update();
+ }
+
+ public List listByWorkflowId(Long workflowId) {
+ return ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowEdge::getWorkflowId, workflowId)
+ .eq(WorkflowEdge::getIsDeleted, false)
+ .list();
+ }
+
+ public List copyByWorkflowId(long workflowId, long targetWorkflow) {
+ List result = new ArrayList<>();
+ self.listByWorkflowId(workflowId).forEach(edge -> {
+ result.add(self.copyEdge(targetWorkflow, edge));
+ });
+ return result;
+ }
+
+ public WorkflowEdge copyEdge(long targetWorkflow, WorkflowEdge sourceEdge) {
+ WorkflowEdge newEdge = new WorkflowEdge();
+ BeanUtils.copyProperties(sourceEdge, newEdge, "id", "uuid", "createTime", "updateTime");
+ newEdge.setUuid(UuidUtil.createShort());
+ newEdge.setWorkflowId(targetWorkflow);
+ baseMapper.insert(newEdge);
+ return getById(newEdge.getId());
+ }
+
+ @Transactional
+ public void deleteEdges(Long workflowId, List uuids) {
+ if (CollectionUtils.isEmpty(uuids)) {
+ return;
+ }
+ for (String uuid : uuids) {
+ WorkflowEdge old = self.getByUuid(uuid);
+ if (null != old && !old.getWorkflowId().equals(workflowId)) {
+ log.error("该边不属于指定的工作流,删除失败,workflowId:{},node workflowId:{}", workflowId, workflowId);
+ throw new BaseException(ErrorEnum.A_PARAMS_ERROR.getInfo());
+ }
+ ChainWrappers.lambdaUpdateChain(baseMapper)
+ .eq(WorkflowEdge::getWorkflowId, workflowId)
+ .eq(WorkflowEdge::getUuid, uuid)
+ .set(WorkflowEdge::getIsDeleted, true)
+ .update();
+ }
+ }
+
+ public WorkflowEdge getByUuid(String uuid) {
+ return ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowEdge::getUuid, uuid)
+ .eq(WorkflowEdge::getIsDeleted, false)
+ .last("limit 1")
+ .one();
+ }
+}
diff --git a/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowNodeService.java b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowNodeService.java
new file mode 100644
index 00000000..cfd1d19a
--- /dev/null
+++ b/ruoyi-modules-api/ruoyi-workflow-api/src/main/java/org/ruoyi/workflow/service/WorkflowNodeService.java
@@ -0,0 +1,245 @@
+package org.ruoyi.workflow.service;
+
+import cn.hutool.core.collection.CollUtil;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.baomidou.mybatisplus.extension.toolkit.ChainWrappers;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.collections4.CollectionUtils;
+import org.ruoyi.common.core.exception.base.BaseException;
+import org.ruoyi.workflow.dto.workflow.WfNodeDto;
+import org.ruoyi.workflow.entity.Workflow;
+import org.ruoyi.workflow.entity.WorkflowComponent;
+import org.ruoyi.workflow.entity.WorkflowNode;
+import org.ruoyi.workflow.enums.ErrorEnum;
+import org.ruoyi.workflow.enums.WfIODataTypeEnum;
+import org.ruoyi.workflow.mapper.WorkflowNodeMapper;
+import org.ruoyi.workflow.util.JsonUtil;
+import org.ruoyi.workflow.util.MPPageUtil;
+import org.ruoyi.workflow.util.UuidUtil;
+import org.ruoyi.workflow.workflow.WfComponentNameEnum;
+import org.ruoyi.workflow.workflow.WfNodeInputConfig;
+import org.ruoyi.workflow.workflow.def.WfNodeIOText;
+import org.springframework.beans.BeanUtils;
+import org.springframework.context.annotation.Lazy;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.ArrayList;
+import java.util.List;
+
+@Slf4j
+@Service
+public class WorkflowNodeService extends ServiceImpl {
+
+ @Lazy
+ @Resource
+ private WorkflowNodeService self;
+
+ @Resource
+ private WorkflowComponentService workflowComponentService;
+
+ public WorkflowNode getStartNode(long workflowId) {
+ return baseMapper.getStartNode(workflowId);
+ }
+
+ public List listDtoByWfId(long workflowId) {
+ List workflowNodeList = ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowNode::getWorkflowId, workflowId)
+ .eq(WorkflowNode::getIsDeleted, false)
+ .list();
+ workflowNodeList.forEach(this::checkAndDecrypt);
+ return MPPageUtil.convertToList(workflowNodeList, WfNodeDto.class, (source, target) -> {
+ target.setInputConfig(JsonUtil.toBean(source.getInputConfig(), ObjectNode.class));
+ target.setNodeConfig(JsonUtil.toBean(source.getNodeConfig(), ObjectNode.class));
+ return target;
+ });
+ }
+
+ public WorkflowNode getByUuid(long workflowId, String uuid) {
+ WorkflowNode node = ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowNode::getWorkflowId, workflowId)
+ .eq(WorkflowNode::getUuid, uuid)
+ .eq(WorkflowNode::getIsDeleted, false)
+ .last("limit 1")
+ .one();
+ checkAndDecrypt(node);
+ return node;
+ }
+
+ public List listByWorkflowId(Long workflowId) {
+ List list = ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowNode::getWorkflowId, workflowId)
+ .eq(WorkflowNode::getIsDeleted, false)
+ .list();
+ list.forEach(this::checkAndDecrypt);
+ return list;
+ }
+
+ public List copyByWorkflowId(long workflowId, long targetWorkflowId) {
+ List result = new ArrayList<>();
+ self.listByWorkflowId(workflowId).forEach(node -> {
+ result.add(self.copyNode(targetWorkflowId, node));
+ });
+ return result;
+ }
+
+ public WorkflowNode copyNode(Long targetWorkflowId, WorkflowNode sourceNode) {
+ WorkflowNode newNode = new WorkflowNode();
+ BeanUtils.copyProperties(sourceNode, newNode, "id", "createTime", "updateTime");
+ newNode.setWorkflowId(targetWorkflowId);
+ baseMapper.insert(newNode);
+
+ return ChainWrappers.lambdaQueryChain(baseMapper)
+ .eq(WorkflowNode::getWorkflowId, targetWorkflowId)
+ .eq(WorkflowNode::getUuid, newNode.getUuid())
+ .eq(WorkflowNode::getIsDeleted, false)
+ .last("limit 1")
+ .one();
+ }
+
+ @Transactional
+ public void createOrUpdateNodes(Long workflowId, List nodes) {
+ List