|
|
@@ -2,25 +2,22 @@ package cn.iocoder.byzs.module.web.controller.admin.ai;
|
|
|
|
|
|
import cn.hutool.core.util.ObjUtil;
|
|
|
import cn.iocoder.byzs.framework.common.pojo.CommonResult;
|
|
|
+import cn.iocoder.byzs.framework.common.pojo.PageResult;
|
|
|
import cn.iocoder.byzs.framework.common.util.object.BeanUtils;
|
|
|
import cn.iocoder.byzs.framework.tenant.core.aop.TenantIgnore;
|
|
|
import cn.iocoder.byzs.module.ai.controller.admin.chat.vo.conversation.AiChatConversationCreateMyReqVO;
|
|
|
-import cn.iocoder.byzs.module.ai.controller.admin.chat.vo.conversation.AiChatConversationRespVO;
|
|
|
import cn.iocoder.byzs.module.ai.controller.admin.chat.vo.message.AiChatMessageSendReqVO;
|
|
|
import cn.iocoder.byzs.module.ai.controller.admin.chat.vo.message.AiChatMessageSendRespVO;
|
|
|
import cn.iocoder.byzs.module.ai.controller.admin.image.vo.AiImageDrawReqVO;
|
|
|
import cn.iocoder.byzs.module.ai.controller.admin.image.vo.AiImageRespVO;
|
|
|
-import cn.iocoder.byzs.module.ai.dal.dataobject.chat.AiChatConversationDO;
|
|
|
+import cn.iocoder.byzs.module.ai.controller.admin.model.vo.chatRole.AiChatRolePageReqVO;
|
|
|
+import cn.iocoder.byzs.module.ai.controller.admin.model.vo.chatRole.AiChatRoleRespVO;
|
|
|
import cn.iocoder.byzs.module.ai.dal.dataobject.image.AiImageDO;
|
|
|
+import cn.iocoder.byzs.module.ai.dal.dataobject.model.AiChatRoleDO;
|
|
|
import cn.iocoder.byzs.module.ai.service.chat.AiChatConversationService;
|
|
|
import cn.iocoder.byzs.module.ai.service.chat.AiChatMessageService;
|
|
|
import cn.iocoder.byzs.module.ai.service.image.AiImageService;
|
|
|
-import cn.iocoder.byzs.module.bjdx.controller.admin.course.vo.CoursePageReqVO;
|
|
|
-import cn.iocoder.byzs.module.bjdx.controller.admin.coursetype.vo.CourseTypeListReqVO;
|
|
|
-import cn.iocoder.byzs.module.bjdx.controller.admin.coursetype.vo.CourseTypeRespVO;
|
|
|
-import cn.iocoder.byzs.module.bjdx.dal.dataobject.coursetype.CourseTypeDO;
|
|
|
-import cn.iocoder.byzs.module.bjdx.service.course.CourseService;
|
|
|
-import cn.iocoder.byzs.module.bjdx.service.coursetype.CourseTypeService;
|
|
|
+import cn.iocoder.byzs.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;
|
|
|
@@ -41,8 +38,6 @@ import static cn.iocoder.byzs.framework.security.core.util.SecurityFrameworkUtil
|
|
|
@RestController
|
|
|
@RequestMapping("/bjdxWeb/ai")
|
|
|
@Validated
|
|
|
-@PermitAll
|
|
|
-@TenantIgnore
|
|
|
public class WebAiController {
|
|
|
|
|
|
@Resource
|
|
|
@@ -51,6 +46,8 @@ public class WebAiController {
|
|
|
private AiChatMessageService chatMessageService;
|
|
|
@Resource
|
|
|
private AiImageService imageService;
|
|
|
+ @Resource
|
|
|
+ private AiChatRoleService chatRoleService;
|
|
|
|
|
|
private Long userId = 1L;
|
|
|
|
|
|
@@ -84,12 +81,16 @@ public class WebAiController {
|
|
|
|
|
|
// ================ 绘图管理 ================
|
|
|
|
|
|
+ @PermitAll
|
|
|
+ @TenantIgnore
|
|
|
@Operation(summary = "生成图片")
|
|
|
@PostMapping("/create-painting")
|
|
|
public CommonResult<Long> drawImage(@Valid @RequestBody AiImageDrawReqVO drawReqVO) {
|
|
|
return success(imageService.drawImage(userId, drawReqVO));
|
|
|
}
|
|
|
|
|
|
+ @PermitAll
|
|
|
+ @TenantIgnore
|
|
|
@GetMapping("/painting-get-my")
|
|
|
@Operation(summary = "绘画-获取绘图记录")
|
|
|
public CommonResult<AiImageRespVO> getImageMy(@RequestParam("id") Long id) {
|
|
|
@@ -99,4 +100,25 @@ public class WebAiController {
|
|
|
}
|
|
|
return success(BeanUtils.toBean(image, AiImageRespVO.class));
|
|
|
}
|
|
|
+
|
|
|
+ @PermitAll
|
|
|
+ @TenantIgnore
|
|
|
+ @GetMapping("/painting-get-mys")
|
|
|
+ @Operation(summary = "获取【我的】绘图记录列表")
|
|
|
+ @Parameter(name = "ids", required = true, description = "绘画编号数组", example = "1024,2048")
|
|
|
+ public CommonResult<List<AiImageRespVO>> getImageListMyByIds(@RequestParam("ids") List<Long> ids) {
|
|
|
+ List<AiImageDO> imageList = imageService.getImageList(ids);
|
|
|
+// imageList.removeIf(item -> !ObjUtil.equal(getLoginUserId(), item.getUserId()));
|
|
|
+ return success(BeanUtils.toBean(imageList, AiImageRespVO.class));
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ @GetMapping("/selectRoleModel")
|
|
|
+ @Operation(summary = "获得聊天角色分页")
|
|
|
+ public CommonResult<PageResult<AiChatRoleRespVO>> selectRoleModel() {
|
|
|
+ AiChatRolePageReqVO pageReqVO = new AiChatRolePageReqVO();
|
|
|
+ pageReqVO.setCategory("老师");
|
|
|
+ PageResult<AiChatRoleDO> pageResult = chatRoleService.getChatRoleMyPage(pageReqVO, getLoginUserId());
|
|
|
+ return success(BeanUtils.toBean(pageResult, AiChatRoleRespVO.class));
|
|
|
+ }
|
|
|
}
|