qiShddlmZddlmZddlmZddlZddlmZddl m Z m Z ddl m Z mZmZmZmZmZmZdd lmZmZdd lmZdd lmZmZdd lmZmZdd lm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddgZ)GddeZ*GddeZ+GddZ,GddZ-GddZ.GddZ/y)) annotations)Optional)LiteralN)_legacy_response)batch_list_paramsbatch_create_params)BodyOmitQueryHeadersNotGivenomit not_given)maybe_transformasync_maybe_transform)cached_property)SyncAPIResourceAsyncAPIResource)to_streamed_response_wrapper"async_to_streamed_response_wrapper)SyncCursorPageAsyncCursorPage)Batch)AsyncPaginatormake_request_options)MetadataBatches AsyncBatchesceZdZdZed dZed dZeeddded ddZ ddded ddZ eeddded  dd Z ddded dd Z y)r;Create large batches of API requests to run asynchronously.ct|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/openai/openai-python#accessing-raw-response-data-eg-headers )BatchesWithRawResponseselfs J/opt/pipecat/venv/lib/python3.12/site-packages/openai/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/openai/openai-python#with_streaming_response )BatchesWithStreamingResponser%s r'with_streaming_responsezBatches.with_streaming_response&s,D11r)Nmetadataoutput_expires_after extra_headers extra_query extra_bodytimeoutc |jdt|||||dtjt |||| t S) 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/responses`, `/v1/chat/completions`, `/v1/embeddings`, `/v1/completions`, `/v1/moderations`, `/v1/images/generations`, and `/v1/images/edits` 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 200 MB in size. metadata: Set of 16 key-value pairs that can be attached to an object. This can be useful for storing additional information about the object in a structured format, and querying for objects via API or the dashboard. Keys are strings with a maximum length of 64 characters. Values are strings with a maximum length of 512 characters. output_expires_after: The expiration policy for the output and/or error file that are generated for a 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 /batchescompletion_windowendpoint input_file_idr/r0r1r2r3r4bodyoptionscast_to)_postrr BatchCreateParamsrr r&r9r:r;r/r0r1r2r3r4s r'createzBatches.create/s_@zz  ): (%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 /batches/r<r?r@ ValueError_getrrr&batch_idr1r2r3r4s r'retrievezBatches.retrievesT.VW_Vbcd dyyz "(+Q[el   r)afterlimitr1r2r3r4c|jdttt||||t ||dt j tSa,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 r7)rQrR)r1r2r3r4query)pager?model) _get_api_listrrrrrBatchListParamsr&rQrRr1r2r3r4s r'listz Batches.lists_B!! &(+'%%!&!&&55  !"  r)c t|std||jd|dt||||tS) 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 rGrH/cancelr<rIrKrArrrMs r'cancelzBatches.cancelsV4VW_Vbcd dzzz )(+Q[el   r))returnr$)rar,r9zLiteral['24h']r:zLiteral['/v1/responses', '/v1/chat/completions', '/v1/embeddings', '/v1/completions', '/v1/moderations', '/v1/images/generations', '/v1/images/edits']r;strr/zOptional[Metadata] | Omitr0z-batch_create_params.OutputExpiresAfter | Omitr1Headers | Noner2 Query | Noner3 Body | Noner4'float | httpx.Timeout | None | NotGivenrar rNrcr1rdr2rer3rfr4rgrar)rQ str | OmitrR int | Omitr1rdr2rer3rfr4rgrazSyncCursorPage[Batch] __name__ __module__ __qualname____doc__rr(r-rrrDrOr[r`r)r'rrsE,,22,/3NR)-$("&;D+P *P  P P ,P LP $&%P &"'P ( )P *9+P , -P p)-$("&;D  &  "   9   H! )-$("&;D2 2  2 &2 "2  2 92  2 t)-$("&;D" " & " ""  " 9"  " r)ceZdZdZed dZed dZeeddded ddZ ddded ddZ eeddded  dd Z ddded dd Z y)rr!ct|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)r6r7r8Nr<r=)rArr rBrrrCs r'rDzAsyncBatches.createsx@ZZ ,): (%2 (,@ $55  )+Q[el      s!0AA AAAAr<c K|std||jd|t||||td{S7w)rFrGrHr<rINrJrMs r'rOzAsyncBatches.retrievecsb.VW_Vbcd dYYz "(+Q[el     s> NN  )@@     %<< LL  '>> NN  r)NrrraNonerlrmrnrrpr)r'r$r$ r)r$ceZdZddZy)rsc2||_tj|j|_tj|j|_tj|j |_tj|j |_yr|)r}rasync_to_raw_response_wrapperrDrOr[r`rs r'rz$AsyncBatchesWithRawResponse.__init__su &DD NN  )FF     %BB LL  'DD NN  r)Nrrrarrrpr)r'rsrsrr)rsceZdZddZy)r,c||_t|j|_t|j|_t|j|_t|j |_yr|)r}rrDrOr[r`rs r'rz%BatchesWithStreamingResponse.__init__s] 2 NN  5     1 LL  3 NN  r)Nrrrpr)r'r,r,rr)r,ceZdZddZy)ruc||_t|j|_t|j|_t|j|_t|j |_yr|)r}rrDrOr[r`rs r'rz*AsyncBatchesWithStreamingResponse.__init__s] 8 NN  ;     7 LL  9 NN  r)Nrrrpr)r'rururr)ru)0 __future__rtypingrtyping_extensionsrhttpxrtypesrr _typesr r r r rrr_utilsrr_compatr _resourcerr _responserr paginationrr types.batchr _base_clientrrtypes.shared_params.metadatar__all__rrr$rsr,rurpr)r'rs#% :JJJ;%9X8?3 n %_ o_ D_ #_ D  $  $  $  r)