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

TASK:batch toProductImage;chatStream;

Browse Source
This commit is contained in:
shahaibo
2025-06-05 14:00:38 +08:00
parent 12dbe90150
commit 4e80493a26
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/com/ai/da/common/RabbitMQ/GenerateConsumer.java
View File

@@ -375,16 +375,16 @@ public class GenerateConsumer {
log.info("toProductImageBatch response : {}", generateResult); log.info("toProductImageBatch response : {}", generateResult);
try { try {
log.info("tasks_id : {} start ", generateResult.get("task_id")); log.info("tasks_id : {} start ", generateResult.get("tasks_id"));
if (!StringUtils.isEmpty(generateResult.getString("progress"))) { if (!StringUtils.isEmpty(generateResult.getString("progress"))) {
String progress = generateResult.getString("progress"); String progress = generateResult.getString("progress");
JSONArray result = generateResult.getJSONArray("result"); JSONObject result = generateResult.getJSONObject("result_data");
String url = null; String url = null;
if (!StringUtils.isEmpty(result)) { if (!StringUtils.isEmpty(result)) {
url = result.getString(0); url = result.getString("image_url");
String taskId = generateResult.getString("tasks_id");
userLikeGroupService.toProductBatch(taskId, url, progress);
} }
String taskId = generateResult.getString("task_id");
userLikeGroupService.toProductBatch(taskId, url, progress);
} else { } else {
// 修改redis中的数据状态为exception // 修改redis中的数据状态为exception
String key = toProductImageResultKey + ":" + generateResult.get("task_id"); String key = toProductImageResultKey + ":" + generateResult.get("task_id");