Expose RAG on Ai

Author: johnsonr

embabel-agent-api/src/main/kotlin/com/embabel/agent/api/common/Ai.kt

@@ -15,6 +15,7 @@
  */
 package com.embabel.agent.api.common
 
+import com.embabel.agent.rag.RagService
 import com.embabel.common.ai.model.*
 import org.springframework.ai.embedding.EmbeddingModel
 
@@ -27,14 +28,24 @@ typealias Embedding = FloatArray
  */
 interface Ai {
 
+    /**
+     * Return an embedding model with the given name
+     */
     fun withEmbeddingModel(model: String): EmbeddingModel =
         withEmbeddingModel(ModelSelectionCriteria.byName(model))
 
+    /**
+     * Return an embedding model matching the given criteria.
+     */
     fun withEmbeddingModel(criteria: ModelSelectionCriteria): EmbeddingModel
 
     fun withDefaultEmbeddingModel(): EmbeddingModel =
         withEmbeddingModel(DefaultModelSelectionCriteria)
 
+    /**
+     * Return the RagService
+     */
+    fun rag(): RagService
 
     /**
      * Get a configurable PromptRunner for this context using

embabel-agent-api/src/main/kotlin/com/embabel/agent/api/common/OperationContext.kt

@@ -21,6 +21,7 @@ import com.embabel.agent.api.dsl.AgentScopeBuilder
 import com.embabel.agent.core.*
 import com.embabel.agent.event.AgenticEventListener
 import com.embabel.agent.prompt.element.ContextualPromptElement
+import com.embabel.agent.rag.RagService
 import com.embabel.common.ai.model.LlmOptions
 import com.embabel.common.ai.model.ModelSelectionCriteria
 import com.embabel.common.ai.prompt.CurrentDate
@@ -232,4 +233,8 @@ private class OperationContextAi(
     override fun withLlm(llm: LlmOptions): PromptRunner {
         return context.promptRunner().withLlm(llm)
     }
+
+    override fun rag(): RagService {
+        return context.processContext.platformServices.ragService
+    }
 }

embabel-agent-rag-lucene/pom.xml

@@ -7,7 +7,7 @@
         <artifactId>embabel-agent-parent</artifactId>
         <version>0.1.2-SNAPSHOT</version>
     </parent>
-    <artifactId>embabel-agent-lucene</artifactId>
+    <artifactId>embabel-agent-rag-lucene</artifactId>
     <packaging>jar</packaging>
     <name>Embabel Agent Rag for Lucene</name>
     <description>Embabel Agent Rag for Lucene</description>
@@ -34,26 +34,26 @@
             <artifactId>lucene-core</artifactId>
             <version>9.11.1</version>
         </dependency>
-        
+
         <dependency>
             <groupId>org.apache.lucene</groupId>
             <artifactId>lucene-analysis-common</artifactId>
             <version>9.11.1</version>
         </dependency>
-        
+
         <dependency>
             <groupId>org.apache.lucene</groupId>
             <artifactId>lucene-queryparser</artifactId>
             <version>9.11.1</version>
         </dependency>
-        
+
         <dependency>
             <groupId>org.apache.lucene</groupId>
             <artifactId>lucene-memory</artifactId>
             <version>9.11.1</version>
         </dependency>
-        
-        
+
+
     </dependencies>
 
     <build>

embabel-agent-rag-lucene/src/main/kotlin/com/embabel/agent/rag/lucene/LuceneRagService.kt

@@ -37,7 +37,7 @@ import java.io.Closeable
 import kotlin.math.sqrt
 import org.springframework.ai.document.Document as SpringAiDocument
 
-class LuceneRagService(
+class LuceneRagService @JvmOverloads constructor(
     override val name: String = "lucene-rag",
     override val description: String = "In-memory Lucene-based RAG service with hybrid text and vector search capabilities",
     private val embeddingModel: EmbeddingModel? = null,