Replace Azure GPT-4 with Azure GPT-4o (#98)

nuclia · Jul 16, 2024 · 0cd0f32 · 0cd0f32
1 parent 9799f91
commit 0cd0f32
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/nua/e2e/regional/test_llm_generate.py b/nua/e2e/regional/test_llm_generate.py
@@ -13,7 +13,7 @@ def test_llm_generate_chatgpt(nua_config):
 
 def test_llm_generate_azure_chatgpt(nua_config):
     np = NucliaPredict()
-    generated = np.generate("Which is the capital of Catalonia?", model="chatgpt-azure")
+    generated = np.generate("Which is the capital of Catalonia?", model="chatgpt-azure-4o")
     assert "Barcelona" in generated.answer
 
 

diff --git a/nua/e2e/regional/test_llm_rag.py b/nua/e2e/regional/test_llm_rag.py
@@ -23,7 +23,7 @@ def test_llm_rag_chatgpt_azure(nua_config):
             "Nuclia CTO is Ramon Navarro",
             "Eudald Camprubí is CEO at the same company as Ramon Navarro",
         ],
-        model="chatgpt-azure",
+        model="chatgpt-azure-4o",
     )
     assert "Eudald" in generated.answer
 

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