iWddlmZddlmZddlmZddlZddlmZddl 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/piyush/.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 |||| ddit 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;r1r2 bearer_authTr3r4r5r6securitybodyoptionscast_to)_postrr BatchCreateParamsrr r(r7r9r;r1r2r3r4r5r6s r)createzBatches.create/s~Dzz  ): (%2 (,@ $5  )+'%'. '   r+r3r4r5r6batch_idc |std||td|t||||dditS)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}rSrGTrHrLrM ValueError_getrrrr(rSr3r4r5r6s r)retrievezBatches.retrieves|. ecW_ccdd dyy /( C C C(+'%'.    r+afterlimitr3r4r5r6r` str | Omitra int | OmitSyncCursorPage[Batch]c|dttt||||t ||dt jdditSa,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`rarGT)r3r4r5r6queryrI)pagerLmodel) _get_api_listrrrrrBatchListParamsr(r`rar3r4r5r6s r)listz Batches.listsB!! &(+'%%!&!&&5 (.   #"   r+c |std||td|t||||dditS) 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 rV/batches/{batch_id}/cancelrXrGTrHrYr[rNrrrr]s r)cancelzBatches.cancels|4 ecW_ccdd dzz 6 J J J(+'%'.    r+)r#r$)r#r,r7r8r9r:r;r<r1r=r2r>r3r?r4r@r5rAr6rBr#r rSr<r3r?r4r@r5rAr6rBr#r)r`rbrarcr3r?r4r@r5rAr6rBr#rd __name__ __module__ __qualname____doc__rr*r/rrrQr^rmrrr+r)rrs!EE,,,_,222_2./3NR)-$("&;D-V V V V V V |)-$("&;D# # # # # # P! )-$("&;D3 3 3 3 3 3 v)-$("&;D& & & & & & & & r+ceZdZdZed(dZed)dZeedddedd*dZ dddedd+dZ eeddded d,d&Z dddedd+d'Z dS)-r r"r#AsyncBatchesWithRawResponsec t|Sr&)r|r's r)r*zAsyncBatches.with_raw_response s+4000r+!AsyncBatchesWithStreamingResponsec t|Sr.)r~r's r)r/z$AsyncBatches.with_streaming_responses1666r+Nr0r7r8r9r:r;r<r1r=r2r>r3r?r4r@r5rAr6rBrc K|dt|||||dtjd{Vt |||| ddit d{VS)rDrErFNrGTrHrJ)rNrr rOrrrPs r)rQzAsyncBatches.create sDZZ ,): (%2 (,@ $5        )+'%'. '          r+rRrSc K|std||td|t||||dditd{VS) rUrVrWrXrGTrHrYNrZr]s r)r^zAsyncBatches.retrievexs. ecW_ccdd dYY /( C C C(+'%'.          r+r_r`rbrarc-AsyncPaginator[Batch, AsyncCursorPage[Batch]]c|dttt||||t ||dt jdditSrf)rjrrrrrrkrls r)rmzAsyncBatches.listsB!!  '(+'%%!&!&&5 (.   #"   r+c K|std||td|t||||dditd{VS) rorVrprXrGTrHrYNrqr]s r)rrzAsyncBatches.cancels4 ecW_ccdd dZZ 6 J J J(+'%'.          r+)r#r|)r#r~rsrt)r`rbrarcr3r?r4r@r5rAr6rBr#rrurzr+r)r r s!EE111_1777_7./3NR)-$("&;D-V V V V V V |)-$("&;D# # # # # # P! )-$("&;D3 3 3 3 3 3 v)-$("&;D& & & & & & & & r+ceZdZddZdS)r$batchesrr#Nonec||_tj|j|_tj|j|_tj|j|_tj|j|_dSN)_batchesrto_raw_response_wrapperrQr^rmrrr(rs r)__init__zBatchesWithRawResponse.__init__sx &> N   )@     %< L   '> N   r+Nrrr#rrvrwrxrrzr+r)r$r$(      r+r$ceZdZddZdS)r|rr r#rc||_tj|j|_tj|j|_tj|j|_tj|j|_dSr)rrasync_to_raw_response_wrapperrQr^rmrrrs r)rz$AsyncBatchesWithRawResponse.__init__sx &D N   )F     %B L   'D N   r+Nrr r#rrrzr+r)r|r| rr+r|ceZdZddZdS)r,rrr#rc||_t|j|_t|j|_t|j|_t|j|_dSr)rrrQr^rmrrrs r)rz%BatchesWithStreamingResponse.__init__ sl 2 N   5     1 L   3 N   r+Nrrrzr+r)r,r,rr+r,ceZdZddZdS)r~rr r#rc||_t|j|_t|j|_t|j|_t|j|_dSr)rrrQr^rmrrrs r)rz*AsyncBatchesWithStreamingResponse.__init__2sl 8 N   ;     7 L   9 N   r+Nrrrzr+r)r~r~1rr+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$r|r,r~rzr+r)rsu#"""""%%%%%% ::::::::JJJJJJJJJJJJJJJJJJJJJJJJJJJJ%%%%%%99999999XXXXXXXX88888888????????333333 n %n n n n n on n n bn n n n n #n n n b        $        $        $          r+