31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from haiku.rag.agents.research.prompts import (
|
|
ITERATIVE_PLAN_PROMPT,
|
|
ITERATIVE_PLAN_PROMPT_WITH_CONTEXT,
|
|
)
|
|
|
|
|
|
def test_iterative_plan_prompt_proposes_first_question():
|
|
"""ITERATIVE_PLAN_PROMPT should instruct to propose the first question."""
|
|
assert "first question" in ITERATIVE_PLAN_PROMPT.lower()
|
|
assert "is_complete=False" in ITERATIVE_PLAN_PROMPT
|
|
|
|
|
|
def test_iterative_plan_prompt_with_context_evaluates_evidence():
|
|
"""ITERATIVE_PLAN_PROMPT_WITH_CONTEXT should evaluate prior answers."""
|
|
assert "prior_answers" in ITERATIVE_PLAN_PROMPT_WITH_CONTEXT
|
|
assert (
|
|
"evaluat" in ITERATIVE_PLAN_PROMPT_WITH_CONTEXT.lower()
|
|
) # matches evaluate/evaluating
|
|
|
|
|
|
def test_prompt_selection_uses_context_prompt_with_prior_answers():
|
|
"""When prior_answers exist, should use ITERATIVE_PLAN_PROMPT_WITH_CONTEXT."""
|
|
has_prior_answers = True
|
|
|
|
effective_plan_prompt = (
|
|
ITERATIVE_PLAN_PROMPT_WITH_CONTEXT
|
|
if has_prior_answers
|
|
else ITERATIVE_PLAN_PROMPT
|
|
)
|
|
|
|
assert effective_plan_prompt == ITERATIVE_PLAN_PROMPT_WITH_CONTEXT
|