o ȳg@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_urlforbid)extrabefore)modevaluesreturncCs$t|dd|d<t|dd|d<|S)z,Validate that api key exists in environment.rINFOBIP_API_KEYrINFOBIP_BASE_URLr)clsrra/var/www/html/chatdoc2/venv/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)selfretry_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'postrr8 status_codeKeyError)r(r-r.r/r8r+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)r6r1rGr/)fieldsr7z /email/3/send)datar9r:r;r/r<r0r=r>) requests_toolbeltrI ImportErrorr,r&r' content_typer?rr8r@rA) r(rErFrGrHrIe form_datarKr+rBrCrrr _send_email_sN     zInfobipAPIWrapper._send_emailsmsr1channelcCs|dkr%|dkr td|dkrtd|dkrtd|j||g|dS|dkrR|dkr1td|dkr9td |dkrAtd |dkrItd |j||||d Std |d)NrSrRz)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)rErFrGrHzChannel z is not supported) ValueErrorrDrQ)r(rHr1r-rGrTrrrruns8zInfobipAPIWrapper.run)rRrRrRrRrS)__name__ __module__ __qualname____doc__rrstr__annotations__rr model_configr classmethodrrrr#r$r,rrDrQrWrrrrr sb   * 2r )r[typingrrrrr#langchain_core.utilsrpydanticrr r requests.adaptersr urllib3.utilr r rrrrs