ja 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_configz8api_key_config parameter is not supported in Gemini API. auth_typez3auth_type parameter is not supported in Gemini API.google_service_account_configzGgoogle_service_account_config parameter is not supported in Gemini API.http_basic_auth_configz@http_basic_auth_config parameter is not supported in Gemini API. oauth_configz6oauth_config parameter is not supported in Gemini API. oidc_configz5oidc_config parameter is not supported in Gemini API.getvsetv ValueErrorrr to_objects _/DATA/AppData/hermes/projects/honcho/.venv/lib/python3.11/site-packages/google/genai/batches.py_AuthConfig_to_mldevr&%s !) + {##/XJ[9+ > >??? +()**6 O P PP + }%%1 J K KK +7899E      +0122> J   +'((4 M N NN + ''3 L M MM 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..Ts3    (i 8 8   r'inlinedEmbedContentResponsesinlined_embed_content_responsescg|]}|Sr.r.r1r2s r%r3z3_BatchJobDestination_from_mldev..cs(       r'r r!r#s @r%_BatchJobDestination_from_mldevr9Gs!!) +())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_vertexrFos#!) ++,--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*Ifile_name parameter is not supported in Gemini Enterprise Agent Platform.r,zQinlined_responses parameter is not supported in Gemini Enterprise Agent Platform.r5z_inlined_embed_content_responses parameter is not supported in Gemini Enterprise Agent Platform.rDrC)r r!r"-_VertexMultimodalDatasetDestination_to_vertexr#s r%_BatchJobDestination_to_vertexrJs!) +z"".()4 hZ+H+HIII + {##/ ,- [9+&&  +'((4  , [>*++  + }%%1      ++,--9      +9:;;G  &    +()**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<z0format parameter is not supported in Gemini API.r?z1gcs_uri parameter is not supported in Gemini API.rBz6bigquery_uri parameter is not supported in Gemini API.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 not supported in Gemini API.r r"r!rUrrr$s` @r%_BatchJobSource_to_mldevr_s> !) +z"". G H HH + {##/ H I II +'((4 M N NN + }%%1ZL${[M"B"BCCC +*+,,8 Z      [+=*>??    +-.//; G   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*rHrXzPinlined_requests parameter is not supported in Gemini Enterprise Agent Platform.rSrQrRrr#s r%_BatchJobSource_to_vertexrasV!) +z"".&'kH:)F)FGGG + {##/[&)4 i[+I+IJJJ +'((4 :& [>*++  + }%%1      +*+,,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_mldevrt0s-!) +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_vertexris!) +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..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..(s3    4 + +   r'roler8r#s @r%_Content_to_mldevrs!) +y!!-      [7)44    +x  ,VHd;99::: r'ci}t|dg"t|ddgt|dgt|dgtdt|dg"t|ddgt|dg|S)Nrfbatchrerpz.dest parameter is not supported in Gemini API.webhook_config webhookConfigrr#s r%_CreateBatchJobConfig_to_mldevr4s!) +'((4 -  [>*++  +x  , E F FF +()**6 /" [+,-- r'c Ri}t|dg!t|dgt|dgt|dgAt|dgttjt|dg|t|dgt d|S)Nrfrerpr{rzNwebhook_config parameter is not supported in Gemini Enterprise Agent Platform.)r r!rJrqt_batch_job_destinationr"r#s r%_CreateBatchJobConfig_to_vertexrNs!) +'((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_mldevris !) +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.(s(       r'r_selfz requests[].*zrequests[].request.*)r r!rqt_contents_for_embed_EmbedContentConfig_to_mldevmovevr^s r%_EmbedContentBatch_to_mldevrs !) + |$$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_dimensionalityoutputDimensionalityrz3mime_type parameter is not supported in Gemini API. auto_truncatez7auto_truncate parameter is not supported in Gemini API. document_ocrz6document_ocr parameter is not supported in Gemini API.audio_track_extractionz@audio_track_extraction parameter is not supported in Gemini API.rr#s r%rr:sY!) + }%%1 z" [;-((  +y!!-w/kG91M1MNNN +/011= -. [2344  + }%%1 J K KK +())5 N O OO +'((4 M N NN +0122> J   r'c i}t|dg!t|dgt|dgt|dg0t|dgt|t|dg||S)Nr*rXrY)r r!rr^s r%rrbs !) + }%%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_mldevrws!) +'((4 M N NN + |$$0YKkJ   [3455  +x  ,VHd;99::: +89::F     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|dg!tdgt|dgt|dgt dt|dgt dt|d g-t|d!gfd"t|d gDt|d#gAt|d#gfd$tjt|d#gDt|d%g/t|d&gtt|d%gt|d'gt d(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|d1gt d2t|d3g!td4gt|d3gt|d5g/td6gtt|d5gt|d7g!td8gt|d7gt|d9gt d:t|d;g!t|d.s3    $D) 4 4   r'toolscVg|]%}ttj|&Sr.)_Tool_to_mldevrqt_toolr\s r%r3z3_GenerateContentConfig_to_mldev..s?    18J55y A A   r' 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 imageConfigenable_enhanced_civic_answersenableEnhancedCivicAnswersmodel_armor_configz J   +)*++7     [+<*=>>    +y!!-        *d; .J.JKK    + ''3 T+ ??KK  +z"". G H HH +()**6   [+;*<==   +-.//;  [0122  +*+,,8%&[;M:N(O(O +())5  ${_,=>>??  +)*++7 C   +)*++7%&[;L:M(N(NOOO +'((4 d;0@AA9MM  +7899E %& [:;<<  +,-..: F   +'((4kN;K)L)LMMM 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..z3    "$ 2 2   r' modelVersion model_versionpromptFeedbackprompt_feedback responseId response_id usageMetadatausage_metadata modelStatus model_statusr8r#s @r%#_GenerateContentResponse_from_mldevrGls!) +)*++7'(${=Nimage_output_options parameter is not supported in Gemini API.rr#s r%r7r7sL!) +'((4]OT+7G%H%HIII + ~&&2[M4 l^#D#DEEE ++,--9 E   +*+,,8 D   +*+,,8 D   +4566B N   +./00< H   r'c Dit|dg5tddgtj|t|dgt|dg@tddgfdtjt|dgDt|dg!tdgt|dgt|dgBtddgt |t|dgtdgiS) Nrnrrc0g|]}t|Sr.)rr0s r%r3z,_InlinedRequest_to_mldev..s3    dI . .   r'rdrgenerationConfig) default_value)r r!rqr t_contentsr8r^s @r%r[r[s` !) +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!rGr#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 pageTokenfilterz0filter parameter is not supported in Gemini API.rr#s r%_ListBatchJobsConfig_to_mldevrrDs!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"". G H HH r'c<i}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dg|S)Nrlrmrnrorprqr8r#s r%_ListBatchJobsConfig_to_vertexrt\s!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".8,d; .K.KLLL r'cji}t|dgtt|dg||SNr)r rrr#s r%!_ListBatchJobsParameters_to_mldevrwts=!) +z"".!${XJ"?"?KKK r'cji}t|dgtt|dg||Srv)r rtr#s r%"_ListBatchJobsParameters_to_vertexrys=!) +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|dgD|S)N retrievalz3retrieval parameter is not supported in Gemini API. computer_use computerUse file_search fileSearch google_search googleSearch google_maps googleMapscode_execution codeExecutionenterprise_web_searchz?enterprise_web_search parameter is not supported in Gemini API.function_declarationsfunctionDeclarationscg|]}|Sr.r.r7s r%r3z"_Tool_to_mldev..`sGGG$GGGr'google_search_retrievalgoogleSearchRetrievalparallel_ai_searchz.vs===$===r')r r"r!rYrQr#s r%rr8s!) + }%%1 J K KK +'((4]OT+7G%H%HIII + ''3\ND}o$F$FGGG +())5 tK/1BCCYOO  + ''3 T+ ??KK  +()**6_%tK:J9K'L'LMMM +/011= I   +/011=  GG${-D,EFFGGG  +1233? ! [4566  +,-..: F   + ''3\ND}o$F$FGGG + ''3 ==${]O<<=== r'c i}t|ddg"t|dgt|ddgt|dg!t|dgt|dg|S)Nr@rAbigquery_destinationrerfr8r#s r%rErE|s!) +-{;<<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}:batchGenerateContentrm?rpostrrinclude_all_fieldsrrrrjkwargs)r _CreateBatchJobParameters _api_clientvertexairget format_maprrpopr http_optionsr convert_to_dictencode_unserializable_typesrbodyjsonloadsrrtBatchJob_from_responsegetattr_verify_response selfrnrzrparameter_model request_dictrequest_url_dictpath query_paramsrrj 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| | SNrz=This method is only supported in the Gemini Developer client.rz{model}:asyncBatchEmbedContentrmrrrrrrrr)r #_CreateEmbeddingsBatchJobParametersrrr"rrrrrrrr rrrrrrrtrrrrrs r%_create_embeddingszBatches._create_embeddingssC? O  0  I  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}rmrrNrrrrrr)r _GetBatchJobParametersrrrKrrrIrrrrr rrrrrrrrtrrrr rrcrrrrrrrrjrrs r%rz Batches.get@s.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}:cancelrmrrNr)r _CancelBatchJobParametersrrrrrrrrrrr rrr rrcrrrrrrrs 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| | SNrrrbatchesrmrrrrrrrrheaders)r _ListBatchJobsParametersrrryrrrwrrrrr rrrrrrrrListBatchJobsResponserr HttpResponserrr rrrrrrrrrjrrs r%_listz Batches._lists4O   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') rrrrrmrrNdeleterrrrrr)r _DeleteBatchJobParametersrrrrrrrrrrr rrrrrrrrDeleteResourceJobrrrrrrrs r%rzBatches.delete+s.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_embeddingss8OOO 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) r~r)rrrr list_requests r%listz Batches.lists7(:L  & !!   r')__name__ __module__ __qualname__rstrrr BatchJobSourceUnionBatchJobSourceUnionDictCreateBatchJobConfigOrDictrrEmbeddingsBatchJobSourceOrDict$CreateEmbeddingsBatchJobConfigOrDictrGetBatchJobConfigOrDictrCancelBatchJobConfigOrDictrListBatchJobsConfigOrDictrrDeleteBatchJobConfigOrDictrrr rrrr.r'r%rrsF ";? QQQc]Q *E,II J Q u78 Q  ~ QQQQl"EI IIIc]I  / I uAB I  ~ IIIIXMQ^^^^"*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 rrr-rrrrtrrrrrs r%rzAsyncBatches._create_embeddingsd se? O  0  I  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}") rrrrrmrrNrrrrrr)r rrrrKrrrIrrrrr rrr-rrrrrtrrrrrs 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') rrrrrmrrNr)r rrrrrrrrrrrr rrr-rs 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 rrrryrrrwrrrrr rrr-rrrrrrrrrrrrrs r%rzAsyncBatches._listU 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') rrrrrmrrNrrrrrrr)r rrrrrrrrrrrr rrr-rrrrrrrrrrrrrs 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", ) rNr r s r%r zAsyncBatches.create s< t/ 5 5C5 O   E.sO4JKKf<r@s$ ''''''''''"""""" &&&&&&......000000......%%%%%%%%  1 2 2 /3tCH~v-.DcN+ #s(^H/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(^ !!!!L/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(^8/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(^%%%%V/3tCH~v-.DcN+ #s(^ ./3tCH~v-.DcN+ #s(^&/3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^6/3rrrtCH~v-.rDcN+r #s(^ rrrrn/3##tCH~v-.#DcN+# #s(^####R/3   tCH~v-. DcN+  #s(^    &/3   tCH~v-. DcN+  #s(^    $/3tCH~v-.DcN+ #s(^(/3tCH~v-.DcN+ #s(^8/3$$tCH~v-.$DcN+$ #s(^$$$$T/3%%%tCH~v-.%DcN+% #s(^ %%%%T/3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^4/3tCH~v-.DcN+ #s(^4/3tCH~v-.DcN+ #s(^/3tCH~v-.DcN+ #s(^/3tCH~v-.DcN+ #s(^8/3tCH~v-.DcN+ #s(^8/3IItCH~v-.IDcN+I #s(^IIII\/3tCH~v-.DcN+ #s(^&/3tCH~v-.DcN+ #s(^>/3AAtCH~v-.ADcN+A #s(^AAAAL/3tCH~v-.DcN+ #s(^(/3tCH~v-.DcN+ #s(^$l l l l l k$l l l ^w w w w w ;)w w w w w r'