U h@shdZddlmZmZmZddlZddlmZmZddl m Z ddl m Z ddl mZGdd d eZdS) z%Util that sends messages via Infobip.)DictListOptionalN) BaseModelroot_validatorget_from_dict_or_env) HTTPAdapter)Retryc@seZdZUdZdZeeed<dZeeed<GdddZ e dd e e d d d Z e jd ddZeeeeedddZeeeeedddZdeeeeeedddZdS)InfobipAPIWrapperz&Wrapper for Infobip API for messaging.Ninfobip_api_keyzhttps://api.infobip.cominfobip_base_urlc@seZdZdZdS)zInfobipAPIWrapper.ConfigZforbidN)__name__ __module__ __qualname__extrarrI/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/infobip.pyConfigsrT)pre)valuesreturncCs$t|dd|d<t|dd|d<|S)z,Validate that api key exists in environment.r ZINFOBIP_API_KEYr ZINFOBIP_BASE_URLr)clsrrrrvalidate_environmentsz&InfobipAPIWrapper.validate_environment)rcCsTtdddddddgd}t|d }t}|d ||jd |jd d |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 WSWntk rYd SXz|dddWStk rYdSXdS)zSend an SMS message.messagescSsg|] }d|iqS)tor).0Z 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%postr r4 status_codeKeyError)r&r*r+r,r4r(response response_jsonrrr _send_sms3s>   zInfobipAPIWrapper._send_sms) from_emailto_emailsubjectbodyrc Cszddlm}Wn,tk r<}ztd|W5d}~XYnX||||d}||d}|} | jd|ji| j|jd|d } | } z | j d kr| d d d WSWnt k rYdSXz| dddWSt k rYdSXdS)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;r r4r<r=) r&rArBrCrDrEeZ form_datarGr(r>r?rrr _send_email]sH   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)rrr__doc__r rstr__annotations__r rrrrr!r"r)rr@rKrQrrrrr s>   + 2r )rRtypingrrrr!Zlangchain_core.pydantic_v1rrZlangchain_core.utilsrZrequests.adaptersr Z urllib3.utilr r rrrrs