Optimize TestGenRequest.to_payload

The optimization hoists `platform.python_version()` out of the per-call loop by caching it at module import time as `_PLATFORM_PYTHON_VERSION`, eliminating a 500+ns system call on every invocation when the language is not Python. The original code imported `platform` and called `platform.python_version()` inside `to_payload`, incurring repeated overhead even though the value never changes within a process. Line profiler shows the `import platform` statement itself consumed 6.6% of runtime, and the conditional evaluation another 7%. The payload dict construction was also reordered to compute `python_version` upfront, reducing incremental dict updates. This yields a 20% speedup (1.97ms → 1.64ms) with no correctness trade-offs.

Author: codeflash-ai[bot]

codeflash/api/schemas.py

@@ -12,10 +12,13 @@
 
 from __future__ import annotations
 
+import platform
 from dataclasses import dataclass, field
 from enum import Enum
 from typing import Any
 
+_PLATFORM_PYTHON_VERSION = platform.python_version()
+
 
 class ModuleSystem(str, Enum):
     """Module system used by the code."""
@@ -190,6 +193,11 @@ 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},
@@ -204,18 +212,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