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

chapter10/staged-system-prompt/evidence.py

@@ -0,0 +1,91 @@
+"""Credential-free raw provider receipts for staged system-prompt add-on."""
+
+from __future__ import annotations
+
+import json
+import time
+from datetime import datetime, timezone
+from pathlib import Path
+from typing import Any
+
+
+def _jsonable(value: Any) -> Any:
+    if hasattr(value, "model_dump"):
+        return value.model_dump(mode="json")
+    return json.loads(json.dumps(value, ensure_ascii=False, default=str))
+
+
+class RecordingCompletions:
+    def __init__(self, owner: "RecordingClient", inner: Any) -> None:
+        self.owner = owner
+        self.inner = inner
+
+    def create(self, **kwargs: Any) -> Any:
+        call_index = len(self.owner.receipts) + 1
+        started = time.perf_counter()
+        record = {
+            "schema_version": "1.0",
+            "call_index": call_index,
+            "started_at_utc": datetime.now(timezone.utc).isoformat(),
+            "provider": self.owner.provider,
+            "base_url": self.owner.base_url,
+            "context": dict(self.owner.context),
+            "request": _jsonable(kwargs),
+            "response": None,
+            "error": None,
+        }
+        try:
+            response = self.inner.create(**kwargs)
+            record["response"] = _jsonable(response)
+            return response
+        except Exception as exc:
+            record["error"] = {
+                "type": type(exc).__name__,
+                "message": str(exc),
+            }
+            raise
+        finally:
+            record["duration_s"] = round(time.perf_counter() - started, 3)
+            path = self.owner.receipt_dir / f"call_{call_index:03d}.json"
+            path.write_text(json.dumps(record, ensure_ascii=False, indent=2), encoding="utf-8")
+            record["receipt_path"] = str(path)
+            self.owner.receipts.append(record)
+
+
+class RecordingChat:
+    def __init__(self, owner: "RecordingClient", inner: Any) -> None:
+        self.completions = RecordingCompletions(owner, inner.completions)
+
+
+class RecordingClient:
+    """Small OpenAI-client proxy that persists each real request and response."""
+
+    def __init__(self, inner: Any, receipt_dir: Path, *, provider: str, base_url: str) -> None:
+        self.receipt_dir = receipt_dir
+        self.receipt_dir.mkdir(parents=True, exist_ok=True)
+        self.provider = provider
+        self.base_url = base_url
+        self.context: dict[str, Any] = {}
+        self.receipts: list[dict[str, Any]] = []
+        self.chat = RecordingChat(self, inner.chat)
+
+    def set_context(self, **context: Any) -> None:
+        self.context = context
+
+    def usage(self) -> dict[str, int]:
+        totals = {
+            "prompt_tokens": 0,
+            "cached_prompt_tokens": 0,
+            "completion_tokens": 0,
+            "requests": 0,
+        }
+        for receipt in self.receipts:
+            response = receipt.get("response") or {}
+            usage = response.get("usage") or {}
+            details = usage.get("prompt_tokens_details") or {}
+            if usage:
+                totals["requests"] += 1
+            totals["prompt_tokens"] += int(usage.get("prompt_tokens") or 0)
+            totals["cached_prompt_tokens"] += int(details.get("cached_tokens") or 0)
+            totals["completion_tokens"] += int(usage.get("completion_tokens") or 0)
+        return totals