Merge branch 'main' into init/install-vscode-extension

Author: KRRT7

.github/workflows/e2e-async.yaml

@@ -0,0 +1,69 @@
+name: E2E - Async
+
+on:
+ pull_request:
+ paths:
+ - '**' # Trigger for all paths
+
+ workflow_dispatch:
+
+jobs:
+ async-optimization:
+ # Dynamically determine if environment is needed only when workflow files change and contributor is external
+ environment: ${{ (github.event_name == 'workflow_dispatch' || (contains(toJSON(github.event.pull_request.files.*.filename), '.github/workflows/') && github.event.pull_request.user.login != 'misrasaurabh1' && github.event.pull_request.user.login != 'KRRT7')) && 'external-trusted-contributors' || '' }}
+
+ runs-on: ubuntu-latest
+ env:
+ CODEFLASH_AIS_SERVER: prod
+ POSTHOG_API_KEY: ${{ secrets.POSTHOG_API_KEY }}
+ CODEFLASH_API_KEY: ${{ secrets.CODEFLASH_API_KEY }}
+ COLUMNS: 110
+ MAX_RETRIES: 3
+ RETRY_DELAY: 5
+ EXPECTED_IMPROVEMENT_PCT: 10
+ CODEFLASH_END_TO_END: 1
+ steps:
+ - name: 🛎️ Checkout
+ uses: actions/checkout@v4
+ with:
+ ref: ${{ github.event.pull_request.head.ref }}
+ repository: ${{ github.event.pull_request.head.repo.full_name }}
+ fetch-depth: 0
+ token: ${{ secrets.GITHUB_TOKEN }}
+
+ - name: Validate PR
+ run: |
+ # Check for any workflow changes
+ if git diff --name-only "${{ github.event.pull_request.base.sha }}" "${{ github.event.pull_request.head.sha }}" | grep -q "^.github/workflows/"; then
+ echo "⚠️ Workflow changes detected."
+
+ # Get the PR author
+ AUTHOR="${{ github.event.pull_request.user.login }}"
+ echo "PR Author: $AUTHOR"
+
+ # Allowlist check
+ if [[ "$AUTHOR" == "misrasaurabh1" || "$AUTHOR" == "KRRT7" ]]; then
+ echo "✅ Authorized user ($AUTHOR). Proceeding."
+ elif [[ "${{ github.event.pull_request.state }}" == "open" ]]; then
+ echo "✅ PR triggered by 'pull_request_target' and is open. Assuming protection rules are in place. Proceeding."
+ else
+ echo "⛔ Unauthorized user ($AUTHOR) attempting to modify workflows. Exiting."
+ exit 1
+ fi
+ else
+ echo "✅ No workflow file changes detected. Proceeding."
+ fi
+
+ - name: Set up Python 3.11 for CLI
+ uses: astral-sh/setup-uv@v5
+ with:
+ python-version: 3.11.6
+
+ - name: Install dependencies (CLI)
+ run: |
+ uv sync
+ 
+ - name: Run Codeflash to optimize async code
+ id: optimize_async_code
+ run: |
+ uv run python tests/scripts/end_to_end_test_async.py

.github/workflows/e2e-bubblesort-pytest-nogit.yaml

@@ -20,7 +20,7 @@ jobs:
  COLUMNS: 110
  MAX_RETRIES: 3
  RETRY_DELAY: 5
- EXPECTED_IMPROVEMENT_PCT: 300
+ EXPECTED_IMPROVEMENT_PCT: 70
  CODEFLASH_END_TO_END: 1
  steps:
  - name: 🛎️ Checkout

.github/workflows/e2e-bubblesort-unittest.yaml

@@ -20,7 +20,7 @@ jobs:
  COLUMNS: 110
  MAX_RETRIES: 3
  RETRY_DELAY: 5
- EXPECTED_IMPROVEMENT_PCT: 300
+ EXPECTED_IMPROVEMENT_PCT: 40
  CODEFLASH_END_TO_END: 1
  steps:
  - name: 🛎️ Checkout

.github/workflows/e2e-init-optimization.yaml

@@ -19,7 +19,7 @@ jobs:
  COLUMNS: 110
  MAX_RETRIES: 3
  RETRY_DELAY: 5
- EXPECTED_IMPROVEMENT_PCT: 30
+ EXPECTED_IMPROVEMENT_PCT: 10
  CODEFLASH_END_TO_END: 1
  steps:
  - name: 🛎️ Checkout

.github/workflows/e2e-topological-sort.yaml

@@ -1,4 +1,4 @@
-name: E2E - Topological Sort
+name: E2E - Topological Sort (Worktree)
 
 on:
  pull_request:
@@ -8,7 +8,7 @@ on:
  workflow_dispatch:
 
 jobs:
- topological-sort-optimization:
+ topological-sort-worktree-optimization:
  # Dynamically determine if environment is needed only when workflow files change and contributor is external
  environment: ${{ (github.event_name == 'workflow_dispatch' || (contains(toJSON(github.event.pull_request.files.*.filename), '.github/workflows/') && github.event.pull_request.user.login != 'misrasaurabh1' && github.event.pull_request.user.login != 'KRRT7')) && 'external-trusted-contributors' || '' }}
  runs-on: ubuntu-latest
@@ -90,4 +90,4 @@ jobs:
  - name: Run Codeflash to optimize code
  id: optimize_code
  run: |
- uv run python tests/scripts/end_to_end_test_topological_sort.py
+ uv run python tests/scripts/end_to_end_test_topological_sort_worktree.py

.github/workflows/unit-tests.yaml

@@ -11,7 +11,7 @@ jobs:
  strategy:
  fail-fast: false
  matrix:
