datarobot-moderations 11.1.16__py3-none-any.whl → 11.1.18__py3-none-any.whl

datarobot_dome/constants.py

@@ -41,6 +41,7 @@ RETRY_COUNT = 10
 MODERATION_CONFIG_FILE_NAME = "moderation_config.yaml"
 DATAROBOT_SERVERLESS_PLATFORM = "datarobotServerless"
 
+RUNTIME_PARAMETER_PREFIX = "MLOPS_RUNTIME_PARAM_"
 SECRET_DEFINITION_PREFIX = "MLOPS_RUNTIME_PARAM_MODERATION"
 OPENAI_SECRET_DEFINITION_SUFFIX = "OPENAI_API_KEY"
 GOOGLE_SERVICE_ACCOUNT_SECRET_DEFINITION_SUFFIX = "GOOGLE_SERVICE_ACCOUNT"
@@ -62,8 +63,8 @@ PROMPT_VECTOR_ATTR = "prompt_vector"
 DATAROBOT_CONFIGURED_ON_PREM_ST_SAAS_URL = "http://datarobot-nginx/api/v2"
 DATAROBOT_ACTUAL_ON_PREM_ST_SAAS_URL = "http://datarobot-prediction-server:80/predApi/v1.0"
 
-
-DISABLE_MODERATION_RUNTIME_PARAM_NAME = "MLOPS_RUNTIME_PARAM_DISABLE_MODERATION"
+DISABLE_MODERATION_RUNTIME_PARAM_NAME = "DISABLE_MODERATION"
+ENABLE_LLM_GATEWAY_INFERENCE_RUNTIME_PARAM_NAME = "ENABLE_LLM_GATEWAY_INFERENCE"
 
 LLM_CONTEXT_COLUMN_NAME = "_LLM_CONTEXT"
 PROMPT_TOKEN_COUNT_COLUMN_NAME_FROM_USAGE = "prompt_token_count_from_usage"
@@ -242,6 +243,11 @@ class AwsModel:
     ANTHROPIC_CLAUDE_3_HAIKU = "anthropic-claude-3-haiku"
     ANTHROPIC_CLAUDE_3_SONNET = "anthropic-claude-3-sonnet"
     ANTHROPIC_CLAUDE_3_OPUS = "anthropic-claude-3-opus"
+    ANTHROPIC_CLAUDE_3_5_SONNET_V1 = "anthropic-claude-3.5-sonnet-v1"
+    ANTHROPIC_CLAUDE_3_5_SONNET_V2 = "anthropic-claude-3.5-sonnet-v2"
+    AMAZON_NOVA_LITE = "amazon-nova-lite"
+    AMAZON_NOVA_MICRO = "amazon-nova-micro"
+    AMAZON_NOVA_PRO = "amazon-nova-pro"
 
     ALL = [
         TITAN,
@@ -249,6 +255,11 @@ class AwsModel:
         ANTHROPIC_CLAUDE_3_HAIKU,
         ANTHROPIC_CLAUDE_3_SONNET,
         ANTHROPIC_CLAUDE_3_OPUS,
+        ANTHROPIC_CLAUDE_3_5_SONNET_V1,
+        ANTHROPIC_CLAUDE_3_5_SONNET_V2,
+        AMAZON_NOVA_LITE,
+        AMAZON_NOVA_MICRO,
+        AMAZON_NOVA_PRO,
     ]
 
 
@@ -266,6 +277,11 @@ class AwsModelVersion:
     ANTHROPIC_CLAUDE_3_HAIKU = "anthropic.claude-3-haiku-20240307-v1:0"
     ANTHROPIC_CLAUDE_3_SONNET = "anthropic.claude-3-sonnet-20240229-v1:0"
     ANTHROPIC_CLAUDE_3_OPUS = "anthropic.claude-3-opus-20240229-v1:0"
+    ANTHROPIC_CLAUDE_3_5_SONNET_V1 = "anthropic.claude-3-5-sonnet-20240620-v1:0"
+    ANTHROPIC_CLAUDE_3_5_SONNET_V2 = "anthropic.claude-3-5-sonnet-20241022-v2:0"
+    AMAZON_NOVA_LITE = "amazon.nova-lite-v1:0"
+    AMAZON_NOVA_MICRO = "amazon.nova-micro-v1:0"
+    AMAZON_NOVA_PRO = "amazon.nova-pro-v1:0"
 
     ALL = [
         TITAN,
@@ -273,6 +289,11 @@ class AwsModelVersion:
         ANTHROPIC_CLAUDE_3_HAIKU,
         ANTHROPIC_CLAUDE_3_SONNET,
         ANTHROPIC_CLAUDE_3_OPUS,
+        ANTHROPIC_CLAUDE_3_5_SONNET_V1,
+        ANTHROPIC_CLAUDE_3_5_SONNET_V2,
+        AMAZON_NOVA_LITE,
+        AMAZON_NOVA_MICRO,
+        AMAZON_NOVA_PRO,
     ]
 
 
