rasa-pro 3.11.0a4.dev2__py3-none-any.whl → 3.11.0rc1__py3-none-any.whl

rasa/telemetry.py

@@ -32,7 +32,13 @@ from rasa.constants import (
     CONFIG_TELEMETRY_ENABLED,
     CONFIG_TELEMETRY_ID,
 )
-from rasa.shared.constants import PROMPT_CONFIG_KEY, PROMPT_TEMPLATE_CONFIG_KEY
+from rasa.shared.constants import (
+    PROMPT_CONFIG_KEY,
+    PROMPT_TEMPLATE_CONFIG_KEY,
+    MODEL_GROUP_CONFIG_KEY,
+    LLM_API_HEALTH_CHECK_ENV_VAR,
+    LLM_API_HEALTH_CHECK_DEFAULT_VALUE,
+)
 from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.shared.constants import DOCS_URL_TELEMETRY, UTTER_ASK_PREFIX
 from rasa.shared.core.flows import Flow
@@ -157,6 +163,7 @@ NUM_LINK_STEPS = "num_link_steps"
 NUM_CALL_STEPS = "num_call_steps"
 NUM_SHARED_SLOTS_BETWEEN_FLOWS = "num_shared_slots_between_flows"
 LLM_COMMAND_GENERATOR_MODEL_NAME = "llm_command_generator_model_name"
