See if ollama is cached

Author: jgchn

.github/actions/run-examples/action.yml

@@ -48,16 +48,16 @@ runs:
       uses: actions/setup-python@v5
       with:
         python-version: ${{ inputs.python-version }}
-    - name: Cache pip
-      uses: actions/cache@v4
-      with:
-        # This path is specific to Ubuntu
-        path: ${{ env.pythonLocation }}
-        # Look to see if there is a cache hit for the setup file
-        key: ${{ inputs.runner-os }}-pip-new3-${{ env.pythonLocation }}-${{ hashFiles('setup.py') }}
-        restore-keys: |
-          ${{ inputs.runner-os }}-pip-new3
-          ${{ inputs.runner-os }}-new3
+    # - name: Cache pip
+    #   uses: actions/cache@v4
+    #   with:
+    #     # This path is specific to Ubuntu
+    #     path: ${{ env.pythonLocation }}
+    #     # Look to see if there is a cache hit for the setup file
+    #     key: ${{ inputs.runner-os }}-pip-new3-${{ env.pythonLocation }}-${{ hashFiles('setup.py') }}
+    #     restore-keys: |
+    #       ${{ inputs.runner-os }}-pip-new3
+    #       ${{ inputs.runner-os }}-new3
     - name: Install dependencies
       shell: bash
       run: pip install --upgrade --upgrade-strategy eager .[all]

.github/workflows/run-examples-prep.yml

@@ -45,6 +45,26 @@ jobs:
         echo "value=$value"
         echo "update_results=$value" >> $GITHUB_OUTPUT
       # Install yq if needed
+    
+    # Cache Ollama before running examples 
+    - name: Restore Ollama model cache
+      uses: actions/cache@v4
+      with:
+        path: ~/.ollama
+        key: ollama-cache-${{ runner.os }}-llama3
+        restore-keys: ollama-cache-${{ runner.os }}
+
+    # Cache pip before running examples
+    - name: Cache pip
+      uses: actions/cache@v4
+      with:
+        # This path is specific to Ubuntu
+        path: ${{ env.pythonLocation }}
+        # Look to see if there is a cache hit for the setup file
+        key: ${{ runner.os }}-pip-new3-${{ env.pythonLocation }}-${{ hashFiles('setup.py') }}
+        restore-keys: |
+          ${{ runner.os }}-pip-new3
+          ${{ runner.os }}-new3
       
     - uses: ./.github/actions/run-examples
       if: ${{ steps.changed-pdl-files.outputs.all_changed_files_count > 0 }}

examples/chatbot/chatbot.pdl

@@ -5,6 +5,7 @@ text:
   message: "What is your query?\n"
 - repeat:
     text:
+    - Hello
     # Send context to Granite model hosted at ollama
     - model: ollama_chat/granite3.2:2b
     # Allow the user to type 'yes', 'no', or anything else, storing

tests/test_examples_run.yaml

@@ -1,4 +1,4 @@
-update_results: false
+update_results: true
 check: []
 skip:
   - examples/cldk/cldk-assistant.pdl