7j}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..X5Lc!f5Lmessage-batches-2024-09-24/v1/messages/batches?beta=truerequestsr8r9r:r;bodyoptionscast_to) rrjoinrr_postrr"BatchCreateParamsrr$r.rLr7r8r9r:r;s r/createzBatches.create1sH  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a zz , *h!79L9^9^_(+Q[el%   r1c $|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://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 r>r?c32K|]}t|ywrArBrDs r/rGz#Batches.retrieve..rHrIrJ1/v1/messages/batches/{message_batch_id}?beta=truemessage_batch_idrMrPrQ ValueErrorrrrRrr_getrrr$r.r]r7r8r9r:r;s r/retrievezBatches.retrieveisD ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a yy M`p q(+Q[el%   r1after_id before_idlimitr7r8r9r:r;c@itdt|r#djtd|Ddgnti|xsi}ddi|xsi}|j dt tt||||t|||dtjtS) 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 r>r?c32K|]}t|ywrArBrDs r/rGzBatches.list..rHrIrJrKrerfrgr8r9r:r;querypagerPmodel) rrrRrr _get_api_listrr$rrr!BatchListParams r.rerfrgr7r8r9r: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://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 rYr>r?c32K|]}t|ywrArBrDs r/rGz!Batches.delete..rHrIrJr[r\rMr^ r`rrrRrr_deleterrr%rbs 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) 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 rYr>r?c32K|]}t|ywrArBrDs r/rGz!Batches.cancel..NrHrIrJ8/v1/messages/batches/{message_batch_id}/cancel?beta=truer\rMr^ r`rrrRrrrSrrr$rbs 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://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 rYr\BNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryr>r?c32K|]}t|ywrArBrDs r/rGz"Batches.results..rHrIrJrMTrPrQstream)r`rc results_urlrprocessing_statusrrrRrrrarrrr&r.r]r7r8r9r:r;batchs r/resultszBatches.results`s!F ^_o^rst t /? @   TUZUlUlTmn "#7QMN O(+Q[el!!CD   r1)returnr,)rr4rLz%Iterable[batch_create_params.Request]r7List[AnthropicBetaParam] | Omitr8Headers | Noner9 Query | Noner: Body | Noner;'float | httpx.Timeout | None | NotGivenrr$r]rCr7rr8rr9rr:rr;rrr$)re str | Omitrfrrg int | Omitr7rr8rr9rr:rr;rrzSyncPage[BetaMessageBatch]r]rCr7rr8rr9rr:rr;rrr%)r]rCr7rr8rr9rr:rr;rrz0JSONLDecoder[BetaMessageBatchIndividualResponse]__name__ __module__ __qualname__rr0r5rrrVrcrtrzrrr1r/r'r's,,2226)-$("&;D6 86 / 6 &6 "6  6 96  6 x26)-$("&;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 ^eZdZed dZed dZeddded ddZeddded ddZ eeeeddded ddZ 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 TKitdt|r#djtd|Ddgnti|xsi}ddi|xsi}|j dt d|itjd{t||||t d{S7$7w) r=r>r?c32K|]}t|ywrArBrDs r/rGz&AsyncBatches.create..rHrIrJrKrLNrMrN) rrrRrrrSrr"rTrr$rUs r/rVzAsyncBatches.createsH  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a ZZ ,,j(-CEXEjEjkk(+Q[el%     k s$A=B(?B$ B(B& B(&B(c @K|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j td|t||||t d {S7w) rXrYr>r?c32K|]}t|ywrArBrDs r/rGz(AsyncBatches.retrieve..rHrIrJr[r\rMr^Nr_rbs r/rczAsyncBatches.retrievesD ^_o^rst t  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a YY M`p q(+Q[el%     BBBBrdc@itdt|r#djtd|Ddgnti|xsi}ddi|xsi}|j dt tt||||t|||dtjtS) rir>r?c32K|]}t|ywrArBrDs r/rGz$AsyncBatches.list..SrHrIrJrKrkrlrn) rrrRrrrqrr$rrr!rrrss r/rtzAsyncBatches.list%sV  $'*hhu5Le5LOkNl/m&n"  "  *+GaML_]_a !! ,+,(+'%%$,%.!& &55  ##"  r1c @K|std|itdt|r#djt d|Ddgnt i|xsi}ddi|xsi}|j td|t||||t d {S7w) rvrYr>r?c32K|]}t|ywrArBrDs r/rGz&AsyncBatches.delete..rHrIrJr[r\rMr^Nrxrbs r/rzzAsyncBatches.deleteosD ^_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) r|rYr>r?c32K|]}t|ywrArBrDs r/rGz&AsyncBatches.cancel..rHrIrJr~r\rMr^Nrrbs 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)rrYr\Nrrrr>r?c32K|]}t|ywrArBrDs r/rGz'AsyncBatches.results..rHrIrJrMTr)r`rcrrrrrrRrrrarrr r&rs r/rzAsyncBatches.resultss9F ^_o^rst tmm5EmFF   TUZUlUlTmn "#7QMN O(+Q[el&&HI    %G$ s"&C:C6CC:1C82C:8C:)rr)rrrr)rerrfrrgrr7rr8rr9rr:rr;rrz=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]r)r]rCr7rr8rr9rr:rr;rrz5AsyncJSONLDecoder[BetaMessageBatchIndividualResponse]rrr1r/r(r(s117726)-$("&;D6 86 / 6 &6 "6  6 96  6 x26)-$("&;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_wrapperrVrcrtrzrrr.batchess r/__init__zBatchesWithRawResponse.__init__(s &>> NN  )@@     %<< LL  '>> NN  '>> NN  (?? OO  r1Nrr'rNonerrrrrr1r/r,r,' r1r,ceZdZddZy)rc||_tj|j|_tj|j|_tj|j |_tj|j |_tj|j|_tj|j|_yrA) rrasync_to_raw_response_wrapperrVrcrtrzrrrs r/rz$AsyncBatchesWithRawResponse.__init__@s &DD NN  )FF     %BB LL  'DD NN  'DD NN  (EE OO  r1Nrr(rrrrr1r/rr?rr1rceZdZddZy)r4cJ||_t|j|_t|j|_t|j|_t|j |_t|j |_t|j|_yrA)rrrVrcrtrzrrrs r/rz%BatchesWithStreamingResponse.__init__Xs 2 NN  5     1 LL  3 NN  3 NN  4 OO  r1Nrrrr1r/r4r4Wrr1r4ceZdZddZy)rcJ||_t|j|_t|j|_t|j|_t|j |_t|j |_t|j|_yrA)rrrVrcrtrzrrrs r/rz*AsyncBatchesWithStreamingResponse.__init__ps 8 NN  ;     7 LL  9 NN  9 NN  : OO  r1Nrrrr1r/rrorr1r)= __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 %B oB J B #B J   0  0  0  r1