Add Chunk count

Author: johnsonr

src/main/java/com/embabel/guide/GuideData.java

@@ -1,7 +1,7 @@
 package com.embabel.guide;
 
 import com.embabel.agent.api.common.LlmReference;
-import com.embabel.agent.rag.WritableRagService;
+import com.embabel.agent.rag.RepositoryRagService;
 import com.embabel.agent.rag.ingestion.DirectoryParsingResult;
 import com.embabel.agent.rag.ingestion.HierarchicalContentReader;
 import com.embabel.agent.rag.tools.RagOptions;
@@ -13,6 +13,7 @@
 import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.PlatformTransactionManager;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.transaction.support.TransactionTemplate;
 
 import java.util.*;
@@ -29,12 +30,12 @@ public class GuideData {
 
     public final List<LlmReference> references = new LinkedList<>();
 
-    private final WritableRagService ragService;
+    private final RepositoryRagService ragService;
 
     private final PlatformTransactionManager platformTransactionManager;
 
     public GuideData(
-            WritableRagService ragService,
+            RepositoryRagService ragService,
             GuideConfig guideConfig,
             PlatformTransactionManager platformTransactionManager) {
         this.ragService = ragService;
@@ -58,11 +59,15 @@ public void provisionDatabase() {
         ragService.provision();
     }
 
+    @Transactional(readOnly = true)
+    public int count() {
+        return ragService.count();
+    }
+
     /**
      * Read all files under this directory
      *
      * @param dir absolute path
-     *            =
      */
     public DirectoryParsingResult readContent(String dir) {
         ragService.provision();

src/main/java/com/embabel/guide/GuideShell.java

@@ -1,7 +1,6 @@
 package com.embabel.guide;
 
 import com.embabel.agent.event.logging.personality.severance.LumonColorPalette;
-import com.embabel.agent.rag.ingestion.ContentElementRepository;
 import com.embabel.agent.shell.TerminalServices;
 import com.embabel.chat.Chatbot;
 import org.springframework.shell.standard.ShellComponent;
@@ -13,8 +12,7 @@
 public record GuideShell(
         TerminalServices terminalServices,
         Chatbot chatbot,
-        GuideData guideData,
-        ContentElementRepository contentElementRepository) {
+        GuideData guideData) {
 
     @ShellMethod("talk to docs")
     public String talk() {
@@ -32,16 +30,14 @@ public String loadDocs() {
         return "Loaded docs: " + directoryParsingResult;
     }
 
-    @ShellMethod("provision Neo")
+    @ShellMethod("provision database")
     public String provision() {
         guideData.provisionDatabase();
         return "Database provisioned";
     }
 
-//    @ShellMethod("show chunks")
-//    public String chunks() {
-//        return (contentElementRepository.findAll().stream()
-//                .map(chunk -> chunk.getId() + ": " + chunk.getText())
-//                .reduce("", (a, b) -> a + "\n" + "*".repeat(80) + "\n" + b);
-//    }
+    @ShellMethod("show chunks")
+    public String chunks() {
+        return guideData.count() + " chunks in the database";
+    }
 }