g_ddlmZddlmZddlZddlmZddlmZm Z m Z m Z m Z ddl mZmZddlmZdd lmZmZdd lmZmZmZmZmZmZmZmZmZmZdd l m!Z!m"Z"dd l#m$Z$dd l%m&Z&m'Z'ddl(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0ddgZ1GddeZ2GddeZ3GddZ4GddZ5GddZ6GddZ7y)) annotations)IterableN)_legacy_response) NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource) BinaryAPIResponseAsyncBinaryAPIResponseStreamedBinaryAPIResponseAsyncStreamedBinaryAPIResponseto_streamed_response_wrapperto_custom_raw_response_wrapper"async_to_streamed_response_wrapper#to_custom_streamed_response_wrapper$async_to_custom_raw_response_wrapper)async_to_custom_streamed_response_wrapper)SyncPage AsyncPage)AnthropicError)AsyncPaginatormake_request_options)MessageBatchIndividualResponsebatch_list_paramsbatch_create_params) JSONLDecoderAsyncJSONLDecoder) MessageBatchBatches AsyncBatchesceZdZed dZed dZddded d dZddded ddZeeeddded ddZ ddded dd Z ddded dd Z y)r&ct|Sa This property can be used as a prefix for any HTTP method call to return the 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 )BatchesWithRawResponseselfs Y/var/www/openai/venv/lib/python3.12/site-packages/anthropic/resources/messages/batches.pywith_raw_responsezBatches.with_raw_response(s&d++ct|Sz 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 )BatchesWithStreamingResponser,s r.with_streaming_responsezBatches.with_streaming_response2s,D11r0N extra_headers extra_query extra_bodytimeoutc |jdtd|itjt ||||t 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. Args: requests: List of requests for prompt completion. Each is an individual request to create a Message. 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 /v1/messages/batchesrequestsr5bodyoptionscast_to)_postr r"BatchCreateParamsrr%r-r=r6r7r8r9s r.createzBatches.create;sK<zz " *h!79L9^9^_(+Q[el!   r0c r|std||jd|t||||tS)0This 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. Args: message_batch_id: ID of the Message Batch. 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 /v1/messages/batches/r5r@rA ValueError_getrr%r-message_batch_idr6r7r8r9s r.retrievezBatches.retrievebsV8 ^_o^rst tyy#$4#5 6(+Q[el!   r0after_id before_idlimitr6r7r8r9c|jdttt||||t |||dt j tSaPList all Message Batches within a Workspace. Most recently created batches are returned first. 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`. 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<)rRrSrT)r6r7r8r9query)pager@model) _get_api_listrr%rr r!BatchListParamsr-rRrSrTr6r7r8r9s r.listz Batches.listsbJ!! ",'(+'%%$,%.!& &55  #"  r0c t|std||jd|dt||||tS)Batches 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. Args: message_batch_id: ID of the Message Batch. 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 rHrI/cancelr5rJrLrBrr%rNs r.cancelzBatches.cancelsYD ^_o^rst tzz#$4#5W =(+Q[el!   r0c |std||j|}|jstd|jddi|xsi}|j |jt ||||dttS)  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. Args: message_batch_id: ID of the Message Batch. 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 rHrOBNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryr5Tr@streamrA) rLrP results_urlrprocessing_statusrMrr#r r-rOr6r7r8r9batchs r.resultszBatches.resultss: ^_o^rst t /? @   TUZUlUlTmn "#7QM)rBr r"rCrr%rDs r.rEzAsyncBatches.create2s`<ZZ ",j(-CEXEjEjkk(+Q[el!     k s!,AA AAAAc K|std||jd|t||||td{S7w)rGrHrIr5rJNrKrNs r.rPzAsyncBatches.retrieveYsd8 ^_o^rst tYY#$4#5 6(+Q[el!     s> NN  )@@     %<< LL  '>> NN  6 OO   r0Nrr&rpNoner{r|r}rr~r0r.r+r+ r0r+ceZdZddZy)rcp||_tj|j|_tj|j|_tj|j |_tj|j |_t|jt|_yr) rrasync_to_raw_response_wrapperrErPr]rbrrorrs r.rz$AsyncBatchesWithRawResponse.__init__,s &DD NN  )FF     %BB LL  'DD NN  < OO "  r0Nrr'rprrr~r0r.rr+rr0rceZdZddZy)r3c ||_t|j|_t|j|_t|j|_t|j |_t |jt|_yr) rrrErPr]rbrrorrs r.rz%BatchesWithStreamingResponse.__init__Bsu 2 NN  5     1 LL  3 NN  ; OO %  r0Nrrr~r0r.r3r3Arr0r3ceZdZddZy)rc ||_t|j|_t|j|_t|j|_t|j |_t |jt|_yr) rrrErPr]rbrrorrs r.rz*AsyncBatchesWithStreamingResponse.__init__Xsv 8 NN  ;     7 LL  9 NN  A OO *  r0Nrrr~r0r.rrWrr0r)8 __future__rtypingrhttpxr_typesrrr r r _utilsr r _compatr _resourcerr _responserrrrrrrrrr paginationrr _exceptionsr _base_clientrrtypes.messagesr r!r"_decoders.jsonlr#r$types.messages.message_batchr%__all__r&r'r+rr3rr~r0r.rs# ??':   .)@dd>8 n %t ot nt #t n  ,  ,  ,  r0