datarobot-moderations 11.1.17__py3-none-any.whl → 11.1.19__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -243,8 +243,8 @@ class AwsModel:
243
243
  ANTHROPIC_CLAUDE_3_HAIKU = "anthropic-claude-3-haiku"
244
244
  ANTHROPIC_CLAUDE_3_SONNET = "anthropic-claude-3-sonnet"
245
245
  ANTHROPIC_CLAUDE_3_OPUS = "anthropic-claude-3-opus"
246
- ANTHROPIC_CLAUDE_3_5_SONNET_V1 = "anthropic-claude-3-sonnet-v1"
247
- ANTHROPIC_CLAUDE_3_5_SONNET_V2 = "anthropic-claude-3-sonnet-v2"
246
+ ANTHROPIC_CLAUDE_3_5_SONNET_V1 = "anthropic-claude-3.5-sonnet-v1"
247
+ ANTHROPIC_CLAUDE_3_5_SONNET_V2 = "anthropic-claude-3.5-sonnet-v2"
248
248
  AMAZON_NOVA_LITE = "amazon-nova-lite"
249
249
  AMAZON_NOVA_MICRO = "amazon-nova-micro"
250
250
  AMAZON_NOVA_PRO = "amazon-nova-pro"
@@ -209,7 +209,7 @@ def get_llm_gateway_client(
209
209
  datarobot_endpoint, datarobot_api_token = get_datarobot_endpoint_and_token()
210
210
  client = ChatOpenAI(
211
211
  # default model is required by ChatOpenAI
212
- model=model or "azure/gpt-4o",
212
+ model=model or "azure/gpt-4o-2024-11-20",
213
213
  api_key=datarobot_api_token,
214
214
  base_url=f"{datarobot_endpoint}/genai/llmgw",
215
215
  # retries are handled by the LLM Gateway
@@ -328,12 +328,7 @@ class Pipeline:
328
328
  asyncio.run(self.async_upload_custom_metrics(url, payload))
329
329
 
330
330
  async def async_upload_custom_metrics(self, url, payload):
331
- upload_task = self.async_http_client.loop.create_task(
332
- self.async_http_client.bulk_upload_custom_metrics(url, payload, self._deployment_id)
333
- )
334
- self.upload_custom_metrics_tasks.add(upload_task)
335
- upload_task.add_done_callback(self.upload_custom_metrics_tasks.discard)
336
- await asyncio.sleep(0)
331
+ await self.async_http_client.bulk_upload_custom_metrics(url, payload, self._deployment_id)
337
332
 
338
333
  def add_aggregate_metrics_to_payload(self, payload):
339
334
  """
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.3
2
2
  Name: datarobot-moderations
3
- Version: 11.1.17
3
+ Version: 11.1.19
4
4
  Summary: DataRobot Monitoring and Moderation framework
5
5
  License: DataRobot Tool and Utility Agreement
6
6
  Author: DataRobot
@@ -1,11 +1,11 @@
1
1
  datarobot_dome/__init__.py,sha256=B5Rx8_CNCNsOpxBbRj27XOXCfRZmvmrAR-NzlzIKnDw,583
2
2
  datarobot_dome/async_http_client.py,sha256=wkB4irwvnchNGzO1bk2C_HWM-GOSB3AUn5TXKl-X0ZI,9649
3
3
  datarobot_dome/chat_helper.py,sha256=BzvtUyZSZxzOqq-5a2wQKhHhr2kMlcP1MFrHaDAeD_o,9671
4
- datarobot_dome/constants.py,sha256=SfZTkfW3rc7oeG2zmYiYxIkUT5dwkay1HXCusZod_-Q,10412
4
+ datarobot_dome/constants.py,sha256=vM2_JkXbn4dkWARCqxNfLriSo0E05LDXVrwNktptpuc,10416
5
5
  datarobot_dome/drum_integration.py,sha256=7gXi29t1Huhio68frMksrkWXfMq08V1Q49MmIRjzgMA,42227
6
6
  datarobot_dome/guard.py,sha256=1INYx17n9ToiB5bzI-jIReUUuqkK_ucxpOx4jQLts6g,33264
7
7
  datarobot_dome/guard_executor.py,sha256=AOI8MZeZETHMoFgBePe0wa2vE9d2975MYQnEDHLZL7s,35462
8
- datarobot_dome/guard_helpers.py,sha256=7jXxRRpO41tumZFZPumpM78qD-hQ-z5yjMtIA8s9_xM,16762
8
+ datarobot_dome/guard_helpers.py,sha256=ajxm-w7MS7eN5DMMO-jbbzjcOYMZ-LvhO53n2NI5_Fk,16773
9
9
  datarobot_dome/guards/__init__.py,sha256=B5Rx8_CNCNsOpxBbRj27XOXCfRZmvmrAR-NzlzIKnDw,583
10
10
  datarobot_dome/guards/guard_llm_mixin.py,sha256=VovlpNZjWIGamF4SSvLF5lzOFyApH5IoOiB_qtCmRg0,12216
11
11
  datarobot_dome/llm.py,sha256=L02OvTrflmD34-FrfXebfF-zzKTeuin7fpne1Cl5psg,5719
@@ -15,10 +15,10 @@ datarobot_dome/metrics/factory.py,sha256=7caa8paI9LuFXDgguXdC4on28V7IwwIsKJT2Z-A
15
15
  datarobot_dome/metrics/metric_scorer.py,sha256=mGxW3NNP93LpbpOiX3MeYyd0YEEjTPE8WVYMGS4SWoY,2516
16
16
  datarobot_dome/pipeline/__init__.py,sha256=B5Rx8_CNCNsOpxBbRj27XOXCfRZmvmrAR-NzlzIKnDw,583
17
17
  datarobot_dome/pipeline/llm_pipeline.py,sha256=fOp_OJnQMDUJH-LKv12kEqli-EqfHjAiSTFqtxzMkhM,19942
18
- datarobot_dome/pipeline/pipeline.py,sha256=_pZ_4K2LMnfYCYj_ur9EwJzo3T-pbO6lFYz1O-_3uQ4,16491
18
+ datarobot_dome/pipeline/pipeline.py,sha256=YrGR3uS7kY9dd30B97cMESBbVNsze25RSoIcjEMqCb0,16251
19
19
  datarobot_dome/pipeline/vdb_pipeline.py,sha256=WTOGn1qe_ZvEcdlvHgeXxl2xTqp7GjfL13c6S-FmAfM,5146
20
20
  datarobot_dome/runtime.py,sha256=FD8wXOweqoQVzbZMh-mucL66xT2kGxPsJUGAcJBgwxw,1468
21
21
  datarobot_dome/streaming.py,sha256=6nYvh6SoxPRLfO6GGdEoHsQuyLP9oX1lDMe8IeGo4lw,17801
22
- datarobot_moderations-11.1.17.dist-info/METADATA,sha256=EZg5IXeC1CXbfEsc4gl2cXd2suSIjj0svvBgGVi4fGQ,4827
23
- datarobot_moderations-11.1.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
24
- datarobot_moderations-11.1.17.dist-info/RECORD,,
22
+ datarobot_moderations-11.1.19.dist-info/METADATA,sha256=a_GR9Rja3SiUi074YFg0dGWv0szX3zF4zR9B7u6DkNE,4827
23
+ datarobot_moderations-11.1.19.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
24
+ datarobot_moderations-11.1.19.dist-info/RECORD,,