
    i^q                         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 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)unmarshal_json_response)AnyDictListMappingOptionalc                   l   e Zd Zddeeeedeedddddee   dee   dee   d	ee   d
eeee	f      dee
   dee   deeej                        deej                      dee   dee   deeeef      dej$                  fdZddeeeedeedddddee   dee   dee   d	ee   d
eeee	f      dee
   dee   deeej                        deej                      dee   dee   deeeef      dej$                  fdZeeededddddee   dej*                  dee   d	e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j,                  fdZeeededddddee   dej*                  dee   d	e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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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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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j,                  fd!Zy)"MistralJobsr   d   FN)page	page_sizemodelagent_idmetadatacreated_aftercreated_by_mestatusretries
server_url
timeout_mshttp_headersr   r   r   r   r   r   r   r   r   r   r   r    returnc                   d}d}|| j                   j                  }|
|
}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |ddg|      }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j(                  d|      a}  Get Batch Jobs

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

        :param page:
        :param page_size:
        :param model:
        :param agent_id:
        :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   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configr+   operation_idoauth2_scopessecurity_source4XX5XXhook_ctxr-   error_status_codesretry_config200*API error occurredUnexpected response received)sdk_configurationr   _get_urlr   %JobsAPIRoutesBatchGetBatchJobsRequest_build_requestr3   r
   rF   
isinstancer   RetryConfig
do_requestr   r   Securitymatch_responser   BatchJobsOutstream_to_textSDKErrorselfr   r   r   r   r   r   r   r   r   r   r   r    r+   r,   r-   reqrF   http_reshttp_res_texts                       T/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/mistralai/mistral_jobs.pylistzMistralJobs.list   s   B //::J!!H}}X}=H>>''	
 !!!'"'$)%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? --!RC  5**33V__!  %u~% # 
 %1CD*6+>+>II%5!00:M//"6-PP%5!00:M//"6-PPoo<hGG    c                \  K   d}d}|| j                   j                  }|
|
}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |ddg|       d{   }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j(                  d|      7 7 ~7 5wr#   )rK   r   rL   r   rM   _build_request_asyncr3   r
   rF   rO   r   rP   do_request_asyncr   r   rR   rS   r   rT   stream_to_text_asyncrV   rW   s                       r\   
list_asynczMistralJobs.list_asyncv   s)    B //::J!!H}}X}=H>>''	
 ''!'"'$)%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. --!RC  5**33V__!  %u~% / 
 
 %1CD*6+>+>II%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG1
$ G Gs8   EH,H&A$H,)H(*A
H,4H*52H,(H,*H,   )r   r   r   timeout_hoursr   r   r   r    input_filesendpointre   c       
            d}d}|	| j                   j                  }	||}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |ddg|      }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j(                  d|      )Q  Create Batch Job

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

        :param input_files:
        :param endpoint:
        :param model:
        :param agent_id:
        :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rf   rg   r   r   r   re   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/@/@) r^   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r    r3   get_serialized_bodyr   r4   r:   &jobs_api_routes_batch_create_batch_jobr<   rA   rB   rC   rG   rH   rI   rJ   )rK   r   rL   r   rq   rN   r3   r
   rF   rO   r   rP   rQ   r   r   rR   rS   r   BatchJobOutrU   rV   rX   rf   rg   r   r   r   re   r   r   r   r    r+   r,   rY   rF   rZ   r[   r-   s                    @r\   createzMistralJobs.create   s   : //::J!!H}}X}=H###'
 !!!'"&$)%)* 2%++44! "! " 
& e%%22%?00==gu001#%HIL?? --!RE  5**33V__!  %u~% # 
 %1CD*6+=+=xHH%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr^   c       
         b  K   d}d}|	| j                   j                  }	||}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |ddg|       d{   }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j(                  d|      7 7 ~7 5w)ri   Nrj   rk   r%   TFr&   r'   c                  R    t        j                   dddt        j                        S rm   ro   rr   s   r\   rs   z*MistralJobs.create_async.<locals>.<lambda>  rt   r^   ru   r4   r:   rw   r<   rA   rB   rC   rG   rH   rI   rJ   )rK   r   rL   r   rq   r`   r3   r
   rF   rO   r   rP   ra   r   r   rR   rS   r   rx   rb   rV   ry   s                    @r\   create_asynczMistralJobs.create_asyncC  s*    : //::J!!H}}X}=H###'
 ''!'"&$)%)* 2%++44! "! ( 
& e%%22%?00==gu001#%HIL.. --!RE  5**33V__!  %u~% / 
 
 %1CD*6+=+=xHH%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG1
