
    pi                       d dl mZ d dlmZmZ d dlmZ d dlZddlm	Z	 ddl
mZmZmZmZmZmZmZ ddlmZmZmZmZ dd	lmZ dd
lmZmZ ddlmZmZ ddlm Z m!Z! ddl"m#Z# ddl$m%Z%m&Z& ddl'm(Z(m)Z) ddl*m+Z+m,Z, ddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 ddgZ5 G d de      Z6 G d de      Z7 G d d      Z8 G d d      Z9 G d d      Z: G d d       Z;y)!    )annotations)ListIterable)chainN   )_legacy_response)BodyOmitQueryHeadersNotGivenomit	not_given)is_givenmaybe_transformstrip_not_givenasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncPage	AsyncPage)AnthropicError)AsyncPaginatormake_request_options)JSONLDecoderAsyncJSONLDecoder)batch_list_paramsbatch_create_params)AnthropicBetaParam)BetaMessageBatch)BetaDeletedMessageBatch)"BetaMessageBatchIndividualResponseBatchesAsyncBatchesc            	      ^   e Zd Zedd       Zedd       Zeddded	 	 	 	 	 	 	 	 	 	 	 	 	 ddZeddded	 	 	 	 	 	 	 	 	 	 	 	 	 ddZ	eeeeddded	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ddZ
eddded	 	 	 	 	 	 	 	 	 	 	 	 	 dd	Zeddded	 	 	 	 	 	 	 	 	 	 	 	 	 dd
Zeddded	 	 	 	 	 	 	 	 	 	 	 	 	 ddZy)r&   c                    t        |       S a  
        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
        )BatchesWithRawResponseselfs    k/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/anthropic/resources/beta/messages/batches.pywith_raw_responsezBatches.with_raw_response   s     &d++    c                    t        |       S 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
        )BatchesWithStreamingResponser,   s    r.   with_streaming_responsezBatches.with_streaming_response(   s     ,D11r0   Nbetasextra_headersextra_query
extra_bodytimeoutc          
     $   i t        dt        |      r#dj                  t        d |D        dg            nt        i      |xs i }ddi|xs i }| j                  dt        d|it        j                        t        ||||      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.

        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,c              3  2   K   | ]  }t        |        y wNstr.0es     r.   	<genexpr>z!Batches.create.<locals>.<genexpr>X        5Lc!f5L   message-batches-2024-09-24/v1/messages/batches?beta=truerequestsr7   r8   r9   r:   bodyoptionscast_to)r   r   joinr   r   _postr   r!   BatchCreateParamsr   r#   r-   rK   r6   r7   r8   r9   r:   s          r.   createzBatches.create1   s    H	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_azz, *h!79L9^9^_(+Q[el %  
 	
r0   c          
        |st        d|      i t        dt        |      r#dj                  t	        d |D        dg            nt
        i      |xs i }ddi|xs i }| j                  d| d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>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z#Batches.retrieve.<locals>.<genexpr>   rG   rH   rI   /v1/messages/batches/
?beta=truerL   rO   rP   	
ValueErrorr   r   rQ   r   r   _getr   r#   r-   message_batch_idr6   r7   r8   r9   r:   s          r.   retrievezBatches.retrievei   s    D  ^_o^rstt	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_ayy#$4#5Z@(+Q[el %  
 	
r0   after_id	before_idlimitr6   r7   r8   r9   r:   c               @   i t        dt        |      r#dj                  t        d |D        dg            nt        i      |xs i }ddi|xs i }| j                  dt        t           t        ||||t        |||dt        j                              t              S )	4  List 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>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   zBatches.list.<locals>.<genexpr>   rG   rH   rI   rJ   rd   re   rf   r7   r8   r9   r:   querypagerO   model)r   r   rQ   r   r   _get_api_listr   r#   r   r   r    BatchListParams	r-   rd   re   rf   r6   r7   r8   r9   r:   s	            r.   listzBatches.list   s    V	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_a!!,*+(+'%%$,%.!&
 &55 ## " 
 	
