rasa-pro 3.11.5__py3-none-any.whl → 3.12.0__py3-none-any.whl

rasa/cli/project_templates/calm/domain/shared.yml

@@ -4,7 +4,4 @@ slots:
   return_value:
     type: any
     mappings:
-      - type: custom
-        action: add_contact
-      - type: custom
-        action: remove_contact
+      - type: controlled

rasa/cli/project_templates/calm/endpoints.yml

@@ -50,9 +50,9 @@ action_endpoint:
 #   type: rephrase
 
 model_groups:
-  - id: openai-gpt-4
+  - id: openai-gpt-4o
     models:
       - provider: openai
-        model: openai/gpt-4
+        model: gpt-4o-2024-11-20
         request_timeout: 7
         max_tokens: 256

rasa/cli/project_templates/tutorial/actions/actions.py

@@ -1,7 +1,8 @@
-from typing import Any, Text, Dict, List
+from typing import Any, Dict, List, Text
+
 from rasa_sdk import Action, Tracker
-from rasa_sdk.executor import CollectingDispatcher
 from rasa_sdk.events import SlotSet
+from rasa_sdk.executor import CollectingDispatcher
 
 
 class ActionCheckSufficientFunds(Action):

rasa/cli/shell.py

@@ -1,18 +1,17 @@
 import argparse
 import logging
 import uuid
-
 from typing import List
 
 from rasa import telemetry
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments import shell as arguments
 from rasa.engine.storage.local_model_storage import LocalModelStorage
+from rasa.exceptions import ModelNotFound
 from rasa.model import get_local_model
 from rasa.shared.constants import ASSISTANT_ID_KEY
 from rasa.shared.data import TrainingType
 from rasa.shared.utils.cli import print_error
-from rasa.exceptions import ModelNotFound
 
 logger = logging.getLogger(__name__)
 
@@ -63,9 +62,9 @@ def add_subparser(
 
 def shell_nlu(args: argparse.Namespace) -> None:
     """Talk with an NLU only bot though the command line."""
+    import rasa.nlu.run
     from rasa.cli.utils import get_validated_path
     from rasa.shared.constants import DEFAULT_MODELS_PATH
-    import rasa.nlu.run
 
     args.connector = "cmdline"
 
@@ -96,7 +95,7 @@ def shell_nlu(args: argparse.Namespace) -> None:
         )
         return
 
-    telemetry.track_shell_started("nlu")
+    telemetry.track_shell_started("nlu", metadata.assistant_id)
     rasa.nlu.run.run_cmdline(model)
 
 
@@ -130,12 +129,12 @@ def shell(args: argparse.Namespace) -> None:
     if metadata.training_type == TrainingType.NLU:
         import rasa.nlu.run
 
-        telemetry.track_shell_started("nlu")
+        telemetry.track_shell_started("nlu", metadata.assistant_id)
 
         rasa.nlu.run.run_cmdline(model)
     else:
         import rasa.cli.run
 
-        telemetry.track_shell_started("rasa")
+        telemetry.track_shell_started("rasa", metadata.assistant_id)
 
         rasa.cli.run.run(args)

rasa/cli/studio/download.py

@@ -7,8 +7,7 @@ from rasa.cli.arguments.default_arguments import (
     add_endpoint_param,
 )
 from rasa.cli.arguments.train import add_data_param, add_domain_param
-from rasa.shared.constants import DEFAULT_ENDPOINTS_PATH, DEFAULT_CONFIG_PATH
-
+from rasa.shared.constants import DEFAULT_CONFIG_PATH, DEFAULT_ENDPOINTS_PATH
 from rasa.studio.download import handle_download
 
 

rasa/cli/studio/studio.py

@@ -3,16 +3,15 @@ from typing import List, Optional, Tuple
 from urllib.parse import ParseResult, urlparse
 
 import questionary
-from rasa.cli import SubParsersAction
 
-import rasa.shared.utils.cli
 import rasa.cli.studio.download
 import rasa.cli.studio.train
 import rasa.cli.studio.upload
+import rasa.shared.utils.cli
+from rasa.cli import SubParsersAction
 from rasa.studio.auth import StudioAuth
 from rasa.studio.config import StudioConfig
 
-
 DEFAULT_REALM_NAME = "rasa-studio"
 
 DEFAULT_CLIENT_ID = "admin-cli"

rasa/cli/studio/train.py

@@ -3,7 +3,6 @@ from typing import List
 
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments.train import set_train_arguments
-
 from rasa.studio.train import handle_train
 
 

