Azure OpenAI: removed a few models from ITs that I can't deploy/access

Yellow-- · Jun 24, 2024 · b31a880 · b31a880
1 parent a1e4b17
commit b31a880
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/...-azure-open-ai/src/test/java/dev/langchain4j/model/azure/AzureOpenAiEmbeddingModelIT.java b/...-azure-open-ai/src/test/java/dev/langchain4j/model/azure/AzureOpenAiEmbeddingModelIT.java
@@ -16,6 +16,7 @@
 
 import static dev.langchain4j.model.azure.AzureOpenAiEmbeddingModelName.TEXT_EMBEDDING_ADA_002;
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.params.provider.EnumSource.Mode.EXCLUDE;
 
 class AzureOpenAiEmbeddingModelIT {
 
@@ -71,7 +72,8 @@ void should_embed_in_batches() {
     }
 
     @ParameterizedTest(name = "Testing model {0}")
-    @EnumSource(AzureOpenAiEmbeddingModelName.class)
+    @EnumSource(value = AzureOpenAiEmbeddingModelName.class,
+            mode = EXCLUDE, names = "TEXT_EMBEDDING_ADA_002_2")
     void should_support_all_string_model_names(AzureOpenAiEmbeddingModelName modelName) {
 
         // given

diff --git a/...j-azure-open-ai/src/test/java/dev/langchain4j/model/azure/AzureOpenAiLanguageModelIT.java b/...j-azure-open-ai/src/test/java/dev/langchain4j/model/azure/AzureOpenAiLanguageModelIT.java
@@ -13,6 +13,7 @@
 import static dev.langchain4j.model.output.FinishReason.LENGTH;
 import static dev.langchain4j.model.output.FinishReason.STOP;
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.jupiter.params.provider.EnumSource.Mode.EXCLUDE;
 
 class AzureOpenAiLanguageModelIT {
 
@@ -57,7 +58,8 @@ void should_generate_answer_and_finish_reason_length() {
     }
 
     @ParameterizedTest(name = "Testing model {0}")
-    @EnumSource(AzureOpenAiLanguageModelName.class)
+    @EnumSource(value = AzureOpenAiLanguageModelName.class,
+            mode = EXCLUDE, names = {"TEXT_DAVINCI_002", "TEXT_DAVINCI_002_1"})
     void should_support_all_string_model_names(AzureOpenAiLanguageModelName modelName) {
 
         // given