|
13 | 13 | build_job_paginated_page_merge_callback, |
14 | 14 | initialize_job_paginated_response, |
15 | 15 | ) |
| 16 | +from ..job_status_utils import is_default_terminal_job_status |
16 | 17 | from ..serialization_utils import ( |
17 | 18 | serialize_model_dump_or_default, |
18 | 19 | serialize_model_dump_to_dict, |
@@ -99,7 +100,7 @@ async def start_and_wait( |
99 | 100 | job_status = await poll_until_terminal_status_async( |
100 | 101 | operation_name=operation_name, |
101 | 102 | get_status=lambda: self.get_status(job_id).status, |
102 | | - is_terminal_status=lambda status: status in {"completed", "failed"}, |
| 103 | + is_terminal_status=is_default_terminal_job_status, |
103 | 104 | poll_interval_seconds=poll_interval_seconds, |
104 | 105 | max_wait_seconds=max_wait_seconds, |
105 | 106 | max_status_failures=max_status_failures, |
@@ -204,7 +205,7 @@ async def start_and_wait( |
204 | 205 | return await wait_for_job_result_async( |
205 | 206 | operation_name=operation_name, |
206 | 207 | get_status=lambda: self.get_status(job_id).status, |
207 | | - is_terminal_status=lambda status: status in {"completed", "failed"}, |
| 208 | + is_terminal_status=is_default_terminal_job_status, |
208 | 209 | fetch_result=lambda: self.get(job_id), |
209 | 210 | poll_interval_seconds=poll_interval_seconds, |
210 | 211 | max_wait_seconds=max_wait_seconds, |
|
0 commit comments