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/cohere.py ADDED
@@ -0,0 +1,182 @@
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
+ Command_A_03_2025Params = TypedDict(
14
+ "Command_A_03_2025Params",
15
+ {
16
+ "max_tokens": Annotated[int, Field(ge=1)],
17
+ "stop_sequences": str,
18
+ "temperature": Annotated[float, Field(ge=0)],
19
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
20
+ "k": Annotated[int, Field(ge=0, le=500)],
21
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
22
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
23
+ "seed": int,
24
+ "response_format.type": Literal["text", "json_object"],
25
+ "logprobs": bool,
26
+ "tool_choice": Literal["REQUIRED", "NONE"],
27
+ "safety_mode": Literal["CONTEXTUAL", "STRICT"],
28
+ },
29
+ total=False,
30
+ )
31
+ setattr(Command_A_03_2025Params, "__pydantic_config__", _PARAMS_CONFIG)
32
+
33
+ Command_A_Plus_05_2026Params = TypedDict(
34
+ "Command_A_Plus_05_2026Params",
35
+ {
36
+ "max_tokens": Annotated[int, Field(ge=1)],
37
+ "stop_sequences": str,
38
+ "temperature": Annotated[float, Field(ge=0)],
39
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
40
+ "k": Annotated[int, Field(ge=0, le=500)],
41
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
42
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
43
+ "seed": int,
44
+ "response_format.type": Literal["text", "json_object"],
45
+ "logprobs": bool,
46
+ "tool_choice": Literal["REQUIRED", "NONE"],
47
+ "safety_mode": Literal["CONTEXTUAL", "STRICT"],
48
+ },
49
+ total=False,
50
+ )
51
+ setattr(Command_A_Plus_05_2026Params, "__pydantic_config__", _PARAMS_CONFIG)
52
+
53
+ Command_A_Reasoning_08_2025Params = TypedDict(
54
+ "Command_A_Reasoning_08_2025Params",
55
+ {
56
+ "max_tokens": Annotated[int, Field(ge=1)],
57
+ "stop_sequences": str,
58
+ "temperature": Annotated[float, Field(ge=0)],
59
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
60
+ "k": Annotated[int, Field(ge=0, le=500)],
61
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
62
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
63
+ "seed": int,
64
+ "thinking.type": Literal["enabled", "disabled"],
65
+ "thinking.token_budget": Annotated[int, Field(ge=1)],
66
+ "response_format.type": Literal["text", "json_object"],
67
+ "logprobs": bool,
68
+ "tool_choice": Literal["REQUIRED", "NONE"],
69
+ "safety_mode": Literal["CONTEXTUAL", "STRICT"],
70
+ },
71
+ total=False,
72
+ )
73
+ setattr(Command_A_Reasoning_08_2025Params, "__pydantic_config__", _PARAMS_CONFIG)
74
+
75
+ Command_A_Translate_08_2025Params = TypedDict(
76
+ "Command_A_Translate_08_2025Params",
77
+ {
78
+ "max_tokens": Annotated[int, Field(ge=1)],
79
+ "stop_sequences": str,
80
+ "temperature": Annotated[float, Field(ge=0)],
81
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
82
+ "k": Annotated[int, Field(ge=0, le=500)],
83
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
84
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
85
+ "seed": int,
86
+ "response_format.type": Literal["text", "json_object"],
87
+ "logprobs": bool,
88
+ "tool_choice": Literal["REQUIRED", "NONE"],
89
+ "safety_mode": Literal["CONTEXTUAL", "STRICT"],
90
+ },
91
+ total=False,
92
+ )
93
+ setattr(Command_A_Translate_08_2025Params, "__pydantic_config__", _PARAMS_CONFIG)
94
+
95
+ Command_A_Vision_07_2025Params = TypedDict(
96
+ "Command_A_Vision_07_2025Params",
97
+ {
98
+ "max_tokens": Annotated[int, Field(ge=1)],
99
+ "stop_sequences": str,
100
+ "temperature": Annotated[float, Field(ge=0)],
101
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
102
+ "k": Annotated[int, Field(ge=0, le=500)],
103
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
104
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
105
+ "seed": int,
106
+ "response_format.type": Literal["text", "json_object"],
107
+ "logprobs": bool,
108
+ "tool_choice": Literal["REQUIRED", "NONE"],
109
+ "safety_mode": Literal["CONTEXTUAL", "STRICT"],
110
+ },
111
+ total=False,
112
+ )
113
+ setattr(Command_A_Vision_07_2025Params, "__pydantic_config__", _PARAMS_CONFIG)
114
+
115
+ Command_R_08_2024Params = TypedDict(
116
+ "Command_R_08_2024Params",
117
+ {
118
+ "max_tokens": Annotated[int, Field(ge=1)],
119
+ "stop_sequences": str,
120
+ "temperature": Annotated[float, Field(ge=0)],
121
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
122
+ "k": Annotated[int, Field(ge=0, le=500)],
123
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
124
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
125
+ "seed": int,
126
+ "response_format.type": Literal["text", "json_object"],
127
+ "logprobs": bool,
128
+ "safety_mode": Literal["CONTEXTUAL", "STRICT", "OFF"],
129
+ },
130
+ total=False,
131
+ )
132
+ setattr(Command_R_08_2024Params, "__pydantic_config__", _PARAMS_CONFIG)
133
+
134
+ Command_R_Plus_08_2024Params = TypedDict(
135
+ "Command_R_Plus_08_2024Params",
136
+ {
137
+ "max_tokens": Annotated[int, Field(ge=1)],
138
+ "stop_sequences": str,
139
+ "temperature": Annotated[float, Field(ge=0)],
140
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
141
+ "k": Annotated[int, Field(ge=0, le=500)],
142
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
143
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
144
+ "seed": int,
145
+ "response_format.type": Literal["text", "json_object"],
146
+ "logprobs": bool,
147
+ "safety_mode": Literal["CONTEXTUAL", "STRICT", "OFF"],
148
+ },
149
+ total=False,
150
+ )
151
+ setattr(Command_R_Plus_08_2024Params, "__pydantic_config__", _PARAMS_CONFIG)
152
+
153
+ Command_R7b_12_2024Params = TypedDict(
154
+ "Command_R7b_12_2024Params",
155
+ {
156
+ "max_tokens": Annotated[int, Field(ge=1)],
157
+ "stop_sequences": str,
158
+ "temperature": Annotated[float, Field(ge=0)],
159
+ "p": Annotated[float, Field(ge=0.01, le=0.99)],
160
+ "k": Annotated[int, Field(ge=0, le=500)],
161
+ "frequency_penalty": Annotated[float, Field(ge=0, le=1)],
162
+ "presence_penalty": Annotated[float, Field(ge=0, le=1)],
163
+ "seed": int,
164
+ "response_format.type": Literal["text", "json_object"],
165
+ "logprobs": bool,
166
+ "tool_choice": Literal["REQUIRED", "NONE"],
167
+ "safety_mode": Literal["CONTEXTUAL", "STRICT"],
168
+ },
169
+ total=False,
170
+ )
171
+ setattr(Command_R7b_12_2024Params, "__pydantic_config__", _PARAMS_CONFIG)
172
+
173
+ __all__ = [
174
+ "Command_A_03_2025Params",
175
+ "Command_A_Plus_05_2026Params",
176
+ "Command_A_Reasoning_08_2025Params",
177
+ "Command_A_Translate_08_2025Params",
178
+ "Command_A_Vision_07_2025Params",
179
+ "Command_R_08_2024Params",
180
+ "Command_R_Plus_08_2024Params",
181
+ "Command_R7b_12_2024Params",
182
+ ]
modelparams/types/deepseek.py ADDED
@@ -0,0 +1,69 @@
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
+ Deepseek_ChatParams = TypedDict(
14
+ "Deepseek_ChatParams",
15
+ {
16
+ "max_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
+ "thinking.type": Literal["disabled", "enabled"],
20
+ },
21
+ total=False,
22
+ )
23
+ setattr(Deepseek_ChatParams, "__pydantic_config__", _PARAMS_CONFIG)
24
+
25
+ Deepseek_ReasonerParams = TypedDict(
26
+ "Deepseek_ReasonerParams",
27
+ {
28
+ "max_tokens": Annotated[int, Field(ge=1)],
29
+ "temperature": Annotated[float, Field(ge=0, le=2)],
30
+ "top_p": Annotated[float, Field(ge=0, le=1)],
31
+ "thinking.type": Literal["enabled", "disabled"],
32
+ "reasoning_effort": Literal["high", "max"],
33
+ },
34
+ total=False,
35
+ )
36
+ setattr(Deepseek_ReasonerParams, "__pydantic_config__", _PARAMS_CONFIG)
37
+
38
+ Deepseek_V4_FlashParams = TypedDict(
39
+ "Deepseek_V4_FlashParams",
40
+ {
41
+ "max_tokens": Annotated[int, Field(ge=1)],
42
+ "temperature": Annotated[float, Field(ge=0, le=2)],
43
+ "top_p": Annotated[float, Field(ge=0, le=1)],
44
+ "thinking.type": Literal["enabled", "disabled"],
45
+ "reasoning_effort": Literal["high", "max"],
46
+ },
47
+ total=False,
48
+ )
49
+ setattr(Deepseek_V4_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
50
+
51
+ Deepseek_V4_ProParams = TypedDict(
52
+ "Deepseek_V4_ProParams",
53
+ {
54
+ "max_tokens": Annotated[int, Field(ge=1)],
55
+ "temperature": Annotated[float, Field(ge=0, le=2)],
56
+ "top_p": Annotated[float, Field(ge=0, le=1)],
57
+ "thinking.type": Literal["enabled", "disabled"],
58
+ "reasoning_effort": Literal["high", "max"],
59
+ },
60
+ total=False,
61
+ )
62
+ setattr(Deepseek_V4_ProParams, "__pydantic_config__", _PARAMS_CONFIG)
63
+
64
+ __all__ = [
65
+ "Deepseek_ChatParams",
66
+ "Deepseek_ReasonerParams",
67
+ "Deepseek_V4_FlashParams",
68
+ "Deepseek_V4_ProParams",
69
+ ]
modelparams/types/google.py ADDED
@@ -0,0 +1,388 @@
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
+ Gemini_2_5_FlashParams = TypedDict(
14
+ "Gemini_2_5_FlashParams",
15
+ {
16
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
17
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
18
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
19
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
20
+ "generationConfig.seed": int,
21
+ "generationConfig.thinkingConfig.thinkingBudget": Annotated[int, Field(ge=-1, le=24576)],
22
+ "generationConfig.thinkingConfig.includeThoughts": bool,
23
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
24
+ },
25
+ total=False,
26
+ )
27
+ setattr(Gemini_2_5_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
28
+
29
+ Gemini_2_5_Flash_LiteParams = TypedDict(
30
+ "Gemini_2_5_Flash_LiteParams",
31
+ {
32
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
33
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
34
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
35
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
36
+ "generationConfig.seed": int,
37
+ "generationConfig.thinkingConfig.thinkingBudget": int,
38
+ "generationConfig.thinkingConfig.includeThoughts": bool,
39
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
40
+ },
41
+ total=False,
42
+ )
43
+ setattr(Gemini_2_5_Flash_LiteParams, "__pydantic_config__", _PARAMS_CONFIG)
44
+
45
+ Gemini_2_5_Flash_Lite_SubscriptionParams = TypedDict(
46
+ "Gemini_2_5_Flash_Lite_SubscriptionParams",
47
+ {
48
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
49
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
50
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
51
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
52
+ "generationConfig.seed": int,
53
+ "generationConfig.thinkingConfig.thinkingBudget": int,
54
+ "generationConfig.thinkingConfig.includeThoughts": bool,
55
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
56
+ },
57
+ total=False,
58
+ )
59
+ setattr(Gemini_2_5_Flash_Lite_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
60
+
61
+ Gemini_2_5_Flash_SubscriptionParams = TypedDict(
62
+ "Gemini_2_5_Flash_SubscriptionParams",
63
+ {
64
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
65
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
66
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
67
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
68
+ "generationConfig.seed": int,
69
+ "generationConfig.thinkingConfig.thinkingBudget": Annotated[int, Field(ge=-1, le=24576)],
70
+ "generationConfig.thinkingConfig.includeThoughts": bool,
71
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
72
+ },
73
+ total=False,
74
+ )
75
+ setattr(Gemini_2_5_Flash_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
76
+
77
+ Gemini_2_5_ProParams = TypedDict(
78
+ "Gemini_2_5_ProParams",
79
+ {
80
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
81
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
82
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
83
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
84
+ "generationConfig.seed": int,
85
+ "generationConfig.thinkingConfig.thinkingBudget": Annotated[int, Field(ge=128, le=32768)],
86
+ "generationConfig.thinkingConfig.includeThoughts": bool,
87
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
88
+ },
89
+ total=False,
90
+ )
91
+ setattr(Gemini_2_5_ProParams, "__pydantic_config__", _PARAMS_CONFIG)
92
+
93
+ Gemini_2_5_Pro_SubscriptionParams = TypedDict(
94
+ "Gemini_2_5_Pro_SubscriptionParams",
95
+ {
96
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
97
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
98
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
99
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
100
+ "generationConfig.seed": int,
101
+ "generationConfig.thinkingConfig.thinkingBudget": Annotated[int, Field(ge=128, le=32768)],
102
+ "generationConfig.thinkingConfig.includeThoughts": bool,
103
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
104
+ },
105
+ total=False,
106
+ )
107
+ setattr(Gemini_2_5_Pro_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
108
+
109
+ Gemini_3_Flash_Preview_SubscriptionParams = TypedDict(
110
+ "Gemini_3_Flash_Preview_SubscriptionParams",
111
+ {
112
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
113
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
114
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
115
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
116
+ "generationConfig.seed": int,
117
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "low", "medium", "high"],
118
+ "generationConfig.thinkingConfig.includeThoughts": bool,
119
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
120
+ },
121
+ total=False,
122
+ )
123
+ setattr(Gemini_3_Flash_Preview_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
124
+
125
+ Gemini_3_1_Flash_LiteParams = TypedDict(
126
+ "Gemini_3_1_Flash_LiteParams",
127
+ {
128
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
129
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
130
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
131
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
132
+ "generationConfig.seed": int,
133
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "low", "medium", "high"],
134
+ "generationConfig.thinkingConfig.includeThoughts": bool,
135
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
136
+ },
137
+ total=False,
138
+ )
139
+ setattr(Gemini_3_1_Flash_LiteParams, "__pydantic_config__", _PARAMS_CONFIG)
140
+
141
+ Gemini_3_1_Flash_Lite_Preview_SubscriptionParams = TypedDict(
142
+ "Gemini_3_1_Flash_Lite_Preview_SubscriptionParams",
143
+ {
144
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
145
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
146
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
147
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
148
+ "generationConfig.seed": int,
149
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "low", "medium", "high"],
150
+ "generationConfig.thinkingConfig.includeThoughts": bool,
151
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
152
+ },
153
+ total=False,
154
+ )
155
+ setattr(Gemini_3_1_Flash_Lite_Preview_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
156
+
157
+ Gemini_3_1_Flash_Lite_SubscriptionParams = TypedDict(
158
+ "Gemini_3_1_Flash_Lite_SubscriptionParams",
159
+ {
160
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
161
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
162
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
163
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
164
+ "generationConfig.seed": int,
165
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "low", "medium", "high"],
166
+ "generationConfig.thinkingConfig.includeThoughts": bool,
167
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
168
+ },
169
+ total=False,
170
+ )
171
+ setattr(Gemini_3_1_Flash_Lite_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
172
+
173
+ Gemini_3_1_Pro_Preview_SubscriptionParams = TypedDict(
174
+ "Gemini_3_1_Pro_Preview_SubscriptionParams",
175
+ {
176
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
177
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
178
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
179
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
180
+ "generationConfig.seed": int,
181
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["low", "high"],
182
+ "generationConfig.thinkingConfig.includeThoughts": bool,
183
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
184
+ },
185
+ total=False,
186
+ )
187
+ setattr(Gemini_3_1_Pro_Preview_SubscriptionParams, "__pydantic_config__", _PARAMS_CONFIG)
188
+
189
+ Gemini_3_5_FlashParams = TypedDict(
190
+ "Gemini_3_5_FlashParams",
191
+ {
192
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
193
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
194
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
195
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
196
+ "generationConfig.seed": int,
197
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "low", "medium", "high"],
198
+ "generationConfig.thinkingConfig.includeThoughts": bool,
199
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
200
+ },
201
+ total=False,
202
+ )
203
+ setattr(Gemini_3_5_FlashParams, "__pydantic_config__", _PARAMS_CONFIG)
204
+
205
+ Gemini_Flash_LatestParams = TypedDict(
206
+ "Gemini_Flash_LatestParams",
207
+ {
208
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
209
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
210
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
211
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
212
+ "generationConfig.seed": int,
213
+ "generationConfig.thinkingConfig.thinkingBudget": int,
214
+ "generationConfig.thinkingConfig.includeThoughts": bool,
215
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
216
+ },
217
+ total=False,
218
+ )
219
+ setattr(Gemini_Flash_LatestParams, "__pydantic_config__", _PARAMS_CONFIG)
220
+
221
+ Gemma_3_12b_ItParams = TypedDict(
222
+ "Gemma_3_12b_ItParams",
223
+ {
224
+ "max_completion_tokens": Annotated[int, Field(ge=1, le=131072)],
225
+ "temperature": float,
226
+ "top_p": Annotated[float, Field(ge=0, le=1)],
227
+ "top_k": Annotated[int, Field(ge=0)],
228
+ },
229
+ total=False,
230
+ )
231
+ setattr(Gemma_3_12b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
232
+
233
+ Gemma_3_1b_ItParams = TypedDict(
234
+ "Gemma_3_1b_ItParams",
235
+ {
236
+ "max_completion_tokens": Annotated[int, Field(ge=1, le=32768)],
237
+ "temperature": float,
238
+ "top_p": Annotated[float, Field(ge=0, le=1)],
239
+ "top_k": Annotated[int, Field(ge=0)],
240
+ },
241
+ total=False,
242
+ )
243
+ setattr(Gemma_3_1b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
244
+
245
+ Gemma_3_27b_ItParams = TypedDict(
246
+ "Gemma_3_27b_ItParams",
247
+ {
248
+ "max_completion_tokens": Annotated[int, Field(ge=1, le=131072)],
249
+ "temperature": float,
250
+ "top_p": Annotated[float, Field(ge=0, le=1)],
251
+ "top_k": Annotated[int, Field(ge=0)],
252
+ },
253
+ total=False,
254
+ )
255
+ setattr(Gemma_3_27b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
256
+
257
+ Gemma_3_4b_ItParams = TypedDict(
258
+ "Gemma_3_4b_ItParams",
259
+ {
260
+ "max_completion_tokens": Annotated[int, Field(ge=1, le=131072)],
261
+ "temperature": float,
262
+ "top_p": Annotated[float, Field(ge=0, le=1)],
263
+ "top_k": Annotated[int, Field(ge=0)],
264
+ },
265
+ total=False,
266
+ )
267
+ setattr(Gemma_3_4b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
268
+
269
+ Gemma_3n_E2b_ItParams = TypedDict(
270
+ "Gemma_3n_E2b_ItParams",
271
+ {
272
+ "max_completion_tokens": Annotated[int, Field(ge=1, le=32768)],
273
+ "temperature": float,
274
+ "top_p": Annotated[float, Field(ge=0, le=1)],
275
+ "top_k": Annotated[int, Field(ge=0)],
276
+ },
277
+ total=False,
278
+ )
279
+ setattr(Gemma_3n_E2b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
280
+
281
+ Gemma_3n_E4b_ItParams = TypedDict(
282
+ "Gemma_3n_E4b_ItParams",
283
+ {
284
+ "max_completion_tokens": Annotated[int, Field(ge=1, le=32768)],
285
+ "temperature": float,
286
+ "top_p": Annotated[float, Field(ge=0, le=1)],
287
+ "top_k": Annotated[int, Field(ge=0)],
288
+ },
289
+ total=False,
290
+ )
291
+ setattr(Gemma_3n_E4b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
292
+
293
+ Gemma_4_12b_ItParams = TypedDict(
294
+ "Gemma_4_12b_ItParams",
295
+ {
296
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
297
+ "temperature": float,
298
+ "top_p": Annotated[float, Field(ge=0, le=1)],
299
+ "top_k": Annotated[int, Field(ge=0)],
300
+ },
301
+ total=False,
302
+ )
303
+ setattr(Gemma_4_12b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
304
+
305
+ Gemma_4_26b_A4b_ItParams = TypedDict(
306
+ "Gemma_4_26b_A4b_ItParams",
307
+ {
308
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
309
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
310
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
311
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
312
+ "generationConfig.seed": int,
313
+ "generationConfig.thinkingConfig.thinkingBudget": int,
314
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "high"],
315
+ "generationConfig.thinkingConfig.includeThoughts": bool,
316
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
317
+ },
318
+ total=False,
319
+ )
320
+ setattr(Gemma_4_26b_A4b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
321
+
322
+ Gemma_4_31b_ItParams = TypedDict(
323
+ "Gemma_4_31b_ItParams",
324
+ {
325
+ "generationConfig.maxOutputTokens": Annotated[int, Field(ge=1, le=65536)],
326
+ "generationConfig.temperature": Annotated[float, Field(ge=0, le=2)],
327
+ "generationConfig.topP": Annotated[float, Field(ge=0, le=1)],
328
+ "generationConfig.topK": Annotated[int, Field(ge=0)],
329
+ "generationConfig.seed": int,
330
+ "generationConfig.thinkingConfig.thinkingBudget": int,
331
+ "generationConfig.thinkingConfig.thinkingLevel": Literal["minimal", "high"],
332
+ "generationConfig.thinkingConfig.includeThoughts": bool,
333
+ "generationConfig.responseMimeType": Literal["text/plain", "application/json"],
334
+ },
335
+ total=False,
336
+ )
337
+ setattr(Gemma_4_31b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
338
+
339
+ Gemma_4_E2b_ItParams = TypedDict(
340
+ "Gemma_4_E2b_ItParams",
341
+ {
342
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
343
+ "temperature": float,
344
+ "top_p": Annotated[float, Field(ge=0, le=1)],
345
+ "top_k": Annotated[int, Field(ge=0)],
346
+ },
347
+ total=False,
348
+ )
349
+ setattr(Gemma_4_E2b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
350
+
351
+ Gemma_4_E4b_ItParams = TypedDict(
352
+ "Gemma_4_E4b_ItParams",
353
+ {
354
+ "max_completion_tokens": Annotated[int, Field(ge=1)],
355
+ "temperature": float,
356
+ "top_p": Annotated[float, Field(ge=0, le=1)],
357
+ "top_k": Annotated[int, Field(ge=0)],
358
+ },
359
+ total=False,
360
+ )
361
+ setattr(Gemma_4_E4b_ItParams, "__pydantic_config__", _PARAMS_CONFIG)
362
+
363
+ __all__ = [
364
+ "Gemini_2_5_FlashParams",
365
+ "Gemini_2_5_Flash_LiteParams",
366
+ "Gemini_2_5_Flash_Lite_SubscriptionParams",
367
+ "Gemini_2_5_Flash_SubscriptionParams",
368
+ "Gemini_2_5_ProParams",
369
+ "Gemini_2_5_Pro_SubscriptionParams",
370
+ "Gemini_3_Flash_Preview_SubscriptionParams",
371
+ "Gemini_3_1_Flash_LiteParams",
372
+ "Gemini_3_1_Flash_Lite_Preview_SubscriptionParams",
373
+ "Gemini_3_1_Flash_Lite_SubscriptionParams",
374
+ "Gemini_3_1_Pro_Preview_SubscriptionParams",
375
+ "Gemini_3_5_FlashParams",
376
+ "Gemini_Flash_LatestParams",
377
+ "Gemma_3_12b_ItParams",
378
+ "Gemma_3_1b_ItParams",
379
+ "Gemma_3_27b_ItParams",
380
+ "Gemma_3_4b_ItParams",
381
+ "Gemma_3n_E2b_ItParams",
382
+ "Gemma_3n_E4b_ItParams",
383
+ "Gemma_4_12b_ItParams",
384
+ "Gemma_4_26b_A4b_ItParams",
385
+ "Gemma_4_31b_ItParams",
386
+ "Gemma_4_E2b_ItParams",
387
+ "Gemma_4_E4b_ItParams",
388
+ ]
modelparams/types/groq.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
+ Qwen3_32bParams = TypedDict(
14
+ "Qwen3_32bParams",
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", "default"],
24
+ "reasoning_format": Literal["hidden", "raw", "parsed"],
25
+ "response_format.type": Literal["text", "json_object"],
26
+ },
27
+ total=False,
28
+ )
29
+ setattr(Qwen3_32bParams, "__pydantic_config__", _PARAMS_CONFIG)
30
+
31
+ __all__ = [
32
+ "Qwen3_32bParams",
33
+ ]