0j ddlZddlZddlmZmZmZddlZddlmZddl m Z ddl m Z m Z ddlmZdeeejdefd efd Zd ed eedfd efd Zded efdZejd eded edfdZdS)N)UnionIteratorNoReturn)NotGiven)DEFAULT_EXECUTE_TIMEOUT_SECONDS)APIStatusErrorAPITimeoutError)is_giventimeoutreturncxt|tjr|j}t |r|t S|S)zCreate a simple float timeout for server responses from the provided timeout. For httpx.Timeout, we only use the read timeout. ) isinstancehttpxTimeoutreadr r)r s Z/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/parallel/lib/_time.pyprepare_timeout_floatr s> '5=)), G  /.. Nrun_idexcc*td|d|)Nz$Fetching task run result for run id z timed out.) TimeoutError)rrs r_raise_timeoutrs QfQQQ R RX[[r status_codec |dvS)aDetermine if an error is retryable. We retry the following HTTP status codes within the SDK: - 408 (Request Timeout): The server timed out waiting for the request - 503 (Service Unavailable): The server is temporarily unable to handle the request - 504 (Gateway Timeout): The gateway server timed out These errors typically indicate temporary issues with the server or network that may resolve upon retry. We don't include 429 (Too Many Requests) as this indicates rate limiting, which requires backing off rather than immediate retries. Note: This is a low-level retry mechanism within the SDK. Customers may want to implement their own retry logic at the application level for other error types. )iii)rs r_is_retryable_errorrs / ))rdeadlinec#Kd}tj|krI dVdS#t$r }|}Yd}~/d}~wt$r!}t |jr|}Yd}~Wd}~wwxYwt ||dS)aNContext manager for handling timeouts and retries when fetching task run results. Args: run_id: The ID of the task run deadline: The absolute time (monotonic) by which the operation must complete Raises: TimeoutError: If the deadline is reached APIStatusError: For non-timeout API errors N)time monotonicr r rrr)rrres rtimeout_retry_contextr$1s#'C .  X % %  EEE F   C HHHH   "1=11     63s$# A"4 A"AAA")r! contextlibtypingrrrr_typesr _constantsr _exceptionsr r _utils._utilsr floatrrstr Exceptionrintboolrcontextmanagerr$rrrr1sd ,,,,,,,,,, 88888899999999$$$$$$ 5 tX)M#N SX    \3\U9d?%;\\\\\*S*T****$  #  8D>      r