o R+ i"@sdZddlZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z dd l mZdd l mZdd l mZdd lmZdd lmZddlZddlmZddlmZddlmZddlZeeZddZ Gddde Z!dS)z3 This module provides a client class for infinite. N)utils)compat) bce_v1_signer) BceBaseClient)bce_http_client)handler) http_headers)http_content_types) http_methods)required)infinite)BceClientError)BceServerError)BceClientConfigurationcCs|jdtjjdkrtd|jdtjjdkr.|}|r(|jd|i| dSt |j |j j d}|j|_| |)Ndz#Can not handle 1xx http status codeZBodyT)Z request_id)statushttpclientCONTINUEr OKread__dict__updatecloserreasonmetadataZbce_request_id status_code) http_responseresponsebodyZbser v/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/baidubce/services/infinite/infinite_client.py_parse_http_response,sr"c@s~eZdZdZdddZ  dddZ   dd d Zdd d Zdd dZe dddZ ddZ dddZ    dddZ dS)InfiniteClientz Infinite sdk client NcCst||dSN)r__init__selfconfigr r r!r%?szInfiniteClient.__init__application/jsonpredictc Cfi}|dur ||d<d|d<||d<tj}|d|}tj|tjti} |jtj |d|| ||dS) aY predict :param endpoint_name: endpoint name :type endpoint_name: string :param body: request data :type body: binary string or dict :param variant_name: variant name or None :type variant_name: string :param content_type: content type,supports application/json,x-image,and x-recordio-protobuf :type content_type: string :param config: None :type config: BceClientConfiguration :param interface: interface_name, several of predict/predict_proba/predict_log_proba/fit_predict/staged_predict/staged_predict are supported depend on frameworks and algorithm used :return: response as following format { Body: 'predict result' } :rtype: baidubce.bce_response.BceResponse Nvariantr*action interface ; charset= /invocations http_method function_namerheadersparamsr( baidubceZDEFAULT_ENCODINGr CONTENT_TYPEBCE_REQUEST_IDuuiduuid4 _send_requestr POST r' endpoint_namerZ variant_name content_typer(r.r5Zdefault_encodingr4r r r!r*Bs$  zInfiniteClient.predictc Cr+) ai debug :param endpoint_name: endpoint name :type endpoint_name: string :param body: request data :type body: binary or dict :param variant_name: variant name or None :type variant_name: string :param content_type: content type,supports application/json,x-image,and x-recordio-protobuf :type content_type: string :param config: None :type config: BceClientConfiguration :param interface: interface_name, several of predict/predict_proba/predict_log_proba/fit_predict/staged_predict/staged_predict are supported depend on frameworks and algorithm used :type config: string :return: response as following format { Body: 'debug info' } :rtype: baidubce.bce_response.BceResponse Nr,debugr-r.r/r0r1r6r>r r r!rAvs$   zInfiniteClient.debugcCs*tjtjtjti}|jtj d||dS)a* get all endpoint :param config: None :type config: BceClientConfiguration :return: response as following format { Body: '{"endpointList":["ep1_name","ep2_name"]}' } :rtype: baidubce.bce_response.BceResponse listr2r3r4r( rr8r JSONr9r:r;r<r GET)r'r(r4r r r!get_endpoint_lists z InfiniteClient.get_endpoint_listcCs.tjtjtjti}|jtj |d||dS)a get endpoint info :param endpoint_name: endpoint name :type endpoint_name: string :param config: None :type config: BceClientConfiguration :return: response as following format { Body: '{ "endpoint_uuid":"ep1", "variant_configs":[ { "variant_uuid":"v1", "variant_name":"v1_name", "...":"..." } ] }' } :rtype: baidubce.bce_response.BceResponse z/inforCrD)r'r?r(r4r r r!get_endpoint_infos z InfiniteClient.get_endpoint_infocCsttjt|Sr$)rZ append_urir Z URL_PREFIXrconvert_to_bytes)r(r3r r r! _get_pathszInfiniteClient._get_pathcCs8|dur ||jSt|j}||||}|Sr$)_convert_configr(copyZmerge_non_none_values)r'r(Z new_configr r r! _merge_configs     zInfiniteClient._merge_configcCsR|dur'|jdurt|j|_|jdur't|jj|j_t|jj|j_|Sr$)ZendpointrrI credentialsZ access_key_idZsecret_access_keyr&r r r!rKs    zInfiniteClient._convert_configc Cs>||}t||}|durt}t|tj|g|||||Sr$)rMr#rJr"r send_requestrsign) r'r2r3rr4r5r(Z body_parserpathr r r!r<s    zInfiniteClient._send_requestr$)Nr)Nr*)NNNNNN)__name__ __module__ __qualname____doc__r%r*rArGrH staticmethodrJrMrKr<r r r r!r#;s*  4 5 #   r#)"rUrLloggingwarningsr7rrZ baidubce.authrZbaidubce.bce_base_clientrZ baidubce.httprrrr r Zbaidubce.utilsr Zbaidubce.servicesr http.clientrZbaidubce.exceptionr rZ!baidubce.bce_client_configurationrr: getLoggerrRZ_loggerr"r#r r r r!s0