
    hQ                    `   d dl mZ d dlmZ d dlmZ d dlZddlmZ ddl	m
Z
mZ ddlmZmZmZ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"m#Z#m$Z$ ddl%m&Z& ddgZ' G d de      Z( G d de      Z) G d d      Z* G d d      Z+ G d d      Z, G d d      Z-y)    )annotations)Optional)LiteralN   )_legacy_response)batch_list_paramsbatch_create_params)	NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)Batch)AsyncPaginatormake_request_options)MetadataBatchesAsyncBatchesc                      e Zd Zedd       Zedd       Zeeddded	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ddZddded	 	 	 	 	 	 	 	 	 	 	 ddZeeddded	 	 	 	 	 	 	 	 	 	 	 	 	 dd	Z	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/openai/openai-python#accessing-raw-response-data-eg-headers
        )BatchesWithRawResponseselfs    V/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/openai/resources/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/openai/openai-python#with_streaming_response
        )BatchesWithStreamingResponser"   s    r$   with_streaming_responsezBatches.with_streaming_response$   s     ,D11r&   Nmetadataoutput_expires_afterextra_headersextra_query
extra_bodytimeoutc       	            | j                  dt        |||||dt        j                        t	        ||||	      t
              S )  
        Creates and executes a batch from an uploaded file of requests

        Args:
          completion_window: The time frame within which the batch should be processed. Currently only `24h`
              is supported.

          endpoint: The endpoint to be used for all requests in the batch. Currently
              `/v1/responses`, `/v1/chat/completions`, `/v1/embeddings`, and `/v1/completions`
              are supported. Note that `/v1/embeddings` batches are also restricted to a
              maximum of 50,000 embedding inputs across all requests in the batch.

          input_file_id: The ID of an uploaded file that contains requests for the new batch.

              See [upload file](https://platform.openai.com/docs/api-reference/files/create)
              for how to upload a file.

              Your input file must be formatted as a
              [JSONL file](https://platform.openai.com/docs/api-reference/batch/request-input),
              and must be uploaded with the purpose `batch`. The file can contain up to 50,000
              requests, and can be up to 200 MB in size.

          metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful
              for storing additional information about the object in a structured format, and
              querying for objects via API or the dashboard.

              Keys are strings with a maximum length of 64 characters. Values are strings with
              a maximum length of 512 characters.

          output_expires_after: The expiration policy for the output and/or error file that are generated for a
              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
        /batchescompletion_windowendpointinput_file_idr,   r-   r.   r/   r0   r1   bodyoptionscast_to)_postr   r	   BatchCreateParamsr   r   
r#   r6   r7   r8   r,   r-   r.   r/   r0   r1   s
             r$   createzBatches.create-   s_    n zz ): (%2 (,@ $55	 )+Q[el   
 	
r&   r9   c          	     r    |st        d|      | j                  d| t        ||||      t              S )F  
        Retrieves a batch.

        Args:
          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
        7Expected a non-empty value for `batch_id` but received 	/batches/r9   r<   r=   
ValueError_getr   r   r#   batch_idr.   r/   r0   r1   s         r$   retrievezBatches.retrievev   sT    . VW_Vbcddyyz"(+Q[el   
 	
r&   afterlimitr.   r/   r0   r1   c                   | j                  dt        t           t        ||||t	        ||dt
        j                              t              S a,  List your organization's batches.

        Args:
          after: A cursor for use in pagination.

        `after` is an object ID that defines your place
              in the list. For instance, if you make a list request and receive 100 objects,
              ending with obj_foo, your subsequent call can include after=obj_foo in order to
              fetch the next page of the list.

          limit: A limit on the number of objects to be returned. Limit can range between 1 and
              100, and the default is 20.

          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
        r4   )rN   rO   )r.   r/   r0   r1   query)pager<   model)_get_api_listr   r   r   r   r   BatchListParamsr#   rN   rO   r.   r/   r0   r1   s          r$   listzBatches.list   s_    B !!&(+'%%!&!& &55 ! " 
 	
r&   c          	     t    |st        d|      | j                  d| dt        ||||      t              S )
  Cancels an in-progress batch.

        The batch will be in status `cancelling` for up to
        10 minutes, before changing to `cancelled`, where it will have partial results
        (if any) available in the output file.

        Args:
          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
        rD   rE   /cancelr9   rF   rH   r>   r   r   rJ   s         r$   cancelzBatches.cancel   sV    4 VW_Vbcddzzz)(+Q[el   
 	
r&   )returnr!   )r^   r)   r6   zLiteral['24h']r7   zULiteral['/v1/responses', '/v1/chat/completions', '/v1/embeddings', '/v1/completions']r8   strr,   zOptional[Metadata] | NotGivenr-   z1batch_create_params.OutputExpiresAfter | NotGivenr.   Headers | Noner/   Query | Noner0   Body | Noner1   'float | httpx.Timeout | None | NotGivenr^   r   rK   r`   r.   ra   r/   rb   r0   rc   r1   rd   r^   r   )rN   str | NotGivenrO   int | NotGivenr.   ra   r/   rb   r0   rc   r1   rd   r^   zSyncCursorPage[Batch]__name__
__module____qualname__r   r%   r*   r
   rA   rL   rX   r]    r&   r$   r   r      s   , , 2 2 3<R[ )-$("&;DG
 *G
 h	G

 G
 0G
 PG
 &G
 "G
  G
 9G
 
G
^ )-$("&;D

 &
 "
  
 9
 

H !* ) )-$("&;D2
 2
 	2
 &2
 "2
  2
 92
 
2
t )-$("&;D"
"
 &"
 ""
  "
 9"
 
"
r&   c                      e Zd Zedd       Zedd       Zeeddded	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 ddZddded	 	 	 	 	 	 	 	 	 	 	 ddZeeddded	 	 	 	 	 	 	 	 	 	 	 	 	 dd	Z	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0r&   c                    t        |       S r(   )!AsyncBatchesWithStreamingResponser"   s    r$   r*   z$AsyncBatches.with_streaming_response   s     166r&   Nr+   c       	           K   | j                  dt        |||||dt        j                         d{   t	        ||||	      t
               d{   S 7 $7 w)r3   r4   r5   Nr9   r:   )r>   r   r	   r?   r   r   r@   s
             r$   rA   zAsyncBatches.create  sx     n ZZ,): (%2 (,@ $55	 	 )+Q[el    
 
 	
	
s!   0AA
AAAAr9   c          	        K   |st        d|      | j                  d| t        ||||      t               d{   S 7 w)rC   rD   rE   r9   rF   NrG   rJ   s         r$   rL   zAsyncBatches.retrieveM  sb     . VW_VbcddYYz"(+Q[el   
 
 	
 
s   <AAArM   c                   | j                  dt        t           t        ||||t	        ||dt
        j                              t              S rQ   )rU   r   r   r   r   r   rV   rW   s          r$   rX   zAsyncBatches.listn  s_    B !! '(+'%%!&!& &55 ! " 
 	
r&   c          	        K   |st        d|      | j                  d| dt        ||||      t               d{   S 7 w)rZ   rD   rE   r[   r9   rF   Nr\   rJ   s         r$   r]   zAsyncBatches.cancel  sd     4 VW_VbcddZZz)(+Q[el    
 
 	
 
s   =AA A)r^   ro   )r^   rq   r_   re   )rN   rf   rO   rg   r.   ra   r/   rb   r0   rc   r1   rd   r^   z-AsyncPaginator[Batch, AsyncCursorPage[Batch]]rh   rl   r&   r$   r   r      s   1 1 7 7 3<R[ )-$("&;DG
 *G
 h	G

 G
 0G
 PG
 &G
 "G
  G
 9G
 
G
^ )-$("&;D

 &
 "
  
 9
 

H !* ) )-$("&;D2
 2
 	2
 &2
 "2
  2
 92
 
72
t )-$("&;D"
"
 &"
 ""
  "
 9"
 
"
r&   c                      e Zd ZddZy)r!   c                2   || _         t        j                  |j                        | _        t        j                  |j                        | _        t        j                  |j
                        | _        t        j                  |j                        | _        y N)_batchesr   to_raw_response_wrapperrA   rL   rX   r]   r#   batchess     r$   __init__zBatchesWithRawResponse.__init__  su    &>>NN
 )@@
 %<<LL
	 '>>NN
