昺h-dZddlZddlmZmZmZmZddlZddlZddl m Z m Z m Z ddl 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)AnyDictListOptional) BaseModelExtraroot_validatorget_from_dict_or_env)RequestSessionc JeZdZUdZeed<eed<dZeed<GddZdeeeffd Z de fd Z d ed ed e deee ffdZ d'dedeedefdZdedefdZ d'dededeedefdZeddedefdZdeefdZdeefdZ d(dededeedefdZ d(dededeedefdZ d(dededeedefdZ d(dededeedefd Zdefd!Zdefd"Zdefd#Zdefd$Zdefd%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_basec eZdZdZejZdS)ZapierNLAWrapper.Configz'Configuration for this pydantic object.N)__name__ __module__ __qualname____doc__rforbidextral/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain_community/utilities/zapier.pyConfigr/s55 rrreturncddd}|jr |dd|jin|d|ji|S)zFormat headers for requests.zapplication/json)Acceptz Content-Type AuthorizationzBearer z X-API-Key)rupdater)selfheaderss r_format_headersz ZapierNLAWrapper._format_headers4st).    - C NN "PD,N"P"PQ     NNK)@A B B Brctj}|j||SN)requestsr r%r#r&)r$sessions 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.)r%N)aiohttp ClientSessionr&requestraise_for_statusjson)r$r,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.r7 preview_onlyT)r#)r$r7r8r:datas 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)r$r=s r_create_action_urlz#ZapierNLAWrapper._create_action_url^s'*IY*I*I*IIIrcz||||}td|||S)NPOSTr4)r<r r@)r$r=r7r8r:r;s r_create_action_requestz'ZapierNLAWrapper._create_action_requestbsI**<NN   # #I . .    rT)prevaluescPd}d|vrd}nd|d<t|dd|}||d<|S)z,Validate that api key exists in environment.NrrZAPIER_NLA_API_KEYr )clsrFzapier_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) GETr?Nresults)r6r)r$r5s 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) r?izrAn unauthorized response occurred. Check that your access token is correct and doesn't need to be refreshed. Err: )r5zLAn unauthorized response occurred. Check that your api key is correct. Err: NrO) r+getrr3r) HTTPError status_coderr4)r$r*r5http_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+rDsendprepare_requestr3r4r$r=r7r8r*r2r5s rrunzZapierNLAWrapper.runsp##%%--ivNN<< 7 7 @ @AA!!###}}x((rcK|d|||||d{V}|dS)rXrBrCNrYr6r@r<r$r=r7r8r5s 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.r:T input_params)r+r#rDrZr[r3r4r\s rpreviewzZapierNLAWrapper.previews ##%%!)r ~t,-----ivtTT<< 7 7 @ @AA!!###}}~..rc K|d|||||dd{V}|dS)rcrBT)r:rCNrYr_r`s 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.)r]r4dumpsr$argsr.r;s r run_as_strzZapierNLAWrapper.run_as_strs*tx(((z$rcTK|j|i|d{V}tj|S)riN)rar4rjrks 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.)rer4rjrks rpreview_as_strzZapierNLAWrapper.preview_as_strs*t|T,V,,z$rcTK|j|i|d{V}tj|S)rqN)rgr4rjrks 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.)rVr4rjr$actionss r list_as_strzZapierNLAWrapper.list_as_strs!))++z'"""rcbK|d{V}tj|S)rvN)rPr4rjrws r alist_as_strzZapierNLAWrapper.alist_as_str%s7 $$$$$$z'"""r)NFr()"rrrrstr__annotations__rrrr&r r+rr6rr<r@r rDr rLrrPrVr]rarergrmrorrrtryr{rrrrrs$#&&&&???? c3h g -c--s-tCQTH~----NS   )1$      JCJCJJJJ"&              ^$40#T$Z####,&*d4j&*&*&*&*RKO))),/)9A$) ))))$KO""",/"9A$" """"&KO / / /,/ /9A$ /  / / / /KO " " ",/ "9A$ "  " " " " S     C              #S#### #C######rr)rr4typingrrrrr0r)langchain_core.pydantic_v1rrr langchain_core.utilsr r r rrrrrs   ,,,,,,,,,,,,GGGGGGGGGG555555%%%%%%%%Q#Q#Q#Q#Q#yQ#Q#Q#Q#Q#r