| | |
| | | import cn.iocoder.yudao.framework.common.pojo.CommonResult; |
| | | import cn.iocoder.yudao.framework.common.pojo.PageResult; |
| | | import cn.iocoder.yudao.framework.common.util.object.BeanUtils; |
| | | import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils; |
| | | import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeyImportExcelVO; |
| | | import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeyImportReqVO; |
| | | import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeyImportRespVO; |
| | | import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeyPageReqVO; |
| | | import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeyRespVO; |
| | | import cn.iocoder.yudao.module.ai.controller.admin.model.vo.apikey.AiApiKeySaveReqVO; |
| | |
| | | import io.swagger.v3.oas.annotations.Operation; |
| | | import io.swagger.v3.oas.annotations.tags.Tag; |
| | | import jakarta.annotation.Resource; |
| | | import jakarta.servlet.http.HttpServletResponse; |
| | | 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.io.IOException; |
| | | import java.util.Arrays; |
| | | import java.util.List; |
| | | |
| | | import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; |
| | | |
| | | @Tag(name = "管理后台 - AI API 密钥") |
| | | @RestController |
| | | @RequestMapping("/ai/api-key") |
| | | @Validated |
| | | public class AiApiKeyController { |
| | | |
| | | @Resource |
| | |
| | | |
| | | @GetMapping("/get") |
| | | @Operation(summary = "获取 API 密钥") |
| | | @PreAuthorize("@ss.hasPermission('ai:api-key:query')") |
| | | public CommonResult<AiApiKeyRespVO> getApiKey(@RequestParam("id") Long id) { |
| | | return success(BeanUtils.toBean(apiKeyService.getApiKey(id), AiApiKeyRespVO.class)); |
| | | } |
| | | |
| | | @GetMapping("/page") |
| | | @Operation(summary = "获取 API 密钥分页") |
| | | @PreAuthorize("@ss.hasPermission('ai:api-key:query')") |
| | | public CommonResult<PageResult<AiApiKeyRespVO>> getApiKeyPage(@Valid AiApiKeyPageReqVO pageReqVO) { |
| | | PageResult<AiApiKeyDO> pageResult = apiKeyService.getApiKeyPage(pageReqVO); |
| | | return success(BeanUtils.toBean(pageResult, AiApiKeyRespVO.class)); |
| | | } |
| | | |
| | | @GetMapping("/get-import-template") |
| | | @Operation(summary = "获得导入 API 密钥模板") |
| | | public void importTemplate(HttpServletResponse response) throws IOException { |
| | | // 手动创建导出 demo |
| | | List<AiApiKeyImportExcelVO> list = Arrays.asList( |
| | | AiApiKeyImportExcelVO.builder().name("示例密钥").apiKey("sk-xxx") |
| | | .platform("TongYi").url("https://dashscope.aliyuncs.com").status(0).build() |
| | | ); |
| | | // 输出 |
| | | ExcelUtils.write(response, "API密钥导入模板.xls", "API密钥列表", AiApiKeyImportExcelVO.class, list); |
| | | } |
| | | |
| | | @PostMapping("/import") |
| | | @Operation(summary = "导入 API 密钥") |
| | | @PreAuthorize("@ss.hasPermission('ai:api-key:import')") |
| | | public CommonResult<AiApiKeyImportRespVO> importExcel(@Valid AiApiKeyImportReqVO importReqVO) |
| | | throws Exception { |
| | | List<AiApiKeyImportExcelVO> list = ExcelUtils.read(importReqVO.getFile(), AiApiKeyImportExcelVO.class); |
| | | return success(apiKeyService.importApiKeyList(list, importReqVO)); |
| | | } |
| | | |
| | | } |