o ZhI @sddlZddlmZmZmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd l mZdd l mZdd lmZmZedZ dSdedeeefdeedefddZ dSdedeeefdeedefddZ dSdedeeefdeedefddZ dSdedeeefdeedefddZ dSdedeeefdeedefddZ dSdedeeefdeedefddZ dSdedeeefdeedefdd Z! dSdedeeefdeedefd!d"Z" dSdedeeefdeedefd#d$Z# dSdedeeefdeedefd%d&Z$ dSdedeeefdeedefd'd(Z% dSdedeeefdeedefd)d*Z& dSdedeeefdeedefd+d,Z' dSdedeeefdeedefd-d.Z( dSdedeeefdeedefd/d0Z) dSdedeeefdeedefd1d2Z* dSdedeeefdeedefd3d4Z+ dSdedeeefdeedefd5d6Z, dSdedeeefdeedefd7d8Z- dSdedeeefdeedefd9d:Z. dSdedeeefdeedefd;d<Z/ dSdedeeefdeedefd=d>Z0 dSdedeeefdeedefd?d@Z1 dSdedeeefdeedefdAdBZ2 dSdedeeefdeedefdCdDZ3 dSdedeeefdeedefdEdFZ4 dSdedeeefdeedefdGdHZ5 dSdedeeefdeedefdIdJZ6 dSdedeeefdeedefdKdLZ7 dSdedeeefdeedefdMdNZ8GdOdPdPej9Z:GdQdRdRej9Z;dS)TN)AnyOptionalUnion) urlencode) _api_module)_common) _extra_utils) _transformers)types) BaseApiClient)get_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.batches api_client from_object parent_objectreturncCPi}t|dgdurtdt|dgdurtdt|dgdur&td|SNformatz0format parameter is not supported in Gemini API.gcs_uriz1gcs_uri parameter is not supported in Gemini API. bigquery_uriz6bigquery_uri parameter is not supported in Gemini API.getv ValueErrorrrr to_objectrK/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/batches.py_BatchJobSource_to_mldev"r!cC~i}t|dgdurt|dgt|dgt|dgdur)t|ddgt|dgt|dgdur=t|ddgt|dg|S) NrinstancesFormatr gcsSourceurisrbigquerySourceinputUrirsetvrrrr _BatchJobSource_to_vertex4s r+cCrrrrrrr _BatchJobDestination_to_mldevJr"r,cCr#) NrpredictionsFormatrgcsDestinationoutputUriPrefixrbigqueryDestination outputUrir)rrrr _BatchJobDestination_to_vertex\s   r2cCsFi}t|dgdurt|dgt|dgt|dgdur!td|S)N display_name displayNamedestz.dest parameter is not supported in Gemini API.rr*rrrrr _CreateBatchJobConfig_to_mldevvs r7c Csdi}t|dgdurt|dgt|dgt|dgdur0t|dgt|t|t|dg||S)Nr3r4r5 outputConfig)rr*r2tZt_batch_job_destinationrrrr _CreateBatchJobConfig_to_vertexs r:cCsfi}t|dgdurtdt|dgdurtdt|dgdur1t|dgt|t|dg||S)Nmodelz/model parameter is not supported in Gemini API.srcz-src parameter is not supported in Gemini API.config)rrr*r7rrrr "_CreateBatchJobParameters_to_mldevsr>c Csi}t|dgdurt|dgt|t|dgt|dgdur4t|dgt|t|t|dg|t|dgdurKt|dgt|t|dg||S)Nr;r< inputConfigr=)rr*r9Zt_modelr+Zt_batch_job_sourcer:rrrr #_CreateBatchJobParameters_to_vertexs4 r@cCFi}t|dgdurtdt|dgdur!t|dgt|dg|SNnamez.name parameter is not supported in Gemini API.r=rrr*rrrr _GetBatchJobParameters_to_mldev rEc C^i}t|dgdurt|ddgt|t|dgt|dgdur-t|dgt|dg|SNrC_urlr=rr*r9Zt_batch_job_namerrrr _GetBatchJobParameters_to_vertexrKcCrArBrDrrrr "_CancelBatchJobParameters_to_mldevrFrMc CrGrHrJrrrr #_CancelBatchJobParameters_to_vertex rLrNcCspi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur6td|S)N page_size_querypageSize page_token pageTokenfilterz0filter parameter is not supported in Gemini API.r6rrrr _ListBatchJobsConfig_to_mldevs rUcCsi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur>t|ddgt|dg|S)NrOrPrQrRrSrTr)rrrr _ListBatchJobsConfig_to_vertex7s rVcCs i}t|dgdurtd|S)Nr=z0config parameter is not supported in Gemini API.rrrrr !_ListBatchJobsParameters_to_mldevPsrWcCs6i}t|dgdurt|dgt|t|dg||S)Nr=)rr*rVrrrr "_ListBatchJobsParameters_to_vertex\srXcCrArBrDrrrr "_DeleteBatchJobParameters_to_mldevnrFrYc CrGrHrJrrrr #_DeleteBatchJobParameters_to_vertex}rLrZcCi}|SNrrrrr _JobError_from_mldevr]cCszi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|S)Ndetailscodemessager)rrrr _JobError_from_vertexsrbcCr[r\rrrrr _BatchJobSource_from_mldevr^rccCi}t|dgdurt|dgt|dgt|ddgdur*t|dgt|ddgt|ddgdur?t|dgt|ddg|S) Nr$rr%r&rr'r(rr)rrrr _BatchJobSource_from_vertexs recCr[r\rrrrr _BatchJobDestination_from_mldevr^rfcCrd) Nr-rr.r/rr0r1rr)rrrr _BatchJobDestination_from_vertexs   rgcCr[r\rrrrr _BatchJob_from_mldevr^rhcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurRt|dgt|t|dg|t|dgduret|dgt|dgt|dgdurxt|d gt|dgt|d gdurt|d gt|d gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|t|dg|t|dgdurt|dgt|t|dg||S)NrCr4r3stateerrorZ createTimeZ create_timeZ startTime start_timeZendTimeend_timeZ updateTimeZ update_timer;r?r<r8r5)rr*rbrergrrrr _BatchJob_from_vertexsTrmcCs.i}t|dgdurt|dgt|dg|S)N nextPageTokennext_page_tokenr)rrrr !_ListBatchJobsResponse_from_mldev6srpcsdit|dgdurtdgt|dgt|dgdur0tdgfddt|dgDS)NrnrobatchPredictionJobs batch_jobscsg|]}t|qSr)rm).0itemrrrr Os z6_ListBatchJobsResponse_from_vertex..r))rrrrrur "_ListBatchJobsResponse_from_vertexBs   rwcCr[r\rrrrr _DeleteResourceJob_from_mldevXr^rxcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur?t|dgt|t|dg||S)NrCdonerj)rr*rbrrrr _DeleteResourceJob_from_vertexbsrzc @eZdZdddededeejdejfddZddd edeej dejfd d Z ddd edeej ddfd d Z dej dejfddZddd edeejdejfddZdddededeejdejfddZdddeej deejfddZdS)BatchesNr=r;r<r=rc Cstj|||d}|jjstdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd} t |t rI|dd} n t |drU|durU|j } t|}t|}|jd ||| } |jjrst|j| } nt|j| } tjj| |d } |j| | S Nr;r<r=6This method is only supported in the Vertex AI client.rIrqrP?r= http_optionspostresponsekwargs)r _CreateBatchJobParameters _api_clientvertexairr@get format_maprpop isinstancedicthasattrrrconvert_to_dictencode_unserializable_typesrequestrmrhBatchJob_from_response model_dump_verify_response selfr;r<r=parameter_model request_dictrequest_url_dictpath query_paramsr response_dict return_valuerrr _create|sH         zBatches._createrCc Ctj||d}|jjstdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}t |t rH|d d}n t |d rT|durT|j }t|}t|}|jd |||} |jjrrt|j| } nt|j| } tjj| |d } |j| | S) aGets a batch job. Args: name (str): A fully-qualified BatchJob resource name or ID. Example: "projects/.../locations/.../batchPredictionJobs/456" or "456" when project and location are initialized in the client. Returns: A BatchJob object that contains details about the batch job. Usage: .. code-block:: python batch_job = client.batches.get(name='123456789') print(f"Batch job: {batch_job.name}, state {batch_job.state}") rCr=rrIbatchPredictionJobs/{name}rPrr=Nrrr)r _GetBatchJobParametersrrrrKrrrrrrrrrrrrrmrhrrrr rrCr=rrrrrrrrrrr rsF         z Batches.getc Cstj||d}|jjstdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}t |t rH|d d}n t |d rT|durT|j }t|}t|}|jd |||} dS) aCancels a batch job. Only available for batch jobs that are running or pending. Args: name (str): A fully-qualified BatchJob resource name or ID. Example: "projects/.../locations/.../batchPredictionJobs/123456789" or "123456789" when project and location are initialized in the client. Usage: .. code-block:: python client.batches.cancel(name='123456789') rrrI!batchPredictionJobs/{name}:cancelrPrr=Nrr)r _CancelBatchJobParametersrrrrNrrrrrrrrrrrr rrCr=rrrrrrrrrr cancels6       zBatches.cancelc Cstj|d}|jjstdt|j|}|d}|r!d|}nd}|d}|r3|dt|}| ddd}t |t rG|dd}n t |drS|durS|j }t|}t|}|jd |||}|jjrqt|j|}nt|j|}tjj||d } |j| | S Nr}rrIrqrPrr=rrr)r _ListBatchJobsParametersrrrrXrrrrrrrrrrrrrwrpListBatchJobsResponserrr rr=rrrrrrrrrrr _list4sL        z Batches._listc Cr) aDeletes a batch job. Args: name (str): A fully-qualified BatchJob resource name or ID. Example: "projects/.../locations/.../batchPredictionJobs/456" or "456" when project and location are initialized in the client. Returns: A DeleteResourceJob object that shows the status of the deletion. Usage: .. code-block:: python client.batches.delete(name='123456789') rrrIrrPrr=Nrdeleter)r _DeleteBatchJobParametersrrrrZrrrrrrrrrrrrrzrxDeleteResourceJobrrrrrrr rjsN        zBatches.deletecCst||}|j|||dS)a{Creates a batch job. Args: model (str): The model to use for the batch job. src (str): The source of the batch job. Currently supports GCS URI(-s) or BigQuery URI. Example: "gs://path/to/input/data" or "bq://projectId.bqDatasetId.bqTableId". config (CreateBatchJobConfig): Optional configuration for the batch job. Returns: A BatchJob object that contains details about the batch job. Usage: .. code-block:: python batch_job = client.batches.create( model="gemini-1.5-flash", src="gs://path/to/input/data", ) print(batch_job.state) rr Zformat_destinationrrr;r<r=rrr creates zBatches.createcCs(|durt}td|j|j|d|S)aLists batch jobs. Args: config (ListBatchJobsConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of batch jobs. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python batch_jobs = client.batches.list(config={"page_size": 10}) for batch_job in batch_jobs: print(f"Batch job: {batch_job.name}, state {batch_job.state}") Nrrr})r ListBatchJobsConfigrrrr=rrr lists z Batches.list)__name__ __module__ __qualname__strrr CreateBatchJobConfigOrDictrrGetBatchJobConfigOrDictrCancelBatchJobConfigOrDictrListBatchJobsConfigOrDictrrDeleteBatchJobConfigOrDictrrrrrrrrr r|zsp 9 I ; : P !r|c @r{) AsyncBatchesNr}r;r<r=rc stj|||d}|jjstdt|j|}|d}|r$d|}nd}|d}|r6|dt|}| ddd} t |t rJ|dd} n t |drV|durV|j } t|}t|}|jd ||| IdH} |jjrwt|j| } nt|j| } tjj| |d } |j| | Sr~)r rrrrr@rrrrrrrrrrr async_requestrmrhrrrrrrrr rsJ          zAsyncBatches._createrCc tj||d}|jjstdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd}t |t rI|d d}n t |d rU|durU|j }t|}t|}|jd |||IdH} |jjrvt|j| } nt|j| } tjj| |d } |j| | S) aGets a batch job. Args: name (str): A fully-qualified BatchJob resource name or ID. Example: "projects/.../locations/.../batchPredictionJobs/456" or "456" when project and location are initialized in the client. Returns: A BatchJob object that contains details about the batch job. Usage: .. code-block:: python batch_job = await client.aio.batches.get(name='123456789') print(f"Batch job: {batch_job.name}, state {batch_job.state}") rrrIrrPrr=Nrrr)r rrrrrKrrrrrrrrrrrrrmrhrrrrrrrr r-sH          zAsyncBatches.getc stj||d}|jjstdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd}t |t rI|d d}n t |d rU|durU|j }t|}t|}|jd |||IdH} dS) aCancels a batch job. Only available for batch jobs that are running or pending. Args: name (str): A fully-qualified BatchJob resource name or ID. Example: "projects/.../locations/.../batchPredictionJobs/123456789" or "123456789" when project and location are initialized in the client. Usage: .. code-block:: python await client.aio.batches.cancel(name='123456789') rrrIrrPrr=Nrr)r rrrrrNrrrrrrrrrrrrrrrr rrs8       zAsyncBatches.cancelc stj|d}|jjstdt|j|}|d}|r"d|}nd}|d}|r4|dt|}| ddd}t |t rH|dd}n t |drT|durT|j }t|}t|}|jd |||IdH}|jjrut|j|}nt|j|}tjj||d } |j| | Sr)r rrrrrXrrrrrrrrrrrrrwrprrrrrrrr rsN         zAsyncBatches._listc r) aDeletes a batch job. Args: name (str): A fully-qualified BatchJob resource name or ID. Example: "projects/.../locations/.../batchPredictionJobs/456" or "456" when project and location are initialized in the client. Returns: A DeleteResourceJob object that shows the status of the deletion. Usage: .. code-block:: python await client.aio.batches.delete(name='123456789') rrrIrrPrr=Nrrr)r rrrrrZrrrrrrrrrrrrrzrxrrrrrrrr rsP         zAsyncBatches.deletecs$t||}|j|||dIdHS)awCreates a batch job asynchronously. Args: model (str): The model to use for the batch job. src (str): The source of the batch job. Currently supports GCS URI(-s) or BigQuery URI. Example: "gs://path/to/input/data" or "bq://projectId.bqDatasetId.bqTableId". config (CreateBatchJobConfig): Optional configuration for the batch job. Returns: A BatchJob object that contains details about the batch job. Usage: .. code-block:: python batch_job = await client.aio.batches.create( model="gemini-1.5-flash", src="gs://path/to/input/data", ) rNrrrrr r.s zAsyncBatches.createcs0|dur t}td|j|j|dIdH|S)a(Lists batch jobs asynchronously. Args: config (ListBatchJobsConfig): Optional configuration for the list request. Returns: A Pager object that contains one page of batch jobs. When iterating over the pager, it automatically fetches the next page if there are more. Usage: .. code-block:: python batch_jobs = await client.aio.batches.list(config={'page_size': 5}) print(f"current page: {batch_jobs.page}") await batch_jobs_pager.next_page() print(f"next page: {batch_jobs_pager.page}") Nrrr})r rrrrrrr rMszAsyncBatches.list)rrrrrr rrrrrrrrrrrrrrrrrrrr rsp 9 I ; : P  rr\)<loggingtypingrrr urllib.parserrrr r r9r rr r rrr*Zpagersrr getLoggerloggerrobjectr!r+r,r2r7r:r>r@rErKrMrNrUrVrWrXrYrZr]rbrcrerfrgrhrmrprwrxrzZ BaseModuler|rrrrr sz                          '                                        ?         {