o +ifH@s4ddlmZddlmZmZddlmZddlZddlm Z ddl m Z m Z ddl mZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lmZm Z ddl!m"Z"ddl#m$Z$m%Z%ddgZ&GdddeZ'GdddeZ(GdddZ)GdddZ*GdddZ+GdddZ,dS)) 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 AsyncBatchesc@eZdZed+ddZed,ddZeddded d-ddZdddedd.dd Zeeddded!d/d'd(Z dddedd.d)d*Z dS)0rreturnBatchesWithRawResponsecCt|SN)rselfr$Q/var/www/html/corbot_env/lib/python3.10/site-packages/openai/resources/batches.pywith_raw_responsezBatches.with_raw_responseBatchesWithStreamingResponsecCr r!)r(r"r$r$r%with_streaming_response#r'zBatches.with_streaming_responseNmetadata extra_headers extra_query extra_bodytimeoutcompletion_windowLiteral['24h']endpointDLiteral['/v1/chat/completions', '/v1/embeddings', '/v1/completions'] input_file_idstrr+#Optional[Dict[str, str]] | NotGivenr,Headers | Noner- Query | Noner. Body | Noner/'float | httpx.Timeout | None | NotGivenrc Cs0|jdt||||dtjt||||dtdS) 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 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 /batchesr0r2r4r+r,r-r.r/bodyoptionscast_to)_postrr BatchCreateParamsrr r#r0r2r4r+r,r-r.r/r$r$r%create's. zBatches.creater>batch_idcCs4|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 /batches/r>rArB ValueError_getrrr#rGr,r-r.r/r$r$r%retrievefszBatches.retrieveafterlimitr,r-r.r/rRstr | NotGivenrSint | NotGivenSyncCursorPage[Batch]c C2|jdttt||||t||dtjdtdSa,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 r<)rRrS)r,r-r.r/query)pagerAmodel) _get_api_listrrrrrBatchListParamsr#rRrSr,r-r.r/r$r$r%list"! z Batches.listcCs6|s td||jd|dt||||dtdS) 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 rIrJ/cancelr>rKrMrCrrrOr$r$r%cancels zBatches.cancel)rr)rr(r0r1r2r3r4r5r+r6r,r7r-r8r.r9r/r:rr rGr5r,r7r-r8r.r9r/r:rr)rRrTrSrUr,r7r-r8r.r9r/r:rrV __name__ __module__ __qualname__rr&r)r rFrPr_rdr$r$r$r%r8   E$:c@r)0rrAsyncBatchesWithRawResponsecCr r!)rlr"r$r$r%r&r'zAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCr r!)rmr"r$r$r%r)r'z$AsyncBatches.with_streaming_responseNr*r0r1r2r3r4r5r+r6r,r7r-r8r.r9r/r:rc s>|jdt||||dtjIdHt||||dtdIdHS)r;r<r=Nr>r?)rCrr rDrrrEr$r$r%rFs .  zAsyncBatches.creater>rGcs<|s td||jd|t||||dtdIdHS)rHrIrJr>rKNrLrOr$r$r%rP(s zAsyncBatches.retrieverQrRrTrSrU-AsyncPaginator[Batch, AsyncCursorPage[Batch]]c CrWrX)r\rrrrrr]r^r$r$r%r_Ir`zAsyncBatches.listcs>|s td||jd|dt||||dtdIdHS)rarIrJrbr>rKNrcrOr$r$r%rd}s  zAsyncBatches.cancel)rrl)rrmrerf)rRrTrSrUr,r7r-r8r.r9r/r:rrnrgr$r$r$r%rrkc@eZdZdddZdS) rbatchesrrNonecCB||_t|j|_t|j|_t|j|_t|j|_dSr!)_batchesrto_raw_response_wrapperrFrPr_rdr#rpr$r$r%__init__ zBatchesWithRawResponse.__init__Nrprrrqrhrirjrvr$r$r$r%rrc@ro) rlrprrrqcCrrr!)rsrasync_to_raw_response_wrapperrFrPr_rdrur$r$r%rvrwz$AsyncBatchesWithRawResponse.__init__Nrprrrqryr$r$r$r%rlrzrlc@ro) r(rprrrqcC:||_t|j|_t|j|_t|j|_t|j|_dSr!)rsrrFrPr_rdrur$r$r%rv z%BatchesWithStreamingResponse.__init__Nrxryr$r$r$r%r(rzr(c@ro) rmrprrrqcCr}r!)rsrrFrPr_rdrur$r$r%rvr~z*AsyncBatchesWithStreamingResponse.__init__Nr|ryr$r$r$r%rmrzrm)- __future__rtypingrrtyping_extensionsrhttpxrtypesrr _typesr r r r r_utilsrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrr__all__rrrrlr(rmr$r$r$r%s.     CC