File: //usr/local/lib/python3.10/dist-packages/langchain/agents/chat/__pycache__/base.cpython-310.pyc
o
���g� � @ s� d dl mZmZmZmZmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d dlmZ d dlmZmZmZ d dlmZ d d lmZ d d
lmZ d d lmZm Z d d l m Z d d
l m Z m!Z!m"Z"m#Z# d dl$m%Z% d dl&m'Z' ededd�G dd� de��Z(dS )� )�Any�List�Optional�Sequence�Tuple)�
deprecated)� AgentAction)�BaseCallbackManager)�BaseLanguageModel)�BasePromptTemplate)�ChatPromptTemplate�HumanMessagePromptTemplate�SystemMessagePromptTemplate)�BaseTool)�Field)�AGENT_DEPRECATION_WARNING)�Agent�AgentOutputParser��ChatOutputParser)�FORMAT_INSTRUCTIONS�
HUMAN_MESSAGE�SYSTEM_MESSAGE_PREFIX�SYSTEM_MESSAGE_SUFFIX)�validate_tools_single_input)�LLMChainz0.1.0z1.0)�message�removalc sl e Zd ZU dZeed�Zeed< e de
fdd��Z e de
fdd��Z d e
eee
f de
f� fd
d � Zed edefd
d��Zedee ddf� fdd� �Ze de
e
fdd��Zeeeeedfdee de
de
de
de
de e
e
de fdd��Z eddeeeedfd e dee d e e de e de
de
de
de
de e
e
d ede!fd d ��Z"e de
fd d!��Z#� Z$S )"� ChatAgentz Chat Agent.)�default_factory�
output_parser�returnc C � dS )z&Prefix to append the observation with.z
Observation: � ��selfr# r# �E/usr/local/lib/python3.10/dist-packages/langchain/agents/chat/base.py�observation_prefix( � z ChatAgent.observation_prefixc C r" )z#Prefix to append the llm call with.zThought:r# r$ r# r# r& �
llm_prefix- r( zChatAgent.llm_prefix�intermediate_stepsc s0 t � �|�}t|t�std��|rd|� �S |S )Nz*agent_scratchpad should be of type string.zhThis was your previous work (but I haven't seen any of it! I only see what you return as final answer):
)�super�_construct_scratchpad�
isinstance�str�
ValueError)r% r* �agent_scratchpad�� __class__r# r& r, 2 s
��z ChatAgent._construct_scratchpad�kwargsc K s t � S �Nr )�clsr3 r# r# r& �_get_default_output_parserA � z$ChatAgent._get_default_output_parser�toolsNc s t � �|� t| j|d� d S )N)�
class_namer8 )r+ �_validate_toolsr �__name__)r5 r8 r1 r# r& r: E s zChatAgent._validate_toolsc C s dgS )Nz Observation:r# r$ r# r# r& �_stopJ r7 zChatAgent._stop�system_message_prefix�system_message_suffix�
human_message�format_instructions�input_variablesc C sv d� dd� |D ��}d� dd� |D ��}|j|d�}d� ||||g�} t�| �t�|�g}
|du r5d d
g}t||
d �S ) ag Create a prompt from a list of tools.
Args:
tools: A list of tools.
system_message_prefix: The system message prefix.
Default is SYSTEM_MESSAGE_PREFIX.
system_message_suffix: The system message suffix.
Default is SYSTEM_MESSAGE_SUFFIX.
human_message: The human message. Default is HUMAN_MESSAGE.
format_instructions: The format instructions.
Default is FORMAT_INSTRUCTIONS.
input_variables: The input variables. Default is None.
Returns:
A prompt template.
�
c S s g | ] }|j � d |j� ��qS )z: )�name� description��.0�toolr# r# r& �
<listcomp>i s z+ChatAgent.create_prompt.<locals>.<listcomp>z, c S � g | ]}|j �qS r# �rC rE r# r# r& rH j � )�
tool_namesz
N�inputr0 )rA �messages)�join�formatr �
from_templater
r ) r5 r8 r= r> r? r@ rA � tool_stringsrL �templaterN r# r# r& �
create_promptN s �� � zChatAgent.create_prompt�llm�callback_managerc
K s^ | � |� | j|||||| d�} t|| |d�} dd� |D �}
|p#| �� }| d| |
|d�|
��S )a= Construct an agent from an LLM and tools.
Args:
llm: The language model.
tools: A list of tools.
callback_manager: The callback manager. Default is None.
output_parser: The output parser. Default is None.
system_message_prefix: The system message prefix.
Default is SYSTEM_MESSAGE_PREFIX.
system_message_suffix: The system message suffix.
Default is SYSTEM_MESSAGE_SUFFIX.
human_message: The human message. Default is HUMAN_MESSAGE.
format_instructions: The format instructions.
Default is FORMAT_INSTRUCTIONS.
input_variables: The input variables. Default is None.
kwargs: Additional keyword arguments.
Returns:
An agent.
)r= r> r? r@ rA )rU �promptrV c S rI r# rJ rE r# r# r& rH � rK z0ChatAgent.from_llm_and_tools.<locals>.<listcomp>)� llm_chain�
allowed_toolsr Nr# )r: rT r r6 )r5 rU r8 rV r r= r> r? r@ rA r3 rW rX rL �_output_parserr# r# r& �from_llm_and_tools| s.
"�� ��z ChatAgent.from_llm_and_toolsc C s t �r4 )r/ r$ r# r# r& � _agent_type� s zChatAgent._agent_type)%r; �
__module__� __qualname__�__doc__r r r r �__annotations__�propertyr. r' r) r r r r, � classmethodr r6 r r r: r< r r r r r r rT r
r r r[ r\ �
__classcell__r# r# r1 r& r s�
�� ������
� �-�������� �
� � �8r N))�typingr r r r r �langchain_core._apir �langchain_core.agentsr �langchain_core.callbacksr � langchain_core.language_modelsr
�langchain_core.promptsr �langchain_core.prompts.chatr r
r �langchain_core.toolsr �pydanticr �langchain._api.deprecationr �langchain.agents.agentr r �#langchain.agents.chat.output_parserr � langchain.agents.chat.promptr r r r �langchain.agents.utilsr �langchain.chains.llmr r r# r# r# r& �<module> s* �