o 6Wj@sddlmZddlmZmZddlmZddlZddlm 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/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddgZ6GdddeZ7GdddeZ8GdddZ9GdddZ:GdddZ;Gdd d Z^z!Batches.create..message-batches-2024-09-24anthropic-betazanthropic-user-profile-idrS/v1/messages/batches?beta=truer>r:r;r<r=bodyoptionscast_to) rrjoinrr_postrr!BatchCreateParamsrr#r.r>r8r9r:r;r<r=r/r/r0create1s(*  zBatches.creater8r:r;r<r=message_batch_idrKcC|s td|itdt|rdtdd|Ddgnti|p%i}ddi|p-i}|jtd|dt||||d t d 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 rSrGcsrHrIrJrLr/r/r0rOrPz#Batches.retrieve..rQ1/v1/messages/batches/{message_batch_id}?beta=truer`rUrXrY ValueErrorrrrZrr_getrrr#r.r`r8r:r;r<r=r/r/r0retrievep*"  zBatches.retrieveafter_id before_idlimitr8r:r;r<r=rnrorp int | OmitSyncPage[BetaMessageBatch]c Citdt|rdtdd|Ddgnti|pi}ddi|p$i}|jdttt||||t |||dt j dtd S) 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 rSrGcsrHrIrJrLr/r/r0rOrPzBatches.list..rQrTrnrorpr:r;r<r=querypagerXmodel) rrrZrr _get_api_listrr#rrr BatchListParams r.rnrorpr8r:r;r<r=r/r/r0list<+ z Batches.listr$cCra)  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 rcrSrGcsrHrIrJrLr/r/r0rOrPz!Batches.delete..rQrdrerUrf rhrrrZrr_deleterrr$rjr/r/r0deleterlzBatches.deletecCra) 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 rcrSrGcsrHrIrJrLr/r/r0rOUrPz!Batches.cancel..rQ8/v1/messages/batches/{message_batch_id}/cancel?beta=truererUrf rhrrrZrrr[rrr#rjr/r/r0cancel(s.( zBatches.cancel0JSONLDecoder[BetaMessageBatchIndividualResponse]cCs|s td||j|d}|jstd|jddi|p i}itdt|r7dtdd |Dd gnt i|p>i}dd i|pFi}|j t |j|dt ||||d t td d 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 rcreBNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryrSrGcsrHrIrJrLr/r/r0rOrPz"Batches.results..rQrUTrXrYstream)rhrk results_urlrprocessing_statusrrrZrrrirrrr%r.r`r8r:r;r<r=batchr/r/r0resultsgs8#    zBatches.results)r)r*)r)r3r>r?r8r@r9rAr:rBr;rCr<rDr=rEr)r#r`rKr8r@r:rBr;rCr<rDr=rEr)r#)rnrArorArprqr8r@r:rBr;rCr<rDr=rEr)rrr`rKr8r@r:rBr;rCr<rDr=rEr)r$)r`rKr8r@r:rBr;rCr<rDr=rEr)r__name__ __module__ __qualname__rr1r5r rr^rkr~rrrr/r/r/r0r&Z   C:N;Cc @r()7r'r)AsyncBatchesWithRawResponsecCr+r,)rr-r/r/r0r1r2zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr+r4)rr-r/r/r0r5r6z$AsyncBatches.with_streaming_responseNr7r>r?r8r@r9rAr:rBr;rCr<rDr=rEr#c sitt|rdtdd|Ddgnt|d|pi}ddi|p&i}|jdtd|itjId Ht ||||d t d Id HS) rFrGcsrHrIrJrLr/r/r0rOrPz&AsyncBatches.create..rQrRrSrTr>NrUrV) rrrZrrr[rr!r\rr#r]r/r/r0r^s**   zAsyncBatches.creater_r`rKc|s td|itdt|rdtdd|Ddgnti|p&i}ddi|p.i}|jtd|dt||||d t d Id HS) rbrcrSrGcsrHrIrJrLr/r/r0rO#rPz(AsyncBatches.retrieve..rQrdrerUrfNrgrjr/r/r0rk,"   zAsyncBatches.retrievermrnrorprq=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]c Crs) rtrSrGcsrHrIrJrLr/r/r0rOarPz$AsyncBatches.list..rQrTrurvrx) rrrZrrr{rr#rrr r|r}r/r/r0r~3rzAsyncBatches.listr$cr) rrcrSrGcsrHrIrJrLr/r/r0rOrPz&AsyncBatches.delete..rQrdrerUrfNrrjr/r/r0r}rzAsyncBatches.deletecr) rrcrSrGcsrHrIrJrLr/r/r0rOrPz&AsyncBatches.cancel..rQrrerUrfNrrjr/r/r0rs0(  zAsyncBatches.cancel5AsyncJSONLDecoder[BetaMessageBatchIndividualResponse]cs|s td||j|dIdH}|jstd|jddi|p$i}itdt|r;dtd d |Dd gnt i|pBi}dd i|pJi}|j t |j|dt ||||d t td dIdHS)rrcreNrrrrSrGcsrHrIrJrLr/r/r0rO#rPz'AsyncBatches.results..rQrUTr)rhrkrrrrrrZrrrirrrr%rr/r/r0rs:#    zAsyncBatches.results)r)r)r)rrr)rnrArorArprqr8r@r:rBr;rCr<rDr=rEr)rr)r`rKr8r@r:rBr;rCr<rDr=rEr)rrr/r/r/r0r'rc@eZdZdddZdS) r*batchesr&r)NonecC^||_t|j|_t|j|_t|j|_t|j|_t|j|_t|j|_dSrI) _batchesrto_raw_response_wrapperr^rkr~rrrr.rr/r/r0__init__6& zBatchesWithRawResponse.__init__Nrr&r)rrrrrr/r/r/r0r*5r*c@r) rrr'r)rcCrrI) rrasync_to_raw_response_wrapperr^rkr~rrrrr/r/r0rNrz$AsyncBatchesWithRawResponse.__init__Nrr'r)rrr/r/r/r0rMrrc@r) r3rr&r)rcCR||_t|j|_t|j|_t|j|_t|j|_t|j|_t|j|_dSrI)rrr^rkr~rrrrr/r/r0rf& z%BatchesWithStreamingResponse.__init__Nrrr/r/r/r0r3err3c@r) rrr'r)rcCrrI)rrr^rkr~rrrrr/r/r0r~rz*AsyncBatchesWithStreamingResponse.__init__Nrrr/r/r/r0r}rr)= __future__rtypingrr itertoolsrhttpxr_typesrr r r r r r_utilsrrrrr_compatr _resourcerr _responserr paginationrr _exceptionsr _base_clientrr_decoders.jsonlrrtypes.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*rr3rr/r/r/r0s@   $