diff --git a/langchain4j-azure-ai-search-spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java b/langchain4j-azure-ai-search-spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java index 8cef7036..00294b90 100644 --- a/langchain4j-azure-ai-search-spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java +++ b/langchain4j-azure-ai-search-spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java @@ -200,12 +200,16 @@ void should_provide_ai_search_embedding_store() { String content6 = "chess"; List contents = asList(content1, content2, content3, content4, content5, content6); + Thread.sleep(3000); + for (String content : contents) { TextSegment textSegment = TextSegment.from(content); Embedding embedding = embeddingModel.embed(content).content(); embeddingStore.add(embedding, textSegment); } - Thread.sleep(2000); + + Thread.sleep(3000); + Embedding relevantEmbedding = embeddingModel.embed("fruit").content(); List> relevant = embeddingStore.findRelevant(relevantEmbedding, 3); assertThat(relevant).hasSize(3);