diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java index 4498b60e9..b6ad1fc87 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/BaseModelController.java @@ -1,5 +1,6 @@ package cn.iocoder.yudao.module.llm.controller.admin.basemodel; +import com.alibaba.fastjson.JSONObject; import org.springframework.web.bind.annotation.*; import javax.annotation.Resource; import org.springframework.validation.annotation.Validated; @@ -53,6 +54,24 @@ public class BaseModelController { return success(true); } + @PutMapping("/active") + @Operation(summary = "启动") + @PreAuthorize("@ss.hasPermission('llm:base-model:update')") + public CommonResult active(@Valid @RequestBody BaseModelSaveReqVO updateReqVO) { + updateReqVO.setIsActive(true); + baseModelService.updateBaseModel(updateReqVO); + return success(true); + } + + @PutMapping("/unactive") + @Operation(summary = "禁用") + @PreAuthorize("@ss.hasPermission('llm:base-model:update')") + public CommonResult unactive(@Valid @RequestBody BaseModelSaveReqVO updateReqVO) { + updateReqVO.setIsActive(false); + baseModelService.updateBaseModel(updateReqVO); + return success(true); + } + @DeleteMapping("/delete") @Operation(summary = "删除基座模型") @Parameter(name = "id", description = "编号", required = true) @@ -71,6 +90,16 @@ public class BaseModelController { return success(BeanUtils.toBean(baseModel, BaseModelRespVO.class)); } + @GetMapping("/getParamModel") + @Operation(summary = "获取参数模板") + @Parameter(name = "id", description = "编号", required = true, example = "1024") + @PreAuthorize("@ss.hasPermission('llm:base-model:query')") + public CommonResult getParamModel(@RequestParam("id") Integer id) { + BaseModelDO baseModel = baseModelService.getBaseModel(id); + //todo 获取参数模板api + return success(new JSONObject()); + } + @GetMapping("/page") @Operation(summary = "获得基座模型分页") @PreAuthorize("@ss.hasPermission('llm:base-model:query')") @@ -79,6 +108,14 @@ public class BaseModelController { return success(BeanUtils.toBean(pageResult, BaseModelRespVO.class)); } + @GetMapping("/list") + @Operation(summary = "获得基座模型列表,下拉菜单") + @PreAuthorize("@ss.hasPermission('llm:base-model:query')") + public CommonResult> getBaseModelList() { + List pageResult = baseModelService.getBaseModelList(); + return success(BeanUtils.toBean(pageResult, BaseModelRespVO.class)); + } + @GetMapping("/export-excel") @Operation(summary = "导出基座模型 Excel") @PreAuthorize("@ss.hasPermission('llm:base-model:export')") @@ -92,4 +129,4 @@ public class BaseModelController { BeanUtils.toBean(list, BaseModelRespVO.class)); } -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelPageReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelPageReqVO.java index 1060568c4..80c498f49 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelPageReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelPageReqVO.java @@ -40,4 +40,13 @@ public class BaseModelPageReqVO extends PageParam { @DateTimeFormat(pattern = FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND) private LocalDateTime[] createTime; -} \ No newline at end of file + @Schema(description = "api地址") + private String apiUrl; + + @Schema(description = "api秘钥") + private String apiKey; + + @Schema(description = "代理") + private String proxy; + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelRespVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelRespVO.java index 37188156c..2548afe2f 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelRespVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelRespVO.java @@ -48,4 +48,14 @@ public class BaseModelRespVO { @ExcelProperty("创建时间") private LocalDateTime createTime; -} \ No newline at end of file + + @Schema(description = "api地址") + private String apiUrl; + + @Schema(description = "api秘钥") + private String apiKey; + + @Schema(description = "代理") + private String proxy; + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java index 934004ad9..1869d9fe5 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/controller/admin/basemodel/vo/BaseModelSaveReqVO.java @@ -39,4 +39,13 @@ public class BaseModelSaveReqVO { @Schema(description = "备注") private String notes; -} \ No newline at end of file + @Schema(description = "api地址") + private String apiUrl; + + @Schema(description = "api秘钥") + private String apiKey; + + @Schema(description = "代理") + private String proxy; + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java index 12ac24434..51173b741 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/dataobject/basemodel/BaseModelDO.java @@ -55,5 +55,17 @@ public class BaseModelDO extends BaseDO { * 备注 */ private String notes; + /** + * api地址 + */ + private String apiUrl; + /** + * api秘钥 + */ + private String apiKey; + /** + * 代理 + */ + private String proxy; -} \ No newline at end of file +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java index 004945000..8eec53a09 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelService.java @@ -52,4 +52,6 @@ public interface BaseModelService { */ PageResult getBaseModelPage(BaseModelPageReqVO pageReqVO); -} \ No newline at end of file + List getBaseModelList(); + +} diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java index 7aeb2e849..a98823cc5 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/basemodel/BaseModelServiceImpl.java @@ -1,5 +1,6 @@ package cn.iocoder.yudao.module.llm.service.basemodel; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import org.springframework.stereotype.Service; import javax.annotation.Resource; import org.springframework.validation.annotation.Validated; @@ -71,4 +72,12 @@ public class BaseModelServiceImpl implements BaseModelService { return baseModelMapper.selectPage(pageReqVO); } -} \ No newline at end of file + @Override + public List getBaseModelList() { + LambdaQueryWrapper queryWrapper = new LambdaQueryWrapper<>(); + queryWrapper.eq(BaseModelDO::getIsActive,0); + List baseModelDOS = baseModelMapper.selectList(queryWrapper); + return baseModelDOS; + } + +}