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

rasa/shared/core/generator.py

@@ -1,49 +1,48 @@
-from collections import defaultdict, namedtuple, deque
-
 import copy
 import logging
 import random
+from collections import defaultdict, deque, namedtuple
 from contextlib import contextmanager
-
-from tqdm import tqdm
 from typing import (
-    Optional,
+    Any,
+    DefaultDict,
+    Deque,
+    Dict,
+    Generator,
+    Iterable,
     List,
-    Text,
+    Optional,
     Set,
-    Dict,
+    Text,
     Tuple,
-    Deque,
-    DefaultDict,
-    Any,
-    Iterable,
-    Generator,
 )
 
+from tqdm import tqdm
+
+import rasa.shared.utils.io
 from rasa.shared.constants import DOCS_URL_STORIES
 from rasa.shared.core.constants import SHOULD_NOT_BE_SET
 from rasa.shared.core.domain import Domain, State
 from rasa.shared.core.events import (
     ActionExecuted,
-    UserUttered,
     ActionReverted,
-    UserUtteranceReverted,
-    Restarted,
+    ActiveLoop,
     Event,
+    Restarted,
     SlotSet,
-    ActiveLoop,
+    UserUtteranceReverted,
+    UserUttered,
 )
-from rasa.shared.core.trackers import DialogueStateTracker, FrozenState
 from rasa.shared.core.slots import Slot
+from rasa.shared.core.trackers import DialogueStateTracker, FrozenState
 from rasa.shared.core.training_data.structures import (
-    StoryGraph,
+    GENERATED_CHECKPOINT_PREFIX,
     STORY_START,
-    StoryStep,
     RuleStep,
-    GENERATED_CHECKPOINT_PREFIX,
+    StoryGraph,
+    StoryStep,
 )
 from rasa.shared.utils.io import is_logging_disabled
-import rasa.shared.utils.io
 
 logger = logging.getLogger(__name__)
 

rasa/shared/core/slot_mappings.py

@@ -1,33 +1,33 @@
 import logging
-from typing import Text, Dict, Any, List, Optional, TYPE_CHECKING, Tuple, cast
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Tuple, cast
 
-from rasa.shared.constants import DOCS_URL_NLU_BASED_SLOTS, IGNORED_INTENTS
 import rasa.shared.utils.io
+from rasa.shared.constants import DOCS_URL_NLU_BASED_SLOTS, IGNORED_INTENTS
+from rasa.shared.core.constants import (
+    ACTIVE_FLOW,
+    ACTIVE_LOOP,
+    MAPPING_CONDITIONS,
+    MAPPING_TYPE,
+    REQUESTED_SLOT,
+    SLOT_MAPPINGS,
+    SlotMappingType,
+)
 from rasa.shared.core.slots import ListSlot, Slot
 from rasa.shared.nlu.constants import (
     ENTITIES,
-    ENTITY_ATTRIBUTE_TYPE,
-    ENTITY_ATTRIBUTE_ROLE,
     ENTITY_ATTRIBUTE_GROUP,
+    ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_VALUE,
     INTENT,
-    NOT_INTENT,
     INTENT_NAME_KEY,
+    NOT_INTENT,
     TEXT,
 )
-from rasa.shared.core.constants import (
-    ACTIVE_FLOW,
-    ACTIVE_LOOP,
-    REQUESTED_SLOT,
-    SLOT_MAPPINGS,
-    MAPPING_TYPE,
-    SlotMappingType,
-    MAPPING_CONDITIONS,
-)
 
 if TYPE_CHECKING:
-    from rasa.shared.core.trackers import DialogueStateTracker
     from rasa.shared.core.domain import Domain
+    from rasa.shared.core.trackers import DialogueStateTracker
     from rasa.shared.nlu.training_data.message import Message
     from rasa.utils.endpoints import EndpointConfig
 

rasa/shared/core/slots.py

@@ -3,14 +3,14 @@ from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Text, Type
 
 import rasa.shared.core.constants
