File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/token_buffer.cpython-310.pyc
o
���g� � @ sj d dl mZmZmZ d dlmZ d dlmZ d dlm Z m
Z
d dl m Z edddd �G d
d � d e ��Z
d S )
� )�Any�Dict�List)�
deprecated)�BaseLanguageModel)� 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< eed< dZ eed< d Z
e ed
< e d e
fd d
��Ze d efdd��Ze d ee fdd��Ze d ee fdd��Zdeee
f d eee
f fdd�Zdeee
f deeef d df� fdd� Z� ZS )� ConversationTokenBufferMemoryz�Conversation chat memory with token limit.
Keeps only the most recent messages in the conversation under the constraint
that the total number of tokens in the conversation does not exceed a certain limit.
�Human� human_prefix�AI� ai_prefix�llm�history�
memory_keyi� �max_token_limit�returnc C s | j r| jS | jS )zString buffer of memory.)�return_messages�buffer_as_messages�
buffer_as_str��self� r �H/usr/local/lib/python3.10/dist-packages/langchain/memory/token_buffer.py�buffer s z$ConversationTokenBufferMemory.bufferc C s t | jj| j| jd�S )z@Exposes the buffer as a string in case return_messages is False.)r r )r � chat_memory�messagesr r r r r r r $ s
�z+ConversationTokenBufferMemory.buffer_as_strc C s | j jS )zIExposes the buffer as a list of messages in case return_messages is True.)r r r r r r r - s z0ConversationTokenBufferMemory.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.ConversationTokenBufferMemory.memory_variables�inputsc C s | j | jiS )zReturn history buffer.)r r )r r" r r r �load_memory_variables: s z3ConversationTokenBufferMemory.load_memory_variables�outputsNc sl t � �||� | jj}| j�|�}|| jkr2g }|| jkr4|�|�d�� | j�|�}|| jks dS dS dS )z6Save context from this conversation to buffer. Pruned.r N) �super� save_contextr r r � get_num_tokens_from_messagesr �append�pop)r r" r$ r �curr_buffer_length�
pruned_memory�� __class__r r r& > s
��z*ConversationTokenBufferMemory.save_context)�__name__�
__module__� __qualname__�__doc__r �str�__annotations__r r r r �int�propertyr r r r r r r! r r# r& �
__classcell__r r r, r r
s"
"2r
N)�typingr r r �langchain_core._apir � langchain_core.language_modelsr �langchain_core.messagesr r � langchain.memory.chat_memoryr r
r r r r �<module> s �