Skip to content
This repository was archived by the owner on Dec 11, 2025. It is now read-only.

Commit 0d8d3fb

Browse files
dragos199993dragos199993
authored
Merge pull request #85 from Not-Diamond/ENG-2409_support-latest-models
Add new model providers
2 parents fd64b62 + 2a488ae commit 0d8d3fb

134 files changed

Lines changed: 32613 additions & 29434 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎notdiamond/llms/providers.py‎

Lines changed: 24 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,12 @@ class NDLLMProviders(Enum):
2323
GPT_4o_MINI_2024_07_18 (NDLLMProvider): refers to 'gpt-4o-mini-2024-07-18' model by OpenAI
2424
GPT_4o_MINI (NDLLMProvider): refers to 'gpt-4o-mini' model by OpenAI
2525
GPT_4_0125_PREVIEW (NDLLMProvider): refers to 'gpt-4-0125-preview' model by OpenAI
26-
GPT_4_5_PREVIEW (NDLLMProvider): refers to 'gpt-4.5-preview' model by OpenAI
27-
GPT_4_5_PREVIEW_2025_02_27 (NDLLMProvider): refers to 'gpt-4.5-preview-2025-02-27' model by OpenAI
26+
GPT_4_1 (NDLLMProvider): refers to 'gpt-4.1' model by OpenAI
27+
GPT_4_1_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-2025-04-14' model by OpenAI
28+
GPT_4_1_MINI (NDLLMProvider): refers to 'gpt-4.1-mini' model by OpenAI
29+
GPT_4_1_MINI_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-mini-2025-04-14' model by OpenAI
30+
GPT_4_1_NANO (NDLLMProvider): refers to 'gpt-4.1-nano' model by OpenAI
31+
GPT_4_1_NANO_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-nano-2025-04-14' model by OpenAI
2832
O1_PREVIEW (NDLLMProvider): refers to 'o1-preview' model by OpenAI
2933
O1_PREVIEW_2024_09_12 (NDLLMProvider): refers to 'o1-preview-2024-09-12' model by OpenAI
3034
O1_MINI (NDLLMProvider): refers to 'o1-mini' model by OpenAI
@@ -38,6 +42,10 @@ class NDLLMProviders(Enum):
3842
CLAUDE_3_7_SONNET_20250219 (NDLLMProvider): refers to 'claude-3-7-sonnet-20250219' model by Anthropic
3943
CLAUDE_3_5_HAIKU_20241022 (NDLLMProvider): refers to 'claude-3-5-haiku-20241022' model by Anthropic
4044
CLAUDE_3_HAIKU_20240307 (NDLLMProvider): refers to 'claude-3-haiku-20240307' model by Anthropic
45+
CLAUDE_OPUS_4_20250514 (NDLLMProvider): refers to 'claude-opus-4-20250514' model by Anthropic
46+
CLAUDE_SONNET_4_20250514 (NDLLMProvider): refers to 'claude-sonnet-4-20250514' model by Anthropic
47+
CLAUDE_OPUS_4_0 (NDLLMProvider): refers to 'claude-opus-4-0' model by Anthropic
48+
CLAUDE_SONNET_4_0 (NDLLMProvider): refers to 'claude-sonnet-4-0' model by Anthropic
4149
4250
GEMINI_PRO (NDLLMProvider): refers to 'gemini-pro' model by Google
4351
GEMINI_1_PRO_LATEST (NDLLMProvider): refers to 'gemini-1.0-pro-latest' model by Google
@@ -46,6 +54,8 @@ class NDLLMProviders(Enum):
4654
GEMINI_15_FLASH_LATEST (NDLLMProvider): refers to 'gemini-1.5-flash-latest' model by Google
4755
GEMINI_20_FLASH (NDLLMProvider): refers to 'gemini-20-flash' model by Google
4856
GEMINI_20_FLASH_001 (NDLLMProvider): refers to 'gemini-20-flash-001' model by Google
57+
GEMINI_25_FLASH (NDLLMProvider): refers to 'gemini-25-flash' model by Google
58+
GEMINI_25_PRO (NDLLMProvider): refers to 'gemini-25-pro' model by Google
4959
5060
COMMAND_R (NDLLMProvider): refers to 'command-r' model by Cohere
5161
COMMAND_R_PLUS (NDLLMProvider): refers to 'command-r-plus' model by Cohere
@@ -99,8 +109,12 @@ class NDLLMProviders(Enum):
99109
GPT_4o_MINI_2024_07_18 = ("openai", "gpt-4o-mini-2024-07-18")
100110
GPT_4o_MINI = ("openai", "gpt-4o-mini")
101111
GPT_4_0125_PREVIEW = ("openai", "gpt-4-0125-preview")
102-
GPT_4_5_PREVIEW = ("openai", "gpt-4.5-preview")
103-
GPT_4_5_PREVIEW_2025_02_27 = ("openai", "gpt-4.5-preview-2025-02-27")
112+
GPT_4_1 = ("openai", "gpt-4.1")
113+
GPT_4_1_2025_04_14 = ("openai", "gpt-4.1-2025-04-14")
114+
GPT_4_1_MINI = ("openai", "gpt-4.1-mini")
115+
GPT_4_1_MINI_2025_04_14 = ("openai", "gpt-4.1-mini-2025-04-14")
116+
GPT_4_1_NANO = ("openai", "gpt-4.1-nano")
117+
GPT_4_1_NANO_2025_04_14 = ("openai", "gpt-4.1-nano-2025-04-14")
104118
O1_PREVIEW = ("openai", "o1-preview")
105119
O1_PREVIEW_2024_09_12 = ("openai", "o1-preview-2024-09-12")
106120
O1_MINI = ("openai", "o1-mini")
@@ -117,6 +131,10 @@ class NDLLMProviders(Enum):
117131
CLAUDE_3_7_SONNET_20250219 = ("anthropic", "claude-3-7-sonnet-20250219")
118132
CLAUDE_3_5_HAIKU_20241022 = ("anthropic", "claude-3-5-haiku-20241022")
119133
CLAUDE_3_HAIKU_20240307 = ("anthropic", "claude-3-haiku-20240307")
134+
CLAUDE_OPUS_4_20250514 = ("anthropic", "claude-opus-4-20250514")
135+
CLAUDE_SONNET_4_20250514 = ("anthropic", "claude-sonnet-4-20250514")
136+
CLAUDE_OPUS_4_0 = ("anthropic", "claude-opus-4-0")
137+
CLAUDE_SONNET_4_0 = ("anthropic", "claude-sonnet-4-0")
120138

