diff --git a/.image/Java监控.jpg b/.image/Java监控.jpg deleted file mode 100644 index 6ad522a..0000000 Binary files a/.image/Java监控.jpg and /dev/null differ diff --git a/.image/MySQL.jpg b/.image/MySQL.jpg deleted file mode 100644 index 64a1940..0000000 Binary files a/.image/MySQL.jpg and /dev/null differ diff --git a/.image/OA请假-列表.jpg b/.image/OA请假-列表.jpg deleted file mode 100644 index 787bb73..0000000 Binary files a/.image/OA请假-列表.jpg and /dev/null differ diff --git a/.image/OA请假-发起.jpg b/.image/OA请假-发起.jpg deleted file mode 100644 index 1a7342d..0000000 Binary files a/.image/OA请假-发起.jpg and /dev/null differ diff --git a/.image/OA请假-详情.jpg b/.image/OA请假-详情.jpg deleted file mode 100644 index a83e7c1..0000000 Binary files a/.image/OA请假-详情.jpg and /dev/null differ diff --git a/.image/Redis.jpg b/.image/Redis.jpg deleted file mode 100644 index 9569352..0000000 Binary files a/.image/Redis.jpg and /dev/null differ diff --git a/.image/admin-uniapp/01.png b/.image/admin-uniapp/01.png deleted file mode 100644 index 0f65d99..0000000 Binary files a/.image/admin-uniapp/01.png and /dev/null differ diff --git a/.image/admin-uniapp/02.png b/.image/admin-uniapp/02.png deleted file mode 100644 index 05ec781..0000000 Binary files a/.image/admin-uniapp/02.png and /dev/null differ diff --git a/.image/admin-uniapp/03.png b/.image/admin-uniapp/03.png deleted file mode 100644 index f400c68..0000000 Binary files a/.image/admin-uniapp/03.png and /dev/null differ diff --git a/.image/admin-uniapp/04.png b/.image/admin-uniapp/04.png deleted file mode 100644 index d5d5ea0..0000000 Binary files a/.image/admin-uniapp/04.png and /dev/null differ diff --git a/.image/admin-uniapp/05.png b/.image/admin-uniapp/05.png deleted file mode 100644 index 1de6d8a..0000000 Binary files a/.image/admin-uniapp/05.png and /dev/null differ diff --git a/.image/admin-uniapp/06.png b/.image/admin-uniapp/06.png deleted file mode 100644 index 400ae90..0000000 Binary files a/.image/admin-uniapp/06.png and /dev/null differ diff --git a/.image/admin-uniapp/07.png b/.image/admin-uniapp/07.png deleted file mode 100644 index 2ed8c0f..0000000 Binary files a/.image/admin-uniapp/07.png and /dev/null differ diff --git a/.image/admin-uniapp/08.png b/.image/admin-uniapp/08.png deleted file mode 100644 index 090e64a..0000000 Binary files a/.image/admin-uniapp/08.png and /dev/null differ diff --git a/.image/admin-uniapp/09.png b/.image/admin-uniapp/09.png deleted file mode 100644 index f2032c8..0000000 Binary files a/.image/admin-uniapp/09.png and /dev/null differ diff --git a/.image/common/ai-feature.png b/.image/common/ai-feature.png deleted file mode 100644 index 1c22dbe..0000000 Binary files a/.image/common/ai-feature.png and /dev/null differ diff --git a/.image/common/ai-preview.gif b/.image/common/ai-preview.gif deleted file mode 100644 index 5f13ac4..0000000 Binary files a/.image/common/ai-preview.gif and /dev/null differ diff --git a/.image/common/bpm-feature.png b/.image/common/bpm-feature.png deleted file mode 100644 index 23787fb..0000000 Binary files a/.image/common/bpm-feature.png and /dev/null differ diff --git a/.image/common/crm-feature.png b/.image/common/crm-feature.png deleted file mode 100644 index e1c9670..0000000 Binary files a/.image/common/crm-feature.png and /dev/null differ diff --git a/.image/common/erp-feature.png b/.image/common/erp-feature.png deleted file mode 100644 index d30b30e..0000000 Binary files a/.image/common/erp-feature.png and /dev/null differ diff --git a/.image/common/infra-feature.png b/.image/common/infra-feature.png deleted file mode 100644 index f5cef50..0000000 Binary files a/.image/common/infra-feature.png and /dev/null differ diff --git a/.image/common/lyzsys-cloud-architecture.png b/.image/common/lyzsys-cloud-architecture.png deleted file mode 100644 index 59416d8..0000000 Binary files a/.image/common/lyzsys-cloud-architecture.png and /dev/null differ diff --git a/.image/common/lyzsys-roadmap.png b/.image/common/lyzsys-roadmap.png deleted file mode 100644 index f4becc9..0000000 Binary files a/.image/common/lyzsys-roadmap.png and /dev/null differ diff --git a/.image/common/mall-feature.png b/.image/common/mall-feature.png deleted file mode 100644 index cca05c0..0000000 Binary files a/.image/common/mall-feature.png and /dev/null differ diff --git a/.image/common/mall-preview.png b/.image/common/mall-preview.png deleted file mode 100644 index e164cd2..0000000 Binary files a/.image/common/mall-preview.png and /dev/null differ diff --git a/.image/common/project-vs.png b/.image/common/project-vs.png deleted file mode 100644 index 561e092..0000000 Binary files a/.image/common/project-vs.png and /dev/null differ diff --git a/.image/common/ruoyi-vue-pro-architecture.png b/.image/common/ruoyi-vue-pro-architecture.png deleted file mode 100644 index 7bd7d59..0000000 Binary files a/.image/common/ruoyi-vue-pro-architecture.png and /dev/null differ diff --git a/.image/common/ruoyi-vue-pro-biz.png b/.image/common/ruoyi-vue-pro-biz.png deleted file mode 100644 index 24a385a..0000000 Binary files a/.image/common/ruoyi-vue-pro-biz.png and /dev/null differ diff --git a/.image/common/system-feature.png b/.image/common/system-feature.png deleted file mode 100644 index 366087c..0000000 Binary files a/.image/common/system-feature.png and /dev/null differ diff --git a/.image/个人中心.jpg b/.image/个人中心.jpg deleted file mode 100644 index ce57f6e..0000000 Binary files a/.image/个人中心.jpg and /dev/null differ diff --git a/.image/代码生成.jpg b/.image/代码生成.jpg deleted file mode 100644 index 751603e..0000000 Binary files a/.image/代码生成.jpg and /dev/null differ diff --git a/.image/令牌管理.jpg b/.image/令牌管理.jpg deleted file mode 100644 index 04abf4d..0000000 Binary files a/.image/令牌管理.jpg and /dev/null differ diff --git a/.image/任务列表-审批.jpg b/.image/任务列表-审批.jpg deleted file mode 100644 index cba312a..0000000 Binary files a/.image/任务列表-审批.jpg and /dev/null differ diff --git a/.image/任务列表-已办.jpg b/.image/任务列表-已办.jpg deleted file mode 100644 index 7a8d0fb..0000000 Binary files a/.image/任务列表-已办.jpg and /dev/null differ diff --git a/.image/任务列表-待办.jpg b/.image/任务列表-待办.jpg deleted file mode 100644 index a90323f..0000000 Binary files a/.image/任务列表-待办.jpg and /dev/null differ diff --git a/.image/任务日志.jpg b/.image/任务日志.jpg deleted file mode 100644 index 599e50a..0000000 Binary files a/.image/任务日志.jpg and /dev/null differ diff --git a/.image/商户信息.jpg b/.image/商户信息.jpg deleted file mode 100644 index 483eace..0000000 Binary files a/.image/商户信息.jpg and /dev/null differ diff --git a/.image/在线用户.jpg b/.image/在线用户.jpg deleted file mode 100644 index b183009..0000000 Binary files a/.image/在线用户.jpg and /dev/null differ diff --git a/.image/大屏设计器-列表.jpg b/.image/大屏设计器-列表.jpg deleted file mode 100644 index 9a45c3b..0000000 Binary files a/.image/大屏设计器-列表.jpg and /dev/null differ diff --git a/.image/大屏设计器-编辑.jpg b/.image/大屏设计器-编辑.jpg deleted file mode 100644 index 63298a0..0000000 Binary files a/.image/大屏设计器-编辑.jpg and /dev/null differ diff --git a/.image/大屏设计器-预览.jpg b/.image/大屏设计器-预览.jpg deleted file mode 100644 index 501d9ea..0000000 Binary files a/.image/大屏设计器-预览.jpg and /dev/null differ diff --git a/.image/字典数据.jpg b/.image/字典数据.jpg deleted file mode 100644 index 8298c89..0000000 Binary files a/.image/字典数据.jpg and /dev/null differ diff --git a/.image/字典类型.jpg b/.image/字典类型.jpg deleted file mode 100644 index 6613392..0000000 Binary files a/.image/字典类型.jpg and /dev/null differ diff --git a/.image/定时任务.jpg b/.image/定时任务.jpg deleted file mode 100644 index d5bbd85..0000000 Binary files a/.image/定时任务.jpg and /dev/null differ diff --git a/.image/岗位管理.jpg b/.image/岗位管理.jpg deleted file mode 100644 index 42b64d2..0000000 Binary files a/.image/岗位管理.jpg and /dev/null differ diff --git a/.image/工作流设计器-bpmn.jpg b/.image/工作流设计器-bpmn.jpg deleted file mode 100644 index 2a61f60..0000000 Binary files a/.image/工作流设计器-bpmn.jpg and /dev/null differ diff --git a/.image/工作流设计器-simple.jpg b/.image/工作流设计器-simple.jpg deleted file mode 100644 index 9ef2c9e..0000000 Binary files a/.image/工作流设计器-simple.jpg and /dev/null differ diff --git a/.image/应用信息-列表.jpg b/.image/应用信息-列表.jpg deleted file mode 100644 index da419a2..0000000 Binary files a/.image/应用信息-列表.jpg and /dev/null differ diff --git a/.image/应用信息-编辑.jpg b/.image/应用信息-编辑.jpg deleted file mode 100644 index 913cfbc..0000000 Binary files a/.image/应用信息-编辑.jpg and /dev/null differ diff --git a/.image/应用管理.jpg b/.image/应用管理.jpg deleted file mode 100644 index 6e7789f..0000000 Binary files a/.image/应用管理.jpg and /dev/null differ diff --git a/.image/我的流程-列表.jpg b/.image/我的流程-列表.jpg deleted file mode 100644 index 223d17a..0000000 Binary files a/.image/我的流程-列表.jpg and /dev/null differ diff --git a/.image/我的流程-发起.jpg b/.image/我的流程-发起.jpg deleted file mode 100644 index 7a83306..0000000 Binary files a/.image/我的流程-发起.jpg and /dev/null differ diff --git a/.image/我的流程-详情.jpg b/.image/我的流程-详情.jpg deleted file mode 100644 index 6a01541..0000000 Binary files a/.image/我的流程-详情.jpg and /dev/null differ diff --git a/.image/报表设计器-图形报表.jpg b/.image/报表设计器-图形报表.jpg deleted file mode 100644 index 681b318..0000000 Binary files a/.image/报表设计器-图形报表.jpg and /dev/null differ diff --git a/.image/报表设计器-打印设计.jpg b/.image/报表设计器-打印设计.jpg deleted file mode 100644 index bb86da6..0000000 Binary files a/.image/报表设计器-打印设计.jpg and /dev/null differ diff --git a/.image/报表设计器-数据报表.jpg b/.image/报表设计器-数据报表.jpg deleted file mode 100644 index 9ca5b9b..0000000 Binary files a/.image/报表设计器-数据报表.jpg and /dev/null differ diff --git a/.image/操作日志.jpg b/.image/操作日志.jpg deleted file mode 100644 index 4a0611a..0000000 Binary files a/.image/操作日志.jpg and /dev/null differ diff --git a/.image/支付订单.jpg b/.image/支付订单.jpg deleted file mode 100644 index 0a56dd7..0000000 Binary files a/.image/支付订单.jpg and /dev/null differ diff --git a/.image/敏感词.jpg b/.image/敏感词.jpg deleted file mode 100644 index 92a5397..0000000 Binary files a/.image/敏感词.jpg and /dev/null differ diff --git a/.image/数据库文档.jpg b/.image/数据库文档.jpg deleted file mode 100644 index a4339d9..0000000 Binary files a/.image/数据库文档.jpg and /dev/null differ diff --git a/.image/文件管理.jpg b/.image/文件管理.jpg deleted file mode 100644 index 054b19f..0000000 Binary files a/.image/文件管理.jpg and /dev/null differ diff --git a/.image/文件管理2.jpg b/.image/文件管理2.jpg deleted file mode 100644 index b12e5c3..0000000 Binary files a/.image/文件管理2.jpg and /dev/null differ diff --git a/.image/文件配置.jpg b/.image/文件配置.jpg deleted file mode 100644 index e618049..0000000 Binary files a/.image/文件配置.jpg and /dev/null differ diff --git a/.image/日志中心.jpg b/.image/日志中心.jpg deleted file mode 100644 index 27c1c6c..0000000 Binary files a/.image/日志中心.jpg and /dev/null differ diff --git a/.image/流程模型-列表.jpg b/.image/流程模型-列表.jpg deleted file mode 100644 index ffdc584..0000000 Binary files a/.image/流程模型-列表.jpg and /dev/null differ diff --git a/.image/流程模型-定义.jpg b/.image/流程模型-定义.jpg deleted file mode 100644 index 18b316c..0000000 Binary files a/.image/流程模型-定义.jpg and /dev/null differ diff --git a/.image/流程模型-设计.jpg b/.image/流程模型-设计.jpg deleted file mode 100644 index 9614969..0000000 Binary files a/.image/流程模型-设计.jpg and /dev/null differ diff --git a/.image/流程表单.jpg b/.image/流程表单.jpg deleted file mode 100644 index 60669c1..0000000 Binary files a/.image/流程表单.jpg and /dev/null differ diff --git a/.image/生成效果.jpg b/.image/生成效果.jpg deleted file mode 100644 index 98ff2cc..0000000 Binary files a/.image/生成效果.jpg and /dev/null differ diff --git a/.image/用户分组.jpg b/.image/用户分组.jpg deleted file mode 100644 index 39af1cd..0000000 Binary files a/.image/用户分组.jpg and /dev/null differ diff --git a/.image/用户管理.jpg b/.image/用户管理.jpg deleted file mode 100644 index 844604a..0000000 Binary files a/.image/用户管理.jpg and /dev/null differ diff --git a/.image/登录.jpg b/.image/登录.jpg deleted file mode 100644 index b782b98..0000000 Binary files a/.image/登录.jpg and /dev/null differ diff --git a/.image/登录日志.jpg b/.image/登录日志.jpg deleted file mode 100644 index 25662d9..0000000 Binary files a/.image/登录日志.jpg and /dev/null differ diff --git a/.image/短信日志.jpg b/.image/短信日志.jpg deleted file mode 100644 index ada8e56..0000000 Binary files a/.image/短信日志.jpg and /dev/null differ diff --git a/.image/短信模板.jpg b/.image/短信模板.jpg deleted file mode 100644 index 09381cc..0000000 Binary files a/.image/短信模板.jpg and /dev/null differ diff --git a/.image/短信渠道.jpg b/.image/短信渠道.jpg deleted file mode 100644 index df3a5c3..0000000 Binary files a/.image/短信渠道.jpg and /dev/null differ diff --git a/.image/租户套餐.png b/.image/租户套餐.png deleted file mode 100644 index 9663167..0000000 Binary files a/.image/租户套餐.png and /dev/null differ diff --git a/.image/租户管理.jpg b/.image/租户管理.jpg deleted file mode 100644 index 647416a..0000000 Binary files a/.image/租户管理.jpg and /dev/null differ diff --git a/.image/系统接口.jpg b/.image/系统接口.jpg deleted file mode 100644 index 6d39d42..0000000 Binary files a/.image/系统接口.jpg and /dev/null differ diff --git a/.image/菜单管理.jpg b/.image/菜单管理.jpg deleted file mode 100644 index ad3b797..0000000 Binary files a/.image/菜单管理.jpg and /dev/null differ diff --git a/.image/表单构建.jpg b/.image/表单构建.jpg deleted file mode 100644 index 81f0374..0000000 Binary files a/.image/表单构建.jpg and /dev/null differ diff --git a/.image/角色管理.jpg b/.image/角色管理.jpg deleted file mode 100644 index eed776e..0000000 Binary files a/.image/角色管理.jpg and /dev/null differ diff --git a/.image/访问日志.jpg b/.image/访问日志.jpg deleted file mode 100644 index ef301aa..0000000 Binary files a/.image/访问日志.jpg and /dev/null differ diff --git a/.image/退款订单.jpg b/.image/退款订单.jpg deleted file mode 100644 index 2c6c6c9..0000000 Binary files a/.image/退款订单.jpg and /dev/null differ diff --git a/.image/通知公告.jpg b/.image/通知公告.jpg deleted file mode 100644 index 97bb42f..0000000 Binary files a/.image/通知公告.jpg and /dev/null differ diff --git a/.image/部门管理.jpg b/.image/部门管理.jpg deleted file mode 100644 index 6eab233..0000000 Binary files a/.image/部门管理.jpg and /dev/null differ diff --git a/.image/配置管理.jpg b/.image/配置管理.jpg deleted file mode 100644 index 0abaec9..0000000 Binary files a/.image/配置管理.jpg and /dev/null differ diff --git a/.image/链路追踪.jpg b/.image/链路追踪.jpg deleted file mode 100644 index 12f7aa8..0000000 Binary files a/.image/链路追踪.jpg and /dev/null differ diff --git a/.image/错误日志.jpg b/.image/错误日志.jpg deleted file mode 100644 index eb615ea..0000000 Binary files a/.image/错误日志.jpg and /dev/null differ diff --git a/.image/错误码管理.jpg b/.image/错误码管理.jpg deleted file mode 100644 index ea91dde..0000000 Binary files a/.image/错误码管理.jpg and /dev/null differ diff --git a/.image/首页.jpg b/.image/首页.jpg deleted file mode 100644 index 10a7fde..0000000 Binary files a/.image/首页.jpg and /dev/null differ diff --git a/lyzsys-module-ai/pom.xml b/lyzsys-module-ai/pom.xml deleted file mode 100644 index e35eb9f..0000000 --- a/lyzsys-module-ai/pom.xml +++ /dev/null @@ -1,263 +0,0 @@ - - - - cn.iocoder.boot - lyzsys - ${revision} - - 4.0.0 - jar - lyzsys-module-ai - - ${project.artifactId} - - ai 模块下，接入 LLM 大模型，支持聊天、绘图、音乐、写作、思维导图等功能。 - 目前已接入各种模型，不限于： - 国内：通义千问、文心一言、讯飞星火、智谱 GLM、DeepSeek - 国外：OpenAI、Ollama、Midjourney、StableDiffusion、Suno - - - 1.1.2 - - 1.1.0.0-RC2 - 1.2.6 - - - - - cn.iocoder.boot - lyzsys-module-system - ${revision} - - - cn.iocoder.boot - lyzsys-module-infra - ${revision} - - - - - - cn.iocoder.boot - lyzsys-spring-boot-starter-biz-tenant - - - - - cn.iocoder.boot - lyzsys-spring-boot-starter-security - - - - - cn.iocoder.boot - lyzsys-spring-boot-starter-mybatis - - - - - cn.iocoder.boot - lyzsys-spring-boot-starter-job - - - - - cn.iocoder.boot - lyzsys-spring-boot-starter-test - - - - - cn.iocoder.boot - lyzsys-spring-boot-starter-excel - - - - - org.springframework.ai - spring-ai-starter-model-openai - ${spring-ai.version} - - - io.swagger.core.v3 - swagger-annotations-jakarta - - - - - org.springframework.ai - spring-ai-starter-model-azure-openai - ${spring-ai.version} - - - org.springframework.ai - spring-ai-starter-model-anthropic - ${spring-ai.version} - - - org.springframework.ai - spring-ai-starter-model-deepseek - ${spring-ai.version} - - - org.springframework.ai - spring-ai-starter-model-ollama - ${spring-ai.version} - - - org.springframework.ai - spring-ai-starter-model-stability-ai - ${spring-ai.version} - - - - org.springframework.ai - spring-ai-starter-model-zhipuai - ${spring-ai.version} - - - org.springframework.ai - spring-ai-starter-model-minimax - ${spring-ai.version} - - - - - com.alibaba.cloud.ai - spring-ai-alibaba-starter-dashscope - ${alibaba-ai.version} - - - - - org.springaicommunity - qianfan-spring-boot-starter - 1.0.0 - - - - org.springaicommunity - moonshot-spring-boot-starter - 1.0.0 - - - - - - org.springframework.ai - spring-ai-starter-vector-store-qdrant - ${spring-ai.version} - - - - - org.springframework.ai - spring-ai-starter-vector-store-redis - ${spring-ai.version} - - - cn.iocoder.boot - lyzsys-spring-boot-starter-redis - - - - - org.springframework.ai - spring-ai-starter-vector-store-milvus - ${spring-ai.version} - - - - org.slf4j - slf4j-reload4j - - - - - - - org.springframework.ai - spring-ai-tika-document-reader - ${spring-ai.version} - - - - spring-cloud-function-context - org.springframework.cloud - - - spring-cloud-function-core - org.springframework.cloud - - - - - - - - - org.springframework.ai - spring-ai-starter-mcp-server-webmvc - ${spring-ai.version} - - - io.swagger.core.v3 - swagger-annotations-jakarta - - - - - - org.springframework.ai - spring-ai-starter-mcp-client - ${spring-ai.version} - - - - - dev.tinyflow - tinyflow-java-core - ${tinyflow.version} - - - com.jfinal - enjoy - - - - com.agentsflex - agents-flex-store-elasticsearch - - - - com.agentsflex - agents-flex-search-engine-es - - - - org.codehaus.groovy - groovy-all - - - - org.slf4j - slf4j-simple - - - org.apache.logging.log4j - log4j-slf4j-impl - - - org.slf4j - slf4j-reload4j - - - - - - diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatConversationController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatConversationController.java deleted file mode 100644 index ad1a700..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatConversationController.java +++ /dev/null @@ -1,118 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat; - -import cn.hutool.core.collection.CollUtil; -import cn.hutool.core.util.ObjUtil; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation.AiChatConversationCreateMyReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation.AiChatConversationPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation.AiChatConversationRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation.AiChatConversationUpdateMyReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.chat.AiChatConversationDO; -import cn.iocoder.lyzsys.module.ai.service.chat.AiChatConversationService; -import cn.iocoder.lyzsys.module.ai.service.chat.AiChatMessageService; -import com.fhs.core.trans.anno.TransMethodResult; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; -import java.util.Map; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertList; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - AI 聊天对话") -@RestController -@RequestMapping("/ai/chat/conversation") -@Validated -public class AiChatConversationController { - - @Resource - private AiChatConversationService chatConversationService; - @Resource - private AiChatMessageService chatMessageService; - - @PostMapping("/create-my") - @Operation(summary = "创建【我的】聊天对话") - public CommonResult createChatConversationMy(@RequestBody @Valid AiChatConversationCreateMyReqVO createReqVO) { - return success(chatConversationService.createChatConversationMy(createReqVO, getLoginUserId())); - } - - @PutMapping("/update-my") - @Operation(summary = "更新【我的】聊天对话") - public CommonResult updateChatConversationMy(@RequestBody @Valid AiChatConversationUpdateMyReqVO updateReqVO) { - chatConversationService.updateChatConversationMy(updateReqVO, getLoginUserId()); - return success(true); - } - - @GetMapping("/my-list") - @Operation(summary = "获得【我的】聊天对话列表") - @TransMethodResult - public CommonResult> getChatConversationMyList() { - List list = chatConversationService.getChatConversationListByUserId(getLoginUserId()); - return success(BeanUtils.toBean(list, AiChatConversationRespVO.class)); - } - - @GetMapping("/get-my") - @Operation(summary = "获得【我的】聊天对话") - @Parameter(name = "id", required = true, description = "对话编号", example = "1024") - @TransMethodResult - public CommonResult getChatConversationMy(@RequestParam("id") Long id) { - AiChatConversationDO conversation = chatConversationService.getChatConversation(id); - if (conversation != null && ObjUtil.notEqual(conversation.getUserId(), getLoginUserId())) { - conversation = null; - } - return success(BeanUtils.toBean(conversation, AiChatConversationRespVO.class)); - } - - @DeleteMapping("/delete-my") - @Operation(summary = "删除聊天对话") - @Parameter(name = "id", required = true, description = "对话编号", example = "1024") - public CommonResult deleteChatConversationMy(@RequestParam("id") Long id) { - chatConversationService.deleteChatConversationMy(id, getLoginUserId()); - return success(true); - } - - @DeleteMapping("/delete-by-unpinned") - @Operation(summary = "删除未置顶的聊天对话") - public CommonResult deleteChatConversationMyByUnpinned() { - chatConversationService.deleteChatConversationMyByUnpinned(getLoginUserId()); - return success(true); - } - - // ========== 对话管理 ========== - - @GetMapping("/page") - @Operation(summary = "获得对话分页", description = "用于【对话管理】菜单") - @PreAuthorize("@ss.hasPermission('ai:chat-conversation:query')") - @TransMethodResult - public CommonResult> getChatConversationPage(AiChatConversationPageReqVO pageReqVO) { - PageResult pageResult = chatConversationService.getChatConversationPage(pageReqVO); - if (CollUtil.isEmpty(pageResult.getList())) { - return success(PageResult.empty()); - } - // 拼接关联数据 - Map messageCountMap = chatMessageService.getChatMessageCountMap( - convertList(pageResult.getList(), AiChatConversationDO::getId)); - return success(BeanUtils.toBean(pageResult, AiChatConversationRespVO.class, - conversation -> conversation.setMessageCount(messageCountMap.getOrDefault(conversation.getId(), 0)))); - } - - @Operation(summary = "管理员删除对话") - @DeleteMapping("/delete-by-admin") - @Parameter(name = "id", required = true, description = "对话编号", example = "1024") - @PreAuthorize("@ss.hasPermission('ai:chat-conversation:delete')") - public CommonResult deleteChatConversationByAdmin(@RequestParam("id") Long id) { - chatConversationService.deleteChatConversationByAdmin(id); - return success(true); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatMessageController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatMessageController.http deleted file mode 100644 index d1abfa4..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatMessageController.http +++ /dev/null @@ -1,66 +0,0 @@ -### 发送消息（段式） -POST {{baseUrl}}/ai/chat/message/send -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "conversationId": "1781604279872581724", - "content": "你是 OpenAI 么？" -} - -### 发送消息（流式） -POST {{baseUrl}}/ai/chat/message/send-stream -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "conversationId": "1781604279872581724", - "content": "1+1=?" -} - -### 发送消息（流式）【带文件】 -POST {{baseUrl}}/ai/chat/message/send-stream -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "conversationId": "1781604279872581797", - "content": "图片里有什么？", - "attachmentUrls": ["http://test.lyzsys.iocoder.cn/1755531278.jpeg"] -} - -### 发送消息（流式）【追问带文件】 -POST {{baseUrl}}/ai/chat/message/send-stream -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "conversationId": "1781604279872581799", - "content": "说下图片里，有哪些字？", - "useContext": true -} - -### 发送消息（流式）【联网搜索】 -POST {{baseUrl}}/ai/chat/message/send-stream -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "conversationId": "1781604279872581799", - "content": "今天是周几？", - "useSearch": true -} - -### 获得指定对话的消息列表 -GET {{baseUrl}}/ai/chat/message/list-by-conversation-id?conversationId=1781604279872581799 -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -### 删除消息 -DELETE {{baseUrl}}/ai/chat/message/delete?id=50 -Authorization: {{token}} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatMessageController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatMessageController.java deleted file mode 100644 index 3645d15..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/AiChatMessageController.java +++ /dev/null @@ -1,157 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat; - -import cn.hutool.core.collection.CollUtil; -import cn.hutool.core.util.ObjUtil; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.collection.MapUtils; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message.AiChatMessagePageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message.AiChatMessageRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message.AiChatMessageSendReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message.AiChatMessageSendRespVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.chat.AiChatConversationDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.chat.AiChatMessageDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDocumentDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeSegmentDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiChatRoleDO; -import cn.iocoder.lyzsys.module.ai.service.chat.AiChatConversationService; -import cn.iocoder.lyzsys.module.ai.service.chat.AiChatMessageService; -import cn.iocoder.lyzsys.module.ai.service.knowledge.AiKnowledgeDocumentService; -import cn.iocoder.lyzsys.module.ai.service.knowledge.AiKnowledgeSegmentService; -import cn.iocoder.lyzsys.module.ai.service.model.AiChatRoleService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import lombok.extern.slf4j.Slf4j; -import org.springframework.http.MediaType; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.web.bind.annotation.*; -import reactor.core.publisher.Flux; - -import java.util.Collections; -import java.util.List; -import java.util.Map; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.*; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - 聊天消息") -@RestController -@RequestMapping("/ai/chat/message") -@Slf4j -public class AiChatMessageController { - - @Resource - private AiChatMessageService chatMessageService; - @Resource - private AiChatConversationService chatConversationService; - @Resource - private AiChatRoleService chatRoleService; - @Resource - private AiKnowledgeSegmentService knowledgeSegmentService; - @Resource - private AiKnowledgeDocumentService knowledgeDocumentService; - - @Operation(summary = "发送消息（段式）", description = "一次性返回，响应较慢") - @PostMapping("/send") - public CommonResult sendMessage(@Valid @RequestBody AiChatMessageSendReqVO sendReqVO) { - return success(chatMessageService.sendMessage(sendReqVO, getLoginUserId())); - } - - @Operation(summary = "发送消息（流式）", description = "流式返回，响应较快") - @PostMapping(value = "/send-stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE) - public Flux> sendChatMessageStream(@Valid @RequestBody AiChatMessageSendReqVO sendReqVO) { - return chatMessageService.sendChatMessageStream(sendReqVO, getLoginUserId()); - } - - @Operation(summary = "获得指定对话的消息列表") - @GetMapping("/list-by-conversation-id") - @Parameter(name = "conversationId", required = true, description = "对话编号", example = "1024") - public CommonResult> getChatMessageListByConversationId( - @RequestParam("conversationId") Long conversationId) { - AiChatConversationDO conversation = chatConversationService.getChatConversation(conversationId); - if (conversation == null || ObjUtil.notEqual(conversation.getUserId(), getLoginUserId())) { - return success(Collections.emptyList()); - } - // 1. 获取消息列表 - List messageList = chatMessageService.getChatMessageListByConversationId(conversationId); - if (CollUtil.isEmpty(messageList)) { - return success(Collections.emptyList()); - } - - // 2. 拼接数据，主要是知识库段落信息 - Map segmentMap = knowledgeSegmentService.getKnowledgeSegmentMap(convertListByFlatMap(messageList, - message -> CollUtil.isEmpty(message.getSegmentIds()) ? null : message.getSegmentIds().stream())); - Map documentMap = knowledgeDocumentService.getKnowledgeDocumentMap( - convertList(segmentMap.values(), AiKnowledgeSegmentDO::getDocumentId)); - List messageVOList = BeanUtils.toBean(messageList, AiChatMessageRespVO.class); - for (int i = 0; i < messageList.size(); i++) { - AiChatMessageDO message = messageList.get(i); - if (CollUtil.isEmpty(message.getSegmentIds())) { - continue; - } - // 设置知识库段落信息 - messageVOList.get(i).setSegments(convertList(message.getSegmentIds(), segmentId -> { - AiKnowledgeSegmentDO segment = segmentMap.get(segmentId); - if (segment == null) { - return null; - } - AiKnowledgeDocumentDO document = documentMap.get(segment.getDocumentId()); - if (document == null) { - return null; - } - return new AiChatMessageRespVO.KnowledgeSegment().setId(segment.getId()).setContent(segment.getContent()) - .setDocumentId(segment.getDocumentId()).setDocumentName(document.getName()); - })); - } - return success(messageVOList); - } - - @Operation(summary = "删除消息") - @DeleteMapping("/delete") - @Parameter(name = "id", required = true, description = "消息编号", example = "1024") - public CommonResult deleteChatMessage(@RequestParam("id") Long id) { - chatMessageService.deleteChatMessage(id, getLoginUserId()); - return success(true); - } - - @Operation(summary = "删除指定对话的消息") - @DeleteMapping("/delete-by-conversation-id") - @Parameter(name = "conversationId", required = true, description = "对话编号", example = "1024") - public CommonResult deleteChatMessageByConversationId(@RequestParam("conversationId") Long conversationId) { - chatMessageService.deleteChatMessageByConversationId(conversationId, getLoginUserId()); - return success(true); - } - - // ========== 对话管理 ========== - - @GetMapping("/page") - @Operation(summary = "获得消息分页", description = "用于【对话管理】菜单") - @PreAuthorize("@ss.hasPermission('ai:chat-conversation:query')") - public CommonResult> getChatMessagePage(AiChatMessagePageReqVO pageReqVO) { - PageResult pageResult = chatMessageService.getChatMessagePage(pageReqVO); - if (CollUtil.isEmpty(pageResult.getList())) { - return success(PageResult.empty()); - } - // 拼接数据 - Map roleMap = chatRoleService.getChatRoleMap( - convertSet(pageResult.getList(), AiChatMessageDO::getRoleId)); - return success(BeanUtils.toBean(pageResult, AiChatMessageRespVO.class, - respVO -> MapUtils.findAndThen(roleMap, respVO.getRoleId(), - role -> respVO.setRoleName(role.getName())))); - } - - @Operation(summary = "管理员删除消息") - @DeleteMapping("/delete-by-admin") - @Parameter(name = "id", required = true, description = "消息编号", example = "1024") - @PreAuthorize("@ss.hasPermission('ai:chat-message:delete')") - public CommonResult deleteChatMessageByAdmin(@RequestParam("id") Long id) { - chatMessageService.deleteChatMessageByAdmin(id); - return success(true); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationCreateMyReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationCreateMyReqVO.java deleted file mode 100644 index e3335b8..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationCreateMyReqVO.java +++ /dev/null @@ -1,16 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 聊天对话创建【我的】 Request VO") -@Data -public class AiChatConversationCreateMyReqVO { - - @Schema(description = "聊天角色编号", example = "666") - private Long roleId; - - @Schema(description = "知识库编号", example = "1204") - private Long knowledgeId; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationPageReqVO.java deleted file mode 100644 index 63e3c1c..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationPageReqVO.java +++ /dev/null @@ -1,26 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 聊天对话的分页 Request VO") -@Data -public class AiChatConversationPageReqVO extends PageParam { - - @Schema(description = "用户编号", example = "1024") - private Long userId; - - @Schema(description = "对话标题", example = "你好") - private String title; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationRespVO.java deleted file mode 100644 index cce397d..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationRespVO.java +++ /dev/null @@ -1,71 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation; - -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiChatRoleDO; -import com.fhs.core.trans.anno.Trans; -import com.fhs.core.trans.constant.TransType; -import com.fhs.core.trans.vo.VO; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 聊天对话 Response VO") -@Data -public class AiChatConversationRespVO implements VO { - - @Schema(description = "对话编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Long id; - - @Schema(description = "用户编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "2048") - private Long userId; - - @Schema(description = "对话标题", requiredMode = Schema.RequiredMode.REQUIRED, example = "我是一个标题") - private String title; - - @Schema(description = "是否置顶", requiredMode = Schema.RequiredMode.REQUIRED, example = "true") - private Boolean pinned; - - @Schema(description = "角色编号", example = "1") - @Trans(type = TransType.SIMPLE, target = AiChatRoleDO.class, fields = {"name", "avatar"}, refs = {"roleName", "roleAvatar"}) - private Long roleId; - - @Schema(description = "模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @Trans(type = TransType.SIMPLE, target = AiModelDO.class, fields = "name", ref = "modelName") - private Long modelId; - - @Schema(description = "模型标志", requiredMode = Schema.RequiredMode.REQUIRED, example = "ERNIE-Bot-turbo-0922") - private String model; - - @Schema(description = "模型名字", requiredMode = Schema.RequiredMode.REQUIRED, example = "张三") - private String modelName; - - @Schema(description = "角色设定", example = "一个快乐的程序员") - private String systemMessage; - - @Schema(description = "温度参数", requiredMode = Schema.RequiredMode.REQUIRED, example = "0.8") - private Double temperature; - - @Schema(description = "单条回复的最大 Token 数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "4096") - private Integer maxTokens; - - @Schema(description = "上下文的最大 Message 数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "10") - private Integer maxContexts; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - - // ========== 关联 role 信息 ========== - - @Schema(description = "角色头像", example = "https://www.iocoder.cn/1.png") - private String roleAvatar; - - @Schema(description = "角色名字", example = "小黄") - private String roleName; - - // ========== 仅在【对话管理】时加载 ========== - - @Schema(description = "消息数量", example = "20") - private Integer messageCount; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationUpdateMyReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationUpdateMyReqVO.java deleted file mode 100644 index 29a5ebf..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/conversation/AiChatConversationUpdateMyReqVO.java +++ /dev/null @@ -1,39 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 聊天对话更新【我的】 Request VO") -@Data -public class AiChatConversationUpdateMyReqVO { - - @Schema(description = "对话编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - @NotNull(message = "对话编号不能为空") - private Long id; - - @Schema(description = "对话标题", example = "我是一个标题") - private String title; - - @Schema(description = "是否置顶", example = "true") - private Boolean pinned; - - @Schema(description = "模型编号", example = "1") - private Long modelId; - - @Schema(description = "知识库编号", example = "1") - private Long knowledgeId; - - @Schema(description = "角色设定", example = "一个快乐的程序员") - private String systemMessage; - - @Schema(description = "温度参数", example = "0.8") - private Double temperature; - - @Schema(description = "单条回复的最大 Token 数量", example = "4096") - private Integer maxTokens; - - @Schema(description = "上下文的最大 Message 数量", example = "10") - private Integer maxContexts; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessagePageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessagePageReqVO.java deleted file mode 100644 index b74d47e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessagePageReqVO.java +++ /dev/null @@ -1,29 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 聊天消息的分页 Request VO") -@Data -public class AiChatMessagePageReqVO extends PageParam { - - @Schema(description = "对话编号", example = "2048") - private Long conversationId; - - @Schema(description = "用户编号", example = "1024") - private Long userId; - - @Schema(description = "消息内容", example = "你好") - private String content; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageRespVO.java deleted file mode 100644 index 59531aa..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageRespVO.java +++ /dev/null @@ -1,85 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message; - -import cn.iocoder.lyzsys.module.ai.framework.ai.core.webserch.AiWebSearchResponse; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; -import java.util.List; - -@Schema(description = "管理后台 - AI 聊天消息 Response VO") -@Data -public class AiChatMessageRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Long id; - - @Schema(description = "对话编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "2048") - private Long conversationId; - - @Schema(description = "回复消息编号", example = "1024") - private Long replyId; - - @Schema(description = "消息类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "role") - private String type; // 参见 MessageType 枚举类 - - @Schema(description = "用户编号", example = "4096") - private Long userId; - - @Schema(description = "角色编号", example = "888") - private Long roleId; - - @Schema(description = "模型标志", requiredMode = Schema.RequiredMode.REQUIRED, example = "gpt-3.5-turbo") - private String model; - - @Schema(description = "模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "123") - private Long modelId; - - @Schema(description = "聊天内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "你好，你好啊") - private String content; - - @Schema(description = "推理内容", example = "要达到这个目标，你需要...") - private String reasoningContent; - - @Schema(description = "是否携带上下文", requiredMode = Schema.RequiredMode.REQUIRED, example = "true") - private Boolean useContext; - - @Schema(description = "知识库段落编号数组", example = "[1,2,3]") - private List segmentIds; - - @Schema(description = "知识库段落数组") - private List segments; - - @Schema(description = "联网搜索的网页内容数组") - private List webSearchPages; - - @Schema(description = "附件 URL 数组", example = "https://www.iocoder.cn/1.png") - private List attachmentUrls; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED, example = "2024-05-12 12:51") - private LocalDateTime createTime; - - // ========== 仅在【对话管理】时加载 ========== - - @Schema(description = "角色名字", example = "小黄") - private String roleName; - - @Schema(description = "知识库段落", example = "Java 开发手册") - @Data - public static class KnowledgeSegment { - - @Schema(description = "段落编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Long id; - - @Schema(description = "切片内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "Java 开发手册") - private String content; - - @Schema(description = "文档编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long documentId; - - @Schema(description = "文档名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "产品使用手册") - private String documentName; - - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageSendReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageSendReqVO.java deleted file mode 100644 index a0e3245..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageSendReqVO.java +++ /dev/null @@ -1,31 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -import java.util.List; - -@Schema(description = "管理后台 - AI 聊天消息发送 Request VO") -@Data -public class AiChatMessageSendReqVO { - - @Schema(description = "聊天对话编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - @NotNull(message = "聊天对话编号不能为空") - private Long conversationId; - - @Schema(description = "聊天内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "帮我写个 Java 算法") - @NotEmpty(message = "聊天内容不能为空") - private String content; - - @Schema(description = "是否携带上下文", example = "true") - private Boolean useContext; - - @Schema(description = "是否联网搜索", example = "true") - private Boolean useSearch; - - @Schema(description = "附件 URL 数组", example = "https://www.iocoder.cn/1.png") - private List attachmentUrls; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageSendRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageSendRespVO.java deleted file mode 100644 index 06ce7fa..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/chat/vo/message/AiChatMessageSendRespVO.java +++ /dev/null @@ -1,50 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message; - -import cn.iocoder.lyzsys.module.ai.framework.ai.core.webserch.AiWebSearchResponse; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; -import java.util.List; - -@Schema(description = "管理后台 - AI 聊天消息发送 Response VO") -@Data -public class AiChatMessageSendRespVO { - - @Schema(description = "发送消息", requiredMode = Schema.RequiredMode.REQUIRED) - private Message send; - - @Schema(description = "接收消息", requiredMode = Schema.RequiredMode.REQUIRED) - private Message receive; - - @Schema(description = "消息") - @Data - public static class Message { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Long id; - - @Schema(description = "消息类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "role") - private String type; // 参见 MessageType 枚举类 - - @Schema(description = "聊天内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "你好，你好啊") - private String content; - - @Schema(description = "推理内容", example = "要达到这个目标，你需要...") - private String reasoningContent; - - @Schema(description = "知识库段落编号数组", example = "[1,2,3]") - private List segmentIds; - - @Schema(description = "知识库段落数组") - private List segments; - - @Schema(description = "联网搜索的网页内容数组") - private List webSearchPages; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/AiImageController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/AiImageController.http deleted file mode 100644 index 9047610..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/AiImageController.http +++ /dev/null @@ -1,42 +0,0 @@ -### 生成图片：OpenAI（DALL） -POST {{baseUrl}}/ai/image/draw -Content-Type: application/json -Authorization: {{token}} - -{ - "platform": "OpenAI", - "prompt": "可爱的小喵星人", - "model": "dall-e-3", - "height": "1024", - "width": "1024", - "options": { - "style": "vivid" - } -} - -### 生成图片：StableDiffusion -POST {{baseUrl}}/ai/image/draw -Content-Type: application/json -Authorization: {{token}} - -{ - "platform": "StableDiffusion", - "prompt": "中国长城", - "model": "stable-diffusion-v1-6", - "height": "1024", - "width": "1024", - "style": "vivid" -} - -### 生成图片：生成图片（Midjourney） -POST {{baseUrl}}/ai/image/midjourney/imagine -Content-Type: application/json -Authorization: {{token}} - -{ - "prompt": "中国旗袍", - "model": "midjourney", - "width": "1", - "height": "1", - "version": "6.0" -} \ No newline at end of file diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/AiImageController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/AiImageController.java deleted file mode 100644 index 2d3e06b..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/AiImageController.java +++ /dev/null @@ -1,139 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image; - -import cn.hutool.core.util.ObjUtil; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.framework.tenant.core.aop.TenantIgnore; -import cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.*; -import cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.midjourney.AiMidjourneyActionReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.midjourney.AiMidjourneyImagineReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.image.AiImageDO; -import cn.iocoder.lyzsys.module.ai.service.image.AiImageService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.annotation.security.PermitAll; -import jakarta.validation.Valid; -import lombok.extern.slf4j.Slf4j; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - AI 绘画") -@RestController -@RequestMapping("/ai/image") -@Slf4j -public class AiImageController { - - @Resource - private AiImageService imageService; - - @GetMapping("/my-page") - @Operation(summary = "获取【我的】绘图分页") - public CommonResult> getImagePageMy(@Validated AiImagePageReqVO pageReqVO) { - PageResult pageResult = imageService.getImagePageMy(getLoginUserId(), pageReqVO); - return success(BeanUtils.toBean(pageResult, AiImageRespVO.class)); - } - - @GetMapping("/public-page") - @Operation(summary = "获取公开的绘图分页") - public CommonResult> getImagePagePublic(AiImagePublicPageReqVO pageReqVO) { - PageResult pageResult = imageService.getImagePagePublic(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiImageRespVO.class)); - } - - @GetMapping("/get-my") - @Operation(summary = "获取【我的】绘图记录") - @Parameter(name = "id", required = true, description = "绘画编号", example = "1024") - public CommonResult getImageMy(@RequestParam("id") Long id) { - AiImageDO image = imageService.getImage(id); - if (image == null || ObjUtil.notEqual(getLoginUserId(), image.getUserId())) { - return success(null); - } - return success(BeanUtils.toBean(image, AiImageRespVO.class)); - } - - @GetMapping("/my-list-by-ids") - @Operation(summary = "获取【我的】绘图记录列表") - @Parameter(name = "ids", required = true, description = "绘画编号数组", example = "1024,2048") - public CommonResult> getImageListMyByIds(@RequestParam("ids") List ids) { - List imageList = imageService.getImageList(ids); - imageList.removeIf(item -> !ObjUtil.equal(getLoginUserId(), item.getUserId())); - return success(BeanUtils.toBean(imageList, AiImageRespVO.class)); - } - - @Operation(summary = "生成图片") - @PostMapping("/draw") - public CommonResult drawImage(@Valid @RequestBody AiImageDrawReqVO drawReqVO) { - return success(imageService.drawImage(getLoginUserId(), drawReqVO)); - } - - @Operation(summary = "删除【我的】绘画记录") - @DeleteMapping("/delete-my") - @Parameter(name = "id", required = true, description = "绘画编号", example = "1024") - public CommonResult deleteImageMy(@RequestParam("id") Long id) { - imageService.deleteImageMy(id, getLoginUserId()); - return success(true); - } - - // ================ midjourney 专属 ================ - - @Operation(summary = "【Midjourney】生成图片") - @PostMapping("/midjourney/imagine") - public CommonResult midjourneyImagine(@Valid @RequestBody AiMidjourneyImagineReqVO reqVO) { - Long imageId = imageService.midjourneyImagine(getLoginUserId(), reqVO); - return success(imageId); - } - - @Operation(summary = "【Midjourney】通知图片进展", description = "由 Midjourney Proxy 回调") - @PostMapping("/midjourney/notify") // 必须是 POST 方法，否则会报错 - @PermitAll - @TenantIgnore - public CommonResult midjourneyNotify(@Valid @RequestBody MidjourneyApi.Notify notify) { - imageService.midjourneyNotify(notify); - return success(true); - } - - @Operation(summary = "【Midjourney】Action 操作（二次生成图片）", description = "例如说：放大、缩小、U1、U2 等") - @PostMapping("/midjourney/action") - public CommonResult midjourneyAction(@Valid @RequestBody AiMidjourneyActionReqVO reqVO) { - Long imageId = imageService.midjourneyAction(getLoginUserId(), reqVO); - return success(imageId); - } - - // ================ 绘图管理 ================ - - @GetMapping("/page") - @Operation(summary = "获得绘画分页") - @PreAuthorize("@ss.hasPermission('ai:image:query')") - public CommonResult> getImagePage(@Valid AiImagePageReqVO pageReqVO) { - PageResult pageResult = imageService.getImagePage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiImageRespVO.class)); - } - - @PutMapping("/update") - @Operation(summary = "更新绘画") - @PreAuthorize("@ss.hasPermission('ai:image:update')") - public CommonResult updateImage(@Valid @RequestBody AiImageUpdateReqVO updateReqVO) { - imageService.updateImage(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除绘画") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:image:delete')") - public CommonResult deleteImage(@RequestParam("id") Long id) { - imageService.deleteImage(id); - return success(true); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageDrawReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageDrawReqVO.java deleted file mode 100644 index c5f9284..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageDrawReqVO.java +++ /dev/null @@ -1,49 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import jakarta.validation.constraints.Size; -import lombok.Data; -import org.springframework.ai.openai.OpenAiImageOptions; -import org.springframework.ai.stabilityai.api.StabilityAiImageOptions; - -import java.util.Map; - -@Schema(description = "管理后台 - AI 绘画 Request VO") -@Data -public class AiImageDrawReqVO { - - @Schema(description = "模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - @NotNull(message = "模型编号不能为空") - private Long modelId; - - @Schema(description = "提示词", requiredMode = Schema.RequiredMode.REQUIRED, example = "画一个长城") - @NotEmpty(message = "提示词不能为空") - @Size(max = 1200, message = "提示词最大 1200") - private String prompt; - - /** - * 1. dall-e-2 模型：256x256、512x512、1024x1024 - * 2. dall-e-3 模型：1024x1024, 1792x1024, 或 1024x1792 - */ - @Schema(description = "图片高度") - @NotNull(message = "图片高度不能为空") - private Integer height; - - @Schema(description = "图片宽度") - @NotNull(message = "图片宽度不能为空") - private Integer width; - - // ========== 各平台绘画的拓展参数 ========== - - /** - * 绘制参数，不同 platform 的不同参数 - * - * 1. {@link OpenAiImageOptions} - * 2. {@link StabilityAiImageOptions} - */ - @Schema(description = "绘制参数") - private Map options; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImagePageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImagePageReqVO.java deleted file mode 100644 index 9ec3ab2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImagePageReqVO.java +++ /dev/null @@ -1,35 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 绘画分页 Request VO") -@Data -public class AiImagePageReqVO extends PageParam { - - @Schema(description = "用户编号", example = "28987") - private Long userId; - - @Schema(description = "平台", example = "OpenAI") - private String platform; - - @Schema(description = "提示词", example = "1") - private String prompt; - - @Schema(description = "绘画状态", example = "1") - private Integer status; - - @Schema(description = "是否发布", example = "1") - private Boolean publicStatus; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImagePublicPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImagePublicPageReqVO.java deleted file mode 100644 index f02ec7d..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImagePublicPageReqVO.java +++ /dev/null @@ -1,14 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 绘画公开的分页 Request VO") -@Data -public class AiImagePublicPageReqVO extends PageParam { - - @Schema(description = "提示词") - private String prompt; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageRespVO.java deleted file mode 100644 index 4e9a0c9..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageRespVO.java +++ /dev/null @@ -1,60 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo; - -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; -import java.util.List; -import java.util.Map; - -@Schema(description = "管理后台 - AI 绘画 Response VO") -@Data -public class AiImageRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Long id; - - @Schema(description = "用户编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Long userId; - - @Schema(description = "平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "OpenAI") - private String platform; // 参见 AiPlatformEnum 枚举 - - @Schema(description = "模型", requiredMode = Schema.RequiredMode.REQUIRED, example = "stable-diffusion-v1-6") - private String model; - - @Schema(description = "提示词", requiredMode = Schema.RequiredMode.REQUIRED, example = "南极的小企鹅") - private String prompt; - - @Schema(description = "图片宽度", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Integer width; - - @Schema(description = "图片高度", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Integer height; - - @Schema(description = "绘画状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "10") - private Integer status; - - @Schema(description = "是否发布", requiredMode = Schema.RequiredMode.REQUIRED, example = "public") - private Boolean publicStatus; - - @Schema(description = "图片地址", example = "https://www.iocoder.cn/1.png") - private String picUrl; - - @Schema(description = "绘画错误信息", example = "图片错误信息") - private String errorMessage; - - @Schema(description = "绘制参数") - private Map options; - - @Schema(description = "mj buttons 按钮") - private List buttons; - - @Schema(description = "完成时间") - private LocalDateTime finishTime; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageUpdateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageUpdateReqVO.java deleted file mode 100644 index 3d299bc..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/AiImageUpdateReqVO.java +++ /dev/null @@ -1,18 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 绘画修改 Request VO") -@Data -public class AiImageUpdateReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "15583") - @NotNull(message = "编号不能为空") - private Long id; - - @Schema(description = "是否发布", example = "true") - private Boolean publicStatus; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/midjourney/AiMidjourneyActionReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/midjourney/AiMidjourneyActionReqVO.java deleted file mode 100644 index f919f3a..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/midjourney/AiMidjourneyActionReqVO.java +++ /dev/null @@ -1,20 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.midjourney; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 绘图操作（Midjourney） Request VO") -@Data -public class AiMidjourneyActionReqVO { - - @Schema(description = "图片编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "图片编号不能为空") - private Long id; - - @Schema(description = "操作按钮编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "MJ::JOB::variation::4::06aa3e66-0e97-49cc-8201-e0295d883de4") - @NotEmpty(message = "操作按钮编号不能为空") - private String customId; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/midjourney/AiMidjourneyImagineReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/midjourney/AiMidjourneyImagineReqVO.java deleted file mode 100644 index 1a0db98..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/image/vo/midjourney/AiMidjourneyImagineReqVO.java +++ /dev/null @@ -1,35 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.midjourney; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 绘画生成（Midjourney） Request VO") -@Data -public class AiMidjourneyImagineReqVO { - - @Schema(description = "提示词", requiredMode = Schema.RequiredMode.REQUIRED, example = "中国神龙") - @NotEmpty(message = "提示词不能为空!") - private String prompt; - - @Schema(description = "模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "模型编号不能为空") - private Long modelId; - - @Schema(description = "图片宽度", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "图片宽度不能为空") - private Integer width; - - @Schema(description = "图片高度", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "图片高度不能为空") - private Integer height; - - @Schema(description = "版本号", requiredMode = Schema.RequiredMode.REQUIRED, example = "6.0") - @NotEmpty(message = "版本号不能为空") - private String version; - - @Schema(description = "参考图", example = "https://www.iocoder.cn/x.png") - private String referImageUrl; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeController.http deleted file mode 100644 index a0f1278..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeController.http +++ /dev/null @@ -1,35 +0,0 @@ -### 创建知识库 -POST {{baseUrl}}/ai/knowledge/create -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "name": "测试标题", - "description": "测试描述", - "embeddingModelId": 30, - "topK": 3, - "similarityThreshold": 0.5, - "status": 0 -} - -### 更新知识库 -PUT {{baseUrl}}/ai/knowledge/update -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "id": 1, - "name": "测试标题（更新）", - "description": "测试描述", - "embeddingModelId": 30, - "topK": 5, - "similarityThreshold": 0.6, - "status": 0 -} - -### 获取知识库分页 -GET {{baseUrl}}/ai/knowledge/page?pageNo=1&pageSize=10 -Authorization: {{token}} -tenant-id: {{adminTenantId}} \ No newline at end of file diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeController.java deleted file mode 100644 index 5284ca4..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeController.java +++ /dev/null @@ -1,84 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge.AiKnowledgePageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge.AiKnowledgeRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge.AiKnowledgeSaveReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDO; -import cn.iocoder.lyzsys.module.ai.service.knowledge.AiKnowledgeService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertList; - -@Tag(name = "管理后台 - AI 知识库") -@RestController -@RequestMapping("/ai/knowledge") -@Validated -public class AiKnowledgeController { - - @Resource - private AiKnowledgeService knowledgeService; - - @GetMapping("/page") - @Operation(summary = "获取知识库分页") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult> getKnowledgePage(@Valid AiKnowledgePageReqVO pageReqVO) { - PageResult pageResult = knowledgeService.getKnowledgePage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiKnowledgeRespVO.class)); - } - - @GetMapping("/get") - @Operation(summary = "获得知识库") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult getKnowledge(@RequestParam("id") Long id) { - AiKnowledgeDO knowledge = knowledgeService.getKnowledge(id); - return success(BeanUtils.toBean(knowledge, AiKnowledgeRespVO.class)); - } - - @PostMapping("/create") - @Operation(summary = "创建知识库") - @PreAuthorize("@ss.hasPermission('ai:knowledge:create')") - public CommonResult createKnowledge(@RequestBody @Valid AiKnowledgeSaveReqVO createReqVO) { - return success(knowledgeService.createKnowledge(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新知识库") - @PreAuthorize("@ss.hasPermission('ai:knowledge:update')") - public CommonResult updateKnowledge(@RequestBody @Valid AiKnowledgeSaveReqVO updateReqVO) { - knowledgeService.updateKnowledge(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除知识库") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:knowledge:delete')") - public CommonResult deleteKnowledge(@RequestParam("id") Long id) { - knowledgeService.deleteKnowledge(id); - return success(true); - } - - @GetMapping("/simple-list") - @Operation(summary = "获得知识库的精简列表") - public CommonResult> getKnowledgeSimpleList() { - List list = knowledgeService.getKnowledgeSimpleListByStatus(CommonStatusEnum.ENABLE.getStatus()); - return success(convertList(list, knowledge -> new AiKnowledgeRespVO() - .setId(knowledge.getId()).setName(knowledge.getName()))); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeDocumentController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeDocumentController.http deleted file mode 100644 index 209e917..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeDocumentController.http +++ /dev/null @@ -1,35 +0,0 @@ -### 创建知识文档 -POST {{baseUrl}}/ai/knowledge/document/create -Content-Type: application/json -Authorization: Bearer {{token}} -tenant-id: {{adminTenantId}} - -{ - "knowledgeId": 2, - "name": "测试文档", - "url": "https://static.iocoder.cn/README.md", - "segmentMaxTokens": 800 -} - -### 批量创建知识文档 -POST {{baseUrl}}/ai/knowledge/document/create-list -Content-Type: application/json -Authorization: Bearer {{token}} -tenant-id: {{adminTenantId}} - -{ - "knowledgeId": 1, - "list": [ - { - "name": "测试文档1", - "url": "https://static.iocoder.cn/README.md", - "segmentMaxTokens": 800 - }, - { - "name": "测试文档2", - "url": "https://static.iocoder.cn/README_lyzsys.md", - "segmentMaxTokens": 400 - } - ] -} - diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeDocumentController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeDocumentController.java deleted file mode 100644 index e0f5cb4..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeDocumentController.java +++ /dev/null @@ -1,90 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge; - -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document.*; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge.AiKnowledgeDocumentCreateReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDocumentDO; -import cn.iocoder.lyzsys.module.ai.service.knowledge.AiKnowledgeDocumentService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; - -@Tag(name = "管理后台 - AI 知识库文档") -@RestController -@RequestMapping("/ai/knowledge/document") -@Validated -public class AiKnowledgeDocumentController { - - @Resource - private AiKnowledgeDocumentService documentService; - - @GetMapping("/page") - @Operation(summary = "获取文档分页") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult> getKnowledgeDocumentPage( - @Valid AiKnowledgeDocumentPageReqVO pageReqVO) { - PageResult pageResult = documentService.getKnowledgeDocumentPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiKnowledgeDocumentRespVO.class)); - } - - @GetMapping("/get") - @Operation(summary = "获取文档详情") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult getKnowledgeDocument(@RequestParam("id") Long id) { - AiKnowledgeDocumentDO document = documentService.getKnowledgeDocument(id); - return success(BeanUtils.toBean(document, AiKnowledgeDocumentRespVO.class)); - } - - @PostMapping("/create") - @Operation(summary = "新建文档（单个）") - @PreAuthorize("@ss.hasPermission('ai:knowledge:create')") - public CommonResult createKnowledgeDocument(@RequestBody @Valid AiKnowledgeDocumentCreateReqVO reqVO) { - Long id = documentService.createKnowledgeDocument(reqVO); - return success(id); - } - - @PostMapping("/create-list") - @Operation(summary = "新建文档（多个）") - @PreAuthorize("@ss.hasPermission('ai:knowledge:create')") - public CommonResult> createKnowledgeDocumentList( - @RequestBody @Valid AiKnowledgeDocumentCreateListReqVO reqVO) { - List ids = documentService.createKnowledgeDocumentList(reqVO); - return success(ids); - } - - @PutMapping("/update") - @Operation(summary = "更新文档") - @PreAuthorize("@ss.hasPermission('ai:knowledge:update')") - public CommonResult updateKnowledgeDocument(@Valid @RequestBody AiKnowledgeDocumentUpdateReqVO reqVO) { - documentService.updateKnowledgeDocument(reqVO); - return success(true); - } - - @PutMapping("/update-status") - @Operation(summary = "更新文档状态") - @PreAuthorize("@ss.hasPermission('ai:knowledge:update')") - public CommonResult updateKnowledgeDocumentStatus( - @Valid @RequestBody AiKnowledgeDocumentUpdateStatusReqVO reqVO) { - documentService.updateKnowledgeDocumentStatus(reqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除文档") - @PreAuthorize("@ss.hasPermission('ai:knowledge:delete')") - public CommonResult deleteKnowledgeDocument(@RequestParam("id") Long id) { - documentService.deleteKnowledgeDocument(id); - return success(true); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeSegmentController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeSegmentController.http deleted file mode 100644 index a24302d..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeSegmentController.http +++ /dev/null @@ -1,17 +0,0 @@ -### 切片内容 -GET {{baseUrl}}/ai/knowledge/segment/split?url=https://static.iocoder.cn/README_lyzsys.md&segmentMaxTokens=800 -Content-Type: application/json -Authorization: Bearer {{token}} -tenant-id: {{adminTenantId}} - -### 搜索段落内容 -GET {{baseUrl}}/ai/knowledge/segment/search?knowledgeId=2&content=如何使用这个产品&topK=5&similarityThreshold=0.1 -Content-Type: application/json -Authorization: Bearer {{token}} -tenant-id: {{adminTenantId}} - -### 获取文档处理列表 -GET {{baseUrl}}/ai/knowledge/segment/get-process-list?documentIds=1,2,3 -Content-Type: application/json -Authorization: Bearer {{token}} -tenant-id: {{adminTenantId}} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeSegmentController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeSegmentController.java deleted file mode 100644 index 03b1577..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/AiKnowledgeSegmentController.java +++ /dev/null @@ -1,130 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge; - -import cn.hutool.core.collection.CollUtil; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.collection.MapUtils; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment.*; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDocumentDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeSegmentDO; -import cn.iocoder.lyzsys.module.ai.service.knowledge.AiKnowledgeDocumentService; -import cn.iocoder.lyzsys.module.ai.service.knowledge.AiKnowledgeSegmentService; -import cn.iocoder.lyzsys.module.ai.service.knowledge.bo.AiKnowledgeSegmentSearchReqBO; -import cn.iocoder.lyzsys.module.ai.service.knowledge.bo.AiKnowledgeSegmentSearchRespBO; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.Parameters; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.hibernate.validator.constraints.URL; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.Collections; -import java.util.List; -import java.util.Map; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertSet; - -@Tag(name = "管理后台 - AI 知识库段落") -@RestController -@RequestMapping("/ai/knowledge/segment") -@Validated -public class AiKnowledgeSegmentController { - - @Resource - private AiKnowledgeSegmentService segmentService; - @Resource - private AiKnowledgeDocumentService documentService; - - @GetMapping("/get") - @Operation(summary = "获取段落详情") - @Parameter(name = "id", description = "段落编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult getKnowledgeSegment(@RequestParam("id") Long id) { - AiKnowledgeSegmentDO segment = segmentService.getKnowledgeSegment(id); - return success(BeanUtils.toBean(segment, AiKnowledgeSegmentRespVO.class)); - } - - @GetMapping("/page") - @Operation(summary = "获取段落分页") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult> getKnowledgeSegmentPage( - @Valid AiKnowledgeSegmentPageReqVO pageReqVO) { - PageResult pageResult = segmentService.getKnowledgeSegmentPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiKnowledgeSegmentRespVO.class)); - } - - @PostMapping("/create") - @Operation(summary = "创建段落") - @PreAuthorize("@ss.hasPermission('ai:knowledge:create')") - public CommonResult createKnowledgeSegment(@Valid @RequestBody AiKnowledgeSegmentSaveReqVO createReqVO) { - return success(segmentService.createKnowledgeSegment(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新段落内容") - @PreAuthorize("@ss.hasPermission('ai:knowledge:update')") - public CommonResult updateKnowledgeSegment(@Valid @RequestBody AiKnowledgeSegmentSaveReqVO reqVO) { - segmentService.updateKnowledgeSegment(reqVO); - return success(true); - } - - @PutMapping("/update-status") - @Operation(summary = "启禁用段落内容") - @PreAuthorize("@ss.hasPermission('ai:knowledge:update')") - public CommonResult updateKnowledgeSegmentStatus( - @Valid @RequestBody AiKnowledgeSegmentUpdateStatusReqVO reqVO) { - segmentService.updateKnowledgeSegmentStatus(reqVO); - return success(true); - } - - @GetMapping("/split") - @Operation(summary = "切片内容") - @Parameters({ - @Parameter(name = "url", description = "文档 URL", required = true), - @Parameter(name = "segmentMaxTokens", description = "分段的最大 Token 数", required = true) - }) - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult> splitContent( - @RequestParam("url") @URL String url, - @RequestParam(value = "segmentMaxTokens") Integer segmentMaxTokens) { - List segments = segmentService.splitContent(url, segmentMaxTokens); - return success(BeanUtils.toBean(segments, AiKnowledgeSegmentRespVO.class)); - } - - @GetMapping("/get-process-list") - @Operation(summary = "获取文档处理列表") - @Parameter(name = "documentIds", description = "文档编号列表", required = true, example = "1,2,3") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult> getKnowledgeSegmentProcessList( - @RequestParam("documentIds") List documentIds) { - List list = segmentService.getKnowledgeSegmentProcessList(documentIds); - return success(list); - } - - @GetMapping("/search") - @Operation(summary = "搜索段落内容") - @PreAuthorize("@ss.hasPermission('ai:knowledge:query')") - public CommonResult> searchKnowledgeSegment( - @Valid AiKnowledgeSegmentSearchReqVO reqVO) { - // 1. 搜索段落 - List segments = segmentService - .searchKnowledgeSegment(BeanUtils.toBean(reqVO, AiKnowledgeSegmentSearchReqBO.class)); - if (CollUtil.isEmpty(segments)) { - return success(Collections.emptyList()); - } - - // 2. 拼接 VO - Map documentMap = documentService.getKnowledgeDocumentMap(convertSet( - segments, AiKnowledgeSegmentSearchRespBO::getDocumentId)); - return success(BeanUtils.toBean(segments, AiKnowledgeSegmentSearchRespVO.class, - segment -> MapUtils.findAndThen(documentMap, segment.getDocumentId(), - document -> segment.setDocumentName(document.getName())))); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentCreateListReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentCreateListReqVO.java deleted file mode 100644 index 0467fd9..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentCreateListReqVO.java +++ /dev/null @@ -1,42 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotBlank; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; -import org.hibernate.validator.constraints.URL; - -import java.util.List; - -@Schema(description = "管理后台 - AI 知识库文档批量创建 Request VO") -@Data -public class AiKnowledgeDocumentCreateListReqVO { - - @Schema(description = "知识库编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1204") - @NotNull(message = "知识库编号不能为空") - private Long knowledgeId; - - @Schema(description = "分段的最大 Token 数", requiredMode = Schema.RequiredMode.REQUIRED, example = "800") - @NotNull(message = "分段的最大 Token 数不能为空") - private Integer segmentMaxTokens; - - @Schema(description = "文档列表", requiredMode = Schema.RequiredMode.REQUIRED) - @NotEmpty(message = "文档列表不能为空") - private List list; - - @Schema(description = "文档") - @Data - public static class Document { - - @Schema(description = "文档名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "三方登陆") - @NotBlank(message = "文档名称不能为空") - private String name; - - @Schema(description = "文档 URL", requiredMode = Schema.RequiredMode.REQUIRED, example = "https://doc.iocoder.cn") - @URL(message = "文档 URL 格式不正确") - private String url; - - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentPageReqVO.java deleted file mode 100644 index be4402f..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentPageReqVO.java +++ /dev/null @@ -1,17 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库文档的分页 Request VO") -@Data -public class AiKnowledgeDocumentPageReqVO extends PageParam { - - @Schema(description = "知识库编号", example = "1") - private Long knowledgeId; - - @Schema(description = "文档名称", example = "Java 开发手册") - private String name; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentRespVO.java deleted file mode 100644 index b76497e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentRespVO.java +++ /dev/null @@ -1,45 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 知识库文档 Response VO") -@Data -public class AiKnowledgeDocumentRespVO { - - @Schema(description = "文档编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long id; - - @Schema(description = "知识库编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long knowledgeId; - - @Schema(description = "文档名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "Java 开发手册") - private String name; - - @Schema(description = "文档 URL", requiredMode = Schema.RequiredMode.REQUIRED, example = "https://doc.iocoder.cn") - private String url; - - @Schema(description = "文档内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "Java 是一门面向对象的语言.....") - private String content; - - @Schema(description = "文档内容长度", requiredMode = Schema.RequiredMode.REQUIRED, example = "2048") - private Integer contentLength; - - @Schema(description = "文档 Token 数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Integer tokens; - - @Schema(description = "分片最大 Token 数", requiredMode = Schema.RequiredMode.REQUIRED, example = "512") - private Integer segmentMaxTokens; - - @Schema(description = "召回次数", requiredMode = Schema.RequiredMode.REQUIRED, example = "10") - private Integer retrievalCount; - - @Schema(description = "文档状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "0") - private Integer status; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentUpdateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentUpdateReqVO.java deleted file mode 100644 index 719838e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentUpdateReqVO.java +++ /dev/null @@ -1,21 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库文档更新 Request VO") -@Data -public class AiKnowledgeDocumentUpdateReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "15583") - @NotNull(message = "编号不能为空") - private Long id; - - @Schema(description = "名称", example = "Java 开发手册") - private String name; - - @Schema(description = "分片最大 Token 数", example = "1000") - private Integer segmentMaxTokens; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentUpdateStatusReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentUpdateStatusReqVO.java deleted file mode 100644 index d8e4269..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/document/AiKnowledgeDocumentUpdateStatusReqVO.java +++ /dev/null @@ -1,22 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库文档更新状态 Request VO") -@Data -public class AiKnowledgeDocumentUpdateStatusReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "15583") - @NotNull(message = "编号不能为空") - private Long id; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "0") - @NotNull(message = "状态不能为空") - @InEnum(CommonStatusEnum.class) - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeDocumentCreateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeDocumentCreateReqVO.java deleted file mode 100644 index 44865fb..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeDocumentCreateReqVO.java +++ /dev/null @@ -1,30 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotBlank; -import jakarta.validation.constraints.NotNull; -import lombok.Data; -import org.hibernate.validator.constraints.URL; - - -@Schema(description = "管理后台 - AI 知识库文档的创建 Request VO") -@Data -public class AiKnowledgeDocumentCreateReqVO { - - @Schema(description = "知识库编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1204") - @NotNull(message = "知识库编号不能为空") - private Long knowledgeId; - - @Schema(description = "文档名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "三方登陆") - @NotBlank(message = "文档名称不能为空") - private String name; - - @Schema(description = "文档 URL", requiredMode = Schema.RequiredMode.REQUIRED, example = "https://doc.iocoder.cn") - @URL(message = "文档 URL 格式不正确") - private String url; - - @Schema(description = "分段的最大 Token 数", requiredMode = Schema.RequiredMode.REQUIRED, example = "800") - @NotNull(message = "分段的最大 Token 数不能为空") - private Integer segmentMaxTokens; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgePageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgePageReqVO.java deleted file mode 100644 index f581bbb..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgePageReqVO.java +++ /dev/null @@ -1,29 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 知识库的分页 Request VO") -@Data -public class AiKnowledgePageReqVO extends PageParam { - - @Schema(description = "知识库名称", example = "芋艿") - private String name; - - @Schema(description = "是否启用", example = "1") - @InEnum(CommonStatusEnum.class) - private Integer status; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeRespVO.java deleted file mode 100644 index ac0c7c8..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeRespVO.java +++ /dev/null @@ -1,39 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 知识库 Response VO") -@Data -public class AiKnowledgeRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long id; - - @Schema(description = "知识库名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "ruoyi-vue-pro 用户指南") - private String name; - - @Schema(description = "知识库描述", example = "帮助你快速构建系统") - private String description; - - @Schema(description = "向量模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "14") - private Long embeddingModelId; - - @Schema(description = "向量模型标识", requiredMode = Schema.RequiredMode.REQUIRED, example = "qwen-72b-chat") - private String embeddingModel; - - @Schema(description = "topK", requiredMode = Schema.RequiredMode.REQUIRED, example = "3") - private Integer topK; - - @Schema(description = "相似度阈值", requiredMode = Schema.RequiredMode.REQUIRED, example = "0.7") - private Double similarityThreshold; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer status; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeSaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeSaveReqVO.java deleted file mode 100644 index 758794a..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/knowledge/AiKnowledgeSaveReqVO.java +++ /dev/null @@ -1,41 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotBlank; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库新增/修改 Request VO") -@Data -public class AiKnowledgeSaveReqVO { - - @Schema(description = "对话编号", example = "1204") - private Long id; - - @Schema(description = "知识库名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "ruoyi-vue-pro 用户指南") - @NotBlank(message = "知识库名称不能为空") - private String name; - - @Schema(description = "知识库描述", requiredMode = Schema.RequiredMode.REQUIRED, example = "存储 ruoyi-vue-pro 操作文档") - private String description; - - @Schema(description = "向量模型编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "向量模型不能为空") - private Long embeddingModelId; - - @Schema(description = "topK", requiredMode = Schema.RequiredMode.REQUIRED, example = "3") - @NotNull(message = "topK 不能为空") - private Integer topK; - - @Schema(description = "相似性阈值", requiredMode = Schema.RequiredMode.REQUIRED, example = "0.5") - @NotNull(message = "相似性阈值不能为空") - private Double similarityThreshold; - - @Schema(description = "是否启用", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "是否启用不能为空") - @InEnum(CommonStatusEnum.class) - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentPageReqVO.java deleted file mode 100644 index 814facf..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentPageReqVO.java +++ /dev/null @@ -1,23 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库分段的分页 Request VO") -@Data -public class AiKnowledgeSegmentPageReqVO extends PageParam { - - @Schema(description = "文档编号", example = "1") - private Long documentId; - - @Schema(description = "分段内容关键字", example = "Java 开发") - private String content; - - @Schema(description = "分段状态", example = "1") - @InEnum(CommonStatusEnum.class) - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentProcessRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentProcessRespVO.java deleted file mode 100644 index fe7a00f..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentProcessRespVO.java +++ /dev/null @@ -1,19 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库段落向量进度 Response VO") -@Data -public class AiKnowledgeSegmentProcessRespVO { - - @Schema(description = "文档编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Long documentId; - - @Schema(description = "总段落数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "10") - private Long count; - - @Schema(description = "已向量化段落数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "5") - private Long embeddingCount; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentRespVO.java deleted file mode 100644 index 6c38b79..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentRespVO.java +++ /dev/null @@ -1,40 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库文档分片 Response VO") -@Data -public class AiKnowledgeSegmentRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long id; - - @Schema(description = "文档编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long documentId; - - @Schema(description = "知识库编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long knowledgeId; - - @Schema(description = "向量库编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1858496a-1dde-4edf-a43e-0aed08f37f8c") - private String vectorId; - - @Schema(description = "切片内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "Java 开发手册") - private String content; - - @Schema(description = "切片内容长度", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Integer contentLength; - - @Schema(description = "token 数量", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - private Integer tokens; - - @Schema(description = "召回次数", requiredMode = Schema.RequiredMode.REQUIRED, example = "10") - private Integer retrievalCount; - - @Schema(description = "文档状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer status; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private Long createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSaveReqVO.java deleted file mode 100644 index 5421b03..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSaveReqVO.java +++ /dev/null @@ -1,21 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import lombok.Data; - -@Schema(description = "管理后台 - AI 新增/修改知识库段落 request VO") -@Data -public class AiKnowledgeSegmentSaveReqVO { - - @Schema(description = "编号", example = "24790") - private Long id; - - @Schema(description = "知识库文档编号", example = "1024") - private Long documentId; - - @Schema(description = "切片内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "Java 开发手册") - @NotEmpty(message = "切片内容不能为空") - private String content; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSearchReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSearchReqVO.java deleted file mode 100644 index 4490edc..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSearchReqVO.java +++ /dev/null @@ -1,27 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; - -@Schema(description = "管理后台 - AI 知识库段落搜索 Request VO") -@Data -public class AiKnowledgeSegmentSearchReqVO { - - @Schema(description = "知识库编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1024") - @NotNull(message = "知识库编号不能为空") - private Long knowledgeId; - - @Schema(description = "内容", requiredMode = Schema.RequiredMode.REQUIRED, example = "如何使用这个产品") - @NotEmpty(message = "内容不能为空") - private String content; - - @Schema(description = "最大返回数量", example = "5") - private Integer topK; - - @Schema(description = "相似度阈值", example = "0.7") - private Double similarityThreshold; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSearchRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSearchRespVO.java deleted file mode 100644 index 23f4a6e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentSearchRespVO.java +++ /dev/null @@ -1,16 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -@Schema(description = "管理后台 - AI 知识库段落搜索 Response VO") -@Data -public class AiKnowledgeSegmentSearchRespVO extends AiKnowledgeSegmentRespVO { - - @Schema(description = "文档名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "产品使用手册") - private String documentName; - - @Schema(description = "相似度分数", requiredMode = Schema.RequiredMode.REQUIRED, example = "0.95") - private Double score; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentUpdateStatusReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentUpdateStatusReqVO.java deleted file mode 100644 index df1a5a3..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/knowledge/vo/segment/AiKnowledgeSegmentUpdateStatusReqVO.java +++ /dev/null @@ -1,22 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - - -@Schema(description = "管理后台 - AI 知识库段落的更新状态 Request VO") -@Data -public class AiKnowledgeSegmentUpdateStatusReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long id; - - @Schema(description = "是否启用", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "是否启用不能为空") - @InEnum(CommonStatusEnum.class) - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/AiMindMapController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/AiMindMapController.java deleted file mode 100644 index 6d2a801..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/AiMindMapController.java +++ /dev/null @@ -1,57 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.mindmap; - -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo.AiMindMapGenerateReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo.AiMindMapPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo.AiMindMapRespVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.mindmap.AiMindMapDO; -import cn.iocoder.lyzsys.module.ai.service.mindmap.AiMindMapService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.http.MediaType; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.web.bind.annotation.*; -import reactor.core.publisher.Flux; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - AI 思维导图") -@RestController -@RequestMapping("/ai/mind-map") -public class AiMindMapController { - - @Resource - private AiMindMapService mindMapService; - - @PostMapping(value = "/generate-stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE) - @Operation(summary = "导图生成（流式）", description = "流式返回，响应较快") - public Flux> generateMindMap(@RequestBody @Valid AiMindMapGenerateReqVO generateReqVO) { - return mindMapService.generateMindMap(generateReqVO, getLoginUserId()); - } - - // ================ 导图管理 ================ - - @DeleteMapping("/delete") - @Operation(summary = "删除思维导图") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:mind-map:delete')") - public CommonResult deleteMindMap(@RequestParam("id") Long id) { - mindMapService.deleteMindMap(id); - return success(true); - } - - @GetMapping("/page") - @Operation(summary = "获得思维导图分页") - @PreAuthorize("@ss.hasPermission('ai:mind-map:query')") - public CommonResult> getMindMapPage(@Valid AiMindMapPageReqVO pageReqVO) { - PageResult pageResult = mindMapService.getMindMapPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiMindMapRespVO.class)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapGenerateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapGenerateReqVO.java deleted file mode 100644 index c6b5ea1..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapGenerateReqVO.java +++ /dev/null @@ -1,15 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotBlank; -import lombok.Data; - -@Schema(description = "管理后台 - AI 思维导图生成 Request VO") -@Data -public class AiMindMapGenerateReqVO { - - @Schema(description = "思维导图内容提示", example = "Java 学习路线") - @NotBlank(message = "思维导图内容提示不能为空") - private String prompt; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapPageReqVO.java deleted file mode 100644 index beb0897..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapPageReqVO.java +++ /dev/null @@ -1,28 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import lombok.EqualsAndHashCode; -import lombok.ToString; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 思维导图分页 Request VO") -@Data -public class AiMindMapPageReqVO extends PageParam { - - @Schema(description = "用户编号", example = "4325") - private Long userId; - - @Schema(description = "生成内容提示", example = "Java 学习路线") - private String prompt; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapRespVO.java deleted file mode 100644 index 8896c65..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/mindmap/vo/AiMindMapRespVO.java +++ /dev/null @@ -1,36 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 思维导图 Response VO") -@Data -public class AiMindMapRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "3373") - private Long id; - - @Schema(description = "用户编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "4325") - private Long userId; - - @Schema(description = "生成内容提示", requiredMode = Schema.RequiredMode.REQUIRED, example = "Java 学习路线") - private String prompt; - - @Schema(description = "生成的思维导图内容") - private String generatedContent; - - @Schema(description = "平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "OpenAI") - private String platform; - - @Schema(description = "模型", requiredMode = Schema.RequiredMode.REQUIRED, example = "gpt-3.5-turbo-0125") - private String model; - - @Schema(description = "错误信息") - private String errorMessage; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiApiKeyController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiApiKeyController.java deleted file mode 100644 index 4e4bae0..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiApiKeyController.java +++ /dev/null @@ -1,83 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model; - -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey.AiApiKeyPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey.AiApiKeyRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey.AiApiKeySaveReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model.AiModelRespVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiApiKeyDO; -import cn.iocoder.lyzsys.module.ai.service.model.AiApiKeyService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertList; - -@Tag(name = "管理后台 - AI API 密钥") -@RestController -@RequestMapping("/ai/api-key") -@Validated -public class AiApiKeyController { - - @Resource - private AiApiKeyService apiKeyService; - - @PostMapping("/create") - @Operation(summary = "创建 API 密钥") - @PreAuthorize("@ss.hasPermission('ai:api-key:create')") - public CommonResult createApiKey(@Valid @RequestBody AiApiKeySaveReqVO createReqVO) { - return success(apiKeyService.createApiKey(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新 API 密钥") - @PreAuthorize("@ss.hasPermission('ai:api-key:update')") - public CommonResult updateApiKey(@Valid @RequestBody AiApiKeySaveReqVO updateReqVO) { - apiKeyService.updateApiKey(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除 API 密钥") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:api-key:delete')") - public CommonResult deleteApiKey(@RequestParam("id") Long id) { - apiKeyService.deleteApiKey(id); - return success(true); - } - - @GetMapping("/get") - @Operation(summary = "获得 API 密钥") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:api-key:query')") - public CommonResult getApiKey(@RequestParam("id") Long id) { - AiApiKeyDO apiKey = apiKeyService.getApiKey(id); - return success(BeanUtils.toBean(apiKey, AiApiKeyRespVO.class)); - } - - @GetMapping("/page") - @Operation(summary = "获得 API 密钥分页") - @PreAuthorize("@ss.hasPermission('ai:api-key:query')") - public CommonResult> getApiKeyPage(@Valid AiApiKeyPageReqVO pageReqVO) { - PageResult pageResult = apiKeyService.getApiKeyPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiApiKeyRespVO.class)); - } - - @GetMapping("/simple-list") - @Operation(summary = "获得 API 密钥分页列表") - public CommonResult> getApiKeySimpleList() { - List list = apiKeyService.getApiKeyList(); - return success(convertList(list, key -> new AiModelRespVO().setId(key.getId()).setName(key.getName()))); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiChatRoleController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiChatRoleController.java deleted file mode 100644 index 63ad2b5..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiChatRoleController.java +++ /dev/null @@ -1,128 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model; - -import cn.hutool.core.util.ObjUtil; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole.AiChatRolePageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole.AiChatRoleRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole.AiChatRoleSaveMyReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole.AiChatRoleSaveReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiChatRoleDO; -import cn.iocoder.lyzsys.module.ai.service.model.AiChatRoleService; -import com.fhs.core.trans.anno.TransMethodResult; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - AI 聊天角色") -@RestController -@RequestMapping("/ai/chat-role") -@Validated -public class AiChatRoleController { - - @Resource - private AiChatRoleService chatRoleService; - - @GetMapping("/my-page") - @Operation(summary = "获得【我的】聊天角色分页") - @TransMethodResult - public CommonResult> getChatRoleMyPage(@Valid AiChatRolePageReqVO pageReqVO) { - PageResult pageResult = chatRoleService.getChatRoleMyPage(pageReqVO, getLoginUserId()); - return success(BeanUtils.toBean(pageResult, AiChatRoleRespVO.class)); - } - - @GetMapping("/get-my") - @Operation(summary = "获得【我的】聊天角色") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @TransMethodResult - public CommonResult getChatRoleMy(@RequestParam("id") Long id) { - AiChatRoleDO chatRole = chatRoleService.getChatRole(id); - if (ObjUtil.notEqual(chatRole.getUserId(), getLoginUserId())) { - return success(null); - } - return success(BeanUtils.toBean(chatRole, AiChatRoleRespVO.class)); - } - - @PostMapping("/create-my") - @Operation(summary = "创建【我的】聊天角色") - public CommonResult createChatRoleMy(@Valid @RequestBody AiChatRoleSaveMyReqVO createReqVO) { - return success(chatRoleService.createChatRoleMy(createReqVO, getLoginUserId())); - } - - @PutMapping("/update-my") - @Operation(summary = "更新【我的】聊天角色") - public CommonResult updateChatRoleMy(@Valid @RequestBody AiChatRoleSaveMyReqVO updateReqVO) { - chatRoleService.updateChatRoleMy(updateReqVO, getLoginUserId()); - return success(true); - } - - @DeleteMapping("/delete-my") - @Operation(summary = "删除【我的】聊天角色") - @Parameter(name = "id", description = "编号", required = true) - public CommonResult deleteChatRoleMy(@RequestParam("id") Long id) { - chatRoleService.deleteChatRoleMy(id, getLoginUserId()); - return success(true); - } - - @GetMapping("/category-list") - @Operation(summary = "获得聊天角色的分类列表") - public CommonResult> getChatRoleCategoryList() { - return success(chatRoleService.getChatRoleCategoryList()); - } - - // ========== 角色管理 ========== - - @PostMapping("/create") - @Operation(summary = "创建聊天角色") - @PreAuthorize("@ss.hasPermission('ai:chat-role:create')") - public CommonResult createChatRole(@Valid @RequestBody AiChatRoleSaveReqVO createReqVO) { - return success(chatRoleService.createChatRole(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新聊天角色") - @PreAuthorize("@ss.hasPermission('ai:chat-role:update')") - public CommonResult updateChatRole(@Valid @RequestBody AiChatRoleSaveReqVO updateReqVO) { - chatRoleService.updateChatRole(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除聊天角色") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:chat-role:delete')") - public CommonResult deleteChatRole(@RequestParam("id") Long id) { - chatRoleService.deleteChatRole(id); - return success(true); - } - - @GetMapping("/get") - @Operation(summary = "获得聊天角色") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:chat-role:query')") - @TransMethodResult - public CommonResult getChatRole(@RequestParam("id") Long id) { - AiChatRoleDO chatRole = chatRoleService.getChatRole(id); - return success(BeanUtils.toBean(chatRole, AiChatRoleRespVO.class)); - } - - @GetMapping("/page") - @Operation(summary = "获得聊天角色分页") - @PreAuthorize("@ss.hasPermission('ai:chat-role:query')") - public CommonResult> getChatRolePage(@Valid AiChatRolePageReqVO pageReqVO) { - PageResult pageResult = chatRoleService.getChatRolePage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiChatRoleRespVO.class)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiModelController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiModelController.java deleted file mode 100644 index 354b5c1..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiModelController.java +++ /dev/null @@ -1,89 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model.AiModelPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model.AiModelRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model.AiModelSaveReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import cn.iocoder.lyzsys.module.ai.service.model.AiModelService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertList; - -@Tag(name = "管理后台 - AI 模型") -@RestController -@RequestMapping("/ai/model") -@Validated -public class AiModelController { - - @Resource - private AiModelService modelService; - - @PostMapping("/create") - @Operation(summary = "创建模型") - @PreAuthorize("@ss.hasPermission('ai:model:create')") - public CommonResult createModel(@Valid @RequestBody AiModelSaveReqVO createReqVO) { - return success(modelService.createModel(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新模型") - @PreAuthorize("@ss.hasPermission('ai:model:update')") - public CommonResult updateModel(@Valid @RequestBody AiModelSaveReqVO updateReqVO) { - modelService.updateModel(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除模型") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:model:delete')") - public CommonResult deleteModel(@RequestParam("id") Long id) { - modelService.deleteModel(id); - return success(true); - } - - @GetMapping("/get") - @Operation(summary = "获得模型") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:model:query')") - public CommonResult getModel(@RequestParam("id") Long id) { - AiModelDO model = modelService.getModel(id); - return success(BeanUtils.toBean(model, AiModelRespVO.class)); - } - - @GetMapping("/page") - @Operation(summary = "获得模型分页") - @PreAuthorize("@ss.hasPermission('ai:model:query')") - public CommonResult> getModelPage(@Valid AiModelPageReqVO pageReqVO) { - PageResult pageResult = modelService.getModelPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiModelRespVO.class)); - } - - @GetMapping("/simple-list") - @Operation(summary = "获得模型列表") - @Parameter(name = "type", description = "类型", required = true, example = "1") - @Parameter(name = "platform", description = "平台", example = "midjourney") - public CommonResult> getModelSimpleList( - @RequestParam("type") Integer type, - @RequestParam(value = "platform", required = false) String platform) { - List list = modelService.getModelListByStatusAndType( - CommonStatusEnum.ENABLE.getStatus(), type, platform); - return success(convertList(list, model -> new AiModelRespVO().setId(model.getId()) - .setName(model.getName()).setModel(model.getModel()).setPlatform(model.getPlatform()))); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiToolController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiToolController.java deleted file mode 100644 index 4e6afa2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/AiToolController.java +++ /dev/null @@ -1,84 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool.AiToolPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool.AiToolRespVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool.AiToolSaveReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiToolDO; -import cn.iocoder.lyzsys.module.ai.service.model.AiToolService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertList; - -@Tag(name = "管理后台 - AI 工具") -@RestController -@RequestMapping("/ai/tool") -@Validated -public class AiToolController { - - @Resource - private AiToolService toolService; - - @PostMapping("/create") - @Operation(summary = "创建工具") - @PreAuthorize("@ss.hasPermission('ai:tool:create')") - public CommonResult createTool(@Valid @RequestBody AiToolSaveReqVO createReqVO) { - return success(toolService.createTool(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新工具") - @PreAuthorize("@ss.hasPermission('ai:tool:update')") - public CommonResult updateTool(@Valid @RequestBody AiToolSaveReqVO updateReqVO) { - toolService.updateTool(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除工具") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:tool:delete')") - public CommonResult deleteTool(@RequestParam("id") Long id) { - toolService.deleteTool(id); - return success(true); - } - - @GetMapping("/get") - @Operation(summary = "获得工具") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:tool:query')") - public CommonResult getTool(@RequestParam("id") Long id) { - AiToolDO tool = toolService.getTool(id); - return success(BeanUtils.toBean(tool, AiToolRespVO.class)); - } - - @GetMapping("/page") - @Operation(summary = "获得工具分页") - @PreAuthorize("@ss.hasPermission('ai:tool:query')") - public CommonResult> getToolPage(@Valid AiToolPageReqVO pageReqVO) { - PageResult pageResult = toolService.getToolPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiToolRespVO.class)); - } - - @GetMapping("/simple-list") - @Operation(summary = "获得工具列表") - public CommonResult> getToolSimpleList() { - List list = toolService.getToolListByStatus(CommonStatusEnum.ENABLE.getStatus()); - return success(convertList(list, tool -> new AiToolRespVO() - .setId(tool.getId()).setName(tool.getName()))); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeyPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeyPageReqVO.java deleted file mode 100644 index e441cc4..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeyPageReqVO.java +++ /dev/null @@ -1,25 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey; - -import lombok.*; -import java.util.*; -import io.swagger.v3.oas.annotations.media.Schema; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import org.springframework.format.annotation.DateTimeFormat; -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI API 密钥分页 Request VO") -@Data -public class AiApiKeyPageReqVO extends PageParam { - - @Schema(description = "名称", example = "文心一言") - private String name; - - @Schema(description = "平台", example = "OpenAI") - private String platform; - - @Schema(description = "状态", example = "1") - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeyRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeyRespVO.java deleted file mode 100644 index 0b82299..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeyRespVO.java +++ /dev/null @@ -1,28 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.*; - -@Schema(description = "管理后台 - AI API 密钥 Response VO") -@Data -public class AiApiKeyRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "23538") - private Long id; - - @Schema(description = "名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "文心一言") - private String name; - - @Schema(description = "密钥", requiredMode = Schema.RequiredMode.REQUIRED, example = "ABC") - private String apiKey; - - @Schema(description = "平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "OpenAI") - private String platform; - - @Schema(description = "自定义 API 地址", example = "https://aip.baidubce.com") - private String url; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeySaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeySaveReqVO.java deleted file mode 100644 index 28f5a02..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/apikey/AiApiKeySaveReqVO.java +++ /dev/null @@ -1,34 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.*; -import java.util.*; -import jakarta.validation.constraints.*; - -@Schema(description = "管理后台 - AI API 密钥新增/修改 Request VO") -@Data -public class AiApiKeySaveReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "23538") - private Long id; - - @Schema(description = "名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "文心一言") - @NotEmpty(message = "名称不能为空") - private String name; - - @Schema(description = "密钥", requiredMode = Schema.RequiredMode.REQUIRED, example = "ABC") - @NotEmpty(message = "密钥不能为空") - private String apiKey; - - @Schema(description = "平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "OpenAI") - @NotEmpty(message = "平台不能为空") - private String platform; - - @Schema(description = "自定义 API 地址", example = "https://aip.baidubce.com") - private String url; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "状态不能为空") - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRolePageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRolePageReqVO.java deleted file mode 100644 index b0648da..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRolePageReqVO.java +++ /dev/null @@ -1,20 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole; - -import lombok.*; -import io.swagger.v3.oas.annotations.media.Schema; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; - -@Schema(description = "管理后台 - AI 聊天角色分页 Request VO") -@Data -public class AiChatRolePageReqVO extends PageParam { - - @Schema(description = "角色名称", example = "李四") - private String name; - - @Schema(description = "角色类别", example = "创作") - private String category; - - @Schema(description = "是否公开", example = "1") - private Boolean publicStatus; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleRespVO.java deleted file mode 100644 index 94671a8..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleRespVO.java +++ /dev/null @@ -1,67 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole; - -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import com.fhs.core.trans.anno.Trans; -import com.fhs.core.trans.constant.TransType; -import com.fhs.core.trans.vo.VO; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; -import java.util.List; - -@Schema(description = "管理后台 - AI 聊天角色 Response VO") -@Data -public class AiChatRoleRespVO implements VO { - - @Schema(description = "角色编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "32746") - private Long id; - - @Schema(description = "用户编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "9442") - private Long userId; - - @Schema(description = "模型编号", example = "17640") - @Trans(type = TransType.SIMPLE, target = AiModelDO.class, fields = { "name", "model" }, refs = { "modelName", "model" }) - private Long modelId; - @Schema(description = "模型名字", example = "张三") - private String modelName; - @Schema(description = "模型标识", example = "gpt-3.5-turbo-0125") - private String model; - - @Schema(description = "角色名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "李四") - private String name; - - @Schema(description = "角色头像", requiredMode = Schema.RequiredMode.REQUIRED, example = "https://www.iocoder.cn/1.png") - private String avatar; - - @Schema(description = "角色类别", requiredMode = Schema.RequiredMode.REQUIRED, example = "创作") - private String category; - - @Schema(description = "角色排序", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer sort; - - @Schema(description = "角色描述", requiredMode = Schema.RequiredMode.REQUIRED, example = "你说的对") - private String description; - - @Schema(description = "角色设定", requiredMode = Schema.RequiredMode.REQUIRED) - private String systemMessage; - - @Schema(description = "引用的知识库编号列表", example = "1,2,3") - private List knowledgeIds; - - @Schema(description = "引用的工具编号列表", example = "1,2,3") - private List toolIds; - - @Schema(description = "引用的 MCP Client 名字列表", example = "filesystem") - private List mcpClientNames; - - @Schema(description = "是否公开", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Boolean publicStatus; - - @Schema(description = "状态", example = "1") - private Integer status; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleSaveMyReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleSaveMyReqVO.java deleted file mode 100644 index 6a3c284..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleSaveMyReqVO.java +++ /dev/null @@ -1,43 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import lombok.Data; -import org.hibernate.validator.constraints.URL; - -import java.util.List; - -@Schema(description = "管理后台 - AI 聊天角色新增/修改【我的】 Request VO") -@Data -public class AiChatRoleSaveMyReqVO { - - @Schema(description = "角色编号", example = "32746") - private Long id; - - @Schema(description = "角色名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "李四") - @NotEmpty(message = "角色名称不能为空") - private String name; - - @Schema(description = "角色头像", requiredMode = Schema.RequiredMode.REQUIRED, example = "https://www.iocoder.cn/1.png") - @NotEmpty(message = "角色头像不能为空") - @URL(message = "角色头像必须是 URL 格式") - private String avatar; - - @Schema(description = "角色描述", requiredMode = Schema.RequiredMode.REQUIRED, example = "你说的对") - @NotEmpty(message = "角色描述不能为空") - private String description; - - @Schema(description = "角色设定", requiredMode = Schema.RequiredMode.REQUIRED, example = "现在开始你扮演一位程序员，你是一名优秀的程序员，具有很强的逻辑思维能力，总能高效的解决问题") - @NotEmpty(message = "角色设定不能为空") - private String systemMessage; - - @Schema(description = "引用的知识库编号列表", example = "1,2,3") - private List knowledgeIds; - - @Schema(description = "引用的工具编号列表", example = "1,2,3") - private List toolIds; - - @Schema(description = "引用的 MCP Client 名字列表", example = "filesystem") - private List mcpClientNames; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleSaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleSaveReqVO.java deleted file mode 100644 index 990edcc..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/chatRole/AiChatRoleSaveReqVO.java +++ /dev/null @@ -1,65 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.*; -import jakarta.validation.constraints.*; -import org.hibernate.validator.constraints.URL; - -import java.util.List; - -@Schema(description = "管理后台 - AI 聊天角色新增/修改 Request VO") -@Data -public class AiChatRoleSaveReqVO { - - @Schema(description = "角色编号", example = "32746") - private Long id; - - @Schema(description = "模型编号", example = "17640") - private Long modelId; - - @Schema(description = "角色名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "李四") - @NotEmpty(message = "角色名称不能为空") - private String name; - - @Schema(description = "角色头像", requiredMode = Schema.RequiredMode.REQUIRED, example = "https://www.iocoder.cn/1.png") - @NotEmpty(message = "角色头像不能为空") - @URL(message = "角色头像必须是 URL 格式") - private String avatar; - - @Schema(description = "角色类别", requiredMode = Schema.RequiredMode.REQUIRED, example = "创作") - @NotEmpty(message = "角色类别不能为空") - private String category; - - @Schema(description = "角色排序", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "角色排序不能为空") - private Integer sort; - - @Schema(description = "角色描述", requiredMode = Schema.RequiredMode.REQUIRED, example = "你说的对") - @NotEmpty(message = "角色描述不能为空") - private String description; - - @Schema(description = "角色设定", requiredMode = Schema.RequiredMode.REQUIRED, example = "现在开始你扮演一位程序员，你是一名优秀的程序员，具有很强的逻辑思维能力，总能高效的解决问题") - @NotEmpty(message = "角色设定不能为空") - private String systemMessage; - - @Schema(description = "引用的知识库编号列表", example = "1,2,3") - private List knowledgeIds; - - @Schema(description = "引用的工具编号列表", example = "1,2,3") - private List toolIds; - - @Schema(description = "引用的 MCP Client 名字列表", example = "filesystem") - private List mcpClientNames; - - @Schema(description = "是否公开", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "是否公开不能为空") - private Boolean publicStatus; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "状态不能为空") - @InEnum(CommonStatusEnum.class) - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelPageReqVO.java deleted file mode 100644 index a5ac9a0..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelPageReqVO.java +++ /dev/null @@ -1,20 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model; - -import lombok.*; -import io.swagger.v3.oas.annotations.media.Schema; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; - -@Schema(description = "管理后台 - API 模型分页 Request VO") -@Data -public class AiModelPageReqVO extends PageParam { - - @Schema(description = "模型名字", example = "张三") - private String name; - - @Schema(description = "模型标识", example = "gpt-3.5-turbo-0125") - private String model; - - @Schema(description = "模型平台", example = "OpenAI") - private String platform; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelRespVO.java deleted file mode 100644 index c542109..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelRespVO.java +++ /dev/null @@ -1,48 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 模型 Response VO") -@Data -public class AiModelRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "2630") - private Long id; - - @Schema(description = "API 秘钥编号", example = "22042") - private Long keyId; - - @Schema(description = "模型名字", example = "张三") - private String name; - - @Schema(description = "模型标识", example = "gpt-3.5-turbo-0125") - private String model; - - @Schema(description = "模型平台", example = "OpenAI") - private String platform; - - @Schema(description = "模型类型", example = "1") - private Integer type; - - @Schema(description = "排序", example = "1") - private Integer sort; - - @Schema(description = "状态", example = "2") - private Integer status; - - @Schema(description = "温度参数", example = "1") - private Double temperature; - - @Schema(description = "单条回复的最大 Token 数量", example = "4096") - private Integer maxTokens; - - @Schema(description = "上下文的最大 Message 数量", example = "8192") - private Integer maxContexts; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelSaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelSaveReqVO.java deleted file mode 100644 index 8a8d2e6..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/model/AiModelSaveReqVO.java +++ /dev/null @@ -1,59 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiModelTypeEnum; -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - API 模型新增/修改 Request VO") -@Data -public class AiModelSaveReqVO { - - @Schema(description = "编号", example = "2630") - private Long id; - - @Schema(description = "API 秘钥编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "22042") - @NotNull(message = "API 秘钥编号不能为空") - private Long keyId; - - @Schema(description = "模型名字", requiredMode = Schema.RequiredMode.REQUIRED, example = "张三") - @NotEmpty(message = "模型名字不能为空") - private String name; - - @Schema(description = "模型标识", requiredMode = Schema.RequiredMode.REQUIRED, example = "gpt-3.5-turbo-0125") - @NotEmpty(message = "模型标识不能为空") - private String model; - - @Schema(description = "模型平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "OpenAI") - @NotEmpty(message = "模型平台不能为空") - @InEnum(AiPlatformEnum.class) - private String platform; - - @Schema(description = "模型类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "模型类型不能为空") - @InEnum(AiModelTypeEnum.class) - private Integer type; - - @Schema(description = "排序", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "排序不能为空") - private Integer sort; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @InEnum(CommonStatusEnum.class) - @NotNull(message = "状态不能为空") - private Integer status; - - @Schema(description = "温度参数", example = "1") - private Double temperature; - - @Schema(description = "单条回复的最大 Token 数量", example = "4096") - private Integer maxTokens; - - @Schema(description = "上下文的最大 Message 数量", example = "8192") - private Integer maxContexts; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolPageReqVO.java deleted file mode 100644 index 9411469..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolPageReqVO.java +++ /dev/null @@ -1,34 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import lombok.EqualsAndHashCode; -import lombok.ToString; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 工具分页 Request VO") -@Data -public class AiToolPageReqVO extends PageParam { - - @Schema(description = "工具名称", example = "王五") - private String name; - - @Schema(description = "工具描述", example = "你猜") - private String description; - - @Schema(description = "状态", example = "1") - @InEnum(CommonStatusEnum.class) - private Integer status; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolRespVO.java deleted file mode 100644 index d0bb075..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolRespVO.java +++ /dev/null @@ -1,27 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.*; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 工具 Response VO") -@Data -public class AiToolRespVO { - - @Schema(description = "工具编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "19661") - private Long id; - - @Schema(description = "工具名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "王五") - private String name; - - @Schema(description = "工具描述", example = "你猜") - private String description; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer status; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolSaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolSaveReqVO.java deleted file mode 100644 index 7db8dbf..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/model/vo/tool/AiToolSaveReqVO.java +++ /dev/null @@ -1,27 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import lombok.Data; - -@Schema(description = "管理后台 - AI 工具新增/修改 Request VO") -@Data -public class AiToolSaveReqVO { - - @Schema(description = "工具编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "19661") - private Long id; - - @Schema(description = "工具名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "王五") - @NotEmpty(message = "工具名称不能为空") - private String name; - - @Schema(description = "工具描述", example = "你猜") - private String description; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @InEnum(CommonStatusEnum.class) - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/AiMusicController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/AiMusicController.http deleted file mode 100644 index ae68c82..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/AiMusicController.http +++ /dev/null @@ -1,26 +0,0 @@ -### 生成音乐：Suno + 歌词模式 -POST {{baseUrl}}/ai/music/generate -Content-Type: application/json -Authorization: {{token}} - -{ - "platform": "Suno", - "generateMode": 2, - "prompt": "创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。", - "model": "chirp-v3.5", - "tags": ["Happy"], - "title": "Happy Song" -} - -### 生成音乐：Suno + 描述模式 -POST {{baseUrl}}/ai/music/generate -Content-Type: application/json -Authorization: {{token}} - -{ - "platform": "Suno", - "generateMode": 1, - "model": "chirp-v3.5", - "prompt": "happy music", - "makeInstrumental": false -} \ No newline at end of file diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/AiMusicController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/AiMusicController.java deleted file mode 100644 index 2041535..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/AiMusicController.java +++ /dev/null @@ -1,98 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.music; - -import cn.hutool.core.util.ObjUtil; -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.music.vo.*; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.music.AiMusicDO; -import cn.iocoder.lyzsys.module.ai.service.music.AiMusicService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.web.bind.annotation.*; - -import java.util.List; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - AI 音乐") -@RestController -@RequestMapping("/ai/music") -public class AiMusicController { - - @Resource - private AiMusicService musicService; - - @GetMapping("/my-page") - @Operation(summary = "获得【我的】音乐分页") - public CommonResult> getMusicMyPage(@Valid AiMusicPageReqVO pageReqVO) { - PageResult pageResult = musicService.getMusicMyPage(pageReqVO, getLoginUserId()); - return success(BeanUtils.toBean(pageResult, AiMusicRespVO.class)); - } - - @PostMapping("/generate") - @Operation(summary = "音乐生成") - public CommonResult> generateMusic(@RequestBody @Valid AiSunoGenerateReqVO reqVO) { - return success(musicService.generateMusic(getLoginUserId(), reqVO)); - } - - @Operation(summary = "删除【我的】音乐记录") - @DeleteMapping("/delete-my") - @Parameter(name = "id", required = true, description = "音乐编号", example = "1024") - public CommonResult deleteMusicMy(@RequestParam("id") Long id) { - musicService.deleteMusicMy(id, getLoginUserId()); - return success(true); - } - - @GetMapping("/get-my") - @Operation(summary = "获取【我的】音乐") - @Parameter(name = "id", required = true, description = "音乐编号", example = "1024") - public CommonResult getMusicMy(@RequestParam("id") Long id) { - AiMusicDO music = musicService.getMusic(id); - if (music == null || ObjUtil.notEqual(getLoginUserId(), music.getUserId())) { - return success(null); - } - return success(BeanUtils.toBean(music, AiMusicRespVO.class)); - } - - @PostMapping("/update-my") - @Operation(summary = "修改【我的】音乐 目前只支持修改标题") - @Parameter(name = "title", required = true, description = "音乐名称", example = "夜空中最亮的星") - public CommonResult updateMy(AiMusicUpdateMyReqVO updateReqVO) { - musicService.updateMyMusic(updateReqVO, getLoginUserId()); - return success(true); - } - - // ================ 音乐管理 ================ - - @GetMapping("/page") - @Operation(summary = "获得音乐分页") - @PreAuthorize("@ss.hasPermission('ai:music:query')") - public CommonResult> getMusicPage(@Valid AiMusicPageReqVO pageReqVO) { - PageResult pageResult = musicService.getMusicPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiMusicRespVO.class)); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除音乐") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:music:delete')") - public CommonResult deleteMusic(@RequestParam("id") Long id) { - musicService.deleteMusic(id); - return success(true); - } - - @PutMapping("/update") - @Operation(summary = "更新音乐") - @PreAuthorize("@ss.hasPermission('ai:music:update')") - public CommonResult updateMusic(@Valid @RequestBody AiMusicUpdateReqVO updateReqVO) { - musicService.updateMusic(updateReqVO); - return success(true); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicPageReqVO.java deleted file mode 100644 index ef873a0..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicPageReqVO.java +++ /dev/null @@ -1,40 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.music.vo; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import cn.iocoder.lyzsys.module.ai.enums.music.AiMusicGenerateModeEnum; -import cn.iocoder.lyzsys.module.ai.enums.music.AiMusicStatusEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 音乐分页 Request VO") -@Data -public class AiMusicPageReqVO extends PageParam { - - @Schema(description = "用户编号", example = "12212") - private Long userId; - - @Schema(description = "音乐名称", example = "夜空中最亮的星") - private String title; - - @Schema(description = "音乐状态", example = "20") - @InEnum(AiMusicStatusEnum.class) - private Integer status; - - @Schema(description = "生成模式", example = "1") - @InEnum(AiMusicGenerateModeEnum.class) - private Integer generateMode; - - @Schema(description = "是否发布", example = "true") - private Boolean publicStatus; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicRespVO.java deleted file mode 100644 index 2bed2e9..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicRespVO.java +++ /dev/null @@ -1,70 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.music.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; -import java.util.List; - -@Schema(description = "管理后台 - AI 音乐 Response VO") -@Data -public class AiMusicRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "24790") - private Long id; - - @Schema(description = "用户编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "12212") - private Long userId; - - @Schema(description = "音乐名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "夜空中最亮的星") - private String title; - - @Schema(description = "歌词", example = "oh~卖糕的") - private String lyric; - - @Schema(description = "图片地址", example = "https://www.iocoder.cn") - private String imageUrl; - - @Schema(description = "音频地址", example = "https://www.iocoder.cn") - private String audioUrl; - - @Schema(description = "视频地址", example = "https://www.iocoder.cn") - private String videoUrl; - - @Schema(description = "音乐状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "20") - private Integer status; - - @Schema(description = "描述词", example = "一首轻快的歌曲") - private String gptDescriptionPrompt; - - @Schema(description = "提示词", example = "创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。") - private String prompt; - - @Schema(description = "模型平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "Suno") - private String platform; - - @Schema(description = "模型", requiredMode = Schema.RequiredMode.REQUIRED, example = "chirp-v3.5") - private String model; - - @Schema(description = "生成模式", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer generateMode; - - @Schema(description = "音乐风格标签") - private List tags; - - @Schema(description = "音乐时长", example = "[\"pop\",\"jazz\",\"punk\"]") - private Double duration; - - @Schema(description = "是否发布", requiredMode = Schema.RequiredMode.REQUIRED, example = "true") - private Boolean publicStatus; - - @Schema(description = "任务编号", example = "11369") - private String taskId; - - @Schema(description = "错误信息") - private String errorMessage; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicUpdateMyReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicUpdateMyReqVO.java deleted file mode 100644 index 1ced23a..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicUpdateMyReqVO.java +++ /dev/null @@ -1,18 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.music.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 修改我的音乐 Request VO") -@Data -public class AiMusicUpdateMyReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "15583") - @NotNull(message = "编号不能为空") - private Long id; - - @Schema(description = "音乐名称", example = "夜空中最亮的星") - private String title; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicUpdateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicUpdateReqVO.java deleted file mode 100644 index ce5d59b..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiMusicUpdateReqVO.java +++ /dev/null @@ -1,18 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.music.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 音乐修改 Request VO") -@Data -public class AiMusicUpdateReqVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "15583") - @NotNull(message = "编号不能为空") - private Long id; - - @Schema(description = "是否发布", example = "true") - private Boolean publicStatus; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiSunoGenerateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiSunoGenerateReqVO.java deleted file mode 100644 index f0686ca..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/music/vo/AiSunoGenerateReqVO.java +++ /dev/null @@ -1,57 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.music.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotBlank; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -import java.util.List; - -@Schema(description = "管理后台 - AI 音乐生成 Request VO") -@Data -public class AiSunoGenerateReqVO { - - @Schema(description = "平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "Suno") - @NotBlank(message = "平台不能为空") - private String platform; // 参见 AiPlatformEnum 枚举 - - /** - * 1. 描述模式：描述词 + 是否纯音乐 + 模型 - * 2. 歌词模式：歌词 + 音乐风格 + 标题 + 模型 - */ - @Schema(description = "生成模式", requiredMode = Schema.RequiredMode.REQUIRED, example = "2") - @NotNull(message = "生成模式不能为空") - private Integer generateMode; // 参见 AiMusicGenerateModeEnum 枚举 - - @Schema(description = "用于生成音乐音频的歌词提示", - example = """ - 1.描述模式：创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。 - 2.歌词模式： - [Verse] - 阳光下奔跑 多么欢快 - 假期就要来 心都飞起来 - 朋友在一旁 笑声又灿烂 - 无忧无虑的 每一天甜蜜 - [Chorus] - 马上放假了 快来庆祝 - 一起去旅行 快去冒险 - 日子太短暂 别再等待 - 马上放假了 梦想起飞 - """) - private String prompt; - - @Schema(description = "是否纯音乐", example = "true") - private Boolean makeInstrumental; - - @Schema(description = "模型", requiredMode = Schema.RequiredMode.REQUIRED, example = "chirp-v3.5") - @NotEmpty(message = "模型不能为空") - private String model; - - @Schema(description = "音乐风格", example = "[\"pop\",\"jazz\",\"punk\"]") - private List tags; - - @Schema(description = "音乐/歌曲名称", example = "夜空中最亮的星") - private String title; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/AiWorkflowController.http b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/AiWorkflowController.http deleted file mode 100644 index 8dc1b0c..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/AiWorkflowController.http +++ /dev/null @@ -1,12 +0,0 @@ -### 测试 AI 工作流 -POST {{baseUrl}}/ai/workflow/test -Content-Type: application/json -Authorization: {{token}} -tenant-id: {{adminTenantId}} - -{ - "id": 4, - "params": { - "message": "1 + 1 = ?" - } -} \ No newline at end of file diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/AiWorkflowController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/AiWorkflowController.java deleted file mode 100644 index 8f0a531..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/AiWorkflowController.java +++ /dev/null @@ -1,77 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.workflow; - -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.workflow.vo.*; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.workflow.AiWorkflowDO; -import cn.iocoder.lyzsys.module.ai.service.workflow.AiWorkflowService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import lombok.extern.slf4j.Slf4j; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.web.bind.annotation.*; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; - -@Tag(name = "管理后台 - AI 工作流") -@RestController -@RequestMapping("/ai/workflow") -@Slf4j -public class AiWorkflowController { - - @Resource - private AiWorkflowService workflowService; - - @PostMapping("/create") - @Operation(summary = "创建 AI 工作流") - @PreAuthorize("@ss.hasPermission('ai:workflow:create')") - public CommonResult createWorkflow(@Valid @RequestBody AiWorkflowSaveReqVO createReqVO) { - return success(workflowService.createWorkflow(createReqVO)); - } - - @PutMapping("/update") - @Operation(summary = "更新 AI 工作流") - @PreAuthorize("@ss.hasPermission('ai:workflow:update')") - public CommonResult updateWorkflow(@Valid @RequestBody AiWorkflowSaveReqVO updateReqVO) { - workflowService.updateWorkflow(updateReqVO); - return success(true); - } - - @DeleteMapping("/delete") - @Operation(summary = "删除 AI 工作流") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:workflow:delete')") - public CommonResult deleteWorkflow(@RequestParam("id") Long id) { - workflowService.deleteWorkflow(id); - return success(true); - } - - @GetMapping("/get") - @Operation(summary = "获得 AI 工作流") - @Parameter(name = "id", description = "编号", required = true, example = "1024") - @PreAuthorize("@ss.hasPermission('ai:workflow:query')") - public CommonResult getWorkflow(@RequestParam("id") Long id) { - AiWorkflowDO workflow = workflowService.getWorkflow(id); - return success(BeanUtils.toBean(workflow, AiWorkflowRespVO.class)); - } - - @GetMapping("/page") - @Operation(summary = "获得 AI 工作流分页") - @PreAuthorize("@ss.hasPermission('ai:workflow:query')") - public CommonResult> getWorkflowPage(@Valid AiWorkflowPageReqVO pageReqVO) { - PageResult pageResult = workflowService.getWorkflowPage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiWorkflowRespVO.class)); - } - - @PostMapping("/test") - @Operation(summary = "测试 AI 工作流") - @PreAuthorize("@ss.hasPermission('ai:workflow:test')") - public CommonResult testWorkflow(@Valid @RequestBody AiWorkflowTestReqVO testReqVO) { - return success(workflowService.testWorkflow(testReqVO)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowPageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowPageReqVO.java deleted file mode 100644 index 158eee6..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowPageReqVO.java +++ /dev/null @@ -1,32 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.workflow.vo; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 工作流分页 Request VO") -@Data -public class AiWorkflowPageReqVO extends PageParam { - - @Schema(description = "名称", example = "工作流") - private String name; - - @Schema(description = "标识", example = "FLOW") - private String code; - - @Schema(description = "状态", example = "1") - @InEnum(CommonStatusEnum.class) - private Integer status; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowRespVO.java deleted file mode 100644 index 90b4016..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowRespVO.java +++ /dev/null @@ -1,33 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.workflow.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 工作流 Response VO") -@Data -public class AiWorkflowRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Long id; - - @Schema(description = "工作流标识", requiredMode = Schema.RequiredMode.REQUIRED, example = "FLOW") - private String code; - - @Schema(description = "工作流名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "工作流") - private String name; - - @Schema(description = "备注", requiredMode = Schema.RequiredMode.REQUIRED, example = "工作流") - private String remark; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - private Integer status; - - @Schema(description = "工作流模型 JSON", requiredMode = Schema.RequiredMode.REQUIRED, example = "{}") - private String graph; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED, example = "时间戳格式") - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowSaveReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowSaveReqVO.java deleted file mode 100644 index be77725..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowSaveReqVO.java +++ /dev/null @@ -1,34 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.workflow.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotEmpty; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 工作流新增/修改 Request VO") -@Data -public class AiWorkflowSaveReqVO { - - @Schema(description = "编号", example = "1") - private Long id; - - @Schema(description = "工作流标识", requiredMode = Schema.RequiredMode.REQUIRED, example = "FLOW") - @NotEmpty(message = "工作流标识不能为空") - private String code; - - @Schema(description = "工作流名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "工作流") - @NotEmpty(message = "工作流名称不能为空") - private String name; - - @Schema(description = "备注", example = "FLOW") - private String remark; - - @Schema(description = "工作流模型", requiredMode = Schema.RequiredMode.REQUIRED, example = "{}") - @NotEmpty(message = "工作流模型不能为空") - private String graph; - - @Schema(description = "状态", requiredMode = Schema.RequiredMode.REQUIRED, example = "FLOW") - @NotNull(message = "状态不能为空") - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowTestReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowTestReqVO.java deleted file mode 100644 index da483f1..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/workflow/vo/AiWorkflowTestReqVO.java +++ /dev/null @@ -1,28 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.workflow.vo; - -import cn.hutool.core.util.StrUtil; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.AssertTrue; -import lombok.Data; - -import java.util.Map; - -@Schema(description = "管理后台 - AI 工作流测试 Request VO") -@Data -public class AiWorkflowTestReqVO { - - @Schema(description = "工作流编号", example = "1024") - private Long id; - - @Schema(description = "工作流模型", example = "{}") - private String graph; - - @Schema(description = "参数", requiredMode = Schema.RequiredMode.REQUIRED, example = "{}") - private Map params; - - @AssertTrue(message = "工作流或模型，必须传递一个") - public boolean isGraphValid() { - return id != null || StrUtil.isNotEmpty(graph); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/AiWriteController.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/AiWriteController.java deleted file mode 100644 index 3a0dc72..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/AiWriteController.java +++ /dev/null @@ -1,57 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.write; - -import cn.iocoder.lyzsys.framework.common.pojo.CommonResult; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.object.BeanUtils; -import cn.iocoder.lyzsys.module.ai.controller.admin.write.vo.AiWriteGenerateReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.write.vo.AiWritePageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.write.vo.AiWriteRespVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.write.AiWriteDO; -import cn.iocoder.lyzsys.module.ai.service.write.AiWriteService; -import io.swagger.v3.oas.annotations.Operation; -import io.swagger.v3.oas.annotations.Parameter; -import io.swagger.v3.oas.annotations.tags.Tag; -import jakarta.annotation.Resource; -import jakarta.validation.Valid; -import org.springframework.http.MediaType; -import org.springframework.security.access.prepost.PreAuthorize; -import org.springframework.web.bind.annotation.*; -import reactor.core.publisher.Flux; - -import static cn.iocoder.lyzsys.framework.common.pojo.CommonResult.success; -import static cn.iocoder.lyzsys.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId; - -@Tag(name = "管理后台 - AI 写作") -@RestController -@RequestMapping("/ai/write") -public class AiWriteController { - - @Resource - private AiWriteService writeService; - - @PostMapping(value = "/generate-stream", produces = MediaType.TEXT_EVENT_STREAM_VALUE) - @Operation(summary = "写作生成（流式）", description = "流式返回，响应较快") - public Flux> generateWriteContent(@RequestBody @Valid AiWriteGenerateReqVO generateReqVO) { - return writeService.generateWriteContent(generateReqVO, getLoginUserId()); - } - - // ================ 写作管理 ================ - - @DeleteMapping("/delete") - @Operation(summary = "删除写作") - @Parameter(name = "id", description = "编号", required = true) - @PreAuthorize("@ss.hasPermission('ai:write:delete')") - public CommonResult deleteWrite(@RequestParam("id") Long id) { - writeService.deleteWrite(id); - return success(true); - } - - @GetMapping("/page") - @Operation(summary = "获得写作分页") - @PreAuthorize("@ss.hasPermission('ai:write:query')") - public CommonResult> getWritePage(@Valid AiWritePageReqVO pageReqVO) { - PageResult pageResult = writeService.getWritePage(pageReqVO); - return success(BeanUtils.toBean(pageResult, AiWriteRespVO.class)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWriteGenerateReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWriteGenerateReqVO.java deleted file mode 100644 index fd55e8b..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWriteGenerateReqVO.java +++ /dev/null @@ -1,39 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.write.vo; - -import cn.iocoder.lyzsys.framework.common.validation.InEnum; -import cn.iocoder.lyzsys.module.ai.enums.write.AiWriteTypeEnum; -import io.swagger.v3.oas.annotations.media.Schema; -import jakarta.validation.constraints.NotNull; -import lombok.Data; - -@Schema(description = "管理后台 - AI 写作生成 Request VO") -@Data -public class AiWriteGenerateReqVO { - - @Schema(description = "写作类型", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @InEnum(value = AiWriteTypeEnum.class, message = "写作类型必须是 {value}") - private Integer type; - - @Schema(description = "写作内容提示", example = "1.撰写：田忌赛马；2.回复：不批") - private String prompt; - - @Schema(description = "原文", example = "领导我要辞职") - private String originalContent; - - @Schema(description = "长度", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "长度不能为空") - private Integer length; - - @Schema(description = "格式", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "格式不能为空") - private Integer format; - - @Schema(description = "语气", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "语气不能为空") - private Integer tone; - - @Schema(description = "语言", requiredMode = Schema.RequiredMode.REQUIRED, example = "1") - @NotNull(message = "语言不能为空") - private Integer language; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWritePageReqVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWritePageReqVO.java deleted file mode 100644 index e08e034..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWritePageReqVO.java +++ /dev/null @@ -1,31 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.write.vo; - -import cn.iocoder.lyzsys.framework.common.pojo.PageParam; -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; -import lombok.EqualsAndHashCode; -import lombok.ToString; -import org.springframework.format.annotation.DateTimeFormat; - -import java.time.LocalDateTime; - -import static cn.iocoder.lyzsys.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND; - -@Schema(description = "管理后台 - AI 写作分页 Request VO") -@Data -public class AiWritePageReqVO extends PageParam { - - @Schema(description = "用户编号", example = "28404") - private Long userId; - - @Schema(description = "写作类型", example = "1") - private Integer type; - - @Schema(description = "平台", example = "TongYi") - private String platform; - - @Schema(description = "创建时间") - @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) - private LocalDateTime[] createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWriteRespVO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWriteRespVO.java deleted file mode 100644 index d98663f..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/admin/write/vo/AiWriteRespVO.java +++ /dev/null @@ -1,54 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.controller.admin.write.vo; - -import io.swagger.v3.oas.annotations.media.Schema; -import lombok.Data; - -import java.time.LocalDateTime; - -@Schema(description = "管理后台 - AI 写作 Response VO") -@Data -public class AiWriteRespVO { - - @Schema(description = "编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "5311") - private Long id; - - @Schema(description = "用户编号", requiredMode = Schema.RequiredMode.REQUIRED, example = "28404") - private Long userId; - - @Schema(description = "写作类型", example = "1") - private Integer type; - - @Schema(description = "平台", requiredMode = Schema.RequiredMode.REQUIRED, example = "TongYi") - private String platform; - - @Schema(description = "模型", requiredMode = Schema.RequiredMode.REQUIRED, example = "qwen") - private String model; - - @Schema(description = "生成内容提示", requiredMode = Schema.RequiredMode.REQUIRED, example = "撰写：田忌赛马") - private String prompt; - - @Schema(description = "生成的内容", example = "你非常不错") - private String generatedContent; - - @Schema(description = "原文", example = "真的么？") - private String originalContent; - - @Schema(description = "长度提示词", example = "1") - private Integer length; - - @Schema(description = "格式提示词", example = "2") - private Integer format; - - @Schema(description = "语气提示词", example = "3") - private Integer tone; - - @Schema(description = "语言提示词", example = "4") - private Integer language; - - @Schema(description = "错误信息") - private String errorMessage; - - @Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED) - private LocalDateTime createTime; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/app/package-info.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/app/package-info.java deleted file mode 100644 index e5a96ca..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/app/package-info.java +++ /dev/null @@ -1,4 +0,0 @@ -/** - * TODO 芋艿：站位，无特殊作用 - */ -package cn.iocoder.lyzsys.module.ai.controller.app; diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/package-info.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/package-info.java deleted file mode 100644 index 8dd73da..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/controller/package-info.java +++ /dev/null @@ -1,6 +0,0 @@ -/** - * 提供 RESTful API 给前端： - * 1. admin 包：提供给管理后台 lyzsys-ui-admin 前端项目 - * 2. app 包：提供给用户 APP lyzsys-ui-app 前端项目，它的 Controller 和 VO 都要添加 App 前缀，用于和管理后台进行区分 - */ -package cn.iocoder.lyzsys.module.ai.controller; diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/chat/AiChatConversationDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/chat/AiChatConversationDO.java deleted file mode 100644 index dc9c741..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/chat/AiChatConversationDO.java +++ /dev/null @@ -1,100 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.chat; - -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiChatRoleDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.*; - -import java.time.LocalDateTime; - -/** - * AI Chat 对话 DO - * - * 用户每次发起 Chat 聊天时，会创建一个 {@link AiChatConversationDO} 对象，将它的消息关联在一起 - * - * @author fansili - * @since 2024/4/14 17:35 - */ -@TableName("ai_chat_conversation") -@KeySequence("ai_chat_conversation_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -@Builder -@NoArgsConstructor -@AllArgsConstructor -public class AiChatConversationDO extends BaseDO { - - public static final String TITLE_DEFAULT = "新对话"; - - /** - * ID 编号，自增 - */ - @TableId - private Long id; - - /** - * 用户编号 - * - * 关联 AdminUserDO 的 userId 字段 - */ - private Long userId; - - /** - * 对话标题 - * - * 默认由系统自动生成，可用户手动修改 - */ - private String title; - /** - * 是否置顶 - */ - private Boolean pinned; - /** - * 置顶时间 - */ - private LocalDateTime pinnedTime; - - /** - * 角色编号 - * - * 关联 {@link AiChatRoleDO#getId()} - */ - private Long roleId; - - /** - * 模型编号 - * - * 关联 {@link AiModelDO#getId()} 字段 - */ - private Long modelId; - /** - * 模型标志 - * - * 冗余 {@link AiModelDO#getModel()} 字段 - */ - private String model; - - // ========== 对话配置 ========== - - /** - * 角色设定 - */ - private String systemMessage; - /** - * 温度参数 - * - * 用于调整生成回复的随机性和多样性程度：较低的温度值会使输出更收敛于高频词汇，较高的则增加多样性 - */ - private Double temperature; - /** - * 单条回复的最大 Token 数量 - */ - private Integer maxTokens; - /** - * 上下文的最大 Message 数量 - */ - private Integer maxContexts; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/chat/AiChatMessageDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/chat/AiChatMessageDO.java deleted file mode 100644 index 6c453b2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/chat/AiChatMessageDO.java +++ /dev/null @@ -1,126 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.chat; - -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.framework.mybatis.core.type.LongListTypeHandler; -import cn.iocoder.lyzsys.framework.mybatis.core.type.StringListTypeHandler; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeSegmentDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiChatRoleDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.webserch.AiWebSearchResponse; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableField; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler; -import lombok.AllArgsConstructor; -import lombok.Builder; -import lombok.Data; -import lombok.NoArgsConstructor; -import org.springframework.ai.chat.messages.MessageType; - -import java.util.List; - -/** - * AI Chat 消息 DO - * - * @since 2024/4/14 17:35 - * @since 2024/4/14 17:35 - */ -@TableName(value = "ai_chat_message", autoResultMap = true) -@KeySequence("ai_chat_message_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -@Builder -@NoArgsConstructor -@AllArgsConstructor -public class AiChatMessageDO extends BaseDO { - - /** - * 编号，作为每条聊天记录的唯一标识符 - */ - @TableId - private Long id; - - /** - * 对话编号 - * - * 关联 {@link AiChatConversationDO#getId()} 字段 - */ - private Long conversationId; - /** - * 回复消息编号 - * - * 关联 {@link #id} 字段 - * - * 大模型回复的消息编号，用于“问答”的关联 - */ - private Long replyId; - - /** - * 消息类型 - * - * 也等价于 OpenAPI 的 role 字段 - * - * 枚举 {@link MessageType} - */ - private String type; - /** - * 用户编号 - * - * 关联 AdminUserDO 的 userId 字段 - */ - private Long userId; - /** - * 角色编号 - * - * 关联 {@link AiChatRoleDO#getId()} 字段 - */ - private Long roleId; - - /** - * 模型标志 - * - * 冗余 {@link AiModelDO#getModel()} - */ - private String model; - /** - * 模型编号 - * - * 关联 {@link AiModelDO#getId()} 字段 - */ - private Long modelId; - - /** - * 聊天内容 - */ - private String content; - /** - * 推理内容 - */ - private String reasoningContent; - - /** - * 是否携带上下文 - */ - private Boolean useContext; - - /** - * 知识库段落编号数组 - * - * 关联 {@link AiKnowledgeSegmentDO#getId()} 字段 - */ - @TableField(typeHandler = LongListTypeHandler.class) - private List segmentIds; - - /** - * 联网搜索的网页内容数组 - */ - @TableField(typeHandler = JacksonTypeHandler.class) - private List webSearchPages; - - /** - * 附件 URL 数组 - */ - @TableField(typeHandler = StringListTypeHandler.class) - private List attachmentUrls; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/image/AiImageDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/image/AiImageDO.java deleted file mode 100644 index 86f025c..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/image/AiImageDO.java +++ /dev/null @@ -1,127 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.image; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import cn.iocoder.lyzsys.module.ai.enums.image.AiImageStatusEnum; -import cn.iocoder.lyzsys.module.system.api.user.dto.AdminUserRespDTO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableField; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler; -import lombok.Data; -import org.springframework.ai.openai.OpenAiImageOptions; -import org.springframework.ai.stabilityai.api.StabilityAiImageOptions; - -import java.time.LocalDateTime; -import java.util.List; -import java.util.Map; - -/** - * AI 绘画 DO - * - * @author fansili - */ -@TableName(value = "ai_image", autoResultMap = true) -@KeySequence("ai_image_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiImageDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - - /** - * 用户编号 - * - * 关联 {@link AdminUserRespDTO#getId()} - */ - private Long userId; - - /** - * 提示词 - */ - private String prompt; - - /** - * 平台 - * - * 枚举 {@link AiPlatformEnum} - */ - private String platform; - /** - * 模型编号 - * - * 关联 {@link AiModelDO#getId()} - */ - private Long modelId; - /** - * 模型标识 - * - * 冗余 {@link AiModelDO#getModel()} - */ - private String model; - - /** - * 图片宽度 - */ - private Integer width; - /** - * 图片高度 - */ - private Integer height; - - /** - * 生成状态 - * - * 枚举 {@link AiImageStatusEnum} - */ - private Integer status; - - /** - * 完成时间 - */ - private LocalDateTime finishTime; - - /** - * 绘画错误信息 - */ - private String errorMessage; - - /** - * 图片地址 - */ - private String picUrl; - /** - * 是否公开 - */ - private Boolean publicStatus; - - /** - * 绘制参数，不同 platform 的不同参数 - * - * 1. {@link OpenAiImageOptions} - * 2. {@link StabilityAiImageOptions} - */ - @TableField(typeHandler = JacksonTypeHandler.class) - private Map options; - - /** - * mj buttons 按钮 - */ - @TableField(typeHandler = JacksonTypeHandler.class) - private List buttons; - - /** - * 任务编号 - * - * 1. midjourney proxy：关联的 task id - */ - private String taskId; - -} - diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeDO.java deleted file mode 100644 index 936f22f..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeDO.java +++ /dev/null @@ -1,64 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.Data; - -/** - * AI 知识库 DO - * - * @author xiaoxin - */ -@TableName(value = "ai_knowledge", autoResultMap = true) -@KeySequence("ai_knowledge_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiKnowledgeDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - /** - * 知识库名称 - */ - private String name; - /** - * 知识库描述 - */ - private String description; - - /** - * 向量模型编号 - * - * 关联 {@link AiModelDO#getId()} - */ - private Long embeddingModelId; - /** - * 模型标识 - * - * 冗余 {@link AiModelDO#getModel()} - */ - private String embeddingModel; - - /** - * topK - */ - private Integer topK; - /** - * 相似度阈值 - */ - private Double similarityThreshold; - - /** - * 状态 - *

- * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeDocumentDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeDocumentDO.java deleted file mode 100644 index 01798b3..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeDocumentDO.java +++ /dev/null @@ -1,69 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.Data; - -/** - * AI 知识库-文档 DO - * - * @author xiaoxin - */ -@TableName(value = "ai_knowledge_document") -@KeySequence("ai_knowledge_document_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiKnowledgeDocumentDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - /** - * 知识库编号 - *

- * 关联 {@link AiKnowledgeDO#getId()} - */ - private Long knowledgeId; - /** - * 文档名称 - */ - private String name; - /** - * 文件 URL - */ - private String url; - /** - * 内容 - */ - private String content; - /** - * 文档长度 - */ - private Integer contentLength; - - /** - * 文档 token 数量 - */ - private Integer tokens; - /** - * 分片最大 Token 数 - */ - private Integer segmentMaxTokens; - - /** - * 召回次数 - */ - private Integer retrievalCount; - - /** - * 状态 - *

- * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeSegmentDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeSegmentDO.java deleted file mode 100644 index 41213e2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/knowledge/AiKnowledgeSegmentDO.java +++ /dev/null @@ -1,72 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.Data; - -/** - * AI 知识库-文档分段 DO - * - * @author xiaoxin - */ -@TableName(value = "ai_knowledge_segment") -@KeySequence("ai_knowledge_segment_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiKnowledgeSegmentDO extends BaseDO { - - /** - * 向量库的编号 - 空值 - */ - public static final String VECTOR_ID_EMPTY = ""; - - /** - * 编号 - */ - @TableId - private Long id; - /** - * 知识库编号 - *

- * 关联 {@link AiKnowledgeDO#getId()} - */ - private Long knowledgeId; - /** - * 文档编号 - *

- * 关联 {@link AiKnowledgeDocumentDO#getId()} - */ - private Long documentId; - /** - * 切片内容 - */ - private String content; - /** - * 切片内容长度 - */ - private Integer contentLength; - - /** - * 向量库的编号 - */ - private String vectorId; - /** - * token 数量 - */ - private Integer tokens; - - /** - * 召回次数 - */ - private Integer retrievalCount; - - /** - * 状态 - *

- * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/mindmap/AiMindMapDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/mindmap/AiMindMapDO.java deleted file mode 100644 index 3c71a0d..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/mindmap/AiMindMapDO.java +++ /dev/null @@ -1,66 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.mindmap; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.Data; - -/** - * AI 思维导图 DO - * - * @author xiaoxin - */ -@TableName(value = "ai_mind_map") -@KeySequence("ai_mind_map_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiMindMapDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - - /** - * 用户编号 - *

- * 关联 AdminUserDO 的 userId 字段 - */ - private Long userId; - - /** - * 平台 - *

- * 枚举 {@link AiPlatformEnum} - */ - private String platform; - /** - * 模型编号 - * - * 关联 {@link AiModelDO#getId()} - */ - private Long modelId; - /** - * 模型 - */ - private String model; - - /** - * 生成内容提示 - */ - private String prompt; - - /** - * 生成的内容 - */ - private String generatedContent; - - /** - * 错误信息 - */ - private String errorMessage; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiApiKeyDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiApiKeyDO.java deleted file mode 100644 index 8181229..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiApiKeyDO.java +++ /dev/null @@ -1,54 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.model; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.*; - -/** - * AI API 秘钥 DO - * - * @author 芋道源码 - */ -@TableName("ai_api_key") -@KeySequence("ai_api_key_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -@Builder -@NoArgsConstructor -@AllArgsConstructor -public class AiApiKeyDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - /** - * 名称 - */ - private String name; - /** - * 密钥 - */ - private String apiKey; - /** - * 平台 - * - * 枚举 {@link AiPlatformEnum} - */ - private String platform; - /** - * API 地址 - */ - private String url; - /** - * 状态 - * - * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiChatRoleDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiChatRoleDO.java deleted file mode 100644 index 3433100..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiChatRoleDO.java +++ /dev/null @@ -1,111 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.model; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.framework.mybatis.core.type.LongListTypeHandler; -import cn.iocoder.lyzsys.framework.mybatis.core.type.StringListTypeHandler; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableField; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.*; - -import java.util.List; - -/** - * AI 聊天角色 DO - * - * @author fansili - * @since 2024/4/24 19:39 - */ -@TableName(value = "ai_chat_role", autoResultMap = true) -@KeySequence("ai_chat_role_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -@Builder -@NoArgsConstructor -@AllArgsConstructor -public class AiChatRoleDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - /** - * 角色名称 - */ - private String name; - /** - * 角色头像 - */ - private String avatar; - /** - * 角色分类 - */ - private String category; - /** - * 角色描述 - */ - private String description; - /** - * 角色设定 - */ - private String systemMessage; - - /** - * 用户编号 - * - * 关联 AdminUserDO 的 userId 字段 - */ - private Long userId; - - /** - * 模型编号 - * - * 关联 {@link AiModelDO#getId()} 字段 - */ - private Long modelId; - - /** - * 引用的知识库编号列表 - * - * 关联 {@link AiKnowledgeDO#getId()} 字段 - */ - @TableField(typeHandler = LongListTypeHandler.class) - private List knowledgeIds; - /** - * 引用的工具编号列表 - * - * 关联 {@link AiToolDO#getId()} 字段 - */ - @TableField(typeHandler = LongListTypeHandler.class) - private List toolIds; - /** - * 引用的 MCP Client 名字列表 - * - * 关联 spring.ai.mcp.client 下的名字 - */ - @TableField(typeHandler = StringListTypeHandler.class) - private List mcpClientNames; - - /** - * 是否公开 - * - * 1. true - 公开；由管理员在【角色管理】所创建 - * 2. false - 私有；由个人在【我的角色】所创建 - */ - private Boolean publicStatus; - - /** - * 排序值 - */ - private Integer sort; - /** - * 状态 - * - * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiModelDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiModelDO.java deleted file mode 100644 index 9468d0d..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiModelDO.java +++ /dev/null @@ -1,88 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.model; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiModelTypeEnum; -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.*; - -/** - * AI 模型 DO - * - * 默认模型：{@link #status} 为开启，并且 {@link #sort} 排序第一 - * - * @author fansili - * @since 2024/4/24 19:39 - */ -@TableName("ai_model") -@KeySequence("ai_model_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -@Builder -@NoArgsConstructor -@AllArgsConstructor -public class AiModelDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - /** - * API 秘钥编号 - * - * 关联 {@link AiApiKeyDO#getId()} - */ - private Long keyId; - /** - * 模型名称 - */ - private String name; - /** - * 模型标志 - */ - private String model; - /** - * 平台 - * - * 枚举 {@link AiPlatformEnum} - */ - private String platform; - /** - * 类型 - * - * 枚举 {@link AiModelTypeEnum} - */ - private Integer type; - - /** - * 排序值 - */ - private Integer sort; - /** - * 状态 - * - * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - - // ========== 对话配置 ========== - - /** - * 温度参数 - * - * 用于调整生成回复的随机性和多样性程度：较低的温度值会使输出更收敛于高频词汇，较高的则增加多样性 - */ - private Double temperature; - /** - * 单条回复的最大 Token 数量 - */ - private Integer maxTokens; - /** - * 上下文的最大 Message 数量 - */ - private Integer maxContexts; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiToolDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiToolDO.java deleted file mode 100644 index daf3793..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/model/AiToolDO.java +++ /dev/null @@ -1,48 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.model; - -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.tool.function.DirectoryListToolFunction; -import cn.iocoder.lyzsys.module.ai.tool.function.WeatherQueryToolFunction; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.*; - -/** - * AI 工具 DO - * - * @author 芋道源码 - */ -@TableName("ai_tool") -@KeySequence("ai_tool_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -@Builder -@NoArgsConstructor -@AllArgsConstructor -public class AiToolDO extends BaseDO { - - /** - * 工具编号 - */ - @TableId - private Long id; - /** - * 工具名称 - * - * 对应 Bean 的名字，例如说： - * 1. {@link DirectoryListToolFunction} 的 Bean 名字是 directory_list - * 2. {@link WeatherQueryToolFunction} 的 Bean 名字是 weather_query - */ - private String name; - /** - * 工具描述 - */ - private String description; - /** - * 状态 - * - * 枚举 {@link cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/music/AiMusicDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/music/AiMusicDO.java deleted file mode 100644 index d108997..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/music/AiMusicDO.java +++ /dev/null @@ -1,119 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.music; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.enums.music.AiMusicGenerateModeEnum; -import cn.iocoder.lyzsys.module.ai.enums.music.AiMusicStatusEnum; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableField; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import com.baomidou.mybatisplus.extension.handlers.JacksonTypeHandler; -import lombok.Data; - -import java.util.List; - -/** - * AI 音乐 DO - * - * @author xiaoxin - */ -@TableName(value = "ai_music", autoResultMap = true) -@KeySequence("ai_music_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiMusicDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - - /** - * 用户编号 - *

- * 关联 AdminUserDO 的 userId 字段 - */ - private Long userId; - - /** - * 音乐名称 - */ - private String title; - - /** - * 歌词 - */ - private String lyric; - - /** - * 图片地址 - */ - private String imageUrl; - /** - * 音频地址 - */ - private String audioUrl; - /** - * 视频地址 - */ - private String videoUrl; - - /** - * 音乐状态 - *

- * 枚举 {@link AiMusicStatusEnum} - */ - private Integer status; - - /** - * 生成模式 - *

- * 枚举 {@link AiMusicGenerateModeEnum} - */ - private Integer generateMode; - - /** - * 描述词 - */ - private String description; - - /** - * 平台 - *

- * 枚举 {@link AiPlatformEnum} - */ - private String platform; - // TODO @芋艿：modelId？ - /** - * 模型 - */ - private String model; - - /** - * 音乐风格标签 - */ - @TableField(typeHandler = JacksonTypeHandler.class) - private List tags; - - /** - * 音乐时长 - */ - private Double duration; - - /** - * 是否公开 - */ - private Boolean publicStatus; - - /** - * 任务编号 - */ - private String taskId; - - /** - * 错误信息 - */ - private String errorMessage; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/workflow/AiWorkflowDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/workflow/AiWorkflowDO.java deleted file mode 100644 index af06f6d..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/workflow/AiWorkflowDO.java +++ /dev/null @@ -1,51 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.workflow; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.Data; - -/** - * AI 工作流 DO - * - * @author lesan - */ -@TableName(value = "ai_workflow", autoResultMap = true) -@KeySequence("ai_workflow") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiWorkflowDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - /** - * 工作流名称 - */ - private String name; - /** - * 工作流标识 - */ - private String code; - - /** - * 工作流模型 JSON 数据 - */ - private String graph; - - /** - * 备注 - */ - private String remark; - - /** - * 状态 - * - * 枚举 {@link CommonStatusEnum} - */ - private Integer status; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/write/AiWriteDO.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/write/AiWriteDO.java deleted file mode 100644 index da607a0..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/dataobject/write/AiWriteDO.java +++ /dev/null @@ -1,104 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.dataobject.write; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.framework.mybatis.core.dataobject.BaseDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import cn.iocoder.lyzsys.module.ai.enums.DictTypeConstants; -import cn.iocoder.lyzsys.module.ai.enums.write.AiWriteTypeEnum; -import com.baomidou.mybatisplus.annotation.KeySequence; -import com.baomidou.mybatisplus.annotation.TableId; -import com.baomidou.mybatisplus.annotation.TableName; -import lombok.Data; - -/** - * AI 写作 DO - * - * @author xiaoxin - */ -@TableName("ai_write") -@KeySequence("ai_write_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。 -@Data -public class AiWriteDO extends BaseDO { - - /** - * 编号 - */ - @TableId - private Long id; - - /** - * 用户编号 - * - * 关联 AdminUserDO 的 userId 字段 - */ - private Long userId; - - /** - * 写作类型 - *

- * 枚举 {@link AiWriteTypeEnum} - */ - private Integer type; - - /** - * 平台 - * - * 枚举 {@link AiPlatformEnum} - */ - private String platform; - /** - * 模型编号 - * - * 关联 {@link AiModelDO#getId()} - */ - private Long modelId; - /** - * 模型 - */ - private String model; - - /** - * 生成内容提示 - */ - private String prompt; - - /** - * 生成的内容 - */ - private String generatedContent; - /** - * 原文 - */ - private String originalContent; - - /** - * 长度提示词 - * - * 字典：{@link DictTypeConstants#AI_WRITE_LENGTH} - */ - private Integer length; - /** - * 格式提示词 - * - * 字典：{@link DictTypeConstants#AI_WRITE_FORMAT} - */ - private Integer format; - /** - * 语气提示词 - * - * 字典：{@link DictTypeConstants#AI_WRITE_TONE} - */ - private Integer tone; - /** - * 语言提示词 - * - * 字典：{@link DictTypeConstants#AI_WRITE_LANGUAGE} - */ - private Integer language; - - /** - * 错误信息 - */ - private String errorMessage; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/chat/AiChatConversationMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/chat/AiChatConversationMapper.java deleted file mode 100644 index 97d1f54..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/chat/AiChatConversationMapper.java +++ /dev/null @@ -1,38 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.chat; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation.AiChatConversationPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.chat.AiChatConversationDO; -import org.apache.ibatis.annotations.Mapper; - -import java.util.List; - -/** - * AI 聊天对话 Mapper - * - * @author 芋道源码 - */ -@Mapper -public interface AiChatConversationMapper extends BaseMapperX { - - default List selectListByUserId(Long userId) { - return selectList(AiChatConversationDO::getUserId, userId); - } - - default List selectListByUserIdAndPinned(Long userId, boolean pinned) { - return selectList(new LambdaQueryWrapperX() - .eq(AiChatConversationDO::getUserId, userId) - .eq(AiChatConversationDO::getPinned, pinned)); - } - - default PageResult selectChatConversationPage(AiChatConversationPageReqVO pageReqVO) { - return selectPage(pageReqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiChatConversationDO::getUserId, pageReqVO.getUserId()) - .likeIfPresent(AiChatConversationDO::getTitle, pageReqVO.getTitle()) - .betweenIfPresent(AiChatConversationDO::getCreateTime, pageReqVO.getCreateTime()) - .orderByDesc(AiChatConversationDO::getId)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/chat/AiChatMessageMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/chat/AiChatMessageMapper.java deleted file mode 100644 index 347a66c..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/chat/AiChatMessageMapper.java +++ /dev/null @@ -1,59 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.chat; - -import cn.hutool.core.collection.CollUtil; -import cn.hutool.core.map.MapUtil; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.conversation.AiChatConversationPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.chat.vo.message.AiChatMessagePageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.chat.AiChatConversationDO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.chat.AiChatMessageDO; -import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; -import org.apache.ibatis.annotations.Mapper; - -import java.util.Collection; -import java.util.Collections; -import java.util.List; -import java.util.Map; - -/** - * AI 聊天对话 Mapper - * - * @author fansili - */ -@Mapper -public interface AiChatMessageMapper extends BaseMapperX { - - default List selectListByConversationId(Long conversationId) { - return selectList(new LambdaQueryWrapperX() - .eq(AiChatMessageDO::getConversationId, conversationId) - .orderByAsc(AiChatMessageDO::getId)); - } - - default Map selectCountMapByConversationId(Collection conversationIds) { - // SQL count 查询 - List> result = selectMaps(new QueryWrapper() - .select("COUNT(id) AS count, conversation_id AS conversationId") - .in("conversation_id", conversationIds) - .groupBy("conversation_id")); - if (CollUtil.isEmpty(result)) { - return Collections.emptyMap(); - } - // 转换数据 - return CollectionUtils.convertMap(result, - record -> MapUtil.getLong(record, "conversationId"), - record -> MapUtil.getInt(record, "count" )); - } - - default PageResult selectPage(AiChatMessagePageReqVO pageReqVO) { - return selectPage(pageReqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiChatMessageDO::getConversationId, pageReqVO.getConversationId()) - .eqIfPresent(AiChatMessageDO::getUserId, pageReqVO.getUserId()) - .likeIfPresent(AiChatMessageDO::getContent, pageReqVO.getContent()) - .betweenIfPresent(AiChatMessageDO::getCreateTime, pageReqVO.getCreateTime()) - .orderByDesc(AiChatMessageDO::getId)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/image/AiImageMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/image/AiImageMapper.java deleted file mode 100644 index 3a0d14a..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/image/AiImageMapper.java +++ /dev/null @@ -1,57 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.image; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.AiImagePageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.image.vo.AiImagePublicPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.image.AiImageDO; -import org.apache.ibatis.annotations.Mapper; - -import java.util.List; - -/** - * AI 绘图 Mapper - * - * @author fansili - */ -@Mapper -public interface AiImageMapper extends BaseMapperX { - - default AiImageDO selectByTaskId(String taskId) { - return selectOne(AiImageDO::getTaskId, taskId); - } - - default PageResult selectPage(AiImagePageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiImageDO::getUserId, reqVO.getUserId()) - .eqIfPresent(AiImageDO::getPlatform, reqVO.getPlatform()) - .eqIfPresent(AiImageDO::getStatus, reqVO.getStatus()) - .eqIfPresent(AiImageDO::getPublicStatus, reqVO.getPublicStatus()) - .betweenIfPresent(AiImageDO::getCreateTime, reqVO.getCreateTime()) - .orderByDesc(AiImageDO::getId)); - } - - default PageResult selectPageMy(Long userId, AiImagePageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiImageDO::getPrompt, reqVO.getPrompt()) - // 情况一：公开 - .eq(Boolean.TRUE.equals(reqVO.getPublicStatus()), AiImageDO::getPublicStatus, reqVO.getPublicStatus()) - // 情况二：私有 - .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiImageDO::getUserId, userId) - .orderByDesc(AiImageDO::getId)); - } - - default PageResult selectPage(AiImagePublicPageReqVO pageReqVO) { - return selectPage(pageReqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiImageDO::getPublicStatus, Boolean.TRUE) - .likeIfPresent(AiImageDO::getPrompt, pageReqVO.getPrompt()) - .orderByDesc(AiImageDO::getId)); - } - - default List selectListByStatusAndPlatform(Integer status, String platform) { - return selectList(AiImageDO::getStatus, status, - AiImageDO::getPlatform, platform); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeDocumentMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeDocumentMapper.java deleted file mode 100644 index e422d4c..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeDocumentMapper.java +++ /dev/null @@ -1,43 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.knowledge; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.document.AiKnowledgeDocumentPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDocumentDO; -import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; -import org.apache.ibatis.annotations.Mapper; - -import java.util.Collection; -import java.util.List; - -/** - * AI 知识库文档 Mapper - * - * @author xiaoxin - */ -@Mapper -public interface AiKnowledgeDocumentMapper extends BaseMapperX { - - default PageResult selectPage(AiKnowledgeDocumentPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiKnowledgeDocumentDO::getKnowledgeId, reqVO.getKnowledgeId()) - .likeIfPresent(AiKnowledgeDocumentDO::getName, reqVO.getName()) - .orderByDesc(AiKnowledgeDocumentDO::getId)); - } - - default void updateRetrievalCountIncr(Collection ids) { - update(new LambdaUpdateWrapper() - .setSql(" retrieval_count = retrieval_count + 1") - .in(AiKnowledgeDocumentDO::getId, ids)); - } - - default List selectListByStatus(Integer status) { - return selectList(AiKnowledgeDocumentDO::getStatus, status); - } - - default List selectListByKnowledgeId(Long knowledgeId) { - return selectList(AiKnowledgeDocumentDO::getKnowledgeId, knowledgeId); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeMapper.java deleted file mode 100644 index b0abd60..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeMapper.java +++ /dev/null @@ -1,32 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.knowledge; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.knowledge.AiKnowledgePageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeDO; -import org.apache.ibatis.annotations.Mapper; - -import java.util.List; - -/** - * AI 知识库 Mapper - * - * @author xiaoxin - */ -@Mapper -public interface AiKnowledgeMapper extends BaseMapperX { - - default PageResult selectPage(AiKnowledgePageReqVO pageReqVO) { - return selectPage(pageReqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiKnowledgeDO::getName, pageReqVO.getName()) - .eqIfPresent(AiKnowledgeDO::getStatus, pageReqVO.getStatus()) - .betweenIfPresent(AiKnowledgeDO::getCreateTime, pageReqVO.getCreateTime()) - .orderByDesc(AiKnowledgeDO::getId)); - } - - default List selectListByStatus(Integer status) { - return selectList(AiKnowledgeDO::getStatus, status); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeSegmentMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeSegmentMapper.java deleted file mode 100644 index 82586bd..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/knowledge/AiKnowledgeSegmentMapper.java +++ /dev/null @@ -1,66 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.knowledge; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.MPJLambdaWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment.AiKnowledgeSegmentPageReqVO; -import cn.iocoder.lyzsys.module.ai.controller.admin.knowledge.vo.segment.AiKnowledgeSegmentProcessRespVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.knowledge.AiKnowledgeSegmentDO; -import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; -import com.github.yulichang.wrapper.MPJLambdaWrapper; -import org.apache.ibatis.annotations.Mapper; - -import java.util.Collection; -import java.util.List; - -/** - * AI 知识库分片 Mapper - * - * @author xiaoxin - */ -@Mapper -public interface AiKnowledgeSegmentMapper extends BaseMapperX { - - default PageResult selectPage(AiKnowledgeSegmentPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .eq(AiKnowledgeSegmentDO::getDocumentId, reqVO.getDocumentId()) - .likeIfPresent(AiKnowledgeSegmentDO::getContent, reqVO.getContent()) - .eqIfPresent(AiKnowledgeSegmentDO::getStatus, reqVO.getStatus()) - .orderByDesc(AiKnowledgeSegmentDO::getId)); - } - - default List selectListByVectorIds(List vectorIds) { - return selectList(new LambdaQueryWrapperX() - .in(AiKnowledgeSegmentDO::getVectorId, vectorIds) - .orderByDesc(AiKnowledgeSegmentDO::getId)); - } - - default List selectListByDocumentId(Long documentId) { - return selectList(new LambdaQueryWrapperX() - .eq(AiKnowledgeSegmentDO::getDocumentId, documentId) - .orderByDesc(AiKnowledgeSegmentDO::getId)); - } - - default List selectListByKnowledgeIdAndStatus(Long knowledgeId, Integer status) { - return selectList(AiKnowledgeSegmentDO::getKnowledgeId, knowledgeId, - AiKnowledgeSegmentDO::getStatus, status); - } - - default List selectProcessList(Collection documentIds) { - MPJLambdaWrapper wrapper = new MPJLambdaWrapperX() - .selectAs(AiKnowledgeSegmentDO::getDocumentId, AiKnowledgeSegmentProcessRespVO::getDocumentId) - .selectCount(AiKnowledgeSegmentDO::getId, "count") - .select("COUNT(CASE WHEN vector_id IS NOT NULL AND vector_id <> '" + AiKnowledgeSegmentDO.VECTOR_ID_EMPTY + "' THEN 1 ELSE NULL END) AS embeddingCount") - .in(AiKnowledgeSegmentDO::getDocumentId, documentIds) - .groupBy(AiKnowledgeSegmentDO::getDocumentId); - return selectJoinList(AiKnowledgeSegmentProcessRespVO.class, wrapper); - } - - default void updateRetrievalCountIncrByIds(List ids) { - update(new LambdaUpdateWrapper() - .setSql(" retrieval_count = retrieval_count + 1") - .in(AiKnowledgeSegmentDO::getId, ids)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/mindmap/AiMindMapMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/mindmap/AiMindMapMapper.java deleted file mode 100644 index bbce0f8..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/mindmap/AiMindMapMapper.java +++ /dev/null @@ -1,26 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.mindmap; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.mindmap.vo.AiMindMapPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.mindmap.AiMindMapDO; -import org.apache.ibatis.annotations.Mapper; - -/** - * AI 思维导图 Mapper - * - * @author xiaoxin - */ -@Mapper -public interface AiMindMapMapper extends BaseMapperX { - - default PageResult selectPage(AiMindMapPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiMindMapDO::getUserId, reqVO.getUserId()) - .eqIfPresent(AiMindMapDO::getPrompt, reqVO.getPrompt()) - .betweenIfPresent(AiMindMapDO::getCreateTime, reqVO.getCreateTime()) - .orderByDesc(AiMindMapDO::getId)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiApiKeyMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiApiKeyMapper.java deleted file mode 100644 index 1e68552..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiApiKeyMapper.java +++ /dev/null @@ -1,35 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.model; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.QueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.apikey.AiApiKeyPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiApiKeyDO; -import org.apache.ibatis.annotations.Mapper; - -/** - * AI API 密钥 Mapper - * - * @author 芋道源码 - */ -@Mapper -public interface AiApiKeyMapper extends BaseMapperX { - - default PageResult selectPage(AiApiKeyPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiApiKeyDO::getName, reqVO.getName()) - .eqIfPresent(AiApiKeyDO::getPlatform, reqVO.getPlatform()) - .eqIfPresent(AiApiKeyDO::getStatus, reqVO.getStatus()) - .orderByDesc(AiApiKeyDO::getId)); - } - - default AiApiKeyDO selectFirstByPlatformAndStatus(String platform, Integer status) { - return selectOne(new QueryWrapperX() - .eq("platform", platform) - .eq("status", status) - .limitN(1) - .orderByAsc("id")); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiChatMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiChatMapper.java deleted file mode 100644 index 93b5ff5..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiChatMapper.java +++ /dev/null @@ -1,47 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.model; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.QueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.model.AiModelPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiModelDO; -import org.apache.ibatis.annotations.Mapper; - -import javax.annotation.Nullable; -import java.util.List; - -/** - * API 模型 Mapper - * - * @author fansili - */ -@Mapper -public interface AiChatMapper extends BaseMapperX { - - default AiModelDO selectFirstByStatus(Integer type, Integer status) { - return selectOne(new QueryWrapperX() - .eq("type", type) - .eq("status", status) - .limitN(1) - .orderByAsc("sort")); - } - - default PageResult selectPage(AiModelPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiModelDO::getName, reqVO.getName()) - .eqIfPresent(AiModelDO::getModel, reqVO.getModel()) - .eqIfPresent(AiModelDO::getPlatform, reqVO.getPlatform()) - .orderByAsc(AiModelDO::getSort)); - } - - default List selectListByStatusAndType(Integer status, Integer type, - @Nullable String platform) { - return selectList(new LambdaQueryWrapperX() - .eq(AiModelDO::getStatus, status) - .eq(AiModelDO::getType, type) - .eqIfPresent(AiModelDO::getPlatform, platform) - .orderByAsc(AiModelDO::getSort)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiChatRoleMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiChatRoleMapper.java deleted file mode 100644 index c3767c8..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiChatRoleMapper.java +++ /dev/null @@ -1,54 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.model; - -import cn.iocoder.lyzsys.framework.common.enums.CommonStatusEnum; -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.chatRole.AiChatRolePageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiChatRoleDO; -import org.apache.ibatis.annotations.Mapper; - -import java.util.List; - -/** - * AI 聊天角色 Mapper - * - * @author fansili - */ -@Mapper -public interface AiChatRoleMapper extends BaseMapperX { - - default PageResult selectPage(AiChatRolePageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiChatRoleDO::getName, reqVO.getName()) - .eqIfPresent(AiChatRoleDO::getCategory, reqVO.getCategory()) - .eqIfPresent(AiChatRoleDO::getPublicStatus, reqVO.getPublicStatus()) - .orderByAsc(AiChatRoleDO::getSort)); - } - - default PageResult selectPageByMy(AiChatRolePageReqVO reqVO, Long userId) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiChatRoleDO::getName, reqVO.getName()) - .eqIfPresent(AiChatRoleDO::getCategory, reqVO.getCategory()) - // 情况一：公开 - .eq(Boolean.TRUE.equals(reqVO.getPublicStatus()), AiChatRoleDO::getPublicStatus, reqVO.getPublicStatus()) - // 情况二：私有 - .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiChatRoleDO::getUserId, userId) - .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiChatRoleDO::getStatus, CommonStatusEnum.ENABLE.getStatus()) - .orderByAsc(AiChatRoleDO::getSort)); - } - - default List selectListGroupByCategory(Integer status) { - return selectList(new LambdaQueryWrapperX() - .select(AiChatRoleDO::getCategory) - .eq(AiChatRoleDO::getStatus, status) - .groupBy(AiChatRoleDO::getCategory)); - } - - default List selectListByName(String name) { - return selectList(new LambdaQueryWrapperX() - .likeIfPresent(AiChatRoleDO::getName, name) - .orderByAsc(AiChatRoleDO::getSort)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiToolMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiToolMapper.java deleted file mode 100644 index 406c376..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/model/AiToolMapper.java +++ /dev/null @@ -1,35 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.model; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.model.vo.tool.AiToolPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.model.AiToolDO; -import org.apache.ibatis.annotations.Mapper; - -import java.util.List; - -/** - * AI 工具 Mapper - * - * @author 芋道源码 - */ -@Mapper -public interface AiToolMapper extends BaseMapperX { - - default PageResult selectPage(AiToolPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .likeIfPresent(AiToolDO::getName, reqVO.getName()) - .eqIfPresent(AiToolDO::getDescription, reqVO.getDescription()) - .eqIfPresent(AiToolDO::getStatus, reqVO.getStatus()) - .betweenIfPresent(AiToolDO::getCreateTime, reqVO.getCreateTime()) - .orderByDesc(AiToolDO::getId)); - } - - default List selectListByStatus(Integer status) { - return selectList(new LambdaQueryWrapperX() - .eq(AiToolDO::getStatus, status) - .orderByDesc(AiToolDO::getId)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/music/AiMusicMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/music/AiMusicMapper.java deleted file mode 100644 index c1f54d3..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/music/AiMusicMapper.java +++ /dev/null @@ -1,44 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.music; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.music.vo.AiMusicPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.music.AiMusicDO; -import org.apache.ibatis.annotations.Mapper; - -import java.util.List; - -/** - * AI 音乐 Mapper - * - * @author xiaoxin - */ -@Mapper -public interface AiMusicMapper extends BaseMapperX { - - default List selectListByStatus(Integer status) { - return selectList(AiMusicDO::getStatus, status); - } - - default PageResult selectPage(AiMusicPageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiMusicDO::getUserId, reqVO.getUserId()) - .eqIfPresent(AiMusicDO::getTitle, reqVO.getTitle()) - .eqIfPresent(AiMusicDO::getStatus, reqVO.getStatus()) - .eqIfPresent(AiMusicDO::getGenerateMode, reqVO.getGenerateMode()) - .betweenIfPresent(AiMusicDO::getCreateTime, reqVO.getCreateTime()) - .eqIfPresent(AiMusicDO::getPublicStatus, reqVO.getPublicStatus()) - .orderByDesc(AiMusicDO::getId)); - } - - default PageResult selectPageByMy(AiMusicPageReqVO reqVO, Long userId) { - return selectPage(reqVO, new LambdaQueryWrapperX() - // 情况一：公开 - .eq(Boolean.TRUE.equals(reqVO.getPublicStatus()), AiMusicDO::getPublicStatus, reqVO.getPublicStatus()) - // 情况二：私有 - .eq(Boolean.FALSE.equals(reqVO.getPublicStatus()), AiMusicDO::getUserId, userId) - .orderByAsc(AiMusicDO::getId)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/workflow/AiWorkflowMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/workflow/AiWorkflowMapper.java deleted file mode 100644 index 12be71e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/workflow/AiWorkflowMapper.java +++ /dev/null @@ -1,30 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.workflow; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.workflow.vo.AiWorkflowPageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.workflow.AiWorkflowDO; -import org.apache.ibatis.annotations.Mapper; - -/** - * AI 工作流 Mapper - * - * @author lesan - */ -@Mapper -public interface AiWorkflowMapper extends BaseMapperX { - - default AiWorkflowDO selectByCode(String code) { - return selectOne(AiWorkflowDO::getCode, code); - } - - default PageResult selectPage(AiWorkflowPageReqVO pageReqVO) { - return selectPage(pageReqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiWorkflowDO::getStatus, pageReqVO.getStatus()) - .likeIfPresent(AiWorkflowDO::getName, pageReqVO.getName()) - .likeIfPresent(AiWorkflowDO::getCode, pageReqVO.getCode()) - .betweenIfPresent(AiWorkflowDO::getCreateTime, pageReqVO.getCreateTime())); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/write/AiWriteMapper.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/write/AiWriteMapper.java deleted file mode 100644 index 5ac5435..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/dal/mysql/write/AiWriteMapper.java +++ /dev/null @@ -1,27 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.dal.mysql.write; - -import cn.iocoder.lyzsys.framework.common.pojo.PageResult; -import cn.iocoder.lyzsys.framework.mybatis.core.mapper.BaseMapperX; -import cn.iocoder.lyzsys.framework.mybatis.core.query.LambdaQueryWrapperX; -import cn.iocoder.lyzsys.module.ai.controller.admin.write.vo.AiWritePageReqVO; -import cn.iocoder.lyzsys.module.ai.dal.dataobject.write.AiWriteDO; -import org.apache.ibatis.annotations.Mapper; - -/** - * AI 写作 Mapper - * - * @author xiaoxin - */ -@Mapper -public interface AiWriteMapper extends BaseMapperX { - - default PageResult selectPage(AiWritePageReqVO reqVO) { - return selectPage(reqVO, new LambdaQueryWrapperX() - .eqIfPresent(AiWriteDO::getUserId, reqVO.getUserId()) - .eqIfPresent(AiWriteDO::getType, reqVO.getType()) - .eqIfPresent(AiWriteDO::getPlatform, reqVO.getPlatform()) - .betweenIfPresent(AiWriteDO::getCreateTime, reqVO.getCreateTime()) - .orderByDesc(AiWriteDO::getId)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/AiChatRoleEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/AiChatRoleEnum.java deleted file mode 100644 index ad38d7b..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/AiChatRoleEnum.java +++ /dev/null @@ -1,50 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums; - -import lombok.AllArgsConstructor; -import lombok.Getter; - -/** - * AI 内置聊天角色的枚举 - * - * @author xiaoxin - */ -@AllArgsConstructor -@Getter -public enum AiChatRoleEnum { - - AI_WRITE_ROLE("写作助手", """ - 你是一位出色的写作助手，能够帮助用户生成创意和灵感，并在用户提供场景和提示词时生成对应的回复。你的任务包括： - 1. 撰写建议：根据用户提供的主题或问题，提供详细的写作建议、情节发展方向、角色设定以及背景描写，确保内容结构清晰、有逻辑。 - 2. 回复生成：根据用户提供的场景和提示词，生成合适的对话或文字回复，确保语气和风格符合场景需求。 - 除此之外不需要除了正文内容外的其他回复，如标题、开头、任何解释性语句或道歉。 - """), - - AI_MIND_MAP_ROLE("导图助手", """ - 你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。markdown 只需要输出一级标题，二级标题，三级标题，四级标题，最多输出四级，除此之外不要输出任何其他 markdown 标记。下面是一个合格的例子： - # Geek-AI 助手 - ## 完整的开源系统 - ### 前端开源 - ### 后端开源 - ## 支持各种大模型 - ### OpenAI - ### Azure - ### 文心一言 - ### 通义千问 - ## 集成多种收费方式 - ### 支付宝 - ### 微信 - 除此之外不要任何解释性语句。 - """), - ; - - /** - * 角色名 - */ - private final String name; - - /** - * 角色设定 - */ - private final String systemMessage; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/AiDocumentSplitStrategyEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/AiDocumentSplitStrategyEnum.java deleted file mode 100644 index ee9c0cb..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/AiDocumentSplitStrategyEnum.java +++ /dev/null @@ -1,53 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums; - -import lombok.AllArgsConstructor; -import lombok.Getter; - -/** - * AI 知识库文档切片策略枚举 - * - * @author runzhen - */ -@AllArgsConstructor -@Getter -public enum AiDocumentSplitStrategyEnum { - - /** - * 自动识别文档类型并选择最佳切片策略 - */ - AUTO("auto", "自动识别"), - - /** - * 基于 Token 数量机械切分（默认策略） - */ - TOKEN("token", "Token 切分"), - - /** - * 按段落切分（以双换行符为分隔） - */ - PARAGRAPH("paragraph", "段落切分"), - - /** - * Markdown QA 格式专用切片器 - * 识别二级标题作为问题，保持问答对完整性 - * 长答案智能切分但保留问题作为上下文 - */ - MARKDOWN_QA("markdown_qa", "Markdown QA 切分"), - - /** - * 语义化切分，保留句子完整性 - * 在段落和句子边界处切分，避免截断 - */ - SEMANTIC("semantic", "语义切分"); - - /** - * 策略代码 - */ - private final String code; - - /** - * 策略名称 - */ - private final String name; - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/DictTypeConstants.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/DictTypeConstants.java deleted file mode 100644 index 0549a3c..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/DictTypeConstants.java +++ /dev/null @@ -1,16 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums; - -/** - * AI 字典类型的枚举类 - * - * @author xiaoxin - */ -public interface DictTypeConstants { - - // ========== AI Write ========== - String AI_WRITE_FORMAT = "ai_write_format"; // 写作格式 - String AI_WRITE_LENGTH = "ai_write_length"; // 写作长度 - String AI_WRITE_LANGUAGE = "ai_write_language"; // 写作语言 - String AI_WRITE_TONE = "ai_write_tone"; // 写作语气 - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/ErrorCodeConstants.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/ErrorCodeConstants.java deleted file mode 100644 index 2f26d77..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/ErrorCodeConstants.java +++ /dev/null @@ -1,68 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums; - -import cn.iocoder.lyzsys.framework.common.exception.ErrorCode; - -/** - * AI 错误码枚举类 - *

- * ai 系统，使用 1-040-000-000 段 - */ -public interface ErrorCodeConstants { - - // ========== API 密钥 1-040-000-000 ========== - ErrorCode API_KEY_NOT_EXISTS = new ErrorCode(1_040_000_000, "API 密钥不存在"); - ErrorCode API_KEY_DISABLE = new ErrorCode(1_040_000_001, "API 密钥已禁用！"); - - // ========== API 模型 1-040-001-000 ========== - ErrorCode MODEL_NOT_EXISTS = new ErrorCode(1_040_001_000, "模型不存在!"); - ErrorCode MODEL_DISABLE = new ErrorCode(1_040_001_001, "模型({})已禁用!"); - ErrorCode MODEL_DEFAULT_NOT_EXISTS = new ErrorCode(1_040_001_002, "操作失败，找不到默认模型"); - ErrorCode MODEL_USE_TYPE_ERROR = new ErrorCode(1_040_001_003, "操作失败，该模型的模型类型不正确"); - - // ========== API 聊天角色 1-040-002-000 ========== - ErrorCode CHAT_ROLE_NOT_EXISTS = new ErrorCode(1_040_002_000, "聊天角色不存在"); - ErrorCode CHAT_ROLE_DISABLE = new ErrorCode(1_040_001_001, "聊天角色({})已禁用!"); - - // ========== API 聊天会话 1-040-003-000 ========== - ErrorCode CHAT_CONVERSATION_NOT_EXISTS = new ErrorCode(1_040_003_000, "对话不存在!"); - ErrorCode CHAT_CONVERSATION_MODEL_ERROR = new ErrorCode(1_040_003_001, "操作失败，该聊天模型的配置不完整"); - - // ========== API 聊天消息 1-040-004-000 ========== - ErrorCode CHAT_MESSAGE_NOT_EXIST = new ErrorCode(1_040_004_000, "消息不存在!"); - ErrorCode CHAT_STREAM_ERROR = new ErrorCode(1_040_004_001, "对话生成异常!"); - - // ========== API 绘画 1-040-005-000 ========== - ErrorCode IMAGE_NOT_EXISTS = new ErrorCode(1_040_005_000, "图片不存在!"); - ErrorCode IMAGE_MIDJOURNEY_SUBMIT_FAIL = new ErrorCode(1_040_005_001, "Midjourney 提交失败!原因：{}"); - ErrorCode IMAGE_CUSTOM_ID_NOT_EXISTS = new ErrorCode(1_040_005_002, "Midjourney 按钮 customId 不存在! {}"); - - // ========== API 音乐 1-040-006-000 ========== - ErrorCode MUSIC_NOT_EXISTS = new ErrorCode(1_040_006_000, "音乐不存在!"); - - // ========== API 写作 1-040-007-000 ========== - ErrorCode WRITE_NOT_EXISTS = new ErrorCode(1_040_007_000, "作文不存在!"); - ErrorCode WRITE_STREAM_ERROR = new ErrorCode(1_040_07_001, "写作生成异常!"); - - // ========== API 思维导图 1-040-008-000 ========== - ErrorCode MIND_MAP_NOT_EXISTS = new ErrorCode(1_040_008_000, "思维导图不存在!"); - - // ========== API 知识库 1-040-009-000 ========== - ErrorCode KNOWLEDGE_NOT_EXISTS = new ErrorCode(1_040_009_000, "知识库不存在!"); - - ErrorCode KNOWLEDGE_DOCUMENT_NOT_EXISTS = new ErrorCode(1_040_009_101, "文档不存在!"); - ErrorCode KNOWLEDGE_DOCUMENT_FILE_EMPTY = new ErrorCode(1_040_009_102, "文档内容为空!"); - ErrorCode KNOWLEDGE_DOCUMENT_FILE_DOWNLOAD_FAIL = new ErrorCode(1_040_009_102, "文件下载失败!"); - ErrorCode KNOWLEDGE_DOCUMENT_FILE_READ_FAIL = new ErrorCode(1_040_009_102, "文档加载失败!"); - - ErrorCode KNOWLEDGE_SEGMENT_NOT_EXISTS = new ErrorCode(1_040_009_202, "段落不存在!"); - ErrorCode KNOWLEDGE_SEGMENT_CONTENT_TOO_LONG = new ErrorCode(1_040_009_203, "内容 Token 数为 {}，超过最大限制 {}"); - - // ========== AI 工具 1-040-010-000 ========== - ErrorCode TOOL_NOT_EXISTS = new ErrorCode(1_040_010_000, "工具不存在"); - ErrorCode TOOL_NAME_NOT_EXISTS = new ErrorCode(1_040_010_001, "工具({})找不到 Bean"); - - // ========== AI 工作流 1-040-011-000 ========== - ErrorCode WORKFLOW_NOT_EXISTS = new ErrorCode(1_040_011_000, "工作流不存在"); - ErrorCode WORKFLOW_CODE_EXISTS = new ErrorCode(1_040_011_001, "工作流标识已存在"); - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/image/AiImageStatusEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/image/AiImageStatusEnum.java deleted file mode 100644 index 69eceb9..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/image/AiImageStatusEnum.java +++ /dev/null @@ -1,37 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums.image; - -import lombok.AllArgsConstructor; -import lombok.Getter; - -/** - * AI 绘画状态的枚举 - * - * @author fansili - */ -@AllArgsConstructor -@Getter -public enum AiImageStatusEnum { - - IN_PROGRESS(10, "进行中"), - SUCCESS(20, "已完成"), - FAIL(30, "已失败"); - - /** - * 状态 - */ - private final Integer status; - /** - * 状态名 - */ - private final String name; - - public static AiImageStatusEnum valueOfStatus(Integer status) { - for (AiImageStatusEnum statusEnum : AiImageStatusEnum.values()) { - if (statusEnum.getStatus().equals(status)) { - return statusEnum; - } - } - throw new IllegalArgumentException("未知会话状态： " + status); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/model/AiModelTypeEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/model/AiModelTypeEnum.java deleted file mode 100644 index fb82cb2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/model/AiModelTypeEnum.java +++ /dev/null @@ -1,41 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums.model; - -import cn.iocoder.lyzsys.framework.common.core.ArrayValuable; -import lombok.Getter; -import lombok.RequiredArgsConstructor; - -import java.util.Arrays; - -/** - * AI 模型类型的枚举 - * - * @author 芋道源码 - */ -@Getter -@RequiredArgsConstructor -public enum AiModelTypeEnum implements ArrayValuable { - - CHAT(1, "对话"), - IMAGE(2, "图片"), - VOICE(3, "语音"), - VIDEO(4, "视频"), - EMBEDDING(5, "向量"), - RERANK(6, "重排序"); - - /** - * 类型 - */ - private final Integer type; - /** - * 类型名 - */ - private final String name; - - public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiModelTypeEnum::getType).toArray(Integer[]::new); - - @Override - public Integer[] array() { - return ARRAYS; - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/model/AiPlatformEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/model/AiPlatformEnum.java deleted file mode 100644 index 2f1e611..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/model/AiPlatformEnum.java +++ /dev/null @@ -1,72 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums.model; - -import cn.iocoder.lyzsys.framework.common.core.ArrayValuable; -import lombok.AllArgsConstructor; -import lombok.Getter; - -import java.util.Arrays; - -/** - * AI 模型平台 - * - * @author fansili - */ -@Getter -@AllArgsConstructor -public enum AiPlatformEnum implements ArrayValuable { - - // ========== 国内平台 ========== - - TONG_YI("TongYi", "通义千问"), // 阿里 - YI_YAN("YiYan", "文心一言"), // 百度 - DEEP_SEEK("DeepSeek", "DeepSeek"), // DeepSeek - ZHI_PU("ZhiPu", "智谱"), // 智谱 AI - XING_HUO("XingHuo", "星火"), // 讯飞 - DOU_BAO("DouBao", "豆包"), // 字节 - HUN_YUAN("HunYuan", "混元"), // 腾讯 - SILICON_FLOW("SiliconFlow", "硅基流动"), // 硅基流动 - MINI_MAX("MiniMax", "MiniMax"), // 稀宇科技 - MOONSHOT("Moonshot", "月之暗面"), // KIMI - BAI_CHUAN("BaiChuan", "百川智能"), // 百川智能 - - // ========== 国外平台 ========== - - OPENAI("OpenAI", "OpenAI"), // OpenAI 官方 - AZURE_OPENAI("AzureOpenAI", "AzureOpenAI"), // OpenAI 微软 - ANTHROPIC("Anthropic", "Anthropic"), // Anthropic Claude - GEMINI("Gemini", "Gemini"), // 谷歌 Gemini - OLLAMA("Ollama", "Ollama"), - - STABLE_DIFFUSION("StableDiffusion", "StableDiffusion"), // Stability AI - MIDJOURNEY("Midjourney", "Midjourney"), // Midjourney - SUNO("Suno", "Suno"), // Suno AI - GROK("Grok","Grok"), // Grok - - ; - - /** - * 平台 - */ - private final String platform; - /** - * 平台名 - */ - private final String name; - - public static final String[] ARRAYS = Arrays.stream(values()).map(AiPlatformEnum::getPlatform).toArray(String[]::new); - - public static AiPlatformEnum validatePlatform(String platform) { - for (AiPlatformEnum platformEnum : AiPlatformEnum.values()) { - if (platformEnum.getPlatform().equals(platform)) { - return platformEnum; - } - } - throw new IllegalArgumentException("非法平台： " + platform); - } - - @Override - public String[] array() { - return ARRAYS; - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/music/AiMusicGenerateModeEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/music/AiMusicGenerateModeEnum.java deleted file mode 100644 index 5dba6a3..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/music/AiMusicGenerateModeEnum.java +++ /dev/null @@ -1,37 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums.music; - -import cn.iocoder.lyzsys.framework.common.core.ArrayValuable; -import lombok.AllArgsConstructor; -import lombok.Getter; - -import java.util.Arrays; - -/** - * AI 音乐生成模式的枚举 - * - * @author xiaoxin - */ -@AllArgsConstructor -@Getter -public enum AiMusicGenerateModeEnum implements ArrayValuable { - - DESCRIPTION(1, "描述模式"), - LYRIC(2, "歌词模式"); - - /** - * 模式 - */ - private final Integer mode; - /** - * 模式名 - */ - private final String name; - - public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiMusicGenerateModeEnum::getMode).toArray(Integer[]::new); - - @Override - public Integer[] array() { - return ARRAYS; - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/music/AiMusicStatusEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/music/AiMusicStatusEnum.java deleted file mode 100644 index c89daa9..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/music/AiMusicStatusEnum.java +++ /dev/null @@ -1,39 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums.music; - -import cn.iocoder.lyzsys.framework.common.core.ArrayValuable; -import lombok.AllArgsConstructor; -import lombok.Getter; - -import java.util.Arrays; - -/** - * AI 音乐状态的枚举 - * - * @author xiaoxin - */ -@AllArgsConstructor -@Getter -public enum AiMusicStatusEnum implements ArrayValuable { - - IN_PROGRESS(10, "进行中"), - SUCCESS(20, "已完成"), - FAIL(30, "已失败"); - - /** - * 状态 - */ - private final Integer status; - - /** - * 状态名 - */ - private final String name; - - public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiMusicStatusEnum::getStatus).toArray(Integer[]::new); - - @Override - public Integer[] array() { - return ARRAYS; - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/write/AiWriteTypeEnum.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/write/AiWriteTypeEnum.java deleted file mode 100644 index 2ea894e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/enums/write/AiWriteTypeEnum.java +++ /dev/null @@ -1,42 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.enums.write; - -import cn.iocoder.lyzsys.framework.common.core.ArrayValuable; -import lombok.AllArgsConstructor; -import lombok.Getter; - -import java.util.Arrays; - -/** - * AI 写作类型的枚举 - * - * @author xiaoxin - */ -@AllArgsConstructor -@Getter -public enum AiWriteTypeEnum implements ArrayValuable { - - WRITING(1, "撰写", "请撰写一篇关于 [{}] 的文章。文章的内容格式：{}，语气：{}，语言：{}，长度：{}。请确保涵盖主要内容，不需要除了正文内容外的其他回复，如标题、额外的解释或道歉。"), - REPLY(2, "回复", "请针对如下内容：[{}] 做个回复。回复内容参考：[{}], 回复格式：{}，语气：{}，语言：{}，长度：{}。不需要除了正文内容外的其他回复，如标题、开头、额外的解释或道歉。"); - - /** - * 类型 - */ - private final Integer type; - /** - * 类型名 - */ - private final String name; - - /** - * 模版 - */ - private final String prompt; - - public static final Integer[] ARRAYS = Arrays.stream(values()).map(AiWriteTypeEnum::getType).toArray(Integer[]::new); - - @Override - public Integer[] array() { - return ARRAYS; - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/config/AiAutoConfiguration.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/config/AiAutoConfiguration.java deleted file mode 100644 index e305e3f..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/config/AiAutoConfiguration.java +++ /dev/null @@ -1,323 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.config; - -import cn.hutool.core.util.StrUtil; -import cn.hutool.extra.spring.SpringUtil; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.AiModelFactory; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.AiModelFactoryImpl; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.baichuan.BaiChuanChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.doubao.DouBaoChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.gemini.GeminiChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.grok.GrokChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.hunyuan.HunYuanChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.siliconflow.SiliconFlowApiConstants; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.siliconflow.SiliconFlowChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.suno.api.SunoApi; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.xinghuo.XingHuoChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.webserch.AiWebSearchClient; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.webserch.bocha.AiBoChaWebSearchClient; -import cn.iocoder.lyzsys.module.ai.tool.method.PersonService; -import io.micrometer.observation.ObservationRegistry; -import lombok.extern.slf4j.Slf4j; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.deepseek.DeepSeekChatModel; -import org.springframework.ai.deepseek.DeepSeekChatOptions; -import org.springframework.ai.deepseek.api.DeepSeekApi; -import org.springframework.ai.embedding.BatchingStrategy; -import org.springframework.ai.embedding.TokenCountBatchingStrategy; -import org.springframework.ai.model.tool.ToolCallingManager; -import org.springframework.ai.openai.OpenAiChatModel; -import org.springframework.ai.openai.OpenAiChatOptions; -import org.springframework.ai.openai.api.OpenAiApi; -import org.springframework.ai.support.ToolCallbacks; -import org.springframework.ai.tokenizer.JTokkitTokenCountEstimator; -import org.springframework.ai.tokenizer.TokenCountEstimator; -import org.springframework.ai.tool.ToolCallback; -import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusServiceClientProperties; -import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusVectorStoreProperties; -import org.springframework.ai.vectorstore.qdrant.autoconfigure.QdrantVectorStoreProperties; -import org.springframework.ai.vectorstore.redis.autoconfigure.RedisVectorStoreProperties; -import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean; -import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; -import org.springframework.boot.context.properties.EnableConfigurationProperties; -import org.springframework.context.annotation.Bean; -import org.springframework.context.annotation.Configuration; - -import java.util.List; -import java.util.Optional; - -/** - * 芋道 AI 自动配置 - * - * @author fansili - */ -@Configuration -@EnableConfigurationProperties({ LyzsysAiProperties.class, - QdrantVectorStoreProperties.class, // 解析 Qdrant 配置 - RedisVectorStoreProperties.class, // 解析 Redis 配置 - MilvusVectorStoreProperties.class, MilvusServiceClientProperties.class // 解析 Milvus 配置 -}) -@Slf4j -public class AiAutoConfiguration { - - @Bean - public AiModelFactory aiModelFactory() { - return new AiModelFactoryImpl(); - } - - @Bean - @ConditionalOnMissingBean - public ObservationRegistry observationRegistry() { - // 特殊：兜底有 ObservationRegistry Bean，避免相关的 ChatModel 创建报错。相关 issue：https://t.zsxq.com/CuPu4 - return ObservationRegistry.NOOP; - } - - // ========== 各种 AI Client 创建 ========== - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.gemini.enable", havingValue = "true") - public GeminiChatModel geminiChatModel(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.Gemini properties = lyzsysAiProperties.getGemini(); - return buildGeminiChatClient(properties); - } - - public GeminiChatModel buildGeminiChatClient(LyzsysAiProperties.Gemini properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(GeminiChatModel.MODEL_DEFAULT); - } - OpenAiChatModel openAiChatModel = OpenAiChatModel.builder() - .openAiApi(OpenAiApi.builder() - .baseUrl(GeminiChatModel.BASE_URL) - .completionsPath(GeminiChatModel.COMPLETE_PATH) - .apiKey(properties.getApiKey()) - .build()) - .defaultOptions(OpenAiChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - .toolCallingManager(getToolCallingManager()) - .build(); - return new GeminiChatModel(openAiChatModel); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.doubao.enable", havingValue = "true") - public DouBaoChatModel douBaoChatClient(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.DouBao properties = lyzsysAiProperties.getDoubao(); - return buildDouBaoChatClient(properties); - } - - public DouBaoChatModel buildDouBaoChatClient(LyzsysAiProperties.DouBao properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(DouBaoChatModel.MODEL_DEFAULT); - } - OpenAiChatModel openAiChatModel = OpenAiChatModel.builder() - .openAiApi(OpenAiApi.builder() - .baseUrl(DouBaoChatModel.BASE_URL) - .completionsPath(DouBaoChatModel.COMPLETE_PATH) - .apiKey(properties.getApiKey()) - .build()) - .defaultOptions(OpenAiChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - .toolCallingManager(getToolCallingManager()) - .build(); - return new DouBaoChatModel(openAiChatModel); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.siliconflow.enable", havingValue = "true") - public SiliconFlowChatModel siliconFlowChatClient(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.SiliconFlow properties = lyzsysAiProperties.getSiliconflow(); - return buildSiliconFlowChatClient(properties); - } - - public SiliconFlowChatModel buildSiliconFlowChatClient(LyzsysAiProperties.SiliconFlow properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(SiliconFlowApiConstants.MODEL_DEFAULT); - } - DeepSeekChatModel openAiChatModel = DeepSeekChatModel.builder() - .deepSeekApi(DeepSeekApi.builder() - .baseUrl(SiliconFlowApiConstants.DEFAULT_BASE_URL) - .apiKey(properties.getApiKey()) - .build()) - .defaultOptions(DeepSeekChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - .toolCallingManager(getToolCallingManager()) - .build(); - return new SiliconFlowChatModel(openAiChatModel); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.hunyuan.enable", havingValue = "true") - public HunYuanChatModel hunYuanChatClient(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.HunYuan properties = lyzsysAiProperties.getHunyuan(); - return buildHunYuanChatClient(properties); - } - - public HunYuanChatModel buildHunYuanChatClient(LyzsysAiProperties.HunYuan properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(HunYuanChatModel.MODEL_DEFAULT); - } - // 特殊：由于混元大模型不提供 deepseek，而是通过知识引擎，所以需要区分下 URL - if (StrUtil.isEmpty(properties.getBaseUrl())) { - properties.setBaseUrl( - StrUtil.startWithIgnoreCase(properties.getModel(), "deepseek") ? HunYuanChatModel.DEEP_SEEK_BASE_URL - : HunYuanChatModel.BASE_URL); - } - // 创建 DeepSeekChatModel、HunYuanChatModel 对象 - DeepSeekChatModel openAiChatModel = DeepSeekChatModel.builder() - .deepSeekApi(DeepSeekApi.builder() - .baseUrl(properties.getBaseUrl()) - .completionsPath(HunYuanChatModel.COMPLETE_PATH) - .apiKey(properties.getApiKey()) - .build()) - .defaultOptions(DeepSeekChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - .toolCallingManager(getToolCallingManager()) - .build(); - return new HunYuanChatModel(openAiChatModel); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.xinghuo.enable", havingValue = "true") - public XingHuoChatModel xingHuoChatClient(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.XingHuo properties = lyzsysAiProperties.getXinghuo(); - return buildXingHuoChatClient(properties); - } - - public XingHuoChatModel buildXingHuoChatClient(LyzsysAiProperties.XingHuo properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(XingHuoChatModel.MODEL_DEFAULT); - } - OpenAiApi.Builder builder = OpenAiApi.builder() - .baseUrl(XingHuoChatModel.BASE_URL_V1) - .apiKey(properties.getAppKey() + ":" + properties.getSecretKey()); - if ("x1".equals(properties.getModel())) { - builder.baseUrl(XingHuoChatModel.BASE_URL_V2) - .completionsPath(XingHuoChatModel.BASE_COMPLETIONS_PATH_V2); - } - OpenAiChatModel openAiChatModel = OpenAiChatModel.builder() - .openAiApi(builder.build()) - .defaultOptions(OpenAiChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - // TODO @芋艿：星火的 function call 有 bug，会报 ToolResponseMessage must have an id 错误！！！ - .toolCallingManager(getToolCallingManager()) - .build(); - return new XingHuoChatModel(openAiChatModel); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.baichuan.enable", havingValue = "true") - public BaiChuanChatModel baiChuanChatClient(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.BaiChuan properties = lyzsysAiProperties.getBaichuan(); - return buildBaiChuanChatClient(properties); - } - - public BaiChuanChatModel buildBaiChuanChatClient(LyzsysAiProperties.BaiChuan properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(BaiChuanChatModel.MODEL_DEFAULT); - } - OpenAiChatModel openAiChatModel = OpenAiChatModel.builder() - .openAiApi(OpenAiApi.builder() - .baseUrl(BaiChuanChatModel.BASE_URL) - .apiKey(properties.getApiKey()) - .build()) - .defaultOptions(OpenAiChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - .toolCallingManager(getToolCallingManager()) - .build(); - return new BaiChuanChatModel(openAiChatModel); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.midjourney.enable", havingValue = "true") - public MidjourneyApi midjourneyApi(LyzsysAiProperties lyzsysAiProperties) { - LyzsysAiProperties.Midjourney config = lyzsysAiProperties.getMidjourney(); - return new MidjourneyApi(config.getBaseUrl(), config.getApiKey(), config.getNotifyUrl()); - } - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.suno.enable", havingValue = "true") - public SunoApi sunoApi(LyzsysAiProperties lyzsysAiProperties) { - return new SunoApi(lyzsysAiProperties.getSuno().getBaseUrl()); - } - - public ChatModel buildGrokChatClient(LyzsysAiProperties.Grok properties) { - if (StrUtil.isEmpty(properties.getModel())) { - properties.setModel(GrokChatModel.MODEL_DEFAULT); - } - OpenAiChatModel openAiChatModel = OpenAiChatModel.builder() - .openAiApi(OpenAiApi.builder() - .baseUrl(Optional.ofNullable(properties.getBaseUrl()) - .orElse(GrokChatModel.BASE_URL)) - .completionsPath(GrokChatModel.COMPLETE_PATH) - .apiKey(properties.getApiKey()) - .build()) - .defaultOptions(OpenAiChatOptions.builder() - .model(properties.getModel()) - .temperature(properties.getTemperature()) - .maxTokens(properties.getMaxTokens()) - .topP(properties.getTopP()) - .build()) - .toolCallingManager(getToolCallingManager()) - .build(); - return new DouBaoChatModel(openAiChatModel); - } - - // ========== RAG 相关 ========== - - @Bean - public TokenCountEstimator tokenCountEstimator() { - return new JTokkitTokenCountEstimator(); - } - - @Bean - public BatchingStrategy batchingStrategy() { - return new TokenCountBatchingStrategy(); - } - - private static ToolCallingManager getToolCallingManager() { - return SpringUtil.getBean(ToolCallingManager.class); - } - - // ========== Web Search 相关 ========== - - @Bean - @ConditionalOnProperty(value = "lyzsys.ai.web-search.enable", havingValue = "true") - public AiWebSearchClient webSearchClient(LyzsysAiProperties lyzsysAiProperties) { - return new AiBoChaWebSearchClient(lyzsysAiProperties.getWebSearch().getApiKey()); - } - - // ========== MCP 相关 ========== - - /** - * 参考自 MCP Server Boot Starter - */ - @Bean - public List toolCallbacks(PersonService personService) { - return List.of(ToolCallbacks.from(personService)); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/config/LyzsysAiProperties.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/config/LyzsysAiProperties.java deleted file mode 100644 index 445f07e..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/config/LyzsysAiProperties.java +++ /dev/null @@ -1,186 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.config; - -import lombok.Data; -import org.springframework.boot.context.properties.ConfigurationProperties; - -/** - * 芋道 AI 配置类 - * - * @author fansili - * @since 1.0 - */ -@ConfigurationProperties(prefix = "lyzsys.ai") -@Data -public class LyzsysAiProperties { - - /** - * 谷歌 Gemini - */ - private Gemini gemini; - - /** - * 字节豆包 - */ - private DouBao doubao; - - /** - * 腾讯混元 - */ - private HunYuan hunyuan; - - /** - * 硅基流动 - */ - private SiliconFlow siliconflow; - - /** - * 讯飞星火 - */ - private XingHuo xinghuo; - - /** - * 百川 - */ - private BaiChuan baichuan; - - /** - * Midjourney 绘图 - */ - private Midjourney midjourney; - - /** - * Suno 音乐 - */ - @SuppressWarnings("SpellCheckingInspection") - private Suno suno; - - /** - * 网络搜索 - */ - private WebSearch webSearch; - - @Data - public static class Gemini { - - private String enable; - private String apiKey; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class DouBao { - - private String enable; - private String apiKey; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class HunYuan { - - private String enable; - private String baseUrl; - private String apiKey; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class SiliconFlow { - - private String enable; - private String apiKey; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class XingHuo { - - private String enable; - private String appId; - private String appKey; - private String secretKey; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class BaiChuan { - - private String enable; - private String apiKey; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class Midjourney { - - private String enable; - private String baseUrl; - - private String apiKey; - private String notifyUrl; - - } - - @Data - public static class Suno { - - private boolean enable; - - private String baseUrl; - - } - - @Data - public static class Grok { - - private String enable; - private String apiKey; - private String baseUrl; - - private String model; - private Double temperature; - private Integer maxTokens; - private Double topP; - - } - - @Data - public static class WebSearch { - - private boolean enable; - - private String apiKey; - - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/AiModelFactory.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/AiModelFactory.java deleted file mode 100644 index f93c6b5..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/AiModelFactory.java +++ /dev/null @@ -1,113 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model; - -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.suno.api.SunoApi; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.embedding.EmbeddingModel; -import org.springframework.ai.image.ImageModel; -import org.springframework.ai.vectorstore.VectorStore; - -import java.util.Map; - -/** - * AI Model 模型工厂的接口类 - * - * @author fansili - */ -public interface AiModelFactory { - - /** - * 基于指定配置，获得 ChatModel 对象 - * - * 如果不存在，则进行创建 - * - * @param platform 平台 - * @param apiKey API KEY - * @param url API URL - * @return ChatModel 对象 - */ - ChatModel getOrCreateChatModel(AiPlatformEnum platform, String apiKey, String url); - - /** - * 基于默认配置，获得 ChatModel 对象 - * - * 默认配置，指的是在 application.yaml 配置文件中的 spring.ai 相关的配置 - * - * @param platform 平台 - * @return ChatModel 对象 - */ - ChatModel getDefaultChatModel(AiPlatformEnum platform); - - /** - * 基于默认配置，获得 ImageModel 对象 - * - * 默认配置，指的是在 application.yaml 配置文件中的 spring.ai 相关的配置 - * - * @param platform 平台 - * @return ImageModel 对象 - */ - ImageModel getDefaultImageModel(AiPlatformEnum platform); - - /** - * 基于指定配置，获得 ImageModel 对象 - * - * 如果不存在，则进行创建 - * - * @param platform 平台 - * @param apiKey API KEY - * @param url API URL - * @return ImageModel 对象 - */ - ImageModel getOrCreateImageModel(AiPlatformEnum platform, String apiKey, String url); - - /** - * 基于指定配置，获得 MidjourneyApi 对象 - * - * 如果不存在，则进行创建 - * - * @param apiKey API KEY - * @param url API URL - * @return MidjourneyApi 对象 - */ - MidjourneyApi getOrCreateMidjourneyApi(String apiKey, String url); - - /** - * 基于指定配置，获得 SunoApi 对象 - * - * 如果不存在，则进行创建 - * - * @param apiKey API KEY - * @param url API URL - * @return SunoApi 对象 - */ - SunoApi getOrCreateSunoApi(String apiKey, String url); - - /** - * 基于指定配置，获得 EmbeddingModel 对象 - * - * 如果不存在，则进行创建 - * - * @param platform 平台 - * @param apiKey API KEY - * @param url API URL - * @param model 模型 - * @return ChatModel 对象 - */ - EmbeddingModel getOrCreateEmbeddingModel(AiPlatformEnum platform, String apiKey, String url, String model); - - /** - * 基于指定配置，获得 VectorStore 对象 - * - * 如果不存在，则进行创建 - * - * @param type 向量存储类型 - * @param embeddingModel 向量模型 - * @param metadataFields 元数据字段 - * @return VectorStore 对象 - */ - VectorStore getOrCreateVectorStore(Class type, - EmbeddingModel embeddingModel, - Map> metadataFields); - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/AiModelFactoryImpl.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/AiModelFactoryImpl.java deleted file mode 100644 index 40500d2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/AiModelFactoryImpl.java +++ /dev/null @@ -1,845 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model; - -import cn.hutool.core.io.FileUtil; -import cn.hutool.core.lang.Assert; -import cn.hutool.core.lang.Singleton; -import cn.hutool.core.lang.func.Func0; -import cn.hutool.core.util.ArrayUtil; -import cn.hutool.core.util.RuntimeUtil; -import cn.hutool.core.util.StrUtil; -import cn.hutool.extra.spring.SpringUtil; -import cn.iocoder.lyzsys.framework.common.util.spring.SpringUtils; -import cn.iocoder.lyzsys.module.ai.enums.model.AiPlatformEnum; -import cn.iocoder.lyzsys.module.ai.framework.ai.config.AiAutoConfiguration; -import cn.iocoder.lyzsys.module.ai.framework.ai.config.LyzsysAiProperties; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.baichuan.BaiChuanChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.doubao.DouBaoChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.gemini.GeminiChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.hunyuan.HunYuanChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api.MidjourneyApi; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.siliconflow.SiliconFlowApiConstants; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.siliconflow.SiliconFlowChatModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.siliconflow.SiliconFlowImageApi; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.siliconflow.SiliconFlowImageModel; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.suno.api.SunoApi; -import cn.iocoder.lyzsys.module.ai.framework.ai.core.model.xinghuo.XingHuoChatModel; -import com.alibaba.cloud.ai.autoconfigure.dashscope.DashScopeChatAutoConfiguration; -import com.alibaba.cloud.ai.autoconfigure.dashscope.DashScopeEmbeddingAutoConfiguration; -import com.alibaba.cloud.ai.autoconfigure.dashscope.DashScopeImageAutoConfiguration; -import com.alibaba.cloud.ai.dashscope.api.DashScopeApi; -import com.alibaba.cloud.ai.dashscope.api.DashScopeImageApi; -import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatModel; -import com.alibaba.cloud.ai.dashscope.chat.DashScopeChatOptions; -import com.alibaba.cloud.ai.dashscope.embedding.DashScopeEmbeddingModel; -import com.alibaba.cloud.ai.dashscope.embedding.DashScopeEmbeddingOptions; -import com.alibaba.cloud.ai.dashscope.image.DashScopeImageModel; -import com.azure.ai.openai.OpenAIClientBuilder; -import com.azure.core.credential.KeyCredential; -import io.micrometer.observation.ObservationRegistry; -import io.milvus.client.MilvusServiceClient; -import io.qdrant.client.QdrantClient; -import io.qdrant.client.QdrantGrpcClient; -import lombok.SneakyThrows; -import org.springaicommunity.moonshot.MoonshotChatModel; -import org.springaicommunity.moonshot.MoonshotChatOptions; -import org.springaicommunity.moonshot.api.MoonshotApi; -import org.springaicommunity.moonshot.autoconfigure.MoonshotChatAutoConfiguration; -import org.springaicommunity.qianfan.QianFanChatModel; -import org.springaicommunity.qianfan.QianFanEmbeddingModel; -import org.springaicommunity.qianfan.QianFanEmbeddingOptions; -import org.springaicommunity.qianfan.QianFanImageModel; -import org.springaicommunity.qianfan.api.QianFanApi; -import org.springaicommunity.qianfan.api.QianFanImageApi; -import org.springaicommunity.qianfan.autoconfigure.QianFanChatAutoConfiguration; -import org.springaicommunity.qianfan.autoconfigure.QianFanEmbeddingAutoConfiguration; -import org.springframework.ai.azure.openai.AzureOpenAiChatModel; -import org.springframework.ai.azure.openai.AzureOpenAiEmbeddingModel; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.deepseek.DeepSeekChatModel; -import org.springframework.ai.deepseek.DeepSeekChatOptions; -import org.springframework.ai.deepseek.api.DeepSeekApi; -import org.springframework.ai.document.MetadataMode; -import org.springframework.ai.embedding.BatchingStrategy; -import org.springframework.ai.embedding.EmbeddingModel; -import org.springframework.ai.embedding.observation.EmbeddingModelObservationConvention; -import org.springframework.ai.image.ImageModel; -import org.springframework.ai.minimax.MiniMaxChatModel; -import org.springframework.ai.minimax.MiniMaxChatOptions; -import org.springframework.ai.minimax.MiniMaxEmbeddingModel; -import org.springframework.ai.minimax.MiniMaxEmbeddingOptions; -import org.springframework.ai.minimax.api.MiniMaxApi; -import org.springframework.ai.model.anthropic.autoconfigure.AnthropicChatAutoConfiguration; -import org.springframework.ai.model.azure.openai.autoconfigure.AzureOpenAiChatAutoConfiguration; -import org.springframework.ai.model.azure.openai.autoconfigure.AzureOpenAiEmbeddingAutoConfiguration; -import org.springframework.ai.model.azure.openai.autoconfigure.AzureOpenAiEmbeddingProperties; -import org.springframework.ai.model.deepseek.autoconfigure.DeepSeekChatAutoConfiguration; -import org.springframework.ai.model.minimax.autoconfigure.MiniMaxChatAutoConfiguration; -import org.springframework.ai.model.minimax.autoconfigure.MiniMaxEmbeddingAutoConfiguration; -import org.springframework.ai.model.ollama.autoconfigure.OllamaChatAutoConfiguration; -import org.springframework.ai.model.openai.autoconfigure.OpenAiChatAutoConfiguration; -import org.springframework.ai.model.openai.autoconfigure.OpenAiEmbeddingAutoConfiguration; -import org.springframework.ai.model.openai.autoconfigure.OpenAiImageAutoConfiguration; -import org.springframework.ai.model.stabilityai.autoconfigure.StabilityAiImageAutoConfiguration; -import org.springframework.ai.model.tool.ToolCallingManager; -import org.springframework.ai.model.zhipuai.autoconfigure.ZhiPuAiChatAutoConfiguration; -import org.springframework.ai.model.zhipuai.autoconfigure.ZhiPuAiEmbeddingAutoConfiguration; -import org.springframework.ai.model.zhipuai.autoconfigure.ZhiPuAiImageAutoConfiguration; -import org.springframework.ai.ollama.OllamaChatModel; -import org.springframework.ai.ollama.OllamaEmbeddingModel; -import org.springframework.ai.ollama.api.OllamaApi; -import org.springframework.ai.ollama.api.OllamaEmbeddingOptions; -import org.springframework.ai.openai.OpenAiChatModel; -import org.springframework.ai.openai.OpenAiEmbeddingModel; -import org.springframework.ai.openai.OpenAiEmbeddingOptions; -import org.springframework.ai.openai.OpenAiImageModel; -import org.springframework.ai.openai.api.OpenAiApi; -import org.springframework.ai.openai.api.OpenAiImageApi; -import org.springframework.ai.openai.api.common.OpenAiApiConstants; -import org.springframework.ai.anthropic.AnthropicChatModel; -import org.springframework.ai.anthropic.api.AnthropicApi; -import org.springframework.ai.stabilityai.StabilityAiImageModel; -import org.springframework.ai.stabilityai.api.StabilityAiApi; -import org.springframework.ai.vectorstore.SimpleVectorStore; -import org.springframework.ai.vectorstore.VectorStore; -import org.springframework.ai.vectorstore.milvus.MilvusVectorStore; -import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusServiceClientConnectionDetails; -import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusServiceClientProperties; -import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusVectorStoreAutoConfiguration; -import org.springframework.ai.vectorstore.milvus.autoconfigure.MilvusVectorStoreProperties; -import org.springframework.ai.vectorstore.observation.DefaultVectorStoreObservationConvention; -import org.springframework.ai.vectorstore.observation.VectorStoreObservationConvention; -import org.springframework.ai.vectorstore.qdrant.QdrantVectorStore; -import org.springframework.ai.vectorstore.qdrant.autoconfigure.QdrantVectorStoreAutoConfiguration; -import org.springframework.ai.vectorstore.qdrant.autoconfigure.QdrantVectorStoreProperties; -import org.springframework.ai.vectorstore.redis.RedisVectorStore; -import org.springframework.ai.vectorstore.redis.autoconfigure.RedisVectorStoreAutoConfiguration; -import org.springframework.ai.vectorstore.redis.autoconfigure.RedisVectorStoreProperties; -import org.springframework.ai.zhipuai.*; -import org.springframework.ai.zhipuai.api.ZhiPuAiApi; -import org.springframework.ai.zhipuai.api.ZhiPuAiImageApi; -import org.springframework.beans.BeansException; -import org.springframework.beans.factory.ObjectProvider; -import org.springframework.boot.autoconfigure.data.redis.RedisProperties; -import org.springframework.web.client.RestClient; -import redis.clients.jedis.JedisPooled; - -import java.io.File; -import java.time.Duration; -import java.util.List; -import java.util.Map; -import java.util.Timer; -import java.util.TimerTask; - -import static cn.iocoder.lyzsys.framework.common.util.collection.CollectionUtils.convertList; -import static org.springframework.ai.retry.RetryUtils.DEFAULT_RETRY_TEMPLATE; - -/** - * AI Model 模型工厂的实现类 - * - * @author 芋道源码 - */ -public class AiModelFactoryImpl implements AiModelFactory { - - @Override - public ChatModel getOrCreateChatModel(AiPlatformEnum platform, String apiKey, String url) { - String cacheKey = buildClientCacheKey(ChatModel.class, platform, apiKey, url); - return Singleton.get(cacheKey, (Func0) () -> { - // noinspection EnhancedSwitchMigration - switch (platform) { - case TONG_YI: - return buildTongYiChatModel(apiKey); - case YI_YAN: - return buildYiYanChatModel(apiKey); - case DEEP_SEEK: - return buildDeepSeekChatModel(apiKey); - case DOU_BAO: - return buildDouBaoChatModel(apiKey); - case HUN_YUAN: - return buildHunYuanChatModel(apiKey, url); - case SILICON_FLOW: - return buildSiliconFlowChatModel(apiKey); - case ZHI_PU: - return buildZhiPuChatModel(apiKey, url); - case MINI_MAX: - return buildMiniMaxChatModel(apiKey, url); - case MOONSHOT: - return buildMoonshotChatModel(apiKey, url); - case XING_HUO: - return buildXingHuoChatModel(apiKey); - case BAI_CHUAN: - return buildBaiChuanChatModel(apiKey); - case OPENAI: - return buildOpenAiChatModel(apiKey, url); - case AZURE_OPENAI: - return buildAzureOpenAiChatModel(apiKey, url); - case ANTHROPIC: - return buildAnthropicChatModel(apiKey, url); - case GEMINI: - return buildGeminiChatModel(apiKey); - case OLLAMA: - return buildOllamaChatModel(url); - case GROK: - return buildGrokChatModel(apiKey,url); - default: - throw new IllegalArgumentException(StrUtil.format("未知平台({})", platform)); - } - }); - } - - @Override - public ChatModel getDefaultChatModel(AiPlatformEnum platform) { - // noinspection EnhancedSwitchMigration - switch (platform) { - case TONG_YI: - return SpringUtil.getBean(DashScopeChatModel.class); - case YI_YAN: - return SpringUtil.getBean(QianFanChatModel.class); - case DEEP_SEEK: - return SpringUtil.getBean(DeepSeekChatModel.class); - case DOU_BAO: - return SpringUtil.getBean(DouBaoChatModel.class); - case HUN_YUAN: - return SpringUtil.getBean(HunYuanChatModel.class); - case SILICON_FLOW: - return SpringUtil.getBean(SiliconFlowChatModel.class); - case ZHI_PU: - return SpringUtil.getBean(ZhiPuAiChatModel.class); - case MINI_MAX: - return SpringUtil.getBean(MiniMaxChatModel.class); - case MOONSHOT: - return SpringUtil.getBean(MoonshotChatModel.class); - case XING_HUO: - return SpringUtil.getBean(XingHuoChatModel.class); - case BAI_CHUAN: - return SpringUtil.getBean(BaiChuanChatModel.class); - case OPENAI: - return SpringUtil.getBean(OpenAiChatModel.class); - case AZURE_OPENAI: - return SpringUtil.getBean(AzureOpenAiChatModel.class); - case ANTHROPIC: - return SpringUtil.getBean(AnthropicChatModel.class); - case GEMINI: - return SpringUtil.getBean(GeminiChatModel.class); - case OLLAMA: - return SpringUtil.getBean(OllamaChatModel.class); - default: - throw new IllegalArgumentException(StrUtil.format("未知平台({})", platform)); - } - } - - @Override - public ImageModel getDefaultImageModel(AiPlatformEnum platform) { - // noinspection EnhancedSwitchMigration - switch (platform) { - case TONG_YI: - return SpringUtil.getBean(DashScopeImageModel.class); - case YI_YAN: - return SpringUtil.getBean(QianFanImageModel.class); - case ZHI_PU: - return SpringUtil.getBean(ZhiPuAiImageModel.class); - case SILICON_FLOW: - return SpringUtil.getBean(SiliconFlowImageModel.class); - case OPENAI: - return SpringUtil.getBean(OpenAiImageModel.class); - case STABLE_DIFFUSION: - return SpringUtil.getBean(StabilityAiImageModel.class); - default: - throw new IllegalArgumentException(StrUtil.format("未知平台({})", platform)); - } - } - - @Override - public ImageModel getOrCreateImageModel(AiPlatformEnum platform, String apiKey, String url) { - // noinspection EnhancedSwitchMigration - switch (platform) { - case TONG_YI: - return buildTongYiImagesModel(apiKey); - case YI_YAN: - return buildQianFanImageModel(apiKey); - case ZHI_PU: - return buildZhiPuAiImageModel(apiKey, url); - case OPENAI: - return buildOpenAiImageModel(apiKey, url); - case SILICON_FLOW: - return buildSiliconFlowImageModel(apiKey,url); - case STABLE_DIFFUSION: - return buildStabilityAiImageModel(apiKey, url); - default: - throw new IllegalArgumentException(StrUtil.format("未知平台({})", platform)); - } - } - - @Override - public MidjourneyApi getOrCreateMidjourneyApi(String apiKey, String url) { - String cacheKey = buildClientCacheKey(MidjourneyApi.class, AiPlatformEnum.MIDJOURNEY.getPlatform(), apiKey, - url); - return Singleton.get(cacheKey, (Func0) () -> { - LyzsysAiProperties.Midjourney properties = SpringUtil.getBean(LyzsysAiProperties.class) - .getMidjourney(); - return new MidjourneyApi(url, apiKey, properties.getNotifyUrl()); - }); - } - - @Override - public SunoApi getOrCreateSunoApi(String apiKey, String url) { - String cacheKey = buildClientCacheKey(SunoApi.class, AiPlatformEnum.SUNO.getPlatform(), apiKey, url); - return Singleton.get(cacheKey, (Func0) () -> new SunoApi(url)); - } - - @Override - @SuppressWarnings("EnhancedSwitchMigration") - public EmbeddingModel getOrCreateEmbeddingModel(AiPlatformEnum platform, String apiKey, String url, String model) { - String cacheKey = buildClientCacheKey(EmbeddingModel.class, platform, apiKey, url, model); - return Singleton.get(cacheKey, (Func0) () -> { - switch (platform) { - case TONG_YI: - return buildTongYiEmbeddingModel(apiKey, model); - case YI_YAN: - return buildYiYanEmbeddingModel(apiKey, model); - case ZHI_PU: - return buildZhiPuEmbeddingModel(apiKey, url, model); - case MINI_MAX: - return buildMiniMaxEmbeddingModel(apiKey, url, model); - case OPENAI: - return buildOpenAiEmbeddingModel(apiKey, url, model); - case AZURE_OPENAI: - return buildAzureOpenAiEmbeddingModel(apiKey, url, model); - case OLLAMA: - return buildOllamaEmbeddingModel(url, model); - default: - throw new IllegalArgumentException(StrUtil.format("未知平台({})", platform)); - } - }); - } - - @Override - public VectorStore getOrCreateVectorStore(Class type, - EmbeddingModel embeddingModel, - Map> metadataFields) { - String cacheKey = buildClientCacheKey(VectorStore.class, embeddingModel, type); - return Singleton.get(cacheKey, (Func0) () -> { - if (type == SimpleVectorStore.class) { - return buildSimpleVectorStore(embeddingModel); - } - if (type == QdrantVectorStore.class) { - return buildQdrantVectorStore(embeddingModel); - } - if (type == RedisVectorStore.class) { - return buildRedisVectorStore(embeddingModel, metadataFields); - } - if (type == MilvusVectorStore.class) { - return buildMilvusVectorStore(embeddingModel); - } - throw new IllegalArgumentException(StrUtil.format("未知类型({})", type)); - }); - } - - private static String buildClientCacheKey(Class clazz, Object... params) { - if (ArrayUtil.isEmpty(params)) { - return clazz.getName(); - } - return StrUtil.format("{}#{}", clazz.getName(), ArrayUtil.join(params, "_")); - } - - // ========== 各种创建 spring-ai 客户端的方法 ========== - - /** - * 可参考 {@link DashScopeChatAutoConfiguration} 的 dashscopeChatModel 方法 - */ - private static DashScopeChatModel buildTongYiChatModel(String key) { - DashScopeApi dashScopeApi = DashScopeApi.builder().apiKey(key).build(); - DashScopeChatOptions options = DashScopeChatOptions.builder().withModel(DashScopeApi.DEFAULT_CHAT_MODEL) - .withTemperature(0.7).build(); - return DashScopeChatModel.builder() - .dashScopeApi(dashScopeApi) - .defaultOptions(options) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link DashScopeImageAutoConfiguration} 的 dashScopeImageModel 方法 - */ - private static DashScopeImageModel buildTongYiImagesModel(String key) { - DashScopeImageApi dashScopeImageApi = DashScopeImageApi.builder().apiKey(key).build(); - return DashScopeImageModel.builder() - .dashScopeApi(dashScopeImageApi) - .build(); - } - - /** - * 可参考 {@link QianFanChatAutoConfiguration} 的 qianFanChatModel 方法 - */ - private static QianFanChatModel buildYiYanChatModel(String key) { - // TODO spring ai qianfan 有 bug，无法使用 https://github.com/spring-ai-community/qianfan/issues/6 - List keys = StrUtil.split(key, '|'); - Assert.equals(keys.size(), 2, "YiYanChatClient 的密钥需要 (appKey|secretKey) 格式"); - String appKey = keys.get(0); - String secretKey = keys.get(1); - QianFanApi qianFanApi = new QianFanApi(appKey, secretKey); - return new QianFanChatModel(qianFanApi); - } - - /** - * 可参考 {@link QianFanEmbeddingAutoConfiguration} 的 qianFanImageModel 方法 - */ - private QianFanImageModel buildQianFanImageModel(String key) { - // TODO spring ai qianfan 有 bug，无法使用 https://github.com/spring-ai-community/qianfan/issues/6 - List keys = StrUtil.split(key, '|'); - Assert.equals(keys.size(), 2, "YiYanChatClient 的密钥需要 (appKey|secretKey) 格式"); - String appKey = keys.get(0); - String secretKey = keys.get(1); - QianFanImageApi qianFanApi = new QianFanImageApi(appKey, secretKey); - return new QianFanImageModel(qianFanApi); - } - - /** - * 可参考 {@link DeepSeekChatAutoConfiguration} 的 deepSeekChatModel 方法 - */ - private static DeepSeekChatModel buildDeepSeekChatModel(String apiKey) { - DeepSeekApi deepSeekApi = DeepSeekApi.builder().apiKey(apiKey).build(); - DeepSeekChatOptions options = DeepSeekChatOptions.builder().model(DeepSeekApi.DEFAULT_CHAT_MODEL) - .temperature(0.7).build(); - return DeepSeekChatModel.builder() - .deepSeekApi(deepSeekApi) - .defaultOptions(options) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link AiAutoConfiguration#douBaoChatClient(LyzsysAiProperties)} - */ - private ChatModel buildDouBaoChatModel(String apiKey) { - LyzsysAiProperties.DouBao properties = new LyzsysAiProperties.DouBao() - .setApiKey(apiKey); - return new AiAutoConfiguration().buildDouBaoChatClient(properties); - } - - /** - * 可参考 {@link AiAutoConfiguration#hunYuanChatClient(LyzsysAiProperties)} - */ - private ChatModel buildHunYuanChatModel(String apiKey, String url) { - LyzsysAiProperties.HunYuan properties = new LyzsysAiProperties.HunYuan() - .setBaseUrl(url).setApiKey(apiKey); - return new AiAutoConfiguration().buildHunYuanChatClient(properties); - } - - /** - * 可参考 {@link AiAutoConfiguration#siliconFlowChatClient(LyzsysAiProperties)} - */ - private ChatModel buildSiliconFlowChatModel(String apiKey) { - LyzsysAiProperties.SiliconFlow properties = new LyzsysAiProperties.SiliconFlow() - .setApiKey(apiKey); - return new AiAutoConfiguration().buildSiliconFlowChatClient(properties); - } - - /** - * 可参考 {@link ZhiPuAiChatAutoConfiguration} 的 zhiPuAiChatModel 方法 - */ - private ZhiPuAiChatModel buildZhiPuChatModel(String apiKey, String url) { - ZhiPuAiApi.Builder zhiPuAiApiBuilder = ZhiPuAiApi.builder().apiKey(apiKey); - if (StrUtil.isNotEmpty(url)) { - zhiPuAiApiBuilder.baseUrl(url); - } - ZhiPuAiChatOptions options = ZhiPuAiChatOptions.builder().model(ZhiPuAiApi.DEFAULT_CHAT_MODEL).temperature(0.7).build(); - return new ZhiPuAiChatModel(zhiPuAiApiBuilder.build(), options, getToolCallingManager(), DEFAULT_RETRY_TEMPLATE, - getObservationRegistry().getIfAvailable()); - } - - /** - * 可参考 {@link ZhiPuAiImageAutoConfiguration} 的 zhiPuAiImageModel 方法 - */ - private ZhiPuAiImageModel buildZhiPuAiImageModel(String apiKey, String url) { - ZhiPuAiImageApi zhiPuAiApi = StrUtil.isEmpty(url) ? new ZhiPuAiImageApi(apiKey) - : new ZhiPuAiImageApi(url, apiKey, RestClient.builder()); - return new ZhiPuAiImageModel(zhiPuAiApi); - } - - /** - * 可参考 {@link MiniMaxChatAutoConfiguration} 的 miniMaxChatModel 方法 - */ - private MiniMaxChatModel buildMiniMaxChatModel(String apiKey, String url) { - MiniMaxApi miniMaxApi = StrUtil.isEmpty(url) ? new MiniMaxApi(apiKey) - : new MiniMaxApi(url, apiKey); - MiniMaxChatOptions options = MiniMaxChatOptions.builder().model(MiniMaxApi.DEFAULT_CHAT_MODEL).temperature(0.7).build(); - return new MiniMaxChatModel(miniMaxApi, options, getToolCallingManager(), DEFAULT_RETRY_TEMPLATE); - } - - /** - * 可参考 {@link MoonshotChatAutoConfiguration} 的 moonshotChatModel 方法 - */ - private MoonshotChatModel buildMoonshotChatModel(String apiKey, String url) { - MoonshotApi.Builder moonshotApiBuilder = MoonshotApi.builder() - .apiKey(apiKey); - if (StrUtil.isNotEmpty(url)) { - moonshotApiBuilder.baseUrl(url); - } - MoonshotChatOptions options = MoonshotChatOptions.builder().model(MoonshotApi.DEFAULT_CHAT_MODEL).build(); - return MoonshotChatModel.builder() - .moonshotApi(moonshotApiBuilder.build()) - .defaultOptions(options) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link AiAutoConfiguration#xingHuoChatClient(LyzsysAiProperties)} - */ - private static XingHuoChatModel buildXingHuoChatModel(String key) { - List keys = StrUtil.split(key, '|'); - Assert.equals(keys.size(), 2, "XingHuoChatClient 的密钥需要 (appKey|secretKey) 格式"); - LyzsysAiProperties.XingHuo properties = new LyzsysAiProperties.XingHuo() - .setAppKey(keys.get(0)).setSecretKey(keys.get(1)); - return new AiAutoConfiguration().buildXingHuoChatClient(properties); - } - - /** - * 可参考 {@link AiAutoConfiguration#baiChuanChatClient(LyzsysAiProperties)} - */ - private BaiChuanChatModel buildBaiChuanChatModel(String apiKey) { - LyzsysAiProperties.BaiChuan properties = new LyzsysAiProperties.BaiChuan() - .setApiKey(apiKey); - return new AiAutoConfiguration().buildBaiChuanChatClient(properties); - } - - /** - * 可参考 {@link OpenAiChatAutoConfiguration} 的 openAiChatModel 方法 - */ - private static OpenAiChatModel buildOpenAiChatModel(String openAiToken, String url) { - url = StrUtil.blankToDefault(url, OpenAiApiConstants.DEFAULT_BASE_URL); - OpenAiApi openAiApi = OpenAiApi.builder().baseUrl(url).apiKey(openAiToken).build(); - return OpenAiChatModel.builder() - .openAiApi(openAiApi) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link AzureOpenAiChatAutoConfiguration} - */ - private static AzureOpenAiChatModel buildAzureOpenAiChatModel(String apiKey, String url) { - // TODO @芋艿：使用前，请测试，暂时没密钥！！！ - OpenAIClientBuilder openAIClientBuilder = new OpenAIClientBuilder() - .endpoint(url).credential(new KeyCredential(apiKey)); - return AzureOpenAiChatModel.builder() - .openAIClientBuilder(openAIClientBuilder) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link AnthropicChatAutoConfiguration} 的 anthropicApi 方法 - */ - private static AnthropicChatModel buildAnthropicChatModel(String apiKey, String url) { - AnthropicApi.Builder builder = AnthropicApi.builder().apiKey(apiKey); - if (StrUtil.isNotEmpty(url)) { - builder.baseUrl(url); - } - AnthropicApi anthropicApi = builder.build(); - return AnthropicChatModel.builder() - .anthropicApi(anthropicApi) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link AiAutoConfiguration#buildGeminiChatClient(LyzsysAiProperties.Gemini)} - */ - private static GeminiChatModel buildGeminiChatModel(String apiKey) { - LyzsysAiProperties.Gemini properties = SpringUtil.getBean(LyzsysAiProperties.class) - .getGemini().setApiKey(apiKey); - return new AiAutoConfiguration().buildGeminiChatClient(properties); - } - - /** - * 可参考 {@link OpenAiImageAutoConfiguration} 的 openAiImageModel 方法 - */ - private OpenAiImageModel buildOpenAiImageModel(String openAiToken, String url) { - url = StrUtil.blankToDefault(url, OpenAiApiConstants.DEFAULT_BASE_URL); - OpenAiImageApi openAiApi = OpenAiImageApi.builder().baseUrl(url).apiKey(openAiToken).build(); - return new OpenAiImageModel(openAiApi); - } - - /** - * 创建 SiliconFlowImageModel 对象 - */ - private SiliconFlowImageModel buildSiliconFlowImageModel(String apiToken, String url) { - url = StrUtil.blankToDefault(url, SiliconFlowApiConstants.DEFAULT_BASE_URL); - SiliconFlowImageApi openAiApi = new SiliconFlowImageApi(url, apiToken); - return new SiliconFlowImageModel(openAiApi); - } - - /** - * 可参考 {@link OllamaChatAutoConfiguration} 的 ollamaChatModel 方法 - */ - private static OllamaChatModel buildOllamaChatModel(String url) { - OllamaApi ollamaApi = OllamaApi.builder().baseUrl(url).build(); - return OllamaChatModel.builder() - .ollamaApi(ollamaApi) - .toolCallingManager(getToolCallingManager()) - .build(); - } - - /** - * 可参考 {@link StabilityAiImageAutoConfiguration} 的 stabilityAiImageModel 方法 - */ - private StabilityAiImageModel buildStabilityAiImageModel(String apiKey, String url) { - url = StrUtil.blankToDefault(url, StabilityAiApi.DEFAULT_BASE_URL); - StabilityAiApi stabilityAiApi = new StabilityAiApi(apiKey, StabilityAiApi.DEFAULT_IMAGE_MODEL, url); - return new StabilityAiImageModel(stabilityAiApi); - } - - private ChatModel buildGrokChatModel(String apiKey,String url) { - LyzsysAiProperties.Grok properties = new LyzsysAiProperties.Grok() - .setBaseUrl(url) - .setApiKey(apiKey); - return new AiAutoConfiguration().buildGrokChatClient(properties); - } - - // ========== 各种创建 EmbeddingModel 的方法 ========== - - /** - * 可参考 {@link DashScopeEmbeddingAutoConfiguration} 的 dashscopeEmbeddingModel 方法 - */ - private DashScopeEmbeddingModel buildTongYiEmbeddingModel(String apiKey, String model) { - DashScopeApi dashScopeApi = DashScopeApi.builder().apiKey(apiKey).build(); - DashScopeEmbeddingOptions dashScopeEmbeddingOptions = DashScopeEmbeddingOptions.builder().withModel(model).build(); - return new DashScopeEmbeddingModel(dashScopeApi, MetadataMode.EMBED, dashScopeEmbeddingOptions); - } - - /** - * 可参考 {@link ZhiPuAiEmbeddingAutoConfiguration} 的 zhiPuAiEmbeddingModel 方法 - */ - private ZhiPuAiEmbeddingModel buildZhiPuEmbeddingModel(String apiKey, String url, String model) { - ZhiPuAiApi.Builder zhiPuAiApiBuilder = ZhiPuAiApi.builder().apiKey(apiKey); - if (StrUtil.isNotEmpty(url)) { - zhiPuAiApiBuilder.baseUrl(url); - } - ZhiPuAiEmbeddingOptions zhiPuAiEmbeddingOptions = ZhiPuAiEmbeddingOptions.builder().model(model).build(); - return new ZhiPuAiEmbeddingModel(zhiPuAiApiBuilder.build(), MetadataMode.EMBED, zhiPuAiEmbeddingOptions); - } - - /** - * 可参考 {@link MiniMaxEmbeddingAutoConfiguration} 的 miniMaxEmbeddingModel 方法 - */ - private EmbeddingModel buildMiniMaxEmbeddingModel(String apiKey, String url, String model) { - MiniMaxApi miniMaxApi = StrUtil.isEmpty(url)? new MiniMaxApi(apiKey) - : new MiniMaxApi(url, apiKey); - MiniMaxEmbeddingOptions miniMaxEmbeddingOptions = MiniMaxEmbeddingOptions.builder().model(model).build(); - return new MiniMaxEmbeddingModel(miniMaxApi, MetadataMode.EMBED, miniMaxEmbeddingOptions); - } - - /** - * 可参考 {@link QianFanEmbeddingAutoConfiguration} 的 qianFanEmbeddingModel 方法 - */ - private QianFanEmbeddingModel buildYiYanEmbeddingModel(String key, String model) { - List keys = StrUtil.split(key, '|'); - Assert.equals(keys.size(), 2, "YiYanChatClient 的密钥需要 (appKey|secretKey) 格式"); - String appKey = keys.get(0); - String secretKey = keys.get(1); - QianFanApi qianFanApi = new QianFanApi(appKey, secretKey); - QianFanEmbeddingOptions qianFanEmbeddingOptions = QianFanEmbeddingOptions.builder().model(model).build(); - return new QianFanEmbeddingModel(qianFanApi, MetadataMode.EMBED, qianFanEmbeddingOptions); - } - - private OllamaEmbeddingModel buildOllamaEmbeddingModel(String url, String model) { - OllamaApi ollamaApi = OllamaApi.builder().baseUrl(url).build(); - OllamaEmbeddingOptions ollamaOptions = OllamaEmbeddingOptions.builder().model(model).build(); - return OllamaEmbeddingModel.builder() - .ollamaApi(ollamaApi) - .defaultOptions(ollamaOptions) - .build(); - } - - /** - * 可参考 {@link OpenAiEmbeddingAutoConfiguration} 的 openAiEmbeddingModel 方法 - */ - private OpenAiEmbeddingModel buildOpenAiEmbeddingModel(String openAiToken, String url, String model) { - url = StrUtil.blankToDefault(url, OpenAiApiConstants.DEFAULT_BASE_URL); - OpenAiApi openAiApi = OpenAiApi.builder().baseUrl(url).apiKey(openAiToken).build(); - OpenAiEmbeddingOptions openAiEmbeddingProperties = OpenAiEmbeddingOptions.builder().model(model).build(); - return new OpenAiEmbeddingModel(openAiApi, MetadataMode.EMBED, openAiEmbeddingProperties); - } - - /** - * 可参考 {@link AzureOpenAiEmbeddingAutoConfiguration} 的 azureOpenAiEmbeddingModel 方法 - */ - private AzureOpenAiEmbeddingModel buildAzureOpenAiEmbeddingModel(String apiKey, String url, String model) { - // TODO @芋艿：手头暂时没密钥，使用建议再测试下 - AzureOpenAiEmbeddingAutoConfiguration azureOpenAiAutoConfiguration = new AzureOpenAiEmbeddingAutoConfiguration(); - // 创建 OpenAIClientBuilder 对象 - OpenAIClientBuilder openAIClientBuilder = new OpenAIClientBuilder() - .endpoint(url).credential(new KeyCredential(apiKey)); - // 获取 AzureOpenAiChatProperties 对象 - AzureOpenAiEmbeddingProperties embeddingProperties = SpringUtil.getBean(AzureOpenAiEmbeddingProperties.class); - return azureOpenAiAutoConfiguration.azureOpenAiEmbeddingModel(openAIClientBuilder, embeddingProperties, - getObservationRegistry(), getEmbeddingModelObservationConvention()); - } - - // ========== 各种创建 VectorStore 的方法 ========== - - /** - * 注意：仅适合本地测试使用，生产建议还是使用 Qdrant、Milvus 等 - */ - @SneakyThrows - @SuppressWarnings("ResultOfMethodCallIgnored") - private SimpleVectorStore buildSimpleVectorStore(EmbeddingModel embeddingModel) { - SimpleVectorStore vectorStore = SimpleVectorStore.builder(embeddingModel).build(); - // 启动加载 - File file = new File(StrUtil.format("{}/vector_store/simple_{}.json", - FileUtil.getUserHomePath(), embeddingModel.getClass().getSimpleName())); - if (!file.exists()) { - FileUtil.mkParentDirs(file); - file.createNewFile(); - } else if (file.length() > 0) { - vectorStore.load(file); - } - // 定时持久化，每分钟一次 - Timer timer = new Timer("SimpleVectorStoreTimer-" + file.getAbsolutePath()); - timer.scheduleAtFixedRate(new TimerTask() { - - @Override - public void run() { - vectorStore.save(file); - } - - }, Duration.ofMinutes(1).toMillis(), Duration.ofMinutes(1).toMillis()); - // 关闭时，进行持久化 - RuntimeUtil.addShutdownHook(() -> vectorStore.save(file)); - return vectorStore; - } - - /** - * 参考 {@link QdrantVectorStoreAutoConfiguration} 的 vectorStore 方法 - */ - @SneakyThrows - private QdrantVectorStore buildQdrantVectorStore(EmbeddingModel embeddingModel) { - QdrantVectorStoreAutoConfiguration configuration = new QdrantVectorStoreAutoConfiguration(); - QdrantVectorStoreProperties properties = SpringUtil.getBean(QdrantVectorStoreProperties.class); - // 参考 QdrantVectorStoreAutoConfiguration 实现，创建 QdrantClient 对象 - QdrantGrpcClient.Builder grpcClientBuilder = QdrantGrpcClient.newBuilder( - properties.getHost(), properties.getPort(), properties.isUseTls()); - if (StrUtil.isNotEmpty(properties.getApiKey())) { - grpcClientBuilder.withApiKey(properties.getApiKey()); - } - QdrantClient qdrantClient = new QdrantClient(grpcClientBuilder.build()); - // 创建 QdrantVectorStore 对象 - QdrantVectorStore vectorStore = configuration.vectorStore(embeddingModel, properties, qdrantClient, - getObservationRegistry(), getCustomObservationConvention(), getBatchingStrategy()); - // 初始化索引 - vectorStore.afterPropertiesSet(); - return vectorStore; - } - - /** - * 参考 {@link RedisVectorStoreAutoConfiguration} 的 vectorStore 方法 - */ - private RedisVectorStore buildRedisVectorStore(EmbeddingModel embeddingModel, - Map> metadataFields) { - // 创建 JedisPooled 对象 - RedisProperties redisProperties = SpringUtils.getBean(RedisProperties.class); - JedisPooled jedisPooled = new JedisPooled(redisProperties.getHost(), redisProperties.getPort(), - redisProperties.getUsername(), redisProperties.getPassword()); - // 创建 RedisVectorStoreProperties 对象 - RedisVectorStoreProperties properties = SpringUtil.getBean(RedisVectorStoreProperties.class); - RedisVectorStore redisVectorStore = RedisVectorStore.builder(jedisPooled, embeddingModel) - .indexName(properties.getIndexName()).prefix(properties.getPrefix()) - .initializeSchema(properties.isInitializeSchema()) - .metadataFields(convertList(metadataFields.entrySet(), entry -> { - String fieldName = entry.getKey(); - Class fieldType = entry.getValue(); - if (Number.class.isAssignableFrom(fieldType)) { - return RedisVectorStore.MetadataField.numeric(fieldName); - } - if (Boolean.class.isAssignableFrom(fieldType)) { - return RedisVectorStore.MetadataField.tag(fieldName); - } - return RedisVectorStore.MetadataField.text(fieldName); - })) - .observationRegistry(getObservationRegistry().getObject()) - .customObservationConvention(getCustomObservationConvention().getObject()) - .batchingStrategy(getBatchingStrategy()) - .build(); - // 初始化索引 - redisVectorStore.afterPropertiesSet(); - return redisVectorStore; - } - - /** - * 参考 {@link MilvusVectorStoreAutoConfiguration} 的 vectorStore 方法 - */ - @SneakyThrows - private MilvusVectorStore buildMilvusVectorStore(EmbeddingModel embeddingModel) { - MilvusVectorStoreAutoConfiguration configuration = new MilvusVectorStoreAutoConfiguration(); - // 获取配置属性 - MilvusVectorStoreProperties serverProperties = SpringUtil.getBean(MilvusVectorStoreProperties.class); - MilvusServiceClientProperties clientProperties = SpringUtil.getBean(MilvusServiceClientProperties.class); - - // 创建 MilvusServiceClient 对象 - MilvusServiceClient milvusClient = configuration.milvusClient(serverProperties, clientProperties, - new MilvusServiceClientConnectionDetails() { - - @Override - public String getHost() { - return clientProperties.getHost(); - } - - @Override - public int getPort() { - return clientProperties.getPort(); - } - - } - ); - // 创建 MilvusVectorStore 对象 - MilvusVectorStore vectorStore = configuration.vectorStore(milvusClient, embeddingModel, serverProperties, - getBatchingStrategy(), getObservationRegistry(), getCustomObservationConvention()); - - // 初始化索引 - vectorStore.afterPropertiesSet(); - return vectorStore; - } - - private static ObjectProvider getObservationRegistry() { - return new ObjectProvider<>() { - - @Override - public ObservationRegistry getObject() throws BeansException { - return SpringUtil.getBean(ObservationRegistry.class); - } - - }; - } - - private static ObjectProvider getCustomObservationConvention() { - return new ObjectProvider<>() { - - @Override - public VectorStoreObservationConvention getObject() throws BeansException { - return new DefaultVectorStoreObservationConvention(); - } - - }; - } - - private static BatchingStrategy getBatchingStrategy() { - return SpringUtil.getBean(BatchingStrategy.class); - } - - private static ToolCallingManager getToolCallingManager() { - return SpringUtil.getBean(ToolCallingManager.class); - } - - private static ObjectProvider getEmbeddingModelObservationConvention() { - return new ObjectProvider<>() { - - @Override - public EmbeddingModelObservationConvention getObject() throws BeansException { - return SpringUtil.getBean(EmbeddingModelObservationConvention.class); - } - - }; - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/baichuan/BaiChuanChatModel.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/baichuan/BaiChuanChatModel.java deleted file mode 100644 index 96f6aa2..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/baichuan/BaiChuanChatModel.java +++ /dev/null @@ -1,45 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model.baichuan; - -import lombok.RequiredArgsConstructor; -import lombok.extern.slf4j.Slf4j; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.chat.model.ChatResponse; -import org.springframework.ai.chat.prompt.ChatOptions; -import org.springframework.ai.chat.prompt.Prompt; -import org.springframework.ai.openai.OpenAiChatModel; -import reactor.core.publisher.Flux; - -/** - * 百川 {@link ChatModel} 实现类 - * - * @author 芋道源码 - */ -@Slf4j -@RequiredArgsConstructor -public class BaiChuanChatModel implements ChatModel { - - public static final String BASE_URL = "https://api.baichuan-ai.com"; - - public static final String MODEL_DEFAULT = "Baichuan4-Turbo"; - - /** - * 兼容 OpenAI 接口，进行复用 - */ - private final OpenAiChatModel openAiChatModel; - - @Override - public ChatResponse call(Prompt prompt) { - return openAiChatModel.call(prompt); - } - - @Override - public Flux stream(Prompt prompt) { - return openAiChatModel.stream(prompt); - } - - @Override - public ChatOptions getDefaultOptions() { - return openAiChatModel.getDefaultOptions(); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/doubao/DouBaoChatModel.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/doubao/DouBaoChatModel.java deleted file mode 100644 index 15ca8b6..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/doubao/DouBaoChatModel.java +++ /dev/null @@ -1,45 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model.doubao; - -import lombok.RequiredArgsConstructor; -import lombok.extern.slf4j.Slf4j; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.chat.model.ChatResponse; -import org.springframework.ai.chat.prompt.ChatOptions; -import org.springframework.ai.chat.prompt.Prompt; -import reactor.core.publisher.Flux; - -/** - * 字节豆包 {@link ChatModel} 实现类 - * - * @author fansili - */ -@Slf4j -@RequiredArgsConstructor -public class DouBaoChatModel implements ChatModel { - - public static final String BASE_URL = "https://ark.cn-beijing.volces.com/api"; - public static final String COMPLETE_PATH = "/v3/chat/completions"; - - public static final String MODEL_DEFAULT = "doubao-1-5-lite-32k-250115"; - - /** - * 兼容 OpenAI 接口，进行复用 - */ - private final ChatModel openAiChatModel; - - @Override - public ChatResponse call(Prompt prompt) { - return openAiChatModel.call(prompt); - } - - @Override - public Flux stream(Prompt prompt) { - return openAiChatModel.stream(prompt); - } - - @Override - public ChatOptions getDefaultOptions() { - return openAiChatModel.getDefaultOptions(); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/gemini/GeminiChatModel.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/gemini/GeminiChatModel.java deleted file mode 100644 index f0d8ffc..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/gemini/GeminiChatModel.java +++ /dev/null @@ -1,46 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model.gemini; - -import lombok.RequiredArgsConstructor; -import lombok.extern.slf4j.Slf4j; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.chat.model.ChatResponse; -import org.springframework.ai.chat.prompt.ChatOptions; -import org.springframework.ai.chat.prompt.Prompt; -import org.springframework.ai.openai.OpenAiChatModel; -import reactor.core.publisher.Flux; - -/** - * 谷歌 Gemini {@link ChatModel} 实现类，基于 Google AI Studio 提供的 OpenAI 兼容方案 - * - * @author 芋道源码 - */ -@Slf4j -@RequiredArgsConstructor -public class GeminiChatModel implements ChatModel { - - public static final String BASE_URL = "https://generativelanguage.googleapis.com/v1beta/openai/"; - public static final String COMPLETE_PATH = "/chat/completions"; - - public static final String MODEL_DEFAULT = "gemini-2.5-flash"; - - /** - * 兼容 OpenAI 接口，进行复用 - */ - private final OpenAiChatModel openAiChatModel; - - @Override - public ChatResponse call(Prompt prompt) { - return openAiChatModel.call(prompt); - } - - @Override - public Flux stream(Prompt prompt) { - return openAiChatModel.stream(prompt); - } - - @Override - public ChatOptions getDefaultOptions() { - return openAiChatModel.getDefaultOptions(); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/grok/GrokChatModel.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/grok/GrokChatModel.java deleted file mode 100644 index a27b2f1..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/grok/GrokChatModel.java +++ /dev/null @@ -1,44 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model.grok; - -import lombok.RequiredArgsConstructor; -import lombok.extern.slf4j.Slf4j; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.chat.model.ChatResponse; -import org.springframework.ai.chat.prompt.ChatOptions; -import org.springframework.ai.chat.prompt.Prompt; -import reactor.core.publisher.Flux; - -/** - * Grok {@link ChatModel} 实现类 - * - * - */ -@Slf4j -@RequiredArgsConstructor -public class GrokChatModel implements ChatModel { - - public static final String BASE_URL = "https://api.x.ai"; - public static final String COMPLETE_PATH = "/v1/chat/completions"; - public static final String MODEL_DEFAULT = "grok-4-fast-reasoning"; - - /** - * 兼容 OpenAI 接口，进行复用 - */ - private final ChatModel openAiChatModel; - - @Override - public ChatResponse call(Prompt prompt) { - return openAiChatModel.call(prompt); - } - - @Override - public Flux stream(Prompt prompt) { - return openAiChatModel.stream(prompt); - } - - @Override - public ChatOptions getDefaultOptions() { - return openAiChatModel.getDefaultOptions(); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/hunyuan/HunYuanChatModel.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/hunyuan/HunYuanChatModel.java deleted file mode 100644 index 77c5324..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/hunyuan/HunYuanChatModel.java +++ /dev/null @@ -1,52 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model.hunyuan; - -import lombok.RequiredArgsConstructor; -import lombok.extern.slf4j.Slf4j; -import org.springframework.ai.chat.model.ChatModel; -import org.springframework.ai.chat.model.ChatResponse; -import org.springframework.ai.chat.prompt.ChatOptions; -import org.springframework.ai.chat.prompt.Prompt; -import reactor.core.publisher.Flux; - -/** - * 腾云混元 {@link ChatModel} 实现类 - * - * 1. 混元大模型：基于 知识引擎原子能力 实现 - * 2. 知识引擎原子能力：基于 知识引擎原子能力 实现 - * - * @author fansili - */ -@Slf4j -@RequiredArgsConstructor -public class HunYuanChatModel implements ChatModel { - - public static final String BASE_URL = "https://api.hunyuan.cloud.tencent.com"; - public static final String COMPLETE_PATH = "/v1/chat/completions"; - - public static final String MODEL_DEFAULT = "hunyuan-turbo"; - - public static final String DEEP_SEEK_BASE_URL = "https://api.lkeap.cloud.tencent.com"; - - public static final String DEEP_SEEK_MODEL_DEFAULT = "deepseek-v3"; - - /** - * 兼容 OpenAI 接口，进行复用 - */ - private final ChatModel openAiChatModel; - - @Override - public ChatResponse call(Prompt prompt) { - return openAiChatModel.call(prompt); - } - - @Override - public Flux stream(Prompt prompt) { - return openAiChatModel.stream(prompt); - } - - @Override - public ChatOptions getDefaultOptions() { - return openAiChatModel.getDefaultOptions(); - } - -} diff --git a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/midjourney/api/MidjourneyApi.java b/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/midjourney/api/MidjourneyApi.java deleted file mode 100644 index 5c5f797..0000000 --- a/lyzsys-module-ai/src/main/java/cn/iocoder/lyzsys/module/ai/framework/ai/core/model/midjourney/api/MidjourneyApi.java +++ /dev/null @@ -1,351 +0,0 @@ -package cn.iocoder.lyzsys.module.ai.framework.ai.core.model.midjourney.api; - -import cn.hutool.core.util.StrUtil; -import cn.iocoder.lyzsys.framework.common.util.json.JsonUtils; -import com.google.common.collect.ImmutableMap; -import com.google.common.collect.Lists; -import lombok.AllArgsConstructor; -import lombok.Data; -import lombok.Getter; -import lombok.extern.slf4j.Slf4j; -import org.springframework.http.HttpRequest; -import org.springframework.http.HttpStatusCode; -import org.springframework.http.MediaType; -import org.springframework.web.reactive.function.client.ClientResponse; -import org.springframework.web.reactive.function.client.WebClient; -import reactor.core.publisher.Mono; - -import java.util.Collection; -import java.util.List; -import java.util.Map; -import java.util.function.Function; -import java.util.function.Predicate; - -/** - * Midjourney API - * - * @author fansili - * @since 1.0 - */ -@Slf4j -public class MidjourneyApi { - - private final Predicate STATUS_PREDICATE = status -> !status.is2xxSuccessful(); - - private final Function>> EXCEPTION_FUNCTION = - reqParam -> response -> response.bodyToMono(String.class).handle((responseBody, sink) -> { - HttpRequest request = response.request(); - log.error("[midjourney-api] 调用失败！请求方式:[{}]，请求地址:[{}]，请求参数:[{}]，响应数据: [{}]", - request.getMethod(), request.getURI(), reqParam, responseBody); - sink.error(new IllegalStateException("[midjourney-api] 调用失败！")); - }); - - private final WebClient webClient; - - /** - * 回调地址 - */ - private final String notifyUrl; - - public MidjourneyApi(String baseUrl, String apiKey, String notifyUrl) { - this.webClient = WebClient.builder() - .baseUrl(baseUrl) - .defaultHeaders(httpHeaders -> { - httpHeaders.setContentType(MediaType.APPLICATION_JSON); - httpHeaders.setBearerAuth(apiKey); - }) - .build(); - this.notifyUrl = notifyUrl; - } - - /** - * imagine - 根据提示词提交绘画任务 - * - * @param request 请求 - * @return 提交结果 - */ - public SubmitResponse imagine(ImagineRequest request) { - if (StrUtil.isEmpty(request.getNotifyHook())) { - request.setNotifyHook(notifyUrl); - } - String response = post("/submit/imagine", request); - return JsonUtils.parseObject(response, SubmitResponse.class); - } - - /** - * action - 放大、缩小、U1、U2... - * - * @param request 请求 - * @return 提交结果 - */ - public SubmitResponse action(ActionRequest request) { - if (StrUtil.isEmpty(request.getNotifyHook())) { - request.setNotifyHook(notifyUrl); - } - String response = post("/submit/action", request); - return JsonUtils.parseObject(response, SubmitResponse.class); - } - - /** - * 批量查询 task 任务 - * - * @param ids 任务编号数组 - * @return task 任务 - */ - public List getTaskList(Collection ids) { - String res = post("/task/list-by-condition", ImmutableMap.of("ids", ids)); - return JsonUtils.parseArray(res, Notify.class); - } - - private String post(String uri, Object body) { - return webClient.post() - .uri(uri) - .body(Mono.just(JsonUtils.toJsonString(body)), String.class) - .retrieve() - .onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(body)) - .bodyToMono(String.class) - .block(); - } - - // ========== record 结构 ========== - - /** - * Imagine 请求（生成图片） - */ - @Data - public static final class ImagineRequest { - - /** - * 垫图(参考图) base64 数组 - */ - private List base64Array; - /** - * 提示词 - */ - private String prompt; - /** - * 通知地址 - */ - private String notifyHook; - /** - * 自定义参数 - */ - private String state; - - public ImagineRequest(List base64Array, String prompt, String notifyHook, String state) { - this.base64Array = base64Array; - this.prompt = prompt; - this.notifyHook = notifyHook; - this.state = state; - } - - public static String buildState(Integer width, Integer height, String version, String model) { - StringBuilder params = new StringBuilder(); - // --ar 来设置尺寸 - params.append(String.format(" --ar %s:%s ", width, height)); - // --niji 模型 - if (ModelEnum.NIJI.getModel().equals(model)) { - params.append(String.format(" --niji %s ", version)); - } else { - params.append(String.format(" --v %s ", version)); - } - return params.toString(); - } - - } - - /** - * Action 请求 - */ - @Data - public static final class ActionRequest { - - private String customId; - private String taskId; - private String notifyHook; - - public ActionRequest(String taskId, String customId, String notifyHook) { - this.customId = customId; - this.taskId = taskId; - this.notifyHook = notifyHook; - } - - } - - /** - * Submit 统一返回 - * - * @param code 状态码: 1(提交成功), 21(已存在), 22(排队中), other(错误) - * @param description 描述 - * @param properties 扩展字段 - * @param result 任务ID - */ - public record SubmitResponse(String code, - String description, - Map properties, - String result) { - } - - /** - * 通知 request - * - * @param id job id - * @param action 任务类型 {@link TaskActionEnum} - * @param status 任务状态 {@link TaskStatusEnum} - * @param prompt 提示词 - * @param promptEn 提示词-英文 - * @param description 任务描述 - * @param state 自定义参数 - * @param submitTime 提交时间 - * @param startTime 开始执行时间 - * @param finishTime 结束时间 - * @param imageUrl 图片url - * @param progress 任务进度 - * @param failReason 失败原因 - * @param buttons 任务完成后的可执行按钮 - */ - public record Notify(String id, - String action, - String status, - - String prompt, - String promptEn, - - String description, - String state, - - Long submitTime, - Long startTime, - Long finishTime, - - String imageUrl, - String progress, - String failReason, - List