o
    ưi                     @   sD   U d dl Z d dlmZ d dlmZ daee ed< G dd dZdS )    N)Optional)CustomAuthSettingscustom_auth_settingsc                   @   sF   e Zd ZdedefddZdeddfddZedee	 fdd	Z
dS )
EnterpriseProxyConfiggeneral_settingsreturnc                    s*   | dd }|d urt| dd}|S )Nr   mode)r   )getr   )selfr   r    r   \/home/app/Keep/.python/lib/python3.10/site-packages/litellm_enterprise/proxy/proxy_server.pyload_custom_auth_settings
   s   z/EnterpriseProxyConfig.load_custom_auth_settingsNc                    s   |  |I d H ad S )N)r   r   )r
   r   r   r   r   load_enterprise_config   s   z,EnterpriseProxyConfig.load_enterprise_configc                  C   s"   ddl m}  d }| rtd}|S )Nr   )premium_userZDOCS_DESCRIPTION)Zlitellm.proxy.proxy_serverr   osgetenv)r   Zdocs_descriptionr   r   r   get_custom_docs_description   s
   
z1EnterpriseProxyConfig.get_custom_docs_description)__name__
__module____qualname__dictr   r   r   staticmethodr   strr   r   r   r   r   r   	   s    

r   )r   typingr   Z+litellm_enterprise.types.proxy.proxy_serverr   r   __annotations__r   r   r   r   r   <module>   s
   
 