rasa-pro 3.10.4__tar.gz → 3.10.5__tar.gz

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rasa-pro
-Version: 3.10.4
+Version: 3.10.5
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Home-page: https://rasa.com
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
@@ -33,7 +33,7 @@ Requires-Dist: apscheduler (>=3.10,<3.11)
 Requires-Dist: attrs (>=23.1,<23.2)
 Requires-Dist: azure-storage-blob (>=12.16.0,<12.17.0)
 Requires-Dist: boto3 (>=1.35.5,<1.36.0)
-Requires-Dist: certifi (>=2024.2.2)
+Requires-Dist: certifi (>=2024.07.04)
 Requires-Dist: cloudpickle (>=2.2.1,<3.1)
 Requires-Dist: colorama (>=0.4.6,<0.5.0) ; sys_platform == "win32"
 Requires-Dist: colorclass (>=2.2,<2.3)
@@ -49,6 +49,7 @@ 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: google-auth (>=2.23.4,<3)
 Requires-Dist: google-cloud-storage (>=2.14.0,<3.0.0)
 Requires-Dist: hvac (>=1.2.1,<2.0.0)
@@ -63,13 +64,13 @@ Requires-Dist: jsonschema (>=4.22)
 Requires-Dist: keras (==2.14.0)
 Requires-Dist: langchain (>=0.2.0,<0.3.0)
 Requires-Dist: langchain-community (>=0.2.0,<0.3.0)
-Requires-Dist: litellm (>=1.44.7,<1.45.0)
+Requires-Dist: litellm (>=1.45.0,<1.46.0)
 Requires-Dist: matplotlib (>=3.7,<3.8)
 Requires-Dist: mattermostwrapper (>=2.2,<2.3)
 Requires-Dist: mlflow (>=2.15.1,<3.0.0) ; extra == "mlflow"
 Requires-Dist: networkx (>=3.1,<3.2)
 Requires-Dist: numpy (>=1.23.5,<1.25.0) ; python_version >= "3.9" and python_version < "3.11"
-Requires-Dist: openai (>=1.40.0,<1.41.0)
+Requires-Dist: openai (>=1.47.0,<1.48.0)
 Requires-Dist: openpyxl (>=3.1.5,<4.0.0)
 Requires-Dist: opentelemetry-api (>=1.16.0,<1.17.0)
 Requires-Dist: opentelemetry-exporter-jaeger (>=1.16.0,<1.17.0)

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rasa-pro"
-version = "3.10.4"
+version = "3.10.5"
 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>",]
@@ -156,10 +156,10 @@ structlog-sentry = "^2.0.3"
 langchain = "~0.2.0"
 langchain-community = "~0.2.0"
 jinja2 = "^3.1.4"
-openai = "~1.40.0"
+openai = "~1.47.0"
 dnspython = "2.6.1"
 wheel = ">=0.40.0"
-certifi = ">=2024.2.2"
+certifi = ">=2024.07.04"
 cryptography = ">=42.0.5"
 jsonpatch = "^1.33"
 importlib-resources = "^6.1.1"
@@ -185,7 +185,7 @@ qdrant-client = "^1.9.0"
 psutil = "^5.9.5"
 tenacity = "~8.4.1" # pinned down because of langchain
 types-protobuf = "4.25.0.20240417"
-litellm = "~1.44.7"
+litellm = "~1.45.0"
 openpyxl = "^3.1.5"
 diskcache = "~5.6.3"
 [[tool.poetry.dependencies.tensorflow-io-gcs-filesystem]]
@@ -228,11 +228,15 @@ optional = true
 version = "^2.15.1"
 optional = true
 
+[[tool.poetry.dependencies.gitpython]]
+version = "~3.1.41"
+optional = true
+
 [tool.poetry.extras]
 spacy = [ "spacy",]
 jieba = [ "jieba",]
 transformers = [ "transformers", "sentencepiece",]
