iTddlmZddlmZddlmZddlZddlmZddl 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/GddZ0dS)) 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)dZeedddedd*dZ dddedd+dZ eeddded d,d&Z dddedd+d'Z dS)-r;Create large batches of API requests to run asynchronously.returnBatchesWithRawResponsec t|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$selfs _/home/ubuntu/.hermes/hermes-agent/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+Nmetadataoutput_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 |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/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;r1r2r3r4r5r6bodyoptionscast_to)_postrr BatchCreateParamsrr r(r7r9r;r1r2r3r4r5r6s r)createzBatches.create/suDzz  ): (%2 (,@ $5  )+Q[el   r+rGbatch_idc |std||td|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/{batch_id}rPrGrJrK ValueError_getrrrr(rPr3r4r5r6s r)retrievezBatches.retrievess. ecW_ccdd dyy /( C C C(+Q[el    r+afterlimitr3r4r5r6r] str | Omitr^ int | OmitSyncCursorPage[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 rE)r]r^)r3r4r5r6query)pagerJmodel) _get_api_listrrrrrBatchListParamsr(r]r^r3r4r5r6s r)listz Batches.listsuB!! &(+'%%!&!&&5    !"   r+c |std||td|t||||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 rS/batches/{batch_id}/cancelrUrGrVrXrLrrrrZs r)cancelzBatches.cancelss4 ecW_ccdd dzz 6 J J J(+Q[el    r+)r#r$)r#r,r7r8r9r:r;r<r1r=r2r>r3r?r4r@r5rAr6rBr#r rPr<r3r?r4r@r5rAr6rBr#r)r]r_r^r`r3r?r4r@r5rAr6rBr#ra __name__ __module__ __qualname____doc__rr*r/rrrOr[rjror+r)rrs!EE,,,_,222_2./3NR)-$("&;D-R R R R R R t)-$("&;D      H! )-$("&;D2 2 2 2 2 2 t)-$("&;D" " " " " " " " r+ceZdZdZed(dZed)dZeedddedd*dZ dddedd+dZ eeddded d,d&Z dddedd+d'Z dS)-r r"r#AsyncBatchesWithRawResponsec t|Sr&)ryr's r)r*zAsyncBatches.with_raw_responses+4000r+!AsyncBatchesWithStreamingResponsec t|Sr.)r{r's r)r/z$AsyncBatches.with_streaming_response s1666r+Nr0r7r8r9r:r;r<r1r=r2r>r3r?r4r@r5rAr6rBrc K|dt|||||dtjd{Vt |||| t d{VS)rDrErFNrGrH)rLrr rMrrrNs r)rOzAsyncBatches.createsDZZ ,): (%2 (,@ $5        )+Q[el          r+rGrPc K|std||td|t||||td{VS)rRrSrTrUrGrVNrWrZs r)r[zAsyncBatches.retrievegs. ecW_ccdd dYY /( C C C(+Q[el          r+r\r]r_r^r`-AsyncPaginator[Batch, AsyncCursorPage[Batch]]c|dttt||||t ||dt jtSrc)rgrrrrrrhris r)rjzAsyncBatches.listsuB!!  '(+'%%!&!&&5    !"   r+c K|std||td|t||||td{VS)rlrSrmrUrGrVNrnrZs r)rozAsyncBatches.cancels4 ecW_ccdd dZZ 6 J J J(+Q[el           r+)r#ry)r#r{rprq)r]r_r^r`r3r?r4r@r5rAr6rBr#rrrrwr+r)r r s!EE111_1777_7./3NR)-$("&;D-R R R R R R t)-$("&;D      H! )-$("&;D2 2 2 2 2 2 t)-$("&;D" " " " " " " " r+ceZdZddZdS)r$batchesrr#Nonec||_tj|j|_tj|j|_tj|j|_tj|j|_dSN)_batchesrto_raw_response_wrapperrOr[rjror(rs r)__init__zBatchesWithRawResponse.__init__sx &> N   )@     %< L   '> N   r+Nrrr#rrsrtrurrwr+r)r$r$(      r+r$ceZdZddZdS)ryrr r#rc||_tj|j|_tj|j|_tj|j|_tj|j|_dSr)rrasync_to_raw_response_wrapperrOr[rjrors r)rz$AsyncBatchesWithRawResponse.__init__sx &D N   )F     %B L   'D N   r+Nrr r#rrrwr+r)ryryrr+ryceZdZddZdS)r,rrr#rc||_t|j|_t|j|_t|j|_t|j|_dSr)rrrOr[rjrors r)rz%BatchesWithStreamingResponse.__init__sl 2 N   5     1 L   3 N   r+Nrrrwr+r)r,r,rr+r,ceZdZddZdS)r{rr r#rc||_t|j|_t|j|_t|j|_t|j|_dSr)rrrOr[rjrors r)rz*AsyncBatchesWithStreamingResponse.__init__sl 8 N   ;     7 L   9 N   r+Nrrrwr+r)r{r{rr+r{)1 __future__rtypingrtyping_extensionsrhttpxrtypesrr _typesr r r r rrr_utilsrrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrrtypes.shared_params.metadatar__all__rr r$ryr,r{rwr+r)rsu#"""""%%%%%% ::::::::JJJJJJJJJJJJJJJJJJJJJJJJJJJJ%%%%%%99999999XXXXXXXX88888888????????333333 n %a a a a a oa a a Ha a a a a #a a a H        $        $        $          r+