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

feat: Add new GPT-4o model and update Claude references #101

Closed
dragos199993 wants to merge 1 commit into from
Closed

feat: Add new GPT-4o model and update Claude references #101

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
6 changes: 2 additions & 4 deletions notdiamond/llms/providers.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ class NDLLMProviders(Enum):
GPT_4_TURBO_2024_04_09 (NDLLMProvider): refers to 'gpt-4-turbo-2024-04-09' model by OpenAI
GPT_4o_2024_05_13 (NDLLMProvider): refers to 'gpt-4o-2024-05-13' model by OpenAI
GPT_4o_2024_08_06 (NDLLMProvider): refers to 'gpt-4o-2024-08-06' model by OpenAI
GPT_4o_2024_11_20 (NDLLMProvider): refers to 'gpt-4o-2024-11-20' model by OpenAI
GPT_4o (NDLLMProvider): refers to 'gpt-4o' model by OpenAI
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
Expand All @@ -31,9 +32,7 @@ class NDLLMProviders(Enum):
GPT_4_1_NANO_2025_04_14 (NDLLMProvider): refers to 'gpt-4.1-nano-2025-04-14' model by OpenAI

CLAUDE_2_1 (NDLLMProvider): refers to 'claude-2.1' model by Anthropic
CLAUDE_3_OPUS_20240229 (NDLLMProvider): refers to 'claude-3-opus-20240229' model by Anthropic
CLAUDE_3_SONNET_20240229 (NDLLMProvider): refers to 'claude-3-sonnet-20240229' model by Anthropic
CLAUDE_3_5_SONNET_20240620 (NDLLMProvider): refers to 'claude-3-5-sonnet-20240620' model by Anthropic
CLAUDE_3_7_SONNET_LATEST (NDLLMProvider): refers to 'claude-3-7-sonnet-latest' model by Anthropic
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
Expand Down Expand Up @@ -103,6 +102,7 @@ class NDLLMProviders(Enum):
GPT_4_TURBO_2024_04_09 = ("openai", "gpt-4-turbo-2024-04-09")
GPT_4o_2024_05_13 = ("openai", "gpt-4o-2024-05-13")
GPT_4o_2024_08_06 = ("openai", "gpt-4o-2024-08-06")
GPT_4o_2024_11_20 = ("openai", "gpt-4o-2024-11-20")
GPT_4o = ("openai", "gpt-4o")
GPT_4o_MINI_2024_07_18 = ("openai", "gpt-4o-mini-2024-07-18")
GPT_4o_MINI = ("openai", "gpt-4o-mini")
Expand All @@ -116,9 +116,7 @@ class NDLLMProviders(Enum):
CHATGPT_4o_LATEST = ("openai", "chatgpt-4o-latest")

