ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter2/local_llm_serving/test_run_experiment.py

@@ -0,0 +1,42 @@
+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