-from rasa.shared.exceptions import RasaException
 import rasa.shared.utils.common
 import rasa.shared.utils.io
 from rasa.shared.constants import (
-    DOCS_URL_SLOTS,
-    DOCS_URL_NLU_BASED_SLOTS,
     DOCS_URL_CATEGORICAL_SLOTS,
+    DOCS_URL_NLU_BASED_SLOTS,
+    DOCS_URL_SLOTS,
 )
+from rasa.shared.exceptions import RasaException
 
 logger = logging.getLogger(__name__)
 

rasa/shared/core/trackers.py

@@ -7,21 +7,21 @@ import time
 from collections import deque
 from enum import Enum
 from typing import (
-    Dict,
-    Text,
+    TYPE_CHECKING,
     Any,
-    Optional,
-    Iterator,
+    Deque,
+    Dict,
+    FrozenSet,
     Generator,
+    Iterable,
+    Iterator,
+    List,
+    Optional,
+    Text,
+    Tuple,
     Type,
     TypeVar,
-    List,
-    Deque,
-    Iterable,
     Union,
-    FrozenSet,
-    Tuple,
-    TYPE_CHECKING,
     cast,
 )
 
@@ -34,50 +34,50 @@ from rasa.shared.constants import (
 from rasa.shared.core import events
 from rasa.shared.core.constants import (
     ACTION_LISTEN_NAME,
-    LOOP_NAME,
-    SHOULD_NOT_BE_SET,
-    PREVIOUS_ACTION,
-    ACTIVE_LOOP,
     ACTION_SESSION_START_NAME,
+    ACTIVE_LOOP,
     FOLLOWUP_ACTION,
+    LOOP_NAME,
+    PREVIOUS_ACTION,
+    SHOULD_NOT_BE_SET,
 )
 from rasa.shared.core.conversation import Dialogue
 from rasa.shared.core.domain import Domain, State
 from rasa.shared.core.events import (
-    DialogueStackUpdated,
-    RoutingSessionEnded,
-    SlotSet,
-    UserUttered,
     ActionExecuted,
-    Event,
-    Restarted,
+    ActionExecutionRejected,
     ActionReverted,
-    UserUtteranceReverted,
-    BotUttered,
     ActiveLoop,
-    SessionStarted,
-    ActionExecutionRejected,
+    BotUttered,
     DefinePrevUserUtteredFeaturization,
+    DialogueStackUpdated,
+    Event,
     FlowStarted,
+    Restarted,
+    RoutingSessionEnded,
+    SessionStarted,
+    SlotSet,
+    UserUtteranceReverted,
+    UserUttered,
 )
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.slots import AnySlot, Slot
 from rasa.shared.nlu.constants import (
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_TYPE,
-    ENTITY_ATTRIBUTE_GROUP,
-    ENTITY_ATTRIBUTE_ROLE,
-    ACTION_TEXT,
     ACTION_NAME,
+    ACTION_TEXT,
     ENTITIES,
+    ENTITY_ATTRIBUTE_GROUP,
+    ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
     METADATA_MODEL_ID,
 )
 
 if TYPE_CHECKING:
+    from rasa.dialogue_understanding.stack.dialogue_stack import DialogueStack
     from rasa.shared.core.events import NLUPredictionData
-    from rasa.shared.core.training_data.structures import Story
     from rasa.shared.core.training_data.story_writer.story_writer import StoryWriter
-    from rasa.dialogue_understanding.stack.dialogue_stack import DialogueStack
+    from rasa.shared.core.training_data.structures import Story
 
     EventTypeAlias = TypeVar("EventTypeAlias", bound=Event)
 

rasa/shared/core/training_data/loading.py

@@ -1,6 +1,6 @@
 import logging
 import os
-from typing import Text, Optional, List, Union
+from typing import List, Optional, Text, Union
 
 import rasa.shared.data
 import rasa.shared.utils.io

rasa/shared/core/training_data/story_reader/story_reader.py

@@ -1,14 +1,14 @@
 import logging
 from pathlib import Path
-from typing import Optional, Dict, Text, List, Any, Union
+from typing import Any, Dict, List, Optional, Text, Union
 
 from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import SlotSet, ActionExecuted, Event
-from rasa.shared.exceptions import RasaCoreException
+from rasa.shared.core.events import ActionExecuted, Event, SlotSet
 from rasa.shared.core.training_data.story_reader.story_step_builder import (
     StoryStepBuilder,
 )
 from rasa.shared.core.training_data.structures import StoryStep
+from rasa.shared.exceptions import RasaCoreException
 
 logger = logging.getLogger(__name__)
 

rasa/shared/core/training_data/story_reader/story_step_builder.py

@@ -1,17 +1,17 @@
 import logging
-from typing import Text, Optional, Dict, Any, List
+from typing import Any, Dict, List, Optional, Text
 
 import rasa.shared.core.training_data.structures
 import rasa.shared.utils.io
 from rasa.shared.constants import DOCS_URL_STORIES
-from rasa.shared.core.events import UserUttered, Event
+from rasa.shared.core.events import Event, UserUttered
 from rasa.shared.core.training_data.structures import (
-    Checkpoint,
     GENERATED_CHECKPOINT_PREFIX,
     GENERATED_HASH_LENGTH,
     STORY_START,
-    StoryStep,
+    Checkpoint,
     RuleStep,
+    StoryStep,
 )
 
 logger = logging.getLogger(__name__)

rasa/shared/core/training_data/story_reader/yaml_story_reader.py

@@ -1,60 +1,58 @@
 import copy
 import functools
 import json
-from json import JSONDecodeError
 import logging
-import structlog
-from pathlib import Path
 import re
+from json import JSONDecodeError
+from pathlib import Path
 from re import Match, Pattern
-from typing import Dict, Text, List, Any, Optional, Union, Tuple
+from typing import Any, Dict, List, Optional, Text, Tuple, Union
+
+import structlog
 
-from rasa.shared.core.domain import Domain
 import rasa.shared.data
-from rasa.shared.core.slots import TextSlot, ListSlot
-from rasa.shared.exceptions import YamlException
 import rasa.shared.utils.io
-from rasa.shared.core.constants import LOOP_NAME
+from rasa.shared.constants import (
+    DOCS_URL_RULES,
+    DOCS_URL_SLOTS,
+    DOCS_URL_STORIES,
+    INTENT_MESSAGE_PREFIX,
+    TEST_STORIES_FILE_PREFIX,
+)
+from rasa.shared.core.constants import LOOP_NAME, RULE_SNIPPET_ACTION_NAME
+from rasa.shared.core.domain import Domain
+from rasa.shared.core.events import (
+    ActiveLoop,
+    DialogueStackUpdated,
+    SlotSet,
+    UserUttered,
+)
+from rasa.shared.core.slots import ListSlot, TextSlot
+from rasa.shared.core.training_data.story_reader.story_reader import StoryReader
+from rasa.shared.core.training_data.structures import StoryStep
+from rasa.shared.exceptions import YamlException
 from rasa.shared.nlu.constants import (
+    ACTION_TEXT,
     ENTITIES,
     ENTITY_ATTRIBUTE_END,
     ENTITY_ATTRIBUTE_START,
     ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_VALUE,
+    EXTRACTOR,
+    FULL_RETRIEVAL_INTENT_NAME_KEY,
     INTENT,
     INTENT_NAME_KEY,
     INTENT_RANKING_KEY,
     PREDICTED_CONFIDENCE_KEY,
-    FULL_RETRIEVAL_INTENT_NAME_KEY,
-    ACTION_TEXT,
     TEXT,
-    EXTRACTOR,
 )
 from rasa.shared.nlu.training_data import entities_parser
-
-from rasa.shared.constants import (
-    INTENT_MESSAGE_PREFIX,
-    DOCS_URL_STORIES,
-    TEST_STORIES_FILE_PREFIX,
-    DOCS_URL_RULES,
-    DOCS_URL_SLOTS,
-)
-
-from rasa.shared.core.constants import RULE_SNIPPET_ACTION_NAME
-from rasa.shared.core.events import (
-    UserUttered,
-    SlotSet,
-    ActiveLoop,
-    DialogueStackUpdated,
-)
-from rasa.shared.core.training_data.story_reader.story_reader import StoryReader
-from rasa.shared.core.training_data.structures import StoryStep
 from rasa.shared.nlu.training_data.message import Message
 from rasa.shared.utils.yaml import (
-    validate_raw_yaml_using_schema_file,
+    is_key_in_yaml,
     read_yaml,
+    validate_raw_yaml_using_schema_file,
     validate_training_data_format_version,
-    is_key_in_yaml,
 )
 
 logger = logging.getLogger(__name__)

rasa/shared/core/training_data/story_writer/yaml_story_writer.py

@@ -1,55 +1,53 @@
+import json
 from collections import OrderedDict
 from pathlib import Path
-from typing import Any, Dict, List, Text, Union, Optional
+from typing import Any, Dict, List, Optional, Text, Union
 
 from ruamel import yaml
-import json
 from ruamel.yaml.comments import CommentedMap
 from ruamel.yaml.scalarstring import DoubleQuotedScalarString, LiteralScalarString
 
-import rasa.shared.utils.io
 import rasa.shared.core.constants
-from rasa.shared.constants import LATEST_TRAINING_DATA_FORMAT_VERSION
 import rasa.shared.core.events
+import rasa.shared.utils.io
+from rasa.shared.constants import LATEST_TRAINING_DATA_FORMAT_VERSION
 from rasa.shared.core.constants import FLOW_HASHES_SLOT
 from rasa.shared.core.events import (
-    UserUttered,
     ActionExecuted,
-    SlotSet,
     ActiveLoop,
-    Event,
     DialogueStackUpdated,
+    Event,
+    SlotSet,
+    UserUttered,
 )
-
 from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
-    KEY_STORIES,
-    KEY_STORY_NAME,
-    KEY_USER_INTENT,
-    KEY_ENTITIES,
     KEY_ACTION,
-    KEY_STEPS,
+    KEY_ACTIVE_LOOP,
+    KEY_BOT_END_TO_END_MESSAGE,
     KEY_CHECKPOINT,
-    KEY_SLOT_NAME,
     KEY_CHECKPOINT_SLOTS,
+    KEY_COMMANDS,
+    KEY_ENTITIES,
     KEY_OR,
-    KEY_USER_MESSAGE,
-    KEY_ACTIVE_LOOP,
-    KEY_BOT_END_TO_END_MESSAGE,
-    KEY_RULES,
-    KEY_RULE_FOR_CONVERSATION_START,
-    KEY_WAIT_FOR_USER_INPUT_AFTER_RULE,
     KEY_RULE_CONDITION,
+    KEY_RULE_FOR_CONVERSATION_START,
     KEY_RULE_NAME,
+    KEY_RULES,
+    KEY_SLOT_NAME,
     KEY_STACK_UPDATE,
-    KEY_COMMANDS,
+    KEY_STEPS,
+    KEY_STORIES,
+    KEY_STORY_NAME,
+    KEY_USER_INTENT,
+    KEY_USER_MESSAGE,
+    KEY_WAIT_FOR_USER_INPUT_AFTER_RULE,
 )
