model-library 0.1.2__py3-none-any.whl → 0.1.4__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.
Files changed (41) hide show
  1. model_library/base/base.py +13 -6
  2. model_library/base/output.py +55 -0
  3. model_library/base/utils.py +3 -2
  4. model_library/config/README.md +169 -0
  5. model_library/config/ai21labs_models.yaml +11 -10
  6. model_library/config/alibaba_models.yaml +21 -22
  7. model_library/config/all_models.json +4708 -2471
  8. model_library/config/amazon_models.yaml +100 -102
  9. model_library/config/anthropic_models.yaml +59 -45
  10. model_library/config/cohere_models.yaml +25 -24
  11. model_library/config/deepseek_models.yaml +28 -25
  12. model_library/config/dummy_model.yaml +9 -7
  13. model_library/config/fireworks_models.yaml +86 -56
  14. model_library/config/google_models.yaml +156 -102
  15. model_library/config/inception_models.yaml +6 -6
  16. model_library/config/kimi_models.yaml +13 -14
  17. model_library/config/minimax_models.yaml +37 -0
  18. model_library/config/mistral_models.yaml +85 -29
  19. model_library/config/openai_models.yaml +192 -159
  20. model_library/config/perplexity_models.yaml +8 -23
  21. model_library/config/together_models.yaml +115 -103
  22. model_library/config/xai_models.yaml +85 -57
  23. model_library/config/zai_models.yaml +23 -15
  24. model_library/exceptions.py +12 -17
  25. model_library/file_utils.py +1 -1
  26. model_library/providers/amazon.py +32 -17
  27. model_library/providers/anthropic.py +2 -6
  28. model_library/providers/google/google.py +35 -29
  29. model_library/providers/minimax.py +33 -0
  30. model_library/providers/mistral.py +10 -1
  31. model_library/providers/openai.py +10 -8
  32. model_library/providers/together.py +18 -211
  33. model_library/register_models.py +36 -38
  34. model_library/registry_utils.py +18 -16
  35. model_library/utils.py +2 -2
  36. {model_library-0.1.2.dist-info → model_library-0.1.4.dist-info}/METADATA +3 -4
  37. model_library-0.1.4.dist-info/RECORD +64 -0
  38. model_library-0.1.2.dist-info/RECORD +0 -61
  39. {model_library-0.1.2.dist-info → model_library-0.1.4.dist-info}/WHEEL +0 -0
  40. {model_library-0.1.2.dist-info → model_library-0.1.4.dist-info}/licenses/LICENSE +0 -0
  41. {model_library-0.1.2.dist-info → model_library-0.1.4.dist-info}/top_level.txt +0 -0
model_library/config/mistral_models.yaml CHANGED
@@ -2,24 +2,25 @@ base-config:
2
2
  company: Mistral
3
3
  documentation_url: https://docs.mistral.ai/getting-started/models/models_overview/
4
4
  open_source: false
5
- class_properties:
6
- supports_images: false
5
+ supports:
6
+ images: false
7
+ files: false
8
+ tools: true
9
+ metadata:
7
10
  available_as_evaluator: false
8
- supports_files: false
9
11
  available_for_everyone: true
10
12
  ignored_for_cost: false
11
- supports_tools: true
12
13
  properties:
13
14
  context_window: null
14
- max_token_output: null
15
+ max_tokens: null
15
16
  training_cutoff: null
16
17
 
17
18
  magistral-models:
18
19
  base-config:
19
20
  properties:
20
21
  reasoning_model: true
21
- class_properties:
22
- supports_temperature: true
22
+ supports:
23
+ temperature: true
23
24
  default_parameters:
24
25
  temperature: 0.7
25
26
 
@@ -30,10 +31,10 @@ magistral-models:
30
31
  release_date: 2025-06-10
31
32
  properties:
32
33
  context_window: 131_072
33
- max_token_output: 40_000
34
+ max_tokens: 40_000
34
35
  training_cutoff: "2024-05"
35
- class_properties:
36
- supports_images: true
36
+ supports:
37
+ images: true
37
38
  costs_per_million_token:
38
39
  input: 2
39
40
  output: 5
@@ -44,9 +45,9 @@ magistral-models:
44
45
  release_date: 2025-09-18
45
46
  properties:
46
47
  context_window: 128_000
47
- max_token_output: 128_000
48
- class_properties:
49
- supports_images: true
48
+ max_tokens: 128_000
49
+ supports:
50
+ images: true
50
51
  costs_per_million_token:
51
52
  input: 2
52
53
  output: 5
@@ -58,23 +59,77 @@ magistral-models:
58
59
  release_date: 2025-09-18
