phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) post_call rules

Browse source
This commit is contained in:
ishaan-jaff 2024-01-15 20:56:25 -08:00
parent 0c25d2e157
commit fcc1e23a05
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/utils.py
View file

@ -1764,11 +1764,14 @@ class Rules:
if type(decision) == bool:
if decision is False:
raise litellm.APIResponseValidationError(message="LLM Response failed post-call-rule check", llm_provider="", model=model) # type: ignore
return True
elif type(decision) == dict:
decision, message = decision["decision"], decision["message"]
if decision is False:
raise litellm.APIResponseValidationError(message=message, llm_provider="", model=model) # type: ignore
decision_val = decision.get("decision", True)
decision_message = decision.get(
"message", "LLM Response failed post-call-rule check"
)
if decision_val is False:
raise litellm.APIResponseValidationError(message=decision_message, llm_provider="", model=model) # type: ignore
return True
####### CLIENT ###################