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

chapter5/agent-creator/reference_agent/tests/test_contract.py

@@ -0,0 +1,74 @@
+from __future__ import annotations
+
+import json
+import copy
+import sys
+from pathlib import Path
+from types import SimpleNamespace
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from agent import GeneratedAgent
+
+
+class FakeCompletions:
+    def __init__(self):
+        self.calls = []
+
+    def create(self, **kwargs):
+        self.calls.append(copy.deepcopy(kwargs))
+        if len(self.calls) == 1:
+            tool_call = SimpleNamespace(
+                id="call-1",
+                function=SimpleNamespace(
+                    name="lookup_domain_fact",
+                    arguments=json.dumps({"query": "purpose"}),
+                ),
+            )
+            message = SimpleNamespace(content=None, tool_calls=[tool_call])
+        else:
+            assert kwargs["messages"][-1]["role"] == "tool"
+            message = SimpleNamespace(content="Verified answer", tool_calls=[])
+        usage = SimpleNamespace(prompt_tokens=10, completion_tokens=3)
+        return SimpleNamespace(choices=[SimpleNamespace(message=message)], usage=usage)
+
+
+def test_standard_tool_loop_keeps_assistant_call_and_tool_result():
+    completions = FakeCompletions()
+    client = SimpleNamespace(chat=SimpleNamespace(completions=completions))
+    result = GeneratedAgent(model="test-model", client=client).run("What is your purpose?")
+    assert result["ok"] is True
+    assert result["answer"] == "Verified answer"
+    second_messages = completions.calls[1]["messages"]
+    assert second_messages[-2]["role"] == "assistant"
+    assert second_messages[-2]["tool_calls"][0]["id"] == "call-1"
+    assert second_messages[-1]["role"] == "tool"
+    assert second_messages[-1]["tool_call_id"] == "call-1"
+    assert result["messages"][:-1] == second_messages
+    assert result["messages"][-1] == {
+        "role": "assistant",
+        "content": "Verified answer",
+    }
+    assert result["usage"] == {
+        "prompt_tokens": 20,
+        "cached_prompt_tokens": 0,
+        "completion_tokens": 6,
+        "requests": 2,
+    }
+
+
+def test_prior_multiturn_history_is_preserved_in_order():
+    completions = FakeCompletions()
+    client = SimpleNamespace(chat=SimpleNamespace(completions=completions))
+    history = [
+        {"role": "user", "content": "Remember owner Mei-Lin."},
+        {"role": "assistant", "content": "Owner Mei-Lin retained."},
+    ]
+
+    result = GeneratedAgent(model="test-model", client=client).run(
+        "Evaluate the release.", history=history
+    )
+
+    first_messages = completions.calls[0]["messages"]
+    assert first_messages[1:3] == history
+    assert result["messages"][1:3] == history

chapter5/agent-creator/reference_agent/tests/test_domain_policy.py

@@ -0,0 +1,63 @@
+from __future__ import annotations
+
+import json
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+from domain_tools import execute_tool
+
+
+ROOT = Path(__file__).resolve().parents[1]
+
+
+def load_spec():
+    return json.loads((ROOT / "domain_spec.json").read_text(encoding="utf-8"))
+
+
+def record(spec, *, identifier, required, status, evidence):
+    return {
+        spec["identifier_field"]: identifier,
+        spec["required_field"]: required,
+        spec["status_field"]: status,
+        spec["evidence_field"]: evidence,
+    }
+
+
+def test_required_nonpassing_record_refuses_with_exact_evidence():
+    spec = load_spec()
+    records = [
+        record(
+            spec,
+            identifier="required-check",
+            required=True,
+            status="failed",
+            evidence="observed failure",
+        )
+    ]
+    result = execute_tool(spec["tool_name"], {spec["records_argument"]: records})
+    assert result["ok"] is True
+    assert result["result"]["approved"] is False
+    assert result["result"]["decision"] == spec["rejected_label"]
+    assert result["result"]["failed_required_records"][0]["evidence"] == "observed failure"
+
+
+def test_only_required_nonpassing_records_block_approval():
+    spec = load_spec()
+    passing = spec["passing_values"][0]
+    records = [
+        record(spec, identifier="required", required=True, status=passing, evidence="ok"),
+        record(spec, identifier="optional", required=False, status="failed", evidence="optional"),
+    ]
+    result = execute_tool(spec["tool_name"], {spec["records_argument"]: records})
+    assert result["ok"] is True
+    assert result["result"]["approved"] is True
+    assert result["result"]["decision"] == spec["approved_label"]
+
+
+def test_missing_or_empty_records_fail_closed():
+    spec = load_spec()
+    result = execute_tool(spec["tool_name"], {})
+    assert result["ok"] is False
+    assert spec["records_argument"] in result["error"]