pull first

Signed-off-by: Adrian Cole <[email protected]>
square · Sep 3, 2024 · a73f63d · a73f63d
1 parent b8172b1
commit a73f63d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -41,15 +41,18 @@ jobs:
       if: matrix.python-version == '3.12'
       id: cache-ollama
       uses: actions/cache@v4
-      with:
+      with: # make the cache key based on where the OLLAMA_MODEL constant is defined.
         path: ~/.ollama/models  # default directory for Ollama models
-        key: ${{ runner.os }}-ollama
+        key: ollama-${{ hashFiles('./src/exchange/providers/ollama.py') }}
 
-    - name: Start Ollama
+    - name: Start Ollama and pull model
       if: matrix.python-version == '3.12'
       run: |
         curl -fsSL https://ollama.com/install.sh | sh
         ollama serve &
+        sleep 5
+        OLLAMA_MODEL=$(uv run python -c "from src.exchange.providers.ollama import OLLAMA_MODEL; print(OLLAMA_MODEL)")
+        ollama pull $OLLAMA_MODEL
 
     - name: Run Ollama tests
       if: matrix.python-version == '3.12'