feat(product-suite): include category in target prompt

This commit is contained in:
chengma
2026-07-17 16:12:03 +08:00
parent ff0ad8bb73
commit 43d50ca629
8 changed files with 14 additions and 10 deletions
+4 -4
View File
@@ -117,11 +117,11 @@ class ProductSuiteTests(unittest.TestCase):
)
self.assertIn(
"生成目标:生成 Shopee 台灣商品白底主圖,純白背景,商品清晰置中,不添加多餘文字。",
"生成目标:白底图,生成 Shopee 台灣商品白底主圖,純白背景,商品清晰置中,不添加多餘文字。",
specs[0]["prompt"],
)
self.assertIn(
"生成目标:生成生活化使用場景圖,展示商品在真實情境中的用途與氛圍,畫面自然可信。",
"生成目标:场景图,生成生活化使用場景圖,展示商品在真實情境中的用途與氛圍,畫面自然可信。",
specs[1]["prompt"],
)
white_prompt = specs[0]["prompt"]
@@ -205,12 +205,12 @@ class ProductSuiteTests(unittest.TestCase):
"draft_hidden",
)
rendered = product_suite.render_product_suite_prompt(default_text, context)
self.assertIn("生成目标:生成自定义分类图片:尺寸图。", rendered)
self.assertIn("生成目标:尺寸图,生成自定义分类图片:尺寸图。", rendered)
self.assertNotIn("draft_", rendered)
self.assertNotIn("套图名称:", rendered)
legacy_template = default_text.replace(
"生成目标:{生成目标}",
"生成目标:{套图名称},{生成目标}",
"套图名称:{套图名称}{补充描述}",
)
self.assertEqual([], product_suite.product_suite_prompt_errors(legacy_template))