(response)
| 12 | |
| 13 | |
| 14 | def usage_dict(response): |
| 15 | usage = getattr(response, "usage", None) |
| 16 | if usage is None: |
| 17 | return {} |
| 18 | if hasattr(usage, "model_dump"): |
| 19 | return usage.model_dump() |
| 20 | if isinstance(usage, dict): |
| 21 | return usage |
| 22 | return { |
| 23 | name: getattr(usage, name) |
| 24 | for name in ("input_tokens", "output_tokens", "total_tokens") |
| 25 | if hasattr(usage, name) |
| 26 | } |
| 27 | |
| 28 | |
| 29 | def debug_response(kind, prompt, response): |