Rasa Addons

A set of power tools to 🚀🚀🚀 your productivity with Rasa

Automated tests
Web chat: a channel to use with our open source web chat widget
Input validation: if you expect Yes or No, make sure your users anser Yes or No
Intent Substitution: avoid random intents when users enter data without semantic consistency (names, brands, time,...)
Custom dispatchers: need to store your Rasa Core templates out of domain file? We've got you covered

Installation

pip install rasa-addons

Web chat channel

from rasa_addons.webchat import WebChatInput, SocketInputChannel

agent = Agent.load(...)
input_channel = WebChatInput(static_assets_path=os.path.join(os.path.dirname(os.path.realpath(__file__)), 'static'))
agent.handle_channel(SocketInputChannel(5500, "/bot", input_channel))

In static you could have an index.html containing the widget snippet that you could access to http://localhost:5500/bot

Validate user input

Use the SuperAgent instead of the native Agent and specify a rules file.

from rasa_addons.superagent import SuperAgent
agent = SuperAgent.load(...,rules_file='rules.yml')

In rules.yml you can add input validation rules

input_validation:
  - after: utter_when_do_you_want_a_wake_up_call
    expected:
      - intents:
        - cancel
      - intents:
        - skeak_to_human
      - intents:
        - enter_time
        entities:
        - time
    error_template: utter_please_provide_time

The following rule will utter the error_template if the user does not reply to utter_when_do_you_want_a_wake_up_call with either /cancel OR /speak_to_human OR /enter_time{"time":"..."} Rules are enforced at the tracker level, so there is no need to retrain when changing them.

Swap intents

Some intents are hard to catch. For example when the user is asked to fill arbitrary data such as a date or a proper noun. The following rule swaps any intent caught after utter_when_do_you_want_a_wake_up_call with enter_data unless...

intent_substitutions:
  - after: utter_when_do_you_want_a_wake_up_call
    intent: enter_data
    unless: frustration|cancel|speak_to_human

Filter entities

Sometimes Rasa NLU CRF extractor will return unexpected entities and those can perturbate your Rasa Core dialogue model because it has never seen this particular combination of intent and entity.

This helper lets you define precisely the entities allowed for every intent in a yaml file. Entities not in the list for a given intent will be cleared. It will only remove entities for intents specifically listed in this section:

allowed_entities:
  book: # intent
    - origin # entity
    - destination
  buy:
    - color
    - product

Get templates out of domain files

If you want to get your templates from another source than the domain, you can do it like this:

Create your dispatcher

class MyDispatcher(Dispatcher):
    def retrieve_template(self, template_name, filled_slots=None, **kwargs):
        """Retrieve a named template from the domain."""

        response = requests.get('api/{template_key}/'.format(...))
        if response.status_code == 200:
            r = response.json()
            if r is not None:
                return self._fill_template_text(r, filled_slots, **kwargs)
            
        else:
            print("error")

Then load your agent

agent = SuperAgent.load(POLICY_PATH,
                        interpreter=interpreter,
                        create_dispatcher=lambda sender_id, output_channel, domain: MyDispatcher(sender_id, output_channel, domain))

Run automated tests (experimental)

You can write test cases as you would write stories, except you should only have utter_... actions.

## chitchat.greet
* chitchat.greet
  - utter_reply_to_greet

## chitchat.how_are_you
* chitchat.how_are_you
  - utter_reply_to_how_are_you

## chitchat.are_you_a_robot
* chitchat.are_you_a_robot
  - utter_reply_to_are_you_a_robot

Then you can run your tests with:

python -m rasa_addons.tests -d domains.yml -m models/dialogue/ -t test_cases/ -r rules.yml

You can put your test cases in different files starting with test (e.g. test_chitchat.md)in a directory.
At this time, it only runs the test and outputs dialogues in the console (errors in red). There is no report (Help wanted). You can also use --distinct to change the sender_id at every test case and --shuffle to shuffle test cases before running the tests.

Hub4IT/rasa-addons