tPjUddlZddlmZddlmZddlZddlmZdZee d<dZ ee d<d Z ee d <d Z ee d <ejZee d <ddlmZmZmZdZGddeZGddeZGddZy)N)Enum)Final)verbose_loggerz!https://deepeval.confident-ai.comDEEPEVAL_BASE_URLz$https://eu.deepeval.confident-ai.comDEEPEVAL_BASE_URL_EUzhttps://api.confident-ai.com API_BASE_URLzhttps://eu.api.confident-ai.comAPI_BASE_URL_EUretryable_exceptions) HTTPHandlerget_async_httpx_clienthttpxSpecialProviderc|jd}|jd}|rtjd||ytjd|y)N exceptiontriesz/Confident AI Error: %s. Retrying: %s time(s)...zRetrying: %s time(s)...)getloggingerror)detailsrrs f/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/deepeval/api.pylog_retry_errorrsB{{;/I;;w'E GTYZ /7ceZdZdZdZdZdZy) HttpMethodsGETPOSTDELETEPUTN)__name__ __module__ __qualname__rrrrrrrr s C D F Crrc8eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) Endpointsz /v1/datasetz /v1/test-runz /v1/tracingz /v1/eventz /v1/feedbackz /v1/promptz/v1/recommend-metricsz /evaluatez/guardz /guardrailsz/generate-baseline-attacksN)rrr DATASET_ENDPOINTTEST_RUN_ENDPOINTTRACING_ENDPOINTEVENT_ENDPOINTFEEDBACK_ENDPOINTPROMPT_ENDPOINTRECOMMEND_ENDPOINTEVALUATE_ENDPOINTGUARD_ENDPOINTGUARDRAILS_ENDPOINTBASELINE_ATTACKS_ENDPOINTr!rrr#r#'sA$&$ N&"O0#N' <rr#cPeZdZd defdZd dedefdZd dedefdZd dedefd Z y) ApiNapi_keyc||_d|d|_|xst|_t |_t tj|_ y)Nzapplication/json)z Content-TypeCONFIDENT_API_KEY) llm_provider) r1_headersr base_api_urlr sync_http_handlerr r LoggingCallbackasync_http_handler)selfr1base_urls r__init__z Api.__init__6sE .!(  %4 !,"8FZFjFj"krmethodurlc|dk7r td |jj||||y#tj$r'}td|j j d}~wt$r}|d}~wwxYw)Nr Only POST requests are supportedr>headersjsonparamsDeepEval logging error: ) Exceptionr7posthttpxHTTPStatusErrorresponsetext)r:r=r>rBrCrDes r _http_requestzApi._http_requestBs V >? ?   " " ' ' (  $$ J6qzz6GHI I G s!2A:"A'' A:3A55A:endpointc|j|j}|j|j||j||}|jdk(r |j Stj|j t|j jd|j#t $r|jcYSwxYw)N)r=r>rBrCrDr) r6valuerMr5 status_coderC ValueErrorrKrdebugrFr)r:r=rNbodyrDr>ress r send_requestzApi.send_requestQs))*8>>*:;''<<MM (  ??c ! xxz!   ,CHHJNN7CHH=> >  xx sB;;CCczK|tjk7r td|j|j} |j j ||j||d{y7#tj$r'}td|jjd}~wt$r}|d}~wwxYww)Nr@rArE) rrrFr6rQr9rGr5rHrIrJrK)r:r=rNrUrDr>rLs ra_send_requestzApi.a_send_requestds [%% %>? ?))*8>>*:; ))..  /   $$ J6qzz6GHI I G sF9B;,A0(A.)A0-B;.A00B8"B%% B81B33B88B;)N)NNN)NN) rrr strr<rMrr#rWrYr!rrr0r05sI l l C c ?;?)?&;)rr0)renumrtypingrrHlitellm._loggingrr__annotations__rrr HTTPErrorr &litellm.llms.custom_httpx.http_handlerr r r rrr#r0r!rrras~ +>5>DeD4 e4::#ooe-8$ = =>>r