o RWjW@sHddlmZddlmZddlmZddlZddlmZddl 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*GdddeZ+GdddeZ,GdddZ-GdddZ.GdddZ/GdddZ0dS)) 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 AsyncBatchesc@eZdZdZed.ddZed/ddZeed d d ed d0ddZ d d d ed d1d"d#Z eed d d ed$d2d*d+Z d d d ed d1d,d-Z d S)3r;Create large batches of API requests to run asynchronously.returnBatchesWithRawResponsecCt|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 )r#selfr(g/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/openai/resources/batches.pywith_raw_responsezBatches.with_raw_responseBatchesWithStreamingResponsecCr$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 )r,r&r(r(r)with_streaming_response&zBatches.with_streaming_responseNmetadataoutput_expires_after extra_headers extra_query extra_bodytimeoutcompletion_windowLiteral['24h']endpointLiteral['/v1/responses', '/v1/chat/completions', '/v1/embeddings', '/v1/completions', '/v1/moderations', '/v1/images/generations', '/v1/images/edits', '/v1/videos'] input_file_idstrr1Optional[Metadata] | Omitr2-batch_create_params.OutputExpiresAfter | Omitr3Headers | Noner4 Query | Noner5 Body | Noner6'float | httpx.Timeout | None | NotGivenrc Cs8|jdt|||||dtjt|||| ddidtdS) 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 /batchesr7r9r;r1r2 bearer_authTr3r4r5r6securitybodyoptionscast_to)_postrrBatchCreateParamsrr r'r7r9r;r1r2r3r4r5r6r(r(r)create/s(B zBatches.creater3r4r5r6batch_idc C<|s td||jtd|dt||||ddidtdS)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}rRrFTrGrKrL ValueError_getrrrr'rRr3r4r5r6r(r(r)retrieves zBatches.retrieveafterlimitr3r4r5r6r_ str | Omitr` int | OmitSyncCursorPage[Batch]c C8|jdttt||||t||dtjddidtdSa,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 rD)r_r`rFT)r3r4r5r6queryrH)pagerKmodel) _get_api_listrrrrrBatchListParamsr'r_r`r3r4r5r6r(r(r)list$!z Batches.listc CrS) 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 rU/batches/{batch_id}/cancelrWrFTrGrXrZrMrrrr\r(r(r)cancels zBatches.cancel)r"r#)r"r,r7r8r9r:r;r<r1r=r2r>r3r?r4r@r5rAr6rBr"r rRr<r3r?r4r@r5rAr6rBr"r)r_rar`rbr3r?r4r@r5rAr6rBr"rc __name__ __module__ __qualname____doc__rr*r.rrrPr]rlrqr(r(r(r)r<  ^(;c@r )3rr!r"AsyncBatchesWithRawResponsecCr$r%)rzr&r(r(r)r* r+zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr$r-)r{r&r(r(r)r.r/z$AsyncBatches.with_streaming_responseNr0r7r8r9r:r;r<r1r=r2r>r3r?r4r@r5rAr6rBrc sF|jdt|||||dtjIdHt|||| ddidtdIdHS)rCrDrENrFTrGrI)rMrrrNrrrOr(r(r)rP s*B  zAsyncBatches.createrQrRc D|s td||jtd|dt||||ddidtdIdHS) rTrUrVrWrFTrGrXNrYr\r(r(r)r]xs  zAsyncBatches.retriever^r_rar`rb-AsyncPaginator[Batch, AsyncCursorPage[Batch]]c Crdre)rirrrrrrjrkr(r(r)rlrmzAsyncBatches.listc r|) rnrUrorWrFTrGrXNrpr\r(r(r)rqs  zAsyncBatches.cancel)r"rz)r"r{rrrs)r_rar`rbr3r?r4r@r5rAr6rBr"r}rtr(r(r(r)r ryc@eZdZdddZdS) r#batchesrr"NonecCB||_t|j|_t|j|_t|j|_t|j|_dSN)_batchesrto_raw_response_wrapperrPr]rlrqr'rr(r(r)__init__ zBatchesWithRawResponse.__init__Nrrr"rrurvrwrr(r(r(r)r#r#c@r~) rzrrr"rcCrr)rrasync_to_raw_response_wrapperrPr]rlrqrr(r(r)rrz$AsyncBatchesWithRawResponse.__init__Nrrr"rrr(r(r(r)rz rrzc@r~) r,rrr"rcC:||_t|j|_t|j|_t|j|_t|j|_dSr)rrrPr]rlrqrr(r(r)r  z%BatchesWithStreamingResponse.__init__Nrrr(r(r(r)r,rr,c@r~) r{rrr"rcCrr)rrrPr]rlrqrr(r(r)r2rz*AsyncBatchesWithStreamingResponse.__init__Nrrr(r(r(r)r{1rr{)1 __future__rtypingrtyping_extensionsrhttpxrtypesrr_typesr r r r r rr_utilsrrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrrtypes.shared_params.metadatar__all__rrr#rzr,r{r(r(r(r)s0    $   rr