Update the test case

langchain4j · Apr 16, 2024 · 13ab9bb · 13ab9bb
1 parent f7873ca
commit 13ab9bb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java b/...spring-boot-starter/src/test/java/dev/langchain4j/azure/aisearch/spring/AutoConfigIT.java
@@ -88,6 +88,8 @@ void should_provide_ai_search_retriever() {
                         Properties.PREFIX + ".content-retriever.api-key=" + AZURE_SEARCH_KEY,
                         Properties.PREFIX + ".content-retriever.endpoint=" + AZURE_SEARCH_ENDPOINT,
                         Properties.PREFIX + ".content-retriever.create-or-update-index=" + "false",
+                        Properties.PREFIX + ".content-retriever.max-results=" + "3",
+                        Properties.PREFIX + ".content-retriever.min-score=" + "0.6",
                         Properties.PREFIX + ".content-retriever.query-type=" + AzureAiSearchQueryType.VECTOR
                 ).withBean(SearchIndex.class, () -> index)
                 .withBean(EmbeddingModel.class, () -> embeddingModel)
@@ -145,6 +147,8 @@ void should_provide_ai_search_retriever() {
                         Properties.PREFIX + ".content-retriever.api-key=" + AZURE_SEARCH_KEY,
                         Properties.PREFIX + ".content-retriever.endpoint=" + AZURE_SEARCH_ENDPOINT,
                         Properties.PREFIX + ".content-retriever.create-or-update-index=" + "false",
+                        Properties.PREFIX + ".content-retriever.max-results=" + "3",
+                        Properties.PREFIX + ".content-retriever.min-score=" + "0.4",
                         Properties.PREFIX + ".content-retriever.query-type=" + AzureAiSearchQueryType.HYBRID_WITH_RERANKING
                 ).withBean(SearchIndex.class, () -> index)
                 .withBean(EmbeddingModel.class, () -> embeddingModel)