
    gn                         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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)AnyDictListMappingOptionalc                   P   e Zd Zddeeedeedddddee   dee   dee   d	eeee	f      d
ee
   dee   deej                     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edeedddddee   dee   dee   d	eeee	f      d
ee
   dee   deej                     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edddddee   dej*                  ded	eeeef      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ededddddee   dej*                  ded	eeeef      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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)!MistralJobsr   d   FN)page	page_sizemodelmetadatacreated_aftercreated_by_mestatusretries
server_url
timeout_mshttp_headersr   r   r   r   r   r   r   r   r   r   r   returnc                4   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,                  ||      ad  Get Batch Jobs

        Get a list of batch jobs for your organization and user.

        :param page:
        :param page_size:
        :param model:
        :param metadata:
        :param created_after:
        :param created_by_me:
        :param status:
        :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   GET/v1/batch/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504$jobs_api_routes_batch_get_batch_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   %JobsAPIRoutesBatchGetBatchJobsRequest_build_requestr1   r
   rB   
isinstancer   RetryConfig
do_requestr   r   Securitymatch_responseunmarshal_jsontextr   BatchJobsOutstream_to_textSDKErrorstatus_codeheadersgetselfr   r   r   r   r   r   r   r   r   r   r   r)   r*   r+   reqrB   http_reshttp_res_textcontent_types                       K/var/www/openai/venv/lib/python3.12/site-packages/mistralai/mistral_jobs.pylistzMistralJobs.list   s   > //::J!!H>>''
 !!!'"'$)%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? C  5**33V__!  %u~% # 
 %1CD''x@S@S7TUU5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
    c                z  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!   )rJ   r   r   rK   _build_request_asyncr1   r
   rB   rM   r   rN   do_request_asyncr   r   rP   rQ   rR   rS   r   rT   stream_to_text_asyncrV   rW   rX   rY   rZ   s                       r`   
list_asynczMistralJobs.list_asynct   s@    > //::J!!H>>''
 ''!'"'$)%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. C  5**33V__!  %u~% / 
 
 %1CD''x@S@S7TUU5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
/
  G Cs8   DH; H4!BH;"H7#AH;<H9=8H;7H;9H;   )r   timeout_hoursr   r   r   r   input_filesendpointri   c       	         :   d}
d}|| j                   j                  }||}
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        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,                  ||      )8  Create Batch Job

        Create a new batch job, it will be queued for processing.

        :param input_files:
        :param endpoint:
        :param model:
        :param metadata:
        :param timeout_hours:
        :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rj   rk   r   r   ri   POSTr#   TFr$   r%   c                  R    t        j                   dddt        j                        S NFjsonr   serialize_request_bodyr   
BatchJobInr+   s   r`   <lambda>z$MistralJobs.create.<locals>.<lambda>  !    (D(Dvv/@/@)rb   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r   r1   get_serialized_bodyr   r2   &jobs_api_routes_batch_create_batch_jobr9   r=   r>   r?   rC   rD   rE   rF   rG   rH   rI   )rJ   r   r   ru   rL   r1   r
   rB   rM   r   rN   rO   r   r   rP   rQ   rR   rS   r   BatchJobOutrU   rV   rW   rX   rY   r[   rj   rk   r   r   ri   r   r   r   r   r)   r*   r\   rB   r]   r^   r_   r+   s                    @r`   createzMistralJobs.create   s   6 //::J!!H###'
 !!!'"&$)%)* 2%++44! "! " 
& e%%22%?00==gu001#%HIL?? E  5**33V__!  %u~% # 
 %1CD''x@R@R7STT5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rb   c       	           K   d}
d}|| j                   j                  }||}
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        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)rm   Nrn   ro   r#   TFr$   r%   c                  R    t        j                   dddt        j                        S rq   rs   rv   s   r`   rw   z*MistralJobs.create_async.<locals>.<lambda>w  rx   rb   ry   r2   r{   r9   r=   r>   r?   rC   rD   rE   rF   rG   rH   rI   )rJ   r   r   ru   rd   r1   r
   rB   rM   r   rN   re   r   r   rP   rQ   rR   rS   r   r|   rf   rV   rW   rX   rY   r}   s                    @r`   create_asynczMistralJobs.create_async?  sB    6 //::J!!H###'
 ''!'"&$)%)* 2%++44! "! ( 
& e%%22%?00==gu001#%HIL.. E  5**33V__!  %u~% / 
 
 %1CD''x@R@R7STT5%.#>"'"<"<X"FFM//$h&:&:M8   ''++N;#88BBoo283G3G2HQ]P^^_`  	
 	
