o ȳgt-@sxdZddlZddlmZmZmZmZddlZddlZddl m Z ddl m Z m Z mZddlmZmZGddde 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 @seZdZUdZeed<eed<dZeed<eddZde eeffd d Z de fd d Z d edede de ee ffddZ d:dedee de fddZdedefddZ  d:dededee defddZeddede de fd d!Zdee fd"d#Zdee fd$d%Z d;dededee de fd&d'Z d;dededee de fd(d)Z d;dededee de fd*d+Z d;dededee de fd,d-Zdefd.d/Zdefd0d1Zdefd2d3Z defd4d5Z!defd6d7Z"defd8d9Z#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)extrareturncCs>ddd}|jr|dd|ji|S|d|ji|S)zFormat headers for requests.zapplication/json)Acceptz Content-Type AuthorizationzBearer z X-API-Key)rupdater)selfheadersr`/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/zapier.py_format_headers4sz ZapierNLAWrapper._format_headerscCst}|j||SN)requestsr rrr)rsessionrrr _get_sessionDszZapierNLAWrapper._get_sessionmethodurlkwargsc stj|d4IdHE}|j||fi|4IdH }||IdHWdIdHWdIdHS1IdHsBwYWdIdHdS1IdHsXwYdS)zMake an async request.)rN)aiohttp ClientSessionrrequestraise_for_statusjson)rr r!r"rresponserrr _arequestIs .zZapierNLAWrapper._arequestNF instructionsparamscCs0|r|ni}|d|i|r|ddi|S)zCreate a payload for an action.r* preview_onlyT)r)rr*r+r,datarrr_create_action_payloadPs z'ZapierNLAWrapper._create_action_payload action_idcCs|jd|dS)zCreate a url for an action.exposed/z /execute/)r)rr/rrr_create_action_url^sz#ZapierNLAWrapper._create_action_urlcCs"||||}td|||dS)NPOSTr')r.r r1)rr/r*r+r,r-rrr_create_action_requestbs z'ZapierNLAWrapper._create_action_requestbefore)modevaluescCs4d}d|vr d}nd|d<t|dd|}||d<|S)z,Validate that api key exists in environment.NrrZAPIER_NLA_API_KEYr)clsr7zapier_nla_api_key_defaultrrrrvalidate_environmentpsz%ZapierNLAWrapper.validate_environmentcs"|d|jdIdH}|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) GETr0Nresults)r)r)rr(rrralistszZapierNLAWrapper.alistc Cs|}z||jd}|Wn+tjy=}z|jdkr7|jr-tjd||dtjd||d|d}~ww|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) r0izrAn 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>) rgetrr&r HTTPError status_coderr')rrr(http_errrrrlists.   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)rr4sendprepare_requestr&r'rr/r*r+rr%r(rrrruns   zZapierNLAWrapper.runcs.|jd|||||dIdH}|dS)rEr2r3NrFr)r1r.rr/r*r+r(rrraruns   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,T input_params)rrr4rGrHr&r'rIrrrpreviews  zZapierNLAWrapper.previewcs2|jd|||j||dddIdH}|dS)rNr2T)r,r3NrFrKrLrrrapreviews zZapierNLAWrapper.apreviewcO|j|i|}t|S)cSame as run, but returns a stringified version of the JSON for insertting back into an LLM.)rJr'dumpsrargsr"r-rrr run_as_str zZapierNLAWrapper.run_as_strc"|j|i|IdH}t|S)rSN)rMr'rTrUrrr arun_as_str s zZapierNLAWrapper.arun_as_strcOrR)gSame as preview, but returns a stringified version of the JSON for insertting back into an LLM.)rPr'rTrUrrrpreview_as_strrXzZapierNLAWrapper.preview_as_strcrY)r[N)rQr'rTrUrrrapreview_as_strs z ZapierNLAWrapper.apreview_as_strcCs|}t|S)dSame as list, but returns a stringified version of the JSON for insertting back into an LLM.)rDr'rTractionsrrr list_as_str s zZapierNLAWrapper.list_as_strcs|IdH}t|S)r^N)r?r'rTr_rrr alist_as_str&s zZapierNLAWrapper.alist_as_str)NFr)$__name__ __module__ __qualname____doc__str__annotations__rr model_configrrr rrr)rr.r1r r4r classmethodr<rr?rDrJrMrPrQrWrZr\r]rarbrrrrr s  "  )      r )rfr'typingrrrrr#rlangchain_core.utilsrpydanticrr r r r r rrrrs