File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/buffer.cpython-310.pyc
o
���gu � @ s� d dl mZmZmZmZ d dlmZ d dlmZm Z d dl
m Z d dl m
Z
mZ d dlmZ eddd d
�G d d � d e
��Zeddd d
�G d
d� de��ZdS )� )�Any�Dict�List�Optional)�
deprecated)� BaseMessage�get_buffer_string)�pre_init)�BaseChatMemory�
BaseMemory)�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 dZdZeed< dZeed< dZeed< e de
fd d
��Z de
fd d �Z d
e
e defdd�Ze defdd��Zdefdd�Ze de
e fdd��Zde
e fdd�Ze de
e fdd��Zdeee
f deee
f fdd �Zdeee
f deee
f fd d �Zd S ) �ConversationBufferMemoryaS A basic memory implementation that simply stores the conversation history.
This stores the entire conversation history in memory without any
additional processing.
Note that additional processing may be required in some situations when the
conversation history is too large to fit in the context window of the model.
�Human� human_prefix�AI� ai_prefix�history�
memory_key�returnc C s | j r| jS | jS )�String buffer of memory.)�return_messages�buffer_as_messages�
buffer_as_str��self� r �B/usr/local/lib/python3.10/dist-packages/langchain/memory/buffer.py�buffer! s z ConversationBufferMemory.bufferc � s$ �| j r | �� I dH S | �� I dH S )r N)r �abuffer_as_messages�abuffer_as_strr r r r �abuffer& s ��� �z ConversationBufferMemory.abuffer�messagesc C s t || j| jd�S )N)r r )r r r �r r$ r r r �_buffer_as_str. s
�z'ConversationBufferMemory._buffer_as_strc C s | � | jj�S )�?Exposes the buffer as a string in case return_messages is True.)r& � chat_memoryr$ r r r r r 5 s z&ConversationBufferMemory.buffer_as_strc � s �| j �� I dH }| �|�S )r' N)r( �
aget_messagesr& r% r r r r" : s �
z'ConversationBufferMemory.abuffer_as_strc C s | j jS )�JExposes the buffer as a list of messages in case return_messages is False.)r( r$ r r r r r ? s z+ConversationBufferMemory.buffer_as_messagesc � s �| j �� I dH S )r* N)r( r) r r r r r! D s �z,ConversationBufferMemory.abuffer_as_messagesc C � | j gS )zMWill always return list of memory variables.
:meta private:
�r r r r r �memory_variablesH s z)ConversationBufferMemory.memory_variables�inputsc C � | j | jiS �zReturn history buffer.�r r �r r. r r r �load_memory_variablesP � z.ConversationBufferMemory.load_memory_variablesc � s �| � � I dH }| j|iS )z9Return key-value pairs given the text input to the chain.N)r# r )r r. r r r r �aload_memory_variablesT s �
z/ConversationBufferMemory.aload_memory_variablesN)�__name__�
__module__� __qualname__�__doc__r �str�__annotations__r r �propertyr r r# r r r& r r" r r! r- r r3 r5 r r r r r s&
"&r c @ s4 e Zd ZU dZdZeed< dZeed< dZeed< dZ e
e ed < dZ e
e ed
< d Z eed < e
d
edefdd��Zedee fdd��Zdeeef deeef fdd�Zdeeef deeef fdd�Zdeeef deeef ddfdd�Zdeeef deeef ddfdd �Zd!d d �Zd!d d �ZdS )"� ConversationStringBufferMemorya� A basic memory implementation that simply stores the conversation history.
This stores the entire conversation history in memory without any
additional processing.
Equivalent to ConversationBufferMemory but tailored more specifically
for string-based conversations rather than chat models.
Note that additional processing may be required in some situations when the
conversation history is too large to fit in the context window of the model.
r r r r � r N�
output_key� input_keyr r �valuesr c C s |� dd�r
td��|S )z*Validate that return messages is not True.r Fz@return_messages must be False for ConversationStringBufferMemory)�get�
ValueError)�clsrA r r r �validate_chainsw s
�z.ConversationStringBufferMemory.validate_chainsc C r+ )zLWill always return list of memory variables.
:meta private:
r, r r r r r- � s z/ConversationStringBufferMemory.memory_variablesr. c C r/ r0 r1 r2 r r r r3 � r4 z4ConversationStringBufferMemory.load_memory_variablesc � s �| � |�S r0 )r3 r2 r r r r5 � s �
z5ConversationStringBufferMemory.aload_memory_variables�outputsc C s� | j du r t|| j�}n| j }| jdu r,t|�dkr#td|�� � ���t|�� �d }n| j}| j� d�|| }| j � d�|| }| j
dd� ||g� 7 _
dS )�.Save context from this conversation to buffer.N� z One output key expected, got r z: �
) r@ r r- r? �lenrC �keys�listr r r �join)r r. rF �prompt_input_keyr? �human�air r r � save_context� s
z+ConversationStringBufferMemory.save_contextc � s �| � ||�S )rG )rQ )r r. rF r r r �
asave_context� s � z,ConversationStringBufferMemory.asave_contextc C s
d| _ dS )zClear memory contents.r> N)r r r r r �clear� s
z$ConversationStringBufferMemory.clearc � s �| � � d S )N)rS r r r r �aclear� s � z%ConversationStringBufferMemory.aclear)r N)r6 r7 r8 r9 r r: r; r r r? r r@ r r r rE r< r r- r r3 r5 rQ rR rS rT r r r r r= Z s2
""&
�
�
�
r= N)�typingr r r r �langchain_core._apir �langchain_core.messagesr r �langchain_core.utilsr � langchain.memory.chat_memoryr
r �langchain.memory.utilsr r r= r r r r �<module> s$ �G