uPj$dZddlZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z ddlmZddlmZGdde ZGd d e ZGd d e ZGd de ZGddZGddZy)z Config repository for database operations on LiteLLM_Config. This repository handles config reconciliation between database values and YAML configmap values. DB values override configmap values except for None values and empty lists. N)MappingSequence)AnyFinalLiteralProtocolcast)verbose_proxy_logger)decrypt_value_helperc8eZdZedefdZedefdZy) _ConfigRowreturncyNselfs k/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/repositories/config_repository.py param_namez_ConfigRow.param_names!$cyrrrs r param_valuez_ConfigRow.param_values%(rN)__name__ __module__ __qualname__propertystrrobjectrrrrr r s( $C$$ (V((rr ceZdZdeeefdedzfdZdeefdZdeeefdeeeeeffdefdZ deeefdedzfdZ y) _ConfigTablewhererNc Kywrrrr!s r find_uniquez_ConfigTable.find_uniquec Kywrrrs r find_manyz_ConfigTable.find_manyr%r&datac Kywrr)rr!r)s rupsertz_ConfigTable.upsert!r%r&c Kywrrr#s rdeletez_ConfigTable.delete#r%r&) rrrrrr r$rr(r+r-rrrr r swV'#s(*;V T@QV:*!5:qWS#X%6qgc7SVX[S[K\F\>]qblqQWS#X%6Q:;LQrr c"eZdZedefdZy) _ConfigDbrcyrrrs rlitellm_configz_ConfigDb.litellm_config's.1rN)rrrrr r1rrrr/r/&s 1 11rr/c"eZdZedefdZy) _PrismaHandlercyrrrs rdbz_PrismaHandle.db,s"rN)rrrrr/r5rrrr3r3+s "I""rr3c eZdZdZdedefdZy) ConfigParamz,Simple wrapper for config parameter from DB.rrc ||_||_yrrr)rrrs r__init__zConfigParam.__init__3s$&rN)rrr__doc__rrr:rrrr7r70s6'3'S'rr7c leZdZdZgdZdefdZedefdZ ede fdZ ede fdZ d e ded zfd Zd e d edefd Zd e defdZdee effdZdededd fdZ ddee efdedee e ffdZdee e fdee e ffdZded eddedefdZ d deded zdefdZdee dd fdZy )!ConfigRepositoryzFRepository for config database operations with reconciliation support.general_settingsrouter_settingslitellm_settingsenvironment_variables prisma_clientc||_yr)_prisma_client)rrCs rr:zConfigRepository.__init__Bs +rrcH|j td|jS)NzFNo DB Connected. See - https://docs.litellm.ai/docs/proxy/virtual_keys)rE RuntimeErrorrs rrCzConfigRepository.prisma_clientEs&    &gh h"""rcB|jjjSr)rCr5r1rs r _config_tablezConfigRepository._config_tableKs!!$$333rc|jSr)rIrs rtablezConfigRepository.tableOs!!!rrNcK|jjd|id{}|y|j}t|trt j |}t||S7Ew)z)Get a config parameter from the database.rr!Nr9)rIr$r isinstancerjsonloadsr7)rrrecordrs r get_paramzConfigRepository.get_paramSsf"00<(( k3 '**[1KjkJJ _s"A,A*AA,rcKt|tstj|n|}|jj d|i||dd|idd{t ||S7w)z'Set a config parameter in the database.rr9r)createupdate)r!r)N)rNrrOdumpsrIr+r7)rrr value_jsons r set_paramzConfigRepository.set_param]su;EkSV;WDJJ{3]h   '',)3JO(*5(   jkJJ sAA(A&A(c|K |jjd|id{y7#t$rYywxYww)z,Delete a config parameter from the database.rrMNTF)rIr- Exception)rrs r delete_paramzConfigRepository.delete_paramisG $$++<2L+M M M N  s(<!-+-<- 9<9<cK|jjd{}i}|D]B}|j}t|trt j |}|||j<D|S7Ow)z,Get all config parameters from the database.N)rIr(rrNrrOrPr)rrecordsresultrQrs rget_all_paramszConfigRepository.get_all_paramsqsp#11;;==+- 4F ,,K+s+"jj5 (3F6$$ %  4  >sA2A0AA2dstsrccJ||fg}|r|j\}}|jD]r\}}| t|trt |dk(r(t|t r6t|j |t r|j|||fn|||<t|ryy)zDeep-merge src into dst, skipping None values and empty lists from src. On conflicts, src (DB) wins, but empty lists are treated as "no value" and don't overwrite the destination. Nr)popitemsrNlistlendictgetappend)rr`rastackdskvs r_deep_merge_dictsz"ConfigRepository._deep_merge_dicts|s c |99;DAq  19a&3q6Q;a&:aeeAh+ELL!A$+AaD renv_varsreturn_original_valueci}|jD];\}}t|trt||d|}|(|||<.t|||<=|S)z,Decrypt environment variables from database.debug)valuekeyexception_typerq)rdrNrr )rrprq decryptedrurtdecrypted_values r_decrypt_env_variablesz'ConfigRepository._decrypt_env_variablessi,. "..* ,JC%%"6#**? # #.%4IcN!$U # ,rcni}|jD]\}}|||<|j}|||<!|S)zLNormalize env variable keys to include both original and uppercase versions.)rdupper)rrp normalizedrurt upper_keys r_normalize_env_variable_keysz-ConfigRepository._normalize_env_variable_keyssG,. "..* *JC#JsO I$)Jy ! *rcurrent_configdb_param_valuec|dk(rr|j|d}|j|}|jD]\}}|tj|<|j dij ||S||vr|||<|St||tr't|tr|j||||S|||<|S)zAUpdate config fields with DB values, handling the merge strategy.rBT)rq) ryr~rdosenviron setdefaultrUrNrgro)rrrrdecrypted_env_varsmerged_env_varsenv_keyrts r_update_config_fieldsz&ConfigRepository._update_config_fieldss 0 0(,(C(CNjn(C(o %)%F%FGY%ZO"1"7"7"9 ,&+ 7# ,  % %&=r B I I/ Z! ! ^ +)7N: &! ! nZ0$ 7J~W[<\  " ">*#=~ N*8N: &r yaml_configstore_model_in_dbcK|durtjd|S|jDcgc]}|j|}}t j |d{}t j|}|D]_}||j}|j} tjd|| |8| ;|j|ttd|| }a|Scc}w7w)aReconcile config from YAML with database overrides. This is the main config reconciliation method that loads config params from the database and merges them with the YAML config. DB values override YAML values except for None values and empty lists. Args: yaml_config: The configuration loaded from YAML file store_model_in_db: Whether to load config from DB Returns: The merged configuration with DB overrides applied TzB'store_model_in_db' is not True, skipping db config reconciliationNzparam_name=%s, param_value=%sr>)rrr)r info CONFIG_PARAMSrRasynciogathercopydeepcopyrrrsrr r) rrrrmtasks responsesconfigresponserrs rreconcile_configz!ConfigRepository.reconcile_configs$ D ( % %&j k 373E3EFaq)FF!(!77 {+! H!,,J"..K & &'F T_ `%+*A33#)#5 # $/4  . 7G7s)*C"CC"C AC"/C"2/C" param_namescKtj|Dcgc]}|j|c}d{ycc}w7 w)zPrefetch config params to warm the cache. This can be called before reconcile_config to ensure all needed params are loaded in a single batch. N)rrrR)rrrms rprefetch_paramsz ConfigRepository.prefetch_paramss4 nn+FQt~~a0FGGGFGsA;AA A)Tr)rrrr;rrr:rr3rCr rIrKrr7rRrXboolr[rgrr_rorryr~rrrrerrrrr=r=8sPM,c,#}## 4|44"|""K#K+2DK K# KC KK KST d3;&7 T(MQV ,EI c3h&T#s(^SRUX %  F*.11 $;1  1fHcHtHrr=)r;rrrOrcollections.abcrrtypingrrrrr litellm._loggingr 0litellm.proxy.common_utils.encrypt_decrypt_utilsr r r r/r3r7r=rrrrsq -661Q))R8R22 #H# ''LHLHr