o
    ưif                     @   s>   d dl m Z mZ d dlZd dlmZ dd ZdefddZdS )	    )datetimetimezoneN) get_next_standardized_reset_timec                   C   s   t tddpdS )z
    Get the budget reset timezone from litellm_settings.
    Falls back to UTC if not specified.

    litellm_settings values are set as attributes on the litellm module
    by proxy_server.py at startup (via setattr(litellm, key, value)).
    r   NUTC)getattrlitellm r   r   `/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/common_utils/timezone_utils.pyget_budget_reset_timezone   s   r
   budget_durationc                 C   s   t | ttjt d}|S )zm
    Get the budget reset time based on the configured timezone.
    Falls back to UTC if not specified.
    )durationcurrent_timeZtimezone_str)r   r   nowr   utcr
   )r   Zreset_atr   r   r	   get_budget_reset_time   s   
r   )r   r   r   Z*litellm.litellm_core_utils.duration_parserr   r
   strr   r   r   r   r	   <module>   s
    