diff --git a/src/routers/flux2_gen_img.py b/src/routers/flux2_gen_img.py index b8f8d10..fc1806f 100644 --- a/src/routers/flux2_gen_img.py +++ b/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) diff --git a/src/server/deep_agent/init_prompt.py b/src/server/deep_agent/init_prompt.py index 61669a4..a06144e 100644 --- a/src/server/deep_agent/init_prompt.py +++ b/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