CLAUDE_2_1 = ("anthropic", "claude-2.1")
CLAUDE_3_OPUS_20240229 = ("anthropic", "claude-3-opus-20240229")
CLAUDE_3_SONNET_20240229 = ("anthropic", "claude-3-sonnet-20240229")
CLAUDE_3_5_SONNET_20240620 = ("anthropic", "claude-3-5-sonnet-20240620")
CLAUDE_3_5_SONNET_20241022 = ("anthropic", "claude-3-5-sonnet-20241022")
CLAUDE_3_5_SONNET_LATEST = ("anthropic", "claude-3-5-sonnet-latest")
CLAUDE_3_7_SONNET_LATEST = ("anthropic", "claude-3-7-sonnet-latest")
Expand Down
14 changes: 5 additions & 9 deletions notdiamond/settings.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@
"gpt-4o",
"gpt-4o-2024-05-13",
"gpt-4o-2024-08-06",
"gpt-4o-2024-11-20",
"gpt-4o-mini",
"gpt-4o-mini-2024-07-18",
"gpt-4-turbo-preview",
Expand All @@ -58,6 +59,7 @@
"gpt-4o",
"gpt-4o-2024-05-13",
"gpt-4o-2024-08-06",
"gpt-4o-2024-11-20",
"gpt-4o-mini",
"gpt-4o-mini-2024-07-18",
"gpt-4-turbo-preview",
Expand All @@ -80,6 +82,7 @@
"gpt-4o",
"gpt-4o-2024-05-13",
"gpt-4o-2024-08-06",
"gpt-4o-2024-11-20",
"gpt-4o-mini",
"gpt-4o-mini-2024-07-18",
"gpt-4-turbo-preview",
Expand All @@ -100,6 +103,7 @@
"gpt-4o": "openai/gpt-4o",
"gpt-4o-2024-05-13": "openai/gpt-4o-2024-05-13",
"gpt-4o-2024-08-06": "openai/gpt-4o-2024-08-06",
"gpt-4o-2024-11-20": "openai/gpt-4o-2024-11-20",
"gpt-4o-mini": "openai/gpt-4o-mini",
"gpt-4o-mini-2024-07-18": "openai/gpt-4o-mini-2024-07-18",
"gpt-4-turbo-preview": "openai/gpt-4-turbo-preview",
Expand All @@ -122,6 +126,7 @@
"gpt-4o": {"input": 5.0, "output": 15.0},
"gpt-4o-2024-05-13": {"input": 5.0, "output": 15.0},
"gpt-4o-2024-08-06": {"input": 2.5, "output": 10.0},
"gpt-4o-2024-11-20": {"input": 2.5, "output": 10.0},
"gpt-4o-mini": {"input": 0.15, "output": 0.6},
"gpt-4o-mini-2024-07-18": {"input": 0.15, "output": 0.6},
"gpt-4-turbo-preview": {"input": 10.0, "output": 30.0},
Expand All @@ -139,11 +144,9 @@
"anthropic": {
"models": [
"claude-2.1",
"claude-3-opus-20240229",
"claude-3-sonnet-20240229",
"claude-3-haiku-20240307",
"claude-3-5-haiku-20241022",
"claude-3-5-sonnet-20240620",
"claude-3-5-sonnet-20241022",
"claude-3-5-sonnet-latest",
"claude-3-7-sonnet-latest",
Expand All @@ -157,10 +160,8 @@
],
"api_key": ANTHROPIC_API_KEY,
"support_tools": [
"claude-3-opus-20240229",
"claude-3-sonnet-20240229",
"claude-3-haiku-20240307",
"claude-3-5-sonnet-20240620",
"claude-3-5-sonnet-20241022",
"claude-3-5-haiku-20241022",
"claude-3-5-sonnet-latest",
Expand All @@ -175,14 +176,11 @@
],
"support_response_model": [
"claude-2.1",
"claude-3-opus-20240229",
],
"openrouter_identifier": {
"claude-2.1": "anthropic/claude-2.1",
"claude-3-opus-20240229": "anthropic/claude-3-opus",
"claude-3-sonnet-20240229": "anthropic/claude-3-sonnet",
"claude-3-haiku-20240307": "anthropic/claude-3-haiku",
"claude-3-5-sonnet-20240620": "anthropic/claude-3.5-sonnet-20240620",
"claude-3-5-sonnet-latest": "anthropic/claude-3.5-sonnet",
"claude-3-5-haiku-20241022": "anthropic/claude-3.5-haiku",
"claude-3-7-sonnet-latest": "anthropic/claude-3.7-sonnet",
Expand All @@ -196,11 +194,9 @@
},
"price": {
"claude-2.1": {"input": 8.0, "output": 24.0},
"claude-3-opus-20240229": {"input": 15.0, "output": 75.0},
"claude-3-sonnet-20240229": {"input": 3.0, "output": 15.0},
"claude-3-haiku-20240307": {"input": 0.25, "output": 1.25},
"claude-3-5-haiku-20241022": {"input": 1.0, "output": 5.0},
"claude-3-5-sonnet-20240620": {"input": 3.0, "output": 15.0},
"claude-3-5-sonnet-20241022": {"input": 3.0, "output": 15.0},
"claude-3-5-sonnet-latest": {"input": 3.0, "output": 15.0},
"claude-3-7-sonnet-latest": {"input": 3.0, "output": 15.0},
Expand Down
4 changes: 2 additions & 2 deletions tests/test_components/test_llms/test_provider.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -53,9 +53,9 @@ def test_existing_openrouter_model():
assert llm_provider.openrouter_model == "mistralai/mistral-large"

llm_provider = LLMConfig(
provider="anthropic", model="claude-3-opus-20240229"
provider="anthropic", model="claude-sonnet-4-0"
)
assert llm_provider.openrouter_model == "anthropic/claude-3-opus"
assert llm_provider.openrouter_model == "anthropic/claude-sonnet-4"


def test_not_existing_openrouter_model():
Expand Down
Loading
Loading