r0   c          
        |st        d|      i t        dt        |      r#dj                  t	        d |D        dg            nt
        i      |xs i }ddi|xs i }| j                  d| d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
        rX   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z!Batches.delete.<locals>.<genexpr>  rG   rH   rI   rZ   r[   rL   r\   	r^   r   r   rQ   r   r   _deleter   r$   r`   s          r.   deletezBatches.delete   s    D  ^_o^rstt	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_a||#$4#5Z@(+Q[el ,  
 	
r0   c          
        |st        d|      i t        dt        |      r#dj                  t	        d |D        dg            nt
        i      |xs i }ddi|xs i }| j                  d| dt        ||||      t        	      S )
r  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.

        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
        rX   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z!Batches.cancel.<locals>.<genexpr>N  rG   rH   rI   rZ   /cancel?beta=truerL   r\   	r^   r   r   rQ   r   r   rR   r   r#   r`   s          r.   cancelzBatches.cancel!  s    P  ^_o^rstt	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_azz#$4#55FG(+Q[el %  
 	
r0   c          
        |st        d|      | j                  |      }|j                  st        d|j                         ddi|xs i }i t        dt        |      r#dj                  t        d |D        d	g            nt        i      |xs i }dd	i|xs i }| j                  |j                  t        ||||
      t        t           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
        rX   ra   BNo `results_url` for the given batch; Has it finished processing? Acceptapplication/binaryr=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z"Batches.results.<locals>.<genexpr>  rG   rH   rI   rL   TrO   rP   stream)r^   rb   results_urlr   processing_statusr   r   rQ   r   r   r_   r   r   r%   r-   ra   r6   r7   r8   r9   r:   batchs           r.   resultszBatches.results^  s   F  ^_o^rstt/?@   TUZUlUlTmn  "#7QM<ORQ	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_ayy(+Q[el !!CD  
 	
r0   )returnr+   )r   r3   rK   z%Iterable[batch_create_params.Request]r6   List[AnthropicBetaParam] | Omitr7   Headers | Noner8   Query | Noner9   Body | Noner:   'float | httpx.Timeout | None | NotGivenr   r#   ra   rB   r6   r   r7   r   r8   r   r9   r   r:   r   r   r#   )rd   
str | Omitre   r   rf   
int | Omitr6   r   r7   r   r8   r   r9   r   r:   r   r   zSyncPage[BetaMessageBatch]ra   rB   r6   r   r7   r   r8   r   r9   r   r:   r   r   r$   )ra   rB   r6   r   r7   r   r8   r   r9   r   r:   r   r   z0JSONLDecoder[BetaMessageBatchIndividualResponse]__name__
__module____qualname__r   r/   r4   r   r   rU   rb   rs   ry   r   r    r0   r.   r&   r&      s   , , 2 2 26 )-$("&;D6
 86
 /	6
 &6
 "6
  6
 96
 
6
x 26 )-$("&;D5
5
 /	5
 &5
 "5
  5
 95
 
5
t  $ $ 15 )-$("&;DH
 H
 	H

 H
 /H
 &H
 "H
  H
 9H
 
$H
\ 26 )-$("&;D5
5
 /	5
 &5
 "5
  5
 95
 
!5
v 26 )-$("&;D;
;
 /	;
 &;
 ";
  ;
 9;
 
;
B 26 )-$("&;D?
?
 /	?
 &?
 "?
  ?
 9?
 
:?
r0   c            	      ^   e Zd Zedd       Zedd       Zeddded	 	 	 	 	 	 	 	 	 	 	 	 	 ddZeddded	 	 	 	 	 	 	 	 	 	 	 	 	 ddZ	eeeeddded	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ddZ
eddded	 	 	 	 	 	 	 	 	 	 	 	 	 dd	Zeddded	 	 	 	 	 	 	 	 	 	 	 	 	 dd
Zeddded	 	 	 	 	 	 	 	 	 	 	 	 	 ddZy)r'   c                    t        |       S r*   )AsyncBatchesWithRawResponser,   s    r.   r/   zAsyncBatches.with_raw_response  s     +400r0   c                    t        |       S r2   )!AsyncBatchesWithStreamingResponser,   s    r.   r4   z$AsyncBatches.with_streaming_response  s     166r0   Nr5   c          
     T  K   i t        dt        |      r#dj                  t        d |D        dg            nt        i      |xs i }ddi|xs i }| j                  dt        d|it        j                         d{   t        ||||      t        	       d{   S 7 $7 w)
