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

chapter4/perception-tools/test_expanded_catalog.py

@@ -0,0 +1,60 @@
+"""Offline contract checks for the real-backed 126-tool perception catalog."""
+
+from __future__ import annotations
+
+import asyncio
+import json
+import sys
+from pathlib import Path
+
+import tiktoken
+
+SRC = Path(__file__).resolve().parent / "src"
+sys.path.insert(0, str(SRC))
+
+import expanded_catalog  # noqa: E402
+import main  # noqa: E402
+
+
+def _schemas() -> list[dict]:
+    tools = asyncio.run(main.mcp.list_tools())
+    return [tool.model_dump(by_alias=True, exclude_none=True, mode="json")
+            for tool in tools]
+
+
+def test_catalog_has_126_unique_complete_schemas_over_50k_tokens():
+    schemas = _schemas()
+    names = {schema["name"] for schema in schemas}
+    rendered = "\n".join(json.dumps(schema, ensure_ascii=False, indent=2)
+                         for schema in schemas)
+    assert len(schemas) == len(names) == 126
+    assert len(expanded_catalog.EXPANDED_SPECS) == 70
+    assert len(expanded_catalog.EXISTING_TOOL_CONTRACTS) == 56
+    assert len(tiktoken.get_encoding("o200k_base").encode(rendered)) > 50_000
+    assert {
+        "web_search", "code_interpreter", "yfinance_quote", "search_news",
+        "arxiv_search", "arxiv_download", "github_list_contributors",
+    } <= names
+    assert all("success" in schema["description"].lower()
+               and "failure" in schema["description"].lower()
+               for schema in schemas)
+
+
+def test_expanded_parameter_descriptions_are_tool_specific():
+    schemas = {schema["name"]: schema for schema in _schemas()}
+    for spec in expanded_catalog.EXPANDED_SPECS:
+        properties = schemas[spec.name]["inputSchema"]["properties"]
+        assert spec.name in properties["query"]["description"]
+        assert spec.name in properties["options_json"]["description"]
+
+
+def test_code_interpreter_nonzero_exit_fails_closed():
+    spec = next(spec for spec in expanded_catalog.EXPANDED_SPECS
+                if spec.name == "code_interpreter")
+    receipt = asyncio.run(expanded_catalog.execute_expanded_tool(
+        spec, 'raise RuntimeError("expected")', '{"timeout": 5}'
+    ))
+    assert receipt["success"] is False
+    assert receipt["error_type"] == "ProcessExecutionError"
+    assert receipt["data"]["returncode"] != 0
+    assert "RuntimeError" in receipt["data"]["stderr"]