a bg@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<ddddZed!d d d ddddddZ e d"d d d ddddddZ e d#d dd ddddddZ e d$d dd ddddddZ e d%d dd dddddd 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_toolszList[BaseTool])returncCs t|jS))Get the tools for all the API operations.)listr)selfr|/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/agent_toolkits/nla/toolkit.py get_tools"szNLAToolkit.get_toolsNFrr zOptional[Requests]boolrz List[NLATool])llmspecrequestsverbosekwargsrc KsV|js gSg}|jD]<}||D],}tjf||||||d|}||q"q|S)r)rpathmethodrrr)pathsZget_methods_for_pathr Zfrom_llm_and_methodappend) rrrrrhttp_operation_toolsrrZ endpoint_toolrrr_get_http_operation_tools&s"   z$NLAToolkit._get_http_operation_toolscKs$|jf||||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)r)r#)clsrrrrrr"rrrfrom_llm_and_spec@s zNLAToolkit.from_llm_and_specstr)r open_api_urlrrrrcKs$t|}|jf||||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$)r from_urlr&)r%rr(rrrrrrrfrom_llm_and_urlZs  zNLAToolkit.from_llm_and_urlr )r ai_pluginrrrrcKs(t|jj}|jf||||d|S)0Instantiate the toolkit from an OpenAPI Spec URLr$)r r)apiurlr&)r%rr+rrrrrrrfrom_llm_and_ai_pluginus z!NLAToolkit.from_llm_and_ai_plugin)r ai_plugin_urlrrrrcKs$t|}|jf||||d|S)r,)rr+rr)r r)r/)r%rr0rrrpluginrrrfrom_llm_and_ai_plugin_urls z%NLAToolkit.from_llm_and_ai_plugin_url)NF)NF)NF)NF)NF)__name__ __module__ __qualname____doc__r r__annotations__r staticmethodr# classmethodr&r*r/r2rrrrrs. rN) __future__rtypingrrrrZlangchain_core.language_modelsrZlangchain_core.toolsrZlangchain_core.tools.baser Zpydanticr Z+langchain_community.agent_toolkits.nla.toolr Z5langchain_community.tools.openapi.utils.openapi_utilsr Z langchain_community.tools.pluginr Z&langchain_community.utilities.requestsrrrrrrs