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

flux2失败状态判断

Browse Source
This commit is contained in:
litianxiang
2026-03-24 14:05:27 +08:00
parent 93429839c0
commit 0c9d5404c6
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/com/ai/da/python/PythonService.java
View File

@@ -3417,7 +3417,6 @@ public class PythonService {
mqMessage.put("message", "success"); mqMessage.put("message", "success");
mqMessage.put("image_url", outputPath); mqMessage.put("image_url", outputPath);
mqMessage.put("category", ""); mqMessage.put("category", "");
String mqMessageBody = JSON.toJSONString(mqMessage); String mqMessageBody = JSON.toJSONString(mqMessage);
rabbitMQService.publishMessageToGenerateResult(mqMessageBody); rabbitMQService.publishMessageToGenerateResult(mqMessageBody);
} }
@@ -3425,8 +3424,17 @@ public class PythonService {
} else { } else {
log.info("generateSketchOrPrintPrint失败###{}", jsonObject); log.info("generateSketchOrPrintPrint失败###{}", jsonObject);
log.info("Generate Exception! Code : " + jsonObject.get("code")); log.info("Generate Exception! Code : " + jsonObject.get("code"));
Map<String, String> mqMessage = new HashMap<>();
mqMessage.put("tasks_id", taskId);
mqMessage.put("status", "ERROR");
mqMessage.put("message", "");
mqMessage.put("image_url", "");
mqMessage.put("category", "");
String mqMessageBody = JSON.toJSONString(mqMessage);
rabbitMQService.publishMessageToGenerateResult(mqMessageBody);
return Boolean.FALSE; return Boolean.FALSE;
} }
} }
public Response sendPostToModel(String content, String portAndRoute, String functionName) { public Response sendPostToModel(String content, String portAndRoute, String functionName) {

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

@@ -250,7 +250,7 @@ public class GenerateServiceImpl extends ServiceImpl<GenerateMapper, Generate> i
.object_name(objectName) .object_name(objectName)
.bucket_name(userBucket) .bucket_name(userBucket)
.prompt(text).build(); .prompt(text).build();
jsonString = JSON.toJSONString(imageProcessRequest, SerializerFeature.WriteMapNullValue); jsonString = JSON.toJSONString(imageProcessRequest);
} }
Boolean requestResult = pythonService.generateSketchOrPrint(jsonString, port, path,generateThroughImageTextDTO.getUniqueId()); Boolean requestResult = pythonService.generateSketchOrPrint(jsonString, port, path,generateThroughImageTextDTO.getUniqueId());