rasa-pro 3.12.0.dev1__py3-none-any.whl → 3.12.0.dev3__py3-none-any.whl

rasa/shared/nlu/training_data/features.py

@@ -2,15 +2,15 @@ from __future__ import annotations
 
 import itertools
 from dataclasses import dataclass
-from typing import Iterable, Union, Text, Optional, List, Any, Tuple, Dict, Set
+from typing import Any, Dict, Iterable, List, Optional, Set, Text, Tuple, Union
 
 import numpy as np
 import scipy.sparse
-from safetensors.numpy import save_file, load_file
+from safetensors.numpy import load_file, save_file
 
 import rasa.shared.nlu.training_data.util
 import rasa.shared.utils.io
-from rasa.shared.nlu.constants import FEATURE_TYPE_SEQUENCE, FEATURE_TYPE_SENTENCE
+from rasa.shared.nlu.constants import FEATURE_TYPE_SENTENCE, FEATURE_TYPE_SEQUENCE
 
 
 @dataclass

rasa/shared/nlu/training_data/formats/__init__.py

@@ -1,4 +1,3 @@
-from rasa.shared.nlu.training_data.formats.rasa_yaml import RasaYAMLReader  # noqa: F401
 from rasa.shared.nlu.training_data.formats.dialogflow import (  # noqa: F401
     DialogflowReader,
 )
@@ -7,4 +6,5 @@ from rasa.shared.nlu.training_data.formats.rasa import (  # noqa: F401
     RasaReader,
     RasaWriter,
 )
+from rasa.shared.nlu.training_data.formats.rasa_yaml import RasaYAMLReader  # noqa: F401
 from rasa.shared.nlu.training_data.formats.wit import WitReader  # noqa: F401

rasa/shared/nlu/training_data/formats/dialogflow.py

@@ -1,16 +1,15 @@
 import logging
 import os
 from pathlib import Path
-from typing import Any, Dict, Optional, Text, List, Tuple, Union
+from typing import Any, Dict, List, Optional, Text, Tuple, Union
 
 import rasa.shared.nlu.training_data.util
+import rasa.shared.utils.io
 from rasa.shared.constants import DOCS_BASE_URL
 from rasa.shared.nlu.training_data.formats.readerwriter import TrainingDataReader
-from rasa.shared.nlu.training_data.util import transform_entity_synonyms
-import rasa.shared.utils.io
-
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
+from rasa.shared.nlu.training_data.util import transform_entity_synonyms
 
 logger = logging.getLogger(__name__)
 

rasa/shared/nlu/training_data/formats/luis.py

@@ -1,22 +1,21 @@
 import logging
 from typing import Any, Dict, List, Text
 
+import rasa.shared.utils.io
 from rasa.shared.nlu.constants import (
-    TEXT,
-    INTENT,
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_END,
     ENTITY_ATTRIBUTE_ROLE,
-    ENTITY_ATTRIBUTE_VALUE,
     ENTITY_ATTRIBUTE_START,
-    ENTITY_ATTRIBUTE_END,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
+    INTENT,
     NO_ENTITY_TAG,
+    TEXT,
 )
 from rasa.shared.nlu.training_data.formats.readerwriter import JsonTrainingDataReader
-import rasa.shared.utils.io
-
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 logger = logging.getLogger(__name__)
 

rasa/shared/nlu/training_data/formats/rasa.py

@@ -2,18 +2,17 @@ import logging
 from collections import defaultdict
 from typing import Any, Dict, Text
 
+import rasa.shared.utils.io
 from rasa.shared.constants import DOCS_URL_MIGRATION_GUIDE
-from rasa.shared.nlu.constants import TEXT, INTENT, ENTITIES
+from rasa.shared.nlu.constants import ENTITIES, INTENT, TEXT
 from rasa.shared.nlu.training_data.formats.readerwriter import (
     JsonTrainingDataReader,
     TrainingDataWriter,
 )
+from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.util import transform_entity_synonyms
 from rasa.shared.utils.io import json_to_string
-
-from rasa.shared.nlu.training_data.training_data import TrainingData
-from rasa.shared.nlu.training_data.message import Message
-import rasa.shared.utils.io
 from rasa.shared.utils.yaml import validate_training_data
 
 logger = logging.getLogger(__name__)

rasa/shared/nlu/training_data/formats/rasa_yaml.py

@@ -2,42 +2,42 @@ import logging
 from collections import OrderedDict
 from pathlib import Path
 from typing import (
-    ClassVar,
-    Text,
     Any,
-    List,
+    Callable,
+    ClassVar,
     Dict,
-    Tuple,
-    Union,
     Iterator,
+    List,
     Optional,
-    Callable,
+    Text,
+    Tuple,
+    Union,
 )
 
