wehub-resource-sync
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from utils.available_models import (
|
|
_model_ids_from_openai_compatible_payload,
|
|
is_together_api_base_url,
|
|
normalize_openai_compatible_base_url,
|
|
)
|
|
|
|
|
|
def test_is_together_api_base_url():
|
|
assert is_together_api_base_url("https://api.together.ai/v1")
|
|
assert is_together_api_base_url("https://api.together.xyz")
|
|
assert not is_together_api_base_url("https://api.fireworks.ai/inference/v1")
|
|
|
|
|
|
def test_model_ids_from_openai_compatible_payload_openai_shape():
|
|
payload = {
|
|
"data": [
|
|
{"id": "meta-llama/Llama-3-8b-chat-hf"},
|
|
{"name": "legacy-name"},
|
|
]
|
|
}
|
|
assert _model_ids_from_openai_compatible_payload(payload) == [
|
|
"meta-llama/Llama-3-8b-chat-hf",
|
|
"legacy-name",
|
|
]
|
|
|
|
|
|
def test_model_ids_from_openai_compatible_payload_top_level_list():
|
|
payload = [
|
|
{"id": "openai/gpt-oss-20b"},
|
|
{"id": "meta-llama/Llama-3-8b-chat-hf"},
|
|
]
|
|
assert _model_ids_from_openai_compatible_payload(payload) == [
|
|
"openai/gpt-oss-20b",
|
|
"meta-llama/Llama-3-8b-chat-hf",
|
|
]
|
|
|
|
|
|
def test_normalize_openai_compatible_base_url_appends_v1():
|
|
assert (
|
|
normalize_openai_compatible_base_url("https://api.together.ai")
|
|
== "https://api.together.ai/v1"
|
|
)
|