o i5@@sHddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z m Z mZmZmZmZddlmZmZmZdd lmZdd lmZmZdd lmZmZmZmZdd l m!Z!dd l"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddgZ*GdddeZ+GdddeZ,GdddZ-GdddZ.GdddZ/GdddZ0dS)) annotations)DictOptional)LiteralN)batch_create_params)BodyOmitQueryHeadersNotGivenomit not_given) path_templatemaybe_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'P/home/kim/smarthome/.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_idr0Optional[Dict[str, str]] | Omitr1Headers | Noner2 Query | Noner3 Body | Noner4'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/batchesr5r7r9r0r1r2r3r4bodyoptionscast_to)_postrrBatchCreateParamsrr r&r5r7r9r0r1r2r3r4r'r'r(create2s* zBatches.createrBbatch_idrcC6|s td||jtd|dt||||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/{batch_id}rKrBrErF ValueError_getrrrr&rKr1r2r3r4r'r'r(retrievem zBatches.retrievercCs|jdt||||dtdS)!List your organization's batches.r@rBrQrTrrr&r1r2r3r4r'r'r(lists z Batches.listrcCrL)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 rN$/openai/v1/batches/{batch_id}/cancelrPrBrQrSrGrrrrUr'r'r(cancelrWzBatches.cancel)r!r")r!r+r5r6r7r8r9r6r0r:r1r;r2r<r3r=r4r>r!r rKr6r1r;r2r<r3r=r4r>r!r r1r;r2r<r3r=r4r>r!r rKr6r1r;r2r<r3r=r4r>r!r __name__ __module__ __qualname__rr)r-r rrJrVr[r_r'r'r'r(r4  A&c@r )-rr!AsyncBatchesWithRawResponsecCr#r$)rir%r'r'r(r)r*zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr#r,)rjr%r'r'r(r-r.z$AsyncBatches.with_streaming_responseNr/r5r6r7r8r9r0r:r1r;r2r<r3r=r4r>rc s>|jdt||||dtjIdHt||||dtdIdHS)r?r@rANrBrC)rGrrrHrrrIr'r'r(rJs *  zAsyncBatches.createrBrKrc>|s td||jtd|dt||||dtdIdHS)rMrNrOrPrBrQNrRrUr'r'r(rV  zAsyncBatches.retrievercs$|jdt||||dtdIdHS)rXr@rBrQNrYrZr'r'r(r[3s  zAsyncBatches.listrcrk)r\rNr]rPrBrQNr^rUr'r'r(r_FrlzAsyncBatches.cancel)r!ri)r!rjr`rarbrcrdr'r'r'r(rrhc@eZdZdddZdS) r"batchesrr!NonecC:||_t|j|_t|j|_t|j|_t|j|_dSN)_batchesrrJrVr[r_r&rnr'r'r(__init__i zBatchesWithRawResponse.__init__Nrnrr!rorerfrgrtr'r'r'r(r"hr"c@rm) rirnrr!rocCrprq)rrrrJrVr[r_rsr'r'r(rt{ruz$AsyncBatchesWithRawResponse.__init__Nrnrr!rorwr'r'r'r(rizrxric@rm) r+rnrr!rocCrprq)rrrrJrVr[r_rsr'r'r(rtruz%BatchesWithStreamingResponse.__init__Nrvrwr'r'r'r(r+rxr+c@rm) rjrnrr!rocCrprq)rrrrJrVr[r_rsr'r'r(rtruz*AsyncBatchesWithStreamingResponse.__init__Nryrwr'r'r'r(rjrxrj)1 __future__rtypingrrZtyping_extensionsrZhttpxtypesr_typesrr r r r r rZ_utilsrrrZ_compatrZ _resourcerr _responserrrrZ _base_clientrZtypes.batch_list_responserZtypes.batch_cancel_responserZtypes.batch_create_responserZtypes.batch_retrieve_responser__all__rrr"rir+rjr'r'r'r(s0   $      &&