uPj1DjUddlZddlZddlZddlZddlmZddlmZmZddl Z ddl Z ddl m Z ddl mZmZddlmZddlmZddlmZdd lmZd d d d dddddddddddZeeeefed<idddddddddd d!d d"d d#d d$d%d&d%d'd%d(d%d)d*d+d*d,d*d-d*d.d/id0d/d1d/d2d/d3d4d5d4d6d4d7d4d8d9d:d9d;d9dd?d>d@d>dAd>dBdCdDdCidEdCdFdCdGdHdIdHdJdHdKdHdLdMdNdMdOdMdPdMdQdRdSdRdTdRdUdRdVdWdXdWdYdWdWdZdZdZdZd>d4d[Zeeeeefefed\<d]ed^ed_eeefdzfd`Z d^ed_e!fdaZ"d^ed_e!fdbZ# ddceeefd^eddeedeeeefdzdfe!d_eeeff dgZ$dhe%eeefd_e%eeeffdiZ&dhe%ed_e!fdjZ'dkeeefdleeefd_eeeffdmZ(dneeefd_edzfdoZ)d^edeeeefd_eeeffdpZ*d]ed_eeefdzfdqZ+dredsed_efdtZ,dredsed_efduZ-GdvdweZ.GdxdyeZ/ddzeee0fd{ed_eee0ffd|Z1d_edzfd}Z2Gd~deZ3y)N)Sequence)AnyFinal)DEFAULT_MAX_RECURSE_DEPTH)BaseLLMModelInfoBaseTokenCounter)BaseLLMExceptionget_secret_str)AllMessageValues)TokenCountResponseg?g?g?gUUUUUU?g?g?g@gUUUUUU?g?g?g @g?gqq?g@)1:11:41:82:33:23:44:14:34:55:48:19:1616:921:9GEMINI_IMAGE_ASPECT_RATIOS)rr)r)r)r )rr)rr)rr )r )r) ))r$0)xr)P) )@ )r+r*r)r-r,)r/r.)r1r0)Xr))` ))rr!r)rr)r r)r"r)r3r2r)r5r4)r7r6)r9r8)@r))@ ))r;r:r)r=r<)r?r>)rAr@)r$r#r)r&r%)r(r')r)r$)r%r)r'`)r$ )r&)rBr%r)rCr')rDr$r))rEr&)iiP)i0i)i` i@)ii )r4)r4rF!GEMINI_IMAGE_SIZE_TO_ASPECT_RATIOsizemodelreturnct|}|y|\}}dt||i}t||}t|rt |r||d<|S||d<|S)N aspectRatio imageSize)_parse_openai_image_size&_map_dimensions_to_gemini_aspect_ratio$_map_dimensions_to_gemini_image_sizeis_gemini_image_modelsupports_gemini_image_size)rHrI dimensionswidthheight image_config image_sizes e/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/gemini/common_utils.py&map_openai_size_to_gemini_image_configrYcsr06JME6(*PQVX^*_`LeZS%STdMdc5jeOe+-M o#_"<'6s'; m$  V?%BD F #    #$U+/; m,/;M/J m,,.1=k1JM+.(,,];$c*/H !%,>!?  $d+'9 m$%++-' U Q QVYapVp!&M# ' If0## 00U  s" C6C6#C<<D,D''D,ruc>ddlm}|j}t}g}|D]tt t s|j %tfd|Dd}||j N||vrS|j||j v|S)NrVertexGeminiConfigc3,K|] }|vs| ywN.0rtools rX z._dedupe_gemini_search_tools..sJ3cTk3Js ) @litellm.llms.vertex_ai.gemini.vertex_and_google_ai_studio_geminir_search_tool_keyssetrxr~appendnextadd)rursearch_tool_keysseen_search_keys deduped_tools search_keyrs @rX_dedupe_gemini_search_toolsrs1BBD(+13M#$%   & J*:JDQ     &  ) ) Z(T"# rZcXddlm}|jtfd|DS)Nrrc3lK|]*ttxrtfdD,yw)c3&K|]}|v ywrrrs rXrz4_has_gemini_search_tool...s-VccTk-VsN)rxr~any)rrrs @rXrz*_has_gemini_search_tool..s-i[_z$%V#-VEU-V*VVis04)rrrr)rurrs @rX_has_gemini_search_toolrs+1BBD ichi iirZnon_default_paramsrcNddlm}|}t|}|jdd|j d}t |t r'|r%|j||}|j||}|j d}|j d}t |tr?t |t r t|s$|j|} |j|| g}|j|t |j dt rt|d|d<|S)Nrrrvru)rdrn) rrr~popr`rxlist _map_function_add_tools_to_optional_paramsr_map_web_search_options'_drop_search_tools_mixed_with_functionsr) rrr gemini_configresult tools_value mapped_toolsrvexisting_tools search_tools rXmap_gemini_image_tools_paramsrs./M - F JJ#T*+//8K+t$+99 ]c9d <4(-D^-T*BBCUV <.s/ ""3'3 C  s #)candidateCountcandidate_countrrrqr)rxr`r~updaterRrr)rIrngeneration_configrVrs ` rX"get_gemini_image_generation_configr s1FQWGX/Y*,L/%%m4d;OM:; %e ,d++7-(! P O".=*+ rZc|dk(ry|jjd\}}}|sy t|}t|}|dks|dkry||fS#t$rYywxYw)Nautoxr)lower partitionint ValueError)rH width_str separator height_strrTrUs rXrNrN*sv v~'+zz|'='=c'B$Iy* 9~J zVq[ &= sA AArTrUc`||ftvr t||fS||z ttfdS)NcTttjt|z Sr)absmathlogr) aspect_ratiorequested_ratios rXz8_map_dimensions_to_gemini_aspect_ratio..Es!TXX.H.VYh.h%i!jrZ)r)rGminr)rTrUrs @rXrOrO>s; v;;0%AA"V^O " j rZcXtj||z}|dkry|dkry|dkryy)Nr'512r$1Kr&2K4K)rsqrt)rTrUeffective_square_sides rXrPrPIs9#'99UV^#<s"t#t# rZc eZdZy) GeminiErrorN)__name__ __module__ __qualname__rrZrXrrTsrZrcdeZdZ ddededeededededzdedzd efd Zed efd Z e ddedzd edzfd Z e ddedzd edzfd Z e ded edzfdZ deeeefd eefdZddedzdedzd eefdZdededeej$zd efdZd edzfdZy)GeminiModelInfoNheadersrImessagesrnlitellm_paramsapi_keyapi_baserJc|S)z8Google AI Studio sends api key via x-goog-api-key headerr)selfrrIrrnrrrs rXvalidate_environmentz$GeminiModelInfo.validate_environmentYs rZcy)Nv1betarrs rX api_versionzGeminiModelInfo.api_versionfsrZc(|xstdxsdS)NGEMINI_API_BASEz)https://generativelanguage.googleapis.comr )rs rX get_api_basezGeminiModelInfo.get_api_basejsk>*;<k@kkrZc:|xstdxs tdSNGOOGLE_API_KEYGEMINI_API_KEYr )rs rX get_api_keyzGeminiModelInfo.get_api_keynsb>*:;bP`AabrZc&|jddS)Ngemini/)replacer\s rXget_base_modelzGeminiModelInfo.get_base_modelrs}}Y++rZmodelscng}|D]-}|djdd}d|z}|j|/|S)Nnamezmodels/rr)rr)rrlitellm_model_namesrIstripped_model_namelitellm_model_names rXprocess_model_namez"GeminiModelInfo.process_model_namevsQ%' ;E"'-"7"7 2"F !*-@!@   & &'9 : ;#"rZctj|}tj|}d|jd}|| t dt j j||d|i}|jdk7r)t d|jd|j|jd }|j|}|S) Nrfz/modelszGEMINI_API_BASE or GEMINI_API_KEY/GOOGLE_API_KEY is not set. Please set the environment variable, to query Gemini's `/models` endpoint.zx-goog-api-key)urlrz1Failed to fetch models from Gemini. Status code: z , Response: r) rrrrrr^module_level_clientr` status_coderzr)rrrendpointresponserrs rX get_modelszGeminiModelInfo.get_models~s"//9!--g6d../w7  wZ "5599*XJ'%w/:    3 &CHDXDXCYYefnfsfsfuevw ! 1%)%<%|||<E|Scc}wcc}wcc}w)aConverts unserializable types in dict to json.dumps() compatible types. This function is called in models.py after calling convert_to_dict(). The convert_to_dict() can convert pydantic object to dict. However, the input to convert_to_dict() is dict mixed of pydantic object and nested dict(the output of converters). So they may be bytes in the dict and they are out of `ser_json_bytes` control in model_dump(mode='json') called in `convert_to_dict`, as well as datetime deserialization in Pydantic json mode. Returns: A dictionary with json.dumps() incompatible type (e.g. bytes datetime) to compatible type (e.g. base64 encoded string, isoformat date string). asciirgc3<K|]}t|tywr)rxbytesrvs rXrz.encode_unserializable_types..s7A:a'7sc3PK|]}t|tj ywr)rxdatetimers rXrz.encode_unserializable_types..sC:a!2!23Cs$&) rrxr~rwrbase64urlsafe_b64encodedecoder isoformatencode_unserializable_typesrall)rrprocessed_datarrdrs rXr$r$sV (( /1N dD ! jjl( U eU #"(":":5"A"H"H"QN3  x00 1"'//"3N3  t $"=eUQY"ON3  t $777\a&bWXv'?'?'B'I'I''R&bs#CUCC>C&Dq{{}&Ds#Z_&`UV'B1eai'P&`s#"'N3 ( 'c&D&`s&+E8-E=Fc2tdxs tdSrr rrZrXget_api_key_from_envr(s * + O~>N/OOrZceZdZdZ ddedzdefdZ ddedeeee fdzdeeee fdzd eee fdzd ed eeee fdzd e dzde dzfd Z y)rz;Token counter implementation for Google AI Studio provider.Ncustom_llm_providerrJc@ddlm}||jjk(S)Nr) LlmProviders)litellm.types.utilsr,GEMINIrd)rr*r,s rXshould_use_token_counting_apiz8GoogleAIStudioTokenCounter.should_use_token_counting_apis 5"l&9&9&?&???rZ model_to_usercontents deployment request_modelrusystemc @Kddl}ddlm} |xsi}|j|j di} ||d} | j | | j d i| d{} | 0t| j dd||| j dd| Sy77w) Nrrr)rIr1 totalTokenstokenizer_usedr) total_tokensr3 model_usedtokenizer_typeoriginal_responser)copy(litellm.llms.gemini.count_tokens.handlerrdeepcopyr`r acount_tokensr ) rr0rr1r2r3rur4r<rcount_tokens_params_requestcount_tokens_paramsrs rX count_tokensz'GoogleAIStudioTokenCounter.count_tokenss W%2 -1]]:>>JZ\^;_-`#! &  $**+>?H8:HH )    %#ZZ q9+'%zz*:B?"(   sA"B$B%8Br)NrNN) rrr__doc__ryrar/rr~rr rBrrZrXrrsE+/@ 4Z@ @-1-1!""tCH~&-"tCH~&- " cNT) "  "DcN#d*"d " d ""rZr)NF)r)4r rrzrcollections.abcrtypingrrrr^litellm.constantsr litellm.llms.base_llm.base_utilsrr)litellm.llms.base_llm.chat.transformationr litellm.secret_managers.mainr litellm.types.llms.openair r-r rr~ryfloat__annotations__rGtuplerrYrarRrQrrrrrrrrNrOrPrrobjectr$r(rrrZrXrOs  $ 7OF762              7E$sEz"23&;H;H%;H%;H% ;H  ;H  ;H%;H%;H;H;H;H5;H;H;H%;H %!;H"#;H$%;H&%';H(%);H*+;H,-;H.%/;H0%1;H23;H45;H6%7;H8%9;H:;;H<=;H>%?;H@%A;HBC;HDE;HF%G;HH%I;HJK;HLM;HN%O;HP%Q;HRS;HTU;HVW;HX5Y;HZ[;H\];H^&_;H`&a;Hbc;Hde;Hf&g;Hhu;H!5eCHos.B)C#D;|  S T#s(^VZEZ  $c$d$""".2&+ , cN, ,sm,#s(^d* , $ ,  #s(^ ,^tDcN';T#s(^@T8j49jjS#XS> #s(^BQS>Q4ZQ( #s(^ #s(^<35c?T+A(#ssSS " I,&I,X#d3;&7##DQTV\Q\L]#LPcDjP-!1-rZ