phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #3112 from BerriAI/litellm_add_alert_types

Browse source 
[Feat] Allow user to select slack alert types to Opt In to
This commit is contained in:
Ishaan Jaff 2024-04-18 16:21:33 -07:00 • committed by GitHub
commit eb04a929e6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 105 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/proxy_server.py
View file

@ -2283,6 +2283,7 @@ class ProxyConfig:
proxy_logging_obj.update_values(
alerting=general_settings.get("alerting", None),
alerting_threshold=general_settings.get("alerting_threshold", 600),
alert_types=general_settings.get("alert_types", None),
redis_cache=redis_usage_cache,
)
### CONNECT TO DATABASE ###
@ -2619,6 +2620,9 @@ class ProxyConfig:
if "alerting" in _general_settings:
general_settings["alerting"] = _general_settings["alerting"]
proxy_logging_obj.alerting = general_settings["alerting"]
if "alert_types" in _general_settings:
general_settings["alert_types"] = _general_settings["alert_types"]
proxy_logging_obj.alert_types = general_settings["alert_types"]
# router settings
_router_settings = config_data.get("router_settings", {})
@ -8196,10 +8200,12 @@ async def update_config(config_info: ConfigYAML):
updated_general_settings = config_info.general_settings.dict(
exclude_none=True
)
config["general_settings"] = {
**updated_general_settings,
**config["general_settings"],
}
_existing_settings = config["general_settings"]
for k, v in updated_general_settings.items():
# overwrite existing settings with updated values
_existing_settings[k] = v
config["general_settings"] = _existing_settings
if config_info.environment_variables is not None:
config.setdefault("environment_variables", {})
@ -8372,7 +8378,15 @@ async def get_config():
)
_slack_env_vars[_var] = _decrypted_value
_data_to_return.append({"name": "slack", "variables": _slack_env_vars})
_alerting_types = proxy_logging_obj.alert_types
_data_to_return.append(
{
"name": "slack",
"variables": _slack_env_vars,
"alerting_types": _alerting_types,
}
)
_router_settings = llm_router.get_settings()
return {