Ngt-dZddlZddlmZmZmZmZddlZddlZddl m Z ddl m Z m Z mZddlmZmZGdde 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)AnyDictListOptionalget_from_dict_or_env) BaseModel ConfigDictmodel_validator)RequestSessionc XeZdZUdZeed<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 fd Zdefd!Zdefd"Zdefd#Z defd$Z!defd%Z"defd&Z#dS))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 |dd|jin|d|ji|S)zFormat headers for requests.zapplication/json)Acceptz Content-Type AuthorizationzBearer z X-API-Key)rupdater)selfheaderss `/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/utilities/zapier.py_format_headersz ZapierNLAWrapper._format_headers4st).    - C NN "PD,N"P"PQ     NNK)@A B B Bctj}|j||SN)requestsr rrr)rsessions r _get_sessionzZapierNLAWrapper._get_sessionDs7"$$t3355666rmethodurlkwargscKtj|4d{V}|j||fi|4d{V}||d{Vcdddd{Vcdddd{VS#1d{VswxYwY dddd{VdS#1d{VswxYwYdS)zMake an async request.)rN)aiohttp ClientSessionrrequestraise_for_statusjson)rr$r%r&r"responses r _arequestzZapierNLAWrapper._arequestIs#(1E1E1G1GHHH - - - - - - -G&wvs==f== - - - - - - -))+++%]]__,,,,,, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -s5B>.B4B> B# #B>&B# 'B>> C CNF instructionsparamscr|r|ni}|d|i|r|ddi|S)zCreate a payload for an action.r/ preview_onlyT)r)rr/r0r2datas r_create_action_payloadz'ZapierNLAWrapper._create_action_payloadPsX 'vvR       0 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'*IY*I*I*IIIrcz||||}td|||S)NPOSTr,)r4r r8)rr5r/r0r2r3s r_create_action_requestz'ZapierNLAWrapper._create_action_requestbsI**<NN   # #I . .    rbefore)modevaluescPd}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_environmentps[ &*" +f 4 4)+ & &68F2 32  &   (:#$ rc^K|d|jdzd{V}|dS)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.alists@&t/G*/TUUUUUUUU ""rc~|} ||jdz}|n]#tj$rK}|jdkr9|jrt jd||t jd|||d}~wwxYw|dS)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#HIIH  % % ' ' ' '!   #s**5",6+366"*  (7,477% N }}y))s1AB"ABB"c|}||||}|||}||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.runsp##%%--ivNN<< 7 7 @ @AA!!###}}x((rcK|d|||||d{V}|dS)rQr:r;NrRr.r8r4rr5r/r0r-s rarunzZapierNLAWrapper.arunsr   # #I . .,,\6BB(         !!rc@|}|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.r2T input_params)r#rr<rSrTr+r,rUs rpreviewzZapierNLAWrapper.previews ##%%!)r ~t,-----ivtTT<< 7 7 @ @AA!!###}}~..rc K|d|||||dd{V}|dS)r\r:T)r2r;NrRrXrYs rapreviewzZapierNLAWrapper.apreviewsx    # #I . .,,\6PT,UU(         !!rcD|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*tx(((z$rcTK|j|i|d{V}tj|S)rbN)rZr,rcrds r arun_as_strzZapierNLAWrapper.arun_as_str s@TY/////////z$rcD|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,,z$rcTK|j|i|d{V}tj|S)rjN)r`r,rcrds rapreview_as_strz ZapierNLAWrapper.apreview_as_strs@ #T]D3F33333333z$rcR|}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!))++z'"""rcbK|d{V}tj|S)roN)rIr,rcrps r alist_as_strzZapierNLAWrapper.alist_as_str&s7 $$$$$$z'"""r)NFr )$__name__ __module__ __qualname____doc__str__annotations__rr model_configrrr r#rr.rr4r8r r<r classmethodrErrIrOrVrZr^r`rfrhrkrmrrrtrrrrs$#&&&&????:Lc3h g -c--s-tCQTH~----NS   )1$      JCJCJJJJ"&              _(###$3[$#0#T$Z####,&*d4j&*&*&*&*RKO))),/)9A$) ))))$KO""",/"9A$" """"&KO / / /,/ /9A$ /  / / / /KO " " ",/ "9A$ "  " " " " S     C              #S#### #C######rr)rxr,typingrrrrr(r!langchain_core.utilsrpydanticr r r r r rr}rrrs   ,,,,,,,,,,,,555555;;;;;;;;;;%%%%%%%%Q#Q#Q#Q#Q#yQ#Q#Q#Q#Q#r