File: //usr/local/lib/python3.10/dist-packages/langchain/memory/__pycache__/vectorstore.cpython-310.pyc
o
���g] � @ s� d Z ddlmZmZmZmZmZmZ ddlm Z ddl
m Z ddl m
Z
ddlmZ ddlmZ ddlmZ e d d
d d �G d
d� de��ZdS )z-Class for a VectorStore-backed memory object.� )�Any�Dict�List�Optional�Sequence�Union)�
deprecated)�Document)�VectorStoreRetriever)�Field)�
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 dZedd�Zeed< dZe ed< dZ
e e ed< d Z e
ed
< eed �Zee ed < ed
ee fdd��Zdee ef d
e fdd�Zdee d
ee eee e f f fdd�Zdee ef d
ee eee e f f fdd�Zdee ef d
ee eee e f f fdd�Zdee ef dee e f d
ee fdd �Z dee ef dee e f d
dfd d �Z dee ef dee e f d
dfd d �Z d%d!d"�Z d%d#d$�Z dS )&�VectorStoreRetrieverMemoryz�Store the conversation history in a vector store and retrieves the relevant
parts of past conversation based on the input.
T��exclude� retriever�history�
memory_keyN� input_keyF� return_docs)�default_factory�exclude_input_keys�returnc C s | j gS )z?The list of keys emitted from the load_memory_variables method.)r ��self� r �G/usr/local/lib/python3.10/dist-packages/langchain/memory/vectorstore.py�memory_variables* s z+VectorStoreRetrieverMemory.memory_variables�inputsc C s | j du r t|| j�S | j S )z!Get the input key for the prompt.N)r r
r )r r! r r r �_get_prompt_input_key/ s
z0VectorStoreRetrieverMemory._get_prompt_input_key�docsc C s* | j sd�dd� |D ��}n|}| j|iS )N�
c S s g | ]}|j �qS r �� page_content)�.0�docr r r �
<listcomp>: s zMVectorStoreRetrieverMemory._documents_to_memory_variables.<locals>.<listcomp>)r �joinr )r r# �resultr r r � _documents_to_memory_variables5 s
z9VectorStoreRetrieverMemory._documents_to_memory_variablesc C s( | � |�}|| }| j�|�}| �|�S )�Return history buffer.)r" r �invoker, �r r! r �queryr# r r r �load_memory_variables? s
z0VectorStoreRetrieverMemory.load_memory_variablesc � s0 �| � |�}|| }| j�|�I dH }| �|�S )r- N)r" r �ainvoker, r/ r r r �aload_memory_variablesH s
�
z1VectorStoreRetrieverMemory.aload_memory_variables�outputsc sd t | j�� � �| j� � fdd�|�� D �}dd� t|�� �t|�� � D �}d�|�}t|d�gS )z0Format context from this conversation to buffer.c s i | ] \}}|� vr||�qS r r �r' �k�vr r r �
<dictcomp>X s z>VectorStoreRetrieverMemory._form_documents.<locals>.<dictcomp>c S s g | ] \}}|� d |� ��qS )z: r r5 r r r r) Y s ��z>VectorStoreRetrieverMemory._form_documents.<locals>.<listcomp>r$ r% )�setr �addr �items�listr* r )r r! r4 �filtered_inputs�textsr& r r r �_form_documentsQ s
�
z*VectorStoreRetrieverMemory._form_documentsc C s | � ||�}| j�|� dS �z.Save context from this conversation to buffer.N)r? r �
add_documents�r r! r4 � documentsr r r � save_context` s z'VectorStoreRetrieverMemory.save_contextc � s$ �| � ||�}| j�|�I dH dS r@ )r? r �aadd_documentsrB r r r �
asave_contexte s � z(VectorStoreRetrieverMemory.asave_contextc C s dS �zNothing to clear.Nr r r r r �clearl s z VectorStoreRetrieverMemory.clearc � s �dS rG r r r r r �aclearo s z!VectorStoreRetrieverMemory.aclear)r N)!�__name__�
__module__� __qualname__�__doc__r r r
�__annotations__r �strr r r �bool�tupler r �propertyr r r r r" r r r, r1 r3 r? rD rF rH rI r r r r r s^
�
�
�
�
�
�
�
�
�&
�
�
�
r N)rM �typingr r r r r r �langchain_core._apir �langchain_core.documentsr �langchain_core.vectorstoresr
�pydanticr � langchain.memory.chat_memoryr �langchain.memory.utilsr
r r r r r �<module> s �