a bg@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|vrPtt|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).0mfchild|/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/retrievers/vespa_retriever.py +z)VespaRetriever._query..id) page_contentmetadata)rqueryr status_code startswith RuntimeErrorformatjsondumpshitspoprrappendr )selfrresponserdocsr$r%rrr _querys&      zVespaRetriever._queryr )r& run_managerrcCs|j}||d<||S)Nr&rcopyr3)r0r&r4rrrr _get_relevant_documents0s z&VespaRetriever._get_relevant_documentsN)_filterz Optional[str])r&r8rcCs>|j}|rd|nd}|d||d<||d<||S)N and ryqlr&r5)r0r&r8rrrr "get_relevant_documents_with_filter7s  z1VespaRetriever.get_relevant_documents_with_filterr)krsourcesr8r:z Optional[int]z"Union[Sequence[str], Literal['*']]z(Union[Sequence[str], Literal['*'], None]) urlrr<rr=r8r:kwargsrcKszddlm} Wnty*tdYn0| |} |} |rR|sH|rRtdnn|dkrhd} d| d<nd|gt|pxg} 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, r9rzselect z from sources z where userQuery()r:r-)rrrr) Zvespa.applicationr@ ImportErrorr6 ValueErrorjoinlist isinstancer)clsr>rr<rr=r8r:r?r@rr_fieldsZ_sourcesrrr from_params@s8     zVespaRetriever.from_params) __name__ __module__ __qualname____doc____annotations__r3r7r; classmethodrJrrrr r s  r ) __future__rr+typingrrrrrrr Zlangchain_core.callbacksr Zlangchain_core.documentsr Zlangchain_core.retrieversr r rrrr s $