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

Add new model providers #85

Merged
dragos199993 merged 4 commits into from
May 30, 2025
Merged

Add new model providers #85

Show file tree
Hide file tree
Changes from 3 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
c64e872
chore: Add new model providers
dragos199993 May 28, 2025
c00897f
chore: Add new model versions for GPT-4.1 and Claude
dragos199993 May 29, 2025
0791da6
chore: Remove deprecated GPT-4.5 model references from settings and p…
dragos199993 May 29, 2025
2a488ae
chore: record cassetes
dragos199993 May 30, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 24 additions & 4 deletions notdiamond/llms/providers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,12 @@ class NDLLMProviders(Enum):
GPT_4o_MINI_2024_07_18 (NDLLMProvider): refers to 'gpt-4o-mini-2024-07-18' model by OpenAI
GPT_4o_MINI (NDLLMProvider): refers to 'gpt-4o-mini' model by OpenAI
GPT_4_0125_PREVIEW (NDLLMProvider): refers to 'gpt-4-0125-preview' model by OpenAI
GPT_4_5_PREVIEW (NDLLMProvider): refers to 'gpt-4.5-preview' model by OpenAI
GPT_4_5_PREVIEW_2025_02_27 (NDLLMProvider): refers to 'gpt-4.5-preview-2025-02-27' model by OpenAI
GPT_4_1 (NDLLMProvider): refers to 'gpt-4.1' model by OpenAI
GPT_4_1_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-2025-04-14' model by OpenAI
GPT_4_1_MINI (NDLLMProvider): refers to 'gpt-4.1-mini' model by OpenAI
GPT_4_1_MINI_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-mini-2025-04-14' model by OpenAI
GPT_4_1_NANO (NDLLMProvider): refers to 'gpt-4.1-nano' model by OpenAI
GPT_4_1_NANO_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-nano-2025-04-14' model by OpenAI
O1_PREVIEW (NDLLMProvider): refers to 'o1-preview' model by OpenAI
O1_PREVIEW_2024_09_12 (NDLLMProvider): refers to 'o1-preview-2024-09-12' model by OpenAI
O1_MINI (NDLLMProvider): refers to 'o1-mini' model by OpenAI
Expand All @@ -38,6 +42,10 @@ class NDLLMProviders(Enum):
CLAUDE_3_7_SONNET_20250219 (NDLLMProvider): refers to 'claude-3-7-sonnet-20250219' model by Anthropic
CLAUDE_3_5_HAIKU_20241022 (NDLLMProvider): refers to 'claude-3-5-haiku-20241022' model by Anthropic
CLAUDE_3_HAIKU_20240307 (NDLLMProvider): refers to 'claude-3-haiku-20240307' model by Anthropic
CLAUDE_OPUS_4_20250514 (NDLLMProvider): refers to 'claude-opus-4-20250514' model by Anthropic
CLAUDE_SONNET_4_20250514 (NDLLMProvider): refers to 'claude-sonnet-4-20250514' model by Anthropic
CLAUDE_OPUS_4_0 (NDLLMProvider): refers to 'claude-opus-4-0' model by Anthropic
CLAUDE_SONNET_4_0 (NDLLMProvider): refers to 'claude-sonnet-4-0' model by Anthropic

GEMINI_PRO (NDLLMProvider): refers to 'gemini-pro' model by Google
GEMINI_1_PRO_LATEST (NDLLMProvider): refers to 'gemini-1.0-pro-latest' model by Google
Expand All @@ -46,6 +54,8 @@ class NDLLMProviders(Enum):
GEMINI_15_FLASH_LATEST (NDLLMProvider): refers to 'gemini-1.5-flash-latest' model by Google
GEMINI_20_FLASH (NDLLMProvider): refers to 'gemini-20-flash' model by Google
GEMINI_20_FLASH_001 (NDLLMProvider): refers to 'gemini-20-flash-001' model by Google
GEMINI_25_FLASH (NDLLMProvider): refers to 'gemini-25-flash' model by Google
GEMINI_25_PRO (NDLLMProvider): refers to 'gemini-25-pro' model by Google

