
    7Mgr                         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  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OptionalUnionc                      e Zd Zddeedeeeee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
j                     fdZddeedeeeee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
j                     fdZdeeeddeddd	de
j"                  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
j8                     fdZdeeeddeddd	de
j"                  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
j8                     fdZeddddedeej                     dee   dee   dee
j>                     f
d Z eddddedeej                     dee   dee   dee
j>                     f
d!Z!eddddedeej                     dee   dee   dee
j>                     f
d"Z"eddddedeej                     dee   dee   dee
j>                     f
d#Z#eddddedeej                     dee   dee   dee
j>                     f
d$Z$eddddedeej                     dee   dee   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_msr   r   r   r   r   r   r   r   r   r   r   r   returnc                6   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
        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_value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_requestr0   r
   rA   
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*   reqrA   http_reshttp_res_textcontent_types                        \E:\wamp.anq\www\apps\calme\nutritwin\resources\nutritwin\Lib\site-packages\mistralai/jobs.pylistz	Jobs.list   s   B //::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    )rI   r   r   rJ   build_request_asyncr0   r
   rA   rL   r   rM   do_request_asyncr   r   rO   rP   rQ   rR   r   rS   stream_to_text_asyncrU   rV   rW   rX   rY   s                        r_   
list_asynczJobs.list_asyncw   sB    B //::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<!H5"BH<#H8$AH<=H:>8H<8H<:H<)	training_filesvalidation_filesr   integrationsrepositories
auto_startr   r   r   hyperparametersrg   rh   ri   rj   rk   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
        Nr   rg   rh   rl   r   ri   rj   rk   POSTr"   TFr#   r$   c                  R    t        j                   dddt        j                        S NFjsonr   serialize_request_bodyr   JobInr*   s   r_   <lambda>zJobs.create.<locals>.<lambda>3      (D(Dvv||)ra   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   get_serialized_bodyr   r1   2jobs_api_routes_fine_tuning_create_fine_tuning_jobr8   r<   r=   r>   rB   rC   rD   rE   rF   rG   rH   ) rI   r   r   rv   r   get_pydantic_modelr   r   TrainingFileTrainingParametersInr	   JobInIntegrationsJobInRepositoriesrK   r0   r
   rA   rL   rM   rN   r   r   rO   rP   rQ   rR   2JobsAPIRoutesFineTuningCreateFineTuningJobResponserT   rU   rV   rW   rX   rZ   r   rl   rg   rh   r   ri   rj   rk   r   r   r   r(   r)   r[   rA   r\   r]   r^   r*   s                      @r_   createzJobs.create   s   V //::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^^_`  	
 	
ra   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)rn   Nro   rp   r"   TFr#   r$   c                  R    t        j                   dddt        j                        S rr   rt   rw   s   r_   rx   z#Jobs.create_async.<locals>.<lambda>  ry   ra   rz   r1   r|   r8   r<   r=   r>   rB   rC   rD   rE   rF   rG   rH   ) rI   r   r   rv   r   r}   r   r   r~   r   r	   r   r   rc   r0   r
   rA   rL   rM   rd   r   r   rO   rP   rQ   rR   r   re   rU   rV   rW   rX   r   s                      @r_   create_asynczJobs.create_asyncb  s    V //::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K2K+BK2K.AK23K048K2.K20K2)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
        Nr   r!   z/v1/fine_tuning/jobs/{job_id}FTr#   r$   r%   r1   /jobs_api_routes_fine_tuning_get_fine_tuning_jobr8   r<   r=   r>   rB   rC   rD   rE   rF   rG   rH   )rI   r   r   .JobsAPIRoutesFineTuningGetFineTuningJobRequestrK   r0   r
   rA   rL   r   rM   rN   r   r   rO   rP   rQ   rR   r   DetailedJobOutrT   rU   rV   rW   rX   rZ   r   r   r   r   r(   r)   r*   r[   rA   r\   r]   r^   s                r_   rX   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^^_`  	
 	
ra   c                l  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   )rI   r   r   r   rc   r0   r
   rA   rL   r   rM   rd   r   r   rO   rP   rQ   rR   r   r   re   rU   rV   rW   rX   r   s                r_   	get_asynczJobs.get_async5  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H4H-BH4H0AH45H268H40H42H4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
        Nr   rp   z$/v1/fine_tuning/jobs/{job_id}/cancelFTr#   r$   r%   r1   2jobs_api_routes_fine_tuning_cancel_fine_tuning_jobr8   r<   r=   r>   rB   rC   rD   rE   rF   rG   rH   )rI   r   r   1JobsAPIRoutesFineTuningCancelFineTuningJobRequestrK   r0   r
   rA   rL   r   rM   rN   r   r   rO   rP   rQ   rR   r   r   rT   rU   rV   rW   rX   r   s                r_   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^^_`  	
 	
