o Zh@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 d e d efd d Zd ejfddZdedeeded efddZdedededed ef ddZ     d!dededededed ef dd ZdS)"InfobipAPIWrapperz&Wrapper for Infobip API for messaging.Ninfobip_api_keyzhttps://api.infobip.cominfobip_base_urlZforbid)extrabefore)modevaluesreturncCs$t|dd|d<t|dd|d<|S)z,Validate that api key exists in environment.rZINFOBIP_API_KEYrZINFOBIP_BASE_URLr)clsrr\/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/utilities/infobip.pyvalidate_environmentsz&InfobipAPIWrapper.validate_environmentcCsNtddgdd}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_sessionsenderdestination_phone_numberstextcCsddd|D||dgi}|}|jddi|j|jd|d}|}z|jd kr8|d d d WSWn tyCYd Swz |dddWStyX YdSw)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 successfully)r(r#r$postrr4 status_codeKeyError)r%r)r*r+r4r'response response_jsonrrr _send_sms5s@     zInfobipAPIWrapper._send_sms from_emailto_emailsubjectbodyc Cszddlm}Wnty}ztd|d}~ww||||d}||d}|} | jd|ji| j|jd|d } | } z| j d krQ| d d d WSWn t y\YdSwz | dddWSt yq YdSw)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_sN     zInfobipAPIWrapper._send_emailsmsr-channelcCs|dkr%|dkr td|dkrtd|dkrtd|j||g|dS|dkrR|dkr1td|dkr9td |dkrAtd |dkrItd |j||||d Std |d)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 sb   * 2r )rUtypingrrrrr Zlangchain_core.utilsrZpydanticrr r Zrequests.adaptersr Z urllib3.utilr r rrrrs