rasa/cli/telemetry.py

@@ -2,11 +2,11 @@ import argparse
 import textwrap
 from typing import List
 
+import rasa.cli.utils
+import rasa.shared.utils.cli
 from rasa import telemetry
 from rasa.cli import SubParsersAction
-import rasa.cli.utils
 from rasa.shared.constants import DOCS_URL_TELEMETRY
-import rasa.shared.utils.cli
 from rasa.utils import licensing
 
 

rasa/cli/test.py

@@ -2,31 +2,31 @@ import argparse
 import asyncio
 import logging
 import os
-from typing import List, Optional, Text, Dict, Union, Any
+from typing import Any, Dict, List, Optional, Text, Union
 
-from rasa.cli import SubParsersAction
+import rasa.cli.utils
 import rasa.shared.data
-from rasa.shared.exceptions import YamlException
-import rasa.shared.utils.io
 import rasa.shared.utils.cli
+import rasa.shared.utils.io
+import rasa.utils.common
+from rasa.cli import SubParsersAction
 from rasa.cli.arguments import test as arguments
 from rasa.core.constants import (
     FAILED_STORIES_FILE,
-    SUCCESSFUL_STORIES_FILE,
     STORIES_WITH_WARNINGS_FILE,
+    SUCCESSFUL_STORIES_FILE,
 )
 from rasa.shared.constants import (
     CONFIG_SCHEMA_FILE,
-    DEFAULT_E2E_TESTS_PATH,
     DEFAULT_CONFIG_PATH,
-    DEFAULT_MODELS_PATH,
     DEFAULT_DATA_PATH,
+    DEFAULT_E2E_TESTS_PATH,
+    DEFAULT_MODELS_PATH,
     DEFAULT_RESULTS_PATH,
 )
-from rasa.shared.utils.yaml import validate_raw_yaml_using_schema_file
-import rasa.cli.utils
-import rasa.utils.common
+from rasa.shared.exceptions import YamlException
 from rasa.shared.importers.importer import TrainingDataImporter
+from rasa.shared.utils.yaml import validate_raw_yaml_using_schema_file
 
 logger = logging.getLogger(__name__)
 
@@ -95,9 +95,9 @@ def _print_core_test_execution_info(args: argparse.Namespace) -> None:
 async def run_core_test_async(args: argparse.Namespace) -> None:
     """Run core tests."""
     from rasa.model_testing import (
-        test_core_models_in_directory,
         test_core,
         test_core_models,
+        test_core_models_in_directory,
     )
 
     stories = rasa.cli.utils.get_validated_path(

rasa/cli/train.py

@@ -24,6 +24,7 @@ from rasa.shared.constants import (
     LLM_CONFIG_KEY,
 )
 from rasa.shared.importers.importer import TrainingDataImporter
+from rasa.shared.utils.common import display_research_study_prompt
 
 structlogger = structlog.getLogger(__name__)
 
@@ -89,6 +90,7 @@ def _check_nlg_endpoint_validity(endpoint: Union[Path, str]) -> None:
                 f"{endpoint}. Please make sure the NLG configuration is correct."
             ),
         )
+        display_research_study_prompt()
         sys.exit(1)
 
 
@@ -155,6 +157,7 @@ def run_training(args: argparse.Namespace, can_exit: bool = False) -> Optional[T
         keep_local_model_copy=args.keep_local_model_copy,
     )
     if training_result.code != 0 and can_exit:
+        display_research_study_prompt()
         sys.exit(training_result.code)
 
     return training_result.model

rasa/cli/utils.py

@@ -1,30 +1,33 @@
-import json
 import argparse
-import structlog
 import importlib
+import json
 import os
 import sys
 import time
 from pathlib import Path
 from types import FrameType
-from typing import Any, Dict, List, Optional, TYPE_CHECKING, Text, Union, overload
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Union, overload
+
 import randomname
+import structlog
 
 import rasa.shared.utils.cli
 import rasa.shared.utils.io
-from rasa.shared.importers.importer import TrainingDataImporter
+from rasa import telemetry
 from rasa.shared.constants import (
     ASSISTANT_ID_DEFAULT_VALUE,
     ASSISTANT_ID_KEY,
     DEFAULT_CONFIG_PATH,
 )
-from rasa import telemetry
+from rasa.shared.importers.importer import TrainingDataImporter
+from rasa.shared.utils.common import display_research_study_prompt
 from rasa.shared.utils.yaml import read_config_file
 from rasa.utils.io import write_yaml
 
 if TYPE_CHECKING:
     from questionary import Question
     from typing_extensions import Literal
