mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
12 lines
317 B
Python
12 lines
317 B
Python
from tenacity import RetryCallState
|
|
from tenacity.stop import stop_base
|
|
|
|
from openhands.utils.shutdown_listener import should_exit
|
|
|
|
|
|
class stop_if_should_exit(stop_base):
|
|
"""Stop if the should_exit flag is set."""
|
|
|
|
def __call__(self, retry_state: 'RetryCallState') -> bool:
|
|
return bool(should_exit())
|