| 19 | self.precheck_step_state[step_name] = self.__class__.INITIAL |
| 20 | |
| 21 | async def precheck_step_done(self, step_name): |
| 22 | async with self._state_lock: |
| 23 | self.precheck_step_state[step_name] = self.__class__.DONE |
| 24 | if step_name == "Decon": |
| 25 | self._decon_event.set() |
| 26 | elif step_name == "ToolSelector": |
| 27 | self._tool_router_event.set() |
| 28 | # Check if all steps are done |
| 29 | if all(state == self.__class__.DONE for state in self.precheck_step_state.values()): |
| 30 | self.handler.pre_checks_done_event.set() |
| 31 | |
| 32 | def set_pre_checks_done(self): |
| 33 | """Synchronous version for compatibility""" |