-import rasa.shared.data
-from rasa.shared.core.domain import Domain
-from rasa.shared.exceptions import YamlException
 from ruamel.yaml import StringIO
 from ruamel.yaml.scalarstring import LiteralScalarString
 
+import rasa.shared.data
+import rasa.shared.nlu.training_data.util
+import rasa.shared.utils.io
 from rasa.shared.constants import (
     DOCS_URL_TRAINING_DATA,
     LATEST_TRAINING_DATA_FORMAT_VERSION,
 )
-from rasa.shared.nlu.constants import METADATA_INTENT, METADATA_EXAMPLE
+from rasa.shared.core.domain import Domain
+from rasa.shared.exceptions import YamlException
+from rasa.shared.nlu.constants import METADATA_EXAMPLE, METADATA_INTENT
 from rasa.shared.nlu.training_data.formats.readerwriter import (
     TrainingDataReader,
     TrainingDataWriter,
 )
-import rasa.shared.utils.io
-import rasa.shared.nlu.training_data.util
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.utils.yaml import (
+    is_key_in_yaml,
+    read_yaml,
     validate_raw_yaml_using_schema_file_with_responses,
     validate_training_data_format_version,
-    read_yaml,
-    is_key_in_yaml,
     write_yaml,
 )
 
@@ -409,9 +409,10 @@ class RasaYAMLWriter(TrainingDataWriter):
         Returns:
             `OrderedDict` containing all training data.
         """
-        from rasa.shared.utils.yaml import KEY_TRAINING_DATA_FORMAT_VERSION
         from ruamel.yaml.scalarstring import DoubleQuotedScalarString
 
+        from rasa.shared.utils.yaml import KEY_TRAINING_DATA_FORMAT_VERSION
+
         nlu_items = []
         nlu_items.extend(cls.process_intents(training_data))
         nlu_items.extend(cls.process_synonyms(training_data))

rasa/shared/nlu/training_data/formats/readerwriter.py

@@ -1,26 +1,23 @@
 import abc
 import json
-from collections import OrderedDict
+import typing
+from collections import OrderedDict, defaultdict
 from pathlib import Path
+from typing import Any, Dict, List, Text, Union
 
 import rasa.shared.nlu.training_data.util
+import rasa.shared.utils.io
 from rasa.shared.constants import INTENT_MESSAGE_PREFIX
-
 from rasa.shared.nlu.constants import (
-    INTENT,
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_END,
-    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_START,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
+    INTENT,
 )
 
-import rasa.shared.utils.io
-import typing
-from typing import Text, Dict, Any, Union, List
-from collections import defaultdict
-
 if typing.TYPE_CHECKING:
     from rasa.shared.nlu.training_data.training_data import TrainingData
 

rasa/shared/nlu/training_data/formats/wit.py

@@ -3,17 +3,16 @@ from typing import Any, Dict, Text
 
 from rasa.shared.core.constants import USER_INTENT_OUT_OF_SCOPE
 from rasa.shared.nlu.constants import (
-    INTENT,
     ENTITIES,
-    TEXT,
     ENTITY_ATTRIBUTE_ROLE,
     ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_VALUE,
+    INTENT,
+    TEXT,
 )
 from rasa.shared.nlu.training_data.formats.readerwriter import JsonTrainingDataReader
-
-from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
+from rasa.shared.nlu.training_data.training_data import TrainingData
 
 logger = logging.getLogger(__name__)
 

rasa/shared/nlu/training_data/loading.py

@@ -2,7 +2,7 @@ import json
 import logging
 import os
 import typing
-from typing import Optional, Text, Callable, Dict, Any, List
+from typing import Any, Callable, Dict, List, Optional, Text
 
 import rasa.shared.utils.io
 from rasa.shared.nlu.training_data.formats.dialogflow import (
@@ -69,11 +69,11 @@ def load_data(resource_name: Text, language: Optional[Text] = "en") -> "Training
 def _reader_factory(fformat: Text) -> Optional["TrainingDataReader"]:
     """Generates the appropriate reader class based on the file format."""
     from rasa.shared.nlu.training_data.formats import (
-        RasaYAMLReader,
-        WitReader,
+        DialogflowReader,
         LuisReader,
         RasaReader,
-        DialogflowReader,
+        RasaYAMLReader,
+        WitReader,
     )
 
     reader: Optional["TrainingDataReader"] = None

rasa/shared/nlu/training_data/lookup_tables_parser.py

@@ -1,4 +1,4 @@
-from typing import Text, List, Dict, Union
+from typing import Dict, List, Text, Union
 
 
 def add_item_to_lookup_tables(

rasa/shared/nlu/training_data/message.py

@@ -1,30 +1,29 @@
-from typing import Any, Optional, Tuple, Text, Dict, Set, List
-
-import typing
 import copy
+import typing
+from typing import Any, Dict, List, Optional, Set, Text, Tuple
 
 import rasa.shared.utils.io
+from rasa.shared.constants import DIAGNOSTIC_DATA
 from rasa.shared.exceptions import RasaException
 from rasa.shared.nlu.constants import (
+    ACTION_NAME,
+    ACTION_TEXT,
     COMMANDS,
-    TEXT,
+    ENTITIES,
+    ENTITY_ATTRIBUTE_END,
+    ENTITY_ATTRIBUTE_START,
+    FEATURE_TYPE_SENTENCE,
+    FEATURE_TYPE_SEQUENCE,
     INTENT,
-    RESPONSE,
     INTENT_RESPONSE_KEY,
     METADATA,
-    METADATA_INTENT,
     METADATA_EXAMPLE,
-    ENTITIES,
-    ENTITY_ATTRIBUTE_START,
-    ENTITY_ATTRIBUTE_END,
+    METADATA_INTENT,
+    RESPONSE,
     RESPONSE_IDENTIFIER_DELIMITER,
-    FEATURE_TYPE_SENTENCE,
-    FEATURE_TYPE_SEQUENCE,
-    ACTION_TEXT,
-    ACTION_NAME,
+    TEXT,
     TEXT_TOKENS,
 )
-from rasa.shared.constants import DIAGNOSTIC_DATA
 
 if typing.TYPE_CHECKING:
     from rasa.shared.nlu.training_data.features import Features

rasa/shared/nlu/training_data/schemas/data_schema.py

@@ -1,4 +1,4 @@
-from typing import Dict, Text, Any
+from typing import Any, Dict, Text
 
 
 def entity_dict_schema() -> Dict[Text, Any]:

rasa/shared/nlu/training_data/schemas/responses.yml

@@ -64,6 +64,7 @@ schema;responses:
                 mapping:
                   type:
                     type: "str"
+                    enum: ['slot']
                   name:
                     type: "str"
                   value:

rasa/shared/nlu/training_data/synonyms_parser.py

@@ -1,9 +1,9 @@
-from typing import Any, Text, List, Dict
+from typing import Any, Dict, List, Text
 
 from rasa.shared.nlu.constants import (
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_END,
+    ENTITY_ATTRIBUTE_START,
+    ENTITY_ATTRIBUTE_VALUE,
 )
 
 

rasa/shared/nlu/training_data/training_data.py

@@ -1,31 +1,30 @@
+import copy
 import logging
+import operator
 import os
-from functools import cached_property
-from pathlib import Path
 import random
 from collections import Counter, OrderedDict
-import copy
+from functools import cached_property
 from os.path import relpath
-from typing import Any, Dict, List, Optional, Set, Text, Tuple, Callable
-import operator
+from pathlib import Path
+from typing import Any, Callable, Dict, List, Optional, Set, Text, Tuple
 
 import rasa.shared.data
 import rasa.shared.utils.io
 from rasa.shared.nlu.constants import (
-    RESPONSE,
-    INTENT_RESPONSE_KEY,
-    ENTITY_ATTRIBUTE_TYPE,
+    ACTION_NAME,
+    ENTITIES,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
-    NO_ENTITY_TAG,
+    ENTITY_ATTRIBUTE_TYPE,
     INTENT,
-    ENTITIES,
+    INTENT_RESPONSE_KEY,
+    NO_ENTITY_TAG,
+    RESPONSE,
     TEXT,
-    ACTION_NAME,
 )
-from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.nlu.training_data import util
-
+from rasa.shared.nlu.training_data.message import Message
 
 DEFAULT_TRAINING_DATA_OUTPUT_PATH = "training_data.yml"
 

rasa/shared/nlu/training_data/util.py

@@ -2,23 +2,24 @@ import json
 import logging
 import os
 import re
-from typing import Any, Dict, Optional, Text, Match, List
+from typing import Any, Dict, List, Match, Optional, Text
 
 import scipy.sparse
+
+import rasa.shared.data
+import rasa.shared.utils.io
+from rasa.shared.constants import UTTER_PREFIX
 from rasa.shared.nlu.constants import (
     ENTITIES,
-    EXTRACTOR,
-    PRETRAINED_EXTRACTORS,
-    ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_END,
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_TYPE,
-    ENTITY_ATTRIBUTE_ROLE,
     ENTITY_ATTRIBUTE_GROUP,
+    ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_START,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
+    EXTRACTOR,
+    PRETRAINED_EXTRACTORS,
 )
-from rasa.shared.constants import UTTER_PREFIX
-import rasa.shared.utils.io
-import rasa.shared.data
 
 logger = logging.getLogger(__name__)