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

Fix unit test to expect correct fallback model

Browse source 
The test was incorrectly expecting granite model as fallback.
Updated to expect all-MiniLM-L6-v2 which is the actual default.
This commit is contained in:
skamenan7 2025-08-12 13:24:48 -04:00 • committed by Sumanth Kamenani
parent 2e3621f32b
commit 68c8d9ace5
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/router/test_embedding_precedence.py
View file

@ -68,11 +68,11 @@ async def test_explicit_override(monkeypatch):
monkeypatch.delenv("LLAMA_STACK_DEFAULT_EMBEDDING_MODEL", raising=False) monkeypatch.delenv("LLAMA_STACK_DEFAULT_EMBEDDING_MODEL", raising=False)
async def test_fallback_to_granite(): async def test_fallback_to_default():
"""Should fallback to granite model when no defaults set.""" """Should fallback to all-MiniLM-L6-v2 when no defaults set."""
router = VectorIORouter(routing_table=_DummyRoutingTable()) router = VectorIORouter(routing_table=_DummyRoutingTable())
model, dim = await router._resolve_embedding_model(None) model, dim = await router._resolve_embedding_model(None)
assert model == "ibm-granite/granite-embedding-125m-english" assert model == "all-MiniLM-L6-v2"
assert dim == 384 assert dim == 384