Fix problems with Lucene (#106)

xebia-functional · May 25, 2023 · f32004d · f32004d
1 parent ec79dd5
commit f32004d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/integrations/lucene/build.gradle.kts b/integrations/lucene/build.gradle.kts
@@ -19,7 +19,7 @@ java {
 
 dependencies {
     implementation(projects.xefCore)
-    implementation(libs.apache.lucene)
+    api(libs.apache.lucene)
 }
 
 tasks.withType<AbstractPublishToMaven> {

diff --git a/integrations/lucene/src/main/kotlin/com/xebia/functional/xef/vectorstores/Lucene.kt b/integrations/lucene/src/main/kotlin/com/xebia/functional/xef/vectorstores/Lucene.kt
@@ -35,7 +35,7 @@ open class Lucene(
   private val requestConfig =
     RequestConfig(EmbeddingModel.TextEmbeddingAda002, RequestConfig.Companion.User("user"))
 
-  override suspend fun addTexts(texts: List<String>) =
+  override suspend fun addTexts(texts: List<String>) {
     texts.forEach {
       val embedding = embeddings?.embedQuery(it, requestConfig)
       val doc =
@@ -45,6 +45,8 @@ open class Lucene(
         }
       writer.addDocument(doc)
     }
+    writer.commit()
+  }
 
   override suspend fun similaritySearch(query: String, limit: Int): List<String> =
     search(FuzzyQuery(Term("contents", query)), limit)