r&   Nr|   r   r^   Noneri   rj   rk   r}   rl   r&   r$   r!   r!         
r&   r!   c                      e Zd ZddZy)ro   c                2   || _         t        j                  |j                        | _        t        j                  |j                        | _        t        j                  |j
                        | _        t        j                  |j                        | _        y rx   )ry   r   async_to_raw_response_wrapperrA   rL   rX   r]   r{   s     r$   r}   z$AsyncBatchesWithRawResponse.__init__  su    &DDNN
 )FF
 %BBLL
	 'DDNN
r&   Nr|   r   r^   r   r   rl   r&   r$   ro   ro     r   r&   ro   c                      e Zd ZddZy)r)   c                    || _         t        |j                        | _        t        |j                        | _        t        |j                        | _        t        |j
                        | _        y rx   )ry   r   rA   rL   rX   r]   r{   s     r$   r}   z%BatchesWithStreamingResponse.__init__  s]    2NN
 5
 1LL
	 3NN
r&   Nr~   r   rl   r&   r$   r)   r)     r   r&   r)   c                      e Zd ZddZy)rq   c                    || _         t        |j                        | _        t        |j                        | _        t        |j                        | _        t        |j
                        | _        y rx   )ry   r   rA   rL   rX   r]   r{   s     r$   r}   z*AsyncBatchesWithStreamingResponse.__init__  s]    8NN
 ;
 7LL
	 9NN
r&   Nr   r   rl   r&   r$   rq   rq     r   r&   rq   ).
__future__r   typingr   typing_extensionsr   httpx r   typesr   r	   _typesr
   r   r   r   r   _utilsr   r   _compatr   	_resourcer   r   	_responser   r   
paginationr   r   types.batchr   _base_clientr   r   types.shared_params.metadatar   __all__r   r   r!   ro   r)   rq   rl   r&   r$   <module>r      s    #  %   : > > ; % 9 X 8  ? 3n
%T
o T
nT
# T
n
 
$
 
$
 
$
 
r&   