opm

pronomial/opm.py

@@ -0,0 +1,9 @@
+from ovos_plugin_manager.coreference import CoreferenceSolverEngine
+
+import pronomial
+
+
+class PronomialCoreferenceSolver(CoreferenceSolverEngine):
+    @classmethod
+    def solve_corefs(cls, text, lang="en"):
+        return pronomial.replace_corefs(text, lang=lang)

setup.py

@@ -1,14 +1,18 @@
 from setuptools import setup
 
+
+PLUGIN_ENTRY_POINT = 'ovos-coref-plugin-pronomial=pronomial.opm:PronomialCoreferenceSolver'
+
 setup(
     name='pronomial',
-    version='0.0.8',
+    version='0.1.0',
     packages=['pronomial', 'pronomial.lang'],
     url='https://github.com/OpenJarbas/pronomial',
     license='apache-2.0',
     author='jarbasAi',
     install_requires=["nltk", "pytest", "quebra_frases"],
     include_package_data=True,
     author_email='[email protected]',
-    description='pronomial postag/word_gender based coreference solver'
+    description='pronomial postag/word_gender based coreference solver',
+    entry_points={'intentbox.coreference': PLUGIN_ENTRY_POINT}
 )