o i@sdZddlmZddlmZddlmZmZddlmZddl m Z m Z ddl m Z dd lmZdd lmZmZmZmZGd d d eZd S)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.)BaseSDK)datetime)modelsutils) HookContext)OptionalNullableUNSET)get_security_from_env)unmarshal_json_response)ListMappingOptionalUnionc"@s4eZdZddeeedeeeeedddddeedeedeed eed eed ee d ee j d eedeedeedee j deedeedeeeefde jfddZddeeedeeeeedddddeedeedeed eed eed ee d ee j d eedeedeedee j deedeedeeeefde jfddZdeeeddeeeedddd dedee je jfdeeee jee jfdeeedeedeeee jee jfdee deed ee jd!eeee jee jfd"eeee jee jfdee j deedeedeeeefde j f d#d$Z!deeeddeeeedddd dedee je jfdeeee jee jfdeeedeedeeee jee jfdee deed ee jd!eeee jee jfd"eeee jee jfdee j deedeedeeeefde j f d%d&Z"edddd'd(edee j deedeedeeeefde j#f d)d*Z$edddd'd(edee j deedeedeeeefde j#f d+d,Z%edddd'd(edee j deedeedeeeefde j&f d-d.Z'edddd'd(edee j deedeedeeeefde j&f d/d0Z(edddd'd(edee j deedeedeeeefde j)f d1d2Z*edddd'd(edee j deedeedeeeefde j)f d3d4Z+dS)5JobsrdFN)page page_sizemodel created_aftercreated_before created_by_mestatus wandb_project wandb_namesuffixretries server_url timeout_ms http_headersrrrrrrrrrrrrrrreturncCsXd}d}| dur |jj} | dur| }n|||}tj||||||||| | d }|jdd|||ddddd||jj| d }| tkrJ|jjturJ|jj} d}t | t j rX| gd f}|j t |j|p`d d gt|jjtjd |ddg|d}t |ddrttj|St |ddrt |}td||t |ddrt |}td||td|aGet Fine Tuning Jobs Get a list of fine-tuning jobs for your organization and user. :param page: The page number of the results to be returned. :param page_size: The number of items to return per page. :param model: The model name used for fine-tuning to filter on. When set, the other results are not displayed. :param created_after: The date/time to filter on. When set, the results for previous creation times are not displayed. :param created_before: :param created_by_me: When set, only return results for jobs created by the API caller. Other results are not displayed. :param status: The current job state to filter on. When set, the other results are not displayed. :param wandb_project: The Weights and Biases project to filter on. When set, the other results are not displayed. :param wandb_name: The Weight and Biases run name to filter on. When set, the other results are not displayed. :param suffix: The model suffix to filter on. When set, the other results are not displayed. :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. N) rrrrrrrrrrGET/v1/fine_tuning/jobsFT user-agentapplication/json methodpathbase_url url_variablesrequestrequest_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuersecurityr4295005025035040jobs_api_routes_fine_tuning_get_fine_tuning_jobsconfigr) operation_id oauth2_scopessecurity_source4XX5XXhook_ctxr+error_status_codes retry_config200*API error occurredUnexpected response received)sdk_configurationr_get_urlr/JobsAPIRoutesFineTuningGetFineTuningJobsRequest_build_requestr1r rD isinstancer RetryConfig do_requestrr Securitymatch_responser JobsOutstream_to_textSDKErrorselfrrrrrrrrrrrrrrr)r*r+reqrDhttp_res http_res_textrZT/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/jobs.pylists~%           z Jobs.listcsld}d}| dur |jj} | dur| }n|||}tj||||||||| | d }|jdd|||ddddd||jj| d }| tkrK|jjturK|jj} d}t | t j rY| gd f}|j t |j|pad d gt|jjtjd |ddg|dIdH}t |ddrttj|St |ddrt |IdH}td||t |ddrt |IdH}td||td|r!)rIrrJrrK_build_request_asyncr1r rDrMrrNdo_request_asyncrr rPrQr rRstream_to_text_asyncrTrUrZrZr[ list_async|s%          zJobs.list_async) training_filesvalidation_filesr integrations auto_startinvalid_sample_skip_percentagejob_type repositoriesclassifier_targetsrrrrhyperparametersrarbrcrdrerfrgrhcsd}d}|dur |jj}| dur| }n|||}tj|t|tttj ||t|t ttj ||| t|tj t| t ttj t| t ttjd |jdd||ddddd||jjfd d |d }| tkry|jjtury|jj} d}t| tjr| gd f}|jt|j|pd dgt|jjtjd|ddg|d}t|ddrttj|St|ddrt|}td||t|ddrt|}td||td|)Create Fine Tuning Job Create a new fine-tuning job, it will be queued for processing. :param model: The name of the model to fine-tune. :param hyperparameters: :param training_files: :param validation_files: A list containing the IDs of uploaded files that contain validation data. If you provide these files, the data is used to generate validation metrics periodically during fine-tuning. These metrics can be viewed in `checkpoints` when getting the status of a running fine-tuning job. The same data should not be present in both train and validation files. :param suffix: A string that will be added to your fine-tuning model name. For example, a suffix of \"my-great-model\" would produce a model name like `ft:open-mistral-7b:my-great-model:xxx...` :param integrations: A list of integrations to enable for your fine-tuning job. :param auto_start: This field will be required in a future release. :param invalid_sample_skip_percentage: :param job_type: :param repositories: :param classifier_targets: :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. N rrarbrrcrdrerfrirgrhPOSTr#TFr$r%ctdddtjSNFjsonrserialize_request_bodyrJobInrZr+rZr[O zJobs.create..r'r(r)r*r+r,r-r.r/r0rr1get_serialized_bodyrr2r82jobs_api_routes_fine_tuning_create_fine_tuning_jobr:r?r@rArErFrGrH)rIrrJrrrrget_pydantic_modelrr TrainingFilerJobInIntegrationsHyperparametersJobInRepositoriesClassifierTargetInrLr1r rDrMrNrOrr rPrQr 2JobsAPIRoutesFineTuningCreateFineTuningJobResponserSrTrVrrirarbrrcrdrerfrgrhrrrrr)r*rWrDrXrYrZrsr[creates6          z Jobs.createcsd}d}|dur |jj}| dur| }n|||}tj|t|tttj ||t|t ttj ||| t|tj t| t ttj t| t ttjd |jdd||ddddd||jjfd d |d }| tkrz|jjturz|jj} d}t| tjr| gd f}|jt|j|pd dgt|jjtjd|ddg|dIdH}t|ddrttj|St|ddrt|IdH}td||t|ddrt|IdH}td||td|)rjNrkrlr#TFr$r%crmrnrprZrsrZr[rtruz#Jobs.create_async..rvr2r8rxr:r?r@rArErFrGrH)rIrrJrrrrryrr rzrr{r|r}r~r]r1r rDrMrNr^rr rPrQr rr_rTrrZrsr[ create_asyncys6         zJobs.create_async)rrrrjob_idc CFd}d}|dur |jj}|dur|}n|||}tj|d}|jdd|||ddddd||jj|d } |tkrA|jjturA|jj}d} t |t j rO|gd f} |j t |j|pWd d gt|jjtjd | ddg| d} t | ddrwttj| St | ddrt | } td| | t | ddrt | } td| | td| aGet Fine Tuning Job Get a fine-tuned job details by its UUID. :param job_id: The ID of the job to analyse. :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. Nrr"z/v1/fine_tuning/jobs/{job_id}FTr$r%r&r2r8/jobs_api_routes_fine_tuning_get_fine_tuning_jobr:r?r@rArErFrGrH)rIrrJr.JobsAPIRoutesFineTuningGetFineTuningJobRequestrLr1r rDrMrrNrOrr rPrQr /JobsAPIRoutesFineTuningGetFineTuningJobResponserSrT rVrrrrrr)r*r+rWrDrXrYrZrZr[getp         zJobs.getc Zd}d}|dur |jj}|dur|}n|||}tj|d}|jdd|||ddddd||jj|d } |tkrB|jjturB|jj}d} t |t j rP|gd f} |j t |j|pXd d gt|jjtjd | ddg| dIdH} t | ddr{ttj| St | ddrt | IdH} td| | t | ddrt | IdH} td| | td| r)rIrrJrrr]r1r rDrMrrNr^rr rPrQr rr_rTrrZrZr[ get_async]r        zJobs.get_asyncc CraCancel Fine Tuning Job Request the cancellation of a fine tuning job. :param job_id: The ID of the job to cancel. :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. Nrrlz$/v1/fine_tuning/jobs/{job_id}/cancelFTr$r%r&r2r82jobs_api_routes_fine_tuning_cancel_fine_tuning_jobr:r?r@rArErFrGrH)rIrrJr1JobsAPIRoutesFineTuningCancelFineTuningJobRequestrLr1r rDrMrrNrOrr rPrQr 2JobsAPIRoutesFineTuningCancelFineTuningJobResponserSrTrrZrZr[cancelrz Jobs.cancelc rr)rIrrJrrr]r1r rDrMrrNr^rr rPrQr rr_rTrrZrZr[ cancel_asyncrzJobs.cancel_asyncc CraStart Fine Tuning Job Request the start of a validated fine tuning job. :param job_id: :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. Nrrlz#/v1/fine_tuning/jobs/{job_id}/startFTr$r%r&r2r81jobs_api_routes_fine_tuning_start_fine_tuning_jobr:r?r@rArErFrGrH)rIrrJr0JobsAPIRoutesFineTuningStartFineTuningJobRequestrLr1r rDrMrrNrOrr rPrQr 1JobsAPIRoutesFineTuningStartFineTuningJobResponserSrTrrZrZr[start\rz Jobs.startc rr)rIrrJrrr]r1r rDrMrrNr^rr rPrQr rr_rTrrZrZr[ start_asyncrzJobs.start_async),__name__ __module__ __qualname__r rintrstrrboolrQueryParamStatusrrNr rRr\r`rr|HyperparametersTypedDictr rzTrainingFileTypedDictr{JobInIntegrationsTypedDictfloatFineTuneableModelTyper}JobInRepositoriesTypedDictr~ClassifierTargetInTypedDictrrrrrrrrrrrrrZrZrZr[r s~      q      u    !     !  Y Y Y Y YrN)__doc__basesdkrr mistralairrmistralai._hooksrmistralai.typesrr mistralai.utilsr 'mistralai.utils.unmarshal_json_responser typingr r rrrrZrZrZr[s