(self, tool_manager=None, fail_after=1)
| 460 | """Context whose inject_tool_message raises after the first N calls.""" |
| 461 | |
| 462 | def __init__(self, tool_manager=None, fail_after=1): |
| 463 | super().__init__(tool_manager=tool_manager) |
| 464 | self._inject_count = 0 |
| 465 | self._fail_after = fail_after |
| 466 | |
| 467 | def inject_tool_message(self, message): |
| 468 | self._inject_count += 1 |