o
    ưi                     @   sJ   d dl mZmZ d dlmZ d dlmZ d dlmZ dedefddZ	d	S )
    )AnyDict)Request)verbose_proxy_logger)UserAPIKeyAuthrequestreturnc                    s   ddl m} td |dpi }td|  |s!tdi }| D ],\}}| j|}|rS|dkr>t|||< q'|dkrOd	d
 |	dD ||< q'|||< q'td|  t
di |}td|  |S )z+
    Handle request from oauth2 proxy.
    r   )general_settingszHandling oauth2 proxy requestoauth2_config_mappingszOauth2 config mappings: z4Oauth2 config mappings not found in general_settingsZ
max_budgetmodelsc                 S   s   g | ]}|  qS  )strip).0modelr   r   [/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/auth/oauth2_proxy_hook.py
<listcomp>$   s    z/handle_oauth2_proxy_request.<locals>.<listcomp>,z1Auth data before creating UserAPIKeyAuth object: zUserAPIKeyAuth object created: Nr   )Zlitellm.proxy.proxy_serverr	   r   debugget
ValueErroritemsheadersfloatsplitr   )r   r	   r
   Z	auth_datakeyheadervalueZuser_api_key_authr   r   r   handle_oauth2_proxy_request	   s0   
r   N)
typingr   r   Zfastapir   Zlitellm._loggingr   Zlitellm.proxy._typesr   r   r   r   r   r   <module>   s
    