COMMAND_R (NDLLMProvider): refers to 'command-r' model by Cohere
COMMAND_R_PLUS (NDLLMProvider): refers to 'command-r-plus' model by Cohere
Expand Down Expand Up @@ -99,8 +109,12 @@ class NDLLMProviders(Enum):
GPT_4o_MINI_2024_07_18 = ("openai", "gpt-4o-mini-2024-07-18")
GPT_4o_MINI = ("openai", "gpt-4o-mini")
GPT_4_0125_PREVIEW = ("openai", "gpt-4-0125-preview")
GPT_4_5_PREVIEW = ("openai", "gpt-4.5-preview")
GPT_4_5_PREVIEW_2025_02_27 = ("openai", "gpt-4.5-preview-2025-02-27")
GPT_4_1 = ("openai", "gpt-4.1")
GPT_4_1_2025_04_14 = ("openai", "gpt-4.1-2025-04-14")
GPT_4_1_MINI = ("openai", "gpt-4.1-mini")
GPT_4_1_MINI_2025_04_14 = ("openai", "gpt-4.1-mini-2025-04-14")
GPT_4_1_NANO = ("openai", "gpt-4.1-nano")
GPT_4_1_NANO_2025_04_14 = ("openai", "gpt-4.1-nano-2025-04-14")
O1_PREVIEW = ("openai", "o1-preview")
O1_PREVIEW_2024_09_12 = ("openai", "o1-preview-2024-09-12")
O1_MINI = ("openai", "o1-mini")
Expand All @@ -117,6 +131,10 @@ class NDLLMProviders(Enum):
CLAUDE_3_7_SONNET_20250219 = ("anthropic", "claude-3-7-sonnet-20250219")
CLAUDE_3_5_HAIKU_20241022 = ("anthropic", "claude-3-5-haiku-20241022")
CLAUDE_3_HAIKU_20240307 = ("anthropic", "claude-3-haiku-20240307")
CLAUDE_OPUS_4_20250514 = ("anthropic", "claude-opus-4-20250514")
CLAUDE_SONNET_4_20250514 = ("anthropic", "claude-sonnet-4-20250514")
CLAUDE_OPUS_4_0 = ("anthropic", "claude-opus-4-0")
CLAUDE_SONNET_4_0 = ("anthropic", "claude-sonnet-4-0")

GEMINI_PRO = ("google", "gemini-pro")
GEMINI_1_PRO_LATEST = ("google", "gemini-1.0-pro-latest")
Expand All @@ -125,6 +143,8 @@ class NDLLMProviders(Enum):
GEMINI_15_FLASH_LATEST = ("google", "gemini-1.5-flash-latest")
GEMINI_20_FLASH = ("google", "gemini-2.0-flash")
GEMINI_20_FLASH_001 = ("google", "gemini-2.0-flash-001")
GEMINI_25_FLASH = ("google", "gemini-2.5-flash")
GEMINI_25_PRO = ("google", "gemini-2.5-pro")

