use lombok in embeddings space

theam · Aug 31, 2023 · 10565e4 · 10565e4
1 parent 10adab3
commit 10565e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/modules/embeddingsspace/build.gradle b/modules/embeddingsspace/build.gradle
@@ -1,5 +1,6 @@
 plugins {
     id 'java'
+    id "io.freefair.lombok" version "8.3"
 }
 
 group = "com.theagilemonkeys.ellmental"

diff --git a/...src/main/java/com/theagilemonkeys/ellmental/embeddingsspace/EmbeddingsSpaceComponent.java b/...src/main/java/com/theagilemonkeys/ellmental/embeddingsspace/EmbeddingsSpaceComponent.java
@@ -3,20 +3,17 @@
 import com.theagilemonkeys.ellmental.core.schema.Embedding;
 import com.theagilemonkeys.ellmental.embeddingsgeneration.EmbeddingsGenerationModel;
 import com.theagilemonkeys.ellmental.embeddingsstore.EmbeddingsStore;
+import lombok.RequiredArgsConstructor;
 
 import java.util.List;
 import java.util.UUID;
 
+@RequiredArgsConstructor
 public class EmbeddingsSpaceComponent {
 
     private final EmbeddingsGenerationModel embeddingsGenerationModel;
     private final EmbeddingsStore embeddingsStore;
 
-    public EmbeddingsSpaceComponent(EmbeddingsGenerationModel embeddingsGenerationModel, EmbeddingsStore embeddingsStore) {
-        this.embeddingsGenerationModel = embeddingsGenerationModel;
-        this.embeddingsStore = embeddingsStore;
-    }
-
     private Embedding generate(String text) {
         return embeddingsGenerationModel.generateEmbedding(text);
     }