9|hk:ddlmZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZddlZddlmZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd l m!Z!ddl"m#Z#er ddl$m%Z%ddl&m'Z'ddZ(ddZ)ddZ* ddZ+GddeZ,eddd d d!dZ-y)") annotationsN) defaultdict) TYPE_CHECKINGAnyCallableDictListOptionalTupleUnion) deprecated)CallbackManagerForChainRun)BaseLanguageModel)JsonOutputFunctionsParser)BasePromptTemplateChatPromptTemplate)get_colored_text)Response)Chain)LLMChain)SequentialChain OpenAPISpec) ParametercRt|dd}t|dd}|r|xs|S|xs|S)Nsummary description)getattr)o prefer_shortrrs h/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain/chains/openai_functions/openapi.py_get_descriptionr"s7aD)G!]D1K%+%  !'!c 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 r! _format_urlrD"s**Z5J$ $ll4(//4 +& c4 Qx3"2Y#-c3 -qS+09+; Q'(M+chhsm;((3- T "!"I,S#F7:yy{Ctq!v{{Aq6*CGCQx3qS sxx( (GQx3cU)qSk]!C51# 5A $B 3:: # ##'Ds-E;ci}g}|D]}|jr|j|j}nAt|jj dj }|j|}|j r|j s|j |_tj|jd||j<|js|j|jd||dS)NrT exclude_noneobject)type propertiesrequired) param_schema get_schemar2contentvaluesmedia_type_schemarjsonloadsnamerKappend)paramsspecrJrKpschemarPs r!_openapi_params_to_json_schemarYIsJH  $ >>__Q^^4F $QYY%5%5%7 8 ; M M __%67F ==!3!3!"F !ZZ  (FG 166 :: OOAFF # $JH MMr#c ddlm}|jsgdfSg}i|jD]x}|j |Dcic]}|j |j f|}}|j|D]+}i}|j||}|j} |j|D]} | | | j | j f<tt} | jD]\} }| | dj|dddd d } | jD]\}}| |s t| ||||<|j!|}|r|j"ri}|j"jD]U\}}|j$s|j'|j$}t)j*|j)d ||<Wt-|dk(r.t|jd\}}|d k(rdnd}|||<n,t-|dkDrdt|j/i|d<|j1|||}|j2|j4d|dd}|j|||j6|j8zd|d<.{ d dfd }||fS#t$r tdwxYwcc}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`.cyNr,r,r#r!z+openapi_spec_to_openai_fn..psr#rUheaderscookiesr8)queryheadercookiepathTrFzapplication/jsonrQdataanyOfrH)rIrJ)rSr parameters)methodr7rSc~ |d} |d}|jdi}t||}d|vr.t|dtrt j |d|d<i||}|d|vr|dj |n||d<|d|vr|dj |n||d<tj||fi|S)Nrir7r8rfr`rU) poprDr1r4rQdumpsupdaterequestsrequest) rSfn_argsr`rUkwargsrir7r8_kwargs_name_to_call_maps r!default_call_apiz3openapi_spec_to_openai_fn..default_call_apis#4(2%e,kk-4 #{+ W GFOT!B"jj9GFO'W''  G# "))'2%, "  7"!((0$*!7w77r#)NN) rSstrrpr4r`Optional[dict]rUrvrqrreturnr)langchain_community.toolsr[ ImportErrorpathsget_parameters_for_pathrSparam_inget_methods_for_path get_operationcopyget_parameters_for_operationrr2r5rTrYget_request_body_for_operationrNrPrMrQrRlenrOfrom_openapi_spec operation_idrbase_urlre)rVr[ functionsrerWr8ri request_argsop op_paramsr;params_by_typename_locparam_loc_to_arg_name param_locarg_name request_body media_types media_typemedia_type_objectrX schema_dictkeyapi_opfnrtrss @r!openapi_spec_to_openai_fnrZsW : ::<I 6-1-I-I$-O ()QVVQZZ ! #  //52 FL##D&1B#((*I::2> @:? 5::u~~67 @(.N(0 6 !x{+2215 6"##% % ! (='B'B'D # 8!),-K&y14.L*   >>rBL 4 4 5A5I5I5O5O5Q1J 1(::!%1B1T1T!U26**"KKTK:3 J/ {#q(.2;3D3D3F.G.J+J $.2D$D&&C(3L%%),3T+:L:L:N5O+PL(!33D$GF++%11$".B   R  4- bj )_2  6v#'!% 888 8 8  8 84 & &&y   H    sJ> K>KcxeZdZUdZded< dZded< dZded< ed d Zed d Z d dd Z y )SimpleRequestChainz5Chain for making a simple request to an API endpoint.rrequest_methodresponseru output_keyfunction input_keyc|jgSr])rselfs r! input_keyszSimpleRequestChain.input_keyssr#c|jgSr])rrs r! output_keyszSimpleRequestChain.output_keyss  r#Nc`|xstj}||jjd}||jjd}t |d}t t j |dd}d|d|z}|j||j||} | jdk7r9| jd | jd |d zd |jd dz} n | j } |j| iS#t$r| j} Y%wxYw)z2Run the logic of this chain and return the output.rS argumentsgreen)indentzCalling endpoint z with arguments: z: z For  zCalled with args: rUr+)rget_noop_managerrrkrrQrlon_textr status_codereasonget Exceptiontextr) rinputs run_manager _run_managerrSargs _pretty_name _pretty_args_text api_responsers r!_callzSimpleRequestChain._calls= #S&@&Q&Q&S dnn%))&1dnn%))+6'g6 ' 4(BGL #L>1CD|SU#!%!4!4T4!@  # #s *++,B|/B/B.CD4&"#&txx"'=&>?@   -',,.** -',, -s7DD-,D-)rwz List[str]r])rDict[str, Any]rz$Optional[CallbackManagerForChainRun]rwr) __name__ __module__ __qualname____doc____annotations__rrpropertyrrrr,r#r!rrsu?/ J 3Is2   !! =A++:+  +r#rz0.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|trP| j | j | jfD] } | |}nt|trtd|t|\} |s td|xstjd}td||d| itd d |d |xsi} |xstfd |}t!d| |g| j"dg|d|S#t$r} td| d} ~ wwxYw#t$r} | d} ~ wt$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)r`rUr,)rSr call_api_fnr`rUs r!r^z#get_openapi_chain..s+ $+ r#)rrr)chainsinput_variablesoutput_variablesrr,)%langchain_community.utilities.openapirryr1rufrom_url from_file from_textr ValueErrorrr from_templaterrrrr)rVrr request_chainllm_chain_kwargsrr`rUrqre conversion openai_fns llm_chainrs `` @r!get_openapi_chainrsTE $   ! !  ! ! J  !$'  dC @GH H7=J  #  )77JF ,/%@   !rI"%7  &M  =)!,,$     [  H     s5C4D4 D= D  D D+D D+*D+)rrr boolrwz Optional[str])r7rur8r4rwru)rUzList[Parameter]rVrrwr4)rVrrwz%Tuple[List[Dict[str, Any]], Callable])NNNNFNN)rVzUnion[OpenAPISpec, str]rzOptional[BaseLanguageModel]rzOptional[BasePromptTemplate]rzOptional[Chain]rOptional[Dict]rrr`rrUrrqrrwr). __future__rrQr- collectionsrtypingrrrrr r r r rnlangchain_core._apir langchain_core.callbacksrlangchain_core.language_modelsr.langchain_core.output_parsers.openai_functionsrlangchain_core.promptsrrlangchain_core.utils.inputrrlangchain.chains.baserlangchain.chains.llmrlangchain.chains.sequentialrrropenapi_pydanticrr"rDrYrrrr,r#r!rs%" #SSS*?<TI7')7A*"$$NN"k' k'*k'\++++\  } (,+/%)'+"!V !V $V )V# V % V  VV VVVVr#