r<   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z&AsyncBatches.create.<locals>.<genexpr>  rG   rH   rI   rJ   rK   NrL   rM   )r   r   rQ   r   r   rR   r   r!   rS   r   r#   rT   s          r.   rU   zAsyncBatches.create  s     H	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_aZZ,,j(-CEXEjEjkk(+Q[el %   
 
 	
k
s$   A=B(?B$
 B(B& B(&B(c          
     2  K   |st        d|      i t        dt        |      r#dj                  t	        d |D        dg            nt
        i      |xs i }ddi|xs i }| j                  d| dt        ||||      t        	       d
{   S 7 w)rW   rX   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z(AsyncBatches.retrieve.<locals>.<genexpr>  rG   rH   rI   rZ   r[   rL   r\   Nr]   r`   s          r.   rb   zAsyncBatches.retrieve  s     D  ^_o^rstt	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_aYY#$4#5Z@(+Q[el %  
 
 	
 
   BBBBrc   c               @   i t        dt        |      r#dj                  t        d |D        dg            nt        i      |xs i }ddi|xs i }| j                  dt        t           t        ||||t        |||dt        j                              t              S )	rh   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z$AsyncBatches.list.<locals>.<genexpr>Q  rG   rH   rI   rJ   rj   rk   rm   )r   r   rQ   r   r   rp   r   r#   r   r   r    rq   rr   s	            r.   rs   zAsyncBatches.list#  s    V	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_a!!,+,(+'%%$,%.!&
 &55 ## " 
 	
r0   c          
     2  K   |st        d|      i t        dt        |      r#dj                  t	        d |D        dg            nt
        i      |xs i }ddi|xs i }| j                  d| dt        ||||      t        	       d
{   S 7 w)ru   rX   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z&AsyncBatches.delete.<locals>.<genexpr>  rG   rH   rI   rZ   r[   rL   r\   Nrw   r`   s          r.   ry   zAsyncBatches.deletem  s     D  ^_o^rstt	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_a\\#$4#5Z@(+Q[el , " 
 
 	
 
r   c          
     2  K   |st        d|      i t        dt        |      r#dj                  t	        d |D        dg            nt
        i      |xs i }ddi|xs i }| j                  d| dt        ||||      t        	       d
{   S 7 w)r{   rX   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z&AsyncBatches.cancel.<locals>.<genexpr>  rG   rH   rI   rZ   r}   rL   r\   Nr~   r`   s          r.   r   zAsyncBatches.cancel  s     P  ^_o^rstt	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_aZZ#$4#55FG(+Q[el %   
 
 	
 
r   c          
       K   |st        d|      | j                  |       d{   }|j                  st        d|j                         ddi|xs i }i t        dt        |      r#dj                  t        d	 |D        d
g            nt        i      |xs i }dd
i|xs i }| j                  |j                  t        ||||      t        t           d       d{   S 7 7 w)r   rX   r   Nr   r   r   r=   r>   c              3  2   K   | ]  }t        |        y wr@   rA   rC   s     r.   rF   z'AsyncBatches.results.<locals>.<genexpr>  rG   rH   rI   rL   Tr   )r^   rb   r   r   r   r   r   rQ   r   r   r_   r   r   r%   r   s           r.   r   zAsyncBatches.results  s1    F  ^_o^rsttmm5EmFF   TUZUlUlTmn  "#7QM<ORQ	
$ '*hhu5Le5LOkNl/m&n"	
 "	
 *+GaML_]_aYY(+Q[el &&HI  
 
 	
% G$
s"   &C/C+B=C/&C-'C/-C/)r   r   )r   r   r   r   )rd   r   re   r   rf   r   r6   r   r7   r   r8   r   r9   r   r:   r   r   z=AsyncPaginator[BetaMessageBatch, AsyncPage[BetaMessageBatch]]r   )ra   rB   r6   r   r7   r   r8   r   r9   r   r:   r   r   z5AsyncJSONLDecoder[BetaMessageBatchIndividualResponse]r   r   r0   r.   r'   r'     s   1 1 7 7 26 )-$("&;D6
 86
 /	6
 &6
 "6
  6
 96
 
6
x 26 )-$("&;D5
5
 /	5
 &5
 "5
  5
 95
 
5
t  $ $ 15 )-$("&;DH
 H
 	H

 H
 /H
 &H
 "H
  H
 9H
 
GH
\ 26 )-$("&;D5
5
 /	5
 &5
 "5
  5
 95
 
!5
v 26 )-$("&;D;
;
 /	;
 &;
 ";
  ;
 9;
 
;
B 26 )-$("&;D?
?
 /	?
 &?
 "?
  ?
 9?
 
??
r0   c                      e Zd ZddZy)r+   c                z   || _         t        j                  |j                        | _        t        j                  |j                        | _        t        j                  |j
                        | _        t        j                  |j                        | _        t        j                  |j                        | _        y r@   )_batchesr   to_raw_response_wrapperrU   rb   rs   ry   r   r-   batchess     r.   __init__zBatchesWithRawResponse.__init__$  s    &>>NN
 )@@
 %<<LL
	 '>>NN
 '>>NN
r0   Nr   r&   r   Noner   r   r   r   r   r0   r.   r+   r+   #      
r0   r+   c                      e Zd ZddZy)r   c                z   || _         t        j                  |j                        | _        t        j                  |j                        | _        t        j                  |j
                        | _        t        j                  |j                        | _        t        j                  |j                        | _        y r@   )r   r   async_to_raw_response_wrapperrU   rb   rs   ry   r   r   s     r.   r   z$AsyncBatchesWithRawResponse.__init__9  s    &DDNN
 )FF
 %BBLL
	 'DDNN
 'DDNN
r0   Nr   r'   r   r   r   r   r0   r.   r   r   8  r   r0   r   c                      e Zd ZddZy)r3   c                   || _         t        |j                        | _        t        |j                        | _        t        |j                        | _        t        |j
                        | _        t        |j                        | _        y r@   )r   r   rU   rb   rs   ry   r   r   s     r.   r   z%BatchesWithStreamingResponse.__init__N  sr    2NN
 5
 1LL
	 3NN
 3NN
r0   Nr   r   r   r0   r.   r3   r3   M  r   r0   r3   c                      e Zd ZddZy)r   c                   || _         t        |j                        | _        t        |j                        | _        t        |j                        | _        t        |j
                        | _        t        |j                        | _        y r@   )r   r   rU   rb   rs   ry   r   r   s     r.   r   z*AsyncBatchesWithStreamingResponse.__init__c  sr    8NN
 ;
 7LL
	 9NN
 9NN
r0   Nr   r   r   r0   r.   r   r   b  r   r0   r   )<
__future__r   typingr   r   	itertoolsr   httpx r   _typesr	   r
   r   r   r   r   r   _utilsr   r   r   r   _compatr   	_resourcer   r   	_responser   r   
paginationr   r   _exceptionsr   _base_clientr   r   _decoders.jsonlr   r   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+   r   r3   r   r   r0   r.   <module>r      s    # !   ! L L L X X ' ; Z . * A ? J = G V mn
%@
o @
F@
# @
F
 
*
 
*
 
*
 
r0   