TASK:batch toProductImage;chatStream;

This commit is contained in:
shahaibo
2025-06-05 17:06:11 +08:00
parent 38ca810068
commit 249351bf52
4 changed files with 97 additions and 65 deletions

View File

@@ -4314,7 +4314,7 @@ public class PythonService {
throw new BusinessException("toProductImage.interface.exception");
}
public Boolean relightBatch(String url, String taskId, String prompt, String direction, String relightType) {
public Boolean relightBatch(String taskIdBatch, List<BatchParamDTO> paramList, String userId) {
// todo 限流校验
// AccessLimitUtils.validate("design",5);
OkHttpClient client = new OkHttpClient().newBuilder()
@@ -4326,12 +4326,9 @@ public class PythonService {
MediaType mediaType = MediaType.parse("application/json");
//关闭FastJson的引用检测 防止出现$ref 现象
Map<String, Object> map = new HashMap<>();
map.put("tasks_id", taskId);
map.put("image_url", url);
map.put("prompt", prompt);
map.put("direction", direction);
map.put("product_type", relightType);
map.put("batch_size", 1);
map.put("batch_tasks_id", taskIdBatch);
map.put("user_id", userId);
map.put("batch_data_list", paramList);
log.info("relightImage请求python 参数:####{}", map);
String param = JSON.toJSONString(map, SerializerFeature.WriteNullStringAsEmpty);
log.info(param);