chore: use uv caches (#32919)

Especially helpful for the text splitters tests where we're installing
pytorch (expensive and slow slow slow). Should speed up CI by 5-10 mins.

w/o caches, CI taking 20 minutes 😨 
w/ caches, CI taking 3 minutes

.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

@@ -145,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

.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'