diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index a2f98d01..2d81ba41 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -2,6 +2,9 @@ name: Build
 
 on: [push, pull_request]
 
+env:
+    OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+
 jobs:
     test:
         runs-on: ubuntu-latest
@@ -20,7 +23,7 @@ jobs:
             - name: Setup Graphviz
               uses: ts-graphviz/setup-graphviz@v2
             - name: Serve Ollama Model
-              uses: phil65/ollama-github-action@v1
+              uses: pydantic/ollama-action@v3
               with:
                   model: "smollm2:360m"
             - name: Install uv
diff --git a/.github/workflows/documentation.yml b/.github/workflows/documentation.yml
index 30f44c1f..15521129 100644
--- a/.github/workflows/documentation.yml
+++ b/.github/workflows/documentation.yml
@@ -21,6 +21,9 @@ defaults:
     run:
         shell: bash
 
+env:
+    OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+
 jobs:
     build:
         runs-on: ubuntu-latest
@@ -36,7 +39,7 @@ jobs:
             - name: Setup Graphviz
               uses: ts-graphviz/setup-graphviz@v2
             - name: Serve Ollama Model
-              uses: phil65/ollama-github-action@v1
+              uses: pydantic/ollama-action@v3
               with:
                   model: "smollm2:360m"
             - name: Install uv