$ G Gs8   EH/H)A$H/,H+-A
H/7H-82H/+H/-H/)r   r   r   r    job_idc                
   d}d}|| j                   j                  }||}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |	ddg|
      }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j(                  d|      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(   r4   r:   #jobs_api_routes_batch_get_batch_jobr<   rA   rB   rC   rG   rH   rI   rJ   )rK   r   rL   r   $JobsAPIRoutesBatchGetBatchJobRequestrN   r3   r
   rF   rO   r   rP   rQ   r   r   rR   rS   r   rx   rU   rV   rX   r~   r   r   r   r    r+   r,   r-   rY   rF   rZ   r[   s                r\   getzMistralJobs.get  s   & //::J!!H}}X}=H==
 !!*'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? --!RB  5**33V__!  %u~% # 
 %1CD*6+=+=xHH%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr^   c                N  K   d}d}|| j                   j                  }||}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |	ddg|
       d{   }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j(                  d|      7 7 ~7 5wr   )rK   r   rL   r   r   r`   r3   r
   rF   rO   r   rP   ra   r   r   rR   rS   r   rx   rb   rV   r   s                r\   	get_asynczMistralJobs.get_async  s    & //::J!!H}}X}=H==
 ''*'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. --!RB  5**33V__!  %u~% / 
 
 %1CD*6+=+=xHH%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG1
$ G G8   D;H%=H>A$H%"H!#A
H%-H#.2H%!H%#H%c                
   d}d}|| j                   j                  }||}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |	ddg|
      }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j                   |dd      r,t        j&                  |      }t        j(                  d||      t        j(                  d|      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   rk   z/v1/batch/jobs/{job_id}/cancelFTr&   r'   r(   r4   r:   &jobs_api_routes_batch_cancel_batch_jobr<   rA   rB   rC   rG   rH   rI   rJ   )rK   r   rL   r   'JobsAPIRoutesBatchCancelBatchJobRequestrN   r3   r
   rF   rO   r   rP   rQ   r   r   rR   rS   r   rx   rU   rV   r   s                r\   cancelzMistralJobs.cancelN  s   & //::J!!H}}X}=H@@
 !!1'"'$(%)* 2%++44! " 
  e%%22%?00==gu001#%HIL?? --!RE  5**33V__!  %u~% # 
 %1CD*6+=+=xHH%5!00:M//"6-PP%5!00:M//"6-PPoo<hGGr^   c                N  K   d}d}|| j                   j                  }||}n| 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        | j                   |xs ddg t        | j                   j                  t        j                              |	ddg|
       d{   }t        j                   |dd      rt#        t        j$                  |      S t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j                   |dd      r4t        j&                  |       d{   }t        j(                  d||      t        j(                  d|      7 7 ~7 5wr   )rK   r   rL   r   r   r`   r3   r
   rF   rO   r   rP   ra   r   r   rR   rS   r   rx   rb   rV   r   s                r\   cancel_asynczMistralJobs.cancel_async  s    & //::J!!H}}X}=H@@
 ''1'"'$(%)* 2%++44! ( 
  e%%22%?00==gu001#%HIL.. --!RE  5**33V__!  %u~% / 
 
 %1CD*6+=+=xHH%5"'"<"<X"FFM//"6-PP%5"'"<"<X"FFM//"6-PPoo<hGG1
$ G Gr   )__name__
__module____qualname__r
   r   intr	   strr   r   r   boolr   r   BatchJobStatusr   rP   r   rT   r]   rc   APIEndpointrx   rz   r}   r   r   r   r    r^   r\   r   r      sK     #&',*/5:49(-@E7<$($(48fH smfH C=	fH
  $fH #3'fH #4S>2fH (1fH  ~fH !f&;&;!<=fH "%"3"34fH SMfH SMfH wsCx01fH 
		fHV  #&',*/5:49(-@E7<$($(48fH smfH C=	fH
  $fH #3'fH #4S>2fH (1fH  ~fH !f&;&;!<=fH "%"3"34fH SMfH SMfH wsCx01fH 
		fHZ (-*/5:')7<$($(48cH #YcH $$	cH
  $cH #3'cH #4S>2cH  }cH "%"3"34cH SMcH SMcH wsCx01cH 
		cHT (-*/5:')7<$($(48cH #YcH $$	cH
  $cH #3'cH #4S>2cH  }cH "%"3"34cH SMcH SMcH wsCx01cH 
		cHR 8=$($(48QH QH "%"3"34	QH
 SMQH SMQH wsCx01QH 
		QHn 8=$($(48QH QH "%"3"34	QH
 SMQH SMQH wsCx01QH 
		QHn 8=$($(48QH QH "%"3"34	QH
 SMQH SMQH wsCx01QH 
		QHn 8=$($(48QH QH "%"3"34	QH
 SMQH SMQH wsCx01QH 
		QHr^   r   N)__doc__basesdkr   r   	mistralair   r   mistralai._hooksr   mistralai.typesr	   r
   mistralai.utilsr   'mistralai.utils.unmarshal_json_responser   typingr   r   r   r   r   r   r   r^   r\   <module>r      s3    G   # ( 3 1 K 5 5eH' eHr^   