^=i?hddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZdd lmZdd lmZmZdd lmZmZmZmZdd lm Z dd l!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddgZ)GddeZ*GddeZ+GddZ,GddZ-GddZ.GddZ/y)) annotations)DictOptional)LiteralN)batch_create_params)BodyOmitQueryHeadersNotGivenomit not_given)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_raw_response_wrapperto_streamed_response_wrapperasync_to_raw_response_wrapper"async_to_streamed_response_wrapper)make_request_options)BatchListResponse)BatchCancelResponse)BatchCreateResponse)BatchRetrieveResponseBatches AsyncBatchesceZdZed dZed dZeddded d dZddded d dZ ddded ddZ ddded dd Z y)rct|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/groq/groq-python#accessing-raw-response-data-eg-headers )BatchesWithRawResponseselfs S/home/rose/Desktop/poly/venv/lib/python3.12/site-packages/groq/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/groq/groq-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. [Learn more](/docs/batch). Args: completion_window: The time frame within which the batch should be processed. Durations from `24h` to `7d` are supported. endpoint: The endpoint to be used for all requests in the batch. Currently `/v1/chat/completions` is supported. input_file_id: The ID of an uploaded file that contains requests for the new batch. See [upload file](/docs/api-reference#files-upload) for how to upload a file. Your input file must be formatted as a [JSONL file](/docs/batch), and must be uploaded with the purpose `batch`. The file can be up to 100 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 /openai/v1/batchescompletion_windowendpoint input_file_idr.r/r0r1r2bodyoptionscast_to)_postrrBatchCreateParamsrr r%r7r8r9r.r/r0r1r2s r&createzBatches.create2s\Tzz  ): (%2 (  $55)+Q[el(  r(r:c 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 /openai/v1/batches/r:r=r> ValueError_getrrr%batch_idr/r0r1r2s r&retrievezBatches.retrievemsT.VW_Vbcd dyy!( ,(+Q[el*   r(c L|jdt||||tS)!List your organization's batches.r5r:rGrJrrr%r/r0r1r2s r&listz Batches.lists5yy (+Q[el&   r(c t|std||jd|dt||||tS)D Cancels 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 rErF/cancelr:rGrIr?rrrKs r&cancelzBatches.cancelsV.VW_Vbcd dzz!(7 3(+Q[el(   r()returnr#)rXr+r7strr8zLiteral['/v1/chat/completions']r9rZr.zOptional[Dict[str, str]] | Omitr/Headers | Noner0 Query | Noner1 Body | Noner2'float | httpx.Timeout | None | NotGivenrXr rLrZr/r[r0r\r1r]r2r^rXr r/r[r0r\r1r]r2r^rXr rLrZr/r[r0r\r1r]r2r^rXr __name__ __module__ __qualname__rr'r,rrrBrMrRrWr(r&rrs{,,2259)-$("&;D9 9 2 9  9 2 9 &9 "9  9 99  9 B)-$("&;D  &  "   9   L)-$("&;D &  "    9   2)-$("&;D  &  "   9   r(ceZdZed dZed dZeddded d dZddded d dZ ddded ddZ 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)r4r5r6Nr:r;)r?rrr@rrrAs r&rBzAsyncBatches.createsuTZZ ,): (%2 (  $55)+Q[el(     s!/AA AAAAr:c K|std||jd|t||||td{S7w)rDrErFr:rGNrHrKs r&rMzAsyncBatches.retrievesb.VW_Vbcd dYY!( ,(+Q[el*     srs#!% 'JJJ;%9 09==A n %b ob Jb #b J  $  $  $  r(