AXjddlmZddlmZmZddlmZddlZddlm 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..^5Lec!femessage-batches-2024-09-24anthropic-betazanthropic-user-profile-idrL/v1/messages/batches?beta=truerequestsr9r:r;r<bodyoptionscast_to) rrjoinrr_postrr"BatchCreateParamsrr$r.rNr7r8r9r:r;r<s r/createzBatches.create1sT   '*hhu5Le5LOkNl/m&n"1@   "  *+GaML_]_a zz , *h!79L9^9^_(+Q[el%   r1r7r9r:r;r<c $|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j 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 rLr?c32K|]}t|ywrArBrDs r/rGz#Batches.retrieve..rHrIrJ1/v1/messages/batches/{message_batch_id}?beta=truemessage_batch_idrOrRrS ValueErrorrrrTrr_getrrr$r.r`r7r9r:r;r<s r/retrievezBatches.retrievepsD ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a yy M`p q(+Q[el%   r1after_id before_idlimitr7r9r:r;r<c@itdt|r#djtd|Ddgnti|xsi}ddi|xsi}|j dt tt||||t|||dtjtS) 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 rLr?c32K|]}t|ywrArBrDs r/rGzBatches.list..rHrIrJrMrhrirjr9r:r;r<querypagerRmodel) rrrTrr _get_api_listrr$rrr!BatchListParams r.rhrirjr7r9r:r;r<s r/listz Batches.listsV  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a !! ,*+(+'%%$,%.!& &55  ##"  r1c $|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j 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 r\rLr?c32K|]}t|ywrArBrDs r/rGz!Batches.delete..rHrIrJr^r_rOra rcrrrTrr_deleterrr%res r/deletezBatches.deletesD ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a || M`p q(+Q[el,   r1c $|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j 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 r\rLr?c32K|]}t|ywrArBrDs r/rGz!Batches.cancel..UrHrIrJ8/v1/messages/batches/{message_batch_id}/cancel?beta=truer_rOra rcrrrTrrrUrrr$res r/cancelzBatches.cancel(sP ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a zz J]m )+Q[el%  r1c |std||j|}|jstd|jddi|xsi}it dt |r#djtd|Dd gnti|xsi}dd i|xsi}|jt|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 r\r_BNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryrLr?c32K|]}t|ywrArBrDs r/rGz"Batches.results..rHrIrJrOTrRrSstream)rcrf results_urlrprocessing_statusrrrTrrrdrrrr&r.r`r7r9r:r;r<batchs r/resultszBatches.resultsgs!F ^_o^rst t /? @   TUZUlUlTmn "#7QMN O(+Q[el!!CD   r1)returnr,)rr4rNz%Iterable[batch_create_params.Request]r7List[AnthropicBetaParam] | Omitr8 str | Omitr9Headers | Noner: Query | Noner; Body | Noner<'float | httpx.Timeout | None | NotGivenrr$r`rCr7rr9rr:rr;rr<rrr$)rhrrirrj int | Omitr7rr9rr:rr;rr<rrzSyncPage[BetaMessageBatch]r`rCr7rr9rr:rr;rr<rrr%)r`rCr7rr9rr:rr;rr<rrz0JSONLDecoder[BetaMessageBatchIndividualResponse]__name__ __module__ __qualname__rr0r5rrrXrfrwr}rrr1r/r'r's,,2226&*)-$("&;D= 8= / = $ = &= "=  = 9=  = F26)-$("&;D5 5 / 5 &5 "5  5 95  5 t $ $ 15)-$("&;DH H  H  H / H &H "H  H 9H  $H \26)-$("&;D5 5 / 5 &5 "5  5 95  !5 v26)-$("&;D= = / = &= "=  = 9=  = F26)-$("&;D? ? / ? &? "?  ? 9?  :? r1c deZdZed dZeddZeeddded ddZeddded ddZ eeeeddded dd Z eddded dd Z eddded dd Z eddded dd Z y)r(ct|Sr+)AsyncBatchesWithRawResponser-s r/r0zAsyncBatches.with_raw_responses+400r1ct|Sr3)!AsyncBatchesWithStreamingResponser-s r/r5z$AsyncBatches.with_streaming_responses166r1Nr6c VKitt|r#djtd|Ddgnt|d|xsi}ddi|xsi}|j dt d|itjd{t|||| t d{S7$7w) r>r?c32K|]}t|ywrArBrDs r/rGz&AsyncBatches.create..rHrIrJrKrLrMrNNrOrP) rrrTrrrUrr"rVrr$rWs r/rXzAsyncBatches.createsT   '*hhu5Le5LOkNl/m&n"1@   "  *+GaML_]_a ZZ ,,j(-CEXEjEjkk(+Q[el%     k s$A>B)B% B) B'!B)'B)rYc @K|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j td|t||||t d {S7w) r[r\rLr?c32K|]}t|ywrArBrDs r/rGz(AsyncBatches.retrieve..#rHrIrJr^r_rOraNrbres r/rfzAsyncBatches.retrievesD ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a YY M`p q(+Q[el%     BBBBrgc@itdt|r#djtd|Ddgnti|xsi}ddi|xsi}|j dt tt||||t|||dtjtS) rlrLr?c32K|]}t|ywrArBrDs r/rGz$AsyncBatches.list..arHrIrJrMrnrorq) rrrTrrrtrr$rrr!rurvs r/rwzAsyncBatches.list3sV  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a !! ,+,(+'%%$,%.!& &55  ##"  r1c @K|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j td|t||||t d {S7w) ryr\rLr?c32K|]}t|ywrArBrDs r/rGz&AsyncBatches.delete..rHrIrJr^r_rOraNr{res r/r}zAsyncBatches.delete}sD ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a \\ M`p q(+Q[el, "    rc @K|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j td|t||||t d {S7w) rr\rLr?c32K|]}t|ywrArBrDs r/rGz&AsyncBatches.cancel..rHrIrJrr_rOraNrres r/rzAsyncBatches.cancelsP ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a ZZ J]m )+Q[el%     rc K|std||j|d{}|jstd|jddi|xsi}it dt |r#djtd |Dd gnti|xsi}dd i|xsi}|jt|j|t|||| ttd d{S77w)rr\r_NrrrrLr?c32K|]}t|ywrArBrDs r/rGz'AsyncBatches.results..#rHrIrJrOTr)rcrfrrrrrrTrrrdrrr r&rs r/rzAsyncBatches.resultss9F ^_o^rst tmm5EmFF   TUZUlUlTmn "#7QMN O(+Q[el&&HI    %G$ s"&C:C6CC:1C82C:8C:)rr)rrrr)rhrrirrjrr7rr9rr:rr;rr<rrz=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]r)r`rCr7rr9rr:rr;rr<rrz5AsyncJSONLDecoder[BetaMessageBatchIndividualResponse]rrr1r/r(r(s117726&*)-$("&;D= 8= / = $ = &= "=  = 9=  = F26)-$("&;D5 5 / 5 &5 "5  5 95  5 t $ $ 15)-$("&;DH H  H  H / H &H "H  H 9H  GH \26)-$("&;D5 5 / 5 &5 "5  5 95  !5 v26)-$("&;D= = / = &= "=  = 9=  = F26)-$("&;D? ? / ? &? "?  ? 9?  ?? r1ceZdZddZy)r,c||_tj|j|_tj|j|_tj|j |_tj|j |_tj|j|_tj|j|_yrA) _batchesrto_raw_response_wrapperrXrfrwr}rrr.batchess r/__init__zBatchesWithRawResponse.__init__6s &>> NN  )@@     %<< LL  '>> NN  '>> NN  (?? OO  r1Nrr'rNonerrrrrr1r/r,r,5 r1r,ceZdZddZy)rc||_tj|j|_tj|j|_tj|j |_tj|j |_tj|j|_tj|j|_yrA) rrasync_to_raw_response_wrapperrXrfrwr}rrrs r/rz$AsyncBatchesWithRawResponse.__init__Ns &DD NN  )FF     %BB LL  'DD NN  'DD NN  (EE OO  r1Nrr(rrrrr1r/rrMrr1rceZdZddZy)r4cJ||_t|j|_t|j|_t|j|_t|j |_t|j |_t|j|_yrA)rrrXrfrwr}rrrs r/rz%BatchesWithStreamingResponse.__init__fs 2 NN  5     1 LL  3 NN  3 NN  4 OO  r1Nrrrr1r/r4r4err1r4ceZdZddZy)rcJ||_t|j|_t|j|_t|j|_t|j |_t|j |_t|j|_yrA)rrrXrfrwr}rrrs r/rz*AsyncBatchesWithStreamingResponse.__init__~s 8 NN  ;     7 LL  9 NN  9 NN  : OO  r1Nrrrr1r/rr}rr1r)= __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,rr4rrr1r/rs#! !LLLgg';Z.*A?J=GVm n %I oI X I #I X   0  0  0  r1