modelparams 0.0.1__py3-none-any.whl

modelparams/_generated/registry.py

@@ -0,0 +1,270 @@
+# AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
+# Do not edit by hand; the YAML catalog under /models is the source of truth.
+
+from typing import Any
+
+from modelparams.types import alibaba
+from modelparams.types import anthropic
+from modelparams.types import cerebras
+from modelparams.types import cohere
+from modelparams.types import deepseek
+from modelparams.types import google
+from modelparams.types import groq
+from modelparams.types import meta
+from modelparams.types import minimax
+from modelparams.types import mistral
+from modelparams.types import moonshot
+from modelparams.types import nvidia
+from modelparams.types import openai
+from modelparams.types import perplexity
+from modelparams.types import thinking_machines
+from modelparams.types import xai
+from modelparams.types import xiaomi
+from modelparams.types import z_ai
+
+from .model_ids import ModelId
+
+PARAM_TYPES: dict[ModelId, Any] = {
+    "alibaba/qwen-flash": alibaba.Qwen_FlashParams,
+    "alibaba/qwen-plus": alibaba.Qwen_PlusParams,
+    "alibaba/qwen3-coder-flash": alibaba.Qwen3_Coder_FlashParams,
+    "alibaba/qwen3-coder-plus": alibaba.Qwen3_Coder_PlusParams,
+    "alibaba/qwen3-max": alibaba.Qwen3_MaxParams,
+    "alibaba/qwen3.5": alibaba.Qwen3_5Params,
+    "alibaba/qwen3.5-flash": alibaba.Qwen3_5_FlashParams,
+    "alibaba/qwen3.6-flash": alibaba.Qwen3_6_FlashParams,
+    "alibaba/qwen3.7-max": alibaba.Qwen3_7_MaxParams,
+    "alibaba/qwen3.7-plus": alibaba.Qwen3_7_PlusParams,
+    "alibaba/qwen3.8-max": alibaba.Qwen3_8_MaxParams,
+    "alibaba/qwq-plus": alibaba.Qwq_PlusParams,
+    "anthropic/claude-3-5-haiku-20241022": anthropic.Claude_3_5_Haiku_20241022Params,
+    "anthropic/claude-3-5-haiku-latest": anthropic.Claude_3_5_Haiku_LatestParams,
+    "anthropic/claude-3-5-sonnet-20241022": anthropic.Claude_3_5_Sonnet_20241022Params,
+    "anthropic/claude-3-5-sonnet-latest": anthropic.Claude_3_5_Sonnet_LatestParams,
+    "anthropic/claude-3-7-sonnet-20250219": anthropic.Claude_3_7_Sonnet_20250219Params,
+    "anthropic/claude-3-7-sonnet-latest": anthropic.Claude_3_7_Sonnet_LatestParams,
+    "anthropic/claude-3-opus-20240229": anthropic.Claude_3_Opus_20240229Params,
+    "anthropic/claude-3-opus-latest": anthropic.Claude_3_Opus_LatestParams,
+    "anthropic/claude-fable-5": anthropic.Claude_Fable_5Params,
+    "anthropic/claude-fable-5-subscription": anthropic.Claude_Fable_5_SubscriptionParams,
+    "anthropic/claude-haiku-4": anthropic.Claude_Haiku_4Params,
+    "anthropic/claude-haiku-4-5": anthropic.Claude_Haiku_4_5Params,
+    "anthropic/claude-haiku-4-5-20251001": anthropic.Claude_Haiku_4_5_20251001Params,
+    "anthropic/claude-haiku-4-5-20251001-subscription": anthropic.Claude_Haiku_4_5_20251001_SubscriptionParams,
+    "anthropic/claude-haiku-4-5-subscription": anthropic.Claude_Haiku_4_5_SubscriptionParams,
+    "anthropic/claude-haiku-4-subscription": anthropic.Claude_Haiku_4_SubscriptionParams,
+    "anthropic/claude-opus-4-1-20250805": anthropic.Claude_Opus_4_1_20250805Params,
+    "anthropic/claude-opus-4-1-20250805-subscription": anthropic.Claude_Opus_4_1_20250805_SubscriptionParams,
+    "anthropic/claude-opus-4-20250514": anthropic.Claude_Opus_4_20250514Params,
+    "anthropic/claude-opus-4-20250514-subscription": anthropic.Claude_Opus_4_20250514_SubscriptionParams,
+    "anthropic/claude-opus-4-5-20251101": anthropic.Claude_Opus_4_5_20251101Params,
+    "anthropic/claude-opus-4-5-20251101-subscription": anthropic.Claude_Opus_4_5_20251101_SubscriptionParams,
+    "anthropic/claude-opus-4-6": anthropic.Claude_Opus_4_6Params,
+    "anthropic/claude-opus-4-6-subscription": anthropic.Claude_Opus_4_6_SubscriptionParams,
+    "anthropic/claude-opus-4-7": anthropic.Claude_Opus_4_7Params,
+    "anthropic/claude-opus-4-7-subscription": anthropic.Claude_Opus_4_7_SubscriptionParams,
+    "anthropic/claude-opus-4-8": anthropic.Claude_Opus_4_8Params,
+    "anthropic/claude-opus-4-8-subscription": anthropic.Claude_Opus_4_8_SubscriptionParams,
+    "anthropic/claude-opus-4-subscription": anthropic.Claude_Opus_4_SubscriptionParams,
+    "anthropic/claude-opus-5": anthropic.Claude_Opus_5Params,
+    "anthropic/claude-opus-5-subscription": anthropic.Claude_Opus_5_SubscriptionParams,
+    "anthropic/claude-sonnet-4-20250514": anthropic.Claude_Sonnet_4_20250514Params,
+    "anthropic/claude-sonnet-4-20250514-subscription": anthropic.Claude_Sonnet_4_20250514_SubscriptionParams,
+    "anthropic/claude-sonnet-4-5": anthropic.Claude_Sonnet_4_5Params,
+    "anthropic/claude-sonnet-4-5-20250929": anthropic.Claude_Sonnet_4_5_20250929Params,
+    "anthropic/claude-sonnet-4-5-20250929-subscription": anthropic.Claude_Sonnet_4_5_20250929_SubscriptionParams,
+    "anthropic/claude-sonnet-4-5-subscription": anthropic.Claude_Sonnet_4_5_SubscriptionParams,
+    "anthropic/claude-sonnet-4-6": anthropic.Claude_Sonnet_4_6Params,
+    "anthropic/claude-sonnet-4-6-subscription": anthropic.Claude_Sonnet_4_6_SubscriptionParams,
+    "anthropic/claude-sonnet-4-subscription": anthropic.Claude_Sonnet_4_SubscriptionParams,
+    "anthropic/claude-sonnet-5": anthropic.Claude_Sonnet_5Params,
+    "cerebras/zai-glm-4.7": cerebras.Zai_Glm_4_7Params,
+    "cohere/command-a-03-2025": cohere.Command_A_03_2025Params,
+    "cohere/command-a-plus-05-2026": cohere.Command_A_Plus_05_2026Params,
+    "cohere/command-a-reasoning-08-2025": cohere.Command_A_Reasoning_08_2025Params,
+    "cohere/command-a-translate-08-2025": cohere.Command_A_Translate_08_2025Params,
+    "cohere/command-a-vision-07-2025": cohere.Command_A_Vision_07_2025Params,
+    "cohere/command-r-08-2024": cohere.Command_R_08_2024Params,
+    "cohere/command-r-plus-08-2024": cohere.Command_R_Plus_08_2024Params,
+    "cohere/command-r7b-12-2024": cohere.Command_R7b_12_2024Params,
+    "deepseek/deepseek-chat": deepseek.Deepseek_ChatParams,
+    "deepseek/deepseek-reasoner": deepseek.Deepseek_ReasonerParams,
+    "deepseek/deepseek-v4-flash": deepseek.Deepseek_V4_FlashParams,
+    "deepseek/deepseek-v4-pro": deepseek.Deepseek_V4_ProParams,
+    "google/gemini-2.5-flash": google.Gemini_2_5_FlashParams,
+    "google/gemini-2.5-flash-lite": google.Gemini_2_5_Flash_LiteParams,
+    "google/gemini-2.5-flash-lite-subscription": google.Gemini_2_5_Flash_Lite_SubscriptionParams,
+    "google/gemini-2.5-flash-subscription": google.Gemini_2_5_Flash_SubscriptionParams,
+    "google/gemini-2.5-pro": google.Gemini_2_5_ProParams,
+    "google/gemini-2.5-pro-subscription": google.Gemini_2_5_Pro_SubscriptionParams,
+    "google/gemini-3-flash-preview-subscription": google.Gemini_3_Flash_Preview_SubscriptionParams,
+    "google/gemini-3.1-flash-lite": google.Gemini_3_1_Flash_LiteParams,
+    "google/gemini-3.1-flash-lite-preview-subscription": google.Gemini_3_1_Flash_Lite_Preview_SubscriptionParams,
+    "google/gemini-3.1-flash-lite-subscription": google.Gemini_3_1_Flash_Lite_SubscriptionParams,
+    "google/gemini-3.1-pro-preview-subscription": google.Gemini_3_1_Pro_Preview_SubscriptionParams,
+    "google/gemini-3.5-flash": google.Gemini_3_5_FlashParams,
+    "google/gemini-flash-latest": google.Gemini_Flash_LatestParams,
+    "google/gemma-3-12b-it": google.Gemma_3_12b_ItParams,
+    "google/gemma-3-1b-it": google.Gemma_3_1b_ItParams,
+    "google/gemma-3-27b-it": google.Gemma_3_27b_ItParams,
+    "google/gemma-3-4b-it": google.Gemma_3_4b_ItParams,
+    "google/gemma-3n-E2B-it": google.Gemma_3n_E2b_ItParams,
+    "google/gemma-3n-E4B-it": google.Gemma_3n_E4b_ItParams,
+    "google/gemma-4-12B-it": google.Gemma_4_12b_ItParams,
+    "google/gemma-4-26b-a4b-it": google.Gemma_4_26b_A4b_ItParams,
+    "google/gemma-4-31b-it": google.Gemma_4_31b_ItParams,
+    "google/gemma-4-E2B-it": google.Gemma_4_E2b_ItParams,
+    "google/gemma-4-E4B-it": google.Gemma_4_E4b_ItParams,
+    "groq/qwen3-32b": groq.Qwen3_32bParams,
+    "meta/Llama-3.3-70B-Instruct": meta.Llama_3_3_70b_InstructParams,
+    "meta/Llama-3.3-8B-Instruct": meta.Llama_3_3_8b_InstructParams,
+    "meta/Llama-4-Maverick-17B-128E-Instruct-FP8": meta.Llama_4_Maverick_17b_128e_Instruct_Fp8Params,
+    "meta/Llama-4-Scout-17B-16E-Instruct-FP8": meta.Llama_4_Scout_17b_16e_Instruct_Fp8Params,
+    "minimax/minimax-m2": minimax.Minimax_M2Params,
+    "minimax/MiniMax-M2-subscription": minimax.Minimax_M2_SubscriptionParams,
+    "minimax/minimax-m2.1": minimax.Minimax_M2_1Params,
+    "minimax/minimax-m2.1-highspeed": minimax.Minimax_M2_1_HighspeedParams,
+    "minimax/MiniMax-M2.1-highspeed-subscription": minimax.Minimax_M2_1_Highspeed_SubscriptionParams,
+    "minimax/MiniMax-M2.1-subscription": minimax.Minimax_M2_1_SubscriptionParams,
+    "minimax/minimax-m2.5": minimax.Minimax_M2_5Params,
+    "minimax/minimax-m2.5-highspeed": minimax.Minimax_M2_5_HighspeedParams,
+    "minimax/MiniMax-M2.5-highspeed-subscription": minimax.Minimax_M2_5_Highspeed_SubscriptionParams,
+    "minimax/MiniMax-M2.5-subscription": minimax.Minimax_M2_5_SubscriptionParams,
+    "minimax/minimax-m2.7": minimax.Minimax_M2_7Params,
+    "minimax/minimax-m2.7-highspeed": minimax.Minimax_M2_7_HighspeedParams,
+    "minimax/MiniMax-M2.7-highspeed-subscription": minimax.Minimax_M2_7_Highspeed_SubscriptionParams,
+    "minimax/MiniMax-M2.7-subscription": minimax.Minimax_M2_7_SubscriptionParams,
+    "minimax/minimax-m3": minimax.Minimax_M3Params,
+    "minimax/MiniMax-M3-subscription": minimax.Minimax_M3_SubscriptionParams,
+    "mistral/codestral-latest": mistral.Codestral_LatestParams,
+    "mistral/devstral-2512": mistral.Devstral_2512Params,
+    "mistral/devstral-latest": mistral.Devstral_LatestParams,
+    "mistral/magistral-medium-latest": mistral.Magistral_Medium_LatestParams,
+    "mistral/magistral-small-latest": mistral.Magistral_Small_LatestParams,
+    "mistral/ministral-14b-latest": mistral.Ministral_14b_LatestParams,
+    "mistral/ministral-3b-latest": mistral.Ministral_3b_LatestParams,
+    "mistral/ministral-8b-latest": mistral.Ministral_8b_LatestParams,
+    "mistral/mistral-large-latest": mistral.Mistral_Large_LatestParams,
+    "mistral/mistral-medium-3.5": mistral.Mistral_Medium_3_5Params,
+    "mistral/mistral-medium-latest": mistral.Mistral_Medium_LatestParams,
+    "mistral/mistral-small-latest": mistral.Mistral_Small_LatestParams,
+    "mistral/open-mistral-nemo": mistral.Open_Mistral_NemoParams,
+    "moonshot/kimi-k2.5": moonshot.Kimi_K2_5Params,
+    "moonshot/kimi-k2.6": moonshot.Kimi_K2_6Params,
+    "moonshot/kimi-k2.6-subscription": moonshot.Kimi_K2_6_SubscriptionParams,
+    "moonshot/kimi-k2.7-code": moonshot.Kimi_K2_7_CodeParams,
+    "moonshot/kimi-k2.7-code-highspeed": moonshot.Kimi_K2_7_Code_HighspeedParams,
+    "moonshot/kimi-k2.7-code-highspeed-subscription": moonshot.Kimi_K2_7_Code_Highspeed_SubscriptionParams,
+    "moonshot/kimi-k2.7-code-subscription": moonshot.Kimi_K2_7_Code_SubscriptionParams,
+    "moonshot/kimi-k3": moonshot.Kimi_K3Params,
+    "moonshot/moonshot-v1-128k": moonshot.Moonshot_V1_128kParams,
+    "moonshot/moonshot-v1-32k": moonshot.Moonshot_V1_32kParams,
+    "moonshot/moonshot-v1-8k": moonshot.Moonshot_V1_8kParams,
+    "nvidia/gliner-pii": nvidia.Gliner_PiiParams,
+    "nvidia/llama-3.1-nemoguard-8b-topic-control": nvidia.Llama_3_1_Nemoguard_8b_Topic_ControlParams,
+    "nvidia/llama-3.1-nemotron-nano-8b-v1": nvidia.Llama_3_1_Nemotron_Nano_8b_V1Params,
+    "nvidia/llama-3.1-nemotron-safety-guard-8b-v3": nvidia.Llama_3_1_Nemotron_Safety_Guard_8b_V3Params,
+    "nvidia/llama-3.1-nemotron-ultra-253b-v1": nvidia.Llama_3_1_Nemotron_Ultra_253b_V1Params,
+    "nvidia/llama-3.3-nemotron-super-49b-v1": nvidia.Llama_3_3_Nemotron_Super_49b_V1Params,
+    "nvidia/llama-3.3-nemotron-super-49b-v1.5": nvidia.Llama_3_3_Nemotron_Super_49b_V1_5Params,
+    "nvidia/nemoguard-jailbreak-detect": nvidia.Nemoguard_Jailbreak_DetectParams,
+    "nvidia/nemotron-3-nano-30b-a3b": nvidia.Nemotron_3_Nano_30b_A3bParams,
+    "nvidia/nemotron-3-super-120b-a12b": nvidia.Nemotron_3_Super_120b_A12bParams,
+    "nvidia/nemotron-3-ultra-550b-a55b": nvidia.Nemotron_3_Ultra_550b_A55bParams,
+    "nvidia/nemotron-3-ultra-subscription": nvidia.Nemotron_3_Ultra_SubscriptionParams,
+    "nvidia/nemotron-content-safety-reasoning-4b": nvidia.Nemotron_Content_Safety_Reasoning_4bParams,
+    "nvidia/nemotron-mini-4b-instruct": nvidia.Nemotron_Mini_4b_InstructParams,
+    "nvidia/riva-translate-4b-instruct-v1.1": nvidia.Riva_Translate_4b_Instruct_V1_1Params,
+    "nvidia/usdcode-llama-3.1-70b-instruct": nvidia.Usdcode_Llama_3_1_70b_InstructParams,
+    "openai/chatgpt-4o-latest": openai.Chatgpt_4o_LatestParams,
+    "openai/gpt-3.5-turbo": openai.Gpt_3_5_TurboParams,
+    "openai/gpt-4-turbo": openai.Gpt_4_TurboParams,
+    "openai/gpt-4-turbo-2024-04-09": openai.Gpt_4_Turbo_2024_04_09Params,
+    "openai/gpt-4.1": openai.Gpt_4_1Params,
+    "openai/gpt-4.1-mini": openai.Gpt_4_1_MiniParams,
+    "openai/gpt-4.1-nano": openai.Gpt_4_1_NanoParams,
+    "openai/gpt-4o": openai.Gpt_4oParams,
+    "openai/gpt-4o-2024-11-20": openai.Gpt_4o_2024_11_20Params,
+    "openai/gpt-4o-mini": openai.Gpt_4o_MiniParams,
+    "openai/gpt-5": openai.Gpt_5Params,
+    "openai/gpt-5-chat-latest": openai.Gpt_5_Chat_LatestParams,
+    "openai/gpt-5-mini": openai.Gpt_5_MiniParams,
+    "openai/gpt-5-nano": openai.Gpt_5_NanoParams,
+    "openai/gpt-5.1": openai.Gpt_5_1Params,
+    "openai/gpt-5.1-codex-max-subscription": openai.Gpt_5_1_Codex_Max_SubscriptionParams,
+    "openai/gpt-5.1-codex-subscription": openai.Gpt_5_1_Codex_SubscriptionParams,
+    "openai/gpt-5.2": openai.Gpt_5_2Params,
+    "openai/gpt-5.2-codex-subscription": openai.Gpt_5_2_Codex_SubscriptionParams,
+    "openai/gpt-5.2-subscription": openai.Gpt_5_2_SubscriptionParams,
+    "openai/gpt-5.3-codex": openai.Gpt_5_3_CodexParams,
+    "openai/gpt-5.3-codex-spark-subscription": openai.Gpt_5_3_Codex_Spark_SubscriptionParams,
+    "openai/gpt-5.3-codex-subscription": openai.Gpt_5_3_Codex_SubscriptionParams,
+    "openai/gpt-5.4": openai.Gpt_5_4Params,
+    "openai/gpt-5.4-mini": openai.Gpt_5_4_MiniParams,
+    "openai/gpt-5.4-mini-subscription": openai.Gpt_5_4_Mini_SubscriptionParams,
+    "openai/gpt-5.4-nano": openai.Gpt_5_4_NanoParams,
+    "openai/gpt-5.4-pro": openai.Gpt_5_4_ProParams,
+    "openai/gpt-5.4-pro-subscription": openai.Gpt_5_4_Pro_SubscriptionParams,
+    "openai/gpt-5.4-subscription": openai.Gpt_5_4_SubscriptionParams,
+    "openai/gpt-5.5": openai.Gpt_5_5Params,
+    "openai/gpt-5.5-pro": openai.Gpt_5_5_ProParams,
+    "openai/gpt-5.5-pro-subscription": openai.Gpt_5_5_Pro_SubscriptionParams,
+    "openai/gpt-5.5-subscription": openai.Gpt_5_5_SubscriptionParams,
+    "openai/gpt-5.6-luna": openai.Gpt_5_6_LunaParams,
+    "openai/gpt-5.6-luna-subscription": openai.Gpt_5_6_Luna_SubscriptionParams,
+    "openai/gpt-5.6-sol": openai.Gpt_5_6_SolParams,
+    "openai/gpt-5.6-sol-subscription": openai.Gpt_5_6_Sol_SubscriptionParams,
+    "openai/gpt-5.6-terra": openai.Gpt_5_6_TerraParams,
+    "openai/gpt-5.6-terra-subscription": openai.Gpt_5_6_Terra_SubscriptionParams,
+    "openai/gpt-oss-120b": openai.Gpt_Oss_120bParams,
+    "openai/gpt-oss-20b": openai.Gpt_Oss_20bParams,
+    "openai/gpt-oss-safeguard-120b": openai.Gpt_Oss_Safeguard_120bParams,
+    "openai/gpt-oss-safeguard-20b": openai.Gpt_Oss_Safeguard_20bParams,
+    "openai/o1": openai.O1Params,
+    "openai/o1-mini": openai.O1_MiniParams,
+    "openai/o1-preview": openai.O1_PreviewParams,
+    "openai/o3": openai.O3Params,
+    "openai/o3-mini": openai.O3_MiniParams,
+    "openai/o3-pro": openai.O3_ProParams,
+    "openai/o4-mini": openai.O4_MiniParams,
+    "perplexity/sonar": perplexity.SonarParams,
+    "perplexity/sonar-deep-research": perplexity.Sonar_Deep_ResearchParams,
+    "perplexity/sonar-pro": perplexity.Sonar_ProParams,
+    "perplexity/sonar-reasoning-pro": perplexity.Sonar_Reasoning_ProParams,
+    "thinking-machines/Inkling": thinking_machines.InklingParams,
+    "xai/grok-4.20-0309-non-reasoning": xai.Grok_4_20_0309_Non_ReasoningParams,
+    "xai/grok-4.20-0309-non-reasoning-subscription": xai.Grok_4_20_0309_Non_Reasoning_SubscriptionParams,
+    "xai/grok-4.20-0309-reasoning": xai.Grok_4_20_0309_ReasoningParams,
+    "xai/grok-4.20-0309-reasoning-subscription": xai.Grok_4_20_0309_Reasoning_SubscriptionParams,
+    "xai/grok-4.20-multi-agent-0309": xai.Grok_4_20_Multi_Agent_0309Params,
+    "xai/grok-4.3": xai.Grok_4_3Params,
+    "xai/grok-4.3-subscription": xai.Grok_4_3_SubscriptionParams,
+    "xai/grok-4.5": xai.Grok_4_5Params,
+    "xai/grok-4.5-subscription": xai.Grok_4_5_SubscriptionParams,
+    "xai/grok-build-0.1": xai.Grok_Build_0_1Params,
+    "xai/grok-build-0.1-subscription": xai.Grok_Build_0_1_SubscriptionParams,
+    "xiaomi/mimo-v2.5": xiaomi.Mimo_V2_5Params,
+    "xiaomi/mimo-v2.5-pro": xiaomi.Mimo_V2_5_ProParams,
+    "xiaomi/mimo-v2.5-subscription": xiaomi.Mimo_V2_5_SubscriptionParams,
+    "z-ai/glm-4.5": z_ai.Glm_4_5Params,
+    "z-ai/glm-4.5-air": z_ai.Glm_4_5_AirParams,
+    "z-ai/glm-4.5-air-subscription": z_ai.Glm_4_5_Air_SubscriptionParams,
+    "z-ai/glm-4.5-airx": z_ai.Glm_4_5_AirxParams,
+    "z-ai/glm-4.5-flash": z_ai.Glm_4_5_FlashParams,
+    "z-ai/glm-4.5-subscription": z_ai.Glm_4_5_SubscriptionParams,
+    "z-ai/glm-4.5-x": z_ai.Glm_4_5_XParams,
+    "z-ai/glm-4.6": z_ai.Glm_4_6Params,
+    "z-ai/glm-4.6-subscription": z_ai.Glm_4_6_SubscriptionParams,
+    "z-ai/glm-4.7": z_ai.Glm_4_7Params,
+    "z-ai/glm-4.7-flash": z_ai.Glm_4_7_FlashParams,
+    "z-ai/glm-4.7-flashx": z_ai.Glm_4_7_FlashxParams,
+    "z-ai/glm-4.7-subscription": z_ai.Glm_4_7_SubscriptionParams,
+    "z-ai/glm-5": z_ai.Glm_5Params,
+    "z-ai/glm-5-subscription": z_ai.Glm_5_SubscriptionParams,
+    "z-ai/glm-5-turbo": z_ai.Glm_5_TurboParams,
+    "z-ai/glm-5-turbo-subscription": z_ai.Glm_5_Turbo_SubscriptionParams,
+    "z-ai/glm-5.1": z_ai.Glm_5_1Params,
+    "z-ai/glm-5.1-subscription": z_ai.Glm_5_1_SubscriptionParams,
+    "z-ai/glm-5.2": z_ai.Glm_5_2Params,
+    "z-ai/glm-5.2-subscription": z_ai.Glm_5_2_SubscriptionParams,
+}

