gzLXddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z ddl mZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lmZm Z ddl!m"Z"ddl#m$Z$m%Z%ddgZ&GddeZ'GddeZ(GddZ)GddZ*GddZ+GddZ,y)) annotations)DictOptional)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_optionsBatches AsyncBatchesceZdZed dZed dZeddded d dZddded ddZeeddded dd Z ddded dd Z y)rct|Sa This property can be used as a prefix for any HTTP method call to return the 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 )BatchesWithRawResponseselfs M/var/www/openai/venv/lib/python3.12/site-packages/openai/resources/batches.pywith_raw_responsezBatches.with_raw_responses&d++ct|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 )BatchesWithStreamingResponser"s r$with_streaming_responsezBatches.with_streaming_response)s,D11r&Nmetadata extra_headers extra_query extra_bodytimeoutc |jdt||||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/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: Optional custom metadata for the 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 /batchescompletion_windowendpoint input_file_idr,r-r.r/r0bodyoptionscast_to)_postrr BatchCreateParamsrr r#r5r6r7r,r-r.r/r0s r$createzBatches.create2s\\zz  ): (%2 (  $55)+Q[el  r&r8c r|std||jd|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/r8r;r< ValueError_getrrr#batch_idr-r.r/r0s r$retrievezBatches.retrieveqsT.VW_Vbcd dyyz "(+Q[el   r&afterlimitr-r.r/r0c|jdttt||||t ||dt j tSa,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 r3)rMrN)r-r.r/r0query)pager;model) _get_api_listrrrrr BatchListParamsr#rMrNr-r.r/r0s r$listz Batches.lists_B!! &(+'%%!&!&&55  !"  r&c t|std||jd|dt||||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 rCrD/cancelr8rErGr=rrrIs r$cancelzBatches.cancelsV4VW_Vbcd dzzz )(+Q[el   r&)returnr!)r]r)r5zLiteral['24h']r6zDLiteral['/v1/chat/completions', '/v1/embeddings', '/v1/completions']r7strr,z#Optional[Dict[str, str]] | NotGivenr-Headers | Noner. Query | Noner/ Body | Noner0'float | httpx.Timeout | None | NotGivenr]r rJr_r-r`r.rar/rbr0rcr]r)rMstr | NotGivenrNint | NotGivenr-r`r.rar/rbr0rcr]zSyncCursorPage[Batch] __name__ __module__ __qualname__rr%r*r r@rKrWr\r&r$rrs,,229B)-$("&;D= *= W =  = 6 = &= "=  = 9=  = J)-$("&;D  &  "   9   H!* ))-$("&;D2 2  2 &2 "2  2 92  2 t)-$("&;D" " & " ""  " 9"  " r&ceZdZed dZed dZeddded d dZddded ddZeeddded dd Z ddded dd Z y)rct|Sr )AsyncBatchesWithRawResponser"s r$r%zAsyncBatches.with_raw_responses+400r&ct|Sr()!AsyncBatchesWithStreamingResponser"s r$r*z$AsyncBatches.with_streaming_responses166r&Nr+c K|jdt||||dtjd{t ||||t d{S7$7w)r2r3r4Nr8r9)r=rr r>rrr?s r$r@zAsyncBatches.createsu\ZZ ,): (%2 (  $55)+Q[el     s!/AA AAAAr8c K|std||jd|t||||td{S7w)rBrCrDr8rENrFrIs r$rKzAsyncBatches.retrieve>sb.VW_Vbcd dYYz "(+Q[el     s> NN  )@@     %<< LL  '>> NN  r&Nr{rr]Nonerhrirjr|rkr&r$r!r! r&r!ceZdZddZy)rnc2||_tj|j|_tj|j|_tj|j |_tj|j |_yrw)rxrasync_to_raw_response_wrapperr@rKrWr\rzs r$r|z$AsyncBatchesWithRawResponse.__init__su &DD NN  )FF     %BB LL  'DD NN  r&Nr{rr]r~rrkr&r$rnrnrr&rnceZdZddZy)r)c||_t|j|_t|j|_t|j|_t|j |_yrw)rxrr@rKrWr\rzs r$r|z%BatchesWithStreamingResponse.__init__s] 2 NN  5     1 LL  3 NN  r&Nr}rrkr&r$r)r)rr&r)ceZdZddZy)rpc||_t|j|_t|j|_t|j|_t|j |_yrw)rxrr@rKrWr\rzs r$r|z*AsyncBatchesWithStreamingResponse.__init__s] 8 NN  ;     7 LL  9 NN  r&Nrrrkr&r$rprprr&rp)- __future__rtypingrrtyping_extensionsrhttpxrtypesr r _typesr r r rr_utilsrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrr__all__rrr!rnr)rprkr&r$rs#!% :>>&9X8 n %J oJ ZJ #J Z  $  $  $  r&