File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/combined.cpython-310.pyc
o
���g� � @ sX d dl Z d dlmZmZmZmZ d dlmZ d dlm Z d dl
m Z G dd� de�Z dS )� N)�Any�Dict�List�Set)�
BaseMemory)�field_validator)�BaseChatMemoryc @ s� e Zd ZU dZee ed< ed�edee dee fdd���Z ed�edee dee fdd���Z
e dee fd d
��Z
d ee ef dee e f fd d
�Zd ee ef dee e f ddfdd�Zddd�ZdS )�CombinedMemoryz+Combining multiple memories' data together.�memories�value�returnc C sB t � }|D ]}|�|j�}|rtd|� d���|t |j�O }q|S )NzThe same variables zL are found in multiplememory object, which is not allowed by CombinedMemory.)�set� intersection�memory_variables�
ValueError)�clsr �
all_variables�val�overlap� r �D/usr/local/lib/python3.10/dist-packages/langchain/memory/combined.py� check_repeated_memory_variable s
�z-CombinedMemory.check_repeated_memory_variablec C s2 |D ]}t |t�r|jdu rt�d|� �� q|S )zHCheck that if memories are of type BaseChatMemory that input keys exist.NzWWhen using CombinedMemory, input keys should be so the input is known. Was not set on )�
isinstancer � input_key�warnings�warn)r r r r r r �check_input_key! s
���z CombinedMemory.check_input_keyc C s" g }| j D ]}|�|j� q|S )z5All the memory variables that this instance provides.)r
�extendr )�selfr �memoryr r r r / s
z CombinedMemory.memory_variables�inputsc C sP i }| j D ] }|�|�}|�� D ]\}}||v r td|� d���|||< qq|S )z Load all vars from sub-memories.z
The variable z# is repeated in the CombinedMemory.)r
�load_memory_variables�itemsr )r r � memory_datar �data�keyr r r r r! ; s
�
�z$CombinedMemory.load_memory_variables�outputsNc C s | j D ]}|�||� qdS )z0Save context from this session for every memory.N)r
� save_context)r r r& r r r r r' K s
�zCombinedMemory.save_contextc C s | j D ]}|�� qdS )z1Clear context from this session for every memory.N)r
�clear)r r r r r r( Q s
�zCombinedMemory.clear)r N)�__name__�
__module__� __qualname__�__doc__r r �__annotations__r � classmethodr r �property�strr r r r! r' r( r r r r r
s&
�� " &r )
r �typingr r r r �langchain_core.memoryr �pydanticr � langchain.memory.chat_memoryr r r r r r �<module> s