⚡️ Speed up method APIRequestor._parse_retry_after_header by 37%

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,