o
    ưi                     @   sL   d Z ddlmZmZ ddlmZ ddlmZ ddlmZ G dd deZ	dS )	z
Base class for Additional Logging Utils for CustomLoggers 

- Health Check for the logging util
- Get Request / Response Payload for the logging util
    )ABCabstractmethod)datetime)Optional)IntegrationHealthCheckStatusc                
       sX   e Zd Z fddZedefddZededee	 dee	 dee
 fd	d
Z  ZS )AdditionalLoggingUtilsc                    s   t    d S )N)super__init__self	__class__ d/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/additional_logging_utils.pyr	      s   zAdditionalLoggingUtils.__init__returnc                       dS )z1
        Check if the service is healthy
        Nr   r
   r   r   r   async_health_check   s   z)AdditionalLoggingUtils.async_health_check
request_idstart_time_utcend_time_utcc                    r   )zO
        Get the request and response payload for a given `request_id`
        Nr   )r   r   r   r   r   r   r   get_request_response_payload   s   
z3AdditionalLoggingUtils.get_request_response_payload)__name__
__module____qualname__r	   r   r   r   strr   r   dictr   __classcell__r   r   r   r   r      s    r   N)
__doc__abcr   r   r   typingr   Z,litellm.types.integrations.base_health_checkr   r   r   r   r   r   <module>   s    