o
    "ch:+                  
   @  s  U d dl mZ d dl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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9 ddl:m;Z;m<Z<m=Z= ddl>m?Z@ ddlAmBZB g dZCejDsddlEmFZF ddlGmHZImJZJ ddlKmLZL ddlMmNZNmOZO ddlPT ddlQmRZRmSZS e@  eT ZUeCD ]ZVeVWdszdeUeV _XW q eYeZfy   Y qw qd dlZd dlZ[d dl\Z]ddl:m%Z%m&Z& da^de_d< da`de_d< daade_d< dabde_d< dZcd e_d!< e%add"e_d#< e&aed$e_d%< dafd&e_d'< dagd(e_d)< dahd*e_d+< e[jid, Zjekejejlmd-and.e_d/< ejlmd0aode_d1< ejlmd2apde_d3< ejlmd4aqde_d5< dZrd6e_d7< G d8d9 d9eZsG d:d; d;eseNZtG d<d= d=e*ZudOd@dAZvdOdBdCZwdOdDdEZxdadFe_dG< dPdIdJZydQdLdMZzddNl{m|Z|m}Z}m~Z~mZmZmZmZmZmZmZmZmZmZmZmZmZmZ dS )R    )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InvalidWebhookSignatureErrorContentFilterFinishReasonError)DefaultHttpxClientDefaultAioHttpClientDefaultAsyncHttpxClient)setup_logging)HttpxBinaryResponseContent)*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.   r0   r/   )	resources)azurepydantic_function_tool)VERSION)AzureOpenAIAsyncAzureOpenAI)*)AssistantEventHandlerAsyncAssistantEventHandler__openai)r   r   
str | Noneapi_keyorganizationprojectwebhook_secretz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=   r4   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                      sp  e Zd Zeed+ddZejd,ddZeed+dd	Zejd,d
d	Zeed+ddZejd,ddZeed+ddZ	e	jd,ddZ	eed- fddZ
e
jd. fddZ
eed/ddZejd0ddZeed1ddZejd2ddZeed3d d!Zejd4d"d!Zeed5d$d%Zejd6d&d%Zeed7 fd(d)Zejd8d*d)Z  ZS )9_ModuleClientreturnr>   c                 C     t S Nr?   self r]   M/home/air/goalskill/back/venv/lib/python3.10/site-packages/openai/__init__.pyr?         z_ModuleClient.api_keyvalueNonec                 C     |a d S rY   rZ   r\   r`   r]   r]   r^   r?         c                 C  rX   rY   r@   r[   r]   r]   r^   r@      r_   z_ModuleClient.organizationc                 C  rb   rY   re   rc   r]   r]   r^   r@      rd   c                 C  rX   rY   rA   r[   r]   r]   r^   rA      r_   z_ModuleClient.projectc                 C  rb   rY   rf   rc   r]   r]   r^   rA      rd   c                 C  rX   rY   rB   r[   r]   r]   r^   rB      r_   z_ModuleClient.webhook_secretc                 C  rb   rY   rg   rc   r]   r]   r^   rB      rd   
_httpx.URLc                   s   t d ur	tt S t j S rY   )rC   _httpxURLsuperr[   	__class__r]   r^   rC      s   
z_ModuleClient.base_urlurl_httpx.URL | strc                   s   |t  _d S rY   )rk   rC   )r\   rn   rl   r]   r^   rC      s   rD   c                 C  rX   rY   rE   r[   r]   r]   r^   rE      r_   z_ModuleClient.timeoutc                 C  rb   rY   rp   rc   r]   r]   r^   rE      rd   rF   c                 C  rX   rY   rG   r[   r]   r]   r^   rG      r_   z_ModuleClient.max_retriesc                 C  rb   rY   rq   rc   r]   r]   r^   rG      rd   rH   c                 C  rX   rY   rI   r[   r]   r]   r^   _custom_headers   r_   z_ModuleClient._custom_headersc                 C  rb   rY   rr   rc   r]   r]   r^   rs      rd   rJ   c                 C  rX   rY   rK   r[   r]   r]   r^   _custom_query   r_   z_ModuleClient._custom_queryc                 C  rb   rY   rt   rc   r]   r]   r^   ru      rd   _httpx.Clientc                   s   t pt jS rY   )rL   rk   _clientr[   rl   r]   r^   rw     s   z_ModuleClient._clientc                 C  rb   rY   )rL   rc   r]   r]   r^   rw     rd   )rW   r>   )r`   r>   rW   ra   )rW   rh   )rn   ro   rW   ra   )rW   rD   )r`   rD   rW   ra   )rW   rF   )r`   rF   rW   ra   )rW   rH   )r`   rH   rW   ra   )rW   rJ   )r`   rJ   rW   ra   )rW   rv   )r`   rv   rW   ra   )__name__
__module____qualname__propertyr   r?   setterr@   rA   rB   rC   rE   rG   rs   ru   rw   __classcell__r]   r]   rl   r^   rV      sf    rV   c                   @  s   e Zd ZdS )_AzureModuleClientN)rx   ry   rz   r]   r]   r]   r^   r~     s    r~   c                      s   e Zd Zd fddZ  ZS ) _AmbiguousModuleClientUsageErrorrW   ra   c                   s   t  d d S )NzAmbiguous use of module client; please set `openai.api_type` or the `OPENAI_API_TYPE` environment variable to `openai` or `azure`)rk   __init__r[   rl   r]   r^   r     s   z)_AmbiguousModuleClientUsageError.__init__rW   ra   )rx   ry   rz   r   r}   r]   r]   rl   r^   r     s    r   rW   boolc                   C  s   t jdd uS )NOPENAI_API_KEY)_osenvirongetr]   r]   r]   r^   _has_openai_credentials  s   r   c                   C  s   t d uptjdd uS )NAZURE_OPENAI_API_KEY)rR   r   r   r   r]   r]   r]   r^   _has_azure_credentials  s   r   c                   C  s    t jdd uptd uptd uS )NrS   )r   r   r   rT   rU   r]   r]   r]   r^   _has_azure_ad_credentials   s
   r   zOpenAI | Nonerw   r   c                  C  s   t d u rztd u rtjdatd u rtjdatd u r"tjdatd u rTt } t	 }t
 }| r8|s5|r8t td us@td urKtjdd urKt |sO|rRdandatdkrjtttttttttttttda t S tttttttttttd
