rasa-pro 3.12.22__py3-none-any.whl → 3.13.0__py3-none-any.whl

rasa/__main__.py

@@ -9,6 +9,7 @@ from rasa_sdk import __version__ as rasa_sdk_version
 
 import rasa.telemetry
 import rasa.utils.io
+import rasa.utils.licensing
 import rasa.utils.tensorflow.environment as tf_env
 from rasa import version
 from rasa.cli import (
@@ -105,11 +106,12 @@ def main(raw_arguments: Optional[List[str]] = None) -> None:
         raw_arguments: Arguments to parse. If not provided,
             arguments will be taken from the command line.
     """
+    rasa.utils.licensing.validate_license_from_env()
+
     warn_if_rasa_plus_package_installed()
     parse_last_positional_argument_as_model_path()
     arg_parser = create_argument_parser()
     cmdline_arguments = arg_parser.parse_args(raw_arguments)
-
     log_level = getattr(cmdline_arguments, "loglevel", None)
     logging_config_file = getattr(cmdline_arguments, "logging_config_file", None)
     configure_logging_and_warnings(
@@ -141,9 +143,6 @@ def main(raw_arguments: Optional[List[str]] = None) -> None:
             if not is_studio_command:
                 plugin_manager().hook.init_telemetry(endpoints_file=endpoints_file)
                 plugin_manager().hook.init_managers(endpoints_file=endpoints_file)
-                plugin_manager().hook.init_anonymization_pipeline(
-                    endpoints_file=endpoints_file
-                )
 
             cmdline_arguments.func(cmdline_arguments)
         elif hasattr(cmdline_arguments, "version"):

rasa/api.py

@@ -38,7 +38,7 @@ def run(
     """
     import rasa.core.run
     import rasa.shared.utils.common
-    from rasa.core.utils import AvailableEndpoints
+    from rasa.core.available_endpoints import AvailableEndpoints
     from rasa.shared.constants import DOCS_BASE_URL
     from rasa.shared.utils.cli import print_warning
 

rasa/cli/dialogue_understanding_test.py

@@ -16,9 +16,9 @@ from rasa.cli.arguments.default_arguments import (
     add_remote_storage_param,
 )
 from rasa.core.agent import Agent
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.exceptions import AgentNotReady
 from rasa.core.processor import MessageProcessor
-from rasa.core.utils import AvailableEndpoints
 from rasa.dialogue_understanding.commands import Command
 from rasa.dialogue_understanding.generator import LLMBasedCommandGenerator
 from rasa.dialogue_understanding.generator.command_parser import DEFAULT_COMMANDS

rasa/cli/e2e_test.py

@@ -18,8 +18,8 @@ from rasa.cli.arguments.default_arguments import (
     add_model_param,
     add_remote_storage_param,
 )
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.exceptions import AgentNotReady
-from rasa.core.utils import AvailableEndpoints
 from rasa.e2e_test.aggregate_test_stats_calculator import (
     AggregateTestStatsCalculator,
 )
@@ -51,7 +51,6 @@ from rasa.e2e_test.utils.io import (
 from rasa.e2e_test.utils.validation import validate_model_path
 from rasa.exceptions import RasaException
 from rasa.shared.constants import DEFAULT_ENDPOINTS_PATH, DEFAULT_MODELS_PATH
-from rasa.utils.beta import ensure_beta_feature_is_enabled
 from rasa.utils.endpoints import EndpointConfig
 
 RASA_PRO_BETA_FINE_TUNING_RECIPE_ENV_VAR_NAME = "RASA_PRO_BETA_FINE_TUNING_RECIPE"
@@ -155,12 +154,6 @@ def execute_e2e_tests(args: argparse.Namespace) -> None:
     Args:
         args: Commandline arguments.
     """
-    if args.coverage_report:
-        ensure_beta_feature_is_enabled(
-            "LLM fine-tuning recipe",
-            env_flag=RASA_PRO_BETA_FINE_TUNING_RECIPE_ENV_VAR_NAME,
-        )
-
     args.endpoints = rasa.cli.utils.get_validated_path(
         args.endpoints, "endpoints", DEFAULT_ENDPOINTS_PATH, True
     )

rasa/cli/evaluate.py

@@ -6,10 +6,10 @@ import rasa.cli.arguments.evaluate as arguments
 import rasa.shared.utils.cli
 from rasa import telemetry
 from rasa.cli import SubParsersAction
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.evaluation.marker_base import Marker, OperatorMarker
 from rasa.core.evaluation.marker_tracker_loader import MarkerTrackerLoader
-from rasa.core.tracker_store import TrackerStore
-from rasa.core.utils import AvailableEndpoints
+from rasa.core.tracker_stores.tracker_store import TrackerStore
 from rasa.shared.core.domain import Domain
 
 STATS_OVERALL_SUFFIX = "-overall.csv"

rasa/cli/export.py

@@ -16,10 +16,10 @@ from rasa.shared.constants import DOCS_URL_EVENT_BROKERS, DOCS_URL_TRACKER_STORE
 from rasa.shared.exceptions import RasaException
 
 if typing.TYPE_CHECKING:
+    from rasa.core.available_endpoints import AvailableEndpoints
     from rasa.core.brokers.broker import EventBroker
     from rasa.core.exporter import Exporter
-    from rasa.core.tracker_store import TrackerStore
-    from rasa.core.utils import AvailableEndpoints
+    from rasa.core.tracker_stores.tracker_store import TrackerStore
 
 logger = logging.getLogger(__name__)
 
@@ -67,7 +67,7 @@ def _get_tracker_store(endpoints: "AvailableEndpoints") -> "TrackerStore":
             f"Exiting. "
         )
 
-    from rasa.core.tracker_store import TrackerStore
+    from rasa.core.tracker_stores.tracker_store import TrackerStore
 
     return TrackerStore.create(endpoints.tracker_store)
 
@@ -184,6 +184,7 @@ async def _export_trackers(args: argparse.Namespace) -> None:
     event_broker = await _get_event_broker(endpoints)
     _prepare_event_broker(event_broker)
     requested_conversation_ids = _get_requested_conversation_ids(args.conversation_ids)
+    is_pii_enabled = True if endpoints.privacy else False
 
     from rasa.core.exporter import Exporter
 
@@ -191,6 +192,7 @@ async def _export_trackers(args: argparse.Namespace) -> None:
         tracker_store,
         event_broker,
         args.endpoints,
+        is_pii_enabled,
         requested_conversation_ids,
         args.minimum_timestamp,
         args.maximum_timestamp,

rasa/cli/inspect.py

@@ -9,6 +9,7 @@ from rasa import telemetry
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments import shell as arguments
 from rasa.core import constants
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.exceptions import ModelNotFound
 from rasa.model import get_local_model
@@ -83,6 +84,12 @@ def inspect(args: argparse.Namespace) -> None:
 
     model = get_validated_path(args.model, "model", DEFAULT_MODELS_PATH)
 
+    # Load endpoints with proper endpoint file location
+    # This will initialise the endpoints singleton properly so that
+    # it can be used safely throughout the codebase with
+    # `AvailableEndpoints.get_instance()`
+    AvailableEndpoints.get_instance(args.endpoints)
+
     try:
         model = get_local_model(model)
     except ModelNotFound:

rasa/cli/llm_fine_tuning.py

@@ -20,8 +20,8 @@ from rasa.cli.e2e_test import (
     read_test_cases,
     validate_model_path,
 )
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.exceptions import AgentNotReady
-from rasa.core.utils import AvailableEndpoints
 from rasa.dialogue_understanding.generator.llm_based_command_generator import (
     LLMBasedCommandGenerator,
 )

rasa/cli/project_templates/default/config.yml

@@ -1,44 +1,17 @@
 # The config recipe.
-# https://rasa.com/docs/rasa-pro/nlu-based-assistants/model-configuration/
 recipe: default.v1
 
 # The assistant project unique identifier
 # This default value must be replaced with a unique assistant name within your deployment
 assistant_id: placeholder_default
 
-# Configuration for Rasa NLU.
-# https://rasa.com/docs/rasa-pro/nlu-based-assistants/components/
 language: en
-
 pipeline:
-# # No configuration for the NLU pipeline was provided. The following default pipeline was used to train your model.
-# # If you'd like to customize it, uncomment and adjust the pipeline.
-# # See https://rasa.com/docs/rasa-pro/nlu-based-assistants/tuning-your-model for more information.
-#   - name: WhitespaceTokenizer
-#   - name: RegexFeaturizer
-#   - name: LexicalSyntacticFeaturizer
-#   - name: CountVectorsFeaturizer
-#   - name: CountVectorsFeaturizer
-#     analyzer: char_wb
-#     min_ngram: 1
-#     max_ngram: 4
-#   - name: DIETClassifier
-#     epochs: 100
-#   - name: EntitySynonymMapper
-#   - name: ResponseSelector
-#     epochs: 100
-#   - name: FallbackClassifier
-#     threshold: 0.3
-#     ambiguity_threshold: 0.1
+- name: CompactLLMCommandGenerator
+  llm:
+    model_group: openai-gpt-4o
 
 # Configuration for Rasa Core.
-# https://rasa.com/docs/rasa-pro/concepts/policies/policy-overview/
 policies:
-# # No configuration for policies was provided. The following default policies were used to train your model.
-# # If you'd like to customize them, uncomment and adjust the policies.
-# # See https://rasa.com/docs/rasa-pro/concepts/policies/policy-overview/ for more information.
-#   - name: MemoizationPolicy
-#   - name: TEDPolicy
-#     max_history: 5
-#     epochs: 100
-#   - name: RulePolicy
+- name: FlowPolicy
+- name: IntentlessPolicy

rasa/cli/project_templates/{calm → default}/e2e_tests/cancelations/user_cancels_during_a_correction.yml

@@ -13,4 +13,4 @@ test_cases:
       - utter: utter_corrected_previous_input
       - utter: utter_ask_add_contact_name
       - user: I changed my mind, stop.
-      - utter: utter_flow_cancelled_rasa
+      - utter: utter_flow_cancelled_rasa

rasa/cli/project_templates/{calm → default}/e2e_tests/cancelations/user_changes_mind_on_a_whim.yml

@@ -4,4 +4,4 @@ test_cases:
       - user: I want to add contact
       - utter: utter_ask_add_contact_handle
       - user: Stop
-      - utter: utter_flow_cancelled_rasa
+      - utter: utter_flow_cancelled_rasa

rasa/cli/project_templates/{calm → default}/e2e_tests/corrections/user_corrects_contact_handle.yml

@@ -17,4 +17,4 @@ test_cases:
           - add_contact_name: "Barbar"
       - utter: utter_ask_add_contact_confirmation
       - user: Yes
-      - utter: utter_contact_added
+      - utter: utter_contact_added

rasa/cli/project_templates/{calm → default}/e2e_tests/corrections/user_corrects_contact_name.yml

@@ -16,4 +16,4 @@ test_cases:
       - utter: utter_corrected_previous_input
       - utter: utter_ask_add_contact_confirmation
       - user: Yes
-      - utter: utter_contact_added
+      - utter: utter_contact_added

rasa/cli/project_templates/{calm → default}/e2e_tests/happy_paths/user_adds_contact_to_their_list.yml

@@ -12,4 +12,4 @@ test_cases:
           - add_contact_name: Bart
       - utter: utter_ask_add_contact_confirmation
       - user: Yes
-      - utter: utter_contact_added
+      - utter: utter_contact_added

rasa/cli/project_templates/{calm → default}/e2e_tests/happy_paths/user_lists_contacts.yml

@@ -2,4 +2,4 @@ test_cases:
   - test_case: user lists their contacts
     steps:
       - user: Please show me my contacts
-      - utter: utter_list_contacts
+      - utter: utter_list_contacts

rasa/cli/project_templates/{calm → default}/e2e_tests/happy_paths/user_removes_contact.yml

@@ -8,4 +8,4 @@ test_cases:
           - remove_contact_handle: "@MaryLu"
       - utter: utter_ask_remove_contact_confirmation
       - user: Yes
-      - utter: utter_remove_contact_success
+      - utter: utter_remove_contact_success

rasa/cli/project_templates/{calm → default}/e2e_tests/happy_paths/user_removes_contact_from_list.yml

@@ -9,4 +9,4 @@ test_cases:
           - remove_contact_handle: "@MaryLu"
       - utter: utter_ask_remove_contact_confirmation
       - user: yes
-      - utter: utter_remove_contact_success
+      - utter: utter_remove_contact_success

rasa/cli/project_templates/default/endpoints.yml

@@ -10,8 +10,8 @@
 # Server which runs your custom actions.
 # https://rasa.com/docs/rasa-pro/concepts/custom-actions
 
-#action_endpoint:
-#  url: "http://localhost:5055/webhook"
+action_endpoint:
+  actions_module: "actions"
 
 # Tracker store which is used to store the conversations.
 # By default the conversations are stored in memory.
@@ -40,3 +40,19 @@
 #  username: username
 #  password: password
 #  queue: queue
+
+# The lines below activate contextual rephrasing, using the default OpenAI language model.
+# Ensure the OPENAI_API_KEY is set to prevent any missing API key errors.
+# For more details, refer to the documentation:
+# https://rasa.com/docs/rasa-pro/concepts/contextual-response-rephraser
+# To enable the rephraser, remove the comment symbols in the lines below.
+#nlg:
+#   type: rephrase
+
+model_groups:
+  - id: openai-gpt-4o
+    models:
+      - provider: openai
+        model: gpt-4o-2024-11-20
+        request_timeout: 7
+        max_tokens: 256

rasa/cli/project_templates/defaults.py

@@ -0,0 +1,133 @@
+from __future__ import annotations
+
+import tempfile
+from typing import Any, Dict, List, Optional, Text
+
+from pydantic import BaseModel, ConfigDict, Field
+
+from rasa.dialogue_understanding.patterns.domain_for_patterns import (
+    ContextField,
+    generate_domain_for_default_patterns,
+)
+from rasa.shared.core.domain import Domain
+from rasa.shared.importers.importer import FlowSyncImporter, TrainingDataImporter
+from rasa.shared.utils.llm import SystemPrompts, get_system_default_prompts
+from rasa.shared.utils.yaml import read_yaml, write_yaml
+
+
+class Response(BaseModel):
+    text: str = Field(..., description="The human-readable response string.")
+    metadata: Optional[Dict[str, Any]] = Field(
+        None, description="Optional metadata used by Rasa at runtime."
+    )
+
+
+class Slot(BaseModel):
+    name: Text = Field(..., description="Name of the slot.")
+    type: Text = Field(..., description="Type of the slot.")
+    mappings: Optional[List[Dict[str, Any]]] = Field(
+        None, description="Mappings for the slot."
+    )
+    model_config = ConfigDict(extra="allow")
+
+
+class PatternFlow(BaseModel):
+    description: str = Field(..., description="Description of the flow.")
+    name: Text = Field(..., description="Name of the flow.")
+    steps: List[Any] = Field(..., description="List of steps in the flow.")
+    nlu_trigger: Optional[Any] = Field(None, description="NLU trigger for the flow.")
+    persisted_slots: Optional[List[Text]] = Field(None, description="Slots to persist.")
+
+
+class PatternDefaults(BaseModel):
+    """Content of `default_flows_for_patterns.yml` after it is loaded from disk."""
+
+    responses: Dict[Text, List[Response]] = Field(
+        default_factory=dict,
+        description="Response templates available out-of-the-box.",
+    )
+    slots: Dict[Text, Slot] = Field(
+        default_factory=dict, description="Slot definitions used by the default flows."
+    )
+    flows: Dict[Text, PatternFlow] = Field(
+        default_factory=dict, description="Default flows (conversation patterns)."
+    )
+
+
+class RasaDefaults(PatternDefaults):
+    """Full set of defaults that Rasa injects when a project is created."""
+
+    prompts: SystemPrompts = Field(..., description="Built-in system prompts.")
+    actions: List[Text] = Field(..., description="Default actions shipped by Rasa.")
+    intents: List[Text] = Field(..., description="Default intents shipped by Rasa.")
+    contexts: Dict[Text, ContextField] = Field(
+        ..., description="Context fields that can appear on the conversation stack."
+    )
+
+    class Config:
+        extra = "forbid"
+        validate_assignment = True
+
+
+def _get_domain_from_importer(config: Dict[Text, Any]) -> Domain:
+    """Get the domain from the TrainingDataImporter.
+
+    Args:
+        config: The config.yml file data.
+
+    Returns:
+        A Domain object .
+    """
+    with tempfile.NamedTemporaryFile("w+", suffix=".yml") as tmp:
+        write_yaml(config, tmp.name)
+        importer = TrainingDataImporter.load_from_config(
+            domain_path=FlowSyncImporter.default_pattern_path(),
+            config_path=tmp.name,
+        )
+        return importer.get_domain()
+
+
+def get_pattern_defaults(config: Dict[Text, Any]) -> PatternDefaults:
+    """Get the default flows, responses and slots for patterns.
+
+    Args:
+        config: The config.yml file data.
+
+    Returns:
+        A PatternDefaults object containing the defaults.
+    """
+    domain = _get_domain_from_importer(config)
+    domain_dict = domain.as_dict()
+
+    # Make sure all the slots are exported, including the builtin slots
+    domain_dict["slots"] = {slot.name: slot.to_dict() for slot in domain.slots}
+
+    return PatternDefaults(**domain_dict)
+
+
+def get_rasa_defaults(config_yaml: Text, endpoints_yaml: Text) -> RasaDefaults:
+    """Get the default values for a Rasa project.
+
+    Args:
+        config_yaml: The content of the config.yml file.
+        endpoints_yaml: The content of the endpoints.yml file.
+
+    Returns:
+        A RasaDefaults object containing the default values for the project.
+    """
+    config = read_yaml(config_yaml)
+    endpoints = read_yaml(endpoints_yaml)
+
+    prompts = get_system_default_prompts(config, endpoints)
+    pattern_domain = generate_domain_for_default_patterns()
+    pattern_defaults = get_pattern_defaults(config)
+
+    return RasaDefaults(
+        prompts=prompts,
+        actions=pattern_domain.actions,
+        intents=pattern_domain.intents,
+        contexts=pattern_domain.contexts,
+        responses=pattern_defaults.responses,
+        slots=pattern_defaults.slots,
+        flows=pattern_defaults.flows,
+    )

rasa/cli/project_templates/tutorial/config.yml

@@ -1,7 +1,7 @@
 recipe: default.v1
 language: en
 pipeline:
-- name: SingleStepLLMCommandGenerator
+- name: CompactLLMCommandGenerator
   llm:
     model_group: rasa_command_generation_model
   flow_retrieval:

rasa/cli/project_templates/tutorial/endpoints.yml

@@ -51,5 +51,5 @@ model_groups:
   - id: rasa_command_generation_model
     models:
       - provider: rasa
-        model: rasa/cmd_gen_codellama_13b_calm_demo
+        model: rasa/command-generator-llama-3.1-8b-instruct
         api_base: "https://tutorial-llm.rasa.ai"

rasa/cli/run.py

@@ -109,7 +109,7 @@ def run(args: argparse.Namespace) -> None:
     # configured
 
     import rasa.model
-    from rasa.core.utils import AvailableEndpoints
+    from rasa.core.available_endpoints import AvailableEndpoints
 
     # start server if remote storage is configured
     if args.remote_storage is not None:

rasa/cli/scaffold.py

@@ -20,18 +20,17 @@ from rasa.shared.utils.cli import print_error_and_exit, print_success
 
 
 class ProjectTemplateName(Enum):
-    """Enum of the different project templates."""
+    """Enum of CALM project templates."""
 
     DEFAULT = "default"
     TUTORIAL = "tutorial"
-    CALM = "calm"
 
     def __str__(self) -> str:
         return self.value
 
 
 template_domain_path = defaultdict(lambda: DEFAULT_DOMAIN_PATH)
-template_domain_path[ProjectTemplateName.CALM] = "domain"
+template_domain_path[ProjectTemplateName.DEFAULT] = "domain"
 
 
 def add_subparser(

rasa/cli/shell.py

@@ -6,6 +6,7 @@ from typing import List
 from rasa import telemetry
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments import shell as arguments
+from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.exceptions import ModelNotFound
 from rasa.model import get_local_model
@@ -105,7 +106,11 @@ def shell(args: argparse.Namespace) -> None:
     from rasa.shared.constants import DEFAULT_MODELS_PATH
 
     args.connector = "cmdline"
-
+    # Load endpoints with proper endpoint file location
+    # This will initialise the endpoints singleton properly so that
+    # it can be used safely throughout the codebase with
+    # `AvailableEndpoints.get_instance()`
+    AvailableEndpoints.get_instance(args.endpoints)
     model = get_validated_path(args.model, "model", DEFAULT_MODELS_PATH)
 
     try:

rasa/cli/studio/download.py

@@ -2,12 +2,6 @@ import argparse
 from typing import List
 
 from rasa.cli import SubParsersAction
-from rasa.cli.arguments.default_arguments import (
-    add_config_param,
-    add_endpoint_param,
-)
-from rasa.cli.arguments.train import add_data_param, add_domain_param
-from rasa.shared.constants import DEFAULT_CONFIG_PATH, DEFAULT_ENDPOINTS_PATH
 from rasa.studio.download import handle_download
 
 
@@ -37,25 +31,9 @@ def add_subparser(
 
 def set_studio_download_arguments(parser: argparse.ArgumentParser) -> None:
     """Add arguments for running `rasa studio download`."""
-    add_domain_param(parser)
-    add_data_param(parser)
-    add_config_param(parser, default=DEFAULT_CONFIG_PATH)
-    add_endpoint_param(
-        parser,
-        "Configuration file for the model endpoints.",
-        default=DEFAULT_ENDPOINTS_PATH,
-    )
-
     parser.add_argument(
         "assistant_name",
         default=None,
-        nargs=1,
         type=str,
         help="Name of the assistant on Rasa Studio",
     )
-
-    parser.add_argument(
-        "--overwrite",
-        action="store_true",
-        help="Overwrite local data with data from Rasa Studio",
-    )

rasa/cli/studio/link.py

@@ -0,0 +1,36 @@
+import argparse
+from typing import List, Text
+
+from rasa.cli import SubParsersAction
+from rasa.shared.constants import (
+    DEFAULT_DOMAIN_PATH,
+)
+from rasa.studio.link import handle_link
+
+
+def add_subparser(
+    subparsers: SubParsersAction,
+    parents: List[argparse.ArgumentParser],
+    domain: Text = DEFAULT_DOMAIN_PATH,
+) -> None:
+    """Register the `rasa studio link` command with the main CLI.
+
+    Args:
+        subparsers: subparser we are going to attach to
+        parents: Parent parsers, needed to ensure tree structure in argparse
+        domain: Path to the assistant's domain file.
+    """
+    link_parser = subparsers.add_parser(
+        "link",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Link the current project to an existing Rasa Studio assistant.",
+    )
+
+    link_parser.add_argument(
+        "assistant_name",
+        type=str,
+        help="Name of the assistant in Rasa Studio.",
+    )
+    link_parser.set_defaults(func=handle_link)

rasa/cli/studio/pull.py

@@ -0,0 +1,79 @@
+import argparse
+from typing import List, Text
+
+from rasa.cli import SubParsersAction
+from rasa.cli.arguments.default_arguments import (
+    add_config_param,
+    add_data_param,
+    add_domain_param,
+    add_endpoint_param,
+)
+from rasa.shared.constants import (
+    DEFAULT_CONFIG_PATH,
+    DEFAULT_DOMAIN_PATH,
+    DEFAULT_ENDPOINTS_PATH,
+)
+from rasa.studio.pull.pull import (
+    handle_pull,
+    handle_pull_config,
+    handle_pull_endpoints,
+)
+
+
+def add_subparser(
+    subparsers: SubParsersAction,
+    parents: List[argparse.ArgumentParser],
+    domain: Text = DEFAULT_DOMAIN_PATH,
+) -> None:
+    """Register `rasa studio pull` and its sub-commands.
+
+    Args:
+        subparsers: subparser we are going to attach to
+        parents: Parent parsers, needed to ensure tree structure in argparse
+        domain: Path to the assistant's domain file.
+    """
+    pull_parser = subparsers.add_parser(
+        "pull",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Pull assistant data from Rasa Studio.",
+    )
+    pull_subparsers = pull_parser.add_subparsers()
+
+    # Arguments for pulling configuration only
+    config_parser = pull_subparsers.add_parser(
+        "config",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Pull only the assistant configuration file.",
+    )
+    add_config_param(config_parser, default=DEFAULT_CONFIG_PATH)
+    config_parser.set_defaults(func=handle_pull_config)
+
+    # Arguments for pulling endpoints only
+    endpoints_parser = pull_subparsers.add_parser(
+        "endpoints",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Pull only the endpoints configuration file.",
+    )
+    add_endpoint_param(
+        endpoints_parser,
+        "Configuration file for the model endpoints.",
+        default=DEFAULT_ENDPOINTS_PATH,
+    )
+    endpoints_parser.set_defaults(func=handle_pull_endpoints)
+
+    # Arguments for pulling the whole assistant data
+    add_domain_param(pull_parser, domain)
+    add_data_param(pull_parser)
+    add_config_param(pull_parser, default=DEFAULT_CONFIG_PATH)
+    add_endpoint_param(
+        pull_parser,
+        "Configuration file for the model endpoints.",
+        default=DEFAULT_ENDPOINTS_PATH,
+    )
+    pull_parser.set_defaults(func=handle_pull)