added paraneter to current state of tracker

RasaHQ · Feb 7, 2018 · 60f17ca · 60f17ca
1 parent 991b2af
commit 60f17ca
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/rasa_core/trackers.py b/rasa_core/trackers.py
@@ -79,12 +79,18 @@ def __init__(self, sender_id, slots,
     ###
     # Public tracker interface
     ###
-    def current_state(self, should_include_events=False):
-        # type: (bool) -> Dict[Text, Any]
+    def current_state(self,
+                      should_include_events=False,
+                      only_events_after_latest_restart=False):
+        # type: (bool, bool) -> Dict[Text, Any]
         """Returns the current tracker state as an object."""
 
         if should_include_events:
-            evts = [e.as_dict() for e in self.events]
+            if only_events_after_latest_restart:
+                es = self.events
+            else:
+                es = self.events_after_latest_restart()
+            evts = [e.as_dict() for e in es]
         else:
             evts = None