wehub-resource-sync
48 lines
2.6 KiB
Python
48 lines
2.6 KiB
Python
from services.chat.prompts import _trim_block, build_system_prompt
|
|
|
|
|
|
def test_trim_block_returns_empty_for_blank_text():
|
|
assert _trim_block("label", "") == ""
|
|
assert _trim_block("label", " \n\t") == ""
|
|
|
|
|
|
def test_build_system_prompt_includes_trimmed_memory_blocks():
|
|
system_prompt = build_system_prompt(
|
|
presentation_memory_context=" Prior deck decision ",
|
|
chat_memory_context="\nEarlier user request\n",
|
|
)
|
|
|
|
assert "Deck memory (background only; may be partial or stale):" in system_prompt
|
|
assert "Chat memory (earlier messages in this conversation):" in system_prompt
|
|
assert "Common prompts" in system_prompt
|
|
assert "If grouped text or elements still cannot be moved cleanly inside the group" in system_prompt
|
|
assert "\nPrior deck decision\n" in system_prompt
|
|
assert "\nEarlier user request\n" in system_prompt
|
|
|
|
|
|
def test_build_system_prompt_omits_empty_memory_blocks():
|
|
system_prompt = build_system_prompt("", " ")
|
|
|
|
assert "Deck memory (background only; may be partial or stale):" not in system_prompt
|
|
assert "Chat memory (earlier messages in this conversation):" not in system_prompt
|
|
assert "Tool Protocol" in system_prompt
|
|
assert "getAvailableBlocks" in system_prompt
|
|
|
|
|
|
def test_system_prompt_preserves_existing_content_for_additive_edits():
|
|
system_prompt = build_system_prompt("", "")
|
|
|
|
assert "Treat add/insert/include requests as additive" in system_prompt
|
|
assert "preserve existing substantive charts, tables, images, text, icons" in system_prompt
|
|
assert "prioritize copying and adapting an existing block/component shape" in system_prompt
|
|
assert "Use primitive elements only when the task cannot be achieved from existing blocks" in system_prompt
|
|
assert "Treat existing template blocks as the default style source" in system_prompt
|
|
assert "search for reusable title/header text blocks first" in system_prompt
|
|
assert "Prefer this over fetching a whole layout or schema just to find one block" in system_prompt
|
|
assert "For new table or chart requests, getAvailableBlocks is mandatory before inserting" in system_prompt
|
|
assert "Do not use addElement to create a new table or chart while a reusable block exists" in system_prompt
|
|
assert "Do not add primitive title, header, subtitle" in system_prompt
|
|
assert "compose it from reusable blocks in this order: title/header block" in system_prompt
|
|
assert "update or add only those requested text elements" in system_prompt
|
|
assert "Use deleteElement, deleteComponent, or deleteSlide only when deletion is explicitly requested" in system_prompt
|