o i& @s ddlZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd l mZdd l mZdd l mZddlmZmZedZ drdeeeefefdeeeefdeeeffddZ drdeeeefefdeeeefdeeeffddZ drdeeeefefdeeeefdeeeffddZ! drdeeeefefdeeeefdeeeffddZ" drdedeeeefefdeeeefdeeeffddZ# drdeeeefefdeeeefdeeeffddZ$ drdeeeefefdeeeefdeeeffd d!Z% drdeeeefefdeeeefdeeeffd"d#Z& drdeeeefefdeeeefdeeeffd$d%Z' drdedeeeefefdeeeefdeeeffd&d'Z( drdedeeeefefdeeeefdeeeffd(d)Z) drdeeeefefdeeeefdeeeffd*d+Z* drdeeeefefdeeeefdeeeffd,d-Z+ drdeeeefefdeeeefdeeeffd.d/Z, drdeeeefefdeeeefdeeeffd0d1Z- drdeeeefefdeeeefdeeeffd2d3Z. drdedeeeefefdeeeefdeeeffd4d5Z/ drdedeeeefefdeeeefdeeeffd6d7Z0 drdeeeefefdeeeefdeeeffd8d9Z1 drdedeeeefefdeeeefdeeeffd:d;Z2 drdedeeeefefdeeeefdeeeffdd?Z4 drdeeeefefdeeeefdeeeffd@dAZ5 drdeeeefefdeeeefdeeeffdBdCZ6 drdedeeeefefdeeeefdeeeffdDdEZ7 drdeeeefefdeeeefdeeeffdFdGZ8 drdedeeeefefdeeeefdeeeffdHdIZ9 drdeeeefefdeeeefdeeeffdJdKZ: drdedeeeefefdeeeefdeeeffdLdMZ; drdeeeefefdeeeefdeeeffdNdOZ< drdedeeeefefdeeeefdeeeffdPdQZ= drdedeeeefefdeeeefdeeeffdRdSZ> drdeeeefefdeeeefdeeeffdTdUZ? drdeeeefefdeeeefdeeeffdVdWZ@ drdedeeeefefdeeeefdeeeffdXdYZA drdeeeefefdeeeefdeeeffdZd[ZB drdeeeefefdeeeefdeeeffd\d]ZC drdeeeefefdeeeefdeeeffd^d_ZD drdeeeefefdeeeefdeeeffd`daZE drdeeeefefdeeeefdeeeffdbdcZF drdeeeefefdeeeefdeeeffdddeZG drdeeeefefdeeeefdeeeffdfdgZH drdeeeefefdeeeefdeeeffdhdiZI drdeeeefefdeeeefdeeeffdjdkZJ drdeeeefefdeeeefdeeeffdldmZKGdndodoe jLZMGdpdqdqe jLZNdS)sN)AnyOptionalUnion) urlencode) _api_module)_common) _extra_utils) _transformers)types) BaseApiClient)get_value_by_path)move_value_by_path)set_value_by_path) AsyncPagerPagerzgoogle_genai.batches from_object parent_objectreturncsit|dgdurtdgt|dgt|ddgdur1tdgfddt|ddgDt|ddgdurKtdgd dt|ddgDS) N responsesFile file_nameinlinedResponsesinlined_responsescg|]}t|qS)_InlinedResponse_from_mldev.0item to_objectrZ/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/google/genai/batches.py 3z3_BatchJobDestination_from_mldev..inlinedEmbedContentResponsesinlined_embed_content_responsescSg|]}|qSrrrrrr!r"Bgetvsetvrrrrr!_BatchJobDestination_from_mldev&s6   r,cCi}t|dgdurt|dgt|dgt|ddgdur*t|dgt|ddgt|ddgdur?t|dgt|ddg|S) NpredictionsFormatformatgcsDestinationoutputUriPrefixgcs_uribigqueryDestination outputUri bigquery_urir(rrr rrr! _BatchJobDestination_from_vertexNs   r7cCsi}t|dgdurt|dgt|dgt|dgdur)t|ddgt|dgt|dgdur=t|ddgt|dgt|d gdurItd t|d gdurUtd t|d gduratd|S)Nr/r.r2r0r1r5r3r4r2file_name parameter is not supported in Vertex AI.rz:inlined_responses parameter is not supported in Vertex AI.r%zHinlined_embed_content_responses parameter is not supported in Vertex AI.r)r* ValueErrorr6rrr!_BatchJobDestination_to_vertexgs4  r;cCr-) NinstancesFormatr/ gcsSourceurisr2bigquerySourceinputUrir5r(r6rrr!_BatchJobSource_from_vertexs rA api_clientcsit|dgdurtdt|dgdurtdt|dgdur&tdt|dgdur9tdgt|dgt|d gdurUtd d gfd d t|d gDS) Nr/z0format parameter is not supported in Gemini API.r2z1gcs_uri parameter is not supported in Gemini API.r5z6bigquery_uri parameter is not supported in Gemini API.rfileNameinlined_requestsrequestscsg|]}t|qSr)_InlinedRequest_to_mldevrrBr rr!r"s z,_BatchJobSource_to_mldev..r)r:r*rBrrrrGr!_BatchJobSource_to_mldevs$   rJcCsi}t|dgdurt|dgt|dgt|dgdur)t|ddgt|dgt|dgdur=t|ddgt|dgt|d gdurItd t|d gdurUtd |S) Nr/r<r2r=r>r5r?r@rr8rDz9inlined_requests parameter is not supported in Vertex AI.r9r6rrr!_BatchJobSource_to_vertexs$ rKc Csfi}t|dgdurt|dgt|dgt|ddgdur*t|dgt|ddgt|ddgdurBt|dgtt|ddgt|ddgdurWt|dgt|ddgt|ddgdurlt|d gt|ddgt|dd gdurt|d gt|dd gt|dd gdurt|d gt|dd gt|dd gdurt|dgttt|dd g||S)Nnamemetadata displayName display_namestate createTime create_timeendTimeend_time updateTime update_timemodeloutputdest)r)r*t t_job_stater,t_recv_batch_job_destinationr6rrr!_BatchJob_from_mldevsV     r]c Csi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur>t|dgtt|dgt|dgdurQt|dgt|dgt|dgdurdt|dgt|dgt|dgdurwt|d gt|dgt|d gdurt|d gt|d gt|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgtt|dg|t|dgdurt|dgttt|dg|t|dgdurt|dgt|dg|S)NrLrNrOrPerrorrQrR startTime start_timerSrTrUrVrW inputConfigsrc outputConfigrYcompletionStatscompletion_stats)r)r*rZr[rAr7r\r6rrr!_BatchJob_from_vertexsR  rfcCsli}t|dgdurt|dgt|dgt|dgdur!tdt|dgdur4t|dgt|dg|S)NdatarO6display_name parameter is not supported in Gemini API. mime_typemimeTyper9r6rrr!_Blob_to_mldevSrkc C8i}t|dgdurt|ddgt|t|dg|SNrL_urlr)r*rZt_batch_job_namerBrrr rrr!"_CancelBatchJobParameters_to_mldevdrsc Crmrnrprrrrr!#_CancelBatchJobParameters_to_vertextrtrucCsi}t|dgdurt|dgt|dgt|dgdur+t|dgtt|dg|t|dgdur>t|dgt|dgt|dgdurQt|dgt|dgt|dgdurdt|d gt|dgt|d gdurwt|d gt|d gt|d gdurt|d gt|d gt|d gdurt|dgt|d gt|dgdurt|dgddt|dgDt|dgdurt|dgt|dg|S)NcontentcitationMetadatacitation_metadata tokenCount token_count finishReason finish_reason avgLogprobs avg_logprobsgroundingMetadatagrounding_metadataindexlogprobsResultlogprobs_result safetyRatingssafety_ratingscSr&rrrrrr!r"z)_Candidate_from_mldev..urlContextMetadataurl_context_metadata)r)r*_CitationMetadata_from_mldevr6rrr!_Candidate_from_mldevsP   rcCs8i}t|dgdurt|dgddt|dgD|S)NcitationSources citationscSr&rrrrrr!r"rz0_CitationMetadata_from_mldev..r(r6rrr!rsrcsbit|dgdurtdgfddt|dgDt|dgdur/tdgt|dgS)Npartscrr)_Part_to_mldevrrrr!r"r#z%_Content_to_mldev..roler(r+rrr!_Content_to_mldevs   rcCsHi}t|dgdurt|ddgt|dgt|dgdur"td|S)NrObatchrNrYz.dest parameter is not supported in Gemini API.r9r6rrr!_CreateBatchJobConfig_to_mldevs rc Cs`i}t|dgdurt|dgt|dgt|dgdur.t|dgttt|dg||S)NrOrNrYrc)r)r*r;rZt_batch_job_destinationr6rrr!_CreateBatchJobConfig_to_vertexsrc Csi}t|dgdurt|ddgt|t|dgt|dgdur6t|ddgt|t|t|dg|t|dgdurGtt|dg||SNrWrorbrraconfig)r)r*rZt_modelrJt_batch_job_sourcerrrrrr!"_CreateBatchJobParameters_to_mldevs( rc Csi}t|dgdurt|dgt|t|dgt|dgdur3t|dgtt|t|dg|t|dgdurDtt|dg||S)NrWrbrar)r)r*rZrrKrrrrrrr!#_CreateBatchJobParameters_to_vertex&s& rcCs0i}t|dgdurt|ddgt|dg|S)NrOrrNr(r6rrr!(_CreateEmbeddingsBatchJobConfig_to_mldevCs rc Csi}t|dgdurt|ddgt|t|dgt|dgdur2t|ddgt|t|dg|t|dgdurCtt|dg||Sr)r)r*rZr"_EmbeddingsBatchJobSource_to_mldevrrrrrr!,_CreateEmbeddingsBatchJobParameters_to_mldevSs( rc Crmrnrprrrrr!"_DeleteBatchJobParameters_to_mldevqrtrc Crmrnrprrrrr!#_DeleteBatchJobParameters_to_vertexrtrcCi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurNt|dgt|dg|SNsdkHttpResponsesdk_http_responserLdoner^r(r6rrr!_DeleteResourceJob_from_mldevrcCrrr(r6rrr!_DeleteResourceJob_from_vertexrrc Cs|i}t|dgdurt|gdddt|t|dgDt|dgdur.r_selfz requests[].*zrequests[].request.*)r)r*rZt_contents_for_embed_EmbedContentConfig_to_mldevmovevrrrrr!_EmbedContentBatch_to_mldevs&  rcCsi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur>t|ddgt|dgt|dgdurJtdt|d gdurVtd |S) N task_typertaskTypetitleoutput_dimensionalityoutputDimensionalityriz3mime_type parameter is not supported in Gemini API. auto_truncatez7auto_truncate parameter is not supported in Gemini API.r9r6rrr!rs(  rcCs\i}t|dgdurt|dgt|dgt|dgdur,t|dgt|t|dg||S)NrrDrE)r)r*rrrrrr!rsrcCsli}t|dgdurtdt|dgdur!t|dgt|dgt|dgdur4t|dgt|dg|S)NrOrhfile_urifileUririrjrHr6rrr!_FileData_to_mldevsrc s^it|dgdurt|dgttt|dgt|dgdur.tdgt|dgt|dgdurAtdgt|dgt|dgdurTtdgt|dgt|dgdurgtd gt|dgt|d gdurztd gt|d gt|d gdurtd gt|d gt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgt|dgt|dgdurtdgtt|dgt|dgdur+tdgt|dgt|dgdur8tdt|dgdurEtdt|d gdur`t|d!gfd"d#t|d gDt|d$gdurt|d$gfd%d#tt|d$gDt|d&gdurt|d'gt|d&gt|d(gdurtd)t|d*gdurt|d+gtt|d*gt|d,gdurtd-gt|d,gt|d.gdurtd/gt|d.gt|d0gdurtd1gt t|d0gt|d2gdurtd3t|d4gdurtd5gt|d4gt|d6gdur-td7gt|d6gS)8Nsystem_instructionsystemInstruction temperaturetop_ptopPtop_ktopKcandidate_countcandidateCountmax_output_tokensmaxOutputTokensstop_sequences stopSequencesresponse_logprobsresponseLogprobslogprobspresence_penaltypresencePenaltyfrequency_penaltyfrequencyPenaltyseedresponse_mime_typeresponseMimeTyperesponse_schemaresponseSchemaresponse_json_schemaresponseJsonSchemarouting_configz8routing_config parameter is not supported in Gemini API.model_selection_configz@model_selection_config parameter is not supported in Gemini API.safety_settingssafetySettingscrr)_SafetySetting_to_mldevrrrr!r"}r#z3_GenerateContentConfig_to_mldev..toolscsg|] }tt|qSr)_Tool_to_mldevrZt_toolrrGrr!r"s tool_config toolConfiglabelsz0labels parameter is not supported in Gemini API.cached_content cachedContentresponse_modalitiesresponseModalitiesmedia_resolutionmediaResolution speech_config speechConfigaudio_timestampz9audio_timestamp parameter is not supported in Gemini API.thinking_configthinkingConfig image_config imageConfig) r)r*rrZ t_contentt_schemar:t_toolst_cached_content_namet_speech_configrIrrGr!_GenerateContentConfig_to_mldevs           rcsit|dgdurtdgt|dgt|dgdur/tdgfddt|dgDt|dgdurBtdgt|dgt|dgdurUtd gt|dgt|d gdurhtd gt|d gt|d gdur{td gt|d gS)Nrr candidatescrr)rrrrr!r"r#z7_GenerateContentResponse_from_mldev.. modelVersion model_versionpromptFeedbackprompt_feedback responseId response_id usageMetadatausage_metadatar(r+rrr!#_GenerateContentResponse_from_mldevs,   rc Crmrnrprrrrr!_GetBatchJobParameters_to_mldevrtrc Crmrnrprrrrr! _GetBatchJobParameters_to_vertexrtrcCsFi}t|dgdurtdt|dgdur!t|dgt|dg|S)N auth_configz5auth_config parameter is not supported in Gemini API. enable_widget enableWidgetrHr6rrr!_GoogleMaps_to_mldevs rcCs^i}t|dgdurtdt|dgdurtdt|dgdur-t|dgt|dg|S)Nexclude_domainsz9exclude_domains parameter is not supported in Gemini API.blocking_confidencez=blocking_confidence parameter is not supported in Gemini API.time_range_filtertimeRangeFilterrHr6rrr!_GoogleSearch_to_mldevsrc sit|dgdurtddgt|t|dgt|dgdur8tddgfddtt|dgDt|dgdurKtdgt|dgt|dgduritddgt|t|dgtdgid S) NrWrrcrr)rrrrr!r"6r#z,_InlinedRequest_to_mldev..rMrgenerationConfig) default_value)r)r*rZr t_contentsrrIrrr!rF%s8    rFcCsZi}t|dgdurt|dgtt|dg|t|dgdur+t|dgt|dg|S)Nresponser^)r)r*rr6rrr!rMs rcCspi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur6td|S)N page_size_querypageSize page_token pageTokenfilterz0filter parameter is not supported in Gemini API.r9r6rrr!_ListBatchJobsConfig_to_mldevas rcCsi}t|dgdurt|ddgt|dgt|dgdur*t|ddgt|dgt|dgdur>t|ddgt|dg|S)Nr r r rrrr(r6rrr!_ListBatchJobsConfig_to_vertexys rcC*i}t|dgdurtt|dg||SNr)r)rr6rrr!!_ListBatchJobsParameters_to_mldevrcCrr)r)rr6rrr!"_ListBatchJobsParameters_to_vertexrrcit|dgdurtdgt|dgt|dgdur(tdgt|dgt|dgdurBtdgfddt|dgDS) Nrr nextPageTokennext_page_token operations batch_jobscrr)r]rrrr!r"r#z5_ListBatchJobsResponse_from_mldev..r(r+rrr!!_ListBatchJobsResponse_from_mldev    rcr) NrrrrbatchPredictionJobsrcrr)rfrrrr!r"r#z6_ListBatchJobsResponse_from_vertex..r(r+rrr!"_ListBatchJobsResponse_from_vertexrr cCsi}t|dgdurt|dgt|dgt|dgdur(t|dgt|dgt|dgdur;t|dgt|dgt|dgdurQt|dgtt|dg|t|d gdurdt|d gt|d gt|d gdurzt|d gtt|d g|t|d gdurt|d gt|d gt|dgdurt|dgt|dgt|dgdurt|dgt|dgt|dgdurt|dgt|dg|S)N function_call functionCallcode_execution_resultcodeExecutionResultexecutable_codeexecutableCode file_datafileDatafunction_responsefunctionResponse inline_data inlineDatatextthoughtthought_signaturethoughtSignaturevideo_metadata videoMetadata)r)r*rrkr6rrr!rsT   rcCsli}t|dgdurt|dgt|dgt|dgdur!tdt|dgdur4t|dgt|dg|S)Ncategorymethodz0method parameter is not supported in Gemini API. thresholdr9r6rrr!rrlrcCs~i}t|dgdurt|dgddt|dgDt|dgdur&tdt|dgdur9t|dgt|dgt|d gdurLt|d gt|d gt|d gdur_t|d gt|d gt|d gdurrt|dgt|d gt|dgdur~tdt|dgdurt|dgtt|dg|t|dgdurt|dgtt|dg|t|dgdurt|dgt|dg|S)Nfunction_declarationsfunctionDeclarationscSr&rrrrrr!r"/rz"_Tool_to_mldev.. retrievalz3retrieval parameter is not supported in Gemini API.google_search_retrievalgoogleSearchRetrieval computer_use computerUse file_search fileSearchcode_execution codeExecutionenterprise_web_searchz?enterprise_web_search parameter is not supported in Gemini API. google_maps googleMaps google_search googleSearch url_context urlContext)r)r*r:rrr6rrr!r&sP rc @xeZdZddddeedeejejfdeej dej fddZ ddddeedej deej dej fd d Zdd d edeejdej fd dZdd d edeejddfddZdd deejdejfddZdd d edeejdejfddZdd dedejdeej dej fddZdd dedej deej dej fddZdd deejdeej fddZdS)BatchesNrWrrWrbrrc Cs<tj|||d}|jjr"t|j|}|d}|rd|}nd}nt|j|}|d}|r5d|}nd}|d}|rG|dt|}| ddd} |j dur^|j j dur^|j j } t |}t |}|jd||| } | jsvint| j} |jjrt| } |jjst| } tjj| |d } |j| | S NrWrbrrorz{model}:batchGenerateContentr ?rpostr kwargs)r _CreateBatchJobParameters _api_clientvertexairget format_maprrpopr http_optionsrconvert_to_dictencode_unserializable_typesrbodyjsonloadsrfr]BatchJob_from_response model_dump_verify_response selfrWrbrparameter_model request_dictrequest_url_dictpath query_paramsrWr  response_dict return_valuerrr!_create`sV           zBatches._createc Cstj|||d}|jjrtdt|j|}|d}|r#d|}nd}|d}|r5|dt|}| ddd} |j durL|j j durL|j j } t |}t |}|jd||| } | jsdint| j} |jjsrt| } tjj| |d } |j| | S NrLz=This method is only supported in the Gemini Developer client.roz{model}:asyncBatchEmbedContentr rMrrNrO)r #_CreateEmbeddingsBatchJobParametersrRrSr:rrTrUrrVrrWrrXrYrrZr[r\r]r]r^r_r`rarrr!_create_embeddingssJ         zBatches._create_embeddingsrrLc Cs:tj||d}|jjr!t|j|}|d}|rd|}nd}nt|j|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j j }t |}t |}|jd |||} | jsuint| j} |jjrt| } |jjst| } tjj| |d } |j| | S) a6Gets 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 Vertex AI client. Or "batches/abc" using the Gemini Developer AI 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}") rLrrobatchPredictionJobs/{name}batches/{name}r rMrNrTrO)r _GetBatchJobParametersrRrSrrTrUrrrVrrWrrXrYrrZr[r\rfr]r]r^r_r` rbrLrrcrdrerfrgrWr rhrirrr!rTsP           z Batches.getc Cstj||d}|jjr!t|j|}|d}|rd|}nd}nt|j|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j 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/456" or "456" when project and location are initialized in the Vertex AI client. Or "batches/abc" using the Gemini Developer AI client. Usage: .. code-block:: python client.batches.cancel(name='123456789') roro!batchPredictionJobs/{name}:cancelbatches/{name}:cancelr rMrNrN)r _CancelBatchJobParametersrRrSrurTrUrsrrVrrWrrXrYr rbrLrrcrdrerfrgrWr rrr!cancel0s@          zBatches.cancelc Cs@tj|d}|jjrt|}|d}|rd|}nd}nt|}|d}|r/d|}nd}|d}|rA|dt|}| ddd}|j durX|j j durX|j j }t |}t |}|jd|||}|jspint|j} |jjr~t| } |jjst| } tjj| |d } tj|jd | _|j| | S Nrnrorbatchesr rMrrTrOheaders)r _ListBatchJobsParametersrRrSrrTrUrrrVrrWrrXrYrrZr[r\r rListBatchJobsResponser^r_ HttpResponser|rr` rbrrcrdrerfrgrWr rhrirrr!_listtsL           z Batches._listc CsJtj||d}|jjr!t|j|}|d}|rd|}nd}nt|j|}|d}|r4d|}nd}|d}|rF|dt|}| ddd}|j dur]|j j dur]|j j }t |}t |}|jd |||} | jsuint| j} |jjrt| } |jjst| } tjj| |d } tj| jd | _|j| | 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') rororprqr rMrNdeleterOr{)r _DeleteBatchJobParametersrRrSrrTrUrrrVrrWrrXrYrrZr[r\rrDeleteResourceJobr^r_rr|rr`rsrrr!rsZ           zBatches.deletecCsTt|j|}tj|||d}|jjr"t||j}|j |||dS|j |||dS)aCreates a batch job. Args: model (str): The model to use for the batch job. src: The source of the batch job. Currently Vertex AI supports GCS URI(-s) or BigQuery URI. Example: "gs://path/to/input/data" or "bq://projectId.bqDatasetId.bqTableId". Gemini Developer API supports List of inlined_request, or file name. Example: "files/file_name". 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-2.0-flash-001", src="gs://path/to/input/data", ) print(batch_job.state) rL rZrrRr rQrSr format_destinationrrjrbrWrbrrcrrr!createszBatches.createcCsXddl}|jdtjddt|j|}tj|||d}|jj r$t d|j |||dS)a]**Experimental** Creates an embedding batch job. Args: model (str): The model to use for the batch job. src: Gemini Developer API supports List of inlined_request, or file name. Example: "files/file_name". 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_embeddings( model="text-embedding-004", src="files/my_embedding_input", ) print(batch_job.state) rNJbatches.create_embeddings() is experimental and may change without notice.r3 stacklevelrL5Vertex AI does not support batches.create_embeddings.) warningswarnrExperimentalWarningrZt_embedding_batch_job_sourcerRr rlrSr:rm)rbrWrbrrrcrrr!create_embeddings0szBatches.create_embeddingscCs(|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}") Nrrn)r ListBatchJobsConfigrrrbrrrr!listbs z Batches.list)__name__ __module__ __qualname__rstrrr BatchJobSourceUnionBatchJobSourceUnionDictCreateBatchJobConfigOrDictr]rjEmbeddingsBatchJobSourceOrDict$CreateEmbeddingsBatchJobConfigOrDictrmGetBatchJobConfigOrDictrTCancelBatchJobConfigOrDictrxListBatchJobsConfigOrDictr~rDeleteBatchJobConfigOrDictrrrrrrrrrr!rI^s G = T E > Z 2 3rIc @rH) AsyncBatchesNrJrWrbrrc sDtj|||d}|jjr#t|j|}|d}|r d|}nd}nt|j|}|d}|r6d|}nd}|d}|rH|dt|}| ddd} |j dur_|j j dur_|j j } t |}t |}|jd||| IdH} | jszint| j} |jjrt| } |jjst| } tjj| |d } |j| | SrK)r rQrRrSrrTrUrrrVrrWrrXrY async_requestrZr[r\rfr]r]r^r_r`rarrr!rjsX            zAsyncBatches._createc stj|||d}|jjrtdt|j|}|d}|r$d|}nd}|d}|r6|dt|}| ddd} |j durM|j j durM|j j } t |}t |}|jd||| IdH} | jshint| j} |jjsvt| } tjj| |d } |j| | Srk)r rlrRrSr:rrTrUrrVrrWrrXrYrrZr[r\r]r]r^r_r`rarrr!rmsL          zAsyncBatches._create_embeddingsrnrLc sBtj||d}|jjr"t|j|}|d}|rd|}nd}nt|j|}|d}|r5d|}nd}|d}|rG|dt|}| ddd}|j dur^|j j dur^|j j }t |}t |}|jd |||IdH} | jsyint| j} |jjrt| } |jjst| } tjj| |d } |j| | S) a@Gets 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 Vertex AI client. Or "batches/abc" using the Gemini Developer AI 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}") rororprqr rMrNrTrO)r rrrRrSrrTrUrrrVrrWrrXrYrrZr[r\rfr]r]r^r_r`rsrrr!rTsV            zAsyncBatches.getc stj||d}|jjr"t|j|}|d}|rd|}nd}nt|j|}|d}|r5d|}nd}|d}|rG|dt|}| ddd}|j dur^|j j dur^|j 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/456" or "456" when project and location are initialized in the Vertex AI client. Or "batches/abc" using the Gemini Developer AI client. Usage: .. code-block:: python await client.aio.batches.cancel(name='123456789') rorortrur rMrNrN)r rvrRrSrurTrUrsrrVrrWrrXrYrrwrrr!rxTsB          zAsyncBatches.cancelc sHtj|d}|jjrt|}|d}|rd|}nd}nt|}|d}|r0d|}nd}|d}|rB|dt|}| ddd}|j durY|j j durY|j j }t |}t |}|jd|||IdH}|jstint|j} |jjrt| } |jjst| } tjj| |d } tj|jd | _|j| | Sry)r r}rRrSrrTrUrrrVrrWrrXrYrrZr[r\r rr~r^r_rr|rr`rrrr!rsR            zAsyncBatches._listc sRtj||d}|jjr"t|j|}|d}|rd|}nd}nt|j|}|d}|r5d|}nd}|d}|rG|dt|}| ddd}|j dur^|j j dur^|j j }t |}t |}|jd |||IdH} | jsyint| j} |jjrt| } |jjst| } tjj| |d } tj| jd | _|j| | 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 await client.aio.batches.delete(name='123456789') rororprqr rMrNrrOr{)r rrRrSrrTrUrrrVrrWrrXrYrrZr[r\rrrr^r_rr|rr`rsrrr!rs\            zAsyncBatches.deletecsbt|j|}tj|||d}|jjr&t||j}|j |||dIdHS|j |||dIdHS)aCreates a batch job asynchronously. Args: model (str): The model to use for the batch job. src: The source of the batch job. Currently Vertex AI supports GCS URI(-s) or BigQuery URI. Example: "gs://path/to/input/data" or "bq://projectId.bqDatasetId.bqTableId". Gemini Develop API supports List of inlined_request, or file name. Example: "files/file_name". 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-2.0-flash-001", src="gs://path/to/input/data", ) rLNrrrrr!r) szAsyncBatches.createcsddl}|jdtjddt|j|}tj|||d}d}|j dur.|j j dur.|j j }|jj r6t d|j |||dIdHS)ao**Experimental** Creates an asynchronously embedding batch job. Args: model (str): The model to use for the batch job. src: Gemini Developer API supports inlined_requests, or file name. Example: "files/file_name". 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_embeddings( model="text-embedding-004", src="files/my_embedding_input", ) print(batch_job.state) rNrrrrLr)rrrrrZrrRr rlrrWrSr:rm)rbrWrbrrrcrWrrr!rU s(  zAsyncBatches.create_embeddingscs0|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}") Nrrn)r rrrrrrr!r szAsyncBatches.list)rrrrrrr rrrr]rjrrrmrrTrrxrr~rrrrrrrrrrrr!rs G = V E @ Z 1 :r)N)Or[loggingtypingrrr urllib.parserrrr r rZr rRr r r)rrrr*pagersrr getLoggerloggerdictrobjectr,r7r;rArJrKr]rfrkrsrurrrrrrrrrrrrrrrrrrrrrrrrFrrrrrrr rrr BaseModulerIrrrrr!s4             *    )    !    ;  =        :          !      !            #      !  &          *                  <    8&