How do I keep a record of each Strale call in an AI agent?

# pip install straleio pydantic-ai
from datetime import datetime, timezone

from straleio import Strale, DoRequest
from pydantic_ai import Agent

strale = Strale(api_key="sk_live_YOUR_KEY")

audit_log = []

agent = Agent("openai:gpt-4o")


@agent.tool_plain
def sanctions_check_with_record(name: str, context: str = "") -> dict:
    """Search a name against sanctions lists and keep a record of the call."""
    result = strale.do(DoRequest(
        capability_slug="sanctions-check",
        inputs={"name": name},
        max_price_cents=30,
    ))

    audit_log.append({
        "timestamp": datetime.now(timezone.utc).isoformat(),
        "tool": result.capability_used,
        "name_searched": name,
        "context": context,
        "transaction_id": result.transaction_id,
        "latency_ms": result.latency_ms,
        "matches_found": len(result.output.get("matches", [])),
    })

    return result.output


response = agent.run_sync("Search 'Acme Corp' against sanctions lists for invoice INV-2024-001")
print(f"Result: {response.output}")
print(f"Records: {audit_log}")