hOhddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ejNe(Z)dZ*dZ+dZ,Gdde-eZ.de e#e&ffdZ/de-fdZ0GddeZ1y)N)Enum) HTTPStatus)AnyDictListOptionalTuple) ClientTimeout)Document)get_runtime_environment)get_from_dict_or_env)VectorStoreRetriever) BaseModel)Responserequest)RequestException)App AuthContextContext FrameworkPromptQaRuntimez0.1.1zhttp://localhost:8000zhttps://api.daxa.aiceZdZdZdZdZdZy)Routesz2Routes available for the Pebblo API as enumerator.z/v1/app/discoverz /v1/promptz/v1/prompt/governanceN)__name__ __module__ __qualname____doc__retrieval_app_discoverpromptprompt_governances/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/langchain_community/chains/pebblo_retrieval/utilities.pyrr%s</ F/r$rreturnc  t}td|jdd}tj}t |j tjd|jdd|j|jt|jdd|jd d }d |jvrd |_ d |_ tjd|tjd|||fS)zFetch the current Framework and Runtime details. Returns: Tuple[Framework, Runtime]: Framework and Runtime for the current app instance. langchainlibrary_versionN)nameversionPWDplatformunknownruntimeruntime_version)hostpathr-os os_versioniplanguagelanguage_versionDarwindesktopzMac OSXz framework zruntime )r rgetr-unamernoder3environsystemr+get_iptyper/loggerdebug) runtime_env frameworkr;r/s r% get_runtimerE-s *+K +//2CT"JI NN E ZZ ZZ Y7 <<== 8I6$):IF G7::  # LL:i[)* LL8G9%& g r$cddl}|j} |j|}|S#t$r|jd}Y|SwxYw)zJFetch local runtime ip address. Returns: str: IP address rN localhost)socket gethostname gethostbyname Exception)rHr1 public_ips r%r?r?LsY     D6((.   6((5 6s)AAcNeZdZUdZeeed< dZeed< eeed< eeed< deffd Z d e d d fd Z d&d ede dededee deedeeefdeded d fdZ d&d ede dededee deedeeefdeded d fdZded eeeeefffdZded eeeeefffdZd&ded efdZe d'dedededeed ed eef d!Zeded"eed d fd#Ze d'dedededeed ed ef d$Z d&d ede dededee deedeeefdeded efd%ZxZ S)(PebbloRetrievalAPIWrapperz!Wrapper for Pebblo Retrieval API.api_keylocalclassifier_locationclassifier_url cloud_urlkwargsc t|ddd|d<t|ddt|d<t|ddt|d<t|d i|y) z%Validate that api key in environment.rOPEBBLO_API_KEYrRPEBBLO_CLASSIFIER_URLrSPEBBLO_CLOUD_URLNr#)r _DEFAULT_CLASSIFIER_URL_DEFAULT_PEBBLO_CLOUD_URLsuper__init__)selfrT __class__s r%r]z"PebbloRetrievalAPIWrapper.__init__hsh0 I/ y$8 $&=?V$  3 K!35N { "6"r$appr&NcLd}|jd}|jdk(rL|j}|jtj j }|jd|||}|jr|jd}|rAtj|jjd}|jd|i|jdti|jtj j }|jd|||}yy) z Send app discovery request to Pebblo server & cloud. Args: app (App): App instance to be discovered. NT exclude_unsetrPPOST cloud_requestpebblo_server_versionpebblo_client_version)dictrQ _make_headersrRrr value make_requestrOjsonloadstextr:updatePLUGIN_VERSIONrS) r^r` pebblo_resppayloadheadersapp_discover_urlrgpebblo_cloud_url_s r%send_app_discoverz+PebbloRetrievalAPIWrapper.send_app_discoverus (((.  # #w .((*G&&'(E(E(K(K'LM ++F4DgwWK <<((t(.>.@.D.DZQS.T +1<1A1A1C1G1G!12 .//r$cJKd|i}idd}d}|jdk(r|j}|jtjj }|j d|||d{}|r*|jdi|d<|jd d|d <||fS74w) rr!rrTrPrdNrr)rQrjrRrr"rkrr:rs r%acheck_prompt_validityz0PebbloRetrievalAPIWrapper.acheck_prompt_validity0s"#H--/ B $  # #w .((*G&&'(@(@(F(F'GH !% 2 2*G^!K.9ooj".M +1<PQ1R .// sA*B#,B!-5B#rfcddd}|r@|jr|jd|ji|Stjd|S)z Generate headers for the request. args: cloud_request (bool): flag indicating whether the request is for Pebblo cloud. returns: dict: Headers for the request. zapplication/json)Acceptz Content-Typez x-api-keyz,API key is missing for Pebblo cloud request.)rOrprAr)r^rfrts r%rjz'PebbloRetrievalAPIWrapper._make_headersQsM).  || T\\:;MNr$methodurlrtrstimeoutc > t|||||}tjd||jjt t |jj r|jj ngt |j|jtjk\r$tjd|j|S|jtjk\r$tjd|j|S|jtjk7r"tjd|j|S#t$rtjd|Yyt$r }tjd|Yd}~yd}~wwxYw) a Make a request to the Pebblo server/cloud API. Args: method (str): HTTP method (GET, POST, PUT, DELETE, etc.). url (str): URL for the request. headers (dict): Headers for the request. payload (Optional[dict]): Payload for the request (for POST, PUT, etc.). timeout (int): Timeout for the request in seconds. Returns: Optional[Response]: Response object if the request is successful. )rrrtrmrz5Request: method %s, url %s, len %s response status %sPebblo Server: Error $Pebblo received an invalid payload: -Pebblo returned an unexpected response code: Unable to reach server %sz'An Exception caught in make_request: %sN)rrArBrstrlenbody status_coderINTERNAL_SERVER_ERRORr BAD_REQUESTroOKrrK)rrrtrsrresponsees r%rlz&PebbloRetrievalAPIWrapper.make_requesthsd* I3gwH LLG  $$C1A1A1F1F((--BOPH(()  ##z'G'GG!6x7K7K6LMNO%%)?)??!Ehmm_UVO %%6C++,. O = NN6 < I NNDa H H Is,CEAEAEF4F<FFrrc|r|jdi}|j|jdijdi|jdd|jdi}|j|jdijdi|jdd|jdg}|D]}|jddyi|d<i|d<g|d<y)z Update the payload with response, prompt and context from Pebblo response. Args: payload (dict): Payload to be updated. pebblo_resp (Optional[dict]): Response from Pebblo server. rretrieval_datadataNr!contextdoc)r:rppop)rsrrrr!r context_datas r%rz.PebbloRetrievalAPIWrapper.update_cloud_payloads {{:r2H OOKOO,x>O'PQ!J,B,BBB8==/R"JMM9K'/1+3--/$9M%:: : :(! ) ::$%:%:::: : : : :(!  = NN6 < J NNEq I I JsG$E<EE<E&EE& CEE E# E&.E /E&3 E<>E$?E<GE<E& E E&E! EE! E&$E<&E9,E/ -E94E<8G9E<<GGG%G;GGGc ft||D cgc]} t| trt| jj d| jj d| j |jjj| jj dc} t||j di|j dd| t| ||r |jnd |rt|d r |jng|j } | jd Scc} w)a Build the QA payload for the prompt. Args: app_name (str): Name of the app. retriever (VectorStoreRetriever): Retriever instance. question (str): Question asked in the prompt. answer (str): Answer generated by the model. auth_context (Optional[AuthContext]): Authentication context. docs (List[Document]): List of documents retrieved. prompt_entities (Dict[str, Any]): Entities present in the prompt. prompt_time (str): Time when the prompt was generated. prompt_gov_enabled (bool): Whether prompt governance is enabled. Returns: dict: The QA payload for the prompt. full_pathsource pb_checksum)retrieved_fromr vector_dbrrrr)rrrr)rr. user_auth)r*rr!rruseruser_identitiesrQTrb)r isinstancer rmetadatar: page_content vectorstorer_rruser_idhasattrrrQri) r^ryrzr{r|r}r~rrrrqas r%rz1PebbloRetrievalAPIWrapper.build_prompt_qa_payloads:  c8,#&<<#3#3#S\\%5%5h%?$(('33==FF # 0 0 ?  (,,Z<+// qA#5  (#)5%%9 k B)22 $ 8 85 8wwTw**5 sBD. )F)N)!rrrrrr__annotations__rQrr]rrxrrrr rboolrrr rrrirj staticmethodintrrlrrr __classcell__)r_s@r%rNrN\s2+ c]"&&ISM!&}! # #NSNTNR$);V;V(;V ;V  ;V {+ ;V8n;Vc3h;V;V!;V ;VN$)<V<V(<V <V  <V {+ <V8n<Vc3h<V<V!<V <V|0c0eD$sCx.rs  33!-65<&0   8 $110S$0U9g-.>   B+ B+r$