Source code for deeppavlov.models.go_bot.nlg.nlg_manager_interface

from abc import ABCMeta, abstractmethod

from deeppavlov.models.go_bot.dto.dataset_features import BatchDialoguesFeatures
from deeppavlov.models.go_bot.nlg.dto.nlg_response_interface import NLGResponseInterface
from deeppavlov.models.go_bot.policy.dto.policy_prediction import PolicyPrediction

[docs]class NLGManagerInterface(metaclass=ABCMeta):
[docs] @abstractmethod def get_action_id(self, action_text) -> int: """ Looks up for an ID relevant to the passed action text in the list of known actions and their ids. Args: action_text: the text for which an ID needs to be returned. Returns: an ID corresponding to the passed action text """ pass
[docs] @abstractmethod def get_api_call_action_id(self) -> int: """ Returns: an ID corresponding to the api call action """ pass
@abstractmethod def decode_response(self, utterance_batch_features: BatchDialoguesFeatures, policy_prediction: PolicyPrediction, tracker_slotfilled_state) -> NLGResponseInterface: # todo: docstring pass
[docs] @abstractmethod def num_of_known_actions(self) -> int: """ Returns: the number of actions known to the NLG module """ pass