Install Codeflash Github action

.github/workflows/codeflash.yml

@@ -0,0 +1,48 @@
+name: Codeflash Optimization
+on:
+  pull_request:
+    paths:
+      - '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
+      - uses: denoland/setup-deno@v2
+        with:
+          deno-version: v2.x
+      - name: 📦 Install Dependencies
+        run: uv sync --frozen --all-extras --all-packages --group lint --group docs
+      - name: Functions to Optimize
+        id: discover_step
+        run: |
+          uv run python -c "
+          from codeflash.cli_cmds.cli import parse_args, process_pyproject_config
+          from codeflash.verification.verification_utils import TestConfig
+          from codeflash.discovery.functions_to_optimize import get_functions_to_optimize
+          import os
+          args = process_pyproject_config(parse_args())
+          t = TestConfig(tests_root=args.tests_root, tests_project_rootdir=args.test_project_root, project_root_path=args.project_root,  test_framework=args.test_framework, pytest_cmd=args.pytest_cmd,)
+          f = get_functions_to_optimize(test_cfg=t, ignore_paths=args.ignore_paths, project_root=args.project_root, module_root=args.module_root,optimize_all=None, replay_test=None, file=None, only_get_this_function=None)
+          functions = ' '.join([fto.function_name for ftol in f[0].values() for fto in ftol])
+          open(os.environ.get('GITHUB_OUTPUT'), 'a').write(f'functions_to_optimize={functions}')
+          "
+      - name: ⚡️Codeflash Optimization
+        run: uv run codeflash optimize --only-functions ${{ steps.discover_step.outputs.functions_to_optimize }} -m pytest --inline-snapshot=disable tests/test_examples.py

pydantic_ai_slim/pydantic_ai/profiles/google.py

@@ -10,6 +10,8 @@
 
 def google_model_profile(model_name: str) -> ModelProfile | None:
     """Get the model profile for a Google model."""
+    import time
+    time.sleep(0.001)
     return ModelProfile(
         json_schema_transformer=GoogleJsonSchemaTransformer,
         supports_json_schema_output=True,

pydantic_ai_slim/pyproject.toml

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

pyproject.toml

@@ -283,3 +283,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"]