121139
GEMINI_PRO = ("google", "gemini-pro")
122140
GEMINI_1_PRO_LATEST = ("google", "gemini-1.0-pro-latest")
@@ -125,6 +143,8 @@ class NDLLMProviders(Enum):
125143
GEMINI_15_FLASH_LATEST = ("google", "gemini-1.5-flash-latest")
126144
GEMINI_20_FLASH = ("google", "gemini-2.0-flash")
127145
GEMINI_20_FLASH_001 = ("google", "gemini-2.0-flash-001")
146+
GEMINI_25_FLASH = ("google", "gemini-2.5-flash")
147+
GEMINI_25_PRO = ("google", "gemini-2.5-pro")
128148

129149
COMMAND_R = ("cohere", "command-r")
130150
COMMAND_R_PLUS = ("cohere", "command-r-plus")

‎notdiamond/settings.py‎

Lines changed: 56 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,12 @@
3939
"gpt-4-turbo-preview",
4040
"gpt-4-0125-preview",
4141
"gpt-4-1106-preview",
42-
"gpt-4.5-preview",
43-
"gpt-4.5-preview-2025-02-27",
42+
"gpt-4.1",
43+
"gpt-4.1-2025-04-14",
44+
"gpt-4.1-mini",
45+
"gpt-4.1-mini-2025-04-14",
46+
"gpt-4.1-nano",
47+
"gpt-4.1-nano-2025-04-14",
4448
"o1-preview",
4549
"o1-preview-2024-09-12",
4650
"o1-mini",
@@ -63,8 +67,12 @@
6367
"gpt-4-turbo-preview",
6468
"gpt-4-0125-preview",
6569
"gpt-4-1106-preview",
66-
"gpt-4.5-preview",
67-
"gpt-4.5-preview-2025-02-27",
70+
"gpt-4.1",
71+
"gpt-4.1-2025-04-14",
72+
"gpt-4.1-mini",
73+
"gpt-4.1-mini-2025-04-14",
74+
"gpt-4.1-nano",
75+
"gpt-4.1-nano-2025-04-14",
6876
],
6977
"support_response_model": [
7078
"gpt-3.5-turbo",
@@ -86,8 +94,12 @@
8694
"o1-mini",
8795
"o1-mini-2024-09-12",
8896
"chatgpt-4o-latest",
89-
"gpt-4.5-preview",
90-
"gpt-4.5-preview-2025-02-27",
97+
"gpt-4.1",
98+
"gpt-4.1-2025-04-14",
99+
"gpt-4.1-mini",
100+
"gpt-4.1-mini-2025-04-14",
101+
"gpt-4.1-nano",
102+
"gpt-4.1-nano-2025-04-14",
91103
],
92104
"openrouter_identifier": {
93105
"gpt-3.5-turbo": "openai/gpt-3.5-turbo",
@@ -105,8 +117,12 @@
105117
"o1-mini": "openai/o1-mini",
106118
"o1-mini-2024-09-12": "openai/o1-mini-2024-09-12",
107119
"chatgpt-4o-latest": "openai/chatgpt-4o-latest",
108-
"gpt-4.5-preview": "openai/gpt-4.5-preview",
109-
"gpt-4.5-preview-2025-02-27": "openai/gpt-4.5-preview-2025-02-27",
120+
"gpt-4.1": "openai/gpt-4.1",
121+
"gpt-4.1-2025-04-14": "openai/gpt-4.1-2025-04-14",
122+
"gpt-4.1-mini": "openai/gpt-4.1-mini",
123+
"gpt-4.1-mini-2025-04-14": "openai/gpt-4.1-mini-2025-04-14",
124+
"gpt-4.1-nano": "openai/gpt-4.1-nano",
125+
"gpt-4.1-nano-2025-04-14": "openai/gpt-4.1-nano-2025-04-14",
110126
},
111127
"price": {
112128
"gpt-3.5-turbo": {"input": 0.5, "output": 1.5},
@@ -128,8 +144,12 @@
128144
"o1-mini": {"input": 3.0, "output": 12.0},
129145
"o1-mini-2024-09-12": {"input": 3.0, "output": 12.0},
130146
"chatgpt-4o-latest": {"input": 5.0, "output": 15.0},
131-
"gpt-4.5-preview": {"input": 75.0, "output": 150.0},
132-
"gpt-4.5-preview-2025-02-27": {"input": 75.0, "output": 150.0},
147+
"gpt-4.1": {"input": 2.0, "output": 8.0},
148+
"gpt-4.1-mini": {"input": 0.5, "output": 1.6},
149+
"gpt-4.1-nano": {"input": 0.1, "output": 0.4},
150+
"gpt-4.1-2025-04-14": {"input": 2.0, "output": 8.0},
151+
"gpt-4.1-mini-2025-04-14": {"input": 0.5, "output": 1.6},
152+
"gpt-4.1-nano-2025-04-14": {"input": 0.1, "output": 0.4},
133153
},
134154
},
135155
"anthropic": {
@@ -144,6 +164,10 @@
144164
"claude-3-5-sonnet-latest",
145165
"claude-3-7-sonnet-latest",
146166
"claude-3-7-sonnet-20250219",
167+
"claude-opus-4-20250514",
168+
"claude-sonnet-4-20250514",
169+
"claude-opus-4-0",
170+
"claude-sonnet-4-0",
147171
],
148172
"api_key": ANTHROPIC_API_KEY,
149173
"support_tools": [
@@ -156,6 +180,10 @@
156180
"claude-3-5-sonnet-latest",
157181
"claude-3-7-sonnet-latest",
158182
"claude-3-7-sonnet-20250219",
183+
"claude-opus-4-20250514",
184+
"claude-sonnet-4-20250514",
185+
"claude-opus-4-0",
186+
"claude-sonnet-4-0",
159187
],
160188
"support_response_model": [
161189
"claude-2.1",
@@ -171,6 +199,10 @@
171199
"claude-3-5-haiku-20241022": "anthropic/claude-3.5-haiku",
172200
"claude-3-7-sonnet-latest": "anthropic/claude-3.7-sonnet",
173201
"claude-3-7-sonnet-20250219": "anthropic/claude-3.7-sonnet",
202+
"claude-opus-4-20250514": "anthropic/claude-opus-4",
203+
"claude-sonnet-4-20250514": "anthropic/claude-sonnet-4",
204+
"claude-opus-4-0": "anthropic/claude-opus-4",
205+
"claude-sonnet-4-0": "anthropic/claude-sonnet-4",
174206
},
175207
"price": {
176208
"claude-2.1": {"input": 8.0, "output": 24.0},
@@ -183,6 +215,10 @@
183215
"claude-3-5-sonnet-latest": {"input": 3.0, "output": 15.0},
184216
"claude-3-7-sonnet-latest": {"input": 3.0, "output": 15.0},
185217
"claude-3-7-sonnet-20250219": {"input": 3.0, "output": 15.0},
218+
"claude-opus-4-20250514": {"input": 15.0, "output": 75.0},
219+
"claude-sonnet-4-20250514": {"input": 3.0, "output": 15.0},
220+
"claude-opus-4-0": {"input": 15.0, "output": 75.0},
221+
"claude-sonnet-4-0": {"input": 3.0, "output": 15.0},
186222
},
187223
},
188224
"google": {
@@ -194,6 +230,8 @@
194230
"gemini-1.5-flash-latest",
195231
"gemini-2.0-flash",
196232
"gemini-2.0-flash-001",
233+
"gemini-2.5-flash",
234+
"gemini-2.5-pro",
197235
],
198236
"api_key": GOOGLE_API_KEY,
199237
"support_tools": [
@@ -202,6 +240,8 @@
202240
"gemini-1.5-pro-latest",
203241
"gemini-1.5-pro-exp-0801",
204242
"gemini-1.5-flash-latest",
243+
"gemini-2.5-flash",
244+
"gemini-2.5-pro",
205245
],
206246
"support_response_model": [
207247
"gemini-pro",
@@ -211,6 +251,8 @@
211251
"gemini-1.5-flash-latest",
212252
"gemini-2.0-flash",
213253
"gemini-2.0-flash-001",
254+
"gemini-2.5-flash",
255+
"gemini-2.5-pro",
214256
],
215257
"openrouter_identifier": {
216258
"gemini-pro": "google/gemini-pro",
@@ -220,6 +262,8 @@
220262
"gemini-1.5-flash-latest": "google/gemini-flash-1.5", #
221263
"gemini-2.0-flash": "google/gemini-2.0-flash", #
222264
"gemini-2.0-flash-001": "google/gemini-2.0-flash", #
265+
"gemini-2.5-flash": "google/gemini-2.5-flash", #
266+
"gemini-2.5-pro": "google/gemini-2.5-pro", #
223267
},
224268
"price": {
225269
"gemini-pro": {"input": 0.5, "output": 1.5},
@@ -229,6 +273,8 @@
229273
"gemini-1.5-flash-latest": {"input": 0.35, "output": 1.05},
230274
"gemini-2.0-flash": {"input": 0.10, "output": 0.40},
231275
"gemini-2.0-flash-001": {"input": 0.10, "output": 0.40},
276+
"gemini-2.5-flash": {"input": 0.15, "output": 0.6},
277+
"gemini-2.5-pro": {"input": 2.0, "output": 10.0},
232278
},
233279
},
234280
"cohere": {

0 commit comments

Comments
 (0)