fix(tokenlab): align model metadata with schema

This commit is contained in:
hedging8563 2026-07-09 14:42:01 +08:00
parent 04571530bc
commit ede53bd65b
3 changed files with 38 additions and 30 deletions

View file

@ -45025,8 +45025,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45038,8 +45037,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true
@ -45057,8 +45058,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45089,8 +45089,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45121,8 +45120,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45134,8 +45132,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true
@ -45153,8 +45153,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45166,8 +45165,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true
@ -45185,8 +45186,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45217,8 +45217,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45230,8 +45229,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true

View file

@ -45258,8 +45258,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45271,8 +45270,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true
@ -45290,8 +45291,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45322,8 +45322,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45354,8 +45353,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45367,8 +45365,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true
@ -45386,8 +45386,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45399,8 +45398,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true
@ -45418,8 +45419,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45450,8 +45450,7 @@
"source": "https://api.tokenlab.sh/v1/models",
"supported_endpoints": [
"/v1/chat/completions",
"/v1/responses",
"/v1/messages"
"/v1/responses"
],
"supported_modalities": [
"text",
@ -45463,8 +45462,10 @@
"supports_function_calling": true,
"supports_native_streaming": true,
"supports_prompt_caching": true,
"supports_adaptive_thinking": true,
"supports_reasoning": true,
"supports_response_schema": true,
"supports_sampling_params": false,
"supports_system_messages": true,
"supports_tool_choice": true,
"supports_vision": true

View file

@ -147,9 +147,15 @@ class TestTokenLabCostMap:
assert info["litellm_provider"] == "tokenlab"
assert info["mode"] in {"chat", "embedding"}
def test_claude_models_advertise_messages_endpoint(self):
def test_claude_model_cost_map_keeps_schema_supported_endpoints(self):
from litellm.llms.openai_like.json_loader import JSONProviderRegistry
endpoints = litellm.model_cost["tokenlab/claude-opus-4-8"]["supported_endpoints"]
assert "/v1/messages" in endpoints
assert endpoints == ["/v1/chat/completions", "/v1/responses"]
provider_config = JSONProviderRegistry.get("tokenlab")
assert provider_config is not None
assert "/v1/messages" in provider_config.supported_endpoints
def test_chat_cost_is_wired(self):
prompt_cost, completion_cost = litellm.cost_per_token(