+
     from rasa.validator import Validator
 
 structlogger = structlog.get_logger()
@@ -189,6 +192,7 @@ def validate_config_path(
                 f"Use '--config' to specify a valid config file."
             ),
         )
+        display_research_study_prompt()
         sys.exit(1)
 
     return str(config)
@@ -221,6 +225,7 @@ def validate_mandatory_config_keys(
                 )
             ),
         )
+        display_research_study_prompt()
         sys.exit(1)
 
     return str(config)
@@ -246,6 +251,7 @@ def validate_files(
     importer: TrainingDataImporter,
     stories_only: bool = False,
     flows_only: bool = False,
+    translations_only: bool = False,
 ) -> None:
     """Validates either the story structure or the entire project.
 
@@ -255,6 +261,7 @@ def validate_files(
         importer: The `TrainingDataImporter` to use to load the training data.
         stories_only: If `True`, only the story structure is validated.
         flows_only: If `True`, only the flows are validated.
+        translations_only: If `True`, only the translations data is validated.
     """
     from rasa.validator import Validator
 
@@ -264,12 +271,15 @@ def validate_files(
         all_good = _validate_story_structure(validator, max_history, fail_on_warnings)
     elif flows_only:
         all_good = validator.verify_flows()
+    elif translations_only:
+        all_good = validator.verify_translations()
     else:
         if importer.get_domain().is_empty():
             structlogger.error(
                 "cli.validate_files.empty_domain",
                 event_info="Encountered empty domain during validation.",
             )
+            display_research_study_prompt()
             sys.exit(1)
 
         valid_domain = _validate_domain(validator)
@@ -278,6 +288,7 @@ def validate_files(
             validator, max_history, fail_on_warnings
         )
         valid_flows = validator.verify_flows()
+        valid_translations = validator.verify_translations(summary_mode=True)
         valid_CALM_slot_mappings = validator.validate_CALM_slot_mappings()
 
         all_good = (
@@ -285,6 +296,7 @@ def validate_files(
             and valid_nlu
             and valid_stories
             and valid_flows
+            and valid_translations
             and valid_CALM_slot_mappings
         )
 
@@ -296,6 +308,7 @@ def validate_files(
             "cli.validate_files.project_validation_error",
             event_info="Project validation completed with errors.",
         )
+        display_research_study_prompt()
         sys.exit(1)
 
 
@@ -307,6 +320,10 @@ def _validate_domain(validator: "Validator") -> bool:
     valid_slot_mappings = validator.verify_slot_mappings()
     valid_responses = validator.check_for_no_empty_parenthesis_in_responses()
     valid_buttons = validator.validate_button_payloads()
+    valid_slot_validation = validator.verify_slot_validation()
+    valid_conditional_responses = (
+        validator.validate_conditional_response_variation_predicates()
+    )
     return (
         valid_domain_validity
         and valid_actions_in_stories_rules
@@ -315,6 +332,8 @@ def _validate_domain(validator: "Validator") -> bool:
         and valid_slot_mappings
         and valid_responses
         and valid_buttons
+        and valid_slot_validation
+        and valid_conditional_responses
     )
 
 
@@ -364,6 +383,7 @@ def cancel_cause_not_found(
             f"with '--{parameter}'."
         ),
     )
+    display_research_study_prompt()
     sys.exit(1)
 
 

rasa/constants.py

@@ -27,7 +27,6 @@ ENV_LOG_LEVEL_LIBRARIES = "LOG_LEVEL_LIBRARIES"
 ENV_LOG_LEVEL_MATPLOTLIB = "LOG_LEVEL_MATPLOTLIB"
 ENV_LOG_LEVEL_RABBITMQ = "LOG_LEVEL_RABBITMQ"
 ENV_LOG_LEVEL_KAFKA = "LOG_LEVEL_KAFKA"
-ENV_LOG_LEVEL_MLFLOW = "LOG_LEVEL_MLFLOW"
 
 DEFAULT_SANIC_WORKERS = 1
 ENV_SANIC_WORKERS = "SANIC_WORKERS"

rasa/core/__init__.py

@@ -1,7 +1,6 @@
 import logging
 
 import rasa
-
 from rasa.core.nlg.contextual_response_rephraser import ContextualResponseRephraser
 from rasa.core.policies.enterprise_search_policy import EnterpriseSearchPolicy
 from rasa.core.policies.intentless_policy import IntentlessPolicy