(self)
| 33 | self.all_times = [] |
| 34 | |
| 35 | def get_stats(self) -> dict[str, int]: |
| 36 | return { |
| 37 | "calls": self.cnt, |
| 38 | "input_tokens": self.input_tokens, |
| 39 | "output_tokens": self.output_tokens, |
| 40 | "all_times": self.all_times, |
| 41 | } |
| 42 | |
| 43 | |
| 44 | class AsyncStatsCallbackHandler(AsyncCallbackHandler): |