-full = [ "spacy", "transformers", "sentencepiece", "jieba",]
+full = [ "spacy", "transformers", "sentencepiece", "jieba", "gitpython",]
 gh-release-notes = [ "github3.py",]
 metal = [ "tensorflow-metal",]
 mlflow = [ "mlflow",]

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/core/policies/intentless_policy.py

@@ -80,6 +80,8 @@ from rasa.utils.ml_utils import (
     persist_faiss_vector_store,
     response_for_template,
 )
+from rasa.dialogue_understanding.patterns.chitchat import FLOW_PATTERN_CHITCHAT
+from rasa.shared.core.constants import ACTION_TRIGGER_CHITCHAT
 from rasa.utils.log_utils import log_llm
 
 if TYPE_CHECKING:
@@ -174,6 +176,21 @@ def filter_responses(responses: Responses, forms: Forms, flows: FlowsList) -> Re
         for name, variants in responses.data.items()
         if name not in combined_responses
     }
+
+    pattern_chitchat = flows.flow_by_id(FLOW_PATTERN_CHITCHAT)
+
+    # The following condition is highly unlikely, but mypy requires the case
+    # of pattern_chitchat == None to be addressed
+    if not pattern_chitchat:
+        return Responses(data=filtered_responses)
+
+    # if action_trigger_chitchat, filter out "utter_free_chitchat_response"
+    has_action_trigger_chitchat = pattern_chitchat.has_action_step(
+        ACTION_TRIGGER_CHITCHAT
+    )
+    if has_action_trigger_chitchat:
+        filtered_responses.pop("utter_free_chitchat_response", None)
+
     return Responses(data=filtered_responses)
 
 
@@ -709,6 +726,7 @@ class IntentlessPolicy(Policy):
             number_of_samples=NUMBER_OF_CONVERSATION_SAMPLES,
             max_number_of_tokens=MAX_NUMBER_OF_TOKENS_FOR_SAMPLES,
         )
+
         extra_ai_responses = self.extract_ai_responses(conversation_samples)
 
         # put conversation responses in front of sampled examples,

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/dialogue_understanding/processor/command_processor.py

@@ -1,6 +1,7 @@
 from typing import List, Optional, Type, Set, Dict
 
 import structlog
