[incubator-nlpcraft] branch NLPCRAFT-471 updated: WIP.

[sergeykamov]

This is an automated email from the ASF dual-hosted git repository.

sergeykamov pushed a commit to branch NLPCRAFT-471
in repository https://gitbox.apache.org/repos/asf/incubator-nlpcraft.git


The following commit(s) were added to refs/heads/NLPCRAFT-471 by this push:
     new fd6dc70  WIP.
fd6dc70 is described below

commit fd6dc70ae76bc83c5660aeaf42e884fa74c4dadc
Author: Sergey Kamov <skhdlem...@gmail.com>
AuthorDate: Mon Dec 27 16:21:26 2021 +0300

    WIP.
---
 .../nlp/token/enricher/NCOpenNlpTokenEnricher.java | 30 +++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git 
a/nlpcraft/src/main/scala/org/apache/nlpcraft/internal/nlp/token/enricher/NCOpenNlpTokenEnricher.java
 
b/nlpcraft/src/main/scala/org/apache/nlpcraft/internal/nlp/token/enricher/NCOpenNlpTokenEnricher.java
index 5db8704..710abbd 100644
--- 
a/nlpcraft/src/main/scala/org/apache/nlpcraft/internal/nlp/token/enricher/NCOpenNlpTokenEnricher.java
+++ 
b/nlpcraft/src/main/scala/org/apache/nlpcraft/internal/nlp/token/enricher/NCOpenNlpTokenEnricher.java
@@ -23,6 +23,8 @@ import org.apache.nlpcraft.NCToken;
 import org.apache.nlpcraft.NCTokenEnricher;
 import 
org.apache.nlpcraft.internal.nlp.token.enricher.impl.NCOpenNlpTokenEnricherImpl;
 
+import java.io.File;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
@@ -32,7 +34,9 @@ import java.util.Objects;
  *  - opennlp:name, values - look at constructor keys)
  *  - opennlp:probability, 0..1 probability
  *
- *  Models can be download here: http://opennlp.sourceforge.net/models-1.5/
+ *  Models can be download here: http://opennlp.sourceforge.net/models-1.5/ or 
trained.
+ *
+ *  TODO: which constructors should we keep?
  */
 public class NCOpenNlpTokenEnricher implements NCTokenEnricher {
     private final NCOpenNlpTokenEnricherImpl impl;
@@ -48,6 +52,30 @@ public class NCOpenNlpTokenEnricher implements 
NCTokenEnricher {
         this.impl = new NCOpenNlpTokenEnricherImpl(models);
     }
 
+    /**
+     *
+     * @param name
+     * @param modelSrc
+     */
+    public NCOpenNlpTokenEnricher(String name, String modelSrc) {
+        Objects.requireNonNull(name, "Name cannot be null.");
+        Objects.requireNonNull(modelSrc, "Model source cannot be null.");
+
+        this.impl = new NCOpenNlpTokenEnricherImpl(new HashMap<>() {{ 
put(name, modelSrc); }});
+    }
+
+    /**
+     *
+     * @param name
+     * @param modelFile
+     */
+    public NCOpenNlpTokenEnricher(String name, File modelFile) {
+        Objects.requireNonNull(name, "Name cannot be null.");
+        Objects.requireNonNull(modelFile, "Model file cannot be null.");
+
+        this.impl = new NCOpenNlpTokenEnricherImpl(new HashMap<>() {{ 
put(name, modelFile.getAbsolutePath()); }});
+    }
+
     @Override
     public void start() {
         impl.start();