Skip to content

Commit 849e5e0

Browse files
Update generative models (#152)
* Update generative models * Update models.py
1 parent 9f305f4 commit 849e5e0

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

nuclia_e2e/nuclia_e2e/models.py

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
"gemini-1-5-flash",
2929
# "gemini-1-5-flash-vision", EXCLUDED because it shares same implementation as non-vision
3030
"mistral",
31-
"azure-mistral",
3231
"chatgpt4",
3332
"chatgpt4o",
3433
"chatgpt4o-mini",
@@ -42,6 +41,8 @@
4241
"deepseek-chat",
4342
# "azure-deepseek-r1", EXCLUDED as it is too slow
4443
"azure-mistral-large-2",
44+
"gcp-claude-3-5-sonnet-v2",
45+
"gcp-claude-3-7-sonnet",
4546
]
4647

4748

@@ -75,6 +76,10 @@ def is_reasoning_model(model: str) -> bool:
7576
"chatgpt-azure-4-turbo",
7677
"chatgpt-azure-4o",
7778
"chatgpt-azure-4o-mini",
79+
# "chatgpt-azure-o1-preview",
80+
# "chatgpt-azure-o1-mini",
81+
# "chatgpt-azure-o1", EXCLUDED because it almost always fails with timeouts
82+
"chatgpt-azure-o3-mini",
7883
"claude-3",
7984
"claude-3-fast",
8085
"claude-3-5-fast",
@@ -86,12 +91,19 @@ def is_reasoning_model(model: str) -> bool:
8691
"gemini-1-5-flash",
8792
"gemini-1-5-flash-vision",
8893
# "mistral",
89-
"azure-mistral",
9094
"chatgpt4",
9195
"chatgpt4o",
9296
"chatgpt4o-mini",
9397
# "chatgpt-o1-preview", WHY EXCLUDED ?
9498
# "chatgpt-o1-mini", WHY EXCLUDED ?
95-
# "llama-3.2-90b-vision-instruct-maas"
96-
# "huggingface" # excluded because it need a key
99+
# "chatgpt-o1",
100+
"chatgpt-o3-mini",
101+
# "huggingface"
102+
"llama-3.2-90b-vision-instruct-maas",
103+
# "deepseek-reasoner",
104+
"deepseek-chat",
105+
# "azure-deepseek-r1",
106+
"azure-mistral-large-2",
107+
"gcp-claude-3-5-sonnet-v2",
108+
"gcp-claude-3-7-sonnet",
97109
]

0 commit comments

Comments
 (0)