aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: flux2 增加状态码 ; fix:

Browse Source
This commit is contained in:
zcr
2026-03-24 14:35:39 +08:00
committed by zchen
parent 674514ec11
commit 4395d67288
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/api/api_generate_image.py
View File

@@ -63,10 +63,14 @@ async def generate_image_flux2_klein(request_item: GenerateImageFlux2KleinModel)
f"http://{settings.FLUX2_GEN_IMG_MODEL_URL}/predict", f"http://{settings.FLUX2_GEN_IMG_MODEL_URL}/predict",
json=request_item.model_dump(), json=request_item.model_dump(),
) )
result = resp.json() 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) 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: except Exception as e:
logger.warning(f"generate_image_flux2_gen_img Run Exception @@@@@@:{e}") logger.warning(f"generate_image_flux2_gen_img Run Exception @@@@@@:{e}")