- python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
+ python-version: ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
  continue-on-error: true
  runs-on: ubuntu-latest
  steps:
@@ -24,10 +24,13 @@ jobs:
  uses: astral-sh/setup-uv@v5
  with:
  python-version: ${{ matrix.python-version }}
- version: "0.5.30"
 
  - name: install dependencies
  run: uv sync
 
+ - name: Install test-only dependencies (Python 3.13)
+ if: matrix.python-version == '3.13'
+ run: uv sync --group tests
+
  - name: Unit tests
- run: uv run pytest tests/
+ run: uv run pytest tests/

.github/workflows/windows-unit-tests.yml

@@ -0,0 +1,30 @@
+name: windows-unit-tests
+
+on:
+ push:
+ branches: [main]
+ pull_request:
+ workflow_dispatch:
+
+jobs:
+ windows-unit-tests:
+ continue-on-error: true
+ runs-on: windows-latest
+ env:
+ PYTHONIOENCODING: utf-8
+ steps:
+ - uses: actions/checkout@v4
+ with:
+ fetch-depth: 0
+ token: ${{ secrets.GITHUB_TOKEN }}
+
+ - name: Install uv
+ uses: astral-sh/setup-uv@v5
+ with:
+ python-version: "3.13"
+
+ - name: install dependencies
+ run: uv sync
+
+ - name: Unit tests
+ run: uv run pytest tests/

README.md

@@ -17,7 +17,7 @@ How to use Codeflash -
 - Automate optimizing all __future__ code you will write by installing Codeflash as a GitHub action.
 - Optimize a Python workflow `python myscript.py` end-to-end by running `codeflash optimize myscript.py`
 
-Codeflash is used by top engineering teams at [Pydantic](https://github.com/pydantic/pydantic/pulls?q=is%3Apr+author%3Amisrasaurabh1+is%3Amerged), [Langflow](https://github.com/langflow-ai/langflow/issues?q=state%3Aclosed%20is%3Apr%20author%3Amisrasaurabh1), [Roboflow](https://github.com/roboflow/inference/pulls?q=is%3Apr+is%3Amerged+codeflash+sort%3Acreated-asc), [Albumentations](https://github.com/albumentations-team/albumentations/issues?q=state%3Amerged%20is%3Apr%20author%3Akrrt7%20OR%20state%3Amerged%20is%3Apr%20author%3Aaseembits93%20) and many others to ship performant, expert level code.
+Codeflash is used by top engineering teams at **Pydantic** [(PRs Merged)](https://github.com/pydantic/pydantic/pulls?q=is%3Apr+author%3Amisrasaurabh1+is%3Amerged), **Roboflow** [(PRs Merged 1](https://github.com/roboflow/inference/issues?q=state%3Aclosed%20is%3Apr%20author%3Amisrasaurabh1%20is%3Amerged), [PRs Merged 2)](https://github.com/roboflow/inference/issues?q=state%3Amerged%20is%3Apr%20author%3Acodeflash-ai%5Bbot%5D), **Unstructured** [(PRs Merged 1](https://github.com/Unstructured-IO/unstructured/pulls?q=is%3Apr+Explanation+and+details+in%3Abody+is%3Amerged), [PRs Merged 2)](https://github.com/Unstructured-IO/unstructured-ingest/pulls?q=is%3Apr+Explanation+and+details+in%3Abody+is%3Amerged), **Langflow** [(PRs Merged)](https://github.com/langflow-ai/langflow/issues?q=state%3Aclosed%20is%3Apr%20author%3Amisrasaurabh1) and many others to ship performant, expert level code.
 
 Codeflash is great at optimizing AI Agents, Computer Vision algorithms, PyTorch code, numerical code, backend code or anything else you might write with Python.
 

code_to_optimize/async_bubble_sort.py

@@ -0,0 +1,43 @@
+import asyncio
+from typing import List, Union
+
+
+async def async_sorter(lst: List[Union[int, float]]) -> List[Union[int, float]]:
+ """
+ Async bubble sort implementation for testing.
+ """
+ print("codeflash stdout: Async sorting list")
+ 
+ await asyncio.sleep(0.01)
+ 
+ n = len(lst)
+ for i in range(n):
+ for j in range(0, n - i - 1):
+ if lst[j] > lst[j + 1]:
+ lst[j], lst[j + 1] = lst[j + 1], lst[j]
+ 
+ result = lst.copy()
+ print(f"result: {result}")
+ return result
+
+
+class AsyncBubbleSorter:
+ """Class with async sorting method for testing."""
+ 
+ async def sorter(self, lst: List[Union[int, float]]) -> List[Union[int, float]]:
+ """
+ Async bubble sort implementation within a class.
+ """
+ print("codeflash stdout: AsyncBubbleSorter.sorter() called")
+ 
+ # Add some async delay
+ await asyncio.sleep(0.005)
+ 
+ n = len(lst)
+ for i in range(n):
+ for j in range(0, n - i - 1):
+ if lst[j] > lst[j + 1]:
+ lst[j], lst[j + 1] = lst[j + 1], lst[j]
+ 
+ result = lst.copy()
+ return result

code_to_optimize/code_directories/async_e2e/main.py

@@ -0,0 +1,16 @@
+import time
+import asyncio
+
+
+async def retry_with_backoff(func, max_retries=3):
+ if max_retries < 1:
+ raise ValueError("max_retries must be at least 1")
+ last_exception = None
+ for attempt in range(max_retries):
+ try:
+ return await func()
+ except Exception as e:
+ last_exception = e
+ if attempt < max_retries - 1:
+ time.sleep(0.0001 * attempt)
+ raise last_exception