update models

nuclia · Dec 19, 2024 · a8cf8ad · a8cf8ad
1 parent 6e09473
commit a8cf8ad
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nua/e2e/regional/models.py b/nua/e2e/regional/models.py
@@ -1,16 +1,17 @@
 ALL_LLMS = [  # Extracted from learning/libraries/learning_models/generative.py
     "chatgpt-azure-4o",
     "chatgpt-azure-4-turbo",
+    "chatgpt-azure-4o-mini",
     "claude-3",
     "claude-3-fast",
     "claude-3-5-fast",
+    "claude-3-5-small",
     "gemini-1-5-pro",
     "gemini-1-5-pro-vision",
     "gemini-1-5-flash",
     "gemini-1-5-flash-vision",
     "mistral",
     "azure-mistral",
-    "chatgpt-vision",
     "chatgpt4",
     "chatgpt4o",
     "chatgpt4o-mini",
@@ -31,12 +32,15 @@
 LLM_WITH_JSON_OUTPUT_SUPPORT = [
     "chatgpt-azure-4-turbo",
     "chatgpt-azure-4o",
+    "chatgpt-azure-4o-mini",
     "claude-3",
     "claude-3-fast",
     "claude-3-5-fast",
+    "claude-3-5-small",
     "chatgpt4",
     "chatgpt4o",
     "chatgpt4o-mini",
     "gemini-1-5-pro",
+    "gemini-1-5-flash",
     "azure-mistral",
 ]