
    g;                         d Z ddlmZ ddlmZ ddlmZmZ ddlmZ ddl	m
Z
mZ ddlmZ dd	lmZmZmZmZ  G d
 de      Zy)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.   )BaseSDK    )datetime)modelsutils)HookContext)OptionalNullableUNSET)get_security_from_env)ListMappingOptionalUnionc                   X   e Zd Zddeedeeeeedddddee   dee   dee   d	ee   d
ee	   dee
j                     dee   dee   dee   deej                     dee   dee   deeeef      dee
j                     fdZddeedeeeeedddddee   dee   dee   d	ee   d
ee	   dee
j                     dee   dee   dee   deej                     dee   dee   deeeef      dee
j                     fdZdeeeddedddd
dedee
j&                  e
j(                  f   deeee
j,                     ee
j.                     f      deee      dee   deeee
j0                     ee
j2                     f      deeee
j4                     ee
j6                     f      dee	   deej                     dee   dee   deeeef      dee
j8                     fdZdeeeddedddd
dedee
j&                  e
j(                  f   deeee
j,                     ee
j.                     f      deee      dee   deeee
j0                     ee
j2                     f      deeee
j4                     ee
j6                     f      dee	   deej                     dee   dee   deeeef      dee
j8                     fdZeddddd edeej                     dee   dee   deeeef      dee
j>                     fd!Z eddddd edeej                     dee   dee   deeeef      dee
j>                     fd"Z!eddddd edeej                     dee   dee   deeeef      dee
j>                     fd#Z"eddddd edeej                     dee   dee   deeeef      dee
j>                     fd$Z#eddddd edeej                     dee   dee   deeeef      dee
j>                     fd%Z$eddddd edeej                     dee   dee   deeeef      dee
j>                     fd&Z%y)'Jobsr   d   FN)page	page_sizemodelcreated_aftercreated_by_mestatuswandb_project
wandb_namesuffixretries
server_url
timeout_mshttp_headersr   r   r   r   r   r   r   r   r   r   r   r   r   returnc                8   d}d}|| j                   j                  }||}t        j                  |||||||||		      }| j	                  dd|||ddddd|| j                   j
                  |	      }|
t        k(  r2| j                   j                  t        ur| j                   j                  }
d}t        |
t        j                        r|
g d
f}| j                  t        dg t        | j                   j
                  t        j                              |ddg|      }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r7t        j(                  |      }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |      }t        j*                  d|j,                   d| d|j,                  ||      a|  Get 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_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)	r   r   r   r   r   r   r   r   r   GET/v1/fine_tuning/jobsFT
user-agentapplication/jsonmethodpathbase_urlurl_variablesrequestrequest_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuer   securityr   4295005025035040jobs_api_routes_fine_tuning_get_fine_tuning_jobsoperation_idoauth2_scopessecurity_source4XX5XXhook_ctxr,   error_status_codesretry_config200*API error occurredContent-Type$Unexpected response received (code: , type: ))sdk_configurationr   r   /JobsAPIRoutesFineTuningGetFineTuningJobsRequest_build_requestr2   r
   rC   
isinstancer   RetryConfig
do_requestr   r   Securitymatch_responseunmarshal_jsontextr   JobsOutstream_to_textSDKErrorstatus_codeheadersgetselfr   r   r   r   r   r   r   r   r   r   r   r   r   r*   r+   r,   reqrC   http_reshttp_res_textcontent_types                         C/var/www/openai/venv/lib/python3.12/site-packages/mistralai/jobs.pylistz	Jobs.list   s!   F //::J!!HHH'''!

 !!''"'$)%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? O  5**33V__!  %u~% # 
 %1CD''x7OPP5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
    c                ~  K   d}d}|| j                   j                  }||}t        j                  |||||||||		      }| j	                  dd|||ddddd|| j                   j
                  |	      }|
