From 68c8d9ace56e6c83281a2b2dc965e697b17cc7d7 Mon Sep 17 00:00:00 2001 From: skamenan7 Date: Tue, 12 Aug 2025 13:24:48 -0400 Subject: [PATCH] Fix unit test to expect correct fallback model The test was incorrectly expecting granite model as fallback. Updated to expect all-MiniLM-L6-v2 which is the actual default. --- tests/unit/router/test_embedding_precedence.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/router/test_embedding_precedence.py b/tests/unit/router/test_embedding_precedence.py index 62061157f..27c039865 100644 --- a/tests/unit/router/test_embedding_precedence.py +++ b/tests/unit/router/test_embedding_precedence.py @@ -68,11 +68,11 @@ async def test_explicit_override(monkeypatch): monkeypatch.delenv("LLAMA_STACK_DEFAULT_EMBEDDING_MODEL", raising=False) -async def test_fallback_to_granite(): - """Should fallback to granite model when no defaults set.""" +async def test_fallback_to_default(): + """Should fallback to all-MiniLM-L6-v2 when no defaults set.""" router = VectorIORouter(routing_table=_DummyRoutingTable()) 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