rasa-pro 3.12.10.dev1__tar.gz → 3.12.12.dev1__tar.gz

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rasa-pro
-Version: 3.12.10.dev1
+Version: 3.12.12.dev1
 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
@@ -63,8 +63,7 @@ Requires-Dist: keras (==2.14.0)
 Requires-Dist: langchain (>=0.2.17,<0.3.0)
 Requires-Dist: langchain-community (>=0.2.19,<0.3.0)
 Requires-Dist: langcodes (>=3.5.0,<4.0.0)
-Requires-Dist: langfuse (>=2.60.2,<2.61.0)
-Requires-Dist: litellm (>=1.68.0,<1.69.0)
+Requires-Dist: litellm (>=1.69.0,<1.70.0)
 Requires-Dist: matplotlib (>=3.7,<3.8)
 Requires-Dist: mattermostwrapper (>=2.2,<2.3)
 Requires-Dist: networkx (>=3.1,<3.2)

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rasa-pro"
-version = "3.12.10.dev1"
+version = "3.12.12.dev1"
 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,7 +13,16 @@ 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.txt", "README.md", "rasa/shared/core/training_data/visualization.html", "rasa/cli/default_config.yml", "rasa/shared/importers/*", "rasa/utils/schemas/*", "rasa/keys", "rasa/dialogue_understanding/classifiers/command_prompt_template.jinja2",]
+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.packages]]
 include = "rasa"
@@ -183,7 +192,7 @@ qdrant-client = "~1.9.1"
 psutil = "^5.9.5"
 tenacity = "~8.4.1" # pinned down because of langchain
 types-protobuf = "4.25.0.20240417"
-litellm = "~1.68.0"
+litellm = "~1.69.0"
 openpyxl = "^3.1.5"
 diskcache = "~5.6.3"
 skops = "~0.10.0"
@@ -195,7 +204,6 @@ scikit-learn = "~1.5.1"
 demoji = "^1.1.0"
 azure-identity = "~1.19.0"
 langcodes = "^3.5.0"
-langfuse = "~2.60.2"
 
 [[tool.poetry.dependencies.tensorflow-io-gcs-filesystem]]
 version = "==0.31" # newer versions don't provide wheels for Windows
@@ -332,9 +340,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.12.12.dev1/rasa/__init__.py

@@ -0,0 +1,9 @@
+import logging
+
+from rasa import version
+
+# define the version before the other imports since these need it
+__version__ = version.__version__
+
+
+logging.getLogger(__name__).addHandler(logging.NullHandler())

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/api.py

@@ -81,6 +81,7 @@ def train(
     remote_storage: Optional[StorageType] = None,
     file_importer: Optional["TrainingDataImporter"] = None,
     keep_local_model_copy: bool = False,
+    remote_root_only: bool = False,
 ) -> "TrainingResult":
     """Runs Rasa Core and NLU training in `async` loop.
 
@@ -108,6 +109,8 @@ def train(
             If it is not provided, a new instance will be created.
         keep_local_model_copy: If `True` the model will be stored locally even if
             remote storage is configured.
+        remote_root_only: If `True`, the model will be stored in the root of the
+            remote model storage.
 
     Returns:
         An instance of `TrainingResult`.
@@ -131,6 +134,7 @@ def train(
             remote_storage=remote_storage,
             file_importer=file_importer,
             keep_local_model_copy=keep_local_model_copy,
+            remote_root_only=remote_root_only,
         )
     )
 

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/cli/arguments/default_arguments.py

@@ -172,7 +172,7 @@ def add_remote_storage_param(
 ) -> None:
     parser.add_argument(
         "--remote-storage",
-        help="Remote storage which should be used to store/load the model."
+        help="Remote storage which should be used to store/load the model. "
         f"Supported storages are: {RemoteStorageType.list()}. "
         "You can also provide your own implementation of the `Persistor` interface.",
         required=required,
@@ -180,6 +180,18 @@ def add_remote_storage_param(
     )
 
 
+def add_remote_root_only_param(
+    parser: argparse.ArgumentParser, required: bool = False
+) -> None:
+    parser.add_argument(
+        "--remote-root-only",
+        action="store_true",
+        help="If set, models will be stored only at the root directory "
+        "of the remote storage.",
+        required=required,
+    )
+
+
 def parse_remote_storage_arg(value: str) -> StorageType:
     try:
         return parse_remote_storage(value)

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/cli/arguments/train.py

@@ -7,6 +7,7 @@ from rasa.cli.arguments.default_arguments import (
     add_endpoint_param,
     add_nlu_data_param,
     add_out_param,
+    add_remote_root_only_param,
     add_remote_storage_param,
     add_stories_param,
 )
@@ -41,6 +42,7 @@ def set_train_arguments(parser: argparse.ArgumentParser) -> None:
         parser, help_text="Configuration file for the connectors as a yml file."
     )
     add_remote_storage_param(parser)
+    add_remote_root_only_param(parser)
 
 
 def set_train_core_arguments(parser: argparse.ArgumentParser) -> None:

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/cli/train.py

@@ -155,6 +155,7 @@ def run_training(args: argparse.Namespace, can_exit: bool = False) -> Optional[T
         remote_storage=args.remote_storage,
         file_importer=training_data_importer,
         keep_local_model_copy=args.keep_local_model_copy,
+        remote_root_only=args.remote_root_only,
     )
     if training_result.code != 0 and can_exit:
         display_research_study_prompt()

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/constants.py

@@ -42,3 +42,5 @@ DEFAULT_BUCKET_NAME = "rasa-models"
 
 HTTP_STATUS_FORBIDDEN = 403
 HTTP_STATUS_NOT_FOUND = 404
+
+RASA_REMOTE_STORAGE_ENV_VAR_NAME = "RASA_REMOTE_STORAGE"

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/core/actions/action.py

@@ -898,7 +898,7 @@ class RemoteAction(Action):
                 draft["buttons"].extend(buttons)
 
             # Avoid overwriting `draft` values with empty values
-            response = {k: v for k, v in response.items() if v}
+            response = {k: v for k, v in response.items() if v is not None}
             draft.update(response)
             bot_messages.append(create_bot_utterance(draft))
 

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/core/channels/voice_ready/audiocodes.py

@@ -6,7 +6,18 @@ import uuid
 from collections import defaultdict
 from dataclasses import asdict
 from datetime import datetime, timedelta, timezone
-from typing import Any, Awaitable, Callable, Dict, List, Optional, Set, Text, Union
+from typing import (
+    Any,
+    Awaitable,
+    Callable,
+    Dict,
+    List,
+    Optional,
+    Set,
+    Text,
+    Tuple,
+    Union,
+)
 
 import structlog
 from jsonschema import ValidationError, validate
@@ -76,35 +87,45 @@ class Conversation:
 
     @staticmethod
     def get_metadata(activity: Dict[Text, Any]) -> Optional[Dict[Text, Any]]:
-        """Get metadata from the activity."""
-        return asdict(map_call_params(activity["parameters"]))
+        """Get metadata from the activity.
+
+        ONLY used for activities NOT for events (see _handle_event)."""
+        return activity.get("parameters")
 
     @staticmethod
-    def _handle_event(event: Dict[Text, Any]) -> Text:
-        """Handle start and DTMF event and return the corresponding text."""
+    def _handle_event(event: Dict[Text, Any]) -> Tuple[Text, Dict[Text, Any]]:
+        """Handle events and return a tuple of text and metadata.
+
+        Args:
+            event: The event to handle.
+
+        Returns:
+            Tuple of text and metadata.
+            text is either /session_start or /vaig_event_<event_name>
+            metadata is a dictionary with the event parameters.
+        """
         structlogger.debug("audiocodes.handle.event", event_payload=event)
         if "name" not in event:
             structlogger.warning(
                 "audiocodes.handle.event.no_name_key", event_payload=event
             )
-            return ""
+            return "", {}
 
         if event["name"] == EVENT_START:
             text = f"{INTENT_MESSAGE_PREFIX}{USER_INTENT_SESSION_START}"
+            metadata = asdict(map_call_params(event.get("parameters", {})))
         elif event["name"] == EVENT_DTMF:
             text = f"{INTENT_MESSAGE_PREFIX}vaig_event_DTMF"
-            event_params = {"value": event["value"]}
-            text += json.dumps(event_params)
+            metadata = {"value": event["value"]}
         else:
             # handle other events described by Audiocodes
             # https://techdocs.audiocodes.com/voice-ai-connect/#VAIG_Combined/inactivity-detection.htm?TocPath=Bot%2520integration%257CReceiving%2520notifications%257C_____3
             text = f"{INTENT_MESSAGE_PREFIX}vaig_event_{event['name']}"
-            event_params = {**event.get("parameters", {})}
+            metadata = {**event.get("parameters", {})}
             if "value" in event:
-                event_params["value"] = event["value"]
-            text += json.dumps(event_params)
+                metadata["value"] = event["value"]
 
-        return text
+        return text, metadata
 
     def is_active_conversation(self, now: datetime, delta: timedelta) -> bool:
         """Check if the conversation is active."""
@@ -139,21 +160,29 @@ class Conversation:
                 structlogger.warning(
                     "audiocodes.handle.activities.duplicate_activity",
                     activity_id=activity[ACTIVITY_ID_KEY],
+                    event_info=(
+                        "Audiocodes might send duplicate activities if the bot has not "
+                        "responded to the previous one or responded too late. Please "
+                        "consider enabling the `use_websocket` option to use"
+                        " Audiocodes Asynchronous API."
+                    ),
                 )
                 continue
             self.activity_ids.append(activity[ACTIVITY_ID_KEY])
             if activity["type"] == ACTIVITY_MESSAGE:
                 text = activity["text"]
+                metadata = self.get_metadata(activity)
             elif activity["type"] == ACTIVITY_EVENT:
-                text = self._handle_event(activity)
+                text, metadata = self._handle_event(activity)
             else:
                 structlogger.warning(
                     "audiocodes.handle.activities.unknown_activity_type",
                     activity=activity,
                 )
+                continue
+
             if not text:
                 continue
-            metadata = self.get_metadata(activity)
             user_msg = UserMessage(
                 text=text,
                 input_channel=input_channel_name,
@@ -392,30 +421,41 @@ class AudiocodesInput(InputChannel):
                     "audiocodes.on_activities.no_conversation", request=request.json
                 )
                 return response.json({})
-            elif conversation.ws:
+
+            if self.use_websocket:
+                # send an empty response for this request
+                # activities are processed in the background
+                # chat response is sent via the websocket
                 ac_output: Union[WebsocketOutput, AudiocodesOutput] = WebsocketOutput(
                     conversation.ws, conversation_id
                 )
-                response_json = {}
-            else:
-                # handle non websocket case where messages get returned in json
-                ac_output = AudiocodesOutput()
-                response_json = {
+                self._create_task(
+                    conversation_id,
+                    conversation.handle_activities(
+                        request.json,
+                        input_channel_name=self.name(),
+                        output_channel=ac_output,
+                        on_new_message=on_new_message,
+                    ),
+                )
+                return response.json({})
+
+            # without websockets, this becomes a blocking call
+            # and the response is sent back to the Audiocodes server
+            # after the activities are processed
+            ac_output = AudiocodesOutput()
+            await conversation.handle_activities(
+                request.json,
+                input_channel_name=self.name(),
+                output_channel=ac_output,
+                on_new_message=on_new_message,
+            )
+            return response.json(
+                {
                     "conversation": conversation_id,
                     "activities": ac_output.messages,
                 }
-
-            # start a background task to handle activities
-            self._create_task(
-                conversation_id,
-                conversation.handle_activities(
-                    request.json,
-                    input_channel_name=self.name(),
-                    output_channel=ac_output,
-                    on_new_message=on_new_message,
-                ),
             )
-            return response.json(response_json)
 
         @ac_webhook.route(
             "/conversation/<conversation_id>/disconnect", methods=["POST"]

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/core/persistor.py

@@ -121,10 +121,12 @@ def get_persistor(storage: StorageType) -> Optional[Persistor]:
 class Persistor(abc.ABC):
     """Store models in cloud and fetch them when needed."""
 
-    def persist(self, trained_model: str) -> None:
+    def persist(self, trained_model: str, remote_root_only: bool = False) -> None:
         """Uploads a trained model persisted in the `target_dir` to cloud storage."""
         absolute_file_key = self._create_file_key(trained_model)
-        file_key = Path(absolute_file_key).name
+        file_key = (
+            Path(absolute_file_key).name if remote_root_only else absolute_file_key
+        )
         self._persist_tar(file_key, trained_model)
 
     def retrieve(self, model_name: Text, target_path: Text) -> Text:
@@ -143,30 +145,32 @@ class Persistor(abc.ABC):
             # ensure backward compatibility
             tar_name = self._tar_name(model_name)
         tar_name = self._create_file_key(tar_name)
-        target_filename = os.path.basename(tar_name)
-        self._retrieve_tar(target_filename)
-        self._copy(os.path.basename(tar_name), target_path)
+        self._retrieve_tar(tar_name, target_path)
 
         if os.path.isdir(target_path):
             return os.path.join(target_path, model_name)
 
         return target_path
 
-    def size_of_persisted_model(self, model_name: Text) -> int:
+    def size_of_persisted_model(
+        self, model_name: Text, target_path: Optional[str] = None
+    ) -> int:
         """Returns the size of the model that has been persisted to cloud storage.
 
         Args:
             model_name: The name of the model to retrieve.
+            target_path: The path to which the model should be saved.
         """
         tar_name = model_name
         if not model_name.endswith(MODEL_ARCHIVE_EXTENSION):
             # ensure backward compatibility
             tar_name = self._tar_name(model_name)
         tar_name = self._create_file_key(tar_name)
-        target_filename = os.path.basename(tar_name)
-        return self._retrieve_tar_size(target_filename)
+        return self._retrieve_tar_size(tar_name, target_path)
 
-    def _retrieve_tar_size(self, filename: Text) -> int:
+    def _retrieve_tar_size(
+        self, filename: Text, target_path: Optional[str] = None
+    ) -> int:
         """Returns the size of the model that has been persisted to cloud storage."""
         structlogger.warning(
             "persistor.retrieve_tar_size.not_implemented",
@@ -179,11 +183,11 @@ class Persistor(abc.ABC):
                 "size directly from the cloud storage."
             ),
         )
-        self._retrieve_tar(filename)
+        self._retrieve_tar(filename, target_path)
         return os.path.getsize(os.path.basename(filename))
 
     @abc.abstractmethod
-    def _retrieve_tar(self, filename: Text) -> None:
+    def _retrieve_tar(self, filename: str, target_path: Optional[str] = None) -> None:
         """Downloads a model previously persisted to cloud storage."""
         raise NotImplementedError
 
@@ -302,7 +306,9 @@ class AWSPersistor(Persistor):
         with open(tar_path, "rb") as f:
             self.s3.Object(self.bucket_name, file_key).put(Body=f)
 
-    def _retrieve_tar_size(self, model_path: Text) -> int:
+    def _retrieve_tar_size(
+        self, model_path: Text, target_path: Optional[str] = None
+    ) -> int:
         """Returns the size of the model that has been persisted to s3."""
         try:
             obj = self.s3.Object(self.bucket_name, model_path)
@@ -310,7 +316,9 @@ class AWSPersistor(Persistor):
         except Exception:
             raise ModelNotFound()
 
-    def _retrieve_tar(self, target_filename: str) -> None:
+    def _retrieve_tar(
+        self, target_filename: str, target_path: Optional[str] = None
+    ) -> None:
         """Downloads a model that has previously been persisted to s3."""
         from botocore import exceptions
 
@@ -320,8 +328,14 @@ class AWSPersistor(Persistor):
             f"in the bucket."
         )
 
+        tar_name = (
+            os.path.join(target_path, os.path.basename(target_filename))
+            if target_path
+            else os.path.basename(target_filename)
+        )
+
         try:
-            with open(target_filename, "wb") as f:
+            with open(tar_name, "wb") as f:
                 self.bucket.download_fileobj(target_filename, f)
 
             structlogger.debug(
@@ -425,7 +439,9 @@ class GCSPersistor(Persistor):
         blob = self.bucket.blob(file_key)
         blob.upload_from_filename(tar_path)
 
-    def _retrieve_tar_size(self, target_filename: Text) -> int:
+    def _retrieve_tar_size(
+        self, target_filename: Text, target_path: Optional[str] = None
+    ) -> int:
         """Returns the size of the model that has been persisted to GCS."""
         try:
             blob = self.bucket.blob(target_filename)
@@ -433,13 +449,22 @@ class GCSPersistor(Persistor):
         except Exception:
             raise ModelNotFound()
 
-    def _retrieve_tar(self, target_filename: Text) -> None:
+    def _retrieve_tar(
+        self, target_filename: str, target_path: Optional[str] = None
+    ) -> None:
         """Downloads a model that has previously been persisted to GCS."""
         from google.api_core import exceptions
 
         blob = self.bucket.blob(target_filename)
+
+        destination = (
+            os.path.join(target_path, os.path.basename(target_filename))
+            if target_path
+            else target_filename
+        )
+
         try:
-            blob.download_to_filename(target_filename)
+            blob.download_to_filename(destination)
 
             structlogger.debug(
                 "gcs_persistor.retrieve_tar.object_found", object_key=target_filename
@@ -500,7 +525,9 @@ class AzurePersistor(Persistor):
         with open(tar_path, "rb") as data:
             self._container_client().upload_blob(name=file_key, data=data)
 
-    def _retrieve_tar_size(self, target_filename: Text) -> int:
+    def _retrieve_tar_size(
+        self, target_filename: Text, target_path: Optional[str] = None
+    ) -> int:
         """Returns the size of the model that has been persisted to Azure."""
         try:
             blob_client = self._container_client().get_blob_client(target_filename)
@@ -509,12 +536,20 @@ class AzurePersistor(Persistor):
         except Exception:
             raise ModelNotFound()
 
-    def _retrieve_tar(self, target_filename: Text) -> None:
+    def _retrieve_tar(
+        self, target_filename: Text, target_path: Optional[str] = None
+    ) -> None:
         """Downloads a model that has previously been persisted to Azure."""
         from azure.core.exceptions import AzureError
 
+        destination = (
+            os.path.join(target_path, os.path.basename(target_filename))
+            if target_path
+            else target_filename
+        )
+
         try:
-            with open(target_filename, "wb") as model_file:
+            with open(destination, "wb") as model_file:
                 blob_client = self._container_client().get_blob_client(target_filename)
                 download_stream = blob_client.download_blob()
                 model_file.write(download_stream.readall())

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/core/policies/intentless_policy.py

@@ -715,9 +715,7 @@ class IntentlessPolicy(LLMHealthCheckMixin, EmbeddingsHealthCheckMixin, Policy):
                 final_response_examples.append(resp)
 
         llm_response = await self.generate_answer(
-            final_response_examples,
-            conversation_samples,
-            history,
+            final_response_examples, conversation_samples, history
         )
         if not llm_response:
             structlogger.debug("intentless_policy.prediction.skip_llm_fail")

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/dialogue_understanding/coexistence/llm_based_router.py

@@ -166,6 +166,7 @@ class LLMBasedRouter(LLMHealthCheckMixin, GraphComponent):
         **kwargs: Any,
     ) -> "LLMBasedRouter":
         """Loads trained component (see parent class for full docstring)."""
+
         # Perform health check on the resolved LLM client config
         llm_config = resolve_model_client_config(config.get(LLM_CONFIG_KEY, {}))
         cls.perform_llm_health_check(

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/dialogue_understanding/generator/llm_based_command_generator.py

@@ -1,8 +1,6 @@
 from abc import ABC, abstractmethod
-from asyncio import Lock
 from functools import lru_cache
 from typing import Any, Dict, List, Optional, Set, Text, Tuple, Union
-from uuid import UUID, uuid4
 
 import structlog
 from jinja2 import Environment, Template, select_autoescape
@@ -91,9 +89,6 @@ class LLMBasedCommandGenerator(
         else:
             self.flow_retrieval = None
 
-        self.sender_id_to_session_id_mapping: Dict[str, UUID] = {}
-        self._lock = Lock()
-
     ### Abstract methods
     @staticmethod
     @abstractmethod
@@ -230,7 +225,8 @@ class LLMBasedCommandGenerator(
 
     @lru_cache
     def compile_template(self, template: str) -> Template:
-        """Compile the prompt template and register custom filters.
+        """
+        Compile the prompt template and register custom filters.
         Compiling the template is an expensive operation,
         so we cache the result.
         """
@@ -332,9 +328,7 @@ class LLMBasedCommandGenerator(
 
     @measure_llm_latency
     async def invoke_llm(
-        self,
-        prompt: Union[List[dict], List[str], str],
-        metadata: Optional[Dict[str, Any]] = None,
+        self, prompt: Union[List[dict], List[str], str]
     ) -> Optional[LLMResponse]:
         """Use LLM to generate a response.
 
@@ -347,7 +341,6 @@ class LLMBasedCommandGenerator(
                 - a list of messages. Each message is a string and will be formatted
                     as a user message.
                 - a single message as a string which will be formatted as user message.
-            metadata: Optional metadata to be passed to the LLM call.
 
         Returns:
             An LLMResponse object.
@@ -359,7 +352,7 @@ class LLMBasedCommandGenerator(
             self.config.get(LLM_CONFIG_KEY), self.get_default_llm_config()
         )
         try:
-            return await llm.acompletion(prompt, metadata)
+            return await llm.acompletion(prompt)
         except Exception as e:
             # unfortunately, langchain does not wrap LLM exceptions which means
             # we have to catch all exceptions here
@@ -662,7 +655,3 @@ class LLMBasedCommandGenerator(
     def get_default_llm_config() -> Dict[str, Any]:
         """Get the default LLM config for the command generator."""
         return DEFAULT_LLM_CONFIG
-
-    async def _get_or_create_session_id(self, sender_id: str) -> UUID:
-        async with self._lock:
-            return self.sender_id_to_session_id_mapping.setdefault(sender_id, uuid4())

{rasa_pro-3.12.10.dev1 → rasa_pro-3.12.12.dev1}/rasa/dialogue_understanding/generator/llm_command_generator.py

@@ -55,9 +55,7 @@ class LLMCommandGenerator(SingleStepLLMCommandGenerator):
         )
 
     async def invoke_llm(
-        self,
-        prompt: Union[List[dict], List[str], str],
-        metadata: Optional[Dict[str, Any]] = None,
+        self, prompt: Union[List[dict], List[str], str]
     ) -> Optional[LLMResponse]:
         try:
             return await super().invoke_llm(prompt)