o i|@s8ddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.GdddZ/GdddZ0dS)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)construct_type)BadRequestError)ClientClosedRequestError)ForbiddenError)GatewayTimeoutError)InternalServerError)InvalidTokenError) NotFoundError)NotImplementedError)ServiceUnavailableError)TooManyRequestsError)UnauthorizedError)UnprocessableEntityError)CompatibleEndpoint)GetModelResponse)ListModelsResponsec@eZdZdefddZdddedejede e fd d Z dddddd d eje d ejedeje dejedejede ef ddZdS)RawModelsClientclient_wrappercC ||_dSN_client_wrapperselfrr$^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/models/raw_client.py__init__ zRawModelsClient.__init__Nrequest_optionsmodelr)returnc Cs|jjjdt|d|d}zd|jkrdkr1nntttt| d}t ||dWS|jdkrQt t |j ttjtjttjtj| dd |jd krqtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkr3tt |j ttjtjttjtj| dd |jdkrTtt |j ttjtjttjtj| dd |jdkrutt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)a= Returns the details of a model, provided its name. Parameters ---------- model : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetModelResponse] OK v1/models/GETmethodr),type_object_responsedataheadersbody status_coder:r;)r! httpx_clientrequestr rHtypingcastrr jsonrr dictr:OptionalAnyrrrrrrr rrrrrrtextr#r*r) _response_data_response_jsonr$r$r%get!s0                                      zRawModelsClient.get page_size page_tokenendpoint default_onlyr)rXrYrZr[c Cs|jjjdd||||d|d}zd|jkrdkr2nntttt|d}t ||dWS|jd krRt t |j ttj tjttj tj|dd |jd krrtt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkr4tt |j ttj tjttj tj|dd |jdkrUtt |j ttj tjttj tj|dd |jdkrvtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |}Wntyt|jt |j |jdwt|jt |j |d)a Returns a list of models available for use. Parameters ---------- page_size : typing.Optional[float] Maximum number of models to include in a page Defaults to `20`, min value of `1`, max value of `1000`. page_token : typing.Optional[str] Page token provided in the `next_page_token` field of a previous response. endpoint : typing.Optional[CompatibleEndpoint] When provided, filters the list of models to only those that are compatible with the specified endpoint. default_only : typing.Optional[bool] When provided, filters the list of models to only the default model to the endpoint. This parameter is only valid when `endpoint` is provided. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListModelsResponse] OK v1/modelsr-rXrYrZr[r/paramsr)r0r1r2r5r8r9r<r=r>r?r@rArBrCrDrErFrG)r!rIrJrHrKrLrr rMrr rNr:rOrPrrrrrrr rrrrrrrQ r#rXrYrZr[r)rSrTrUr$r$r%lists:#                                      zRawModelsClient.list)__name__ __module__ __qualname__rr&strrKrOr rrrVfloatrboolrrar$r$r$r%r: .rc@r)AsyncRawModelsClientrcCrrr r"r$r$r%r&r'zAsyncRawModelsClient.__init__Nr(r*r)r+c s|jjjdt|d|dIdH}zd|jkr dkr5nntttt| d}t ||dWS|jd krUt t |j ttjtjttjtj| dd |jd krutt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkr7tt |j ttjtjttjtj| dd |jdkrXtt |j ttjtjttjtj| dd |jdkrytt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)aB Returns the details of a model, provided its name. Parameters ---------- model : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetModelResponse] OK r,r-r.Nr0r1r2r5r8r9r<r=r>r?r@rArBrCrDrErFrG)r!rIrJr rHrKrLrr rMrr rNr:rOrPrrrrrrr rrrrrrrQrRr$r$r%rVs2                                       zAsyncRawModelsClient.getrWrXrYrZr[c s|jjjdd||||d|dIdH}zd|jkr!dkr6nntttt|d}t ||d WS|jd krVt t |j ttj tjttj tj|dd |jd krvtt |j ttj tjttj tj|dd |jd krtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkr8tt |j ttj tjttj tj|dd |jdkrYtt |j ttj tjttj tj|dd |jdkrztt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |jdkrtt |j ttj tjttj tj|dd |}Wntyt|jt |j |jdwt|jt |j |d)a Returns a list of models available for use. Parameters ---------- page_size : typing.Optional[float] Maximum number of models to include in a page Defaults to `20`, min value of `1`, max value of `1000`. page_token : typing.Optional[str] Page token provided in the `next_page_token` field of a previous response. endpoint : typing.Optional[CompatibleEndpoint] When provided, filters the list of models to only those that are compatible with the specified endpoint. default_only : typing.Optional[bool] When provided, filters the list of models to only the default model to the endpoint. This parameter is only valid when `endpoint` is provided. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListModelsResponse] OK r\r-r]r^Nr0r1r2r5r8r9r<r=r>r?r@rArBrCrDrErFrG)r!rIrJrHrKrLrr rMrr rNr:rOrPrrrrrrr rrrrrrrQr`r$r$r%ra;s<#                                       zAsyncRawModelsClient.list)rbrcrdrr&rerKrOr rrrVrfrrgrrar$r$r$r%rirhri)1rK json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.unchecked_base_modelr errors.bad_request_errorr "errors.client_closed_request_errorr errors.forbidden_errorrerrors.gateway_timeout_errorrerrors.internal_server_errorrerrors.invalid_token_errorrerrors.not_found_errorrerrors.not_implemented_errorr errors.service_unavailable_errorrerrors.too_many_requests_errorrerrors.unauthorized_errorr!errors.unprocessable_entity_errorrtypes.compatible_endpointrtypes.get_model_responsertypes.list_models_responserrrir$r$r$r%s6                    r