File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/summary.cpython-310.pyc
o
���gb � @ s d dl mZ d dlmZmZmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d dlmZmZmZ d d lmZ d d
lmZ d d lmZ d d lm Z d d
l m Z d dl m Z edddd�G dd� de��Z!edddd�G dd� de e!��Z"e"�#� dS )� )� annotations)�Any�Dict�List�Type)�
deprecated)� BaseCache)� Callbacks)�BaseChatMessageHistory)�BaseLanguageModel)� BaseMessage�
SystemMessage�get_buffer_string)�BasePromptTemplate)�pre_init)� BaseModel)�LLMChain)�BaseChatMemory)�SUMMARY_PROMPTz0.2.12z1.0z�Refer here for how to incorporate summaries of conversation history: https://langchain-ai.github.io/langgraph/how-tos/memory/add-summary-conversation-history/)�since�removal�messagec @ s^ e Zd ZU dZdZded< dZded< ded< eZd ed
< e Z
d ed < ddd�Z ddd�Z dS )�SummarizerMixinzMixin for summarizer.�Human�str� human_prefix�AI� ai_prefixr �llmr �promptzType[BaseMessage]�summary_message_cls�messages�List[BaseMessage]�existing_summary�returnc C s0 t || j| jd�}t| j| jd�}|j||d�S �N)r r )r r )�summary� new_lines)r r r r r r �predict��selfr! r# r' �chain� r, �C/usr/local/lib/python3.10/dist-packages/langchain/memory/summary.py�predict_new_summary% s �z#SummarizerMixin.predict_new_summaryc � s8 �t || j| jd�}t| j| jd�}|j||d�I d H S r% )r r r r r r �apredictr) r, r, r- �apredict_new_summary1 s ��z$SummarizerMixin.apredict_new_summaryN)r! r" r# r r$ r )
�__name__�
__module__� __qualname__�__doc__r �__annotations__r r r r
r r. r0 r, r, r, r- r s
r z0.3.1z1.0.0z_Please see the migration guide at: https://python.langchain.com/docs/versions/migrating_memory/c s� e Zd ZU dZdZded< dZded< edd�d&dd��Ze d'dd��Z
d(dd�Z e d)d d ��Z
d*� fd"d#� Zd+� fd$d%� Z� ZS ),�ConversationSummaryMemoryz�Continually summarizes the conversation history.
The summary is updated after each conversation turn.
The implementations returns a summary of the conversation history which
can be used to provide context to the model.
� r �buffer�history�
memory_key� )�summarize_stepr r � chat_memoryr
r< �int�kwargsr r$ c K sR | d||d�|��}t dt|jj�|�D ]}|�|jj||| � |j�|_q|S )N)r r= r r, )�range�lenr= r! r. r8 )�clsr r= r< r? �obj�ir, r, r- �
from_messagesQ s �z'ConversationSummaryMemory.from_messages� List[str]c C s | j gS )zMWill always return list of memory variables.
:meta private:
)r: �r* r, r, r- �memory_variablesa s z*ConversationSummaryMemory.memory_variables�inputs�Dict[str, Any]c C s( | j r | j| jd�g}n| j}| j|iS )zReturn history buffer.)�content)�return_messagesr r8 r: )r* rI r8 r, r, r- �load_memory_variablesi s
z/ConversationSummaryMemory.load_memory_variables�valuesr c C s8 |d j }ddh}|t|�krtd|� d|� d���|S )z4Validate that prompt input variables are consistent.r r& r' z:Got unexpected prompt input variables. The prompt expects z, but it should have �.)�input_variables�set�
ValueError)rB rN �prompt_variables�
expected_keysr, r, r- � validate_prompt_input_variablesq s
���z9ConversationSummaryMemory.validate_prompt_input_variables�outputs�Dict[str, str]�Nonec s. t � �||� | �| jjdd� | j�| _dS )z.Save context from this conversation to buffer.���N)�super� save_contextr. r= r! r8 )r* rI rV �� __class__r, r- r[ } s
�z&ConversationSummaryMemory.save_contextc s t � �� d| _dS )zClear memory contents.r7 N)rZ �clearr8 rG r\ r, r- r^ � s
z ConversationSummaryMemory.clear)
r r r= r
r< r> r? r r$ r6 )r$ rF )rI rJ r$ rJ )rN r r$ r )rI rJ rV rW r$ rX )r$ rX )r1 r2 r3 r4 r8 r5 r: � classmethodrE �propertyrH rM r rU r[ r^ �
__classcell__r, r, r\ r- r6 > s
�
r6 N)$�
__future__r �typingr r r r �langchain_core._apir �langchain_core.cachesr �langchain_core.callbacksr �langchain_core.chat_historyr
� langchain_core.language_modelsr �langchain_core.messagesr r
r �langchain_core.promptsr �langchain_core.utilsr �pydanticr �langchain.chains.llmr � langchain.memory.chat_memoryr �langchain.memory.promptr r r6 �
model_rebuildr, r, r, r- �<module> s6 �"� D