t        k(  r2| j                   j                  t        ur| j                   j                  }
d}t        |
t        j                        r|
g d
f}| j                  t        dg t        | j                   j
                  t        j                              |ddg|       d{   }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r?t        j(                  |       d{   }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |       d{   }t        j*                  d|j,                   d| d|j,                  ||      7 7 7 <wr"   )rK   r   r   rL   _build_request_asyncr2   r
   rC   rN   r   rO   do_request_asyncr   r   rQ   rR   rS   rT   r   rU   stream_to_text_asyncrW   rX   rY   rZ   r[   s                         ra   
list_asynczJobs.list_asyncz   sE    F //::J!!HHH'''!

 ''''"'$)%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. O  5**33V__!  %u~% / 
 
 %1CD''x7OPP5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
/
  G Cs8   D H="H6#BH=$H9%AH=>H;?8H=9H=;H=)
training_filesvalidation_filesr   integrationsrepositories
auto_startr   r   r   r   hyperparametersri   rj   rk   rl   rm   c                   d}d}|| j                   j                  }|
|
}t        j                  |t	        j
                  |t        t        t        j                              |t	        j
                  |t        j                        |t	        j
                  |t        t        t        j                              t	        j
                  |t        t        t        j                              |      | j                  dd||ddddd|| j                   j                  fd	|
      }|	t        k(  r2| j                   j                   t        ur| j                   j                   }	d}t#        |	t        j$                        r|	g df}| j'                  t)        dg t+        | j                   j                  t        j,                              |ddg|      }t	        j.                  |dd      r5t	        j0                  |j2                  t        t        j4                           S t	        j.                  |ddgd      r7t	        j6                  |      }t        j8                  d|j:                  ||      |j<                  j?                  d      }t	        j6                  |      }t        j8                  d|j:                   d| d|j:                  ||      )  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: The fine-tuning hyperparameter settings used in a fine-tune job.
        :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 repositories:
        :param auto_start: This field will be required in a future release.
        :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r   ri   rj   rn   r   rk   rl   rm   POSTr$   TFr%   r&   c                  R    t        j                   dddt        j                        S NFjsonr   serialize_request_bodyr   JobInr,   s   ra   <lambda>zJobs.create.<locals>.<lambda><      (D(Dvv||)rc   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r   r2   get_serialized_bodyr   r3   2jobs_api_routes_fine_tuning_create_fine_tuning_jobr:   r>   r?   r@   rD   rE   rF   rG   rH   rI   rJ   ) rK   r   r   rx   r   get_pydantic_modelr   r   TrainingFileTrainingParametersInr	   JobInIntegrationsJobInRepositoriesrM   r2   r
   rC   rN   rO   rP   r   r   rQ   rR   rS   rT   2JobsAPIRoutesFineTuningCreateFineTuningJobResponserV   rW   rX   rY   rZ   r\   r   rn   ri   rj   r   rk   rl   rm   r   r   r   r   r*   r+   r]   rC   r^   r_   r`   r,   s                       @ra   createzJobs.create   s   Z //::J!!H,, 33f.A.A)B C .!44!<!< 11.tF4L4L/MN 11htF,D,D'EF "!
& !!''"&$)%)* 2%++44! "! " 
& e%%22%?00==gu001#%HIL?? Q  5**33V__!  %u~% # 
 %1CD''RRS  5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rc   c                  K   d}d}|| j                   j                  }|
|
}t        j                  |t	        j
                  |t        t        t        j                              |t	        j
                  |t        j                        |t	        j
                  |t        t        t        j                              t	        j
                  |t        t        t        j                              |      | j                  dd||ddddd|| j                   j                  fd	|
      }|	t        k(  r2| j                   j                   t        ur| j                   j                   }	d}t#        |	t        j$                        r|	g df}| j'                  t)        dg t+        | j                   j                  t        j,                              |ddg|       d{   }t	        j.                  |dd      r5t	        j0                  |j2                  t        t        j4                           S t	        j.                  |ddgd      r?t	        j6                  |       d{   }t        j8                  d|j:                  ||      |j<                  j?                  d      }t	        j6                  |       d{   }t        j8                  d|j:                   d| d|j:                  ||      7 7 7 <w)rp   Nrq   rr   r$   TFr%   r&   c                  R    t        j                   dddt        j                        S rt   rv   ry   s   ra   rz   z#Jobs.create_async.<locals>.<lambda>  r{   rc   r|   r3   r~   r:   r>   r?   r@   rD   rE   rF   rG   rH   rI   rJ   ) rK   r   r   rx   r   r   r   r   r   r   r	   r   r   re   r2   r
   rC   rN   rO   rf   r   r   rQ   rR   rS   rT   r   rg   rW   rX   rY   rZ   r   s                       @ra   create_asynczJobs.create_asynck  s    Z //::J!!H,, 33f.A.A)B C .!44!<!< 11.tF4L4L/MN 11htF,D,D'EF "!
& ''''"&$)%)* 2%++44! "! ( 
& e%%22%?00==gu001#%HIL.. Q  5**33V__!  %u~% / 
 
 %1CD''RRS  5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
5
& G Cs8   GK3K,BK3K/AK34K158K3/K31K3)r   r   r   r   job_idc                (   d}d}|| j                   j                  }||}t        j                  |      }| j	                  dd|||ddddd|| j                   j
                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        dg t        | j                   j
                  t        j                              |	ddg|
      }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r7t        j(                  |      }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |      }t        j*                  d|j,                   d| d|j,                  ||      a  Get 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.
        Nr   r#   z/v1/fine_tuning/jobs/{job_id}FTr%   r&   r'   r3   /jobs_api_routes_fine_tuning_get_fine_tuning_jobr:   r>   r?   r@   rD   rE   rF   rG   rH   rI   rJ   )rK   r   r   .JobsAPIRoutesFineTuningGetFineTuningJobRequestrM   r2   r
   rC   rN   r   rO   rP   r   r   rQ   rR   rS   rT   r   DetailedJobOutrV   rW   rX   rY   rZ   r\   r   r   r   r   r   r*   r+   r,   r]   rC   r^   r_   r`   s                 ra   rZ   zJobs.get  s
   & //::J!!HGG
 !!0'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? N  5**33V__!  %u~% # 
 %1CD''x@U@U7VWW5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rc   c                n  K   d}d}|| j                   j                  }||}t        j                  |      }| j	                  dd|||ddddd|| j                   j
                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        dg t        | j                   j
                  t        j                              |	ddg|
       d{   }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r?t        j(                  |       d{   }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |       d{   }t        j*                  d|j,                   d| d|j,                  ||      7 7 7 <wr   )rK   r   r   r   re   r2   r
   rC   rN   r   rO   rf   r   r   rQ   rR   rS   rT   r   r   rg   rW   rX   rY   rZ   r   s                 ra   	get_asynczJobs.get_asyncD  s.    & //::J!!HGG
 ''0'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. N  5**33V__!  %u~% / 
 
 %1CD''x@U@U7VWW5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
/
  G C8   DH5H.BH5H1AH56H378H51H53H5c                (   d}d}|| j                   j                  }||}t        j                  |      }| j	                  dd|||ddddd|| j                   j
                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        dg t        | j                   j
                  t        j                              |	ddg|
      }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r7t        j(                  |      }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |      }t        j*                  d|j,                   d| d|j,                  ||      a  Cancel 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.
        Nr   rr   z$/v1/fine_tuning/jobs/{job_id}/cancelFTr%   r&   r'   r3   2jobs_api_routes_fine_tuning_cancel_fine_tuning_jobr:   r>   r?   r@   rD   rE   rF   rG   rH   rI   rJ   )rK   r   r   1JobsAPIRoutesFineTuningCancelFineTuningJobRequestrM   r2   r
   rC   rN   r   rO   rP   r   r   rQ   rR   rS   rT   r   r   rV   rW   rX   rY   rZ   r   s                 ra   cancelzJobs.cancel  s
   & //::J!!HJJ
 !!7'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? Q  5**33V__!  %u~% # 
 %1CD''x@U@U7VWW5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rc   c                n  K   d}d}|| j                   j                  }||}t        j                  |      }| j	                  dd|||ddddd|| j                   j
                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        dg t        | j                   j
                  t        j                              |	ddg|
       d{   }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r?t        j(                  |       d{   }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |       d{   }t        j*                  d|j,                   d| d|j,                  ||      7 7 7 <wr   )rK   r   r   r   re   r2   r
   rC   rN   r   rO   rf   r   r   rQ   rR   rS   rT   r   r   rg   rW   rX   rY   rZ   r   s                 ra   cancel_asynczJobs.cancel_async  s.    & //::J!!HJJ
 ''7'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. Q  5**33V__!  %u~% / 
 
 %1CD''x@U@U7VWW5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
/
  G Cr   c                (   d}d}|| j                   j                  }||}t        j                  |      }| j	                  dd|||ddddd|| j                   j
                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        dg t        | j                   j
                  t        j                              |	ddg|
      }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r7t        j(                  |      }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |      }t        j*                  d|j,                   d| d|j,                  ||      a  Start 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.
        Nr   rr   z#/v1/fine_tuning/jobs/{job_id}/startFTr%   r&   r'   r3   1jobs_api_routes_fine_tuning_start_fine_tuning_jobr:   r>   r?   r@   rD   rE   rF   rG   rH   rI   rJ   )rK   r   r   0JobsAPIRoutesFineTuningStartFineTuningJobRequestrM   r2   r
   rC   rN   r   rO   rP   r   r   rQ   rR   rS   rT   r   r   rV   rW   rX   rY   rZ   r   s                 ra   startz
Jobs.startC  s
   & //::J!!HII
 !!6'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? P  5**33V__!  %u~% # 
 %1CD''x@U@U7VWW5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rc   c                n  K   d}d}|| j                   j                  }||}t        j                  |      }| j	                  dd|||ddddd|| j                   j
                  |	      }	|t        k(  r2| j                   j                  t        ur| j                   j                  }d}
t        |t        j                        r|g d
f}
| j                  t        dg t        | j                   j
                  t        j                              |	ddg|
       d{   }t        j                  |dd      r5t        j                   |j"                  t$        t        j&                           S t        j                  |ddgd      r?t        j(                  |       d{   }t        j*                  d|j,                  ||      |j.                  j1                  d      }t        j(                  |       d{   }t        j*                  d|j,                   d| d|j,                  ||      7 7 7 <wr   )rK   r   r   r   re   r2   r
   rC   rN   r   rO   rf   r   r   rQ   rR   rS   rT   r   r   rg   rW   rX   rY   rZ   r   s                 ra   start_asynczJobs.start_async  s.    & //::J!!HII
 ''6'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. P  5**33V__!  %u~% / 
 
 %1CD''x@U@U7VWW5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
/
  G Cr   )&__name__
__module____qualname__r
   r   intr	   strr   boolr   QueryParamStatusr   rO   r   rU   rb   rh   r   r   TrainingParametersInTypedDictr   r   TrainingFileTypedDictr   JobInIntegrationsTypedDictr   JobInRepositoriesTypedDictr   r   r   r   rZ   r   r   r   r   r    rc   ra   r   r      sy     #&',49(-<A/4,1(-7<$($(48k
 smk
 C=	k

  $k
 (1k
  ~k
 !!8!89k
 (,k
 %S)k
 !%k
 "%"3"34k
 SMk
 SMk
 wsCx01k
  
&..	!!k
`  #&',49(-<A/4,1(-7<$($(48k
 smk
 C=	k

  $k
 (1k
  ~k
 !!8!89k
 (,k
 %S)k
 !%k
 "%"3"34k
 SMk
 SMk
 wsCx01k
  
&..	!!k
l 8=(-
 
 %)7<$($(485B
 B
 '')M)MM
	B
 !$v**+T&2N2N-OOP
B
 +495B
 !%B
 'V--.V5V5V0WW
B
" V--.V5V5V0WW
#B
, TN-B
. "%"3"34/B
0 SM1B
2 SM3B
4 wsCx015B
6 
&KK	L7B
Z 8=(-
 
 %)7<$($(485B
 B
 '')M)MM
	B
 !$v**+T&2N2N-OOP
B
 +495B
 !%B
 'V--.V5V5V0WW
B
" V--.V5V5V0WW
#B
, TN-B
. "%"3"34/B
0 SM1B
2 SM3B
4 wsCx015B
6 
&KK	L7B
P 8=$($(48S
 S
 "%"3"34	S

 SMS
 SMS
 wsCx01S
 
&''	(S
r 8=$($(48S
 S
 "%"3"34	S

 SMS
 SMS
 wsCx01S
 
&''	(S
r 8=$($(48S
 S
 "%"3"34	S

 SMS
 SMS
 wsCx01S
 
&''	(S
r 8=$($(48S
 S
 "%"3"34	S

 SMS
 SMS
 wsCx01S
 
&''	(S
r 8=$($(48S
 S
 "%"3"34	S

 SMS
 SMS
 wsCx01S
 
&''	(S
r 8=$($(48S
 S
 "%"3"34	S

 SMS
 SMS
 wsCx01S
 
&''	(S
rc   r   N)__doc__basesdkr   r   	mistralair   r   mistralai._hooksr   mistralai.typesr	   r
   mistralai.utilsr   typingr   r   r   r   r   r   rc   ra   <module>r      s.    G   # ( 3 1 1 1_
7 _
rc   