File: //usr/local/lib/python3.10/dist-packages/langchain/agents/__pycache__/loading.cpython-310.pyc
o
���g� � @ sh d Z ddlZddlZddlmZ ddlmZmZmZm Z ddl
Z
ddl m Z ddl
mZ ddlmZ ddlmZmZ dd lmZ dd
lmZmZ e�e�Zd Z d e d
edee dede eef f
dd�Z e ddd� d d e d
ee deee dede eef f
dd��Z e ddd�de e ef dede eef fdd��Z!de e ef dede eef fd d �Z"dS ) z!Functionality for loading agents.� N)�Path)�Any�List�Optional�Union)�
deprecated)�BaseLanguageModel)�Tool)�BaseMultiActionAgent�BaseSingleActionAgent)�AGENT_TO_CLASS)�
load_chain�load_chain_from_configzHhttps://raw.githubusercontent.com/hwchase17/langchain-hub/master/agents/�config�llm�tools�kwargs�returnc K sJ | � d�}|tvrtd|� d���t| }i | �|�}|j||fi |��S )N�_type�Loading � agent not supported)�popr �
ValueError�from_llm_and_tools)r r r r � config_type� agent_cls�combined_config� r �C/usr/local/lib/python3.10/dist-packages/langchain/agents/loading.py�_load_agent_from_tools s
r z0.1.0z1.0)�removalc K s� d| vrt d��| �dd�}|r*|du rt d��|du r t d��t| ||fi |��S | �d�}|tvr;t d|� d ���t| }d
| v rMt| �d
��| d
< nd | v r[t| �d ��| d
< nt d ��d
| v rkt�d� | d
= i | �|�}|di |��S )a� Load agent from Config Dict.
Args:
config: Config dict to load agent from.
llm: Language model to use as the agent.
tools: List of tools this agent has access to.
kwargs: Additional keyword arguments passed to the agent executor.
Returns:
An agent executor.
Raises:
ValueError: If agent type is not specified in the config.
r z$Must specify an agent Type in config�load_from_llm_and_toolsFNzFIf `load_from_llm_and_tools` is set to True, then LLM must be providedzHIf `load_from_llm_and_tools` is set to True, then tools must be providedr r � llm_chain�llm_chain_pathz<One of `llm_chain` and `llm_chain_path` should be specified.�
output_parserzZCurrently loading output parsers on agent is not supported, will just use the default one.r )r r r r r r
�logger�warning)r r r r �load_from_toolsr r r r r r �load_agent_from_config" s: ��
� r( �pathc K s, t | t�r| �d�rtd��t| fi |��S )aY Unified method for loading an agent from LangChainHub or local fs.
Args:
path: Path to the agent file.
kwargs: Additional keyword arguments passed to the agent executor.
Returns:
An agent executor.
Raises:
RuntimeError: If loading from the deprecated github-based
Hub is attempted.
zlc://z�Loading from the deprecated github-based Hub is no longer supported. Please use the new LangChain Hub at https://smith.langchain.com/hub instead.)�
isinstance�str�
startswith� RuntimeError�_load_agent_from_file)r) r r r r �
load_agent\ s
�r/ �filec K s� ddh}t | t�rt| �}n| }|jdd� dkr3t|��
}t�|�}W d � n1 s-w Y n,|jdd� dkrWt|d��
}t�|�}W d � n1 sQw Y nt d|� d���t
|fi |��S )zLoad agent from file.�json�yaml� N�rz&Unsupported file type, must be one of �.) r* r+ r �suffix�openr1 �loadr2 � safe_loadr r( )r0 r �valid_suffixes� file_path�fr r r r r. v s
�� ��r. )NN)#�__doc__r1 �logging�pathlibr �typingr r r r r2 �langchain_core._apir � langchain_core.language_modelsr �langchain_core.toolsr �langchain.agents.agentr
r �langchain.agents.typesr �langchain.chains.loadingr
r � getLogger�__file__r% �URL_BASE�dictr r( r+ r/ r. r r r r �<module> sl
����
�
���
��
�
9
��
�
��
