h:xddlmZddlZddlZddlmZddlmZmZm Z m Z m Z ddl Z ddl mZddlmZddlmZddlmZdd lmZmZdd lmZdd l mZdd lmZdd lmZddlm Z er ddl!m"Z"ddl#m$Z$ddZ%ddZ& ddZ'GddeZ(eddd d ddZ)y) ) annotationsN) defaultdict) TYPE_CHECKINGAnyCallableOptionalUnion) deprecated)CallbackManagerForChainRun)BaseLanguageModel)JsonOutputFunctionsParser)BasePromptTemplateChatPromptTemplate)get_colored_text)Response)Chain)LLMChain)SequentialChain OpenAPISpec) Parameterc tjd|}i}|D]J}|jdjd}||}t |t rj|ddk(r!|ddk(rdnd}d|j |z}n|ddk(r'|ddk(r|d nd}|d |j |z}ndj |}nt |trr|ddk(rd nd} |jD cgc]\} } | j | | f} } } |ddk(rd}d}n|ddk(rd}d}nd}d }||j | z }n!|ddk(rd|}n|ddk(r d|d |}n|}|||<M|jd i|Scc} } w) Nz{(.*?)}z.;*r.,;=) refindalllstriprstrip isinstancelistjoindictitemsformat) url path_paramsexpected_path_param new_paramsparam clean_paramvalsepnew_valkv_sepkvkv_strss g/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langchain/chains/openai_functions/openapi.py _format_urlr9s**Z5J$$ll4(//4 +& c4 Qx3"2Y#-c3 -qS+09+; Q'(M+chhsm;((3- T "!"I,S#F7:yy{Ctq!v{{Aq6*CGCQx3qS sxx( (G 1X_#iG 1X_+au-GG# 5?$@ 3:: # ##%Ds-E;ci}g}|D]}|jr|j|j}nGtt|jj j }|j|}|jr|js|j|_tj|jd||j<|js|j|jd||dS)NT exclude_noneobject)type propertiesrequired) param_schema get_schemanextitercontentvaluesmedia_type_schema descriptionjsonloadsnamer@append)paramsspecr?r@pschemarGs r8_openapi_params_to_json_schemarQ@sJH  $ >>__Q^^4F $T!))*:*:*<%= > P P __%67F ==!3!3!"F !ZZ  (FG 166 :: OOAFF # $JH MMc ddlm}|jsgdfSg}i|jD]~}|j |Dcic]}|j |j f|}}|j|D]1}i} |j||} |j} |j| D]} | | | j | j f<tt} | jD]\}}| |dj|ddd d d }|jD]\}}| |s t| ||| |<|j!| }|r|j"ri}|j"jD]U\}}|j$s|j'|j$}t)j*|j)d ||<Wt-|dk(r4t/t1|j\}}|dk(rdnd}|| |<n,t-|dkDrdt|j3i| d<|j5|||}|j6|j8d| dd}|j|||j:|j<zd|d<4 d dfd }||fS#t$r}d}t||d}~wwxYwcc}w)a4Convert 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`.NcyNr r rRr8z+openapi_spec_to_openai_fn..hsrRrMheaderscookiesr,)queryheadercookiepathTr;zapplication/jsonrIdataanyOfr=)r>r?)rKrH parameters)methodr+rKc |d} |d}|jdi}t||}d|vr.t|dtrt j |d|d<i||} |d| vr| dj |n|| d<|d| vr| dj |n|| d<tj||fi| d|iS)Nrbr+r,r_rYrMtimeout) popr9r%r(rIdumpsupdaterequestsrequest) rKfn_argsrYrMrdkwargsrbr+r,_kwargs_name_to_call_maps r8default_call_apiz3openapi_spec_to_openai_fn..default_call_apis#4(2%e,kk-4 #{+ W GFOT!B"jj9GFO'W''  G# "))'2%, "  7"!((0$*!HwHHHrR)NN)rKstrrjr(rYOptional[dict]rMrqrdz Optional[int]rkrreturnr)langchain_community.toolsrT ImportErrorpathsget_parameters_for_pathrKparam_inget_methods_for_path get_operationcopyget_parameters_for_operationrr&r)rLrQget_request_body_for_operationrErGrBrIrJlenrCrDrFfrom_openapi_spec operation_idrHbase_urlr^)rNrTemsg functionsr^rOr,rb request_argsop op_paramsr/params_by_typename_locparam_loc_to_arg_name param_locarg_name request_body media_types media_typemedia_type_objectrP schema_dictkeyapi_opfnrnrms @r8openapi_spec_to_openai_fnrQsu&: ::<I 7-1-I-I$-O ()QVVQZZ ! #  //53 FL##D&1B#((*I::2> @:? 5::u~~67 @(.N(0 6 !x{+2215 6"##% % ! (='B'B'D # 8!),-K&y1.L*   >>rBL 4 4 5A5I5I5O5O5Q1J 1(::!%1B1T1T!U26**"KKTK:3 J/ {#q(.24 8I8I8K3L.M+J $.2D$D&&C(3L%%),3T+:L:L:N5O+PL(!33D$GF++%11$".B   R  4- bj )a3  7x#'!%!# III I I  I  I I6 & && & H #A% & sK K& K#KK#cxeZdZUdZded< dZded< dZded< ed d Zed d Z d dd Z y )SimpleRequestChainz5Chain for making a simple request to an API endpoint.rrequest_methodresponserp output_keyfunction input_keyc|jgSrV)rselfs r8 input_keyszSimpleRequestChain.input_keyssrRc|jgSrV)rrs r8 output_keyszSimpleRequestChain.output_keyss  rRNc |xstj}||jjd}||jjd}t |d}t t j |dd}d|d|z}|j||j||} | jtjjk7r2| jd| jd |d |jd d } n | j } |j"| iS#t$r| j } Y%wxYw) z2Run the logic of this chain and return the output.rK argumentsgreen)indentzCalling endpoint z with arguments: z: z For z Called with args: rMr)r get_noop_managerrrerrIrfon_textr status_coderhcodesokreasonget Exceptiontextr) rinputs run_manager _run_managerrKargs _pretty_name _pretty_args_text api_responsers r8_callzSimpleRequestChain._calls9 #S&@&Q&Q&S dnn%))&1dnn%))+6'g6 ' 4(BGL #L>1CD|SU#!%!4!4T4!@  # #x~~'8'8 8++,B|/B/B.C%%)XXh%;$<>   -',,.** -',, -sD&&D>=D>)rrz list[str]rV)rdict[str, Any]rz$Optional[CallbackManagerForChainRun]rrr) __name__ __module__ __qualname____doc____annotations__rrpropertyrrrr rRr8rrsu?/ J 3Is2   !! =A++:+  +rRrz0.2.13zThis function is deprecated and will be removed in langchain 1.0. See API reference for replacement: https://api.python.langchain.com/en/latest/chains/langchain.chains.openai_functions.openapi.get_openapi_chain.htmlz1.0)sincemessageremovalc ^ ddlm} t|trR| j | j | jfD] } | |}nt|trd|} t| t|\} |s d} t| |xstjd}td||d| itd d |d |xsi}|xstfd |}t!d||g|j"dg|d|S#t$r} d} t| | d} ~ wwxYw#t$rt$rYwxYw)a Create a chain for querying an API from a OpenAPI spec. Note: this class is deprecated. See below for a replacement implementation. The benefits of this implementation are: - Uses LLM tool calling features to encourage properly-formatted API requests; - Includes async support. .. code-block:: python from typing import Any from langchain.chains.openai_functions.openapi import openapi_spec_to_openai_fn from langchain_community.utilities.openapi import OpenAPISpec from langchain_core.prompts import ChatPromptTemplate from langchain_openai import ChatOpenAI # Define API spec. Can be JSON or YAML api_spec = """ { "openapi": "3.1.0", "info": { "title": "JSONPlaceholder API", "version": "1.0.0" }, "servers": [ { "url": "https://jsonplaceholder.typicode.com" } ], "paths": { "/posts": { "get": { "summary": "Get posts", "parameters": [ { "name": "_limit", "in": "query", "required": false, "schema": { "type": "integer", "example": 2 }, "description": "Limit the number of results" } ] } } } } """ parsed_spec = OpenAPISpec.from_text(api_spec) openai_fns, call_api_fn = openapi_spec_to_openai_fn(parsed_spec) tools = [ {"type": "function", "function": fn} for fn in openai_fns ] prompt = ChatPromptTemplate.from_template( "Use the provided APIs to respond to this user query:\n\n{query}" ) llm = ChatOpenAI(model="gpt-4o-mini", temperature=0).bind_tools(tools) def _execute_tool(message) -> Any: if tool_calls := message.tool_calls: tool_call = message.tool_calls[0] response = call_api_fn(name=tool_call["name"], fn_args=tool_call["args"]) response.raise_for_status() return response.json() else: return message.content chain = prompt | llm | _execute_tool .. code-block:: python response = chain.invoke({"query": "Get me top two posts."}) Args: spec: OpenAPISpec or url/file/text string corresponding to one. llm: language model, should be an OpenAI function-calling model, e.g. `ChatOpenAI(model="gpt-3.5-turbo-0613")`. prompt: Main prompt template to use. request_chain: Chain for taking the functions output and executing the request. rrzqCould not import langchain_community.utilities.openapi. Please install it with `pip install langchain-community`.Nz!Unable to parse spec from source zkMust provide an LLM for this chain.For example, from langchain_openai import ChatOpenAI llm = ChatOpenAI() z>Use the provided API's to respond to this user query: {query}rF) args_onlyr)llmprompt llm_kwargs output_parserrverbosec||S)N)rYrMr )rKr call_api_fnrYrMs r8rWz#get_openapi_chain..s+   + rR)rrr)chainsinput_variablesoutput_variablesrr )%langchain_community.utilities.openapirrtr%rpfrom_url from_file from_textr ValueErrorrr from_templaterr rrr)rNrr request_chainllm_chain_kwargsrrYrMrkrrr conversion openai_fns llm_chainrs `` @r8get_openapi_chainrsT&E$   ! !  ! ! J  !$'  dC 5dVrs" #@@*?<TI7')7A*#$LN"n' n'*n'b++++\  } (,+/%)'+"!\ !\ $\ )\# \ % \  \\ \\\\rR