codeflash workflow for pydantic ai

Author: aseembits93

.github/workflows/codeflash.yml

@@ -0,0 +1,32 @@
+name: Codeflash
+on:
+  pull_request:
+    paths:
+      # So that this workflow only runs when code within the target module is modified
+      - 'pydantic_ai_slim/pydantic_ai/**'
+  workflow_dispatch:
+concurrency:
+  # Any new push to the PR will cancel the previous run, so that only the latest code is optimized
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+jobs:
+  optimize:
+    name: Optimize new Python code
+    # Don't run codeflash on codeflash-ai[bot] commits, prevent duplicate optimizations
+    if: ${{ github.actor != 'codeflash-ai[bot]' }}
+    runs-on: ubuntu-latest
+    env:
+      CODEFLASH_API_KEY: ${{ secrets.CODEFLASH_API_KEY }}
+    steps:
+      - name: 🛎️ Checkout
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - name: 🐍 Setup UV
+        uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true
+      - name: 📦 Install Dependencies
+        run: uv sync --package pydantic-ai-slim
+      - name: ⚡️Codeflash Optimization
+        run: uv run codeflash

pydantic_ai_slim/pyproject.toml

@@ -103,6 +103,7 @@ dev = [
     "boto3-stubs[bedrock-runtime]",
     "strict-no-cover>=0.1.1",
     "pytest-xdist>=3.6.1",
+    "codeflash>=0.15.4"
 ]
 
 [tool.hatch.metadata]

pyproject.toml

@@ -287,3 +287,10 @@ check-hidden = true
 # Ignore "formatting" like **L**anguage
 ignore-regex = '\*\*[A-Z]\*\*[a-z]+\b'
 ignore-words-list = 'asend,aci'
+
+[tool.codeflash]
+module-root = "pydantic_ai_slim/pydantic_ai"
+tests-root = "tests"
+test-framework = "pytest"
+ignore-paths = []
+formatter-cmds = ["ruff check --exit-zero --fix $file", "ruff format $file"]