`i ddlZddlmZmZmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd l mZdd l mZdd lmZmZej.dZ d&dedeeefdeedefdZ d&dedeeefdeedefdZ d&dedeeefdeedefdZ d&dedeeefdeedefdZ d&dedeeefdeedefdZ d&dedeeefdeedefdZ d&dedeeefdeedefdZ! d&dedeeefdeedefdZ" d&dedeeefdeedefdZ# d&dedeeefdeedefdZ$ d&dedeeefdeedefdZ% d&dedeeefdeedefdZ& d&dedeeefdeedefdZ' d&dedeeefdeedefd Z( d&dedeeefdeedefd!Z)Gd"d#ejTZ+Gd$d%ejTZ,y)'N)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_objectreturnci}t|dgt|dgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dg|S) NformatinstancesFormatgcs_uri gcsSourceuris bigquery_uribigquerySourceinputUrigetvsetvrrr to_objects Y/var/www/html/ts-aiml.yecor.com/venv/lib/python3.12/site-packages/google/genai/batches.py_BatchJobSource_to_vertexr%"s !) +z".&'kH:)FG + {#/[&)4 i[+IJ +'(4 :& [>*+ ci}t|dgt|dgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dg|S) NrpredictionsFormatrgcsDestinationoutputUriPrefixrbigqueryDestination outputUrirr"s r$_BatchJobDestination_to_vertexr-8s !) +z".()4 hZ+HI + {#/ ,- [9+&  +'(4  , [>*+ r&ci}t|dgt|dgt|dgt|dg8t|dgt|tj|t|dg||S)N display_name displayNamedest outputConfig)r r!r-tt_batch_job_destinationr"s r$_CreateBatchJobConfig_to_vertexr5Rs !) +'(4kN;K)LM +x , &  % %j${VH2M N   r&cji}t|dg-t|dgtj|t|dgt|dg8t|dgt |tj |t|dg|t|dg$t|dgt |t|dg||S)Nmodelsrc inputConfigconfig)r r!r3t_modelr%t_batch_job_sourcer5r"s r$#_CreateBatchJobParameters_to_vertexr=js !) +y!-   *d; :;  +w+ !  T+w-G H   +z".  ' [8*5y  r&c i}t|dg.t|ddgtj|t|dgt|dgt|dgt|dg|SNname_urlr:r r!r3t_batch_job_namer"s r$ _GetBatchJobParameters_to_vertexrDr !) +x ,  :tK&'BC  +z".XJ[8* => r&c i}t|dg.t|ddgtj|t|dgt|dgt|dgt|dg|Sr?rBr"s r$#_CancelBatchJobParameters_to_vertexrGrEr&ci}t|dgt|ddgt|dgt|dgt|ddgt|dgt|dgt|ddgt|dg|S)N page_size_querypageSize page_token pageTokenfilterrr"s r$_ListBatchJobsConfig_to_vertexrOs !) + }%1*-tK+/O + ~&2 ; [<.)  +z".8,d; .KL r&c li}t|dg$t|dgt|t|dg||S)Nr:)r r!rOr"s r$"_ListBatchJobsParameters_to_vertexrQsH !) +z".  & [8*5y  r&c i}t|dg.t|ddgtj|t|dgt|dgt|dgt|dg|Sr?rBr"s r$#_DeleteBatchJobParameters_to_vertexrSrEr&ci}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg|S)Ndetailscodemessagerr"s r$_JobError_from_vertexrXs !) + {#/YKkI;!?@ +x ,VHd;9: + {#/YKkI;!?@ r&c i}t|dgt|dgt|dgt|ddgt|dgt|ddgt|ddgt|dgt|ddg|S) Nrrrrrrrrrr"s r$_BatchJobSource_from_vertexrZs !) +)*+7XJ[3D2E FG + V,-9YKkK3H!IJ +(*56B  [+Z89 r&c i}t|dgt|dgt|dgt|ddgt|dgt|ddgt|ddgt|dgt|ddg|S) Nr(rr)r*rr+r,rrr"s r$ _BatchJobDestination_from_vertexr\s !) ++,-9XJ[3F2G HI +(*;<=I   [+->?@  +-{;<H  [0+>? r&c i}t|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dgt|dg$t|dgt|t|dg|t|dgt|dgt|dgt|dgt|d gt|dgt|d gt|d gt|d gt|d gt|d gt|d gt|dgt|dgt|dgt|dg$t|dgt|t|dg|t|dg$t|dgt |t|dg||S)Nr@r0r/stateerror createTime create_time startTime start_timeendTimeend_time updateTime update_timer7r9r8r2r1)r r!rXrZr\r"s r$_BatchJob_from_vertexrh4s !) +x ,VHd;9: + '3^$d;&HI +y!-WItK';< +y!-   [7)4i  + ~&2]OT+ ~%FG + }%1\ND{m$DE + {#/ZL${YK"@A + ~&2]OT+ ~%FG +y!-WItK';< + '3 # [=/:I  +'(4 ( [>*:;Y  r&c i}t|dgt|dgt|dgt|dg2t|dgt|dgDcgc]}t|||c}|Scc}w)N nextPageTokennext_page_tokenbatchPredictionJobs batch_jobs)r r!rh)rrrr#items r$"_ListBatchJobsResponse_from_vertexrops !) +()5&'kO;L)MN +-./; [+@*AB  "*dI >    s A* c i}t|dgt|dgt|dgt|dgt|dgt|dgt|dg$t|dgt|t|dg||S)Nr@doner_)r r!rXr"s r$_DeleteResourceJob_from_vertexrrs !) +x ,VHd;9: +x ,VHd;9: +y!-   [7)4i  r&c eZdZdddededeej dejfdZdddedeejdejfd Z dddedeejddfd Z dejdejfd Zdddedeej dej"fd Zdddededeej dejfd ZdddeejdeejfdZy)BatchesNr:r7r8r:rcLtj|||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd} |j,|jj|jj} tj|}tj|}|jjd||| } |jjrt!|j| } tj"j%| |j' } |jj)| | S Nr7r8r:6This method is only supported in the Vertex AI client.rArlrJ?r:postresponsekwargs)r _CreateBatchJobParameters _api_clientvertexai ValueErrorr=get format_maprpopr: http_optionsr convert_to_dictencode_unserializable_typesrequestrhBatchJob_from_response model_dump_verify_response selfr7r8r:parameter_model request_dictrequest_url_dictpath query_paramsr response_dict return_values r$_createzBatches._creates55 O    $ $ O PP8   Ol&))&1 $//0@A$##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL$$,,lLM   +D,<,>00'A'A'C1L %%l3 r&r@cJtj||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||} |jjrt!|j| } tj"j%| |j' } |jj)| | 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}") r@r:ryrAbatchPredictionJobs/{name}rJrzr:Nrr|)r _GetBatchJobParametersrrrrDrrrrr:rr rrrrhrrrr rr@r:rrrrrrrrs r$rz Batches.gets*22 O    $ $ O PP5   Ol&))&1 +667GH+##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL$$,, t\<M   +D,<,>00'A'A'C1L %%l3 r&c\tj||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||} y) 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') rryrA!batchPredictionJobs/{name}:cancelrJrzr:Nr{)r _CancelBatchJobParametersrrrrGrrrrr:rr rrr rr@r:rrrrrrrs r$cancelzBatches.cancels$,55 O    $ $ O PP8   Ol&))&1 2==>NO2##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL$$,,lLMr&cHtj|}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd|||}|jjrt!|j|}tj"j%||j' } |jj)| | S NruryrArlrJrzr:rr|)r _ListBatchJobsParametersrrrrQrrrrr:rr rrrroListBatchJobsResponserrr rr:rrrrrrrrs r$_listz Batches._listWs44O    $ $ O PP7   Ol&))&1 $//0@A$##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL$$,, t\<M   8   Mm..=='A'A'C>L %%l3 r&cJtj||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||} |jjrt!|j| } tj"j%| |j' } |jj)| | S) 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') rryrArrJrzr:Ndeleter|)r _DeleteBatchJobParametersrrrrSrrrrr:rr rrrrrDeleteResourceJobrrrrs r$rzBatches.deletes.55 O    $ $ O PP8   Ol&))&1 +667GH+##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL$$,,$ lM   4   Mm**99'A'A'C:L %%l3 r&cVtj||}|j|||S)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) rxr format_destinationrrr7r8r:s r$createzBatches.creates+: , ,S& 9F <>00'A'A'C1L %%l3 sD1F23F04A=F2r@cfKtj||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{} |jjrt!|j| } tj"j%| |j' } |jj)| | S7|w) 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}") rryrArrJrzr:Nrr|)r rrrrrDrrrrr:rr rrrrhrrrrrs r$rzAsyncBatches.getIs*22 O    $ $ O PP5   Ol&))&1 +667GH+##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL**88 t\<M   +D,<,>00'A'A'C1L %%l3 D0F12F/3A=F1cxKtj||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{} y7w) 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') rryrArrJrzr:Nr{)r rrrrrGrrrrr:rr rrrrs r$rzAsyncBatches.cancels2,55 O    $ $ O PP8   Ol&))&1 2==>NO2##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL**88lLMsD0D:2D83D:cdKtj|}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd|||d{}|jjrt!|j|}tj"j%||j' } |jj)| | S7|wr)r rrrrrQrrrrr:rr rrrrorrrrrs r$rzAsyncBatches._lists44O    $ $ O PP7   Ol&))&1 $//0@A$##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL**88 t\<M   8   Mm..=='A'A'C>L %%l3 sD/F01F.2A=F0cfKtj||}|jjs t dt |j|}|j d}|rdj|}nd}|j d}|r|dt|}|jddd}|j,|jj|jj}tj|}tj|}|jjd |||d{} |jjrt!|j| } tj"j%| |j' } |jj)| | S7|w) 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') rryrArrJrzr:Nrr|)r rrrrrSrrrrr:rr rrrrrrrrrrs r$rzAsyncBatches.deletes.55 O    $ $ O PP8   Ol&))&1 +667GH+##H-LvQy./ 0dXt$04L*  " " / / ;$++88l**<8L66|DL**88$ lM   4   Mm**99'A'A'C:L %%l3 rcrKtj||}|j|||d{S7w)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", ) rxNrrs r$rzAsyncBatches.createDs58 , ,S& 9FEs6B BB Bs .757cK|tj}td|j|j|d{|S7 w)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}") Nrmru)r rrrrs r$rzAsyncBatches.listcsK*~((*f  jjj''   (s=A A A )rrrrrr rrrrrrrrrrrrrrrrrr&r$rrs<@ 55  5 u778 5  ~~ 5pMQBB"*5+H+H"IB ~~BP<@ : :u778 :  :x1661 ""1n<@ F Fu778 F   FZ<@ CC  C u778 C  ~~ C@DH ? ?@%..!r&r)N)-loggingtypingrrr urllib.parserrr r r r3r rr rr rr!pagersrr getLoggerloggerdictobjectr%r-r5r=rDrGrOrQrSrXrZr\rhrorr BaseModulertrrr&r$rs$''" &..%   1 2 %)tV|$D> 2%)tV|$D> :%)tV|$D> 6%)!!tV|$!D>! !N%)tV|$D> ,%)tV|$D> ,%)tV|$D> 8%)tV|$D> *%)tV|$D> ,%)tV|$D> *%)tV|$D> 2%)tV|$D> :%)99tV|$9D>9 9~%)tV|$D> 2%)tV|$D> 0ok$$od o;))or&