o 0Xxi @s(ddlmZmZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd l m Z dd lmZdd lmZdd lmZmZ dRde deeefdedefddZ dRde deeefdedefddZ dRde deeefdedefddZ dRde deeefdedefddZ dRde deeefdedefddZ dRde deeefdedefddZ dRde deeefdedefddZ dRde deeefdedefd d!Z dRde deeefdedefd"d#Z dRde deeefdedefd$d%Z dRde deeefdedefd&d'Z! dRde deeefdedefd(d)Z" dRde deeefdedefd*d+Z# dRde deeefdedefd,d-Z$ dRde deeefdedefd.d/Z% dRde deeefdedefd0d1Z& dRde deeefdedefd2d3Z' dRde deeefdedefd4d5Z( dRde deeefdedefd6d7Z) dRde deeefdedefd8d9Z* dRde deeefdedefd:d;Z+ dRde deeefdedefdd?Z- dRde deeefdedefd@dAZ. dRde deeefdedefdBdCZ/ dRde deeefdedefdDdEZ0 dRde deeefdedefdFdGZ1 dRde deeefdedefdHdIZ2 dRde deeefdedefdJdKZ3 dRde deeefdedefdLdMZ4GdNdOdOej5Z6GdPdQdQej5Z7d S)S)OptionalUnion) urlencode) _api_module)_common) _extra_utils) _transformers)types) ApiClient)get_value_by_path)set_value_by_path) AsyncPagerPagerN 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_objectrS/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/batches.py_BatchJobSource_to_mldevr cC~i}t|dgdurt|dgt|dgt|dgdur)t|ddgt|dgt|dgdur=t|ddgt|dg|S) NrinstancesFormatr gcsSourceurisrbigquerySourceinputUrirsetvrrrr_BatchJobSource_to_vertex1s r*cCrrrrrrr_BatchJobDestination_to_mldevGr!r+cCr") NrpredictionsFormatrgcsDestinationoutputUriPrefixrbigqueryDestination outputUrir(rrrr_BatchJobDestination_to_vertexYs   r1cCsFi}t|dgdurt|dgt|dgt|dgdur!td|S)N display_name displayNamedestz.dest parameter is not supported in Gemini API.rr)rrrrr_CreateBatchJobConfig_to_mldevss r6c Csdi}t|dgdurt|dgt|dgt|dgdur0t|dgt|t|t|dg||S)Nr2r3r4 outputConfig)rr)r1tt_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)r6rrrr"_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)r8t_modelr*t_batch_job_sourcer:rrrr#_CreateBatchJobParameters_to_vertexs4 rBcCFi}t|dgdurtdt|dgdur!t|dgt|dg|SNnamez.name parameter is not supported in Gemini API.r=rrr)rrrr_GetBatchJobParameters_to_mldev rGc C^i}t|dgdurt|ddgt|t|dgt|dgdur-t|dgt|dg|SNrE_urlr=rr)r8t_batch_job_namerrrr _GetBatchJobParameters_to_vertexrNcCrCrDrFrrrr"_CancelBatchJobParameters_to_mldevrHrPc CrIrJrLrrrr#_CancelBatchJobParameters_to_vertexrOrQcCspi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur6td|S)N page_size_querypageSize page_token pageTokenfilterz0filter parameter is not supported in Gemini API.r5rrrr_ListBatchJobsConfig_to_mldevs rXcCsi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur>t|ddgt|dg|S)NrRrSrTrUrVrWr(rrrr_ListBatchJobsConfig_to_vertex4s rYcCs i}t|dgdurtd|S)Nr=z0config parameter is not supported in Gemini API.rrrrr!_ListBatchJobsParameters_to_mldevMsrZcCs6i}t|dgdurt|dgt|t|dg||S)Nr=)rr)rYrrrr"_ListBatchJobsParameters_to_vertexYsr[cCrCrDrFrrrr"_DeleteBatchJobParameters_to_mldevkrHr\c CrIrJrLrrrr#_DeleteBatchJobParameters_to_vertexzrOr]cCi}|SNrrrrr_JobError_from_mldevr`cCszi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dg|S)Ndetailscodemessager(rrrr_JobError_from_vertexsrecCr^r_rrrrr_BatchJobSource_from_mldevrarfcCi}t|dgdurt|dgt|dgt|ddgdur*t|dgt|ddgt|ddgdur?t|dgt|ddg|S) Nr#rr$r%rr&r'rr(rrrr_BatchJobSource_from_vertexs rhcCr^r_rrrrr_BatchJobDestination_from_mldevraricCrg) Nr,rr-r.rr/r0rr(rrrr _BatchJobDestination_from_vertexs   rjcCr^r_rrrrr_BatchJob_from_mldevrarkcCsi}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)NrEr3r2stateerror createTime create_time startTime start_timeendTimeend_time updateTime update_timer;r?r<r7r4)rr)rerhrjrrrr_BatchJob_from_vertexsTrvcCs.i}t|dgdurt|dgt|dg|S)N nextPageTokennext_page_tokenr(rrrr!_ListBatchJobsResponse_from_mldev3srycsdit|dgdurtdgt|dgt|dgdur0tdgfddt|dgDS)NrwrxbatchPredictionJobs batch_jobscsg|]}t|qSr)rv).0itemrrrr Ls z6_ListBatchJobsResponse_from_vertex..r()rrrrr~r"_ListBatchJobsResponse_from_vertex?s   rcCr^r_rrrrr_DeleteResourceJob_from_mldevUrarcCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur?t|dgt|t|dg||S)NrEdonerm)rr)rerrrr_DeleteResourceJob_from_vertex_src @eZdZdddededeejdejfddZddd edeej dejfd d Z ddd edeej ddfd d Z dej dejfddZddd edeejdejfddZdddededeejdejfddZdddeej deejfddZdS)BatchesNr=r;r<r=rc Cstj|||d}|jjstdt|j|}d|d}|d}|r.|dt|}| ddd}t |t rB|dd}nt |drJ|j }t|}t|}|jd |||} |jjrht|j| } nt|j| } tjj| |d } |j| | S Nr;r<r=6This method is only supported in the Vertex AI client.rzrKrS?r= http_optionspostresponsekwargs)r _CreateBatchJobParameters _api_clientvertexairrB format_mapgetrpop isinstancedicthasattrrrconvert_to_dictencode_unserializable_typesrequestrvrkBatchJob_from_response_verify_response selfr;r<r=parameter_model request_dictpath query_paramsr response_dict return_valuerrr_createysB        zBatches._createrEc Ctj||d}|jjstdt|j|}d|d}|d}|r-|dt|}| ddd}t |t rA|d d}nt |d rI|j }t|}t|}|jd |||}|jjrgt|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}") rEr=rbatchPredictionJobs/{name}rKrSrr=Nrrr)r _GetBatchJobParametersrrrrNrrrrrrrrrrrrrvrkrrr rrEr=rrrrrrrrrrrs@        z Batches.getc Cstj||d}|jjstdt|j|}d|d}|d}|r-|dt|}| ddd}t |t rA|d d}nt |d rI|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') rrz!batchPredictionJobs/{name}:cancelrKrSrr=Nrr)r _CancelBatchJobParametersrrrrQrrrrrrrrrrrr rrEr=rrrrrrrrrcancels4      zBatches.cancelc Cstj|d}|jjstdt|j|}d|d}|d}|r,|dt|}| ddd}t |t r@|dd}nt |drH|j }t|}t|}|jd |||}|jjrft|j|}nt|j|}tjj||d }|j||S NrrrzrKrSrr=rrr)r _ListBatchJobsParametersrrrr[rrrrrrrrrrrrrryListBatchJobsResponserr rr=rrrrrrrrrr_list$sF       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') rrrrKrSrr=Nrdeleter)r _DeleteBatchJobParametersrrrr]rrrrrrrrrrrrrrDeleteResourceJobrrrrrrrUsH       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) rrformat_destinationrrr;r<r=rrrcreates zBatches.createcCstd|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}") r{r)rrrr=rrrlists  z Batches.list)__name__ __module__ __qualname__strrr CreateBatchJobConfigOrDictrrGetBatchJobConfigOrDictrCancelBatchJobConfigOrDictrListBatchJobsConfigOrDictrrDeleteBatchJobConfigOrDictrrrrrrrrrrwsp 4 D 8 5 K !rc @r) AsyncBatchesNrr;r<r=rc stj|||d}|jjstdt|j|}d|d}|d}|r/|dt|}| ddd}t |t rC|dd}nt |drK|j }t|}t|}|jd |||IdH} |jjrlt|j| } nt|j| } tjj| |d } |j| | Sr)r rrrrrBrrrrrrrrrrr async_requestrvrkrrrrrrrrsD         zAsyncBatches._createrEc tj||d}|jjstdt|j|}d|d}|d}|r.|dt|}| ddd}t |t rB|d d}nt |d rJ|j }t|}t|}|jd |||IdH}|jjrkt|j|}nt|j|}tjj||d } |j| | Sr)r rrrrrNrrrrrrrrrrrrrvrkrrrrrrrr sB         zAsyncBatches.getc stj||d}|jjstdt|j|}d|d}|d}|r.|dt|}| ddd}t |t rB|d d}nt |d rJ|j }t|}t|}|jd |||IdH}dSr)r rrrrrQrrrrrrrrrrrrrrrrrLs6      zAsyncBatches.cancelc stj|d}|jjstdt|j|}d|d}|d}|r-|dt|}| ddd}t |t rA|dd}nt |drI|j }t|}t|}|jd |||IdH}|jjrjt|j|}nt|j|}tjj||d }|j||Sr)r rrrrr[rrrrrrrrrrrrrryrrrrrrrrsH        zAsyncBatches._listc rr)r rrrrr]rrrrrrrrrrrrrrrrrrrrrrsJ        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", ) rNrrrrrrs zAsyncBatches.createcs 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}") r{rN)rrrrrrrszAsyncBatches.list)rrrrrr rrrrrrrrrrrrrrrrrrrrrsp 4 D 8 5 K  rr_)8typingrr urllib.parserrrrr r8r rr r rr r)pagersrrrobjectr r*r+r1r6r:r>rBrGrNrPrQrXrYrZr[r\r]r`rerfrhrirjrkrvryrrr BaseModulerrrrrrsv                         '                                        ?         b