first version of github actions

Author: misrasaurabh1

codeflash/cli_cmds/cmd_init.py

@@ -26,16 +26,13 @@
 from codeflash.cli_cmds.cli_common import apologize_and_exit
 from codeflash.cli_cmds.console import console, logger
 from codeflash.cli_cmds.extension import install_vscode_extension
+
 # Import JS/TS init module
 from codeflash.cli_cmds.init_javascript import (
-    JsPackageManager,
-    JSSetupInfo,
     ProjectLanguage,
     detect_project_language,
     determine_js_package_manager,
-    get_js_codeflash_command,
     get_js_dependency_installation_commands,
-    get_js_runtime_setup_string,
     init_js_project,
 )
 from codeflash.code_utils.code_utils import validate_relative_directory_path
@@ -798,8 +795,16 @@ def install_github_actions(override_formatter_check: bool = False) -> None:  # n
 
         # Generate workflow content AFTER user confirmation
         logger.info("[cmd_init.py:install_github_actions] User confirmed, generating workflow content...")
+
+        # Select the appropriate workflow template based on project language
+        project_language = detect_project_language_for_workflow(Path.cwd())
+        if project_language in ("javascript", "typescript"):
+            workflow_template = "codeflash-optimize-js.yaml"
+        else:
+            workflow_template = "codeflash-optimize.yaml"
+
         optimize_yml_content = (
-            files("codeflash").joinpath("cli_cmds", "workflows", "codeflash-optimize.yaml").read_text(encoding="utf-8")
+            files("codeflash").joinpath("cli_cmds", "workflows", workflow_template).read_text(encoding="utf-8")
         )
         materialized_optimize_yml_content = generate_dynamic_workflow_content(
             optimize_yml_content, config, git_root, benchmark_mode
@@ -1337,7 +1342,15 @@ def generate_dynamic_workflow_content(
     module_path = str(Path(config["module_root"]).relative_to(git_root) / "**")
     optimize_yml_content = optimize_yml_content.replace("{{ codeflash_module_path }}", module_path)
 
-    # Get working directory
+    # Detect project language
+    project_language = detect_project_language_for_workflow(Path.cwd())
+
+    # For JavaScript/TypeScript projects, use static template customization
+    # (AI-generated steps are currently Python-only)
+    if project_language in ("javascript", "typescript"):
+        return customize_codeflash_yaml_content(optimize_yml_content, config, git_root, benchmark_mode)
+
+    # Python project - try AI-generated steps
     toml_path = Path.cwd() / "pyproject.toml"
     try:
         with toml_path.open(encoding="utf8") as pyproject_file:
@@ -1503,13 +1516,20 @@ def _customize_python_workflow_content(
         codeflash_cmd += " --benchmark"
     return optimize_yml_content.replace("{{ codeflash_command }}", codeflash_cmd)
 
-# TODO:{claude} Refactor and move to support for language specific 
+# TODO:{claude} Refactor and move to support for language specific
 def _customize_js_workflow_content(
     optimize_yml_content: str,
     git_root: Path,
     benchmark_mode: bool = False,  # noqa: FBT001, FBT002
 ) -> str:
     """Customize workflow content for JavaScript/TypeScript projects."""
+    from codeflash.cli_cmds.init_javascript import (
+        get_js_codeflash_install_step,
+        get_js_codeflash_run_command,
+        get_js_runtime_setup_steps,
+        is_codeflash_dependency,
+    )
+
     project_root = Path.cwd()
     package_json_path = project_root / "package.json"
 
@@ -1531,19 +1551,24 @@ def _customize_js_workflow_content(
 
     optimize_yml_content = optimize_yml_content.replace("{{ working_directory }}", working_dir)
 
-    # Determine package manager
+    # Determine package manager and codeflash dependency status
     pkg_manager = determine_js_package_manager(project_root)
+    codeflash_is_dep = is_codeflash_dependency(project_root)
 
-    # Setup runtime environment
-    runtime_setup = get_js_runtime_setup_string(pkg_manager)
-    optimize_yml_content = optimize_yml_content.replace("{{ setup_runtime_environment }}", runtime_setup)
+    # Setup runtime environment (Node.js/Bun)
+    runtime_setup = get_js_runtime_setup_steps(pkg_manager)
+    optimize_yml_content = optimize_yml_content.replace("{{ setup_runtime_steps }}", runtime_setup)
 
     # Install dependencies
     install_deps_cmd = get_js_dependency_installation_commands(pkg_manager)
     optimize_yml_content = optimize_yml_content.replace("{{ install_dependencies_command }}", install_deps_cmd)
 
-    # Codeflash command
-    codeflash_cmd = get_js_codeflash_command(pkg_manager)
+    # Install codeflash step (only if not a dependency)
+    install_codeflash = get_js_codeflash_install_step(pkg_manager, is_dependency=codeflash_is_dep)
+    optimize_yml_content = optimize_yml_content.replace("{{ install_codeflash_step }}", install_codeflash)
+
+    # Codeflash run command
+    codeflash_cmd = get_js_codeflash_run_command(pkg_manager, is_dependency=codeflash_is_dep)
     if benchmark_mode:
         codeflash_cmd += " --benchmark"
     return optimize_yml_content.replace("{{ codeflash_command }}", codeflash_cmd)

codeflash/cli_cmds/init_javascript.py

@@ -516,37 +516,127 @@ def configure_package_json(setup_info: JSSetupInfo) -> bool:
 # ============================================================================
 
 
-def get_js_runtime_setup_string(pkg_manager: JsPackageManager) -> str:
-    """Generate the appropriate Node.js setup step for GitHub Actions."""
+def is_codeflash_dependency(project_root: Path) -> bool:
+    """Check if codeflash is listed as a dependency in package.json."""
+    package_json_path = project_root / "package.json"
+    if not package_json_path.exists():
+        return False
+
+    try:
+        with package_json_path.open(encoding="utf8") as f:
+            package_data = json.load(f)
+    except (json.JSONDecodeError, OSError):
+        return False
+
+    deps = package_data.get("dependencies", {})
+    dev_deps = package_data.get("devDependencies", {})
+    return "codeflash" in deps or "codeflash" in dev_deps
+
+
+def get_js_runtime_setup_steps(pkg_manager: JsPackageManager) -> str:
+    """Generate the appropriate Node.js/Bun setup steps for GitHub Actions.
+
+    Returns properly indented YAML steps for the workflow template.
+    """
     if pkg_manager == JsPackageManager.BUN:
-        return """name: 🥟 Setup Bun
+        return """- name: 🥟 Setup Bun
         uses: oven-sh/setup-bun@v2
         with:
           bun-version: latest"""
+
     if pkg_manager == JsPackageManager.PNPM:
-        return """name: 📦 Setup pnpm
+        return """- name: 📦 Setup pnpm
         uses: pnpm/action-setup@v4
         with:
           version: 9
       - name: 🟢 Setup Node.js
         uses: actions/setup-node@v4
         with:
-          node-version: '20'
+          node-version: '22'
           cache: 'pnpm'"""
+
     if pkg_manager == JsPackageManager.YARN:
-        return """name: 🟢 Setup Node.js
+        return """- name: 🟢 Setup Node.js
         uses: actions/setup-node@v4
         with:
-          node-version: '20'
+          node-version: '22'
           cache: 'yarn'"""
+
     # NPM or UNKNOWN
-    return """name: 🟢 Setup Node.js
+    return """- name: 🟢 Setup Node.js
         uses: actions/setup-node@v4
         with:
-          node-version: '20'
+          node-version: '22'
           cache: 'npm'"""
 
 
+def get_js_codeflash_install_step(pkg_manager: JsPackageManager, *, is_dependency: bool) -> str:
+    """Generate the codeflash installation step if not already a dependency.
+
+    Args:
+        pkg_manager: The package manager being used.
+        is_dependency: Whether codeflash is already in package.json dependencies.
+
+    Returns:
+        YAML step string for installing codeflash, or empty string if not needed.
+
+    """
+    if is_dependency:
+        # 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
+    return """- name: 📥 Install Codeflash
+        run: npm install -g codeflash"""
+
+
+def get_js_codeflash_run_command(pkg_manager: JsPackageManager, *, is_dependency: bool) -> str:
+    """Generate the codeflash run command for GitHub Actions.
+
+    Args:
+        pkg_manager: The package manager being used.
+        is_dependency: Whether codeflash is in package.json dependencies.
+
+    Returns:
+        Command string to run codeflash.
+
+    """
+    if is_dependency:
+        # Use package manager's run command for local dependency
+        if pkg_manager == JsPackageManager.BUN:
+            return "bun run codeflash"
+        if pkg_manager == JsPackageManager.PNPM:
+            return "pnpm exec codeflash"
+        if pkg_manager == JsPackageManager.YARN:
+            return "yarn codeflash"
+        # NPM
+        return "npx codeflash"
+
+    # Globally installed - just run directly
+    return "codeflash"
+
+
+def get_js_runtime_setup_string(pkg_manager: JsPackageManager) -> str:
+    """Generate the appropriate Node.js setup step for GitHub Actions.
+
+    Deprecated: Use get_js_runtime_setup_steps instead.
+    """
+    return get_js_runtime_setup_steps(pkg_manager)
+
+
 def get_js_dependency_installation_commands(pkg_manager: JsPackageManager) -> str:
     """Generate commands to install JavaScript/TypeScript dependencies."""
     if pkg_manager == JsPackageManager.BUN: