o ȳg@sddlmZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZGd d d e Zd S)) annotations)AnyListOptionalSequence)BaseLanguageModel)BaseTool) BaseToolkit)Field)NLATool) OpenAPISpec)AIPlugin)Requestsc@seZdZUdZedZded< d%ddZe d&d'ddZ e d&d(ddZ e d&d)ddZ e d&d*d d!Z e d&d+d#d$Zd S), NLAToolkitaNatural Language API Toolkit. *Security Note*: This toolkit creates tools that enable making calls to an Open API compliant API. The tools created by this toolkit may be able to make GET, POST, PATCH, PUT, DELETE requests to any of the exposed endpoints on the API. Control access to who can use this toolkit. See https://python.langchain.com/docs/security for more information. .zSequence[NLATool] nla_toolsreturnList[BaseTool]cCs t|jS))Get the tools for all the API operations.)listr)selfrj/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/agent_toolkits/nla/toolkit.py get_tools"s zNLAToolkit.get_toolsNFllmrspecr requestsOptional[Requests]verboseboolkwargsr List[NLATool]c KsV|jsgSg}|jD]}||D]}tjd||||||d|}||qq |S)r)rpathmethodrrrNr)pathsget_methods_for_pathr from_llm_and_methodappend) rrrrrhttp_operation_toolsr!r" endpoint_toolrrr_get_http_operation_tools&s$    z$NLAToolkit._get_http_operation_toolscKs$|jd||||d|}||dS)aInstantiate the toolkit by creating tools for each operation. Args: llm: The language model to use. spec: The OpenAPI spec. requests: Optional requests object. Default is None. verbose: Whether to print verbose output. Default is False. kwargs: Additional arguments. Returns: The toolkit. rrrr)rNr)r))clsrrrrrr'rrrfrom_llm_and_spec@s  zNLAToolkit.from_llm_and_spec open_api_urlstrcK$t|}|jd||||d|S)aInstantiate the toolkit from an OpenAPI Spec URL. Args: llm: The language model to use. open_api_url: The URL of the OpenAPI spec. requests: Optional requests object. Default is None. verbose: Whether to print verbose output. Default is False. kwargs: Additional arguments. Returns: The toolkit. r*Nr)r from_urlr,)r+rr-rrrrrrrfrom_llm_and_urlZs zNLAToolkit.from_llm_and_url ai_pluginr cKs(t|jj}|jd||||d|S)0Instantiate the toolkit from an OpenAPI Spec URLr*Nr)r r0apiurlr,)r+rr2rrrrrrrfrom_llm_and_ai_pluginus z!NLAToolkit.from_llm_and_ai_plugin ai_plugin_urlcKr/)r3)rr2rrNr)r r0r6)r+rr7rrrpluginrrrfrom_llm_and_ai_plugin_urls z%NLAToolkit.from_llm_and_ai_plugin_url)rr)NF) rrrr rrrrrrrr ) rrrr rrrrrrrr) rrr-r.rrrrrrrr) rrr2r rrrrrrrr) rrr7r.rrrrrrrr)__name__ __module__ __qualname____doc__r r__annotations__r staticmethodr) classmethodr,r1r6r9rrrrrs2  rN) __future__rtypingrrrrlangchain_core.language_modelsrlangchain_core.toolsrlangchain_core.tools.baser pydanticr +langchain_community.agent_toolkits.nla.toolr 5langchain_community.tools.openapi.utils.openapi_utilsr langchain_community.tools.pluginr &langchain_community.utilities.requestsrrrrrrs