-
Notifications
You must be signed in to change notification settings - Fork 4.7k
/
domain.py
1973 lines (1673 loc) · 73.4 KB
/
domain.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
import copy
import collections
import json
import logging
import os
from pathlib import Path
from typing import (
Any,
Dict,
List,
NoReturn,
Optional,
Set,
Text,
Tuple,
Union,
TYPE_CHECKING,
Iterable,
MutableMapping,
NamedTuple,
Callable,
cast,
)
from dataclasses import dataclass
from ruamel.yaml.scalarstring import DoubleQuotedScalarString
from rasa.shared.constants import (
DEFAULT_SESSION_EXPIRATION_TIME_IN_MINUTES,
DEFAULT_CARRY_OVER_SLOTS_TO_NEW_SESSION,
DOMAIN_SCHEMA_FILE,
DOCS_URL_DOMAINS,
DOCS_URL_FORMS,
LATEST_TRAINING_DATA_FORMAT_VERSION,
DOCS_URL_RESPONSES,
REQUIRED_SLOTS_KEY,
IGNORED_INTENTS,
RESPONSE_CONDITION,
)
import rasa.shared.core.constants
from rasa.shared.core.constants import (
ACTION_SHOULD_SEND_DOMAIN,
SlotMappingType,
MAPPING_TYPE,
MAPPING_CONDITIONS,
ACTIVE_LOOP,
)
from rasa.shared.exceptions import (
RasaException,
YamlException,
YamlSyntaxException,
)
import rasa.shared.utils.validation
import rasa.shared.utils.io
import rasa.shared.utils.common
import rasa.shared.core.slot_mappings
from rasa.shared.core.events import SlotSet, UserUttered
from rasa.shared.core.slots import Slot, CategoricalSlot, TextSlot, AnySlot, ListSlot
from rasa.shared.utils.validation import KEY_TRAINING_DATA_FORMAT_VERSION
from rasa.shared.nlu.constants import (
ENTITY_ATTRIBUTE_TYPE,
ENTITY_ATTRIBUTE_ROLE,
ENTITY_ATTRIBUTE_GROUP,
RESPONSE_IDENTIFIER_DELIMITER,
INTENT_NAME_KEY,
ENTITIES,
)
if TYPE_CHECKING:
from rasa.shared.core.trackers import DialogueStateTracker
CARRY_OVER_SLOTS_KEY = "carry_over_slots_to_new_session"
SESSION_EXPIRATION_TIME_KEY = "session_expiration_time"
SESSION_CONFIG_KEY = "session_config"
USED_ENTITIES_KEY = "used_entities"
USE_ENTITIES_KEY = "use_entities"
IGNORE_ENTITIES_KEY = "ignore_entities"
IS_RETRIEVAL_INTENT_KEY = "is_retrieval_intent"
ENTITY_ROLES_KEY = "roles"
ENTITY_GROUPS_KEY = "groups"
ENTITY_FEATURIZATION_KEY = "influence_conversation"
KEY_SLOTS = "slots"
KEY_INTENTS = "intents"
KEY_ENTITIES = "entities"
KEY_RESPONSES = "responses"
KEY_ACTIONS = "actions"
KEY_FORMS = "forms"
KEY_E2E_ACTIONS = "e2e_actions"
KEY_RESPONSES_TEXT = "text"
ALL_DOMAIN_KEYS = [
KEY_SLOTS,
KEY_FORMS,
KEY_ACTIONS,
KEY_ENTITIES,
KEY_INTENTS,
KEY_RESPONSES,
KEY_E2E_ACTIONS,
SESSION_CONFIG_KEY,
]
PREV_PREFIX = "prev_"
# State is a dictionary with keys (USER, PREVIOUS_ACTION, SLOTS, ACTIVE_LOOP)
# representing the origin of a SubState;
# the values are SubStates, that contain the information needed for featurization
SubStateValue = Union[Text, Tuple[Union[float, Text], ...]]
SubState = MutableMapping[Text, SubStateValue]
State = Dict[Text, SubState]
logger = logging.getLogger(__name__)
class InvalidDomain(RasaException):
"""Exception that can be raised when domain is not valid."""
class ActionNotFoundException(ValueError, RasaException):
"""Raised when an action name could not be found."""
class SessionConfig(NamedTuple):
"""The Session Configuration."""
session_expiration_time: float # in minutes
carry_over_slots: bool
@staticmethod
def default() -> "SessionConfig":
"""Returns the SessionConfig with the default values."""
return SessionConfig(
DEFAULT_SESSION_EXPIRATION_TIME_IN_MINUTES,
DEFAULT_CARRY_OVER_SLOTS_TO_NEW_SESSION,
)
def are_sessions_enabled(self) -> bool:
"""Returns a boolean value depending on the value of session_expiration_time."""
return self.session_expiration_time > 0
def as_dict(self) -> Dict:
"""Return serialized `SessionConfig`."""
return {
"session_expiration_time": self.session_expiration_time,
"carry_over_slots_to_new_session": self.carry_over_slots,
}
@dataclass
class EntityProperties:
"""Class for keeping track of the properties of entities in the domain."""
entities: List[Text]
roles: Dict[Text, List[Text]]
groups: Dict[Text, List[Text]]
default_ignored_entities: List[Text]
class Domain:
"""The domain specifies the universe in which the bot's policy acts.
A Domain subclass provides the actions the bot can take, the intents
and entities it can recognise.
"""
@classmethod
def empty(cls) -> "Domain":
"""Returns empty Domain."""
return Domain.from_dict({})
@classmethod
def load(cls, paths: Union[List[Union[Path, Text]], Text, Path]) -> "Domain":
"""Returns loaded Domain after merging all domain files."""
if not paths:
raise InvalidDomain(
"No domain file was specified. Please specify a path "
"to a valid domain file."
)
elif not isinstance(paths, list) and not isinstance(paths, set):
paths = [paths]
domain = Domain.empty()
for path in paths:
other = cls.from_path(path)
domain = domain.merge(other)
return domain
@classmethod
def from_path(cls, path: Union[Text, Path]) -> "Domain":
"""Loads the `Domain` from a path."""
path = os.path.abspath(path)
if os.path.isfile(path):
domain = cls.from_file(path)
elif os.path.isdir(path):
domain = cls.from_directory(path)
else:
raise InvalidDomain(
"Failed to load domain specification from '{}'. "
"File not found!".format(os.path.abspath(path))
)
return domain
@classmethod
def from_file(cls, path: Text) -> "Domain":
"""Loads the `Domain` from a YAML file."""
return cls.from_yaml(rasa.shared.utils.io.read_file(path), path)
@classmethod
def from_yaml(cls, yaml: Text, original_filename: Text = "") -> "Domain":
"""Loads the `Domain` from YAML text after validating it."""
try:
rasa.shared.utils.validation.validate_yaml_schema(yaml, DOMAIN_SCHEMA_FILE)
data = rasa.shared.utils.io.read_yaml(yaml)
if not rasa.shared.utils.validation.validate_training_data_format_version(
data, original_filename
):
return Domain.empty()
return cls.from_dict(data)
except YamlException as e:
e.filename = original_filename
raise e
@classmethod
def from_dict(cls, data: Dict) -> "Domain":
"""Deserializes and creates domain.
Args:
data: The serialized domain.
Returns:
The instantiated `Domain` object.
"""
duplicates = data.pop("duplicates", None)
if duplicates:
warn_about_duplicates_found_during_domain_merging(duplicates)
responses = data.get(KEY_RESPONSES, {})
domain_slots = data.get(KEY_SLOTS, {})
if domain_slots:
rasa.shared.core.slot_mappings.validate_slot_mappings(domain_slots)
slots = cls.collect_slots(domain_slots)
domain_actions = data.get(KEY_ACTIONS, [])
actions = cls._collect_action_names(domain_actions)
additional_arguments = {
**data.get("config", {}),
"actions_which_explicitly_need_domain": cls._collect_actions_which_explicitly_need_domain( # noqa: E501
domain_actions
),
}
session_config = cls._get_session_config(data.get(SESSION_CONFIG_KEY, {}))
intents = data.get(KEY_INTENTS, {})
forms = data.get(KEY_FORMS, {})
_validate_forms(forms)
return cls(
intents=intents,
entities=data.get(KEY_ENTITIES, {}),
slots=slots,
responses=responses,
action_names=actions,
forms=data.get(KEY_FORMS, {}),
data=Domain._cleaned_data(data),
action_texts=data.get(KEY_E2E_ACTIONS, []),
session_config=session_config,
**additional_arguments,
)
@staticmethod
def _get_session_config(session_config: Dict) -> SessionConfig:
session_expiration_time_min = session_config.get(SESSION_EXPIRATION_TIME_KEY)
if session_expiration_time_min is None:
session_expiration_time_min = DEFAULT_SESSION_EXPIRATION_TIME_IN_MINUTES
carry_over_slots = session_config.get(
CARRY_OVER_SLOTS_KEY, DEFAULT_CARRY_OVER_SLOTS_TO_NEW_SESSION
)
return SessionConfig(session_expiration_time_min, carry_over_slots)
@classmethod
def from_directory(cls, path: Text) -> "Domain":
"""Loads and merges multiple domain files recursively from a directory tree."""
combined: Dict[Text, Any] = {}
for root, _, files in os.walk(path, followlinks=True):
for file in files:
full_path = os.path.join(root, file)
if Domain.is_domain_file(full_path):
_ = Domain.from_file(full_path) # does the validation here only
other_dict = rasa.shared.utils.io.read_yaml(
rasa.shared.utils.io.read_file(full_path)
)
combined = Domain.merge_domain_dicts(other_dict, combined)
domain = Domain.from_dict(combined)
return domain
def merge(
self,
domain: Optional["Domain"],
override: bool = False,
) -> "Domain":
"""Merges this domain dict with another one, combining their attributes.
This method merges domain dicts, and ensures all attributes (like ``intents``,
``entities``, and ``actions``) are known to the Domain when the
object is created.
List attributes like ``intents`` and ``actions`` are deduped
and merged. Single attributes are taken from `domain1` unless
override is `True`, in which case they are taken from `domain2`.
"""
if not domain or domain.is_empty():
return self
if self.is_empty():
return domain
merged_dict = self.__class__.merge_domain_dicts(
domain.as_dict(), self.as_dict(), override
)
return Domain.from_dict(merged_dict)
@staticmethod
def merge_domain_dicts(
domain_dict: Dict,
combined: Dict,
override: bool = False,
) -> Dict:
"""Combines two domain dictionaries."""
if not domain_dict:
return combined
if not combined:
return domain_dict
if override:
config = domain_dict.get("config", {})
for key, val in config.items():
combined["config"][key] = val
if (
override
or combined.get(SESSION_CONFIG_KEY) == SessionConfig.default().as_dict()
or combined.get(SESSION_CONFIG_KEY) is None
) and domain_dict.get(SESSION_CONFIG_KEY) not in [
None,
SessionConfig.default().as_dict(),
]:
combined[SESSION_CONFIG_KEY] = domain_dict[SESSION_CONFIG_KEY]
# remove existing forms from new actions
for form in combined.get(KEY_FORMS, []):
if form in domain_dict.get(KEY_ACTIONS, []):
domain_dict[KEY_ACTIONS].remove(form)
duplicates: Dict[Text, List[Text]] = {}
merge_func_mappings: Dict[Text, Callable[..., Any]] = {
KEY_INTENTS: rasa.shared.utils.common.merge_lists_of_dicts,
KEY_ENTITIES: rasa.shared.utils.common.merge_lists_of_dicts,
KEY_ACTIONS: rasa.shared.utils.common.merge_lists_of_dicts,
KEY_E2E_ACTIONS: rasa.shared.utils.common.merge_lists,
KEY_FORMS: rasa.shared.utils.common.merge_dicts,
KEY_RESPONSES: rasa.shared.utils.common.merge_dicts,
KEY_SLOTS: rasa.shared.utils.common.merge_dicts,
}
for key, merge_func in merge_func_mappings.items():
duplicates[key] = rasa.shared.utils.common.extract_duplicates(
combined.get(key, []), domain_dict.get(key, [])
)
default: Union[List[Any], Dict[Text, Any]] = (
{} if merge_func == rasa.shared.utils.common.merge_dicts else []
)
combined[key] = merge_func(
combined.get(key, default), domain_dict.get(key, default), override
)
if duplicates:
duplicates = rasa.shared.utils.common.clean_duplicates(duplicates)
combined.update({"duplicates": duplicates})
return combined
def _preprocess_domain_dict(
self,
data: Dict,
store_entities_as_slots: bool,
session_config: SessionConfig,
) -> Dict:
data = self._add_default_keys_to_domain_dict(
data,
store_entities_as_slots,
session_config,
)
data = self._sanitize_intents_in_domain_dict(data)
return data
@staticmethod
def _add_default_keys_to_domain_dict(
data: Dict,
store_entities_as_slots: bool,
session_config: SessionConfig,
) -> Dict:
# add the config, session_config and training data version defaults
# if not included in the original domain dict
if "config" not in data and not store_entities_as_slots:
data.update(
{"config": {"store_entities_as_slots": store_entities_as_slots}}
)
if SESSION_CONFIG_KEY not in data:
data.update(
{
SESSION_CONFIG_KEY: {
SESSION_EXPIRATION_TIME_KEY: (
session_config.session_expiration_time
),
CARRY_OVER_SLOTS_KEY: session_config.carry_over_slots,
}
}
)
if KEY_TRAINING_DATA_FORMAT_VERSION not in data:
data.update(
{
KEY_TRAINING_DATA_FORMAT_VERSION: DoubleQuotedScalarString(
LATEST_TRAINING_DATA_FORMAT_VERSION
)
}
)
return data
@staticmethod
def _reset_intent_flags(intent: Dict[Text, Any]) -> None:
for intent_property in intent.values():
if (
USE_ENTITIES_KEY in intent_property.keys()
and not intent_property[USE_ENTITIES_KEY]
):
intent_property[USE_ENTITIES_KEY] = []
if (
IGNORE_ENTITIES_KEY in intent_property.keys()
and not intent_property[IGNORE_ENTITIES_KEY]
):
intent_property[IGNORE_ENTITIES_KEY] = []
@staticmethod
def _sanitize_intents_in_domain_dict(data: Dict[Text, Any]) -> Dict[Text, Any]:
if not data.get(KEY_INTENTS):
return data
for intent in data.get(KEY_INTENTS, []):
if isinstance(intent, dict):
Domain._reset_intent_flags(intent)
data[KEY_INTENTS] = Domain._sort_intent_names_alphabetical_order(
intents=data.get(KEY_INTENTS)
)
return data
@staticmethod
def collect_slots(slot_dict: Dict[Text, Any]) -> List[Slot]:
"""Collects a list of slots from a dictionary."""
slots = []
# make a copy to not alter the input dictionary
slot_dict = copy.deepcopy(slot_dict)
# Don't sort the slots, see https://github.com/RasaHQ/rasa-x/issues/3900
for slot_name in slot_dict:
slot_type = slot_dict[slot_name].pop("type", None)
slot_class = Slot.resolve_by_type(slot_type)
slot = slot_class(slot_name, **slot_dict[slot_name])
slots.append(slot)
return slots
@staticmethod
def _transform_intent_properties_for_internal_use(
intent: Dict[Text, Any], entity_properties: EntityProperties
) -> Dict[Text, Any]:
"""Transforms the intent's parameters in a format suitable for internal use.
When an intent is retrieved from the `domain.yml` file, it contains two
parameters, the `use_entities` and the `ignore_entities` parameter.
With the values of these two parameters the Domain class is updated, a new
parameter is added to the intent called `used_entities` and the two
previous parameters are deleted. This happens because internally only the
parameter `used_entities` is needed to list all the entities that should be
used for this intent.
Args:
intent: The intent as retrieved from the `domain.yml` file thus having two
parameters, the `use_entities` and the `ignore_entities` parameter.
entity_properties: Entity properties as provided by the domain file.
Returns:
The intent with the new format thus having only one parameter called
`used_entities` since this is the expected format of the intent
when used internally.
"""
name, properties = next(iter(intent.items()))
if properties:
properties.setdefault(USE_ENTITIES_KEY, True)
else:
raise InvalidDomain(
f"In the `domain.yml` file, the intent '{name}' cannot have value of"
f" `{type(properties)}`. If you have placed a ':' character after the"
f" intent's name without adding any additional parameters to this"
f" intent then you would need to remove the ':' character. Please see"
f" {rasa.shared.constants.DOCS_URL_DOMAINS} for more information on how"
f" to correctly add `intents` in the `domain` and"
f" {rasa.shared.constants.DOCS_URL_INTENTS} for examples on"
f" when to use the ':' character after an intent's name."
)
properties.setdefault(
IGNORE_ENTITIES_KEY, entity_properties.default_ignored_entities
)
if not properties[USE_ENTITIES_KEY]: # this covers False, None and []
properties[USE_ENTITIES_KEY] = []
# `use_entities` is either a list of explicitly included entities
# or `True` if all should be included
# if the listed entities have a role or group label, concatenate the entity
# label with the corresponding role or group label to make sure roles and
# groups can also influence the dialogue predictions
if properties[USE_ENTITIES_KEY] is True:
included_entities = set(entity_properties.entities) - set(
entity_properties.default_ignored_entities
)
included_entities.update(
Domain.concatenate_entity_labels(entity_properties.roles)
)
included_entities.update(
Domain.concatenate_entity_labels(entity_properties.groups)
)
else:
included_entities = set(properties[USE_ENTITIES_KEY])
for entity in list(included_entities):
included_entities.update(
Domain.concatenate_entity_labels(entity_properties.roles, entity)
)
included_entities.update(
Domain.concatenate_entity_labels(entity_properties.groups, entity)
)
excluded_entities = set(properties[IGNORE_ENTITIES_KEY])
for entity in list(excluded_entities):
excluded_entities.update(
Domain.concatenate_entity_labels(entity_properties.roles, entity)
)
excluded_entities.update(
Domain.concatenate_entity_labels(entity_properties.groups, entity)
)
used_entities = list(included_entities - excluded_entities)
used_entities.sort()
# Only print warning for ambiguous configurations if entities were included
# explicitly.
explicitly_included = isinstance(properties[USE_ENTITIES_KEY], list)
ambiguous_entities = included_entities.intersection(excluded_entities)
if explicitly_included and ambiguous_entities:
rasa.shared.utils.io.raise_warning(
f"Entities: '{ambiguous_entities}' are explicitly included and"
f" excluded for intent '{name}'."
f"Excluding takes precedence in this case. "
f"Please resolve that ambiguity.",
docs=f"{DOCS_URL_DOMAINS}",
)
properties[USED_ENTITIES_KEY] = used_entities
del properties[USE_ENTITIES_KEY]
del properties[IGNORE_ENTITIES_KEY]
return intent
@rasa.shared.utils.common.lazy_property
def retrieval_intents(self) -> List[Text]:
"""List retrieval intents present in the domain."""
return [
intent
for intent in self.intent_properties
if self.intent_properties[intent].get(IS_RETRIEVAL_INTENT_KEY)
]
@classmethod
def collect_entity_properties(
cls, domain_entities: List[Union[Text, Dict[Text, Any]]]
) -> EntityProperties:
"""Get entity properties for a domain from what is provided by a domain file.
Args:
domain_entities: The entities as provided by a domain file.
Returns:
An instance of EntityProperties.
"""
entity_properties = EntityProperties([], {}, {}, [])
for entity in domain_entities:
if isinstance(entity, str):
entity_properties.entities.append(entity)
elif isinstance(entity, dict):
for _entity, sub_labels in entity.items():
entity_properties.entities.append(_entity)
if sub_labels:
if ENTITY_ROLES_KEY in sub_labels:
entity_properties.roles[_entity] = sub_labels[
ENTITY_ROLES_KEY
]
if ENTITY_GROUPS_KEY in sub_labels:
entity_properties.groups[_entity] = sub_labels[
ENTITY_GROUPS_KEY
]
if (
ENTITY_FEATURIZATION_KEY in sub_labels
and sub_labels[ENTITY_FEATURIZATION_KEY] is False
):
entity_properties.default_ignored_entities.append(_entity)
else:
raise InvalidDomain(
f"In the `domain.yml` file, the entity '{_entity}' cannot"
f" have value of `{type(sub_labels)}`. If you have placed a"
f" ':' character after the entity `{_entity}` without"
f" adding any additional parameters to this entity then you"
f" would need to remove the ':' character. Please see"
f" {rasa.shared.constants.DOCS_URL_DOMAINS} for more"
f" information on how to correctly add `entities` in the"
f" `domain` and {rasa.shared.constants.DOCS_URL_ENTITIES}"
f" for examples on when to use the ':' character after an"
f" entity's name."
)
else:
raise InvalidDomain(
f"Invalid domain. Entity is invalid, type of entity '{entity}' "
f"not supported: '{type(entity).__name__}'"
)
return entity_properties
@classmethod
def collect_intent_properties(
cls,
intents: List[Union[Text, Dict[Text, Any]]],
entity_properties: EntityProperties,
) -> Dict[Text, Dict[Text, Union[bool, List]]]:
"""Get intent properties for a domain from what is provided by a domain file.
Args:
intents: The intents as provided by a domain file.
entity_properties: Entity properties as provided by the domain file.
Returns:
The intent properties to be stored in the domain.
"""
# make a copy to not alter the input argument
intents = copy.deepcopy(intents)
intent_properties: Dict[Text, Any] = {}
duplicates = set()
for intent in intents:
intent_name, properties = cls._intent_properties(intent, entity_properties)
if intent_name in intent_properties.keys():
duplicates.add(intent_name)
intent_properties.update(properties)
if duplicates:
raise InvalidDomain(
f"Intents are not unique! Found multiple intents "
f"with name(s) {sorted(duplicates)}. "
f"Either rename or remove the duplicate ones."
)
cls._add_default_intents(intent_properties, entity_properties)
return intent_properties
@classmethod
def _intent_properties(
cls, intent: Union[Text, Dict[Text, Any]], entity_properties: EntityProperties
) -> Tuple[Text, Dict[Text, Any]]:
if not isinstance(intent, dict):
intent_name = intent
intent = {
intent_name: {
USE_ENTITIES_KEY: True,
IGNORE_ENTITIES_KEY: entity_properties.default_ignored_entities,
}
}
else:
intent_name = next(iter(intent.keys()))
return (
intent_name,
cls._transform_intent_properties_for_internal_use(
intent, entity_properties
),
)
@classmethod
def _add_default_intents(
cls,
intent_properties: Dict[Text, Dict[Text, Union[bool, List]]],
entity_properties: EntityProperties,
) -> None:
for intent_name in rasa.shared.core.constants.DEFAULT_INTENTS:
if intent_name not in intent_properties:
_, properties = cls._intent_properties(intent_name, entity_properties)
intent_properties.update(properties)
def __init__(
self,
intents: Union[Set[Text], List[Text], List[Dict[Text, Any]]],
entities: List[Union[Text, Dict[Text, Any]]],
slots: List[Slot],
responses: Dict[Text, List[Dict[Text, Any]]],
action_names: List[Text],
forms: Union[Dict[Text, Any], List[Text]],
data: Dict,
action_texts: Optional[List[Text]] = None,
store_entities_as_slots: bool = True,
session_config: SessionConfig = SessionConfig.default(),
**kwargs: Any,
) -> None:
"""Create a `Domain`.
Args:
intents: Intent labels.
entities: The names of entities which might be present in user messages.
slots: Slots to store information during the conversation.
responses: Bot responses. If an action with the same name is executed, it
will send the matching response to the user.
action_names: Names of custom actions.
forms: Form names and their slot mappings.
data: original domain dict representation.
action_texts: End-to-End bot utterances from end-to-end stories.
store_entities_as_slots: If `True` Rasa will automatically create `SlotSet`
events for entities if there are slots with the same name as the entity.
session_config: Configuration for conversation sessions. Conversations are
restarted at the end of a session.
"""
self.entity_properties = self.collect_entity_properties(entities)
self.intent_properties = self.collect_intent_properties(
intents, self.entity_properties
)
self.overridden_default_intents = self._collect_overridden_default_intents(
intents
)
self.form_names, self.forms, overridden_form_actions = self._initialize_forms(
forms
)
action_names += overridden_form_actions
self.responses = responses
self.action_texts = action_texts if action_texts is not None else []
data_copy = copy.deepcopy(data)
self._data = self._preprocess_domain_dict(
data_copy,
store_entities_as_slots,
session_config,
)
self.session_config = session_config
self._custom_actions = action_names
self._actions_which_explicitly_need_domain = (
kwargs.get("actions_which_explicitly_need_domain") or []
)
# only includes custom actions and utterance actions
self.user_actions = self._combine_with_responses(action_names, responses)
# includes all action names (custom, utterance, default actions and forms)
# and action texts from end-to-end bot utterances
self.action_names_or_texts = (
self._combine_user_with_default_actions(self.user_actions)
+ [
form_name
for form_name in self.form_names
if form_name not in self._custom_actions
]
+ self.action_texts
)
self._user_slots = copy.copy(slots)
self.slots = slots
self._add_default_slots()
self.store_entities_as_slots = store_entities_as_slots
self._check_domain_sanity()
def __deepcopy__(self, memo: Optional[Dict[int, Any]]) -> "Domain":
"""Enables making a deep copy of the `Domain` using `copy.deepcopy`.
See https://docs.python.org/3/library/copy.html#copy.deepcopy
for more implementation.
Args:
memo: Optional dictionary of objects already copied during the current
copying pass.
Returns:
A deep copy of the current domain.
"""
domain_dict = self.as_dict()
return self.__class__.from_dict(copy.deepcopy(domain_dict, memo))
def count_conditional_response_variations(self) -> int:
"""Returns count of conditional response variations."""
count = 0
for response_variations in self.responses.values():
for variation in response_variations:
if RESPONSE_CONDITION in variation:
count += 1
return count
@staticmethod
def _collect_overridden_default_intents(
intents: Union[Set[Text], List[Text], List[Dict[Text, Any]]]
) -> List[Text]:
"""Collects the default intents overridden by the user.
Args:
intents: User-provided intents.
Returns:
User-defined intents that are default intents.
"""
intent_names: Set[Text] = {
next(iter(intent.keys())) if isinstance(intent, dict) else intent
for intent in intents
}
return sorted(
intent_names.intersection(set(rasa.shared.core.constants.DEFAULT_INTENTS))
)
@staticmethod
def _initialize_forms(
forms: Dict[Text, Any]
) -> Tuple[List[Text], Dict[Text, Any], List[Text]]:
"""Retrieves the initial values for the Domain's form fields.
Args:
forms: Parsed content of the `forms` section in the domain.
Returns:
The form names, a mapping of form names and required slots, and custom
actions.
Returning custom actions for each forms means that Rasa Open Source should
not use the default `FormAction` for the forms, but rather a custom action
for it. This can e.g. be used to run the deprecated Rasa Open Source 1
`FormAction` which is implemented in the Rasa SDK.
"""
for form_name, form_data in forms.items():
if form_data is not None and REQUIRED_SLOTS_KEY not in form_data:
forms[form_name] = {REQUIRED_SLOTS_KEY: form_data}
return list(forms.keys()), forms, []
def __hash__(self) -> int:
"""Returns a unique hash for the domain."""
return int(self.fingerprint(), 16)
def fingerprint(self) -> Text:
"""Returns a unique hash for the domain which is stable across python runs.
Returns:
fingerprint of the domain
"""
self_as_dict = self.as_dict()
transformed_intents: List[Text] = []
for intent in self_as_dict.get(KEY_INTENTS, []):
if isinstance(intent, dict):
transformed_intents.append(*intent.keys())
elif isinstance(intent, str):
transformed_intents.append(intent)
self_as_dict[KEY_INTENTS] = sorted(transformed_intents)
self_as_dict[KEY_ACTIONS] = self.action_names_or_texts
return rasa.shared.utils.io.get_dictionary_fingerprint(self_as_dict)
@staticmethod
def _sort_intent_names_alphabetical_order(
intents: List[Union[Text, Dict]]
) -> List[Union[Text, Dict]]:
def sort(elem: Union[Text, Dict]) -> Union[Text, Dict]:
if isinstance(elem, dict):
return next(iter(elem.keys()))
elif isinstance(elem, str):
return elem
sorted_intents = sorted(intents, key=sort)
return sorted_intents
@rasa.shared.utils.common.lazy_property
def user_actions_and_forms(self) -> List[Text]:
"""Returns combination of user actions and forms."""
return self.user_actions + self.form_names
@rasa.shared.utils.common.lazy_property
def num_actions(self) -> int:
"""Returns the number of available actions."""
# noinspection PyTypeChecker
return len(self.action_names_or_texts)
@rasa.shared.utils.common.lazy_property
def num_states(self) -> int:
"""Number of used input states for the action prediction."""
return len(self.input_states)
@rasa.shared.utils.common.lazy_property
def retrieval_intent_responses(self) -> Dict[Text, List[Dict[Text, Any]]]:
"""Return only the responses which are defined for retrieval intents."""
return dict(
filter(
lambda intent_response: self.is_retrieval_intent_response(
intent_response
),
self.responses.items(),
)
)
@staticmethod
def is_retrieval_intent_response(
response: Tuple[Text, List[Dict[Text, Any]]]
) -> bool:
"""Check if the response is for a retrieval intent.
These responses have a `/` symbol in their name. Use that to filter them from
the rest.
"""
return RESPONSE_IDENTIFIER_DELIMITER in response[0]
def _add_default_slots(self) -> None:
"""Sets up the default slots and slot values for the domain."""
self._add_requested_slot()
self._add_knowledge_base_slots()
self._add_categorical_slot_default_value()
self._add_session_metadata_slot()
def _add_categorical_slot_default_value(self) -> None:
"""Add a default value to all categorical slots.
All unseen values found for the slot will be mapped to this default value
for featurization.
"""
for slot in [s for s in self.slots if isinstance(s, CategoricalSlot)]:
slot.add_default_value()
def _add_requested_slot(self) -> None:
"""Add a slot called `requested_slot` to the list of slots.
The value of this slot will hold the name of the slot which the user
needs to fill in next (either explicitly or implicitly) as part of a form.
"""
if self.form_names and rasa.shared.core.constants.REQUESTED_SLOT not in [
slot.name for slot in self.slots
]:
self.slots.append(
TextSlot(
rasa.shared.core.constants.REQUESTED_SLOT,
mappings=[],
influence_conversation=False,
)
)
def _add_knowledge_base_slots(self) -> None:
"""Add slots for the knowledge base action to slots.
Slots are only added if the default knowledge base action name is present.
As soon as the knowledge base action is not experimental anymore, we should
consider creating a new section in the domain file dedicated to knowledge
base slots.
"""
if (
rasa.shared.core.constants.DEFAULT_KNOWLEDGE_BASE_ACTION
in self.action_names_or_texts
):
logger.warning(
"You are using an experimental feature: Action '{}'!".format(
rasa.shared.core.constants.DEFAULT_KNOWLEDGE_BASE_ACTION