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

Commit c64e872

Browse files
dragos199993dragos199993
committed
chore: Add new model providers
1 parent fd64b62 commit c64e872

2 files changed

Lines changed: 47 additions & 0 deletions

File tree

‎notdiamond/llms/providers.py‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,9 @@ 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_1 (NDLLMProvider): refers to 'gpt-4.1' model by OpenAI
27+
GPT_4_1_MINI (NDLLMProvider): refers to 'gpt-4.1-mini' model by OpenAI
28+
GPT_4_1_NANO (NDLLMProvider): refers to 'gpt-4.1-nano' model by OpenAI
2629
GPT_4_5_PREVIEW (NDLLMProvider): refers to 'gpt-4.5-preview' model by OpenAI
2730
GPT_4_5_PREVIEW_2025_02_27 (NDLLMProvider): refers to 'gpt-4.5-preview-2025-02-27' model by OpenAI
2831
O1_PREVIEW (NDLLMProvider): refers to 'o1-preview' model by OpenAI
@@ -38,6 +41,8 @@ class NDLLMProviders(Enum):
3841
CLAUDE_3_7_SONNET_20250219 (NDLLMProvider): refers to 'claude-3-7-sonnet-20250219' model by Anthropic
3942
CLAUDE_3_5_HAIKU_20241022 (NDLLMProvider): refers to 'claude-3-5-haiku-20241022' model by Anthropic
4043
CLAUDE_3_HAIKU_20240307 (NDLLMProvider): refers to 'claude-3-haiku-20240307' model by Anthropic
44+
CLAUDE_OPUS_4_20250514 (NDLLMProvider): refers to 'claude-opus-4-20250514' model by Anthropic
45+
CLAUDE_SONNET_4_20250514 (NDLLMProvider): refers to 'claude-sonnet-4-20250514' model by Anthropic
4146
4247
GEMINI_PRO (NDLLMProvider): refers to 'gemini-pro' model by Google
4348
GEMINI_1_PRO_LATEST (NDLLMProvider): refers to 'gemini-1.0-pro-latest' model by Google
@@ -46,6 +51,8 @@ class NDLLMProviders(Enum):
4651
GEMINI_15_FLASH_LATEST (NDLLMProvider): refers to 'gemini-1.5-flash-latest' model by Google
4752
GEMINI_20_FLASH (NDLLMProvider): refers to 'gemini-20-flash' model by Google
4853
GEMINI_20_FLASH_001 (NDLLMProvider): refers to 'gemini-20-flash-001' model by Google
54+
GEMINI_25_FLASH (NDLLMProvider): refers to 'gemini-25-flash' model by Google
55+
GEMINI_25_PRO (NDLLMProvider): refers to 'gemini-25-pro' model by Google
4956
5057
COMMAND_R (NDLLMProvider): refers to 'command-r' model by Cohere
5158
COMMAND_R_PLUS (NDLLMProvider): refers to 'command-r-plus' model by Cohere
@@ -99,6 +106,9 @@ class NDLLMProviders(Enum):
99106
GPT_4o_MINI_2024_07_18 = ("openai", "gpt-4o-mini-2024-07-18")
100107
GPT_4o_MINI = ("openai", "gpt-4o-mini")
101108
GPT_4_0125_PREVIEW = ("openai", "gpt-4-0125-preview")
109+
GPT_4_1 = ("openai", "gpt-4.1")
110+
GPT_4_1_MINI = ("openai", "gpt-4.1-mini")
111+
GPT_4_1_NANO = ("openai", "gpt-4.1-nano")
102112
GPT_4_5_PREVIEW = ("openai", "gpt-4.5-preview")
103113
GPT_4_5_PREVIEW_2025_02_27 = ("openai", "gpt-4.5-preview-2025-02-27")
104114
O1_PREVIEW = ("openai", "o1-preview")
@@ -117,6 +127,8 @@ class NDLLMProviders(Enum):
117127
CLAUDE_3_7_SONNET_20250219 = ("anthropic", "claude-3-7-sonnet-20250219")
118128
CLAUDE_3_5_HAIKU_20241022 = ("anthropic", "claude-3-5-haiku-20241022")
119129
CLAUDE_3_HAIKU_20240307 = ("anthropic", "claude-3-haiku-20240307")
130+
CLAUDE_OPUS_4_20250514 = ("anthropic", "claude-opus-4-20250514")
131+
CLAUDE_SONNET_4_20250514 = ("anthropic", "claude-sonnet-4-20250514")
120132