-
 from rasa.shared.core.training_data.story_writer.story_writer import StoryWriter
 from rasa.shared.core.training_data.structures import (
-    StoryStep,
-    Checkpoint,
     STORY_START,
+    Checkpoint,
     RuleStep,
+    StoryStep,
 )
 from rasa.shared.utils.yaml import write_yaml
 

rasa/shared/core/training_data/structures.py

@@ -1,21 +1,20 @@
 import json
 import logging
-from collections import deque, defaultdict
-
-import uuid
 import typing
+import uuid
+from collections import defaultdict, deque
 from typing import (
-    List,
-    Text,
+    Any,
     Deque,
     Dict,
+    List,
     Optional,
-    Tuple,
-    Any,
+    Sequence,
     Set,
-    ValuesView,
+    Text,
+    Tuple,
     Union,
-    Sequence,
+    ValuesView,
 )
 
 import rasa.shared.utils.io
@@ -27,16 +26,15 @@ from rasa.shared.core.constants import (
 from rasa.shared.core.conversation import Dialogue
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.events import (
-    UserUttered,
     ActionExecuted,
     Event,
     SessionStarted,
     SlotSet,
+    UserUttered,
 )
 from rasa.shared.core.trackers import DialogueStateTracker
 from rasa.shared.exceptions import RasaCoreException
 
-
 if typing.TYPE_CHECKING:
     import networkx as nx
 
@@ -740,9 +738,10 @@ class StoryGraph:
 
     def visualize(self, output_file: Optional[Text] = None) -> "nx.MultiDiGraph":
         import networkx as nx
-        from rasa.shared.core.training_data import visualization
         from colorhash import ColorHash
 
+        from rasa.shared.core.training_data import visualization
+
         graph = nx.MultiDiGraph()
         next_node_idx = [0]
         nodes = {"STORY_START": 0, "STORY_END": -1}

rasa/shared/core/training_data/visualization.py

@@ -1,15 +1,14 @@
-from collections import defaultdict, deque
-
 import random
+from collections import defaultdict, deque
 from typing import (
+    TYPE_CHECKING,
     Any,
-    Text,
-    List,
     Deque,
     Dict,
+    List,
     Optional,
     Set,
-    TYPE_CHECKING,
+    Text,
     Union,
     cast,
 )
@@ -18,22 +17,23 @@ import rasa.shared.utils.io
 from rasa.shared.constants import INTENT_MESSAGE_PREFIX
 from rasa.shared.core.constants import ACTION_LISTEN_NAME
 from rasa.shared.core.domain import Domain
-from rasa.shared.core.events import UserUttered, ActionExecuted, Event
+from rasa.shared.core.events import ActionExecuted, Event, UserUttered
 from rasa.shared.core.generator import TrainingDataGenerator
 from rasa.shared.core.training_data.structures import StoryGraph, StoryStep
 from rasa.shared.nlu.constants import (
+    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_VALUE,
     INTENT,
-    TEXT,
-    ENTITY_ATTRIBUTE_TYPE,
     INTENT_NAME_KEY,
+    TEXT,
 )
 
 if TYPE_CHECKING:
-    from rasa.shared.nlu.training_data.training_data import TrainingData
-    from rasa.shared.nlu.training_data.message import Message
     import networkx
 
+    from rasa.shared.nlu.training_data.message import Message
+    from rasa.shared.nlu.training_data.training_data import TrainingData
+
 EDGE_NONE_LABEL = "NONE"
 
 START_NODE_ID = 0

rasa/shared/data.py

@@ -5,16 +5,16 @@ import uuid
 from enum import Enum
 from pathlib import Path
 from typing import (
+    TYPE_CHECKING,
     Any,
+    Callable,
+    Iterable,
+    List,
+    Optional,
     Protocol,
-    TYPE_CHECKING,
+    Set,
     Text,
-    Optional,
     Union,
-    List,
-    Callable,
-    Set,
-    Iterable,
     runtime_checkable,
 )
 

rasa/shared/engine/caching.py

@@ -1,7 +1,6 @@
 import os
 from pathlib import Path
 
-
 DEFAULT_CACHE_LOCATION = Path(".rasa", "cache")
 DEFAULT_CACHE_NAME = "cache.db"
 DEFAULT_CACHE_SIZE_MB = 1000

rasa/shared/exceptions.py

@@ -1,11 +1,11 @@
 import json
-from typing import Optional, Text, Dict, Any
+from typing import Any, Dict, Optional, Text
 
 import jsonschema
 from ruamel.yaml.error import (
     MarkedYAMLError,
-    MarkedYAMLWarning,
     MarkedYAMLFutureWarning,
+    MarkedYAMLWarning,
 )
 
 

rasa/shared/importers/rasa.py

@@ -3,18 +3,17 @@ import os
 from typing import Dict, List, Optional, Text, Union
 
 import rasa.shared.core.flows.yaml_flows_io
-from rasa.shared.core.flows import FlowsList
-
 import rasa.shared.data
 import rasa.shared.utils.io
+from rasa.shared.core.domain import Domain, InvalidDomain
+from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
+    YAMLStoryReader,
+)
 from rasa.shared.core.training_data.structures import StoryGraph
 from rasa.shared.importers import utils
 from rasa.shared.importers.importer import TrainingDataImporter
 from rasa.shared.nlu.training_data.training_data import TrainingData
