Try release of LC4j

Author: clun

langchain4j-astradb/src/test/java/dev/langchain4j/store/embedding/astradb/GettingStartedGuideTestIT.java

@@ -139,15 +139,15 @@ public void should_search_content_retriever() {
 
             ContentRetriever contentRetriever = EmbeddingStoreContentRetriever.builder()
                     .embeddingStore(storeJohnny)
-                    //.embeddingModel()
+                    .embeddingModel(AstraDBTestSupport.createopenAIEmbeddingModel(OpenAiEmbeddingModelName.TEXT_EMBEDDING_3_SMALL))
                     .maxResults(2)
                     .minScore(0.5)
                     .build();
 
             // configuring it to use the components we've created above.
             Assistant ai = AiServices.builder(Assistant.class)
                     .contentRetriever(contentRetriever)
-                    //.chatLanguageModel(getChatLanguageModelChatBison())
+                    .chatLanguageModel(AstraDBTestSupport.createOpenAIChatLanguageModel(OpenAiChatModelName.GPT_4_O))
                     //.chatMemory(MessageWindowChatMemory.withMaxMessages(10))
                     .build();
 

langchain4j-astradb/src/test/java/dev/langchain4j/store/embedding/astradb/GettingStartedGuideVectorizedTestIT.java

@@ -3,7 +3,9 @@
 import com.datastax.astra.client.Database;
 import com.datastax.astra.client.model.CollectionOptions;
 import com.datastax.astra.client.model.SimilarityMetric;
+import com.datastax.astra.langchain4j.Assistant;
 import com.datastax.astra.langchain4j.AstraDBTestSupport;
+import com.datastax.astra.langchain4j.rag.AstraVectorizeContentRetriever;
 import com.datastax.astra.langchain4j.rag.AstraVectorizeIngestor;
 import com.datastax.astra.langchain4j.store.embedding.AstraDbEmbeddingStore;
 import com.datastax.astra.langchain4j.store.embedding.EmbeddingSearchRequestAstra;
@@ -12,6 +14,13 @@
 import dev.langchain4j.data.document.parser.TextDocumentParser;
 import dev.langchain4j.data.document.splitter.DocumentSplitters;
 import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.openai.OpenAiChatModelName;
+import dev.langchain4j.model.openai.OpenAiEmbeddingModelName;
+import dev.langchain4j.rag.DefaultRetrievalAugmentor;
+import dev.langchain4j.rag.RetrievalAugmentor;
+import dev.langchain4j.rag.content.retriever.ContentRetriever;
+import dev.langchain4j.rag.content.retriever.EmbeddingStoreContentRetriever;
+import dev.langchain4j.service.AiServices;
 import dev.langchain4j.store.embedding.EmbeddingMatch;
 import lombok.extern.slf4j.Slf4j;
 import org.junit.jupiter.api.BeforeAll;
@@ -81,6 +90,23 @@ public void should_ingest_documents() {
 
     @Test
     @Order(2)
+    public void should_search_with_content_retriever() {
+        AstraVectorizeContentRetriever contentRetriever = AstraVectorizeContentRetriever.builder()
+                .embeddingStore(embeddingStoreVectorizeNVidia)
+                .maxResults(2)
+                .minScore(0.5)
+                .build();
+        // configuring it to use the components we've created above.
+        Assistant ai = AiServices.builder(Assistant.class)
+                .contentRetriever(contentRetriever)
+                .chatLanguageModel(AstraDBTestSupport.createOpenAIChatLanguageModel(OpenAiChatModelName.GPT_4_O))
+                .build();
+        String response = ai.answer("Who is Johnny?");
+        System.out.println(response);
+    }
+
+    @Test
+    @Order(3)
     public void should_search_results() {
         String question = "Who is Johnny?";
 
@@ -107,4 +133,14 @@ public void should_search_results() {
 
         System.out.println(ragContext);
     }
+
+    @Test
+    @Order(4)
+    public void should_search_withResultAggregator() {
+        String question = "Who is Johnny?";
+        // Our guy for advanced RAG
+        RetrievalAugmentor retrievalAugmentor = DefaultRetrievalAugmentor.builder()
+                .contentRetriever(AstraVectorizeContentRetriever.from(embeddingStoreVectorizeNVidia))
+                .build();
+    }
 }