@@ -288,6 +309,11 @@ AWS_MODEL_TO_AWS_MODEL_VERSION_MAP = {
     AwsModel.ANTHROPIC_CLAUDE_3_HAIKU: AwsModelVersion.ANTHROPIC_CLAUDE_3_HAIKU,
     AwsModel.ANTHROPIC_CLAUDE_3_SONNET: AwsModelVersion.ANTHROPIC_CLAUDE_3_SONNET,
     AwsModel.ANTHROPIC_CLAUDE_3_OPUS: AwsModelVersion.ANTHROPIC_CLAUDE_3_OPUS,
+    AwsModel.ANTHROPIC_CLAUDE_3_5_SONNET_V1: AwsModelVersion.ANTHROPIC_CLAUDE_3_5_SONNET_V1,
+    AwsModel.ANTHROPIC_CLAUDE_3_5_SONNET_V2: AwsModelVersion.ANTHROPIC_CLAUDE_3_5_SONNET_V2,
+    AwsModel.AMAZON_NOVA_LITE: AwsModelVersion.AMAZON_NOVA_LITE,
+    AwsModel.AMAZON_NOVA_MICRO: AwsModelVersion.AMAZON_NOVA_MICRO,
+    AwsModel.AMAZON_NOVA_PRO: AwsModelVersion.AMAZON_NOVA_PRO,
 }
 
 

datarobot_dome/drum_integration.py

@@ -55,6 +55,7 @@ from datarobot_dome.constants import TargetType
 from datarobot_dome.guard_executor import AsyncGuardExecutor
 from datarobot_dome.pipeline.llm_pipeline import LLMPipeline
 from datarobot_dome.pipeline.vdb_pipeline import VDBPipeline
+from datarobot_dome.runtime import get_runtime_parameter_value_bool
 from datarobot_dome.streaming import ModerationIterator
 from datarobot_dome.streaming import StreamingContextBuilder
 
@@ -915,13 +916,11 @@ def init(model_dir: str = os.getcwd()):
         pipeline: A Guard pipeline object required to enforce moderations while
             scoring on user data
     """
-    disable_moderation_runtime_value = json.loads(
-        os.environ.get(DISABLE_MODERATION_RUNTIME_PARAM_NAME, "{}")
+    disable_moderation_runtime_value = get_runtime_parameter_value_bool(
+        param_name=DISABLE_MODERATION_RUNTIME_PARAM_NAME,
+        default_value=False,
     )
-    if (
-        "payload" in disable_moderation_runtime_value
-        and disable_moderation_runtime_value["payload"]
-    ):
+    if disable_moderation_runtime_value:
         _logger.warning("Moderation is disabled via runtime parameter on the model")
         return None
 
@@ -999,6 +998,8 @@ class VdbModerationPipeline(ModerationPipeline):
 def moderation_pipeline_factory(
     target_type: str, model_dir: str = os.getcwd()
 ) -> Optional[ModerationPipeline]:
+    # Disable ragas tracking while loading the module.
+    os.environ["RAGAS_DO_NOT_TRACK"] = "true"
     if target_type in TargetType.guards():
         pipeline = init(model_dir=model_dir)
         if pipeline:

datarobot_dome/guard_helpers.py

@@ -34,6 +34,7 @@ from ragas.metrics import AgentGoalAccuracyWithoutReference
 from rouge_score import rouge_scorer
 
 from datarobot_dome.constants import AWS_MODEL_TO_AWS_MODEL_VERSION_MAP
+from datarobot_dome.constants import ENABLE_LLM_GATEWAY_INFERENCE_RUNTIME_PARAM_NAME
 from datarobot_dome.constants import GOOGLE_MODEL_TO_GOOGLE_MODEL_VERSION_MAP
 from datarobot_dome.constants import LOGGER_NAME_PREFIX
 from datarobot_dome.constants import PROMPT_TOKEN_COUNT_COLUMN_NAME_FROM_USAGE
@@ -42,6 +43,7 @@ from datarobot_dome.constants import AwsModel
 from datarobot_dome.constants import GoogleModel
 from datarobot_dome.constants import GuardLLMType
 from datarobot_dome.llm import DataRobotLLM
+from datarobot_dome.runtime import get_runtime_parameter_value_bool
 
 # Ideally, we want to return confidence score between 0.0 and 100.0,
 # but for ROUGE-1 guard, UI allows the user to configure value between
@@ -227,17 +229,27 @@ def get_llm_gateway_client(
     return client
 
 
-def use_llm_gateway_inference(llm_type: GuardLLMType):
+def use_llm_gateway_inference(llm_type: str):
     """