-from rasa.shared.core.domain import InvalidDomain, Domain
-from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
-    YAMLStoryReader,
-)
 from rasa.shared.utils.common import cached_method
 from rasa.shared.utils.yaml import read_model_configuration
 

rasa/shared/importers/utils.py

@@ -1,4 +1,4 @@
-from typing import Iterable, Text, Optional, List
+from typing import Iterable, List, Optional, Text
 
 from rasa.shared.core.domain import Domain
 from rasa.shared.core.flows import FlowsList

rasa/shared/nlu/constants.py

@@ -6,6 +6,9 @@ PREDICTED_COMMANDS = "predicted_commands"
 PROMPTS = "prompts"
 KEY_USER_PROMPT = "user_prompt"
 KEY_SYSTEM_PROMPT = "system_prompt"
+KEY_LLM_RESPONSE_METADATA = "llm_response_metadata"
+KEY_PROMPT_NAME = "prompt_name"
+KEY_COMPONENT_NAME = "component_name"
 LLM_COMMANDS = "llm_commands"  # needed for fine-tuning
 LLM_PROMPT = "llm_prompt"  # needed for fine-tuning
 FLOWS_FROM_SEMANTIC_SEARCH = "flows_from_semantic_search"

rasa/shared/nlu/training_data/entities_parser.py

