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/anthropic.py ADDED
@@ -0,0 +1,620 @@
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
+ Claude_3_5_Haiku_20241022Params = TypedDict(
14
+ "Claude_3_5_Haiku_20241022Params",
15
+ {
16
+ "max_tokens": Annotated[int, Field(ge=1)],
17
+ "temperature": Annotated[float, Field(ge=0, le=1)],
18
+ "top_p": Annotated[float, Field(ge=0, le=1)],
19
+ "top_k": Annotated[int, Field(ge=0)],
20
+ },
21
+ total=False,
22
+ )
23
+ setattr(Claude_3_5_Haiku_20241022Params, "__pydantic_config__", _PARAMS_CONFIG)
24
+
25
+ Claude_3_5_Haiku_LatestParams = TypedDict(
26
+ "Claude_3_5_Haiku_LatestParams",
27
+ {
28
+ "max_tokens": Annotated[int, Field(ge=1)],
29
+ "temperature": Annotated[float, Field(ge=0, le=1)],
30
+ "top_p": Annotated[float, Field(ge=0, le=1)],
31
+ "top_k": Annotated[int, Field(ge=0)],
32
+ },
33
+ total=False,
34
+ )
35
+ setattr(Claude_3_5_Haiku_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
36
+
37
+ Claude_3_5_Sonnet_20241022Params = TypedDict(
38
+ "Claude_3_5_Sonnet_20241022Params",
39
+ {
40
+ "max_tokens": Annotated[int, Field(ge=1)],
41
+ "temperature": Annotated[float, Field(ge=0, le=1)],
42
+ "top_p": Annotated[float, Field(ge=0, le=1)],
43
+ "top_k": Annotated[int, Field(ge=0)],
44
+ },
45
+ total=False,
46
+ )
47
+ setattr(Claude_3_5_Sonnet_20241022Params, "__pydantic_config__", _PARAMS_CONFIG)
48
+
49
+ Claude_3_5_Sonnet_LatestParams = TypedDict(
50
+ "Claude_3_5_Sonnet_LatestParams",
51
+ {
52
+ "max_tokens": Annotated[int, Field(ge=1)],
53
+ "temperature": Annotated[float, Field(ge=0, le=1)],
54
+ "top_p": Annotated[float, Field(ge=0, le=1)],
55
+ "top_k": Annotated[int, Field(ge=0)],
56
+ },
57
+ total=False,
58
+ )
59
+ setattr(Claude_3_5_Sonnet_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
60
+
61
+ Claude_3_7_Sonnet_20250219Params = TypedDict(
62
+ "Claude_3_7_Sonnet_20250219Params",
63
+ {
64
+ "max_tokens": Annotated[int, Field(ge=1)],
65
+ "temperature": Annotated[float, Field(ge=0, le=1)],
66
+ "top_p": Annotated[float, Field(ge=0, le=1)],
67
+ "top_k": Annotated[int, Field(ge=0)],
68
+ "thinking.type": Literal["disabled", "enabled"],
69
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
70
+ },
71
+ total=False,
72
+ )
73
+ setattr(Claude_3_7_Sonnet_20250219Params, "__pydantic_config__", _PARAMS_CONFIG)
74
+
75
+ Claude_3_7_Sonnet_LatestParams = TypedDict(
76
+ "Claude_3_7_Sonnet_LatestParams",
77
+ {
78
+ "max_tokens": Annotated[int, Field(ge=1)],
79
+ "temperature": Annotated[float, Field(ge=0, le=1)],
80
+ "top_p": Annotated[float, Field(ge=0, le=1)],
81
+ "top_k": Annotated[int, Field(ge=0)],
82
+ "thinking.type": Literal["disabled", "enabled"],
83
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
84
+ },
85
+ total=False,
86
+ )
87
+ setattr(Claude_3_7_Sonnet_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
88
+
89
+ Claude_3_Opus_20240229Params = TypedDict(
90
+ "Claude_3_Opus_20240229Params",
91
+ {
92
+ "max_tokens": Annotated[int, Field(ge=1)],
93
+ "temperature": Annotated[float, Field(ge=0, le=1)],
94
+ "top_p": Annotated[float, Field(ge=0, le=1)],
95
+ "top_k": Annotated[int, Field(ge=0)],
96
+ },
97
+ total=False,
98
+ )
99
+ setattr(Claude_3_Opus_20240229Params, "__pydantic_config__", _PARAMS_CONFIG)
100
+
101
+ Claude_3_Opus_LatestParams = TypedDict(
102
+ "Claude_3_Opus_LatestParams",
103
+ {
104
+ "max_tokens": Annotated[int, Field(ge=1)],
105
+ "temperature": Annotated[float, Field(ge=0, le=1)],
106
+ "top_p": Annotated[float, Field(ge=0, le=1)],
107
+ "top_k": Annotated[int, Field(ge=0)],
108
+ },
109
+ total=False,
110
+ )
111
+ setattr(Claude_3_Opus_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
112
+
113
+ Claude_Fable_5Params = TypedDict(
114
+ "Claude_Fable_5Params",
115
+ {
116
+ "max_tokens": Annotated[int, Field(ge=1)],
117
+ "thinking.type": Literal["adaptive"],
118
+ "thinking.display": Literal["summarized", "omitted"],
119
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
120
+ },
121
+ total=False,
122
+ )
123
+ setattr(Claude_Fable_5Params, "__pydantic_config__", _PARAMS_CONFIG)
124
+
125
+ Claude_Fable_5_SubscriptionParams = TypedDict(
126
+ "Claude_Fable_5_SubscriptionParams",
127
+ {
128
+ "max_tokens": Annotated[int, Field(ge=1)],
129
+ "thinking.type": Literal["adaptive"],
130
+ "thinking.display": Literal["summarized", "omitted"],
131
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
132
+ },
133
+ total=False,
134
+ )
135
+ setattr(Claude_Fable_5_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
136
+
137
+ Claude_Haiku_4Params = TypedDict(
138
+ "Claude_Haiku_4Params",
139
+ {
140
+ "max_tokens": Annotated[int, Field(ge=1)],
141
+ "temperature": Annotated[float, Field(ge=0, le=1)],
142
+ "top_p": Annotated[float, Field(ge=0, le=1)],
143
+ "top_k": Annotated[int, Field(ge=0)],
144
+ "thinking.type": Literal["disabled", "enabled"],
145
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
146
+ },
147
+ total=False,
148
+ )
149
+ setattr(Claude_Haiku_4Params, "__pydantic_config__", _PARAMS_CONFIG)
150
+
151
+ Claude_Haiku_4_5Params = TypedDict(
152
+ "Claude_Haiku_4_5Params",
153
+ {
154
+ "max_tokens": Annotated[int, Field(ge=1)],
155
+ "temperature": Annotated[float, Field(ge=0, le=1)],
156
+ "top_p": Annotated[float, Field(ge=0, le=1)],
157
+ "top_k": Annotated[int, Field(ge=0)],
158
+ "thinking.type": Literal["disabled", "enabled"],
159
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
160
+ },
161
+ total=False,
162
+ )
163
+ setattr(Claude_Haiku_4_5Params, "__pydantic_config__", _PARAMS_CONFIG)
164
+
165
+ Claude_Haiku_4_5_20251001Params = TypedDict(
166
+ "Claude_Haiku_4_5_20251001Params",
167
+ {
168
+ "max_tokens": Annotated[int, Field(ge=1)],
169
+ "temperature": Annotated[float, Field(ge=0, le=1)],
170
+ "top_p": Annotated[float, Field(ge=0, le=1)],
171
+ "top_k": Annotated[int, Field(ge=0)],
172
+ "thinking.type": Literal["disabled", "enabled"],
173
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
174
+ },
175
+ total=False,
176
+ )
177
+ setattr(Claude_Haiku_4_5_20251001Params, "__pydantic_config__", _PARAMS_CONFIG)
178
+
179
+ Claude_Haiku_4_5_20251001_SubscriptionParams = TypedDict(
180
+ "Claude_Haiku_4_5_20251001_SubscriptionParams",
181
+ {
182
+ "max_tokens": Annotated[int, Field(ge=1)],
183
+ "temperature": Annotated[float, Field(ge=0, le=1)],
184
+ "top_p": Annotated[float, Field(ge=0, le=1)],
185
+ "top_k": Annotated[int, Field(ge=0)],
186
+ "thinking.type": Literal["disabled", "enabled"],
187
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
188
+ },
189
+ total=False,
190
+ )
191
+ setattr(Claude_Haiku_4_5_20251001_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
192
+
193
+ Claude_Haiku_4_5_SubscriptionParams = TypedDict(
194
+ "Claude_Haiku_4_5_SubscriptionParams",
195
+ {
196
+ "max_tokens": Annotated[int, Field(ge=1)],
197
+ "temperature": Annotated[float, Field(ge=0, le=1)],
198
+ "top_p": Annotated[float, Field(ge=0, le=1)],
199
+ "top_k": Annotated[int, Field(ge=0)],
200
+ "thinking.type": Literal["disabled", "enabled"],
201
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
202
+ },
203
+ total=False,
204
+ )
205
+ setattr(Claude_Haiku_4_5_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
206
+
207
+ Claude_Haiku_4_SubscriptionParams = TypedDict(
208
+ "Claude_Haiku_4_SubscriptionParams",
209
+ {
210
+ "max_tokens": Annotated[int, Field(ge=1)],
211
+ "temperature": Annotated[float, Field(ge=0, le=1)],
212
+ "top_p": Annotated[float, Field(ge=0, le=1)],
213
+ "top_k": Annotated[int, Field(ge=0)],
214
+ "thinking.type": Literal["disabled", "enabled"],
215
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
216
+ },
217
+ total=False,
218
+ )
219
+ setattr(Claude_Haiku_4_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
220
+
221
+ Claude_Opus_4_1_20250805Params = TypedDict(
222
+ "Claude_Opus_4_1_20250805Params",
223
+ {
224
+ "max_tokens": Annotated[int, Field(ge=1)],
225
+ "temperature": Annotated[float, Field(ge=0, le=1)],
226
+ "top_p": Annotated[float, Field(ge=0, le=1)],
227
+ "top_k": Annotated[int, Field(ge=0)],
228
+ "thinking.type": Literal["disabled", "enabled"],
229
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
230
+ "thinking.display": Literal["summarized", "omitted"],
231
+ },
232
+ total=False,
233
+ )
234
+ setattr(Claude_Opus_4_1_20250805Params, "__pydantic_config__", _PARAMS_CONFIG)
235
+
236
+ Claude_Opus_4_1_20250805_SubscriptionParams = TypedDict(
237
+ "Claude_Opus_4_1_20250805_SubscriptionParams",
238
+ {
239
+ "max_tokens": Annotated[int, Field(ge=1)],
240
+ "temperature": Annotated[float, Field(ge=0, le=1)],
241
+ "top_p": Annotated[float, Field(ge=0, le=1)],
242
+ "top_k": Annotated[int, Field(ge=0)],
243
+ "thinking.type": Literal["disabled", "enabled"],
244
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
245
+ "thinking.display": Literal["summarized", "omitted"],
246
+ },
247
+ total=False,
248
+ )
249
+ setattr(Claude_Opus_4_1_20250805_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
250
+
251
+ Claude_Opus_4_20250514Params = TypedDict(
252
+ "Claude_Opus_4_20250514Params",
253
+ {
254
+ "max_tokens": Annotated[int, Field(ge=1)],
255
+ "temperature": Annotated[float, Field(ge=0, le=1)],
256
+ "top_p": Annotated[float, Field(ge=0, le=1)],
257
+ "top_k": Annotated[int, Field(ge=0)],
258
+ "thinking.type": Literal["disabled", "enabled"],
259
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
260
+ "thinking.display": Literal["summarized", "omitted"],
261
+ },
262
+ total=False,
263
+ )
264
+ setattr(Claude_Opus_4_20250514Params, "__pydantic_config__", _PARAMS_CONFIG)
265
+
266
+ Claude_Opus_4_20250514_SubscriptionParams = TypedDict(
267
+ "Claude_Opus_4_20250514_SubscriptionParams",
268
+ {
269
+ "max_tokens": Annotated[int, Field(ge=1)],
270
+ "temperature": Annotated[float, Field(ge=0, le=1)],
271
+ "top_p": Annotated[float, Field(ge=0, le=1)],
272
+ "top_k": Annotated[int, Field(ge=0)],
273
+ "thinking.type": Literal["disabled", "enabled"],
274
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
275
+ "thinking.display": Literal["summarized", "omitted"],
276
+ },
277
+ total=False,
278
+ )
279
+ setattr(Claude_Opus_4_20250514_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
280
+
281
+ Claude_Opus_4_5_20251101Params = TypedDict(
282
+ "Claude_Opus_4_5_20251101Params",
283
+ {
284
+ "max_tokens": Annotated[int, Field(ge=1)],
285
+ "temperature": Annotated[float, Field(ge=0, le=1)],
286
+ "top_p": Annotated[float, Field(ge=0, le=1)],
287
+ "top_k": Annotated[int, Field(ge=0)],
288
+ "thinking.type": Literal["disabled", "enabled"],
289
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
290
+ "thinking.display": Literal["summarized", "omitted"],
291
+ "output_config.effort": Literal["low", "medium", "high"],
292
+ },
293
+ total=False,
294
+ )
295
+ setattr(Claude_Opus_4_5_20251101Params, "__pydantic_config__", _PARAMS_CONFIG)
296
+
297
+ Claude_Opus_4_5_20251101_SubscriptionParams = TypedDict(
298
+ "Claude_Opus_4_5_20251101_SubscriptionParams",
299
+ {
300
+ "max_tokens": Annotated[int, Field(ge=1)],
301
+ "temperature": Annotated[float, Field(ge=0, le=1)],
302
+ "top_p": Annotated[float, Field(ge=0, le=1)],
303
+ "top_k": Annotated[int, Field(ge=0)],
304
+ "thinking.type": Literal["disabled", "enabled"],
305
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
306
+ "thinking.display": Literal["summarized", "omitted"],
307
+ "output_config.effort": Literal["low", "medium", "high"],
308
+ },
309
+ total=False,
310
+ )
311
+ setattr(Claude_Opus_4_5_20251101_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
312
+
313
+ Claude_Opus_4_6Params = TypedDict(
314
+ "Claude_Opus_4_6Params",
315
+ {
316
+ "max_tokens": Annotated[int, Field(ge=1)],
317
+ "temperature": Annotated[float, Field(ge=0, le=1)],
318
+ "top_p": Annotated[float, Field(ge=0, le=1)],
319
+ "top_k": Annotated[int, Field(ge=0)],
320
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
321
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
322
+ "thinking.display": Literal["summarized", "omitted"],
323
+ "output_config.effort": Literal["low", "medium", "high", "max"],
324
+ },
325
+ total=False,
326
+ )
327
+ setattr(Claude_Opus_4_6Params, "__pydantic_config__", _PARAMS_CONFIG)
328
+
329
+ Claude_Opus_4_6_SubscriptionParams = TypedDict(
330
+ "Claude_Opus_4_6_SubscriptionParams",
331
+ {
332
+ "max_tokens": Annotated[int, Field(ge=1)],
333
+ "temperature": Annotated[float, Field(ge=0, le=1)],
334
+ "top_p": Annotated[float, Field(ge=0, le=1)],
335
+ "top_k": Annotated[int, Field(ge=0)],
336
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
337
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
338
+ "thinking.display": Literal["summarized", "omitted"],
339
+ "output_config.effort": Literal["low", "medium", "high", "max"],
340
+ },
341
+ total=False,
342
+ )
343
+ setattr(Claude_Opus_4_6_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
344
+
345
+ Claude_Opus_4_7Params = TypedDict(
346
+ "Claude_Opus_4_7Params",
347
+ {
348
+ "max_tokens": Annotated[int, Field(ge=1)],
349
+ "thinking.type": Literal["disabled", "adaptive"],
350
+ "thinking.display": Literal["summarized", "omitted"],
351
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
352
+ },
353
+ total=False,
354
+ )
355
+ setattr(Claude_Opus_4_7Params, "__pydantic_config__", _PARAMS_CONFIG)
356
+
357
+ Claude_Opus_4_7_SubscriptionParams = TypedDict(
358
+ "Claude_Opus_4_7_SubscriptionParams",
359
+ {
360
+ "max_tokens": Annotated[int, Field(ge=1)],
361
+ "thinking.type": Literal["disabled", "adaptive"],
362
+ "thinking.display": Literal["summarized", "omitted"],
363
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
364
+ },
365
+ total=False,
366
+ )
367
+ setattr(Claude_Opus_4_7_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
368
+
369
+ Claude_Opus_4_8Params = TypedDict(
370
+ "Claude_Opus_4_8Params",
371
+ {
372
+ "max_tokens": Annotated[int, Field(ge=1)],
373
+ "thinking.type": Literal["disabled", "adaptive"],
374
+ "thinking.display": Literal["summarized", "omitted"],
375
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
376
+ },
377
+ total=False,
378
+ )
379
+ setattr(Claude_Opus_4_8Params, "__pydantic_config__", _PARAMS_CONFIG)
380
+
381
+ Claude_Opus_4_8_SubscriptionParams = TypedDict(
382
+ "Claude_Opus_4_8_SubscriptionParams",
383
+ {
384
+ "max_tokens": Annotated[int, Field(ge=1)],
385
+ "thinking.type": Literal["disabled", "adaptive"],
386
+ "thinking.display": Literal["summarized", "omitted"],
387
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
388
+ },
389
+ total=False,
390
+ )
391
+ setattr(Claude_Opus_4_8_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
392
+
393
+ Claude_Opus_4_SubscriptionParams = TypedDict(
394
+ "Claude_Opus_4_SubscriptionParams",
395
+ {
396
+ "max_tokens": Annotated[int, Field(ge=1)],
397
+ "temperature": Annotated[float, Field(ge=0, le=1)],
398
+ "top_p": Annotated[float, Field(ge=0, le=1)],
399
+ "top_k": Annotated[int, Field(ge=0)],
400
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
401
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
402
+ },
403
+ total=False,
404
+ )
405
+ setattr(Claude_Opus_4_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
406
+
407
+ Claude_Opus_5Params = TypedDict(
408
+ "Claude_Opus_5Params",
409
+ {
410
+ "max_tokens": Annotated[int, Field(ge=1)],
411
+ "thinking.type": Literal["disabled", "adaptive"],
412
+ "thinking.display": Literal["summarized", "omitted"],
413
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
414
+ },
415
+ total=False,
416
+ )
417
+ setattr(Claude_Opus_5Params, "__pydantic_config__", _PARAMS_CONFIG)
418
+
419
+ Claude_Opus_5_SubscriptionParams = TypedDict(
420
+ "Claude_Opus_5_SubscriptionParams",
421
+ {
422
+ "max_tokens": Annotated[int, Field(ge=1)],
423
+ "thinking.type": Literal["disabled", "adaptive"],
424
+ "thinking.display": Literal["summarized", "omitted"],
425
+ "output_config.effort": Literal["low", "medium", "high", "xhigh", "max"],
426
+ },
427
+ total=False,
428
+ )
429
+ setattr(Claude_Opus_5_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
430
+
431
+ Claude_Sonnet_4_20250514Params = TypedDict(
432
+ "Claude_Sonnet_4_20250514Params",
433
+ {
434
+ "max_tokens": Annotated[int, Field(ge=1)],
435
+ "temperature": Annotated[float, Field(ge=0, le=1)],
436
+ "top_p": Annotated[float, Field(ge=0, le=1)],
437
+ "top_k": Annotated[int, Field(ge=0)],
438
+ "thinking.type": Literal["disabled", "enabled"],
439
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
440
+ "thinking.display": Literal["summarized", "omitted"],
441
+ },
442
+ total=False,
443
+ )
444
+ setattr(Claude_Sonnet_4_20250514Params, "__pydantic_config__", _PARAMS_CONFIG)
445
+
446
+ Claude_Sonnet_4_20250514_SubscriptionParams = TypedDict(
447
+ "Claude_Sonnet_4_20250514_SubscriptionParams",
448
+ {
449
+ "max_tokens": Annotated[int, Field(ge=1)],
450
+ "temperature": Annotated[float, Field(ge=0, le=1)],
451
+ "top_p": Annotated[float, Field(ge=0, le=1)],
452
+ "top_k": Annotated[int, Field(ge=0)],
453
+ "thinking.type": Literal["disabled", "enabled"],
454
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
455
+ "thinking.display": Literal["summarized", "omitted"],
456
+ },
457
+ total=False,
458
+ )
459
+ setattr(Claude_Sonnet_4_20250514_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
460
+
461
+ Claude_Sonnet_4_5Params = TypedDict(
462
+ "Claude_Sonnet_4_5Params",
463
+ {
464
+ "max_tokens": Annotated[int, Field(ge=1)],
465
+ "temperature": Annotated[float, Field(ge=0, le=1)],
466
+ "top_p": Annotated[float, Field(ge=0, le=1)],
467
+ "top_k": Annotated[int, Field(ge=0)],
468
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
469
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
470
+ },
471
+ total=False,
472
+ )
473
+ setattr(Claude_Sonnet_4_5Params, "__pydantic_config__", _PARAMS_CONFIG)
474
+
475
+ Claude_Sonnet_4_5_20250929Params = TypedDict(
476
+ "Claude_Sonnet_4_5_20250929Params",
477
+ {
478
+ "max_tokens": Annotated[int, Field(ge=1)],
479
+ "temperature": Annotated[float, Field(ge=0, le=1)],
480
+ "top_p": Annotated[float, Field(ge=0, le=1)],
481
+ "top_k": Annotated[int, Field(ge=0)],
482
+ "thinking.type": Literal["disabled", "enabled"],
483
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
484
+ },
485
+ total=False,
486
+ )
487
+ setattr(Claude_Sonnet_4_5_20250929Params, "__pydantic_config__", _PARAMS_CONFIG)
488
+
489
+ Claude_Sonnet_4_5_20250929_SubscriptionParams = TypedDict(
490
+ "Claude_Sonnet_4_5_20250929_SubscriptionParams",
491
+ {
492
+ "max_tokens": Annotated[int, Field(ge=1)],
493
+ "temperature": Annotated[float, Field(ge=0, le=1)],
494
+ "top_p": Annotated[float, Field(ge=0, le=1)],
495
+ "top_k": Annotated[int, Field(ge=0)],
496
+ "thinking.type": Literal["disabled", "enabled"],
497
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
498
+ },
499
+ total=False,
500
+ )
501
+ setattr(Claude_Sonnet_4_5_20250929_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
502
+
503
+ Claude_Sonnet_4_5_SubscriptionParams = TypedDict(
504
+ "Claude_Sonnet_4_5_SubscriptionParams",
505
+ {
506
+ "max_tokens": Annotated[int, Field(ge=1)],
507
+ "temperature": Annotated[float, Field(ge=0, le=1)],
508
+ "top_p": Annotated[float, Field(ge=0, le=1)],
509
+ "top_k": Annotated[int, Field(ge=0)],
510
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
511
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
512
+ },
513
+ total=False,
514
+ )
515
+ setattr(Claude_Sonnet_4_5_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
516
+
517
+ Claude_Sonnet_4_6Params = TypedDict(
518
+ "Claude_Sonnet_4_6Params",
519
+ {
520
+ "max_tokens": Annotated[int, Field(ge=1)],
521
+ "temperature": Annotated[float, Field(ge=0, le=1)],
522
+ "top_p": Annotated[float, Field(ge=0, le=1)],
523
+ "top_k": Annotated[int, Field(ge=0)],
524
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
525
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
526
+ "thinking.display": Literal["summarized", "omitted"],
527
+ "output_config.effort": Literal["low", "medium", "high", "max"],
528
+ },
529
+ total=False,
530
+ )
531
+ setattr(Claude_Sonnet_4_6Params, "__pydantic_config__", _PARAMS_CONFIG)
532
+
533
+ Claude_Sonnet_4_6_SubscriptionParams = TypedDict(
534
+ "Claude_Sonnet_4_6_SubscriptionParams",
535
+ {
536
+ "max_tokens": Annotated[int, Field(ge=1)],
537
+ "temperature": Annotated[float, Field(ge=0, le=1)],
538
+ "top_p": Annotated[float, Field(ge=0, le=1)],
539
+ "top_k": Annotated[int, Field(ge=0)],
540
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
541
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
542
+ "thinking.display": Literal["summarized", "omitted"],
543
+ "output_config.effort": Literal["low", "medium", "high", "max"],
544
+ },
545
+ total=False,
546
+ )
547
+ setattr(Claude_Sonnet_4_6_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
548
+
549
+ Claude_Sonnet_4_SubscriptionParams = TypedDict(
550
+ "Claude_Sonnet_4_SubscriptionParams",
551
+ {
552
+ "max_tokens": Annotated[int, Field(ge=1)],
553
+ "temperature": Annotated[float, Field(ge=0, le=1)],
554
+ "top_p": Annotated[float, Field(ge=0, le=1)],
555
+ "top_k": Annotated[int, Field(ge=0)],
556
+ "thinking.type": Literal["disabled", "adaptive", "enabled"],
557
+ "thinking.budget_tokens": Annotated[int, Field(ge=1024)],
558
+ },
559
+ total=False,
560
+ )
561
+ setattr(Claude_Sonnet_4_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
562
+
563
+ Claude_Sonnet_5Params = TypedDict(
564
+ "Claude_Sonnet_5Params",
565
+ {
566
+ "max_tokens": Annotated[int, Field(ge=1)],
567
+ "temperature": Annotated[float, Field(ge=0, le=1)],
568
+ "top_p": Annotated[float, Field(ge=0, le=1)],
569
+ "top_k": Annotated[int, Field(ge=0)],
570
+ "thinking.type": Literal["disabled", "adaptive"],
571
+ "thinking.display": Literal["summarized", "omitted"],
572
+ "output_config.effort": Literal["low", "medium", "high", "max"],
573
+ },
574
+ total=False,
575
+ )
576
+ setattr(Claude_Sonnet_5Params, "__pydantic_config__", _PARAMS_CONFIG)
577
+
578
+ __all__ = [
579
+ "Claude_3_5_Haiku_20241022Params",
580
+ "Claude_3_5_Haiku_LatestParams",
581
+ "Claude_3_5_Sonnet_20241022Params",
582
+ "Claude_3_5_Sonnet_LatestParams",
583
+ "Claude_3_7_Sonnet_20250219Params",
584
+ "Claude_3_7_Sonnet_LatestParams",
585
+ "Claude_3_Opus_20240229Params",
586
+ "Claude_3_Opus_LatestParams",
587
+ "Claude_Fable_5Params",
588
+ "Claude_Fable_5_SubscriptionParams",
589
+ "Claude_Haiku_4Params",
590
+ "Claude_Haiku_4_5Params",
591
+ "Claude_Haiku_4_5_20251001Params",
592
+ "Claude_Haiku_4_5_20251001_SubscriptionParams",
593
+ "Claude_Haiku_4_5_SubscriptionParams",
594
+ "Claude_Haiku_4_SubscriptionParams",
595
+ "Claude_Opus_4_1_20250805Params",
596
+ "Claude_Opus_4_1_20250805_SubscriptionParams",
597
+ "Claude_Opus_4_20250514Params",
598
+ "Claude_Opus_4_20250514_SubscriptionParams",
599
+ "Claude_Opus_4_5_20251101Params",
600
+ "Claude_Opus_4_5_20251101_SubscriptionParams",
601
+ "Claude_Opus_4_6Params",
602
+ "Claude_Opus_4_6_SubscriptionParams",
603
+ "Claude_Opus_4_7Params",
604
+ "Claude_Opus_4_7_SubscriptionParams",
605
+ "Claude_Opus_4_8Params",
606
+ "Claude_Opus_4_8_SubscriptionParams",
607
+ "Claude_Opus_4_SubscriptionParams",
608
+ "Claude_Opus_5Params",
609
+ "Claude_Opus_5_SubscriptionParams",
610
+ "Claude_Sonnet_4_20250514Params",
611
+ "Claude_Sonnet_4_20250514_SubscriptionParams",
612
+ "Claude_Sonnet_4_5Params",
613
+ "Claude_Sonnet_4_5_20250929Params",
614
+ "Claude_Sonnet_4_5_20250929_SubscriptionParams",
615
+ "Claude_Sonnet_4_5_SubscriptionParams",
616
+ "Claude_Sonnet_4_6Params",
617
+ "Claude_Sonnet_4_6_SubscriptionParams",
618
+ "Claude_Sonnet_4_SubscriptionParams",
619
+ "Claude_Sonnet_5Params",
620
+ ]
modelparams/types/cerebras.py ADDED
@@ -0,0 +1,33 @@
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
+ Zai_Glm_4_7Params = TypedDict(
14
+ "Zai_Glm_4_7Params",
15
+ {
16
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
17
+ "temperature": Annotated[float, Field(ge=0, le=2)],
18
+ "top_p": Annotated[float, Field(ge=0, le=1)],
19
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
20
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
21
+ "seed": int,
22
+ "stop": str,
23
+ "reasoning_effort": Literal["none", "low", "medium", "high"],
24
+ "clear_thinking": bool,
25
+ "response_format.type": Literal["text", "json_object"],
26
+ },
27
+ total=False,
28
+ )
29
+ setattr(Zai_Glm_4_7Params, "__pydantic_config__", _PARAMS_CONFIG)
30
+
31
+ __all__ = [
32
+ "Zai_Glm_4_7Params",
33
+ ]