feature: allow access to the current NLUService

spokestack · Dec 14, 2020 · 5433f64 · 5433f64
1 parent b4f2367
commit 5433f64
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/main/java/io/spokestack/spokestack/nlu/NLUManager.java b/src/main/java/io/spokestack/spokestack/nlu/NLUManager.java
@@ -62,6 +62,15 @@ private NLUService buildService() throws Exception {
         return (NLUService) constructed;
     }
 
+    /**
+     * Get the NLU service currently in use.
+     *
+     * @return The current NLU service.
+     */
+    public NLUService getNlu() {
+        return nlu;
+    }
+
     /**
      * Releases resources in use by the NLU module.
      */

diff --git a/src/test/java/io/spokestack/spokestack/SpokestackTest.java b/src/test/java/io/spokestack/spokestack/SpokestackTest.java
@@ -318,6 +318,7 @@ public void testClose() throws Exception {
         // modules don't work after close()
         spokestack.close();
         assertNull(spokestack.getTts().getTtsService());
+        assertNull(spokestack.getNlu().getNlu());
         assertThrows(
               IllegalStateException.class,
               () -> spokestack.classify("test"));
@@ -330,6 +331,7 @@ public void testClose() throws Exception {
         // restart supported
         spokestack.prepare();
         assertNotNull(spokestack.getTts().getTtsService());
+        assertNotNull(spokestack.getNlu().getNlu());
         assertDoesNotThrow(() -> spokestack.classify("test"));
         assertDoesNotThrow(() -> spokestack.synthesize(request));
     }