33 lines
997 B
Python
33 lines
997 B
Python
|
"""Default prompts: breaking news, ignore futures/market tape."""
|
||
|
|
|
||
|
|
from summarizer import MAP_PROMPT_DEFAULT, RECAP_PROMPT_DEFAULT, SUMMARY_PROMPT_DEFAULT
|
||
|
|
|
||
|
|
|
||
|
|
def _assert_breaking_not_futures(prompt: str) -> None:
|
||
|
|
p = prompt.lower()
|
||
|
|
assert "breaking" in p
|
||
|
|
assert "futures" in p
|
||
|
|
assert "ignore" in p or "do not" in p or "not" in p
|
||
|
|
assert "es=f" not in p
|
||
|
|
assert "yfinance" not in p
|
||
|
|
|
||
|
|
|
||
|
|
def test_map_prompt_focuses_on_breaking_news_not_futures():
|
||
|
|
_assert_breaking_not_futures(MAP_PROMPT_DEFAULT)
|
||
|
|
|
||
|
|
|
||
|
|
def test_summary_prompt_focuses_on_breaking_news_not_futures():
|
||
|
|
_assert_breaking_not_futures(SUMMARY_PROMPT_DEFAULT)
|
||
|
|
p = SUMMARY_PROMPT_DEFAULT.lower()
|
||
|
|
assert "commodity" in p or "market" in p
|
||
|
|
|
||
|
|
|
||
|
|
def test_recap_prompt_is_daily_24h_breaking_news():
|
||
|
|
p = RECAP_PROMPT_DEFAULT.lower()
|
||
|
|
_assert_breaking_not_futures(RECAP_PROMPT_DEFAULT)
|
||
|
|
assert "daily" in p
|
||
|
|
assert "24" in p
|
||
|
|
assert "summary_en" in p
|
||
|
|
assert "ticker" in p
|
||
|
|
assert "map_items" in p
|