59
60
  properties:
60
61
  context_window: 128_000
61
- max_token_output: 128_000
62
- class_properties:
63
- supports_images: true
62
+ max_tokens: 128_000
63
+ supports:
64
+ images: true
64
65
  costs_per_million_token:
65
66
  input: 0.5
66
67
  output: 1.5
67
68
 
69
+ # Standard Mistral Models
68
70
  mistralai-models:
69
71
  base-config:
70
72
  properties:
71
73
  reasoning_model: false
72
- class_properties:
73
- supports_temperature: true
74
+ max_tokens: 256_000
75
+ supports:
76
+ temperature: true
74
77
  default_parameters:
75
78
  temperature: 0.7
76
79
 
77
- # Standard Mistral Models
80
+ mistralai/mistral-large-2512:
81
+ label: Mistral Large 3
82
+ description: Mistral's latest flagship model
83
+ release_date: 2025-12-02
84
+ open_source: true
85
+ properties:
86
+ context_window: 256_000
87
+ max_tokens: 256_000
88
+ supports:
89
+ images: true
90
+ tools: true
91
+ costs_per_million_token:
92
+ input: 0.5
93
+ output: 1.5
94
+ default_parameters:
95
+ temperature: 0.065
96
+
97
+ mistralai/devstral-2512:
98
+ label: Devstral 2
99
+ description: Enterprise grade coding model that excels at using tools to explore codebases and power software engineering agents.
100
+ documentation_url: https://docs.mistral.ai/models/devstral-2-25-12
101
+ release_date: 2025-12-09
102
+ open_source: true
103
+ properties:
104
+ context_window: 256_000
105
+ max_tokens: 256_000
106
+ supports:
107
+ images: false
108
+ tools: true
109
+ costs_per_million_token:
110
+ input: 0.4
111
+ output: 2.0
112
+ default_parameters:
113
+ temperature: 0.2
114
+
115
+ mistralai/labs-devstral-small-2512:
116
+ label: Devstral Small 2
117
+ description: Open source coding model that excels at using tools to explore codebases and power software engineering agents.
118
+ documentation_url: https://docs.mistral.ai/models/devstral-small-2-25-12
119
+ release_date: 2025-12-09
120
+ open_source: true
121
+ properties:
122
+ context_window: 256_000
123
+ max_tokens: 256_000
124
+ supports:
125
+ images: false
126
+ tools: true
127
+ costs_per_million_token:
128
+ input: 0.1
129
+ output: 0.3
130
+ default_parameters:
131
+ temperature: 0.2
132
+
78
133
  mistralai/mistral-medium-2505:
79
134
  label: Mistral Medium 3.1 (05/2025)
80
135
  description: Mistral Medium 2505
@@ -82,10 +137,10 @@ mistralai-models:
82
137
  release_date: 2025-05-07
83
138
  properties:
84
139
  context_window: 131_072
85
- max_token_output: 32768
140
+ max_tokens: 32768
86
141
  training_cutoff: "2024-05"
87
- class_properties:
88
- supports_images: true
142
+ supports:
143
+ images: true
89
144
  costs_per_million_token:
90
145
  input: 0.4
91
146
  output: 2.0
@@ -97,10 +152,10 @@ mistralai-models:
97
152
  release_date: 2025-03-17
98
153
  properties:
99
154
  context_window: 131_072
100
- max_token_output: 8192
155
+ max_tokens: 8192
101
156
  training_cutoff: "2024-05"
102
- class_properties:
103
- supports_images: true
157
+ supports:
158
+ images: true
104
159
  costs_per_million_token:
105
160
  input: 0.075
106
161
  output: 0.3
@@ -111,8 +166,9 @@ mistralai-models:
111
166
  release_date: 2024-11-18
112
167
  properties:
113
168
  context_window: 128_000
114
- class_properties:
115
- supports_images: true
169
+ supports:
170
+ images: true
171
+ metadata:
116
172
  deprecated: true
117
173
  costs_per_million_token:
118
174
  input: 2.00
@@ -124,7 +180,7 @@ mistralai-models:
124
180
  release_date: 2024-11-18
125
181
  properties:
126
182
  context_window: 128_000
127
- class_properties:
183
+ metadata:
128
184
  deprecated: true
129
185
  costs_per_million_token:
130
186
  input: 2.00
@@ -136,7 +192,7 @@ mistralai-models:
136
192
  release_date: 2024-09-17
137
193
  properties:
138
194
  context_window: 32_000
139
- class_properties:
195
+ metadata:
140
196
  deprecated: true
141
197
  costs_per_million_token:
142
198
  input: 0.20