scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style(grayswan): apply ruff format to post-call context changes

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-29 21:56:57 +00:00
parent 5b74a0375b
commit 8e93f092af
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/guardrails/guardrail_hooks/grayswan/grayswan.py
View file

@ -278,11 +278,7 @@ class GraySwanGuardrail(CustomGuardrail):
messages: Final = (
*context,
*(_MonitorMessage(role=role, content=text) for text in texts),
*(
(_MonitorMessage(role="assistant", tool_calls=response_tool_calls),)
if response_tool_calls
else ()
),
*((_MonitorMessage(role="assistant", tool_calls=response_tool_calls),) if response_tool_calls else ()),
)
# Get dynamic params from request metadata
@ -622,9 +618,7 @@ class GraySwanGuardrail(CustomGuardrail):
request_route: Final = (
litellm_metadata.get("user_api_key_request_route") if isinstance(litellm_metadata, Mapping) else None
)
route_call_types: Final = (
get_call_types_for_route(request_route) if isinstance(request_route, str) else None
)
route_call_types: Final = get_call_types_for_route(request_route) if isinstance(request_route, str) else None
call_type: Final = (
(route_call_types[0].value if route_call_types else None)
or (logging_obj.call_type if logging_obj is not None else None)