feat : enable metadata filtering

rajadilipkolli · Jul 6, 2024 · 6848740 · 6848740
1 parent 8e8b314
commit 6848740
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/...ores/pgvector-langchain4j/src/main/java/com/learning/ai/service/PgVectorStoreService.java b/...ores/pgvector-langchain4j/src/main/java/com/learning/ai/service/PgVectorStoreService.java
@@ -37,7 +37,7 @@ public AIChatResponse queryEmbeddingStore(String question, Integer userId) {
         }
         EmbeddingSearchRequest embeddingSearchRequest = embeddingSearchRequestBuilder.build();
         EmbeddingSearchResult<TextSegment> relevant = embeddingStore.search(embeddingSearchRequest);
-        EmbeddingMatch<TextSegment> embeddingMatch = relevant.matches().get(0);
+        EmbeddingMatch<TextSegment> embeddingMatch = relevant.matches().getFirst();
 
         LOGGER.info("Score : {}", embeddingMatch.score()); // 0.8144288608390052
         String answer = embeddingMatch.embedded().text();

diff --git a/...es/pgvector-langchain4j/src/test/java/com/learning/ai/controller/TestQueryController.java b/...es/pgvector-langchain4j/src/test/java/com/learning/ai/controller/TestQueryController.java
@@ -6,7 +6,6 @@
 
 import com.learning.ai.config.AbstractIntegrationTest;
 import org.hamcrest.Matchers;
-import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
 
 class TestQueryController extends AbstractIntegrationTest {
@@ -21,7 +20,6 @@ void queryEmbeddedStore() throws Exception {
     }
 
     @Test
-    @Disabled("Fixed in later version of langchain4j > 0.30.0")
     void queryEmbeddedStoreWithMetadata() throws Exception {
         mockMvc.perform(get("/api/ai/query")
                         .param("question", "What is your favourite sport")