wanjia9506
/
langchain4j
mirror of https://github.com/langchain4j/langchain4j.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

stabilize Azure AI Search tests

This commit is contained in:
LangChain4j 2024-08-24 15:10:25 +02:00
parent a2f3432d17
commit d5bc868854
4 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain4j-azure-ai-search/src/test/java/dev/langchain4j/rag/content/retriever/azure/search/AzureAiSearchContentRetrieverIT.java
View File

@ -14,6 +14,7 @@ import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.EmbeddingStoreWithFilteringIT;
import org.awaitility.core.ThrowingRunnable;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.slf4j.Logger;
@ -83,8 +84,9 @@ public class AzureAiSearchContentRetrieverIT extends EmbeddingStoreWithFiltering
}
@BeforeEach
void setUp() {
void beforeEach() throws InterruptedException {
clearStore();
Thread.sleep(2_000);
}
@Test
@ -122,6 +124,7 @@ public class AzureAiSearchContentRetrieverIT extends EmbeddingStoreWithFiltering
}
@Test
@Disabled("no semantic ranker in a free Azure tier")
void testAllTypesOfSearch() {
String content1 = "This book is about politics";
String content2 = "Cats sleeps a lot.";
@ -261,6 +264,7 @@ public class AzureAiSearchContentRetrieverIT extends EmbeddingStoreWithFiltering
}
@Test
@Disabled("no semantic ranker in a free Azure tier")
void testAddEmbeddingsAndRetrieveRelevantWithHybridAndReranking() {
String content1 = "Albert Camus (7 November 1913 – 4 January 1960) was a French philosopher, author, dramatist, journalist, world federalist, and political activist. He was the recipient of the 1957 Nobel Prize in Literature at the age of 44, the second-youngest recipient in history. His works include The Stranger, The Plague, The Myth of Sisyphus, The Fall, and The Rebel.\n" +
"\n" +

6
langchain4j-azure-ai-search/src/test/java/dev/langchain4j/rag/content/retriever/azure/search/AzureAiSearchContentRetrieverRemovalIT.java
View File

@ -7,6 +7,7 @@ import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.EmbeddingStoreWithRemovalIT;
import org.awaitility.core.ThrowingRunnable;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@ -40,6 +41,11 @@ public class AzureAiSearchContentRetrieverRemovalIT extends EmbeddingStoreWithRe
return this.embeddingModel;
}
@BeforeEach
void beforeEach() throws InterruptedException {
Thread.sleep(2_000);
}
@AfterEach
void afterEach() {
try {

6
langchain4j-azure-ai-search/src/test/java/dev/langchain4j/store/embedding/azure/search/AzureAiSearchEmbeddingStoreIT.java
View File

@ -13,6 +13,7 @@ import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.EmbeddingStoreWithFilteringIT;
import org.awaitility.core.ThrowingRunnable;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.slf4j.Logger;
@ -54,6 +55,11 @@ public class AzureAiSearchEmbeddingStoreIT extends EmbeddingStoreWithFilteringIT
return embeddingModel;
}
@BeforeEach
void beforeEach() throws InterruptedException {
Thread.sleep(2_000);
}
@AfterEach
void afterEach() {
try {

6
langchain4j-azure-ai-search/src/test/java/dev/langchain4j/store/embedding/azure/search/AzureAiSearchEmbeddingStoreRemovalIT.java
View File

@ -7,6 +7,7 @@ import dev.langchain4j.store.embedding.EmbeddingStore;
import dev.langchain4j.store.embedding.EmbeddingStoreWithRemovalIT;
import org.awaitility.core.ThrowingRunnable;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@ -37,6 +38,11 @@ public class AzureAiSearchEmbeddingStoreRemovalIT extends EmbeddingStoreWithRemo
return embeddingModel;
}
@BeforeEach
void beforeEach() throws InterruptedException {
Thread.sleep(2_000);
}
@AfterEach
void afterEach() {
try {