llama-stack-mirror/.github/actions/setup-ollama/action.yml

23 lines
690 B
YAML

name: Setup Ollama
description: Start Ollama
inputs:
suite:
description: 'Test suite to use: base, responses, vision, etc.'
required: false
default: ''
runs:
using: "composite"
steps:
- name: Start Ollama
shell: bash
run: |
if [ "${{ inputs.suite }}" == "vision" ]; then
image="ollama-with-vision-model"
else
image="ollama-with-models"
fi
echo "Starting Ollama with image: $image"
docker run -d --name ollama -p 11434:11434 docker.io/llamastack/$image
echo "Verifying Ollama status..."
timeout 30 bash -c 'while ! curl -s -L http://127.0.0.1:11434; do sleep 1 && echo "."; done'