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/nvidia.py ADDED
@@ -0,0 +1,244 @@
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
+ Gliner_PiiParams = TypedDict(
14
+ "Gliner_PiiParams",
15
+ {
16
+ "threshold": Annotated[float, Field(ge=0, le=1)],
17
+ "chunk_length": Annotated[int, Field(ge=1, le=2048)],
18
+ "overlap": Annotated[int, Field(ge=0, le=512)],
19
+ "flat_ner": bool,
20
+ },
21
+ total=False,
22
+ )
23
+ setattr(Gliner_PiiParams, "__pydantic_config__", _PARAMS_CONFIG)
24
+
25
+ Llama_3_1_Nemoguard_8b_Topic_ControlParams = TypedDict(
26
+ "Llama_3_1_Nemoguard_8b_Topic_ControlParams",
27
+ {
28
+ "temperature": Annotated[float, Field(ge=0, le=2)],
29
+ "top_p": Annotated[float, Field(le=1)],
30
+ "max_tokens": Annotated[int, Field(ge=1)],
31
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
32
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
33
+ "stop": str,
34
+ },
35
+ total=False,
36
+ )
37
+ setattr(Llama_3_1_Nemoguard_8b_Topic_ControlParams, "__pydantic_config__", _PARAMS_CONFIG)
38
+
39
+ Llama_3_1_Nemotron_Nano_8b_V1Params = TypedDict(
40
+ "Llama_3_1_Nemotron_Nano_8b_V1Params",
41
+ {
42
+ "temperature": Annotated[float, Field(ge=0, le=1)],
43
+ "top_p": Annotated[float, Field(le=1)],
44
+ "max_tokens": Annotated[int, Field(ge=1, le=16384)],
45
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
46
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
47
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
48
+ "stop": str,
49
+ },
50
+ total=False,
51
+ )
52
+ setattr(Llama_3_1_Nemotron_Nano_8b_V1Params, "__pydantic_config__", _PARAMS_CONFIG)
53
+
54
+ Llama_3_1_Nemotron_Safety_Guard_8b_V3Params = TypedDict(
55
+ "Llama_3_1_Nemotron_Safety_Guard_8b_V3Params",
56
+ {
57
+ "temperature": Annotated[float, Field(ge=0, le=1)],
58
+ },
59
+ total=False,
60
+ )
61
+ setattr(Llama_3_1_Nemotron_Safety_Guard_8b_V3Params, "__pydantic_config__", _PARAMS_CONFIG)
62
+
63
+ Llama_3_1_Nemotron_Ultra_253b_V1Params = TypedDict(
64
+ "Llama_3_1_Nemotron_Ultra_253b_V1Params",
65
+ {
66
+ "temperature": Annotated[float, Field(ge=0, le=1)],
67
+ "top_p": Annotated[float, Field(le=1)],
68
+ "max_tokens": Annotated[int, Field(ge=1, le=16384)],
69
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
70
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
71
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
72
+ "stop": str,
73
+ },
74
+ total=False,
75
+ )
76
+ setattr(Llama_3_1_Nemotron_Ultra_253b_V1Params, "__pydantic_config__", _PARAMS_CONFIG)
77
+
78
+ Llama_3_3_Nemotron_Super_49b_V1Params = TypedDict(
79
+ "Llama_3_3_Nemotron_Super_49b_V1Params",
80
+ {
81
+ "temperature": Annotated[float, Field(ge=0, le=1)],
82
+ "top_p": Annotated[float, Field(le=1)],
83
+ "max_tokens": Annotated[int, Field(ge=1, le=16384)],
84
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
85
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
86
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
87
+ "stop": str,
88
+ },
89
+ total=False,
90
+ )
91
+ setattr(Llama_3_3_Nemotron_Super_49b_V1Params, "__pydantic_config__", _PARAMS_CONFIG)
92
+
93
+ Llama_3_3_Nemotron_Super_49b_V1_5Params = TypedDict(
94
+ "Llama_3_3_Nemotron_Super_49b_V1_5Params",
95
+ {
96
+ "temperature": Annotated[float, Field(ge=0, le=1)],
97
+ "top_p": Annotated[float, Field(le=1)],
98
+ "max_tokens": Annotated[int, Field(ge=1, le=65536)],
99
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
100
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
101
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
102
+ "stop": str,
103
+ },
104
+ total=False,
105
+ )
106
+ setattr(Llama_3_3_Nemotron_Super_49b_V1_5Params, "__pydantic_config__", _PARAMS_CONFIG)
107
+
108
+ Nemoguard_Jailbreak_DetectParams = TypedDict(
109
+ "Nemoguard_Jailbreak_DetectParams",
110
+ {
111
+
112
+ },
113
+ total=False,
114
+ )
115
+ setattr(Nemoguard_Jailbreak_DetectParams, "__pydantic_config__", _PARAMS_CONFIG)
116
+
117
+ Nemotron_3_Nano_30b_A3bParams = TypedDict(
118
+ "Nemotron_3_Nano_30b_A3bParams",
119
+ {
120
+ "temperature": Annotated[float, Field(le=1)],
121
+ "top_p": Annotated[float, Field(le=1)],
122
+ "max_tokens": Annotated[int, Field(ge=1, le=32768)],
123
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
124
+ "stop": str,
125
+ },
126
+ total=False,
127
+ )
128
+ setattr(Nemotron_3_Nano_30b_A3bParams, "__pydantic_config__", _PARAMS_CONFIG)
129
+
130
+ Nemotron_3_Super_120b_A12bParams = TypedDict(
131
+ "Nemotron_3_Super_120b_A12bParams",
132
+ {
133
+ "temperature": Annotated[float, Field(le=1)],
134
+ "top_p": Annotated[float, Field(le=1)],
135
+ "max_tokens": Annotated[int, Field(ge=1, le=32768)],
136
+ "reasoning_effort": Literal["none", "low", "high"],
137
+ "reasoning_budget": Annotated[int, Field(ge=-1, le=32768)],
138
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
139
+ "stop": str,
140
+ },
141
+ total=False,
142
+ )
143
+ setattr(Nemotron_3_Super_120b_A12bParams, "__pydantic_config__", _PARAMS_CONFIG)
144
+
145
+ Nemotron_3_Ultra_550b_A55bParams = TypedDict(
146
+ "Nemotron_3_Ultra_550b_A55bParams",
147
+ {
148
+ "temperature": Annotated[float, Field(le=1)],
149
+ "top_p": Annotated[float, Field(le=1)],
150
+ "max_tokens": Annotated[int, Field(ge=1, le=32768)],
151
+ "reasoning_effort": Literal["none", "medium", "high"],
152
+ "reasoning_budget": Annotated[int, Field(ge=-1, le=32768)],
153
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
154
+ "stop": str,
155
+ },
156
+ total=False,
157
+ )
158
+ setattr(Nemotron_3_Ultra_550b_A55bParams, "__pydantic_config__", _PARAMS_CONFIG)
159
+
160
+ Nemotron_3_Ultra_SubscriptionParams = TypedDict(
161
+ "Nemotron_3_Ultra_SubscriptionParams",
162
+ {
163
+ "temperature": Annotated[float, Field(le=1)],
164
+ "top_p": Annotated[float, Field(le=1)],
165
+ "max_tokens": Annotated[int, Field(ge=1, le=32768)],
166
+ "reasoning_effort": Literal["none", "medium", "high"],
167
+ "reasoning_budget": Annotated[int, Field(ge=-1, le=32768)],
168
+ "stop": str,
169
+ },
170
+ total=False,
171
+ )
172
+ setattr(Nemotron_3_Ultra_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
173
+
174
+ Nemotron_Content_Safety_Reasoning_4bParams = TypedDict(
175
+ "Nemotron_Content_Safety_Reasoning_4bParams",
176
+ {
177
+ "temperature": Annotated[float, Field(le=1)],
178
+ "top_p": Annotated[float, Field(le=1)],
179
+ "max_tokens": Annotated[int, Field(ge=1, le=32768)],
180
+ "seed": Annotated[int, Field(ge=0, le=18446744073709552000)],
181
+ "stop": str,
182
+ },
183
+ total=False,
184
+ )
185
+ setattr(Nemotron_Content_Safety_Reasoning_4bParams, "__pydantic_config__", _PARAMS_CONFIG)
186
+
187
+ Nemotron_Mini_4b_InstructParams = TypedDict(
188
+ "Nemotron_Mini_4b_InstructParams",
189
+ {
190
+ "temperature": Annotated[float, Field(ge=0, le=1)],
191
+ "top_p": Annotated[float, Field(le=1)],
192
+ "max_tokens": Annotated[int, Field(ge=1, le=4096)],
193
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
194
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
195
+ "stop": str,
196
+ },
197
+ total=False,
198
+ )
199
+ setattr(Nemotron_Mini_4b_InstructParams, "__pydantic_config__", _PARAMS_CONFIG)
200
+
201
+ Riva_Translate_4b_Instruct_V1_1Params = TypedDict(
202
+ "Riva_Translate_4b_Instruct_V1_1Params",
203
+ {
204
+ "temperature": Annotated[float, Field(ge=0, le=1)],
205
+ "top_p": Annotated[float, Field(le=1)],
206
+ "max_tokens": Annotated[int, Field(ge=1, le=4096)],
207
+ "frequency_penalty": Annotated[float, Field(ge=-2, le=2)],
208
+ "presence_penalty": Annotated[float, Field(ge=-2, le=2)],
209
+ "stop": str,
210
+ },
211
+ total=False,
212
+ )
213
+ setattr(Riva_Translate_4b_Instruct_V1_1Params, "__pydantic_config__", _PARAMS_CONFIG)
214
+
215
+ Usdcode_Llama_3_1_70b_InstructParams = TypedDict(
216
+ "Usdcode_Llama_3_1_70b_InstructParams",
217
+ {
218
+ "temperature": Annotated[float, Field(ge=0, le=1)],
219
+ "top_p": Annotated[float, Field(le=1)],
220
+ "max_tokens": Annotated[int, Field(ge=1, le=2048)],
221
+ "expert_type": Literal["auto", "code", "knowledge", "helperfunction"],
222
+ },
223
+ total=False,
224
+ )
225
+ setattr(Usdcode_Llama_3_1_70b_InstructParams, "__pydantic_config__", _PARAMS_CONFIG)
226
+
227
+ __all__ = [
228
+ "Gliner_PiiParams",
229
+ "Llama_3_1_Nemoguard_8b_Topic_ControlParams",
230
+ "Llama_3_1_Nemotron_Nano_8b_V1Params",
231
+ "Llama_3_1_Nemotron_Safety_Guard_8b_V3Params",
232
+ "Llama_3_1_Nemotron_Ultra_253b_V1Params",
233
+ "Llama_3_3_Nemotron_Super_49b_V1Params",
234
+ "Llama_3_3_Nemotron_Super_49b_V1_5Params",
235
+ "Nemoguard_Jailbreak_DetectParams",
236
+ "Nemotron_3_Nano_30b_A3bParams",
237
+ "Nemotron_3_Super_120b_A12bParams",
238
+ "Nemotron_3_Ultra_550b_A55bParams",
239
+ "Nemotron_3_Ultra_SubscriptionParams",
240
+ "Nemotron_Content_Safety_Reasoning_4bParams",
241
+ "Nemotron_Mini_4b_InstructParams",
242
+ "Riva_Translate_4b_Instruct_V1_1Params",
243
+ "Usdcode_Llama_3_1_70b_InstructParams",
244
+ ]