liqiang
b119135836
Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""conversation_history_limit=0 must omit history, not include all via [-0:]."""
|
|
from types import SimpleNamespace
|
|
|
|
from agent import AgenticRAG
|
|
|
|
|
|
def test_history_limit_zero_omits_history():
|
|
agent = object.__new__(AgenticRAG)
|
|
agent.config = SimpleNamespace(
|
|
agent=SimpleNamespace(conversation_history_limit=0)
|
|
)
|
|
agent._get_system_prompt = lambda: "sys"
|
|
agent.conversation_history = [
|
|
{"role": "user", "content": "old1"},
|
|
{"role": "assistant", "content": "old2"},
|
|
]
|
|
msgs = agent._build_messages("new question")
|
|
assert msgs == [
|
|
{"role": "system", "content": "sys"},
|
|
{"role": "user", "content": "new question"},
|
|
]
|
|
|
|
|
|
def test_positive_history_limit_still_keeps_tail():
|
|
agent = object.__new__(AgenticRAG)
|
|
agent.config = SimpleNamespace(
|
|
agent=SimpleNamespace(conversation_history_limit=1)
|
|
)
|
|
agent._get_system_prompt = lambda: "sys"
|
|
agent.conversation_history = [
|
|
{"role": "user", "content": "old1"},
|
|
{"role": "assistant", "content": "old2"},
|
|
]
|
|
msgs = agent._build_messages("new question")
|
|
assert msgs == [
|
|
{"role": "system", "content": "sys"},
|
|
{"role": "assistant", "content": "old2"},
|
|
{"role": "user", "content": "new question"},
|
|
]
|