add prompt assistant

This commit is contained in:
2026-05-07 09:51:47 -06:00
parent a46848b1c9
commit 22c3b65e28
4 changed files with 115 additions and 8 deletions

View File

@@ -297,10 +297,16 @@ class PromptHistoryTest(TestCase):
self.assertIsNone(data[0]["code_js"])
from prompt.llm import parse_guidance_response
from prompt.llm import GUIDANCE_SYSTEM_PROMPT, parse_guidance_response
class ParseGuidanceResponseTest(TestCase):
def test_guidance_prompt_asks_for_bold_keywords(self):
self.assertIn("Markdown", GUIDANCE_SYSTEM_PROMPT)
self.assertIn("[READY] 不要加粗", GUIDANCE_SYSTEM_PROMPT)
for keyword in ("**主题**", "**视觉**", "**交互**", "**内容**", "**可以生成**"):
self.assertIn(keyword, GUIDANCE_SYSTEM_PROMPT)
def test_ready_prefix_with_newline_stripped(self):
content, is_ready = parse_guidance_response("[READY]\n很好,可以生成了!")
self.assertEqual(content, "很好,可以生成了!")