From 4395d672885b4fb40950cc7e3df6b2fb5ca4295d Mon Sep 17 00:00:00 2001 From: zcr Date: Tue, 24 Mar 2026 14:35:39 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20flux2=20=E5=A2=9E=E5=8A=A0=E7=8A=B6?= =?UTF-8?q?=E6=80=81=E7=A0=81=20;=20fix:?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/api/api_generate_image.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/app/api/api_generate_image.py b/app/api/api_generate_image.py index c3332b6..5f7ff4b 100644 --- a/app/api/api_generate_image.py +++ b/app/api/api_generate_image.py @@ -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(), ) - result = resp.json() - - logger.info(f"generate_image_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"generate_image_flux2_gen_img Run Exception @@@@@@:{e}")