feat: add UserEntity support for android

Author: spoeck

android/src/main/java/de/innfactory/apiai/RNApiAiModule.java

@@ -24,8 +24,11 @@
 import ai.api.model.AIResponse;
 import ai.api.model.Result;
 import ai.api.model.AIContext;
+import ai.api.model.Entity;
+import ai.api.model.EntityEntry;
 import ai.api.RequestExtras;
 
+
 /**
  * Created by Anton Spöck on 2017-07-21
  **/
@@ -58,7 +61,7 @@ public class RNApiAiModule extends ReactContextBaseJavaModule implements AIListe
             AIConfiguration.RecognitionEngine.System);
     private AIDataService aiDataService;
     private List<AIContext> contexts;
-
+    private List<Entity> entities;
     private Callback onResultCallback;
     private Callback onErrorCallback;
     private Callback onListeningStartedCallback;
@@ -85,7 +88,14 @@ public void setConfiguration(String clientAccessToken, String languageTag) {
     @ReactMethod
     public void setContextsAsJson(String contextsAsJson) {
         Gson gson= new Gson();
-        contexts = gson.fromJson(contextsAsJson, new TypeToken<List<AIContext>>(){}.getType());
+        contexts = gson.fromJson(contextsAsJson, new TypeToken<List<Entity>>(){}.getType());
+    }
+
+
+    @ReactMethod
+    public void setEntitiesAsJson(String userEntitiesAsJson) throws AIServiceException {
+        Gson gson= new Gson();
+        entities = gson.fromJson(userEntitiesAsJson, new TypeToken<List<Entity>>(){}.getType());
     }
 
 
@@ -103,10 +113,11 @@ public void run() {
                 aiService.setListener(RNApiAiModule.this);
 
                 // set contexts
-                if (contexts != null) {
-                    RequestExtras requestExtras = new RequestExtras(contexts, null);
+                if (contexts != null || entities != null) {
+                    RequestExtras requestExtras = new RequestExtras(contexts, entities);
                     aiService.startListening(requestExtras);
                     contexts = null;
+                    entities = null;
                 } else {
 
                     // start listening without context
@@ -268,10 +279,11 @@ protected AIResponse doInBackground(AIRequest... requests) {
                     AIResponse response = null;
 
                     // set contexts
-                    if (contexts != null) {
-                        RequestExtras requestExtras = new RequestExtras(contexts, null);
+                    if (contexts != null || entities != null) {
+                        RequestExtras requestExtras = new RequestExtras(contexts, entities);
                         response = aiDataService.request(aiRequest, requestExtras);
                         contexts = null;
+                        entities = null;
                     } else {
 
                         // start request without context

example/index.android.js

@@ -23,7 +23,7 @@ export default class App extends Component {
       console.log(ApiAi);
 
       ApiAi.setConfiguration(
-          "57b6ce865e6e4b138a74a88cfd8bc526", ApiAi.LANG_GERMAN
+          "INSERT_YOUR_CLIENT_ACCESS_TOKEN_HERE", ApiAi.LANG_GERMAN
       );
 
 
@@ -37,6 +37,23 @@ export default class App extends Component {
       }];
 
       ApiAi.setContexts(contexts);
+
+
+      const entities = [{
+          "name":"shop",
+          "extend":true,
+          "entries":[
+              {
+                  "value":"Media Markt",
+                  "synonyms":[
+                      "Media Markt",
+                  ]
+              }
+          ]
+      }];
+
+
+      ApiAi.setEntities(entities);
   }
 
 

example/package.json

@@ -9,7 +9,7 @@
   "dependencies": {
     "react": "16.0.0-alpha.12",
     "react-native": "0.47.1",
-    "react-native-api-ai": "^1.1.0"
+    "react-native-api-ai": "../"
   },
   "devDependencies": {
     "babel-jest": "20.0.3",

index.android.js

@@ -6,6 +6,10 @@ let ApiAi = NativeModules.ApiAi;
 
 ApiAi.setContexts = (contexts) => {
     ApiAi.setContextsAsJson(JSON.stringify(contexts))
-}
+};
+
+ApiAi.setEntities = (entities) => {
+    ApiAi.setEntitiesAsJson(JSON.stringify(entities))
+};
 
 module.exports = ApiAi;