Skip to content

Commit 651a7a3

Browse files
update test model ids suite
Signed-off-by: AngeloDanducci <angelo.danducci.ii@ibm.com>
1 parent 7c0b9cc commit 651a7a3

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

test/backends/test_model_ids.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,9 @@ def test_ollama_model_names_exist(const_name: str, ollama_name: str) -> None:
9090
# Strip any tag so we query the base model name.
9191
base_name = ollama_name.split(":")[0]
9292
tag = ollama_name.split(":")[1] if ":" in ollama_name else "latest"
93-
registry_url = f"https://registry.ollama.ai/v2/{base_name}/manifests/{tag}"
93+
# Unnamespaced models (no "/" in base_name) live under the "library" namespace.
94+
registry_name = base_name if "/" in base_name else f"library/{base_name}"
95+
registry_url = f"https://registry.ollama.ai/v2/{registry_name}/manifests/{tag}"
9496
try:
9597
req = urllib.request.Request(registry_url, method="HEAD")
9698
urllib.request.urlopen(req, timeout=10)

0 commit comments

Comments
 (0)