jWlddlmZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZmZmZmZmZmZdd lmZmZmZdd lmZdd lmZmZdd lmZmZdd l m!Z!m"Z"ddl#m$Z$ddl%m&Z&m'Z'ddl(m)Z)ddgZ*GddeZ+GddeZ,GddZ-GddZ.GddZ/GddZ0y)) annotations)Optional)LiteralN)_legacy_response)batch_list_paramsbatch_create_params)BodyOmitQueryHeadersNotGivenomit not_given) path_templatemaybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)Batch)AsyncPaginatormake_request_options)MetadataBatches AsyncBatchesceZdZdZed dZed dZeeddded ddZ ddded ddZ eeddded  dd Z ddded dd Z y)r;Create large batches of API requests to run asynchronously.ct|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/openai/openai-python#accessing-raw-response-data-eg-headers )BatchesWithRawResponseselfs w/var/www/internaltesting/Shamanth/prverification/HIS_auto/venv/lib/python3.12/site-packages/openai/resources/batches.pywith_raw_responsezBatches.with_raw_responses&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/openai/openai-python#with_streaming_response )BatchesWithStreamingResponser&s r(with_streaming_responsezBatches.with_streaming_response&s,D11r*Nmetadataoutput_expires_after extra_headers extra_query extra_bodytimeoutc |jdt|||||dtjt |||| ddit 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`, `/v1/completions`, `/v1/moderations`, `/v1/images/generations`, `/v1/images/edits`, and `/v1/videos` 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 /batchescompletion_windowendpoint input_file_idr0r1 bearer_authTr2r3r4r5securitybodyoptionscast_to)_postrr BatchCreateParamsrr r'r:r;r<r0r1r2r3r4r5s r(createzBatches.create/shDzz  ): (%2 (,@ $55 )+'%'. '  r*r2r3r4r5c |std||jtd|t||||dditS)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/{batch_id}batch_idr=Tr>rBrC ValueError_getrrrr'rNr2r3r4r5s r(retrievezBatches.retrieves^.VW_Vbcd dyy /( C(+'%'.   r*afterlimitr2r3r4r5c|jdttt||||t ||dt j dditSa,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 r8)rVrWr=T)r2r3r4r5queryr?)pagerBmodel) _get_api_listrrrrrBatchListParamsr'rVrWr2r3r4r5s r(listz Batches.listsiB!! &(+'%%!&!&&55 (. #"  r*c |std||jtd|t||||dditS) 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 rK/batches/{batch_id}/cancelrMr=Tr>rOrQrDrrrrSs r(cancelzBatches.cancels^4VW_Vbcd dzz 6 J(+'%'.   r*)returnr%)rfr-r:zLiteral['24h']r;zLiteral['/v1/responses', '/v1/chat/completions', '/v1/embeddings', '/v1/completions', '/v1/moderations', '/v1/images/generations', '/v1/images/edits', '/v1/videos']r<strr0zOptional[Metadata] | Omitr1z-batch_create_params.OutputExpiresAfter | Omitr2Headers | Noner3 Query | Noner4 Body | Noner5'float | httpx.Timeout | None | NotGivenrfr rNrhr2rir3rjr4rkr5rlrfr)rV str | OmitrW int | Omitr2rir3rjr4rkr5rlrfzSyncCursorPage[Batch] __name__ __module__ __qualname____doc__rr)r.rrrGrTr`rer*r(rrsE,,22./3NR)-$("&;D-V *V   V V ,V L!V &&'V (")V * +V ,9-V . /V |)-$("&;D# # & # "#  # 9#  # P! )-$("&;D3 3  3 &3 "3  3 93  3 v)-$("&;D& & & & "&  & 9&  & r*ceZdZdZed dZed dZeeddded ddZ ddded ddZ eeddded  dd Z ddded dd Z y)r r"ct|Sr$)AsyncBatchesWithRawResponser&s r(r)zAsyncBatches.with_raw_response s+400r*ct|Sr,)!AsyncBatchesWithStreamingResponser&s r(r.z$AsyncBatches.with_streaming_responses166r*Nr/c K|jdt|||||dtjd{t |||| ddit d{S7'7w)r7r8r9Nr=Tr>r@)rDrr rErrrFs r(rGzAsyncBatches.create sDZZ ,): (%2 (,@ $55  )+'%'. '      s!0AA "AAAArHc K|std||jtd|t||||dditd{S7w) rJrKrLrMr=Tr>rONrPrSs r(rTzAsyncBatches.retrievexsl.VW_Vbcd dYY /( C(+'%'.     AA A ArUc|jdttt||||t ||dt j dditSrY)r]rrrrrr^r_s r(r`zAsyncBatches.listsiB!!  '(+'%%!&!&&55 (. #"  r*c K|std||jtd|t||||dditd{S7w) rbrKrcrMr=Tr>rONrdrSs r(rezAsyncBatches.cancelsl4VW_Vbcd dZZ 6 J(+'%'.     r})rfrx)rfrzrgrm)rVrnrWror2rir3rjr4rkr5rlrfz-AsyncPaginator[Batch, AsyncCursorPage[Batch]]rprur*r(r r sE1177./3NR)-$("&;D-V *V   V V ,V L!V &&'V (")V * +V ,9-V . /V |)-$("&;D# # & # "#  # 9#  # P! )-$("&;D3 3  3 &3 "3  3 93  73 v)-$("&;D& & & & "&  & 9&  & r*ceZdZddZy)r%c2||_tj|j|_tj|j|_tj|j |_tj|j |_yN)_batchesrto_raw_response_wrapperrGrTr`rer'batchess r(__init__zBatchesWithRawResponse.__init__su &>> NN  )@@     %<< LL  '>> NN  r*NrrrfNonerqrrrsrrur*r(r%r% r*r%ceZdZddZy)rxc2||_tj|j|_tj|j|_tj|j |_tj|j |_yr)rrasync_to_raw_response_wrapperrGrTr`rers r(rz$AsyncBatchesWithRawResponse.__init__su &DD NN  )FF     %BB LL  'DD NN  r*Nrr rfrrrur*r(rxrx rr*rxceZdZddZy)r-c||_t|j|_t|j|_t|j|_t|j |_yr)rrrGrTr`rers r(rz%BatchesWithStreamingResponse.__init__ s] 2 NN  5     1 LL  3 NN  r*Nrrrur*r(r-r-rr*r-ceZdZddZy)rzc||_t|j|_t|j|_t|j|_t|j |_yr)rrrGrTr`rers r(rz*AsyncBatchesWithStreamingResponse.__init__2s] 8 NN  ;     7 LL  9 NN  r*Nrrrur*r(rzrz1rr*rz)1 __future__rtypingrtyping_extensionsrhttpxrtypesrr _typesr r r r rrr_utilsrrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrrtypes.shared_params.metadatar__all__rr r%rxr-rzrur*r(rs#% :JJJJJ%9X8?3 n %n on bn #n b  $  $  $  r*