litellm-enterprise 0.1.16__tar.gz → 0.1.17__tar.gz

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: litellm-enterprise
-Version: 0.1.16
+Version: 0.1.17
 Summary: Package for LiteLLM Enterprise features
 Author: BerriAI
 Requires-Python: >=3.8, !=2.7.*, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*, !=3.6.*, !=3.7.*
@@ -9,6 +9,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Project-URL: Documentation, https://docs.litellm.ai
 Project-URL: Homepage, https://litellm.ai
 Project-URL: Repository, https://github.com/BerriAI/litellm

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/litellm_enterprise/enterprise_callbacks/llama_guard.py

@@ -105,6 +105,7 @@ class _ENTERPRISE_LlamaGuard(CustomLogger):
             "moderation",
             "audio_transcription",
             "responses",
+            "mcp_call",
         ],
     ):
         """

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/litellm_enterprise/enterprise_callbacks/llm_guard.py

@@ -127,6 +127,7 @@ class _ENTERPRISE_LLMGuard(CustomLogger):
             "moderation",
             "audio_transcription",
             "responses",
+            "mcp_call",
         ],
     ):
         """

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/litellm_enterprise/enterprise_callbacks/pagerduty/pagerduty.py

@@ -147,6 +147,7 @@ class PagerDutyAlerting(SlackAlerting):
             "audio_transcription",
             "pass_through_endpoint",
             "rerank",
+            "mcp_call",
         ],
     ) -> Optional[Union[Exception, str, dict]]:
         """

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/litellm_enterprise/integrations/prometheus.py

@@ -122,19 +122,19 @@ class PrometheusLogger(CustomLogger):
 
             # Counter for total_output_tokens
             self.litellm_tokens_metric = self._counter_factory(
-                "litellm_total_tokens",
+                "litellm_total_tokens_metric",
                 "Total number of input + output tokens from LLM requests",
                 labelnames=self.get_labels_for_metric("litellm_total_tokens_metric"),
             )
 
             self.litellm_input_tokens_metric = self._counter_factory(
-                "litellm_input_tokens",
+                "litellm_input_tokens_metric",
                 "Total number of input tokens from LLM requests",
                 labelnames=self.get_labels_for_metric("litellm_input_tokens_metric"),
             )
 
             self.litellm_output_tokens_metric = self._counter_factory(
-                "litellm_output_tokens",
+                "litellm_output_tokens_metric",
                 "Total number of output tokens from LLM requests",
                 labelnames=self.get_labels_for_metric("litellm_output_tokens_metric"),
             )

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/litellm_enterprise/proxy/auth/user_api_key_auth.py

@@ -3,7 +3,7 @@ from typing import Any, Optional
 from fastapi import Request
 
 from litellm._logging import verbose_proxy_logger
-from litellm.proxy._types import UserAPIKeyAuth
+from litellm.proxy._types import ProxyException, UserAPIKeyAuth
 
 
 async def enterprise_custom_auth(
@@ -24,6 +24,8 @@ async def enterprise_custom_auth(
     elif custom_auth_settings["mode"] == "auto":
         try:
             return await user_custom_auth(request, api_key)
+        except ProxyException as e:
+            raise e
         except Exception as e:
             verbose_proxy_logger.debug(
                 f"Error in custom auth, checking litellm auth: {e}"

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/litellm_enterprise/proxy/hooks/managed_files.py

@@ -290,6 +290,7 @@ class _PROXY_LiteLLMManagedFiles(CustomLogger, BaseFileEndpoints):
             "aretrieve_fine_tuning_job",
             "alist_fine_tuning_jobs",
             "acancel_fine_tuning_job",
+            "mcp_call",
         ],
     ) -> Union[Exception, str, Dict, None]:
         """

{litellm_enterprise-0.1.16 → litellm_enterprise-0.1.17}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "litellm-enterprise"
-version = "0.1.16"
+version = "0.1.17"
 description = "Package for LiteLLM Enterprise features"
 authors = ["BerriAI"]
 readme = "README.md"
@@ -22,7 +22,7 @@ requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.commitizen]
-version = "0.1.16"
+version = "0.1.17"
 version_files = [
     "pyproject.toml:version",
     "../requirements.txt:litellm-enterprise==",