
    g'                       U d dl mZ d dlZd dlmZ ddlmZ ddlm	Z	m
Z
mZmZmZmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZ dd	lmZ dd
lmZmZ ddlm Z m!Z! ddl"m#Z#m$Z$m%Z% ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6 ddl7m8Z8m9Z9 ddl:m;Z< g dZ=ddl>m?Z@mAZA ddlBmCZC ddlDmEZEmFZF ddlG ddlHmIZImJZJ  e<         eK       ZLe=D ]  ZMeMj                  d      r	 deLeM   _O        ! d dlRZSd dlZTd dlUZVddl7m#Z#m$Z$ daWdeXd<   daYdeXd<   daZdeXd<   dZ[deXd<   e#a\deXd <   e$a]d!eXd"<   da^d#eXd$<   da_d%eXd&<   da`d'eXd(<   eTj                  d)   Zb eSj                  ebej                  j                  d*            afd+eXd,<   ej                  j                  d-      agdeXd.<   ej                  j                  d/      ahdeXd0<   ej                  j                  d1      aideXd2<   dZjd3eXd4<    G d5 d6e      Zk G d7 d8ekeE      Zl G d9 d:e(      ZmdCd;ZndCd<ZodCd=Zpdad>eXd?<   dDd@ZqdEdAZrddBlsmtZtmuZumvZvmwZwmxZxmyZymzZzm{Z{m|Z|m}Z}m~Z~ y# ePeQf$ r Y ~w xY w)F    )annotationsN)override   )types)	NOT_GIVENOmitNoneTypeNotGiven	TransportProxiesTypes)file_from_path)	ClientOpenAIStreamTimeoutr   AsyncClientAsyncOpenAIAsyncStreamRequestOptions)	BaseModel)	__title____version__)APIResponseAsyncAPIResponse)DEFAULT_TIMEOUTDEFAULT_MAX_RETRIESDEFAULT_CONNECTION_LIMITS)APIErrorOpenAIErrorConflictErrorNotFoundErrorAPIStatusErrorRateLimitErrorAPITimeoutErrorBadRequestErrorAPIConnectionErrorAuthenticationErrorInternalServerErrorPermissionDeniedErrorLengthFinishReasonErrorUnprocessableEntityErrorAPIResponseValidationErrorContentFilterFinishReasonError)DefaultHttpxClientDefaultAsyncHttpxClient)setup_logging)(r   r   r   r	   r   r   r
   r   r   r   r   r"   r$   r&   r,   r%   r'   r)   r!   r    r+   r#   r(   r*   r-   r   r   r   r   r   r   r   r   r   r   r   r   r   r.   r/   )azurepydantic_function_tool)VERSION)AzureOpenAIAsyncAzureOpenAI)*)AssistantEventHandlerAsyncAssistantEventHandler__openai)r   r   
str | Noneapi_keyorganizationprojectzstr | _httpx.URL | Nonebase_urlfloat | Timeout | Nonetimeoutintmax_retries_t.Mapping[str, str] | Nonedefault_headers_t.Mapping[str, object] | Nonedefault_queryz_httpx.Client | Nonehttp_client)r:   r1   OPENAI_API_TYPEz_ApiType | Noneapi_typeOPENAI_API_VERSIONapi_versionAZURE_OPENAI_ENDPOINTazure_endpointAZURE_OPENAI_AD_TOKENazure_ad_tokenz"_azure.AzureADTokenProvider | Noneazure_ad_token_providerc                  r    e Zd Zeedd              Zej                  dd       Zeedd              Zej                  dd       Zeedd              Zej                  dd       Zeed fd              Z	e	j                  d fd       Z	eedd	              Z
e
j                  dd
       Z
