^h(SSKJrJrJrJrJrJr SSKrSSKJ r J r J r SSK J r SSKJr "SS\ 5rS\S \S S \4S jrS\S \S S \4S jrSSKJrJr \SrS\S\S\\S\\S\\S\SS \\\44SjrS\S\S\\S\\S \\\44 SjrS\\S \4SjrS\4SjrSrSrSrS \4S!jrS"\S \4S#jr g)$)DictListLiteralOptionalTupleUnionN)supports_response_schemasupports_system_messagesverbose_logger)BaseLLMException)PartTypec b^\rSrSrSS\S\S\\\\ R44U4Sjjjr Sr U=r $) VertexAIError status_codemessageheadersc">[TU]X!US9 g)N)rrr)super__init__)selfrrr __class__s ]/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/vertex_ai/common_utils.pyrVertexAIError.__init__ s 7S)N)__name__ __module__ __qualname____firstlineno__intstrrrrhttpxHeadersr__static_attributes__ __classcell__)rs@rrr sF 9= TTT%emm 345 TTrrmodelcustom_llm_provider) vertex_aivertex_ai_betageminireturncUnUS:XaSn[XS9nU$![a;n[R"SR [ U555 SnSnAU$SnAff=f)Nr*r)r'r(zUnable to identify if system message supported. Defaulting to 'False'. Received error message - {} Add it here - https://github.com/BerriAI/litellm/blob/main/model_prices_and_context_window.jsonF)r Exceptionr warningformatr")r'r(_custom_llm_providersupports_system_messagees rget_supports_system_messager5s~ (2 "2 2#. ":#  #" ( R Y YA  #( ""(s A0AAc,UnUS:XaSn[XS9nU$)Nr*r)r.)r )r'r(r2_supports_response_schemas rget_supports_response_schemar8)s//..* 8! %$r)rr)chat embeddingbatch_embeddingmodestreamvertex_projectvertex_locationvertex_api_version)v1v1beta1c SnSnUS:XagSnUSLaSnSUSUSUSUS US US 3 nOSUSUSUSUS US U3 nUR5(aSUSUSUSUS US U3 nUSLaUS - nOBUS :Xa<SnSUSUSUS US U3 nUR5(aSUSUSUSUS US U3 nU(aU(d[SU35eXg4$)Nr9generateContentTstreamGenerateContentzhttps://z-aiplatform.googleapis.com/z /projects/z /locations/z/publishers/google/models/:z?alt=ssez /endpoints/r:predictz'-aiplatform.googleapis.com/v1/projects/z,Unable to get vertex url/endpoint for mode: )isdigit ValueError)r<r'r=r>r?r@urlendpoints r_get_vertex_urlrL<sTC"H v~$ T>.H_--HI[H\\fgufvwBCRBSSmnsmttuv~uGHC_--HI[H\\fgufvwBCRBSSmnsmttuv~u@C ==??_--HI[H\\fgufvwBCRBSS^_d^eefgofpqC~z!  ))PQ_P``kl{k|}WX]W^^_`h_ij ==??_--HI[H\\fgufvwBCRBSS^_d^eefgofpqC hGvNOO =rgemini_api_keycSRU5nUS:Xa4SnUSLaSnSRXEU5nXe4$SRXEU5nXe4$US:XaS nSRXEU5nXe4$US :XaS nSRXEU5nWW4$) Nz models/{}r9rDTrEzEhttps://generativelanguage.googleapis.com/v1beta/{}:{}?key={}&alt=ssez=https://generativelanguage.googleapis.com/v1beta/{}:{}?key={}r: embedContentr;batchEmbedContents)r1)r<r'r=rM_gemini_model_namerKrJs r_get_gemini_urlrRcs %++E2 v~$ T>.HY``"nC( =PVV&.  =  !MTT .  = " "'MTT .  =rpartsc`SnUH%nSU;dM URS5(dM#SnM' U$)z check that user_content has 'text' parameter. - Known Vertex Error: Unable to submit request because it must have a text parameter. - 'text' param needs to be len > 0 - Relevant Issue: https://github.com/BerriAI/litellm/issues/5515 FtextT)get)rShas_text_paramparts r_check_text_in_contentrYs5N T>dhhv..!N r parameterscURS05nUR5Hup#[X15 M [X5 [U5 [ U5 [ USS9 [ USS9 U$)z This is a modified version of https://github.com/google-gemini/generative-ai-python/blob/8f77cc6ac99937cd3a81299ecf79608b91b06bbb/google/generativeai/types/content_types.py#L419 z$defstitle) field_namez$schema)popitems unpack_defsconvert_to_nullableadd_object_type strip_field)rZdefsnamevalues r_build_vertex_schemargsi >>'2 &Dzz| E $ !  #J  w/y rcNURSS5nUcgUR5Hup4URSS5nUb'XRS5Sn[Xa5 XbU'MAURSS5nUbR[ U5HAupU RSS5nUcMXRS5Sn[Xa5 XgU'MC MURSS5n U cMU RSS5nUcMXRS5Sn[Xa5 XdS'M g)N propertiesz$refzdefs/anyOfr_)rVr_splitr` enumerate) schemardrirerfref_keyrefanyofiatyper_s rr`r`s*L$/J!'') ))FD)  }}W-b12C  ""t   '4(  %e,))FD1&}}W5b9:C*"!H -  '4(  ii-G"==1"56C&!$g3*rcURSS5nUbb[U5S:wa [S5eUup#USS0:XaURU5 O%USS0:XaURU5 O [S5eSUS'UR SS5nUb$UR 5HupV[ U5 M UR S S5nUb [ U5 gg) NrkzInvalid input: Type Unions are not supported, except for `Optional` types. Please provide an `Optional` type or a non-Union type.typenullTnullablerir_)r^lenrIupdaterVr_ra)rnrqabrirerfr_s rraras JJw %E  u:?I   MM!  66" " MM! I "zL$/J%++-KD  &. JJw %E E"rcURSS5nUbGSU;aUScURSS5 SUS'UR5Hup#[U5 M URSS5nUb [U5 gg)Nrirequiredobjectrvr_)rVr^r_rb)rnrirerfr_s rrbrbsL$/J  F:$6$> JJz4 (!v%++-KD E ". JJw %E rr]cURUS5 URSS5nUb$UR5Hup4[XA5 M URSS5nUb [XQ5 gg)Nrir_)r^rVr_rc)rnr]rirerfr_s rrcrcsh JJz4 L$/J%++-KD  *. JJw %E E&rvertex_datetimecdSSKJn URUS5n[UR55$)z Converts a Vertex AI datetime string to an OpenAI datetime integer vertex_datetime: str = "2024-12-04T21:53:12.120184Z" returns: int = 1722729192 r)datetimez%Y-%m-%dT%H:%M:%S.%fZ)rstrptimer! timestamp)rrdts r+_convert_vertex_datetime_to_openai_datetimer s-"   ?,C DB r||~ r)!typingrrrrrrr#litellmr r r )litellm.llms.base_llm.chat.transformationr litellm.types.llms.vertex_air rr"boolr5r8all_gemini_url_modesrLrRrYdictrgr`rarbrcr!rrrrrsw>> VVF1T$T# #%,-T%U# #* %  %%,-T%U %  %%EF$ $ $ TN$SM $ c] $  0 $ 38_$N   TNSM   38_ D $x. T T:B#<  'C '   r