Source code for alex.components.nlg.common

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from __future__ import unicode_literals

from alex.components.nlg.exceptions import TemplateNLGException
from alex.components.nlg.template import TemplateNLG, TectoTemplateNLG


[docs]def get_nlg_type(cfg):
    return cfg['NLG']['type']


[docs]def nlg_factory(nlg_type, cfg):
    nlg = None

    # do not forget to maintain all supported dialogue managers
    if nlg_type == 'Template':
        nlg = TemplateNLG(cfg)
    elif nlg_type == 'TectoTemplate':
        nlg = TectoTemplateNLG(cfg)
    else:
        try:
            nlg = nlg_type(cfg)
        except NameError:
            raise TemplateNLGException('Unsupported NLG: %s' % nlg_type)

    return nlg