|
12 | 12 | import rasa
|
13 | 13 | import rasa.utils.io
|
14 | 14 | import rasa.core.utils
|
15 |
| -from rasa.constants import DEFAULT_DOMAIN_PATH, LEGACY_DOCS_BASE_URL, ENV_SANIC_BACKLOG |
| 15 | +from rasa.constants import ( |
| 16 | + DEFAULT_DOMAIN_PATH, |
| 17 | + LEGACY_DOCS_BASE_URL, |
| 18 | + ENV_SANIC_BACKLOG, |
| 19 | + DEFAULT_CORE_SUBDIRECTORY_NAME, |
| 20 | +) |
16 | 21 | from rasa.core import constants, jobs, training
|
17 | 22 | from rasa.core.channels.channel import InputChannel, OutputChannel, UserMessage
|
18 | 23 | from rasa.core.constants import DEFAULT_REQUEST_TIMEOUT
|
|
25 | 30 | from rasa.core.policies.memoization import MemoizationPolicy
|
26 | 31 | from rasa.core.policies.policy import Policy
|
27 | 32 | from rasa.core.processor import MessageProcessor
|
28 |
| -from rasa.core.tracker_store import InMemoryTrackerStore, TrackerStore |
| 33 | +from rasa.core.tracker_store import ( |
| 34 | + InMemoryTrackerStore, |
| 35 | + TrackerStore, |
| 36 | + FailSafeTrackerStore, |
| 37 | +) |
29 | 38 | from rasa.core.trackers import DialogueStateTracker
|
30 | 39 | from rasa.exceptions import ModelNotFound
|
31 | 40 | from rasa.importers.importer import TrainingDataImporter
|
@@ -764,8 +773,8 @@ def persist(self, model_path: Text, dump_flattened_stories: bool = False) -> Non
|
764 | 773 | if not self.is_core_ready():
|
765 | 774 | raise AgentNotReady("Can't persist without a policy ensemble.")
|
766 | 775 |
|
767 |
| - if not model_path.endswith("core"): |
768 |
| - model_path = os.path.join(model_path, "core") |
| 776 | + if not model_path.endswith(DEFAULT_CORE_SUBDIRECTORY_NAME): |
| 777 | + model_path = os.path.join(model_path, DEFAULT_CORE_SUBDIRECTORY_NAME) |
769 | 778 |
|
770 | 779 | self._clear_model_directory(model_path)
|
771 | 780 |
|
@@ -851,9 +860,11 @@ def create_tracker_store(
|
851 | 860 | ) -> TrackerStore:
|
852 | 861 | if store is not None:
|
853 | 862 | store.domain = domain
|
854 |
| - return store |
| 863 | + tracker_store = store |
855 | 864 | else:
|
856 |
| - return InMemoryTrackerStore(domain) |
| 865 | + tracker_store = InMemoryTrackerStore(domain) |
| 866 | + |
| 867 | + return FailSafeTrackerStore(tracker_store) |
857 | 868 |
|
858 | 869 | @staticmethod
|
859 | 870 | def _create_lock_store(store: Optional[LockStore]) -> LockStore:
|
|
0 commit comments