Context manager that times tool invocations.
(self, node_id: str, tool_name: str)
| 453 | |
| 454 | @contextmanager |
| 455 | def tool_timer(self, node_id: str, tool_name: str): |
| 456 | """Context manager that times tool invocations.""" |
| 457 | self.__init_timers__() |
| 458 | start_time = time.time() |
| 459 | try: |
| 460 | yield |
| 461 | finally: |
| 462 | end_time = time.time() |
| 463 | duration = (end_time - start_time) |
| 464 | self._timers[f"tool_{node_id}_{tool_name}"] = duration |
| 465 | |
| 466 | @contextmanager |
| 467 | def thinking_timer(self, node_id: str, stage: str): |
no test coverage detected