o +i@sddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZmZmZdd lmZdd lmZmZdd lmZmZdd lm 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.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddgZ5GdddeZ6GdddeZ7GdddZ8GdddZ9GdddZ:Gdd d Z;dS)!) annotations)ListIterable)chainN)_legacy_response)BodyOmitQueryHeadersNotGivenomit not_given)is_givenmaybe_transformstrip_not_givenasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncPage AsyncPage)AnthropicError)AsyncPaginatormake_request_options) JSONLDecoderAsyncJSONLDecoder)batch_list_paramsbatch_create_params)AnthropicBetaParam)BetaMessageBatch)BetaDeletedMessageBatch)"BetaMessageBatchIndividualResponseBatches AsyncBatchesc @eZdZed.ddZed/ddZeddded d0ddZeddded d1ddZ eeeedddedd2d$d%Z eddded d3d'd(Z eddded d1d)d*Z eddded d4d,d-Z dS)5r%returnBatchesWithRawResponsecCt|Sa This property can be used as a prefix for any HTTP method call to return the raw response object instead of the parsed content. For more information, see https://www.github.com/anthropics/anthropic-sdk-python#accessing-raw-response-data-eg-headers )r)selfr.f/var/www/html/psymed-ai/venv/lib/python3.10/site-packages/anthropic/resources/beta/messages/batches.pywith_raw_responsezBatches.with_raw_responseBatchesWithStreamingResponsecCr*z An alternative to `.with_raw_response` that doesn't eagerly read the response body. For more information, see https://www.github.com/anthropics/anthropic-sdk-python#with_streaming_response )r2r,r.r.r/with_streaming_response(zBatches.with_streaming_responseNbetas extra_headers extra_query extra_bodytimeoutrequests%Iterable[batch_create_params.Request]r7List[AnthropicBetaParam] | Omitr8Headers | Noner9 Query | Noner: Body | Noner;'float | httpx.Timeout | None | NotGivenr"c Csvitdt|rdtdd|Ddgnti|pi}ddi|p$i}|jdtd|itjt ||||dt d S)  Send a batch of Message creation requests. The Message Batches API can be used to process multiple Messages API requests at once. Once a Message Batch is created, it begins processing immediately. Batches can take up to 24 hours to complete. Learn more about the Message Batches API in our [user guide](https://docs.claude.com/en/docs/build-with-claude/batch-processing) Args: requests: List of requests for prompt completion. Each is an individual request to create a Message. 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 anthropic-beta,cs|]}t|VqdSNstr.0er.r.r/ Xz!Batches.create..message-batches-2024-09-24/v1/messages/batches?beta=truer<r8r9r:r;bodyoptionscast_to) rrjoinrr_postrr BatchCreateParamsrr"r-r<r7r8r9r:r;r.r.r/create1s($ zBatches.createmessage_batch_idrIcC|s td|itdt|rdtdd|Ddgnti|p%i}ddi|p-i}|jd|dt||||d td 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 rDrEcsrFrGrHrJr.r.r/rMrNz#Batches.retrieve..rO/v1/messages/batches/ ?beta=truerQrTrU ValueErrorrrrVrr_getrr"r-r[r7r8r9r:r;r.r.r/retrievei*"  zBatches.retrieveafter_id before_idlimitr7r8r9r:r;ri str | Omitrjrk int | OmitSyncPage[BetaMessageBatch]c Citdt|rdtdd|Ddgnti|pi}ddi|p$i}|jdttt||||t |||dt j dtd S) 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 rDrEcsrFrGrHrJr.r.r/rMrNzBatches.list..rOrPrirjrkr8r9r:r;querypagerTmodel) rrrVrr _get_api_listrr"rrrBatchListParams r-rirjrkr7r8r9r:r;r.r.r/list<+ z Batches.listr#cCr\)  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^rDrEcsrFrGrHrJr.r.r/rMrNz!Batches.delete..rOr_r`rQra rcrrrVrr_deleterr#rer.r.r/deletergzBatches.deletecCr\) 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^rDrEcsrFrGrHrJr.r.r/rMNrNz!Batches.cancel..rOr_/cancel?beta=truerQra rcrrrVrrrWrr"rer.r.r/cancel!s*(  zBatches.cancel0JSONLDecoder[BetaMessageBatchIndividualResponse]cCs|s td||j|d}|jstd|jddi|p i}itdt|r7dtdd |Dd gnt i|p>i}dd i|pFi}|j |jt ||||d t t d 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://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^r[BNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryrDrEcsrFrGrHrJr.r.r/rMrNz"Batches.results..rOrQTrTrUstream)rcrf results_urlrprocessing_statusrrrVrrrdrrr$r-r[r7r8r9r:r;batchr.r.r/results^s8#   zBatches.results)r(r))r(r2r<r=r7r>r8r?r9r@r:rAr;rBr(r"r[rIr7r>r8r?r9r@r:rAr;rBr(r")rirlrjrlrkrmr7r>r8r?r9r@r:rAr;rBr(rnr[rIr7r>r8r?r9r@r:rAr;rBr(r#)r[rIr7r>r8r?r9r@r:rAr;rBr(r__name__ __module__ __qualname__rr0r4r rrZrfrzrrrr.r.r.r/r%X   <:N;Ac @r')5r&r(AsyncBatchesWithRawResponsecCr*r+)rr,r.r.r/r0r1zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr*r3)rr,r.r.r/r4r5z$AsyncBatches.with_streaming_responseNr6r<r=r7r>r8r?r9r@r:rAr;rBr"c sitdt|rdtdd|Ddgnti|pi}ddi|p%i}|jdtd|itjIdHt ||||d t d IdHS) rCrDrEcsrFrGrHrJr.r.r/rMrNz&AsyncBatches.create..rOrPr<NrQrR) rrrVrrrWrr rXrr"rYr.r.r/rZs*$  zAsyncBatches.creater[rIc|s td|itdt|rdtdd|Ddgnti|p&i}ddi|p.i}|jd|dt||||d td Id HS) r]r^rDrEcsrFrGrHrJr.r.r/rMrNz(AsyncBatches.retrieve..rOr_r`rQraNrbrer.r.r/rf,"   zAsyncBatches.retrieverhrirlrjrkrm=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]c Cro) rprDrEcsrFrGrHrJr.r.r/rMQrNz$AsyncBatches.list..rOrPrqrrrt) rrrVrrrwrr"rrrrxryr.r.r/rz#r{zAsyncBatches.listr#cr) r|r^rDrEcsrFrGrHrJr.r.r/rMrNz&AsyncBatches.delete..rOr_r`rQraNr}rer.r.r/rmrzAsyncBatches.deletecr) rr^rDrEcsrFrGrHrJr.r.r/rMrNz&AsyncBatches.cancel..rOr_rrQraNrrer.r.r/rs,(   zAsyncBatches.cancel5AsyncJSONLDecoder[BetaMessageBatchIndividualResponse]cs|s td||j|dIdH}|jstd|jddi|p$i}itdt|r;dtd d |Dd gnt i|pBi}dd i|pJi}|j |jt ||||d t t d dIdHS)rr^rNrrrrDrEcsrFrGrHrJr.r.r/rMrNz'AsyncBatches.results..rOrQTr)rcrfrrrrrrVrrrdrrr$rr.r.r/rs:#   zAsyncBatches.results)r(r)r(rrr)rirlrjrlrkrmr7r>r8r?r9r@r:rAr;rBr(rr)r[rIr7r>r8r?r9r@r:rAr;rBr(rrr.r.r.r/r&rc@eZdZdddZdS) r)batchesr%r(NonecCP||_t|j|_t|j|_t|j|_t|j|_t|j|_dSrG)_batchesrto_raw_response_wrapperrZrfrzrrr-rr.r.r/__init__$  zBatchesWithRawResponse.__init__Nrr%r(rrrrrr.r.r.r/r)#r)c@r) rrr&r(rcCrrG)rrasync_to_raw_response_wrapperrZrfrzrrrr.r.r/r9rz$AsyncBatchesWithRawResponse.__init__Nrr&r(rrr.r.r.r/r8rrc@r) r2rr%r(rcCF||_t|j|_t|j|_t|j|_t|j|_t|j|_dSrG)rrrZrfrzrrrr.r.r/rN  z%BatchesWithStreamingResponse.__init__Nrrr.r.r.r/r2Mrr2c@r) rrr&r(rcCrrG)rrrZrfrzrrrr.r.r/rcrz*AsyncBatchesWithStreamingResponse.__init__Nrrr.r.r.r/rbrr)< __future__rtypingrr itertoolsrhttpxr_typesrr r r r r r_utilsrrrr_compatr _resourcerr _responserr paginationrr _exceptionsr _base_clientrr_decoders.jsonlrrtypes.beta.messagesrr 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)rr2rr.r.r.r/s@   $