aidlab/aida_back
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix : fast || high

Browse Source
This commit is contained in:
徐佩
2024-12-01 16:32:51 +08:00
parent a237b4041d
commit b836e7fd67
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/com/ai/da/model/dto/GenerateToPythonDTO.java
View File

@@ -29,12 +29,15 @@ public class GenerateToPythonDTO {
private String svg; private String svg;
public GenerateToPythonDTO(String tasks_id, String prompt, String image_url, String mode, String category, String gender) { private String version;
public GenerateToPythonDTO(String tasks_id, String prompt, String image_url, String mode, String category, String gender, String version) {
this.image_url = image_url; this.image_url = image_url;
this.category = category; this.category = category;
this.prompt = prompt; this.prompt = prompt;
this.mode = mode; this.mode = mode;
this.tasks_id = tasks_id; this.tasks_id = tasks_id;
this.gender = gender; this.gender = gender;
this.version = version;
} }
} }

8
src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java
View File

@@ -150,6 +150,14 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
String port = generateServicePort; String port = generateServicePort;
String jsonString = ""; String jsonString = "";
HashMap<String, String> params = new HashMap<>(); HashMap<String, String> params = new HashMap<>();
String version = null;
if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("high")){
version = "high";
params.put("version","high");
}else if (!StringUtil.isNullOrEmpty(generateThroughImageTextDTO.getVersion()) && generateThroughImageTextDTO.getVersion().equals("fast")){
version = "fast";
params.put("version","fast");
}
// 3.1 确定不同类型的印花分别调哪个接口 // 3.1 确定不同类型的印花分别调哪个接口
if (generateThroughImageTextDTO.getLevel1Type().equals(PRINT_BOARD.getRealName())) { if (generateThroughImageTextDTO.getLevel1Type().equals(PRINT_BOARD.getRealName())) {
switch (generateThroughImageTextDTO.getLevel2Type()) { switch (generateThroughImageTextDTO.getLevel2Type()) {