don't retry if API returns task failure (#8111)

This commit is contained in:
Christian Byrne 2025-05-13 22:28:30 -07:00 committed by GitHub
parent 98ff01e148
commit f3ff5c40db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1105,7 +1105,7 @@ class PollingOperation(Generic[T, R]):
except Exception as e:
# For other errors, increment count and potentially abort
consecutive_errors += 1
if consecutive_errors >= max_consecutive_errors:
if consecutive_errors >= max_consecutive_errors or status == TaskStatus.FAILED:
raise Exception(
f"Polling aborted after {consecutive_errors} consecutive errors: {str(e)}"
) from e