+LLM_COMMAND_GENERATOR_MODEL_GROUP_ID = "llm_command_generator_model_group_id"
 LLM_COMMAND_GENERATOR_CUSTOM_PROMPT_USED = "llm_command_generator_custom_prompt_used"
 MULTI_STEP_LLM_COMMAND_GENERATOR_HANDLE_FLOWS_PROMPT_USED = (
     "multi_step_llm_command_generator_custom_handle_flows_prompt_used"
@@ -166,6 +173,7 @@ MULTI_STEP_LLM_COMMAND_GENERATOR_FILL_SLOTS_PROMPT_USED = (
 )
 FLOW_RETRIEVAL_ENABLED = "flow_retrieval_enabled"
 FLOW_RETRIEVAL_EMBEDDING_MODEL_NAME = "flow_retrieval_embedding_model_name"
+FLOW_RETRIEVAL_EMBEDDING_MODEL_GROUP_ID = "flow_retrieval_embedding_model_group_id"
 TRACING_BACKEND = "tracing_backend"
 METRICS_BACKEND = "metrics_backend"
 VERSION = "version"
@@ -960,6 +968,13 @@ def track_model_training(
         "policies": config.get("policies"),
         "train_schema": config.get("train_schema"),
         "predict_schema": config.get("predict_schema"),
+        "model_groups": rasa.core.utils.AvailableEndpoints.get_instance().model_groups,
+        "api_health_check_enabled": (
+            os.getenv(
+                LLM_API_HEALTH_CHECK_ENV_VAR, LLM_API_HEALTH_CHECK_DEFAULT_VALUE
+            ).lower()
+            == "true"
+        ),
         "num_intent_examples": len(nlu_data.intent_examples),
         "num_entity_examples": len(nlu_data.entity_examples),
         "num_actions": len(domain.action_names_or_texts),
@@ -1114,46 +1129,76 @@ def _get_llm_command_generator_config(config: Dict[str, Any]) -> Optional[Dict]:
                 return component
         return None
 
-    def extract_settings(component: Dict) -> Dict:
-        """Extracts the settings from the command generator component."""
+    def extract_llm_command_generator_llm_client_settings(component: Dict) -> Dict:
+        """Extracts settings related to LLM command generator."""
         llm_config = component.get(LLM_CONFIG_KEY, {})
-        llm_model_name = (
-            llm_config.get(MODEL_CONFIG_KEY)
-            or llm_config.get(MODEL_NAME_CONFIG_KEY)
-            or DEFAULT_LLM_CONFIG[MODEL_CONFIG_KEY]
+        llm_model_group_id = llm_config.get(MODEL_GROUP_CONFIG_KEY)
+        llm_model_name = llm_config.get(MODEL_CONFIG_KEY) or llm_config.get(
+            MODEL_NAME_CONFIG_KEY
+        )
+        if llm_model_group_id is None and llm_model_name is None:
+            llm_model_name = DEFAULT_LLM_CONFIG[MODEL_CONFIG_KEY]
+
+        custom_prompt_used = (
+            PROMPT_CONFIG_KEY in component or PROMPT_TEMPLATE_CONFIG_KEY in component
         )
+        return {
+            LLM_COMMAND_GENERATOR_MODEL_NAME: llm_model_name,
+            LLM_COMMAND_GENERATOR_MODEL_GROUP_ID: llm_model_group_id,
+            LLM_COMMAND_GENERATOR_CUSTOM_PROMPT_USED: custom_prompt_used,
+        }
+
+    def extract_multistep_command_generator_prompt_settings(component: Dict) -> Dict:
+        """Extracts settings related to multistep command generator."""
+        prompt_templates = component.get("prompt_templates", {})
+        handle_flows_prompt_used = HANDLE_FLOWS_KEY in prompt_templates
+        fill_slots_prompt_used = FILL_SLOTS_KEY in prompt_templates
+        return {
+            MULTI_STEP_LLM_COMMAND_GENERATOR_HANDLE_FLOWS_PROMPT_USED: handle_flows_prompt_used,  # noqa: E501
+            MULTI_STEP_LLM_COMMAND_GENERATOR_FILL_SLOTS_PROMPT_USED: fill_slots_prompt_used,  # noqa: E501
+        }
+
+    def extract_flow_retrieval_settings(component: Dict) -> Dict:
+        """Extracts settings related to flow retrieval."""
         flow_retrieval_config = component.get(FLOW_RETRIEVAL_KEY, {})
         flow_retrieval_enabled = flow_retrieval_config.get("active", True)
-        flow_retrieval_embeddings_config = flow_retrieval_config.get(
+        embeddings_config = flow_retrieval_config.get(
             EMBEDDINGS_CONFIG_KEY, DEFAULT_EMBEDDINGS_CONFIG
         )
         flow_retrieval_embedding_model_name = (
             (
-                flow_retrieval_embeddings_config.get(MODEL_NAME_CONFIG_KEY)
-                or flow_retrieval_embeddings_config.get(MODEL_CONFIG_KEY)
+                embeddings_config.get(MODEL_NAME_CONFIG_KEY)
+                or embeddings_config.get(MODEL_CONFIG_KEY)
             )
             if flow_retrieval_enabled
             else None
         )
+        flow_retrieval_embedding_model_group_id = embeddings_config.get(
+            MODEL_GROUP_CONFIG_KEY
+        )
         return {
-            LLM_COMMAND_GENERATOR_MODEL_NAME: llm_model_name,
-            LLM_COMMAND_GENERATOR_CUSTOM_PROMPT_USED: PROMPT_CONFIG_KEY in component
-            or PROMPT_TEMPLATE_CONFIG_KEY in component,
-            MULTI_STEP_LLM_COMMAND_GENERATOR_HANDLE_FLOWS_PROMPT_USED: HANDLE_FLOWS_KEY
-            in component.get("prompt_templates", {}),
-            MULTI_STEP_LLM_COMMAND_GENERATOR_FILL_SLOTS_PROMPT_USED: FILL_SLOTS_KEY
-            in component.get("prompt_templates", {}),
             FLOW_RETRIEVAL_ENABLED: flow_retrieval_enabled,
             FLOW_RETRIEVAL_EMBEDDING_MODEL_NAME: flow_retrieval_embedding_model_name,
+            FLOW_RETRIEVAL_EMBEDDING_MODEL_GROUP_ID: flow_retrieval_embedding_model_group_id,  # noqa: E501
         }
 
+    def extract_settings(component: Dict) -> Dict:
+        """Extracts the settings from the command generator component."""
+        settings = {}
+        settings.update(extract_llm_command_generator_llm_client_settings(component))
+        settings.update(extract_multistep_command_generator_prompt_settings(component))
+        settings.update(extract_flow_retrieval_settings(component))
+        return settings
+
     command_generator_config = {
         LLM_COMMAND_GENERATOR_MODEL_NAME: None,
+        LLM_COMMAND_GENERATOR_MODEL_GROUP_ID: None,
         LLM_COMMAND_GENERATOR_CUSTOM_PROMPT_USED: None,
         MULTI_STEP_LLM_COMMAND_GENERATOR_HANDLE_FLOWS_PROMPT_USED: None,
         MULTI_STEP_LLM_COMMAND_GENERATOR_FILL_SLOTS_PROMPT_USED: None,
         FLOW_RETRIEVAL_ENABLED: None,
         FLOW_RETRIEVAL_EMBEDDING_MODEL_NAME: None,
+        FLOW_RETRIEVAL_EMBEDDING_MODEL_GROUP_ID: None,
     }
 
     pipeline = config.get("pipeline", [])
@@ -1553,6 +1598,7 @@ def track_response_rephrase(
     custom_prompt_template: Optional[str],
     llm_type: Optional[str],
     llm_model: Optional[str],
+    llm_model_group_id: Optional[str],
 ) -> None:
     """Track when a user rephrases a response."""
     _track(
@@ -1562,6 +1608,7 @@ def track_response_rephrase(
             "custom_prompt_template": custom_prompt_template,
             "llm_type": llm_type,
             "llm_model": llm_model,
+            "llm_model_group_id": llm_model_group_id,
         },
     )
 
@@ -1576,8 +1623,10 @@ def track_intentless_policy_train() -> None:
 def track_intentless_policy_train_completed(
     embeddings_type: Optional[str],
     embeddings_model: Optional[str],
+    embeddings_model_group_id: Optional[str],
     llm_type: Optional[str],
     llm_model: Optional[str],
+    llm_model_group_id: Optional[str],
 ) -> None:
     """Track when a user trains a policy."""
     _track(
@@ -1585,8 +1634,10 @@ def track_intentless_policy_train_completed(
         {
             "embeddings_type": embeddings_type,
             "embeddings_model": embeddings_model,
+            "embeddings_model_group_id": embeddings_model_group_id,
             "llm_type": llm_type,
             "llm_model": llm_model,
+            "llm_model_group_id": llm_model_group_id,
         },
     )
 
@@ -1595,8 +1646,10 @@ def track_intentless_policy_train_completed(
 def track_intentless_policy_predict(
     embeddings_type: Optional[str],
     embeddings_model: Optional[str],
+    embeddings_model_group_id: Optional[str],
     llm_type: Optional[str],
     llm_model: Optional[str],
+    llm_model_group_id: Optional[str],
     score: float,
 ) -> None:
     """Track when a user trains a policy."""
@@ -1605,8 +1658,10 @@ def track_intentless_policy_predict(
         {
             "embeddings_type": embeddings_type,
             "embeddings_model": embeddings_model,
+            "embeddings_model_group_id": embeddings_model_group_id,
             "llm_type": llm_type,
             "llm_model": llm_model,
+            "llm_model_group_id": llm_model_group_id,
             "score": score,
         },
     )
@@ -1696,8 +1751,10 @@ def track_enterprise_search_policy_train_completed(
     vector_store_type: Optional[str],
     embeddings_type: Optional[str],
     embeddings_model: Optional[str],
+    embeddings_model_group_id: Optional[str],
     llm_type: Optional[str],
     llm_model: Optional[str],
+    llm_model_group_id: Optional[str],
     citation_enabled: Optional[bool],
 ) -> None:
     """Track when a user completes training Enterprise Search policy."""
@@ -1707,8 +1764,10 @@ def track_enterprise_search_policy_train_completed(
             "vector_store_type": vector_store_type,
             "embeddings_type": embeddings_type,
             "embeddings_model": embeddings_model,
+            "embeddings_model_group_id": embeddings_model_group_id,
             "llm_type": llm_type,
             "llm_model": llm_model,
+            "llm_model_group_id": llm_model_group_id,
             "citation_enabled": citation_enabled,
         },
     )
@@ -1719,8 +1778,10 @@ def track_enterprise_search_policy_predict(
     vector_store_type: Optional[str],
     embeddings_type: Optional[str],
     embeddings_model: Optional[str],
+    embeddings_model_group_id: Optional[str],
     llm_type: Optional[str],
     llm_model: Optional[str],
+    llm_model_group_id: Optional[str],
     citation_enabled: Optional[bool],
 ) -> None:
     """Track when a user predicts the next action using Enterprise Search policy."""
@@ -1730,8 +1791,10 @@ def track_enterprise_search_policy_predict(
             "vector_store_type": vector_store_type,
             "embeddings_type": embeddings_type,
             "embeddings_model": embeddings_model,
+            "embeddings_model_group_id": embeddings_model_group_id,
             "llm_type": llm_type,
             "llm_model": llm_model,
+            "llm_model_group_id": llm_model_group_id,
             "citation_enabled": citation_enabled,
         },
     )

rasa/tracing/instrumentation/attribute_extractors.py

@@ -5,6 +5,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Text, Tuple, Union
 
 import tiktoken
 from numpy import ndarray
+from rasa.dialogue_understanding.generator.constants import FLOW_RETRIEVAL_KEY
 from rasa_sdk.grpc_py import action_webhook_pb2
 
 from rasa.core.actions.action import DirectCustomActionExecutor
@@ -19,6 +20,7 @@ from rasa.core.processor import MessageProcessor
 from rasa.core.tracker_store import TrackerStore
 from rasa.dialogue_understanding.commands import Command
 from rasa.dialogue_understanding.stack.dialogue_stack import DialogueStack
+from rasa.dialogue_understanding.generator import LLMBasedCommandGenerator
 from rasa.engine.graph import ExecutionContext, GraphModelConfiguration, GraphNode
 from rasa.engine.training.graph_trainer import GraphTrainer
 from rasa.shared.constants import (
@@ -27,6 +29,8 @@ from rasa.shared.constants import (
     PROVIDER_CONFIG_KEY,
     TIMEOUT_CONFIG_KEY,
     DEPLOYMENT_CONFIG_KEY,
+    MODEL_GROUP_ID_CONFIG_KEY,
+    LLM_CONFIG_KEY,
 )
 from rasa.shared.core.constants import REQUESTED_SLOT
 from rasa.shared.core.domain import Domain
@@ -50,10 +54,7 @@ if TYPE_CHECKING:
     from rasa.core.policies.enterprise_search_policy import EnterpriseSearchPolicy
     from rasa.core.policies.intentless_policy import IntentlessPolicy
     from rasa.core.policies.policy import PolicyPrediction
-    from rasa.dialogue_understanding.generator import (
-        CommandGenerator,
-        LLMBasedCommandGenerator,
-    )
+    from rasa.dialogue_understanding.generator import CommandGenerator
 
 # This file contains all attribute extractors for tracing instrumentation.
 # These are functions that are applied to the arguments of the wrapped function to be
@@ -300,22 +301,49 @@ def extract_attrs_for_command(
     }
 
 
-def extract_llm_config(self: Any, default_llm_config: Dict[str, Any]) -> Dict[str, Any]:
+def extract_llm_config(
+    self: Any,
+    default_llm_config: Dict[str, Any],
+    default_embeddings_config: Dict[str, Any],
+) -> Dict[str, Any]:
     if isinstance(self, ContextualResponseRephraser):
-        config = self.nlg_endpoint.kwargs
+        # ContextualResponseRephraser is not a graph component, so it's
+        # not having a full config.
+        config = {"llm": self.llm_config}
     else:
         config = self.config
 
     llm_property = combine_custom_and_default_config(
-        config.get("llm"), default_llm_config
+        config.get(LLM_CONFIG_KEY), default_llm_config
     )
 
+    if isinstance(self, LLMBasedCommandGenerator):
+        flow_retrieval_config = config.get(FLOW_RETRIEVAL_KEY, {}) or {}
+        embeddings_property = combine_custom_and_default_config(
+            flow_retrieval_config.get(EMBEDDINGS_CONFIG_KEY),
+            default_embeddings_config,
+        )
+    else:
+        embeddings_property = combine_custom_and_default_config(
+            config.get(EMBEDDINGS_CONFIG_KEY), default_embeddings_config
+        )
+
     attributes = {
         "class_name": self.__class__.__name__,
+        # llm client attributes
         "llm_model": str(llm_property.get(MODEL_CONFIG_KEY)),
         "llm_type": str(llm_property.get(PROVIDER_CONFIG_KEY)),
-        "embeddings": json.dumps(config.get(EMBEDDINGS_CONFIG_KEY, {})),
+        "llm_model_group_id": str(llm_property.get(MODEL_GROUP_ID_CONFIG_KEY)),
         "llm_temperature": str(llm_property.get("temperature")),
+        "llm_request_timeout": str(llm_property.get(TIMEOUT_CONFIG_KEY)),
+        # embedding client attributes
+        "embeddings_model": str(embeddings_property.get(MODEL_CONFIG_KEY)),
+        "embeddings_type": str(embeddings_property.get(PROVIDER_CONFIG_KEY)),
+        "embeddings_model_group_id": str(
+            embeddings_property.get(MODEL_GROUP_ID_CONFIG_KEY)
+        ),
+        # TODO: Keeping this to avoid potential breaking changes
+        "embeddings": json.dumps(embeddings_property, sort_keys=True),
         "request_timeout": str(llm_property.get(TIMEOUT_CONFIG_KEY)),
     }
 
@@ -329,11 +357,16 @@ def extract_attrs_for_llm_based_command_generator(
     self: "LLMBasedCommandGenerator",
     prompt: str,
 ) -> Dict[str, Any]:
-    from rasa.dialogue_understanding.generator.constants import (
-        DEFAULT_LLM_CONFIG,
+    from rasa.dialogue_understanding.generator.constants import DEFAULT_LLM_CONFIG
+    from rasa.dialogue_understanding.generator.flow_retrieval import (
+        DEFAULT_EMBEDDINGS_CONFIG,
     )
 
-    attributes = extract_llm_config(self, default_llm_config=DEFAULT_LLM_CONFIG)
+    attributes = extract_llm_config(
+        self,
+        default_llm_config=DEFAULT_LLM_CONFIG,
+        default_embeddings_config=DEFAULT_EMBEDDINGS_CONFIG,
+    )
 
     return extend_attributes_with_prompt_tokens_length(self, attributes, prompt)
 
@@ -344,7 +377,12 @@ def extract_attrs_for_contextual_response_rephraser(
 ) -> Dict[str, Any]:
     from rasa.core.nlg.contextual_response_rephraser import DEFAULT_LLM_CONFIG
 
-    attributes = extract_llm_config(self, default_llm_config=DEFAULT_LLM_CONFIG)
+    attributes = extract_llm_config(
+        self,
+        default_llm_config=DEFAULT_LLM_CONFIG,
+        # rephraser is not using embeddings
+        default_embeddings_config={},
+    )
 
     return extend_attributes_with_prompt_tokens_length(self, attributes, prompt)
 
@@ -355,7 +393,12 @@ def extract_attrs_for_create_history(
 ) -> Dict[str, Any]:
     from rasa.core.nlg.contextual_response_rephraser import DEFAULT_LLM_CONFIG
 
-    return extract_llm_config(self, default_llm_config=DEFAULT_LLM_CONFIG)
+    return extract_llm_config(
+        self,
+        default_llm_config=DEFAULT_LLM_CONFIG,
+        # rephraser is not using embeddings
+        default_embeddings_config={},
+    )
 
 
 def extract_attrs_for_generate(
@@ -580,9 +623,16 @@ def extract_attrs_for_intentless_policy_find_closest_response(
 def extract_attrs_for_intentless_policy_generate_llm_answer(
     self: "IntentlessPolicy", llm: "BaseLLM", prompt: str
 ) -> Dict[str, Any]:
-    from rasa.core.policies.intentless_policy import DEFAULT_LLM_CONFIG
+    from rasa.core.policies.intentless_policy import (
+        DEFAULT_LLM_CONFIG,
+        DEFAULT_EMBEDDINGS_CONFIG,
+    )
 
-    attributes = extract_llm_config(self, default_llm_config=DEFAULT_LLM_CONFIG)
+    attributes = extract_llm_config(
+        self,
+        default_llm_config=DEFAULT_LLM_CONFIG,
+        default_embeddings_config=DEFAULT_EMBEDDINGS_CONFIG,
+    )
 
     return extend_attributes_with_prompt_tokens_length(self, attributes, prompt)
 
@@ -590,9 +640,16 @@ def extract_attrs_for_intentless_policy_generate_llm_answer(
 def extract_attrs_for_enterprise_search_generate_llm_answer(
     self: "EnterpriseSearchPolicy", llm: "BaseLLM", prompt: str
 ) -> Dict[str, Any]:
-    from rasa.core.policies.enterprise_search_policy import DEFAULT_LLM_CONFIG
+    from rasa.core.policies.enterprise_search_policy import (
+        DEFAULT_LLM_CONFIG,
+        DEFAULT_EMBEDDINGS_CONFIG,
+    )
 
-    attributes = extract_llm_config(self, default_llm_config=DEFAULT_LLM_CONFIG)
+    attributes = extract_llm_config(
+        self,
+        default_llm_config=DEFAULT_LLM_CONFIG,
+        default_embeddings_config=DEFAULT_EMBEDDINGS_CONFIG,
+    )
 
     return extend_attributes_with_prompt_tokens_length(self, attributes, prompt)
 

rasa/utils/io.py

@@ -2,7 +2,6 @@ import asyncio
 import filecmp
 import logging
 import os
-import pickle
 import tempfile
 import warnings
 import re
@@ -98,29 +97,6 @@ def enable_async_loop_debugging(
     return event_loop
 
 
-def pickle_dump(filename: Union[Text, Path], obj: Any) -> None:
-    """Saves object to file.
-
-    Args:
-        filename: the filename to save the object to
-        obj: the object to store
-    """
-    with open(filename, "wb") as f:
-        pickle.dump(obj, f)
-
-
-def pickle_load(filename: Union[Text, Path]) -> Any:
-    """Loads an object from a file.
-
-    Args:
-        filename: the filename to load the object from
-
-    Returns: the loaded object
-    """
-    with open(filename, "rb") as f:
-        return pickle.load(f)
-
-
 def create_temporary_file(data: Any, suffix: Text = "", mode: Text = "w+") -> Text:
     """Creates a tempfile.NamedTemporaryFile object for data."""
     encoding = None if "b" in mode else rasa.shared.utils.io.DEFAULT_ENCODING
@@ -191,48 +167,6 @@ def create_validator(
     return FunctionValidator
 
 
-def json_unpickle(
-    file_name: Union[Text, Path], encode_non_string_keys: bool = False
-) -> Any:
-    """Unpickle an object from file using json.
-
-    Args:
-        file_name: the file to load the object from
-        encode_non_string_keys: If set to `True` then jsonpickle will encode non-string
-          dictionary keys instead of coercing them into strings via `repr()`.
-
-    Returns: the object
-    """
-    import jsonpickle.ext.numpy as jsonpickle_numpy
-    import jsonpickle
-
-    jsonpickle_numpy.register_handlers()
-
-    file_content = rasa.shared.utils.io.read_file(file_name)
-    return jsonpickle.loads(file_content, keys=encode_non_string_keys)
-
-
-def json_pickle(
-    file_name: Union[Text, Path], obj: Any, encode_non_string_keys: bool = False
-) -> None:
-    """Pickle an object to a file using json.
-
-    Args:
-        file_name: the file to store the object to
-        obj: the object to store
-        encode_non_string_keys: If set to `True` then jsonpickle will encode non-string
-          dictionary keys instead of coercing them into strings via `repr()`.
-    """
-    import jsonpickle.ext.numpy as jsonpickle_numpy
-    import jsonpickle
-
-    jsonpickle_numpy.register_handlers()
-
-    rasa.shared.utils.io.write_text_file(
-        jsonpickle.dumps(obj, keys=encode_non_string_keys), file_name
-    )
-
-
 def get_emoji_regex() -> Pattern:
     """Returns regex to identify emojis."""
     return re.compile(

rasa/utils/log_utils.py

@@ -78,6 +78,7 @@ def _anonymizer(
 
 def configure_structlog(
     log_level: Optional[int] = None,
+    include_time: bool = False,
 ) -> None:
     """Configure logging of the server."""
     if log_level is None:  # Log level NOTSET is 0 so we use `is None` here
@@ -114,6 +115,9 @@ def configure_structlog(
         SentryProcessor(event_level=logging.FATAL),
     ]
 
+    if include_time:
+        shared_processors.append(structlog.processors.TimeStamper(fmt="iso"))
+
     if not FORCE_JSON_LOGGING and sys.stderr.isatty():
         # Pretty printing when we run in a terminal session.
         # Automatically prints pretty tracebacks when "rich" is installed
@@ -143,11 +147,14 @@ def configure_structlog(
         # logger.
         cache_logger_on_first_use=True,
     )
+    # doing logger creation inline, to prevent usage of unconfigured logger
+    structlog.get_logger().debug("structlog.configured")
 
 
 def log_llm(logger: Any, log_module: str, log_event: str, **kwargs: Any) -> None:
-    """Logs LLM-specific events depending on a flag passed through an environment
-    variable. If the module's flag is set to INFO (e.g.
+    """Logs LLM-specific events depending on a flag passed through an env var.
+
+    If the module's flag is set to INFO (e.g.
     LOG_PROMPT_LLM_COMMAND_GENERATOR=INFO), its prompt is logged at INFO level,
     overriding the general log level setting.