7|hvddlmZddlZddlmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZGddeZy) ) annotationsN)AnyDictListLiteralOptionalSequenceUnion)CallbackManagerForRetrieverRun)Document) BaseRetrieverceZdZUdZded< ded< ded< ded < dd Z dd Zd d  ddZed dd d d d ddZ y )VespaRetrieverz`Vespa` retriever.rapprbodystr content_fieldz Sequence[str]metadata_fieldscl|jj|}t|jj ds$t dj |j|jd}d|vr!t tj|dg}|jD]}|dj|jd}|jdk(r|d}n,|jDcic]}||dj|}}|d|d<|jt|| |Scc}w) N2z2Could not retrieve data from Vespa. Error code: {}rooterrorsfields*id) page_contentmetadata)rqueryr status_code startswith RuntimeErrorformatjsondumpshitspoprrgetappendr ) selfrresponserdocschildrrmfs m/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/retrievers/vespa_retriever.py_queryzVespaRetriever._querys(88>>$'8''(33C8DKK((  }}V$ t tzz$x.9: :]] PE ?..t/A/A2FL##s* ?BFBVBVWBBh 3 3B 77WW"4[HTN KKlXN O P Xs+D1cb|jj}||d<|j|S)Nrrcopyr0)r*r run_managerrs r/_get_relevant_documentsz&VespaRetriever._get_relevant_documents0s,yy~~W {{4  N)_filterc|jj}|rd|nd}|d|z|d<||d<|j|S)N and ryqlrr2)r*rr7rs r/"get_relevant_documents_with_filterz1VespaRetriever.get_relevant_documents_with_filter7sNyy~~'.E'#B5kG+U W {{4  r6)krsourcesr7r:c  ddlm} | |} |j} |r|s|r t d|dk(rd} d| d<n"dj |gt |xsgz} t|trdj |nd} |rd |nd }d | d | d |}|| d<|r|| d<|| | ||S#t$r tdwxYw)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) vespa.applicationr@ ImportErrorr3 ValueErrorjoinlist isinstancer )clsurlrr=rr>r7r:kwargsr@rr_fields_sourcess r/ from_paramszVespaRetriever.from_params@s@  / Cj{{} Gw  #%")Y))]Od?;Pb6Q$QR-7-Jtyy)PSH+2gY'GG9N8*z(Union[Sequence[str], Literal['*'], None]r7rQr:rQrKrrOr) __name__ __module__ __qualname____doc____annotations__r0r5r; classmethodrNr<r6r/rr s H% J$""'2!!*H! !7;!!&3! !  >@<@!%!= = =  = < = := = = =  = = r6r) __future__rr$typingrrrrrr r langchain_core.callbacksr langchain_core.documentsr langchain_core.retrieversr rr<r6r/r]s+" FFFC-3s ]s r6