Merge pull request #1778 from codeflash-ai/fix/github_actions_init

[Fix] Github Actions init

Author: Saga4

codeflash/cli_cmds/init_javascript.py

@@ -690,22 +690,9 @@ def get_js_codeflash_install_step(pkg_manager: JsPackageManager, *, is_dependenc
         # Codeflash will be installed with other dependencies
         return ""
 
-    # Need to install codeflash separately
-    if pkg_manager == JsPackageManager.BUN:
-        return """- name: 📥 Install Codeflash
-        run: bun add -g codeflash"""
-
-    if pkg_manager == JsPackageManager.PNPM:
-        return """- name: 📥 Install Codeflash
-        run: pnpm add -g codeflash"""
-
-    if pkg_manager == JsPackageManager.YARN:
-        return """- name: 📥 Install Codeflash
-        run: yarn global add codeflash"""
-
-    # NPM or UNKNOWN
+    # Install codeflash via uv (Python + uv are set up in the workflow)
     return """- name: 📥 Install Codeflash
-        run: npm install -g codeflash"""
+        run: uv tool install codeflash"""
 
 
 def get_js_codeflash_run_command(pkg_manager: JsPackageManager, *, is_dependency: bool) -> str:

codeflash/cli_cmds/workflows/codeflash-optimize-js.yaml

@@ -27,6 +27,12 @@ jobs:
         uses: actions/checkout@v4
         with:
           fetch-depth: 0
+      - name: 🐍 Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.12'
+      - name: 📦 Setup uv
+        uses: astral-sh/setup-uv@v4
       {{ setup_runtime_steps }}
       - name: 📦 Install Dependencies
         run: {{ install_dependencies_command }}

codeflash/cli_cmds/workflows/codeflash-optimize.yaml

@@ -27,7 +27,7 @@ jobs:
         uses: actions/checkout@v4
         with:
           fetch-depth: 0
-      - {{ setup_python_dependency_manager }}
+      - {{ setup_runtime_environment }}
       - name: 📦 Install Dependencies
         run: {{ install_dependencies_command }}
       - name: ⚡️Codeflash Optimization

tests/code_utils/test_js_workflow_helpers.py

@@ -131,30 +131,13 @@ def test_returns_empty_when_is_dependency(self) -> None:
 
         assert result == ""
 
-    def test_npm_global_install(self) -> None:
-        """Should generate npm global install when not a dependency."""
-        result = get_js_codeflash_install_step(JsPackageManager.NPM, is_dependency=False)
+    def test_uv_tool_install_when_not_dependency(self) -> None:
+        """Should generate uv tool install when not a dependency, regardless of package manager."""
+        for pkg_manager in (JsPackageManager.NPM, JsPackageManager.YARN, JsPackageManager.PNPM, JsPackageManager.BUN):
+            result = get_js_codeflash_install_step(pkg_manager, is_dependency=False)
 
-        assert "Install Codeflash" in result
-        assert "npm install -g codeflash" in result
-
-    def test_yarn_global_install(self) -> None:
-        """Should generate yarn global install when not a dependency."""
-        result = get_js_codeflash_install_step(JsPackageManager.YARN, is_dependency=False)
-
-        assert "yarn global add codeflash" in result
-
-    def test_pnpm_global_install(self) -> None:
-        """Should generate pnpm global install when not a dependency."""
-        result = get_js_codeflash_install_step(JsPackageManager.PNPM, is_dependency=False)
-
-        assert "pnpm add -g codeflash" in result
-
-    def test_bun_global_install(self) -> None:
-        """Should generate bun global install when not a dependency."""
-        result = get_js_codeflash_install_step(JsPackageManager.BUN, is_dependency=False)
-
-        assert "bun add -g codeflash" in result
+            assert "Install Codeflash" in result
+            assert "uv tool install codeflash" in result
 
 
 class TestGetJsCodeflashRunCommand: