⚡️ Speed up method APIRequestor._parse_retry_after_header by 37%

[codeflash-ai]

📄 37% (0.37x) speedup for APIRequestor._parse_retry_after_header in src/together/abstract/api_requestor.py

⏱️ Runtime : 9.54 microseconds → 6.97 microseconds (best of 37 runs)

📝 Explanation and details

The optimized code achieves a 36% speedup by restructuring the parsing logic to minimize redundant operations and avoid unnecessary work:

Key optimizations:

1. Eliminated redundant dict lookups: The original code called response_headers.get("retry-after-ms", None) inside a try-except block, then potentially looked up the same key again. The optimized version stores the result in retry_ms_header once and checks for None before attempting conversion.

2. Avoided unnecessary string conversions: The original code unconditionally called str(response_headers.get("retry-after")) even when the header might not exist or might be successfully parsed as a float. The optimized version only converts to string when needed for date parsing.

3. Reduced exception handling overhead: By checking if retry_ms_header is not None and if retry_header is not None before attempting type conversions, the optimized code avoids triggering TypeError exceptions when headers don't exist, which is expensive in Python.

4. Streamlined control flow: The optimized version has clearer conditional branches that short-circuit when values are None, avoiding unnecessary processing steps.

Performance benefits by test case:

• Empty headers or missing keys: Faster due to fewer exception catches and string conversions
• Valid numeric headers: Faster due to single dict lookup and direct null checks
• Date parsing fallback: Still efficient but only executes when necessary

The line profiler shows the optimized version spends less time on dict lookups (14.1% vs 19.9%) and has better distributed execution across fewer code paths.

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	🔘 None Found
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	✅ 8 Passed
📊 Tests Coverage	88.2%

🔎 Concolic Coverage Tests and Runtime

Test File::Test Function	Original ⏱️	Optimized ⏱️	Speedup
codeflash_concolic_atws5rsq/tmpjhs12g13/test_concolic_coverage.py::test_APIRequestor__parse_retry_after_header	5.24μs	4.04μs	29.5%✅
codeflash_concolic_atws5rsq/tmpjhs12g13/test_concolic_coverage.py::test_APIRequestor__parse_retry_after_header_2	1.17μs	1.17μs	0.428%✅
codeflash_concolic_atws5rsq/tmpjhs12g13/test_concolic_coverage.py::test_APIRequestor__parse_retry_after_header_3	581ns	561ns	3.57%✅
codeflash_concolic_atws5rsq/tmpjhs12g13/test_concolic_coverage.py::test_APIRequestor__parse_retry_after_header_4	2.55μs	1.20μs	113%✅

To edit these changes git checkout codeflash/optimize-APIRequestor._parse_retry_after_header-mgzuus6u and push.