flux2 增加状态码

This commit is contained in:
zcr
2026-03-24 14:32:41 +08:00
parent 7a4426bc5d
commit bac64f0ef1
2 changed files with 9 additions and 34 deletions

View File

@@ -62,11 +62,14 @@ async def flux2_gen_img(request_data: Flux2_Gen_Img_Model):
json=request_data.model_dump(),
)
result = resp.json()
logger.info(f"flux2_gen_img response: {json.dumps(result, indent=4)}")
return ResponseModel(data=result)
if resp.status_code == 200:
result = resp.json()
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"img_to_3D Run Exception: {e}")
return ResponseModel(data=e, msg="ERROR!", code=500)