tj~ddlmZddlZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZGddeZdS) ) annotationsN)AnyDictListLiteralOptionalSequenceUnion)CallbackManagerForRetrieverRun)Document) BaseRetrieverceZdZUdZded< ded< ded< ded < d!d Zd"dZddd#dZeddddddd$d Z dS)%VespaRetrieverz`Vespa` retriever.rapprbodystr content_fieldz Sequence[str]metadata_fieldsreturnList[Document]cN|j|}t|jds't d|j|jd}d|vr't tj|dg}|j D]yd |j d}|j dkr d}nfd|j D}d |d <| t|| z|S) N2z2Could not retrieve data from Vespa. Error code: {}rooterrorsfields*cHi|]}|d|S)r)get).0mfchilds l/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/retrievers/vespa_retriever.py z)VespaRetriever._query..+s-WWWBBh 3 3B 7 7WWWid) page_contentmetadata)rqueryr status_code startswith RuntimeErrorformatjsondumpshitspoprrappendr )selfrresponserdocsr'r(r"s @r#_queryzVespaRetriever._querys48>>$''8'((33C88 DKK(  }V$ t  tz$x.99:: :] P PE ?..t/A2FFL#s** ?WWWW$BVWWW"4[HTN KKlXNNN O O O O r%r) run_managerr ch|j}||d<||S)Nr)rcopyr6)r3r)r7rs r#_get_relevant_documentsz&VespaRetriever._get_relevant_documents0s0y~~W {{4   r%N)_filterr< Optional[str]c|j}|rd|nd}|d|z|d<||d<||S)N and ryqlr)r9)r3r)r<rs r#"get_relevant_documents_with_filterz1VespaRetriever.get_relevant_documents_with_filter7sZy~~'.6#'###B5kG+U W {{4   r%)krsourcesr<r@urlrC Optional[int]"Union[Sequence[str], Literal['*']]rD(Union[Sequence[str], Literal['*'], None]r@kwargsc  ddlm} n#t$rtdwxYw| |} |} |r|s|rt d|dkrd} d| d<n(d|gt |pgz} t|trd|nd} |rd |nd }d | d | d |}|| d<|r|| d<|| | ||S)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, r?rzselect z from sources z where userQuery()r@r0)rrrr) vespa.applicationrK ImportErrorr: ValueErrorjoinlist isinstancer )clsrErrCrrDr<r@rIrKrr_fields_sourcess r# from_paramszVespaRetriever.from_params@su@  / / / / / / /   U  eCjj{{}}  YG Yw Y  #%%")Y))]Od?;Pb6Q6Q$QRR-7-J-JStyy)))PSH+2:'g'''GXGXX8XXwXXCU  DLs'+     s #)rrrr)r)rr7r rr)r)rr<r=rr)rErrrrCrFrrGrDrHr<r=r@r=rIrrr) __name__ __module__ __qualname____doc____annotations__r6r;rA classmethodrWrBr%r#rr s HHH%JJJ$""""'2!!!!7;!!!!!!  >@<@!%!= = = = = [= = = r%r) __future__rr.typingrrrrrr r langchain_core.callbacksr langchain_core.documentsr langchain_core.retrieversr rrBr%r#rcs"""""" FFFFFFFFFFFFFFFFFFCCCCCC------333333s s s s s ]s s s s s r%