File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/buffer_window.cpython-310.pyc
o
���g� � @ sb d dl mZmZmZmZ d dlmZ d dlmZm Z d dl
m Z edddd�G d d
� d
e ��Z d S ) � )�Any�Dict�List�Union)�
deprecated)� BaseMessage�get_buffer_string)�BaseChatMemoryz0.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 dZdZeed< dZeed< dZeed< dZ e
ed < e d
e ee
e f fd d ��Ze d
efd
d��Ze d
e
e fdd��Ze d
e
e fdd��Zdeeef d
eeef fdd�ZdS )� ConversationBufferWindowMemoryz�Use to keep track of the last k turns of a conversation.
If the number of messages in the conversation is more than the maximum number
of messages to keep, the oldest messages are dropped.
�Human� human_prefix�AI� ai_prefix�history�
memory_key� �k�returnc C s | j r| jS | jS )zString buffer of memory.)�return_messages�buffer_as_messages�
buffer_as_str��self� r �I/usr/local/lib/python3.10/dist-packages/langchain/memory/buffer_window.py�buffer s z%ConversationBufferWindowMemory.bufferc C s8 | j dkr| jj| j d d� ng }t|| j| jd�S )z@Exposes the buffer as a string in case return_messages is False.r � N)r r )r � chat_memory�messagesr r r )r r! r r r r # s &�z,ConversationBufferWindowMemory.buffer_as_strc C s& | j dkr| jj| j d d� S g S )zIExposes the buffer as a list of messages in case return_messages is True.r r N)r r r! r r r r r - s &z1ConversationBufferWindowMemory.buffer_as_messagesc C s | j gS )zMWill always return list of memory variables.
:meta private:
)r r r r r �memory_variables2 s z/ConversationBufferWindowMemory.memory_variables�inputsc C s | j | jiS )zReturn history buffer.)r r )r r# r r r �load_memory_variables: s z4ConversationBufferWindowMemory.load_memory_variablesN)�__name__�
__module__� __qualname__�__doc__r �str�__annotations__r r r �int�propertyr r r r r r r" r r r$ r r r r r
s
&r
N)
�typingr r r r �langchain_core._apir �langchain_core.messagesr r � langchain.memory.chat_memoryr r
r r r r �<module> s �