@@ -1,20 +1,19 @@
+import logging
 import re
 from json import JSONDecodeError
-from typing import Text, List, Dict, Match, Optional, NamedTuple, Any
-import logging
+from typing import Any, Dict, List, Match, NamedTuple, Optional, Text
 
 import rasa.shared.nlu.training_data.util
 from rasa.shared.constants import DOCS_URL_TRAINING_DATA_NLU
 from rasa.shared.exceptions import InvalidEntityFormatException
 from rasa.shared.nlu.constants import (
-    ENTITY_ATTRIBUTE_VALUE,
-    ENTITY_ATTRIBUTE_TYPE,
     ENTITY_ATTRIBUTE_GROUP,
     ENTITY_ATTRIBUTE_ROLE,
+    ENTITY_ATTRIBUTE_TYPE,
+    ENTITY_ATTRIBUTE_VALUE,
 )
 from rasa.shared.nlu.training_data.message import Message
 
-
 GROUP_ENTITY_VALUE = "value"
 GROUP_ENTITY_TYPE = "entity"
 GROUP_ENTITY_DICT = "entity_dict"
@@ -167,8 +166,9 @@ def get_validated_dict(json_str: Text) -> Dict[Text, Text]:
         Deserialized and validated `json_str`.
     """
     import json
-    from rasa.shared.utils.yaml import validate_training_data
+
     import rasa.shared.nlu.training_data.schemas.data_schema as schema
+    from rasa.shared.utils.yaml import validate_training_data
 
     # add {} as they are not part of the regex
     try: