Merge pull request #1826 from codeflash-ai/codeflash/optimize-pr1199-2026-03-13T01.03.06

⚡️ Speed up method `TestGenRequest.to_payload` by 20% in PR #1199 (`omni-java`)

Author: claude[bot]

codeflash/api/schemas.py

@@ -17,6 +17,8 @@
 from enum import Enum
 from typing import Any
 
+_PLATFORM_PYTHON_VERSION = platform.python_version()
+
 
 class ModuleSystem(str, Enum):
     """Module system used by the code."""
@@ -189,6 +191,9 @@ class TestGenRequest:
 
     def to_payload(self) -> dict[str, Any]:
         """Convert to API payload dict, maintaining backward compatibility."""
+        # Backward compat: backend still expects python_version
+        python_version = self.language_info.version if self.language_info.name == "python" else _PLATFORM_PYTHON_VERSION
+
         payload = {
             "source_code_being_tested": self.source_code,
             "function_to_optimize": {"function_name": self.function_name, "is_async": self.is_async},
@@ -203,18 +208,10 @@ def to_payload(self) -> dict[str, Any]:
             "codeflash_version": self.codeflash_version,
             "is_async": self.is_async,
             "is_numerical_code": self.is_numerical_code,
+            "language_version": self.language_info.version,
+            "python_version": python_version,
         }
 
-        # Add language version (canonical for all languages)
-        payload["language_version"] = self.language_info.version
-
-        # Backward compat: backend still expects python_version
-        import platform
-
-        payload["python_version"] = (
-            self.language_info.version if self.language_info.name == "python" else platform.python_version()
-        )
-
         # Module system for JS/TS
         if self.language_info.module_system != ModuleSystem.UNKNOWN:
             payload["module_system"] = self.language_info.module_system.value