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

init custom guardrail class

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-23 10:54:42 -07:00
parent 7d30188f84
commit 1b1e0f2d77
2 changed files with 26 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -1959,7 +1959,9 @@ class ProxyConfig:
# Guardrail settings
guardrails_v2 = config.get("guardrails", None)
if guardrails_v2:
init_guardrails_v2(all_guardrails=guardrails_v2)
init_guardrails_v2(
all_guardrails=guardrails_v2, config_file_path=config_file_path
)
return router, router.get_model_list(), general_settings
def get_model_info_with_id(self, model, db_model=False) -> RouterModelInfo: