phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

re use duration_in_seconds

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-23 12:44:28 -08:00
parent 653d16e158
commit 2b9ff03cd0
1 changed files with 2 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/router_strategy/provider_budgets.py
View file

@ -25,6 +25,7 @@ from litellm._logging import verbose_router_logger
from litellm.caching.caching import DualCache from litellm.caching.caching import DualCache
from litellm.integrations.custom_logger import CustomLogger from litellm.integrations.custom_logger import CustomLogger
from litellm.litellm_core_utils.core_helpers import _get_parent_otel_span_from_kwargs from litellm.litellm_core_utils.core_helpers import _get_parent_otel_span_from_kwargs
from litellm.litellm_core_utils.duration_parser import duration_in_seconds
from litellm.router_utils.cooldown_callbacks import ( from litellm.router_utils.cooldown_callbacks import (
_get_prometheus_logger_from_callbacks, _get_prometheus_logger_from_callbacks,
) )
@ -207,7 +208,7 @@ class ProviderBudgetLimiting(CustomLogger):
) )
spend_key = f"provider_spend:{custom_llm_provider}:{budget_config.time_period}" spend_key = f"provider_spend:{custom_llm_provider}:{budget_config.time_period}"
ttl_seconds = self.get_ttl_seconds(budget_config.time_period) ttl_seconds = duration_in_seconds(duration=budget_config.time_period)
verbose_router_logger.debug( verbose_router_logger.debug(
f"Incrementing spend for {spend_key} by {response_cost}, ttl: {ttl_seconds}" f"Incrementing spend for {spend_key} by {response_cost}, ttl: {ttl_seconds}"
) )
@ -242,15 +243,6 @@ class ProviderBudgetLimiting(CustomLogger):
return None return None
return custom_llm_provider return custom_llm_provider
def get_ttl_seconds(self, time_period: str) -> int:
"""
Convert time period (e.g., '1d', '30d') to seconds for Redis TTL
"""
if time_period.endswith("d"):
days = int(time_period[:-1])
return days * 24 * 60 * 60
raise ValueError(f"Unsupported time period format: {time_period}")
def _track_provider_remaining_budget_prometheus( def _track_provider_remaining_budget_prometheus(
self, provider: str, spend: float, budget_limit: float self, provider: str, spend: float, budget_limit: float
): ):