.github/workflows/codeflash-optimize.yaml

@@ -35,13 +35,9 @@ jobs:
 
       - name: 📦 Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: ⚡️Codeflash Optimization
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run codeflash --benchmark
+          uv run codeflash --benchmark

.github/workflows/end-to-end-test-benchmark-bubblesort.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -62,13 +61,8 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
-
+          uv sync
       - name: Run Codeflash to optimize code
         id: optimize_code_with_benchmarks
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_benchmark_sort.py
+          uv run python tests/scripts/end_to_end_test_benchmark_sort.py

.github/workflows/end-to-end-test-bubblesort-pytest-no-git.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -61,11 +60,8 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
-
+          uv sync
+    
       - name: Remove .git
         run: |
           if [ -d ".git" ]; then
@@ -85,5 +81,4 @@ jobs:
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_bubblesort_pytest.py
+          uv run python tests/scripts/end_to_end_test_bubblesort_pytest.py

.github/workflows/end-to-end-test-bubblesort-unittest.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -61,13 +60,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_bubblesort_unittest.py
+          uv run python tests/scripts/end_to_end_test_bubblesort_unittest.py

.github/workflows/end-to-end-test-coverage.yaml

@@ -1,8 +1,7 @@
 name: Coverage E2E
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -59,14 +58,10 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
-          poetry add black  # my-best-repo in end_to_end_test_coverage.py is configured to use black
+          uv sync
+          uv add black  # my-best-repo in end_to_end_test_coverage.py is configured to use black
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_coverage.py
+          uv run python tests/scripts/end_to_end_test_coverage.py

.github/workflows/end-to-end-test-futurehouse.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -61,13 +60,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_futurehouse.py
+          uv run python tests/scripts/end_to_end_test_futurehouse.py

.github/workflows/end-to-end-test-init-optim.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
   workflow_dispatch:
@@ -60,13 +59,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_init_optimization.py
+          uv run python tests/scripts/end_to_end_test_init_optimization.py

.github/workflows/end-to-end-test-tracer-replay.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
   workflow_dispatch:
@@ -61,13 +60,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_tracer_replay.py
+          uv run python tests/scripts/end_to_end_test_tracer_replay.py

.github/workflows/end-to-end-topological-sort-test.yaml

@@ -1,8 +1,7 @@
 name: end-to-end-test
 
 on:
-  # Use pull_request_target for everything to ensure access to secrets
-  pull_request_target:
+  pull_request:
     paths:
       - '**'  # Trigger for all paths
 
@@ -86,13 +85,9 @@ jobs:
 
       - name: Install dependencies (CLI)
         run: |
-          uv tool install poetry
-          uv venv
-          source .venv/bin/activate
-          poetry install --with dev
+          uv sync
 
       - name: Run Codeflash to optimize code
         id: optimize_code
         run: |
-          source .venv/bin/activate
-          poetry run python tests/scripts/end_to_end_test_topological_sort.py
+          uv run python tests/scripts/end_to_end_test_topological_sort.py

.github/workflows/mypy.yml

@@ -21,11 +21,10 @@ jobs:
         with:
           version: "0.5.30"
 
-      - name: install poetry as a tool
-        run: uv tool install poetry
+      - name: sync uv
+        run: |
+          uv sync
 
-      - name: install dependencies
-        run: uvx poetry install --with dev
 
       - name: Run mypy on allowlist
-        run: uvx poetry run mypy --non-interactive --config-file pyproject.toml @mypy_allowlist.txt
+        run: uv run mypy --non-interactive --config-file pyproject.toml @mypy_allowlist.txt

.github/workflows/unit-tests.yaml

@@ -25,17 +25,9 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           version: "0.5.30"
-      - name: install poetry as a tool
-        run: uv tool install poetry
 
       - name: install dependencies
-        run: uvx poetry install --with dev
+        run: uv sync
 
       - name: Unit tests
-        run: uvx poetry run pytest tests/ --cov --cov-report=xml --benchmark-skip -m "not ci_skip"
-
-      - name: Upload coverage reports to Codecov
-        uses: codecov/codecov-action@v5
-        if: matrix.python-version == '3.12.1'
-        with:
-          token: ${{ secrets.CODECOV_TOKEN }}
+        run: uv run pytest tests/ --benchmark-skip -m "not ci_skip"

README.md

@@ -65,10 +65,10 @@ https://github.com/user-attachments/assets/38f44f4e-be1c-4f84-8db9-63d5ee3e61e5
 
 Join our community for support and discussions. If you have any questions, feel free to reach out to us using one of the following methods:
 