COMMAND_R = ("cohere", "command-r")
COMMAND_R_PLUS = ("cohere", "command-r-plus")
Expand Down
66 changes: 56 additions & 10 deletions notdiamond/settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,8 +39,12 @@
"gpt-4-turbo-preview",
"gpt-4-0125-preview",
"gpt-4-1106-preview",
"gpt-4.5-preview",
"gpt-4.5-preview-2025-02-27",
"gpt-4.1",
"gpt-4.1-2025-04-14",
"gpt-4.1-mini",
"gpt-4.1-mini-2025-04-14",
"gpt-4.1-nano",
"gpt-4.1-nano-2025-04-14",
"o1-preview",
"o1-preview-2024-09-12",
"o1-mini",
Expand All @@ -63,8 +67,12 @@
"gpt-4-turbo-preview",
"gpt-4-0125-preview",
"gpt-4-1106-preview",
"gpt-4.5-preview",
"gpt-4.5-preview-2025-02-27",
"gpt-4.1",
"gpt-4.1-2025-04-14",
"gpt-4.1-mini",
"gpt-4.1-mini-2025-04-14",
"gpt-4.1-nano",
"gpt-4.1-nano-2025-04-14",
],
"support_response_model": [
"gpt-3.5-turbo",
Expand All @@ -86,8 +94,12 @@
"o1-mini",
"o1-mini-2024-09-12",
"chatgpt-4o-latest",
"gpt-4.5-preview",
"gpt-4.5-preview-2025-02-27",
"gpt-4.1",
"gpt-4.1-2025-04-14",
"gpt-4.1-mini",
"gpt-4.1-mini-2025-04-14",
"gpt-4.1-nano",
"gpt-4.1-nano-2025-04-14",
],
"openrouter_identifier": {
"gpt-3.5-turbo": "openai/gpt-3.5-turbo",
Expand All @@ -105,8 +117,12 @@
"o1-mini": "openai/o1-mini",
"o1-mini-2024-09-12": "openai/o1-mini-2024-09-12",
"chatgpt-4o-latest": "openai/chatgpt-4o-latest",
"gpt-4.5-preview": "openai/gpt-4.5-preview",
"gpt-4.5-preview-2025-02-27": "openai/gpt-4.5-preview-2025-02-27",
"gpt-4.1": "openai/gpt-4.1",
"gpt-4.1-2025-04-14": "openai/gpt-4.1-2025-04-14",
"gpt-4.1-mini": "openai/gpt-4.1-mini",
"gpt-4.1-mini-2025-04-14": "openai/gpt-4.1-mini-2025-04-14",
"gpt-4.1-nano": "openai/gpt-4.1-nano",
"gpt-4.1-nano-2025-04-14": "openai/gpt-4.1-nano-2025-04-14",
},
"price": {
"gpt-3.5-turbo": {"input": 0.5, "output": 1.5},
Expand All @@ -128,8 +144,12 @@
"o1-mini": {"input": 3.0, "output": 12.0},
"o1-mini-2024-09-12": {"input": 3.0, "output": 12.0},
"chatgpt-4o-latest": {"input": 5.0, "output": 15.0},
"gpt-4.5-preview": {"input": 75.0, "output": 150.0},
"gpt-4.5-preview-2025-02-27": {"input": 75.0, "output": 150.0},
"gpt-4.1": {"input": 2.0, "output": 8.0},
"gpt-4.1-mini": {"input": 0.5, "output": 1.6},
"gpt-4.1-nano": {"input": 0.1, "output": 0.4},
"gpt-4.1-2025-04-14": {"input": 2.0, "output": 8.0},
"gpt-4.1-mini-2025-04-14": {"input": 0.5, "output": 1.6},
"gpt-4.1-nano-2025-04-14": {"input": 0.1, "output": 0.4},
},
},
"anthropic": {
Expand All @@ -144,6 +164,10 @@
"claude-3-5-sonnet-latest",
"claude-3-7-sonnet-latest",
"claude-3-7-sonnet-20250219",
"claude-opus-4-20250514",
"claude-sonnet-4-20250514",
"claude-opus-4-0",
"claude-sonnet-4-0",
],
"api_key": ANTHROPIC_API_KEY,
"support_tools": [
Expand All @@ -156,6 +180,10 @@
"claude-3-5-sonnet-latest",
"claude-3-7-sonnet-latest",
"claude-3-7-sonnet-20250219",
"claude-opus-4-20250514",
"claude-sonnet-4-20250514",
"claude-opus-4-0",
"claude-sonnet-4-0",
],
"support_response_model": [
"claude-2.1",
Expand All @@ -171,6 +199,10 @@
"claude-3-5-haiku-20241022": "anthropic/claude-3.5-haiku",
"claude-3-7-sonnet-latest": "anthropic/claude-3.7-sonnet",
"claude-3-7-sonnet-20250219": "anthropic/claude-3.7-sonnet",
"claude-opus-4-20250514": "anthropic/claude-opus-4",
"claude-sonnet-4-20250514": "anthropic/claude-sonnet-4",
"claude-opus-4-0": "anthropic/claude-opus-4",
"claude-sonnet-4-0": "anthropic/claude-sonnet-4",
},
"price": {
"claude-2.1": {"input": 8.0, "output": 24.0},
Expand All @@ -183,6 +215,10 @@
"claude-3-5-sonnet-latest": {"input": 3.0, "output": 15.0},
"claude-3-7-sonnet-latest": {"input": 3.0, "output": 15.0},
"claude-3-7-sonnet-20250219": {"input": 3.0, "output": 15.0},
"claude-opus-4-20250514": {"input": 15.0, "output": 75.0},
"claude-sonnet-4-20250514": {"input": 3.0, "output": 15.0},
"claude-opus-4-0": {"input": 15.0, "output": 75.0},
"claude-sonnet-4-0": {"input": 3.0, "output": 15.0},
},
},
"google": {
Expand All @@ -194,6 +230,8 @@
"gemini-1.5-flash-latest",
"gemini-2.0-flash",
"gemini-2.0-flash-001",
"gemini-2.5-flash",
"gemini-2.5-pro",
],
"api_key": GOOGLE_API_KEY,
"support_tools": [
Expand All @@ -202,6 +240,8 @@
"gemini-1.5-pro-latest",
"gemini-1.5-pro-exp-0801",
"gemini-1.5-flash-latest",
"gemini-2.5-flash",
"gemini-2.5-pro",
],
"support_response_model": [
"gemini-pro",
Expand All @@ -211,6 +251,8 @@
"gemini-1.5-flash-latest",
"gemini-2.0-flash",
"gemini-2.0-flash-001",
"gemini-2.5-flash",
"gemini-2.5-pro",
],
"openrouter_identifier": {
"gemini-pro": "google/gemini-pro",
Expand All @@ -220,6 +262,8 @@
"gemini-1.5-flash-latest": "google/gemini-flash-1.5", #
"gemini-2.0-flash": "google/gemini-2.0-flash", #
"gemini-2.0-flash-001": "google/gemini-2.0-flash", #
"gemini-2.5-flash": "google/gemini-2.5-flash", #
"gemini-2.5-pro": "google/gemini-2.5-pro", #
},
"price": {
"gemini-pro": {"input": 0.5, "output": 1.5},
Expand All @@ -229,6 +273,8 @@
"gemini-1.5-flash-latest": {"input": 0.35, "output": 1.05},
"gemini-2.0-flash": {"input": 0.10, "output": 0.40},
"gemini-2.0-flash-001": {"input": 0.10, "output": 0.40},
"gemini-2.5-flash": {"input": 0.15, "output": 0.6},
"gemini-2.5-pro": {"input": 2.0, "output": 10.0},
},
},
"cohere": {
Expand Down
Loading