a
    bg                     @   s\   d dl mZmZ d dlZd dlZd dlmZmZ d dlm	Z	 d dl
mZ G dd deZdS )    )ListOptionalN)#AsyncCallbackManagerForRetrieverRunCallbackManagerForRetrieverRun)Document)BaseRetrieverc                   @   s   e Zd ZU dZeed< dZee ed< dZ	eed< dZ
eed< d	Zeed
< dZeed< eeee dddZeeee dddZdS )RemoteLangChainRetrieverz`LangChain API` retriever.urlNheadersmessage	input_keyresponseresponse_keypage_contentpage_content_keymetadatametadata_key)queryrun_managerreturnc                   s:   t j j j|i jd}| } fdd| j D S )N)jsonr
   c                    s$   g | ]}t | j | j d qS )r   r   r   r   r   .0rself }/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/retrievers/remote_retriever.py
<listcomp>$   s   zDRemoteLangChainRetriever._get_relevant_documents.<locals>.<listcomp>)requestspostr	   r   r
   r   r   )r   r   r   r   resultr   r   r   _get_relevant_documents   s    
z0RemoteLangChainRetriever._get_relevant_documentsc             
      s   t  4 I d H t}|jd j j j|id4 I d H $}| I d H }W d   I d H  qp1 I d H sf0    Y  W d   I d H  q1 I d H s0    Y   fdd| j D S )NPOST)r
   r   c                    s$   g | ]}t | j | j d qS r   r   r   r   r   r   r    3   s   zERemoteLangChainRetriever._aget_relevant_documents.<locals>.<listcomp>)aiohttpZClientSessionrequestr	   r
   r   r   r   )r   r   r   sessionr   r#   r   r   r   _aget_relevant_documents+   s    b
z1RemoteLangChainRetriever._aget_relevant_documents)__name__
__module____qualname____doc__str__annotations__r
   r   dictr   r   r   r   r   r   r   r$   r   r)   r   r   r   r   r      s   
r   )typingr   r   r&   r!   Zlangchain_core.callbacksr   r   Zlangchain_core.documentsr   Zlangchain_core.retrieversr   r   r   r   r   r   <module>   s   