modelparams/catalog.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+from collections.abc import Mapping
+from importlib.resources import files
+from types import MappingProxyType
+from typing import cast
+
+from pydantic import TypeAdapter
+
+from ._generated.model_ids import MODEL_IDS, ModelId, Provider
+from .models import CatalogEntry, JsonPrimitive, Parameter
+
+_catalog_bytes = files("modelparams._generated").joinpath("catalog.json").read_bytes()
+CATALOG = TypeAdapter(tuple[CatalogEntry, ...]).validate_json(_catalog_bytes)
+
+
+def _entry_id(entry: CatalogEntry) -> str:
+    suffix = "" if entry.auth_type == "api_key" else "-subscription"
+    return f"{entry.provider}/{entry.model}{suffix}"
+
+
+BY_ID: Mapping[ModelId, CatalogEntry] = MappingProxyType(
+    cast(dict[ModelId, CatalogEntry], {_entry_id(entry): entry for entry in CATALOG})
+)
+DEFAULTS: Mapping[ModelId, Mapping[str, JsonPrimitive]] = MappingProxyType(
+    {
+        model_id: MappingProxyType(
+            {
+                parameter.path: parameter.default
+                for parameter in entry.params
+                if "default" in parameter.model_fields_set
+            }
+        )
+        for model_id, entry in BY_ID.items()
+    }
+)
+
+
+def get_model(model_id: ModelId) -> CatalogEntry:
+    return BY_ID[model_id]
+
+
+def get_defaults(model_id: ModelId) -> Mapping[str, JsonPrimitive]:
+    return DEFAULTS[model_id]
+
+
+def list_models(provider: Provider | None = None) -> tuple[ModelId, ...]:
+    if provider is None:
+        return MODEL_IDS
+    prefix = f"{provider}/"
+    return tuple(model_id for model_id in MODEL_IDS if model_id.startswith(prefix))
+
+
+def get_param(model_id: ModelId, path: str) -> Parameter | None:
+    return next((parameter for parameter in BY_ID[model_id].params if parameter.path == path), None)
+
+
+def list_all_models() -> tuple[CatalogEntry, ...]:
+    return CATALOG

