⚡️ Speed up function get_dependency_manager_installation_string by 127% in PR #208 (bump-gha-uv-version)

Here's an optimized version of your program with reduced runtime, specifically targeting the bottlenecks.

### **Optimizations performed:**
1. **Constant Folding:** The sys.version_info fetch and string formatting for the Python version is only necessary for the non-UV case and does not need to be constructed unless used.  
2. **Precomputed Templates:** The output strings are constants and can be stored as such, to avoid reconstructing them on each call.  
3. **Avoid Unnecessary Formatting:** For the setup-python path, the version string is constant across invocations of the same interpreter, so can use lazy-static initialization.
4. **Reduced Function Overhead:** Restructured code to minimize code execution paths and avoid unnecessary work.

---



---

**Summary of changes:**  
- Only creates formatted strings once per interpreter lifetime, so on repeated heavy calls, time and allocations are minimized.
- `sys.version_info` is not re-accessed on every call.
- Preserved all logical comments; comments were added only where code was optimized.

This rewrite should significantly improve per-call performance of `get_dependency_manager_installation_string()`.

Author: codeflash-ai[bot]

codeflash/cli_cmds/cmd_init.py

@@ -239,7 +239,7 @@ def collect_setup_info() -> SetupInfo:
         else:
             apologize_and_exit()
     else:
-        tests_root = Path(curdir) / Path(cast(str, tests_root_answer))
+        tests_root = Path(curdir) / Path(cast("str", tests_root_answer))
     tests_root = tests_root.relative_to(curdir)
     ph("cli-tests-root-provided")
 
@@ -302,7 +302,7 @@ def collect_setup_info() -> SetupInfo:
     elif benchmarks_answer == no_benchmarks_option:
         benchmarks_root = None
     else:
-        benchmarks_root = tests_root / Path(cast(str, benchmarks_answer))
+        benchmarks_root = tests_root / Path(cast("str", benchmarks_answer))
 
     # TODO: Implement other benchmark framework options
     # if benchmarks_root:
@@ -354,9 +354,9 @@ def collect_setup_info() -> SetupInfo:
         module_root=str(module_root),
         tests_root=str(tests_root),
         benchmarks_root=str(benchmarks_root) if benchmarks_root else None,
-        test_framework=cast(str, test_framework),
+        test_framework=cast("str", test_framework),
         ignore_paths=ignore_paths,
-        formatter=cast(str, formatter),
+        formatter=cast("str", formatter),
         git_remote=str(git_remote),
     )
 
@@ -466,7 +466,7 @@ def check_for_toml_or_setup_file() -> str | None:
             click.echo("⏩️ Skipping pyproject.toml creation.")
             apologize_and_exit()
     click.echo()
-    return cast(str, project_name)
+    return cast("str", project_name)
 
 
 def install_github_actions(override_formatter_check: bool = False) -> None:
@@ -619,17 +619,10 @@ def get_dependency_installation_commands(dep_manager: DependencyManager) -> tupl
 
 
 def get_dependency_manager_installation_string(dep_manager: DependencyManager) -> str:
-    py_version = sys.version_info
-    python_version_string = f"'{py_version.major}.{py_version.minor}'"
+    # Return precomputed string to avoid repeated formatting and runtime cost
     if dep_manager == DependencyManager.UV:
-        return """name: 🐍 Setup UV
-        uses: astral-sh/setup-uv@v6
-        with:
-          enable-cache: true"""
-    return f"""name: 🐍 Set up Python
-        uses: actions/setup-python@v5
-        with:
-          python-version: {python_version_string}"""
+        return _UV_INSTALL_STR
+    return _SETUP_PYTHON_INSTALL_STR
 
 
 def get_github_action_working_directory(toml_path: Path, git_root: Path) -> str:
@@ -966,3 +959,16 @@ def ask_for_telemetry() -> bool:
     )
 
     return enable_telemetry
+
+
+_UV_INSTALL_STR = """name: 🐍 Setup UV
+        uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true"""
+
+_PYTHON_VERSION_STRING = f"'{sys.version_info.major}.{sys.version_info.minor}'"
+
+_SETUP_PYTHON_INSTALL_STR = f"""name: 🐍 Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: {_PYTHON_VERSION_STRING}"""