Spaces:
No application file
No application file
Upload .github\workflows\vdb-tests.yml with huggingface_hub
Browse files
.github//workflows//vdb-tests.yml
ADDED
|
@@ -0,0 +1,86 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
name: Run VDB Tests
|
| 2 |
+
|
| 3 |
+
on:
|
| 4 |
+
pull_request:
|
| 5 |
+
branches:
|
| 6 |
+
- main
|
| 7 |
+
paths:
|
| 8 |
+
- api/core/rag/datasource/**
|
| 9 |
+
- docker/**
|
| 10 |
+
- .github/workflows/vdb-tests.yml
|
| 11 |
+
- api/poetry.lock
|
| 12 |
+
- api/pyproject.toml
|
| 13 |
+
|
| 14 |
+
concurrency:
|
| 15 |
+
group: vdb-tests-${{ github.head_ref || github.run_id }}
|
| 16 |
+
cancel-in-progress: true
|
| 17 |
+
|
| 18 |
+
jobs:
|
| 19 |
+
test:
|
| 20 |
+
name: VDB Tests
|
| 21 |
+
runs-on: ubuntu-latest
|
| 22 |
+
strategy:
|
| 23 |
+
matrix:
|
| 24 |
+
python-version:
|
| 25 |
+
- "3.11"
|
| 26 |
+
- "3.12"
|
| 27 |
+
|
| 28 |
+
steps:
|
| 29 |
+
- name: Checkout code
|
| 30 |
+
uses: actions/checkout@v4
|
| 31 |
+
with:
|
| 32 |
+
fetch-depth: 0
|
| 33 |
+
persist-credentials: false
|
| 34 |
+
|
| 35 |
+
- name: Setup Poetry and Python ${{ matrix.python-version }}
|
| 36 |
+
uses: ./.github/actions/setup-poetry
|
| 37 |
+
with:
|
| 38 |
+
python-version: ${{ matrix.python-version }}
|
| 39 |
+
poetry-lockfile: api/poetry.lock
|
| 40 |
+
|
| 41 |
+
- name: Check Poetry lockfile
|
| 42 |
+
run: |
|
| 43 |
+
poetry check -C api --lock
|
| 44 |
+
poetry show -C api
|
| 45 |
+
|
| 46 |
+
- name: Install dependencies
|
| 47 |
+
run: poetry install -C api --with dev
|
| 48 |
+
|
| 49 |
+
- name: Set up dotenvs
|
| 50 |
+
run: |
|
| 51 |
+
cp docker/.env.example docker/.env
|
| 52 |
+
cp docker/middleware.env.example docker/middleware.env
|
| 53 |
+
|
| 54 |
+
- name: Expose Service Ports
|
| 55 |
+
run: sh .github/workflows/expose_service_ports.sh
|
| 56 |
+
|
| 57 |
+
- name: Set up Vector Store (TiDB)
|
| 58 |
+
uses: hoverkraft-tech/compose-action@v2.0.2
|
| 59 |
+
with:
|
| 60 |
+
compose-file: docker/tidb/docker-compose.yaml
|
| 61 |
+
services: |
|
| 62 |
+
tidb
|
| 63 |
+
tiflash
|
| 64 |
+
|
| 65 |
+
- name: Set up Vector Stores (Weaviate, Qdrant, PGVector, Milvus, PgVecto-RS, Chroma, MyScale, ElasticSearch, Couchbase)
|
| 66 |
+
uses: hoverkraft-tech/compose-action@v2.0.2
|
| 67 |
+
with:
|
| 68 |
+
compose-file: |
|
| 69 |
+
docker/docker-compose.yaml
|
| 70 |
+
services: |
|
| 71 |
+
weaviate
|
| 72 |
+
qdrant
|
| 73 |
+
couchbase-server
|
| 74 |
+
etcd
|
| 75 |
+
minio
|
| 76 |
+
milvus-standalone
|
| 77 |
+
pgvecto-rs
|
| 78 |
+
pgvector
|
| 79 |
+
chroma
|
| 80 |
+
elasticsearch
|
| 81 |
+
|
| 82 |
+
- name: Check TiDB Ready
|
| 83 |
+
run: poetry run -P api python api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py
|
| 84 |
+
|
| 85 |
+
- name: Test Vector Stores
|
| 86 |
+
run: poetry run -P api bash dev/pytest/pytest_vdb.sh
|