
    NgW.                       d Z ddlmZ ddlZddlZddl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 ddlmZmZ ddlmZmZ dd	lmZ dd
lmZ  ej        e          Z eddd           G d de                      ZdS )zAzure OpenAI chat wrapper.    )annotationsN)Any	AwaitableCallableDictListUnion)
deprecated)
ChatResult)get_from_dict_or_envpre_init)	BaseModelField)
ChatOpenAI)is_openai_v1z0.0.10z1.0z langchain_openai.AzureChatOpenAI)sinceremovalalternative_importc                      e Zd ZU dZdZded<   	  edd          Zded<   	  edd	          Zd
ed<   	  edd          Z	ded<   	 dZ
ded<   	 dZded<   	 dZded<   	 dZd
ed<   	 dZd
ed<   	 dZded<   	 ed(d            Zed)d            Zed* fd            Zed*d             Zed* fd!            Zed+d"            Zed*d#            Zd, fd'Z xZS )-AzureChatOpenAIa  `Azure OpenAI` Chat Completion API.

    To use this class you
    must have a deployed model on Azure OpenAI. Use `deployment_name` in the
    constructor to refer to the "Model deployment name" in the Azure portal.

    In addition, you should have the ``openai`` python package installed, and the
    following environment variables set or passed in constructor in lower case:
    - ``AZURE_OPENAI_API_KEY``
    - ``AZURE_OPENAI_ENDPOINT``
    - ``AZURE_OPENAI_AD_TOKEN``
    - ``OPENAI_API_VERSION``
    - ``OPENAI_PROXY``

    For example, if you have `gpt-35-turbo` deployed, with the deployment name
    `35-turbo-dev`, the constructor should look like:

    .. code-block:: python

        AzureChatOpenAI(
            azure_deployment="35-turbo-dev",
            openai_api_version="2023-05-15",
        )

    Be aware the API version may change.

    You can also specify the version of the model using ``model_version`` constructor
    parameter, as Azure OpenAI doesn't return model version with the response.

    Default is empty. When you specify the version, it will be appended to the
    model name in the response. Setting correct version will help you to calculate the
    cost properly. Model version is not validated, so make sure you set it correctly
    to get the correct cost.

    Any parameters that are valid to be passed to the openai.create call can be passed
    in, even if not explicitly saved on this class.
    NzUnion[str, None]azure_endpointazure_deployment)defaultaliasdeployment_name api_versionstropenai_api_versionapi_keyopenai_api_keyazure_ad_tokenzUnion[Callable[[], str], None]azure_ad_token_providerz)Union[Callable[[], Awaitable[str]], None]azure_ad_async_token_providermodel_versionopenai_api_typeTboolvalidate_base_urlreturn	List[str]c                
    g dS )z*Get the namespace of the langchain object.)	langchainchat_modelsazure_openai )clss    h/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chat_models/azure_openai.pyget_lc_namespacez AzureChatOpenAI.get_lc_namespacen   s     <;;;    valuesr   c                   |d         dk     rt          d          |d         dk    r|d         rt          d          |d         p't          j        d          pt          j        d          |d<   |d	         pt          j        d
          |d	<   |d         pt          j        d          |d<   |d         p't          j        d          pt          j        d          |d<   |d         pt          j        d          |d<   |d         pt          j        d          |d<   t          |ddd          |d<   t          |ddd          |d<   	 ddl}n# t
          $ r t          d          w xY wt                      re|d	         }|r|d         rd|vrB|d	                             d           dz   |d	<   t          j	        d!| d"|d	          d#           |d$         rct          j	        d%           |d$         |d	         vr:t          j	        d&| d"|d	          d#           |d	xx         d'|d$         z   z  cc<   d|d$<   |d         |d         |d$         |d         |d         |d(         |d         |d	         |d)         |d*         |d+         |d,         |d-         d.} |j
        d2i |j        j        |d/<   |d0         }|r||d(<    |j        d2i |j        j        |d1<   n
|j        |d/<   |S )3z?Validate that api key and python package exists in environment.n   zn must be at least 1.	streamingzn must be 1 when streaming.r!   AZURE_OPENAI_API_KEYOPENAI_API_KEYopenai_api_baseOPENAI_API_BASEr   OPENAI_API_VERSIONopenai_organizationOPENAI_ORG_IDOPENAI_ORGANIZATIONr   AZURE_OPENAI_ENDPOINTr"   AZURE_OPENAI_AD_TOKENr&   OPENAI_API_TYPEazure)r   openai_proxyOPENAI_PROXYr   r   NzTCould not import openai python package. Please install it with `pip install openai`.r(   z/openai/zAs of openai>=1.0.0, Azure endpoints should be specified via the `azure_endpoint` param not `openai_api_base` (or alias `base_url`). Updating `openai_api_base` from z to .r   zAs of openai>=1.0.0, if `deployment_name` (or alias `azure_deployment`) is specified then `openai_api_base` (or alias `base_url`) should not be. Instead use `deployment_name` (or alias `azure_deployment`) and `azure_endpoint`.zAs of openai>=1.0.0, if `openai_api_base` (or alias `base_url`) is specified it is expected to be of the form https://example-resource.azure.openai.com/openai/deployments/example-deployment. Updating z/deployments/r#   request_timeoutmax_retriesdefault_headersdefault_queryhttp_client)r   r   r   r    r"   r#   organizationbase_urltimeoutrJ   rK   rL   rM   clientr$   async_clientr/   )
ValueErrorosgetenvr   openaiImportErrorr   rstripwarningswarnAzureOpenAIchatcompletionsAsyncAzureOpenAIChatCompletion)r0   r4   rV   r;   client_paramsr$   s         r1   validate_environmentz$AzureChatOpenAI.validate_environments   sg    #;??4555#;??vk2?:;;; #$ +y/00+y)** 	 
 %++<$= %
