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
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
import json
|
|
|
|
from run_experiment import normalize_tool_call, parse_tool_calls, sha256_text
|
|
|
|
|
|
def test_parse_multiple_raw_tool_calls():
|
|
raw = (
|
|
'<tool_call>\n{"name":"get_current_time","arguments":{"city":"Vancouver"}}\n</tool_call>'
|
|
'<tool_call>\n{"name":"get_weather","arguments":{"city":"Vancouver"}}\n</tool_call>'
|
|
)
|
|
calls = parse_tool_calls(raw)
|
|
assert [item["name"] for item in calls] == ["get_current_time", "get_weather"]
|
|
|
|
|
|
def test_normalize_small_model_city_arguments():
|
|
time_call = normalize_tool_call(
|
|
{"name": "get_current_time", "arguments": {"city": "Vancouver"}}
|
|
)
|
|
weather_call = normalize_tool_call(
|
|
{"name": "get_weather", "arguments": {"city": "Vancouver"}}
|
|
)
|
|
assert time_call == {
|
|
"name": "get_current_time",
|
|
"arguments": {"timezone": "America/Vancouver"},
|
|
}
|
|
assert weather_call == {
|
|
"name": "get_current_temperature",
|
|
"arguments": {"location": "Vancouver, Canada", "unit": "celsius"},
|
|
}
|
|
|
|
|
|
def test_normalize_does_not_hide_an_explicitly_wrong_timezone():
|
|
call = normalize_tool_call(
|
|
{"name": "get_current_time", "arguments": {"timezone": "America/New_York"}}
|
|
)
|
|
assert call["arguments"]["timezone"] == "America/New_York"
|
|
|
|
|
|
def test_protocol_is_valid_json_and_hashable():
|
|
raw = b'{"experiment_id":"2-1"}'
|
|
assert json.loads(raw)["experiment_id"] == "2-1"
|
|
assert len(sha256_text(raw.decode())) == 64
|