phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-12 04:00:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(tests): limit vector store providers for record mode in CI tests

Browse source 
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.
This commit is contained in:
Derek Higgins 2025-10-24 00:30:16 +01:00
parent 9916cb3b17
commit b268b3b620
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/conftest.py
View file

@ -363,8 +363,8 @@ def vector_provider_wrapper(func):
return func(*args, **kwargs) return func(*args, **kwargs)
# For replay tests, only use providers that are available in ci-tests environment # For CI tests (replay/record), only use providers that are available in ci-tests environment
if os.environ.get("LLAMA_STACK_TEST_INFERENCE_MODE") == "replay": if os.environ.get("LLAMA_STACK_TEST_INFERENCE_MODE") in ("replay", "record"):
all_providers = ["faiss", "sqlite-vec"] all_providers = ["faiss", "sqlite-vec"]
else: else:
# For live tests, try all providers (they'll skip if not available) # For live tests, try all providers (they'll skip if not available)