TCjn ddlZddlZddlmZmZmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lmZdd lmZdd l mZdd l mZdd l mZddlmZmZejdZ dJdeeeefefdeeeefdeeeffdZ dJdeeeefefdeeeefdeeeffdZ dJdeeeefefdeeeefdeeeffdZ! dJdeeeefefdeeeefdeeeffdZ" dJdeeeefefdeeeefdeeeffdZ# dJdedeeeefefdeeeefdeeeffdZ$ dJdeeeefefdeeeefdeeeffdZ% dJdeeeefefdeeeefdeeeffdZ& dJdeeeefefdeeeefdeeeffdZ' dJdedeeeefefdeeeefdeeeffdZ( dJdedeeeefefdeeeefdeeeffdZ) dJdeeeefefdeeeefdeeeffdZ* dJdeeeefefdeeeefdeeeffd Z+ dJdeeeefefdeeeefdeeeffd!Z, dJdeeeefefdeeeefdeeeffd"Z- dJdeeeefefdeeeefdeeeffd#Z. dJdedeeeefefdeeeefdeeeffd$Z/ dJdedeeeefefdeeeefdeeeffd%Z0 dJdeeeefefdeeeefdeeeffd&Z1 dJdedeeeefefdeeeefdeeeffd'Z2 dJdedeeeefefdeeeefdeeeffd(Z3 dJdedeeeefefdeeeefdeeeffd)Z4 dJdeeeefefdeeeefdeeeffd*Z5 dJdeeeefefdeeeefdeeeffd+Z6 dJdedeeeefefdeeeefdeeeffd,Z7 dJdeeeefefdeeeefdeeeffd-Z8 dJdedeeeefefdeeeefdeeeffd.Z9 dJdeeeefefdeeeefdeeeffd/Z: dJdeeeefefdeeeefdeeeffd0Z; dJdedeeeefefdeeeefdeeeffd1Z< dJdeeeefefdeeeefdeeeffd2Z= dJdedeeeefefdeeeefdeeeffd3Z> dJdedeeeefefdeeeefdeeeffd4Z? dJdeeeefefdeeeefdeeeffd5Z@ dJdeeeefefdeeeefdeeeffd6ZA dJdeeeefefdeeeefdeeeffd7ZB dJdedeeeefefdeeeefdeeeffd8ZC dJdeeeefefdeeeefdeeeffd9ZD dJdeeeefefdeeeefdeeeffd:ZE dJdeeeefefdeeeefdeeeffd;ZF dJdeeeefefdeeeefdeeeffd<ZG dJdeeeefefdeeeefdeeeffd=ZH dJdeeeefefdeeeefdeeeffd>ZI dJdeeeefefdeeeefdeeeffd?ZJ dJdeeeefefdeeeefdeeeffd@ZK dJdeeeefefdeeeefdeeeffdAZL dJdeeeefefdeeeefdeeeffdBZM dJdeeeefefdeeeefdeeeffdCZN dJdeeeefefdeeeefdeeeffdDZO dJdeeeefefdeeeefdeeeffdEZPGdFdGe jQZRGdHdIe jQZSdS)KN)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 XC:\Projects\Historical\FacebookAutoPoster\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_vertexr`sV!) +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_mldevrsGs-!) +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)Nrbrdrerferrorrgrh startTime start_timerirjrkrlrm inputConfigsrc outputConfigrocompletionStatscompletion_stats outputInfo output_info)r r!rprqrTrFrrr#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_metadatacontinuationTokencontinuation_token)r r!_CitationMetadata_from_mldevr#s r%_Candidate_from_mldevrs!) + {##/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  ++,--9  [./00 r'c i}t|dg+t|dgdt|dgD|S)NcitationSources citationscg|]}|Sr.r.r7s r%r3z0_CitationMetadata_from_mldev..&sAAA$AAAr'r8r#s r%rrs^!) +)*++7  AA${->,?@@AAA r'c it|dg-tdgfdt|dgDt|dg!tdgt|dgS)Npartsc0g|]}t|Sr.)_Part_to_mldevr0s r%r3z%_Content_to_mldev..5s3    4 + +   r'roler8r#s @r%_Content_to_mldevr,s!) +y!!-      [7)44    +x  ,VHd;99::: r'ci}t|dg"t|ddgt|dgt|dgtdt|dg"t|ddgt|dg|S)Nrebatchrdrozldest parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.webhook_config webhookConfigrr#s r%_CreateBatchJobConfig_to_mldevrAs!) +'((4 -  [>*++  +x  ,  3    +()**6 /" [+,-- r'c Ri}t|dg!t|dgt|dgt|dgAt|dgttjt|dg|t|dgt d|S)Nrerdrorzrzvwebhook_config parameter is only supported in Gemini Developer API mode, not in Gemini Enterprise Agent Platform mode.)r r!rJrpt_batch_job_destinationr"r#s r%_CreateBatchJobConfig_to_vertexr^s!) +'((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||SNrmrryrrxconfig)r r!rpt_modelr^t_batch_job_sourcerr]s r%"_CreateBatchJobParameters_to_mldevrys !) +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)Nrmryrxr)r r!rprr`rrr]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)Nrerrdr8r#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!rpr"_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_responserbdonerur8r#s r%_DeleteResourceJob_from_mldevr!) +)*++7'(${=N.8s(       r'r_selfz requests[].*zrequests[].request.*)r r!rpt_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_dimensionalityoutputDimensionality mime_typezqmime_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%rrJsz!) + }%%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%rr{s !) + }%%1[M4 k]#C#CDDD +*+,,8  # [+=*>??   r'ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dgtdt|dgtd|S)Nidargsrb partial_argsztpartial_args parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode. will_continuezuwill_continue parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.rr#s r%_FunctionCall_to_mldevrs!) +v*TFDtf55666 +x  ,VHd;99::: +x  ,VHd;99::: +'((4  <    +())5  <   r'ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dgtd|S)Nallowed_function_namesallowedFunctionNamesmodestream_function_call_argumentszstream_function_call_arguments parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.rr#s r%_FunctionCallingConfig_to_mldevrs!) +0122>   [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'g!t|d'gt|d'gt|d(g4t|d)gtj t|d(gt|d*g!td+gt|d*gt|d,g!td-gt|d,gt|d.g3td/gtj t|d.gt|d0gtd1t|d2g!td3gt|d2gt|d4g/td5gtt|d4gt|d6g!td7gt|d6gt|d8gtd9t|d:g!t|d;gt|d:gt|dg!t|d?gt|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..'s3    $D) 4 4   r'toolscVg|]%}ttj|&Sr.)_Tool_to_mldevrpt_toolr\s r%r3z3_GenerateContentConfig_to_mldev..1s?    18J55y A A   r' tool_config toolConfiglabelscached_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_configaudioTranscriptionConfigrr) r r!rrp t_contentt_schema t_json_schemar"t_tools_ToolConfig_to_mldevt_cached_content_namet_speech_config_ImageConfig_to_mldevr]s` @r%_GenerateContentConfig_to_mldevr3sY !) +,-..:  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"". Dxj$A$ABBB +()**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  +,-..:  [/011 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_mldevrBs!) +)*++7'(${=N.Es3    dI . .   r'rcrgenerationConfig) default_value)r r!rpr t_contentsr3r]s @r%r[r[4s` !) +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)Nresponsercru)r r!rBr#s r%r/r/\s!) + |$$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_mldevrnss!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".  3   r'c<i}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dg|S)Nrhrirjrkrlrmr8r#s r%_ListBatchJobsConfig_to_vertexrps!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".8,d; .K.KLLL r'cji}t|dgtt|dg||SNr)r rnr#s r%!_ListBatchJobsParameters_to_mldevrss=!) +z"".!${XJ"?"?KKK r'cji}t|dgtt|dg||Srr)r rpr#s r%"_ListBatchJobsParameters_to_vertexrus=!) +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.)rsr0s r%r3z5_ListBatchJobsResponse_from_mldev..s3    !y 1 1   r'r8r#s @r%!_ListBatchJobsResponse_from_mldevr|s!) +)*++7'(${=N.r7r'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!rUrMr#s r%rrts!) + }%%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_destinationrdrer8r#s r%rErEs!) +-{;<<H   [0+>??  + ''3^$d;&H&HIII r'ci}t|dg"t|ddgt|dgt|dg!t|dgt|dg|S)Nrr@rArerdr8r#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)BatchesNrmrrmryrrc |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| | SNrmryrrr~z{model}:batchGenerateContentri?rpostrrinclude_all_fieldsrrrrfkwargs)r _CreateBatchJobParameters _api_clientvertexairget format_maprrpopr http_optionsr convert_to_dictencode_unserializable_typesrbodyjsonloadsrrsBatchJob_from_responsegetattr_verify_response selfrmryrparameter_model request_dictrequest_url_dictpath query_paramsrrf response_dict return_values r%_createzBatches._creates5 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}:asyncBatchEmbedContentrirrrrrrrr)r #_CreateEmbeddingsBatchJobParametersrrr"rrrrrrrr rrrrrrrsrrrrrs 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'rrbc 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}") rbrrbatchPredictionJobs/{name}batches/{name}rirrNrrrrrr)r _GetBatchJobParametersrrrFrrrDrrrrr rrrrrrrrsrrrr rrbrrrrrrrrfrrs 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}:cancelrirrNr)r _CancelBatchJobParametersrrrrrrrrrrr rrr rrbrrrrrrrs r%cancelzBatches.cancels.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| | SNrrr~batchesrirrrrrrrrheaders)r _ListBatchJobsParametersrrrurrrsrrrrr rrrrrrrr|ListBatchJobsResponserr HttpResponserrr rrrrrrrrrfrrs r%_listz Batches._list*s4O   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') rrrrrirrNdeleterrrrrr)r _DeleteBatchJobParametersrrrrrrrrrrr rrrrrrrrDeleteResourceJobrrrrrrrs r%r zBatches.deletess.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 rprrr rrr format_destinationrrrrmryrrs 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 ExperimentalWarningrpt_embedding_batch_job_sourcerr rrr"r)rrmryrrrs 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) rzr)rrrr list_requests r%listz Batches.list: s7(:L  & !!   r')__name__ __module__ __qualname__rstrrr BatchJobSourceUnionBatchJobSourceUnionDictCreateBatchJobConfigOrDictrrEmbeddingsBatchJobSourceOrDict$CreateEmbeddingsBatchJobConfigOrDictrGetBatchJobConfigOrDictrCancelBatchJobConfigOrDictrListBatchJobsConfigOrDictrrDeleteBatchJobConfigOrDictr r rrrrr.r'r%rrsF ";? 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) AsyncBatchesNrrmryrrc 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_requestrrrrrsrrrrrs r%rzAsyncBatches._createY 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 rrr0rrrrsrrrrrs 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'rrbc 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}") rrrrrirrNrrrrrr)r rrrrFrrrDrrrrr rrr0rrrrrsrrrrrs 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') rrrrrirrNr)r rrrrrrrrrrrr rrr0rs r%rzAsyncBatches.cancelZ 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 rrrrurrrsrrrrr rrr0rrrrr|rrrrrrrrs r%rzAsyncBatches._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') rrrrrirrNr rrrrrr)r r rrrrrrrrrrr rrr0rrrrrr rrrrrrrs r%r zAsyncBatches.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", ) rNr rs r%rzAsyncBatches.createM s< t/ 5 5C5 O   E.sO4JKKf<rCs$ ''''''''''"""""" &&&&&&......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(^;;;;B/3   tCH~v-. DcN+  #s(^    &/3   tCH~v-. DcN+  #s(^    $/3<<tCH~v-.<DcN+< #s(^<<<<B/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(^6/3FFFtCH~v-.FDcN+F #s(^ FFFFV/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/3PPtCH~v-.PDcN+P #s(^PPPPj/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'