eedd              Zej                  dd       Zeedd              Zej                  dd       Zeedd              Zej                  dd       Zeed fd              Zej                  d d       Z xZS )!_ModuleClientc                    t         S Nr<   selfs    D/var/www/openai/venv/lib/python3.12/site-packages/openai/__init__.pyr<   z_ModuleClient.api_key   	         c                    |a y rU   rV   rX   values     rY   r<   z_ModuleClient.api_key   	     r[   c                    t         S rU   r=   rW   s    rY   r=   z_ModuleClient.organization   s
     r[   c                    |a y rU   ra   r]   s     rY   r=   z_ModuleClient.organization   s	     r[   c                    t         S rU   r>   rW   s    rY   r>   z_ModuleClient.project   rZ   r[   c                    |a y rU   rd   r]   s     rY   r>   z_ModuleClient.project   r_   r[   c                V    t         t        j                  t               S t        |    S rU   )r?   _httpxURLsuperrX   	__class__s    rY   r?   z_ModuleClient.base_url   s%     ::h''wr[   c                $    |t               _        y rU   )ri   r?   )rX   urlrk   s     rY   r?   z_ModuleClient.base_url   s    r[   c                    t         S rU   rA   rW   s    rY   rA   z_ModuleClient.timeout   rZ   r[   c                    |a y rU   ro   r]   s     rY   rA   z_ModuleClient.timeout   r_   r[   c                    t         S rU   rC   rW   s    rY   rC   z_ModuleClient.max_retries   s
     r[   c                    |a y rU   rr   r]   s     rY   rC   z_ModuleClient.max_retries   	     r[   c                    t         S rU   rE   rW   s    rY   _custom_headersz_ModuleClient._custom_headers   s
     r[   c                    |a y rU   rv   r]   s     rY   rw   z_ModuleClient._custom_headers   s	      r[   c                    t         S rU   rG   rW   s    rY   _custom_queryz_ModuleClient._custom_query   s
     r[   c                    |a y rU   rz   r]   s     rY   r{   z_ModuleClient._custom_query   s	     r[   c                (    t         xs
 t        |   S rU   )rH   ri   _clientrj   s    rY   r~   z_ModuleClient._client   s     -ego-r[   c                    |a y rU   )rH   r]   s     rY   r~   z_ModuleClient._client   rt   r[   )returnr;   )r^   r;   r   None)r   z
_httpx.URL)rm   z_httpx.URL | strr   r   )r   r@   )r^   r@   r   r   )r   rB   )r^   rB   r   r   )r   rD   )r^   rD   r   r   )r   rF   )r^   rF   r   r   )r   _httpx.Client)r^   r   r   r   )__name__
__module____qualname__propertyr   r<   setterr=   r>   r?   rA   rC   rw   r{   r~   __classcell__rk   s   @rY   rS   rS      s       ^^ 
     
    ^^ 
      __     ^^ 
     
       
     
 .  . ^^ r[   rS   c                      e Zd Zy)_AzureModuleClientN)r   r   r    r[   rY   r   r      s    r[   r   c                        e Zd Zd fdZ xZS ) _AmbiguousModuleClientUsageErrorc                $    t         |   d       y )NzAmbiguous use of module client; please set `openai.api_type` or the `OPENAI_API_TYPE` environment variable to `openai` or `azure`)ri   __init__rj   s    rY   r   z)_AmbiguousModuleClientUsageError.__init__   s     P	
r[   r   r   )r   r   r   r   r   r   s   @rY   r   r      s    
 
r[   r   c                 D    t         j                  j                  d      d uS )NOPENAI_API_KEY)_osenvirongetr   r[   rY   _has_openai_credentialsr     s    ;;??+,D88r[   c                 X    t         d uxs! t        j                  j                  d      d uS )NAZURE_OPENAI_API_KEY)rN   r   r   r   r   r[   rY   _has_azure_credentialsr     s&    %\9O)PX\)\\r[   c                 l    t         j                  j                  d      d uxs t        d uxs t        d uS )NrO   )r   r   r   rP   rQ   r   r[   rY   _has_azure_ad_credentialsr     s7    /0< 	/%	/"$.r[   zOpenAI | Noner~   c                    t         vt        t        j                  j	                  d      at
        t        j                  j	                  d      at        t        j                  j	                  d      at        mt               } t               }t               }| r|s|r
t               t
        t        *t        j                  j	                  d      	 
t               |s|rdandat        dk(  rMt        t        t        t        t
        t        t        t         t"        t$        t&        t(        t*              a t         S t-        t        t        t.        t         t"        t$        t&        t(        t*        	      a t         S t         S )	NrM   rO   rK   r   r1   r:   )rL   rN   r<   rP   rQ   r=   r?   rA   rC   rE   rG   rH   )	r<   r=   r>   r?   rA   rC   rE   rG   rH   )r~   rN   r   r   r   rP   rL   rJ   r   r   r   r   rQ   r   r<   r=   r?   rA   rC   rE   rG   rH   rS   r>   )
has_openai	has_azurehas_azure_ads      rY   _load_clientr     s9     ! [[__-DEN! [[__-DEN++//*>?K02J.0I46LyL688*.E.QWZWbWbWfWf&XX 788L"#w('--(?)!' /+'G N%#+'#

 Nr[   c                     d a y rU   )r~   r   r[   rY   _reset_clientr   X  s	     Gr[   )betachataudiofilesimagesmodelsbatches
embeddingscompletionsfine_tuningmoderations)r   bool)r   r   r   )
__future__r   osr   typing_extensionsr    r   _typesr   r   r	   r
   r   r   _utilsr   r~   r   r   r   r   r   r   r   r   _modelsr   _versionr   r   	_responser   r   
_constantsr   r   r   _exceptionsr   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   _base_clientr.   r/   _utils._logsr0   _setup_logging__all__libr1   _azurer2   versionr3   	lib.azurer4   r5   lib._old_apilib.streamingr7   r8   locals__locals__name
startswithr   	TypeErrorAttributeErrortyping_t_tehttpxrg   r<   __annotations__r=   r>   r?   rA   rC   rE   rG   rH   Literal_ApiTypecastr   r   rJ   rL   rN   rP   rQ   rS   r   r   r   r   r   r   r   _module_clientr   r   r   r   r   r   r   r   r   r   r   r   r[   rY   <module>r      s    #  &  P P " v v v  , W W W    $ F 9)V S ' W 
   8FT"	*2HV'     > j  $(
! ("1	 1&S &/3, 304- 4$(! (;;()#BGGHckkoo>O.PQ/ Q++//*>?Z ? [[__-DE
 E [[__-DE
 E>B ; BfF fR 
{ 
9]  ?D   u >* 		s   #
III