File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/chat_memory.cpython-310.pyc
o
���gw
� @ s� d dl Z d dlmZ d dlmZmZmZmZ d dlm Z d dl
m Z m Z d dl
mZ d dlmZmZ d dlmZ d d lmZ e d
d d d
�G dd� dee��ZdS )� N)�ABC)�Any�Dict�Optional�Tuple)�
deprecated)�BaseChatMessageHistory�InMemoryChatMessageHistory)�
BaseMemory)� AIMessage� HumanMessage)�Field)�get_prompt_input_keyz0.3.1z1.0.0z_Please see the migration guide at: https://python.langchain.com/docs/versions/migrating_memory/)�since�removal�messagec @ s� e Zd ZU dZeed�Zeed< dZ e
e ed< dZ e
e ed< dZ
eed< d ee ef d
ee e f d ee e f fd d
�Zd ee ef d
ee e f d dfdd�Zd ee ef d
ee e f d dfdd�Zddd�Zddd�ZdS )�BaseChatMemoryax Abstract base class for chat memory.
**ATTENTION** This abstraction was created prior to when chat models had
native tool calling capabilities.
It does **NOT** support native tool calling capabilities for chat models and
will fail SILENTLY if used with a chat model that has native tool calling.
DO NOT USE THIS ABSTRACTION FOR NEW CODE.
)�default_factory� chat_memoryN�
output_key� input_keyF�return_messages�inputs�outputs�returnc C s� | j d u r t|| j�}n| j }| jd u rDt|�dkr#t|�� �d }n$d|v r:d}t�d| j j
� d|�� � d�� n
t d|�� � d���| j}|| || fS ) N� r �output�'z ' got multiple output keys: zc. The default 'output' key is being used. If this is not desired, please manually set 'output_key'.zGot multiple output keys: zT, cannot determine which to store in memory. Please set the 'output_key' explicitly.) r r �memory_variablesr �len�list�keys�warnings�warn� __class__�__name__�
ValueError)�selfr r �prompt_input_keyr � r) �G/usr/local/lib/python3.10/dist-packages/langchain/memory/chat_memory.py�_get_input_output+ s$
���z BaseChatMemory._get_input_outputc C s0 | � ||�\}}| j�t|d�t|d�g� dS �z.Save context from this conversation to buffer.)�contentN)r+ r � add_messagesr r �r' r r � input_str�
output_strr) r) r* � save_contextF s ��zBaseChatMemory.save_contextc � s8 �| � ||�\}}| j�t|d�t|d�g�I dH dS r, )r+ r �
aadd_messagesr r r/ r) r) r* �
asave_contextP s ���z BaseChatMemory.asave_contextc C s | j �� dS �zClear memory contents.N)r �clear�r' r) r) r* r6 \ s zBaseChatMemory.clearc � s �| j �� I dH dS r5 )r �aclearr7 r) r) r* r8 ` s �zBaseChatMemory.aclear)r N)r% �
__module__� __qualname__�__doc__r
r r r �__annotations__r r �strr r �boolr r r r+ r2 r4 r6 r8 r) r) r) r* r s2
�
�
�
�&
�
�
�
r )r" �abcr �typingr r r r �langchain_core._apir �langchain_core.chat_historyr r �langchain_core.memoryr
�langchain_core.messagesr r �pydanticr
�langchain.memory.utilsr r r) r) r) r* �<module> s �