gddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddlm1Z1ejdejfdZ4GddZ5Gd d!Z6y)"N)SyncClientWrapper)RequestOptions)ListEmbedJobResponse)construct_type)BadRequestError)UnauthorizedError)ForbiddenError) NotFoundError)UnprocessableEntityError)TooManyRequestsError)InvalidTokenError)ClientClosedRequestError)InternalServerError)NotImplementedError)ServiceUnavailableError)GatewayTimeoutError)JSONDecodeError)ApiError)EmbedInputType) EmbeddingType)CreateEmbedJobRequestTruncate)CreateEmbedJobResponse)EmbedJob)jsonable_encoder)AsyncClientWrapper.cfeZdZdefdZdddej edefdZ e e e ddd e d e d e d ej e d ej ejedej edej edefdZddde dej edefdZddde dej eddfdZy)EmbedJobsClientclient_wrapperc||_yN_client_wrapperselfr s M/var/www/openai/venv/lib/python3.12/site-packages/cohere/embed_jobs/client.py__init__zEmbedJobsClient.__init__# -Nrequest_optionsr,returnc |jjjdd|} d|jcxkrdkr:nn7t j t tt |jS|jdk(rvtt j tjtjttjtj|j|jdk(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvt!t j tjtjttjtj|j|jd k(rvt#t j tjtjttjtj|j|jdk(rvt%t j tjtjttjtj|j|jdk(rvt't j tjtjttjtj|j|jdk(rvt)t j tjtjttjtj|j|jdk(rvt+t j tjtjttjtj|j|jdk(rvt-t j tjtjttjtj|j|j}t1|j|#t.$r"t1|j|j2wxYw)a! The list embed job endpoint allows users to view all embed jobs history for that specific user. Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- ListEmbedJobResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.embed_jobs.list() v1/embed-jobsGETmethodr,,type_object_ status_codebodyr$ httpx_clientrequestrEtypingcastrrjsonrOptionalAnyr r r r r rrrrrrrrtextr&r, _response_response_jsons r'listzEmbedJobsClient.list&s0((55== +> C Si++1c1{{("2 ) 0$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NN Sy'<'<9>>R R SsA[9Y [+\nameembedding_typestruncater,model dataset_id input_typerUrVrWc 4|jjjdd||||||dddi|t} d|jcxkrdkr:nn7t j ttt|j S|jd k(rvtt j t jt jtt jt j|j |jd k(rvtt j t jt jtt jt j|j |jd k(rvtt j t jt jtt jt j|j |jd k(rvtt j t jt jtt jt j|j |jdk(rvt!t j t jt jtt jt j|j |jdk(rvt#t j t jt jtt jt j|j |jdk(rvt%t j t jt jtt jt j|j |jdk(rvt't j t jt jtt jt j|j |jdk(rvt)t j t jt jtt jt j|j |jdk(rvt+t j t jt jtt jt j|j |jdk(rvt-t j t jt jtt jt j|j |jdk(rvt/t j t jt jtt jt j|j |j} t3|j| #t0$r"t3|j|j4wxYw)az This API launches an async Embed job for a [Dataset](https://docs.cohere.com/docs/datasets) of type `embed-input`. The result of a completed embed job is new Dataset of type `embed-output`, which contains the original text entries and the corresponding embeddings. Parameters ---------- model : str ID of the embedding model. Available models and corresponding embedding dimensions: - `embed-english-v3.0` : 1024 - `embed-multilingual-v3.0` : 1024 - `embed-english-light-v3.0` : 384 - `embed-multilingual-light-v3.0` : 384 dataset_id : str ID of a [Dataset](https://docs.cohere.com/docs/datasets). The Dataset must be of type `embed-input` and must have a validation status `Validated` input_type : EmbedInputType name : typing.Optional[str] The name of the embed job. embedding_types : typing.Optional[typing.Sequence[EmbeddingType]] Specifies the types of embeddings you want to get back. Not required and default is None, which returns the Embed Floats response type. Can be one or more of the following types. * `"float"`: Use this when you want to get back the default float embeddings. Valid for all models. * `"int8"`: Use this when you want to get back signed int8 embeddings. Valid for only v3 models. * `"uint8"`: Use this when you want to get back unsigned int8 embeddings. Valid for only v3 models. * `"binary"`: Use this when you want to get back signed binary embeddings. Valid for only v3 models. * `"ubinary"`: Use this when you want to get back unsigned binary embeddings. Valid for only v3 models. truncate : typing.Optional[CreateEmbedJobRequestTruncate] One of `START|END` to specify how the API will handle inputs longer than the maximum token length. Passing `START` will discard the start of the input. `END` will discard the end of the input. In both cases, input is discarded until the remaining input is exactly the maximum input token length for the model. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- CreateEmbedJobResponse OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.embed_jobs.create( model="model", dataset_id="dataset_id", input_type="search_document", ) r/POSTrXrYrZrUrVrW content-typeapplication/jsonr2rLheadersr,omitr3r4r5r8r9r:r;r<r=r>r?r@rArBrCrDr$rHrIOMITrErJrKrrrLrrMrNr r r r r rrrrrrrrrO r&rXrYrZrUrVrWr,rQrRs r'createzEmbedJobsClient.creates,P((55== ((#2$  2,> "C Si++1c1{{*"4 ) 0$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NN Sy'<'<9>>R R SsA[, Y [,,+\idc ,|jjjdt|d|} d|jcxkrdkr:nn7t j ttt|jS|jdk(rvtt j t jt jtt jt j|j|jdk(rvtt j t jt jtt jt j|j|jd k(rvtt j t jt jtt jt j|j|jd k(rvtt j t jt jtt jt j|j|jd k(rvt!t j t jt jtt jt j|j|jd k(rvt#t j t jt jtt jt j|j|jd k(rvt%t j t jt jtt jt j|j|jdk(rvt't j t jt jtt jt j|j|jdk(rvt)t j t jt jtt jt j|j|jdk(rvt+t j t jt jtt jt j|j|jdk(rvt-t j t jt jtt jt j|j|jdk(rvt/t j t jt jtt jt j|j|j}t3|j|#t0$r"t3|j|j4wxYw)aa This API retrieves the details about an embed job started by the same user. Parameters ---------- id : str The ID of the embed job to retrieve. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- EmbedJob OK Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.embed_jobs.get( id="id", ) v1/embed-jobs/r0r1r3r4r5r8r9r:r;r<r=r>r?r@rArBrCrDr$rHrIrrErJrKrrrLrrMrNr r r r r rrrrrrrrrOr&rgr,rQrRs r'getzEmbedJobsClient.gets:((55==-b12 3+> C Si++1c1{{"& ) 0$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NN Sy'<'<9>>R R SsA[(Y [((+\c  |jjjdt|dd|} d|jcxkrdkry|jdk(rvt t jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jdk(rvt!t jt jt jtt jt j|j |jdk(rvt#t jt jt jtt jt j|j |jdk(rvt%t jt jt jtt jt j|j |jdk(rvt't jt jt jtt jt j|j |jdk(rvt)t jt jt jtt jt j|j |jdk(rvt+t jt jt jtt jt j|j |jdk(rvt-t jt jt jtt jt j|j |j}t1|j|#t.$r"t1|j|j2wxYw)a$ This API allows users to cancel an active embed job. Once invoked, the embedding process will be terminated, and users will be charged for the embeddings processed up to the cancellation point. It's important to note that partial results will not be available to users after cancellation. Parameters ---------- id : str The ID of the embed job to cancel. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- from cohere import Client client = Client( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) client.embed_jobs.cancel( id="id", ) ri/cancelr\r1r3r4Nr8r5r9r:r;r<r=r>r?r@rArBrCrDr$rHrIrrErrJrKrMrNrrLr r r r r rrrrrrrrrOrks r'cancelzEmbedJobsClient.cancelPs8((55==-b12' :+> } Si++1c12$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NN Sy'<'<9>>R R SsZ1 Y Z11+[)__name__ __module__ __qualname__rr(rJrMrrrSrdstrrSequencerrrrfrrlrpr*r'rr"sE.*;.JNaOv~'FaORfaOR&*KOCG;?]O]O ]O # ]O ooc" ]O )GH]O//"?@]O 8]O ]O~RVfOcfOv~/NfOZbfOPUY_O_O&//.2Q_O]a_Or*rcfeZdZdefdZdddej edefdZ e e e ddd e d e d e d ej e d ej ejedej edej edefdZddde dej edefdZddde dej eddfdZy)AsyncEmbedJobsClientr c||_yr"r#r%s r'r(zAsyncEmbedJobsClient.__init__r)r*Nr+r,r-c 2K|jjjdd|d{} d|jcxkrdkr:nn7t j t tt |jS|jdk(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvtt j tjtjttjtj|j|jd k(rvt!t j tjtjttjtj|j|jdk(rvt#t j tjtjttjtj|j|jdk(rvt%t j tjtjttjtj|j|jdk(rvt't j tjtjttjtj|j|jdk(rvt)t j tjtjttjtj|j|jdk(rvt+t j tjtjttjtj|j|jdk(rvt-t j tjtjttjtj|j|j}t1|j|7#t.$r"t1|j|j2wxYww)a The list embed job endpoint allows users to view all embed jobs history for that specific user. Parameters ---------- request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- ListEmbedJobResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.embed_jobs.list() asyncio.run(main()) r/r0r1Nr3r4r5r8r9r:r;r<r=r>r?r@rArBrCrDrGrPs r'rSzAsyncEmbedJobsClient.lists@..;;CC +D  C Si++1c1{{("2 ) 0$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NNS N Sy'<'<9>>R R Ss4,\[&\A[)\Y [)\)+\\rTrXrYrZrUrVrWc RK|jjjdd||||||dddi|td{} d|jcxkrd kr:nn7t j ttt|j S|jd k(rvtt j t jt jtt jt j|j |jd k(rvtt j t jt jtt jt j|j |jd k(rvtt j t jt jtt jt j|j |jdk(rvtt j t jt jtt jt j|j |jdk(rvt!t j t jt jtt jt j|j |jdk(rvt#t j t jt jtt jt j|j |jdk(rvt%t j t jt jtt jt j|j |jdk(rvt't j t jt jtt jt j|j |jdk(rvt)t j t jt jtt jt j|j |jdk(rvt+t j t jt jtt jt j|j |jdk(rvt-t j t jt jtt jt j|j |jdk(rvt/t j t jt jtt jt j|j |j} t3|j| 7#t0$r"t3|j|j4wxYww)a This API launches an async Embed job for a [Dataset](https://docs.cohere.com/docs/datasets) of type `embed-input`. The result of a completed embed job is new Dataset of type `embed-output`, which contains the original text entries and the corresponding embeddings. Parameters ---------- model : str ID of the embedding model. Available models and corresponding embedding dimensions: - `embed-english-v3.0` : 1024 - `embed-multilingual-v3.0` : 1024 - `embed-english-light-v3.0` : 384 - `embed-multilingual-light-v3.0` : 384 dataset_id : str ID of a [Dataset](https://docs.cohere.com/docs/datasets). The Dataset must be of type `embed-input` and must have a validation status `Validated` input_type : EmbedInputType name : typing.Optional[str] The name of the embed job. embedding_types : typing.Optional[typing.Sequence[EmbeddingType]] Specifies the types of embeddings you want to get back. Not required and default is None, which returns the Embed Floats response type. Can be one or more of the following types. * `"float"`: Use this when you want to get back the default float embeddings. Valid for all models. * `"int8"`: Use this when you want to get back signed int8 embeddings. Valid for only v3 models. * `"uint8"`: Use this when you want to get back unsigned int8 embeddings. Valid for only v3 models. * `"binary"`: Use this when you want to get back signed binary embeddings. Valid for only v3 models. * `"ubinary"`: Use this when you want to get back unsigned binary embeddings. Valid for only v3 models. truncate : typing.Optional[CreateEmbedJobRequestTruncate] One of `START|END` to specify how the API will handle inputs longer than the maximum token length. Passing `START` will discard the start of the input. `END` will discard the end of the input. In both cases, input is discarded until the remaining input is exactly the maximum input token length for the model. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- CreateEmbedJobResponse OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.embed_jobs.create( model="model", dataset_id="dataset_id", input_type="search_document", ) asyncio.run(main()) r/r\r]r^r_r`Nr3r4r5r8r9r:r;r<r=r>r?r@rArBrCrDrcres r'rfzAsyncEmbedJobsClient.creates<`..;;CC ((#2$  2,D  "C Si++1c1{{*"4 ) 0$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NNk f Sy'<'<9>>R R Ss5<\'[6\'A[9\'Y [9\'9+\$$\'rgc JK|jjjdt|d|d{} d|jcxkrdkr:nn7t j ttt|jS|jdk(rvtt j t jt jtt jt j|j|jd k(rvtt j t jt jtt jt j|j|jd k(rvtt j t jt jtt jt j|j|jd k(rvtt j t jt jtt jt j|j|jd k(rvt!t j t jt jtt jt j|j|jd k(rvt#t j t jt jtt jt j|j|jdk(rvt%t j t jt jtt jt j|j|jdk(rvt't j t jt jtt jt j|j|jdk(rvt)t j t jt jtt jt j|j|jdk(rvt+t j t jt jtt jt j|j|jdk(rvt-t j t jt jtt jt j|j|jdk(rvt/t j t jt jtt jt j|j|j}t3|j|7#t0$r"t3|j|j4wxYww)a This API retrieves the details about an embed job started by the same user. Parameters ---------- id : str The ID of the embed job to retrieve. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- EmbedJob OK Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.embed_jobs.get( id="id", ) asyncio.run(main()) rir0r1Nr3r4r5r8r9r:r;r<r=r>r?r@rArBrCrDrjrks r'rlzAsyncEmbedJobsClient.gets!J..;;CC-b12 3+D  C Si++1c1{{"& ) 0$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NNS N Sy'<'<9>>R R Ss58\#[2\#A[5\#Y [5\#5+\  \#c K|jjjdt|dd|d{} d|jcxkrdkry|jdk(rvt t jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jd k(rvtt jt jt jtt jt j|j |jdk(rvt!t jt jt jtt jt j|j |jdk(rvt#t jt jt jtt jt j|j |jdk(rvt%t jt jt jtt jt j|j |jdk(rvt't jt jt jtt jt j|j |jdk(rvt)t jt jt jtt jt j|j |jdk(rvt+t jt jt jtt jt j|j |jdk(rvt-t jt jt jtt jt j|j |j}t1|j|7#t.$r"t1|j|j2wxYww)a This API allows users to cancel an active embed job. Once invoked, the embedding process will be terminated, and users will be charged for the embeddings processed up to the cancellation point. It's important to note that partial results will not be available to users after cancellation. Parameters ---------- id : str The ID of the embed job to cancel. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- None Examples -------- import asyncio from cohere import AsyncClient client = AsyncClient( client_name="YOUR_CLIENT_NAME", token="YOUR_TOKEN", ) async def main() -> None: await client.embed_jobs.cancel( id="id", ) asyncio.run(main()) rirnr\r1Nr3r4r8r5r9r:r;r<r=r>r?r@rArBrCrDrorks r'rpzAsyncEmbedJobsClient.cancel8sH..;;CC-b12' :+D  } Si++1c12$$+%KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+$KK 3&"(//&**"=$-NN$4$$+#KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+*KK 3&"(//&**"=$-NN$4$$+'KK 3&"(//&**"=$-NN$4$$+.KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4$$+-KK 3&"(//&**"=$-NN$4$$+)KK 3&"(//&**"=$-NN$4'^^-N9#8#8~NNG B Sy'<'<9>>R R Ss49[,Z;[,Z>[,Y Z>$[,>+[))[,)rqrrrsrr(rJrMrrrSrdrtrrurrrrfrrlrprvr*r'rxrxsE.*<.PTiOV__^-LiOXliOb&*KOCG;?eOeO eO # eO ooc" eO )GHeO//"?@eO 8eO eONX\nOCnOV__^5TnO`hnO`[_gOsgO8WgOcggOr*rx)7rJcore.client_wrapperrcore.request_optionsrtypes.list_embed_job_responsercore.unchecked_base_modelrerrors.bad_request_errorrerrors.unauthorized_errorr errors.forbidden_errorr errors.not_found_errorr !errors.unprocessable_entity_errorr errors.too_many_requests_errorr errors.invalid_token_errorr"errors.client_closed_request_errorrerrors.internal_server_errorrerrors.not_implemented_errorr errors.service_unavailable_errorrerrors.gateway_timeout_errorr json.decoderrcore.api_errorrtypes.embed_input_typertypes.embedding_typer'types.create_embed_job_request_truncatertypes.create_embed_job_responsertypes.embed_jobrcore.jsonable_encoderrrrKrNrdrrxrvr*r'rs31@66932HA:I>>F>(%30RD&44v{{6::s#M OM O`m Om Or*