i}ddlmZddlmZmZddlmZddlZddlm 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..X(5L5Lc!ff5L5L5L5L5L5Lr2message-batches-2024-09-24/v1/messages/batches?beta=truer=r9r:r;r<bodyoptionscast_to) rrjoinrr_postrr"BatchCreateParamsrr$r/r=r8r9r:r;r<s r0createzBatches.create1sH  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a zz , *h!79L9^__(+Q[el%    r2message_batch_idrKc 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://docs.claude.com/en/docs/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 rFrGc34K|]}t|VdSrIrJrLs r0rOz#Batches.retrieve..rPr2rQ1/v1/messages/batches/{message_batch_id}?beta=truer]rSrVrW ValueErrorrrrXrr_getrrr$r/r]r8r9r:r;r<s r0retrievezBatches.retrieveisD  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a yy M`p q q q(+Q[el%    r2after_id before_idlimitr8r9r:r;r<rk str | Omitrlrm int | OmitSyncPage[BetaMessageBatch]c|itdt|r.dtd|Ddgnti|pi}ddi|pi}|dt tt||||t|||dtj tS) 4List 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://docs.claude.com/en/docs/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 rFrGc34K|]}t|VdSrIrJrLs r0rOzBatches.list..rPr2rQrRrkrlrmr9r:r;r<querypagerVmodel) rrrXrr _get_api_listrr$rrr!BatchListParams r/rkrlrmr8r9r: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://docs.claude.com/en/docs/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 r`rFrGc34K|]}t|VdSrIrJrLs r0rOz!Batches.delete..rPr2rQrbrcrSrd rfrrrXrr_deleterrr%rhs 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) rBatches 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://docs.claude.com/en/docs/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 r`rFrGc34K|]}t|VdSrIrJrLs r0rOz!Batches.cancel..NrPr2rQ8/v1/messages/batches/{message_batch_id}/cancel?beta=truercrSrd rfrrrXrrrYrrr$rhs 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://docs.claude.com/en/docs/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 r`rcBNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryrFrGc34K|]}t|VdSrIrJrLs r0rOz"Batches.results..rPr2rQrSTrVrWstream)rfri results_urlrprocessing_statusrrrXrrrgrrrr&r/r]r8r9r:r;r<batchs r0resultszBatches.results`s^F  us_osstt t /? @@   nUZUlnn "#7QMN O O O(+Q[el!!CD    r2)r*r+)r*r3r=r>r8r?r9r@r:rAr;rBr<rCr*r$r]rKr8r?r9r@r:rAr;rBr<rCr*r$)rkrnrlrnrmror8r?r9r@r:rAr;rBr<rCr*rpr]rKr8r?r9r@r:rAr;rBr<rCr*r%)r]rKr8r?r9r@r:rAr;rBr<rCr*r__name__ __module__ __qualname__rr1r6rrr\rir}rrrr2r0r'r's,,,_,222_226)-$("&;D6 6 6 6 6 6 x26)-$("&;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'dZedddedd(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?r9r@r:rAr;rBr<rCr$c rKitdt|r.dtd|Ddgnti|pi}ddi|pi}|dt d|itjd{Vt||||t d{VS) rErFrGc34K|]}t|VdSrIrJrLs r0rOz&AsyncBatches.create..rPr2rQrRr=NrSrT) rrrXrrrYrr"rZrr$r[s r0r\zAsyncBatches.createsH  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a ZZ ,,j(-CEXEjkkkkkkkk(+Q[el%           r2r]rKc vK|std|itdt|r.dt d|Ddgnt i|pi}ddi|pi}|td|t||||t d {VS) r_r`rFrGc34K|]}t|VdSrIrJrLs r0rOz(AsyncBatches.retrieve..rPr2rQrbrcrSrdNrerhs r0rizAsyncBatches.retrievesD  us_osstt t  $'#chhu5L5Le5L5L5LOkNl/m/m&n&n&n"  "  *+GaML_]_a YY M`p q q q(+Q[el%          r2rjrkrnrlrmro=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]c|itdt|r.dtd|Ddgnti|pi}ddi|pi}|dt tt||||t|||dtj tS) rrrFrGc34K|]}t|VdSrIrJrLs r0rOz$AsyncBatches.list..SrPr2rQrRrtrurw) rrrXrrrzrr$rrr!r{r|s r0r}zAsyncBatches.list%sV  $'#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) rr`rFrGc34K|]}t|VdSrIrJrLs r0rOz&AsyncBatches.delete..rPr2rQrbrcrSrdNrrhs r0rzAsyncBatches.deleteosD  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) rr`rFrGc34K|]}t|VdSrIrJrLs r0rOz&AsyncBatches.cancel..rPr2rQrrcrSrdNrrhs 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)rr`rcNrrrrFrGc34K|]}t|VdSrIrJrLs r0rOz'AsyncBatches.results..rPr2rQrSTr)rfrirrrrrrXrrrgrrr r&rs r0rzAsyncBatches.resultssF  us_osstt tmm5EmFFFFFFFF   nUZUlnn "#7QMN O O O(+Q[el&&HI          r2)r*r)r*rrr)rkrnrlrnrmror8r?r9r@r:rAr;rBr<rCr*rr)r]rKr8r?r9r@r:rAr;rBr<rCr*rrrr2r0r(r(s111_1777_726)-$("&;D6 6 6 6 6 6 x26)-$("&;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|_dSrI) _batchesrto_raw_response_wrapperr\rir}rrrr/rs r0__init__zBatchesWithRawResponse.__init__(s &> N   )@     %< L   '> N   '> N   (? O   r2Nrr'r*rrrrrrr2r0r+r+'(      r2r+ceZdZddZdS)rrr(r*rc|||_tj|j|_tj|j|_tj|j|_tj|j|_tj|j|_tj|j|_dSrI) rrasync_to_raw_response_wrapperr\rir}rrrrs r0rz$AsyncBatchesWithRawResponse.__init__@s &D N   )F     %B L   'D N   'D N   (E O   r2Nrr(r*rrrr2r0rr?rr2rceZdZddZdS)r3rr'r*rc@||_t|j|_t|j|_t|j|_t|j|_t|j|_t|j|_dSrI)rrr\rir}rrrrs r0rz%BatchesWithStreamingResponse.__init__Xs 2 N   5     1 L   3 N   3 N   4 O   r2Nrrrr2r0r3r3Wrr2r3ceZdZddZdS)rrr(r*rc@||_t|j|_t|j|_t|j|_t|j|_t|j|_t|j|_dSrI)rrr\rir}rrrrs r0rz*AsyncBatchesWithStreamingResponse.__init__ps 8 N   ;     7 L   9 N   9 N   : O   r2Nrrrr2r0rrorr2r)= __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 %B B B B B oB B B J B B B B B #B B B J         0        0        0          r2