121133
GEMINI_PRO = ("google", "gemini-pro")
122134
GEMINI_1_PRO_LATEST = ("google", "gemini-1.0-pro-latest")
@@ -125,6 +137,8 @@ class NDLLMProviders(Enum):
125137
GEMINI_15_FLASH_LATEST = ("google", "gemini-1.5-flash-latest")
126138
GEMINI_20_FLASH = ("google", "gemini-2.0-flash")
127139
GEMINI_20_FLASH_001 = ("google", "gemini-2.0-flash-001")
140+
GEMINI_25_FLASH = ("google", "gemini-2.5-flash")
141+
GEMINI_25_PRO = ("google", "gemini-2.5-pro")
128142

129143
COMMAND_R = ("cohere", "command-r")
130144
COMMAND_R_PLUS = ("cohere", "command-r-plus")

‎notdiamond/settings.py‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,9 @@
3939
"gpt-4-turbo-preview",
4040
"gpt-4-0125-preview",
4141
"gpt-4-1106-preview",
42+
"gpt-4.1",
43+
"gpt-4.1-mini",
44+
"gpt-4.1-nano",
4245
"gpt-4.5-preview",
4346
"gpt-4.5-preview-2025-02-27",
4447
"o1-preview",
@@ -65,6 +68,9 @@
6568
"gpt-4-1106-preview",
6669
"gpt-4.5-preview",
6770
"gpt-4.5-preview-2025-02-27",
71+
"gpt-4.1",
72+
"gpt-4.1-mini",
73+
"gpt-4.1-nano",
6874
],
6975
"support_response_model": [
7076
"gpt-3.5-turbo",
@@ -88,6 +94,9 @@
8894
"chatgpt-4o-latest",
8995
"gpt-4.5-preview",
9096
"gpt-4.5-preview-2025-02-27",
97+
"gpt-4.1",
98+
"gpt-4.1-mini",
99+
"gpt-4.1-nano",
91100
],
92101
"openrouter_identifier": {
93102
"gpt-3.5-turbo": "openai/gpt-3.5-turbo",
@@ -107,6 +116,9 @@
107116
"chatgpt-4o-latest": "openai/chatgpt-4o-latest",
108117
"gpt-4.5-preview": "openai/gpt-4.5-preview",
109118
"gpt-4.5-preview-2025-02-27": "openai/gpt-4.5-preview-2025-02-27",
119+
"gpt-4.1": "openai/gpt-4.1",
120+
"gpt-4.1-mini": "openai/gpt-4.1-mini",
121+
"gpt-4.1-nano": "openai/gpt-4.1-nano",
110122
},
111123
"price": {
112124
"gpt-3.5-turbo": {"input": 0.5, "output": 1.5},
@@ -130,6 +142,9 @@
130142
"chatgpt-4o-latest": {"input": 5.0, "output": 15.0},
131143
"gpt-4.5-preview": {"input": 75.0, "output": 150.0},
132144
"gpt-4.5-preview-2025-02-27": {"input": 75.0, "output": 150.0},
145+
"gpt-4.1": {"input": 2.0, "output": 8.0},
146+
"gpt-4.1-mini": {"input": 0.5, "output": 1.6},
147+
"gpt-4.1-nano": {"input": 0.1, "output": 0.4},
133148
},
134149
},
135150
"anthropic": {
@@ -144,6 +159,8 @@
144159
"claude-3-5-sonnet-latest",
145160
"claude-3-7-sonnet-latest",
146161
"claude-3-7-sonnet-20250219",
162+
"claude-opus-4-20250514",
163+
"claude-sonnet-4-20250514",
147164
],
148165
"api_key": ANTHROPIC_API_KEY,
149166
"support_tools": [
@@ -156,6 +173,8 @@
156173
"claude-3-5-sonnet-latest",
157174
"claude-3-7-sonnet-latest",
158175
"claude-3-7-sonnet-20250219",
176+
"claude-opus-4-20250514",
177+
"claude-sonnet-4-20250514",
159178
],
160179
"support_response_model": [
161180
"claude-2.1",
@@ -171,6 +190,8 @@
171190
"claude-3-5-haiku-20241022": "anthropic/claude-3.5-haiku",
172191
"claude-3-7-sonnet-latest": "anthropic/claude-3.7-sonnet",
173192
"claude-3-7-sonnet-20250219": "anthropic/claude-3.7-sonnet",
193+
"claude-opus-4-20250514": "anthropic/claude-opus-4",
194+
"claude-sonnet-4-20250514": "anthropic/claude-sonnet-4",
174195
},
175196
"price": {
176197
"claude-2.1": {"input": 8.0, "output": 24.0},
@@ -183,6 +204,8 @@
183204
"claude-3-5-sonnet-latest": {"input": 3.0, "output": 15.0},
184205
"claude-3-7-sonnet-latest": {"input": 3.0, "output": 15.0},
185206
"claude-3-7-sonnet-20250219": {"input": 3.0, "output": 15.0},
207+
"claude-opus-4-20250514": {"input": 15.0, "output": 75.0},
208+
"claude-sonnet-4-20250514": {"input": 3.0, "output": 15.0},
186209
},
187210
},
188211
"google": {
@@ -194,6 +217,8 @@
194217
"gemini-1.5-flash-latest",
195218
"gemini-2.0-flash",
196219
"gemini-2.0-flash-001",
220+
"gemini-2.5-flash",
221+
"gemini-2.5-pro",
197222
],
198223
"api_key": GOOGLE_API_KEY,
199224
"support_tools": [
@@ -202,6 +227,8 @@
202227
"gemini-1.5-pro-latest",
203228
"gemini-1.5-pro-exp-0801",
204229
"gemini-1.5-flash-latest",
230+
"gemini-2.5-flash",
231+
"gemini-2.5-pro",
205232
],
206233
"support_response_model": [
207234
"gemini-pro",
@@ -211,6 +238,8 @@
211238
"gemini-1.5-flash-latest",
212239
"gemini-2.0-flash",
213240
"gemini-2.0-flash-001",
241+
"gemini-2.5-flash",
242+
"gemini-2.5-pro",
214243
],
215244
"openrouter_identifier": {
216245
"gemini-pro": "google/gemini-pro",
@@ -220,6 +249,8 @@
220249
"gemini-1.5-flash-latest": "google/gemini-flash-1.5", #
221250
"gemini-2.0-flash": "google/gemini-2.0-flash", #
222251
"gemini-2.0-flash-001": "google/gemini-2.0-flash", #
252+
"gemini-2.5-flash": "google/gemini-2.5-flash", #
253+
"gemini-2.5-pro": "google/gemini-2.5-pro", #
223254
},
224255
"price": {
225256
"gemini-pro": {"input": 0.5, "output": 1.5},
@@ -229,6 +260,8 @@
229260
"gemini-1.5-flash-latest": {"input": 0.35, "output": 1.05},
230261
"gemini-2.0-flash": {"input": 0.10, "output": 0.40},
231262
"gemini-2.0-flash-001": {"input": 0.10, "output": 0.40},
263+
"gemini-2.5-flash": {"input": 0.15, "output": 0.6},
264+
"gemini-2.5-pro": {"input": 2.0, "output": 10.0},
232265
},
233266
},
234267
"cohere": {

0 commit comments

Comments
 (0)