o iS@s"ddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(e)ej*dZ+GdddZ,GdddZ-dS)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)'convert_and_respect_annotation_metadata)construct_type)BadRequestError)ForbiddenError)InternalServerError) NotFoundError)ServiceUnavailableError)UnauthorizedError)Batch)CancelBatchResponse)CreateBatchResponse)GetBatchResponse)ListBatchesResponse.c @eZdZdefddZddddddejedejedejed eje d e e f d d Z dd de d eje d e efddZdd ded eje d e efddZdd ded eje d e efddZdS)RawBatchesClientclient_wrappercC ||_dSN_client_wrapperselfrr"_/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/batches/raw_client.py__init__ zRawBatchesClient.__init__N page_size page_tokenorder_byrequest_optionsr'r(r)r*returnc Cs*|jjjdd|||d|d}zd|jkrdkr0nntttt|d}t ||dWS|jd krPt t |j ttj tjttj tj|dd |jd krptt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |}Wnty t|jt |j |jdwt|jt |j |d)a List the batches for the current user Parameters ---------- page_size : typing.Optional[int] The maximum number of batches to return. The service may return fewer than this value. If unspecified, at most 50 batches will be returned. The maximum value is 1000; values above 1000 will be coerced to 1000. page_token : typing.Optional[str] A page token, received from a previous `ListBatches` call. Provide this to retrieve the subsequent page. order_by : typing.Optional[str] Batches can be ordered by creation time or last updated time. Use `created_at` for creation time or `updated_at` for last updated time. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListBatchesResponse] A successful response. v2/batchesGETr'r(r)methodparamsr*,type_object_responsedataheadersbody status_coder<r=)r httpx_clientrequestrDtypingcastrr jsonrr dictr<OptionalAnyrrrrrrrtextr!r'r(r)r* _response_data_response_jsonr"r"r#list!s#                    zRawBatchesClient.listr*rFc Cs4|jjjddt|tddddi|td}zd|jkr d kr5nntt t t | d }t ||d WS|jd krUt t|jttjtjt tjtj| d d |jdkrutt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d | }Wntyt|jt|j|jdwt|jt|j|d)ag Creates and executes a batch from an uploaded dataset of requests Parameters ---------- request : Batch request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateBatchResponse] A successful response. r,POSTwriter6 annotation direction content-typeapplication/jsonr0rIr<r*omitr2r3r4r7r:r;r>r?r@rArBrC)rrErFr rOMITrDrGrHrr rIrr rJr<rKrLrrrrrrrrMr!rFr*rOrPrQr"r"r#creates                     zRawBatchesClient.createidc Cs*|jjjdt|d|d}zd|jkrdkr0nntttt| d}t ||dWS|jdkrPt t |j ttjtjttjtj| dd |jd krptt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wnty t|jt |j |jdwt|jt |j |d)aG Retrieves a batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetBatchResponse] A successful response. v2/batches/r-r0r*r2r3r4r7r:r;r>r?r@rArBrC)rrErFr rDrGrHrr rIrr rJr<rKrLrrrrrrrrMr!r`r*rOrPrQr"r"r#retrieve s                    zRawBatchesClient.retrievec Cs,|jjjdt|dd|d}zd|jkrdkr1nntttt| d}t ||dWS|jd krQt t |j ttjtjttjtj| dd |jd krqtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wnty t|jt |j |jdwt|jt |j |d)aU Cancels an in-progress batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CancelBatchResponse] A successful response. ra:cancelrTrbr2r3r4r7r:r;r>r?r@rArBrC)rrErFr rDrGrHrr rIrr rJr<rKrLrrrrrrrrMrcr"r"r#cancelus                   zRawBatchesClient.cancel)__name__ __module__ __qualname__rr$rGrKintstrr rrrRrrr_rrdrrfr"r"r"r#rR  n jrc @r)AsyncRawBatchesClientrcCrrrr r"r"r#r$r%zAsyncRawBatchesClient.__init__Nr&r'r(r)r*r+c s2|jjjdd|||d|dIdH}zd|jkrdkr4nntttt|d}t ||d WS|jd krTt t |j ttj tjttj tj|dd |jd krttt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |}Wnty t|jt |j |jdwt|jt |j |d)a List the batches for the current user Parameters ---------- page_size : typing.Optional[int] The maximum number of batches to return. The service may return fewer than this value. If unspecified, at most 50 batches will be returned. The maximum value is 1000; values above 1000 will be coerced to 1000. page_token : typing.Optional[str] A page token, received from a previous `ListBatches` call. Provide this to retrieve the subsequent page. order_by : typing.Optional[str] Batches can be ordered by creation time or last updated time. Use `created_at` for creation time or `updated_at` for last updated time. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListBatchesResponse] A successful response. r,r-r.r/Nr2r3r4r7r:r;r>r?r@rArBrC)rrErFrDrGrHrr rIrr rJr<rKrLrrrrrrrrMrNr"r"r#rRs#                     zAsyncRawBatchesClient.listrSrFc s<|jjjddt|tddddi|tdIdH}zd |jkr$d kr9nntt t t | d }t ||d WS|jd krYt t|jttjtjt tjtj| d d|jdkrytt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d| }Wntyt|jt|j|jdwt|jt|j|d)al Creates and executes a batch from an uploaded dataset of requests Parameters ---------- request : Batch request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CreateBatchResponse] A successful response. r,rTrUrVrYrZr[Nr2r3r4r7r:r;r>r?r@rArBrC)rrErFr rr]rDrGrHrr rIrr rJr<rKrLrrrrrrrrMr^r"r"r#r_as                      zAsyncRawBatchesClient.creater`c s2|jjjdt|d|dIdH}zd|jkrdkr4nntttt| d}t ||dWS|jd krTt t |j ttjtjttjtj| dd |jd krttt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wnty t|jt |j |jdwt|jt |j |d)aL Retrieves a batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetBatchResponse] A successful response. rar-rbNr2r3r4r7r:r;r>r?r@rArBrC)rrErFr rDrGrHrr rIrr rJr<rKrLrrrrrrrrMrcr"r"r#rds                     zAsyncRawBatchesClient.retrievec s4|jjjdt|dd|dIdH}zd|jkr dkr5nntttt| d}t ||d WS|jd krUt t |j ttjtjttjtj| dd |jd krutt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)aZ Cancels an in-progress batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CancelBatchResponse] A successful response. rarerTrbNr2r3r4r7r:r;r>r?r@rArBrC)rrErFr rDrGrHrr rIrr rJr<rKrLrrrrrrrrMrcr"r"r#rf7s                    zAsyncRawBatchesClient.cancel)rgrhrirr$rGrKrjrkr rrrRrrr_rrdrrfr"r"r"r#rmrlrm).rG json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.serializationr core.unchecked_base_modelr errors.bad_request_errorr errors.forbidden_errorrerrors.internal_server_errorrerrors.not_found_errorr errors.service_unavailable_errorrerrors.unauthorized_errorr types.batchrtypes.cancel_batch_responsertypes.create_batch_responsertypes.get_batch_responsertypes.list_batches_responserrHrLr]rrmr"r"r"r#s4                 E