feat(product-suite): include category in target prompt
This commit is contained in:
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user