rasa-pro 3.13.0.dev7__tar.gz → 3.13.0.dev8__tar.gz

{rasa_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rasa-pro
-Version: 3.13.0.dev7
+Version: 3.13.0.dev8
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
 Author: Rasa Technologies GmbH
@@ -18,6 +18,7 @@ Provides-Extra: full
 Provides-Extra: gh-release-notes
 Provides-Extra: jieba
 Provides-Extra: metal
+Provides-Extra: pii
 Provides-Extra: spacy
 Provides-Extra: transformers
 Requires-Dist: CacheControl (>=0.14.2,<0.15.0)
@@ -45,10 +46,10 @@ Requires-Dist: demoji (>=1.1.0,<2.0.0)
 Requires-Dist: diskcache (>=5.6.3,<5.7.0)
 Requires-Dist: dnspython (==2.6.1)
 Requires-Dist: faiss-cpu (>=1.7.4,<2.0.0)
-Requires-Dist: faker (>=26.0.0,<27.0.0)
 Requires-Dist: fbmessenger (>=6.0.0,<6.1.0)
 Requires-Dist: github3.py (>=3.2.0,<3.3.0) ; extra == "gh-release-notes"
 Requires-Dist: gitpython (>=3.1.41,<3.2.0) ; extra == "full"
+Requires-Dist: gliner (>=0.2.20,<0.3.0) ; extra == "pii"
 Requires-Dist: google-auth (>=2.23.4,<3)
 Requires-Dist: google-cloud-storage (>=2.14.0,<2.15.0)
 Requires-Dist: hvac (>=1.2.1,<2.0.0)
@@ -68,6 +69,7 @@ Requires-Dist: matplotlib (>=3.7,<3.8)
 Requires-Dist: mattermostwrapper (>=2.2,<2.3)
 Requires-Dist: networkx (>=3.1,<3.2)
 Requires-Dist: numpy (>=1.26.4,<1.27.0)
+Requires-Dist: onnxruntime (==1.19.2) ; extra == "pii"
 Requires-Dist: openai (>=1.68.2,<1.69.0)
 Requires-Dist: openpyxl (>=3.1.5,<4.0.0)
 Requires-Dist: opentelemetry-api (>=1.16.0,<1.17.0)
@@ -78,13 +80,10 @@ Requires-Dist: packaging (>=23.2,<23.3)
 Requires-Dist: pep440-version-utils (>=1.1.0,<1.2.0)
 Requires-Dist: pluggy (>=1.2.0,<2.0.0)
 Requires-Dist: portalocker (>=2.7.0,<3.0.0)
-Requires-Dist: presidio-analyzer (>=2.2.33,<2.2.34)
-Requires-Dist: presidio-anonymizer (>=2.2.354,<3.0.0)
 Requires-Dist: prompt-toolkit (>=3.0.28,<3.0.29)
 Requires-Dist: protobuf (>=4.23.3,<4.25.4)
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: psycopg2-binary (>=2.9.10,<2.10.0)
-Requires-Dist: pycountry (>=22.3.5,<23.0.0)
 Requires-Dist: pydot (>=1.4,<1.5)
 Requires-Dist: pykwalify (>=1.8,<1.9)
 Requires-Dist: pymilvus (>=2.4.1,<2.4.2)
@@ -138,7 +137,7 @@ Requires-Dist: tensorflow_hub (>=0.13.0,<0.14.0)
 Requires-Dist: terminaltables (>=3.1.10,<3.2.0)
 Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
-Requires-Dist: transformers (>=4.36.2,<4.37.0) ; extra == "transformers" or extra == "full"
+Requires-Dist: transformers (>=4.38.2,<4.39.0) ; extra == "transformers" or extra == "full"
 Requires-Dist: twilio (>=8.4,<8.5)
 Requires-Dist: types-protobuf (==4.25.0.20240417)
 Requires-Dist: typing-extensions (>=4.7.1,<5.0.0)

{rasa_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rasa-pro"
-version = "3.13.0.dev7"
+version = "3.13.0.dev8"
 description = "State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development."
 authors = [ "Rasa Technologies GmbH <hi@rasa.com>",]
 maintainers = [ "Tom Bocklisch <tom@rasa.com>",]
@@ -13,17 +13,44 @@ repository = "https://github.com/rasahq/rasa"
 documentation = "https://rasa.com/docs"
 classifiers = [ "Development Status :: 5 - Production/Stable", "Intended Audience :: Developers", "Topic :: Software Development :: Libraries",]
 keywords = [ "nlp", "machine-learning", "machine-learning-library", "bot", "bots", "botkit", "rasa conversational-agents", "conversational-ai", "chatbot", "chatbot-framework", "bot-framework",]
-include = [
-  # license and readme should not be included in wheels, since they would end up in the root library folder
-  "LICENSE.txt", "README.md",
-  { path = "rasa/shared/core/training_data/visualization.html", format = ["sdist", "wheel"] },
-  { path = "rasa/cli/default_config.yml", format = ["sdist", "wheel"] },
-  { path = "rasa/shared/importers/*", format = ["sdist", "wheel"] },
-  { path = "rasa/utils/schemas/*", format = ["sdist", "wheel"] },
-  { path = "rasa/dialogue_understanding/classifiers/command_prompt_template.jinja2", format = ["sdist", "wheel"] },
-  { path = "rasa/keys", format = ["sdist", "wheel"] }
-]
 readme = "README.md"
+
+[[tool.poetry.include]]
+path = "LICENSE.txt"
+# license should not be included in wheels,
+# since it would end up in the root library folder
+format = "sdist"
+
+[[tool.poetry.include]]
+path = "README.md"
+# readme should not be included in wheels,
+# since it would end up in the root library folder
+format = "sdist"
+
+[[tool.poetry.include]]
+path = "rasa/shared/core/training_data/visualization.html"
+format = ["sdist", "wheel"]
+
+[[tool.poetry.include]]
+path = "rasa/cli/default_config.yml"
+format = ["sdist", "wheel"]
+
+[[tool.poetry.include]]
+path = "rasa/shared/importers/*"
+format = ["sdist", "wheel"]
+
+[[tool.poetry.include]]
+path = "rasa/utils/schemas/*"
+format = ["sdist", "wheel"]
+
+[[tool.poetry.include]]
+path = "rasa/dialogue_understanding/classifiers/command_prompt_template.jinja2"
+format = ["sdist", "wheel"]
+
+[[tool.poetry.include]]
+path = "rasa/keys"
+format = ["sdist", "wheel"]
+
 [[tool.poetry.packages]]
 include = "rasa"
 
@@ -180,13 +207,9 @@ pep440-version-utils = "~1.1.0"
 hvac = "^1.2.1"
 rich = "^13.4.2"
 cvg-python-sdk = "^0.5.1"
-presidio-analyzer = "^2.2.33, <2.2.34"
-presidio-anonymizer = "^2.2.354"
-faker = "^26.0.0"
 python-keycloak = "^3.12.0"
 tiktoken = "~0.7.0"
 faiss-cpu = "^1.7.4"
-pycountry = "^22.3.5"
 pymilvus = ">=2.4.1,<2.4.2" # pinned because 2.4.2 introduces milvus-lite dependency that fails on windows
 qdrant-client = "~1.9.1"
 psutil = "^5.9.5"
@@ -205,6 +228,14 @@ demoji = "^1.1.0"
 azure-identity = "~1.19.0"
 langcodes = "^3.5.0"
 
+[[tool.poetry.dependencies.gliner]]
+version = "~0.2.20"
+optional = true
+
+[[tool.poetry.dependencies.onnxruntime]]
+version = "1.19.2" # pinned to avoid breaking changes in versions >= 1.20.0 with Python 3.9
+optional = true
+
 [[tool.poetry.dependencies.tensorflow-io-gcs-filesystem]]
 version = "==0.31" # newer versions don't provide wheels for Windows
 markers = "sys_platform == 'win32'"
@@ -236,6 +267,7 @@ transformers = [ "transformers", "sentencepiece",]
 full = [ "spacy", "transformers", "sentencepiece", "jieba", "gitpython",]
 gh-release-notes = [ "github3.py",]
 metal = [ "tensorflow-metal",]
+pii = ["gliner", "onnxruntime"]
 
 [tool.poetry.scripts]
 rasa = "rasa.__main__:main"
@@ -313,7 +345,7 @@ version = "~3.2.0"
 optional = true
 
 [tool.poetry.dependencies.transformers]
-version = "~4.36.2"
+version = "~4.38.2"
 optional = true
 
 [tool.poetry.dependencies.sentencepiece]
@@ -340,8 +372,6 @@ convention = "google"
 "tests/shared/core/test_domain.py" = [ "E501",]
 
 
-
-
 [tool.poetry.group.dev.dependencies]
 ruff = ">=0.3.5,<0.4.0"
 docker = "~7.1.0"

{rasa_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/rasa/__main__.py

@@ -143,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_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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,
 )

{rasa_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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_stores.tracker_store import TrackerStore
-from rasa.core.utils import AvailableEndpoints
 from rasa.shared.core.domain import Domain
 
 STATS_OVERALL_SUFFIX = "-overall.csv"

{rasa_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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_stores.tracker_store import TrackerStore
-    from rasa.core.utils import AvailableEndpoints
 
 logger = logging.getLogger(__name__)
 

{rasa_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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,
 )
@@ -208,10 +208,7 @@ def prepare_llm_fine_tuning_data(args: argparse.Namespace) -> None:
         sys.exit(0)
 
     flows = asyncio.run(e2e_test_runner.agent.processor.get_flows())
-    llm_command_generator_config = _get_llm_command_generator_config(e2e_test_runner)
-    llm_command_generator: Type[LLMBasedCommandGenerator] = _get_llm_command_generator(
-        e2e_test_runner
-    )
+    _validate_llm_command_generator_present(e2e_test_runner)
 
     # set up storage context
     storage_context = create_storage_context(StorageType.FILE, output_dir)
@@ -242,11 +239,11 @@ def prepare_llm_fine_tuning_data(args: argparse.Namespace) -> None:
             rephrase_config,
             args.num_rephrases,
             flows,
-            llm_command_generator,
-            llm_command_generator_config,
+            e2e_test_runner.agent,
             storage_context,
         )
     )
+
     statistics["num_passing_rephrased_user_messages"] = sum(
         [conversation.get_number_of_rephrases(True) for conversation in conversations]
     )
@@ -257,7 +254,11 @@ def prepare_llm_fine_tuning_data(args: argparse.Namespace) -> None:
 
     # 3. create fine-tuning dataset
     log_start_of_module("LLM Data Preparation")
-    llm_fine_tuning_data = convert_to_fine_tuning_data(conversations, storage_context)
+    llm_fine_tuning_data = asyncio.run(
+        convert_to_fine_tuning_data(
+            conversations, storage_context, e2e_test_runner.agent
+        )
+    )
     statistics["num_ft_data_points"] = len(llm_fine_tuning_data)
     log_end_of_module("LLM Data Preparation", statistics)
 
@@ -311,9 +312,9 @@ def _get_llm_command_generator_config(e2e_test_runner: E2ETestRunner) -> Dict[st
     sys.exit(1)
 
 
-def _get_llm_command_generator(
+def _validate_llm_command_generator_present(
     e2e_test_runner: E2ETestRunner,
-) -> Type[LLMBasedCommandGenerator]:
+) -> None:
     train_schema = e2e_test_runner.agent.processor.model_metadata.train_schema  # type: ignore
 
     for _, node in train_schema.nodes.items():
@@ -322,7 +323,7 @@ def _get_llm_command_generator(
         ) and not node.matches_type(
             MultiStepLLMCommandGenerator, include_subtypes=True
         ):
-            return cast(Type[LLMBasedCommandGenerator], node.uses)
+            return
 
     rasa.shared.utils.cli.print_error(
         "The provided model is not trained using 'SingleStepLLMCommandGenerator' or "

rasa_pro-3.13.0.dev8/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_pro-3.13.0.dev7 → rasa_pro-3.13.0.dev8}/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_pro-3.13.0.dev8/rasa/cli/studio/link.py

@@ -0,0 +1,53 @@
+import argparse
+from typing import List, Text
+
+from rasa.cli import SubParsersAction
+from rasa.cli.studio.upload import (
+    add_config_param,
+    add_data_param,
+    add_domain_param,
+    add_endpoint_param,
+)
+from rasa.shared.constants import (
+    DEFAULT_DOMAIN_PATH,
+    DEFAULT_ENDPOINTS_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",
+        nargs=1,
+        type=str,
+        help="Name of the assistant in Rasa Studio.",
+    )
+
+    add_domain_param(link_parser, domain)
+    add_data_param(link_parser)
+    add_config_param(link_parser)
+    add_endpoint_param(
+        link_parser,
+        "Configuration file for the model endpoints.",
+        default=DEFAULT_ENDPOINTS_PATH,
+    )
+    link_parser.set_defaults(func=handle_link)

rasa_pro-3.13.0.dev8/rasa/cli/studio/pull.py

@@ -0,0 +1,78 @@
+import argparse
+from typing import List, Text
+
+from rasa.cli import SubParsersAction
+from rasa.cli.arguments.default_arguments import (
+    add_config_param,
+    add_endpoint_param,
+)
+from rasa.cli.studio.upload import add_data_param, add_domain_param
+from rasa.shared.constants import (
+    DEFAULT_CONFIG_PATH,
+    DEFAULT_DOMAIN_PATH,
+    DEFAULT_ENDPOINTS_PATH,
+)
+from rasa.studio.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)

rasa_pro-3.13.0.dev8/rasa/cli/studio/push.py

@@ -0,0 +1,78 @@
+import argparse
+from typing import List, Text
+
+from rasa.cli import SubParsersAction
+from rasa.cli.studio.upload import (
+    add_config_param,
+    add_data_param,
+    add_domain_param,
+    add_endpoint_param,
+)
+from rasa.shared.constants import (
+    DEFAULT_DOMAIN_PATH,
+    DEFAULT_ENDPOINTS_PATH,
+)
+from rasa.studio.push import (
+    handle_push,
+    handle_push_config,
+    handle_push_endpoints,
+)
+
+
+def add_subparser(
+    subparsers: SubParsersAction,
+    parents: List[argparse.ArgumentParser],
+    domain: Text = DEFAULT_DOMAIN_PATH,
+) -> None:
+    """Register the `rasa studio push` command and its sub-commands.
+
+    Args:
+        subparsers: The subparsers to add the command to.
+        parents: Parent parsers, needed to ensure tree structure in argparse
+        domain: Path to the assistant's domain file.
+    """
+    push_parser = subparsers.add_parser(
+        "push",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Push assistant data to Rasa Studio.",
+    )
+    push_subparsers = push_parser.add_subparsers()
+
+    # Arguments for pushing configuration only
+    config_parser = push_subparsers.add_parser(
+        "config",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Push only the assistant configuration file.",
+    )
+    add_config_param(config_parser)
+    config_parser.set_defaults(func=handle_push_config)
+
+    # Arguments for pushing endpoints only
+    endpoints_parser = push_subparsers.add_parser(
+        "endpoints",
+        parents=parents,
+        conflict_handler="resolve",
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
+        help="Push 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_push_endpoints)
+
+    # Arguments for pushing the whole assistant data
+    add_domain_param(push_parser, domain)
+    add_data_param(push_parser)
+    add_config_param(push_parser)
+    add_endpoint_param(
+        push_parser,
+        "Configuration file for the model endpoints.",
+        default=DEFAULT_ENDPOINTS_PATH,
+    )
+    push_parser.set_defaults(func=handle_push)