modelparams 0.0.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of modelparams might be problematic. Click here for more details.

Files changed (32) hide show
  1. modelparams/__init__.py +60 -0
  2. modelparams/_generated/__init__.py +2 -0
  3. modelparams/_generated/catalog.json +18338 -0
  4. modelparams/_generated/model_ids.py +536 -0
  5. modelparams/_generated/registry.py +270 -0
  6. modelparams/catalog.py +59 -0
  7. modelparams/models.py +62 -0
  8. modelparams/py.typed +1 -0
  9. modelparams/types/__init__.py +4 -0
  10. modelparams/types/alibaba.py +183 -0
  11. modelparams/types/anthropic.py +620 -0
  12. modelparams/types/cerebras.py +33 -0
  13. modelparams/types/cohere.py +182 -0
  14. modelparams/types/deepseek.py +69 -0
  15. modelparams/types/google.py +388 -0
  16. modelparams/types/groq.py +33 -0
  17. modelparams/types/meta.py +78 -0
  18. modelparams/types/minimax.py +214 -0
  19. modelparams/types/mistral.py +250 -0
  20. modelparams/types/moonshot.py +154 -0
  21. modelparams/types/nvidia.py +244 -0
  22. modelparams/types/openai.py +610 -0
  23. modelparams/types/perplexity.py +98 -0
  24. modelparams/types/thinking_machines.py +28 -0
  25. modelparams/types/xai.py +174 -0
  26. modelparams/types/xiaomi.py +59 -0
  27. modelparams/types/z_ai.py +331 -0
  28. modelparams/validation.py +19 -0
  29. modelparams-0.0.1.dist-info/METADATA +104 -0
  30. modelparams-0.0.1.dist-info/RECORD +32 -0
  31. modelparams-0.0.1.dist-info/WHEEL +4 -0
  32. modelparams-0.0.1.dist-info/licenses/LICENSE +21 -0
