o $ivQ@sDddlmZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZmZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddgZ)GdddeZ*GdddeZ+GdddZ,GdddZ-GdddZ.GdddZ/dS)) annotations)Optional)LiteralN)_legacy_response)batch_list_paramsbatch_create_params)BodyOmitQueryHeadersNotGivenomit not_given)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)Batch)AsyncPaginatormake_request_options)MetadataBatches AsyncBatchesc@eZdZed-ddZed.ddZeeddded d/ddZdddedd0d!d"Z eeddded#d1d)d*Z dddedd0d+d,Z dS)2rreturnBatchesWithRawResponsecCt|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&X/var/www/html/flask_server/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']endpointULiteral['/v1/responses', '/v1/chat/completions', '/v1/embeddings', '/v1/completions'] input_file_idstrr/Optional[Metadata] | Omitr0-batch_create_params.OutputExpiresAfter | Omitr1Headers | Noner2 Query | Noner3 Body | Noner4'float | httpx.Timeout | None | NotGivenrc Cs2|jdt|||||dtjt|||| dtdS) 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 /batchesr5r7r9r/r0r1r2r3r4bodyoptionscast_to)_postrrBatchCreateParamsrr r%r5r7r9r/r0r1r2r3r4r&r&r'create-s 7 zBatches.createrDbatch_idcCs4|s td||jd|t||||dtdS)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/rDrGrH ValueError_getrrr%rMr1r2r3r4r&r&r'retrievevszBatches.retrieveafterlimitr1r2r3r4rX str | OmitrY int | OmitSyncCursorPage[Batch]c C2|jdttt||||t||dtjdtdSa,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 rB)rXrY)r1r2r3r4query)pagerGmodel) _get_api_listrrrrrBatchListParamsr%rXrYr1r2r3r4r&r&r'list"! z Batches.listcCs6|s td||jd|dt||||dtdS) 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 rOrP/cancelrDrQrSrIrrrUr&r&r'cancels zBatches.cancel)r r!)r r*r5r6r7r8r9r:r/r;r0r<r1r=r2r>r3r?r4r@r r rMr:r1r=r2r>r3r?r4r@r r)rXrZrYr[r1r=r2r>r3r?r4r@r r\ __name__ __module__ __qualname__rr(r,rrrLrVrerjr&r&r&r'r:  O$:c@r)2rr AsyncBatchesWithRawResponsecCr"r#)rrr$r&r&r'r(r)zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr"r+)rsr$r&r&r'r,r-z$AsyncBatches.with_streaming_responseNr.r5r6r7r8r9r:r/r;r0r<r1r=r2r>r3r?r4r@rc s@|jdt|||||dtjIdHt|||| dtdIdHS)rArBrCNrDrE)rIrrrJrrrKr&r&r'rLs"7  zAsyncBatches.createrDrMcs<|s td||jd|t||||dtdIdHS)rNrOrPrDrQNrRrUr&r&r'rVMs zAsyncBatches.retrieverWrXrZrYr[-AsyncPaginator[Batch, AsyncCursorPage[Batch]]c Cr]r^)rbrrrrrrcrdr&r&r'renrfzAsyncBatches.listcs>|s td||jd|dt||||dtdIdHS)rgrOrPrhrDrQNrirUr&r&r'rjs  zAsyncBatches.cancel)r rr)r rsrkrl)rXrZrYr[r1r=r2r>r3r?r4r@r rtrmr&r&r&r'rrqc@eZdZdddZdS) r!batchesrr NonecCB||_t|j|_t|j|_t|j|_t|j|_dSN)_batchesrto_raw_response_wrapperrLrVrerjr%rvr&r&r'__init__ zBatchesWithRawResponse.__init__Nrvrr rwrnrorpr}r&r&r&r'r!r!c@ru) rrrvrr rwcCrxry)rzrasync_to_raw_response_wrapperrLrVrerjr|r&r&r'r}r~z$AsyncBatchesWithRawResponse.__init__Nrvrr rwrr&r&r&r'rrrrrc@ru) r*rvrr rwcC:||_t|j|_t|j|_t|j|_t|j|_dSry)rzrrLrVrerjr|r&r&r'r} z%BatchesWithStreamingResponse.__init__Nrrr&r&r&r'r*rr*c@ru) rsrvrr rwcCrry)rzrrLrVrerjr|r&r&r'r}rz*AsyncBatchesWithStreamingResponse.__init__Nrrr&r&r&r'rsrrs)0 __future__rtypingrtyping_extensionsrhttpxrtypesrr_typesr r r r r rr_utilsrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrrtypes.shared_params.metadatar__all__rrr!rrr*rsr&r&r&r's0    $   XX