remove chatgpt-azure-3" (#117)

nuclia · Nov 12, 2024 · 187ae03 · 187ae03
1 parent 1a1945c
commit 187ae03
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/nua/e2e/regional/models.py b/nua/e2e/regional/models.py
@@ -1,7 +1,6 @@
 ALL_LLMS = [  # Extracted from learning/libraries/learning_models/generative.py
     "chatgpt-azure-4o",
     "chatgpt-azure-4-turbo",
-    "chatgpt-azure-3",
     "claude-3",
     "claude-3-fast",
     "claude-3-5-fast",

diff --git a/nua/e2e/regional/test_llm_summarize.py b/nua/e2e/regional/test_llm_summarize.py
@@ -27,7 +27,7 @@
 
 def test_summarize_chatgpt(nua_config):
     np = NucliaPredict()
-    embed = np.summarize(DATA, model="chatgpt-azure-3")
+    embed = np.summarize(DATA, model="chatgpt4o")
     assert "Manresa" in embed.summary
     assert "Barcelona" in embed.summary