toolbox/named_entity_recognization/named_entity_recognization.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
from typing import Callable, Dict, List, Tuple, Union

from toolbox.named_entity_recognization.pyltp_ner import pyltp_ner


language_to_engines = {
    "chinese": ["pyltp"]
}


engine_to_tagger: Dict[str, Callable] = {
    "pyltp": pyltp_ner
}


def ner(text: str, language: str, engine: str):
    ner_tagger = engine_to_tagger.get(engine)
    if ner_tagger is None:
        raise AssertionError(f"engine {engine} not supported.")

    words, postags, ner_tags = ner_tagger(text, language)
    return words, postags, ner_tags


if __name__ == "__main__":
    pass