modelparams/models.py

@@ -0,0 +1,62 @@
+from __future__ import annotations
+
+from typing import Literal, TypeAlias
+
+from pydantic import BaseModel, ConfigDict, Field
+
+JsonPrimitive: TypeAlias = str | int | float | bool | None
+JsonNumber: TypeAlias = int | float
+AuthType: TypeAlias = Literal["api_key", "subscription"]
+ParamType: TypeAlias = Literal["boolean", "enum", "integer", "number", "string"]
+ParamGroup: TypeAlias = Literal[
+    "generation_length",
+    "sampling",
+    "reasoning",
+    "tooling",
+    "output_format",
+    "observability",
+    "provider_metadata",
+]
+
+
+class FrozenModel(BaseModel):
+    model_config = ConfigDict(frozen=True, extra="forbid", populate_by_name=True)
+
+
+class ParamRange(FrozenModel):
+    min: JsonNumber | None = None
+    max: JsonNumber | None = None
+    step: JsonNumber | None = None
+
+
+class ApplicabilityCondition(FrozenModel):
+    not_: JsonPrimitive | tuple[JsonPrimitive, ...] = Field(alias="not")
+
+
+ApplicabilityValue: TypeAlias = JsonPrimitive | tuple[JsonPrimitive, ...] | ApplicabilityCondition
+ApplicabilityRule: TypeAlias = dict[str, ApplicabilityValue]
+ApplicabilityRuleSet: TypeAlias = ApplicabilityRule | tuple[ApplicabilityRule, ...]
+
+
+class Applicability(FrozenModel):
+    only: ApplicabilityRuleSet | None = None
+    except_: ApplicabilityRuleSet | None = Field(default=None, alias="except")
+
+
+class Parameter(FrozenModel):
+    path: str
+    label: str
+    description: str
+    group: ParamGroup
+    type: ParamType
+    default: JsonPrimitive = None
+    range: ParamRange | None = None
+    values: tuple[JsonPrimitive, ...] | None = None
+    applicability: Applicability | None = None
+
+
+class CatalogEntry(FrozenModel):
+    provider: str
+    auth_type: AuthType = Field(alias="authType")
+    model: str
+    params: tuple[Parameter, ...]

