Source code for intents.service_connector

"""
Connectors allow *Intents* Agent definitions to operate with real cloud services
such as Dialogflow, Lex or Azure Bot Services. Currently, only one connector is
provided with this library, and this is for Dialogflow ES:
:mod:`intents.connectors.dialogflow_es.connector`.

.. note::

    Details about the Connector interface are only useful if you need to develop your own Service Connector (please consider raising a pull request if this is the case). If you just need to use the included Dialogflow Connector you can jump to its documentation page right away: :mod:`intents.connectors.dialogflow_es.connector`

Connectors are used to operate with the cloud version of the Agent, and
specifically to:

* Export an :class:`intents.Agent` in a format that is natively readable by the
  Service
* Predict User messages on the Cloud Agent
* Trigger Intents on the Cloud Agent

More details can be found in the :class:`Connector` interface.
"""
from uuid import uuid1
from abc import ABC, abstractmethod
from typing import Dict, Any, List
from dataclasses import dataclass, field

from intents import Intent, Agent, Entity
from intents.language import IntentResponse, IntentResponseGroup
from intents.model.intent import IntentParameterMetadata, _IntentMetaclass
from intents.model.entity import EntityMixin, SystemEntityMixin, _EntityMetaclass

[docs]class EntityMapping(ABC): """ An Entity Mapping is a (de-)serializer for predicted Entities. Most of the times a Mapping is not needed as the Entity can be mapped directly to its type (e.g. "3" -> `Number(3)`). However, prediction services such as Dialogflow may define system entities of structured types; a notable example is Dialogflow's *sys.person* entity, which is returned as `{"name": "John"}` and therefore needs custom logic to be mapped to `Person("John")`. This is modelled in :class:`intents.connectors.dialogflow_es.entities.PersonEntityMapping`. Another notable scenario is Date/Time objects. A mapping can be used to convert time strings from the Service format to python objects. For Dialogflow ES, this is modelled in :class:`intents.connectors.dialogflow_es.entities.DateTimeEntityMapping`. """ @property @abstractmethod def entity_cls(self) -> _EntityMetaclass: """ This is the internal entity type that is being mapped. >>> mapping = StringEntityMapping(Sys.Integer, 'sys.number-integer') >>> mapping.entity_cls Sys.Integer """ @property @abstractmethod def service_name(self) -> str: """ This is the name of the Entity in the Prediction Service domain. >>> mapping = StringEntityMapping(Sys.Integer, 'sys.number-integer') >>> mapping.service_name 'sys.number-integer' """
[docs] @abstractmethod def from_service(self, service_data: Any) -> SystemEntityMixin: """ De-serialize the Service representation of an Entity (typically the value that is returned at prediction time) to an instance of one of the internal Entity classes in :class:`intents.model.entities` >>> date_mapping.from_service("2021-07-11") Sys.Date(2021, 7, 11) :param service_data: A parameter value, as it is returned by the Service in a prediction/trigger response :return: the parameter value, modelled as one of the System Entity classes """
[docs] @abstractmethod def to_service(self, entity: SystemEntityMixin) -> Any: """ Serialize a System Entity instance into a Service representation (typically, to be sent as a parameter of a trigger request) >>> date_mapping.to_service(Sys.Date(2021, 7, 11)) "2021-07-11" Args: entity: The System Entity to serialize Returns: The serialized Entity that can be sent to Service (e.g. in a trigger request) """
[docs]class StringEntityMapping(EntityMapping): """ This is a basic :class:`EntityMapping` that reads values as they are sent by the prediction service (e.g. `"3"` -> `Sys.Integer("3")`), and serializes by simple string conversion (e.g. `Sys.Integer(3)` -> "3"). The System Entity to use must be defined when instantiating the mapping, for instance: >>> StringEntityMapping(Sys.Integer, "sys.number-integer") """ entity_cls: _EntityMetaclass = None service_name: str = None def __init__(self, entity_cls: _EntityMetaclass, service_name: str): """ :param service_name: name of the Entity in the Prediction Service """ self.entity_cls = entity_cls self.service_name = service_name
[docs] def from_service(self, service_data: Any) -> SystemEntityMixin: return self.entity_cls(service_data)
[docs] def to_service(self, entity: SystemEntityMixin) -> Any: return str(entity)
[docs]class ServiceEntityMappings(dict): """ Models a list of entity mappings, in the form of a dict where the key is a System entity class (inherits from :class:`SystemEntityMixin`) and the value is a :class:`EntityMapping`. In addition to a standard dict, these features are added: * Instantiate from a list of mappings with :meth:`from_list` * Consistency check: a mapping list must cover all the entities defined in the framework (TODO) * Shortcut to define StringEntityMappings like `(Sys.Integer, 'sys.number-integer')` (TODO) """ @classmethod def from_list(cls, mapping_list: List[EntityMapping]) -> "ServiceEntityMappings": result = cls() for mapping in mapping_list: if mapping in result: raise ValueError(f"Mapping {mapping} already defined in list: {mapping_list}") result[mapping.entity_cls] = mapping return result
[docs]@dataclass class Prediction: """ One of the core uses of Service Connectors is to predict user utterances, or programmatically trigger intents. This class models the result of such predictions and triggers. You will typically obtain `Prediction` objects from :class:`Connector` methods :meth:`~Connector.predict` and :meth:`~Connector.trigger`. Args: intent: An instance of the predicted Intent confidence: A confidence value on the service prediction fulfillment_message_dict: A map of Intent Responses, as they were returned by the Service. Consider using :meth:`Prediction.fulfillment_messages` for convenience fulfillment_text: A plain-text version of the response """ intent: Intent confidence: float fulfillment_message_dict: Dict[IntentResponseGroup, List[IntentResponse]] = field(repr=False) fulfillment_text: str = None
[docs] def fulfillment_messages( self, response_group: IntentResponseGroup=IntentResponseGroup.RICH ) -> List[IntentResponse]: """ Return a list of fulfillment messages that are suitable for the given Response Group. The following scenarios may happen: * :class:`language.IntentResponseGroup.DEFAULT` is requested -> Message in the `DEFAULT` group will be returned * :class:`language.IntentResponseGroup.RICH` is requested * `RICH` messages are defined -> `RICH` messages are returned * No `RICH` message is defined -> `DEFAULT` messages are returned If present, messages in the "rich" group will be returned: >>> prediction.fulfillment_messages() [TextIntentResponse(choices=['I like travelling too! How can I help?']), QuickRepliesIntentResponse(replies=['Recommend a hotel', 'Send holiday photo', 'Where the station?'])] Alternatively, I can ask for plain-text default messages: >>> from intents.language import IntentResponseGroup >>> prediction.fulfillment_messages(IntentResponseGroup.DEFAULT) [TextIntentResponse(choices=['Nice, I can send you holiday pictures, or recommend an hotel'])] Args: response_group: The Response Group to fetch responses for """ if response_group == IntentResponseGroup.RICH and \ not self.fulfillment_message_dict.get(response_group): response_group = IntentResponseGroup.DEFAULT return self.fulfillment_message_dict.get(response_group, [])
[docs]def deserialize_intent_parameters( service_parameters: Dict[str, Any], intent_cls: _IntentMetaclass, mappings: ServiceEntityMappings ) -> Dict[str, EntityMixin]: """ Cast parameters from Service format to Intents format according to the given schema. Typically this happens when a Connector has to turn prediction parameters into *Intents* entities. Args: service_parameters: The parameters dict, as it is returned by a Prediction Service intent_cls: The Intent parameters will be matched against mappings: The Service Entity Mappings, to deserialize parameter values """ result = {} schema = intent_cls.parameter_schema for param_name, param_value in service_parameters.items(): if param_name not in schema: raise ValueError(f"Found parameter {param_name} in Service Prediction, but Intent class does not define it.") param_metadata = schema[param_name] mapping_cls = param_metadata.entity_cls if issubclass(mapping_cls, Entity): # Custom Entity mapping_cls = Entity mapping = mappings[mapping_cls] try: if param_metadata.is_list: if not isinstance(param_value, list): raise ValueError(f"Parameter {param_name} is defined as List, but returned value is not of 'list' type: {param_value}") result[param_name] = [mapping.from_service(x) for x in param_value] else: result[param_name] = mapping.from_service(param_value) except Exception as exc: raise RuntimeError(f"Failed to match parameter '{param_name}' with value '{param_value}' against schema {schema}. See source exception above for details.") from exc return result
[docs]class Connector(ABC): """ Connect the given Agent to a Prediction Service. :param agent_cls: The Agent to connect :param default_session: A default session ID (conversation channel) for predictions :param default_language: A default language for predictions """ agent_cls: type(Agent) default_session: str default_language: str def __init__(self, agent_cls: type(Agent), default_session: str=None, default_language: str="en"): if not default_session: default_session = f"py-{str(uuid1())}" self.agent_cls = agent_cls self.default_session = default_session self.default_language = default_language @property @abstractmethod def entity_mappings(self) -> ServiceEntityMappings: """ A Service Connector must know the Entity Mappings of its Prediction Service. They will be used to lookup entity names during export. """
[docs] @abstractmethod def predict(self, message: str, session: str=None, language: str=None) -> Prediction: """ Predict the given User message in the given session using the given language. When `session` or `language` are None, `predict` will use the default values that are specified in :meth:`__init__`. *predict* will return an instance of :class:`Prediction`, representing the service response. >>> from intents.connectors import DialogflowEsConnector >>> from example_agent import ExampleAgent >>> df = DialogflowEsConnector('/path/to/service-account.json', ExampleAgent) >>> prediction = df.predict("Hi, my name is Guido") >>> prediction.intent UserNameGive(user_name='Guido') >>> prediction.intent.user_name "Guido" >>> prediction.fulfillment_text "Hi Guido, I'm Bot" >>> prediction.confidence 0.86 Args: message: The User message to predict session: Any string identifying a conversation language: A ISO 639-1 language code (e.g. "en") """
[docs] @abstractmethod def trigger(self, intent: Intent, session: str=None, language: str=None) -> Prediction: """ Trigger the given Intent in the given session using the given language. When `session` or `language` are None, `predict` will use the default values that are specified in :meth:`__init__`. >>> from intents.connectors import DialogflowEsConnector >>> from example_agent import ExampleAgent, smalltalk >>> df = DialogflowEsConnector('/path/to/service-account.json', ExampleAgent) >>> prediction = df.trigger(smalltalk.AgentNameGive(agent_name='Alice')) >>> prediction.intent AgentNameGive(agent_name='Alice') >>> prediction.fulfillment_text "Howdy Human, I'm Alice" >>> prediction.confidence 1.0 Args: intent: The Intent instance to trigger session: Any string identifying a conversation language: A ISO 639-1 language code (e.g. "en") """
[docs] @abstractmethod def upload(self): """ Upload the connected Agent to the Prediction Service. """
[docs] @abstractmethod def export(self, destination: str): """ Export the connected Agent in a format that can be read and imported natively by the Prediction service. For instance, the Dialogflow service will produce a ZIP export that can be imported from the Dialogflow console. Note that you can also directly upload the Agent with :meth:`upload`. :param destination: destination path of the exported Agent """
def _entity_service_name(self, entity_cls: SystemEntityMixin) -> str: """ Return the name of the given entity in the specific service; this can be the class name itself, or an :class:`EntityMapping` lookup in the case of System Entities. For instance, a :class:`Sys.Person` Entity will need to be looked up in the mappings to find out its service name (`sys.person` in Dialogflow, `AMAZON.Person` in Alexa, and so on). A custom entity (e.g. `PizzaType`) can use its class name instead. """ if issubclass(entity_cls, SystemEntityMixin): return self.entity_mappings[entity_cls].service_name return entity_cls.name