ra   c                l  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   )rI   r   r   r   rc   r0   r
   rA   rL   r   rM   rd   r   r   rO   rP   rQ   rR   r   r   re   rU   rV   rW   rX   r   s                r_   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
        Nr   rp   z#/v1/fine_tuning/jobs/{job_id}/startFTr#   r$   r%   r1   1jobs_api_routes_fine_tuning_start_fine_tuning_jobr8   r<   r=   r>   rB   rC   rD   rE   rF   rG   rH   )rI   r   r   0JobsAPIRoutesFineTuningStartFineTuningJobRequestrK   r0   r
   rA   rL   r   rM   rN   r   r   rO   rP   rQ   rR   r   r   rT   rU   rV   rW   rX   r   s                r_   startz
Jobs.start+  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^^_`  	
 	
ra   c                l  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   )rI   r   r   r   rc   r0   r
   rA   rL   r   rM   rd   r   r   rO   rP   rQ   rR   r   r   re   rU   rV   rW   rX   r   s                r_   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   rM   rS   r`   rf   FineTuneableModelr   r   TrainingParametersInTypedDictr   r~   TrainingFileTypedDictr   JobInIntegrationsTypedDictr   JobInRepositoriesTypedDictr   r   r   r   rX   r   r   r   r   r    ra   r_   r   r      sm     #&',49(-<A/4,1(-7<$($(h
 smh
 C=	h

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

  $h
 (1h
  ~h
 !!8!89h
 (,h
 %S)h
 !%h
 "%"3"34h
 SMh
 SMh
 
&..	!h
f 8=(-
 
 %)7<$($(3
 ''
 '')M)MM
	
 !$v**+T&2N2N-OOP

 +495
 !%
 'V--.V5V5V0WW

" V--.V5V5V0WW
#
, TN-
. "%"3"34/
0 SM1
2 SM3
4 
&KK	L5
T 8=(-
 
 %)7<$($(3
 ''
 '')M)MM
	
 !$v**+T&2N2N-OOP

 +495
 !%
 'V--.V5V5V0WW

" V--.V5V5V0WW
#
, TN-
. "%"3"34/
0 SM1
2 SM3
4 
&KK	L5
J 8=$($(P
 P
 "%"3"34	P

 SMP
 SMP
 
&''	(P
l 8=$($(P
 P
 "%"3"34	P

 SMP
 SMP
 
&''	(P
l 8=$($(P
 P
 "%"3"34	P

 SMP
 SMP
 
&''	(P
l 8=$($(P
 P
 "%"3"34	P

 SMP
 SMP
 
&''	(P
l 8=$($(P
 P
 "%"3"34	P

 SMP
 SMP
 
&''	(P
l 8=$($(P
 P
 "%"3"34	P

 SMP
 SMP
 
&''	(P
ra   r   N)__doc__basesdkr   r   	mistralair   r   mistralai._hooksr   mistralai.typesr	   r
   mistralai.utilsr   typingr   r   r   r   r   ra   r_   <module>r      s.    G   # ( 3 1 ( (A
7 A
ra   