modelparams/py.typed

@@ -0,0 +1 @@
+

modelparams/types/__init__.py

@@ -0,0 +1,4 @@
+# AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
+# Do not edit by hand; the YAML catalog under /models is the source of truth.
+
+__all__ = ["alibaba", "anthropic", "cerebras", "cohere", "deepseek", "google", "groq", "meta", "minimax", "mistral", "moonshot", "nvidia", "openai", "perplexity", "thinking_machines", "xai", "xiaomi", "z_ai"]

modelparams/types/alibaba.py

@@ -0,0 +1,183 @@
+# AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
+# Do not edit by hand; the YAML catalog under /models is the source of truth.
+
+# ruff: noqa: F401
+
+from typing import Annotated, Literal
+
+from pydantic import ConfigDict, Field
+from typing_extensions import TypedDict
+
+_PARAMS_CONFIG = ConfigDict(strict=True, extra="forbid")
+
+Qwen_FlashParams = TypedDict(
+    "Qwen_FlashParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+        "extra_body.chat_template_kwargs.enable_thinking": bool,
+    },
+    total=False,
+)
+setattr(Qwen_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen_PlusParams = TypedDict(
+    "Qwen_PlusParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+        "extra_body.chat_template_kwargs.enable_thinking": bool,
+    },
+    total=False,
+)
+setattr(Qwen_PlusParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_Coder_FlashParams = TypedDict(
+    "Qwen3_Coder_FlashParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwen3_Coder_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_Coder_PlusParams = TypedDict(
+    "Qwen3_Coder_PlusParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwen3_Coder_PlusParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_MaxParams = TypedDict(
+    "Qwen3_MaxParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+        "extra_body.chat_template_kwargs.enable_thinking": bool,
+    },
+    total=False,
+)
+setattr(Qwen3_MaxParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_5Params = TypedDict(
+    "Qwen3_5Params",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+        "extra_body.chat_template_kwargs.enable_thinking": bool,
+    },
+    total=False,
+)
+setattr(Qwen3_5Params, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_5_FlashParams = TypedDict(
+    "Qwen3_5_FlashParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+        "extra_body.chat_template_kwargs.enable_thinking": bool,
+    },
+    total=False,
+)
+setattr(Qwen3_5_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_6_FlashParams = TypedDict(
+    "Qwen3_6_FlashParams",
+    {
+        "max_completion_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=0)],
+        "extra_body.enable_thinking": bool,
+        "extra_body.thinking_budget": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwen3_6_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_7_MaxParams = TypedDict(
+    "Qwen3_7_MaxParams",
+    {
+        "max_completion_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=0)],
+        "extra_body.enable_thinking": bool,
+        "extra_body.thinking_budget": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwen3_7_MaxParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_7_PlusParams = TypedDict(
+    "Qwen3_7_PlusParams",
+    {
+        "max_completion_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=0)],
+        "extra_body.enable_thinking": bool,
+        "extra_body.thinking_budget": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwen3_7_PlusParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwen3_8_MaxParams = TypedDict(
+    "Qwen3_8_MaxParams",
+    {
+        "max_completion_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=0)],
+        "extra_body.enable_thinking": bool,
+        "extra_body.thinking_budget": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwen3_8_MaxParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+Qwq_PlusParams = TypedDict(
+    "Qwq_PlusParams",
+    {
+        "max_tokens": Annotated[int, Field(ge=1)],
+        "temperature": Annotated[float, Field(ge=0, le=2)],
+        "top_p": Annotated[float, Field(ge=0, le=1)],
+        "extra_body.top_k": Annotated[int, Field(ge=1)],
+    },
+    total=False,
+)
+setattr(Qwq_PlusParams, "__pydantic_config__", _PARAMS_CONFIG)
+
+__all__ = [
+    "Qwen_FlashParams",
+    "Qwen_PlusParams",
+    "Qwen3_Coder_FlashParams",
+    "Qwen3_Coder_PlusParams",
+    "Qwen3_MaxParams",
+    "Qwen3_5Params",
+    "Qwen3_5_FlashParams",
+    "Qwen3_6_FlashParams",
+    "Qwen3_7_MaxParams",
+    "Qwen3_7_PlusParams",
+    "Qwen3_8_MaxParams",
+    "Qwq_PlusParams",
+]