Add LLM memory-related interfaces

src/main/java/org/myrobotlab/document/Document.java

@@ -17,7 +17,7 @@
 public class Document {
 
   private String id;
-  private HashMap<String, ArrayList<Object>> data;
+  private final HashMap<String, ArrayList<Object>> data;
   private ProcessingStatus status;
 
   public Document(String id) {
@@ -27,11 +27,7 @@ public Document(String id) {
   }
 
   public ArrayList<Object> getField(String fieldName) {
-    if (data.containsKey(fieldName)) {
-      return data.get(fieldName);
-    } else {
-      return null;
-    }
+    return data.getOrDefault(fieldName, null);
   }
 
   public void setField(String fieldName, ArrayList<Object> value) {
@@ -151,9 +147,7 @@ public boolean equals(Object obj) {
         return false;
     } else if (!id.equals(other.id))
       return false;
-    if (status != other.status)
-      return false;
-    return true;
+    return status == other.status;
   }
 
   @Override

src/main/java/org/myrobotlab/service/EasyBert.java

@@ -0,0 +1,53 @@
+package org.myrobotlab.service;
+
+import com.google.common.primitives.Floats;
+import com.robrua.nlp.bert.Bert;
+import org.myrobotlab.framework.Service;
+import org.myrobotlab.service.interfaces.TextEmbeddingGenerator;
+
+import java.io.File;
+import java.util.List;
+
+public class EasyBert extends Service implements TextEmbeddingGenerator {
+
+    private Bert bert;
+    public final String DEFAULT_BERT_MODEL = "com/robrua/nlp/easy-bert/bert-uncased-L-12-H-768-A-12";
+
+
+    /**
+     * Constructor of service, reservedkey typically is a services name and inId
+     * will be its process id
+     *
+     * @param reservedKey the service name
+     * @param inId        process id
+     */
+    public EasyBert(String reservedKey, String inId) {
+        super(reservedKey, inId);
+        bert = Bert.load(DEFAULT_BERT_MODEL);
+    }
+
+    @Override
+    public List<Float> generateEmbeddings(String words) {
+        List<Float> embeddings = Floats.asList(bert.embedSequence(words));
+        invoke("publishEmbeddings", embeddings);
+        return embeddings;
+    }
+
+    @Override
+    public List<Float> publishEmbeddings(List<Float> embeddings) {
+        return embeddings;
+    }
+
+    @Override
+    public void onText(String text) throws Exception {
+        generateEmbeddings(text);
+    }
+
+    public void setBertModel(String resource) {
+        bert = Bert.load(resource);
+    }
+
+    public void setBertModel(File model) {
+        bert = Bert.load(model);
+    }
+}