-    USE the LLM gateway if its runtime parameter is available and enabled
-    DATAROBOT and NIM LLM types are not supported by the gateway
+    Determine whether the given LLM should use the LLM Gateway for inference.
+
+    `DATAROBOT` and `NIM LLM` types are not supported by the gateway.
+
+    Parameters
+    ----------
+    llm_type
+        The type of the LLM used in the guard.
+
+    Returns
+    -------
+    True if LLM Gateway should be used, False otherwise.
     """
-    if json.loads(os.environ.get("ENABLE_LLM_GATEWAY_INFERENCE", "false")) and llm_type not in [
-        GuardLLMType.DATAROBOT,
-        GuardLLMType.NIM,
-    ]:
-        return True
-    return False
+    is_enabled_by_runtime_parameter = get_runtime_parameter_value_bool(
+        param_name=ENABLE_LLM_GATEWAY_INFERENCE_RUNTIME_PARAM_NAME,
+        default_value=False,
+    )
+    is_compatible_llm = llm_type not in [GuardLLMType.DATAROBOT, GuardLLMType.NIM]
+    return is_enabled_by_runtime_parameter and is_compatible_llm
 
 
 def get_azure_openai_client(

datarobot_dome/pipeline/pipeline.py

@@ -328,12 +328,7 @@ class Pipeline:
         asyncio.run(self.async_upload_custom_metrics(url, payload))
 
     async def async_upload_custom_metrics(self, url, payload):
-        upload_task = self.async_http_client.loop.create_task(
-            self.async_http_client.bulk_upload_custom_metrics(url, payload, self._deployment_id)
-        )
-        self.upload_custom_metrics_tasks.add(upload_task)
-        upload_task.add_done_callback(self.upload_custom_metrics_tasks.discard)
-        await asyncio.sleep(0)
+        await self.async_http_client.bulk_upload_custom_metrics(url, payload, self._deployment_id)
 
     def add_aggregate_metrics_to_payload(self, payload):
         """

datarobot_dome/runtime.py

@@ -0,0 +1,42 @@
+#  ---------------------------------------------------------------------------------
+#  Copyright (c) 2025 DataRobot, Inc. and its affiliates. All rights reserved.
+#  Last updated 2025.
+#
+#  DataRobot, Inc. Confidential.
+#  This is proprietary source code of DataRobot, Inc. and its affiliates.
+#
+#  This file and its contents are subject to DataRobot Tool and Utility Agreement.
+#  For details, see
+#  https://www.datarobot.com/wp-content/uploads/2021/07/DataRobot-Tool-and-Utility-Agreement.pdf.
+#  ---------------------------------------------------------------------------------
+import json
+import os
+
+from datarobot_dome.constants import RUNTIME_PARAMETER_PREFIX
+
+
+def get_runtime_parameter_value_bool(param_name: str, default_value: bool) -> bool:
+    """
+    Retrieve the value of a boolean-typed model runtime parameter with the specified name.
+
+    Parameters
+    ----------
+    param_name
+        The name of the model runtime parameter to retrieve (without the env variable prefix).
+    default_value
+        The default value to return if the model runtime parameter is undefined or underspecified.
+
+    Returns
+    -------
+    The parsed runtime parameter value.
+    """
+    env_var_name = f"{RUNTIME_PARAMETER_PREFIX}{param_name}"
+    param_body = json.loads(os.environ.get(env_var_name, "{}"))
+
+    if not param_body:
+        return default_value
+
+    if "payload" not in param_body:
+        return default_value
+
+    return bool(param_body["payload"])

