(self)
| 9 | Add more stats as needed.""" |
| 10 | |
| 11 | def __init__(self) -> None: |
| 12 | super().__init__() |
| 13 | self.cnt = 0 |
| 14 | self.input_tokens = 0 |
| 15 | self.output_tokens = 0 |
| 16 | self.all_times = [] |
| 17 | self.start_time = 0 |
| 18 | |
| 19 | def on_chat_model_start(self, serialized, prompts, **kwargs): |
| 20 | self.start_time = time.time() |