🎉

ovos_plugin_manager/embeddings.py

@@ -1,4 +1,4 @@
-from ovos_plugin_manager.templates.embeddings import EmbeddingsDB, FaceEmbeddingsRecognizer, VoiceEmbeddingsRecognizer
+from ovos_plugin_manager.templates.embeddings import EmbeddingsDB, TextEmbeddingRecognizer, FaceEmbeddingsRecognizer, VoiceEmbeddingsRecognizer
 from ovos_plugin_manager.utils import PluginTypes
 
 
@@ -57,3 +57,22 @@ def load_face_embeddings_plugin(module_name: str) -> type(FaceEmbeddingsRecogniz
     """
     from ovos_plugin_manager.utils import load_plugin
     return load_plugin(module_name, PluginTypes.FACE_EMBEDDINGS)
+
+
+def find_text_embeddings_plugins() -> dict:
+    """
+    Find all installed plugins
+    @return: dict plugin names to entrypoints
+    """
+    from ovos_plugin_manager.utils import find_plugins
+    return find_plugins(PluginTypes.TEXT_EMBEDDINGS)
+
+
+def load_text_embeddings_plugin(module_name: str) -> type(TextEmbeddingRecognizer):
+    """
+    Get an uninstantiated class for the requested module_name
+    @param module_name: Plugin entrypoint name to load
+    @return: Uninstantiated class
+    """
+    from ovos_plugin_manager.utils import load_plugin
+    return load_plugin(module_name, PluginTypes.TEXT_EMBEDDINGS)

ovos_plugin_manager/templates/embeddings.py

@@ -40,6 +40,27 @@ def distance(self, embeddings_a: np.ndarray,
             raise ValueError("Unsupported metric")
 
 
+class TextEmbeddingRecognizer(EmbeddingsDB):
+    def __init__(self, db: EmbeddingsDB):
+        self.db = db
+
+    @abc.abstractmethod
+    def get_text_embeddings(self, text: str) -> np.ndarray:
+        return NotImplemented
+
+    def add_document(self, document: str) -> None:
+        embeddings = self.get_text_embeddings(document)
+        self.db.add_embeddings(document, embeddings)
+
+    def delete_document(self, document: str) -> None:
+        embeddings = self.get_text_embeddings(document)
+        self.db.delete_embedding(embeddings)
+
+    def query_document(self, document, top_k: int = 5) -> List[Tuple[str, float]]:
+        embeddings = self.get_text_embeddings(document)
+        return self.db.query(embeddings, top_k)
+
+
 class FaceEmbeddingsRecognizer:
     def __init__(self, db: EmbeddingsDB, thresh: float = 0.15):
         self.db = db

ovos_plugin_manager/utils/__init__.py

@@ -25,6 +25,7 @@ class PluginTypes(str, Enum):
     EMBEDDINGS = "opm.embeddings"
     FACE_EMBEDDINGS = "opm.embeddings.face"
     VOICE_EMBEDDINGS = "opm.embeddings.voice"
+    TEXT_EMBEDDINGS = "opm.embeddings.text"
     GUI = "ovos.plugin.gui"
     PHAL = "ovos.plugin.phal"
     ADMIN = "ovos.plugin.phal.admin"
@@ -65,6 +66,7 @@ class PluginConfigTypes(str, Enum):
     EMBEDDINGS = "opm.embeddings.config"
     FACE_EMBEDDINGS = "opm.embeddings.face.config"
     VOICE_EMBEDDINGS = "opm.embeddings.voice.config"
+    TEXT_EMBEDDINGS = "opm.embeddings.text.config"
     GUI = "ovos.plugin.gui.config"
     PHAL = "ovos.plugin.phal.config"
     ADMIN = "ovos.plugin.phal.admin.config"