modelparams/types/meta.py ADDED
@@ -0,0 +1,78 @@
1
+ # AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
2
+ # Do not edit by hand; the YAML catalog under /models is the source of truth.
3
+
4
+ # ruff: noqa: F401
5
+
6
+ from typing import Annotated, Literal
7
+
8
+ from pydantic import ConfigDict, Field
9
+ from typing_extensions import TypedDict
10
+
11
+ _PARAMS_CONFIG = ConfigDict(strict=True, extra="forbid")
12
+
13
+ Llama_3_3_70b_InstructParams = TypedDict(
14
+ "Llama_3_3_70b_InstructParams",
15
+ {
16
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
17
+ "temperature": float,
18
+ "top_p": float,
19
+ "top_k": int,
20
+ "repetition_penalty": float,
21
+ "response_format.type": Literal["text", "json_schema"],
22
+ "tool_choice": Literal["auto", "none", "required"],
23
+ },
24
+ total=False,
25
+ )
26
+ setattr(Llama_3_3_70b_InstructParams, "__pydantic_config__", _PARAMS_CONFIG)
27
+
28
+ Llama_3_3_8b_InstructParams = TypedDict(
29
+ "Llama_3_3_8b_InstructParams",
30
+ {
31
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
32
+ "temperature": float,
33
+ "top_p": float,
34
+ "top_k": int,
35
+ "repetition_penalty": float,
36
+ "response_format.type": Literal["text", "json_schema"],
37
+ "tool_choice": Literal["auto", "none", "required"],
38
+ },
39
+ total=False,
40
+ )
41
+ setattr(Llama_3_3_8b_InstructParams, "__pydantic_config__", _PARAMS_CONFIG)
42
+
43
+ Llama_4_Maverick_17b_128e_Instruct_Fp8Params = TypedDict(
44
+ "Llama_4_Maverick_17b_128e_Instruct_Fp8Params",
45
+ {
46
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
47
+ "temperature": float,
48
+ "top_p": float,
49
+ "top_k": int,
50
+ "repetition_penalty": float,
51
+ "response_format.type": Literal["text", "json_schema"],
52
+ "tool_choice": Literal["auto", "none", "required"],
53
+ },
54
+ total=False,
55
+ )
56
+ setattr(Llama_4_Maverick_17b_128e_Instruct_Fp8Params, "__pydantic_config__", _PARAMS_CONFIG)
57
+
58
+ Llama_4_Scout_17b_16e_Instruct_Fp8Params = TypedDict(
59
+ "Llama_4_Scout_17b_16e_Instruct_Fp8Params",
60
+ {
61
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
62
+ "temperature": float,
63
+ "top_p": float,
64
+ "top_k": int,
65
+ "repetition_penalty": float,
66
+ "response_format.type": Literal["text", "json_schema"],
67
+ "tool_choice": Literal["auto", "none", "required"],
68
+ },
69
+ total=False,
70
+ )
71
+ setattr(Llama_4_Scout_17b_16e_Instruct_Fp8Params, "__pydantic_config__", _PARAMS_CONFIG)
72
+
73
+ __all__ = [
74
+ "Llama_3_3_70b_InstructParams",
75
+ "Llama_3_3_8b_InstructParams",
76
+ "Llama_4_Maverick_17b_128e_Instruct_Fp8Params",
77
+ "Llama_4_Scout_17b_16e_Instruct_Fp8Params",
78
+ ]
modelparams/types/minimax.py ADDED
@@ -0,0 +1,214 @@
1
+ # AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
2
+ # Do not edit by hand; the YAML catalog under /models is the source of truth.
3
+
4
+ # ruff: noqa: F401
5
+
6
+ from typing import Annotated, Literal
7
+
8
+ from pydantic import ConfigDict, Field
9
+ from typing_extensions import TypedDict
10
+
11
+ _PARAMS_CONFIG = ConfigDict(strict=True, extra="forbid")
12
+
13
+ Minimax_M2Params = TypedDict(
14
+ "Minimax_M2Params",
15
+ {
16
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
17
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
18
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
19
+ "reasoning_split": bool,
20
+ },
21
+ total=False,
22
+ )
23
+ setattr(Minimax_M2Params, "__pydantic_config__", _PARAMS_CONFIG)
24
+
25
+ Minimax_M2_SubscriptionParams = TypedDict(
26
+ "Minimax_M2_SubscriptionParams",
27
+ {
28
+ "max_tokens": Annotated[int, Field(ge=1)],
29
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
30
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
31
+ },
32
+ total=False,
33
+ )
34
+ setattr(Minimax_M2_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
35
+
36
+ Minimax_M2_1Params = TypedDict(
37
+ "Minimax_M2_1Params",
38
+ {
39
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
40
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
41
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
42
+ "reasoning_split": bool,
43
+ },
44
+ total=False,
45
+ )
46
+ setattr(Minimax_M2_1Params, "__pydantic_config__", _PARAMS_CONFIG)
47
+
48
+ Minimax_M2_1_HighspeedParams = TypedDict(
49
+ "Minimax_M2_1_HighspeedParams",
50
+ {
51
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
52
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
53
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
54
+ "reasoning_split": bool,
55
+ },
56
+ total=False,
57
+ )
58
+ setattr(Minimax_M2_1_HighspeedParams, "__pydantic_config__", _PARAMS_CONFIG)
59
+
60
+ Minimax_M2_1_Highspeed_SubscriptionParams = TypedDict(
61
+ "Minimax_M2_1_Highspeed_SubscriptionParams",
62
+ {
63
+ "max_tokens": Annotated[int, Field(ge=1)],
64
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
65
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
66
+ },
67
+ total=False,
68
+ )
69
+ setattr(Minimax_M2_1_Highspeed_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
70
+
71
+ Minimax_M2_1_SubscriptionParams = TypedDict(
72
+ "Minimax_M2_1_SubscriptionParams",
73
+ {
74
+ "max_tokens": Annotated[int, Field(ge=1)],
75
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
76
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
77
+ },
78
+ total=False,
79
+ )
80
+ setattr(Minimax_M2_1_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
81
+
82
+ Minimax_M2_5Params = TypedDict(
83
+ "Minimax_M2_5Params",
84
+ {
85
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
86
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
87
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
88
+ "reasoning_split": bool,
89
+ },
90
+ total=False,
91
+ )
92
+ setattr(Minimax_M2_5Params, "__pydantic_config__", _PARAMS_CONFIG)
93
+
94
+ Minimax_M2_5_HighspeedParams = TypedDict(
95
+ "Minimax_M2_5_HighspeedParams",
96
+ {
97
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
98
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
99
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
100
+ "reasoning_split": bool,
101
+ },
102
+ total=False,
103
+ )
104
+ setattr(Minimax_M2_5_HighspeedParams, "__pydantic_config__", _PARAMS_CONFIG)
105
+
106
+ Minimax_M2_5_Highspeed_SubscriptionParams = TypedDict(
107
+ "Minimax_M2_5_Highspeed_SubscriptionParams",
108
+ {
109
+ "max_tokens": Annotated[int, Field(ge=1)],
110
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
111
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
112
+ },
113
+ total=False,
114
+ )
115
+ setattr(Minimax_M2_5_Highspeed_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
116
+
117
+ Minimax_M2_5_SubscriptionParams = TypedDict(
118
+ "Minimax_M2_5_SubscriptionParams",
119
+ {
120
+ "max_tokens": Annotated[int, Field(ge=1)],
121
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
122
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
123
+ },
124
+ total=False,
125
+ )
126
+ setattr(Minimax_M2_5_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
127
+
128
+ Minimax_M2_7Params = TypedDict(
129
+ "Minimax_M2_7Params",
130
+ {
131
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
132
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
133
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
134
+ "reasoning_split": bool,
135
+ },
136
+ total=False,
137
+ )
138
+ setattr(Minimax_M2_7Params, "__pydantic_config__", _PARAMS_CONFIG)
139
+
140
+ Minimax_M2_7_HighspeedParams = TypedDict(
141
+ "Minimax_M2_7_HighspeedParams",
142
+ {
143
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
144
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
145
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
146
+ "reasoning_split": bool,
147
+ },
148
+ total=False,
149
+ )
150
+ setattr(Minimax_M2_7_HighspeedParams, "__pydantic_config__", _PARAMS_CONFIG)
151
+
152
+ Minimax_M2_7_Highspeed_SubscriptionParams = TypedDict(
153
+ "Minimax_M2_7_Highspeed_SubscriptionParams",
154
+ {
155
+ "max_tokens": Annotated[int, Field(ge=1)],
156
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
157
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
158
+ },
159
+ total=False,
160
+ )
161
+ setattr(Minimax_M2_7_Highspeed_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
162
+
163
+ Minimax_M2_7_SubscriptionParams = TypedDict(
164
+ "Minimax_M2_7_SubscriptionParams",
165
+ {
166
+ "max_tokens": Annotated[int, Field(ge=1)],
167
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
168
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
169
+ },
170
+ total=False,
171
+ )
172
+ setattr(Minimax_M2_7_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
173
+
174
+ Minimax_M3Params = TypedDict(
175
+ "Minimax_M3Params",
176
+ {
177
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
178
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
179
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
180
+ "reasoning_split": bool,
181
+ },
182
+ total=False,
183
+ )
184
+ setattr(Minimax_M3Params, "__pydantic_config__", _PARAMS_CONFIG)
185
+
186
+ Minimax_M3_SubscriptionParams = TypedDict(
187
+ "Minimax_M3_SubscriptionParams",
188
+ {
189
+ "max_tokens": Annotated[int, Field(ge=1)],
190
+ "temperature": Annotated[float, Field(ge=0.01, le=1)],
191
+ "top_p": Annotated[float, Field(ge=0.01, le=1)],
192
+ },
193
+ total=False,
194
+ )
195
+ setattr(Minimax_M3_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
196
+
197
+ __all__ = [
198
+ "Minimax_M2Params",
199
+ "Minimax_M2_SubscriptionParams",
200
+ "Minimax_M2_1Params",
201
+ "Minimax_M2_1_HighspeedParams",
202
+ "Minimax_M2_1_Highspeed_SubscriptionParams",
203
+ "Minimax_M2_1_SubscriptionParams",
204
+ "Minimax_M2_5Params",
205
+ "Minimax_M2_5_HighspeedParams",
206
+ "Minimax_M2_5_Highspeed_SubscriptionParams",
207
+ "Minimax_M2_5_SubscriptionParams",
208
+ "Minimax_M2_7Params",
209
+ "Minimax_M2_7_HighspeedParams",
210
+ "Minimax_M2_7_Highspeed_SubscriptionParams",
211
+ "Minimax_M2_7_SubscriptionParams",
212
+ "Minimax_M3Params",
213
+ "Minimax_M3_SubscriptionParams",
214
+ ]
modelparams/types/mistral.py ADDED
@@ -0,0 +1,250 @@
1
+ # AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
2
+ # Do not edit by hand; the YAML catalog under /models is the source of truth.
3
+
4
+ # ruff: noqa: F401
5
+
6
+ from typing import Annotated, Literal
7
+
8
+ from pydantic import ConfigDict, Field
9
+ from typing_extensions import TypedDict
10
+
11
+ _PARAMS_CONFIG = ConfigDict(strict=True, extra="forbid")
12
+
13
+ Codestral_LatestParams = TypedDict(
14
+ "Codestral_LatestParams",
15
+ {
16
+ "max_tokens": Annotated[int, Field(ge=1)],
17
+ "stop": str,
18
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
19
+ "top_p": Annotated[float, Field(ge=0, le=1)],
20
+ "random_seed": Annotated[int, Field(ge=0)],
21
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
22
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
23
+ "response_format.type": Literal["text", "json_object"],
24
+ "safe_prompt": bool,
25
+ },
26
+ total=False,
27
+ )
28
+ setattr(Codestral_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
29
+
30
+ Devstral_2512Params = TypedDict(
31
+ "Devstral_2512Params",
32
+ {
33
+ "max_tokens": Annotated[int, Field(ge=1)],
34
+ "stop": str,
35
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
36
+ "top_p": Annotated[float, Field(ge=0, le=1)],
37
+ "random_seed": Annotated[int, Field(ge=0)],
38
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
39
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
40
+ "response_format.type": Literal["text", "json_object"],
41
+ "safe_prompt": bool,
42
+ },
43
+ total=False,
44
+ )
45
+ setattr(Devstral_2512Params, "__pydantic_config__", _PARAMS_CONFIG)
46
+
47
+ Devstral_LatestParams = TypedDict(
48
+ "Devstral_LatestParams",
49
+ {
50
+ "max_tokens": Annotated[int, Field(ge=1)],
51
+ "stop": str,
52
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
53
+ "top_p": Annotated[float, Field(ge=0, le=1)],
54
+ "random_seed": Annotated[int, Field(ge=0)],
55
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
56
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
57
+ "response_format.type": Literal["text", "json_object"],
58
+ "safe_prompt": bool,
59
+ },
60
+ total=False,
61
+ )
62
+ setattr(Devstral_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
63
+
64
+ Magistral_Medium_LatestParams = TypedDict(
65
+ "Magistral_Medium_LatestParams",
66
+ {
67
+ "max_tokens": Annotated[int, Field(ge=1)],
68
+ "stop": str,
69
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
70
+ "top_p": Annotated[float, Field(ge=0, le=1)],
71
+ "random_seed": Annotated[int, Field(ge=0)],
72
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
73
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
74
+ "prompt_mode": Literal["reasoning"],
75
+ "response_format.type": Literal["text", "json_object"],
76
+ "safe_prompt": bool,
77
+ },
78
+ total=False,
79
+ )
80
+ setattr(Magistral_Medium_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
81
+
82
+ Magistral_Small_LatestParams = TypedDict(
83
+ "Magistral_Small_LatestParams",
84
+ {
85
+ "max_tokens": Annotated[int, Field(ge=1)],
86
+ "stop": str,
87
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
88
+ "top_p": Annotated[float, Field(ge=0, le=1)],
89
+ "random_seed": Annotated[int, Field(ge=0)],
90
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
91
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
92
+ "prompt_mode": Literal["reasoning"],
93
+ "response_format.type": Literal["text", "json_object"],
94
+ "safe_prompt": bool,
95
+ },
96
+ total=False,
97
+ )
98
+ setattr(Magistral_Small_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
99
+
100
+ Ministral_14b_LatestParams = TypedDict(
101
+ "Ministral_14b_LatestParams",
102
+ {
103
+ "max_tokens": Annotated[int, Field(ge=1)],
104
+ "stop": str,
105
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
106
+ "top_p": Annotated[float, Field(ge=0, le=1)],
107
+ "random_seed": Annotated[int, Field(ge=0)],
108
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
109
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
110
+ "response_format.type": Literal["text", "json_object"],
111
+ "safe_prompt": bool,
112
+ },
113
+ total=False,
114
+ )
115
+ setattr(Ministral_14b_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
116
+
117
+ Ministral_3b_LatestParams = TypedDict(
118
+ "Ministral_3b_LatestParams",
119
+ {
120
+ "max_tokens": Annotated[int, Field(ge=1)],
121
+ "stop": str,
122
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
123
+ "top_p": Annotated[float, Field(ge=0, le=1)],
124
+ "random_seed": Annotated[int, Field(ge=0)],
125
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
126
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
127
+ "response_format.type": Literal["text", "json_object"],
128
+ "safe_prompt": bool,
129
+ },
130
+ total=False,
131
+ )
132
+ setattr(Ministral_3b_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
133
+
134
+ Ministral_8b_LatestParams = TypedDict(
135
+ "Ministral_8b_LatestParams",
136
+ {
137
+ "max_tokens": Annotated[int, Field(ge=1)],
138
+ "stop": str,
139
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
140
+ "top_p": Annotated[float, Field(ge=0, le=1)],
141
+ "random_seed": Annotated[int, Field(ge=0)],
142
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
143
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
144
+ "response_format.type": Literal["text", "json_object"],
145
+ "safe_prompt": bool,
146
+ },
147
+ total=False,
148
+ )
149
+ setattr(Ministral_8b_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
150
+
151
+ Mistral_Large_LatestParams = TypedDict(
152
+ "Mistral_Large_LatestParams",
153
+ {
154
+ "max_tokens": Annotated[int, Field(ge=1)],
155
+ "stop": str,
156
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
157
+ "top_p": Annotated[float, Field(ge=0, le=1)],
158
+ "random_seed": Annotated[int, Field(ge=0)],
159
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
160
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
161
+ "response_format.type": Literal["text", "json_object"],
162
+ "safe_prompt": bool,
163
+ },
164
+ total=False,
165
+ )
166
+ setattr(Mistral_Large_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
167
+
168
+ Mistral_Medium_3_5Params = TypedDict(
169
+ "Mistral_Medium_3_5Params",
170
+ {
171
+ "max_tokens": Annotated[int, Field(ge=1)],
172
+ "stop": str,
173
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
174
+ "top_p": Annotated[float, Field(ge=0, le=1)],
175
+ "random_seed": Annotated[int, Field(ge=0)],
176
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
177
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
178
+ "response_format.type": Literal["text", "json_object"],
179
+ "safe_prompt": bool,
180
+ },
181
+ total=False,
182
+ )
183
+ setattr(Mistral_Medium_3_5Params, "__pydantic_config__", _PARAMS_CONFIG)
184
+
185
+ Mistral_Medium_LatestParams = TypedDict(
186
+ "Mistral_Medium_LatestParams",
187
+ {
188
+ "max_tokens": Annotated[int, Field(ge=1)],
189
+ "stop": str,
190
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
191
+ "top_p": Annotated[float, Field(ge=0, le=1)],
192
+ "random_seed": Annotated[int, Field(ge=0)],
193
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
194
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
195
+ "response_format.type": Literal["text", "json_object"],
196
+ "safe_prompt": bool,
197
+ },
198
+ total=False,
199
+ )
200
+ setattr(Mistral_Medium_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
201
+
202
+ Mistral_Small_LatestParams = TypedDict(
203
+ "Mistral_Small_LatestParams",
204
+ {
205
+ "max_tokens": Annotated[int, Field(ge=1)],
206
+ "stop": str,
207
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
208
+ "top_p": Annotated[float, Field(ge=0, le=1)],
209
+ "random_seed": Annotated[int, Field(ge=0)],
210
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
211
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
212
+ "response_format.type": Literal["text", "json_object"],
213
+ "safe_prompt": bool,
214
+ },
215
+ total=False,
216
+ )
217
+ setattr(Mistral_Small_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
218
+
219
+ Open_Mistral_NemoParams = TypedDict(
220
+ "Open_Mistral_NemoParams",
221
+ {
222
+ "max_tokens": Annotated[int, Field(ge=1)],
223
+ "stop": str,
224
+ "temperature": Annotated[float, Field(ge=0, le=1.5)],
225
+ "top_p": Annotated[float, Field(ge=0, le=1)],
226
+ "random_seed": Annotated[int, Field(ge=0)],
227
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
228
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
229
+ "response_format.type": Literal["text", "json_object"],
230
+ "safe_prompt": bool,
231
+ },
232
+ total=False,
233
+ )
234
+ setattr(Open_Mistral_NemoParams, "__pydantic_config__", _PARAMS_CONFIG)
235
+
236
+ __all__ = [
237
+ "Codestral_LatestParams",
238
+ "Devstral_2512Params",
239
+ "Devstral_LatestParams",
240
+ "Magistral_Medium_LatestParams",
241
+ "Magistral_Small_LatestParams",
242
+ "Ministral_14b_LatestParams",
243
+ "Ministral_3b_LatestParams",
244
+ "Ministral_8b_LatestParams",
245
+ "Mistral_Large_LatestParams",
246
+ "Mistral_Medium_3_5Params",
247
+ "Mistral_Medium_LatestParams",
248
+ "Mistral_Small_LatestParams",
249
+ "Open_Mistral_NemoParams",
250
+ ]
modelparams/types/moonshot.py ADDED
@@ -0,0 +1,154 @@
1
+ # AUTO-GENERATED by packages/modelparams-python/scripts/codegen.ts.
2
+ # Do not edit by hand; the YAML catalog under /models is the source of truth.
3
+
4
+ # ruff: noqa: F401
5
+
6
+ from typing import Annotated, Literal
7
+
8
+ from pydantic import ConfigDict, Field
9
+ from typing_extensions import TypedDict
10
+
11
+ _PARAMS_CONFIG = ConfigDict(strict=True, extra="forbid")
12
+
13
+ Kimi_K2_5Params = TypedDict(
14
+ "Kimi_K2_5Params",
15
+ {
16
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
17
+ "thinking.type": Literal["enabled", "disabled"],
18
+ "response_format.type": Literal["text", "json_object"],
19
+ },
20
+ total=False,
21
+ )
22
+ setattr(Kimi_K2_5Params, "__pydantic_config__", _PARAMS_CONFIG)
23
+
24
+ Kimi_K2_6Params = TypedDict(
25
+ "Kimi_K2_6Params",
26
+ {
27
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
28
+ "thinking.type": Literal["enabled", "disabled"],
29
+ "response_format.type": Literal["text", "json_object"],
30
+ },
31
+ total=False,
32
+ )
33
+ setattr(Kimi_K2_6Params, "__pydantic_config__", _PARAMS_CONFIG)
34
+
35
+ Kimi_K2_6_SubscriptionParams = TypedDict(
36
+ "Kimi_K2_6_SubscriptionParams",
37
+ {
38
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
39
+ "thinking.type": Literal["enabled", "disabled"],
40
+ "response_format.type": Literal["text", "json_object"],
41
+ },
42
+ total=False,
43
+ )
44
+ setattr(Kimi_K2_6_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
45
+
46
+ Kimi_K2_7_CodeParams = TypedDict(
47
+ "Kimi_K2_7_CodeParams",
48
+ {
49
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
50
+ "response_format.type": Literal["text", "json_object", "json_schema"],
51
+ },
52
+ total=False,
53
+ )
54
+ setattr(Kimi_K2_7_CodeParams, "__pydantic_config__", _PARAMS_CONFIG)
55
+
56
+ Kimi_K2_7_Code_HighspeedParams = TypedDict(
57
+ "Kimi_K2_7_Code_HighspeedParams",
58
+ {
59
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
60
+ "response_format.type": Literal["text", "json_object", "json_schema"],
61
+ },
62
+ total=False,
63
+ )
64
+ setattr(Kimi_K2_7_Code_HighspeedParams, "__pydantic_config__", _PARAMS_CONFIG)
65
+
66
+ Kimi_K2_7_Code_Highspeed_SubscriptionParams = TypedDict(
67
+ "Kimi_K2_7_Code_Highspeed_SubscriptionParams",
68
+ {
69
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
70
+ "response_format.type": Literal["text", "json_object"],
71
+ },
72
+ total=False,
73
+ )
74
+ setattr(Kimi_K2_7_Code_Highspeed_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
75
+
76
+ Kimi_K2_7_Code_SubscriptionParams = TypedDict(
77
+ "Kimi_K2_7_Code_SubscriptionParams",
78
+ {
79
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
80
+ "response_format.type": Literal["text", "json_object"],
81
+ },
82
+ total=False,
83
+ )
84
+ setattr(Kimi_K2_7_Code_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
85
+
86
+ Kimi_K3Params = TypedDict(
87
+ "Kimi_K3Params",
88
+ {
89
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
90
+ "thinking.type": Literal["enabled", "disabled"],
91
+ "response_format.type": Literal["text", "json_object", "json_schema"],
92
+ },
93
+ total=False,
94
+ )
95
+ setattr(Kimi_K3Params, "__pydantic_config__", _PARAMS_CONFIG)
96
+
97
+ Moonshot_V1_128kParams = TypedDict(
98
+ "Moonshot_V1_128kParams",
99
+ {
100
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
101
+ "temperature": Annotated[float, Field(ge=0, le=1)],
102
+ "top_p": Annotated[float, Field(ge=0, le=1)],
103
+ "n": Annotated[int, Field(ge=1, le=5)],
104
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
105
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
106
+ "response_format.type": Literal["text", "json_object"],
107
+ },
108
+ total=False,
109
+ )
110
+ setattr(Moonshot_V1_128kParams, "__pydantic_config__", _PARAMS_CONFIG)
111
+
112
+ Moonshot_V1_32kParams = TypedDict(
113
+ "Moonshot_V1_32kParams",
114
+ {
115
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
116
+ "temperature": Annotated[float, Field(ge=0, le=1)],
117
+ "top_p": Annotated[float, Field(ge=0, le=1)],
118
+ "n": Annotated[int, Field(ge=1, le=5)],
119
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
120
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
121
+ "response_format.type": Literal["text", "json_object"],
122
+ },
123
+ total=False,
124
+ )
125
+ setattr(Moonshot_V1_32kParams, "__pydantic_config__", _PARAMS_CONFIG)
126
+
127
+ Moonshot_V1_8kParams = TypedDict(
128
+ "Moonshot_V1_8kParams",
129
+ {
130
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
131
+ "temperature": Annotated[float, Field(ge=0, le=1)],
132
+ "top_p": Annotated[float, Field(ge=0, le=1)],
133
+ "n": Annotated[int, Field(ge=1, le=5)],
134
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
135
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
136
+ "response_format.type": Literal["text", "json_object"],
137
+ },
138
+ total=False,
139
+ )
140
+ setattr(Moonshot_V1_8kParams, "__pydantic_config__", _PARAMS_CONFIG)
141
+
142
+ __all__ = [
143
+ "Kimi_K2_5Params",
144
+ "Kimi_K2_6Params",
145
+ "Kimi_K2_6_SubscriptionParams",
146
+ "Kimi_K2_7_CodeParams",
147
+ "Kimi_K2_7_Code_HighspeedParams",
148
+ "Kimi_K2_7_Code_Highspeed_SubscriptionParams",
149
+ "Kimi_K2_7_Code_SubscriptionParams",
150
+ "Kimi_K3Params",
151
+ "Moonshot_V1_128kParams",
152
+ "Moonshot_V1_32kParams",
153
+ "Moonshot_V1_8kParams",
154
+ ]