o Æ°i+ã @sddlZddlZddlmZmZmZmZmZerddlm Z ddl m Z ddl m Z dedefdd „Zd eeeefd ed defd d„Z dd eeeefdeedeeeeffdd„Zdedefdd„Z dd eeeefdeedeeeeffdd„ZdS)éN)Ú TYPE_CHECKINGÚDictÚListÚOptionalÚUnion)ÚOpenAIFileObject)ÚCredentialLiteLLMParams)Úverbose_loggerÚlitellm_paramsÚreturncCs*tdi|¤Ž}t t | ¡¡ ¡¡ ¡S)zX Hash of the credential params, used for mapping the file id to the right model N©)rÚhashlibÚsha256ÚjsonÚdumpsZ model_dumpÚencodeÚ hexdigest)r Zsensitive_paramsr r úX/home/app/Keep/.python/lib/python3.10/site-packages/litellm/router_utils/common_utils.pyÚ,get_litellm_params_sensitive_credential_hash s ÿþrÚhealthy_deploymentsÚ responsesrcCsfi}t|tƒrt||ƒD]\}}|j|| di¡ d¡<q |St|tƒr1| ¡D]\}}|||<q(|S)z} Create a mapping of model name to file id { "model_id": "file_id", "model_id": "file_id", } Ú model_infoÚid)Ú isinstanceÚlistÚziprÚgetÚdictÚitems)rrZmodel_file_id_mappingÚ deploymentÚresponseZmodel_idZfile_idr r rÚadd_model_file_id_mappingss ÿ ý r!Úrequest_kwargscs¦|dur|S| d¡p i}| d¡pi}| d¡p| d¡}tƒ‰t|tƒr(|S|D]}| d¡p2i}| d¡}|dur=q*||krIˆ | d¡¡q*‡fdd „|DƒS) zM If a model has a team_id Only use if request is from that team NÚmetadataÚlitellm_metadataZuser_api_key_team_idrZteam_idrcs&g|]}| di¡ d¡ˆvr|‘qS)rr©r)Ú.0r©Z ids_to_remover rÚ Hs þz,filter_team_based_models..)rÚsetrrÚadd)rr"r#r$Zrequest_team_idrZ _model_infoZ model_team_idr r'rÚfilter_team_based_models,s* ÿ  € þr+rcCs | di¡}d|vr|dSdS)a› Check if a deployment supports web search. Priority: 1. Check config-level override in model_info.supports_web_search 2. Default to True (assume supported unless explicitly disabled) Note: Ideally we'd fall back to litellm.supports_web_search() but model_prices_and_context_window.json doesn't have supports_web_search tags on all models yet. TODO: backfill and add fallback. rZsupports_web_searchTr%)rrr r rÚ_deployment_supports_web_searchNs r,cCs–|dur|St|tƒr |Sd}| d¡pg}|D]}| d¡dks(| d¡dkr,d}nq|s1|Sdd „|Dƒ}t|ƒd krIt|ƒd krIt d ¡|S) z[ If the request is websearch, filter out deployments that don't support web search NFÚtoolsÚtypeZ web_searchZweb_search_previewTcSsg|]}t|ƒr|‘qSr )r,)r&Údr r rr(|sz1filter_web_search_deployments..rz-No deployments support web search for request)rrrÚlenr Úwarning)rr"Zis_web_search_requestr-ZtoolZfinal_deploymentsr r rÚfilter_web_search_deploymentsbs" þ r2)N)r rÚtypingrrrrrZlitellm.types.llms.openairZlitellm.types.routerrZlitellm._loggingr rÚstrrr!r+Úboolr,r2r r r rÚs@    ÿÿ þþÿþ ý"þÿþý