diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py index 0bb38481f..683ca7fe9 100644 --- a/tests/integration/conftest.py +++ b/tests/integration/conftest.py @@ -371,7 +371,19 @@ def vector_provider_wrapper(func): inference_mode = os.environ.get("LLAMA_STACK_TEST_INFERENCE_MODE") if inference_mode == "live": # For live tests, try all providers (they'll skip if not available) - all_providers = ["faiss", "sqlite-vec", "milvus", "chromadb", "pgvector", "weaviate", "qdrant", "elasticsearch"] + all_providers = [ + "faiss", + "sqlite-vec", + "milvus", + "chromadb", + "pgvector", + "weaviate", + "qdrant", + "elasticsearch", + ] + else: + # For CI tests (replay/record), only use providers that are available in ci-tests environment + all_providers = ["faiss", "sqlite-vec"] return pytest.mark.parametrize("vector_io_provider_id", all_providers)(wrapper)