Ngj:rddlmZddlZddlZddlmZddlmZmZm 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'd9dZ(d:dZ)d;d!Z*dd8Z-dS)?) annotationsN) defaultdict) TYPE_CHECKINGAnyCallableDictListOptionalTupleUnion) deprecated)CallbackManagerForChainRun)BaseLanguageModel)JsonOutputFunctionsParser)BasePromptTemplateChatPromptTemplate)get_colored_text)Response)Chain)LLMChain)SequentialChain OpenAPISpec) Parameteror prefer_shortboolreturn Optional[str]cZt|dd}t|dd}|r|p|S|p|S)Nsummary description)getattr)rrr!r"s e/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/chains/openai_functions/openapi.py_get_descriptionr%sBaD))G!]D11K&%+%  !'!urlstr path_paramsdictcT tjd|}i}|D]}|dd}||}t |t r|ddkr)|ddkrdnd}d||z}n|ddkr/|ddkr|d nd}|d ||z}nd|}nt |tro|ddkrd nd fd |D} |ddkrd}d}n|ddkrd}d}nd}d }||| z }n)|ddkrd|}n|ddkr d|d |}n|}|||<|j d i|S) Nz{(.*?)}z.;*r.,;=cDg|]\}}||fS)join).0kvkv_seps r$ z_format_url..3s-CCCtq!v{{Aq6**CCCr&r3) refindalllstriprstrip isinstancelistr4r*itemsformat) r'r)expected_path_param new_paramsparam clean_paramvalsepnew_valkv_strsr8s @r$ _format_urlrK"s*Z55J$ $ $ll4((//44 +& c4  Qx3"2Y#--cc3 -qS+09+;+;''''(+++chhsmm;((3-- T " " !"I,,SS#FCCCCsyy{{CCCGQx3qS sxx(( (GGQx3#c))qS1k11C11# 5 3: # # # ##r&paramsList[Parameter]specrci}g}|D]}|jr||j}nFt|jdj}||}|jr|js |j|_tj|d||j <|j r| |j d||dS)NrT exclude_noneobject)type propertiesrequired) param_schema get_schemar@contentvaluesmedia_type_schemar"jsonloadsnamerUappend)rLrNrTrUpschemarZs r$_openapi_params_to_json_schemaraIsJH  $ $ > 8__Q^44FF $QY%5%5%7%7 8 8 ; M __%677F = /!3 /!"F !Z  (F(FGG 16 : $ OOAF # # #JH M MMr&%Tuple[List[Dict[str, Any]], Callable]c ddlm}n#t$rtdwxYw|jsgdfSg}i|jD]}d||D}||D]T}i}|||}|}||D]} | || j | j f<tt} | D]&\} } | | d| 'ddd d d } | D]&\}}| |rt| ||||<'||}|r|jri}|j D]Q\}}|jrE||j}t)j|d ||<Rt-|dkr:t| d\}}|dkrdnd}|||<n9t-|dkr&dt|i|d<||||}|j|jd|dd}||||j|jzd|d<V d d!fd }||fS)"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`.cdSNr3r3r&r$z+openapi_spec_to_openai_fn..ps4r&c,i|]}|j|jf|Sr3)r]param_in)r5r_s r$ z-openapi_spec_to_openai_fn..ts0   ()QVQZ !   r&rLheaderscookiesr))queryheadercookiepathTrPzapplication/jsonr[dataanyOfrR)rSrT)r]r" parameters)methodr'r]Nr(fn_argsr*Optional[dict]kwargsrrc |d} |d}|di}t||}d|vr8t|dtrt j|d|d<i||}|%d|vr|d|n||d<|%d|vr|d|n||d<tj||fi|S)Nrur'r)rrrlrL) poprKr?r*r[dumpsupdaterequestsrequest) r]rvrlrLrxrur'r)_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$*!77w777r&)NN) r]r(rvr*rlrwrLrwrxrrr)langchain_community.toolsrd ImportErrorpathsget_parameters_for_pathget_methods_for_path get_operationcopyget_parameters_for_operationr]rirr@rAr^raget_request_body_for_operationrXrZrWr[r\lenrYfrom_openapi_spec operation_idr"base_urlrq)rNrd functionsrqr)ru request_argsop op_paramsrEparams_by_typename_locr_param_loc_to_arg_name param_locarg_name request_body media_types media_typemedia_type_objectr` schema_dictkeyapi_opfnrrs @r$openapi_spec_to_openai_fnrZs :::::::     H    : <<I 66  -1-I-I$-O-O   //552 2 FL##D&11B#((**I::2>> @ @:? 5:u~677(..N(00 6 6 !x{+2215555!##% %% ! (='B'B'D'D  # 8!),-K&y14..L* >>rBBL Q 4 Q 5A5I5O5O5Q5Q1J 1(:!%1B1T!U!U26*"KKTK::33 J/{##q((.2;3D3D3F3F.G.G.J+J $.2D$D$D&&&C(3L%%%%)),3T+:L:L:N:N5O5O+PL(!33D$GGF+%1$".B   R  4-- bj ) )_2 n#'!% 88888884 & &&s $c|eZdZUdZded< dZded< dZded< edd Zedd Z dddZ d S)SimpleRequestChainz5Chain for making a simple request to an API endpoint.rrequest_methodresponser( output_keyfunction input_keyr List[str]c|jgSrf)rselfs r$ input_keyszSimpleRequestChain.input_keyss r&c|jgSrf)rrs r$ output_keyszSimpleRequestChain.output_keyss   r&NinputsDict[str, Any] run_manager$Optional[CallbackManagerForChainRun]cd|ptj}||jd}||jd}t |d}t t j|dd}d|d|z}|||||} | j dkr3| j d | j d |d zd | d dz} n- | } n#t$r | j } YnwxYw|j| iS)z2Run the logic of this chain and return the output.r] argumentsgreen)indentzCalling endpoint z with arguments: z: z For  zCalled with args: rLr:)rget_noop_managerrrzrr[r{on_textr status_codereasonget Exceptiontextr) rrr _run_managerr]args _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!@!@  #s * *+DD|/BDD"4"""#>txx'<'<>>? H  -',,.. - - -', -**s=DD&%D&)rrrf)rrrrrr) __name__ __module__ __qualname____doc____annotations__rrpropertyrrrr3r&r$rrs??/ J    3I2    X !!!X! =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)sincemessageremovalFUnion[OpenAPISpec, str]llmOptional[BaseLanguageModel]promptOptional[BasePromptTemplate] request_chainOptional[Chain]llm_chain_kwargsOptional[Dict]verboserlrxrc x ddlm} n"#t$r} td| d} ~ wwxYwt|trj| j| j| jfD].} | |}n#t$r} | d} ~ wt$rY+wxYwt|trtd|t|\} |std|ptj d}td||d| itd d |d |pi} |ptfd |}t!d| |g| jdg|d|S)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)rr llm_kwargs output_parserrrc"||S)N)rlrLr3)r]r call_api_fnrlrLs r$rgz#get_openapi_chain..s!++ $+ + + r&)rrr)chainsinput_variablesoutput_variablesrr3)%langchain_community.utilities.openapirrr?r(from_url from_file from_textr ValueErrorrr from_templaterrrrr)rNrrrrrrlrLrxre conversion openai_fns llm_chainrs `` @r$get_openapi_chainrs-TEEEEEEE  H    $I   !  !  J  !z$''        dC  IGGGHH H7==J   #   )7JF ,/%@@@   !rI"%7       &&&M  =)!,$       s. +&+ A'' B1A33 BB)rrrrrr)r'r(r)r*rr()rLrMrNrrr*)rNrrrb)NNNNFNN)rNrrrrrrrrrrrrlrrLrrxrrr). __future__rr[r; collectionsrtypingrrrrr r r r r}langchain_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%rKrarrrr3r&r$rsZ"""""" ######SSSSSSSSSSSSSSSSSSSS******??????<<<<<<TTTTTTIIIIIIII777777''''''))))))777777+AAAAAA******""""$$$$$$$$NNNNN"k'k'k'k'\++++++++++++++++\   } (,+/%)'+"!VVVVVVVr&