Merge branch 'master' into wip-v1.0

.github/actions/uv_setup/action.yml

@@ -1,12 +1,22 @@
-# TODO: https://docs.astral.sh/uv/guides/integration/github/#caching
-
 name: uv-install
-description: Set up Python and uv
+description: Set up Python and uv with caching
 
 inputs:
   python-version:
     description: Python version, supporting MAJOR.MINOR only
     required: true
+  enable-cache:
+    description: Enable caching for uv dependencies
+    required: false
+    default: 'true'
+  cache-suffix:
+    description: Custom cache key suffix for cache invalidation
+    required: false
+    default: ''
+  working-directory:
+    description: Working directory for cache glob scoping
+    required: false
+    default: '**'
 
 env:
   UV_VERSION: "0.5.25"
@@ -15,7 +25,13 @@ runs:
   using: composite
   steps:
     - name: Install uv and set the python version
-      uses: astral-sh/setup-uv@v5
+      uses: astral-sh/setup-uv@v6
       with:
         version: ${{ env.UV_VERSION }}
         python-version: ${{ inputs.python-version }}
+        enable-cache: ${{ inputs.enable-cache }}
+        cache-dependency-glob: |
+          ${{ inputs.working-directory }}/pyproject.toml
+          ${{ inputs.working-directory }}/uv.lock
+          ${{ inputs.working-directory }}/requirements*.txt
+        cache-suffix: ${{ inputs.cache-suffix }}

.github/workflows/_compile_integration_test.yml

@@ -33,6 +33,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ inputs.python-version }}
+          cache-suffix: compile-integration-tests-${{ inputs.working-directory }}
+          working-directory: ${{ inputs.working-directory }}
 
       - name: '📦 Install Integration Dependencies'
         shell: bash

.github/workflows/_integration_test.yml

@@ -34,6 +34,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ inputs.python-version }}
+          cache-suffix: integration-tests-${{ inputs.working-directory }}
+          working-directory: ${{ inputs.working-directory }}
 
       - name: '📦 Install Integration Dependencies'
         shell: bash

.github/workflows/_lint.yml

@@ -39,6 +39,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ inputs.python-version }}
+          cache-suffix: lint-${{ inputs.working-directory }}
+          working-directory: ${{ inputs.working-directory }}
 
       - name: '📦 Install Lint & Typing Dependencies'
         # Also installs dev/lint/test/typing dependencies, to ensure we have

.github/workflows/_test.yml

@@ -39,6 +39,9 @@ jobs:
         id: setup-python
         with:
           python-version: ${{ inputs.python-version }}
+          cache-suffix: test-${{ inputs.working-directory }}
+          working-directory: ${{ inputs.working-directory }}
+
       - name: '📦 Install Test Dependencies'
         shell: bash
         run: uv sync --group test --dev

.github/workflows/_test_doc_imports.yml

@@ -27,6 +27,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ inputs.python-version }}
+          cache-suffix: test-doc-imports-${{ inputs.working-directory }}
+          working-directory: ${{ inputs.working-directory }}
 
       - name: '📦 Install Test Dependencies'
         shell: bash

.github/workflows/_test_pydantic.yml

@@ -40,6 +40,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ inputs.python-version }}
+          cache-suffix: test-pydantic-${{ inputs.working-directory }}
+          working-directory: ${{ inputs.working-directory }}
 
       - name: '📦 Install Test Dependencies'
         shell: bash

.github/workflows/check_diffs.yml

@@ -54,6 +54,7 @@ jobs:
       dependencies: ${{ steps.set-matrix.outputs.dependencies }}
       test-doc-imports: ${{ steps.set-matrix.outputs.test-doc-imports }}
       test-pydantic: ${{ steps.set-matrix.outputs.test-pydantic }}
+      codspeed: ${{ steps.set-matrix.outputs.codspeed }}
   # Run linting only on packages that have changed files
   lint:
     needs: [ build ]
@@ -144,6 +145,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ matrix.job-configs.python-version }}
+          cache-suffix: extended-tests-${{ matrix.job-configs.working-directory }}
+          working-directory: ${{ matrix.job-configs.working-directory }}
 
       - name: '📦 Install Dependencies & Run Extended Tests'
         shell: bash
@@ -166,10 +169,50 @@ jobs:
           # and `set -e` above will cause the step to fail.
           echo "$STATUS" | grep 'nothing to commit, working tree clean'
 
