o iO@s^ddlZddlmZddlmZddlmZmZddlm 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.ddl/m0Z0ddl1m2Z2e3ej4dZ5GdddZ6Gdd d Z7dS)!N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)'convert_and_respect_annotation_metadata)construct_type)BadRequestError)ForbiddenError)InternalServerError) NotFoundError)ServiceUnavailableError)UnauthorizedError)CreateFinetunedModelResponse)DeleteFinetunedModelResponse)FinetunedModel)GetFinetunedModelResponse)ListEventsResponse)ListFinetunedModelsResponse)ListTrainingStepMetricsResponse)Settings)Status)UpdateFinetunedModelResponse.c@eZdZdefddZddddddejedejedejed eje d e e f d d Z dd de d eje d e efddZdd ded eje d e efddZdd ded eje d e efddZedddedededejed eje d e ef ddZddddddedejedejedejed eje d e ef ddZdddddedejedejed eje d e ef d d!ZdS)"RawFinetuningClientclient_wrappercC ||_dSN_client_wrapperselfr r'b/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/cohere/finetuning/raw_client.py__init__# zRawFinetuningClient.__init__N page_size page_tokenorder_byrequest_optionsr,r-r.r/returnc Cs*|jjjdd|||d|d}zd|jkrdkr0nntttt|d}t ||dWS|jd krPt t |j ttj tjttj tj|dd |jd krptt |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 |}Wnty t|jt |j |jdwt|jt |j |d)a Returns a list of fine-tuned models that the user has access to. Parameters ---------- page_size : typing.Optional[int] Maximum number of results to be returned by the server. If 0, defaults to 50. page_token : typing.Optional[str] Request a specific page of the list results. order_by : typing.Optional[str] Comma separated list of fields. For example: "created_at,name". The default sorting order is ascending. To specify descending order for a field, append " desc" to the field name. For example: "created_at desc,name". Supported sorting fields: - created_at (default) request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListFinetunedModelsResponse] A successful response. v1/finetuning/finetuned-modelsGETr,r-r.methodparamsr/,type_object_responsedataheadersbody status_coderArB)r$ httpx_clientrequestrItypingcastrr jsonrr dictrAOptionalAnyrrrrrrrtextr&r,r-r.r/ _response_data_response_jsonr'r'r(list_finetuned_models&s$                    z)RawFinetuningClient.list_finetuned_modelsr/rKc Cs4|jjjddt|tddddi|td}zd|jkr d kr5nntt t t | d }t ||d WS|jd krUt t|jttjtjt tjtj| d d |jdkrutt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d | }Wntyt|jt|j|jdwt|jt|j|d)a Creates a new fine-tuned model. The model will be trained on the dataset specified in the request body. The training process may take some time, and the model will be available once the training is complete. Parameters ---------- request : FinetunedModel request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateFinetunedModelResponse] A successful response. r1POSTwriter; annotation direction content-typeapplication/jsonr5rNrAr/omitr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rOMITrIrLrMrr rNrr rOrArPrQrrrrrrrrRr&rKr/rTrUrVr'r'r(create_finetuned_models                     z*RawFinetuningClient.create_finetuned_modelidc C*|jjjdt|d|d}zd|jkrdkr0nntttt| d}t ||dWS|jdkrPt t |j ttjtjttjtj| dd |jd krptt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wnty t|jt |j |jdwt|jt |j |d)ap Retrieve a fine-tuned model by its ID. Parameters ---------- id : str The fine-tuned model ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetFinetunedModelResponse] A successful response. v1/finetuning/finetuned-models/r2r5r/r7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRr&rer/rTrUrVr'r'r(get_finetuned_models                    z'RawFinetuningClient.get_finetuned_modelc Crf)a Deletes a fine-tuned model. The model will be removed from the system and will no longer be available for use. This operation is irreversible. Parameters ---------- id : str The fine-tuned model ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[DeleteFinetunedModelResponse] A successful response. rgDELETErhr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRrir'r'r(delete_finetuned_model{s                    z*RawFinetuningClient.delete_finetuned_modelstatusr/namesettingsrnc CsF|jjjdt|d|t|tdd|dddi|td}zd |jkr)d kr>nnt t t t | d }t ||d WS|jd kr^tt|jt tjtjt tjtj| d d|jdkr~tt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d| }Wntyt|jt|j|jdwt|jt|j|d)a Updates the fine-tuned model with the given ID. The model will be updated with the new settings and name provided in the request body. Parameters ---------- id : str FinetunedModel ID. name : str FinetunedModel name (e.g. `foobar`). settings : Settings FinetunedModel settings such as dataset, hyperparameters... status : typing.Optional[Status] Current stage in the life-cycle of the fine-tuned model. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[UpdateFinetunedModelResponse] A successful response. rgPATCHrZr[rorprnr^r_r`r7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr r rrbrIrLrMrr rNrr rOrArPrQrrrrrrrrR r&rerorprnr/rTrUrVr'r'r(update_finetuned_models"                    z*RawFinetuningClient.update_finetuned_modelfinetuned_model_idc Cs6|jjjdt|dd|||d|d}zd|jkr!dkr6nntttt| d}t ||d WS|jd krVt t |j ttjtjttjtj| dd |jd krvtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)a Returns a list of events that occurred during the life-cycle of the fine-tuned model. The events are ordered by creation time, with the most recent event first. The list can be paginated using `page_size` and `page_token` parameters. Parameters ---------- finetuned_model_id : str The parent fine-tuned model ID. page_size : typing.Optional[int] Maximum number of results to be returned by the server. If 0, defaults to 50. page_token : typing.Optional[str] Request a specific page of the list results. order_by : typing.Optional[str] Comma separated list of fields. For example: "created_at,name". The default sorting order is ascending. To specify descending order for a field, append " desc" to the field name. For example: "created_at desc,name". Supported sorting fields: - created_at (default) request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListEventsResponse] A successful response. rg/eventsr2r3r4r7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrR r&rur,r-r.r/rTrUrVr'r'r( list_eventshs*                    zRawFinetuningClient.list_eventsr,r-r/c Cs4|jjjdt|dd||d|d}zd|jkr dkr5nntttt| d}t ||d WS|jd krUt t |j ttjtjttjtj| dd |jd krutt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wntyt|jt |j |jdwt|jt |j |d)aJ Returns a list of metrics measured during the training of a fine-tuned model. The metrics are ordered by step number, with the most recent step first. The list can be paginated using `page_size` and `page_token` parameters. Parameters ---------- finetuned_model_id : str The parent fine-tuned model ID. page_size : typing.Optional[int] Maximum number of results to be returned by the server. If 0, defaults to 50. page_token : typing.Optional[str] Request a specific page of the list results. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[ListTrainingStepMetricsResponse] A successful response. rg/training-step-metricsr2r,r-r4r7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRr&rur,r-r/rTrUrVr'r'r(list_training_step_metricss!                    z.RawFinetuningClient.list_training_step_metrics)__name__ __module__ __qualname__rr)rLrPintstrr rrrWrrrdrrjrrlrbrrrrtrrxrr}r'r'r'r(r"  n j p   rc@r)"AsyncRawFinetuningClientr cCr!r"r#r%r'r'r(r)jr*z!AsyncRawFinetuningClient.__init__Nr+r,r-r.r/r0c s2|jjjdd|||d|dIdH}zd|jkrdkr4nntttt|d}t ||d WS|jd krTt t |j ttj tjttj tj|dd |jd krttt |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 |}Wnty t|jt |j |jdwt|jt |j |d)a Returns a list of fine-tuned models that the user has access to. Parameters ---------- page_size : typing.Optional[int] Maximum number of results to be returned by the server. If 0, defaults to 50. page_token : typing.Optional[str] Request a specific page of the list results. order_by : typing.Optional[str] Comma separated list of fields. For example: "created_at,name". The default sorting order is ascending. To specify descending order for a field, append " desc" to the field name. For example: "created_at desc,name". Supported sorting fields: - created_at (default) request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListFinetunedModelsResponse] A successful response. r1r2r3r4Nr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKrIrLrMrr rNrr rOrArPrQrrrrrrrrRrSr'r'r(rWms$                     z.AsyncRawFinetuningClient.list_finetuned_modelsrXrKc s<|jjjddt|tddddi|tdIdH}zd |jkr$d kr9nntt t t | d }t ||d WS|jd krYt t|jttjtjt tjtj| d d|jdkrytt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d|jdkrtt|jttjtjt tjtj| d d| }Wntyt|jt|j|jdwt|jt|j|d)a  Creates a new fine-tuned model. The model will be trained on the dataset specified in the request body. The training process may take some time, and the model will be available once the training is complete. Parameters ---------- request : FinetunedModel request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CreateFinetunedModelResponse] A successful response. r1rYrZr[r^r_r`Nr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rrbrIrLrMrr rNrr rOrArPrQrrrrrrrrRrcr'r'r(rds                      z/AsyncRawFinetuningClient.create_finetuned_modelrec 2|jjjdt|d|dIdH}zd|jkrdkr4nntttt| d}t ||dWS|jd krTt t |j ttjtjttjtj| dd |jd krttt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jd krtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd |jdkrtt |j ttjtjttjtj| dd | }Wnty t|jt |j |jdwt|jt |j |d)au Retrieve a fine-tuned model by its ID. Parameters ---------- id : str The fine-tuned model ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetFinetunedModelResponse] A successful response. rgr2rhNr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRrir'r'r(rjYs                     z,AsyncRawFinetuningClient.get_finetuned_modelc r)a Deletes a fine-tuned model. The model will be removed from the system and will no longer be available for use. This operation is irreversible. Parameters ---------- id : str The fine-tuned model ID. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[DeleteFinetunedModelResponse] A successful response. rgrkrhNr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRrir'r'r(rls                     z/AsyncRawFinetuningClient.delete_finetuned_modelrmrorprnc sP|jjjdt|d|t|tdd|dddi|tdId H}zd |jkr-d krBnnt t t t | d }t ||d WS|jdkrbtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d|jdkrtt|jt tjtjt tjtj| d d| }Wntyt|jt|j|jdwt|jt|j|d)a Updates the fine-tuned model with the given ID. The model will be updated with the new settings and name provided in the request body. Parameters ---------- id : str FinetunedModel ID. name : str FinetunedModel name (e.g. `foobar`). settings : Settings FinetunedModel settings such as dataset, hyperparameters... status : typing.Optional[Status] Current stage in the life-cycle of the fine-tuned model. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[UpdateFinetunedModelResponse] A successful response. rgrqrZr[rrr^r_r`Nr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr r rrbrIrLrMrr rNrr rOrArPrQrrrrrrrrRrsr'r'r(rt,s"                     z/AsyncRawFinetuningClient.update_finetuned_modelruc s>|jjjdt|dd|||d|dIdH}zd|jkr%dkr:nntttt| d }t ||d WS|jd krZt t |j ttjtjttjtj| d d |jd krztt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d | }Wntyt|jt |j |jdwt|jt |j |d)a Returns a list of events that occurred during the life-cycle of the fine-tuned model. The events are ordered by creation time, with the most recent event first. The list can be paginated using `page_size` and `page_token` parameters. Parameters ---------- finetuned_model_id : str The parent fine-tuned model ID. page_size : typing.Optional[int] Maximum number of results to be returned by the server. If 0, defaults to 50. page_token : typing.Optional[str] Request a specific page of the list results. order_by : typing.Optional[str] Comma separated list of fields. For example: "created_at,name". The default sorting order is ascending. To specify descending order for a field, append " desc" to the field name. For example: "created_at desc,name". Supported sorting fields: - created_at (default) request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListEventsResponse] A successful response. rgrvr2r3r4Nr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRrwr'r'r(rxs*                     z$AsyncRawFinetuningClient.list_eventsryc s<|jjjdt|dd||d|dIdH}zd|jkr$dkr9nntttt| d }t ||d WS|jd krYt t |j ttjtjttjtj| d d |jd krytt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d |jdkrtt |j ttjtjttjtj| d d | }Wntyt|jt |j |jdwt|jt |j |d)aO Returns a list of metrics measured during the training of a fine-tuned model. The metrics are ordered by step number, with the most recent step first. The list can be paginated using `page_size` and `page_token` parameters. Parameters ---------- finetuned_model_id : str The parent fine-tuned model ID. page_size : typing.Optional[int] Maximum number of results to be returned by the server. If 0, defaults to 50. page_token : typing.Optional[str] Request a specific page of the list results. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[ListTrainingStepMetricsResponse] A successful response. rgrzr2r{r4Nr7r8r9r<r?r@rCrDrErFrGrH)r$rJrKr rIrLrMrr rNrr rOrArPrQrrrrrrrrRr|r'r'r(r}4s!                     z3AsyncRawFinetuningClient.list_training_step_metrics)r~rrrr)rLrPrrr rrrWrrrdrrjrrlrbrrrrtrrxrr}r'r'r'r(rirr)8rL json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.serializationr core.unchecked_base_modelr errors.bad_request_errorr errors.forbidden_errorrerrors.internal_server_errorrerrors.not_found_errorr errors.service_unavailable_errorrerrors.unauthorized_errorr0finetuning.types.create_finetuned_model_responser0finetuning.types.delete_finetuned_model_responser finetuning.types.finetuned_modelr-finetuning.types.get_finetuned_model_responser%finetuning.types.list_events_responser/finetuning.types.list_finetuned_models_responser4finetuning.types.list_training_step_metrics_responserfinetuning.types.settingsrfinetuning.types.statusr0finetuning.types.update_finetuned_model_responserrMrQrbrrr'r'r'r(sD                      M