chengma/cmshoppe
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: rename bundled cover prompt template

Browse Source
This commit is contained in:
chengma
2026-07-17 18:24:58 +08:00
parent d1cd50f5c5
commit d853db1c8a
8 changed files with 93 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_prompts.py
+16 -2
View File
@@ -120,10 +120,10 @@ class PromptTests(TempDirMixin, unittest.TestCase):
prompts.load_title_prompt(title_path),
prompts.load_title_template("默认", title_templates_dir),
)
self.assertEqual(["papa1"], prompts.list_cover_templates(cover_dir))
self.assertEqual(["cover_prompt"], prompts.list_cover_templates(cover_dir))
self.assertIn(
"商品标题:{新标题}",
prompts.load_cover_template("papa1", cover_dir),
prompts.load_cover_template("cover_prompt", cover_dir),
)
self.assert_removed(temp_dir)
@@ -153,6 +153,20 @@ class PromptTests(TempDirMixin, unittest.TestCase):
self.assert_removed(temp_dir)
def test_ensure_default_prompts_keeps_legacy_papa1_template_unchanged(self):
with self.make_temp_dir() as temp_dir:
title_path = os.path.join(temp_dir, "title_prompt.txt")
cover_dir = os.path.join(temp_dir, "prompts", "cover")
prompts.save_cover_template("papa1", "历史封面提示词", cover_dir)
prompts.ensure_default_prompts(title_path, cover_dir)
self.assertEqual(["papa1"], prompts.list_cover_templates(cover_dir))
self.assertEqual("历史封面提示词", prompts.load_cover_template("papa1", cover_dir))
self.assertFalse(os.path.exists(os.path.join(cover_dir, "cover_prompt.txt")))
self.assert_removed(temp_dir)
def test_ensure_default_prompts_accepts_explicit_title_templates_dir(self):
with self.make_temp_dir() as temp_dir:
title_path = os.path.join(temp_dir, "custom", "title_prompt.txt")