tj3ddlmZddlZddlZddlZddlmZddlmZddl m Z m Z ddl Z ddl mZddlmZddlmZdd lmZdd lmZmZdd lmZdd l mZmZdd lmZddlmZddl m!Z!ddl"m#Z#e r ddl$m%Z%ddl&m'Z'ej(e)Z*dZ+d8dZ,d9dZ-edd e+!d:d#Z.edd e+!Gd$d%eZ/ed&d e+! d;dz_format_url..6s-CCCtq!v{{Aq6**CCCr%) refindalllstriprstrip isinstancelistr&ritemsformat) rrexpected_path_param new_paramsparam clean_paramvalsepnew_valkv_strsr*s @r+ _format_urlr?%s*Z55J$$$ll4((//44 +& c4  Qx3"2Y#--cc3 -qS+09+;+;''''(+++chhsmm;((3-- T " " !"I,,SS#FCCCCsyy{{CCCGQx3qS sxx(( (GG 1X__#iiGG 1X__-+----GGG# 5 3: # # # ##r-paramslist[Parameter]specrci}g}|D]}|jr||j}nMtt|jj}||}|jr|js |j|_tj |d||j <|j r| |j d||dS)NT exclude_noneobject)type propertiesrequired) param_schema get_schemanextitercontentvaluesmedia_type_schema descriptionjsonloadsnamerIappend)r@rBrHrIpschemarPs r+_openapi_params_to_json_schemarXKsJH  $ $ > 8__Q^44FF $T!)*:*:*<*<%=%= > > P __%677F = /!3 /!"F !Z  (F(FGG 16 : $ OOAF # # #JH M MMr-z1.0.4z2.0.0)sinceremovaladdendum%tuple[list[dict[str, Any]], Callable]c ddlm}n$#t$r}d}t||d}~wwxYw|jsgdfSg}i|jD]}d||D}||D]\}i}|||} |} || D]} | | | j | j f<tt} | D]&\} }| | d|'dd d d d }| D]&\}}| |rt| ||||<'|| }|r|jri}|j D]Q\}}|jrE||j}t)j|d ||<Rt-|dkrAt/t1| \}}|dkrdnd}|||<n9t-|dkr&dt|i|d<||||}|j|jd|dd}||||j|jzd|d<^ d#d$fd" }||fS)%abOpenAPI spec to OpenAI function JSON Schema. Convert a valid OpenAPI spec to the JSON Schema format expected for OpenAI functions. Args: spec: OpenAPI spec to convert. Returns: Tuple of the OpenAI functions JSON schema and a default function for executing a request based on the OpenAI function schema. r) APIOperationzeCould not import langchain_community.tools. Please install it with `pip install langchain-community`.NcdSNr%r%r-r+z+openapi_spec_to_openai_fn..zs4r-c,i|]}|j|jf|Sr%)rTparam_in)r'rVs r+ z-openapi_spec_to_openai_fn..~s0   ()QVQZ !   r-r@headerscookiesr)queryheadercookiepathTrDzapplication/jsonrRdataanyOfrF)rGrH)rTrQ parameters)methodrrTrfn_argsr dict | Nonetimeout int | Nonekwargsrrc |d} |d}|di}t||}d|vr8t|dtrt j|d|d<i||} |%d| vr| d|n|| d<|%d| vr| d|n|| d<tj||fi| d|iS)Nrorrrlrfr@rs) popr?r3rrRdumpsupdaterequestsrequest) rTrqrfr@rsrurorr_kwargs_name_to_call_maps r+default_call_apiz3openapi_spec_to_openai_fn..default_call_apis#4(2%e,kk-44 #{++ W  GFOT!B!B "j99GFO'W''  G## "))'2222%, "  7""!((0000$*!HHwHHHHHHr-)NNrp)rTrrqrrfrrr@rrrsrtrurrr)langchain_community.toolsr^ ImportErrorpathsget_parameters_for_pathget_methods_for_path get_operationcopyget_parameters_for_operationrTrcrr4r5rUrXget_request_body_for_operationrNrPrKrRrSlenrLrMrOfrom_openapi_spec operation_idrQbase_urlrk)rBr^emsg functionsrkrro request_argsop op_paramsr9params_by_typename_locrVparam_loc_to_arg_name param_locarg_name request_body media_types media_typemedia_type_objectrW schema_dictkeyapi_opfnr~r}s @r+openapi_spec_to_openai_fnr\s(&::::::: &&& H #A% & : <<I 77  -1-I-I$-O-O   //553 3 FL##D&11B#((**I::2>> @ @:? 5:u~677(..N(00 6 6 !x{+2215555!##% %% ! (='B'B'D'D  # 8!),-K&y1..L* >>rBBL Q 4 Q 5A5I5O5O5Q5Q1J 1(:!%1B1T!U!U26*"KKTK::33 J/{##q((.24 8I8I8K8K3L3L.M.M+J $.2D$D$D&&&C(3L%%%%)),3T+:L:L:N:N5O5O+PL(!33D$GGF+%1$".B   R  4-- bj ) )a3 p $" IIIIIII6 & &&s +&+ceZdZUdZded< dZded< dZded< eedd Z eedd Z dddZ d S)SimpleRequestChainz5Chain for making a simple request to an API endpoint.rrequest_methodresponser output_keyfunction input_keyr list[str]c|jgSr`)rselfs r+ input_keyszSimpleRequestChain.input_keyssr-c|jgSr`)rrs r+ output_keyszSimpleRequestChain.output_keyss  r-Ninputsdict[str, Any] run_manager!CallbackManagerForChainRun | Nonec |ptj}||jd}||jd}t |d}t t j|dd}d|d|z}|||||} | j tj j kr,| j d| j d |d |d d } nY | } nC#t$r | j} Yn0t"$r$t$d | j} YnwxYw|j| iS)z2Run the logic of this chain and return the output.rT argumentsgreen)indentzCalling endpoint z with arguments: z: z For z Called with args: r@r.z)Unexpected error parsing response as JSON)r get_noop_managerrrwrrRrxon_textr status_coderzcodesokreasongetrtext Exception_logger exceptionr) rrr _run_managerrTargs _pretty_name _pretty_args_text api_responsers r+_callzSimpleRequestChain._calls #S&@&Q&S&S dn%))&11dn%))+66'g66 ' 4(B(B(BGLL DLDDD|SU###!%!4!4T4!@!@  #x~'8 8 8+>>|/B>>>>%)XXh%;%;>> H  -',,.." - - -', - - -!!"MNNN', -**sDE-*EE)rrr`)rrrrrr) __name__ __module__ __qualname____doc____annotations__rrpropertyrrrrr%r-r+rrs @?/ J    3I2     XX  !!!XX! :>+++++++r-rz0.2.13FOpenAPISpec | strllmBaseLanguageModel | NonepromptBasePromptTemplate | None request_chain Chain | Nonellm_chain_kwargsrrverboseboolrfrurrc  ddlm} n$#t$r} d} t| | d} ~ wwxYwt|tr| j| j| jfD]K} | |}n<#t$rt$r%t d| j dYHwxYwt|trd|} t| t|\} |sd } t| |ptjd }t!d||d | it#d d|d|pi}|pt%fd|}t'd||g|jdg|d|S)aCreate a chain for querying an API from a OpenAPI spec. !!! warning "Deprecated" This function and all related utilities in this module are deprecated. Use LLM tool calling features directly with an HTTP client instead. Args: spec: OpenAPISpec or url/file/text string corresponding to one. llm: language model, should be an OpenAI function-calling model. prompt: Main prompt template to use. request_chain: Chain for taking the functions output and executing the request. params: Request parameters. headers: Request headers. verbose: Whether to run the chain in verbose mode. llm_chain_kwargs: LLM chain additional keyword arguments. **kwargs: Additional keyword arguments to pass to the chain. rrzqCould not import langchain_community.utilities.openapi. Please install it with `pip install langchain-community`.Nz$Parse spec failed for OpenAPISpec.%sT)exc_infoz!Unable to parse spec from source zmMust provide an LLM for this chain.For example, from langchain_openai import ChatOpenAI model = ChatOpenAI() z>Use the provided API's to respond to this user query: {query}rF) args_onlyr)rr llm_kwargs output_parserrrc"||S)N)rfr@r%)rTr call_api_fnrfr@s r+raz#get_openapi_chain..as$++   + + + r-)rrr)chainsinput_variablesoutput_variablesrr%)%langchain_community.utilities.openapirrr3rfrom_url from_file from_textrrdebugr ValueErrorrr from_templaterr rrr)rBrrrrrrfr@rurrr conversion openai_fns llm_chainrs `` @r+get_openapi_chainrsTD&EEEEEEE &&& H #A% & $"   !  !   J !z$''       :'!  dC  "rs""""""  ######$$$$$$%%%%%%%%******??????<<<<<<TTTTTTIIIIIIII777777........&&&&&&//////111111??????+AAAAAA****** ' H % %0#$#$#$#$LNNNN"    ! p'p'p'  p'f    ! 2+2+2+2+2+2+2+  2+j    !%)(,"&$(\\\\  \\\r-