a fH@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 @seZdZeddddZeddddZeddded d d d d dddddd ddZdddedd ddddddddZeedddeddddddddddd Z dddedd ddddddd!d"Z dS)#rBatchesWithRawResponsereturncCst|SN)rselfr#Q/var/www/ai-form-bot/venv/lib/python3.9/site-packages/openai/resources/batches.pywith_raw_responseszBatches.with_raw_responseBatchesWithStreamingResponsecCst|Sr )r&r!r#r#r$with_streaming_response#szBatches.with_streaming_responseNmetadata extra_headers extra_query extra_bodytimeoutLiteral['24h']DLiteral['/v1/chat/completions', '/v1/embeddings', '/v1/completions']str#Optional[Dict[str, str]] | NotGivenHeaders | None Query | None Body | None'float | httpx.Timeout | None | NotGivenr completion_windowendpoint input_file_idr)r*r+r,r-rc 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 /batchesr7r8r9r)r*r+r,r-bodyoptionscast_to)_postrr BatchCreateParamsrr r"r7r8r9r)r*r+r,r-r#r#r$create's. zBatches.creater=batch_idr*r+r,r-rcCs4|std||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=r@rA ValueError_getrrr"rGr*r+r,r-r#r#r$retrievefszBatches.retrieveafterlimitr*r+r,r-str | NotGivenint | NotGivenzSyncCursorPage[Batch]rRrSr*r+r,r-rc Cs2|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)pager@model) _get_api_listrrrrrBatchListParamsr"rRrSr*r+r,r-r#r#r$lists"! z Batches.listcCs6|std||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=rKrMrBrrrOr#r#r$cancels zBatches.cancel __name__ __module__ __qualname__rr%r'r rErPr^rbr#r#r#r$rs6 "E$:c @seZdZeddddZeddddZeddded d d d d dddddd ddZdddedd ddddddddZeedddeddddddddddd Z dddedd ddddddd!d"Z dS)#rAsyncBatchesWithRawResponsercCst|Sr )rgr!r#r#r$r%szAsyncBatches.with_raw_response!AsyncBatchesWithStreamingResponsecCst|Sr )rhr!r#r#r$r'sz$AsyncBatches.with_streaming_responseNr(r.r/r0r1r2r3r4r5rr6c s<|jdt||||dtjIdHt||||dtdIdHS)r:r;r<Nr=r>)rBrr rCrrrDr#r#r$rEs. zAsyncBatches.creater=rFcs:|std||jd|t||||dtdIdHS)rHrIrJr=rKNrLrOr#r#r$rP(szAsyncBatches.retrieverQrTrUz-AsyncPaginator[Batch, AsyncCursorPage[Batch]]rVc Cs2|jdttt||||t||dtjdtdSrW)r[rrrrrr\r]r#r#r$r^Is"! zAsyncBatches.listcs<|std||jd|dt||||dtdIdHS)r_rIrJr`r=rKNrarOr#r#r$rb}s zAsyncBatches.cancelrcr#r#r#r$rs6 "E$:c@seZdZdddddZdS)rrNonebatchesrcCsB||_t|j|_t|j|_t|j|_t|j|_dSr )_batchesrZto_raw_response_wrapperrErPr^rbr"rkr#r#r$__init__szBatchesWithRawResponse.__init__Nrdrerfrnr#r#r#r$rsrc@seZdZdddddZdS)rgrrirjcCsB||_t|j|_t|j|_t|j|_t|j|_dSr )rlrZasync_to_raw_response_wrapperrErPr^rbrmr#r#r$rnsz$AsyncBatchesWithRawResponse.__init__Nror#r#r#r$rgsrgc@seZdZdddddZdS)r&rrirjcCs:||_t|j|_t|j|_t|j|_t|j|_dSr )rlrrErPr^rbrmr#r#r$rnsz%BatchesWithStreamingResponse.__init__Nror#r#r#r$r&sr&c@seZdZdddddZdS)rhrrirjcCs:||_t|j|_t|j|_t|j|_t|j|_dSr )rlrrErPr^rbrmr#r#r$rnsz*AsyncBatchesWithStreamingResponse.__init__Nror#r#r#r$rhsrh)- __future__rtypingrrtyping_extensionsrZhttpxrtypesrr _typesr r r r rZ_utilsrr_compatrZ _resourcerr _responserrZ paginationrrZ types.batchrZ _base_clientrr__all__rrrrgr&rhr#r#r#r$s,     CC