Fix ruff lint and format issues in retry-after code

Signed-off-by: debu-sinha <debusinha2009@gmail.com>

Author: debu-sinha

llama-index-integrations/embeddings/llama-index-embeddings-openai/llama_index/embeddings/openai/utils.py

@@ -40,7 +40,8 @@
 
 
 class _WaitRetryAfter(wait_base):
-    """Wait strategy that respects the Retry-After header on RateLimitError.
+    """
+    Wait strategy that respects the Retry-After header on RateLimitError.
 
     When the last exception is an ``openai.RateLimitError`` whose HTTP response
     contains a ``Retry-After`` header, the wait time is taken from that header
@@ -61,7 +62,8 @@ def __call__(self, retry_state: RetryCallState) -> float:
 
 
 def _parse_retry_after(exc: openai.RateLimitError) -> Optional[float]:
-    """Extract the Retry-After value (in seconds) from a RateLimitError.
+    """
+    Extract the Retry-After value (in seconds) from a RateLimitError.
 
     Returns ``None`` when the header is missing or cannot be parsed.
     """

llama-index-integrations/embeddings/llama-index-embeddings-openai/tests/test_retry_after.py

@@ -66,7 +66,9 @@ def test_parse_retry_after_missing_header():
 
 
 def test_parse_retry_after_non_numeric():
-    exc = _make_rate_limit_error(headers={"Retry-After": "Wed, 21 Oct 2025 07:28:00 GMT"})
+    exc = _make_rate_limit_error(
+        headers={"Retry-After": "Wed, 21 Oct 2025 07:28:00 GMT"}
+    )
     assert _parse_retry_after(exc) is None
 
 
@@ -127,7 +129,9 @@ def test_wait_retry_after_falls_back_for_non_rate_limit_error():
     fallback = MagicMock(return_value=7.0)
     strategy = _WaitRetryAfter(fallback)
 
-    exc = openai.APITimeoutError(request=httpx.Request("POST", "https://api.openai.com"))
+    exc = openai.APITimeoutError(
+        request=httpx.Request("POST", "https://api.openai.com")
+    )
     rs = _make_retry_state(exc)
     assert strategy(rs) == 7.0
     fallback.assert_called_once_with(rs)

llama-index-integrations/llms/llama-index-llms-openai/llama_index/llms/openai/utils.py

@@ -259,7 +259,8 @@ def is_json_schema_supported(model: str) -> bool:
 
 
 class _WaitRetryAfter(wait_base):
-    """Wait strategy that respects the Retry-After header on RateLimitError.
+    """
+    Wait strategy that respects the Retry-After header on RateLimitError.
 
     When the last exception is an ``openai.RateLimitError`` whose HTTP response
     contains a ``Retry-After`` header, the wait time is taken from that header
@@ -280,7 +281,8 @@ def __call__(self, retry_state: RetryCallState) -> float:
 
 
 def _parse_retry_after(exc: openai.RateLimitError) -> Optional[float]:
-    """Extract the Retry-After value (in seconds) from a RateLimitError.
+    """
+    Extract the Retry-After value (in seconds) from a RateLimitError.
 
     Returns ``None`` when the header is missing or cannot be parsed.
     """

llama-index-integrations/llms/llama-index-llms-openai/tests/test_retry_after.py

@@ -66,7 +66,9 @@ def test_parse_retry_after_missing_header():
 
 
 def test_parse_retry_after_non_numeric():
-    exc = _make_rate_limit_error(headers={"Retry-After": "Wed, 21 Oct 2025 07:28:00 GMT"})
+    exc = _make_rate_limit_error(
+        headers={"Retry-After": "Wed, 21 Oct 2025 07:28:00 GMT"}
+    )
     assert _parse_retry_after(exc) is None
 
 
@@ -126,7 +128,9 @@ def test_wait_retry_after_falls_back_for_non_rate_limit_error():
     fallback = MagicMock(return_value=7.0)
     strategy = _WaitRetryAfter(fallback)
 
-    exc = openai.APITimeoutError(request=httpx.Request("POST", "https://api.openai.com"))
+    exc = openai.APITimeoutError(
+        request=httpx.Request("POST", "https://api.openai.com")
+    )
     rs = _make_retry_state(exc)
     assert strategy(rs) == 7.0
     fallback.assert_called_once_with(rs)