34 lines
1.4 KiB
Python
34 lines
1.4 KiB
Python
|
"""Tests for anonymous custom endpoint fallback in get_available_models().
|
||
|
|
|
||
|
|
Verifies that when /v1/models probe fails for an anonymous custom endpoint
|
||
|
|
(configured via bare base_url, not custom_providers[]), the provider group
|
||
|
|
is still added to the picker with an empty model list so the user can type
|
||
|
|
a model ID manually (#2542).
|
||
|
|
"""
|
||
|
|
|
||
|
|
import re
|
||
|
|
|
||
|
|
REPO = __file__.rsplit("/", 2)[-3] if "/" in __file__ else "."
|
||
|
|
CONFIG_PY = open(f"{REPO}/api/config.py").read() if REPO != "." else ""
|
||
|
|
|
||
|
|
|
||
|
|
def test_custom_anonymous_endpoint_empty_models_fallback_in_get_available_models():
|
||
|
|
"""get_available_models() must add a 'custom' provider group with empty
|
||
|
|
models when cfg_base_url is set but the /v1/models probe returned nothing
|
||
|
|
(anonymous endpoint — no match in custom_providers[])."""
|
||
|
|
assert 'elif pid == "custom" and cfg_base_url:' in CONFIG_PY, (
|
||
|
|
"Anonymous custom endpoints with failed /v1/models probe must get "
|
||
|
|
"an empty-model fallback group instead of being silently dropped (#2542)"
|
||
|
|
)
|
||
|
|
# Verify the empty models dict is correct
|
||
|
|
assert '"models": []' in CONFIG_PY.split('elif pid == "custom" and cfg_base_url:')[1].split("groups.append")[1][:200], (
|
||
|
|
"The fallback group must have an empty models list"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_revert_custom_endpoint_fallback_comment():
|
||
|
|
"""The #2542 comment should reference the issue."""
|
||
|
|
assert "#2542" in CONFIG_PY, (
|
||
|
|
"The fallback should reference the issue number for traceability"
|
||
|
|
)
|