feat: align product suite prompts with reference images
This commit is contained in:
@@ -168,11 +168,10 @@ class ProductSuiteTests(unittest.TestCase):
|
||||
template_text=template,
|
||||
)
|
||||
self.assertIn(
|
||||
"参考图规则:使用第一張上傳圖作為主商品圖,其餘圖片只作為參考。",
|
||||
"参考图规则:当前上传图片是本任务唯一主参考图;保持商品主体、款式、颜色和关键细节准确;不编造用户与参考图均未提供的信息。",
|
||||
single_primary,
|
||||
)
|
||||
self.assertNotIn("当前上传图片是本任务唯一主参考图", single_primary)
|
||||
self.assertNotIn("参考图规则:使用", single_primary)
|
||||
self.assertNotIn("第2至", single_primary)
|
||||
|
||||
settings["per_image_primary"] = True
|
||||
per_image_primary = product_suite.build_suite_prompt(
|
||||
@@ -187,9 +186,23 @@ class ProductSuiteTests(unittest.TestCase):
|
||||
"参考图规则:当前上传图片是本任务唯一主参考图;保持商品主体、款式、颜色和关键细节准确;不编造用户与参考图均未提供的信息。",
|
||||
per_image_primary,
|
||||
)
|
||||
self.assertNotIn("使用第一張上傳圖作為主商品圖", per_image_primary)
|
||||
self.assertNotIn("第2至", per_image_primary)
|
||||
self.assertNotIn("序号2", per_image_primary)
|
||||
|
||||
multi_reference = product_suite.build_suite_prompt(
|
||||
"卖点",
|
||||
product_suite.default_suite_settings(),
|
||||
"场景图",
|
||||
"51100639510",
|
||||
reference_asset_count=2,
|
||||
template_text=template,
|
||||
)
|
||||
self.assertIn(
|
||||
"参考图规则:第1张为主商品图,请保持其主体、款式、颜色和关键细节准确;第2至3张仅作为风格、构图、场景或排版参考,不得替换主商品;不编造用户与参考图均未提供的信息。",
|
||||
multi_reference,
|
||||
)
|
||||
self.assertEqual(7, product_suite.reference_asset_count({}, 9))
|
||||
|
||||
def test_product_suite_template_validation_and_custom_category_rendering(self):
|
||||
default_text = prompts.load_default_product_suite_prompt()
|
||||
self.assertEqual([], product_suite.product_suite_prompt_errors(default_text))
|
||||
|
||||
Reference in New Issue
Block a user