+  # Run codspeed benchmarks only on packages that have changed files
+  codspeed:
+    name: '⚡ CodSpeed Benchmarks'
+    needs: [ build ]
+    if: ${{ needs.build.outputs.codspeed != '[]' && !contains(github.event.pull_request.labels.*.name, 'codspeed-ignore') }}
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        job-configs: ${{ fromJson(needs.build.outputs.codspeed) }}
+      fail-fast: false
+    steps:
+      - uses: actions/checkout@v5
+
+      # We have to use 3.12 as 3.13 is not yet supported
+      - name: '📦 Install UV Package Manager'
+        uses: astral-sh/setup-uv@v6
+        with:
+          python-version: "3.12"
+
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+
+      - name: '📦 Install Test Dependencies'
+        run: uv sync --group test
+        working-directory: ${{ matrix.job-configs.working-directory }}
+
+      - name: '⚡ Run Benchmarks: ${{ matrix.job-configs.working-directory }}'
+        uses: CodSpeedHQ/action@v4
+        with:
+          token: ${{ secrets.CODSPEED_TOKEN }}
+          run: |
+            cd ${{ matrix.job-configs.working-directory }}
+            if [ "${{ matrix.job-configs.working-directory }}" = "libs/core" ]; then
+              uv run --no-sync pytest ./tests/benchmarks --codspeed
+            else
+              uv run --no-sync pytest ./tests/ --codspeed
+            fi
+          mode: ${{ matrix.job-configs.working-directory == 'libs/core' && 'walltime' || 'instrumentation' }}
+
   # Final status check - ensures all required jobs passed before allowing merge
   ci_success:
     name: '✅ CI Success'
-    needs: [build, lint, test, compile-integration-tests, extended-tests, test-doc-imports, test-pydantic]
+    needs: [build, lint, test, compile-integration-tests, extended-tests, test-doc-imports, test-pydantic, codspeed]
     if: |
       always()
     runs-on: ubuntu-latest

.github/workflows/codspeed.yml

@@ -1,66 +0,0 @@
-name: '⚡ CodSpeed'
-
-on:
-  push:
-    branches:
-        - master
-  pull_request:
-  workflow_dispatch:
-
-permissions:
-  contents: read
-
-env:
-  AZURE_OPENAI_CHAT_DEPLOYMENT_NAME: foo
-  AZURE_OPENAI_LEGACY_CHAT_DEPLOYMENT_NAME: foo
-  DEEPSEEK_API_KEY: foo
-  FIREWORKS_API_KEY: foo
-
-jobs:
-  codspeed:
-    name: 'Benchmark'
-    runs-on: ubuntu-latest
-    if: ${{ !contains(github.event.pull_request.labels.*.name, 'codspeed-ignore') }}
-    strategy:
-      matrix:
-        include:
-          - working-directory: libs/core
-            mode: walltime
-          - working-directory: libs/partners/openai
-          - working-directory: libs/partners/anthropic
-          - working-directory: libs/partners/deepseek
-          - working-directory: libs/partners/fireworks
-          - working-directory: libs/partners/xai
-          - working-directory: libs/partners/mistralai
-          - working-directory: libs/partners/groq
-      fail-fast: false
-
-    steps:
-      - uses: actions/checkout@v5
-
-      # We have to use 3.12 as 3.13 is not yet supported
-      - name: '📦 Install UV Package Manager'
-        uses: astral-sh/setup-uv@v6
-        with:
-          python-version: "3.12"
-
-      - uses: actions/setup-python@v6
-        with:
-          python-version: "3.12"
-
-      - name: '📦 Install Test Dependencies'
-        run: uv sync --group test
-        working-directory: ${{ matrix.working-directory }}
-
-      - name: '⚡ Run Benchmarks: ${{ matrix.working-directory }}'
-        uses: CodSpeedHQ/action@v3
-        with:
-          token: ${{ secrets.CODSPEED_TOKEN }}
-          run: |
-            cd ${{ matrix.working-directory }}
-            if [ "${{ matrix.working-directory }}" = "libs/core" ]; then
-              uv run --no-sync pytest ./tests/benchmarks --codspeed
-            else
-              uv run --no-sync pytest ./tests/ --codspeed
-            fi
-          mode: ${{ matrix.mode || 'instrumentation' }}

.github/workflows/run_notebooks.yml

@@ -32,6 +32,8 @@ jobs:
         uses: "./.github/actions/uv_setup"
         with:
           python-version: ${{ github.event.inputs.python_version || '3.11' }}
+          cache-suffix: run-notebooks-${{ github.event.inputs.working-directory || 'all' }}
+          working-directory: ${{ github.event.inputs.working-directory || '**' }}
 
       - name: '🔐 Authenticate to Google Cloud'
         id: 'auth'