a t S t S )	NrQ   rS   rO   r   r4   r=   )rP   rR   r?   rT   rU   r@   rC   rE   rG   rI   rK   rL   )
r?   r@   rA   rB   rC   rE   rG   rI   rK   rL   )rw   rR   r   r   r   rT   rP   rN   r   r   r   r   rU   r~   r?   r@   rC   rE   rG   rI   rK   rL   rV   rA   rB   )
has_openai	has_azurehas_azure_adr]   r]   r^   _load_client+  sh   r   ra   c                   C  s   d a d S rY   )rw   r]   r]   r]   r^   _reset_clientn  s   r   )betachataudioevalsfilesimagesmodelsbatchesuploadswebhooks	responses
containers
embeddingscompletionsfine_tuningmoderationsvector_stores)rW   r   )rW   r   r   )
__future__r   osr   typing_ttyping_extensionsr    r   _typesr   r   r   r	   r
   r   _utilsr   rw   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,   r-   _base_clientr.   r/   r0   _utils._logsr1   _setup_logging_legacy_responser2   __all__TYPE_CHECKING_utils._resources_proxyr3   libr4   _azurer5   versionr6   	lib.azurer7   r8   lib._old_apilib.streamingr:   r;   locals__locals__name
startswithry   	TypeErrorAttributeError_tehttpxri   r?   __annotations__r@   rA   rB   rC   rE   rG   rI   rK   rL   Literal_ApiTypecastr   r   rN   rP   rR   rT   rU   rV   r~   r   r   r   r   r   r   _module_clientr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r]   r]   r]   r^   <module>   s    ,L-

t




CP