U ™ÀÂhQ-ã@stdZddlZddlmZmZmZmZddlZddlZddl m Z m Z ddl m Z ddlmZmZGdd„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)Ú BaseModelÚroot_validator©Úget_from_dict_or_env)ÚRequestÚSessionc@s¸eZdZUdZeed<eed<dZeed<Gdd„dƒZeeefdœd d „Z e dœd d „Z eee eee fd œdd„Z d8eeeedœdd„Zeedœdd„Zd9eeeeedœdd„Zeddeedœdd„ƒZeedœd d!„Zeedœd"d#„Zd:eeeeedœd$d%„Zd;eeeeedœ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_basec@seZdZdZdS)zZapierNLAWrapper.ConfigZforbidN)Ú__name__Ú __module__Ú __qualname__Úextra©rrúH/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/utilities/zapier.pyÚConfig0sr)Ú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ÚheadersrrrÚ_format_headers3sþÿz ZapierNLAWrapper._format_headerscCst ¡}|j | ¡¡|S)N)Úrequestsr rrr)rÚsessionrrrÚ _get_sessionCszZapierNLAWrapper._get_session)ÚmethodÚurlÚkwargsrc ËsŽtj| ¡d4IdHšf}|j||f|Ž4IdHš<}| ¡| ¡IdHW5QIdHR£W5QIdHR£SQIdHRXW5QIdHRXdS)zMake an async request.)rN)ÚaiohttpZ ClientSessionrÚrequestÚraise_for_statusÚjson)rr!r"r#rÚresponserrrÚ _arequestHszZapierNLAWrapper._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_payloadOs ÿÿ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 r1)rr/r*r+r,r-rrrÚ_create_action_requestas ýz'ZapierNLAWrapper._create_action_requestT)Úpre)ÚvaluesrcCs4d}d|krd}nd|d<t|dd|ƒ}||d<|S)z,Validate that api key exists in environment.NrÚr ZZAPIER_NLA_API_KEYr)Úclsr6Zzapier_nla_api_key_defaultr rrrÚvalidate_environmentosü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) ÚGETr0NÚresults)r)r)rr(rrrÚalistˆszZapierNLAWrapper.alistc Cs’| ¡}z| |jd¡}| ¡Wn`tjk r„}z@|jdkrp|jr\tjd|›|d‚tjd|›|d‚|‚W5d}~XYnX| ¡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) r0i‘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: Nr;) 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)r r4ÚsendÚprepare_requestr&r'©rr/r*r+rr%r(rrrÚrunÆs zZapierNLAWrapper.runcÃs,|jd| |¡| ||¡dIdH}|dS)rBr2r3NrC©r)r1r.©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)r rr4rDrEr&r'rFrrrÚpreviewés zZapierNLAWrapper.previewcÃs0|jd| |¡|j||dddIdH}|dS)rKr2T)r,r3NrCrHrIrrrÚapreview÷s ý zZapierNLAWrapper.apreviewcOs|j||Ž}t |¡S)úcSame as run, but returns a stringified version of the JSON for insertting back into an LLM.)rGr'Údumps©rÚargsr#r-rrrÚ run_as_strs zZapierNLAWrapper.run_as_strcÏs|j||ŽIdH}t |¡S)rNN)rJr'rOrPrrrÚ arun_as_str szZapierNLAWrapper.arun_as_strcOs|j||Ž}t |¡S)úgSame as preview, but returns a stringified version of the JSON for insertting back into an LLM.)rLr'rOrPrrrÚpreview_as_strs zZapierNLAWrapper.preview_as_strcÏs|j||ŽIdH}t |¡S)rTN)rMr'rOrPrrrÚ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.)rAr'rO©rÚactionsrrrÚ list_as_strszZapierNLAWrapper.list_as_strcÃs| ¡IdH}t |¡S)rWN)r<r'rOrXrrrÚ alist_as_str$szZapierNLAWrapper.alist_as_str)NF)NF)N)N)N)N)"rrrÚ__doc__ÚstrÚ__annotations__rrrrr r rr)rr.r1r r4rr9rr<rArGrJrLrMrRrSrUrVrZr[rrrrr s~  ÿþ ûú )ÿþ ÿþ ÿþ ÿþ þ r )r\r'Útypingrrrrr$rZlangchain_core.pydantic_v1rrZlangchain_core.utilsr r r r rrrrÚs