{datarobot_moderations-11.1.16.dist-info → datarobot_moderations-11.1.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: datarobot-moderations
-Version: 11.1.16
+Version: 11.1.18
 Summary: DataRobot Monitoring and Moderation framework
 License: DataRobot Tool and Utility Agreement
 Author: DataRobot

{datarobot_moderations-11.1.16.dist-info → datarobot_moderations-11.1.18.dist-info}/RECORD

@@ -1,11 +1,11 @@
 datarobot_dome/__init__.py,sha256=B5Rx8_CNCNsOpxBbRj27XOXCfRZmvmrAR-NzlzIKnDw,583
 datarobot_dome/async_http_client.py,sha256=wkB4irwvnchNGzO1bk2C_HWM-GOSB3AUn5TXKl-X0ZI,9649
 datarobot_dome/chat_helper.py,sha256=BzvtUyZSZxzOqq-5a2wQKhHhr2kMlcP1MFrHaDAeD_o,9671
-datarobot_dome/constants.py,sha256=mnSa8rUAha4XlsS2lwPmFCkH2RzfSL_MMkErsWHqIbA,9040
-datarobot_dome/drum_integration.py,sha256=nULpLYVMiS5vihfNUyuq-nvZpgXrQibQbVu2UMAscu8,42102
+datarobot_dome/constants.py,sha256=vM2_JkXbn4dkWARCqxNfLriSo0E05LDXVrwNktptpuc,10416
+datarobot_dome/drum_integration.py,sha256=7gXi29t1Huhio68frMksrkWXfMq08V1Q49MmIRjzgMA,42227
 datarobot_dome/guard.py,sha256=1INYx17n9ToiB5bzI-jIReUUuqkK_ucxpOx4jQLts6g,33264
 datarobot_dome/guard_executor.py,sha256=AOI8MZeZETHMoFgBePe0wa2vE9d2975MYQnEDHLZL7s,35462
-datarobot_dome/guard_helpers.py,sha256=Bfdi8gow2_TAVzRHYUDqEfcG5bWx2KR1dnpxt9E850Y,16311
+datarobot_dome/guard_helpers.py,sha256=7jXxRRpO41tumZFZPumpM78qD-hQ-z5yjMtIA8s9_xM,16762
 datarobot_dome/guards/__init__.py,sha256=B5Rx8_CNCNsOpxBbRj27XOXCfRZmvmrAR-NzlzIKnDw,583
 datarobot_dome/guards/guard_llm_mixin.py,sha256=VovlpNZjWIGamF4SSvLF5lzOFyApH5IoOiB_qtCmRg0,12216
 datarobot_dome/llm.py,sha256=L02OvTrflmD34-FrfXebfF-zzKTeuin7fpne1Cl5psg,5719
@@ -15,9 +15,10 @@ datarobot_dome/metrics/factory.py,sha256=7caa8paI9LuFXDgguXdC4on28V7IwwIsKJT2Z-A
 datarobot_dome/metrics/metric_scorer.py,sha256=mGxW3NNP93LpbpOiX3MeYyd0YEEjTPE8WVYMGS4SWoY,2516
 datarobot_dome/pipeline/__init__.py,sha256=B5Rx8_CNCNsOpxBbRj27XOXCfRZmvmrAR-NzlzIKnDw,583
 datarobot_dome/pipeline/llm_pipeline.py,sha256=fOp_OJnQMDUJH-LKv12kEqli-EqfHjAiSTFqtxzMkhM,19942
-datarobot_dome/pipeline/pipeline.py,sha256=_pZ_4K2LMnfYCYj_ur9EwJzo3T-pbO6lFYz1O-_3uQ4,16491
+datarobot_dome/pipeline/pipeline.py,sha256=YrGR3uS7kY9dd30B97cMESBbVNsze25RSoIcjEMqCb0,16251
 datarobot_dome/pipeline/vdb_pipeline.py,sha256=WTOGn1qe_ZvEcdlvHgeXxl2xTqp7GjfL13c6S-FmAfM,5146
+datarobot_dome/runtime.py,sha256=FD8wXOweqoQVzbZMh-mucL66xT2kGxPsJUGAcJBgwxw,1468
 datarobot_dome/streaming.py,sha256=6nYvh6SoxPRLfO6GGdEoHsQuyLP9oX1lDMe8IeGo4lw,17801
-datarobot_moderations-11.1.16.dist-info/METADATA,sha256=WBOhNlF-pwpbJ2PvlkhwqlKbF3vtWkbNdu6vjqvmvaQ,4827
-datarobot_moderations-11.1.16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-datarobot_moderations-11.1.16.dist-info/RECORD,,
+datarobot_moderations-11.1.18.dist-info/METADATA,sha256=pfqGL7fkmf04cOw9vyCL216SWkIm9XQSCNs1r36OUAI,4827
+datarobot_moderations-11.1.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+datarobot_moderations-11.1.18.dist-info/RECORD,,