쎒j~ ddlZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd l mZdd l mZdd l mZddlmZmZejdZ dLdeeeefefdeeeefdeeeffdZ dLdeeeefefdeeeefdeeeffdZ dLdeeeefefdeeeefdeeeffdZ! dLdeeeefefdeeeefdeeeffdZ" dLdeeeefefdeeeefdeeeffdZ# dLdedeeeefefdeeeefdeeeffdZ$ dLdeeeefefdeeeefdeeeffdZ% dLdeeeefefdeeeefdeeeffdZ& dLdeeeefefdeeeefdeeeffdZ' dLdeeeefefdeeeefdeeeffdZ( dLdedeeeefefdeeeefdeeeffdZ) dLdedeeeefefdeeeefdeeeffdZ* dLdeeeefefdeeeefdeeeffd Z+ dLdeeeefefdeeeefdeeeffd!Z, dLdeeeefefdeeeefdeeeffd"Z- dLdeeeefefdeeeefdeeeffd#Z. dLdeeeefefdeeeefdeeeffd$Z/ dLdedeeeefefdeeeefdeeeffd%Z0 dLdedeeeefefdeeeefdeeeffd&Z1 dLdeeeefefdeeeefdeeeffd'Z2 dLdedeeeefefdeeeefdeeeffd(Z3 dLdedeeeefefdeeeefdeeeffd)Z4 dLdedeeeefefdeeeefdeeeffd*Z5 dLdeeeefefdeeeefdeeeffd+Z6 dLdeeeefefdeeeefdeeeffd,Z7 dLdedeeeefefdeeeefdeeeffd-Z8 dLdeeeefefdeeeefdeeeffd.Z9 dLdedeeeefefdeeeefdeeeffd/Z: dLdeeeefefdeeeefdeeeffd0Z; dLdeeeefefdeeeefdeeeffd1Z< dLdeeeefefdeeeefdeeeffd2Z= dLdedeeeefefdeeeefdeeeffd3Z> dLdeeeefefdeeeefdeeeffd4Z? dLdedeeeefefdeeeefdeeeffd5Z@ dLdedeeeefefdeeeefdeeeffd6ZA dLdeeeefefdeeeefdeeeffd7ZB dLdeeeefefdeeeefdeeeffd8ZC dLdeeeefefdeeeefdeeeffd9ZD dLdedeeeefefdeeeefdeeeffd:ZE dLdeeeefefdeeeefdeeeffd;ZF dLdeeeefefdeeeefdeeeffd<ZG dLdeeeefefdeeeefdeeeffd=ZH dLdeeeefefdeeeefdeeeffd>ZI dLdeeeefefdeeeefdeeeffd?ZJ dLdeeeefefdeeeefdeeeffd@ZK dLdeeeefefdeeeefdeeeffdAZL dLdeeeefefdeeeefdeeeffdBZM dLdeeeefefdeeeefdeeeffdCZN dLdeeeefefdeeeefdeeeffdDZO dLdeeeefefdeeeefdeeeffdEZP dLdeeeefefdeeeefdeeeffdFZQ dLdeeeefefdeeeefdeeeffdGZRGdHdIe jSZTGdJdKe jSZUdS)MN)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_objectreturnci}t|dg!t|dgt|dgt|dgtdt|dgtdt|dgtdt|d gtd t|d gtd t|d gtd|S)Napi_keyapiKeyapi_key_configzvapi_key_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. auth_typezqauth_type parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.google_service_account_configzgoogle_service_account_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.http_basic_auth_configz~http_basic_auth_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. oauth_configztoauth_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. oidc_configzsoidc_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.getvsetv ValueErrorrr to_objects BC:\Projects\DSCons\.venv\Lib\site-packages\google\genai\batches.py_AuthConfig_to_mldevr&%sL!) + {##/XJ[9+ > >??? +()**6  <    + }%%1  <    +7899E  M    +0122>  M    +'((4  <    + ''3  <   c nit|dg!tdgt|dgt|ddg.tdgfdt|ddgDt|ddg ,tdgdt|ddgDS) N responsesFile file_nameinlinedResponsesinlined_responsesc0g|]}t|S)_InlinedResponse_from_mldev.0itemr$s r% z3_BatchJobDestination_from_mldev..as3    (i 8 8   r'inlinedEmbedContentResponsesinlined_embed_content_responsescg|]}|Sr.r.r1r2s r%r3z3_BatchJobDestination_from_mldev..ps(       r'r r!r#s @r%_BatchJobDestination_from_mldevr9Ts!!) +())5[M4 o5F#G#GHHH +*,>?@@L     02DE       ;79KLMM   *+  /1CD       r'c i}t|dg!t|dgt|dgt|ddg"t|dgt|ddgt|ddg"t|dgt|ddgt|d g/t|d gtt|d g||S) NpredictionsFormatformatgcsDestinationoutputUriPrefixgcs_uribigqueryDestination outputUri bigquery_uri"vertexMultimodalDatasetDestinationvertex_dataset)r r!/_VertexMultimodalDatasetDestination_from_vertexr#s r% _BatchJobDestination_from_vertexrF|s#!) ++,--9XJ[3F2G H HIII +(*;<==I   [+->?@@  +-{;<<H  [0+>??  +<=>>J 7 CD E Ey   r'c zi}t|dg!t|dgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dgt|d gtd t|d gtd t|d gtdt|dg/t|dgtt|dg||S)Nr<r;r?r=r>rBr@rAr*qfile_name parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.r,zyinlined_responses parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.r5zinlined_embed_content_responses parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rDrC)r r!r"-_VertexMultimodalDatasetDestination_to_vertexr#s r%_BatchJobDestination_to_vertexrJs!) +z"".()4 hZ+H+HIII + {##/ ,- [9+&&  +'((4  , [>*++  + }%%1  9    ++,--9  ?    +9:;;G  M    +()**6 -.5 /0 1 19   r'c i}t|dg!t|dgt|dgt|ddg"t|dgt|ddgt|ddg"t|dgt|ddgt|d d g "t|d gt|d d g|S) NinstancesFormatr< gcsSourceurisr?bigquerySourceinputUrirBvertexMultimodalDatasetSource datasetNamevertex_dataset_namer8r#s r%_BatchJobSource_from_vertexrTs!) +)*++7XJ[3D2E F FGGG + V,--9YKkK3H!I!IJJJ +(*566B  [+Z899 ;8-HII    [:MJKK r' api_clientc it|dgtdt|dgtdt|dgtdt|dg!tdgt|dgt|d g/td d gfd t|d gDt|d gtd S)Nr<znformat parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r?zogcs_uri parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.rBztbigquery_uri parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r*fileNameinlined_requestsrequestsc2g|]}t|Sr.)_InlinedRequest_to_mldevr1r2rUr$s r%r3z,_BatchJobSource_to_mldev..s5    %Zy A A   r'rSz{vertex_dataset_name parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.r r"r!rUrrr$s` @r%_BatchJobSource_to_mldevr_s_ !) +z"".  3    + {##/  <    +'((4  <    + }%%1ZL${[M"B"BCCC +*+,,8 Z      [+=*>??    +-.//;  B   r'c i}t|dg!t|dgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dgt|d gtd t|d gtd t|d g"t|ddgt|d g|S)Nr<rLr?rMrNrBrOrPr*rHrXzxinlined_requests parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rSrQrRrr#s r%_BatchJobSource_to_vertexrasV!) +z"".&'kH:)F)FGGG + {##/[&)4 i[+I+IJJJ +'((4 :& [>*++  + }%%1  9    +*+,,8  ?    +-.//; (-8 [0122 r'c i}t|dg!t|dgt|dgt|ddg"t|dgt|ddgt|ddg4t|dgtjt|ddgt|ddg"t|dgt|ddgt|ddg"t|d gt|ddgt|dd g"t|d gt|dd gt|dd g"t|d gt|dd gt|dd gBt|dgt tjt|dd g||S)Nnamemetadata displayName display_namestate createTime create_timeendTimeend_time updateTime update_timemodeloutputdest)r r!t t_job_stater9t_recv_batch_job_destinationr#s r%_BatchJob_from_mldevrtGs-!) +x  ,VHd;99::: + M233?  [:}566  + G,--9   d;W(=>>??  + L122>  [:|455  + I.//;ZL${Z4K"L"LMMM + L122>  [:|455  + G,--9WItK*g1FGGHHH + H-..: ' *[:x"899          r'c i}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg3t|dgtjt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|d gt|dgt|d g!t|d gt|d gt|d g!t|d gt|d gt|dg!t|dgt|dgt|dg/t|dgt t|dg|t|dgAt|dgt tjt|dg|t|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nrcrerfrgerrorrhri startTime start_timerjrkrlrmrn inputConfigsrc outputConfigrpcompletionStatscompletion_stats outputInfo output_info)r r!rqrrrTrFrsr#s r%_BatchJob_from_vertexrs!) +x  ,VHd;99::: + ''3^$d;&H&HIII +y!!-WIq}T+y-I-IJJKKK +y!!-WItK';;<<< + ~&&2]OT+ ~%F%FGGG + }%%1\ND{m$D$DEEE + {##/ZL${YK"@"@AAA + ~&&2]OT+ ~%F%FGGG +y!!-WItK';;<<< + ''3 # }o . .    +'((4 ( *4 n=M+N+N O O    +)*++7&'k.s???$???r'urlContextMetadataurl_context_metadata)r r!_CitationMetadata_from_mldevr#s r%_Candidate_from_mldevrsh!) + {##/YKkI;!?!?@@@ +*+,,8 $ 12 3 3Y   + ~&&2]OT+ ~%F%FGGG +'((4_%tK.9I'J'JKKK ++,--9  [./00  + ''3^$d;&H&HIII +y!!-WItK';;<<< +()**6&'k>???  +,-..:   [/011 r'c i}t|dg+t|dgdt|dgD|S)NcitationSources citationscg|]}|Sr.r.r7s r%r3z0_CitationMetadata_from_mldev..3sAAA$AAAr'r8r#s r%rr*s^!) +)*++7  AA${->,?@@AAA r'c it|dg-tdgfdt|dgDt|dg!tdgt|dgS)Npartsc0g|]}t|Sr.)_Part_to_mldevr0s r%r3z%_Content_to_mldev..Bs3    4 + +   r'roler8r#s @r%_Content_to_mldevr9s!) +y!!-      [7)44    +x  ,VHd;99::: r'ci}t|dg"t|ddgt|dgt|dgtdt|dg"t|ddgt|dg|S)Nrfbatchrerpzldest parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.webhook_config webhookConfigrr#s r%_CreateBatchJobConfig_to_mldevrNs!) +'((4 -  [>*++  +x  ,  3    +()**6 /" [+,-- r'c Ri}t|dg!t|dgt|dgt|dgAt|dgttjt|dg|t|dgt d|S)Nrfrerpr{rzvwebhook_config parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.)r r!rJrqt_batch_job_destinationr"r#s r%_CreateBatchJobConfig_to_vertexrks!) +'((4kN;K)L)LMMM +x  , & %d;&A&A B BI   +()**6  ?   r'ci}t|dg5t|ddgtj|t|dgt|dgDt|ddgt |tj|t|dg|t|dgt t|dg||SNrnrrzrryconfig)r r!rqt_modelr_t_batch_job_sourcerr^s r%"_CreateBatchJobParameters_to_mldevrs !) +y!!-  *d; ::;;  +w+ -   T+w-G-G H H    +z""."4 hZ#@#@)LLL r'c i}t|dg4t|dgtj|t|dgt|dgBt|dgt tj|t|dg|t|dgt t|dg||S)Nrnrzryr)r r!rqrrarrr^s r%#_CreateBatchJobParameters_to_vertexrs !) +y!!-   *d; ::;;  +w+ ! T+w-G-G H H    +z"".#Dxj$A$A9MMM r'cpi}t|dg"t|ddgt|dg|S)Nrfrrer8r#s r%(_CreateEmbeddingsBatchJobConfig_to_mldevrsQ!) +'((4 -  [>*++ r'c zi}t|dg5t|ddgtj|t|dgt|dg1t|ddgt |t|dg|t|dgt t|dg||Sr)r r!rqr"_EmbeddingsBatchJobSource_to_mldevrr^s r%,_CreateEmbeddingsBatchJobParameters_to_mldevrs !) +y!!-  *d; ::;;  +w+ - * [5'22I   +z""., [8*%%y r'c i}t|dg5t|ddgtj|t|dg|Srrr^s r%"_DeleteBatchJobParameters_to_mldevrrr'c i}t|dg5t|ddgtj|t|dg|Srrr^s r%#_DeleteBatchJobParameters_to_vertexrrr'ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dg|SNsdkHttpResponsesdk_http_responsercdonervr8r#s r%_DeleteResourceJob_from_mldevr!) +)*++7'(${=N.Es(       r'r_selfz requests[].*zrequests[].request.*)r r!rqt_contents_for_embed_EmbedContentConfig_to_mldevmovevr^s r%_EmbedContentBatch_to_mldevr;s !) + |$$0,,,  .Dzl;;       +z"".  $T+z%B%BINN  )n&<=>>> r'c<i}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dgtdt|d gtd t|d gtd t|d gtd|S)N task_typertaskTypetitleoutput_dimensionalityoutputDimensionalityrzqmime_type parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. auto_truncatezuauto_truncate parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. document_ocrztdocument_ocr parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.audio_track_extractionz~audio_track_extraction parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.rr#s r%rrWsz!) + }%%1 z" [;-((  +y!!-w/kG91M1MNNN +/011= -. [2344  + }%%1  <    +())5  <    +'((4  <    +0122>  M   r'c i}t|dg!t|dgt|dgt|dg0t|dgt|t|dg||S)Nr*rXrY)r r!rr^s r%rrs !) + }%%1[M4 k]#C#CDDD +*+,,8  # [+=*>??   r'ci}t|dgtdt|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nrfrfile_urifileUrirrr]r#s r%_FileData_to_mldevrs!) +'((4  <    + |$$0YKkJ   [3455  +x  ,VHd;99::: +89::F  M   r'c > it|dgAt|dgttjt|dgt|dg!tdgt|dgt|dg!tdgt|dgt|dg!tdgt|dgt|dg!td gt|dgt|d g!td gt|d gt|d g!td gt|d gt|dg!tdgt|dgt|dg!tdgt|dgt|dg!tdgt|dgt|dg!tdgt|dgt|dg!tdgt|dgt|dg!tdgt|dgt|dg4tdgtjt|dgt|dg3tdgtjt|dgt|dgtdt|dgtdt|d g-t|d!gfd"t|d gDt|d#gAt|d#gfd$tjt|d#gDt|d%g/t|d&gtt|d%gt|d'gtd(t|d)g4t|d*gtj t|d)gt|d+g!td,gt|d+gt|d-g!td.gt|d-gt|d/g3td0gtj t|d/gt|d1gtd2t|d3g!td4gt|d3gt|d5g/td6gtt|d5gt|d7g!td8gt|d7gt|d9gtd:t|d;g!t|dgt|d=gS)?Nsystem_instructionsystemInstruction temperaturetop_ptopPtop_ktopKcandidate_countcandidateCountmax_output_tokensmaxOutputTokensstop_sequences stopSequencesresponse_logprobsresponseLogprobslogprobspresence_penaltypresencePenaltyfrequency_penaltyfrequencyPenaltyseedresponse_mime_typeresponseMimeTyperesponse_schemaresponseSchemaresponse_json_schemaresponseJsonSchemarouting_configzvrouting_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.model_selection_configz~model_selection_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.safety_settingssafetySettingsc0g|]}t|Sr.)_SafetySetting_to_mldevr0s r%r3z3_GenerateContentConfig_to_mldev..Hs3    $D) 4 4   r'toolscVg|]%}ttj|&Sr.)_Tool_to_mldevrqt_toolr\s r%r3z3_GenerateContentConfig_to_mldev..Rs?    18J55y A A   r' tool_config toolConfiglabelsznlabels parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.cached_content cachedContentresponse_modalitiesresponseModalitiesmedia_resolutionmediaResolution speech_config speechConfigaudio_timestampzwaudio_timestamp parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.thinking_configthinkingConfig image_config imageConfigenable_enhanced_civic_answersenableEnhancedCivicAnswersmodel_armor_configzzmodel_armor_config parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. service_tier serviceTieraudio_transcription_configaudioTranscriptionConfig) r r!rrq t_contentt_schema t_json_schemar"t_tools_ToolConfig_to_mldevt_cached_content_namet_speech_config_ImageConfig_to_mldevr^s` @r%_GenerateContentConfig_to_mldevr;s !) +,-..:  K[+?*@AA B BI   + ''3]OT+ %G%GHHH +y!!-VHd; ::;;; +y!!-VHd; ::;;; +)*++7%&[;L:M(N(NOOO ++,--9%&[;N:O(P(P +()**6_%tK:J9K'L'LMMM ++,--9  [./00  + |$$0ZL${ZL"A"ABBB +*+,,8%&[;M:N(O(O ++,--9  [./00  +x  ,VHd;99::: +,-..:  [/011  +)*++7  :tK2C1DEEFF  +./00<  [+A*BCCDD  +()**6  <    +0122>  M    +)*++7     [+<*=>>    +y!!-        *d; .J.JKK    + ''3 T+ ??KK  +z"".  3    +()**6   [+;*<==   +-.//;  [0122  +*+,,8%&[;M:N(O(O +())5  ${_,=>>??  +)*++7  <    +)*++7%&[;L:M(N(NOOO +'((4 d;0@AA9MM  +7899E %& [:;<<  +,-..:  B    +'((4kN;K)L)LMMM +4566B #$ [7899 r'c it|dg!tdgt|dgt|dg-tdgfdt|dgDt|dg!tdgt|dgt|dg!tdgt|dgt|d g!td gt|d gt|d g!td gt|d gt|d g!tdgt|d gS)Nrr candidatesc0g|]}t|Sr.)rr0s r%r3z7_GenerateContentResponse_from_mldev..3    "$ 2 2   r' modelVersion model_versionpromptFeedbackprompt_feedback responseId response_id usageMetadatausage_metadata modelStatus model_statusr8r#s @r%#_GenerateContentResponse_from_mldevrJs!) +)*++7'(${=N.bs3    dI . .   r'rdrgenerationConfig) default_value)r r!rqr t_contentsr;r^s @r%r[r[Qs` !) +y!!- G *d; ::;;  + |$$0 J     T+ |%D%DEE    + |$$0ZL${ZL"A"ABBB +z"". &''  xj ) ) YKr : : :   r'c Ri}t|dg/t|dgtt|dg|t|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nresponserdrv)r r!rJr#s r%r/r/ys!) + |$$0  + zl + +Y   + |$$0ZL${ZL"A"ABBB +y!!-WItK';;<<< r'ci}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dgtd|S)N page_size_querypageSize page_token pageTokenfilterznfilter parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.rr#s r%_ListBatchJobsConfig_to_mldevrvs!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".  3   r'c<i}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dg|S)Nrprqrrrsrtrur8r#s r%_ListBatchJobsConfig_to_vertexrxs!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".8,d; .K.KLLL r'cji}t|dgtt|dg||SNr)r rvr#s r%!_ListBatchJobsParameters_to_mldevr{s=!) +z"".!${XJ"?"?KKK r'cji}t|dgtt|dg||Srz)r rxr#s r%"_ListBatchJobsParameters_to_vertexr}s=!) +z""."4 hZ#@#@)LLL r'c Pit|dg!tdgt|dgt|dg!tdgt|dgt|dg-tdgfdt|dgDS)Nrr nextPageTokennext_page_token operations batch_jobsc0g|]}t|Sr.)rtr0s r%r3z5_ListBatchJobsResponse_from_mldev..s3    !y 1 1   r'r8r#s @r%!_ListBatchJobsResponse_from_mldevrs!) +)*++7'(${=N.r?r'r8r#s @r%"_ListBatchJobsResponse_from_vertexrs!) +)*++7'(${=N?@@L +, [ABCC r'c i}t|dgtdt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg/t|dgtt|dg|t|d g/t|d gt t|d g|t|d g!t|d gt|d gt|d gtdt|dg+t|dgdt|dgDt|dg!t|dgt|dgt|dgtdt|dg!t|dgt|dgt|dg+t|dgdt|dgDt|dgtd|S)N retrievalzqretrieval parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. computer_use computerUse file_search fileSearch google_search googleSearch google_maps googleMapscode_execution codeExecutionenterprise_web_searchz}enterprise_web_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.function_declarationsfunctionDeclarationscg|]}|Sr.r.r7s r%r3z"_Tool_to_mldev..sGGG$GGGr'google_search_retrievalgoogleSearchRetrievalparallel_ai_searchzzparallel_ai_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. url_context urlContext mcp_servers mcpServerscg|]}|Sr.r.r7s r%r3z"_Tool_to_mldev..s===$===r' exa_ai_searchzuexa_ai_search parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.)r r"r!r]rUr#s r%rrs!) + }%%1  <    +'((4]OT+7G%H%HIII + ''3\ND}o$F$FGGG +())5 tK/1BCCYOO  + ''3 T+ ??KK  +()**6_%tK:J9K'L'LMMM +/011=  B    +/011=  GG${-D,EFFGGG  +1233? ! [4566  +,-..:  B    + ''3\ND}o$F$FGGG + ''3 ==${]O<<===  +())5  <   r'c i}t|ddg"t|dgt|ddgt|dg!t|dgt|dg|S)Nr@rAbigquery_destinationrerfr8r#s r%rErEs!) +-{;<<H   [0+>??  + ''3^$d;&H&HIII r'ci}t|dg"t|ddgt|dgt|dg!t|dgt|dg|S)Nrr@rArfrer8r#s r%rIrIs!) +./00<  , [1233  +'((4]OT+7G%H%HIII r'c eZdZddddeedeejejfdeej dej fdZ ddddeedej deej dej fdZdd d edeejdej fd Zdd d edeejddfd Zdd deejdejfd Zdd d edeejdejfdZdd dedejdeej dej fdZdd dedej deej dej fdZdd deejdeej fdZdS)BatchesNrnrrnrzrrc |tj|||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd||| } | jsint#j| j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } |j| | SNrnrzrrrz{model}:batchGenerateContentrq?rpostrrinclude_all_fieldsrrrrnkwargs)r _CreateBatchJobParameters _api_clientvertexairget format_maprrpopr http_optionsr convert_to_dictencode_unserializable_typesrbodyjsonloadsrrtBatchJob_from_responsegetattr_verify_response selfrnrzrparameter_model request_dictrequest_url_dictpath query_paramsrrn response_dict return_values r%_createzBatches._create s5 O  .8  Ol&))&11 %$//0@AA$7  Ol&))&11 .-889IJJ-##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM  ;+M::m   $:*=99m>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 r'c tj|||}|jjrt dt |j|}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd||| } | jsint#j| j} |jjst'| } tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini } |j| | SNrziThis method is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.rz{model}:asyncBatchEmbedContentrqrrrrrrrr)r #_CreateEmbeddingsBatchJobParametersrrr"rrrrrrrr rrrrrrrtrrrrrs r%_create_embeddingszBatches._create_embeddings^sE? O  0  4   B  Ol&))&11 0/::;KLL/##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM   $:*=99m>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 r'rrcc ztj||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||} | jsint#j| j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini} |j| | S)aWGets 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 Gemini Enterprise Agent Platform 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}") rcrrbatchPredictionJobs/{name}batches/{name}rqrrNrrrrrr)r _GetBatchJobParametersrrrNrrrLrrrrr rrrrrrrrtrrrr rrcrrrrrrrrnrrs r%rz Batches.gets.2 O   5  Ol&))&11 ,+667GHH+4  Ol&))&11  **+;<<##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''t\<PPH&mJBBHM1J1JM  ;+M::m   $:*=99m>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 r'ctj||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|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') rr!batchPredictionJobs/{name}:cancelbatches/{name}:cancelrqrrNr)r _CancelBatchJobParametersrrrrrrrrrrr rrr rrcrrrrrrrs r%cancelzBatches.cancel s.5 O  '8  Ol&))&11 32==>NOO27  Ol&))&11 '&112BCC&##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDLVT<FFFFFr'c tj|}|jjr?t |}|d}|rd|}nAd}n>t|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||}|jsint#j|j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } tj|j| _|j| | SNrrrbatchesrqrrrrrrrrheaders)r _ListBatchJobsParametersrrr}rrr{rrrrr rrrrrrrrListBatchJobsResponserr HttpResponserrr rrrrrrrrrnrrs r%_listz Batches._listLs4O   7HHl%))&11 %$//0@AA$6GGl%))&11 ##$455##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''t\<PPH&mJBBHM1J1JM  H8GGm   $G7 FFm.== ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !>L$&+%7 &&&L" %%l333 r'c tj||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||} | jsint#j| j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini} tj| j| _|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') rrrrrqrrNdeleterrrrrr)r _DeleteBatchJobParametersrrrrrrrrrrr rrrrrrrrDeleteResourceJobrrr rrrrs r%rzBatches.deletes.5 O   8  Ol&))&11 ,+667GHH+7  Ol&))&11  **+;<<##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''$ lH'mJBBHM1J1JM  D4]CCm   $C3MBBm*99 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !:L$&+%7 &&&L" %%l333 r'ctj|j|}tj|||}|jjr2t j||j}| |||S| |||S)a Creates a batch job. Args: model (str): The model to use for the batch job. src: The source of the batch job. Currently Gemini Enterprise Agent Platform 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) r rqrrr rrr format_destinationrrrrnrzrrs r%createzBatches.creates> t/ 5 5C5 O   ?.sO4JKKf \\3v\ > >> \\3v\ > >>r'cddl}|dtjdt j|j|}tj|||}|jj rtd| |||S)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.r stacklevelrlGemini Enterprise Agent Platform (previously known as Vertex AI) does not support batches.create_embeddings.) warningswarnr ExperimentalWarningrqt_embedding_batch_job_sourcerr rrr"r)rrnrzrrrs r%create_embeddingszBatches.create_embeddings' s8OOO MM ,  ()93 ? ?C? O   J  9    $ $5c& $ I IIr'c\|j}td||||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 config = {'page_size': 10} for batch_job in client.batches.list(config): print(batch_job.name) rr)r rrr list_requests r%listz Batches.list\ s7(:L  & !!   r')__name__ __module__ __qualname__rstrrr BatchJobSourceUnionBatchJobSourceUnionDictCreateBatchJobConfigOrDictrrEmbeddingsBatchJobSourceOrDict$CreateEmbeddingsBatchJobConfigOrDictrGetBatchJobConfigOrDictrCancelBatchJobConfigOrDictrListBatchJobsConfigOrDictr r DeleteBatchJobConfigOrDictrrrr!rr%r.r'r%rr sF ";? QQQc]Q *E,II J Q u78 Q  ~ QQQQl"EI JJJc]J  / J uAB J  ~ JJJJZMQ^^^^"*5+H"I^ ~^^^^H<@ @G@G@G @Gu78 @G  @G@G@G@GFDHGGG ?@G "GGGGZ<@ bbb bu78 b   bbbbR<@ ,?,?,?,?  ( ,? u78 ,?  ~ ,?,?,?,?fFJ 3J3J3J3J  / 3J uAB 3J  ~ 3J3J3J3JlDH ?@ U^r'rc eZdZddddeedeejejfdeej dej fdZ ddddeedej deej dej fdZdd d edeejdej fd Zdd d edeejddfd Zdd deejdejfd Zdd d edeejdejfdZdd dedejdeej dej fdZdd dedej deej dej fdZdd deejdeej fdZdS) AsyncBatchesNrrnrzrrc Ktj|||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd||| d{V} | jsint#j| j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } |j| | Sr)r rrrrrrrrrrrr rr async_requestrrrrrtrrrrrs r%rzAsyncBatches._create{ s5 O  .8  Ol&))&11 %$//0@AA$7  Ol&))&11 .-889IJJ-##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM  ;+M::m   $:*=99m>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 r'c Ktj|||}|jjrt dt |j|}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd||| d{V} | jsint#j| j} |jjst'| } tj| t-|ddrDdt-|j d dt-|j d dt-|j d dd ini } |j| | Sr)r rrrr"rrrrrrrr rrr6rrrrtrrrrrs r%rzAsyncBatches._create_embeddings sg? O  0  4   B  Ol&))&11 0/::;KLL/##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM   $:*=99m>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 r'rrcc Ktj||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{V} | jsint#j| j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini} |j| | S)aaGets 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 Gemini Enterprise Agent Platform 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}") rrrrrqrrNrrrrrr)r rrrrNrrrLrrrrr rrr6rrrrrtrrrrrs r%rzAsyncBatches.get s.2 O   5  Ol&))&11 ,+667GHH+4  Ol&))&11  **+;<<##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM  ;+M::m   $:*=99m>00 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !1L& %%l333 r'cKtj||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{VdS) 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') rrrrrqrrNr)r rrrrrrrrrrrr rrr6rs r%rzAsyncBatches.cancel| s.5 O  '8  Ol&))&11 32==>NOO27  Ol&))&11 '&112BCC&##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL   ( (lL  r'c Ktj|}|jjr?t |}|d}|rd|}nAd}n>t|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd|||d{V}|jsint#j|j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini } tj|j| _|j| | Sr)r r rrr}rrr{rrrrr rrr6rrrrrr rrr rrrr s r%r zAsyncBatches._list s4O   7HHl%))&11 %$//0@AA$6GGl%))&11 ##$455##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM  H8GGm   $G7 FFm.== ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !>L$&+%7 &&&L" %%l333 r'c Ktj||}|jjrEt |j|}|d}|rd|}nGd}nDt|j|}|d}|rd|}nd}|d}|r|dt|}| ddd}|j |j j |j j }tj |}tj|}|jd |||d{V} | jsint#j| j} |jjrt'| } |jjst)| } tj| t/|ddrDdt/|j d dt/|j d dt/|j d dd ini} tj| j| _|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') rrrrrqrrNrrrrrrr)r rrrrrrrrrrrr rrr6rrrrrrrrr rrrrs r%rzAsyncBatches.delete s.5 O   8  Ol&))&11 ,+667GHH+7  Ol&))&11  **+;<<##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33$ lH'mJBBHM1J1JM  D4]CCm   $C3MBBm*99 ?Hd 3 3 #*#*,=t$$)0#*,BD))'.#*,@$''     !:L$&+%7 &&&L" %%l333 r'c,Ktj|j|}tj|||}|jjr8t j||j}| |||d{VS| |||d{VS)aCreates a batch job asynchronously. Args: model (str): The model to use for the batch job. src: The source of the batch job. Currently Gemini Enterprise Agent Platform 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", ) rNrrs r%rzAsyncBatches.createo s< t/ 5 5C5 O   E.sO4JKKf<rIs$ ''''''''''"""""" &&&&&&......000000......%%%%%%%%  1 2 2 /3,,tCH~v-.,DcN+, #s(^,,,,b/3%%tCH~v-.%DcN+% #s(^%%%%T/3tCH~v-.DcN+ #s(^H/311tCH~v-.1DcN+1 #s(^1111l/3tCH~v-.DcN+ #s(^D/3+++tCH~v-.+DcN++ #s(^ ++++`/3%%tCH~v-.%DcN+% #s(^%%%%T/366tCH~v-.6DcN+6 #s(^6666v/3;;tCH~v-.;DcN+; #s(^;;;;@/3tCH~v-.DcN+ #s(^./3   tCH~v-. DcN+  #s(^    &/3   tCH~v-. DcN+  #s(^    $/355tCH~v-.5DcN+5 #s(^5555t/3  tCH~v-. DcN+  #s(^    "/3tCH~v-.DcN+ #s(^./3tCH~v-.DcN+ #s(^>/3tCH~v-.DcN+ #s(^</3tCH~v-.DcN+ #s(^ B/3tCH~v-.DcN+ #s(^ >/3  tCH~v-. DcN+  #s(^    &/3tCH~v-.DcN+ #s(^ B/3   tCH~v-. DcN+  #s(^    &/3   tCH~v-. DcN+  #s(^    $/3tCH~v-.DcN+ #s(^0/3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^ </3..tCH~v-..DcN+. #s(^....h/3tCH~v-.DcN+ #s(^ ./3tCH~v-.DcN+ #s(^,/3tCH~v-.DcN+ #s(^>/3tCH~v-.DcN+ #s(^6/3BBBtCH~v-.BDcN+B #s(^ BBBBN/3##tCH~v-.#DcN+# #s(^####R/3   tCH~v-. DcN+  #s(^    &/3   tCH~v-. DcN+  #s(^    $/3tCH~v-.DcN+ #s(^4/3tCH~v-.DcN+ #s(^</3))tCH~v-.)DcN+) #s(^))))^/3%%%tCH~v-.%DcN+% #s(^ %%%%T/3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^:/3tCH~v-.DcN+ #s(^4/3tCH~v-.DcN+ #s(^/3tCH~v-.DcN+ #s(^/3tCH~v-.DcN+ #s(^8/3tCH~v-.DcN+ #s(^8/3UUtCH~v-.UDcN+U #s(^UUUUt/3tCH~v-.DcN+ #s(^,/3tCH~v-.DcN+ #s(^>/3LLtCH~v-.LDcN+L #s(^LLLLb/3tCH~v-.DcN+ #s(^(/3tCH~v-.DcN+ #s(^$m m m m m k$m m m `x x x x x ;)x x x x x r'