From 8eb656d77a5d5554f1a8883b6aaa007918f5976d Mon Sep 17 00:00:00 2001
From: Raja Kolli <rajadilipkolli@gmail.com>
Date: Mon, 27 May 2024 05:55:58 +0000
Subject: [PATCH] polish : use lamda for document enricher

---
 rag/rag-springai-ollama-llm/pom.xml             | 14 +++++++-------
 .../service/DataIndexerService.java             | 17 ++++++-----------
 rag/rag-springai-openai-llm/pom.xml             |  8 ++++----
 3 files changed, 17 insertions(+), 22 deletions(-)

diff --git a/rag/rag-springai-ollama-llm/pom.xml b/rag/rag-springai-ollama-llm/pom.xml
index 0de8e89..852a367 100644
--- a/rag/rag-springai-ollama-llm/pom.xml
+++ b/rag/rag-springai-ollama-llm/pom.xml
@@ -70,10 +70,10 @@
             <scope>test</scope>
         </dependency>
         <dependency>
-			<groupId>org.springframework.ai</groupId>
-			<artifactId>spring-ai-spring-boot-testcontainers</artifactId>
-			<scope>test</scope>
-		</dependency>
+            <groupId>org.springframework.ai</groupId>
+            <artifactId>spring-ai-spring-boot-testcontainers</artifactId>
+            <scope>test</scope>
+        </dependency>
         <dependency>
             <groupId>org.testcontainers</groupId>
             <artifactId>junit-jupiter</artifactId>
@@ -127,9 +127,9 @@
                         <palantirJavaFormat>
                             <version>2.47.0</version>
                         </palantirJavaFormat>
-                        <importOrder />
-                        <removeUnusedImports />
-                        <formatAnnotations />
+                        <importOrder/>
+                        <removeUnusedImports/>
+                        <formatAnnotations/>
                     </java>
                 </configuration>
                 <executions>
diff --git a/rag/rag-springai-ollama-llm/src/main/java/com/learning/ai/llmragwithspringai/service/DataIndexerService.java b/rag/rag-springai-ollama-llm/src/main/java/com/learning/ai/llmragwithspringai/service/DataIndexerService.java
index d07c35a..e2d2ea7 100644
--- a/rag/rag-springai-ollama-llm/src/main/java/com/learning/ai/llmragwithspringai/service/DataIndexerService.java
+++ b/rag/rag-springai-ollama-llm/src/main/java/com/learning/ai/llmragwithspringai/service/DataIndexerService.java
@@ -1,11 +1,9 @@
 package com.learning.ai.llmragwithspringai.service;
 
-import java.util.List;
 import java.util.Map;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.ai.chat.prompt.transformer.TransformerContentType;
-import org.springframework.ai.document.Document;
 import org.springframework.ai.document.DocumentReader;
 import org.springframework.ai.document.DocumentTransformer;
 import org.springframework.ai.reader.ExtractedTextFormatter;
@@ -53,15 +51,12 @@ public void loadData(Resource documentResource) {
         }
         if (documentReader != null) {
             LOGGER.info("Loading text document to redis vector database");
-            var metadataEnricher = new DocumentTransformer() {
-                @Override
-                public List<Document> apply(List<Document> documents) {
-                    documents.forEach(d -> {
-                        Map<String, Object> metadata = d.getMetadata();
-                        metadata.put(TransformerContentType.EXTERNAL_KNOWLEDGE, "true");
-                    });
-                    return documents;
-                }
+            DocumentTransformer metadataEnricher = documents -> {
+                documents.forEach(d -> {
+                    Map<String, Object> metadata = d.getMetadata();
+                    metadata.put(TransformerContentType.EXTERNAL_KNOWLEDGE, "true");
+                });
+                return documents;
             };
             vectorStore.accept(metadataEnricher.apply(tokenTextSplitter.apply(documentReader.get())));
             LOGGER.info("Loaded document to redis vector database.");
diff --git a/rag/rag-springai-openai-llm/pom.xml b/rag/rag-springai-openai-llm/pom.xml
index 98cd183..53bbfd7 100644
--- a/rag/rag-springai-openai-llm/pom.xml
+++ b/rag/rag-springai-openai-llm/pom.xml
@@ -120,11 +120,11 @@
                 <configuration>
                     <java>
                         <palantirJavaFormat>
-                            <version>2.40.0</version>
+                            <version>2.47.0</version>
                         </palantirJavaFormat>
-                        <importOrder />
-                        <removeUnusedImports />
-                        <formatAnnotations />
+                        <importOrder/>
+                        <removeUnusedImports/>
+                        <formatAnnotations/>
                     </java>
                 </configuration>
                 <executions>