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

Added elasticsearch in conftest

Browse source
This commit is contained in:
Enrico Zimuel 2025-11-04 14:33:41 +01:00
commit 1478e672c8
No known key found for this signature in database
GPG key ID: 6CB203F6934A69F1
374 changed files with 52786 additions and 80518 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/integration/conftest.py
View file

@ -368,12 +368,22 @@ def vector_provider_wrapper(func):
return func(*args, **kwargs)
# For CI tests (replay/record), only use providers that are available in ci-tests environment
if os.environ.get("LLAMA_STACK_TEST_INFERENCE_MODE") in ("replay", "record"):
all_providers = ["faiss", "sqlite-vec"]
else:
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)