phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-11 13:44:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 32930868de into 2f58d87c22

Browse source
This commit is contained in:
Sumanth Kamenani 2025-09-24 09:30:04 +02:00 • committed by GitHub
commit 689f1db815
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 284 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
tests/integration/conftest.py
View file

@ -94,6 +94,19 @@ def pytest_configure(config):
if not current:
setattr(config.option, dest, value)
# After processing CLI --env overrides, ensure global default embedding model is set for vector-store operations
embedding_model_opt = config.getoption("--embedding-model") or "sentence-transformers/all-MiniLM-L6-v2"
if embedding_model_opt and not os.getenv("LLAMA_STACK_DEFAULT_EMBEDDING_MODEL"):
# Use first value in comma-separated list (if any)
default_model = embedding_model_opt.split(",")[0].strip()
os.environ["LLAMA_STACK_DEFAULT_EMBEDDING_MODEL"] = default_model
logger.info(f"Setting LLAMA_STACK_DEFAULT_EMBEDDING_MODEL={default_model}")
embedding_dim_opt = config.getoption("--embedding-dimension") or 384
if not os.getenv("LLAMA_STACK_DEFAULT_EMBEDDING_DIMENSION") and embedding_dim_opt:
os.environ["LLAMA_STACK_DEFAULT_EMBEDDING_DIMENSION"] = str(embedding_dim_opt)
logger.info(f"Setting LLAMA_STACK_DEFAULT_EMBEDDING_DIMENSION={embedding_dim_opt}")
def pytest_addoption(parser):
parser.addoption(