mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
1d2a616be7
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Graham Neubig <neubig@gmail.com>
12 lines
319 B
Python
12 lines
319 B
Python
from tenacity import RetryCallState
|
|
from tenacity.stop import stop_base
|
|
|
|
from openhands.runtime.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 should_exit()
|