tj,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 eZdZUdZeed<eed<dZeed<edZde eeffd Z de fd Z d ed ed e de ee ffdZ d)dedee dede fdZdedefdZ d)dededee dedef dZedede de fdZdee fdZdee fdZ d*dededee de fdZ d*dededee de fdZ d*dededee de fd Z d*dededee de fd!Zd"e d e defd#Zd"e d e defd$Z d"e d e defd%Z!d"e d e 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 b/var/www/html/CA-Chatbot/venv/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 instructionsparams preview_onlycr|r|ni}|d|i|r|ddi|S)zCreate a payload for an action.r/r1T)r)rr/r0r1datas 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_urlas'*IY*I*I*IIIrcz||||}td|||S)NPOSTr,)r4r r8)rr5r/r0r1r3s r_create_action_requestz'ZapierNLAWrapper._create_action_requestesI**<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_environmentss[ &*" +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.r1T input_params)r#rr<rSrTr+r,rUs rpreviewzZapierNLAWrapper.previews ##%%!)r ~t,-----ivtTT<< 7 7 @ @AA!!###}}~..rc K|d|||||dd{V}|dS)r\r:T)r1r;NrRrXrYs rapreviewzZapierNLAWrapper.apreviewsx    # #I . .,,\6PT,UU(         !!rargscD|j|i|}tj|S)cSame as run, but returns a stringified version of the JSON for insertting back into an LLM.)rVr,dumpsrrar&r3s r run_as_strzZapierNLAWrapper.run_as_str s*tx(((z$rcTK|j|i|d{V}tj|S)rcN)rZr,rdres r arun_as_strzZapierNLAWrapper.arun_as_strs@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,rdres rpreview_as_strzZapierNLAWrapper.preview_as_strs*t|T,V,,z$rcTK|j|i|d{V}tj|S)rjN)r`r,rdres 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,rdractionss r list_as_strzZapierNLAWrapper.list_as_str!s!))++z'"""rcbK|d{V}tj|S)roN)rIr,rdrps r alist_as_strzZapierNLAWrapper.alist_as_str's7 $$$$$$z'"""r)NFr )%__name__ __module__ __qualname____doc__str__annotations__rr model_configrrr r#rr.rboolr4r8r r<r classmethodrErrIrOrVrZr^r`rfrhrkrmrrrtrrrrs$#&&&&????:Lc3h g -c--s-tCQTH~----"&"    "JCJCJJJJ"&"                _(###$3[$#0#T$Z####,&*d4j&*&*&*&*RKO))),/)9A$) ))))$KO""",/"9A$" """"&KO / / /,/ /9A$ /  / / / /KO " " ",/ "9A$ "  " " " "  s s     s c c     C 3 3     3 # #    #S#### #C######rr)rxr,typingrrrrr(r!langchain_core.utilsrpydanticr r r r r rr~rrrs   ,,,,,,,,,,,,555555;;;;;;;;;;%%%%%%%%R#R#R#R#R#yR#R#R#R#R#r