Persist tags to output

Author: gunthercox

chatterbot/chatterbot.py

@@ -232,6 +232,8 @@ def __init__(self, statement, count=1):
             persona='bot:' + self.name
         )
 
+        response.add_tags(*result.get_tags())
+
         response.confidence = result.confidence
 
         return response

tests/test_chatbot.py

@@ -222,6 +222,16 @@ def test_response_with_tags_added(self):
         self.assertIsLength(results, 2)
         self.assertIn('test', results[0].get_tags())
 
+    def test_response_preserves_tags(self):
+        """
+        The response returned from the chatbot should preserve tags
+        from the statement that was used to generated the response.
+        """
+        self._create_with_search_text(text='Hello', tags=['test'])
+        response = self.chatbot.get_response('Hello')
+
+        self.assertEqual(response.get_tags(), ['test'])
+
     def test_get_response_with_text_and_kwargs(self):
         self.chatbot.get_response('Hello', conversation='greetings')