7|ht-~dZddlZddlmZmZmZmZddlZddlZddl m Z ddl m Z m Z mZddlmZmZGdde Zy) 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)AnyDictListOptionalget_from_dict_or_env) BaseModel ConfigDictmodel_validator)RequestSessionc eZdZUdZeed<eed<dZeed<edZde eeffd Z de fd Z d ed ed e de ee ffdZ d&dedee de fdZdedefdZ d&dededee defdZedede de fdZdee fdZdee fdZ d'dededee de fdZ d'dededee de fdZ d'dededee de fdZ d'dededee de fdZdefd Zdefd!Zdefd"Z defd#Z!defd$Z"defd%Z#y)(ZapierNLAWrapperaWrapper 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_keyzapier_nla_oauth_access_tokenzhttps://nla.zapier.com/api/v1/zapier_nla_api_baseforbid)extrareturncddd}|jr"|jdd|ji|S|jd|ji|S)zFormat headers for requests.zapplication/json)Acceptz Content-Type AuthorizationzBearer z X-API-Key)rupdater)selfheaderss c/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/utilities/zapier.py_format_headersz ZapierNLAWrapper._format_headers4sc).   - - NN GD,N,N+O"PQ   NNK)@)@A Bctj}|jj|j |SN)requestsr rrr)rsessions r _get_sessionzZapierNLAWrapper._get_sessionDs0""$t3356rmethodurlkwargscKtj|j4d{}|j||fi|4d{}|j |j d{cdddd{cdddd{S7k7P7*77 #1d{7swYnxYwdddd{7y#1d{7swYyxYww)zMake an async request.)rN)aiohttp ClientSessionrrequestraise_for_statusjson)rr$r%r&r"responses r _arequestzZapierNLAWrapper._arequestIs((1E1E1GH - -G&wvs=f= - -))+%]]_, - - - - - - -, - - - - - - - - - -s)CBCCB C $B!0B1B!4 CBC CBCCB!CC!B3 'B*(B3 /C6 CCCCC CCN instructionsparamscb|r|ni}|jd|i|r|jddi|S)zCreate a payload for an action.r/ preview_onlyT)r)rr/r0r2datas r_create_action_payloadz'ZapierNLAWrapper._create_action_payloadPs> vR     KK. / r action_idc(|jd|dzS)zCreate a url for an action.exposed/z /execute/)r)rr5s r_create_action_urlz#ZapierNLAWrapper._create_action_url^s''HYKy*IIIrcb|j|||}td|j||S)NPOSTr,)r4r r8)rr5r/r0r2r3s r_create_action_requestz'ZapierNLAWrapper._create_action_requestbs9**<N   # #I .  rbefore)modevaluescHd}d|vrd}nd|d<t|dd|}||d<|S)z,Validate that api key exists in environment.NrrZAPIER_NLA_API_KEYr)clsr?zapier_nla_api_key_defaultrs rvalidate_environmentz%ZapierNLAWrapper.validate_environmentpsP &*" +f 4)+ &68F2 32  &  (:#$ rcfK|jd|jdzd{}|dS7 w)aReturns 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) GETr7Nresults)r.r)rr-s ralistzZapierNLAWrapper.alists7&t/G/G*/TUU ""Vs #1/ 1c~|j} |j|jdz}|j|jdS#tj $rV}j dk(r@|jrt j d||t j d|||d}~wwxYw)aReturns 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) r7izrAn 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: NrH) r#getrr+r! HTTPError status_coderr,)rr"r-http_errs rlistzZapierNLAWrapper.lists&##% {{4#;#;j#HIH  % % ' }}y))!! ##s*55",,++3*6"*  ((,,4:7% N s.AB<&AB77B<c|j}|j|||}|j|j|}|j |j 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#r<sendprepare_requestr+r,rr5r/r0r"r*r-s rrunzZapierNLAWrapper.runs\##%--ivN<< 7 7 @A!!#}}x((rcK|jd|j||j||d{}|dS7 w)rQr:r;NrRr.r8r4rr5r/r0r-s rarunzZapierNLAWrapper.arunsS   # #I .,,\6B(   !!  s7AA Ac|j}|r|ni}|jddi|j|||d}|j|j |}|j |j 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.r2T input_params)r#rr<rSrTr+r,rUs rpreviewzZapierNLAWrapper.previewsx ##%!r ~t,---ivtT<< 7 7 @A!!#}}~..rc K|jd|j||j||dd{}|dS7 w)r\r:T)r2r;NrRrXrYs rapreviewzZapierNLAWrapper.apreviewsY    # #I .,,\6PT,U(   !!  s9AA AcP|j|i|}tj|S)cSame as run, but returns a stringified version of the JSON for insertting back into an LLM.)rVr,dumpsrargsr&r3s r run_as_strzZapierNLAWrapper.run_as_strs(txx((zz$rclK|j|i|d{}tj|S7w)rbN)rZr,rcrds r arun_as_strzZapierNLAWrapper.arun_as_str s4TYY///zz$0 424cP|j|i|}tj|S)gSame as preview, but returns a stringified version of the JSON for insertting back into an LLM.)r^r,rcrds rpreview_as_strzZapierNLAWrapper.preview_as_strs(t||T,V,zz$rclK|j|i|d{}tj|S7w)rkN)r`r,rcrds rapreview_as_strz ZapierNLAWrapper.apreview_as_strs4 #T]]D3F33zz$4ricL|j}tj|S)dSame as list, but returns a stringified version of the JSON for insertting back into an LLM.)rOr,rcractionss r list_as_strzZapierNLAWrapper.list_as_str s))+zz'""rchK|jd{}tj|S7w)rpN)rIr,rcrqs r alist_as_strzZapierNLAWrapper.alist_as_str&s* $zz'""%s 202)NFr )$__name__ __module__ __qualname____doc__str__annotations__rr model_configrrr r#rr.rr4r8r r<r classmethodrErrIrOrVrZr^r`rfrhrlrnrsrurrrrsE$#&&??Lc3h g -c--s-tCQTH~-NS  )1$   JCJCJ"&          (#$3$0#T$Z#,&*d4j&*RKO)),/)9A$) )$KO"",/"9A$" "&KO / /,/ /9A$ /  /KO " ",/ "9A$ "  " S  C     #S# #C#rr)ryr,typingrrrrr(r!langchain_core.utilsrpydanticr r r r r rr~rrrs5  ,,5;;%Q#yQ#r