mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-27 06:28:50 +00:00
chore: Enabling teste for Weaviate
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
062c6a419a
commit
3c4a9cc624
1 changed files with 28 additions and 1 deletions
|
@ -22,7 +22,7 @@ jobs:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
strategy:
|
strategy:
|
||||||
matrix:
|
matrix:
|
||||||
vector-io-provider: ["inline::faiss", "inline::sqlite-vec", "inline::milvus", "remote::chromadb", "remote::pgvector"]
|
vector-io-provider: ["inline::faiss", "inline::sqlite-vec", "inline::milvus", "remote::chromadb", "remote::pgvector", "remote::weaviate"]
|
||||||
python-version: ["3.12", "3.13"]
|
python-version: ["3.12", "3.13"]
|
||||||
fail-fast: false # we want to run all tests regardless of failure
|
fail-fast: false # we want to run all tests regardless of failure
|
||||||
|
|
||||||
|
@ -46,6 +46,14 @@ jobs:
|
||||||
-e ANONYMIZED_TELEMETRY=FALSE \
|
-e ANONYMIZED_TELEMETRY=FALSE \
|
||||||
chromadb/chroma:latest
|
chromadb/chroma:latest
|
||||||
|
|
||||||
|
- name: Setup Weaviate
|
||||||
|
if: matrix.vector-io-provider == 'remote::weaviate'
|
||||||
|
run: |
|
||||||
|
docker run --rm -d --pull always \
|
||||||
|
--name weaviate \
|
||||||
|
-p 8080:8080 -p 50051:50051 \
|
||||||
|
cr.weaviate.io/semitechnologies/weaviate:1.32.0
|
||||||
|
|
||||||
- name: Start PGVector DB
|
- name: Start PGVector DB
|
||||||
if: matrix.vector-io-provider == 'remote::pgvector'
|
if: matrix.vector-io-provider == 'remote::pgvector'
|
||||||
run: |
|
run: |
|
||||||
|
@ -91,6 +99,21 @@ jobs:
|
||||||
docker logs chromadb
|
docker logs chromadb
|
||||||
exit 1
|
exit 1
|
||||||
|
|
||||||
|
- name: Wait for Weaviate to be ready
|
||||||
|
if: matrix.vector-io-provider == 'remote::weaviate'
|
||||||
|
run: |
|
||||||
|
echo "Waiting for Weaviate to be ready..."
|
||||||
|
for i in {1..30}; do
|
||||||
|
if curl -s http://localhost:8080 | grep -q "https://weaviate.io/developers/weaviate/current/"; then
|
||||||
|
echo "Weaviate is ready!"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
sleep 2
|
||||||
|
done
|
||||||
|
echo "Weaviate failed to start"
|
||||||
|
docker logs weaviate
|
||||||
|
exit 1
|
||||||
|
|
||||||
- name: Build Llama Stack
|
- name: Build Llama Stack
|
||||||
run: |
|
run: |
|
||||||
uv run llama stack build --template ci-tests --image-type venv
|
uv run llama stack build --template ci-tests --image-type venv
|
||||||
|
@ -111,6 +134,10 @@ jobs:
|
||||||
PGVECTOR_DB: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
|
PGVECTOR_DB: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
|
||||||
PGVECTOR_USER: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
|
PGVECTOR_USER: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
|
||||||
PGVECTOR_PASSWORD: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
|
PGVECTOR_PASSWORD: ${{ matrix.vector-io-provider == 'remote::pgvector' && 'llamastack' || '' }}
|
||||||
|
ENABLE_WEAVIATE: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'true' || '' }}
|
||||||
|
WEAVIATE_API_KEY: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'llamastack' || '' }}
|
||||||
|
WEAVIATE_CLUSTER_URL: ${{ matrix.vector-io-provider == 'remote::weaviate' && 'http://localhost:8080' || '' }}
|
||||||
|
|
||||||
run: |
|
run: |
|
||||||
uv run pytest -sv --stack-config="inference=inline::sentence-transformers,vector_io=${{ matrix.vector-io-provider }}" \
|
uv run pytest -sv --stack-config="inference=inline::sentence-transformers,vector_io=${{ matrix.vector-io-provider }}" \
|
||||||
tests/integration/vector_io \
|
tests/integration/vector_io \
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue