37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""Pin the Mistral provider-logo pattern to cover Mixtral and Ministral.
|
|
|
|
The pattern was /mistral/i, which does not match "mixtral" (note the x) or
|
|
"ministral" -- Mistral AI's flagship MoE and edge families -- so those models
|
|
rendered with no provider logo unless they carried a "mistralai/" prefix.
|
|
"""
|
|
import json
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
_REPO = Path(__file__).resolve().parent.parent
|
|
_HELPER = _REPO / "static" / "js" / "providers.js"
|
|
pytestmark = pytest.mark.skipif(not shutil.which("node"), reason="node not on PATH")
|
|
|
|
|
|
def _has_logo(model):
|
|
js = (
|
|
f"import {{ providerLogo }} from '{_HELPER.as_posix()}';"
|
|
f"console.log(JSON.stringify(providerLogo({json.dumps(model)}) !== null));"
|
|
)
|
|
p = subprocess.run(["node", "--input-type=module"], input=js, capture_output=True, text=True, cwd=str(_REPO), timeout=30)
|
|
assert p.returncode == 0, p.stderr
|
|
return json.loads(p.stdout.strip())
|
|
|
|
|
|
def test_mixtral_ministral_get_a_logo():
|
|
assert _has_logo("mixtral-8x7b") is True
|
|
assert _has_logo("ministral-8b") is True
|
|
assert _has_logo("mistral-large-latest") is True
|
|
|
|
|
|
def test_unknown_vendor_has_no_logo():
|
|
assert _has_logo("totally-unknown-model-xyz") is False
|