o i^q@sdZddlmZddlmZddlmZmZddlmZddl m Z m Z ddl m Z dd lmZdd lmZmZmZmZmZGd d d eZd S)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.)BaseSDK)datetime)modelsutils) HookContext)OptionalNullableUNSET)get_security_from_env)unmarshal_json_response)AnyDictListMappingOptionalc@seZdZddeeeedeedddd deedeedeed eed eeee fd ee d ee d ee e jdeejdeedeedeeeefde jfddZddeeeedeedddd deedeedeed eed eeee fd ee d ee d ee e jdeejdeedeedeeeefde jfddZeeededdddde ede jdeed eed eeeefdeedeejdeedeedeeeefde jfddZeeededdddde ede jdeed eed eeeefdeedeejdeedeedeeeefde jfddZedddd d!edeejdeedeedeeeefde jf d"d#Zedddd d!edeejdeedeedeeeefde jf d$d%Zedddd d!edeejdeedeedeeeefde jf d&d'Zedddd d!edeejdeedeedeeeefde jf d(d)ZdS)* MistralJobsrdFN) page page_sizemodelagent_idmetadata created_after created_by_mestatusretries server_url timeout_ms http_headersrrrrrrrrrrrrreturnc CsTd} d}| dur |jj} | dur| } n|| |} tj||||||||d}|jdd| ||ddddd| |jj| d }| tkrH|jjturH|jj} d}t | t j rV| gd f}|j t |j| p^d d gt|jjtjd |ddg|d}t |ddr~ttj|St |ddrt |}td||t |ddrt |}td||td|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)rrrrrrrrGET/v1/batch/jobsFT user-agentapplication/json methodpathbase_url url_variablesrequestrequest_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuersecurityr429500502503504$jobs_api_routes_batch_get_batch_jobsconfigr( operation_id oauth2_scopessecurity_source4XX5XXhook_ctxr*error_status_codes retry_config200*API error occurredUnexpected response received)sdk_configurationr_get_urlr%JobsAPIRoutesBatchGetBatchJobsRequest_build_requestr0r rC isinstancer RetryConfig do_requestrr Securitymatch_responser BatchJobsOutstream_to_textSDKErrorselfrrrrrrrrrrrrr(r)r*reqrChttp_res http_res_textrY\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/mistral_jobs.pylistsz!           zMistralJobs.listc shd} d}| dur |jj} | dur| } n|| |} tj||||||||d}|jdd| ||ddddd| |jj| d }| tkrI|jjturI|jj} d}t | t j rW| gd f}|j t |j| p_d d gt|jjtjd |ddg|dIdH}t |ddrttj|St |ddrt |IdH}td||t |ddrt |IdH}td||td|r )rHrrIrrJ_build_request_asyncr0r rCrLrrMdo_request_asyncrr rOrPr rQstream_to_text_asyncrSrTrYrYrZ list_asyncvs|!          zMistralJobs.list_async)rrr timeout_hoursrrrr input_filesendpointrac sZd} d} | dur |jj} |dur|} n|| | } tj||||||d|jdd| | ddddd| |jjfd d | d } |tkrK|jjturK|jj}d}t |t j rY|gd f}|j t |j| pad dgt|jjtjd| ddg|d}t |ddrttj|St |ddrt |}td||t |ddrt |}td||td|)QCreate 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. NrbrcrrrraPOSTr"TFr#r$ctdddtjSNFjsonrserialize_request_bodyr BatchJobInrYr*rYrZ z$MistralJobs.create..r&r'r(r)r*r+r,r-r.r/rr0get_serialized_bodyrr1r7&jobs_api_routes_batch_create_batch_jobr9r>r?r@rDrErFrG)rHrrIrrlrKr0r rCrLrrMrNrr rOrPr BatchJobOutrRrSrUrbrcrrrrarrrrr(r)rVrCrWrXrYrmrZcreatesx            zMistralJobs.createc snd} d} | dur |jj} |dur|} n|| | } tj||||||d|jdd| | ddddd| |jjfd d | d } |tkrL|jjturL|jj}d}t |t j rZ|gd f}|j t |j| pbd dgt|jjtjd| ddg|dIdH}t |ddrttj|St |ddrt |IdH}td||t |ddrt |IdH}td||td|)rdNrerfr"TFr#r$crgrhrjrYrmrYrZrnroz*MistralJobs.create_async..rpr1r7rrr9r>r?r@rDrErFrG)rHrrIrrlr\r0r rCrLrrMr]rr rOrPr rsr^rSrtrYrmrZ create_asyncCsz           zMistralJobs.create_async)rrrrjob_idc CFd}d}|dur |jj}|dur|}n|||}tj|d}|jdd|||ddddd||jj|d } |tkrA|jjturA|jj}d} t |t j rO|gd f} |j t |j|pWd d gt|jjtjd | ddg| d} t | ddrwttj| St | ddrt | } td| | t | ddrt | } td| | td| aGet 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. Nrwr!z/v1/batch/jobs/{job_id}FTr#r$r%r1r7#jobs_api_routes_batch_get_batch_jobr9r>r?r@rDrErFrG)rHrrIr$JobsAPIRoutesBatchGetBatchJobRequestrKr0r rCrLrrMrNrr rOrPr rsrRrS rUrwrrrrr(r)r*rVrCrWrXrYrYrZgetl          zMistralJobs.getc Zd}d}|dur |jj}|dur|}n|||}tj|d}|jdd|||ddddd||jj|d } |tkrB|jjturB|jj}d} t |t j rP|gd f} |j t |j|pXd d gt|jjtjd | ddg| dIdH} t | ddr{ttj| St | ddrt | IdH} td| | t | ddrt | IdH} td| | td| ry)rHrrIrr|r\r0r rCrLrrMr]rr rOrPr rsr^rSr}rYrYrZ get_asyncn         zMistralJobs.get_asyncc CrxaCancel 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. Nrzrfz/v1/batch/jobs/{job_id}/cancelFTr#r$r%r1r7&jobs_api_routes_batch_cancel_batch_jobr9r>r?r@rDrErFrG)rHrrIr'JobsAPIRoutesBatchCancelBatchJobRequestrKr0r rCrLrrMrNrr rOrPr rsrRrSr}rYrYrZcancelNrzMistralJobs.cancelc rr)rHrrIrrr\r0r rCrLrrMr]rr rOrPr rsr^rSr}rYrYrZ cancel_asyncrzMistralJobs.cancel_async)__name__ __module__ __qualname__r rintrstrr r rboolrrBatchJobStatusrrMrrQr[r_ APIEndpointrsrurvr~rrrrYrYrYrZr s      k      m     j     i W W WrN)__doc__basesdkrr mistralairrmistralai._hooksrmistralai.typesrr mistralai.utilsr 'mistralai.utils.unmarshal_json_responser typingr r rrrrrYrYrYrZs