U h@spddlmZddlZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZGdddeZdS) ) annotationsN)AnyDictListLiteralOptionalSequenceUnion)CallbackManagerForRetrieverRun)Document) BaseRetrieverc @seZdZUdZded<ded<ded<ded <dd d d d Zddd dddZddddd dddZedddddddddddddddd ddZ dS) VespaRetrieverz`Vespa` retriever.rapprbodystr content_fieldz Sequence[str]metadata_fieldszList[Document])rreturncs|j|}t|jds,td|j|jd}d|krPtt|dg}|j D]\d |j d}|j dkrd}nfdd |j D}d |d <| t||d qZ|S) N2z2Could not retrieve data from Vespa. Error code: {}rooterrorsfields*csi|]}|d|qS)r)get).0ZmfchildR/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/retrievers/vespa_retriever.py +sz)VespaRetriever._query..id) page_contentmetadata)rqueryr status_code startswith RuntimeErrorformatjsondumpshitspoprrappendr )selfrresponserZdocsr"r#rrr_querys&      zVespaRetriever._queryr )r$ run_managerrcCs|j}||d<||S)Nr$rcopyr0)r.r$r1rrrr_get_relevant_documents0s z&VespaRetriever._get_relevant_documentsN)_filterz Optional[str])r$r5rcCs>|j}|rd|nd}|d||d<||d<||S)N and ryqlr$r2)r.r$r5rrrr"get_relevant_documents_with_filter7s  z1VespaRetriever.get_relevant_documents_with_filterr)krsourcesr5r7z Optional[int]z"Union[Sequence[str], Literal['*']]z(Union[Sequence[str], Literal['*'], None]) urlrr9rr:r5r7kwargsrcKszddlm} Wntk r,tdYnX| |} |} |rT|sJ|rTtdnn|dkrjd} d| d<nd|gt|pzg} t|trd|nd} |rd |nd }d | d | d |}|| d<|r|| d<|| | ||dS)aInstantiate retriever from params. Args: url (str): Vespa app URL. content_field (str): Field in results to return as Document page_content. k (Optional[int]): Number of Documents to return. Defaults to None. metadata_fields(Sequence[str] or "*"): Fields in results to include in document metadata. Defaults to empty tuple (). sources (Sequence[str] or "*" or None): Sources to retrieve from. Defaults to None. _filter (Optional[str]): Document filter condition expressed in YQL. Defaults to None. yql (Optional[str]): Full YQL query to be used. Should not be specified if _filter or sources are specified. Defaults to None. kwargs (Any): Keyword arguments added to query body. Returns: VespaRetriever: Instantiated VespaRetriever. r)VespazCpyvespa is not installed, please install with `pip install pyvespa`zKyql should only be specified if both sources and _filter are not specified.rshortsummaryz, r6rzselect z from sources z where userQuery()r7r+)rrrr) Zvespa.applicationr= ImportErrorr3 ValueErrorjoinlist isinstancer)clsr;rr9rr:r5r7r<r=rr_fieldsZ_sourcesrrr from_params@s8    zVespaRetriever.from_params) __name__ __module__ __qualname____doc____annotations__r0r4r8 classmethodrGrrrrr s  r ) __future__rr)typingrrrrrrr Zlangchain_core.callbacksr Zlangchain_core.documentsr Zlangchain_core.retrieversr r rrrrs $