܎jddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZmZmZmZdd lmZdd lmZmZdd lmZmZdd l m!Z!m"Z"dd l#m$Z$ddl%m&Z&m'Z'ddl(m)Z)m*Z*ddl+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddgZ6GddeZ7GddeZ8GddZ9GddZ:GddZ;Gdd Zz!Batches.create..^(5L5Lc!ff5L5L5L5L5L5Lr2message-batches-2024-09-24anthropic-betazanthropic-user-profile-idrT/v1/messages/batches?beta=truer>r:r;r<r=bodyoptionscast_to) rrjoinrr_postrr"BatchCreateParamsrr$r/r>r8r9r:r;r<r=s r0createzBatches.create1sT   '#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"1@   "  *+GaML_]_a zz , *h!79L9^__(+Q[el%    r2r8r:r;r<r=message_batch_idrLc f|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t S) This endpoint is idempotent and can be used to poll for Message Batch completion. To access the results of a Message Batch, make a request to the `results_url` field in the response. Learn more about the Message Batches API in our [user guide](https://platform.claude.com/docs/en/build-with-claude/batch-processing) Args: message_batch_id: ID of the Message Batch. betas: Optional header to specify the beta version(s) you want to use. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds ?Expected a non-empty value for `message_batch_id` but received rTrHc34K|]}t|VdSrJrKrMs r0rPz#Batches.retrieve..rQr2rR1/v1/messages/batches/{message_batch_id}?beta=truerarVrYrZ ValueErrorrrr[rr_getrrr$r/rar8r:r;r<r=s r0retrievezBatches.retrievepsD  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a yy M`p q q q(+Q[el%    r2after_id before_idlimitr8r:r;r<r=rorprq int | OmitSyncPage[BetaMessageBatch]c|itdt|r.dtd|Ddgnti|pi}ddi|pi}|dt tt||||t|||dtj tS) 8List all Message Batches within a Workspace. Most recently created batches are returned first. Learn more about the Message Batches API in our [user guide](https://platform.claude.com/docs/en/build-with-claude/batch-processing) Args: after_id: ID of the object to use as a cursor for pagination. When provided, returns the page of results immediately after this object. before_id: ID of the object to use as a cursor for pagination. When provided, returns the page of results immediately before this object. limit: Number of items to return per page. Defaults to `20`. Ranges from `1` to `1000`. betas: Optional header to specify the beta version(s) you want to use. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds rTrHc34K|]}t|VdSrJrKrMs r0rPzBatches.list..rQr2rRrUrorprqr:r;r<r=querypagerYmodel) rrr[rr _get_api_listrr$rrr!BatchListParams r/rorprqr8r:r;r<r=s r0listz Batches.listsV  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a !! ,*+(+'%%$,%.!& &5    ##"   r2r%c f|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t S)  Delete a Message Batch. Message Batches can only be deleted once they've finished processing. If you'd like to delete an in-progress batch, you must first cancel it. Learn more about the Message Batches API in our [user guide](https://platform.claude.com/docs/en/build-with-claude/batch-processing) Args: message_batch_id: ID of the Message Batch. betas: Optional header to specify the beta version(s) you want to use. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds rdrTrHc34K|]}t|VdSrJrKrMs r0rPz!Batches.delete..rQr2rRrfrgrVrh rjrrr[rr_deleterrr%rls r0deletezBatches.deletesD  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a || M`p q q q(+Q[el,    r2c f|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t S) vBatches may be canceled any time before processing ends. Once cancellation is initiated, the batch enters a `canceling` state, at which time the system may complete any in-progress, non-interruptible requests before finalizing cancellation. The number of canceled requests is specified in `request_counts`. To determine which requests were canceled, check the individual results within the batch. Note that cancellation may not result in any canceled requests if they were non-interruptible. Learn more about the Message Batches API in our [user guide](https://platform.claude.com/docs/en/build-with-claude/batch-processing) Args: message_batch_id: ID of the Message Batch. betas: Optional header to specify the beta version(s) you want to use. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds rdrTrHc34K|]}t|VdSrJrKrMs r0rPz!Batches.cancel..UrQr2rR8/v1/messages/batches/{message_batch_id}/cancel?beta=truergrVrh rjrrr[rrr\rrr$rls r0cancelzBatches.cancel(sP  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a zz J]m   )+Q[el%   r20JSONLDecoder[BetaMessageBatchIndividualResponse]c |std|||}|jstd|jddi|pi}it dt |r.dtd|Dd gnti|pi}dd i|pi}| t|j|t|||| ttd S)  Streams the results of a Message Batch as a `.jsonl` file. Each line in the file is a JSON object containing the result of a single request in the Message Batch. Results are not guaranteed to be in the same order as requests. Use the `custom_id` field to match results to requests. Learn more about the Message Batches API in our [user guide](https://platform.claude.com/docs/en/build-with-claude/batch-processing) Args: message_batch_id: ID of the Message Batch. betas: Optional header to specify the beta version(s) you want to use. extra_headers: Send extra headers extra_query: Add additional query parameters to the request extra_body: Add additional JSON properties to the request timeout: Override the client-level default timeout for this request, in seconds rdrgBNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryrTrHc34K|]}t|VdSrJrKrMs r0rPz"Batches.results..rQr2rRrVTrYrZstream)rjrm results_urlrprocessing_statusrrr[rrrkrrrr&r/rar8r:r;r<r=batchs r0resultszBatches.resultsgs^F  us_osstt t /? @@   nUZUlnn "#7QMN O O O(+Q[el!!CD    r2)r*r+)r*r3r>r?r8r@r9rAr:rBr;rCr<rDr=rEr*r$rarLr8r@r:rBr;rCr<rDr=rEr*r$)rorArprArqrrr8r@r:rBr;rCr<rDr=rEr*rsrarLr8r@r:rBr;rCr<rDr=rEr*r%)rarLr8r@r:rBr;rCr<rDr=rEr*r__name__ __module__ __qualname__rr1r6rrr_rmrrrrr2r0r'r's,,,_,222_226&*)-$("&;D= = = = = = F26)-$("&;D5 5 5 5 5 5 t $ $ 15)-$("&;DH H H H H H \26)-$("&;D5 5 5 5 5 5 v26)-$("&;D= = = = = = F26)-$("&;D? ? ? ? ? ? ? ? r2c eZdZed(dZed)dZeedddedd*dZedddedd+dZ eeeedddedd,d"Z edddedd-d$Z edddedd+d%Z edddedd.d'Z dS)/r(r*AsyncBatchesWithRawResponsec t|Sr-)rr.s r0r1zAsyncBatches.with_raw_responses+4000r2!AsyncBatchesWithStreamingResponsec t|Sr5)rr.s r0r6z$AsyncBatches.with_streaming_responses1666r2Nr7r>r?r8r@r9rAr:rBr;rCr<rDr=rEr$c tKitt|r.dtd|Ddgnt|d|pi}ddi|pi}|dt d|itjd{Vt|||| t d{VS) rGrHc34K|]}t|VdSrJrKrMs r0rPz&AsyncBatches.create..rQr2rRrSrTrUr>NrVrW) rrr[rrr\rr"r]rr$r^s r0r_zAsyncBatches.createsT   '#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"1@   "  *+GaML_]_a ZZ ,,j(-CEXEjkkkkkkkk(+Q[el%           r2r`rarLc vK|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t d {VS) rcrdrTrHc34K|]}t|VdSrJrKrMs r0rPz(AsyncBatches.retrieve..#rQr2rRrfrgrVrhNrirls r0rmzAsyncBatches.retrievesD  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a YY M`p q q q(+Q[el%          r2rnrorprqrr=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]c|itdt|r.dtd|Ddgnti|pi}ddi|pi}|dt tt||||t|||dtj tS) rurTrHc34K|]}t|VdSrJrKrMs r0rPz$AsyncBatches.list..arQr2rRrUrwrxrz) rrr[rrr}rr$rrr!r~rs r0rzAsyncBatches.list3sV  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a !! ,+,(+'%%$,%.!& &5    ##"   r2r%c vK|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t d {VS) rrdrTrHc34K|]}t|VdSrJrKrMs r0rPz&AsyncBatches.delete..rQr2rRrfrgrVrhNrrls r0rzAsyncBatches.delete}sD  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a \\ M`p q q q(+Q[el, "         r2c vK|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t d {VS) rrdrTrHc34K|]}t|VdSrJrKrMs r0rPz&AsyncBatches.cancel..rQr2rRrrgrVrhNrrls r0rzAsyncBatches.cancels&P  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a ZZ J]m   )+Q[el%          r25AsyncJSONLDecoder[BetaMessageBatchIndividualResponse]c K|std|||d{V}|jstd|jddi|pi}it dt |r.dtd |Dd gnti|pi}dd i|pi}| t|j|t|||| ttd d{VS)rrdrgNrrrrTrHc34K|]}t|VdSrJrKrMs r0rPz'AsyncBatches.results..#rQr2rRrVTr)rjrmrrrrrr[rrrkrrr r&rs r0rzAsyncBatches.resultssF  us_osstt tmm5EmFFFFFFFF   nUZUlnn "#7QMN O O O(+Q[el&&HI          r2)r*r)r*rrr)rorArprArqrrr8r@r:rBr;rCr<rDr=rEr*rr)rarLr8r@r:rBr;rCr<rDr=rEr*rrrr2r0r(r(s111_1777_726&*)-$("&;D= = = = = = F26)-$("&;D5 5 5 5 5 5 t $ $ 15)-$("&;DH H H H H H \26)-$("&;D5 5 5 5 5 5 v26)-$("&;D= = = = = = F26)-$("&;D? ? ? ? ? ? ? ? r2ceZdZddZdS)r+batchesr'r*Nonec|||_tj|j|_tj|j|_tj|j|_tj|j|_tj|j|_tj|j|_dSrJ) _batchesrto_raw_response_wrapperr_rmrrrrr/rs r0__init__zBatchesWithRawResponse.__init__6s &> N   )@     %< L   '> N   '> N   (? O   r2Nrr'r*rrrrrrr2r0r+r+5(      r2r+ceZdZddZdS)rrr(r*rc|||_tj|j|_tj|j|_tj|j|_tj|j|_tj|j|_tj|j|_dSrJ) rrasync_to_raw_response_wrapperr_rmrrrrrs r0rz$AsyncBatchesWithRawResponse.__init__Ns &D N   )F     %B L   'D N   'D N   (E O   r2Nrr(r*rrrr2r0rrMrr2rceZdZddZdS)r3rr'r*rc@||_t|j|_t|j|_t|j|_t|j|_t|j|_t|j|_dSrJ)rrr_rmrrrrrs r0rz%BatchesWithStreamingResponse.__init__fs 2 N   5     1 L   3 N   3 N   4 O   r2Nrrrr2r0r3r3err2r3ceZdZddZdS)rrr(r*rc@||_t|j|_t|j|_t|j|_t|j|_t|j|_t|j|_dSrJ)rrr_rmrrrrrs r0rz*AsyncBatchesWithStreamingResponse.__init__~s 8 N   ;     7 L   9 N   9 N   : O   r2Nrrrr2r0rr}rr2r)= __future__rtypingrr itertoolsrhttpxr_typesr r r r r rr_utilsrrrrr_compatr _resourcerr _responserr paginationrr _exceptionsr _base_clientrr_decoders.jsonlrr types.beta.messagesr!r"types.anthropic_beta_paramr#&types.beta.messages.beta_message_batchr$.types.beta.messages.beta_deleted_message_batchr%:types.beta.messages.beta_message_batch_individual_responser&__all__r'r(r+rr3rrr2r0rs#"""""!!!!!!!! !!!!!!LLLLLLLLLLLLLLLLLLgggggggggggggg'''''';;;;;;;;ZZZZZZZZ........******AAAAAAAA????????JJJJJJJJ======GGGGGGVVVVVVmmmmmm n %I I I I I oI I I X I I I I I #I I I X         0        0        0          r2