+from rasa.shared.core.training_data.structures import StoryGraph
 from rasa.dialogue_understanding.commands import (
     CancelFlowCommand,
     ClarifyCommand,
@@ -179,6 +180,7 @@ def execute_commands(
     tracker: DialogueStateTracker,
     all_flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> List[Event]:
     """Executes a list of commands.
 
@@ -187,6 +189,7 @@ def execute_commands(
         tracker: The tracker to execute the commands on.
         all_flows: All flows.
         execution_context: Information about the single graph run.
+        story_graph: StoryGraph object with stories available for training.
 
     Returns:
         A list of the events that were created.
@@ -194,7 +197,9 @@ def execute_commands(
     commands: List[Command] = get_commands_from_tracker(tracker)
     original_tracker = tracker.copy()
 
-    commands = clean_up_commands(commands, tracker, all_flows, execution_context)
+    commands = clean_up_commands(
+        commands, tracker, all_flows, execution_context, story_graph
+    )
 
     updated_flows = find_updated_flows(tracker, all_flows)
     if updated_flows:
@@ -326,6 +331,7 @@ def clean_up_commands(
     tracker: DialogueStateTracker,
     all_flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> List[Command]:
     """Clean up a list of commands.
 
@@ -340,6 +346,7 @@ def clean_up_commands(
         tracker: The tracker to clean up the commands for.
         all_flows: All flows.
         execution_context: Information about a single graph run.
+        story_graph: StoryGraph object with stories available for training.
 
     Returns:
     The cleaned up commands.
@@ -386,7 +393,7 @@ def clean_up_commands(
         # handle chitchat command differently from other free-form answer commands
         elif isinstance(command, ChitChatAnswerCommand):
             clean_commands = clean_up_chitchat_command(
-                clean_commands, command, all_flows, execution_context
+                clean_commands, command, all_flows, execution_context, story_graph
             )
 
         elif isinstance(command, FreeFormAnswerCommand):
@@ -562,6 +569,7 @@ def clean_up_chitchat_command(
     command: ChitChatAnswerCommand,
     flows: FlowsList,
     execution_context: ExecutionContext,
+    story_graph: Optional[StoryGraph] = None,
 ) -> List[Command]:
     """Clean up a chitchat answer command.
 
@@ -573,6 +581,7 @@ def clean_up_chitchat_command(
         command: The command to clean up.
         flows: All flows.
         execution_context: Information about a single graph run.
+        story_graph: StoryGraph object with stories available for training.
     Returns:
         The cleaned up commands.
     """
@@ -598,7 +607,11 @@ def clean_up_chitchat_command(
     )
     defines_intentless_policy = execution_context.has_node(IntentlessPolicy)
 
-    if has_action_trigger_chitchat and not defines_intentless_policy:
+    has_e2e_stories = True if (story_graph and story_graph.has_e2e_stories()) else False
+
+    if (has_action_trigger_chitchat and not defines_intentless_policy) or (
+        defines_intentless_policy and not has_e2e_stories
+    ):
         resulting_commands.insert(
             0, CannotHandleCommand(RASA_PATTERN_CANNOT_HANDLE_CHITCHAT)
         )

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/dialogue_understanding/processor/command_processor_component.py

@@ -9,6 +9,7 @@ from rasa.engine.storage.storage import ModelStorage
 from rasa.shared.core.events import Event
 from rasa.shared.core.flows import FlowsList
 from rasa.shared.core.trackers import DialogueStateTracker
+from rasa.shared.core.training_data.structures import StoryGraph
 
 
 class CommandProcessorComponent(GraphComponent):
@@ -31,9 +32,12 @@ class CommandProcessorComponent(GraphComponent):
         return cls(execution_context)
 
     def execute_commands(
-        self, tracker: DialogueStateTracker, flows: FlowsList
+        self,
+        tracker: DialogueStateTracker,
+        flows: FlowsList,
+        story_graph: StoryGraph,
     ) -> List[Event]:
         """Execute commands to update tracker state."""
         return rasa.dialogue_understanding.processor.command_processor.execute_commands(
-            tracker, flows, self._execution_context
+            tracker, flows, self._execution_context, story_graph
         )

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/engine/recipes/default_recipe.py

@@ -405,7 +405,7 @@ class DefaultV1Recipe(Recipe):
             return {}
 
         def resolver_name_from_parameter(parameter: str) -> str:
-            # we got a couple special cases to handle wher the parameter name
+            # we got a couple special cases to handle where the parameter name
             # doesn't match the provider name
             if "training_trackers" == parameter:
                 return "training_tracker_provider"
@@ -597,7 +597,7 @@ class DefaultV1Recipe(Recipe):
             needs={"importer": "finetuning_validator"},
             uses=StoryGraphProvider,
             constructor_name="create",
-            fn="provide",
+            fn="provide_train",
             config={"exclusion_percentage": cli_parameters.get("exclusion_percentage")},
             is_input=True,
         )
@@ -882,6 +882,14 @@ class DefaultV1Recipe(Recipe):
             config={},
             resource=Resource("domain_provider"),
         )
+        predict_nodes["story_graph_provider"] = SchemaNode(
+            **DEFAULT_PREDICT_KWARGS,
+            needs={},
+            uses=StoryGraphProvider,
+            fn="provide_inference",
+            config={},
+            resource=Resource("story_graph_provider"),
+        )
         predict_nodes["flows_provider"] = SchemaNode(
             **DEFAULT_PREDICT_KWARGS,
             needs={},

rasa_pro-3.10.5/rasa/graph_components/providers/story_graph_provider.py

@@ -0,0 +1,96 @@
+from __future__ import annotations
+from typing import Dict, Text, Any, List
+
+from rasa.engine.graph import GraphComponent, ExecutionContext
+from rasa.engine.storage.resource import Resource
+from rasa.engine.storage.storage import ModelStorage
+from rasa.shared.core.training_data.structures import StoryGraph, StoryStep
+from rasa.shared.importers.importer import TrainingDataImporter
+from rasa.shared.core.training_data.story_writer.yaml_story_writer import (
+    YAMLStoryWriter,
+)
+from rasa.shared.core.training_data.story_reader.yaml_story_reader import (
+    YAMLStoryReader,
+)
+
+
+STORIES_PERSISTENCE_FILE_NAME = "stories.yml"
+
+
+class StoryGraphProvider(GraphComponent):
+    """Provides the training data from stories."""
+
+    def __init__(
+        self,
+        config: Dict[Text, Any],
+        model_storage: ModelStorage,
+        resource: Resource,
+        stories: StoryGraph = None,
+    ) -> None:
+        """Creates provider from config."""
+        self._config = config
+        self._model_storage = model_storage
+        self._resource = resource
+        self._stories = stories
+
+    @staticmethod
+    def get_default_config() -> Dict[Text, Any]:
+        """Returns default configuration (see parent class for full docstring)."""
+        return {"exclusion_percentage": None}
+
+    @classmethod
+    def create(
+        cls,
+        config: Dict[Text, Any],
+        model_storage: ModelStorage,
+        resource: Resource,
+        execution_context: ExecutionContext,
+    ) -> StoryGraphProvider:
+        """Creates component (see parent class for full docstring)."""
+        return cls(config, model_storage, resource)
+
+    @classmethod
+    def load(
+        cls,
+        config: Dict[Text, Any],
+        model_storage: ModelStorage,
+        resource: Resource,
+        execution_context: ExecutionContext,
+        **kwargs: Any,
+    ) -> StoryGraphProvider:
+        """Creates provider using a persisted version of itself."""
+        with model_storage.read_from(resource) as resource_directory:
+            reader = YAMLStoryReader()
+            story_steps = reader.read_from_file(
+                resource_directory / STORIES_PERSISTENCE_FILE_NAME
+            )
+        stories = StoryGraph(story_steps)
+        return cls(config, model_storage, resource, stories)
+
+    def _persist(self, story_steps: List[StoryStep]) -> None:
+        """Persists flows to model storage."""
+        with self._model_storage.write_to(self._resource) as resource_directory:
+            writer = YAMLStoryWriter()
+            writer.dump(
+                resource_directory / STORIES_PERSISTENCE_FILE_NAME,
+                story_steps,
+            )
+
+    def provide_train(self, importer: TrainingDataImporter) -> StoryGraph:
+        """Provides the story graph from the training data.
+
+        Args:
+            importer: instance of TrainingDataImporter.
+
+        Returns:
+            The story graph containing stories and rules used for training.
+        """
+        stories = importer.get_stories(**self._config)
+        self._persist(stories.story_steps)
+        return stories
+
+    def provide_inference(self) -> StoryGraph:
+        """Provides the stories configuration during inference."""
+        if self._stories is None:
+            self._stories = StoryGraph([])
+        return self._stories

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/nlu/classifiers/logistic_regression_classifier.py

@@ -3,7 +3,9 @@ from typing import Any, Text, Dict, List, Type, Tuple
 import joblib
 import structlog
 from scipy.sparse import hstack, vstack, csr_matrix
+from sklearn.exceptions import NotFittedError
 from sklearn.linear_model import LogisticRegression
+from sklearn.utils.validation import check_is_fitted
 
 from rasa.engine.storage.resource import Resource
 from rasa.engine.storage.storage import ModelStorage
@@ -154,6 +156,17 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
 
     def process(self, messages: List[Message]) -> List[Message]:
         """Return the most likely intent and its probability for a message."""
+        # Check if the classifier is trained
+        if not self.is_trained():
+            structlogger.warning(
+                "logistic_regression_classifier.not_trained.skip_intent_prediction",
+                event_info=(
+                    f"The '{self.__class__.__name__}' is not trained. "
+                    f"Skipping intent prediction."
+                ),
+            )
+            return messages
+
         X = self._create_X(messages)
         probas = self.clf.predict_proba(X)
         for idx, message in enumerate(messages):
@@ -216,3 +229,12 @@ class LogisticRegressionClassifier(IntentClassifier, GraphComponent):
     def validate_config(cls, config: Dict[Text, Any]) -> None:
         """Validates that the component is configured properly."""
         pass
+
+    def is_trained(self) -> bool:
+        """Checks if the model has been trained."""
+        try:
+            # This will raise a NotFittedError if the classifier isn't fitted
+            check_is_fitted(self.clf)
+            return True
+        except NotFittedError:
+            return False

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/nlu/featurizers/sparse_featurizer/count_vectors_featurizer.py

@@ -14,6 +14,7 @@ from rasa.nlu.featurizers.sparse_featurizer.sparse_featurizer import SparseFeatu
 from rasa.nlu.utils.spacy_utils import SpacyModel
 from rasa.shared.constants import DOCS_URL_COMPONENTS
 import rasa.utils.io as io_utils
+from sklearn.exceptions import NotFittedError
 from sklearn.feature_extraction.text import CountVectorizer
 from rasa.shared.nlu.training_data.training_data import TrainingData
 from rasa.shared.nlu.training_data.message import Message
@@ -551,36 +552,39 @@ class CountVectorsFeaturizer(SparseFeaturizer, GraphComponent):
         sequence_features: List[Optional[scipy.sparse.spmatrix]] = []
         sentence_features: List[Optional[scipy.sparse.spmatrix]] = []
 
-        for i, tokens in enumerate(all_tokens):
-            if not tokens:
-                # nothing to featurize
-                sequence_features.append(None)
-                sentence_features.append(None)
-                continue
-
-            # vectorizer.transform returns a sparse matrix of size
-            # [n_samples, n_features]
-            # set input to list of tokens if sequence should be returned
-            # otherwise join all tokens to a single string and pass that as a list
-            if not tokens:
-                # attribute is not set (e.g. response not present)
-                sequence_features.append(None)
-                sentence_features.append(None)
-                continue
-
-            seq_vec = self.vectorizers[attribute].transform(tokens)
-            seq_vec.sort_indices()
-
-            sequence_features.append(seq_vec.tocoo())
-
-            if attribute in DENSE_FEATURIZABLE_ATTRIBUTES:
-                tokens_text = [" ".join(tokens)]
-                sentence_vec = self.vectorizers[attribute].transform(tokens_text)
-                sentence_vec.sort_indices()
-
-                sentence_features.append(sentence_vec.tocoo())
-            else:
-                sentence_features.append(None)
+        try:
+            for i, tokens in enumerate(all_tokens):
+                # vectorizer.transform returns a sparse matrix of size
+                # [n_samples, n_features]
+                # set input to list of tokens if sequence should be returned
+                # otherwise join all tokens to a single string and pass that as a list
+                if not tokens:
+                    # attribute is not set (e.g. response not present)
+                    sequence_features.append(None)
+                    sentence_features.append(None)
+                    continue
+
+                seq_vec = self.vectorizers[attribute].transform(tokens)
+                seq_vec.sort_indices()
+
+                sequence_features.append(seq_vec.tocoo())
+
+                if attribute in DENSE_FEATURIZABLE_ATTRIBUTES:
+                    tokens_text = [" ".join(tokens)]
+                    sentence_vec = self.vectorizers[attribute].transform(tokens_text)
+                    sentence_vec.sort_indices()
+
+                    sentence_features.append(sentence_vec.tocoo())
+                else:
+                    sentence_features.append(None)
+        except NotFittedError:
+            logger.warning(
+                f"Unable to train CountVectorizer for message "
+                f"attribute - {attribute}, since the call to sklearn's "
+                f"`.fit()` method failed. Leaving an untrained "
+                f"CountVectorizer for it."
+            )
+            return [None], [None]
 
         return sequence_features, sentence_features
 

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/shared/constants.py

@@ -181,6 +181,7 @@ PROMPT_TEMPLATE_CONFIG_KEY = "prompt_template"
 
 STREAM_CONFIG_KEY = "stream"
 N_REPHRASES_CONFIG_KEY = "n"
+USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY = "use_chat_completions_endpoint"
 
 AZURE_API_KEY_ENV_VAR = "AZURE_API_KEY"
 AZURE_AD_TOKEN_ENV_VAR = "AZURE_AD_TOKEN"

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/shared/core/domain.py

@@ -790,13 +790,18 @@ class Domain:
             }
         else:
             intent_name = next(iter(intent.keys()))
-
-        return (
-            intent_name,
-            cls._transform_intent_properties_for_internal_use(
-                intent, entity_properties
-            ),
-        )
+        try:
+            return (
+                intent_name,
+                cls._transform_intent_properties_for_internal_use(
+                    intent, entity_properties
+                ),
+            )
+        except AttributeError:
+            raise InvalidDomain(
+                f"Detected invalid intent definition: {intent}. "
+                f"Please make sure all intent definitions are valid."
+            )
 
     @classmethod
     def _add_default_intents(

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/shared/core/training_data/structures.py

@@ -804,6 +804,26 @@ class StoryGraph:
         """Returns text representation of object."""
         return f"{self.__class__.__name__}: {len(self.story_steps)} story steps"
 
+    def has_e2e_stories(self) -> bool:
+        """
+        Checks if there are end-to-end (E2E) stories present in the story steps.
+
+        An E2E story is determined by checking if any `UserUttered` event has
+        associated text within the story steps.
+
+        Returns:
+            bool: True if any E2E story (i.e., a `UserUttered` event with text)
+            is found, False otherwise.
+        """
+        if not self.story_steps:
+            return False
+        for story_step in self.story_steps:
+            for event in story_step.events:
+                if isinstance(event, UserUttered):
+                    if event.text:
+                        return True
+        return False
+
 
 def generate_id(prefix: Text = "", max_chars: Optional[int] = None) -> Text:
     """Generate a random UUID.

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/shared/providers/_configs/self_hosted_llm_client_config.py

@@ -21,6 +21,7 @@ from rasa.shared.constants import (
     PROVIDER_CONFIG_KEY,
     OPENAI_PROVIDER,
     SELF_HOSTED_PROVIDER,
+    USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY,
 )
 from rasa.shared.providers._configs.utils import (
     raise_deprecation_warnings,
@@ -71,6 +72,7 @@ class SelfHostedLLMClientConfig:
     api_base: str
     api_version: Optional[str] = None
     api_type: Optional[str] = OPENAI_PROVIDER
+    use_chat_completions_endpoint: Optional[bool] = True
     extra_parameters: dict = field(default_factory=dict)
 
     def __post_init__(self) -> None:
@@ -140,6 +142,9 @@ class SelfHostedLLMClientConfig:
             # Optional parameters
             api_type=config.pop(API_TYPE_CONFIG_KEY, OPENAI_PROVIDER),
             api_version=config.pop(API_VERSION_CONFIG_KEY, None),
+            use_chat_completions_endpoint=config.pop(
+                USE_CHAT_COMPLETIONS_ENDPOINT_CONFIG_KEY, True
+            ),
             # The rest of parameters (e.g. model parameters) are considered
             # as extra parameters
             extra_parameters=config,

{rasa_pro-3.10.4 → rasa_pro-3.10.5}/rasa/shared/providers/llm/_base_litellm_client.py

@@ -3,7 +3,11 @@ from typing import Dict, List, Any, Union
 
 import logging
 import structlog
-from litellm import completion, acompletion, validate_environment
+from litellm import (
+    completion,
+    acompletion,
+    validate_environment,
+)
 
 from rasa.shared.exceptions import (
     ProviderClientAPIException,