diff --git a/src/together/abstract/api_requestor.py b/src/together/abstract/api_requestor.py index 7e37eaf8..3e5e2e69 100644 --- a/src/together/abstract/api_requestor.py +++ b/src/together/abstract/api_requestor.py @@ -126,28 +126,33 @@ def _parse_retry_after_header( # First, try the non-standard `retry-after-ms` header for milliseconds, # which is more precise than integer-seconds `retry-after` - try: - retry_ms_header = response_headers.get("retry-after-ms", None) - return float(retry_ms_header) / 1000 - except (TypeError, ValueError): - pass + retry_ms_header = response_headers.get("retry-after-ms") + if retry_ms_header is not None: + try: + return float(retry_ms_header) / 1000 + except (TypeError, ValueError): + pass # Next, try parsing `retry-after` header as seconds (allowing nonstandard floats). - retry_header = str(response_headers.get("retry-after")) - try: - # note: the spec indicates that this should only ever be an integer - # but if someone sends a float there's no reason for us to not respect it - return float(retry_header) - except (TypeError, ValueError): - pass + retry_header = response_headers.get("retry-after") + if retry_header is not None: + try: + # note: the spec indicates that this should only ever be an integer + # but if someone sends a float there's no reason for us to not respect it + return float(retry_header) + except (TypeError, ValueError): + pass # Last, try parsing `retry-after` as a date. - retry_date_tuple = email.utils.parsedate_tz(retry_header) - if retry_date_tuple is None: - return None + if retry_header is not None: + retry_date_tuple = email.utils.parsedate_tz(str(retry_header)) + if retry_date_tuple is None: + return None + + retry_date = email.utils.mktime_tz(retry_date_tuple) + return float(retry_date - time.time()) - retry_date = email.utils.mktime_tz(retry_date_tuple) - return float(retry_date - time.time()) + return None def _calculate_retry_timeout( self,