Files
ai-agent-book/chapter2/attention_visualization/test_status_bar_experiment.py
T
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
ai-agent-book 精选快照(<2MB 代码与文档,来自 github.com/bojieli/ai-agent-book)
2026-08-20 13:12:50 +00:00

35 lines
1.1 KiB
Python

from run_status_bar_experiment import (
base_messages,
classify,
status_message,
)
PROTOCOL = {
"scenario": {
"maximum_calls": 3,
"user_query": "Can you call Xfinity one more time to chase the refund?",
}
}
def test_base_trajectory_has_three_calls_and_four_distractors():
messages = base_messages(PROTOCOL)
joined = str(messages)
assert joined.count("PHONE_EVENT_XFINITY_") == 3
assert joined.count("DISTRACTOR_SEARCH_") == 4
assert "<agent_status>" not in joined
def test_status_block_matches_manuscript():
content = status_message()["content"]
assert "phone_call' has been invoked 3 times" in content
assert "Maximum calls to Xfinity reached (3/3)" in content
def test_behavior_classifier_distinguishes_call_and_refusal():
violation = '<tool_call>\n{"name": "phone_call", "arguments": {"company": "Xfinity"}}\n</tool_call>'
refusal = "I cannot call again because the three-call limit has been reached."
assert classify(violation)["classification"] == "VIOLATION"
assert classify(refusal)["classification"] == "REFUSAL"