o i)@sddlZddlmZmZddlmZddlmZmZddl m Z ddl m Z dd l mZdd lmZdd lmZeejd ZGd ddZGdddZdS)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)AsyncRawBatchesClientRawBatchesClient)Batch)CancelBatchResponse)CreateBatchResponse)GetBatchResponse)ListBatchesResponse.c @eZdZdefddZedefddZdddddd ej e d ej e d ej e d ej e de f d dZddded ej e defddZddde d ej e defddZddde d ej e defddZdS) BatchesClientclient_wrappercCt|d|_dSN)r)r _raw_clientselfrr[/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/batches/client.py__init__zBatchesClient.__init__returncC|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawBatchesClient rrrrrwith_raw_response zBatchesClient.with_raw_responseN page_size page_tokenorder_byrequest_optionsr!r"r#r$cCs|jj||||d}|jS)a List the batches for the current user Parameters ---------- page_size : typing.Optional[int] The maximum number of batches to return. The service may return fewer than this value. If unspecified, at most 50 batches will be returned. The maximum value is 1000; values above 1000 will be coerced to 1000. page_token : typing.Optional[str] A page token, received from a previous `ListBatches` call. Provide this to retrieve the subsequent page. order_by : typing.Optional[str] Batches can be ordered by creation time or last updated time. Use `created_at` for creation time or `updated_at` for last updated time. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- ListBatchesResponse A successful response. Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.batches.list() r rlistdatarr!r"r#r$ _responserrrr&!s-zBatchesClient.listr$requestcC|jj||d}|jS)a Creates and executes a batch from an uploaded dataset of requests Parameters ---------- request : Batch request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- CreateBatchResponse A successful response. Examples -------- from cohere import Client from cohere.batches import Batch client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.batches.create( request=Batch( name="name", input_dataset_id="input_dataset_id", model="model", ), ) r+r$rcreater'rr+r$r)rrrr/Ss!zBatchesClient.createidcCr,)a. Retrieves a batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetBatchResponse A successful response. Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.batches.retrieve( id="id", ) r*rretriever'rr1r$r)rrrr3wzBatchesClient.retrievecCr,)a: Cancels an in-progress batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- CancelBatchResponse A successful response. Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.batches.cancel( id="id", ) r*rcancelr'r4rrrr7r5zBatchesClient.cancel)__name__ __module__ __qualname__rrpropertyrrtypingOptionalintstrrr r&r r r/r r3r r7rrrrrs,  "2"$& rc @r)AsyncBatchesClientrcCrr)rrrrrrrrzAsyncBatchesClient.__init__rcCr)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawBatchesClient rrrrrrrz$AsyncBatchesClient.with_raw_responseNr r!r"r#r$cs"|jj||||dIdH}|jS)a List the batches for the current user Parameters ---------- page_size : typing.Optional[int] The maximum number of batches to return. The service may return fewer than this value. If unspecified, at most 50 batches will be returned. The maximum value is 1000; values above 1000 will be coerced to 1000. page_token : typing.Optional[str] A page token, received from a previous `ListBatches` call. Provide this to retrieve the subsequent page. order_by : typing.Optional[str] Batches can be ordered by creation time or last updated time. Use `created_at` for creation time or `updated_at` for last updated time. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- ListBatchesResponse A successful response. Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.batches.list() asyncio.run(main()) r Nr%r(rrrr&s 5 zAsyncBatchesClient.listr*r+c|jj||dIdH}|jS)a Creates and executes a batch from an uploaded dataset of requests Parameters ---------- request : Batch request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- CreateBatchResponse A successful response. Examples -------- import asyncio from cohere import AsyncClient from cohere.batches import Batch client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.batches.create( request=Batch( name="name", input_dataset_id="input_dataset_id", model="model", ), ) asyncio.run(main()) r-Nr.r0rrrr/s+zAsyncBatchesClient.creater1crA)a Retrieves a batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- GetBatchResponse A successful response. Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.batches.retrieve( id="id", ) asyncio.run(main()) r*Nr2r4rrrr3/%zAsyncBatchesClient.retrievecrA)a Cancels an in-progress batch Parameters ---------- id : str The batch ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- CancelBatchResponse A successful response. Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.batches.cancel( id="id", ) asyncio.run(main()) r*Nr6r4rrrr7WrBzAsyncBatchesClient.cancel)r8r9r:rrr;rrr<r=r>r?rr r&r r r/r r3r r7rrrrr@s:  ; ".&(r@)r<core.client_wrapperrrcore.request_optionsr raw_clientrr types.batchr types.cancel_batch_responser types.create_batch_responser types.get_batch_responser types.list_batches_responser castAnyOMITrr@rrrrs      '