feat: flux2 增加状态码 ; fix:
This commit is contained in:
@@ -63,10 +63,14 @@ async def generate_image_flux2_klein(request_item: GenerateImageFlux2KleinModel)
|
||||
f"http://{settings.FLUX2_GEN_IMG_MODEL_URL}/predict",
|
||||
json=request_item.model_dump(),
|
||||
)
|
||||
if resp.status_code == 200:
|
||||
result = resp.json()
|
||||
|
||||
logger.info(f"generate_image_flux2_gen_img response: {json.dumps(result, indent=4)}")
|
||||
logger.info(f"flux2_gen_img response: {json.dumps(result, indent=4)}")
|
||||
return ResponseModel(data=result)
|
||||
else:
|
||||
error = resp.json()
|
||||
logger.info(f"flux2_gen_img response: {json.dumps(error, indent=4)}")
|
||||
return ResponseModel(data=error, msg="ERROR!", code=500)
|
||||
|
||||
except Exception as e:
|
||||
logger.warning(f"generate_image_flux2_gen_img Run Exception @@@@@@:{e}")
|
||||
|
||||
Reference in New Issue
Block a user