a bŠÝgt-ã@sxdZddlZddlmZmZmZmZddlZddlZddl m Z ddl m Z m Z mZddlmZmZGdd„de ƒZdS) aUtil that can interact with Zapier NLA. Full docs here: https://nla.zapier.com/start/ Note: this wrapper currently only implemented the `api_key` auth method for testing and server-side production use cases (using the developer's connected accounts on Zapier.com) For use-cases where LangChain + Zapier NLA is powering a user-facing application, and LangChain needs access to the end-user's connected accounts on Zapier.com, you'll need to use oauth. Review the full docs above and reach out to nla@zapier.com for developer support. éN)ÚAnyÚDictÚListÚOptional©Úget_from_dict_or_env)Ú BaseModelÚ ConfigDictÚmodel_validator)ÚRequestÚSessionc@s¸eZdZUdZeed<eed<dZeed<eddZe eefdœd d „Z e dœd d „Z eee e ee fd œdd„Zd8eee e dœdd„Zeedœdd„Zd9eeee edœdd„Zeddee e dœdd„ƒƒZee dœd d!„Zee dœd"d#„Zd:eeee e dœd$d%„Zd;eeee e dœd&d'„ZdÚZapierNLAWrapperaðWrapper for Zapier NLA. Full docs here: https://nla.zapier.com/start/ This wrapper supports both API Key and OAuth Credential auth methods. API Key is the fastest way to get started using this wrapper. Call this wrapper with either `zapier_nla_api_key` or `zapier_nla_oauth_access_token` arguments, or set the `ZAPIER_NLA_API_KEY` environment variable. If both arguments are set, the Access Token will take precedence. For use-cases where LangChain + Zapier NLA is powering a user-facing application, and LangChain needs access to the end-user's connected accounts on Zapier.com, you'll need to use OAuth. Review the full docs above to learn how to create your own provider and generate credentials. Úzapier_nla_api_keyÚzapier_nla_oauth_access_tokenzhttps://nla.zapier.com/api/v1/Úzapier_nla_api_baseZforbid)Úextra)ÚreturncCs<dddœ}|jr(| dd|j›i¡n| d|ji¡|S)zFormat headers for requests.zapplication/json)ÚAcceptz Content-TypeÚ AuthorizationzBearer z X-API-Key)rÚupdater)ÚselfÚheaders©rúr/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/zapier.pyÚ_format_headers4sþÿz ZapierNLAWrapper._format_headerscCst ¡}|j | ¡¡|S)N)Úrequestsr rrr)rÚsessionrrrÚ _get_sessionDszZapierNLAWrapper._get_session)ÚmethodÚurlÚkwargsrc Ësºtj| ¡d4IdHšˆ}|j||fi|¤Ž4IdHš@}| ¡| ¡IdHWdƒIdHWdƒIdHS1IdHs‚0YWdƒIdHq¶1IdHs¬0YdS)zMake an async request.)rN)ÚaiohttpZ ClientSessionrÚrequestÚraise_for_statusÚjson)rrrr rÚresponserrrÚ _arequestIszZapierNLAWrapper._arequestNF)Ú instructionsÚparamsrcCs0|r|ni}| d|i¡|r,| ddi¡|S)zCreate a payload for an action.r'Ú preview_onlyT)r)rr'r(r)ÚdatarrrÚ_create_action_payloadPs ÿÿz'ZapierNLAWrapper._create_action_payload)Ú action_idrcCs|jd|›dS)zCreate a url for an action.úexposed/z /execute/)r)rr,rrrÚ_create_action_url^sz#ZapierNLAWrapper._create_action_url)r,r'r(rcCs"| |||¡}td| |¡|dS)NÚPOST©r$)r+r r.)rr,r'r(r)r*rrrÚ_create_action_requestbs ýz'ZapierNLAWrapper._create_action_requestÚbefore)Úmode)ÚvaluesrcCs4d}d|vrd}nd|d<t|dd|ƒ}||d<|S)z,Validate that api key exists in environment.NrÚrZZAPIER_NLA_API_KEYr)Úclsr4Zzapier_nla_api_key_defaultrrrrÚvalidate_environmentpsüz%ZapierNLAWrapper.validate_environmentcÃs | d|jd¡IdH}|dS)a´Returns a list of all exposed (enabled) actions associated with current user (associated with the set api_key). Change your exposed actions here: https://nla.zapier.com/demo/start/ The return list can be empty if no actions exposed. Else will contain a list of action objects: [{ "id": str, "description": str, "params": Dict[str, str] }] `params` will always contain an `instructions` key, the only required param. All others optional and if provided will override any AI guesses (see "understanding the AI guessing flow" here: https://nla.zapier.com/api/v1/docs) ÚGETr-NÚresults)r&r)rr%rrrÚalistŠszZapierNLAWrapper.alistc Cs”| ¡}z| |jd¡}| ¡Wnbtjy†}zH|jdkrn|jrZtjd|›|d‚tjd|›|d‚|‚WYd}~n d}~00| ¡dS)aÇReturns a list of all exposed (enabled) actions associated with current user (associated with the set api_key). Change your exposed actions here: https://nla.zapier.com/demo/start/ The return list can be empty if no actions exposed. Else will contain a list of action objects: [{ "id": str, "description": str, "params": Dict[str, str] }] `params` will always contain an `instructions` key, the only required param. All others optional and if provided will override any AI guesses (see "understanding the AI guessing flow" here: https://nla.zapier.com/docs/using-the-api#ai-guessing) r-i‘zrAn unauthorized response occurred. Check that your access token is correct and doesn't need to be refreshed. Err: )r%zLAn unauthorized response occurred. Check that your api key is correct. Err: Nr9) rÚgetrr#rÚ HTTPErrorÚ status_coderr$)rrr%Úhttp_errrrrÚlist s*  þüÿýzZapierNLAWrapper.listcCs:| ¡}| |||¡}| | |¡¡}| ¡| ¡dS)áExecutes an action that is identified by action_id, must be exposed (enabled) by the current user (associated with the set api_key). Change your exposed actions here: https://nla.zapier.com/demo/start/ The return JSON is guaranteed to be less than ~500 words (350 tokens) making it safe to inject into the prompt of another LLM call. Úresult)rr1ÚsendÚprepare_requestr#r$©rr,r'r(rr"r%rrrÚrunÈs zZapierNLAWrapper.runcÃs,|jd| |¡| ||¡dIdH}|dS)r@r/r0NrA©r&r.r+©rr,r'r(r%rrrÚarunÙs  ý zZapierNLAWrapper.aruncCsV| ¡}|r|ni}| ddi¡| |||d¡}| | |¡¡}| ¡| ¡dS)úËSame as run, but instead of actually executing the action, will instead return a preview of params that have been guessed by the AI in case you need to explicitly review before executing.r)TZ input_params)rrr1rBrCr#r$rDrrrÚpreviewës zZapierNLAWrapper.previewcÃs0|jd| |¡|j||dddIdH}|dS)rIr/T)r)r0NrArFrGrrrÚapreviewùs ý zZapierNLAWrapper.apreviewcOs|j|i|¤Ž}t |¡S)úcSame as run, but returns a stringified version of the JSON for insertting back into an LLM.)rEr$Údumps©rÚargsr r*rrrÚ run_as_strszZapierNLAWrapper.run_as_strcÏs |j|i|¤ŽIdH}t |¡S)rLN)rHr$rMrNrrrÚ arun_as_str szZapierNLAWrapper.arun_as_strcOs|j|i|¤Ž}t |¡S)úgSame as preview, but returns a stringified version of the JSON for insertting back into an LLM.)rJr$rMrNrrrÚpreview_as_strszZapierNLAWrapper.preview_as_strcÏs |j|i|¤ŽIdH}t |¡S)rRN)rKr$rMrNrrrÚapreview_as_strsz ZapierNLAWrapper.apreview_as_strcCs| ¡}t |¡S)údSame as list, but returns a stringified version of the JSON for insertting back into an LLM.)r?r$rM©rÚactionsrrrÚ list_as_str szZapierNLAWrapper.list_as_strcÃs| ¡IdH}t |¡S)rUN)r:r$rMrVrrrÚ alist_as_str&szZapierNLAWrapper.alist_as_str)NF)NF)N)N)N)N)$Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚstrÚ__annotations__rr Z model_configrrr rrr&rr+r.r r1r Ú classmethodr7rr:r?rErHrJrKrPrQrSrTrXrYrrrrr sp  ÿÿþ ûú )ÿ þ ÿ þ ÿ þ ÿ þ þ r )r]r$Útypingrrrrr!rZlangchain_core.utilsrZpydanticrr r r r r rrrrÚs