Skip to content

Commit 00d8414

Browse files
authored
fix(tests): limit vector store providers for record mode in CI tests (#3898)
The vector_provider_wrapper was only limiting providers to faiss/sqlite-vec for replay mode, but CI tests also run in record mode with the same limited set of providers. This caused test failures when trying to test against milvus, chromadb, pgvector, weaviate, and qdrant which aren't configured in the record job.
1 parent 7c0e434 commit 00d8414

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integration/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -363,8 +363,8 @@ def wrapper(*args, **kwargs):
363363

364364
return func(*args, **kwargs)
365365

366-
# For replay tests, only use providers that are available in ci-tests environment
367-
if os.environ.get("LLAMA_STACK_TEST_INFERENCE_MODE") == "replay":
366+
# For CI tests (replay/record), only use providers that are available in ci-tests environment
367+
if os.environ.get("LLAMA_STACK_TEST_INFERENCE_MODE") in ("replay", "record"):
368368
all_providers = ["faiss", "sqlite-vec"]
369369
else:
370370
# For live tests, try all providers (they'll skip if not available)

0 commit comments

Comments
 (0)