diff --git a/.github/workflows/vdb-tests.yml b/.github/workflows/vdb-tests.yml index c7ded0ba850348..3e9532a383e1cd 100644 --- a/.github/workflows/vdb-tests.yml +++ b/.github/workflows/vdb-tests.yml @@ -51,7 +51,24 @@ jobs: - name: Expose Service Ports run: sh .github/workflows/expose_service_ports.sh - - name: Set up Vector Stores (TiDB) + - name: Set up Vector Stores (Weaviate, Qdrant, PGVector, Milvus, PgVecto-RS, Chroma, MyScale, ElasticSearch, Couchbase) + uses: hoverkraft-tech/compose-action@v2.0.2 + with: + compose-file: | + docker/docker-compose.yaml + services: | + weaviate + qdrant + couchbase-server + etcd + minio + milvus-standalone + pgvecto-rs + pgvector + chroma + elasticsearch + + - name: Set up Vector Store (TiDB) uses: hoverkraft-tech/compose-action@v2.0.2 with: compose-file: | @@ -65,4 +82,4 @@ jobs: run: poetry run -C api python api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py - name: Test Vector Stores - run: poetry run -C api pytest api/tests/integration_tests/vdb/tidb_vector + run: poetry run -C api bash dev/pytest/pytest_vdb.sh diff --git a/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py b/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py index a4286232ceebdf..84e8feb6db37ec 100644 --- a/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py +++ b/api/tests/integration_tests/vdb/tidb_vector/check_tiflash_ready.py @@ -37,7 +37,7 @@ def main(): for attempt in range(attempts): try: create_table() - break # 如果成功,则退出循环 + break except Exception as e: print(f"TiFlash is not ready. Exception: {e}") print(f"Attempt {attempt + 1} failed,retry in {retry_wait_seconds} seconds...") diff --git a/docker/tidb/config/tiflash-learner-nightly.toml b/docker/tidb/config/tiflash-learner-nightly.toml index 60daa3f428d97d..9bbb1569a4613c 100644 --- a/docker/tidb/config/tiflash-learner-nightly.toml +++ b/docker/tidb/config/tiflash-learner-nightly.toml @@ -1,11 +1,5 @@ log-file = "/logs/tiflash_tikv.log" -[readpool] - -[readpool.coprocessor] - -[readpool.storage] - [server] engine-addr = "tiflash:4030" addr = "0.0.0.0:20280" @@ -14,32 +8,3 @@ status-addr = "tiflash:20292" [storage] data-dir = "/data/flash" - -[pd] - -[metric] - -[raftstore] -capacity = "10GB" - -[coprocessor] - -[rocksdb] -wal-dir = "" - -[rocksdb.defaultcf] - -[rocksdb.lockcf] - -[rocksdb.writecf] - -[raftdb] - -[raftdb.defaultcf] - -[security] -ca-path = "" -cert-path = "" -key-path = "" - -[import] \ No newline at end of file