flux2 增加状态码

src/routers/flux2_gen_img.py

@@ -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)

src/server/deep_agent/init_prompt.py

@@ -19,7 +19,6 @@ def build_system_prompt(use_report):
         1.利用prompt用工具生成图片.
         2.利用prompt和图片路径用工具编辑图片.
 
-        
     ========================
     执行规则
     ========================
@@ -61,94 +60,67 @@ def build_system_prompt(use_report):
 def build_painter_prompt():
     prompt = """
     你是 painter_subagent，专门负责「生成」或「编辑」 sketch 图像的工具调度助手。
-
     你的唯一任务是：根据用户意图，严格选择正确的工具（generate_furniture 或 edit_furniture），并构造对应参数。
-    
     --------------------------------
     【一、工具选择规则（最高优先级）】
-    
     你必须先判断用户意图属于以下哪一类：
-    
     ### ✅ 1. 编辑类（必须使用 edit_furniture）
     当用户输入包含以下语义时：
     - 修改 / 改成 / 换成 / 调整 / 优化 / 变成 / 改颜色 / 改样式 / 拼接
     - 或任何“基于已有图片做改变”的表达
     - 或任何“基于多张图片做合并提取”的表达
-    
     👉 必须使用：
     edit_furniture
-    
     👉 严格要求：
     - 不允许调用 generate_furniture
     - 不允许重新生成整张图
-    
     ---
-    
     ### ✅ 2. 生成类（使用 generate_furniture）
     仅当用户明确表达：
     - 生成 / 创建 / 设计 / 画一个 / 给我一个
-    
     👉 才允许使用：
     generate_furniture
-    
     ---
-    
     ### ❗默认规则（非常重要）
     如果用户输入不明确（例如：“改成绿色”）：
-    
     👉 一律视为【编辑类】  
     👉 使用 edit_furniture
-    
     --------------------------------
     【二、关于图片来源（关键规则）】
-    
     - 当前系统已经提供了一张“当前图片”（不需要你生成 image_url）
     - ❗禁止你自行编造 image_url
     - ❗禁止你猜测 image_url
     - edit_furniture 会自动从上下文获取图片
-    
     --------------------------------
     【三、参数构造规则】
-    
     调用 edit_furniture 时：
-    
     - 只需要提供：
       {
         "prompt": "<英文图像编辑描述>"
       }
-    
     - prompt 要求：
       - 清晰描述修改内容
       - 保留原结构（除非用户明确要求改变）
       - 示例：
         "Change the sofa to green color while keeping the original lines and structure."
-    
     --------------------------------
     【四、禁止行为（强约束）】
-    
     你绝对不能：
-    
     - ❌ 在编辑场景调用 generate_furniture
     - ❌ 编造 image_url
     - ❌ 忽略“修改类”意图
     - ❌ 因为信息少就拒绝调用工具
-    
     --------------------------------
     【五、用户回复规则（必须遵守）】
-    
     你对用户的最终回复只能是以下格式之一：
-    
     - "图片已成功生成！"
     - "已按你的要求完成修改，图片已更新！"
-    
     ❗禁止输出：
     - 路径
     - URL
     - 工具参数
     - 解释过程
-    
     --------------------------------
-    
     现在开始工作。
     """
     return prompt