$i> 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' dJdeeeefefdeeeefdeeeffdZ( dJdedeeeefefdeeeefdeeeffdZ) dJdedeeeefefdeeeefdeeeffdZ* dJdeeeefefdeeeefdeeeffd Z+ dJdeeeefefdeeeefdeeeffd!Z, dJdeeeefefdeeeefdeeeffd"Z- dJdeeeefefdeeeefdeeeffd#Z. dJdeeeefefdeeeefdeeeffd$Z/ dJdedeeeefefdeeeefdeeeffd%Z0 dJdedeeeefefdeeeefdeeeffd&Z1 dJdeeeefefdeeeefdeeeffd'Z2 dJdedeeeefefdeeeefdeeeffd(Z3 dJdedeeeefefdeeeefdeeeffd)Z4 dJdedeeeefefdeeeefdeeeffd*Z5 dJdeeeefefdeeeefdeeeffd+Z6 dJdeeeefefdeeeefdeeeffd,Z7 dJdedeeeefefdeeeefdeeeffd-Z8 dJdeeeefefdeeeefdeeeffd.Z9 dJdedeeeefefdeeeefdeeeffd/Z: dJdeeeefefdeeeefdeeeffd0Z; dJdeeeefefdeeeefdeeeffd1Z< dJdeeeefefdeeeefdeeeffd2Z= dJdedeeeefefdeeeefdeeeffd3Z> dJdeeeefefdeeeefdeeeffd4Z? dJdedeeeefefdeeeefdeeeffd5Z@ dJdedeeeefefdeeeefdeeeffd6ZA dJdeeeefefdeeeefdeeeffd7ZB dJdeeeefefdeeeefdeeeffd8ZC dJdeeeefefdeeeefdeeeffd9ZD dJdedeeeefefdeeeefdeeeffd: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_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 ^/var/www/html/jarvis.com/web/backend/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..Us3    (i 8 8   r'inlinedEmbedContentResponsesinlined_embed_content_responsescg|]}|Sr.r.r1r2s r%r3z3_BatchJobDestination_from_mldev..ds(       r'r r!r#s @r%_BatchJobDestination_from_mldevr9Hs!!) +())5[M4 o5F#G#GHHH +*,>?@@L     02DE       ;79KLMM   *+  /1CD       r'c >i}t|dg!t|dgt|dgt|ddg"t|dgt|ddgt|ddg"t|dgt|ddg|S) NpredictionsFormatformatgcsDestinationoutputUriPrefixgcs_uribigqueryDestination outputUri bigquery_urir8r#s r% _BatchJobDestination_from_vertexrCps!) ++,--9XJ[3F2G H HIII +(*;<==I   [+->?@@  +-{;<<H  [0+>?? r'ci}t|dg!t|dgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dgt|d gtd t|d gtd t|d gtd|S)Nr<r;r?r=r>rBr@rAr*2file_name parameter is not supported in Vertex AI.r,z:inlined_responses parameter is not supported in Vertex AI.r5zHinlined_embed_content_responses parameter is not supported in Vertex AI.rr#s r%_BatchJobDestination_to_vertexrFs?!) +z"".()4 hZ+H+HIII + {##/ ,- [9+&&  +'((4  , [>*++  + }%%1 I J JJ ++,--9 D   +9:;;G     r'c >i}t|dg!t|dgt|dgt|ddg"t|dgt|ddgt|ddg"t|dgt|ddg|S) NinstancesFormatr< gcsSourceurisr?bigquerySourceinputUrirBr8r#s r%_BatchJobSource_from_vertexrMs!) +)*++7XJ[3D2E F FGGG + V,--9YKkK3H!I!IJJJ +(*566B  [+Z899 r' api_clientc it|dgtdt|dgtdt|dgtdt|dg!tdgt|dgt|d g/td d gfd t|d gDS) 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_mldevr1r2rNr$s r%r3z,_BatchJobSource_to_mldev..s5    %Zy A A   r'r r"r!rNrrr$s` @r%_BatchJobSource_to_mldevrXs !) +z"". G H HH + {##/ H I II +'((4 M N NN + }%%1ZL${[M"B"BCCC +*+,,8 Z      [+=*>??    r'ci}t|dg!t|dgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dgt|d gtd t|d gtd |S) Nr<rHr?rIrJrBrKrLr*rErQz9inlined_requests parameter is not supported in Vertex AI.rr#s r%_BatchJobSource_to_vertexrZs!) +z"".&'kH:)F)FGGG + {##/[&)4 i[+I+IJJJ +'((4 :& [>*++  + }%%1 I J JJ +*+,,8 C   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_mldevrms-!) +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|dg|S)Nr\r^r_r`errorrarb startTime start_timercrdrerfrg inputConfigsrc outputConfigricompletionStatscompletion_stats)r r!rjrkrMrCrlr#s r%_BatchJob_from_vertexrw:s!) +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|dgtd|S)Nr_batchr^riz.dest parameter is not supported in Gemini API.rr#s r%_CreateBatchJobConfig_to_mldevrsr!) +'((4 -  [>*++  +x  , E F FF r'c i}t|dg!t|dgt|dgt|dgAt|dgttjt|dg||S)Nr_r^rirt)r r!rFrjt_batch_job_destinationr#s r%_CreateBatchJobConfig_to_vertexrs!) +'((4kN;K)L)LMMM +x  , & %d;&A&A B BI   r'ci}t|dg5t|ddgtj|t|dgt|dgDt|ddgt |tj|t|dg|t|dgt t|dg||SNrgrrsrrrconfig)r r!rjt_modelrXt_batch_job_sourcerrWs r%"_CreateBatchJobParameters_to_mldevr*s !) +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)Nrgrsrrr)r r!rjrrZrrrWs r%#_CreateBatchJobParameters_to_vertexrHs !) +y!!-   *d; ::;;  +w+ ! T+w-G-G H H    +z"".#Dxj$A$A9MMM r'cpi}t|dg"t|ddgt|dg|S)Nr_rr^r8r#s r%(_CreateEmbeddingsBatchJobConfig_to_mldevresQ!) +'((4 -  [>*++ r'c zi}t|dg5t|ddgtj|t|dgt|dg1t|ddgt |t|dg|t|dgt t|dg||Sr)r r!rjr"_EmbeddingsBatchJobSource_to_mldevrrWs r%,_CreateEmbeddingsBatchJobParameters_to_mldevrus !) +y!!-  *d; ::;;  +w+ - * [5'22I   +z""., [8*%%y r'c i}t|dg5t|ddgtj|t|dg|SrrrWs r%"_DeleteBatchJobParameters_to_mldevrrr'c i}t|dg5t|ddgtj|t|dg|SrrrWs r%#_DeleteBatchJobParameters_to_vertexrrr'ci}t|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dgt|dg!t|dgt|dg|SNsdkHttpResponsesdk_http_responser\doneror8r#s r%_DeleteResourceJob_from_mldevr!) +)*++7'(${=N.s(       r'r_selfz requests[].*zrequests[].request.*)r r!rjt_contents_for_embed_EmbedContentConfig_to_mldevmovevrWs r%_EmbedContentBatch_to_mldevrs !) + |$$0,,,  .Dzl;;       +z"".  $T+z%B%BINN  )n&<=>>> r'ci}t|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dg"t|ddgt|dgt|dgtdt|d gtd |S) N task_typertaskTypetitleoutput_dimensionalityoutputDimensionalityr{z3mime_type parameter is not supported in Gemini API. auto_truncatez7auto_truncate parameter is not supported in Gemini API.rr#s r%rrs !) + }%%1 z" [;-((  +y!!-w/kG91M1MNNN +/011= -. [2344  + }%%1 J K KK +())5 N O OO r'c i}t|dg!t|dgt|dgt|dg0t|dgt|t|dg||S)Nr*rQrR)r r!rrWs r%rrs !) + }%%1[M4 k]#C#CDDD +*+,,8  # [+=*>??   r'ci}t|dgtdt|dg!t|dgt|dgt|dg!t|dgt|dg|S)Nr_rzfile_urifileUrir{r|rVr#s r%_FileData_to_mldevr0s!) +'((4 M N NN + |$$0YKkJ   [3455  +x  ,VHd;99::: +89::F     r'c R 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:S);Nsystem_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_settingssafetySettingsc0g|]}t|Sr.)_SafetySetting_to_mldevr0s r%r3z3_GenerateContentConfig_to_mldev..s3    $D) 4 4   r'toolscVg|]%}ttj|&Sr.)_Tool_to_mldevrjt_toolrUs 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   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 gS) Nrr candidatesc0g|]}t|Sr.)rr0s r%r3z7_GenerateContentResponse_from_mldev..03    "$ 2 2   r' modelVersion model_versionpromptFeedbackprompt_feedback responseId response_id usageMetadatausage_metadatar8r#s @r%#_GenerateContentResponse_from_mldevr6"s!) +)*++7'(${=Nimage_output_options parameter is not supported in Gemini API.rr#s r%r(r(sL!) +'((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) Nrgrrc0g|]}t|Sr.)rr0s r%r3z,_InlinedRequest_to_mldev..s3    dI . .   r'r]rgenerationConfig) default_value)r r!rjr t_contentsr)rWs @r%rTrTs` !) +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)Nresponser]ro)r r!r6r#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_mldevras!) + }%%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)Nr[r\r]r^r_r`r8r#s r%_ListBatchJobsConfig_to_vertexrcs!) + }%%1*-tK+/O/O + ~&&2 ; [<.))  +z"".8,d; .K.KLLL r'cji}t|dgtt|dg||SNr)r rar#s r%!_ListBatchJobsParameters_to_mldevrf's=!) +z"".!${XJ"?"?KKK r'cji}t|dgtt|dg||Sre)r rcr#s r%"_ListBatchJobsParameters_to_vertexrh2s=!) +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.)rmr0s r%r3z5_ListBatchJobsResponse_from_mldev..Ns3    !y 1 1   r'r8r#s @r%!_ListBatchJobsResponse_from_mldevro=s!) +)*++7'(${=N.hr-r'r8r#s @r%"_ListBatchJobsResponse_from_vertexrsWs!) +)*++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.&s===$===r')r r"r!rHr@r#s r%r r s!) + }%%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 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)BatchesNrgrrgrsrrctj|||}|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| | } |j| | S Nrgrsrrrqz{model}:batchGenerateContentr\?rpostrYkwargs)r _CreateBatchJobParameters _api_clientvertexairget format_maprrpopr http_optionsr convert_to_dictencode_unserializable_typesrbodyjsonloadsrwrmBatchJob_from_response model_dump_verify_response selfrgrsrparameter_model request_dictrequest_url_dictpath query_paramsrrY response_dict return_values r%_createzBatches._create.s(5 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'A'A'C'C1L %%l333 r'cTtj|||}|jjrt dt |j|}|d}|rd|}nd}|d}|r|dt|}| ddd} |j |j j |j j } tj |}tj|}|jd||| } | jsint#j| j} |jjst'| } tj| | } |j| | S Nrz=This method is only supported in the Gemini Developer client.rz{model}:asyncBatchEmbedContentr\rrrr)r #_CreateEmbeddingsBatchJobParametersrrr"rrrrrrrr rrrrrrrmrrrrrs r%_create_embeddingszBatches._create_embeddingsrs? O  0  I  B  Ol&))&11 0/::;KLL/##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLH'mJBBHM1J1JM   $:*=99m>00'A'A'C'C1L %%l333 r'rr\ctj||}|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| | } |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}") r\rrbatchPredictionJobs/{name}batches/{name}r\rrNrr)r _GetBatchJobParametersrrr:rrr8rrrrr rrrrrrrwrmrrrr rr\rrrrrrrrYrrs 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'A'A'C'C1L %%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}:cancelr\rrNr)r _CancelBatchJobParametersrrrrrrrrrrr rrr rr\rrrrrrrrYs r%cancelzBatches.cancels.5 O  '8  Ol&))&11 32==>NOO27  Ol&))&11 '&112BCC&##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''lLHHHr'ctj|}|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| | } tj|j | _|j| | S Nrrrqbatchesr\rrrrheaders)r _ListBatchJobsParametersrrrhrrrfrrrrr rrrrrrrsroListBatchJobsResponserr HttpResponserrr rrrrrrrrrYrrs r%_listz Batches._listBs(4O   7HHl%))&11 %$//0@AA$6GGl%))&11 ##$455##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL''t\<PPH&mJBBHM1J1JM  H8GGm   $G7 FFm.=='A'A'C'C>L&+%7 &&&L" %%l333 r'c2tj||}|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| | } 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') rrrrr\rrNdeleterr)r _DeleteBatchJobParametersrrrrrrrrrrr rrrrrrrrDeleteResourceJobrrrrrrrs r%rzBatches.delete|sJ.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'A'A'C'C:L&+%7 &&&L" %%l333 r'ctj|j|}tj|||}|jjr2t j||j}| |||S| |||S)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) r rjrrr rrr format_destinationrrrrgrsrrs 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 stacklevelr5Vertex AI does not support batches.create_embeddings.) warningswarnr ExperimentalWarningrjt_embedding_batch_job_sourcerr rrr"r)rrgrsrrrs r%create_embeddingszBatches.create_embeddingss8OOO MM ,  ()93 ? ?C? O   J N O OO  $ $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) rmr)rrrr list_requests r%listz Batches.list0s7(:L  & !!   r')__name__ __module__ __qualname__rstrrr BatchJobSourceUnionBatchJobSourceUnionDictCreateBatchJobConfigOrDictrrEmbeddingsBatchJobSourceOrDict$CreateEmbeddingsBatchJobConfigOrDictrGetBatchJobConfigOrDictrCancelBatchJobConfigOrDictrListBatchJobsConfigOrDictrrDeleteBatchJobConfigOrDictrrrrrrr.r'r%rr,s+ ";? BBBc]B *E,II J B u78 B  ~ BBBBN"EI :::c]:  / : uAB :  ~ ::::zMQNNNN"*5+H"IN ~NNNNh<@ BBB Bu78 B  BBBBJDH888 ?@8 "8888|<@ SSS Su78 S   SSSSt<@ +?+?+?+?  ( +? u78 +?  ~ +?+?+?+?dFJ 0J0J0J0J  / 0J uAB 0J  ~ 0J0J0J0JfDH ?@ 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) AsyncBatchesNrrgrsrrcKtj|||}|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| | } |j| | Sr)r rrrrrrrrrrrr rr async_requestrrrrwrmrrrrrs r%rzAsyncBatches._createOsJ5 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'A'A'C'C1L %%l333 r'cdKtj|||}|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| | } |j| | Sr)r rrrr"rrrrrrrr rrrrrrrmrrrrrs r%rzAsyncBatches._create_embeddingss? O  0  I  B  Ol&))&11 0/::;KLL/##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLH'mJBBHM1J1JM   $:*=99m>00'A'A'C'C1L %%l333 r'rr\cKtj||}|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| | } |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}") rrrrr\rrNrr)r rrrr:rrr8rrrrr rrrrrrrwrmrrrrrs r%rzAsyncBatches.getsG,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'A'A'C'C1L %%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{V} 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') rrrrr\rrNr)r rrrrrrrrrrrr rrrrs r%rzAsyncBatches.cancel! s.5 O  '8  Ol&))&11 32==>NOO27  Ol&))&11 '&112BCC&##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33lLHHHr'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| | } tj|j | _|j| | Sr)r rrrrhrrrfrrrrr rrrrrrrsrorrrrrrrrs r%rzAsyncBatches._liste sQ4O   7HHl%))&11 %$//0@AA$6GGl%))&11 ##$455##H--L1 0 0y.. 0 0dXt$$$04L*  " / ;$+8l*<88L6|DDL%33 t\<H'mJBBHM1J1JM  H8GGm   $G7 FFm.=='A'A'C'C>L&+%7 &&&L" %%l333 r'cBKtj||}|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| | } 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') rrrrr\rrNrrr)r rrrrrrrrrrrr rrrrrrrrrrrrrrrrs r%rzAsyncBatches.delete sl.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'A'A'C'C: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 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", ) rNrrs r%rzAsyncBatches.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(^6/3$$tCH~v-.$DcN+$ #s(^$$$$R/3tCH~v-.DcN+ #s(^0/3tCH~v-.DcN+ #s(^ B/3tCH~v-.DcN+ #s(^>/366tCH~v-.6DcN+6 #s(^6666v/388tCH~v-.8DcN+8 #s(^8888z/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(^0/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(^ </3tCH~v-.DcN+ #s(^F/3tCH~v-.DcN+ #s(^ ./3tCH~v-.DcN+ #s(^&/3tCH~v-.DcN+ #s(^2/3tCH~v-.DcN+ #s(^6/3oootCH~v-.oDcN+o #s(^ ooooh/3  tCH~v-. DcN+  #s(^    L/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/3FFtCH~v-.FDcN+F #s(^FFFFV/3tCH~v-.DcN+ #s(^&/3tCH~v-.DcN+ #s(^>/3AAtCH~v-.ADcN+A #s(^AAAAH^^^^^k$^^^Bggggg;)gggggr'