scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix(proxy): add None guard to guardrail branch of streaming iterator hook

Browse source 
Apply the same None guard to the apply_guardrail branch for consistency,
as flagged in PR review. All three callback dispatch branches now guard
against None returns to prevent streaming chain breakage.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
tombii 2026-03-03 22:24:16 +01:00
parent 7ea99461dd
commit ba8459fda0
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/utils.py
View file

@ -2122,13 +2122,15 @@ class ProxyLogging:
current_response = _new_response
elif "apply_guardrail" in type(_callback).__dict__:
request_data["guardrail_to_apply"] = callback
current_response = (
_new_response = (
unified_guardrail.async_post_call_streaming_iterator_hook(
user_api_key_dict=user_api_key_dict,
request_data=request_data,
response=current_response,
)
)
if _new_response is not None:
current_response = _new_response
else:
_new_response = (
_callback.async_post_call_streaming_iterator_hook(