U ܟjuO@s<ddlmZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lmZmZddl 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) NOT_GIVENBodyQueryHeadersNotGiven)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)Batch)AsyncPaginatormake_request_options)MetadataBatches AsyncBatchesc @seZdZeddddZeddddZeddded d d d d dddddd ddZdddedd ddddddddZeedddeddddddddddd Z dddedd ddddddd!d"Z dS)#rBatchesWithRawResponsereturncCst|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 )rselfr#batch_idr+r,r-r.rcCs4|std||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/r>rArB ValueError_getrrr"rHr+r,r-r.r#r#r$retrievetszBatches.retrieveafterlimitr+r,r-r.str | NotGivenint | NotGivenzSyncCursorPage[Batch]rSrTr+r,r-r.rc Cs2|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 r<)rSrT)r+r,r-r.query)pagerAmodel) _get_api_listrrrrrBatchListParamsr"rSrTr+r,r-r.r#r#r$lists"! z Batches.listcCs6|std||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 rJrK/cancelr>rLrNrCrrrPr#r#r$cancels zBatches.cancel __name__ __module__ __qualname__rr%r(r rFrQr_rcr#r#r#r$rs6 "J$:c @seZdZeddddZeddddZeddded d d d d dddddd ddZdddedd ddddddddZeedddeddddddddddd Z dddedd ddddddd!d"Z dS)#rAsyncBatchesWithRawResponsercCst|Sr )rhr!r#r#r$r%szAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCst|Sr')rir!r#r#r$r(sz$AsyncBatches.with_streaming_responseNr)r/r0r1r2r3r4r5r6rr7c s<|jdt||||dtjIdHt||||dtdIdHS)r;r<r=Nr>r?)rCrrrDrrrEr#r#r$rFs$3 zAsyncBatches.creater>rGcs:|std||jd|t||||dtdIdHS)rIrJrKr>rLNrMrPr#r#r$rQFszAsyncBatches.retrieverRrUrVz-AsyncPaginator[Batch, AsyncCursorPage[Batch]]rWc Cs2|jdttt||||t||dtjdtdSrX)r\rrrrrr]r^r#r#r$r_gs"! zAsyncBatches.listcs<|std||jd|dt||||dtdIdHS)r`rJrKrar>rLNrbrPr#r#r$rcs zAsyncBatches.cancelrdr#r#r#r$rs6 "J$:c@seZdZdddddZdS)rrNonebatchesrcCsB||_t|j|_t|j|_t|j|_t|j|_dSN)_batchesrZto_raw_response_wrapperrFrQr_rcr"rlr#r#r$__init__szBatchesWithRawResponse.__init__Nrerfrgrpr#r#r#r$rsrc@seZdZdddddZdS)rhrrjrkcCsB||_t|j|_t|j|_t|j|_t|j|_dSrm)rnrZasync_to_raw_response_wrapperrFrQr_rcror#r#r$rpsz$AsyncBatchesWithRawResponse.__init__Nrqr#r#r#r$rhsrhc@seZdZdddddZdS)r&rrjrkcCs:||_t|j|_t|j|_t|j|_t|j|_dSrm)rnrrFrQr_rcror#r#r$rpsz%BatchesWithStreamingResponse.__init__Nrqr#r#r#r$r&sr&c@seZdZdddddZdS)rirrjrkcCs:||_t|j|_t|j|_t|j|_t|j|_dSrm)rnrrFrQr_rcror#r#r$rpsz*AsyncBatchesWithStreamingResponse.__init__Nrqr#r#r#r$risri). __future__rtypingrtyping_extensionsrZhttpxrtypesrr_typesr r r r r _utilsrr_compatrZ _resourcerr _responserrZ paginationrrZ types.batchrZ _base_clientrrZtypes.shared_params.metadatar__all__rrrrhr&rir#r#r#r$s.       SS