NgOxddlZddlZddlZddlZddlmZddlmZddlm 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&ej'e(Z)dZ*dZ+dZ,Gdde-eZ.de e#e&ffdZ/de-fdZ0GddeZ1dS)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.aiceZdZdZdZdZdZdS)Routesz2Routes available for the Pebblo API as enumerator.z/v1/app/discoverz /v1/promptz/v1/prompt/governanceN)__name__ __module__ __qualname____doc__retrieval_app_discoverpromptprompt_governanceq/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/chains/pebblo_retrieval/utilities.pyrr%s'<</ F/r$rreturnc 4t}td|dd}tj}t |jtjd|dd|j |j t|dd|d d }d |jvrd |_ d |_ td|td|||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"J"JI N  E Z Z Y77 <= 88I66$):IFF   G7:  # LL)i))*** LL%G%%&&& g r$cddl}|} ||}n%#t$r|d}YnwxYw|S)zJFetch local runtime ip address. Returns: str: IP address rN localhost)socket gethostname gethostbyname Exception)rHr1 public_ips r%r?r?Lsp MMM     D6((.. 666((55 6 s0AAceZdZUdZeeed< dZeed< eeed< eeed< deffd Z d e d d fd Z d(dede dededee deedeeefdeded d fdZ d(dede dededee deedeeefdeded d fdZded eeeeefffdZded eeeeefffdZd(ded efdZe d)deded ed!eed"ed eef d#Zed!ed$eed d fd%Ze d)deded ed!eed"ed ef d&Z d(dede 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<tjd i|dS) 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__hs0 I/  y$8 $&=?V$ $  3 K!35N  { ""6"""""r$appr&Nc<d}|d}|jdkrF|}|jtjj}|d|||}|jr|d}|rCtj |j  d}| d|i| dti|jtjj}|d|||}dSdS) 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_discoverus9 (((..  #w . .((**G&M(E(KMM ++F4DgwWWK < N((t(<.>.@.@.D.DZQS.T.T +1<1A1A1C1C1G1G!122 .//r$c:Kd|i}idd}d}|jdkr|}|jtjj}|d|||d{V}|r2|di|d<|d d|d <||fS) rr!rrTrPrdNrr)rQrjrRrr"rkrr:rs r%acheck_prompt_validityz0PebbloRetrievalAPIWrapper.acheck_prompt_validity0s"#H--/ B B $  #w . .((**G&H(@(FHH !% 2 2*G^!!K S.9ooj".M.M +1<PQ1R1R .//r$rfcddd}|r>|jr|d|jintd|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_headersQs`).    O| O T\:;;;;MNNNr$methodurlrtrstimeoutc   t|||||}td||jjt t |jjr |jjngt |j|jtj kr#t d|jno|jtj kr#t d|j n7|jtj kr"t d|j|S#t$rt d|Yn1t$r%}t d|Yd}~nd}~wwxYwdS) 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_requesths* I3gwH LLG $C1A1FN(--BOOPPH())    #z'GGGMx7KMMNNNN%)???UhmUUVVVV%66.+.. O = = = NN6 < < < < < I I I NNDa H H H H H H H H ItsD(D++%F FE;;Frrc4|r|di}||didi|dd|di}||didi|dd|dg}|D]}|dddSi|d<i|d<g|d<dS)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(  ${{:r22H OOKOO,zEPebbloRetrievalAPIWrapper.build_prompt_qa_payload..s   c8,, #&<#3#3#S\%5%5h%?%?$$('3=F # 0 0 ? ?    r$rrr)rrrr)rr. user_auth)r*rr!rruseruser_identitiesrQTrb)rrr:user_idhasattrrrQri) r^ryrzr{r|r}r~rrrqas ` r%rz1PebbloRetrievalAPIWrapper.build_prompt_qa_payloads:        (,,Z<<+// qAA#5  (((#)5D%%9 ' k B BL22 $ 85   8wwTw***r$)F)Nr)!rrrrrr__annotations__rQrr]rrxrrrr rboolrrr rrrirj staticmethodintrrlrrr __classcell__)r_s@r%rNrN\s:++ c]"&&&&ISM!!!&}! # # # # # # #NSNTNNNNR$);V;V;V(;V ;V  ;V {+ ;V8n;Vc3h;V;V!;V ;V;V;V;VN$)<V<V<V(<V <V  <V {+ <V8n<Vc3h<V<V!<V <V<V<V<V|0c0eD$sCx.rs?  33333333333333!!!!!!------666666555555<<<<<<&&&&&&&&000000  8 $ $1100000S$000U9g-.>      B+B+B+B+B+ B+B+B+B+B+r$