DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

增加app 应用名称 增删改查 简单逻辑

Browse Source
This commit is contained in:
limin 2024-12-18 17:04:44 +08:00
parent 77c57b9a42
commit 2b7ad2ac81
10 changed files with 511 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
yudao-module-llm/yudao-module-llm-api/src/main/java/cn/iocoder/yudao/module/llm/enums/ErrorCodeConstants.java
View File

@ -12,4 +12,7 @@ public interface ErrorCodeConstants {
ErrorCode LABEL_NOT_EXISTS = new ErrorCode(10004,"标签不存在");
ErrorCode FINE_TUNING_TASK_NOT_EXISTS = new ErrorCode(10005, "微调任务不存在");
ErrorCode APPLICATION_NOT_EXISTS = new ErrorCode(10005, "应用名称服务不存在");
}

95
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/application/ApplicationController.java Normal file
View File

@ -0,0 +1,95 @@
package cn.iocoder.yudao.module.llm.controller.admin.application;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.security.access.prepost.PreAuthorize;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Operation;
import javax.validation.constraints.*;
import javax.validation.*;
import javax.servlet.http.*;
import java.util.*;
import java.io.IOException;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.*;
import cn.iocoder.yudao.module.llm.controller.admin.application.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.application.ApplicationDO;
import cn.iocoder.yudao.module.llm.service.application.ApplicationService;
@Tag(name = "管理后台 - 大模型应用")
@RestController
@RequestMapping("/llm/application")
@Validated
public class ApplicationController {
@Resource
private ApplicationService applicationService;
@PostMapping("/create")
@Operation(summary = "创建大模型应用")
@PreAuthorize("@ss.hasPermission('llm:application:create')")
public CommonResult<Long> createApplication(@Valid @RequestBody ApplicationSaveReqVO createReqVO) {
return success(applicationService.createApplication(createReqVO));
}
@PutMapping("/update")
@Operation(summary = "更新大模型应用")
@PreAuthorize("@ss.hasPermission('llm:application:update')")
public CommonResult<Boolean> updateApplication(@Valid @RequestBody ApplicationSaveReqVO updateReqVO) {
applicationService.updateApplication(updateReqVO);
return success(true);
}
@DeleteMapping("/delete")
@Operation(summary = "删除大模型应用")
@Parameter(name = "id", description = "编号", required = true)
@PreAuthorize("@ss.hasPermission('llm:application:delete')")
public CommonResult<Boolean> deleteApplication(@RequestParam("id") Long id) {
applicationService.deleteApplication(id);
return success(true);
}
@GetMapping("/get")
@Operation(summary = "获得大模型应用")
@Parameter(name = "id", description = "编号", required = true, example = "1024")
@PreAuthorize("@ss.hasPermission('llm:application:query')")
public CommonResult<ApplicationRespVO> getApplication(@RequestParam("id") Long id) {
ApplicationDO application = applicationService.getApplication(id);
return success(BeanUtils.toBean(application, ApplicationRespVO.class));
}
@GetMapping("/page")
@Operation(summary = "获得大模型应用分页")
@PreAuthorize("@ss.hasPermission('llm:application:query')")
public CommonResult<PageResult<ApplicationRespVO>> getApplicationPage(@Valid ApplicationPageReqVO pageReqVO) {
PageResult<ApplicationDO> pageResult = applicationService.getApplicationPage(pageReqVO);
return success(BeanUtils.toBean(pageResult, ApplicationRespVO.class));
}
@GetMapping("/export-excel")
@Operation(summary = "导出大模型应用 Excel")
@PreAuthorize("@ss.hasPermission('llm:application:export')")
@ApiAccessLog(operateType = EXPORT)
public void exportApplicationExcel(@Valid ApplicationPageReqVO pageReqVO,
HttpServletResponse response) throws IOException {
pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
List<ApplicationDO> list = applicationService.getApplicationPage(pageReqVO).getList();
// 导出 Excel
ExcelUtils.write(response, "大模型应用.xls", "数据", ApplicationRespVO.class,
BeanUtils.toBean(list, ApplicationRespVO.class));
}
}

