tPj zUddlmZddlmZmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZmZddlZdd lmZdd lmZmZmZmZerdd lmZe iZd ed <eddGddZddZddZd dZ d!dZ!d"dZ"d#dZ# d$dZ$d%dZ%edf d&dZ&d'dZ'd(dZ( d)dZ)y)*) annotations)MappingSequence) dataclass)datedatetimetimezone)groupby)MappingProxyType) TYPE_CHECKINGFinalN)verbose_logger)DEFAULT_DEPRECATION_WARN_DAYSDeprecationStatusModelDeprecationInfoModelDeprecationResponse)RouterzFinal[Mapping[str, object]]_NO_MODEL_METADATAT)frozenslotsc,eZdZUded<ded<ded<y)_ResolvedDeprecationrdeprecation_date str | None litellm_modellitellm_providerN)__name__ __module__ __qualname____annotations__q/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/common_utils/model_deprecation.pyrrs  r"rct|tr|jSt|tr|St|tsy tj|j S#t $rYywxYwN) isinstancerrstr fromisoformatstrip ValueError) raw_values r#_parse_deprecation_dater, se)X&~~)T" i %!!)//"344 s"A(( A43A4c.t|tr|sytjj |}t|t syt |j d}|y|j d}t||t|tr|SdS)Nrrrrr)r&r'litellm model_costgetrr,r) model_keyentryparsedproviders r#_cost_map_lookupr6-s i %Y%%)))4E eW %+EII6H,IJF ~ii 23H %/#%> EI r"cT|j|}t|tr|StSr%)r1r&rr) deploymentkeyvalues r#_mapping_fieldr;>s%>>#&Eug.5F4FFr"ct|d}t|djd}t|jd}|H|jd}t|t |t r|ndt |t r|SdSt |t rd|vr|j dd d nd}td t|jd t|t|fDdS) zIResolve a deployment's deprecation date, preferring its explicit override model_infolitellm_paramsmodelrNrr./c3$K|]}|| ywr%r!).0resolveds r# z2_resolve_deployment_deprecation..Us   #   base_model) r;r1r,rr&r'splitnextr6)r8r= raw_modeloverrider5 unprefixeds r#_resolve_deployment_deprecationrMCs'z<@J%j2BCGGPI-jnn=O.PQH$..);<#%'1)S'A)t)3Hc)BX  IM  7AC6PUX\eUe Q/2koJ  ! !=> + ,    r"c|dkry||kryy)Nr deprecatedimminentupcomingr!) days_untilwarn_within_dayss r# _classifyrTbsA~%% r"c  |jd}t|tr|syt|}|y|j|z j }t ||j|j|t|||jS)N model_name)rVrrdays_until_deprecationstatusr) r1r&r'rMrdaysrrrTr)r8todayrSrVrDrRs r# _build_infor[js"|4J j# &j5jAH!22U:@@J ,,!22)%56!22  r"cZt|d}tdt|dDS)zGReport a model group carrying the same date on several deployments oncec2|j|jfSr%rVrr?s r#z_dedupe..su7G7GI_I_6`r"r9c38K|]\}}t|ywr%)rI)rC_groups r#rEz_dedupe..s 5U sc2|j|jfSr%r^r_s r#r`z_dedupe..sHXHXZ_ZpZpGqr")sortedtupler )modelsordereds r#_deduperj~s3F(`aG $+G9q$r r"cDttfd|DdS)Nc3BK|]}|jk(s|ywr%)rX)rCr?rXs r#rEz_bucket..s Au%,,&*@U Asc|jSr%)rr_s r#r`z_bucket..s e44r"ra)rgrf)rhrXs `r#_bucketrns#  A A4  r"ctjtj}|xs|j ||j xsdnd}t tdfd|DD}tjdt|t|tt|dt|dt|d|S) zGBucket every deployment carrying a deprecation date by how urgent it isr!c3$K|]}|| ywr%r!)rCinfos r#rEz-collect_model_deprecations..s   rFc38K|]}t|ywr%)r[)rCr8effective_todayrSs r#rEz-collect_model_deprecations..srXbZBRSrsz=model_deprecation: %d/%d deployments carry a deprecation daterOrPrQ)rOrPrQrS checked_at) rnowr utcrget_model_listrjrgrdebuglenrrn) llm_routerrSrZ snapshot_time deploymentsdedupedrss ` @r#collect_model_deprecationsr~s $<< 5M":m&8&8&:O@J@V*335;\^K  rfqr  GG G  K $7L1*-*-)  r"cf|jddjddjddS)zRNeutralize Slack control characters so a model name cannot forge a mention or link&z&z>)replace)r:s r#_escape_slack_mrkdwnrs- ==g & . .sF ; C CC PPr"c (|jdkrdt|jdnd|jd}dt|jd|jrt|jndd |j j d |d S) Nrzalready deprecated zd agozin du• `z ` (provider: unknownz , deprecates z, ))rWabsrrVrr isoformat)rqsuffixs r# _format_entryrs  & & * c$"="=>?uE4../q 1 $T__567EIEZEZ*4+@+@A`ijk++55786(! Er"c|js |jsy|jrdgd|jDnd}|jr%d|jdgd|jDnd}djd g||d S) zRender the alert for the deprecated and imminent buckets, None when both are empty Upcoming models are left out of the alert to keep it actionable. Nz *Already deprecated:*c32K|]}t|ywr%rrCis r#rEz3format_deprecation_alert_message..s%T1mA&6%Tr!z *Deprecating within z days:*c32K|]}t|ywr%rrs r#rEz3format_deprecation_alert_message..s :1mA :r u"*⚠️ Model Deprecation Warning*zl Plan migrations to a supported model. See https://docs.litellm.ai/docs/proxy/model_management for guidance.)rOrPrSjoin)snapshotdeprecated_sectionimminent_sections r# format_deprecation_alert_messagers   x'8'8ZbYlYl "U%T@S@S%TUrt   %X%>%>$?w G :(9(9 :   99 0    P   r")r+objectreturn date | None)r2rr_ResolvedDeprecation | None)r8Mapping[str, object]r9r'rr)r8rrr)rRintrSrrr)r8rrZrrSrrzModelDeprecationInfo | None)rhSequence[ModelDeprecationInfo]r tuple[ModelDeprecationInfo, ...])rhrrXrrr)rzz Router | NonerSrrZrrr)r:r'rr')rqrrr')rrrr)* __future__rcollections.abcrr dataclassesrrrr itertoolsr typesr typingr r r/litellm._loggingr%litellm.types.proxy.model_deprecationrrrrlitellm.routerrrr rr,r6r;rMrTr[rjrnr~rrrr!r"r#rs"-!--"'+%2B22F/F $d#!!$! "G $ >( *%:  BQ &r"