(job_id: str, usage: Any)
| 119 | |
| 120 | |
| 121 | def _sync_token_usage(job_id: str, usage: Any) -> None: |
| 122 | requests = int(getattr(usage, 'requests', 0) or 0) |
| 123 | input_tokens = int(getattr(usage, 'input_tokens', 0) or 0) |
| 124 | output_tokens = int(getattr(usage, 'output_tokens', 0) or 0) |
| 125 | total_tokens = int(getattr(usage, 'total_tokens', 0) or 0) |
| 126 | |
| 127 | def apply(job): |
| 128 | job.usage.token.requests = requests |
| 129 | job.usage.token.input_tokens = input_tokens |
| 130 | job.usage.token.output_tokens = output_tokens |
| 131 | job.usage.token.total_tokens = total_tokens |
| 132 | |
| 133 | mutate_job_state(job_id, apply) |
| 134 | |
| 135 | |
| 136 | def _coerce_dict_rows(value: Any) -> list[dict[str, Any]]: |
no test coverage detected