feat: fix product suite reference prompt rule

This commit is contained in:
chengma
2026-07-17 17:27:08 +08:00
parent e4f0733333
commit e77b16ae2e
6 changed files with 48 additions and 39 deletions
+5 -1
View File
@@ -449,7 +449,11 @@ class ProductSuiteGuiTests(TempDirMixin, unittest.TestCase):
), mock.patch.object(tab, "_start_thread", return_value=object()):
self.assertTrue(tab.start_generation(state))
self.assertEqual(expected, state.worker.job_specs[0]["prompt"])
self.assertIn("第2至3张仅作为风格、构图、场景或排版参考", expected)
self.assertIn(
"参考图规则:使用第一張上傳圖作為主商品圖,其餘圖片只作為參考。",
expected,
)
self.assertNotIn("第2至", expected)
state.worker = None
state.thread = None