/
  G Cs8   D H>#H7$BH>%H:&AH>?H< 8H>:H><H>)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 Batch Job

        Get a batch job details by its UUID.

        :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.
        Nr   r"   z/v1/batch/jobs/{job_id}FTr$   r%   r&   r2   #jobs_api_routes_batch_get_batch_jobr9   r=   r>   r?   rC   rD   rE   rF   rG   rH   rI   )rJ   r   r   $JobsAPIRoutesBatchGetBatchJobRequestrL   r1   r
   rB   rM   r   rN   rO   r   r   rP   rQ   rR   rS   r   r|   rU   rV   rW   rX   rY   r[   r   r   r   r   r   r)   r*   r+   r\   rB   r]   r^   r_   s                 r`   rY   zMistralJobs.get  s
   & //::J!!H==
 !!*'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? B  5**33V__!  %u~% # 
 %1CD''x@R@R7STT5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rb   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   )rJ   r   r   r   rd   r1   r
   rB   rM   r   rN   re   r   r   rP   rQ   rR   rS   r   r|   rf   rV   rW   rX   rY   r   s                 r`   	get_asynczMistralJobs.get_async  s.    & //::J!!H==
 ''*'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. B  5**33V__!  %u~% / 
 
 %1CD''x@R@R7STT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 Batch Job

        Request the cancellation of a batch 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   ro   z/v1/batch/jobs/{job_id}/cancelFTr$   r%   r&   r2   &jobs_api_routes_batch_cancel_batch_jobr9   r=   r>   r?   rC   rD   rE   rF   rG   rH   rI   )rJ   r   r   'JobsAPIRoutesBatchCancelBatchJobRequestrL   r1   r
   rB   rM   r   rN   rO   r   r   rP   rQ   rR   rS   r   r|   rU   rV   rW   rX   rY   r   s                 r`   cancelzMistralJobs.cancelM  s
   & //::J!!H@@
 !!1'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? E  5**33V__!  %u~% # 
 %1CD''x@R@R7STT5%.#>!00:M//$h&:&:M8   ''++N;,,X6oo283G3G2HQ]P^^_`  	
 	
rb   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   )rJ   r   r   r   rd   r1   r
   rB   rM   r   rN   re   r   r   rP   rQ   rR   rS   r   r|   rf   rV   rW   rX   rY   r   s                 r`   cancel_asynczMistralJobs.cancel_async  s.    & //::J!!H@@
 ''1'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. E  5**33V__!  %u~% / 
 
 %1CD''x@R@R7STT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   r   r   boolr   BatchJobStatusr   rN   r   rT   ra   rg   r   APIEndpointr|   r~   r   rY   r   r   r    rb   r`   r   r      s     #&',5:49(-:?7<$($(48e
 sme
 C=	e

  $e
 #4S>2e
 (1e
  ~e
 !!6!67e
 "%"3"34e
 SMe
 SMe
 wsCx01e
 
&%%	&e
T  #&',5:49(-:?7<$($(48e
 sme
 C=	e

  $e
 #4S>2e
 (1e
  ~e
 !!6!67e
 "%"3"34e
 SMe
 SMe
 wsCx01e
 
&%%	&e
Z 6;')7<$($(48b
 #Yb
 $$	b

 b
 #4S>2b
  }b
 "%"3"34b
 SMb
 SMb
 wsCx01b
 
&$$	%b
T 6;')7<$($(48b
 #Yb
 $$	b

 b
 #4S>2b
  }b
 "%"3"34b
 SMb
 SMb
 wsCx01b
 
&$$	%b
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
rb   r   N)__doc__basesdkr   r   	mistralair   r   mistralai._hooksr   mistralai.typesr	   r
   mistralai.utilsr   typingr   r   r   r   r   r   r   rb   r`   <module>r      s.    G   # ( 3 1 5 5i
' i
rb   