21
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/application/vo/ApplicationPageReqVO.java Normal file
View File

@ -0,0 +1,21 @@
package cn.iocoder.yudao.module.llm.controller.admin.application.vo;
import lombok.*;
import java.util.*;
import io.swagger.v3.oas.annotations.media.Schema;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import static cn.iocoder.yudao.framework.common.util.date.DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND;
@Schema(description = "管理后台 - 大模型应用分页 Request VO")
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
public class ApplicationPageReqVO extends PageParam {
@Schema(description = "应用名称", example = "赵六")
private String appName;
}

78
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/application/vo/ApplicationRespVO.java Normal file
View File

@ -0,0 +1,78 @@
package cn.iocoder.yudao.module.llm.controller.admin.application.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.*;
import java.util.*;
import org.springframework.format.annotation.DateTimeFormat;
import java.time.LocalDateTime;
import com.alibaba.excel.annotation.*;
import cn.iocoder.yudao.framework.excel.core.annotations.DictFormat;
import cn.iocoder.yudao.framework.excel.core.convert.DictConvert;
@Schema(description = "管理后台 - 大模型应用 Response VO")
@Data
@ExcelIgnoreUnannotated
public class ApplicationRespVO {
@Schema(description = "应用ID", requiredMode = Schema.RequiredMode.REQUIRED, example = "31452")
@ExcelProperty("应用ID")
private Long id;
@Schema(description = "应用名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "赵六")
@ExcelProperty("应用名称")
private String appName;
@Schema(description = "应用简介")
@ExcelProperty("应用简介")
private String appIntro;
@Schema(description = "应用分类，使用字典（llm_application_category）", requiredMode = Schema.RequiredMode.REQUIRED)
@ExcelProperty(value = "应用分类，使用字典（llm_application_category）", converter = DictConvert.class)
@DictFormat("llm_application_category") // TODO 代码优化：建议设置到对应的 DictTypeConstants 枚举类中
private Long appCategory;
@Schema(description = "应用标签，保存多个标签ID，用逗号分隔", requiredMode = Schema.RequiredMode.REQUIRED)
@ExcelProperty("应用标签，保存多个标签ID，用逗号分隔")
private String appLabel;
@Schema(description = "模型服务，模型服务ID")
@ExcelProperty("模型服务，模型服务ID")
private Long appModel;
@Schema(description = "随机性")
@ExcelProperty("随机性")
private Integer temperature;
@Schema(description = "采样性")
@ExcelProperty("采样性")
private Integer topP;
@Schema(description = "单次回复限制")
@ExcelProperty("单次回复限制")
private Integer maxTokens;
@Schema(description = "系统Prompt")
@ExcelProperty("系统Prompt")
private String prompt;
@Schema(description = "向量知识库，知识库ID")
@ExcelProperty("向量知识库，知识库ID")
private Long retrievalAugmentedGeneration;
@Schema(description = "应用API地址", example = "https://www.iocoder.cn")
@ExcelProperty("应用API地址")
private String appUrl;
@Schema(description = "应用图标")
@ExcelProperty("应用图标")
private String appLogo;
@Schema(description = "聊天人次", example = "8556")
@ExcelProperty("聊天人次")
private Long chatCount;
@Schema(description = "创建时间", requiredMode = Schema.RequiredMode.REQUIRED)
@ExcelProperty("创建时间")
private LocalDateTime createTime;
}

57
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/application/vo/ApplicationSaveReqVO.java Normal file
View File

@ -0,0 +1,57 @@
package cn.iocoder.yudao.module.llm.controller.admin.application.vo;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.*;
import java.util.*;
import javax.validation.constraints.*;
@Schema(description = "管理后台 - 大模型应用新增/修改 Request VO")
@Data
public class ApplicationSaveReqVO {
@Schema(description = "应用ID", requiredMode = Schema.RequiredMode.REQUIRED, example = "31452")
private Long id;
@Schema(description = "应用名称", requiredMode = Schema.RequiredMode.REQUIRED, example = "赵六")
@NotEmpty(message = "应用名称不能为空")
private String appName;
@Schema(description = "应用简介")
private String appIntro;
@Schema(description = "应用分类，使用字典（llm_application_category）", requiredMode = Schema.RequiredMode.REQUIRED)
@NotNull(message = "应用分类，使用字典（llm_application_category）不能为空")
private Long appCategory;
@Schema(description = "应用标签，保存多个标签ID，用逗号分隔", requiredMode = Schema.RequiredMode.REQUIRED)
@NotEmpty(message = "应用标签，保存多个标签ID，用逗号分隔不能为空")
private String appLabel;
@Schema(description = "模型服务，模型服务ID")
private Long appModel;
@Schema(description = "随机性")
private Integer temperature;
@Schema(description = "采样性")
private Integer topP;
@Schema(description = "单次回复限制")
private Integer maxTokens;
@Schema(description = "系统Prompt")
private String prompt;
@Schema(description = "向量知识库，知识库ID")
private Long retrievalAugmentedGeneration;
@Schema(description = "应用API地址", example = "https://www.iocoder.cn")
private String appUrl;
@Schema(description = "应用图标")
private String appLogo;
@Schema(description = "聊天人次", example = "8556")
private Long chatCount;
}

85
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/application/ApplicationDO.java Normal file
View File

@ -0,0 +1,85 @@
package cn.iocoder.yudao.module.llm.dal.dataobject.application;
import lombok.*;
import java.util.*;
import java.time.LocalDateTime;
import java.time.LocalDateTime;
import com.baomidou.mybatisplus.annotation.*;
import cn.iocoder.yudao.framework.mybatis.core.dataobject.BaseDO;
/**
* 大模型应用 DO
*
* @author 华大大模型
*/
@TableName("llm_application")
@KeySequence("llm_application_seq") // 用于 Oracle、PostgreSQL、Kingbase、DB2、H2 数据库的主键自增。如果是 MySQL 等数据库，可不写。
@Data
@EqualsAndHashCode(callSuper = true)
@ToString(callSuper = true)
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class ApplicationDO extends BaseDO {
/**
* 应用ID
*/
@TableId
private Long id;
/**
* 应用名称
*/
private String appName;
/**
* 应用简介
*/
private String appIntro;
/**
* 应用分类，使用字典（llm_application_category）
*
* 枚举 {@link TODO llm_application_category 对应的类}
*/
private Long appCategory;
/**
* 应用标签，保存多个标签ID，用逗号分隔
*/
private String appLabel;
/**
* 模型服务，模型服务ID
*/
private Long appModel;
/**
* 随机性
*/
private Integer temperature;
/**
* 采样性
*/
private Integer topP;
/**
* 单次回复限制
*/
private Integer maxTokens;
/**
* 系统Prompt
*/
private String prompt;
/**
* 向量知识库，知识库ID
*/
private Long retrievalAugmentedGeneration;
/**
* 应用API地址
*/
private String appUrl;
/**
* 应用图标
*/
private String appLogo;
/**
* 聊天人次
*/
private Long chatCount;
}

27
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/application/ApplicationMapper.java Normal file
View File

@ -0,0 +1,27 @@
package cn.iocoder.yudao.module.llm.dal.mysql.application;
import java.util.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX;
import cn.iocoder.yudao.module.llm.dal.dataobject.application.ApplicationDO;
import org.apache.ibatis.annotations.Mapper;
import cn.iocoder.yudao.module.llm.controller.admin.application.vo.*;
/**
* 大模型应用 Mapper
*
* @author 华大大模型
*/
@Mapper
public interface ApplicationMapper extends BaseMapperX<ApplicationDO> {
default PageResult<ApplicationDO> selectPage(ApplicationPageReqVO reqVO) {
return selectPage(reqVO, new LambdaQueryWrapperX<ApplicationDO>()
.likeIfPresent(ApplicationDO::getAppName, reqVO.getAppName())
.eq(ApplicationDO::getDeleted,false)
.orderByDesc(ApplicationDO::getId));
}
}

55
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/application/ApplicationService.java Normal file
View File

@ -0,0 +1,55 @@
package cn.iocoder.yudao.module.llm.service.application;
import java.util.*;
import javax.validation.*;
import cn.iocoder.yudao.module.llm.controller.admin.application.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.application.ApplicationDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
/**
* 大模型应用 Service 接口
*
* @author 华大大模型
*/
public interface ApplicationService {
/**
* 创建大模型应用
*
* @param createReqVO 创建信息
* @return 编号
*/
Long createApplication(@Valid ApplicationSaveReqVO createReqVO);
/**
* 更新大模型应用
*
* @param updateReqVO 更新信息
*/
void updateApplication(@Valid ApplicationSaveReqVO updateReqVO);
/**
* 删除大模型应用
*
* @param id 编号
*/
void deleteApplication(Long id);
/**
* 获得大模型应用
*
* @param id 编号
* @return 大模型应用
*/
ApplicationDO getApplication(Long id);
/**
* 获得大模型应用分页
*
* @param pageReqVO 分页查询
* @return 大模型应用分页
*/
PageResult<ApplicationDO> getApplicationPage(ApplicationPageReqVO pageReqVO);
}

78
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/application/ApplicationServiceImpl.java Normal file
View File

@ -0,0 +1,78 @@
package cn.iocoder.yudao.module.llm.service.application;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;
import java.util.*;
import cn.iocoder.yudao.module.llm.controller.admin.application.vo.*;
import cn.iocoder.yudao.module.llm.dal.dataobject.application.ApplicationDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.llm.dal.mysql.application.ApplicationMapper;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.llm.enums.ErrorCodeConstants.*;
/**
* 大模型应用 Service 实现类
*
* @author 华大大模型
*/
@Service
@Validated
public class ApplicationServiceImpl implements ApplicationService {
@Resource
private ApplicationMapper applicationMapper;
@Override
public Long createApplication(ApplicationSaveReqVO createReqVO) {
// 插入
ApplicationDO application = BeanUtils.toBean(createReqVO, ApplicationDO.class);
applicationMapper.insert(application);
// 返回
return application.getId();
}
@Override
public void updateApplication(ApplicationSaveReqVO updateReqVO) {
// 校验存在
validateApplicationExists(updateReqVO.getId());
// 更新
ApplicationDO updateObj = BeanUtils.toBean(updateReqVO, ApplicationDO.class);
applicationMapper.updateById(updateObj);
}
@Override
public void deleteApplication(Long id) {
// 校验存在
validateApplicationExists(id);
// 删除
LambdaUpdateWrapper<ApplicationDO> wrapper = new LambdaUpdateWrapper<>();
wrapper.eq(ApplicationDO::getId, id)
.set(ApplicationDO::getDeleted,true);
applicationMapper.update(null, wrapper);
}
private void validateApplicationExists(Long id) {
if (applicationMapper.selectById(id) == null) {
throw exception(APPLICATION_NOT_EXISTS);
}
}
@Override
public ApplicationDO getApplication(Long id) {
return applicationMapper.selectById(id);
}
@Override
public PageResult<ApplicationDO> getApplicationPage(ApplicationPageReqVO pageReqVO) {
return applicationMapper.selectPage(pageReqVO);
}
}

12
yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/application/ApplicationMapper.xml Normal file
View File

@ -0,0 +1,12 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="cn.iocoder.yudao.module.llm.dal.mysql.application.ApplicationMapper">
<!--
一般情况下，尽可能使用 Mapper 进行 CRUD 增删改查即可。
无法满足的场景，例如说多表关联查询，才使用 XML 编写 SQL。
代码生成器暂时只生成 Mapper XML 文件本身，更多推荐 MybatisX 快速开发插件来生成查询。
文档可见：https://www.iocoder.cn/MyBatis/x-plugins/
-->
</mapper>