This commit is contained in:
Hardik Shah 2025-06-26 13:54:56 -07:00
parent 6b251432c3
commit 058bda0ee7

View file

@ -35,10 +35,8 @@ jobs:
with:
python-version: ${{ matrix.python-version }}
- name: Setup ollama
uses: ./.github/actions/setup-ollama
- name: Setup Chroma
if: matrix.vector-io-provider == 'remote::chromadb'
run: |
docker run --rm -d --pull always \
--name chromadb \
@ -49,6 +47,7 @@ jobs:
chromadb/chroma:latest
- name: Wait for ChromaDB to be ready
if: matrix.vector-io-provider == 'remote::chromadb'
run: |
echo "Waiting for ChromaDB to be ready..."
for i in {1..30}; do
@ -74,11 +73,10 @@ jobs:
- name: Run Vector IO Integration Tests
env:
ENABLE_CHROMADB: true
CHROMADB_URL: http://localhost:8000
OLLAMA_URL: "http://0.0.0.0:11434"
ENABLE_CHROMADB: ${{ matrix.vector-io-provider == 'remote::chromadb' && 'true' || '' }}
CHROMADB_URL: ${{ matrix.vector-io-provider == 'remote::chromadb' && 'http://localhost:8000' || '' }}
run: |
uv run pytest -sv --stack-config="inference=ollama,vector_io=${{ matrix.vector-io-provider }}" \
uv run pytest -sv --stack-config="inference=sentence-transformers,vector_io=${{ matrix.vector-io-provider }}" \
tests/integration/vector_io \
--embedding-model all-MiniLM-L6-v2
@ -94,7 +92,7 @@ jobs:
echo "SANITIZED_PROVIDER=${matrix.vector-io-provider//:/_}" >> $GITHUB_ENV
- name: Write ChromaDB logs to file
if: ${{ always() }}
if: ${{ always() && matrix.vector-io-provider == 'remote::chromadb' }}
run: |
docker logs chromadb > chromadb.log