Merge pull request #1510 from RasaHQ/update_examples

Update examples

.gitignore

@@ -41,6 +41,8 @@ tmp/
 tokens.dat
 graph.png
 graph.html
+story_graph.html
+story_graph.dot
 debug.md
 examples/restaurantbot/models*
 examples/moodbot/*.png

CHANGELOG.rst

@@ -41,6 +41,7 @@ Fixed
 - fix error during interactive learning which was caused by actions which
   dispatched messages using ``dispatcher.utter_custom_message``
 - re-added missing ``python-engineio`` dependency
+- fixed not working examples in ``examples/``
 
 
 [0.12.3] - 2018-12-03

examples/concertbot/Makefile

@@ -7,6 +7,10 @@ help:
 	@echo "        Runs the core server."
 	@echo "    run-actions"
 	@echo "        Runs the action server."
+	@echo "    train-interactive"
+	@echo "        Train the bot with interactive learning."
+	@echo "    visualize"
+	@echo "        Show your stories as a graph."
 
 run:
 	make run-actions&
@@ -21,6 +25,9 @@ run-actions:
 train-core:
 	python train.py
 
-run-interactive:
+train-interactive:
 	make run-actions&
 	python train_interactive.py
+
+visualize:
+	python visualize.py

examples/concertbot/data/stories.md

@@ -21,9 +21,15 @@
     - slot{"concerts": [{"artist": "Foo Fighters", "reviews": 4.5}]}
 
 ## compare_reviews_venues
+* search_venues
+    - action_search_venues
+    - slot{"venues": [{"name": "Big Arena", "reviews": 4.5}]}
 * compare_reviews
     - action_show_venue_reviews
 
 ## compare_reviews_concerts
+* search_concerts
+    - action_search_concerts
+    - slot{"concerts": [{"artist": "Foo Fighters", "reviews": 4.5}]}
 * compare_reviews
     - action_show_concert_reviews

examples/concertbot/policy_config.yml

@@ -0,0 +1,7 @@
+policies:
+  - name: KerasPolicy
+    epochs: 200
+    batch_size: 50
+    max_training_samples: 300
+  - name: FallbackPolicy
+  - name: MemoizationPolicy

examples/concertbot/train.py

@@ -1,25 +1,26 @@
-from rasa_core import utils
-from rasa_core.agent import Agent
-from rasa_core.policies.keras_policy import KerasPolicy
-from rasa_core.policies.memoization import MemoizationPolicy
+from rasa_core import train, utils
 
-if __name__ == '__main__':
-    utils.configure_colored_logging(loglevel="INFO")
+import logging
+logger = logging.getLogger(__name__)
 
-    training_data_file = 'data/stories.md'
-    model_path = 'models/dialogue'
 
-    agent = Agent("domain.yml",
-                  policies=[MemoizationPolicy(), KerasPolicy()])
+def train_dialogue(domain_file='domain.yml',
+                   stories_file='data/stories.md',
+                   model_path='models/dialogue',
+                   policy_config='policy_config.yml'):
+    return train.train_dialogue_model(domain_file=domain_file,
+                                      stories_file=stories_file,
+                                      output_path=model_path,
+                                      policy_config=policy_config,
+                                      kwargs={'augmentation_factor': 50,
+                                              'validation_split': 0.2}
+                                      )
 
-    training_data = agent.load_data(training_data_file)
 
-    agent.train(
-        training_data,
-        augmentation_factor=50,
-        epochs=500,
-        batch_size=10,
-        validation_split=0.2
-    )
+if __name__ == '__main__':
+    train_dialogue()
 
-    agent.persist(model_path)
+    utils.configure_colored_logging(loglevel="INFO")
+    logger.info("This example does not include NLU data."
+                "Please specify the desired intent with a preceding '/', e.g."
+                "'/greet' .")

examples/concertbot/train_interactive.py

@@ -10,13 +10,14 @@ def train_agent():
     return train.train_dialogue_model(domain_file="domain.yml",
                                       stories_file="data/stories.md",
                                       output_path="models/dialogue",
-                                      kwargs={"batch_size": 50,
-                                              "epochs": 200,
-                                              "max_training_samples": 300
-                                              })
+                                      policy_config='policy_config.yml'
+                                      )
 
 
 if __name__ == '__main__':
     utils.configure_colored_logging(loglevel="INFO")
     agent = train_agent()
+    logger.info("This example does not include NLU data."
+                "Please specify the desired intent with a preceding '/', e.g."
+                "'/greet' .")
     interactive.run_interactive_learning(agent)

examples/moodbot/Makefile

@@ -3,8 +3,12 @@ help:
 	@echo "        Train the natural language understanding using Rasa NLU."
 	@echo "    train-core"
 	@echo "        Train a dialogue model using Rasa core."
-	@echo "    run"
-	@echo "        Spin up a server that serves as an endpoint to receive facebook user messages."
+	@echo "    run-{channel}"
+	@echo "        Spin up a server that uses the connector for {channel}."
+	@echo "    evaluate-nlu"
+	@echo "        Evaluates the nlu data."
+	@echo "    visualize"
+	@echo "        Show your stories as a graph."
 
 train-nlu:
 	python -m rasa_nlu.train -c nlu_model_config.yml --fixed_model_name current \
@@ -25,6 +29,15 @@ run-telegram:
 run-rocketchat:
 	python -m rasa_core.run -d models/dialogue -u models/nlu/current -p 5002 -c rocketchat --credentials rocketchat_credentials.yml
 
+run-mattermost:
+	python -m rasa_core.run -d models/dialogue -u models/nlu/current -p 5002 -c mattermost --credentials credentials.yml
+
+run-twilio:
+	python -m rasa_core.run -d models/dialogue -u models/nlu/current -p 5002 -c twilio --credentials credentials.yml
+
+run-webexteams:
+	python -m rasa_core.run -d models/dialogue -u models/nlu/current -p 5002 -c webexteams --credentials credentials.yml
+
 run-cmdline:
 	python -m rasa_core.run -d models/dialogue -u models/nlu/current
 

examples/moodbot/rocketchat_credentials.yml

@@ -1,4 +1,5 @@
-user: "your_bot_user"
-password: "you_bot_pass"
-server_url: "localhost:3000"
-ssl: false
+rocketchat:
+  user: "your_bot_user"
+  password: "you_bot_pass"
+  server_url: "localhost:3000"
+  ssl: false

examples/restaurantbot/Makefile

@@ -1,10 +1,14 @@
 help:
+	@echo "    run"
+	@echo "        Runs the bot on the command line."
+	@echo "    run-actions"
+	@echo "        Runs the action server."
+	@echo "    run-core"
+	@echo "        Runs the core server."
 	@echo "    train-nlu"
 	@echo "        Train the natural language understanding using Rasa NLU."
 	@echo "    train-core"
 	@echo "        Train a dialogue model using Rasa core."
-	@echo "    run"
-	@echo "        Runs the bot on the command line."
 
 run:
 	make run-actions&

tests/test_examples.py

@@ -89,3 +89,13 @@ def test_formbot_example():
     httpretty.disable()
 
     assert responses[0]['text'] == 'chitchat'
+
+
+def test_concertbot_training():
+    from examples.concertbot.train import train_dialogue
+
+    assert train_dialogue(domain_file='examples/concertbot/domain.yml',
+                          stories_file='examples/concertbot/data/stories.md',
+                          model_path='examples/concertbot/models/dialogue',
+                          policy_config='examples/concertbot/'
+                                        'policy_config.yml')