a bg@spdZddlmZmZmZmZddlZddlmZddl m Z m Z m Z ddl mZddlmZGdd d e ZdS) z%Util that sends messages via Infobip.)AnyDictListOptionalNget_from_dict_or_env) BaseModel ConfigDictmodel_validator) HTTPAdapter)Retryc@seZdZUdZdZeeed<dZeeed<e ddZ e dd e e ed d d Zejd ddZeeeeedddZeeeeedddZdeeeeeedddZdS)InfobipAPIWrapperz&Wrapper for Infobip API for messaging.Ninfobip_api_keyzhttps://api.infobip.cominfobip_base_urlZforbid)extrabefore)mode)valuesreturncCs$t|dd|d<t|dd|d<|S)z,Validate that api key exists in environment.rZINFOBIP_API_KEYrZINFOBIP_BASE_URLr)clsrrs/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/infobip.pyvalidate_environmentsz&InfobipAPIWrapper.validate_environment)rcCsNtddgdd}t|d}t}|d||jd|jdd |S) z0Get a requests session with the correct headers.)iiiii)totalbackoff_factorstatus_forcelist) max_retrieszhttps://zApp zinfobip-langchain-community) Authorizationz User-Agent)r r requestsSessionmountheadersupdater)selfZretry_strategyadaptersessionrrr_get_requests_session"s   z'InfobipAPIWrapper._get_requests_session)senderdestination_phone_numberstextrcCsddd|D||dgi}|}|jddi|j|jd|d}|}z |jd krp|d d d WSWntyYd S0z|dddWStyYdS0dS)zSend an SMS message.messagescSsg|] }d|iqS)tor).0 destinationrrr <sz/InfobipAPIWrapper._send_sms..) destinationsfromr+ Content-Typezapplication/jsonz/sms/2/text/advanced)json requestErrorserviceExceptionr+Failed to send messager messageIdECould not get message ID from response, message was sent successfullyN)r(r#r$postrr4 status_codeKeyError)r%r)r*r+r4r'response response_jsonrrr _send_sms5s<     zInfobipAPIWrapper._send_sms) from_emailto_emailsubjectbodyrc Cszddlm}Wn.ty>}ztd|WYd}~n d}~00||||d}||d}|} | jd|ji| j|jd|d } | } z | j d kr| d d d WSWnt yYdS0z| dddWSt yYdS0dS)zSend an email message.r)MultipartEncoderz^Unable to import requests_toolbelt, please install it with `pip install -U requests-toolbelt`.N)r2r-rCr+)fieldsr3z /email/3/send)datar5r6r7r+r8r,r9r:) Zrequests_toolbeltrE ImportErrorr(r#r$ content_typer;rr4r<r=) r%rArBrCrDrEeZ form_datarGr'r>r?rrr _send_email_sF     zInfobipAPIWrapper._send_emailsms)rDr-r)rCchannelrcCs|dkrJ|dkrtd|dkr(td|dkr8td|j||g|dS|dkr|dkrbtd|dkrrtd |dkrtd |dkrtd |j||||d Std |ddS)NrMrLz)Sender must be specified for SMS messagesz.Destination must be specified for SMS messagesz'Body must be specified for SMS messages)r)r*r+emailz+Sender must be specified for email messagesz0Destination must be specified for email messagesz,Subject must be specified for email messagesz)Body must be specified for email messages)rArBrCrDzChannel z is not supported) ValueErrorr@rK)r%rDr-r)rCrNrrrruns8zInfobipAPIWrapper.run)rLrLrLrLrM)__name__ __module__ __qualname____doc__rrstr__annotations__rr Z model_configr classmethodrrrr r!r(rr@rKrQrrrrr s:    + 2r )rUtypingrrrrr Zlangchain_core.utilsrZpydanticrr r Zrequests.adaptersr Z urllib3.utilr r rrrrs