o i?@sDddlmZddlmZmZddlmZddlZddlm 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)GdddeZ*GdddeZ+GdddZ,GdddZ-GdddZ.GdddZ/dS)) 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 AsyncBatchesc@eZdZed'ddZed(ddZeddded d)ddZdddedd*dd Z dddedd+d"d#Z dddedd,d%d&Z dS)-rreturnBatchesWithRawResponsecCt|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 )r!selfr&\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/groq/resources/batches.pywith_raw_responsezBatches.with_raw_responseBatchesWithStreamingResponsecCr"z 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 )r*r$r&r&r'with_streaming_response)zBatches.with_streaming_responseNmetadata extra_headers extra_query extra_bodytimeoutcompletion_windowstrendpointLiteral['/v1/chat/completions'] input_file_idr/Optional[Dict[str, str]] | Omitr0Headers | Noner1 Query | Noner2 Body | Noner3'float | httpx.Timeout | None | NotGivenrc Cs0|jdt||||dtjt||||dtdS)_ 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/batchesr4r6r8r/r0r1r2r3bodyoptionscast_to)_postrrBatchCreateParamsrr r%r4r6r8r/r0r1r2r3r&r&r'create2s* zBatches.createrAbatch_idrcCs4|s td||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 /openai/v1/batches/rArDrE ValueError_getrrr%rJr0r1r2r3r&r&r'retrievemszBatches.retrievercCs|jdt||||dtdS)!List your organization's batches.r?rArNrQrrr%r0r1r2r3r&r&r'lists z Batches.listrcCs6|s td||jd|dt||||dtdS)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 rLrM/cancelrArNrPrFrrrRr&r&r'cancels zBatches.cancel)r r!)r r*r4r5r6r7r8r5r/r9r0r:r1r;r2r<r3r=r r rJr5r0r:r1r;r2r<r3r=r r r0r:r1r;r2r<r3r=r r rJr5r0r:r1r;r2r<r3r=r r __name__ __module__ __qualname__rr(r,r rrIrSrWr[r&r&r&r'r4  A&c@r)-rr AsyncBatchesWithRawResponsecCr"r#)rer$r&r&r'r(r)zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr"r+)rfr$r&r&r'r,r-z$AsyncBatches.with_streaming_responseNr.r4r5r6r7r8r/r9r0r:r1r;r2r<r3r=rc s>|jdt||||dtjIdHt||||dtdIdHS)r>r?r@NrArB)rFrrrGrrrHr&r&r'rIs *  zAsyncBatches.createrArJrcs<|s td||jd|t||||dtdIdHS)rKrLrMrArNNrOrRr&r&r'rSs zAsyncBatches.retrievercs$|jdt||||dtdIdHS)rTr?rArNNrUrVr&r&r'rW3s  zAsyncBatches.listrcs>|s td||jd|dt||||dtdIdHS)rXrLrMrYrArNNrZrRr&r&r'r[Fs  zAsyncBatches.cancel)r re)r rfr\r]r^r_r`r&r&r&r'rrdc@eZdZdddZdS) r!batchesrr NonecC:||_t|j|_t|j|_t|j|_t|j|_dSN)_batchesrrIrSrWr[r%rhr&r&r'__init__i zBatchesWithRawResponse.__init__Nrhrr rirarbrcrnr&r&r&r'r!hr!c@rg) rerhrr ricCrjrk)rlrrIrSrWr[rmr&r&r'rn{roz$AsyncBatchesWithRawResponse.__init__Nrhrr rirqr&r&r&r'rezrrrec@rg) r*rhrr ricCrjrk)rlrrIrSrWr[rmr&r&r'rnroz%BatchesWithStreamingResponse.__init__Nrprqr&r&r&r'r*rrr*c@rg) rfrhrr ricCrjrk)rlrrIrSrWr[rmr&r&r'rnroz*AsyncBatchesWithStreamingResponse.__init__Nrsrqr&r&r&r'rfrrrf)0 __future__rtypingrrtyping_extensionsrhttpxtypesr_typesrr r r r r r_utilsrr_compatr _resourcerr _responserrrr _base_clientrtypes.batch_list_responsertypes.batch_cancel_responsertypes.batch_create_responsertypes.batch_retrieve_responser__all__rrr!rer*rfr&r&r&r's0   $      &&