diff --git a/src/main/java/com/ai/da/common/enums/PythonToJavaApiOperationTypeEnum.java b/src/main/java/com/ai/da/common/enums/PythonToJavaApiOperationTypeEnum.java index a3c4bb6a..dbef6022 100644 --- a/src/main/java/com/ai/da/common/enums/PythonToJavaApiOperationTypeEnum.java +++ b/src/main/java/com/ai/da/common/enums/PythonToJavaApiOperationTypeEnum.java @@ -4,7 +4,7 @@ import java.util.stream.Stream; /** * @author yanglei - * @description python调用java 操作数类型 generatePrint ->生成印花 designCollection ->设计collection + * @description python调用java 操作数类型 generatePrint ->生成印花 designCollection ->设计collection generateSketch->设计草图 * @create 2022-10-3 17:33 **/ public enum PythonToJavaApiOperationTypeEnum { @@ -19,7 +19,12 @@ public enum PythonToJavaApiOperationTypeEnum { /** * 设计collection */ - DESIGN_COLLECTION("designCollection"); + DESIGN_COLLECTION("designCollection"), + + /** + * 生成草图 + */ + GENERATE_SKETCH("generateSketch"); private String realName; diff --git a/src/main/java/com/ai/da/controller/PythonController.java b/src/main/java/com/ai/da/controller/PythonController.java index 0c823be1..6791dace 100644 --- a/src/main/java/com/ai/da/controller/PythonController.java +++ b/src/main/java/com/ai/da/controller/PythonController.java @@ -46,7 +46,8 @@ public class PythonController { @PostMapping("/saveGeneratePicture") public Response upload(@RequestParam("file") MultipartFile file, @ApiParam("操作类型 generatePrint ->生成印花 " + - "designCollection ->设计collection generateAdvancedDesign ->生成高级design") + "designCollection ->设计collection generateAdvancedDesign ->生成高级design"+ + "generateSketch -> 生成草图") @RequestParam(value = "operateType") String operateType) { return Response.success(pythonService.upload(file, operateType)); }