o /ifT(@sddlZddlZddlZddlZddlZddlmZmZm Z ddl Z ddl m Z ddlmZddl m Z ddlmZddlmZmZdZd Zd Zd d geed dZehdZGddde ZGdddZdS)N) HTTPAdapterResponseRetry) ClientSession)r)Enum) API_KEY_ENV)KEY_FILE_HANDLER KeyValueszhttps://api.confident-ai.comiiii>GETPOSTDELETEc@seZdZdZdZdZdZdS) Endpointsz /v1/datasetz /v1/test-runz /v1/eventz /v1/feedbackN)__name__ __module__ __qualname__DATASET_ENDPOINTTEST_RUN_ENDPOINTEVENT_ENDPOINTFEEDBACK_ENDPOINTrrE/var/www/html/corbot_env/lib/python3.10/site-packages/deepeval/api.pyrs rc@seZdZdZeeddddddfdefddZe         d!de fdd Z e d e fd d Z      d"d dZ d#ddZd$ddZd%ddZd%ddZe d#dedefddZe dedefddZ      d"ddZ d$dd ZdS)&Apiz3Internal Api reference for handling http operationsNapi_keycCs|dkr ttj}|dks|durtd||_|jdf|_d|||d|_d||i|_ |p4t |_ ||_ ||_ ||_dS)NrzPlease provide a valid API Key.application/json) Content-Type User-AgentCONFIDENT_API_KEYr)r fetch_datar API_KEY ValueErrorr_auth_generate_useragent_headers_headers_multipart_form_data API_BASE_URL base_api_urlverifyproxiescert)selfruser_agent_extensionapi_instance_urlr)r*r+rrr__init__"s      z Api.__init__returnc  Cst} tttttdd} t| d} | d| | r| nd| _ |r#|nd| _ | r.| j | z|p2i}|p6d}| j |||||||d}|WStyV}zt||d}~ww)NF)totalbackoff_factorstatus_forcelistallowed_methodsraise_on_status) max_retrieszhttps://T)methodurlheadersparamsjsonfilesdata)requestsSessionrHTTP_TOTAL_RETRIESHTTP_RETRY_BACKOFF_FACTORHTTP_STATUS_FORCE_LISTHTTP_RETRY_ALLOWED_METHODSrmountr+r)r*updaterequest Exception)r7r8r9authr:bodyr<r=r)r*r+httpsretry_strategyadapterreserrrrr _http_requestEs>     zApi._http_requestrMc Csnz |d|j}Wnty$}z t||j}WYd}~nd}~ww|jdkr3td|tt |)NerrorizDeprecation Warning: ) r;gettextr"print status_codewarningswarnDeprecationWarningrG)rMmessageerrr_raise_on_responsews zApi._raise_on_responsec  Cs|j|} |j|| |||||||j|j|jd } d} | jdkr=z| } W| Sty<} z| WYd} ~ Sd} ~ ww| jdkro| dd} t | d }|dkrid |d <| ||||||||St d dS|| | S) z0Generic HTTP request method with error handling.)r)r*r+NirXzConflict occurred.z [y/N]: yT overwritezAborted.)r(rOr)r*r+rTr;r"rQinputstriplower _api_requestrSrZ)r,r7endpointr9rHr:rIr<r=r8rMr;rYrX user_inputrrrrasF     zApi._api_requestcCs|jd||j|j|dS)zGeneric GET Request Wrapperr )r9rHr:rar%r#)r,rbr:rrr get_requestszApi.get_requestc Cs*|jd||dur |jn|j|j|||dS)zGeneric POST Request Wrapperr Nr9rHrIr<r=)rar%r&r#r,rbrIr<r=rrr post_requestszApi.post_requestcCs|jd||j|j||dS)zGeneric DELETE Request Wrapperr)r9rHr:rIrd)r,rbr:rIrrrdelete_requestszApi.delete_requestcCs|jd|||j|j|dS)zGeneric PUT Request WrapperPUT)rIr9rHr:rd)r,rbrIr:rrr put_requestszApi.put_request extensioncCs6t}t}dtdd|d||g}|S)aGenerates UserAgent parameter with module, Python and OS details Args: extension (str, optional): Option to extend UserAgent with source system Returns: str: Generated UserAgent parameter with platform versions  NzPython/zOS/)platformpython_versionjoinfilter)rlro os_platform user_agentrrrr$s  zApi._generate_useragentrRcCstjj|ddS)alProject and Batch names can be a part of URL, which causes an error in case of a special character used. Quotation assures the right object to be retrieved from API. `quote_string('a bc/def')` -> `a%20bc%2Fdef` Args: text (str): Input text to be quoted Returns: str: Quoted text in return r)safe)urllibparsequote)rRrrr quote_stringszApi.quote_stringc  s|j|} t4IdH} z|r(t}|D] \} } |j| | | dq| j|| |||dIdH} | jdkrkz| IdH}|WWWdIdHSt yj| IdHYWWdIdHSw| j dd}d|vr~| IdH}n| IdH}| jdkrt d |t d | jd |WWdIdHdSty}ztd ||d}~ww1IdHswYdS) z=Generic asynchronous HTTP request method with error handling.N)filename)r7r8r9r:r;r[rrrizError 400: Bad Request - zError z: zHTTP request failed: )r(raiohttpFormDataitems add_fieldrFstatusr;r"rRr9rQrSrG)r,r7rbr9rHr:rIr<r=r8session file_name file_contentrMr; content_typeerror_responserNrrr_api_request_asyncsP    )zApi._api_request_asyncc s2|jd||dur |jn|j|j|||dIdHS)z)Generic asynchronous POST Request Wrapperr Nrf)rr%r&r#rgrrrpost_request_asyncHs zApi.post_request_async) NNNNNNNNN)NNNNNN)N)NNN)NN)rrr__doc__osgetenvrstrr/ staticmethodrrOrZrarerhrirkr$rxrrrrrrrsd  # 1 2    :r)rrn urllib.parserur>rUrequests.adaptersrrrrzrenumrdeepeval.constantsrdeepeval.key_handlerrr r'r@rAlistrangerB frozensetrCrrrrrrs&