@gzL~ddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z ddl mZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lmZm Z ddl!m"Z"ddl#m$Z$m%Z%ddgZ&GddeZ'GddeZ(GddZ)GddZ*GddZ+GddZ,dS)) annotations)DictOptional)LiteralN)_legacy_response)batch_list_paramsbatch_create_params) NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)Batch)AsyncPaginatormake_request_optionsBatches AsyncBatchesceZdZed%dZed&dZedddedd'dZdddedd(dZeedddedd)d#Z dddedd(d$Z dS)*rreturnBatchesWithRawResponsec t|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/openai/openai-python#accessing-raw-response-data-eg-headers )r selfs L/var/www/piapp/venv/lib/python3.11/site-packages/openai/resources/batches.pywith_raw_responsezBatches.with_raw_responses&d+++BatchesWithStreamingResponsec t|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 )r(r#s r%with_streaming_responsezBatches.with_streaming_response)s,D111r'Nmetadata extra_headers extra_query extra_bodytimeoutcompletion_windowLiteral['24h']endpointDLiteral['/v1/chat/completions', '/v1/embeddings', '/v1/completions'] input_file_idstrr-#Optional[Dict[str, str]] | NotGivenr.Headers | Noner/ Query | Noner0 Body | Noner1'float | httpx.Timeout | None | NotGivenrc |dt||||dtjt ||||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/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: Optional custom metadata for the 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 /batchesr2r4r6r-r.r/r0r1bodyoptionscast_to)_postrr BatchCreateParamsrr r$r2r4r6r-r.r/r0r1s r%createzBatches.create2sr\zz  ): (%2 (  $5)+Q[el   r'rAbatch_idc |std||d|t||||tS)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/rArDrE ValueError_getrrr$rJr.r/r0r1s r%retrievezBatches.retrieveqsm. ecW_ccdd dyy " " "(+Q[el    r'afterlimitr.r/r0r1rVstr | NotGivenrWint | NotGivenSyncCursorPage[Batch]c|dttt||||t ||dt jtSa,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 r?)rVrW)r.r/r0r1query)pagerDmodel) _get_api_listrrrrr BatchListParamsr$rVrWr.r/r0r1s r%listz Batches.listsuB!! &(+'%%!&!&&5    !"   r'c |std||d|dt||||tS) 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 rMrN/cancelrArOrQrFrrrSs r%cancelzBatches.cancelsp4 ecW_ccdd dzz ) ) ) )(+Q[el    r')rr )rr(r2r3r4r5r6r7r-r8r.r9r/r:r0r;r1r<rr rJr7r.r9r/r:r0r;r1r<rr)rVrXrWrYr.r9r/r:r0r;r1r<rrZ __name__ __module__ __qualname__rr&r+r rIrTrcrhr'r%rrs,,,_,222_29B)-$("&;D= = = = = = J)-$("&;D      H!* ))-$("&;D2 2 2 2 2 2 t)-$("&;D" " " " " " " " r'ceZdZed%dZed&dZedddedd'dZdddedd(dZeedddedd)d#Z dddedd(d$Z dS)*rrAsyncBatchesWithRawResponsec t|Sr")rqr#s r%r&zAsyncBatches.with_raw_responses+4000r'!AsyncBatchesWithStreamingResponsec t|Sr*)rsr#s r%r+z$AsyncBatches.with_streaming_responses1666r'Nr,r2r3r4r5r6r7r-r8r.r9r/r:r0r;r1r<rc K|dt||||dtjd{Vt ||||t d{VS)r>r?r@NrArB)rFrr rGrrrHs r%rIzAsyncBatches.creates\ZZ ,): (%2 (  $5)+Q[el          r'rArJc K|std||d|t||||td{VS)rLrMrNrArONrPrSs r%rTzAsyncBatches.retrieve>s. ecW_ccdd dYY " " "(+Q[el          r'rUrVrXrWrY-AsyncPaginator[Batch, AsyncCursorPage[Batch]]c|dttt||||t ||dt jtSr\)r`rrrrr rarbs r%rczAsyncBatches.list_suB!!  '(+'%%!&!&&5    !"   r'c K|std||d|dt||||td{VS)rerMrNrfrArONrgrSs r%rhzAsyncBatches.cancels4 ecW_ccdd dZZ ) ) ) )(+Q[el           r')rrq)rrsrirj)rVrXrWrYr.r9r/r:r0r;r1r<rrwrkror'r%rrs111_1777_79B)-$("&;D= = = = = = J)-$("&;D      H!* ))-$("&;D2 2 2 2 2 2 t)-$("&;D" " " " " " " " r'ceZdZddZdS)r batchesrrNonec||_tj|j|_tj|j|_tj|j|_tj|j|_dSN)_batchesrto_raw_response_wrapperrIrTrcrhr$r{s r%__init__zBatchesWithRawResponse.__init__sx &> N   )@     %< L   '> N   r'Nr{rrr|rlrmrnrror'r%r r (      r'r ceZdZddZdS)rqr{rrr|c||_tj|j|_tj|j|_tj|j|_tj|j|_dSr~)rrasync_to_raw_response_wrapperrIrTrcrhrs r%rz$AsyncBatchesWithRawResponse.__init__sx &D N   )F     %B L   'D N   r'Nr{rrr|rror'r%rqrqrr'rqceZdZddZdS)r(r{rrr|c||_t|j|_t|j|_t|j|_t|j|_dSr~)rrrIrTrcrhrs r%rz%BatchesWithStreamingResponse.__init__sl 2 N   5     1 L   3 N   r'Nrrror'r%r(r(rr'r(ceZdZddZdS)rsr{rrr|c||_t|j|_t|j|_t|j|_t|j|_dSr~)rrrIrTrcrhrs r%rz*AsyncBatchesWithStreamingResponse.__init__sl 8 N   ;     7 L   9 N   r'Nrrror'r%rsrsrr'rs)- __future__rtypingrrtyping_extensionsrhttpxrtypesr r _typesr r r rr_utilsrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrr__all__rrr rqr(rsror'r%rs{#"""""!!!!!!!!%%%%%% ::::::::>>>>>>>>>>>>>>&%%%%%99999999XXXXXXXX88888888 n %J J J J J oJ J J ZJ J J J J #J J J Z        $        $        $          r'