+- [Free live Installation Support](https://calendly.com/codeflash-saurabh/codeflash-setup)
 - [Join our Discord](https://www.codeflash.ai/discord)
 - [Follow us on Twitter](https://x.com/codeflashAI)
 - [Follow us on Linkedin](https://www.linkedin.com/in/saurabh-misra/)
-- [Email founders](mailto:saurabh@codeflash.ai)
 
 ## License
 

code_to_optimize/code_directories/simple_tracer_e2e/workload.py

@@ -21,6 +21,45 @@ def test_threadpool() -> None:
     for r in result:
         print(r)
 
+class AlexNet:
+    def __init__(self, num_classes=1000):
+        self.num_classes = num_classes
+        self.features_size = 256 * 6 * 6
+
+    def forward(self, x):
+        features = self._extract_features(x)
+
+        output = self._classify(features)
+        return output
+
+    def _extract_features(self, x):
+        result = []
+        for i in range(len(x)):
+            pass
+
+        return result
+
+    def _classify(self, features):
+        total = sum(features)
+        return [total % self.num_classes for _ in features]
+
+class SimpleModel:
+    @staticmethod
+    def predict(data):
+        return [x * 2 for x in data]
+
+    @classmethod
+    def create_default(cls):
+        return cls()
+
+def test_models():
+    model = AlexNet(num_classes=10)
+    input_data = [1, 2, 3, 4, 5]
+    result = model.forward(input_data)
+
+    model2 = SimpleModel.create_default()
+    prediction = model2.predict(input_data)
 
 if __name__ == "__main__":
     test_threadpool()
+    test_models()

codeflash/LICENSE

@@ -3,7 +3,7 @@ Business Source License 1.1
 Parameters
 
 Licensor:             CodeFlash Inc.
-Licensed Work:        Codeflash Client version 0.12.x
+Licensed Work:        Codeflash Client version 0.14.x
                       The Licensed Work is (c) 2024 CodeFlash Inc.
 
 Additional Use Grant: None. Production use of the Licensed Work is only permitted
@@ -13,7 +13,7 @@ Additional Use Grant: None. Production use of the Licensed Work is only permitte
                       Platform. Please visit codeflash.ai for further
                       information.
 
-Change Date:          2029-04-23
+Change Date:          2029-06-09
 
 Change License:       MIT
 

codeflash/api/aiservice.py

@@ -11,6 +11,7 @@
 
 from codeflash.cli_cmds.console import console, logger
 from codeflash.code_utils.env_utils import get_codeflash_api_key
+from codeflash.code_utils.git_utils import get_last_commit_author_if_pr_exists, get_repo_owner_and_name
 from codeflash.models.models import OptimizedCandidate
 from codeflash.telemetry.posthog_cf import ph
 from codeflash.version import __version__ as codeflash_version
@@ -97,6 +98,12 @@ def optimize_python_code(  # noqa: D417
 
         """
         start_time = time.perf_counter()
+        try:
+            git_repo_owner, git_repo_name = get_repo_owner_and_name()
+        except Exception as e:
+            logger.warning(f"Could not determine repo owner and name: {e}")
+            git_repo_owner, git_repo_name = None, None
+
         payload = {
             "source_code": source_code,
             "dependency_code": dependency_code,
@@ -105,6 +112,9 @@ def optimize_python_code(  # noqa: D417
             "python_version": platform.python_version(),
             "experiment_metadata": experiment_metadata,
             "codeflash_version": codeflash_version,
+            "current_username": get_last_commit_author_if_pr_exists(None),
+            "repo_owner": git_repo_owner,
+            "repo_name": git_repo_name,
         }
 
         logger.info("Generating optimized candidates…")
@@ -118,7 +128,7 @@ def optimize_python_code(  # noqa: D417
 
         if response.status_code == 200:
             optimizations_json = response.json()["optimizations"]
-            logger.info(f"Generated {len(optimizations_json)} candidates.")
+            logger.info(f"Generated {len(optimizations_json)} candidate optimizations.")
             console.rule()
             end_time = time.perf_counter()
             logger.debug(f"Generating optimizations took {end_time - start_time:.2f} seconds.")
@@ -189,7 +199,7 @@ def optimize_python_code_line_profiler(  # noqa: D417
 
         if response.status_code == 200:
             optimizations_json = response.json()["optimizations"]
-            logger.info(f"Generated {len(optimizations_json)} candidates.")
+            logger.info(f"Generated {len(optimizations_json)} candidate optimizations.")
             console.rule()
             return [
                 OptimizedCandidate(