B
 B
 ! (..B'C (
ry H
 H
#$
 () 0y))0y.// 	$%
 $**:#; $
ry#@
 @
  $**:#; $
ry#@
 @
  %9%'8'%
 %
 %
 ! "6NNB"
 "
 "
~	MMMM 	 	 	?  	
 >> >	5 %%67O 56*=#> 5O330188==	I ,- MM +M M 177H0IM M M   +, 5M0   /0?P8QQQ < )8	< <
  &&78< < <   0111+f5F.GG111 15F,-%&:;"()9":$*+<$=!"23"()9":+12K+L &'< ="#45!"34%m4#)*;#<!'!8%m4 M  2v1BBMBBGSF8,23R,S), Y;X78%<V%< & && &; >""  &4F8s   E E-Dict[str, Any]c                    t                      rt                      j        S i t                      j        d| j        iS )z2Get the default parameters for calling OpenAI API.engine)r   super_default_paramsr   self	__class__s    r1   rf   zAzureChatOpenAI._default_params   sI     >> 	77**'')$.  r3   c                    i | j         S )zGet the identifying parameters.)rf   rh   s    r1   _identifying_paramsz#AzureChatOpenAI._identifying_params   s     ($&''r3   c                    t                      rt                      j        S i t                      j        | j        | j        dS )z1Get the config params used for the openai client.)api_typer   )r   re   _client_paramsr&   r   rg   s    r1   ro   zAzureChatOpenAI._client_params   sR     >> 	77))''( 0#6   r3   c                    dS )Nzazure-openai-chatr/   rk   s    r1   	_llm_typezAzureChatOpenAI._llm_type  s    ""r3   c                     | j         | j        dS )Nr&   r   rs   rk   s    r1   lc_attributeszAzureChatOpenAI.lc_attributes  s      $3"&"9
 
 	
r3   responseUnion[dict, BaseModel]r   c                   t          |t                    s|                                }|d         D ]+}|                    dd           dk    rt          d          ,t	                                          |          }d|v rF|d         }| j        r| d| j         }|j        $t          |j        t                    r
||j        d<   |S )Nchoicesfinish_reasoncontent_filterzKAzure has not provided the response due to a content filter being triggeredmodel-
model_name)
isinstancedictgetrS   re   _create_chat_resultr%   
llm_output)rh   ru   reschat_resultr{   ri   s        r1   r   z#AzureChatOpenAI._create_chat_result  s    (D)) 	'}}HI& 	 	Cww--1AAA &   B
 gg11(;;hW%E! 8 774#577%1j&7 71 8=&|4r3   )r)   r*   )r4   r   r)   r   )r)   rb   )r)   r   )ru   rv   r)   r   )__name__
__module____qualname____doc__r   __annotations__r   r   r   r!   r"   r#   r$   r%   r&   r(   classmethodr2   r   ra   propertyrf   rl   ro   rq   rt   r   __classcell__)ri   s   @r1   r   r      sX        $ $L (,N++++ ).dBT(U(U(UOUUUU
 $eBmDDDDDDDS',uT'K'K'KNKKKKU'+N++++ ?CBBBB
 PT!SSSS M+O+""""" < < < [< p p p Xpd      X ( ( ( X( 	 	 	 	 	 X	 # # # X# 
 
 
 X
         r3   r   )r   
__future__r   loggingrT   rY   typingr   r   r   r   r   r	   langchain_core._api.deprecationr
   langchain_core.outputsr   langchain_core.utilsr   r   pydanticr   r   &langchain_community.chat_models.openair    langchain_community.utils.openair   	getLoggerr   loggerr   r/   r3   r1   <module>r      sY       " " " " " "  				  > > > > > > > > > > > > > > > > 6 6 6 6 6 6 - - - - - - ? ? ? ? ? ? ? ? % % % % % % % % = = = = = = 9 9 9 9 9 9		8	$	$ 
9  
H H H H Hj H H 
H H Hr3   